mirrors/cpython
1
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-08-04 00:48:58 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 8

Issue #6697: Fixed instances of _PyUnicode_AsString() result not checked for NULL

Browse source
This commit is contained in:
Alexander Belopolsky 2010-12-08 23:31:48 +00:00
parent 1b2bd3b348
commit e239d23e8c
13 changed files with 144 additions and 78 deletions
Download patch file Download diff file Expand all files Collapse all files

8
Lib/test/datetimetester.py
View file

@ -2508,11 +2508,17 @@ class TestTimeTZ(TestTime, TZInfoBase, unittest.TestCase):
# Check that an invalid tzname result raises an exception. # Check that an invalid tzname result raises an exception.
class Badtzname(tzinfo): class Badtzname(tzinfo):
def tzname(self, dt): return 42 tz = 42
def tzname(self, dt): return self.tz
t = time(2, 3, 4, tzinfo=Badtzname()) t = time(2, 3, 4, tzinfo=Badtzname())
self.assertEqual(t.strftime("%H:%M:%S"), "02:03:04") self.assertEqual(t.strftime("%H:%M:%S"), "02:03:04")
self.assertRaises(TypeError, t.strftime, "%Z") self.assertRaises(TypeError, t.strftime, "%Z")
# Issue #6697:
if '_Fast' in str(type(self)):
Badtzname.tz = '\ud800'
self.assertRaises(ValueError, t.strftime, "%Z")
def test_hash_edge_cases(self): def test_hash_edge_cases(self):
# Offsets that overflow a basic time. # Offsets that overflow a basic time.
t1 = self.theclass(0, 1, 2, 3, tzinfo=FixedOffset(1439, "")) t1 = self.theclass(0, 1, 2, 3, tzinfo=FixedOffset(1439, ""))

2
Lib/test/test_pyexpat.py
View file

@ -203,6 +203,8 @@ class ParseTest(unittest.TestCase):
operations = out.out operations = out.out
self._verify_parse_output(operations) self._verify_parse_output(operations)
# Issue #6697.
self.assertRaises(AttributeError, getattr, parser, '\uD800')
def test_parse_file(self): def test_parse_file(self):
# Try parsing a file # Try parsing a file

2
Lib/test/test_socket.py
View file

@ -667,6 +667,8 @@ class GeneralModuleTests(unittest.TestCase):
type=socket.SOCK_STREAM, proto=0, type=socket.SOCK_STREAM, proto=0,
flags=socket.AI_PASSIVE) flags=socket.AI_PASSIVE)
self.assertEqual(a, b) self.assertEqual(a, b)
# Issue #6697.
self.assertRaises(UnicodeEncodeError, socket.getaddrinfo, 'localhost', '\uD800')
def test_getnameinfo(self): def test_getnameinfo(self):
# only IP addresses are allowed # only IP addresses are allowed

2
Lib/test/test_syslog.py
View file

@ -11,6 +11,8 @@ class Test(unittest.TestCase):
def test_openlog(self): def test_openlog(self):
syslog.openlog('python') syslog.openlog('python')
# Issue #6697.
self.assertRaises(UnicodeEncodeError, syslog.openlog, '\uD800')
def test_syslog(self): def test_syslog(self):
syslog.openlog('python') syslog.openlog('python')

18
Lib/test/test_xml_etree_c.py
View file

@ -8,10 +8,26 @@ cET = support.import_module('xml.etree.cElementTree')
# cElementTree specific tests # cElementTree specific tests
def sanity(): def sanity():
""" r"""
Import sanity. Import sanity.
>>> from xml.etree import cElementTree >>> from xml.etree import cElementTree
Issue #6697.
>>> e = cElementTree.Element('a')
>>> getattr(e, '\uD800') # doctest: +ELLIPSIS
Traceback (most recent call last):
...
UnicodeEncodeError: ...
>>> p = cElementTree.XMLParser()
>>> p.version.split()[0]
'Expat'
>>> getattr(p, '\uD800')
Traceback (most recent call last):
...
AttributeError: 'XMLParser' object has no attribute '\ud800'
""" """

3
Modules/_datetimemodule.c
View file

