diff options
author | Martin Panter <vadmium+py@gmail.com> | 2015-10-07 10:01:04 +0000 |
---|---|---|
committer | Martin Panter <vadmium+py@gmail.com> | 2015-10-07 10:01:04 +0000 |
commit | 6e1d0fd29a658fa402340dbc031e4fc7db202ee7 (patch) | |
tree | 43991206a47c92f8b2ac1d4a1a938f31e5294f08 /Python/pystrhex.c | |
parent | ffee692e0ddd36e128c7dc3632b1e795a9a8a289 (diff) | |
parent | 51dc66135fd222c634512dfacba7a799372d230f (diff) | |
download | cpython-6e1d0fd29a658fa402340dbc031e4fc7db202ee7.tar.gz |
Issue #25286: Merge dictionary view glossary from 3.4 into 3.5
Diffstat (limited to 'Python/pystrhex.c')
-rw-r--r-- | Python/pystrhex.c | 61 |
1 files changed, 61 insertions, 0 deletions
diff --git a/Python/pystrhex.c b/Python/pystrhex.c new file mode 100644 index 0000000000..1259ed12df --- /dev/null +++ b/Python/pystrhex.c @@ -0,0 +1,61 @@ +/* bytes to hex implementation */ + +#include "Python.h" + +static PyObject *_Py_strhex_impl(const char* argbuf, const Py_ssize_t arglen, + int return_bytes) +{ + PyObject *retval; + Py_UCS1* retbuf; + Py_ssize_t i, j; + + assert(arglen >= 0); + if (arglen > PY_SSIZE_T_MAX / 2) + return PyErr_NoMemory(); + + if (return_bytes) { + /* If _PyBytes_FromSize() were public we could avoid malloc+copy. */ + retbuf = (Py_UCS1*) PyMem_Malloc(arglen*2); + if (!retbuf) + return PyErr_NoMemory(); + retval = NULL; /* silence a compiler warning, assigned later. */ + } else { + retval = PyUnicode_New(arglen*2, 127); + if (!retval) + return NULL; + retbuf = PyUnicode_1BYTE_DATA(retval); + } + + /* make hex version of string, taken from shamodule.c */ + for (i=j=0; i < arglen; i++) { + unsigned char c; + c = (argbuf[i] >> 4) & 0xf; + retbuf[j++] = Py_hexdigits[c]; + c = argbuf[i] & 0xf; + retbuf[j++] = Py_hexdigits[c]; + } + + if (return_bytes) { + retval = PyBytes_FromStringAndSize((const char *)retbuf, arglen*2); + PyMem_Free(retbuf); + } +#ifdef Py_DEBUG + else { + assert(_PyUnicode_CheckConsistency(retval, 1)); + } +#endif + + return retval; +} + +PyAPI_FUNC(PyObject *) _Py_strhex(const char* argbuf, const Py_ssize_t arglen) +{ + return _Py_strhex_impl(argbuf, arglen, 0); +} + +/* Same as above but returns a bytes() instead of str() to avoid the + * need to decode the str() when bytes are needed. */ +PyAPI_FUNC(PyObject *) _Py_strhex_bytes(const char* argbuf, const Py_ssize_t arglen) +{ + return _Py_strhex_impl(argbuf, arglen, 1); +} |