mirror of https://github.com/python/cpython
bpo-33817: Fix _PyBytes_Resize() for empty bytes object. (GH-11516)
Add also tests for PyUnicode_FromFormat() and PyBytes_FromFormat() with empty result.
This commit is contained in:
parent
d0d3e99120
commit
44cc4822bb
|
@ -1001,6 +1001,12 @@ class BytesTest(BaseBytesTest, unittest.TestCase):
|
||||||
self.assertRaises(OverflowError,
|
self.assertRaises(OverflowError,
|
||||||
PyBytes_FromFormat, b'%c', c_int(256))
|
PyBytes_FromFormat, b'%c', c_int(256))
|
||||||
|
|
||||||
|
# Issue #33817: empty strings
|
||||||
|
self.assertEqual(PyBytes_FromFormat(b''),
|
||||||
|
b'')
|
||||||
|
self.assertEqual(PyBytes_FromFormat(b'%s', b''),
|
||||||
|
b'')
|
||||||
|
|
||||||
def test_bytes_blocking(self):
|
def test_bytes_blocking(self):
|
||||||
class IterationBlocked(list):
|
class IterationBlocked(list):
|
||||||
__bytes__ = None
|
__bytes__ = None
|
||||||
|
|
|
@ -2680,6 +2680,12 @@ class CAPITest(unittest.TestCase):
|
||||||
check_format('%.%s',
|
check_format('%.%s',
|
||||||
b'%.%s', b'abc')
|
b'%.%s', b'abc')
|
||||||
|
|
||||||
|
# Issue #33817: empty strings
|
||||||
|
check_format('',
|
||||||
|
b'')
|
||||||
|
check_format('',
|
||||||
|
b'%s', b'')
|
||||||
|
|
||||||
# Test PyUnicode_AsWideChar()
|
# Test PyUnicode_AsWideChar()
|
||||||
@support.cpython_only
|
@support.cpython_only
|
||||||
def test_aswidechar(self):
|
def test_aswidechar(self):
|
||||||
|
|
|
@ -0,0 +1 @@
|
||||||
|
Fixed :c:func:`_PyBytes_Resize` for empty bytes objects.
|
|
@ -2991,9 +2991,22 @@ _PyBytes_Resize(PyObject **pv, Py_ssize_t newsize)
|
||||||
/* return early if newsize equals to v->ob_size */
|
/* return early if newsize equals to v->ob_size */
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
if (Py_SIZE(v) == 0) {
|
||||||
|
if (newsize == 0) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
*pv = _PyBytes_FromSize(newsize, 0);
|
||||||
|
Py_DECREF(v);
|
||||||
|
return (*pv == NULL) ? -1 : 0;
|
||||||
|
}
|
||||||
if (Py_REFCNT(v) != 1) {
|
if (Py_REFCNT(v) != 1) {
|
||||||
goto error;
|
goto error;
|
||||||
}
|
}
|
||||||
|
if (newsize == 0) {
|
||||||
|
*pv = _PyBytes_FromSize(0, 0);
|
||||||
|
Py_DECREF(v);
|
||||||
|
return (*pv == NULL) ? -1 : 0;
|
||||||
|
}
|
||||||
/* XXX UNREF/NEWREF interface should be more symmetrical */
|
/* XXX UNREF/NEWREF interface should be more symmetrical */
|
||||||
_Py_DEC_REFTOTAL;
|
_Py_DEC_REFTOTAL;
|
||||||
_Py_ForgetReference(v);
|
_Py_ForgetReference(v);
|
||||||
|
|
Loading…
Reference in New Issue