@ -1257,7 +1257,8 @@ wrap_strftime(PyObject *object, PyObject *format, PyObject *timetuple,
assert(PyUnicode_Check(Zreplacement)); assert(PyUnicode_Check(Zreplacement));
ptoappend = _PyUnicode_AsStringAndSize(Zreplacement, ptoappend = _PyUnicode_AsStringAndSize(Zreplacement,
&ntoappend); &ntoappend);
ntoappend = Py_SIZE(Zreplacement); if (ptoappend == NULL)
goto Done;
} }
else if (ch == 'f') { else if (ch == 'f') {
/* format microseconds */ /* format microseconds */

47
Modules/_elementtree.c
View file

@ -1483,6 +1483,9 @@ element_getattro(ElementObject* self, PyObject* nameobj)
if (PyUnicode_Check(nameobj)) if (PyUnicode_Check(nameobj))
name = _PyUnicode_AsString(nameobj); name = _PyUnicode_AsString(nameobj);
if (name == NULL)
return NULL;
/* handle common attributes first */ /* handle common attributes first */
if (strcmp(name, "tag") == 0) { if (strcmp(name, "tag") == 0) {
@ -2139,7 +2142,7 @@ expat_set_error(const char* message, int line, int column)
PyObject *position; PyObject *position;
char buffer[256]; char buffer[256];
sprintf(buffer, "%s: line %d, column %d", message, line, column); sprintf(buffer, "%.100s: line %d, column %d", message, line, column);
error = PyObject_CallFunction(elementtree_parseerror_obj, "s", buffer); error = PyObject_CallFunction(elementtree_parseerror_obj, "s", buffer);
if (!error) if (!error)
@ -2194,8 +2197,8 @@ expat_default_handler(XMLParserObject* self, const XML_Char* data_in,
Py_XDECREF(res); Py_XDECREF(res);
} else if (!PyErr_Occurred()) { } else if (!PyErr_Occurred()) {
/* Report the first error, not the last */ /* Report the first error, not the last */
char message[128]; char message[128] = "undefined entity ";
sprintf(message, "undefined entity &%.100s;", _PyUnicode_AsString(key)); strncat(message, data_in, data_len < 100?data_len:100);
expat_set_error( expat_set_error(
message, message,
EXPAT(GetErrorLineNumber)(self->parser), EXPAT(GetErrorLineNumber)(self->parser),
@ -2796,29 +2799,25 @@ static PyMethodDef xmlparser_methods[] = {
static PyObject* static PyObject*
xmlparser_getattro(XMLParserObject* self, PyObject* nameobj) xmlparser_getattro(XMLParserObject* self, PyObject* nameobj)
{ {
PyObject* res; if (PyUnicode_Check(nameobj)) {
char *name = ""; PyObject* res;
if (PyUnicode_CompareWithASCIIString(nameobj, "entity") == 0)
if (PyUnicode_Check(nameobj)) res = self->entity;
name = _PyUnicode_AsString(nameobj); else if (PyUnicode_CompareWithASCIIString(nameobj, "target") == 0)
res = self->target;
PyErr_Clear(); else if (PyUnicode_CompareWithASCIIString(nameobj, "version") == 0) {
return PyUnicode_FromFormat(
if (strcmp(name, "entity") == 0) "Expat %d.%d.%d", XML_MAJOR_VERSION,
res = self->entity;
else if (strcmp(name, "target") == 0)
res = self->target;
else if (strcmp(name, "version") == 0) {
char buffer[100];
sprintf(buffer, "Expat %d.%d.%d", XML_MAJOR_VERSION,
XML_MINOR_VERSION, XML_MICRO_VERSION); XML_MINOR_VERSION, XML_MICRO_VERSION);
return PyUnicode_DecodeUTF8(buffer, strlen(buffer), "strict"); }
} else { else
return PyObject_GenericGetAttr((PyObject*) self, nameobj); goto generic;
}
Py_INCREF(res); Py_INCREF(res);
return res; return res;
}
generic:
return PyObject_GenericGetAttr((PyObject*) self, nameobj);
} }
static PyTypeObject XMLParser_Type = { static PyTypeObject XMLParser_Type = {

9
Modules/_lsprof.c
View file

@ -178,7 +178,16 @@ normalizeUserObj(PyObject *obj)
PyObject *mod = fn->m_module; PyObject *mod = fn->m_module;
const char *modname; const char *modname;
if (mod && PyUnicode_Check(mod)) { if (mod && PyUnicode_Check(mod)) {
/* XXX: The following will truncate module names with embedded
* null-characters. It is unlikely that this can happen in
* practice and the concequences are not serious enough to
* introduce extra checks here.
*/
modname = _PyUnicode_AsString(mod); modname = _PyUnicode_AsString(mod);
if (modname == NULL) {
modname = "<encoding error>";
PyErr_Clear();
}
} }
else if (mod && PyModule_Check(mod)) { else if (mod && PyModule_Check(mod)) {
modname = PyModule_GetName(mod); modname = PyModule_GetName(mod);

17
Modules/_testcapimodule.c
View file

@ -1741,15 +1741,16 @@ test_string_from_format(PyObject *self, PyObject *args)
{ {
PyObject *result; PyObject *result;
char *msg; char *msg;
static const Py_UNICODE one[] = {'1', 0};
#define CHECK_1_FORMAT(FORMAT, TYPE) \ #define CHECK_1_FORMAT(FORMAT, TYPE) \
result = PyUnicode_FromFormat(FORMAT, (TYPE)1); \ result = PyUnicode_FromFormat(FORMAT, (TYPE)1); \
if (result == NULL) \ if (result == NULL) \
return NULL; \ return NULL; \
if (strcmp(_PyUnicode_AsString(result), "1")) { \ if (Py_UNICODE_strcmp(PyUnicode_AS_UNICODE(result), one)) { \
msg = FORMAT " failed at 1"; \ msg = FORMAT " failed at 1"; \
goto Fail; \ goto Fail; \
} \ } \
Py_DECREF(result) Py_DECREF(result)
CHECK_1_FORMAT("%d", int); CHECK_1_FORMAT("%d", int);

13
Modules/parsermodule.c
View file

@ -792,6 +792,11 @@ build_node_children(PyObject *tuple, node *root, int *line_num)
} }
} }
temp_str = _PyUnicode_AsStringAndSize(temp, &len); temp_str = _PyUnicode_AsStringAndSize(temp, &len);
if (temp_str == NULL) {
Py_DECREF(temp);
Py_XDECREF(elem);
return 0;
}
strn = (char *)PyObject_MALLOC(len + 1); strn = (char *)PyObject_MALLOC(len + 1);
if (strn != NULL) if (strn != NULL)
(void) memcpy(strn, temp_str, len + 1); (void) memcpy(strn, temp_str, len + 1);
@ -870,6 +875,8 @@ build_node_tree(PyObject *tuple)
encoding = PySequence_GetItem(tuple, 2); encoding = PySequence_GetItem(tuple, 2);
/* tuple isn't borrowed anymore here, need to DECREF */ /* tuple isn't borrowed anymore here, need to DECREF */
tuple = PySequence_GetSlice(tuple, 0, 2); tuple = PySequence_GetSlice(tuple, 0, 2);
if (tuple == NULL)
return NULL;
} }
res = PyNode_New(num); res = PyNode_New(num);
if (res != NULL) { if (res != NULL) {
@ -881,6 +888,12 @@ build_node_tree(PyObject *tuple)
Py_ssize_t len; Py_ssize_t len;
const char *temp; const char *temp;
temp = _PyUnicode_AsStringAndSize(encoding, &len); temp = _PyUnicode_AsStringAndSize(encoding, &len);
if (temp == NULL) {
Py_DECREF(res);
Py_DECREF(encoding);
Py_DECREF(tuple);
return NULL;
}
res->n_str = (char *)PyObject_MALLOC(len + 1); res->n_str = (char *)PyObject_MALLOC(len + 1);
if (res->n_str != NULL && temp != NULL) if (res->n_str != NULL && temp != NULL)
(void) memcpy(res->n_str, temp, len + 1); (void) memcpy(res->n_str, temp, len + 1);

68
Modules/pyexpat.c
View file

@ -1215,11 +1215,12 @@ xmlparse_dealloc(xmlparseobject *self)
} }
static int static int
handlername2int(const char *name) handlername2int(PyObject *name)
{ {
int i; int i;
for (i = 0; handler_info[i].name != NULL; i++) { for (i = 0; handler_info[i].name != NULL; i++) {
if (strcmp(name, handler_info[i].name) == 0) { if (PyUnicode_CompareWithASCIIString(
name, handler_info[i].name) == 0) {
return i; return i;
} }
} }
@ -1237,13 +1238,13 @@ get_pybool(int istrue)
static PyObject * static PyObject *
xmlparse_getattro(xmlparseobject *self, PyObject *nameobj) xmlparse_getattro(xmlparseobject *self, PyObject *nameobj)
{ {
char *name = ""; Py_UNICODE *name;
int handlernum = -1; int handlernum = -1;
if (PyUnicode_Check(nameobj)) if (!PyUnicode_Check(nameobj))
name = _PyUnicode_AsString(nameobj); goto generic;
handlernum = handlername2int(name); handlernum = handlername2int(nameobj);
if (handlernum != -1) { if (handlernum != -1) {
PyObject *result = self->handlers[handlernum]; PyObject *result = self->handlers[handlernum];
@ -1252,46 +1253,48 @@ xmlparse_getattro(xmlparseobject *self, PyObject *nameobj)
Py_INCREF(result); Py_INCREF(result);
return result; return result;
} }
name = PyUnicode_AS_UNICODE(nameobj);
if (name[0] == 'E') { if (name[0] == 'E') {
if (strcmp(name, "ErrorCode") == 0) if (PyUnicode_CompareWithASCIIString(nameobj, "ErrorCode") == 0)
return PyLong_FromLong((long) return PyLong_FromLong((long)
XML_GetErrorCode(self->itself)); XML_GetErrorCode(self->itself));
if (strcmp(name, "ErrorLineNumber") == 0) if (PyUnicode_CompareWithASCIIString(nameobj, "ErrorLineNumber") == 0)
return PyLong_FromLong((long) return PyLong_FromLong((long)
XML_GetErrorLineNumber(self->itself)); XML_GetErrorLineNumber(self->itself));
if (strcmp(name, "ErrorColumnNumber") == 0) if (PyUnicode_CompareWithASCIIString(nameobj, "ErrorColumnNumber") == 0)
return PyLong_FromLong((long) return PyLong_FromLong((long)
XML_GetErrorColumnNumber(self->itself)); XML_GetErrorColumnNumber(self->itself));
if (strcmp(name, "ErrorByteIndex") == 0) if (PyUnicode_CompareWithASCIIString(nameobj, "ErrorByteIndex") == 0)
return PyLong_FromLong((long) return PyLong_FromLong((long)
XML_GetErrorByteIndex(self->itself)); XML_GetErrorByteIndex(self->itself));
} }
if (name[0] == 'C') { if (name[0] == 'C') {
if (strcmp(name, "CurrentLineNumber") == 0) if (PyUnicode_CompareWithASCIIString(nameobj, "CurrentLineNumber") == 0)
return PyLong_FromLong((long) return PyLong_FromLong((long)
XML_GetCurrentLineNumber(self->itself)); XML_GetCurrentLineNumber(self->itself));
if (strcmp(name, "CurrentColumnNumber") == 0) if (PyUnicode_CompareWithASCIIString(nameobj, "CurrentColumnNumber") == 0)
return PyLong_FromLong((long) return PyLong_FromLong((long)
XML_GetCurrentColumnNumber(self->itself)); XML_GetCurrentColumnNumber(self->itself));
if (strcmp(name, "CurrentByteIndex") == 0) if (PyUnicode_CompareWithASCIIString(nameobj, "CurrentByteIndex") == 0)
return PyLong_FromLong((long) return PyLong_FromLong((long)
XML_GetCurrentByteIndex(self->itself)); XML_GetCurrentByteIndex(self->itself));
} }
if (name[0] == 'b') { if (name[0] == 'b') {
if (strcmp(name, "buffer_size") == 0) if (PyUnicode_CompareWithASCIIString(nameobj, "buffer_size") == 0)
return PyLong_FromLong((long) self->buffer_size); return PyLong_FromLong((long) self->buffer_size);
if (strcmp(name, "buffer_text") == 0) if (PyUnicode_CompareWithASCIIString(nameobj, "buffer_text") == 0)
return get_pybool(self->buffer != NULL); return get_pybool(self->buffer != NULL);
if (strcmp(name, "buffer_used") == 0) if (PyUnicode_CompareWithASCIIString(nameobj, "buffer_used") == 0)
return PyLong_FromLong((long) self->buffer_used); return PyLong_FromLong((long) self->buffer_used);
} }
if (strcmp(name, "namespace_prefixes") == 0) if (PyUnicode_CompareWithASCIIString(nameobj, "namespace_prefixes") == 0)
return get_pybool(self->ns_prefixes); return get_pybool(self->ns_prefixes);
if (strcmp(name, "ordered_attributes") == 0) if (PyUnicode_CompareWithASCIIString(nameobj, "ordered_attributes") == 0)
return get_pybool(self->ordered_attributes); return get_pybool(self->ordered_attributes);
if (strcmp(name, "specified_attributes") == 0) if (PyUnicode_CompareWithASCIIString(nameobj, "specified_attributes") == 0)
return get_pybool((long) self->specified_attributes); return get_pybool((long) self->specified_attributes);
if (strcmp(name, "intern") == 0) { if (PyUnicode_CompareWithASCIIString(nameobj, "intern") == 0) {
if (self->intern == NULL) { if (self->intern == NULL) {
Py_INCREF(Py_None); Py_INCREF(Py_None);
return Py_None; return Py_None;
@ -1301,7 +1304,7 @@ xmlparse_getattro(xmlparseobject *self, PyObject *nameobj)
return self->intern; return self->intern;
} }
} }
generic:
return PyObject_GenericGetAttr((PyObject*)self, nameobj); return PyObject_GenericGetAttr((PyObject*)self, nameobj);
} }
@ -1352,7 +1355,7 @@ xmlparse_dir(PyObject *self, PyObject* noargs)
} }
static int static int
sethandler(xmlparseobject *self, const char *name, PyObject* v) sethandler(xmlparseobject *self, PyObject *name, PyObject* v)
{ {
int handlernum = handlername2int(name); int handlernum = handlername2int(name);
if (handlernum >= 0) { if (handlernum >= 0) {
@ -1388,14 +1391,15 @@ sethandler(xmlparseobject *self, const char *name, PyObject* v)
} }
static int static int
xmlparse_setattr(xmlparseobject *self, char *name, PyObject *v) xmlparse_setattro(xmlparseobject *self, PyObject *name, PyObject *v)
{ {
/* Set attribute 'name' to value 'v'. v==NULL means delete */ /* Set attribute 'name' to value 'v'. v==NULL means delete */
if (v == NULL) { if (v == NULL) {
PyErr_SetString(PyExc_RuntimeError, "Cannot delete attribute"); PyErr_SetString(PyExc_RuntimeError, "Cannot delete attribute");
return -1; return -1;
} }
if (strcmp(name, "buffer_text") == 0) { assert(PyUnicode_Check(name));
if (PyUnicode_CompareWithASCIIString(name, "buffer_text") == 0) {
if (PyObject_IsTrue(v)) { if (PyObject_IsTrue(v)) {
if (self->buffer == NULL) { if (self->buffer == NULL) {
self->buffer = malloc(self->buffer_size); self->buffer = malloc(self->buffer_size);
@ -1414,7 +1418,7 @@ xmlparse_setattr(xmlparseobject *self, char *name, PyObject *v)
} }
return 0; return 0;
} }
if (strcmp(name, "namespace_prefixes") == 0) { if (PyUnicode_CompareWithASCIIString(name, "namespace_prefixes") == 0) {
if (PyObject_IsTrue(v)) if (PyObject_IsTrue(v))
self->ns_prefixes = 1; self->ns_prefixes = 1;
else else
@ -1422,14 +1426,14 @@ xmlparse_setattr(xmlparseobject *self, char *name, PyObject *v)
XML_SetReturnNSTriplet(self->itself, self->ns_prefixes); XML_SetReturnNSTriplet(self->itself, self->ns_prefixes);
return 0; return 0;
} }
if (strcmp(name, "ordered_attributes") == 0) { if (PyUnicode_CompareWithASCIIString(name, "ordered_attributes") == 0) {
if (PyObject_IsTrue(v)) if (PyObject_IsTrue(v))
self->ordered_attributes = 1; self->ordered_attributes = 1;
else else
self->ordered_attributes = 0; self->ordered_attributes = 0;
return 0; return 0;
} }
if (strcmp(name, "specified_attributes") == 0) { if (PyUnicode_CompareWithASCIIString(name, "specified_attributes") == 0) {
if (PyObject_IsTrue(v)) if (PyObject_IsTrue(v))
self->specified_attributes = 1; self->specified_attributes = 1;
else else
@ -1437,7 +1441,7 @@ xmlparse_setattr(xmlparseobject *self, char *name, PyObject *v)
return 0; return 0;
} }
if (strcmp(name, "buffer_size") == 0) { if (PyUnicode_CompareWithASCIIString(name, "buffer_size") == 0) {
long new_buffer_size; long new_buffer_size;
if (!PyLong_Check(v)) { if (!PyLong_Check(v)) {
PyErr_SetString(PyExc_TypeError, "buffer_size must be an integer"); PyErr_SetString(PyExc_TypeError, "buffer_size must be an integer");
@ -1480,7 +1484,7 @@ xmlparse_setattr(xmlparseobject *self, char *name, PyObject *v)
return 0; return 0;
} }
if (strcmp(name, "CharacterDataHandler") == 0) { if (PyUnicode_CompareWithASCIIString(name, "CharacterDataHandler") == 0) {
/* If we're changing the character data handler, flush all /* If we're changing the character data handler, flush all
* cached data with the old handler. Not sure there's a * cached data with the old handler. Not sure there's a
* "right" thing to do, though, but this probably won't * "right" thing to do, though, but this probably won't
@ -1492,7 +1496,7 @@ xmlparse_setattr(xmlparseobject *self, char *name, PyObject *v)
if (sethandler(self, name, v)) { if (sethandler(self, name, v)) {
return 0; return 0;
} }
PyErr_SetString(PyExc_AttributeError, name); PyErr_SetObject(PyExc_AttributeError, name);
return -1; return -1;
} }
@ -1524,7 +1528,7 @@ static PyTypeObject Xmlparsetype = {
(destructor)xmlparse_dealloc, /*tp_dealloc*/ (destructor)xmlparse_dealloc, /*tp_dealloc*/
(printfunc)0, /*tp_print*/ (printfunc)0, /*tp_print*/
0, /*tp_getattr*/ 0, /*tp_getattr*/
(setattrfunc)xmlparse_setattr, /*tp_setattr*/ 0, /*tp_setattr*/
0, /*tp_reserved*/ 0, /*tp_reserved*/
(reprfunc)0, /*tp_repr*/ (reprfunc)0, /*tp_repr*/
0, /*tp_as_number*/ 0, /*tp_as_number*/
@ -1534,7 +1538,7 @@ static PyTypeObject Xmlparsetype = {
(ternaryfunc)0, /*tp_call*/ (ternaryfunc)0, /*tp_call*/
(reprfunc)0, /*tp_str*/ (reprfunc)0, /*tp_str*/
(getattrofunc)xmlparse_getattro, /* tp_getattro */ (getattrofunc)xmlparse_getattro, /* tp_getattro */
0, /* tp_setattro */ (setattrofunc)xmlparse_setattro, /* tp_setattro */
0, /* tp_as_buffer */ 0, /* tp_as_buffer */
Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC, /*tp_flags*/ Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC, /*tp_flags*/
Xmlparsetype__doc__, /* tp_doc - Documentation string */ Xmlparsetype__doc__, /* tp_doc - Documentation string */

8
Modules/socketmodule.c
View file

@ -1406,9 +1406,9 @@ getsockaddrarg(PySocketSockObject *s, PyObject *args,
{ {
struct sockaddr_hci *addr = (struct sockaddr_hci *)addr_ret; struct sockaddr_hci *addr = (struct sockaddr_hci *)addr_ret;
#if defined(__NetBSD__) || defined(__DragonFly__) #if defined(__NetBSD__) || defined(__DragonFly__)
char *straddr = PyBytes_AS_STRING(args); char *straddr = PyBytes_AS_STRING(args);
_BT_HCI_MEMB(addr, family) = AF_BLUETOOTH; _BT_HCI_MEMB(addr, family) = AF_BLUETOOTH;
if (straddr == NULL) { if (straddr == NULL) {
PyErr_SetString(socket_error, "getsockaddrarg: " PyErr_SetString(socket_error, "getsockaddrarg: "
"wrong format"); "wrong format");
@ -4022,8 +4022,10 @@ socket_getaddrinfo(PyObject *self, PyObject *args, PyObject* kwargs)
pptr = pbuf; pptr = pbuf;
} else if (PyUnicode_Check(pobj)) { } else if (PyUnicode_Check(pobj)) {
pptr = _PyUnicode_AsString(pobj); pptr = _PyUnicode_AsString(pobj);
if (pptr == NULL)
goto err;
} else if (PyBytes_Check(pobj)) { } else if (PyBytes_Check(pobj)) {
pptr = PyBytes_AsString(pobj); pptr = PyBytes_AS_STRING(pobj);
} else if (pobj == Py_None) { } else if (pobj == Py_None) {
pptr = (char *)NULL; pptr = (char *)NULL;
} else { } else {

25
Modules/syslogmodule.c
View file

@ -68,9 +68,9 @@ syslog_get_argv(void)
* is optional. * is optional.
*/ */
Py_ssize_t argv_len; Py_ssize_t argv_len, scriptlen;
PyObject *scriptobj; PyObject *scriptobj;
char *atslash; Py_UNICODE *atslash, *atstart;
PyObject *argv = PySys_GetObject("argv"); PyObject *argv = PySys_GetObject("argv");
if (argv == NULL) { if (argv == NULL) {
@ -90,13 +90,16 @@ syslog_get_argv(void)
if (!PyUnicode_Check(scriptobj)) { if (!PyUnicode_Check(scriptobj)) {
return(NULL); return(NULL);
} }
if (PyUnicode_GET_SIZE(scriptobj) == 0) { scriptlen = PyUnicode_GET_SIZE(scriptobj);
if (scriptlen == 0) {
return(NULL); return(NULL);
} }
atslash = strrchr(_PyUnicode_AsString(scriptobj), SEP); atstart = PyUnicode_AS_UNICODE(scriptobj);
atslash = Py_UNICODE_strrchr(atstart, SEP);
if (atslash) { if (atslash) {
return(PyUnicode_FromString(atslash + 1)); return(PyUnicode_FromUnicode(atslash + 1,
scriptlen - (atslash - atstart) - 1));
} else { } else {
Py_INCREF(scriptobj); Py_INCREF(scriptobj);
return(scriptobj); return(scriptobj);
@ -113,6 +116,7 @@ syslog_openlog(PyObject * self, PyObject * args, PyObject *kwds)
long facility = LOG_USER; long facility = LOG_USER;
PyObject *new_S_ident_o = NULL; PyObject *new_S_ident_o = NULL;
static char *keywords[] = {"ident", "logoption", "facility", 0}; static char *keywords[] = {"ident", "logoption", "facility", 0};
char *ident = NULL;
if (!PyArg_ParseTupleAndKeywords(args, kwds, if (!PyArg_ParseTupleAndKeywords(args, kwds,
"|Ull:openlog", keywords, &new_S_ident_o, &logopt, &facility)) "|Ull:openlog", keywords, &new_S_ident_o, &logopt, &facility))
@ -120,12 +124,12 @@ syslog_openlog(PyObject * self, PyObject * args, PyObject *kwds)
if (new_S_ident_o) { if (new_S_ident_o) {
Py_INCREF(new_S_ident_o); Py_INCREF(new_S_ident_o);
} }
/* get sys.argv[0] or NULL if we can't for some reason */ /* get sys.argv[0] or NULL if we can't for some reason */
if (!new_S_ident_o) { if (!new_S_ident_o) {
new_S_ident_o = syslog_get_argv(); new_S_ident_o = syslog_get_argv();
} }
Py_XDECREF(S_ident_o); Py_XDECREF(S_ident_o);
S_ident_o = new_S_ident_o; S_ident_o = new_S_ident_o;
@ -134,8 +138,13 @@ syslog_openlog(PyObject * self, PyObject * args, PyObject *kwds)
* make a copy, and syslog(3) later uses it. We can't garbagecollect it * make a copy, and syslog(3) later uses it. We can't garbagecollect it
* If NULL, just let openlog figure it out (probably using C argv[0]). * If NULL, just let openlog figure it out (probably using C argv[0]).
*/ */
if (S_ident_o) {
ident = _PyUnicode_AsString(S_ident_o);
if (ident == NULL)
return NULL;
}
openlog(S_ident_o ? _PyUnicode_AsString(S_ident_o) : NULL, logopt, facility); openlog(ident, logopt, facility);
S_log_open = 1; S_log_open = 1;
Py_INCREF(Py_None); Py_INCREF(Py_None);