languages/cpython
languages/cpython
1
0
Fork 0
mirror of https://github.com/python/cpython synced 2024-09-16 02:19:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

bpo-43296: Handle sqlite3_value_blob() errors (GH-24674)

Browse source
This commit is contained in:
Erlend Egeberg Aasland 2021-04-14 23:09:11 +02:00 committed by GitHub
parent e07f4ab26a
commit 5cb601f956
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 24 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
Lib/sqlite3/test/userfunctions.py
View file

@ -276,6 +276,10 @@ def test_any_arguments(self):
val = cur.fetchone()[0]
self.assertEqual(val, 2)
def test_empty_blob(self):
cur = self.con.execute("select isblob(x'')")
self.assertTrue(cur.fetchone()[0])
# Regarding deterministic functions:
#
# Between 3.8.3 and 3.15.0, deterministic functions were only used to

3
Misc/NEWS.d/next/Library/2021-03-01-13-01-33.bpo-43296.p_gU6T.rst Normal file
View file

@ -0,0 +1,3 @@
Improve :mod:`sqlite3` error handling: ``sqlite3_value_blob()`` errors that
set ``SQLITE_NOMEM`` now raise :exc:`MemoryError`. Patch by Erlend E.
Aasland.

24
Modules/_sqlite/connection.c
View file

@ -546,7 +546,6 @@ _pysqlite_build_py_params(sqlite3_context *context, int argc,
sqlite3_value* cur_value;
PyObject* cur_py_value;
const char* val_str;
Py_ssize_t buflen;
args = PyTuple_New(argc);
if (!args) {
@ -571,19 +570,26 @@ _pysqlite_build_py_params(sqlite3_context *context, int argc,
cur_py_value = Py_NewRef(Py_None);
}
break;
case SQLITE_BLOB:
buflen = sqlite3_value_bytes(cur_value);
cur_py_value = PyBytes_FromStringAndSize(
sqlite3_value_blob(cur_value), buflen);
case SQLITE_BLOB: {
sqlite3 *db = sqlite3_context_db_handle(context);
const void *blob = sqlite3_value_blob(cur_value);
if (blob == NULL && sqlite3_errcode(db) == SQLITE_NOMEM) {
PyErr_NoMemory();
goto error;
}
Py_ssize_t size = sqlite3_value_bytes(cur_value);
cur_py_value = PyBytes_FromStringAndSize(blob, size);
break;
}
case SQLITE_NULL:
default:
cur_py_value = Py_NewRef(Py_None);
}
if (!cur_py_value) {
Py_DECREF(args);
return NULL;
goto error;
}
PyTuple_SetItem(args, i, cur_py_value);
@ -591,6 +597,10 @@ _pysqlite_build_py_params(sqlite3_context *context, int argc,
}
return args;
error:
Py_DECREF(args);
return NULL;
}
static void