mirror of
https://github.com/python/cpython.git
synced 2025-12-08 06:10:17 +00:00
[3.9] gh-80254: Disallow recursive usage of cursors in sqlite3 converters (#92278)
* [3.9] gh-80254: Disallow recursive usage of cursors in `sqlite3` converters
(cherry picked from commit c908dc5b47)
Co-authored-by: Sergey Fedoseev <fedoseev.sergey@gmail.com>
Co-authored-by: Jelle Zijlstra <jelle.zijlstra@gmail.com>
* Fix ref leak in pysqlite_cursor_iternext
* Explicitly free resources at test tearDown()
This commit is contained in:
parent
d82a769a1d
commit
7d17a7b352
3 changed files with 77 additions and 14 deletions
|
|
@ -28,6 +28,9 @@
|
|||
import functools
|
||||
from test import support
|
||||
|
||||
from unittest.mock import patch
|
||||
|
||||
|
||||
class RegressionTests(unittest.TestCase):
|
||||
def setUp(self):
|
||||
self.con = sqlite.connect(":memory:")
|
||||
|
|
@ -413,10 +416,50 @@ def log(self, *args):
|
|||
|
||||
|
||||
|
||||
class RecursiveUseOfCursors(unittest.TestCase):
|
||||
# GH-80254: sqlite3 should not segfault for recursive use of cursors.
|
||||
msg = "Recursive use of cursors not allowed"
|
||||
|
||||
def setUp(self):
|
||||
self.con = sqlite.connect(":memory:",
|
||||
detect_types=sqlite.PARSE_COLNAMES)
|
||||
self.cur = self.con.cursor()
|
||||
self.cur.execute("create table test(x foo)")
|
||||
self.cur.executemany("insert into test(x) values (?)",
|
||||
[("foo",), ("bar",)])
|
||||
|
||||
def tearDown(self):
|
||||
self.cur.close()
|
||||
self.con.close()
|
||||
del self.cur
|
||||
del self.con
|
||||
|
||||
def test_recursive_cursor_init(self):
|
||||
conv = lambda x: self.cur.__init__(self.con)
|
||||
with patch.dict(sqlite.converters, {"INIT": conv}):
|
||||
with self.assertRaisesRegex(sqlite.ProgrammingError, self.msg):
|
||||
self.cur.execute(f'select x as "x [INIT]", x from test')
|
||||
|
||||
def test_recursive_cursor_close(self):
|
||||
conv = lambda x: self.cur.close()
|
||||
with patch.dict(sqlite.converters, {"CLOSE": conv}):
|
||||
with self.assertRaisesRegex(sqlite.ProgrammingError, self.msg):
|
||||
self.cur.execute(f'select x as "x [CLOSE]", x from test')
|
||||
|
||||
def test_recursive_cursor_fetch(self):
|
||||
conv = lambda x, l=[]: self.cur.fetchone() if l else l.append(None)
|
||||
with patch.dict(sqlite.converters, {"ITER": conv}):
|
||||
self.cur.execute(f'select x as "x [ITER]", x from test')
|
||||
with self.assertRaisesRegex(sqlite.ProgrammingError, self.msg):
|
||||
self.cur.fetchall()
|
||||
|
||||
|
||||
def suite():
|
||||
regression_suite = unittest.makeSuite(RegressionTests, "Check")
|
||||
recursive_cursor = unittest.makeSuite(RecursiveUseOfCursors)
|
||||
return unittest.TestSuite((
|
||||
regression_suite,
|
||||
recursive_cursor,
|
||||
))
|
||||
|
||||
def test():
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue