diff options
author | Mark Hammond <mhammond@skippinet.com.au> | 2000-07-28 03:44:41 +0000 |
---|---|---|
committer | Mark Hammond <mhammond@skippinet.com.au> | 2000-07-28 03:44:41 +0000 |
commit | c8738eba8a6829e3867011a18b24a9590719e1dd (patch) | |
tree | f92c996e5b560dca7a59b039764100395c32b02a /PC/_winreg.c | |
parent | f1ac41fb4eb67c0fd0d25ff8259cf286c7a50fff (diff) | |
download | cpython-c8738eba8a6829e3867011a18b24a9590719e1dd.tar.gz |
Allow any object supporting the buffer protocol to be written as a binary object.
Diffstat (limited to 'PC/_winreg.c')
-rw-r--r-- | PC/_winreg.c | 18 |
1 files changed, 11 insertions, 7 deletions
diff --git a/PC/_winreg.c b/PC/_winreg.c index 84bf3c4876..edfadac77c 100644 --- a/PC/_winreg.c +++ b/PC/_winreg.c @@ -830,19 +830,23 @@ Py2Reg(PyObject *value, DWORD typ, BYTE **retDataBuf, DWORD *retDataSize) if (value == Py_None) *retDataSize = 0; else { - if (!PyString_Check(value)) - return 0; - *retDataSize = PyString_Size(value); + void *src_buf; + PyBufferProcs *pb = value->ob_type->tp_as_buffer; + if (pb==NULL) { + PyErr_Format(PyExc_TypeError, + "Objects of type '%s' can not " + "be used as binary registry values", + value->ob_type->tp_name); + return FALSE; + } + *retDataSize = (*pb->bf_getreadbuffer)(value, 0, &src_buf); *retDataBuf = (BYTE *)PyMem_NEW(char, *retDataSize); if (*retDataBuf==NULL){ PyErr_NoMemory(); return FALSE; } - memcpy(*retDataBuf, - PyString_AS_STRING( - (PyStringObject *)value), - *retDataSize); + memcpy(*retDataBuf, src_buf, *retDataSize); } break; } |