Commit 12a5a479 authored by Georg Brandl's avatar Georg Brandl

Bug #1545497: when given an explicit base, int() did ignore NULs

embedded in the string to convert.
parent fc6fa407
...@@ -729,6 +729,11 @@ class BuiltinTest(unittest.TestCase): ...@@ -729,6 +729,11 @@ class BuiltinTest(unittest.TestCase):
self.assertRaises(ValueError, int, '123\0') self.assertRaises(ValueError, int, '123\0')
self.assertRaises(ValueError, int, '53', 40) self.assertRaises(ValueError, int, '53', 40)
# SF bug 1545497: embedded NULs were not detected with
# explicit base
self.assertRaises(ValueError, int, '123\0', 10)
self.assertRaises(ValueError, int, '123\x00 245', 20)
x = int('1' * 600) x = int('1' * 600)
self.assert_(isinstance(x, long)) self.assert_(isinstance(x, long))
......
...@@ -12,6 +12,9 @@ What's New in Python 2.6 alpha 1? ...@@ -12,6 +12,9 @@ What's New in Python 2.6 alpha 1?
Core and builtins Core and builtins
----------------- -----------------
- Bug #1545497: when given an explicit base, int() did ignore NULs
embedded in the string to convert.
- Bug #1569998: break inside a try statement (outside a loop) is now - Bug #1569998: break inside a try statement (outside a loop) is now
recognized and rejected. recognized and rejected.
......
...@@ -987,8 +987,25 @@ int_new(PyTypeObject *type, PyObject *args, PyObject *kwds) ...@@ -987,8 +987,25 @@ int_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
return PyInt_FromLong(0L); return PyInt_FromLong(0L);
if (base == -909) if (base == -909)
return PyNumber_Int(x); return PyNumber_Int(x);
if (PyString_Check(x)) if (PyString_Check(x)) {
return PyInt_FromString(PyString_AS_STRING(x), NULL, base); /* Since PyInt_FromString doesn't have a length parameter,
* check here for possible NULs in the string. */
char *string = PyString_AS_STRING(x);
if (strlen(string) != PyString_Size(x)) {
/* create a repr() of the input string,
* just like PyInt_FromString does */
PyObject *srepr;
srepr = PyObject_Repr(x);
if (srepr == NULL)
return NULL;
PyErr_Format(PyExc_ValueError,
"invalid literal for int() with base %d: %s",
base, PyString_AS_STRING(srepr));
Py_DECREF(srepr);
return NULL;
}
return PyInt_FromString(string, NULL, base);
}
#ifdef Py_USING_UNICODE #ifdef Py_USING_UNICODE
if (PyUnicode_Check(x)) if (PyUnicode_Check(x))
return PyInt_FromUnicode(PyUnicode_AS_UNICODE(x), return PyInt_FromUnicode(PyUnicode_AS_UNICODE(x),
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment