traverseda
/
cpython
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fix escape_encode to return the correct consumed size

This commit is contained in:
Philip Jenvey 2010-04-05 02:51:51 +00:00
parent dd194ab37b
commit 034b0acdd3
2 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
Lib/test/test_codecs.py
View File

@ -835,6 +835,9 @@ class UnicodeInternalTest(unittest.TestCase):
self.assertEquals(encoder(u"a")[1], 1)
self.assertEquals(encoder(u"\xe9\u0142")[1], 2)
encoder = codecs.getencoder("string-escape")
self.assertEquals(encoder(r'\x00')[1], 4)
# From http://www.gnu.org/software/libidn/draft-josefsson-idn-test-vectors.html
nameprep_tests = [
# 3.1 Map to nothing.

9
Modules/_codecsmodule.c
View File

@ -179,12 +179,13 @@ escape_encode(PyObject *self,
PyObject *str;
const char *errors = NULL;
char *buf;
Py_ssize_t len;
Py_ssize_t consumed, len;
if (!PyArg_ParseTuple(args, "O!|z:escape_encode",
&PyString_Type, &str, &errors))
if (!PyArg_ParseTuple(args, "S|z:escape_encode",
&str, &errors))
return NULL;
consumed = PyString_GET_SIZE(str);
str = PyString_Repr(str, 0);
if (!str)
return NULL;
@ -196,7 +197,7 @@ escape_encode(PyObject *self,
if (_PyString_Resize(&str, len-2) < 0)
return NULL;
return codec_tuple(str, PyString_Size(str));
return codec_tuple(str, consumed);
}
#ifdef Py_USING_UNICODE