Add PyUnicode_AsStringAndSize(), which is like PyUnicode_AsString() but

has an extra (optional) output parameter through which it returns the size.
Use this in a few places where I used PyUnicode_AsString() + strlen(),
and in one new place (which fixes test_pep263).
This commit is contained in:
Guido van Rossum 2007-08-29 13:53:23 +00:00
parent 9befa93b04
commit 7d1df6c9b1
4 changed files with 31 additions and 17 deletions

View File

@ -641,20 +641,25 @@ PyAPI_FUNC(PyObject*) PyUnicode_FromOrdinal(int ordinal);
PyAPI_FUNC(PyObject *) _PyUnicode_AsDefaultEncodedString( PyAPI_FUNC(PyObject *) _PyUnicode_AsDefaultEncodedString(
PyObject *, const char *); PyObject *, const char *);
/* Return a char* holding the default encoded value of the /* Return a char* holding the UTF-8 encoded value of the
Unicode object. Unicode object.
DEPRECATED: use PyUnicode_AsStringAndSize() instead.
*/ */
PyAPI_FUNC(char *) PyUnicode_AsStringAndSize(PyObject*, Py_ssize_t *);
/* Returns the UTF-8 encoding, and its size.
If the output argument is NULL, no size is stored.
*/
PyAPI_FUNC(char *) PyUnicode_AsString(PyObject*); PyAPI_FUNC(char *) PyUnicode_AsString(PyObject*);
/* Returns the UTF-8 encoding.
/* Returns the currently active default encoding. This is equivalent to PyUnicode_AsStringAndSize(x, NULL).
The default encoding is currently implemented as run-time settable
process global. This may change in future versions of the
interpreter to become a parameter which is managed on a per-thread
basis.
*/ */
PyAPI_FUNC(const char*) PyUnicode_GetDefaultEncoding(void); PyAPI_FUNC(const char*) PyUnicode_GetDefaultEncoding(void);

View File

@ -50,12 +50,11 @@ int pysqlite_statement_create(pysqlite_Statement* self, pysqlite_Connection* con
self->st = NULL; self->st = NULL;
self->in_use = 0; self->in_use = 0;
sql_cstr = PyUnicode_AsString(sql); sql_cstr = PyUnicode_AsStringAndSize(sql, &sql_cstr_len);
if (sql_cstr == NULL) { if (sql_cstr == NULL) {
rc = PYSQLITE_SQL_WRONG_TYPE; rc = PYSQLITE_SQL_WRONG_TYPE;
return rc; return rc;
} }
sql_cstr_len = strlen(sql_cstr); /* XXX */
self->in_weakreflist = NULL; self->in_weakreflist = NULL;
Py_INCREF(sql); Py_INCREF(sql);
@ -216,12 +215,11 @@ int pysqlite_statement_recompile(pysqlite_Statement* self, PyObject* params)
Py_ssize_t sql_len; Py_ssize_t sql_len;
sqlite3_stmt* new_st; sqlite3_stmt* new_st;
sql_cstr = PyUnicode_AsString(self->sql); sql_cstr = PyUnicode_AsStringAndSize(self->sql, &sql_len);
if (sql_cstr == NULL) { if (sql_cstr == NULL) {
rc = PYSQLITE_SQL_WRONG_TYPE; rc = PYSQLITE_SQL_WRONG_TYPE;
return rc; return rc;
} }
sql_len = strlen(sql_cstr); /* XXXX */
rc = sqlite3_prepare(self->db, rc = sqlite3_prepare(self->db,
sql_cstr, sql_cstr,

View File

@ -1184,16 +1184,25 @@ PyObject *_PyUnicode_AsDefaultEncodedString(PyObject *unicode,
} }
char* char*
PyUnicode_AsString(PyObject *unicode) PyUnicode_AsStringAndSize(PyObject *unicode, Py_ssize_t *psize)
{ {
PyObject *str8;
if (!PyUnicode_Check(unicode)) { if (!PyUnicode_Check(unicode)) {
PyErr_BadArgument(); PyErr_BadArgument();
return NULL; return NULL;
} }
unicode = _PyUnicode_AsDefaultEncodedString(unicode, NULL); str8 = _PyUnicode_AsDefaultEncodedString(unicode, NULL);
if (!unicode) if (str8 == NULL)
return NULL; return NULL;
return PyString_AsString(unicode); if (psize != NULL)
*psize = PyString_GET_SIZE(str8);
return PyString_AS_STRING(str8);
}
char*
PyUnicode_AsString(PyObject *unicode)
{
return PyUnicode_AsStringAndSize(unicode, NULL);
} }
Py_UNICODE *PyUnicode_AsUnicode(PyObject *unicode) Py_UNICODE *PyUnicode_AsUnicode(PyObject *unicode)
@ -8098,6 +8107,7 @@ unicode_buffer_getbuffer(PyUnicodeObject *self, PyBuffer *view, int flags)
if (flags & PyBUF_CHARACTER) { if (flags & PyBUF_CHARACTER) {
PyErr_SetString(PyExc_SystemError, "can't use str as char buffer"); PyErr_SetString(PyExc_SystemError, "can't use str as char buffer");
abort();
return -1; return -1;
} }
return PyBuffer_FillInfo(view, (void *)self->str, return PyBuffer_FillInfo(view, (void *)self->str,

View File

@ -383,7 +383,8 @@ fp_readl(char *s, int size, struct tok_state *tok)
goto error; goto error;
allocated = 1; allocated = 1;
} }
if (PyObject_AsCharBuffer(bufobj, &buf, &buflen) < 0) { buf = PyUnicode_AsStringAndSize(bufobj, &buflen);
if (buf == NULL) {
goto error; goto error;
} }
if (buflen > size) { if (buflen > size) {