`Record.get()` with invalid positional argument count segfaults
Nessuno ha ancora preso questa issue.
Valutazione
- Difficoltà
- 2/5
- Tempo stimato
- 1-3 ore
- Idoneità per principianti
- 88/100
Direzione di ricerca
Inizia in asyncpg/protocol/record/recordobj.c, in record_get(), quindi esamina tests/test_record.py::test_record_get. Compila l’estensione con il comando setup.py documentato e riproduci le chiamate con zero e tre argomenti usando l’internal record helper. Il lavoro è completato quando entrambe le chiamate non valide sollevano TypeError senza causare un arresto anomalo del processo e i test di regressione passano.
Scritto dal modello di indicizzazione a partire dal testo della issue.
Descrizione
Calling asyncpg.Record.get() with an invalid number of positional arguments can crash the Python process instead of raising TypeError.
Confirmed crashing calls:
record.get()record.get("a", 2, 3)
Valid and separately handled cases behave as expected:
record.get("a")returns the value.record.get("a", default=2)raisesTypeError: Record.get() takes no keyword arguments.
Affected Component
- File:
asyncpg/protocol/record/recordobj.c - Function:
record_get() - Method exposed as:
asyncpg.Record.get - Observed commit:
db8ecc2a38e16fb0c090aef6f5506547c2831c24
Impact
This is a native crash / process-level denial of service in the CPython extension. It is not a PostgreSQL wire-level remote issue by itself; it requires same-process Python code to call Record.get() with an invalid positional argument count. This can still matter for applications that expose generic object dispatch, plugins, scripting hooks, template helpers, or RPC-style method invocation over returned records.
Root Cause
In record_get(), the invalid positional argument-count branch sets a Python exception but continues execution:
if (nargs == 2) {
key = args[0];
defval = args[1];
} else if (nargs == 1) {
key = args[0];
} else {
PyErr_Format(PyExc_TypeError,
"Record.get() expected 1 or 2 arguments, got %zd",
nargs);
}
key is not initialized in that branch. The function then reaches:
res = record_item_by_name((ApgRecordObject *)self, key, &val);
As a result, an uninitialized PyObject *key is passed to record_item_by_name(), causing a native crash.
The release build also emits:
asyncpg/protocol/record/recordobj.c:702:11: warning: 'key' may be used uninitialized [-Wmaybe-uninitialized]
Steps to Reproduce
Build asyncpg from source:
git submodule update --init --recursive
python setup.py build_ext --inplace
Minimal repro without requiring a PostgreSQL server, using the same internal record helper used by tests/test_record.py:
PYTHONPATH=. python -u - <<'PY'
from asyncpg.protocol.protocol import _create_record as Record
r = Record({"a": 0}, (1,))
print("before")
r.get()
print("after")
PY
A three-positional-argument variant also crashes:
PYTHONPATH=. python -u - <<'PY'
from asyncpg.protocol.protocol import _create_record as Record
r = Record({"a": 0}, (1,))
print("before")
r.get("a", 2, 3)
print("after")
PY
A public API variant can be reproduced by fetching any row and then calling the invalid method form:
import asyncio
import asyncpg
async def main():
conn = await asyncpg.connect()
try:
row = await conn.fetchrow("select 1 as a")
row.get()
finally:
await conn.close()
asyncio.run(main())
Expected Result
Invalid positional argument counts should raise a Python exception, for example:
TypeError: Record.get() expected 1 or 2 arguments, got 0
and:
TypeError: Record.get() expected 1 or 2 arguments, got 3
Actual Result
On a release build, both invalid calls segfault:
before no args
Segmentation fault (core dumped)
before three args
Segmentation fault (core dumped)
Local verification exited with code 139 for both r.get() and r.get("a", 2, 3).
With ASAN, the invalid argument-count path produced:
AddressSanitizer:DEADLYSIGNAL
ERROR: AddressSanitizer: SEGV on unknown address
Suggested Fix
Return immediately after setting the argument-count error:
} else {
PyErr_Format(PyExc_TypeError,
"Record.get() expected 1 or 2 arguments, got %zd",
nargs);
return NULL;
}
It would also be useful to add regression coverage to tests/test_record.py::test_record_get:
with self.assertRaises(TypeError):
r.get()
with self.assertRaises(TypeError):
r.get("a", 2, 3)
- Lingua principale
- Python
- Stelle
- 8.1k
- Fork
- 469
- Merge medio
- 2g 20h
- PR unite (30g)
- 9
Guida per i contributori
Nessuna guida per i contributori indicizzata per questo repository
Come iniziare
- Leggi tutta la issue e poi la guida ai contributi del progetto.
- Commenta sulla issue per dire che te ne occupi tu — evita che due persone facciano lo stesso lavoro.
- Fai un fork del repository e lavora su un branch.
- Apri una pull request che faccia riferimento al numero della issue.
Altre issue di MagicStack/asyncpg
-
Difficoltà 2/5 1-3 ore Idoneità per principianti 78/100
MagicStack/asyncpg#1357 ·
-
Difficoltà 4/5 3-5 giorni Idoneità per principianti 48/100
MagicStack/asyncpg#1354 ·
-
Difficoltà 3/5 1-2 giorni Idoneità per principianti 72/100
MagicStack/asyncpg#1342 ·
-
Difficoltà 3/5 1-2 giorni Idoneità per principianti 56/100
MagicStack/asyncpg#1340 · 1 commento ·
-
Difficoltà 2/5 1-3 ore Idoneità per principianti 28/100
MagicStack/asyncpg#1337 ·
Tutte le issue di MagicStack/asyncpg
Issue simili
-
enhancement
Difficoltà 2/5 1-3 ore Idoneità per principianti 70/100
canonical/paas-charm#368 · 1 commento ·
-
Difficoltà 2/5 1-3 ore Idoneità per principianti 75/100
-
tech debt
Difficoltà 2/5 1-3 ore Idoneità per principianti 75/100
-
addition to tracking list Aperta
Difficoltà 1/5 Meno di un'ora Idoneità per principianti 90/100
StevenBlack/hosts#3256 ·
-
Difficoltà 1/5 Meno di un'ora Idoneità per principianti 90/100
qualcomm/qai-appbuilder#275 ·