Skip to content

Commit

Permalink
bpo-33817: Fix _PyBytes_Resize() for empty bytes object. (GH-11516)
Browse files Browse the repository at this point in the history
Add also tests for PyUnicode_FromFormat() and PyBytes_FromFormat()
with empty result.
(cherry picked from commit 44cc482)

Co-authored-by: Serhiy Storchaka <storchaka@gmail.com>
  • Loading branch information
miss-islington and serhiy-storchaka committed Jan 12, 2019
1 parent 059997d commit d39c192
Show file tree
Hide file tree
Showing 4 changed files with 26 additions and 0 deletions.
6 changes: 6 additions & 0 deletions Lib/test/test_bytes.py
Expand Up @@ -999,6 +999,12 @@ def ptr_formatter(ptr):
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
Expand Down
6 changes: 6 additions & 0 deletions Lib/test/test_unicode.py
Expand Up @@ -2676,6 +2676,12 @@ def check_format(expected, format, *args):
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):
Expand Down
@@ -0,0 +1 @@
Fixed :c:func:`_PyBytes_Resize` for empty bytes objects.
13 changes: 13 additions & 0 deletions Objects/bytesobject.c
Expand Up @@ -2990,9 +2990,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);
Expand Down

0 comments on commit d39c192

Please sign in to comment.