1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
|
/* work around pypy3.9 v7.3.9 bug with Py_LIMITED_API */
#ifdef PYPY_VERSION
#ifndef Py_None
#define Py_None (&_Py_NoneStruct)
#endif
#endif
/*
* Get string data from Python object.
*/
static Py_ssize_t get_buffer(PyObject *obj, unsigned char **buf_p, PyObject **tmp_obj_p)
{
PyObject *str = NULL;
Py_ssize_t res;
/* check for None */
if (obj == Py_None) {
PyErr_Format(PyExc_TypeError, "None is not allowed");
return -1;
}
/* quick path for bytes */
if (PyBytes_Check(obj)) {
if (PyBytes_AsStringAndSize(obj, (char**)buf_p, &res) < 0)
return -1;
return res;
}
/* convert to bytes */
if (PyUnicode_Check(obj)) {
/* no direct string access in abi3 */
*tmp_obj_p = PyUnicode_AsUTF8String(obj);
} else if (PyMemoryView_Check(obj) || PyByteArray_Check(obj)) {
/* no direct buffer access in abi3 */
*tmp_obj_p = PyBytes_FromObject(obj);
} else {
/* Not a string-like object, run str() or it. */
str = PyObject_Str(obj);
if (str == NULL)
return -1;
*tmp_obj_p = PyUnicode_AsUTF8String(str);
Py_CLEAR(str);
}
if (*tmp_obj_p == NULL)
return -1;
if (PyBytes_AsStringAndSize(*tmp_obj_p, (char**)buf_p, &res) < 0)
return -1;
return res;
}
|
