Mercurial > cpython-lru_cache-weakref
diff Modules/_sqlite/statement.c @ 41669:d12811decf94
Use AsCharBuffer to get C strings out of Python strings.
| author | Guido van Rossum <guido@python.org> |
|---|---|
| date | Wed, 09 May 2007 23:37:01 +0000 |
| parents | 47f9efdc0855 |
| children | e70cdce0e8b2 |
line wrap: on
line diff
--- a/Modules/_sqlite/statement.c Wed May 09 23:36:14 2007 +0000 +++ b/Modules/_sqlite/statement.c Wed May 09 23:37:01 2007 +0000 @@ -44,30 +44,20 @@ { const char* tail; int rc; - PyObject* sql_str; - char* sql_cstr; + const char* sql_cstr; + Py_ssize_t sql_cstr_len; self->st = NULL; self->in_use = 0; - if (PyString_Check(sql)) { - sql_str = sql; - Py_INCREF(sql_str); - } else if (PyUnicode_Check(sql)) { - sql_str = PyUnicode_AsUTF8String(sql); - if (!sql_str) { - rc = PYSQLITE_SQL_WRONG_TYPE; - return rc; - } - } else { + if (PyObject_AsCharBuffer(sql, &sql_cstr, &sql_cstr_len) < 0) { rc = PYSQLITE_SQL_WRONG_TYPE; return rc; } self->in_weakreflist = NULL; - self->sql = sql_str; - - sql_cstr = PyString_AsString(sql_str); + Py_INCREF(sql); + self->sql = sql; rc = sqlite3_prepare(connection->db, sql_cstr, @@ -219,10 +209,14 @@ { const char* tail; int rc; - char* sql_cstr; + const char* sql_cstr; + Py_ssize_t sql_len; sqlite3_stmt* new_st; - sql_cstr = PyString_AsString(self->sql); + if (PyObject_AsCharBuffer(self->sql, &sql_cstr, &sql_len) < 0) { + rc = PYSQLITE_SQL_WRONG_TYPE; + return rc; + } rc = sqlite3_prepare(self->db, sql_cstr,