mirrors/cpython
1
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-12-10 11:00:14 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 11

Factorize _PyBytes_FromList() and _PyBytes_FromTuple() code using a C macro

Browse source
This commit is contained in:
Victor Stinner 2015-10-14 13:50:40 +02:00
parent f2eafa323b
commit 3c50ce39bf
1 changed files with 35 additions and 54 deletions
Download patch file Download diff file Expand all files Collapse all files

89
Objects/bytesobject.c
View file

@ -3408,68 +3408,49 @@ fail:
return NULL; return NULL;
} }
#define _PyBytes_FROM_LIST_BODY(x, GET_ITEM) \
do { \
PyObject *bytes; \
Py_ssize_t i; \
Py_ssize_t value; \
char *str; \
PyObject *item; \
\
bytes = PyBytes_FromStringAndSize(NULL, Py_SIZE(x)); \
if (bytes == NULL) \
return NULL; \
str = ((PyBytesObject *)bytes)->ob_sval; \
\
for (i = 0; i < Py_SIZE(x); i++) { \
item = GET_ITEM((x), i); \
value = PyNumber_AsSsize_t(item, PyExc_ValueError); \
if (value == -1 && PyErr_Occurred()) \
goto error; \
\
if (value < 0 || value >= 256) { \
PyErr_SetString(PyExc_ValueError, \
"bytes must be in range(0, 256)"); \
goto error; \
} \
*str++ = (char) value; \
} \
return bytes; \
\
error: \
Py_DECREF(bytes); \
return NULL; \
} while (0)
static PyObject* static PyObject*
_PyBytes_FromList(PyObject *x) _PyBytes_FromList(PyObject *x)
{ {
PyObject *new; _PyBytes_FROM_LIST_BODY(x, PyList_GET_ITEM);
Py_ssize_t i;
Py_ssize_t value;
char *str;
new = PyBytes_FromStringAndSize(NULL, Py_SIZE(x));
if (new == NULL)
return NULL;
str = ((PyBytesObject *)new)->ob_sval;
for (i = 0; i < Py_SIZE(x); i++) {
value = PyNumber_AsSsize_t(PyList_GET_ITEM(x, i), PyExc_ValueError);
if (value == -1 && PyErr_Occurred())
goto error;
if (value < 0 || value >= 256) {
PyErr_SetString(PyExc_ValueError,
"bytes must be in range(0, 256)");
goto error;
}
*str++ = (char) value;
}
return new;
error:
Py_DECREF(new);
return NULL;
} }
static PyObject* static PyObject*
_PyBytes_FromTuple(PyObject *x) _PyBytes_FromTuple(PyObject *x)
{ {
PyObject *new; _PyBytes_FROM_LIST_BODY(x, PyTuple_GET_ITEM);
Py_ssize_t i;
Py_ssize_t value;
char *str;
new = PyBytes_FromStringAndSize(NULL, Py_SIZE(x));
if (new == NULL)
return NULL;
str = ((PyBytesObject *)new)->ob_sval;
for (i = 0; i < Py_SIZE(x); i++) {
value = PyNumber_AsSsize_t(PyTuple_GET_ITEM(x, i), PyExc_ValueError);
if (value == -1 && PyErr_Occurred())
goto error;
if (value < 0 || value >= 256) {
PyErr_SetString(PyExc_ValueError,
"bytes must be in range(0, 256)");
goto error;
}
*str++ = (char) value;
}
return new;
error:
Py_DECREF(new);
return NULL;
} }
static PyObject * static PyObject *