[Python-checkins] cpython: PyUnicode_WriteChar() raises IndexError on invalid index
victor.stinner
python-checkins at python.org
Sun Oct 2 01:14:17 CEST 2011
http://hg.python.org/cpython/rev/99aa46107a22
changeset: 72575:99aa46107a22
user: Victor Stinner <victor.stinner at haypocalc.com>
date: Sun Oct 02 00:34:53 2011 +0200
summary:
PyUnicode_WriteChar() raises IndexError on invalid index
PyUnicode_WriteChar() raises also a ValueError if the string has more than 1
reference.
files:
Include/unicodeobject.h | 4 +++-
Objects/unicodeobject.c | 28 +++++++++++++++++++++-------
2 files changed, 24 insertions(+), 8 deletions(-)
diff --git a/Include/unicodeobject.h b/Include/unicodeobject.h
--- a/Include/unicodeobject.h
+++ b/Include/unicodeobject.h
@@ -647,7 +647,9 @@
);
/* Write a character to the string. The string must have been created through
- PyUnicode_New, must not be shared, and must not have been hashed yet. */
+ PyUnicode_New, must not be shared, and must not have been hashed yet.
+
+ Return 0 on success, -1 on error. */
PyAPI_FUNC(int) PyUnicode_WriteChar(
PyObject *unicode,
diff --git a/Objects/unicodeobject.c b/Objects/unicodeobject.c
--- a/Objects/unicodeobject.c
+++ b/Objects/unicodeobject.c
@@ -622,6 +622,19 @@
}
#endif
+static int
+_PyUnicode_Dirty(PyObject *unicode)
+{
+ assert(PyUnicode_Check(unicode));
+ if (Py_REFCNT(unicode) != 1) {
+ PyErr_SetString(PyExc_ValueError,
+ "Cannot modify a string having more than 1 reference");
+ return -1;
+ }
+ _PyUnicode_DIRTY(unicode);
+ return 0;
+}
+
Py_ssize_t
PyUnicode_CopyCharacters(PyObject *to, Py_ssize_t to_start,
PyObject *from, Py_ssize_t from_start,
@@ -651,12 +664,8 @@
if (how_many == 0)
return 0;
- if (Py_REFCNT(to) != 1) {
- PyErr_SetString(PyExc_ValueError,
- "Cannot modify a string having more than 1 reference");
+ if (_PyUnicode_Dirty(to))
return -1;
- }
- _PyUnicode_DIRTY(to);
from_kind = PyUnicode_KIND(from);
from_data = PyUnicode_DATA(from);
@@ -2855,10 +2864,15 @@
PyUnicode_WriteChar(PyObject *unicode, Py_ssize_t index, Py_UCS4 ch)
{
if (!PyUnicode_Check(unicode) || !PyUnicode_IS_COMPACT(unicode)) {
- return PyErr_BadArgument();
+ PyErr_BadArgument();
return -1;
}
-
+ if (index < 0 || index >= _PyUnicode_LENGTH(unicode)) {
+ PyErr_SetString(PyExc_IndexError, "string index out of range");
+ return -1;
+ }
+ if (_PyUnicode_Dirty(unicode))
+ return -1;
PyUnicode_WRITE(PyUnicode_KIND(unicode), PyUnicode_DATA(unicode),
index, ch);
return 0;
--
Repository URL: http://hg.python.org/cpython
More information about the Python-checkins
mailing list