mirror of
https://github.com/python/cpython.git
synced 2026-04-27 06:10:59 +00:00
gh-148973: fix segfault on mismatch between consts size and oparg in compiler (#148974)
This commit is contained in:
parent
db0ee44b93
commit
c650b51c32
5 changed files with 95 additions and 6 deletions
|
|
@ -1309,6 +1309,14 @@ get_const_value(int opcode, int oparg, PyObject *co_consts)
|
|||
PyObject *constant = NULL;
|
||||
assert(loads_const(opcode));
|
||||
if (opcode == LOAD_CONST) {
|
||||
assert(PyList_Check(co_consts));
|
||||
Py_ssize_t n = PyList_GET_SIZE(co_consts);
|
||||
if (oparg < 0 || oparg >= n) {
|
||||
PyErr_Format(PyExc_ValueError,
|
||||
"LOAD_CONST index %d is out of range for consts (len=%zd)",
|
||||
oparg, n);
|
||||
return NULL;
|
||||
}
|
||||
constant = PyList_GET_ITEM(co_consts, oparg);
|
||||
}
|
||||
if (opcode == LOAD_SMALL_INT) {
|
||||
|
|
@ -2167,6 +2175,9 @@ basicblock_optimize_load_const(PyObject *const_cache, basicblock *bb, PyObject *
|
|||
cfg_instr *inst = &bb->b_instr[i];
|
||||
if (inst->i_opcode == LOAD_CONST) {
|
||||
PyObject *constant = get_const_value(inst->i_opcode, inst->i_oparg, consts);
|
||||
if (constant == NULL) {
|
||||
return ERROR;
|
||||
}
|
||||
int res = maybe_instr_make_load_smallint(inst, constant, consts, const_cache);
|
||||
Py_DECREF(constant);
|
||||
if (res < 0) {
|
||||
|
|
@ -4064,6 +4075,10 @@ _PyCompile_OptimizeCfg(PyObject *seq, PyObject *consts, int nlocals)
|
|||
PyErr_SetString(PyExc_ValueError, "expected an instruction sequence");
|
||||
return NULL;
|
||||
}
|
||||
if (!PyList_Check(consts)) {
|
||||
PyErr_SetString(PyExc_TypeError, "consts must be a list");
|
||||
return NULL;
|
||||
}
|
||||
PyObject *const_cache = PyDict_New();
|
||||
if (const_cache == NULL) {
|
||||
return NULL;
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue