2022-11-05 08:40:43 +08:00
|
|
|
// This file is generated by Tools/cases_generator/generate_cases.py
|
2023-03-04 12:59:21 +08:00
|
|
|
// from:
|
|
|
|
// Python/bytecodes.c
|
2022-11-03 12:31:26 +08:00
|
|
|
// Do not edit!
|
|
|
|
|
|
|
|
TARGET(NOP) {
|
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(RESUME) {
|
2023-09-07 21:39:03 +08:00
|
|
|
PREDICTED(RESUME);
|
|
|
|
static_assert(0 == 0, "incorrect cache size");
|
|
|
|
TIER_ONE_ONLY
|
2023-08-17 18:16:03 +08:00
|
|
|
assert(frame == tstate->current_frame);
|
2023-06-14 20:46:37 +08:00
|
|
|
if (_PyFrame_GetCode(frame)->_co_instrumentation_version != tstate->interp->monitoring_version) {
|
|
|
|
int err = _Py_Instrument(_PyFrame_GetCode(frame), tstate->interp);
|
2023-04-12 19:04:55 +08:00
|
|
|
if (err) goto error;
|
|
|
|
next_instr--;
|
|
|
|
}
|
2023-09-07 21:39:03 +08:00
|
|
|
else {
|
|
|
|
if (oparg < 2) {
|
|
|
|
CHECK_EVAL_BREAKER();
|
|
|
|
}
|
|
|
|
next_instr[-1].op.code = RESUME_CHECK;
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
2023-04-12 19:04:55 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2023-09-07 21:39:03 +08:00
|
|
|
TARGET(RESUME_CHECK) {
|
|
|
|
#if defined(__EMSCRIPTEN__)
|
2023-09-09 00:54:45 +08:00
|
|
|
DEOPT_IF(_Py_emscripten_signal_clock == 0, RESUME);
|
|
|
|
_Py_emscripten_signal_clock -= Py_EMSCRIPTEN_SIGNAL_HANDLING;
|
2023-09-07 21:39:03 +08:00
|
|
|
#endif
|
|
|
|
/* Possibly combine these two checks */
|
|
|
|
DEOPT_IF(_PyFrame_GetCode(frame)->_co_instrumentation_version
|
|
|
|
!= tstate->interp->monitoring_version, RESUME);
|
|
|
|
DEOPT_IF(_Py_atomic_load_relaxed_int32(&tstate->interp->ceval.eval_breaker), RESUME);
|
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2023-04-12 19:04:55 +08:00
|
|
|
TARGET(INSTRUMENTED_RESUME) {
|
|
|
|
/* Possible performance enhancement:
|
|
|
|
* We need to check the eval breaker anyway, can we
|
|
|
|
* combine the instrument verison check and the eval breaker test?
|
|
|
|
*/
|
2023-06-14 20:46:37 +08:00
|
|
|
if (_PyFrame_GetCode(frame)->_co_instrumentation_version != tstate->interp->monitoring_version) {
|
|
|
|
if (_Py_Instrument(_PyFrame_GetCode(frame), tstate->interp)) {
|
2023-04-12 19:04:55 +08:00
|
|
|
goto error;
|
|
|
|
}
|
|
|
|
next_instr--;
|
|
|
|
}
|
|
|
|
else {
|
2023-07-04 04:28:27 +08:00
|
|
|
if (oparg < 2) {
|
|
|
|
CHECK_EVAL_BREAKER();
|
|
|
|
}
|
2023-04-12 19:04:55 +08:00
|
|
|
_PyFrame_SetStackPointer(frame, stack_pointer);
|
|
|
|
int err = _Py_call_instrumentation(
|
|
|
|
tstate, oparg > 0, frame, next_instr-1);
|
|
|
|
stack_pointer = _PyFrame_GetStackPointer(frame);
|
|
|
|
if (err) goto error;
|
|
|
|
if (frame->prev_instr != next_instr-1) {
|
|
|
|
/* Instrumentation has jumped */
|
|
|
|
next_instr = frame->prev_instr;
|
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
}
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(LOAD_FAST_CHECK) {
|
2022-11-09 00:22:56 +08:00
|
|
|
PyObject *value;
|
|
|
|
value = GETLOCAL(oparg);
|
|
|
|
if (value == NULL) goto unbound_local_error;
|
2022-11-03 12:31:26 +08:00
|
|
|
Py_INCREF(value);
|
2022-11-09 00:22:56 +08:00
|
|
|
STACK_GROW(1);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = value;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(LOAD_FAST) {
|
2022-11-09 00:22:56 +08:00
|
|
|
PyObject *value;
|
|
|
|
value = GETLOCAL(oparg);
|
2022-11-03 12:31:26 +08:00
|
|
|
assert(value != NULL);
|
|
|
|
Py_INCREF(value);
|
2022-11-09 00:22:56 +08:00
|
|
|
STACK_GROW(1);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = value;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2023-05-10 01:02:14 +08:00
|
|
|
TARGET(LOAD_FAST_AND_CLEAR) {
|
|
|
|
PyObject *value;
|
|
|
|
value = GETLOCAL(oparg);
|
|
|
|
// do not use SETLOCAL here, it decrefs the old value
|
|
|
|
GETLOCAL(oparg) = NULL;
|
|
|
|
STACK_GROW(1);
|
|
|
|
stack_pointer[-1] = value;
|
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2023-06-05 18:07:04 +08:00
|
|
|
TARGET(LOAD_FAST_LOAD_FAST) {
|
|
|
|
PyObject *value1;
|
|
|
|
PyObject *value2;
|
|
|
|
uint32_t oparg1 = oparg >> 4;
|
|
|
|
uint32_t oparg2 = oparg & 15;
|
|
|
|
value1 = GETLOCAL(oparg1);
|
|
|
|
value2 = GETLOCAL(oparg2);
|
|
|
|
Py_INCREF(value1);
|
|
|
|
Py_INCREF(value2);
|
|
|
|
STACK_GROW(2);
|
|
|
|
stack_pointer[-2] = value1;
|
2023-08-05 00:35:56 +08:00
|
|
|
stack_pointer[-1] = value2;
|
2023-06-05 18:07:04 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2022-11-03 12:31:26 +08:00
|
|
|
TARGET(LOAD_CONST) {
|
2022-11-09 00:22:56 +08:00
|
|
|
PyObject *value;
|
2023-06-14 04:42:03 +08:00
|
|
|
value = GETITEM(FRAME_CO_CONSTS, oparg);
|
2022-11-03 12:31:26 +08:00
|
|
|
Py_INCREF(value);
|
2022-11-09 00:22:56 +08:00
|
|
|
STACK_GROW(1);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = value;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(STORE_FAST) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *value;
|
|
|
|
value = stack_pointer[-1];
|
2022-11-03 12:31:26 +08:00
|
|
|
SETLOCAL(oparg, value);
|
2022-11-09 00:22:56 +08:00
|
|
|
STACK_SHRINK(1);
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2023-06-05 18:07:04 +08:00
|
|
|
TARGET(STORE_FAST_LOAD_FAST) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *value1;
|
2023-06-05 18:07:04 +08:00
|
|
|
PyObject *value2;
|
2023-08-05 00:35:56 +08:00
|
|
|
value1 = stack_pointer[-1];
|
2023-06-05 18:07:04 +08:00
|
|
|
uint32_t oparg1 = oparg >> 4;
|
|
|
|
uint32_t oparg2 = oparg & 15;
|
|
|
|
SETLOCAL(oparg1, value1);
|
|
|
|
value2 = GETLOCAL(oparg2);
|
|
|
|
Py_INCREF(value2);
|
|
|
|
stack_pointer[-1] = value2;
|
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(STORE_FAST_STORE_FAST) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *value1;
|
|
|
|
PyObject *value2;
|
|
|
|
value1 = stack_pointer[-1];
|
|
|
|
value2 = stack_pointer[-2];
|
2023-06-05 18:07:04 +08:00
|
|
|
uint32_t oparg1 = oparg >> 4;
|
|
|
|
uint32_t oparg2 = oparg & 15;
|
|
|
|
SETLOCAL(oparg1, value1);
|
|
|
|
SETLOCAL(oparg2, value2);
|
|
|
|
STACK_SHRINK(2);
|
2022-12-09 07:54:07 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2022-11-03 12:31:26 +08:00
|
|
|
TARGET(POP_TOP) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *value;
|
|
|
|
value = stack_pointer[-1];
|
2022-11-03 12:31:26 +08:00
|
|
|
Py_DECREF(value);
|
2022-11-09 00:22:56 +08:00
|
|
|
STACK_SHRINK(1);
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(PUSH_NULL) {
|
2022-11-11 02:50:57 +08:00
|
|
|
PyObject *res;
|
|
|
|
res = NULL;
|
|
|
|
STACK_GROW(1);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = res;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2022-12-09 07:54:07 +08:00
|
|
|
TARGET(END_FOR) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *value;
|
|
|
|
// POP_TOP
|
|
|
|
value = stack_pointer[-1];
|
2022-12-09 07:54:07 +08:00
|
|
|
{
|
|
|
|
Py_DECREF(value);
|
|
|
|
}
|
2023-08-05 00:35:56 +08:00
|
|
|
// POP_TOP
|
|
|
|
value = stack_pointer[-2];
|
2022-12-09 07:54:07 +08:00
|
|
|
{
|
|
|
|
Py_DECREF(value);
|
|
|
|
}
|
|
|
|
STACK_SHRINK(2);
|
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2023-04-12 19:04:55 +08:00
|
|
|
TARGET(INSTRUMENTED_END_FOR) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *value;
|
|
|
|
PyObject *receiver;
|
|
|
|
value = stack_pointer[-1];
|
|
|
|
receiver = stack_pointer[-2];
|
2023-04-12 19:04:55 +08:00
|
|
|
/* Need to create a fake StopIteration error here,
|
|
|
|
* to conform to PEP 380 */
|
|
|
|
if (PyGen_Check(receiver)) {
|
|
|
|
PyErr_SetObject(PyExc_StopIteration, value);
|
|
|
|
if (monitor_stop_iteration(tstate, frame, next_instr-1)) {
|
|
|
|
goto error;
|
|
|
|
}
|
|
|
|
PyErr_SetRaisedException(NULL);
|
|
|
|
}
|
|
|
|
Py_DECREF(receiver);
|
|
|
|
Py_DECREF(value);
|
|
|
|
STACK_SHRINK(2);
|
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(END_SEND) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *value;
|
|
|
|
PyObject *receiver;
|
|
|
|
value = stack_pointer[-1];
|
|
|
|
receiver = stack_pointer[-2];
|
2023-04-12 19:04:55 +08:00
|
|
|
Py_DECREF(receiver);
|
|
|
|
STACK_SHRINK(1);
|
|
|
|
stack_pointer[-1] = value;
|
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(INSTRUMENTED_END_SEND) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *value;
|
|
|
|
PyObject *receiver;
|
|
|
|
value = stack_pointer[-1];
|
|
|
|
receiver = stack_pointer[-2];
|
2023-04-12 19:04:55 +08:00
|
|
|
if (PyGen_Check(receiver) || PyCoro_CheckExact(receiver)) {
|
|
|
|
PyErr_SetObject(PyExc_StopIteration, value);
|
|
|
|
if (monitor_stop_iteration(tstate, frame, next_instr-1)) {
|
|
|
|
goto error;
|
|
|
|
}
|
|
|
|
PyErr_SetRaisedException(NULL);
|
|
|
|
}
|
|
|
|
Py_DECREF(receiver);
|
|
|
|
STACK_SHRINK(1);
|
|
|
|
stack_pointer[-1] = value;
|
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2022-11-03 12:31:26 +08:00
|
|
|
TARGET(UNARY_NEGATIVE) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *value;
|
2022-11-09 00:22:56 +08:00
|
|
|
PyObject *res;
|
2023-08-05 00:35:56 +08:00
|
|
|
value = stack_pointer[-1];
|
2022-11-09 00:22:56 +08:00
|
|
|
res = PyNumber_Negative(value);
|
2022-11-03 12:31:26 +08:00
|
|
|
Py_DECREF(value);
|
2022-11-09 00:22:56 +08:00
|
|
|
if (res == NULL) goto pop_1_error;
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = res;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(UNARY_NOT) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *value;
|
2022-11-09 00:22:56 +08:00
|
|
|
PyObject *res;
|
2023-08-05 00:35:56 +08:00
|
|
|
value = stack_pointer[-1];
|
2023-06-30 04:49:54 +08:00
|
|
|
assert(PyBool_Check(value));
|
|
|
|
res = Py_IsFalse(value) ? Py_True : Py_False;
|
|
|
|
stack_pointer[-1] = res;
|
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(TO_BOOL) {
|
|
|
|
PREDICTED(TO_BOOL);
|
|
|
|
static_assert(INLINE_CACHE_ENTRIES_TO_BOOL == 3, "incorrect cache size");
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *value;
|
2023-06-30 04:49:54 +08:00
|
|
|
PyObject *res;
|
2023-08-05 00:35:56 +08:00
|
|
|
value = stack_pointer[-1];
|
2023-06-30 04:49:54 +08:00
|
|
|
#if ENABLE_SPECIALIZATION
|
|
|
|
_PyToBoolCache *cache = (_PyToBoolCache *)next_instr;
|
|
|
|
if (ADAPTIVE_COUNTER_IS_ZERO(cache->counter)) {
|
|
|
|
next_instr--;
|
|
|
|
_Py_Specialize_ToBool(value, next_instr);
|
|
|
|
DISPATCH_SAME_OPARG();
|
|
|
|
}
|
|
|
|
STAT_INC(TO_BOOL, deferred);
|
|
|
|
DECREMENT_ADAPTIVE_COUNTER(cache->counter);
|
|
|
|
#endif /* ENABLE_SPECIALIZATION */
|
2022-11-03 12:31:26 +08:00
|
|
|
int err = PyObject_IsTrue(value);
|
|
|
|
Py_DECREF(value);
|
2022-11-09 00:22:56 +08:00
|
|
|
if (err < 0) goto pop_1_error;
|
2023-06-30 04:49:54 +08:00
|
|
|
res = err ? Py_True : Py_False;
|
|
|
|
stack_pointer[-1] = res;
|
|
|
|
next_instr += 3;
|
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(TO_BOOL_BOOL) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *value;
|
|
|
|
value = stack_pointer[-1];
|
2023-06-30 04:49:54 +08:00
|
|
|
DEOPT_IF(!PyBool_Check(value), TO_BOOL);
|
|
|
|
STAT_INC(TO_BOOL, hit);
|
|
|
|
next_instr += 3;
|
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(TO_BOOL_INT) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *value;
|
2023-06-30 04:49:54 +08:00
|
|
|
PyObject *res;
|
2023-08-05 00:35:56 +08:00
|
|
|
value = stack_pointer[-1];
|
2023-06-30 04:49:54 +08:00
|
|
|
DEOPT_IF(!PyLong_CheckExact(value), TO_BOOL);
|
|
|
|
STAT_INC(TO_BOOL, hit);
|
|
|
|
if (_PyLong_IsZero((PyLongObject *)value)) {
|
|
|
|
assert(_Py_IsImmortal(value));
|
|
|
|
res = Py_False;
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
2022-11-09 00:22:56 +08:00
|
|
|
else {
|
2023-06-30 04:49:54 +08:00
|
|
|
Py_DECREF(value);
|
|
|
|
res = Py_True;
|
|
|
|
}
|
|
|
|
stack_pointer[-1] = res;
|
|
|
|
next_instr += 3;
|
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(TO_BOOL_LIST) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *value;
|
2023-06-30 04:49:54 +08:00
|
|
|
PyObject *res;
|
2023-08-05 00:35:56 +08:00
|
|
|
value = stack_pointer[-1];
|
2023-06-30 04:49:54 +08:00
|
|
|
DEOPT_IF(!PyList_CheckExact(value), TO_BOOL);
|
|
|
|
STAT_INC(TO_BOOL, hit);
|
|
|
|
res = Py_SIZE(value) ? Py_True : Py_False;
|
|
|
|
Py_DECREF(value);
|
|
|
|
stack_pointer[-1] = res;
|
|
|
|
next_instr += 3;
|
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(TO_BOOL_NONE) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *value;
|
2023-06-30 04:49:54 +08:00
|
|
|
PyObject *res;
|
2023-08-05 00:35:56 +08:00
|
|
|
value = stack_pointer[-1];
|
2023-06-30 04:49:54 +08:00
|
|
|
// This one is a bit weird, because we expect *some* failures:
|
|
|
|
DEOPT_IF(!Py_IsNone(value), TO_BOOL);
|
|
|
|
STAT_INC(TO_BOOL, hit);
|
|
|
|
res = Py_False;
|
|
|
|
stack_pointer[-1] = res;
|
|
|
|
next_instr += 3;
|
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(TO_BOOL_STR) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *value;
|
2023-06-30 04:49:54 +08:00
|
|
|
PyObject *res;
|
2023-08-05 00:35:56 +08:00
|
|
|
value = stack_pointer[-1];
|
2023-06-30 04:49:54 +08:00
|
|
|
DEOPT_IF(!PyUnicode_CheckExact(value), TO_BOOL);
|
|
|
|
STAT_INC(TO_BOOL, hit);
|
|
|
|
if (value == &_Py_STR(empty)) {
|
|
|
|
assert(_Py_IsImmortal(value));
|
2022-11-09 00:22:56 +08:00
|
|
|
res = Py_False;
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
2023-06-30 04:49:54 +08:00
|
|
|
else {
|
|
|
|
assert(Py_SIZE(value));
|
|
|
|
Py_DECREF(value);
|
|
|
|
res = Py_True;
|
|
|
|
}
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = res;
|
2023-06-30 04:49:54 +08:00
|
|
|
next_instr += 3;
|
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(TO_BOOL_ALWAYS_TRUE) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *value;
|
2023-06-30 04:49:54 +08:00
|
|
|
PyObject *res;
|
2023-08-05 00:35:56 +08:00
|
|
|
value = stack_pointer[-1];
|
2023-06-30 04:49:54 +08:00
|
|
|
uint32_t version = read_u32(&next_instr[1].cache);
|
|
|
|
// This one is a bit weird, because we expect *some* failures:
|
|
|
|
assert(version);
|
|
|
|
DEOPT_IF(Py_TYPE(value)->tp_version_tag != version, TO_BOOL);
|
|
|
|
STAT_INC(TO_BOOL, hit);
|
|
|
|
Py_DECREF(value);
|
|
|
|
res = Py_True;
|
|
|
|
stack_pointer[-1] = res;
|
|
|
|
next_instr += 3;
|
2022-11-09 00:22:56 +08:00
|
|
|
DISPATCH();
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(UNARY_INVERT) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *value;
|
2022-11-09 00:22:56 +08:00
|
|
|
PyObject *res;
|
2023-08-05 00:35:56 +08:00
|
|
|
value = stack_pointer[-1];
|
2022-11-09 00:22:56 +08:00
|
|
|
res = PyNumber_Invert(value);
|
2022-11-03 12:31:26 +08:00
|
|
|
Py_DECREF(value);
|
2022-11-09 00:22:56 +08:00
|
|
|
if (res == NULL) goto pop_1_error;
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = res;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(BINARY_OP_MULTIPLY_INT) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *right;
|
|
|
|
PyObject *left;
|
|
|
|
PyObject *res;
|
|
|
|
// _GUARD_BOTH_INT
|
|
|
|
right = stack_pointer[-1];
|
|
|
|
left = stack_pointer[-2];
|
2023-05-31 23:09:23 +08:00
|
|
|
{
|
|
|
|
DEOPT_IF(!PyLong_CheckExact(left), BINARY_OP);
|
|
|
|
DEOPT_IF(!PyLong_CheckExact(right), BINARY_OP);
|
|
|
|
}
|
2023-08-05 00:35:56 +08:00
|
|
|
// _BINARY_OP_MULTIPLY_INT
|
2023-05-31 23:09:23 +08:00
|
|
|
{
|
|
|
|
STAT_INC(BINARY_OP, hit);
|
|
|
|
res = _PyLong_Multiply((PyLongObject *)left, (PyLongObject *)right);
|
|
|
|
_Py_DECREF_SPECIALIZED(right, (destructor)PyObject_Free);
|
|
|
|
_Py_DECREF_SPECIALIZED(left, (destructor)PyObject_Free);
|
|
|
|
if (res == NULL) goto pop_2_error;
|
|
|
|
}
|
|
|
|
STACK_SHRINK(1);
|
2023-08-05 00:35:56 +08:00
|
|
|
stack_pointer[-1] = res;
|
|
|
|
next_instr += 1;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2023-05-31 23:09:23 +08:00
|
|
|
TARGET(BINARY_OP_ADD_INT) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *right;
|
|
|
|
PyObject *left;
|
|
|
|
PyObject *res;
|
|
|
|
// _GUARD_BOTH_INT
|
|
|
|
right = stack_pointer[-1];
|
|
|
|
left = stack_pointer[-2];
|
2023-05-31 23:09:23 +08:00
|
|
|
{
|
|
|
|
DEOPT_IF(!PyLong_CheckExact(left), BINARY_OP);
|
|
|
|
DEOPT_IF(!PyLong_CheckExact(right), BINARY_OP);
|
|
|
|
}
|
2023-08-05 00:35:56 +08:00
|
|
|
// _BINARY_OP_ADD_INT
|
2023-05-31 23:09:23 +08:00
|
|
|
{
|
|
|
|
STAT_INC(BINARY_OP, hit);
|
|
|
|
res = _PyLong_Add((PyLongObject *)left, (PyLongObject *)right);
|
|
|
|
_Py_DECREF_SPECIALIZED(right, (destructor)PyObject_Free);
|
|
|
|
_Py_DECREF_SPECIALIZED(left, (destructor)PyObject_Free);
|
|
|
|
if (res == NULL) goto pop_2_error;
|
|
|
|
}
|
|
|
|
STACK_SHRINK(1);
|
2023-08-05 00:35:56 +08:00
|
|
|
stack_pointer[-1] = res;
|
|
|
|
next_instr += 1;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(BINARY_OP_SUBTRACT_INT) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *right;
|
|
|
|
PyObject *left;
|
|
|
|
PyObject *res;
|
|
|
|
// _GUARD_BOTH_INT
|
|
|
|
right = stack_pointer[-1];
|
|
|
|
left = stack_pointer[-2];
|
2023-05-31 23:09:23 +08:00
|
|
|
{
|
|
|
|
DEOPT_IF(!PyLong_CheckExact(left), BINARY_OP);
|
|
|
|
DEOPT_IF(!PyLong_CheckExact(right), BINARY_OP);
|
|
|
|
}
|
2023-08-05 00:35:56 +08:00
|
|
|
// _BINARY_OP_SUBTRACT_INT
|
2023-05-31 23:09:23 +08:00
|
|
|
{
|
|
|
|
STAT_INC(BINARY_OP, hit);
|
|
|
|
res = _PyLong_Subtract((PyLongObject *)left, (PyLongObject *)right);
|
|
|
|
_Py_DECREF_SPECIALIZED(right, (destructor)PyObject_Free);
|
|
|
|
_Py_DECREF_SPECIALIZED(left, (destructor)PyObject_Free);
|
|
|
|
if (res == NULL) goto pop_2_error;
|
|
|
|
}
|
|
|
|
STACK_SHRINK(1);
|
2023-08-05 00:35:56 +08:00
|
|
|
stack_pointer[-1] = res;
|
|
|
|
next_instr += 1;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2023-05-31 23:09:23 +08:00
|
|
|
TARGET(BINARY_OP_MULTIPLY_FLOAT) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *right;
|
|
|
|
PyObject *left;
|
|
|
|
PyObject *res;
|
|
|
|
// _GUARD_BOTH_FLOAT
|
|
|
|
right = stack_pointer[-1];
|
|
|
|
left = stack_pointer[-2];
|
2023-05-31 23:09:23 +08:00
|
|
|
{
|
|
|
|
DEOPT_IF(!PyFloat_CheckExact(left), BINARY_OP);
|
|
|
|
DEOPT_IF(!PyFloat_CheckExact(right), BINARY_OP);
|
|
|
|
}
|
2023-08-05 00:35:56 +08:00
|
|
|
// _BINARY_OP_MULTIPLY_FLOAT
|
2023-05-31 23:09:23 +08:00
|
|
|
{
|
|
|
|
STAT_INC(BINARY_OP, hit);
|
|
|
|
double dres =
|
|
|
|
((PyFloatObject *)left)->ob_fval *
|
|
|
|
((PyFloatObject *)right)->ob_fval;
|
|
|
|
DECREF_INPUTS_AND_REUSE_FLOAT(left, right, dres, res);
|
|
|
|
}
|
|
|
|
STACK_SHRINK(1);
|
2023-08-05 00:35:56 +08:00
|
|
|
stack_pointer[-1] = res;
|
|
|
|
next_instr += 1;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2023-05-31 23:09:23 +08:00
|
|
|
TARGET(BINARY_OP_ADD_FLOAT) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *right;
|
|
|
|
PyObject *left;
|
|
|
|
PyObject *res;
|
|
|
|
// _GUARD_BOTH_FLOAT
|
|
|
|
right = stack_pointer[-1];
|
|
|
|
left = stack_pointer[-2];
|
2023-05-31 23:09:23 +08:00
|
|
|
{
|
|
|
|
DEOPT_IF(!PyFloat_CheckExact(left), BINARY_OP);
|
|
|
|
DEOPT_IF(!PyFloat_CheckExact(right), BINARY_OP);
|
|
|
|
}
|
2023-08-05 00:35:56 +08:00
|
|
|
// _BINARY_OP_ADD_FLOAT
|
2023-05-31 23:09:23 +08:00
|
|
|
{
|
|
|
|
STAT_INC(BINARY_OP, hit);
|
|
|
|
double dres =
|
|
|
|
((PyFloatObject *)left)->ob_fval +
|
|
|
|
((PyFloatObject *)right)->ob_fval;
|
|
|
|
DECREF_INPUTS_AND_REUSE_FLOAT(left, right, dres, res);
|
|
|
|
}
|
|
|
|
STACK_SHRINK(1);
|
2023-08-05 00:35:56 +08:00
|
|
|
stack_pointer[-1] = res;
|
|
|
|
next_instr += 1;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2023-05-31 23:09:23 +08:00
|
|
|
TARGET(BINARY_OP_SUBTRACT_FLOAT) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *right;
|
|
|
|
PyObject *left;
|
|
|
|
PyObject *res;
|
|
|
|
// _GUARD_BOTH_FLOAT
|
|
|
|
right = stack_pointer[-1];
|
|
|
|
left = stack_pointer[-2];
|
2023-05-31 23:09:23 +08:00
|
|
|
{
|
|
|
|
DEOPT_IF(!PyFloat_CheckExact(left), BINARY_OP);
|
|
|
|
DEOPT_IF(!PyFloat_CheckExact(right), BINARY_OP);
|
|
|
|
}
|
2023-08-05 00:35:56 +08:00
|
|
|
// _BINARY_OP_SUBTRACT_FLOAT
|
2023-05-31 23:09:23 +08:00
|
|
|
{
|
|
|
|
STAT_INC(BINARY_OP, hit);
|
|
|
|
double dres =
|
|
|
|
((PyFloatObject *)left)->ob_fval -
|
|
|
|
((PyFloatObject *)right)->ob_fval;
|
|
|
|
DECREF_INPUTS_AND_REUSE_FLOAT(left, right, dres, res);
|
|
|
|
}
|
|
|
|
STACK_SHRINK(1);
|
2023-08-05 00:35:56 +08:00
|
|
|
stack_pointer[-1] = res;
|
|
|
|
next_instr += 1;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2023-05-31 23:09:23 +08:00
|
|
|
TARGET(BINARY_OP_ADD_UNICODE) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *right;
|
|
|
|
PyObject *left;
|
|
|
|
PyObject *res;
|
|
|
|
// _GUARD_BOTH_UNICODE
|
|
|
|
right = stack_pointer[-1];
|
|
|
|
left = stack_pointer[-2];
|
2023-05-31 23:09:23 +08:00
|
|
|
{
|
|
|
|
DEOPT_IF(!PyUnicode_CheckExact(left), BINARY_OP);
|
|
|
|
DEOPT_IF(!PyUnicode_CheckExact(right), BINARY_OP);
|
|
|
|
}
|
2023-08-05 00:35:56 +08:00
|
|
|
// _BINARY_OP_ADD_UNICODE
|
2023-05-31 23:09:23 +08:00
|
|
|
{
|
|
|
|
STAT_INC(BINARY_OP, hit);
|
|
|
|
res = PyUnicode_Concat(left, right);
|
|
|
|
_Py_DECREF_SPECIALIZED(left, _PyUnicode_ExactDealloc);
|
|
|
|
_Py_DECREF_SPECIALIZED(right, _PyUnicode_ExactDealloc);
|
|
|
|
if (res == NULL) goto pop_2_error;
|
|
|
|
}
|
|
|
|
STACK_SHRINK(1);
|
2023-08-05 00:35:56 +08:00
|
|
|
stack_pointer[-1] = res;
|
|
|
|
next_instr += 1;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2023-05-31 23:09:23 +08:00
|
|
|
TARGET(BINARY_OP_INPLACE_ADD_UNICODE) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *right;
|
|
|
|
PyObject *left;
|
|
|
|
// _GUARD_BOTH_UNICODE
|
|
|
|
right = stack_pointer[-1];
|
|
|
|
left = stack_pointer[-2];
|
2023-05-31 23:09:23 +08:00
|
|
|
{
|
|
|
|
DEOPT_IF(!PyUnicode_CheckExact(left), BINARY_OP);
|
|
|
|
DEOPT_IF(!PyUnicode_CheckExact(right), BINARY_OP);
|
|
|
|
}
|
2023-08-05 00:35:56 +08:00
|
|
|
// _BINARY_OP_INPLACE_ADD_UNICODE
|
2023-05-31 23:09:23 +08:00
|
|
|
{
|
|
|
|
_Py_CODEUNIT true_next = next_instr[INLINE_CACHE_ENTRIES_BINARY_OP];
|
2023-06-05 18:07:04 +08:00
|
|
|
assert(true_next.op.code == STORE_FAST);
|
2023-05-31 23:09:23 +08:00
|
|
|
PyObject **target_local = &GETLOCAL(true_next.op.arg);
|
|
|
|
DEOPT_IF(*target_local != left, BINARY_OP);
|
|
|
|
STAT_INC(BINARY_OP, hit);
|
|
|
|
/* Handle `left = left + right` or `left += right` for str.
|
|
|
|
*
|
|
|
|
* When possible, extend `left` in place rather than
|
|
|
|
* allocating a new PyUnicodeObject. This attempts to avoid
|
|
|
|
* quadratic behavior when one neglects to use str.join().
|
|
|
|
*
|
|
|
|
* If `left` has only two references remaining (one from
|
|
|
|
* the stack, one in the locals), DECREFing `left` leaves
|
|
|
|
* only the locals reference, so PyUnicode_Append knows
|
|
|
|
* that the string is safe to mutate.
|
|
|
|
*/
|
|
|
|
assert(Py_REFCNT(left) >= 2);
|
|
|
|
_Py_DECREF_NO_DEALLOC(left);
|
|
|
|
PyUnicode_Append(target_local, right);
|
|
|
|
_Py_DECREF_SPECIALIZED(right, _PyUnicode_ExactDealloc);
|
|
|
|
if (*target_local == NULL) goto pop_2_error;
|
|
|
|
// The STORE_FAST is already done.
|
2023-06-15 07:14:22 +08:00
|
|
|
SKIP_OVER(INLINE_CACHE_ENTRIES_BINARY_OP + 1);
|
2023-05-31 23:09:23 +08:00
|
|
|
}
|
|
|
|
STACK_SHRINK(2);
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(BINARY_SUBSCR) {
|
|
|
|
PREDICTED(BINARY_SUBSCR);
|
2023-03-30 06:53:30 +08:00
|
|
|
static_assert(INLINE_CACHE_ENTRIES_BINARY_SUBSCR == 1, "incorrect cache size");
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *sub;
|
|
|
|
PyObject *container;
|
2022-11-09 00:22:56 +08:00
|
|
|
PyObject *res;
|
2023-08-05 00:35:56 +08:00
|
|
|
sub = stack_pointer[-1];
|
|
|
|
container = stack_pointer[-2];
|
2023-01-20 02:14:55 +08:00
|
|
|
#if ENABLE_SPECIALIZATION
|
2022-11-10 02:50:09 +08:00
|
|
|
_PyBinarySubscrCache *cache = (_PyBinarySubscrCache *)next_instr;
|
|
|
|
if (ADAPTIVE_COUNTER_IS_ZERO(cache->counter)) {
|
|
|
|
next_instr--;
|
|
|
|
_Py_Specialize_BinarySubscr(container, sub, next_instr);
|
|
|
|
DISPATCH_SAME_OPARG();
|
|
|
|
}
|
|
|
|
STAT_INC(BINARY_SUBSCR, deferred);
|
|
|
|
DECREMENT_ADAPTIVE_COUNTER(cache->counter);
|
2023-01-20 02:14:55 +08:00
|
|
|
#endif /* ENABLE_SPECIALIZATION */
|
2022-11-09 00:22:56 +08:00
|
|
|
res = PyObject_GetItem(container, sub);
|
2022-11-03 12:31:26 +08:00
|
|
|
Py_DECREF(container);
|
|
|
|
Py_DECREF(sub);
|
2022-11-09 00:22:56 +08:00
|
|
|
if (res == NULL) goto pop_2_error;
|
|
|
|
STACK_SHRINK(1);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = res;
|
2023-03-30 06:53:30 +08:00
|
|
|
next_instr += 1;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(BINARY_SLICE) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *stop;
|
|
|
|
PyObject *start;
|
|
|
|
PyObject *container;
|
2022-11-09 00:22:56 +08:00
|
|
|
PyObject *res;
|
2023-08-05 00:35:56 +08:00
|
|
|
stop = stack_pointer[-1];
|
|
|
|
start = stack_pointer[-2];
|
|
|
|
container = stack_pointer[-3];
|
2022-11-03 12:31:26 +08:00
|
|
|
PyObject *slice = _PyBuildSlice_ConsumeRefs(start, stop);
|
2022-11-09 00:22:56 +08:00
|
|
|
// Can't use ERROR_IF() here, because we haven't
|
|
|
|
// DECREF'ed container yet, and we still own slice.
|
2022-11-03 12:31:26 +08:00
|
|
|
if (slice == NULL) {
|
2022-11-09 00:22:56 +08:00
|
|
|
res = NULL;
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
2022-11-09 00:22:56 +08:00
|
|
|
else {
|
|
|
|
res = PyObject_GetItem(container, slice);
|
|
|
|
Py_DECREF(slice);
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
|
|
|
Py_DECREF(container);
|
2022-11-09 00:22:56 +08:00
|
|
|
if (res == NULL) goto pop_3_error;
|
|
|
|
STACK_SHRINK(2);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = res;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(STORE_SLICE) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *stop;
|
|
|
|
PyObject *start;
|
|
|
|
PyObject *container;
|
|
|
|
PyObject *v;
|
|
|
|
stop = stack_pointer[-1];
|
|
|
|
start = stack_pointer[-2];
|
|
|
|
container = stack_pointer[-3];
|
|
|
|
v = stack_pointer[-4];
|
2022-11-03 12:31:26 +08:00
|
|
|
PyObject *slice = _PyBuildSlice_ConsumeRefs(start, stop);
|
2022-11-09 00:22:56 +08:00
|
|
|
int err;
|
2022-11-03 12:31:26 +08:00
|
|
|
if (slice == NULL) {
|
2022-11-09 00:22:56 +08:00
|
|
|
err = 1;
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
2022-11-09 00:22:56 +08:00
|
|
|
else {
|
|
|
|
err = PyObject_SetItem(container, slice, v);
|
|
|
|
Py_DECREF(slice);
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
|
|
|
Py_DECREF(v);
|
|
|
|
Py_DECREF(container);
|
2022-11-09 00:22:56 +08:00
|
|
|
if (err) goto pop_4_error;
|
|
|
|
STACK_SHRINK(4);
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(BINARY_SUBSCR_LIST_INT) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *sub;
|
|
|
|
PyObject *list;
|
2022-11-18 09:06:07 +08:00
|
|
|
PyObject *res;
|
2023-08-05 00:35:56 +08:00
|
|
|
sub = stack_pointer[-1];
|
|
|
|
list = stack_pointer[-2];
|
2022-11-03 12:31:26 +08:00
|
|
|
DEOPT_IF(!PyLong_CheckExact(sub), BINARY_SUBSCR);
|
|
|
|
DEOPT_IF(!PyList_CheckExact(list), BINARY_SUBSCR);
|
|
|
|
|
|
|
|
// Deopt unless 0 <= sub < PyList_Size(list)
|
2023-03-22 22:49:51 +08:00
|
|
|
DEOPT_IF(!_PyLong_IsNonNegativeCompact((PyLongObject *)sub), BINARY_SUBSCR);
|
2023-01-30 18:03:04 +08:00
|
|
|
Py_ssize_t index = ((PyLongObject*)sub)->long_value.ob_digit[0];
|
2022-11-03 12:31:26 +08:00
|
|
|
DEOPT_IF(index >= PyList_GET_SIZE(list), BINARY_SUBSCR);
|
|
|
|
STAT_INC(BINARY_SUBSCR, hit);
|
2022-11-18 09:06:07 +08:00
|
|
|
res = PyList_GET_ITEM(list, index);
|
2022-11-03 12:31:26 +08:00
|
|
|
assert(res != NULL);
|
|
|
|
Py_INCREF(res);
|
|
|
|
_Py_DECREF_SPECIALIZED(sub, (destructor)PyObject_Free);
|
|
|
|
Py_DECREF(list);
|
2022-11-18 09:06:07 +08:00
|
|
|
STACK_SHRINK(1);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = res;
|
2023-03-30 06:53:30 +08:00
|
|
|
next_instr += 1;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2023-08-09 04:42:43 +08:00
|
|
|
TARGET(BINARY_SUBSCR_STR_INT) {
|
|
|
|
PyObject *sub;
|
|
|
|
PyObject *str;
|
|
|
|
PyObject *res;
|
|
|
|
sub = stack_pointer[-1];
|
|
|
|
str = stack_pointer[-2];
|
|
|
|
DEOPT_IF(!PyLong_CheckExact(sub), BINARY_SUBSCR);
|
|
|
|
DEOPT_IF(!PyUnicode_CheckExact(str), BINARY_SUBSCR);
|
|
|
|
DEOPT_IF(!_PyLong_IsNonNegativeCompact((PyLongObject *)sub), BINARY_SUBSCR);
|
|
|
|
Py_ssize_t index = ((PyLongObject*)sub)->long_value.ob_digit[0];
|
|
|
|
DEOPT_IF(PyUnicode_GET_LENGTH(str) <= index, BINARY_SUBSCR);
|
|
|
|
// Specialize for reading an ASCII character from any string:
|
|
|
|
Py_UCS4 c = PyUnicode_READ_CHAR(str, index);
|
|
|
|
DEOPT_IF(Py_ARRAY_LENGTH(_Py_SINGLETON(strings).ascii) <= c, BINARY_SUBSCR);
|
|
|
|
STAT_INC(BINARY_SUBSCR, hit);
|
|
|
|
res = (PyObject*)&_Py_SINGLETON(strings).ascii[c];
|
|
|
|
_Py_DECREF_SPECIALIZED(sub, (destructor)PyObject_Free);
|
|
|
|
Py_DECREF(str);
|
|
|
|
STACK_SHRINK(1);
|
|
|
|
stack_pointer[-1] = res;
|
|
|
|
next_instr += 1;
|
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2022-11-03 12:31:26 +08:00
|
|
|
TARGET(BINARY_SUBSCR_TUPLE_INT) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *sub;
|
|
|
|
PyObject *tuple;
|
2022-11-18 09:06:07 +08:00
|
|
|
PyObject *res;
|
2023-08-05 00:35:56 +08:00
|
|
|
sub = stack_pointer[-1];
|
|
|
|
tuple = stack_pointer[-2];
|
2022-11-03 12:31:26 +08:00
|
|
|
DEOPT_IF(!PyLong_CheckExact(sub), BINARY_SUBSCR);
|
|
|
|
DEOPT_IF(!PyTuple_CheckExact(tuple), BINARY_SUBSCR);
|
|
|
|
|
|
|
|
// Deopt unless 0 <= sub < PyTuple_Size(list)
|
2023-03-22 22:49:51 +08:00
|
|
|
DEOPT_IF(!_PyLong_IsNonNegativeCompact((PyLongObject *)sub), BINARY_SUBSCR);
|
2023-01-30 18:03:04 +08:00
|
|
|
Py_ssize_t index = ((PyLongObject*)sub)->long_value.ob_digit[0];
|
2022-11-03 12:31:26 +08:00
|
|
|
DEOPT_IF(index >= PyTuple_GET_SIZE(tuple), BINARY_SUBSCR);
|
|
|
|
STAT_INC(BINARY_SUBSCR, hit);
|
2022-11-18 09:06:07 +08:00
|
|
|
res = PyTuple_GET_ITEM(tuple, index);
|
2022-11-03 12:31:26 +08:00
|
|
|
assert(res != NULL);
|
|
|
|
Py_INCREF(res);
|
|
|
|
_Py_DECREF_SPECIALIZED(sub, (destructor)PyObject_Free);
|
|
|
|
Py_DECREF(tuple);
|
2022-11-18 09:06:07 +08:00
|
|
|
STACK_SHRINK(1);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = res;
|
2023-03-30 06:53:30 +08:00
|
|
|
next_instr += 1;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(BINARY_SUBSCR_DICT) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *sub;
|
|
|
|
PyObject *dict;
|
2022-11-18 09:06:07 +08:00
|
|
|
PyObject *res;
|
2023-08-05 00:35:56 +08:00
|
|
|
sub = stack_pointer[-1];
|
|
|
|
dict = stack_pointer[-2];
|
2022-11-18 09:06:07 +08:00
|
|
|
DEOPT_IF(!PyDict_CheckExact(dict), BINARY_SUBSCR);
|
2022-11-03 12:31:26 +08:00
|
|
|
STAT_INC(BINARY_SUBSCR, hit);
|
2022-11-18 09:06:07 +08:00
|
|
|
res = PyDict_GetItemWithError(dict, sub);
|
2022-11-03 12:31:26 +08:00
|
|
|
if (res == NULL) {
|
|
|
|
if (!_PyErr_Occurred(tstate)) {
|
|
|
|
_PyErr_SetKeyError(sub);
|
|
|
|
}
|
2022-11-18 09:06:07 +08:00
|
|
|
Py_DECREF(dict);
|
|
|
|
Py_DECREF(sub);
|
GH-98831: Typed stack effects, and more instructions converted (#99764)
Stack effects can now have a type, e.g. `inst(X, (left, right -- jump/uint64_t)) { ... }`.
Instructions converted to the non-legacy format:
* COMPARE_OP
* COMPARE_OP_FLOAT_JUMP
* COMPARE_OP_INT_JUMP
* COMPARE_OP_STR_JUMP
* STORE_ATTR
* DELETE_ATTR
* STORE_GLOBAL
* STORE_ATTR_INSTANCE_VALUE
* STORE_ATTR_WITH_HINT
* STORE_ATTR_SLOT, and complete the store_attr family
* Complete the store_subscr family: STORE_SUBSCR{,DICT,LIST_INT}
(STORE_SUBSCR was alread half converted,
but wasn't using cache effects yet.)
* DELETE_SUBSCR
* PRINT_EXPR
* INTERPRETER_EXIT (a bit weird, ends in return)
* RETURN_VALUE
* GET_AITER (had to restructure it some)
The original had mysterious `SET_TOP(NULL)` before `goto error`.
I assume those just account for `obj` having been decref'ed,
so I got rid of them in favor of the cleanup implied by `ERROR_IF()`.
* LIST_APPEND (a bit unhappy with it)
* SET_ADD (also a bit unhappy with it)
Various other improvements/refactorings as well.
2022-12-09 05:31:27 +08:00
|
|
|
if (true) goto pop_2_error;
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
2022-11-18 09:06:07 +08:00
|
|
|
Py_INCREF(res); // Do this before DECREF'ing dict, sub
|
2022-11-03 12:31:26 +08:00
|
|
|
Py_DECREF(dict);
|
2022-11-18 09:06:07 +08:00
|
|
|
Py_DECREF(sub);
|
|
|
|
STACK_SHRINK(1);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = res;
|
2023-03-30 06:53:30 +08:00
|
|
|
next_instr += 1;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(BINARY_SUBSCR_GETITEM) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *sub;
|
|
|
|
PyObject *container;
|
|
|
|
sub = stack_pointer[-1];
|
|
|
|
container = stack_pointer[-2];
|
2023-05-13 06:23:13 +08:00
|
|
|
DEOPT_IF(tstate->interp->eval_frame, BINARY_SUBSCR);
|
2022-11-03 12:31:26 +08:00
|
|
|
PyTypeObject *tp = Py_TYPE(container);
|
2023-03-30 06:53:30 +08:00
|
|
|
DEOPT_IF(!PyType_HasFeature(tp, Py_TPFLAGS_HEAPTYPE), BINARY_SUBSCR);
|
|
|
|
PyHeapTypeObject *ht = (PyHeapTypeObject *)tp;
|
|
|
|
PyObject *cached = ht->_spec_cache.getitem;
|
|
|
|
DEOPT_IF(cached == NULL, BINARY_SUBSCR);
|
2022-11-03 12:31:26 +08:00
|
|
|
assert(PyFunction_Check(cached));
|
|
|
|
PyFunctionObject *getitem = (PyFunctionObject *)cached;
|
2023-03-30 06:53:30 +08:00
|
|
|
uint32_t cached_version = ht->_spec_cache.getitem_version;
|
|
|
|
DEOPT_IF(getitem->func_version != cached_version, BINARY_SUBSCR);
|
2022-11-03 12:31:26 +08:00
|
|
|
PyCodeObject *code = (PyCodeObject *)getitem->func_code;
|
|
|
|
assert(code->co_argcount == 2);
|
|
|
|
DEOPT_IF(!_PyThreadState_HasStackSpace(tstate, code->co_framesize), BINARY_SUBSCR);
|
|
|
|
STAT_INC(BINARY_SUBSCR, hit);
|
|
|
|
Py_INCREF(getitem);
|
2023-01-06 22:55:56 +08:00
|
|
|
_PyInterpreterFrame *new_frame = _PyFrame_PushUnchecked(tstate, getitem, 2);
|
2022-11-03 12:31:26 +08:00
|
|
|
STACK_SHRINK(2);
|
|
|
|
new_frame->localsplus[0] = container;
|
|
|
|
new_frame->localsplus[1] = sub;
|
2023-06-15 07:14:22 +08:00
|
|
|
SKIP_OVER(INLINE_CACHE_ENTRIES_BINARY_SUBSCR);
|
2023-04-13 23:19:07 +08:00
|
|
|
frame->return_offset = 0;
|
2022-11-18 03:36:03 +08:00
|
|
|
DISPATCH_INLINED(new_frame);
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(LIST_APPEND) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *v;
|
|
|
|
PyObject *list;
|
|
|
|
v = stack_pointer[-1];
|
|
|
|
list = stack_pointer[-2 - (oparg-1)];
|
GH-98831: Typed stack effects, and more instructions converted (#99764)
Stack effects can now have a type, e.g. `inst(X, (left, right -- jump/uint64_t)) { ... }`.
Instructions converted to the non-legacy format:
* COMPARE_OP
* COMPARE_OP_FLOAT_JUMP
* COMPARE_OP_INT_JUMP
* COMPARE_OP_STR_JUMP
* STORE_ATTR
* DELETE_ATTR
* STORE_GLOBAL
* STORE_ATTR_INSTANCE_VALUE
* STORE_ATTR_WITH_HINT
* STORE_ATTR_SLOT, and complete the store_attr family
* Complete the store_subscr family: STORE_SUBSCR{,DICT,LIST_INT}
(STORE_SUBSCR was alread half converted,
but wasn't using cache effects yet.)
* DELETE_SUBSCR
* PRINT_EXPR
* INTERPRETER_EXIT (a bit weird, ends in return)
* RETURN_VALUE
* GET_AITER (had to restructure it some)
The original had mysterious `SET_TOP(NULL)` before `goto error`.
I assume those just account for `obj` having been decref'ed,
so I got rid of them in favor of the cleanup implied by `ERROR_IF()`.
* LIST_APPEND (a bit unhappy with it)
* SET_ADD (also a bit unhappy with it)
Various other improvements/refactorings as well.
2022-12-09 05:31:27 +08:00
|
|
|
if (_PyList_AppendTakeRef((PyListObject *)list, v) < 0) goto pop_1_error;
|
|
|
|
STACK_SHRINK(1);
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(SET_ADD) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *v;
|
|
|
|
PyObject *set;
|
|
|
|
v = stack_pointer[-1];
|
|
|
|
set = stack_pointer[-2 - (oparg-1)];
|
GH-98831: Typed stack effects, and more instructions converted (#99764)
Stack effects can now have a type, e.g. `inst(X, (left, right -- jump/uint64_t)) { ... }`.
Instructions converted to the non-legacy format:
* COMPARE_OP
* COMPARE_OP_FLOAT_JUMP
* COMPARE_OP_INT_JUMP
* COMPARE_OP_STR_JUMP
* STORE_ATTR
* DELETE_ATTR
* STORE_GLOBAL
* STORE_ATTR_INSTANCE_VALUE
* STORE_ATTR_WITH_HINT
* STORE_ATTR_SLOT, and complete the store_attr family
* Complete the store_subscr family: STORE_SUBSCR{,DICT,LIST_INT}
(STORE_SUBSCR was alread half converted,
but wasn't using cache effects yet.)
* DELETE_SUBSCR
* PRINT_EXPR
* INTERPRETER_EXIT (a bit weird, ends in return)
* RETURN_VALUE
* GET_AITER (had to restructure it some)
The original had mysterious `SET_TOP(NULL)` before `goto error`.
I assume those just account for `obj` having been decref'ed,
so I got rid of them in favor of the cleanup implied by `ERROR_IF()`.
* LIST_APPEND (a bit unhappy with it)
* SET_ADD (also a bit unhappy with it)
Various other improvements/refactorings as well.
2022-12-09 05:31:27 +08:00
|
|
|
int err = PySet_Add(set, v);
|
2022-11-03 12:31:26 +08:00
|
|
|
Py_DECREF(v);
|
GH-98831: Typed stack effects, and more instructions converted (#99764)
Stack effects can now have a type, e.g. `inst(X, (left, right -- jump/uint64_t)) { ... }`.
Instructions converted to the non-legacy format:
* COMPARE_OP
* COMPARE_OP_FLOAT_JUMP
* COMPARE_OP_INT_JUMP
* COMPARE_OP_STR_JUMP
* STORE_ATTR
* DELETE_ATTR
* STORE_GLOBAL
* STORE_ATTR_INSTANCE_VALUE
* STORE_ATTR_WITH_HINT
* STORE_ATTR_SLOT, and complete the store_attr family
* Complete the store_subscr family: STORE_SUBSCR{,DICT,LIST_INT}
(STORE_SUBSCR was alread half converted,
but wasn't using cache effects yet.)
* DELETE_SUBSCR
* PRINT_EXPR
* INTERPRETER_EXIT (a bit weird, ends in return)
* RETURN_VALUE
* GET_AITER (had to restructure it some)
The original had mysterious `SET_TOP(NULL)` before `goto error`.
I assume those just account for `obj` having been decref'ed,
so I got rid of them in favor of the cleanup implied by `ERROR_IF()`.
* LIST_APPEND (a bit unhappy with it)
* SET_ADD (also a bit unhappy with it)
Various other improvements/refactorings as well.
2022-12-09 05:31:27 +08:00
|
|
|
if (err) goto pop_1_error;
|
|
|
|
STACK_SHRINK(1);
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(STORE_SUBSCR) {
|
|
|
|
PREDICTED(STORE_SUBSCR);
|
2023-02-02 02:56:52 +08:00
|
|
|
static_assert(INLINE_CACHE_ENTRIES_STORE_SUBSCR == 1, "incorrect cache size");
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *sub;
|
|
|
|
PyObject *container;
|
|
|
|
PyObject *v;
|
|
|
|
sub = stack_pointer[-1];
|
|
|
|
container = stack_pointer[-2];
|
|
|
|
v = stack_pointer[-3];
|
2023-01-20 02:14:55 +08:00
|
|
|
#if ENABLE_SPECIALIZATION
|
2023-07-18 03:12:33 +08:00
|
|
|
_PyStoreSubscrCache *cache = (_PyStoreSubscrCache *)next_instr;
|
|
|
|
if (ADAPTIVE_COUNTER_IS_ZERO(cache->counter)) {
|
2022-11-10 02:50:09 +08:00
|
|
|
next_instr--;
|
|
|
|
_Py_Specialize_StoreSubscr(container, sub, next_instr);
|
|
|
|
DISPATCH_SAME_OPARG();
|
|
|
|
}
|
|
|
|
STAT_INC(STORE_SUBSCR, deferred);
|
|
|
|
DECREMENT_ADAPTIVE_COUNTER(cache->counter);
|
2023-01-20 02:14:55 +08:00
|
|
|
#endif /* ENABLE_SPECIALIZATION */
|
2022-11-03 12:31:26 +08:00
|
|
|
/* container[sub] = v */
|
2022-11-10 02:50:09 +08:00
|
|
|
int err = PyObject_SetItem(container, sub, v);
|
2022-11-03 12:31:26 +08:00
|
|
|
Py_DECREF(v);
|
|
|
|
Py_DECREF(container);
|
|
|
|
Py_DECREF(sub);
|
GH-98831: Typed stack effects, and more instructions converted (#99764)
Stack effects can now have a type, e.g. `inst(X, (left, right -- jump/uint64_t)) { ... }`.
Instructions converted to the non-legacy format:
* COMPARE_OP
* COMPARE_OP_FLOAT_JUMP
* COMPARE_OP_INT_JUMP
* COMPARE_OP_STR_JUMP
* STORE_ATTR
* DELETE_ATTR
* STORE_GLOBAL
* STORE_ATTR_INSTANCE_VALUE
* STORE_ATTR_WITH_HINT
* STORE_ATTR_SLOT, and complete the store_attr family
* Complete the store_subscr family: STORE_SUBSCR{,DICT,LIST_INT}
(STORE_SUBSCR was alread half converted,
but wasn't using cache effects yet.)
* DELETE_SUBSCR
* PRINT_EXPR
* INTERPRETER_EXIT (a bit weird, ends in return)
* RETURN_VALUE
* GET_AITER (had to restructure it some)
The original had mysterious `SET_TOP(NULL)` before `goto error`.
I assume those just account for `obj` having been decref'ed,
so I got rid of them in favor of the cleanup implied by `ERROR_IF()`.
* LIST_APPEND (a bit unhappy with it)
* SET_ADD (also a bit unhappy with it)
Various other improvements/refactorings as well.
2022-12-09 05:31:27 +08:00
|
|
|
if (err) goto pop_3_error;
|
2022-11-09 00:22:56 +08:00
|
|
|
STACK_SHRINK(3);
|
2023-03-01 00:49:35 +08:00
|
|
|
next_instr += 1;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(STORE_SUBSCR_LIST_INT) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *sub;
|
|
|
|
PyObject *list;
|
|
|
|
PyObject *value;
|
|
|
|
sub = stack_pointer[-1];
|
|
|
|
list = stack_pointer[-2];
|
|
|
|
value = stack_pointer[-3];
|
2022-11-03 12:31:26 +08:00
|
|
|
DEOPT_IF(!PyLong_CheckExact(sub), STORE_SUBSCR);
|
|
|
|
DEOPT_IF(!PyList_CheckExact(list), STORE_SUBSCR);
|
|
|
|
|
|
|
|
// Ensure nonnegative, zero-or-one-digit ints.
|
2023-03-22 22:49:51 +08:00
|
|
|
DEOPT_IF(!_PyLong_IsNonNegativeCompact((PyLongObject *)sub), STORE_SUBSCR);
|
2023-01-30 18:03:04 +08:00
|
|
|
Py_ssize_t index = ((PyLongObject*)sub)->long_value.ob_digit[0];
|
2022-11-03 12:31:26 +08:00
|
|
|
// Ensure index < len(list)
|
|
|
|
DEOPT_IF(index >= PyList_GET_SIZE(list), STORE_SUBSCR);
|
|
|
|
STAT_INC(STORE_SUBSCR, hit);
|
|
|
|
|
|
|
|
PyObject *old_value = PyList_GET_ITEM(list, index);
|
|
|
|
PyList_SET_ITEM(list, index, value);
|
|
|
|
assert(old_value != NULL);
|
|
|
|
Py_DECREF(old_value);
|
|
|
|
_Py_DECREF_SPECIALIZED(sub, (destructor)PyObject_Free);
|
|
|
|
Py_DECREF(list);
|
GH-98831: Typed stack effects, and more instructions converted (#99764)
Stack effects can now have a type, e.g. `inst(X, (left, right -- jump/uint64_t)) { ... }`.
Instructions converted to the non-legacy format:
* COMPARE_OP
* COMPARE_OP_FLOAT_JUMP
* COMPARE_OP_INT_JUMP
* COMPARE_OP_STR_JUMP
* STORE_ATTR
* DELETE_ATTR
* STORE_GLOBAL
* STORE_ATTR_INSTANCE_VALUE
* STORE_ATTR_WITH_HINT
* STORE_ATTR_SLOT, and complete the store_attr family
* Complete the store_subscr family: STORE_SUBSCR{,DICT,LIST_INT}
(STORE_SUBSCR was alread half converted,
but wasn't using cache effects yet.)
* DELETE_SUBSCR
* PRINT_EXPR
* INTERPRETER_EXIT (a bit weird, ends in return)
* RETURN_VALUE
* GET_AITER (had to restructure it some)
The original had mysterious `SET_TOP(NULL)` before `goto error`.
I assume those just account for `obj` having been decref'ed,
so I got rid of them in favor of the cleanup implied by `ERROR_IF()`.
* LIST_APPEND (a bit unhappy with it)
* SET_ADD (also a bit unhappy with it)
Various other improvements/refactorings as well.
2022-12-09 05:31:27 +08:00
|
|
|
STACK_SHRINK(3);
|
2023-03-01 00:49:35 +08:00
|
|
|
next_instr += 1;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(STORE_SUBSCR_DICT) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *sub;
|
|
|
|
PyObject *dict;
|
|
|
|
PyObject *value;
|
|
|
|
sub = stack_pointer[-1];
|
|
|
|
dict = stack_pointer[-2];
|
|
|
|
value = stack_pointer[-3];
|
2022-11-03 12:31:26 +08:00
|
|
|
DEOPT_IF(!PyDict_CheckExact(dict), STORE_SUBSCR);
|
|
|
|
STAT_INC(STORE_SUBSCR, hit);
|
|
|
|
int err = _PyDict_SetItem_Take2((PyDictObject *)dict, sub, value);
|
|
|
|
Py_DECREF(dict);
|
GH-98831: Typed stack effects, and more instructions converted (#99764)
Stack effects can now have a type, e.g. `inst(X, (left, right -- jump/uint64_t)) { ... }`.
Instructions converted to the non-legacy format:
* COMPARE_OP
* COMPARE_OP_FLOAT_JUMP
* COMPARE_OP_INT_JUMP
* COMPARE_OP_STR_JUMP
* STORE_ATTR
* DELETE_ATTR
* STORE_GLOBAL
* STORE_ATTR_INSTANCE_VALUE
* STORE_ATTR_WITH_HINT
* STORE_ATTR_SLOT, and complete the store_attr family
* Complete the store_subscr family: STORE_SUBSCR{,DICT,LIST_INT}
(STORE_SUBSCR was alread half converted,
but wasn't using cache effects yet.)
* DELETE_SUBSCR
* PRINT_EXPR
* INTERPRETER_EXIT (a bit weird, ends in return)
* RETURN_VALUE
* GET_AITER (had to restructure it some)
The original had mysterious `SET_TOP(NULL)` before `goto error`.
I assume those just account for `obj` having been decref'ed,
so I got rid of them in favor of the cleanup implied by `ERROR_IF()`.
* LIST_APPEND (a bit unhappy with it)
* SET_ADD (also a bit unhappy with it)
Various other improvements/refactorings as well.
2022-12-09 05:31:27 +08:00
|
|
|
if (err) goto pop_3_error;
|
|
|
|
STACK_SHRINK(3);
|
2023-03-01 00:49:35 +08:00
|
|
|
next_instr += 1;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(DELETE_SUBSCR) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *sub;
|
|
|
|
PyObject *container;
|
|
|
|
sub = stack_pointer[-1];
|
|
|
|
container = stack_pointer[-2];
|
2022-11-03 12:31:26 +08:00
|
|
|
/* del container[sub] */
|
GH-98831: Typed stack effects, and more instructions converted (#99764)
Stack effects can now have a type, e.g. `inst(X, (left, right -- jump/uint64_t)) { ... }`.
Instructions converted to the non-legacy format:
* COMPARE_OP
* COMPARE_OP_FLOAT_JUMP
* COMPARE_OP_INT_JUMP
* COMPARE_OP_STR_JUMP
* STORE_ATTR
* DELETE_ATTR
* STORE_GLOBAL
* STORE_ATTR_INSTANCE_VALUE
* STORE_ATTR_WITH_HINT
* STORE_ATTR_SLOT, and complete the store_attr family
* Complete the store_subscr family: STORE_SUBSCR{,DICT,LIST_INT}
(STORE_SUBSCR was alread half converted,
but wasn't using cache effects yet.)
* DELETE_SUBSCR
* PRINT_EXPR
* INTERPRETER_EXIT (a bit weird, ends in return)
* RETURN_VALUE
* GET_AITER (had to restructure it some)
The original had mysterious `SET_TOP(NULL)` before `goto error`.
I assume those just account for `obj` having been decref'ed,
so I got rid of them in favor of the cleanup implied by `ERROR_IF()`.
* LIST_APPEND (a bit unhappy with it)
* SET_ADD (also a bit unhappy with it)
Various other improvements/refactorings as well.
2022-12-09 05:31:27 +08:00
|
|
|
int err = PyObject_DelItem(container, sub);
|
2022-11-03 12:31:26 +08:00
|
|
|
Py_DECREF(container);
|
|
|
|
Py_DECREF(sub);
|
GH-98831: Typed stack effects, and more instructions converted (#99764)
Stack effects can now have a type, e.g. `inst(X, (left, right -- jump/uint64_t)) { ... }`.
Instructions converted to the non-legacy format:
* COMPARE_OP
* COMPARE_OP_FLOAT_JUMP
* COMPARE_OP_INT_JUMP
* COMPARE_OP_STR_JUMP
* STORE_ATTR
* DELETE_ATTR
* STORE_GLOBAL
* STORE_ATTR_INSTANCE_VALUE
* STORE_ATTR_WITH_HINT
* STORE_ATTR_SLOT, and complete the store_attr family
* Complete the store_subscr family: STORE_SUBSCR{,DICT,LIST_INT}
(STORE_SUBSCR was alread half converted,
but wasn't using cache effects yet.)
* DELETE_SUBSCR
* PRINT_EXPR
* INTERPRETER_EXIT (a bit weird, ends in return)
* RETURN_VALUE
* GET_AITER (had to restructure it some)
The original had mysterious `SET_TOP(NULL)` before `goto error`.
I assume those just account for `obj` having been decref'ed,
so I got rid of them in favor of the cleanup implied by `ERROR_IF()`.
* LIST_APPEND (a bit unhappy with it)
* SET_ADD (also a bit unhappy with it)
Various other improvements/refactorings as well.
2022-12-09 05:31:27 +08:00
|
|
|
if (err) goto pop_2_error;
|
|
|
|
STACK_SHRINK(2);
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2023-01-06 00:05:51 +08:00
|
|
|
TARGET(CALL_INTRINSIC_1) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *value;
|
2022-11-03 12:31:26 +08:00
|
|
|
PyObject *res;
|
2023-08-05 00:35:56 +08:00
|
|
|
value = stack_pointer[-1];
|
2023-01-06 00:05:51 +08:00
|
|
|
assert(oparg <= MAX_INTRINSIC_1);
|
2023-07-21 00:46:04 +08:00
|
|
|
res = _PyIntrinsics_UnaryFunctions[oparg].func(tstate, value);
|
2022-11-03 12:31:26 +08:00
|
|
|
Py_DECREF(value);
|
GH-98831: Typed stack effects, and more instructions converted (#99764)
Stack effects can now have a type, e.g. `inst(X, (left, right -- jump/uint64_t)) { ... }`.
Instructions converted to the non-legacy format:
* COMPARE_OP
* COMPARE_OP_FLOAT_JUMP
* COMPARE_OP_INT_JUMP
* COMPARE_OP_STR_JUMP
* STORE_ATTR
* DELETE_ATTR
* STORE_GLOBAL
* STORE_ATTR_INSTANCE_VALUE
* STORE_ATTR_WITH_HINT
* STORE_ATTR_SLOT, and complete the store_attr family
* Complete the store_subscr family: STORE_SUBSCR{,DICT,LIST_INT}
(STORE_SUBSCR was alread half converted,
but wasn't using cache effects yet.)
* DELETE_SUBSCR
* PRINT_EXPR
* INTERPRETER_EXIT (a bit weird, ends in return)
* RETURN_VALUE
* GET_AITER (had to restructure it some)
The original had mysterious `SET_TOP(NULL)` before `goto error`.
I assume those just account for `obj` having been decref'ed,
so I got rid of them in favor of the cleanup implied by `ERROR_IF()`.
* LIST_APPEND (a bit unhappy with it)
* SET_ADD (also a bit unhappy with it)
Various other improvements/refactorings as well.
2022-12-09 05:31:27 +08:00
|
|
|
if (res == NULL) goto pop_1_error;
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = res;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2023-02-14 19:54:13 +08:00
|
|
|
TARGET(CALL_INTRINSIC_2) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *value1;
|
|
|
|
PyObject *value2;
|
2023-02-14 19:54:13 +08:00
|
|
|
PyObject *res;
|
2023-08-05 00:35:56 +08:00
|
|
|
value1 = stack_pointer[-1];
|
|
|
|
value2 = stack_pointer[-2];
|
2023-02-14 19:54:13 +08:00
|
|
|
assert(oparg <= MAX_INTRINSIC_2);
|
2023-07-21 00:46:04 +08:00
|
|
|
res = _PyIntrinsics_BinaryFunctions[oparg].func(tstate, value2, value1);
|
2023-02-14 19:54:13 +08:00
|
|
|
Py_DECREF(value2);
|
|
|
|
Py_DECREF(value1);
|
|
|
|
if (res == NULL) goto pop_2_error;
|
|
|
|
STACK_SHRINK(1);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = res;
|
2023-02-14 19:54:13 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2022-11-03 12:31:26 +08:00
|
|
|
TARGET(RAISE_VARARGS) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject **args;
|
|
|
|
args = stack_pointer - oparg;
|
2022-11-03 12:31:26 +08:00
|
|
|
PyObject *cause = NULL, *exc = NULL;
|
|
|
|
switch (oparg) {
|
|
|
|
case 2:
|
2023-01-26 06:29:56 +08:00
|
|
|
cause = args[1];
|
2022-11-03 12:31:26 +08:00
|
|
|
/* fall through */
|
|
|
|
case 1:
|
2023-01-26 06:29:56 +08:00
|
|
|
exc = args[0];
|
2022-11-03 12:31:26 +08:00
|
|
|
/* fall through */
|
|
|
|
case 0:
|
2023-07-27 20:32:30 +08:00
|
|
|
if (do_raise(tstate, exc, cause)) {
|
|
|
|
assert(oparg == 0);
|
|
|
|
monitor_reraise(tstate, frame, next_instr-1);
|
|
|
|
goto exception_unwind;
|
|
|
|
}
|
2022-11-03 12:31:26 +08:00
|
|
|
break;
|
|
|
|
default:
|
|
|
|
_PyErr_SetString(tstate, PyExc_SystemError,
|
|
|
|
"bad RAISE_VARARGS oparg");
|
|
|
|
break;
|
|
|
|
}
|
2023-01-26 06:29:56 +08:00
|
|
|
if (true) { STACK_SHRINK(oparg); goto error; }
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
|
|
|
|
2022-11-10 20:34:57 +08:00
|
|
|
TARGET(INTERPRETER_EXIT) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *retval;
|
|
|
|
retval = stack_pointer[-1];
|
2022-11-10 20:34:57 +08:00
|
|
|
assert(frame == &entry_frame);
|
|
|
|
assert(_PyFrame_IsIncomplete(frame));
|
2023-08-17 18:16:03 +08:00
|
|
|
/* Restore previous frame and return. */
|
|
|
|
tstate->current_frame = frame->previous;
|
2022-11-03 12:31:26 +08:00
|
|
|
assert(!_PyErr_Occurred(tstate));
|
2023-08-04 17:10:29 +08:00
|
|
|
tstate->c_recursion_remaining += PY_EVAL_C_STACK_UNITS;
|
2022-11-03 12:31:26 +08:00
|
|
|
return retval;
|
|
|
|
}
|
|
|
|
|
2022-11-10 20:34:57 +08:00
|
|
|
TARGET(RETURN_VALUE) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *retval;
|
2023-09-12 06:39:19 +08:00
|
|
|
// _SAVE_CURRENT_IP
|
2023-08-18 02:29:58 +08:00
|
|
|
{
|
|
|
|
#if TIER_ONE
|
|
|
|
frame->prev_instr = next_instr - 1;
|
|
|
|
#endif
|
|
|
|
#if TIER_TWO
|
2023-09-12 06:39:19 +08:00
|
|
|
// Relies on a preceding _SET_IP
|
2023-08-18 02:29:58 +08:00
|
|
|
frame->prev_instr--;
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
// _POP_FRAME
|
2023-08-05 00:35:56 +08:00
|
|
|
retval = stack_pointer[-1];
|
GH-98831: Typed stack effects, and more instructions converted (#99764)
Stack effects can now have a type, e.g. `inst(X, (left, right -- jump/uint64_t)) { ... }`.
Instructions converted to the non-legacy format:
* COMPARE_OP
* COMPARE_OP_FLOAT_JUMP
* COMPARE_OP_INT_JUMP
* COMPARE_OP_STR_JUMP
* STORE_ATTR
* DELETE_ATTR
* STORE_GLOBAL
* STORE_ATTR_INSTANCE_VALUE
* STORE_ATTR_WITH_HINT
* STORE_ATTR_SLOT, and complete the store_attr family
* Complete the store_subscr family: STORE_SUBSCR{,DICT,LIST_INT}
(STORE_SUBSCR was alread half converted,
but wasn't using cache effects yet.)
* DELETE_SUBSCR
* PRINT_EXPR
* INTERPRETER_EXIT (a bit weird, ends in return)
* RETURN_VALUE
* GET_AITER (had to restructure it some)
The original had mysterious `SET_TOP(NULL)` before `goto error`.
I assume those just account for `obj` having been decref'ed,
so I got rid of them in favor of the cleanup implied by `ERROR_IF()`.
* LIST_APPEND (a bit unhappy with it)
* SET_ADD (also a bit unhappy with it)
Various other improvements/refactorings as well.
2022-12-09 05:31:27 +08:00
|
|
|
STACK_SHRINK(1);
|
2023-08-18 02:29:58 +08:00
|
|
|
{
|
|
|
|
assert(EMPTY());
|
|
|
|
#if TIER_ONE
|
|
|
|
assert(frame != &entry_frame);
|
|
|
|
#endif
|
2023-08-31 18:34:52 +08:00
|
|
|
STORE_SP();
|
|
|
|
_Py_LeaveRecursiveCallPy(tstate);
|
|
|
|
// GH-99729: We need to unlink the frame *before* clearing it:
|
|
|
|
_PyInterpreterFrame *dying = frame;
|
2023-08-18 02:29:58 +08:00
|
|
|
frame = tstate->current_frame = dying->previous;
|
|
|
|
_PyEval_FrameClearAndPop(tstate, dying);
|
|
|
|
frame->prev_instr += frame->return_offset;
|
|
|
|
_PyFrame_StackPush(frame, retval);
|
2023-08-31 18:34:52 +08:00
|
|
|
LOAD_SP();
|
|
|
|
LOAD_IP();
|
|
|
|
#if LLTRACE && TIER_ONE
|
|
|
|
lltrace = maybe_lltrace_resume_frame(frame, &entry_frame, GLOBALS());
|
|
|
|
if (lltrace < 0) {
|
|
|
|
goto exit_unwind;
|
|
|
|
}
|
|
|
|
#endif
|
2023-08-18 02:29:58 +08:00
|
|
|
}
|
2023-08-31 18:34:52 +08:00
|
|
|
DISPATCH();
|
2023-04-12 19:04:55 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(INSTRUMENTED_RETURN_VALUE) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *retval;
|
|
|
|
retval = stack_pointer[-1];
|
2023-04-12 19:04:55 +08:00
|
|
|
int err = _Py_call_instrumentation_arg(
|
|
|
|
tstate, PY_MONITORING_EVENT_PY_RETURN,
|
|
|
|
frame, next_instr-1, retval);
|
|
|
|
if (err) goto error;
|
|
|
|
STACK_SHRINK(1);
|
|
|
|
assert(EMPTY());
|
|
|
|
_PyFrame_SetStackPointer(frame, stack_pointer);
|
|
|
|
_Py_LeaveRecursiveCallPy(tstate);
|
|
|
|
assert(frame != &entry_frame);
|
|
|
|
// GH-99729: We need to unlink the frame *before* clearing it:
|
|
|
|
_PyInterpreterFrame *dying = frame;
|
2023-08-17 18:16:03 +08:00
|
|
|
frame = tstate->current_frame = dying->previous;
|
2023-08-18 02:29:58 +08:00
|
|
|
_PyEval_FrameClearAndPop(tstate, dying);
|
2023-04-13 23:19:07 +08:00
|
|
|
frame->prev_instr += frame->return_offset;
|
2023-04-12 19:04:55 +08:00
|
|
|
_PyFrame_StackPush(frame, retval);
|
|
|
|
goto resume_frame;
|
2023-02-08 06:32:21 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(RETURN_CONST) {
|
2023-08-18 02:29:58 +08:00
|
|
|
PyObject *value;
|
|
|
|
PyObject *retval;
|
|
|
|
// LOAD_CONST
|
|
|
|
{
|
|
|
|
value = GETITEM(FRAME_CO_CONSTS, oparg);
|
|
|
|
Py_INCREF(value);
|
|
|
|
}
|
2023-09-12 06:39:19 +08:00
|
|
|
// _SAVE_CURRENT_IP
|
2023-08-18 02:29:58 +08:00
|
|
|
{
|
|
|
|
#if TIER_ONE
|
|
|
|
frame->prev_instr = next_instr - 1;
|
|
|
|
#endif
|
|
|
|
#if TIER_TWO
|
2023-09-12 06:39:19 +08:00
|
|
|
// Relies on a preceding _SET_IP
|
2023-08-18 02:29:58 +08:00
|
|
|
frame->prev_instr--;
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
// _POP_FRAME
|
|
|
|
retval = value;
|
|
|
|
{
|
|
|
|
assert(EMPTY());
|
|
|
|
#if TIER_ONE
|
|
|
|
assert(frame != &entry_frame);
|
|
|
|
#endif
|
2023-08-31 18:34:52 +08:00
|
|
|
STORE_SP();
|
|
|
|
_Py_LeaveRecursiveCallPy(tstate);
|
|
|
|
// GH-99729: We need to unlink the frame *before* clearing it:
|
|
|
|
_PyInterpreterFrame *dying = frame;
|
2023-08-18 02:29:58 +08:00
|
|
|
frame = tstate->current_frame = dying->previous;
|
|
|
|
_PyEval_FrameClearAndPop(tstate, dying);
|
|
|
|
frame->prev_instr += frame->return_offset;
|
|
|
|
_PyFrame_StackPush(frame, retval);
|
2023-08-31 18:34:52 +08:00
|
|
|
LOAD_SP();
|
|
|
|
LOAD_IP();
|
|
|
|
#if LLTRACE && TIER_ONE
|
|
|
|
lltrace = maybe_lltrace_resume_frame(frame, &entry_frame, GLOBALS());
|
|
|
|
if (lltrace < 0) {
|
|
|
|
goto exit_unwind;
|
|
|
|
}
|
|
|
|
#endif
|
2023-08-18 02:29:58 +08:00
|
|
|
}
|
2023-08-31 18:34:52 +08:00
|
|
|
DISPATCH();
|
2023-04-12 19:04:55 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(INSTRUMENTED_RETURN_CONST) {
|
2023-06-14 04:42:03 +08:00
|
|
|
PyObject *retval = GETITEM(FRAME_CO_CONSTS, oparg);
|
2023-04-12 19:04:55 +08:00
|
|
|
int err = _Py_call_instrumentation_arg(
|
|
|
|
tstate, PY_MONITORING_EVENT_PY_RETURN,
|
|
|
|
frame, next_instr-1, retval);
|
|
|
|
if (err) goto error;
|
|
|
|
Py_INCREF(retval);
|
|
|
|
assert(EMPTY());
|
|
|
|
_PyFrame_SetStackPointer(frame, stack_pointer);
|
|
|
|
_Py_LeaveRecursiveCallPy(tstate);
|
|
|
|
assert(frame != &entry_frame);
|
|
|
|
// GH-99729: We need to unlink the frame *before* clearing it:
|
|
|
|
_PyInterpreterFrame *dying = frame;
|
2023-08-17 18:16:03 +08:00
|
|
|
frame = tstate->current_frame = dying->previous;
|
2023-08-18 02:29:58 +08:00
|
|
|
_PyEval_FrameClearAndPop(tstate, dying);
|
2023-04-13 23:19:07 +08:00
|
|
|
frame->prev_instr += frame->return_offset;
|
2023-04-12 19:04:55 +08:00
|
|
|
_PyFrame_StackPush(frame, retval);
|
|
|
|
goto resume_frame;
|
2022-11-10 20:34:57 +08:00
|
|
|
}
|
|
|
|
|
2022-11-03 12:31:26 +08:00
|
|
|
TARGET(GET_AITER) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *obj;
|
GH-98831: Typed stack effects, and more instructions converted (#99764)
Stack effects can now have a type, e.g. `inst(X, (left, right -- jump/uint64_t)) { ... }`.
Instructions converted to the non-legacy format:
* COMPARE_OP
* COMPARE_OP_FLOAT_JUMP
* COMPARE_OP_INT_JUMP
* COMPARE_OP_STR_JUMP
* STORE_ATTR
* DELETE_ATTR
* STORE_GLOBAL
* STORE_ATTR_INSTANCE_VALUE
* STORE_ATTR_WITH_HINT
* STORE_ATTR_SLOT, and complete the store_attr family
* Complete the store_subscr family: STORE_SUBSCR{,DICT,LIST_INT}
(STORE_SUBSCR was alread half converted,
but wasn't using cache effects yet.)
* DELETE_SUBSCR
* PRINT_EXPR
* INTERPRETER_EXIT (a bit weird, ends in return)
* RETURN_VALUE
* GET_AITER (had to restructure it some)
The original had mysterious `SET_TOP(NULL)` before `goto error`.
I assume those just account for `obj` having been decref'ed,
so I got rid of them in favor of the cleanup implied by `ERROR_IF()`.
* LIST_APPEND (a bit unhappy with it)
* SET_ADD (also a bit unhappy with it)
Various other improvements/refactorings as well.
2022-12-09 05:31:27 +08:00
|
|
|
PyObject *iter;
|
2023-08-05 00:35:56 +08:00
|
|
|
obj = stack_pointer[-1];
|
2022-11-03 12:31:26 +08:00
|
|
|
unaryfunc getter = NULL;
|
|
|
|
PyTypeObject *type = Py_TYPE(obj);
|
|
|
|
|
|
|
|
if (type->tp_as_async != NULL) {
|
|
|
|
getter = type->tp_as_async->am_aiter;
|
|
|
|
}
|
|
|
|
|
GH-98831: Typed stack effects, and more instructions converted (#99764)
Stack effects can now have a type, e.g. `inst(X, (left, right -- jump/uint64_t)) { ... }`.
Instructions converted to the non-legacy format:
* COMPARE_OP
* COMPARE_OP_FLOAT_JUMP
* COMPARE_OP_INT_JUMP
* COMPARE_OP_STR_JUMP
* STORE_ATTR
* DELETE_ATTR
* STORE_GLOBAL
* STORE_ATTR_INSTANCE_VALUE
* STORE_ATTR_WITH_HINT
* STORE_ATTR_SLOT, and complete the store_attr family
* Complete the store_subscr family: STORE_SUBSCR{,DICT,LIST_INT}
(STORE_SUBSCR was alread half converted,
but wasn't using cache effects yet.)
* DELETE_SUBSCR
* PRINT_EXPR
* INTERPRETER_EXIT (a bit weird, ends in return)
* RETURN_VALUE
* GET_AITER (had to restructure it some)
The original had mysterious `SET_TOP(NULL)` before `goto error`.
I assume those just account for `obj` having been decref'ed,
so I got rid of them in favor of the cleanup implied by `ERROR_IF()`.
* LIST_APPEND (a bit unhappy with it)
* SET_ADD (also a bit unhappy with it)
Various other improvements/refactorings as well.
2022-12-09 05:31:27 +08:00
|
|
|
if (getter == NULL) {
|
2022-11-03 12:31:26 +08:00
|
|
|
_PyErr_Format(tstate, PyExc_TypeError,
|
|
|
|
"'async for' requires an object with "
|
|
|
|
"__aiter__ method, got %.100s",
|
|
|
|
type->tp_name);
|
|
|
|
Py_DECREF(obj);
|
GH-98831: Typed stack effects, and more instructions converted (#99764)
Stack effects can now have a type, e.g. `inst(X, (left, right -- jump/uint64_t)) { ... }`.
Instructions converted to the non-legacy format:
* COMPARE_OP
* COMPARE_OP_FLOAT_JUMP
* COMPARE_OP_INT_JUMP
* COMPARE_OP_STR_JUMP
* STORE_ATTR
* DELETE_ATTR
* STORE_GLOBAL
* STORE_ATTR_INSTANCE_VALUE
* STORE_ATTR_WITH_HINT
* STORE_ATTR_SLOT, and complete the store_attr family
* Complete the store_subscr family: STORE_SUBSCR{,DICT,LIST_INT}
(STORE_SUBSCR was alread half converted,
but wasn't using cache effects yet.)
* DELETE_SUBSCR
* PRINT_EXPR
* INTERPRETER_EXIT (a bit weird, ends in return)
* RETURN_VALUE
* GET_AITER (had to restructure it some)
The original had mysterious `SET_TOP(NULL)` before `goto error`.
I assume those just account for `obj` having been decref'ed,
so I got rid of them in favor of the cleanup implied by `ERROR_IF()`.
* LIST_APPEND (a bit unhappy with it)
* SET_ADD (also a bit unhappy with it)
Various other improvements/refactorings as well.
2022-12-09 05:31:27 +08:00
|
|
|
if (true) goto pop_1_error;
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
|
|
|
|
GH-98831: Typed stack effects, and more instructions converted (#99764)
Stack effects can now have a type, e.g. `inst(X, (left, right -- jump/uint64_t)) { ... }`.
Instructions converted to the non-legacy format:
* COMPARE_OP
* COMPARE_OP_FLOAT_JUMP
* COMPARE_OP_INT_JUMP
* COMPARE_OP_STR_JUMP
* STORE_ATTR
* DELETE_ATTR
* STORE_GLOBAL
* STORE_ATTR_INSTANCE_VALUE
* STORE_ATTR_WITH_HINT
* STORE_ATTR_SLOT, and complete the store_attr family
* Complete the store_subscr family: STORE_SUBSCR{,DICT,LIST_INT}
(STORE_SUBSCR was alread half converted,
but wasn't using cache effects yet.)
* DELETE_SUBSCR
* PRINT_EXPR
* INTERPRETER_EXIT (a bit weird, ends in return)
* RETURN_VALUE
* GET_AITER (had to restructure it some)
The original had mysterious `SET_TOP(NULL)` before `goto error`.
I assume those just account for `obj` having been decref'ed,
so I got rid of them in favor of the cleanup implied by `ERROR_IF()`.
* LIST_APPEND (a bit unhappy with it)
* SET_ADD (also a bit unhappy with it)
Various other improvements/refactorings as well.
2022-12-09 05:31:27 +08:00
|
|
|
iter = (*getter)(obj);
|
|
|
|
Py_DECREF(obj);
|
|
|
|
if (iter == NULL) goto pop_1_error;
|
|
|
|
|
2022-11-03 12:31:26 +08:00
|
|
|
if (Py_TYPE(iter)->tp_as_async == NULL ||
|
|
|
|
Py_TYPE(iter)->tp_as_async->am_anext == NULL) {
|
|
|
|
|
|
|
|
_PyErr_Format(tstate, PyExc_TypeError,
|
|
|
|
"'async for' received an object from __aiter__ "
|
|
|
|
"that does not implement __anext__: %.100s",
|
|
|
|
Py_TYPE(iter)->tp_name);
|
|
|
|
Py_DECREF(iter);
|
GH-98831: Typed stack effects, and more instructions converted (#99764)
Stack effects can now have a type, e.g. `inst(X, (left, right -- jump/uint64_t)) { ... }`.
Instructions converted to the non-legacy format:
* COMPARE_OP
* COMPARE_OP_FLOAT_JUMP
* COMPARE_OP_INT_JUMP
* COMPARE_OP_STR_JUMP
* STORE_ATTR
* DELETE_ATTR
* STORE_GLOBAL
* STORE_ATTR_INSTANCE_VALUE
* STORE_ATTR_WITH_HINT
* STORE_ATTR_SLOT, and complete the store_attr family
* Complete the store_subscr family: STORE_SUBSCR{,DICT,LIST_INT}
(STORE_SUBSCR was alread half converted,
but wasn't using cache effects yet.)
* DELETE_SUBSCR
* PRINT_EXPR
* INTERPRETER_EXIT (a bit weird, ends in return)
* RETURN_VALUE
* GET_AITER (had to restructure it some)
The original had mysterious `SET_TOP(NULL)` before `goto error`.
I assume those just account for `obj` having been decref'ed,
so I got rid of them in favor of the cleanup implied by `ERROR_IF()`.
* LIST_APPEND (a bit unhappy with it)
* SET_ADD (also a bit unhappy with it)
Various other improvements/refactorings as well.
2022-12-09 05:31:27 +08:00
|
|
|
if (true) goto pop_1_error;
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = iter;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(GET_ANEXT) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *aiter;
|
2022-12-28 09:11:03 +08:00
|
|
|
PyObject *awaitable;
|
2023-08-05 00:35:56 +08:00
|
|
|
aiter = stack_pointer[-1];
|
2022-11-03 12:31:26 +08:00
|
|
|
unaryfunc getter = NULL;
|
|
|
|
PyObject *next_iter = NULL;
|
|
|
|
PyTypeObject *type = Py_TYPE(aiter);
|
|
|
|
|
|
|
|
if (PyAsyncGen_CheckExact(aiter)) {
|
|
|
|
awaitable = type->tp_as_async->am_anext(aiter);
|
|
|
|
if (awaitable == NULL) {
|
|
|
|
goto error;
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
if (type->tp_as_async != NULL){
|
|
|
|
getter = type->tp_as_async->am_anext;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (getter != NULL) {
|
|
|
|
next_iter = (*getter)(aiter);
|
|
|
|
if (next_iter == NULL) {
|
|
|
|
goto error;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
else {
|
|
|
|
_PyErr_Format(tstate, PyExc_TypeError,
|
|
|
|
"'async for' requires an iterator with "
|
|
|
|
"__anext__ method, got %.100s",
|
|
|
|
type->tp_name);
|
|
|
|
goto error;
|
|
|
|
}
|
|
|
|
|
|
|
|
awaitable = _PyCoro_GetAwaitableIter(next_iter);
|
|
|
|
if (awaitable == NULL) {
|
|
|
|
_PyErr_FormatFromCause(
|
|
|
|
PyExc_TypeError,
|
|
|
|
"'async for' received an invalid object "
|
|
|
|
"from __anext__: %.100s",
|
|
|
|
Py_TYPE(next_iter)->tp_name);
|
|
|
|
|
|
|
|
Py_DECREF(next_iter);
|
|
|
|
goto error;
|
|
|
|
} else {
|
|
|
|
Py_DECREF(next_iter);
|
|
|
|
}
|
|
|
|
}
|
2022-12-28 09:11:03 +08:00
|
|
|
STACK_GROW(1);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = awaitable;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(GET_AWAITABLE) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *iterable;
|
2022-12-28 09:11:03 +08:00
|
|
|
PyObject *iter;
|
2023-08-05 00:35:56 +08:00
|
|
|
iterable = stack_pointer[-1];
|
2022-12-28 09:11:03 +08:00
|
|
|
iter = _PyCoro_GetAwaitableIter(iterable);
|
2022-11-03 12:31:26 +08:00
|
|
|
|
|
|
|
if (iter == NULL) {
|
2023-07-21 04:37:19 +08:00
|
|
|
_PyEval_FormatAwaitableError(tstate, Py_TYPE(iterable), oparg);
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
Py_DECREF(iterable);
|
|
|
|
|
|
|
|
if (iter != NULL && PyCoro_CheckExact(iter)) {
|
|
|
|
PyObject *yf = _PyGen_yf((PyGenObject*)iter);
|
|
|
|
if (yf != NULL) {
|
|
|
|
/* `iter` is a coroutine object that is being
|
|
|
|
awaited, `yf` is a pointer to the current awaitable
|
|
|
|
being awaited on. */
|
|
|
|
Py_DECREF(yf);
|
|
|
|
Py_CLEAR(iter);
|
|
|
|
_PyErr_SetString(tstate, PyExc_RuntimeError,
|
|
|
|
"coroutine is being awaited already");
|
|
|
|
/* The code below jumps to `error` if `iter` is NULL. */
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-12-28 09:11:03 +08:00
|
|
|
if (iter == NULL) goto pop_1_error;
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = iter;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(SEND) {
|
2023-02-13 19:24:55 +08:00
|
|
|
PREDICTED(SEND);
|
2023-05-11 19:52:06 +08:00
|
|
|
static_assert(INLINE_CACHE_ENTRIES_SEND == 1, "incorrect cache size");
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *v;
|
|
|
|
PyObject *receiver;
|
2023-02-03 19:30:21 +08:00
|
|
|
PyObject *retval;
|
2023-08-05 00:35:56 +08:00
|
|
|
v = stack_pointer[-1];
|
|
|
|
receiver = stack_pointer[-2];
|
2023-02-13 19:24:55 +08:00
|
|
|
#if ENABLE_SPECIALIZATION
|
|
|
|
_PySendCache *cache = (_PySendCache *)next_instr;
|
|
|
|
if (ADAPTIVE_COUNTER_IS_ZERO(cache->counter)) {
|
|
|
|
next_instr--;
|
|
|
|
_Py_Specialize_Send(receiver, next_instr);
|
|
|
|
DISPATCH_SAME_OPARG();
|
|
|
|
}
|
|
|
|
STAT_INC(SEND, deferred);
|
|
|
|
DECREMENT_ADAPTIVE_COUNTER(cache->counter);
|
|
|
|
#endif /* ENABLE_SPECIALIZATION */
|
2022-11-10 20:34:57 +08:00
|
|
|
assert(frame != &entry_frame);
|
2023-05-13 06:23:13 +08:00
|
|
|
if ((tstate->interp->eval_frame == NULL) &&
|
|
|
|
(Py_TYPE(receiver) == &PyGen_Type || Py_TYPE(receiver) == &PyCoro_Type) &&
|
|
|
|
((PyGenObject *)receiver)->gi_frame_state < FRAME_EXECUTING)
|
2023-04-12 19:04:55 +08:00
|
|
|
{
|
|
|
|
PyGenObject *gen = (PyGenObject *)receiver;
|
|
|
|
_PyInterpreterFrame *gen_frame = (_PyInterpreterFrame *)gen->gi_iframe;
|
|
|
|
STACK_SHRINK(1);
|
|
|
|
_PyFrame_StackPush(gen_frame, v);
|
|
|
|
gen->gi_frame_state = FRAME_EXECUTING;
|
|
|
|
gen->gi_exc_state.previous_item = tstate->exc_info;
|
|
|
|
tstate->exc_info = &gen->gi_exc_state;
|
2023-06-15 07:14:22 +08:00
|
|
|
SKIP_OVER(INLINE_CACHE_ENTRIES_SEND);
|
2023-08-17 07:26:43 +08:00
|
|
|
frame->return_offset = oparg;
|
2023-04-12 19:04:55 +08:00
|
|
|
DISPATCH_INLINED(gen_frame);
|
|
|
|
}
|
2023-02-13 19:24:55 +08:00
|
|
|
if (Py_IsNone(v) && PyIter_Check(receiver)) {
|
|
|
|
retval = Py_TYPE(receiver)->tp_iternext(receiver);
|
|
|
|
}
|
|
|
|
else {
|
|
|
|
retval = PyObject_CallMethodOneArg(receiver, &_Py_ID(send), v);
|
|
|
|
}
|
|
|
|
if (retval == NULL) {
|
2023-04-12 19:04:55 +08:00
|
|
|
if (_PyErr_ExceptionMatches(tstate, PyExc_StopIteration)
|
|
|
|
) {
|
|
|
|
monitor_raise(tstate, frame, next_instr-1);
|
|
|
|
}
|
2023-02-13 19:24:55 +08:00
|
|
|
if (_PyGen_FetchStopIterationValue(&retval) == 0) {
|
|
|
|
assert(retval != NULL);
|
|
|
|
JUMPBY(oparg);
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
|
|
|
else {
|
2023-02-13 19:24:55 +08:00
|
|
|
goto error;
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
|
|
|
}
|
2023-02-15 20:21:40 +08:00
|
|
|
Py_DECREF(v);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = retval;
|
|
|
|
next_instr += 1;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2023-02-13 19:24:55 +08:00
|
|
|
TARGET(SEND_GEN) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *v;
|
|
|
|
PyObject *receiver;
|
|
|
|
v = stack_pointer[-1];
|
|
|
|
receiver = stack_pointer[-2];
|
2023-05-13 06:23:13 +08:00
|
|
|
DEOPT_IF(tstate->interp->eval_frame, SEND);
|
2023-02-13 19:24:55 +08:00
|
|
|
PyGenObject *gen = (PyGenObject *)receiver;
|
|
|
|
DEOPT_IF(Py_TYPE(gen) != &PyGen_Type &&
|
|
|
|
Py_TYPE(gen) != &PyCoro_Type, SEND);
|
|
|
|
DEOPT_IF(gen->gi_frame_state >= FRAME_EXECUTING, SEND);
|
|
|
|
STAT_INC(SEND, hit);
|
|
|
|
_PyInterpreterFrame *gen_frame = (_PyInterpreterFrame *)gen->gi_iframe;
|
|
|
|
STACK_SHRINK(1);
|
|
|
|
_PyFrame_StackPush(gen_frame, v);
|
|
|
|
gen->gi_frame_state = FRAME_EXECUTING;
|
|
|
|
gen->gi_exc_state.previous_item = tstate->exc_info;
|
|
|
|
tstate->exc_info = &gen->gi_exc_state;
|
2023-06-15 07:14:22 +08:00
|
|
|
SKIP_OVER(INLINE_CACHE_ENTRIES_SEND);
|
2023-08-17 07:26:43 +08:00
|
|
|
frame->return_offset = oparg;
|
2023-02-13 19:24:55 +08:00
|
|
|
DISPATCH_INLINED(gen_frame);
|
2023-04-12 19:04:55 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(INSTRUMENTED_YIELD_VALUE) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *retval;
|
|
|
|
retval = stack_pointer[-1];
|
2023-04-12 19:04:55 +08:00
|
|
|
assert(frame != &entry_frame);
|
2023-06-14 04:42:03 +08:00
|
|
|
assert(oparg >= 0); /* make the generator identify this as HAS_ARG */
|
2023-04-12 19:04:55 +08:00
|
|
|
PyGenObject *gen = _PyFrame_GetGenerator(frame);
|
|
|
|
gen->gi_frame_state = FRAME_SUSPENDED;
|
|
|
|
_PyFrame_SetStackPointer(frame, stack_pointer - 1);
|
|
|
|
int err = _Py_call_instrumentation_arg(
|
|
|
|
tstate, PY_MONITORING_EVENT_PY_YIELD,
|
|
|
|
frame, next_instr-1, retval);
|
|
|
|
if (err) goto error;
|
|
|
|
tstate->exc_info = gen->gi_exc_state.previous_item;
|
|
|
|
gen->gi_exc_state.previous_item = NULL;
|
|
|
|
_Py_LeaveRecursiveCallPy(tstate);
|
|
|
|
_PyInterpreterFrame *gen_frame = frame;
|
2023-08-17 18:16:03 +08:00
|
|
|
frame = tstate->current_frame = frame->previous;
|
2023-04-12 19:04:55 +08:00
|
|
|
gen_frame->previous = NULL;
|
|
|
|
_PyFrame_StackPush(frame, retval);
|
|
|
|
goto resume_frame;
|
2023-02-13 19:24:55 +08:00
|
|
|
}
|
|
|
|
|
2022-11-03 12:31:26 +08:00
|
|
|
TARGET(YIELD_VALUE) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *retval;
|
|
|
|
retval = stack_pointer[-1];
|
2022-11-03 12:31:26 +08:00
|
|
|
// NOTE: It's important that YIELD_VALUE never raises an exception!
|
|
|
|
// The compiler treats any exception raised here as a failed close()
|
|
|
|
// or throw() call.
|
2023-06-14 04:42:03 +08:00
|
|
|
assert(oparg >= 0); /* make the generator identify this as HAS_ARG */
|
2022-11-10 20:34:57 +08:00
|
|
|
assert(frame != &entry_frame);
|
2022-11-07 22:49:51 +08:00
|
|
|
PyGenObject *gen = _PyFrame_GetGenerator(frame);
|
|
|
|
gen->gi_frame_state = FRAME_SUSPENDED;
|
2022-12-28 09:11:03 +08:00
|
|
|
_PyFrame_SetStackPointer(frame, stack_pointer - 1);
|
2022-11-07 22:49:51 +08:00
|
|
|
tstate->exc_info = gen->gi_exc_state.previous_item;
|
|
|
|
gen->gi_exc_state.previous_item = NULL;
|
2022-11-03 12:31:26 +08:00
|
|
|
_Py_LeaveRecursiveCallPy(tstate);
|
2022-11-10 20:34:57 +08:00
|
|
|
_PyInterpreterFrame *gen_frame = frame;
|
2023-08-17 18:16:03 +08:00
|
|
|
frame = tstate->current_frame = frame->previous;
|
2022-11-10 20:34:57 +08:00
|
|
|
gen_frame->previous = NULL;
|
|
|
|
_PyFrame_StackPush(frame, retval);
|
|
|
|
goto resume_frame;
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(POP_EXCEPT) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *exc_value;
|
|
|
|
exc_value = stack_pointer[-1];
|
2022-11-03 12:31:26 +08:00
|
|
|
_PyErr_StackItem *exc_info = tstate->exc_info;
|
2022-12-28 09:11:03 +08:00
|
|
|
Py_XSETREF(exc_info->exc_value, exc_value);
|
|
|
|
STACK_SHRINK(1);
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(RERAISE) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *exc;
|
|
|
|
PyObject **values;
|
|
|
|
exc = stack_pointer[-1];
|
|
|
|
values = stack_pointer - 1 - oparg;
|
2023-02-02 18:02:57 +08:00
|
|
|
assert(oparg >= 0 && oparg <= 2);
|
2022-11-03 12:31:26 +08:00
|
|
|
if (oparg) {
|
2023-02-02 18:02:57 +08:00
|
|
|
PyObject *lasti = values[0];
|
2022-11-03 12:31:26 +08:00
|
|
|
if (PyLong_Check(lasti)) {
|
2023-06-14 20:46:37 +08:00
|
|
|
frame->prev_instr = _PyCode_CODE(_PyFrame_GetCode(frame)) + PyLong_AsLong(lasti);
|
2022-11-03 12:31:26 +08:00
|
|
|
assert(!_PyErr_Occurred(tstate));
|
|
|
|
}
|
|
|
|
else {
|
|
|
|
assert(PyLong_Check(lasti));
|
|
|
|
_PyErr_SetString(tstate, PyExc_SystemError, "lasti is not an int");
|
|
|
|
goto error;
|
|
|
|
}
|
|
|
|
}
|
2023-02-02 18:02:57 +08:00
|
|
|
assert(exc && PyExceptionInstance_Check(exc));
|
|
|
|
Py_INCREF(exc);
|
2023-03-19 23:17:59 +08:00
|
|
|
_PyErr_SetRaisedException(tstate, exc);
|
2023-07-27 20:32:30 +08:00
|
|
|
monitor_reraise(tstate, frame, next_instr-1);
|
2022-11-03 12:31:26 +08:00
|
|
|
goto exception_unwind;
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(END_ASYNC_FOR) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *exc;
|
|
|
|
PyObject *awaitable;
|
|
|
|
exc = stack_pointer[-1];
|
|
|
|
awaitable = stack_pointer[-2];
|
2023-02-01 02:47:50 +08:00
|
|
|
assert(exc && PyExceptionInstance_Check(exc));
|
|
|
|
if (PyErr_GivenExceptionMatches(exc, PyExc_StopAsyncIteration)) {
|
|
|
|
Py_DECREF(awaitable);
|
|
|
|
Py_DECREF(exc);
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
|
|
|
else {
|
2023-02-01 02:47:50 +08:00
|
|
|
Py_INCREF(exc);
|
2023-03-19 23:17:59 +08:00
|
|
|
_PyErr_SetRaisedException(tstate, exc);
|
2023-07-27 20:32:30 +08:00
|
|
|
monitor_reraise(tstate, frame, next_instr-1);
|
2022-11-03 12:31:26 +08:00
|
|
|
goto exception_unwind;
|
|
|
|
}
|
2023-02-01 02:47:50 +08:00
|
|
|
STACK_SHRINK(2);
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(CLEANUP_THROW) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *exc_value;
|
|
|
|
PyObject *last_sent_val;
|
|
|
|
PyObject *sub_iter;
|
2023-02-13 19:24:55 +08:00
|
|
|
PyObject *none;
|
2023-02-02 18:02:57 +08:00
|
|
|
PyObject *value;
|
2023-08-05 00:35:56 +08:00
|
|
|
exc_value = stack_pointer[-1];
|
|
|
|
last_sent_val = stack_pointer[-2];
|
|
|
|
sub_iter = stack_pointer[-3];
|
2022-11-03 12:31:26 +08:00
|
|
|
assert(throwflag);
|
|
|
|
assert(exc_value && PyExceptionInstance_Check(exc_value));
|
|
|
|
if (PyErr_GivenExceptionMatches(exc_value, PyExc_StopIteration)) {
|
2023-02-02 18:02:57 +08:00
|
|
|
value = Py_NewRef(((PyStopIterationObject *)exc_value)->value);
|
|
|
|
Py_DECREF(sub_iter);
|
|
|
|
Py_DECREF(last_sent_val);
|
|
|
|
Py_DECREF(exc_value);
|
2023-05-17 05:36:02 +08:00
|
|
|
none = Py_None;
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
2022-11-11 02:50:57 +08:00
|
|
|
else {
|
2023-02-08 17:31:12 +08:00
|
|
|
_PyErr_SetRaisedException(tstate, Py_NewRef(exc_value));
|
2023-07-27 20:32:30 +08:00
|
|
|
monitor_reraise(tstate, frame, next_instr-1);
|
2022-11-11 02:50:57 +08:00
|
|
|
goto exception_unwind;
|
|
|
|
}
|
2023-02-13 19:24:55 +08:00
|
|
|
STACK_SHRINK(1);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-2] = none;
|
2023-08-05 00:35:56 +08:00
|
|
|
stack_pointer[-1] = value;
|
2022-11-11 02:50:57 +08:00
|
|
|
DISPATCH();
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(LOAD_ASSERTION_ERROR) {
|
2022-12-28 09:11:03 +08:00
|
|
|
PyObject *value;
|
|
|
|
value = Py_NewRef(PyExc_AssertionError);
|
|
|
|
STACK_GROW(1);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = value;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(LOAD_BUILD_CLASS) {
|
|
|
|
PyObject *bc;
|
2023-07-12 04:04:12 +08:00
|
|
|
if (PyMapping_GetOptionalItem(BUILTINS(), &_Py_ID(__build_class__), &bc) < 0) goto error;
|
|
|
|
if (bc == NULL) {
|
|
|
|
_PyErr_SetString(tstate, PyExc_NameError,
|
|
|
|
"__build_class__ not found");
|
|
|
|
if (true) goto error;
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
2022-12-28 09:11:03 +08:00
|
|
|
STACK_GROW(1);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = bc;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(STORE_NAME) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *v;
|
|
|
|
v = stack_pointer[-1];
|
2023-06-14 04:42:03 +08:00
|
|
|
PyObject *name = GETITEM(FRAME_CO_NAMES, oparg);
|
2022-11-03 12:31:26 +08:00
|
|
|
PyObject *ns = LOCALS();
|
|
|
|
int err;
|
|
|
|
if (ns == NULL) {
|
|
|
|
_PyErr_Format(tstate, PyExc_SystemError,
|
|
|
|
"no locals found when storing %R", name);
|
|
|
|
Py_DECREF(v);
|
2022-12-28 09:11:03 +08:00
|
|
|
if (true) goto pop_1_error;
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
|
|
|
if (PyDict_CheckExact(ns))
|
|
|
|
err = PyDict_SetItem(ns, name, v);
|
|
|
|
else
|
|
|
|
err = PyObject_SetItem(ns, name, v);
|
|
|
|
Py_DECREF(v);
|
2022-12-28 09:11:03 +08:00
|
|
|
if (err) goto pop_1_error;
|
|
|
|
STACK_SHRINK(1);
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(DELETE_NAME) {
|
2023-06-14 04:42:03 +08:00
|
|
|
PyObject *name = GETITEM(FRAME_CO_NAMES, oparg);
|
2022-11-03 12:31:26 +08:00
|
|
|
PyObject *ns = LOCALS();
|
|
|
|
int err;
|
|
|
|
if (ns == NULL) {
|
|
|
|
_PyErr_Format(tstate, PyExc_SystemError,
|
|
|
|
"no locals when deleting %R", name);
|
|
|
|
goto error;
|
|
|
|
}
|
|
|
|
err = PyObject_DelItem(ns, name);
|
2022-11-11 02:50:57 +08:00
|
|
|
// Can't use ERROR_IF here.
|
2022-11-03 12:31:26 +08:00
|
|
|
if (err != 0) {
|
2023-07-21 04:37:19 +08:00
|
|
|
_PyEval_FormatExcCheckArg(tstate, PyExc_NameError,
|
|
|
|
NAME_ERROR_MSG,
|
|
|
|
name);
|
2022-11-03 12:31:26 +08:00
|
|
|
goto error;
|
|
|
|
}
|
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(UNPACK_SEQUENCE) {
|
|
|
|
PREDICTED(UNPACK_SEQUENCE);
|
2023-02-08 07:44:37 +08:00
|
|
|
static_assert(INLINE_CACHE_ENTRIES_UNPACK_SEQUENCE == 1, "incorrect cache size");
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *seq;
|
|
|
|
seq = stack_pointer[-1];
|
2023-01-20 02:14:55 +08:00
|
|
|
#if ENABLE_SPECIALIZATION
|
2022-11-10 02:50:09 +08:00
|
|
|
_PyUnpackSequenceCache *cache = (_PyUnpackSequenceCache *)next_instr;
|
|
|
|
if (ADAPTIVE_COUNTER_IS_ZERO(cache->counter)) {
|
|
|
|
next_instr--;
|
|
|
|
_Py_Specialize_UnpackSequence(seq, next_instr, oparg);
|
|
|
|
DISPATCH_SAME_OPARG();
|
|
|
|
}
|
|
|
|
STAT_INC(UNPACK_SEQUENCE, deferred);
|
|
|
|
DECREMENT_ADAPTIVE_COUNTER(cache->counter);
|
2023-01-20 02:14:55 +08:00
|
|
|
#endif /* ENABLE_SPECIALIZATION */
|
2023-02-08 04:37:43 +08:00
|
|
|
PyObject **top = stack_pointer + oparg - 1;
|
2023-07-21 04:37:19 +08:00
|
|
|
int res = _PyEval_UnpackIterable(tstate, seq, oparg, -1, top);
|
2022-11-03 12:31:26 +08:00
|
|
|
Py_DECREF(seq);
|
2023-02-08 04:37:43 +08:00
|
|
|
if (res == 0) goto pop_1_error;
|
|
|
|
STACK_SHRINK(1);
|
|
|
|
STACK_GROW(oparg);
|
2023-03-01 00:49:35 +08:00
|
|
|
next_instr += 1;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(UNPACK_SEQUENCE_TWO_TUPLE) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *seq;
|
|
|
|
PyObject **values;
|
|
|
|
seq = stack_pointer[-1];
|
|
|
|
values = stack_pointer - 1;
|
2022-11-03 12:31:26 +08:00
|
|
|
DEOPT_IF(!PyTuple_CheckExact(seq), UNPACK_SEQUENCE);
|
|
|
|
DEOPT_IF(PyTuple_GET_SIZE(seq) != 2, UNPACK_SEQUENCE);
|
2023-02-08 07:44:37 +08:00
|
|
|
assert(oparg == 2);
|
2022-11-03 12:31:26 +08:00
|
|
|
STAT_INC(UNPACK_SEQUENCE, hit);
|
2023-02-08 07:44:37 +08:00
|
|
|
values[0] = Py_NewRef(PyTuple_GET_ITEM(seq, 1));
|
|
|
|
values[1] = Py_NewRef(PyTuple_GET_ITEM(seq, 0));
|
2022-11-03 12:31:26 +08:00
|
|
|
Py_DECREF(seq);
|
2023-02-08 07:44:37 +08:00
|
|
|
STACK_SHRINK(1);
|
|
|
|
STACK_GROW(oparg);
|
2023-03-01 00:49:35 +08:00
|
|
|
next_instr += 1;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(UNPACK_SEQUENCE_TUPLE) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *seq;
|
|
|
|
PyObject **values;
|
|
|
|
seq = stack_pointer[-1];
|
|
|
|
values = stack_pointer - 1;
|
2022-11-03 12:31:26 +08:00
|
|
|
DEOPT_IF(!PyTuple_CheckExact(seq), UNPACK_SEQUENCE);
|
|
|
|
DEOPT_IF(PyTuple_GET_SIZE(seq) != oparg, UNPACK_SEQUENCE);
|
|
|
|
STAT_INC(UNPACK_SEQUENCE, hit);
|
|
|
|
PyObject **items = _PyTuple_ITEMS(seq);
|
2023-02-08 07:44:37 +08:00
|
|
|
for (int i = oparg; --i >= 0; ) {
|
|
|
|
*values++ = Py_NewRef(items[i]);
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
|
|
|
Py_DECREF(seq);
|
2023-02-08 07:44:37 +08:00
|
|
|
STACK_SHRINK(1);
|
|
|
|
STACK_GROW(oparg);
|
2023-03-01 00:49:35 +08:00
|
|
|
next_instr += 1;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(UNPACK_SEQUENCE_LIST) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *seq;
|
|
|
|
PyObject **values;
|
|
|
|
seq = stack_pointer[-1];
|
|
|
|
values = stack_pointer - 1;
|
2022-11-03 12:31:26 +08:00
|
|
|
DEOPT_IF(!PyList_CheckExact(seq), UNPACK_SEQUENCE);
|
|
|
|
DEOPT_IF(PyList_GET_SIZE(seq) != oparg, UNPACK_SEQUENCE);
|
|
|
|
STAT_INC(UNPACK_SEQUENCE, hit);
|
|
|
|
PyObject **items = _PyList_ITEMS(seq);
|
2023-02-08 07:44:37 +08:00
|
|
|
for (int i = oparg; --i >= 0; ) {
|
|
|
|
*values++ = Py_NewRef(items[i]);
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
|
|
|
Py_DECREF(seq);
|
2023-02-08 07:44:37 +08:00
|
|
|
STACK_SHRINK(1);
|
|
|
|
STACK_GROW(oparg);
|
2023-03-01 00:49:35 +08:00
|
|
|
next_instr += 1;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(UNPACK_EX) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *seq;
|
|
|
|
seq = stack_pointer[-1];
|
2022-11-03 12:31:26 +08:00
|
|
|
int totalargs = 1 + (oparg & 0xFF) + (oparg >> 8);
|
2023-02-08 04:37:43 +08:00
|
|
|
PyObject **top = stack_pointer + totalargs - 1;
|
2023-07-21 04:37:19 +08:00
|
|
|
int res = _PyEval_UnpackIterable(tstate, seq, oparg & 0xFF, oparg >> 8, top);
|
2022-11-03 12:31:26 +08:00
|
|
|
Py_DECREF(seq);
|
2023-02-08 04:37:43 +08:00
|
|
|
if (res == 0) goto pop_1_error;
|
|
|
|
STACK_GROW((oparg & 0xFF) + (oparg >> 8));
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(STORE_ATTR) {
|
|
|
|
PREDICTED(STORE_ATTR);
|
2023-02-02 02:56:52 +08:00
|
|
|
static_assert(INLINE_CACHE_ENTRIES_STORE_ATTR == 4, "incorrect cache size");
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *owner;
|
|
|
|
PyObject *v;
|
|
|
|
owner = stack_pointer[-1];
|
|
|
|
v = stack_pointer[-2];
|
2023-01-20 02:14:55 +08:00
|
|
|
#if ENABLE_SPECIALIZATION
|
2023-07-18 03:12:33 +08:00
|
|
|
_PyAttrCache *cache = (_PyAttrCache *)next_instr;
|
|
|
|
if (ADAPTIVE_COUNTER_IS_ZERO(cache->counter)) {
|
2023-06-14 04:42:03 +08:00
|
|
|
PyObject *name = GETITEM(FRAME_CO_NAMES, oparg);
|
2022-11-10 02:50:09 +08:00
|
|
|
next_instr--;
|
2022-11-18 07:09:18 +08:00
|
|
|
_Py_Specialize_StoreAttr(owner, next_instr, name);
|
2022-11-10 02:50:09 +08:00
|
|
|
DISPATCH_SAME_OPARG();
|
|
|
|
}
|
|
|
|
STAT_INC(STORE_ATTR, deferred);
|
|
|
|
DECREMENT_ADAPTIVE_COUNTER(cache->counter);
|
2023-01-20 02:14:55 +08:00
|
|
|
#endif /* ENABLE_SPECIALIZATION */
|
2023-06-14 04:42:03 +08:00
|
|
|
PyObject *name = GETITEM(FRAME_CO_NAMES, oparg);
|
GH-98831: Typed stack effects, and more instructions converted (#99764)
Stack effects can now have a type, e.g. `inst(X, (left, right -- jump/uint64_t)) { ... }`.
Instructions converted to the non-legacy format:
* COMPARE_OP
* COMPARE_OP_FLOAT_JUMP
* COMPARE_OP_INT_JUMP
* COMPARE_OP_STR_JUMP
* STORE_ATTR
* DELETE_ATTR
* STORE_GLOBAL
* STORE_ATTR_INSTANCE_VALUE
* STORE_ATTR_WITH_HINT
* STORE_ATTR_SLOT, and complete the store_attr family
* Complete the store_subscr family: STORE_SUBSCR{,DICT,LIST_INT}
(STORE_SUBSCR was alread half converted,
but wasn't using cache effects yet.)
* DELETE_SUBSCR
* PRINT_EXPR
* INTERPRETER_EXIT (a bit weird, ends in return)
* RETURN_VALUE
* GET_AITER (had to restructure it some)
The original had mysterious `SET_TOP(NULL)` before `goto error`.
I assume those just account for `obj` having been decref'ed,
so I got rid of them in favor of the cleanup implied by `ERROR_IF()`.
* LIST_APPEND (a bit unhappy with it)
* SET_ADD (also a bit unhappy with it)
Various other improvements/refactorings as well.
2022-12-09 05:31:27 +08:00
|
|
|
int err = PyObject_SetAttr(owner, name, v);
|
2022-11-03 12:31:26 +08:00
|
|
|
Py_DECREF(v);
|
|
|
|
Py_DECREF(owner);
|
GH-98831: Typed stack effects, and more instructions converted (#99764)
Stack effects can now have a type, e.g. `inst(X, (left, right -- jump/uint64_t)) { ... }`.
Instructions converted to the non-legacy format:
* COMPARE_OP
* COMPARE_OP_FLOAT_JUMP
* COMPARE_OP_INT_JUMP
* COMPARE_OP_STR_JUMP
* STORE_ATTR
* DELETE_ATTR
* STORE_GLOBAL
* STORE_ATTR_INSTANCE_VALUE
* STORE_ATTR_WITH_HINT
* STORE_ATTR_SLOT, and complete the store_attr family
* Complete the store_subscr family: STORE_SUBSCR{,DICT,LIST_INT}
(STORE_SUBSCR was alread half converted,
but wasn't using cache effects yet.)
* DELETE_SUBSCR
* PRINT_EXPR
* INTERPRETER_EXIT (a bit weird, ends in return)
* RETURN_VALUE
* GET_AITER (had to restructure it some)
The original had mysterious `SET_TOP(NULL)` before `goto error`.
I assume those just account for `obj` having been decref'ed,
so I got rid of them in favor of the cleanup implied by `ERROR_IF()`.
* LIST_APPEND (a bit unhappy with it)
* SET_ADD (also a bit unhappy with it)
Various other improvements/refactorings as well.
2022-12-09 05:31:27 +08:00
|
|
|
if (err) goto pop_2_error;
|
|
|
|
STACK_SHRINK(2);
|
2023-03-01 00:49:35 +08:00
|
|
|
next_instr += 4;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(DELETE_ATTR) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *owner;
|
|
|
|
owner = stack_pointer[-1];
|
2023-06-14 04:42:03 +08:00
|
|
|
PyObject *name = GETITEM(FRAME_CO_NAMES, oparg);
|
2023-07-11 17:38:22 +08:00
|
|
|
int err = PyObject_DelAttr(owner, name);
|
2022-11-03 12:31:26 +08:00
|
|
|
Py_DECREF(owner);
|
GH-98831: Typed stack effects, and more instructions converted (#99764)
Stack effects can now have a type, e.g. `inst(X, (left, right -- jump/uint64_t)) { ... }`.
Instructions converted to the non-legacy format:
* COMPARE_OP
* COMPARE_OP_FLOAT_JUMP
* COMPARE_OP_INT_JUMP
* COMPARE_OP_STR_JUMP
* STORE_ATTR
* DELETE_ATTR
* STORE_GLOBAL
* STORE_ATTR_INSTANCE_VALUE
* STORE_ATTR_WITH_HINT
* STORE_ATTR_SLOT, and complete the store_attr family
* Complete the store_subscr family: STORE_SUBSCR{,DICT,LIST_INT}
(STORE_SUBSCR was alread half converted,
but wasn't using cache effects yet.)
* DELETE_SUBSCR
* PRINT_EXPR
* INTERPRETER_EXIT (a bit weird, ends in return)
* RETURN_VALUE
* GET_AITER (had to restructure it some)
The original had mysterious `SET_TOP(NULL)` before `goto error`.
I assume those just account for `obj` having been decref'ed,
so I got rid of them in favor of the cleanup implied by `ERROR_IF()`.
* LIST_APPEND (a bit unhappy with it)
* SET_ADD (also a bit unhappy with it)
Various other improvements/refactorings as well.
2022-12-09 05:31:27 +08:00
|
|
|
if (err) goto pop_1_error;
|
|
|
|
STACK_SHRINK(1);
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(STORE_GLOBAL) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *v;
|
|
|
|
v = stack_pointer[-1];
|
2023-06-14 04:42:03 +08:00
|
|
|
PyObject *name = GETITEM(FRAME_CO_NAMES, oparg);
|
GH-98831: Typed stack effects, and more instructions converted (#99764)
Stack effects can now have a type, e.g. `inst(X, (left, right -- jump/uint64_t)) { ... }`.
Instructions converted to the non-legacy format:
* COMPARE_OP
* COMPARE_OP_FLOAT_JUMP
* COMPARE_OP_INT_JUMP
* COMPARE_OP_STR_JUMP
* STORE_ATTR
* DELETE_ATTR
* STORE_GLOBAL
* STORE_ATTR_INSTANCE_VALUE
* STORE_ATTR_WITH_HINT
* STORE_ATTR_SLOT, and complete the store_attr family
* Complete the store_subscr family: STORE_SUBSCR{,DICT,LIST_INT}
(STORE_SUBSCR was alread half converted,
but wasn't using cache effects yet.)
* DELETE_SUBSCR
* PRINT_EXPR
* INTERPRETER_EXIT (a bit weird, ends in return)
* RETURN_VALUE
* GET_AITER (had to restructure it some)
The original had mysterious `SET_TOP(NULL)` before `goto error`.
I assume those just account for `obj` having been decref'ed,
so I got rid of them in favor of the cleanup implied by `ERROR_IF()`.
* LIST_APPEND (a bit unhappy with it)
* SET_ADD (also a bit unhappy with it)
Various other improvements/refactorings as well.
2022-12-09 05:31:27 +08:00
|
|
|
int err = PyDict_SetItem(GLOBALS(), name, v);
|
2022-11-03 12:31:26 +08:00
|
|
|
Py_DECREF(v);
|
GH-98831: Typed stack effects, and more instructions converted (#99764)
Stack effects can now have a type, e.g. `inst(X, (left, right -- jump/uint64_t)) { ... }`.
Instructions converted to the non-legacy format:
* COMPARE_OP
* COMPARE_OP_FLOAT_JUMP
* COMPARE_OP_INT_JUMP
* COMPARE_OP_STR_JUMP
* STORE_ATTR
* DELETE_ATTR
* STORE_GLOBAL
* STORE_ATTR_INSTANCE_VALUE
* STORE_ATTR_WITH_HINT
* STORE_ATTR_SLOT, and complete the store_attr family
* Complete the store_subscr family: STORE_SUBSCR{,DICT,LIST_INT}
(STORE_SUBSCR was alread half converted,
but wasn't using cache effects yet.)
* DELETE_SUBSCR
* PRINT_EXPR
* INTERPRETER_EXIT (a bit weird, ends in return)
* RETURN_VALUE
* GET_AITER (had to restructure it some)
The original had mysterious `SET_TOP(NULL)` before `goto error`.
I assume those just account for `obj` having been decref'ed,
so I got rid of them in favor of the cleanup implied by `ERROR_IF()`.
* LIST_APPEND (a bit unhappy with it)
* SET_ADD (also a bit unhappy with it)
Various other improvements/refactorings as well.
2022-12-09 05:31:27 +08:00
|
|
|
if (err) goto pop_1_error;
|
|
|
|
STACK_SHRINK(1);
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(DELETE_GLOBAL) {
|
2023-06-14 04:42:03 +08:00
|
|
|
PyObject *name = GETITEM(FRAME_CO_NAMES, oparg);
|
2022-11-03 12:31:26 +08:00
|
|
|
int err;
|
|
|
|
err = PyDict_DelItem(GLOBALS(), name);
|
2022-11-11 02:50:57 +08:00
|
|
|
// Can't use ERROR_IF here.
|
2022-11-03 12:31:26 +08:00
|
|
|
if (err != 0) {
|
|
|
|
if (_PyErr_ExceptionMatches(tstate, PyExc_KeyError)) {
|
2023-07-21 04:37:19 +08:00
|
|
|
_PyEval_FormatExcCheckArg(tstate, PyExc_NameError,
|
|
|
|
NAME_ERROR_MSG, name);
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
|
|
|
goto error;
|
|
|
|
}
|
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2023-05-16 11:36:23 +08:00
|
|
|
TARGET(LOAD_LOCALS) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *locals;
|
|
|
|
locals = LOCALS();
|
|
|
|
if (locals == NULL) {
|
|
|
|
_PyErr_SetString(tstate, PyExc_SystemError,
|
|
|
|
"no locals found");
|
|
|
|
if (true) goto error;
|
2023-05-16 11:36:23 +08:00
|
|
|
}
|
2023-08-05 00:35:56 +08:00
|
|
|
Py_INCREF(locals);
|
2023-05-16 11:36:23 +08:00
|
|
|
STACK_GROW(1);
|
2023-08-05 00:35:56 +08:00
|
|
|
stack_pointer[-1] = locals;
|
2023-05-16 11:36:23 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2023-09-09 10:49:20 +08:00
|
|
|
TARGET(LOAD_FROM_DICT_OR_GLOBALS) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *mod_or_class_dict;
|
|
|
|
PyObject *v;
|
2023-09-09 10:49:20 +08:00
|
|
|
mod_or_class_dict = stack_pointer[-1];
|
|
|
|
PyObject *name = GETITEM(FRAME_CO_NAMES, oparg);
|
|
|
|
if (PyMapping_GetOptionalItem(mod_or_class_dict, name, &v) < 0) {
|
|
|
|
goto error;
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
2023-09-09 10:49:20 +08:00
|
|
|
if (v == NULL) {
|
|
|
|
v = PyDict_GetItemWithError(GLOBALS(), name);
|
|
|
|
if (v != NULL) {
|
|
|
|
Py_INCREF(v);
|
|
|
|
}
|
|
|
|
else if (_PyErr_Occurred(tstate)) {
|
2023-07-12 04:04:12 +08:00
|
|
|
goto error;
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
2023-09-09 10:49:20 +08:00
|
|
|
else {
|
|
|
|
if (PyMapping_GetOptionalItem(BUILTINS(), name, &v) < 0) {
|
2022-11-03 12:31:26 +08:00
|
|
|
goto error;
|
2023-05-16 11:36:23 +08:00
|
|
|
}
|
2023-09-09 10:49:20 +08:00
|
|
|
if (v == NULL) {
|
|
|
|
_PyEval_FormatExcCheckArg(
|
|
|
|
tstate, PyExc_NameError,
|
|
|
|
NAME_ERROR_MSG, name);
|
|
|
|
goto error;
|
2023-05-16 11:36:23 +08:00
|
|
|
}
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
|
|
|
}
|
2023-09-09 10:49:20 +08:00
|
|
|
Py_DECREF(mod_or_class_dict);
|
2023-08-05 00:35:56 +08:00
|
|
|
stack_pointer[-1] = v;
|
2023-05-16 11:36:23 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2023-09-09 10:49:20 +08:00
|
|
|
TARGET(LOAD_NAME) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *v;
|
2023-09-09 10:49:20 +08:00
|
|
|
PyObject *mod_or_class_dict = LOCALS();
|
|
|
|
if (mod_or_class_dict == NULL) {
|
|
|
|
_PyErr_SetString(tstate, PyExc_SystemError,
|
|
|
|
"no locals found");
|
|
|
|
if (true) goto error;
|
|
|
|
}
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *name = GETITEM(FRAME_CO_NAMES, oparg);
|
|
|
|
if (PyMapping_GetOptionalItem(mod_or_class_dict, name, &v) < 0) {
|
|
|
|
goto error;
|
|
|
|
}
|
|
|
|
if (v == NULL) {
|
|
|
|
v = PyDict_GetItemWithError(GLOBALS(), name);
|
|
|
|
if (v != NULL) {
|
|
|
|
Py_INCREF(v);
|
|
|
|
}
|
|
|
|
else if (_PyErr_Occurred(tstate)) {
|
2023-07-12 04:04:12 +08:00
|
|
|
goto error;
|
2023-05-16 11:36:23 +08:00
|
|
|
}
|
2023-08-05 00:35:56 +08:00
|
|
|
else {
|
|
|
|
if (PyMapping_GetOptionalItem(BUILTINS(), name, &v) < 0) {
|
2023-05-16 11:36:23 +08:00
|
|
|
goto error;
|
|
|
|
}
|
2023-08-05 00:35:56 +08:00
|
|
|
if (v == NULL) {
|
|
|
|
_PyEval_FormatExcCheckArg(
|
|
|
|
tstate, PyExc_NameError,
|
|
|
|
NAME_ERROR_MSG, name);
|
|
|
|
goto error;
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2023-09-09 10:49:20 +08:00
|
|
|
STACK_GROW(1);
|
2023-08-05 00:35:56 +08:00
|
|
|
stack_pointer[-1] = v;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(LOAD_GLOBAL) {
|
|
|
|
PREDICTED(LOAD_GLOBAL);
|
2023-03-11 09:01:16 +08:00
|
|
|
static_assert(INLINE_CACHE_ENTRIES_LOAD_GLOBAL == 4, "incorrect cache size");
|
2023-08-10 02:19:39 +08:00
|
|
|
PyObject *res;
|
2023-02-02 05:12:49 +08:00
|
|
|
PyObject *null = NULL;
|
2023-01-20 02:14:55 +08:00
|
|
|
#if ENABLE_SPECIALIZATION
|
2022-11-10 02:50:09 +08:00
|
|
|
_PyLoadGlobalCache *cache = (_PyLoadGlobalCache *)next_instr;
|
|
|
|
if (ADAPTIVE_COUNTER_IS_ZERO(cache->counter)) {
|
2023-06-14 04:42:03 +08:00
|
|
|
PyObject *name = GETITEM(FRAME_CO_NAMES, oparg>>1);
|
2022-11-10 02:50:09 +08:00
|
|
|
next_instr--;
|
|
|
|
_Py_Specialize_LoadGlobal(GLOBALS(), BUILTINS(), next_instr, name);
|
|
|
|
DISPATCH_SAME_OPARG();
|
|
|
|
}
|
|
|
|
STAT_INC(LOAD_GLOBAL, deferred);
|
|
|
|
DECREMENT_ADAPTIVE_COUNTER(cache->counter);
|
2023-01-20 02:14:55 +08:00
|
|
|
#endif /* ENABLE_SPECIALIZATION */
|
2023-06-14 04:42:03 +08:00
|
|
|
PyObject *name = GETITEM(FRAME_CO_NAMES, oparg>>1);
|
2022-11-03 12:31:26 +08:00
|
|
|
if (PyDict_CheckExact(GLOBALS())
|
|
|
|
&& PyDict_CheckExact(BUILTINS()))
|
|
|
|
{
|
2023-08-10 02:19:39 +08:00
|
|
|
res = _PyDict_LoadGlobal((PyDictObject *)GLOBALS(),
|
|
|
|
(PyDictObject *)BUILTINS(),
|
|
|
|
name);
|
|
|
|
if (res == NULL) {
|
2022-11-03 12:31:26 +08:00
|
|
|
if (!_PyErr_Occurred(tstate)) {
|
|
|
|
/* _PyDict_LoadGlobal() returns NULL without raising
|
|
|
|
* an exception if the key doesn't exist */
|
2023-07-21 04:37:19 +08:00
|
|
|
_PyEval_FormatExcCheckArg(tstate, PyExc_NameError,
|
|
|
|
NAME_ERROR_MSG, name);
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
2023-02-02 05:12:49 +08:00
|
|
|
if (true) goto error;
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
2023-08-10 02:19:39 +08:00
|
|
|
Py_INCREF(res);
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
|
|
|
else {
|
|
|
|
/* Slow-path if globals or builtins is not a dict */
|
|
|
|
|
|
|
|
/* namespace 1: globals */
|
2023-08-10 02:19:39 +08:00
|
|
|
if (PyMapping_GetOptionalItem(GLOBALS(), name, &res) < 0) goto error;
|
|
|
|
if (res == NULL) {
|
2022-11-03 12:31:26 +08:00
|
|
|
/* namespace 2: builtins */
|
2023-08-10 02:19:39 +08:00
|
|
|
if (PyMapping_GetOptionalItem(BUILTINS(), name, &res) < 0) goto error;
|
|
|
|
if (res == NULL) {
|
2023-07-21 04:37:19 +08:00
|
|
|
_PyEval_FormatExcCheckArg(
|
2023-07-12 04:04:12 +08:00
|
|
|
tstate, PyExc_NameError,
|
|
|
|
NAME_ERROR_MSG, name);
|
2023-02-02 05:12:49 +08:00
|
|
|
if (true) goto error;
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2023-02-02 05:12:49 +08:00
|
|
|
null = NULL;
|
|
|
|
STACK_GROW(1);
|
|
|
|
STACK_GROW(((oparg & 1) ? 1 : 0));
|
2023-08-10 02:19:39 +08:00
|
|
|
stack_pointer[-1 - (oparg & 1 ? 1 : 0)] = res;
|
|
|
|
if (oparg & 1) { stack_pointer[-(oparg & 1 ? 1 : 0)] = null; }
|
2023-03-11 09:01:16 +08:00
|
|
|
next_instr += 4;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(LOAD_GLOBAL_MODULE) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *res;
|
2023-08-10 02:19:39 +08:00
|
|
|
PyObject *null = NULL;
|
2023-08-05 00:35:56 +08:00
|
|
|
// _GUARD_GLOBALS_VERSION
|
2023-07-12 21:34:14 +08:00
|
|
|
{
|
|
|
|
uint16_t version = read_u16(&next_instr[1].cache);
|
|
|
|
PyDictObject *dict = (PyDictObject *)GLOBALS();
|
|
|
|
DEOPT_IF(!PyDict_CheckExact(dict), LOAD_GLOBAL);
|
|
|
|
DEOPT_IF(dict->ma_keys->dk_version != version, LOAD_GLOBAL);
|
|
|
|
assert(DK_IS_UNICODE(dict->ma_keys));
|
|
|
|
}
|
2023-08-05 00:35:56 +08:00
|
|
|
// _LOAD_GLOBAL_MODULE
|
2023-07-12 21:34:14 +08:00
|
|
|
{
|
|
|
|
uint16_t index = read_u16(&next_instr[3].cache);
|
|
|
|
PyDictObject *dict = (PyDictObject *)GLOBALS();
|
|
|
|
PyDictUnicodeEntry *entries = DK_UNICODE_ENTRIES(dict->ma_keys);
|
|
|
|
res = entries[index].me_value;
|
|
|
|
DEOPT_IF(res == NULL, LOAD_GLOBAL);
|
|
|
|
Py_INCREF(res);
|
|
|
|
STAT_INC(LOAD_GLOBAL, hit);
|
|
|
|
null = NULL;
|
|
|
|
}
|
2023-02-02 05:12:49 +08:00
|
|
|
STACK_GROW(1);
|
|
|
|
STACK_GROW(((oparg & 1) ? 1 : 0));
|
2023-08-10 02:19:39 +08:00
|
|
|
stack_pointer[-1 - (oparg & 1 ? 1 : 0)] = res;
|
|
|
|
if (oparg & 1) { stack_pointer[-(oparg & 1 ? 1 : 0)] = null; }
|
2023-08-05 00:35:56 +08:00
|
|
|
next_instr += 4;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(LOAD_GLOBAL_BUILTIN) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *res;
|
2023-08-10 02:19:39 +08:00
|
|
|
PyObject *null = NULL;
|
2023-08-05 00:35:56 +08:00
|
|
|
// _GUARD_GLOBALS_VERSION
|
2023-07-12 21:34:14 +08:00
|
|
|
{
|
|
|
|
uint16_t version = read_u16(&next_instr[1].cache);
|
|
|
|
PyDictObject *dict = (PyDictObject *)GLOBALS();
|
|
|
|
DEOPT_IF(!PyDict_CheckExact(dict), LOAD_GLOBAL);
|
|
|
|
DEOPT_IF(dict->ma_keys->dk_version != version, LOAD_GLOBAL);
|
|
|
|
assert(DK_IS_UNICODE(dict->ma_keys));
|
|
|
|
}
|
2023-08-05 00:35:56 +08:00
|
|
|
// _GUARD_BUILTINS_VERSION
|
2023-07-12 21:34:14 +08:00
|
|
|
{
|
|
|
|
uint16_t version = read_u16(&next_instr[2].cache);
|
|
|
|
PyDictObject *dict = (PyDictObject *)BUILTINS();
|
|
|
|
DEOPT_IF(!PyDict_CheckExact(dict), LOAD_GLOBAL);
|
|
|
|
DEOPT_IF(dict->ma_keys->dk_version != version, LOAD_GLOBAL);
|
|
|
|
assert(DK_IS_UNICODE(dict->ma_keys));
|
|
|
|
}
|
2023-08-05 00:35:56 +08:00
|
|
|
// _LOAD_GLOBAL_BUILTINS
|
2023-07-12 21:34:14 +08:00
|
|
|
{
|
|
|
|
uint16_t index = read_u16(&next_instr[3].cache);
|
|
|
|
PyDictObject *bdict = (PyDictObject *)BUILTINS();
|
|
|
|
PyDictUnicodeEntry *entries = DK_UNICODE_ENTRIES(bdict->ma_keys);
|
|
|
|
res = entries[index].me_value;
|
|
|
|
DEOPT_IF(res == NULL, LOAD_GLOBAL);
|
|
|
|
Py_INCREF(res);
|
|
|
|
STAT_INC(LOAD_GLOBAL, hit);
|
|
|
|
null = NULL;
|
|
|
|
}
|
2023-02-02 05:12:49 +08:00
|
|
|
STACK_GROW(1);
|
|
|
|
STACK_GROW(((oparg & 1) ? 1 : 0));
|
2023-08-10 02:19:39 +08:00
|
|
|
stack_pointer[-1 - (oparg & 1 ? 1 : 0)] = res;
|
|
|
|
if (oparg & 1) { stack_pointer[-(oparg & 1 ? 1 : 0)] = null; }
|
2023-08-05 00:35:56 +08:00
|
|
|
next_instr += 4;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(DELETE_FAST) {
|
|
|
|
PyObject *v = GETLOCAL(oparg);
|
2022-11-11 02:50:57 +08:00
|
|
|
if (v == NULL) goto unbound_local_error;
|
|
|
|
SETLOCAL(oparg, NULL);
|
|
|
|
DISPATCH();
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(MAKE_CELL) {
|
|
|
|
// "initial" is probably NULL but not if it's an arg (or set
|
|
|
|
// via PyFrame_LocalsToFast() before MAKE_CELL has run).
|
|
|
|
PyObject *initial = GETLOCAL(oparg);
|
|
|
|
PyObject *cell = PyCell_New(initial);
|
|
|
|
if (cell == NULL) {
|
|
|
|
goto resume_with_error;
|
|
|
|
}
|
|
|
|
SETLOCAL(oparg, cell);
|
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(DELETE_DEREF) {
|
|
|
|
PyObject *cell = GETLOCAL(oparg);
|
|
|
|
PyObject *oldobj = PyCell_GET(cell);
|
2022-11-11 02:50:57 +08:00
|
|
|
// Can't use ERROR_IF here.
|
|
|
|
// Fortunately we don't need its superpower.
|
|
|
|
if (oldobj == NULL) {
|
2023-07-21 04:37:19 +08:00
|
|
|
_PyEval_FormatExcUnbound(tstate, _PyFrame_GetCode(frame), oparg);
|
2022-11-11 02:50:57 +08:00
|
|
|
goto error;
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
2022-11-11 02:50:57 +08:00
|
|
|
PyCell_SET(cell, NULL);
|
|
|
|
Py_DECREF(oldobj);
|
|
|
|
DISPATCH();
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
|
|
|
|
2023-05-16 11:36:23 +08:00
|
|
|
TARGET(LOAD_FROM_DICT_OR_DEREF) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *class_dict;
|
2022-12-28 09:11:03 +08:00
|
|
|
PyObject *value;
|
2023-08-05 00:35:56 +08:00
|
|
|
class_dict = stack_pointer[-1];
|
2023-05-16 11:36:23 +08:00
|
|
|
PyObject *name;
|
|
|
|
assert(class_dict);
|
2023-06-14 20:46:37 +08:00
|
|
|
assert(oparg >= 0 && oparg < _PyFrame_GetCode(frame)->co_nlocalsplus);
|
|
|
|
name = PyTuple_GET_ITEM(_PyFrame_GetCode(frame)->co_localsplusnames, oparg);
|
2023-07-12 04:04:12 +08:00
|
|
|
if (PyMapping_GetOptionalItem(class_dict, name, &value) < 0) {
|
|
|
|
Py_DECREF(class_dict);
|
|
|
|
goto error;
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
2023-05-16 11:36:23 +08:00
|
|
|
Py_DECREF(class_dict);
|
2022-11-03 12:31:26 +08:00
|
|
|
if (!value) {
|
|
|
|
PyObject *cell = GETLOCAL(oparg);
|
|
|
|
value = PyCell_GET(cell);
|
|
|
|
if (value == NULL) {
|
2023-07-21 04:37:19 +08:00
|
|
|
_PyEval_FormatExcUnbound(tstate, _PyFrame_GetCode(frame), oparg);
|
2022-11-03 12:31:26 +08:00
|
|
|
goto error;
|
|
|
|
}
|
|
|
|
Py_INCREF(value);
|
|
|
|
}
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = value;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(LOAD_DEREF) {
|
2022-12-28 09:11:03 +08:00
|
|
|
PyObject *value;
|
2022-11-03 12:31:26 +08:00
|
|
|
PyObject *cell = GETLOCAL(oparg);
|
2022-12-28 09:11:03 +08:00
|
|
|
value = PyCell_GET(cell);
|
2022-11-03 12:31:26 +08:00
|
|
|
if (value == NULL) {
|
2023-07-21 04:37:19 +08:00
|
|
|
_PyEval_FormatExcUnbound(tstate, _PyFrame_GetCode(frame), oparg);
|
2022-12-28 09:11:03 +08:00
|
|
|
if (true) goto error;
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
2022-12-28 09:11:03 +08:00
|
|
|
Py_INCREF(value);
|
|
|
|
STACK_GROW(1);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = value;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(STORE_DEREF) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *v;
|
|
|
|
v = stack_pointer[-1];
|
2022-11-03 12:31:26 +08:00
|
|
|
PyObject *cell = GETLOCAL(oparg);
|
|
|
|
PyObject *oldobj = PyCell_GET(cell);
|
|
|
|
PyCell_SET(cell, v);
|
|
|
|
Py_XDECREF(oldobj);
|
2022-12-28 09:11:03 +08:00
|
|
|
STACK_SHRINK(1);
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(COPY_FREE_VARS) {
|
|
|
|
/* Copy closure variables to free variables */
|
2023-06-14 20:46:37 +08:00
|
|
|
PyCodeObject *co = _PyFrame_GetCode(frame);
|
2022-11-03 12:31:26 +08:00
|
|
|
assert(PyFunction_Check(frame->f_funcobj));
|
|
|
|
PyObject *closure = ((PyFunctionObject *)frame->f_funcobj)->func_closure;
|
|
|
|
assert(oparg == co->co_nfreevars);
|
2023-01-04 23:41:39 +08:00
|
|
|
int offset = co->co_nlocalsplus - oparg;
|
2022-11-03 12:31:26 +08:00
|
|
|
for (int i = 0; i < oparg; ++i) {
|
|
|
|
PyObject *o = PyTuple_GET_ITEM(closure, i);
|
2022-11-10 18:25:33 +08:00
|
|
|
frame->localsplus[offset + i] = Py_NewRef(o);
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(BUILD_STRING) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject **pieces;
|
2022-11-03 12:31:26 +08:00
|
|
|
PyObject *str;
|
2023-08-05 00:35:56 +08:00
|
|
|
pieces = stack_pointer - oparg;
|
2023-01-18 07:59:19 +08:00
|
|
|
str = _PyUnicode_JoinArray(&_Py_STR(empty), pieces, oparg);
|
2023-03-14 06:08:45 +08:00
|
|
|
for (int _i = oparg; --_i >= 0;) {
|
|
|
|
Py_DECREF(pieces[_i]);
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
2023-01-18 07:59:19 +08:00
|
|
|
if (str == NULL) { STACK_SHRINK(oparg); goto error; }
|
|
|
|
STACK_SHRINK(oparg);
|
|
|
|
STACK_GROW(1);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = str;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(BUILD_TUPLE) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject **values;
|
2023-01-18 07:59:19 +08:00
|
|
|
PyObject *tup;
|
2023-08-05 00:35:56 +08:00
|
|
|
values = stack_pointer - oparg;
|
2023-01-18 07:59:19 +08:00
|
|
|
tup = _PyTuple_FromArraySteal(values, oparg);
|
|
|
|
if (tup == NULL) { STACK_SHRINK(oparg); goto error; }
|
2022-11-03 12:31:26 +08:00
|
|
|
STACK_SHRINK(oparg);
|
2023-01-18 07:59:19 +08:00
|
|
|
STACK_GROW(1);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = tup;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(BUILD_LIST) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject **values;
|
2023-01-18 07:59:19 +08:00
|
|
|
PyObject *list;
|
2023-08-05 00:35:56 +08:00
|
|
|
values = stack_pointer - oparg;
|
2023-01-18 07:59:19 +08:00
|
|
|
list = _PyList_FromArraySteal(values, oparg);
|
|
|
|
if (list == NULL) { STACK_SHRINK(oparg); goto error; }
|
2023-01-04 02:49:49 +08:00
|
|
|
STACK_SHRINK(oparg);
|
2023-01-18 07:59:19 +08:00
|
|
|
STACK_GROW(1);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = list;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(LIST_EXTEND) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *iterable;
|
|
|
|
PyObject *list;
|
|
|
|
iterable = stack_pointer[-1];
|
|
|
|
list = stack_pointer[-2 - (oparg-1)];
|
2022-11-03 12:31:26 +08:00
|
|
|
PyObject *none_val = _PyList_Extend((PyListObject *)list, iterable);
|
|
|
|
if (none_val == NULL) {
|
|
|
|
if (_PyErr_ExceptionMatches(tstate, PyExc_TypeError) &&
|
|
|
|
(Py_TYPE(iterable)->tp_iter == NULL && !PySequence_Check(iterable)))
|
|
|
|
{
|
|
|
|
_PyErr_Clear(tstate);
|
|
|
|
_PyErr_Format(tstate, PyExc_TypeError,
|
|
|
|
"Value after * must be an iterable, not %.200s",
|
|
|
|
Py_TYPE(iterable)->tp_name);
|
|
|
|
}
|
|
|
|
Py_DECREF(iterable);
|
2022-12-28 09:11:03 +08:00
|
|
|
if (true) goto pop_1_error;
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
2023-05-17 05:36:02 +08:00
|
|
|
assert(Py_IsNone(none_val));
|
2022-11-03 12:31:26 +08:00
|
|
|
Py_DECREF(iterable);
|
2022-12-28 09:11:03 +08:00
|
|
|
STACK_SHRINK(1);
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(SET_UPDATE) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *iterable;
|
|
|
|
PyObject *set;
|
|
|
|
iterable = stack_pointer[-1];
|
|
|
|
set = stack_pointer[-2 - (oparg-1)];
|
2022-11-03 12:31:26 +08:00
|
|
|
int err = _PySet_Update(set, iterable);
|
|
|
|
Py_DECREF(iterable);
|
2022-12-28 09:11:03 +08:00
|
|
|
if (err < 0) goto pop_1_error;
|
|
|
|
STACK_SHRINK(1);
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(BUILD_SET) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject **values;
|
2023-01-18 07:59:19 +08:00
|
|
|
PyObject *set;
|
2023-08-05 00:35:56 +08:00
|
|
|
values = stack_pointer - oparg;
|
2023-01-18 07:59:19 +08:00
|
|
|
set = PySet_New(NULL);
|
2023-02-17 01:46:43 +08:00
|
|
|
if (set == NULL)
|
|
|
|
goto error;
|
2022-11-03 12:31:26 +08:00
|
|
|
int err = 0;
|
2023-01-18 07:59:19 +08:00
|
|
|
for (int i = 0; i < oparg; i++) {
|
|
|
|
PyObject *item = values[i];
|
2022-11-03 12:31:26 +08:00
|
|
|
if (err == 0)
|
|
|
|
err = PySet_Add(set, item);
|
|
|
|
Py_DECREF(item);
|
|
|
|
}
|
|
|
|
if (err != 0) {
|
|
|
|
Py_DECREF(set);
|
2023-01-18 07:59:19 +08:00
|
|
|
if (true) { STACK_SHRINK(oparg); goto error; }
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
2023-01-18 07:59:19 +08:00
|
|
|
STACK_SHRINK(oparg);
|
|
|
|
STACK_GROW(1);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = set;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(BUILD_MAP) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject **values;
|
2023-01-18 07:59:19 +08:00
|
|
|
PyObject *map;
|
2023-08-05 00:35:56 +08:00
|
|
|
values = stack_pointer - oparg*2;
|
2023-01-18 07:59:19 +08:00
|
|
|
map = _PyDict_FromItems(
|
|
|
|
values, 2,
|
|
|
|
values+1, 2,
|
2022-11-03 12:31:26 +08:00
|
|
|
oparg);
|
2023-03-14 06:08:45 +08:00
|
|
|
for (int _i = oparg*2; --_i >= 0;) {
|
|
|
|
Py_DECREF(values[_i]);
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
2023-01-18 07:59:19 +08:00
|
|
|
if (map == NULL) { STACK_SHRINK(oparg*2); goto error; }
|
|
|
|
STACK_SHRINK(oparg*2);
|
|
|
|
STACK_GROW(1);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = map;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(SETUP_ANNOTATIONS) {
|
|
|
|
int err;
|
|
|
|
PyObject *ann_dict;
|
|
|
|
if (LOCALS() == NULL) {
|
|
|
|
_PyErr_Format(tstate, PyExc_SystemError,
|
|
|
|
"no locals found when setting up annotations");
|
2022-12-28 09:11:03 +08:00
|
|
|
if (true) goto error;
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
|
|
|
/* check if __annotations__ in locals()... */
|
|
|
|
if (PyDict_CheckExact(LOCALS())) {
|
|
|
|
ann_dict = _PyDict_GetItemWithError(LOCALS(),
|
|
|
|
&_Py_ID(__annotations__));
|
|
|
|
if (ann_dict == NULL) {
|
2022-12-28 09:11:03 +08:00
|
|
|
if (_PyErr_Occurred(tstate)) goto error;
|
2022-11-03 12:31:26 +08:00
|
|
|
/* ...if not, create a new one */
|
|
|
|
ann_dict = PyDict_New();
|
2022-12-28 09:11:03 +08:00
|
|
|
if (ann_dict == NULL) goto error;
|
2022-11-03 12:31:26 +08:00
|
|
|
err = PyDict_SetItem(LOCALS(), &_Py_ID(__annotations__),
|
|
|
|
ann_dict);
|
|
|
|
Py_DECREF(ann_dict);
|
2022-12-28 09:11:03 +08:00
|
|
|
if (err) goto error;
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
else {
|
|
|
|
/* do the same if locals() is not a dict */
|
2023-07-12 04:04:12 +08:00
|
|
|
if (PyMapping_GetOptionalItem(LOCALS(), &_Py_ID(__annotations__), &ann_dict) < 0) goto error;
|
2022-11-03 12:31:26 +08:00
|
|
|
if (ann_dict == NULL) {
|
|
|
|
ann_dict = PyDict_New();
|
2022-12-28 09:11:03 +08:00
|
|
|
if (ann_dict == NULL) goto error;
|
2022-11-03 12:31:26 +08:00
|
|
|
err = PyObject_SetItem(LOCALS(), &_Py_ID(__annotations__),
|
|
|
|
ann_dict);
|
|
|
|
Py_DECREF(ann_dict);
|
2022-12-28 09:11:03 +08:00
|
|
|
if (err) goto error;
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
|
|
|
else {
|
|
|
|
Py_DECREF(ann_dict);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(BUILD_CONST_KEY_MAP) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *keys;
|
|
|
|
PyObject **values;
|
2022-11-03 12:31:26 +08:00
|
|
|
PyObject *map;
|
2023-08-05 00:35:56 +08:00
|
|
|
keys = stack_pointer[-1];
|
|
|
|
values = stack_pointer - 1 - oparg;
|
2022-11-03 12:31:26 +08:00
|
|
|
if (!PyTuple_CheckExact(keys) ||
|
|
|
|
PyTuple_GET_SIZE(keys) != (Py_ssize_t)oparg) {
|
|
|
|
_PyErr_SetString(tstate, PyExc_SystemError,
|
|
|
|
"bad BUILD_CONST_KEY_MAP keys argument");
|
2023-01-18 07:59:19 +08:00
|
|
|
goto error; // Pop the keys and values.
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
|
|
|
map = _PyDict_FromItems(
|
|
|
|
&PyTuple_GET_ITEM(keys, 0), 1,
|
2023-01-18 07:59:19 +08:00
|
|
|
values, 1, oparg);
|
2023-03-14 06:08:45 +08:00
|
|
|
for (int _i = oparg; --_i >= 0;) {
|
|
|
|
Py_DECREF(values[_i]);
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
2023-03-14 06:08:45 +08:00
|
|
|
Py_DECREF(keys);
|
2023-01-18 07:59:19 +08:00
|
|
|
if (map == NULL) { STACK_SHRINK(oparg); goto pop_1_error; }
|
|
|
|
STACK_SHRINK(oparg);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = map;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(DICT_UPDATE) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *update;
|
2023-08-10 02:19:39 +08:00
|
|
|
PyObject *dict;
|
2023-08-05 00:35:56 +08:00
|
|
|
update = stack_pointer[-1];
|
2023-08-10 02:19:39 +08:00
|
|
|
dict = stack_pointer[-2 - (oparg - 1)];
|
2022-11-03 12:31:26 +08:00
|
|
|
if (PyDict_Update(dict, update) < 0) {
|
|
|
|
if (_PyErr_ExceptionMatches(tstate, PyExc_AttributeError)) {
|
|
|
|
_PyErr_Format(tstate, PyExc_TypeError,
|
|
|
|
"'%.200s' object is not a mapping",
|
|
|
|
Py_TYPE(update)->tp_name);
|
|
|
|
}
|
|
|
|
Py_DECREF(update);
|
2022-12-28 09:11:03 +08:00
|
|
|
if (true) goto pop_1_error;
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
|
|
|
Py_DECREF(update);
|
2022-12-28 09:11:03 +08:00
|
|
|
STACK_SHRINK(1);
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(DICT_MERGE) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *update;
|
2023-08-10 02:19:39 +08:00
|
|
|
PyObject *dict;
|
|
|
|
PyObject *callable;
|
2023-08-05 00:35:56 +08:00
|
|
|
update = stack_pointer[-1];
|
2023-08-10 02:19:39 +08:00
|
|
|
dict = stack_pointer[-2 - (oparg - 1)];
|
|
|
|
callable = stack_pointer[-5 - (oparg - 1)];
|
2022-11-03 12:31:26 +08:00
|
|
|
if (_PyDict_MergeEx(dict, update, 2) < 0) {
|
2023-08-10 02:19:39 +08:00
|
|
|
_PyEval_FormatKwargsError(tstate, callable, update);
|
2022-11-03 12:31:26 +08:00
|
|
|
Py_DECREF(update);
|
2022-12-28 09:11:03 +08:00
|
|
|
if (true) goto pop_1_error;
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
|
|
|
Py_DECREF(update);
|
2022-12-28 09:11:03 +08:00
|
|
|
STACK_SHRINK(1);
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(MAP_ADD) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *value;
|
|
|
|
PyObject *key;
|
2023-08-10 02:19:39 +08:00
|
|
|
PyObject *dict;
|
2023-08-05 00:35:56 +08:00
|
|
|
value = stack_pointer[-1];
|
|
|
|
key = stack_pointer[-2];
|
2023-08-10 02:19:39 +08:00
|
|
|
dict = stack_pointer[-3 - (oparg - 1)];
|
2022-12-28 09:11:03 +08:00
|
|
|
assert(PyDict_CheckExact(dict));
|
|
|
|
/* dict[key] = value */
|
|
|
|
// Do not DECREF INPUTS because the function steals the references
|
|
|
|
if (_PyDict_SetItem_Take2((PyDictObject *)dict, key, value) != 0) goto pop_2_error;
|
2022-11-03 12:31:26 +08:00
|
|
|
STACK_SHRINK(2);
|
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2023-05-17 00:29:00 +08:00
|
|
|
TARGET(INSTRUMENTED_LOAD_SUPER_ATTR) {
|
|
|
|
_PySuperAttrCache *cache = (_PySuperAttrCache *)next_instr;
|
|
|
|
// cancel out the decrement that will happen in LOAD_SUPER_ATTR; we
|
|
|
|
// don't want to specialize instrumented instructions
|
|
|
|
INCREMENT_ADAPTIVE_COUNTER(cache->counter);
|
|
|
|
GO_TO_INSTRUCTION(LOAD_SUPER_ATTR);
|
|
|
|
}
|
|
|
|
|
2023-04-25 06:22:14 +08:00
|
|
|
TARGET(LOAD_SUPER_ATTR) {
|
2023-04-26 01:45:51 +08:00
|
|
|
PREDICTED(LOAD_SUPER_ATTR);
|
2023-05-11 22:08:13 +08:00
|
|
|
static_assert(INLINE_CACHE_ENTRIES_LOAD_SUPER_ATTR == 1, "incorrect cache size");
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *self;
|
|
|
|
PyObject *class;
|
|
|
|
PyObject *global_super;
|
2023-08-10 02:19:39 +08:00
|
|
|
PyObject *attr;
|
|
|
|
PyObject *null = NULL;
|
2023-08-05 00:35:56 +08:00
|
|
|
self = stack_pointer[-1];
|
|
|
|
class = stack_pointer[-2];
|
|
|
|
global_super = stack_pointer[-3];
|
2023-06-14 04:42:03 +08:00
|
|
|
PyObject *name = GETITEM(FRAME_CO_NAMES, oparg >> 2);
|
2023-04-26 01:45:51 +08:00
|
|
|
int load_method = oparg & 1;
|
|
|
|
#if ENABLE_SPECIALIZATION
|
|
|
|
_PySuperAttrCache *cache = (_PySuperAttrCache *)next_instr;
|
|
|
|
if (ADAPTIVE_COUNTER_IS_ZERO(cache->counter)) {
|
|
|
|
next_instr--;
|
2023-05-11 22:08:13 +08:00
|
|
|
_Py_Specialize_LoadSuperAttr(global_super, class, next_instr, load_method);
|
2023-04-26 01:45:51 +08:00
|
|
|
DISPATCH_SAME_OPARG();
|
|
|
|
}
|
|
|
|
STAT_INC(LOAD_SUPER_ATTR, deferred);
|
|
|
|
DECREMENT_ADAPTIVE_COUNTER(cache->counter);
|
|
|
|
#endif /* ENABLE_SPECIALIZATION */
|
|
|
|
|
2023-05-17 00:29:00 +08:00
|
|
|
if (opcode == INSTRUMENTED_LOAD_SUPER_ATTR) {
|
|
|
|
PyObject *arg = oparg & 2 ? class : &_PyInstrumentation_MISSING;
|
|
|
|
int err = _Py_call_instrumentation_2args(
|
|
|
|
tstate, PY_MONITORING_EVENT_CALL,
|
|
|
|
frame, next_instr-1, global_super, arg);
|
|
|
|
if (err) goto pop_3_error;
|
|
|
|
}
|
|
|
|
|
2023-04-26 01:45:51 +08:00
|
|
|
// we make no attempt to optimize here; specializations should
|
|
|
|
// handle any case whose performance we care about
|
|
|
|
PyObject *stack[] = {class, self};
|
|
|
|
PyObject *super = PyObject_Vectorcall(global_super, stack, oparg & 2, NULL);
|
2023-05-17 00:29:00 +08:00
|
|
|
if (opcode == INSTRUMENTED_LOAD_SUPER_ATTR) {
|
|
|
|
PyObject *arg = oparg & 2 ? class : &_PyInstrumentation_MISSING;
|
|
|
|
if (super == NULL) {
|
|
|
|
_Py_call_instrumentation_exc2(
|
|
|
|
tstate, PY_MONITORING_EVENT_C_RAISE,
|
|
|
|
frame, next_instr-1, global_super, arg);
|
|
|
|
}
|
|
|
|
else {
|
|
|
|
int err = _Py_call_instrumentation_2args(
|
|
|
|
tstate, PY_MONITORING_EVENT_C_RETURN,
|
|
|
|
frame, next_instr-1, global_super, arg);
|
|
|
|
if (err < 0) {
|
|
|
|
Py_CLEAR(super);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2023-04-26 01:45:51 +08:00
|
|
|
Py_DECREF(global_super);
|
|
|
|
Py_DECREF(class);
|
|
|
|
Py_DECREF(self);
|
|
|
|
if (super == NULL) goto pop_3_error;
|
2023-08-10 02:19:39 +08:00
|
|
|
attr = PyObject_GetAttr(super, name);
|
2023-04-26 01:45:51 +08:00
|
|
|
Py_DECREF(super);
|
2023-08-10 02:19:39 +08:00
|
|
|
if (attr == NULL) goto pop_3_error;
|
|
|
|
null = NULL;
|
2023-04-25 06:22:14 +08:00
|
|
|
STACK_SHRINK(2);
|
|
|
|
STACK_GROW(((oparg & 1) ? 1 : 0));
|
2023-08-10 02:19:39 +08:00
|
|
|
stack_pointer[-1 - (oparg & 1 ? 1 : 0)] = attr;
|
|
|
|
if (oparg & 1) { stack_pointer[-(oparg & 1 ? 1 : 0)] = null; }
|
2023-05-11 22:08:13 +08:00
|
|
|
next_instr += 1;
|
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(LOAD_SUPER_ATTR_ATTR) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *self;
|
|
|
|
PyObject *class;
|
|
|
|
PyObject *global_super;
|
2023-08-10 02:19:39 +08:00
|
|
|
PyObject *attr;
|
2023-08-05 00:35:56 +08:00
|
|
|
self = stack_pointer[-1];
|
|
|
|
class = stack_pointer[-2];
|
|
|
|
global_super = stack_pointer[-3];
|
2023-05-11 22:08:13 +08:00
|
|
|
assert(!(oparg & 1));
|
|
|
|
DEOPT_IF(global_super != (PyObject *)&PySuper_Type, LOAD_SUPER_ATTR);
|
|
|
|
DEOPT_IF(!PyType_Check(class), LOAD_SUPER_ATTR);
|
|
|
|
STAT_INC(LOAD_SUPER_ATTR, hit);
|
2023-06-14 04:42:03 +08:00
|
|
|
PyObject *name = GETITEM(FRAME_CO_NAMES, oparg >> 2);
|
2023-08-10 02:19:39 +08:00
|
|
|
attr = _PySuper_Lookup((PyTypeObject *)class, self, name, NULL);
|
2023-05-11 22:08:13 +08:00
|
|
|
Py_DECREF(global_super);
|
|
|
|
Py_DECREF(class);
|
|
|
|
Py_DECREF(self);
|
2023-08-10 02:19:39 +08:00
|
|
|
if (attr == NULL) goto pop_3_error;
|
2023-05-11 22:08:13 +08:00
|
|
|
STACK_SHRINK(2);
|
2023-08-13 03:36:46 +08:00
|
|
|
stack_pointer[-1] = attr;
|
2023-05-11 22:08:13 +08:00
|
|
|
next_instr += 1;
|
2023-04-26 01:45:51 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(LOAD_SUPER_ATTR_METHOD) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *self;
|
|
|
|
PyObject *class;
|
|
|
|
PyObject *global_super;
|
2023-08-10 02:19:39 +08:00
|
|
|
PyObject *attr;
|
|
|
|
PyObject *self_or_null;
|
2023-08-05 00:35:56 +08:00
|
|
|
self = stack_pointer[-1];
|
|
|
|
class = stack_pointer[-2];
|
|
|
|
global_super = stack_pointer[-3];
|
2023-05-11 22:08:13 +08:00
|
|
|
assert(oparg & 1);
|
2023-04-26 01:45:51 +08:00
|
|
|
DEOPT_IF(global_super != (PyObject *)&PySuper_Type, LOAD_SUPER_ATTR);
|
|
|
|
DEOPT_IF(!PyType_Check(class), LOAD_SUPER_ATTR);
|
2023-05-11 22:08:13 +08:00
|
|
|
STAT_INC(LOAD_SUPER_ATTR, hit);
|
2023-06-14 04:42:03 +08:00
|
|
|
PyObject *name = GETITEM(FRAME_CO_NAMES, oparg >> 2);
|
2023-05-31 04:36:24 +08:00
|
|
|
PyTypeObject *cls = (PyTypeObject *)class;
|
2023-05-11 22:08:13 +08:00
|
|
|
int method_found = 0;
|
2023-08-10 02:19:39 +08:00
|
|
|
attr = _PySuper_Lookup(cls, self, name,
|
2023-07-25 04:14:56 +08:00
|
|
|
Py_TYPE(self)->tp_getattro == PyObject_GenericGetAttr ? &method_found : NULL);
|
2023-04-26 01:45:51 +08:00
|
|
|
Py_DECREF(global_super);
|
|
|
|
Py_DECREF(class);
|
2023-08-10 02:19:39 +08:00
|
|
|
if (attr == NULL) {
|
2023-05-11 22:08:13 +08:00
|
|
|
Py_DECREF(self);
|
|
|
|
if (true) goto pop_3_error;
|
|
|
|
}
|
|
|
|
if (method_found) {
|
2023-08-10 02:19:39 +08:00
|
|
|
self_or_null = self; // transfer ownership
|
2023-05-11 22:08:13 +08:00
|
|
|
} else {
|
|
|
|
Py_DECREF(self);
|
2023-08-10 02:19:39 +08:00
|
|
|
self_or_null = NULL;
|
2023-05-11 22:08:13 +08:00
|
|
|
}
|
2023-04-26 01:45:51 +08:00
|
|
|
STACK_SHRINK(1);
|
2023-08-10 02:19:39 +08:00
|
|
|
stack_pointer[-2] = attr;
|
|
|
|
stack_pointer[-1] = self_or_null;
|
2023-05-11 22:08:13 +08:00
|
|
|
next_instr += 1;
|
2023-04-25 06:22:14 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2022-11-03 12:31:26 +08:00
|
|
|
TARGET(LOAD_ATTR) {
|
|
|
|
PREDICTED(LOAD_ATTR);
|
2023-02-02 02:56:52 +08:00
|
|
|
static_assert(INLINE_CACHE_ENTRIES_LOAD_ATTR == 9, "incorrect cache size");
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *owner;
|
2023-08-10 02:19:39 +08:00
|
|
|
PyObject *attr;
|
|
|
|
PyObject *self_or_null = NULL;
|
2023-08-05 00:35:56 +08:00
|
|
|
owner = stack_pointer[-1];
|
2023-01-20 02:14:55 +08:00
|
|
|
#if ENABLE_SPECIALIZATION
|
2022-11-10 02:50:09 +08:00
|
|
|
_PyAttrCache *cache = (_PyAttrCache *)next_instr;
|
|
|
|
if (ADAPTIVE_COUNTER_IS_ZERO(cache->counter)) {
|
2023-06-14 04:42:03 +08:00
|
|
|
PyObject *name = GETITEM(FRAME_CO_NAMES, oparg>>1);
|
2022-11-10 02:50:09 +08:00
|
|
|
next_instr--;
|
2022-11-18 07:09:18 +08:00
|
|
|
_Py_Specialize_LoadAttr(owner, next_instr, name);
|
2022-11-10 02:50:09 +08:00
|
|
|
DISPATCH_SAME_OPARG();
|
|
|
|
}
|
|
|
|
STAT_INC(LOAD_ATTR, deferred);
|
|
|
|
DECREMENT_ADAPTIVE_COUNTER(cache->counter);
|
2023-01-20 02:14:55 +08:00
|
|
|
#endif /* ENABLE_SPECIALIZATION */
|
2023-06-14 04:42:03 +08:00
|
|
|
PyObject *name = GETITEM(FRAME_CO_NAMES, oparg >> 1);
|
2022-11-03 12:31:26 +08:00
|
|
|
if (oparg & 1) {
|
2023-01-30 09:28:39 +08:00
|
|
|
/* Designed to work in tandem with CALL, pushes two values. */
|
2023-08-10 02:19:39 +08:00
|
|
|
attr = NULL;
|
|
|
|
if (_PyObject_GetMethod(owner, name, &attr)) {
|
2022-11-03 12:31:26 +08:00
|
|
|
/* We can bypass temporary bound method object.
|
|
|
|
meth is unbound method and obj is self.
|
|
|
|
|
|
|
|
meth | self | arg1 | ... | argN
|
|
|
|
*/
|
2023-08-10 02:19:39 +08:00
|
|
|
assert(attr != NULL); // No errors on this branch
|
|
|
|
self_or_null = owner; // Transfer ownership
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
|
|
|
else {
|
|
|
|
/* meth is not an unbound method (but a regular attr, or
|
|
|
|
something was returned by a descriptor protocol). Set
|
|
|
|
the second element of the stack to NULL, to signal
|
|
|
|
CALL that it's not a method call.
|
|
|
|
|
|
|
|
NULL | meth | arg1 | ... | argN
|
|
|
|
*/
|
|
|
|
Py_DECREF(owner);
|
2023-08-10 02:19:39 +08:00
|
|
|
if (attr == NULL) goto pop_1_error;
|
|
|
|
self_or_null = NULL;
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
|
|
|
}
|
2022-11-11 02:50:57 +08:00
|
|
|
else {
|
2023-01-30 09:28:39 +08:00
|
|
|
/* Classic, pushes one value. */
|
2023-08-10 02:19:39 +08:00
|
|
|
attr = PyObject_GetAttr(owner, name);
|
2022-11-11 02:50:57 +08:00
|
|
|
Py_DECREF(owner);
|
2023-08-10 02:19:39 +08:00
|
|
|
if (attr == NULL) goto pop_1_error;
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
2023-01-30 09:28:39 +08:00
|
|
|
STACK_GROW(((oparg & 1) ? 1 : 0));
|
2023-08-10 02:19:39 +08:00
|
|
|
stack_pointer[-1 - (oparg & 1 ? 1 : 0)] = attr;
|
|
|
|
if (oparg & 1) { stack_pointer[-(oparg & 1 ? 1 : 0)] = self_or_null; }
|
2023-03-01 00:49:35 +08:00
|
|
|
next_instr += 9;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(LOAD_ATTR_INSTANCE_VALUE) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *owner;
|
2023-08-10 02:19:39 +08:00
|
|
|
PyObject *attr;
|
|
|
|
PyObject *null = NULL;
|
2023-08-05 00:35:56 +08:00
|
|
|
// _GUARD_TYPE_VERSION
|
|
|
|
owner = stack_pointer[-1];
|
2023-07-13 23:36:19 +08:00
|
|
|
{
|
|
|
|
uint32_t type_version = read_u32(&next_instr[1].cache);
|
|
|
|
PyTypeObject *tp = Py_TYPE(owner);
|
|
|
|
assert(type_version != 0);
|
|
|
|
DEOPT_IF(tp->tp_version_tag != type_version, LOAD_ATTR);
|
|
|
|
}
|
2023-08-05 00:35:56 +08:00
|
|
|
// _CHECK_MANAGED_OBJECT_HAS_VALUES
|
2023-07-13 23:36:19 +08:00
|
|
|
{
|
|
|
|
assert(Py_TYPE(owner)->tp_dictoffset < 0);
|
|
|
|
assert(Py_TYPE(owner)->tp_flags & Py_TPFLAGS_MANAGED_DICT);
|
2023-08-10 03:14:50 +08:00
|
|
|
PyDictOrValues *dorv = _PyObject_DictOrValuesPointer(owner);
|
|
|
|
DEOPT_IF(!_PyDictOrValues_IsValues(*dorv) &&
|
|
|
|
!_PyObject_MakeInstanceAttributesFromDict(owner, dorv),
|
|
|
|
LOAD_ATTR);
|
2023-07-13 23:36:19 +08:00
|
|
|
}
|
2023-08-05 00:35:56 +08:00
|
|
|
// _LOAD_ATTR_INSTANCE_VALUE
|
2023-07-13 23:36:19 +08:00
|
|
|
{
|
|
|
|
uint16_t index = read_u16(&next_instr[3].cache);
|
|
|
|
PyDictOrValues dorv = *_PyObject_DictOrValuesPointer(owner);
|
2023-08-10 02:19:39 +08:00
|
|
|
attr = _PyDictOrValues_GetValues(dorv)->values[index];
|
|
|
|
DEOPT_IF(attr == NULL, LOAD_ATTR);
|
2023-07-13 23:36:19 +08:00
|
|
|
STAT_INC(LOAD_ATTR, hit);
|
2023-08-10 02:19:39 +08:00
|
|
|
Py_INCREF(attr);
|
|
|
|
null = NULL;
|
2023-07-13 23:36:19 +08:00
|
|
|
Py_DECREF(owner);
|
|
|
|
}
|
|
|
|
STACK_GROW(((oparg & 1) ? 1 : 0));
|
2023-08-10 02:19:39 +08:00
|
|
|
stack_pointer[-1 - (oparg & 1 ? 1 : 0)] = attr;
|
|
|
|
if (oparg & 1) { stack_pointer[-(oparg & 1 ? 1 : 0)] = null; }
|
2023-08-05 00:35:56 +08:00
|
|
|
next_instr += 9;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(LOAD_ATTR_MODULE) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *owner;
|
2023-08-10 02:19:39 +08:00
|
|
|
PyObject *attr;
|
|
|
|
PyObject *null = NULL;
|
2023-08-05 00:35:56 +08:00
|
|
|
owner = stack_pointer[-1];
|
2023-02-02 02:56:52 +08:00
|
|
|
uint32_t type_version = read_u32(&next_instr[1].cache);
|
|
|
|
uint16_t index = read_u16(&next_instr[3].cache);
|
2022-11-03 12:31:26 +08:00
|
|
|
DEOPT_IF(!PyModule_CheckExact(owner), LOAD_ATTR);
|
|
|
|
PyDictObject *dict = (PyDictObject *)((PyModuleObject *)owner)->md_dict;
|
|
|
|
assert(dict != NULL);
|
2023-02-02 02:56:52 +08:00
|
|
|
DEOPT_IF(dict->ma_keys->dk_version != type_version, LOAD_ATTR);
|
2022-11-03 12:31:26 +08:00
|
|
|
assert(dict->ma_keys->dk_kind == DICT_KEYS_UNICODE);
|
2023-02-02 02:56:52 +08:00
|
|
|
assert(index < dict->ma_keys->dk_nentries);
|
|
|
|
PyDictUnicodeEntry *ep = DK_UNICODE_ENTRIES(dict->ma_keys) + index;
|
2023-08-10 02:19:39 +08:00
|
|
|
attr = ep->me_value;
|
|
|
|
DEOPT_IF(attr == NULL, LOAD_ATTR);
|
2022-11-03 12:31:26 +08:00
|
|
|
STAT_INC(LOAD_ATTR, hit);
|
2023-08-10 02:19:39 +08:00
|
|
|
Py_INCREF(attr);
|
|
|
|
null = NULL;
|
2022-11-03 12:31:26 +08:00
|
|
|
Py_DECREF(owner);
|
2023-02-02 02:56:52 +08:00
|
|
|
STACK_GROW(((oparg & 1) ? 1 : 0));
|
2023-08-10 02:19:39 +08:00
|
|
|
stack_pointer[-1 - (oparg & 1 ? 1 : 0)] = attr;
|
|
|
|
if (oparg & 1) { stack_pointer[-(oparg & 1 ? 1 : 0)] = null; }
|
2023-03-01 00:49:35 +08:00
|
|
|
next_instr += 9;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(LOAD_ATTR_WITH_HINT) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *owner;
|
2023-08-10 02:19:39 +08:00
|
|
|
PyObject *attr;
|
|
|
|
PyObject *null = NULL;
|
2023-08-05 00:35:56 +08:00
|
|
|
owner = stack_pointer[-1];
|
2023-02-02 02:56:52 +08:00
|
|
|
uint32_t type_version = read_u32(&next_instr[1].cache);
|
|
|
|
uint16_t index = read_u16(&next_instr[3].cache);
|
2022-11-03 12:31:26 +08:00
|
|
|
PyTypeObject *tp = Py_TYPE(owner);
|
|
|
|
assert(type_version != 0);
|
|
|
|
DEOPT_IF(tp->tp_version_tag != type_version, LOAD_ATTR);
|
|
|
|
assert(tp->tp_flags & Py_TPFLAGS_MANAGED_DICT);
|
|
|
|
PyDictOrValues dorv = *_PyObject_DictOrValuesPointer(owner);
|
|
|
|
DEOPT_IF(_PyDictOrValues_IsValues(dorv), LOAD_ATTR);
|
|
|
|
PyDictObject *dict = (PyDictObject *)_PyDictOrValues_GetDict(dorv);
|
|
|
|
DEOPT_IF(dict == NULL, LOAD_ATTR);
|
|
|
|
assert(PyDict_CheckExact((PyObject *)dict));
|
2023-06-14 04:42:03 +08:00
|
|
|
PyObject *name = GETITEM(FRAME_CO_NAMES, oparg>>1);
|
2023-02-02 02:56:52 +08:00
|
|
|
uint16_t hint = index;
|
2022-11-03 12:31:26 +08:00
|
|
|
DEOPT_IF(hint >= (size_t)dict->ma_keys->dk_nentries, LOAD_ATTR);
|
|
|
|
if (DK_IS_UNICODE(dict->ma_keys)) {
|
|
|
|
PyDictUnicodeEntry *ep = DK_UNICODE_ENTRIES(dict->ma_keys) + hint;
|
|
|
|
DEOPT_IF(ep->me_key != name, LOAD_ATTR);
|
2023-08-10 02:19:39 +08:00
|
|
|
attr = ep->me_value;
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
|
|
|
else {
|
|
|
|
PyDictKeyEntry *ep = DK_ENTRIES(dict->ma_keys) + hint;
|
|
|
|
DEOPT_IF(ep->me_key != name, LOAD_ATTR);
|
2023-08-10 02:19:39 +08:00
|
|
|
attr = ep->me_value;
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
2023-08-10 02:19:39 +08:00
|
|
|
DEOPT_IF(attr == NULL, LOAD_ATTR);
|
2022-11-03 12:31:26 +08:00
|
|
|
STAT_INC(LOAD_ATTR, hit);
|
2023-08-10 02:19:39 +08:00
|
|
|
Py_INCREF(attr);
|
|
|
|
null = NULL;
|
2022-11-03 12:31:26 +08:00
|
|
|
Py_DECREF(owner);
|
2023-02-02 02:56:52 +08:00
|
|
|
STACK_GROW(((oparg & 1) ? 1 : 0));
|
2023-08-10 02:19:39 +08:00
|
|
|
stack_pointer[-1 - (oparg & 1 ? 1 : 0)] = attr;
|
|
|
|
if (oparg & 1) { stack_pointer[-(oparg & 1 ? 1 : 0)] = null; }
|
2023-03-01 00:49:35 +08:00
|
|
|
next_instr += 9;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(LOAD_ATTR_SLOT) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *owner;
|
2023-08-10 02:19:39 +08:00
|
|
|
PyObject *attr;
|
|
|
|
PyObject *null = NULL;
|
2023-09-28 06:27:44 +08:00
|
|
|
// _GUARD_TYPE_VERSION
|
2023-08-05 00:35:56 +08:00
|
|
|
owner = stack_pointer[-1];
|
2023-09-28 06:27:44 +08:00
|
|
|
{
|
|
|
|
uint32_t type_version = read_u32(&next_instr[1].cache);
|
|
|
|
PyTypeObject *tp = Py_TYPE(owner);
|
|
|
|
assert(type_version != 0);
|
|
|
|
DEOPT_IF(tp->tp_version_tag != type_version, LOAD_ATTR);
|
|
|
|
}
|
|
|
|
// _LOAD_ATTR_SLOT
|
|
|
|
{
|
|
|
|
uint16_t index = read_u16(&next_instr[3].cache);
|
|
|
|
char *addr = (char *)owner + index;
|
|
|
|
attr = *(PyObject **)addr;
|
|
|
|
DEOPT_IF(attr == NULL, LOAD_ATTR);
|
|
|
|
STAT_INC(LOAD_ATTR, hit);
|
|
|
|
Py_INCREF(attr);
|
|
|
|
null = NULL;
|
|
|
|
Py_DECREF(owner);
|
|
|
|
}
|
2023-02-02 02:56:52 +08:00
|
|
|
STACK_GROW(((oparg & 1) ? 1 : 0));
|
2023-08-10 02:19:39 +08:00
|
|
|
stack_pointer[-1 - (oparg & 1 ? 1 : 0)] = attr;
|
|
|
|
if (oparg & 1) { stack_pointer[-(oparg & 1 ? 1 : 0)] = null; }
|
2023-03-01 00:49:35 +08:00
|
|
|
next_instr += 9;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(LOAD_ATTR_CLASS) {
|
2023-08-10 02:19:39 +08:00
|
|
|
PyObject *owner;
|
|
|
|
PyObject *attr;
|
|
|
|
PyObject *null = NULL;
|
|
|
|
owner = stack_pointer[-1];
|
2023-02-02 02:56:52 +08:00
|
|
|
uint32_t type_version = read_u32(&next_instr[1].cache);
|
|
|
|
PyObject *descr = read_obj(&next_instr[5].cache);
|
2022-11-03 12:31:26 +08:00
|
|
|
|
2023-08-10 02:19:39 +08:00
|
|
|
DEOPT_IF(!PyType_Check(owner), LOAD_ATTR);
|
|
|
|
DEOPT_IF(((PyTypeObject *)owner)->tp_version_tag != type_version,
|
2022-11-03 12:31:26 +08:00
|
|
|
LOAD_ATTR);
|
|
|
|
assert(type_version != 0);
|
|
|
|
|
|
|
|
STAT_INC(LOAD_ATTR, hit);
|
2023-08-10 02:19:39 +08:00
|
|
|
null = NULL;
|
|
|
|
attr = descr;
|
|
|
|
assert(attr != NULL);
|
|
|
|
Py_INCREF(attr);
|
|
|
|
Py_DECREF(owner);
|
2023-02-02 02:56:52 +08:00
|
|
|
STACK_GROW(((oparg & 1) ? 1 : 0));
|
2023-08-10 02:19:39 +08:00
|
|
|
stack_pointer[-1 - (oparg & 1 ? 1 : 0)] = attr;
|
|
|
|
if (oparg & 1) { stack_pointer[-(oparg & 1 ? 1 : 0)] = null; }
|
2023-03-01 00:49:35 +08:00
|
|
|
next_instr += 9;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(LOAD_ATTR_PROPERTY) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *owner;
|
|
|
|
owner = stack_pointer[-1];
|
2023-02-02 02:56:52 +08:00
|
|
|
uint32_t type_version = read_u32(&next_instr[1].cache);
|
|
|
|
uint32_t func_version = read_u32(&next_instr[3].cache);
|
|
|
|
PyObject *fget = read_obj(&next_instr[5].cache);
|
2023-08-10 02:19:39 +08:00
|
|
|
assert((oparg & 1) == 0);
|
2022-11-03 12:31:26 +08:00
|
|
|
DEOPT_IF(tstate->interp->eval_frame, LOAD_ATTR);
|
|
|
|
|
|
|
|
PyTypeObject *cls = Py_TYPE(owner);
|
|
|
|
DEOPT_IF(cls->tp_version_tag != type_version, LOAD_ATTR);
|
|
|
|
assert(type_version != 0);
|
|
|
|
assert(Py_IS_TYPE(fget, &PyFunction_Type));
|
|
|
|
PyFunctionObject *f = (PyFunctionObject *)fget;
|
|
|
|
assert(func_version != 0);
|
|
|
|
DEOPT_IF(f->func_version != func_version, LOAD_ATTR);
|
|
|
|
PyCodeObject *code = (PyCodeObject *)f->func_code;
|
|
|
|
assert(code->co_argcount == 1);
|
|
|
|
DEOPT_IF(!_PyThreadState_HasStackSpace(tstate, code->co_framesize), LOAD_ATTR);
|
|
|
|
STAT_INC(LOAD_ATTR, hit);
|
|
|
|
Py_INCREF(fget);
|
2023-01-06 22:55:56 +08:00
|
|
|
_PyInterpreterFrame *new_frame = _PyFrame_PushUnchecked(tstate, f, 1);
|
2023-02-02 02:56:52 +08:00
|
|
|
// Manipulate stack directly because we exit with DISPATCH_INLINED().
|
2023-08-10 02:19:39 +08:00
|
|
|
STACK_SHRINK(1);
|
2022-11-03 12:31:26 +08:00
|
|
|
new_frame->localsplus[0] = owner;
|
2023-06-15 07:14:22 +08:00
|
|
|
SKIP_OVER(INLINE_CACHE_ENTRIES_LOAD_ATTR);
|
2023-04-13 23:19:07 +08:00
|
|
|
frame->return_offset = 0;
|
2022-11-18 03:36:03 +08:00
|
|
|
DISPATCH_INLINED(new_frame);
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(LOAD_ATTR_GETATTRIBUTE_OVERRIDDEN) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *owner;
|
|
|
|
owner = stack_pointer[-1];
|
2023-02-02 02:56:52 +08:00
|
|
|
uint32_t type_version = read_u32(&next_instr[1].cache);
|
|
|
|
uint32_t func_version = read_u32(&next_instr[3].cache);
|
|
|
|
PyObject *getattribute = read_obj(&next_instr[5].cache);
|
2023-08-10 02:19:39 +08:00
|
|
|
assert((oparg & 1) == 0);
|
2022-11-03 12:31:26 +08:00
|
|
|
DEOPT_IF(tstate->interp->eval_frame, LOAD_ATTR);
|
|
|
|
PyTypeObject *cls = Py_TYPE(owner);
|
|
|
|
DEOPT_IF(cls->tp_version_tag != type_version, LOAD_ATTR);
|
|
|
|
assert(type_version != 0);
|
|
|
|
assert(Py_IS_TYPE(getattribute, &PyFunction_Type));
|
|
|
|
PyFunctionObject *f = (PyFunctionObject *)getattribute;
|
|
|
|
assert(func_version != 0);
|
|
|
|
DEOPT_IF(f->func_version != func_version, LOAD_ATTR);
|
|
|
|
PyCodeObject *code = (PyCodeObject *)f->func_code;
|
|
|
|
assert(code->co_argcount == 2);
|
2022-11-10 02:50:09 +08:00
|
|
|
DEOPT_IF(!_PyThreadState_HasStackSpace(tstate, code->co_framesize), LOAD_ATTR);
|
2022-11-03 12:31:26 +08:00
|
|
|
STAT_INC(LOAD_ATTR, hit);
|
|
|
|
|
2023-06-14 04:42:03 +08:00
|
|
|
PyObject *name = GETITEM(FRAME_CO_NAMES, oparg >> 1);
|
2022-11-03 12:31:26 +08:00
|
|
|
Py_INCREF(f);
|
2023-01-06 22:55:56 +08:00
|
|
|
_PyInterpreterFrame *new_frame = _PyFrame_PushUnchecked(tstate, f, 2);
|
2023-02-02 02:56:52 +08:00
|
|
|
// Manipulate stack directly because we exit with DISPATCH_INLINED().
|
2023-08-10 02:19:39 +08:00
|
|
|
STACK_SHRINK(1);
|
2022-11-03 12:31:26 +08:00
|
|
|
new_frame->localsplus[0] = owner;
|
2022-11-10 18:25:33 +08:00
|
|
|
new_frame->localsplus[1] = Py_NewRef(name);
|
2023-06-15 07:14:22 +08:00
|
|
|
SKIP_OVER(INLINE_CACHE_ENTRIES_LOAD_ATTR);
|
2023-04-13 23:19:07 +08:00
|
|
|
frame->return_offset = 0;
|
2022-11-18 03:36:03 +08:00
|
|
|
DISPATCH_INLINED(new_frame);
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(STORE_ATTR_INSTANCE_VALUE) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *owner;
|
|
|
|
PyObject *value;
|
2023-09-28 06:27:44 +08:00
|
|
|
// _GUARD_TYPE_VERSION_STORE
|
2023-08-05 00:35:56 +08:00
|
|
|
owner = stack_pointer[-1];
|
2023-09-28 06:27:44 +08:00
|
|
|
{
|
|
|
|
uint32_t type_version = read_u32(&next_instr[1].cache);
|
|
|
|
PyTypeObject *tp = Py_TYPE(owner);
|
|
|
|
assert(type_version != 0);
|
|
|
|
DEOPT_IF(tp->tp_version_tag != type_version, STORE_ATTR);
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
2023-09-28 06:27:44 +08:00
|
|
|
// _GUARD_DORV_VALUES
|
|
|
|
{
|
2023-09-28 22:31:32 +08:00
|
|
|
assert(Py_TYPE(owner)->tp_flags & Py_TPFLAGS_MANAGED_DICT);
|
2023-09-28 06:27:44 +08:00
|
|
|
PyDictOrValues dorv = *_PyObject_DictOrValuesPointer(owner);
|
|
|
|
DEOPT_IF(!_PyDictOrValues_IsValues(dorv), STORE_ATTR);
|
|
|
|
}
|
|
|
|
// _STORE_ATTR_INSTANCE_VALUE
|
|
|
|
value = stack_pointer[-2];
|
|
|
|
{
|
|
|
|
uint16_t index = read_u16(&next_instr[3].cache);
|
|
|
|
PyDictOrValues dorv = *_PyObject_DictOrValuesPointer(owner);
|
|
|
|
STAT_INC(STORE_ATTR, hit);
|
|
|
|
PyDictValues *values = _PyDictOrValues_GetValues(dorv);
|
|
|
|
PyObject *old_value = values->values[index];
|
|
|
|
values->values[index] = value;
|
|
|
|
if (old_value == NULL) {
|
|
|
|
_PyDictValues_AddToInsertionOrder(values, index);
|
|
|
|
}
|
|
|
|
else {
|
|
|
|
Py_DECREF(old_value);
|
|
|
|
}
|
|
|
|
Py_DECREF(owner);
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
GH-98831: Typed stack effects, and more instructions converted (#99764)
Stack effects can now have a type, e.g. `inst(X, (left, right -- jump/uint64_t)) { ... }`.
Instructions converted to the non-legacy format:
* COMPARE_OP
* COMPARE_OP_FLOAT_JUMP
* COMPARE_OP_INT_JUMP
* COMPARE_OP_STR_JUMP
* STORE_ATTR
* DELETE_ATTR
* STORE_GLOBAL
* STORE_ATTR_INSTANCE_VALUE
* STORE_ATTR_WITH_HINT
* STORE_ATTR_SLOT, and complete the store_attr family
* Complete the store_subscr family: STORE_SUBSCR{,DICT,LIST_INT}
(STORE_SUBSCR was alread half converted,
but wasn't using cache effects yet.)
* DELETE_SUBSCR
* PRINT_EXPR
* INTERPRETER_EXIT (a bit weird, ends in return)
* RETURN_VALUE
* GET_AITER (had to restructure it some)
The original had mysterious `SET_TOP(NULL)` before `goto error`.
I assume those just account for `obj` having been decref'ed,
so I got rid of them in favor of the cleanup implied by `ERROR_IF()`.
* LIST_APPEND (a bit unhappy with it)
* SET_ADD (also a bit unhappy with it)
Various other improvements/refactorings as well.
2022-12-09 05:31:27 +08:00
|
|
|
STACK_SHRINK(2);
|
2023-03-01 00:49:35 +08:00
|
|
|
next_instr += 4;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(STORE_ATTR_WITH_HINT) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *owner;
|
|
|
|
PyObject *value;
|
|
|
|
owner = stack_pointer[-1];
|
|
|
|
value = stack_pointer[-2];
|
2022-12-14 19:12:53 +08:00
|
|
|
uint32_t type_version = read_u32(&next_instr[1].cache);
|
|
|
|
uint16_t hint = read_u16(&next_instr[3].cache);
|
2022-11-03 12:31:26 +08:00
|
|
|
PyTypeObject *tp = Py_TYPE(owner);
|
|
|
|
assert(type_version != 0);
|
|
|
|
DEOPT_IF(tp->tp_version_tag != type_version, STORE_ATTR);
|
|
|
|
assert(tp->tp_flags & Py_TPFLAGS_MANAGED_DICT);
|
|
|
|
PyDictOrValues dorv = *_PyObject_DictOrValuesPointer(owner);
|
2022-11-10 02:50:09 +08:00
|
|
|
DEOPT_IF(_PyDictOrValues_IsValues(dorv), STORE_ATTR);
|
2022-11-03 12:31:26 +08:00
|
|
|
PyDictObject *dict = (PyDictObject *)_PyDictOrValues_GetDict(dorv);
|
|
|
|
DEOPT_IF(dict == NULL, STORE_ATTR);
|
|
|
|
assert(PyDict_CheckExact((PyObject *)dict));
|
2023-06-14 04:42:03 +08:00
|
|
|
PyObject *name = GETITEM(FRAME_CO_NAMES, oparg);
|
2022-11-03 12:31:26 +08:00
|
|
|
DEOPT_IF(hint >= (size_t)dict->ma_keys->dk_nentries, STORE_ATTR);
|
GH-98831: Typed stack effects, and more instructions converted (#99764)
Stack effects can now have a type, e.g. `inst(X, (left, right -- jump/uint64_t)) { ... }`.
Instructions converted to the non-legacy format:
* COMPARE_OP
* COMPARE_OP_FLOAT_JUMP
* COMPARE_OP_INT_JUMP
* COMPARE_OP_STR_JUMP
* STORE_ATTR
* DELETE_ATTR
* STORE_GLOBAL
* STORE_ATTR_INSTANCE_VALUE
* STORE_ATTR_WITH_HINT
* STORE_ATTR_SLOT, and complete the store_attr family
* Complete the store_subscr family: STORE_SUBSCR{,DICT,LIST_INT}
(STORE_SUBSCR was alread half converted,
but wasn't using cache effects yet.)
* DELETE_SUBSCR
* PRINT_EXPR
* INTERPRETER_EXIT (a bit weird, ends in return)
* RETURN_VALUE
* GET_AITER (had to restructure it some)
The original had mysterious `SET_TOP(NULL)` before `goto error`.
I assume those just account for `obj` having been decref'ed,
so I got rid of them in favor of the cleanup implied by `ERROR_IF()`.
* LIST_APPEND (a bit unhappy with it)
* SET_ADD (also a bit unhappy with it)
Various other improvements/refactorings as well.
2022-12-09 05:31:27 +08:00
|
|
|
PyObject *old_value;
|
2022-11-03 12:31:26 +08:00
|
|
|
uint64_t new_version;
|
|
|
|
if (DK_IS_UNICODE(dict->ma_keys)) {
|
|
|
|
PyDictUnicodeEntry *ep = DK_UNICODE_ENTRIES(dict->ma_keys) + hint;
|
|
|
|
DEOPT_IF(ep->me_key != name, STORE_ATTR);
|
|
|
|
old_value = ep->me_value;
|
|
|
|
DEOPT_IF(old_value == NULL, STORE_ATTR);
|
2023-03-09 23:16:30 +08:00
|
|
|
new_version = _PyDict_NotifyEvent(tstate->interp, PyDict_EVENT_MODIFIED, dict, name, value);
|
2022-11-03 12:31:26 +08:00
|
|
|
ep->me_value = value;
|
|
|
|
}
|
|
|
|
else {
|
|
|
|
PyDictKeyEntry *ep = DK_ENTRIES(dict->ma_keys) + hint;
|
|
|
|
DEOPT_IF(ep->me_key != name, STORE_ATTR);
|
|
|
|
old_value = ep->me_value;
|
|
|
|
DEOPT_IF(old_value == NULL, STORE_ATTR);
|
2023-03-09 23:16:30 +08:00
|
|
|
new_version = _PyDict_NotifyEvent(tstate->interp, PyDict_EVENT_MODIFIED, dict, name, value);
|
2022-11-03 12:31:26 +08:00
|
|
|
ep->me_value = value;
|
|
|
|
}
|
|
|
|
Py_DECREF(old_value);
|
|
|
|
STAT_INC(STORE_ATTR, hit);
|
|
|
|
/* Ensure dict is GC tracked if it needs to be */
|
|
|
|
if (!_PyObject_GC_IS_TRACKED(dict) && _PyObject_GC_MAY_BE_TRACKED(value)) {
|
|
|
|
_PyObject_GC_TRACK(dict);
|
|
|
|
}
|
|
|
|
/* PEP 509 */
|
|
|
|
dict->ma_version_tag = new_version;
|
|
|
|
Py_DECREF(owner);
|
GH-98831: Typed stack effects, and more instructions converted (#99764)
Stack effects can now have a type, e.g. `inst(X, (left, right -- jump/uint64_t)) { ... }`.
Instructions converted to the non-legacy format:
* COMPARE_OP
* COMPARE_OP_FLOAT_JUMP
* COMPARE_OP_INT_JUMP
* COMPARE_OP_STR_JUMP
* STORE_ATTR
* DELETE_ATTR
* STORE_GLOBAL
* STORE_ATTR_INSTANCE_VALUE
* STORE_ATTR_WITH_HINT
* STORE_ATTR_SLOT, and complete the store_attr family
* Complete the store_subscr family: STORE_SUBSCR{,DICT,LIST_INT}
(STORE_SUBSCR was alread half converted,
but wasn't using cache effects yet.)
* DELETE_SUBSCR
* PRINT_EXPR
* INTERPRETER_EXIT (a bit weird, ends in return)
* RETURN_VALUE
* GET_AITER (had to restructure it some)
The original had mysterious `SET_TOP(NULL)` before `goto error`.
I assume those just account for `obj` having been decref'ed,
so I got rid of them in favor of the cleanup implied by `ERROR_IF()`.
* LIST_APPEND (a bit unhappy with it)
* SET_ADD (also a bit unhappy with it)
Various other improvements/refactorings as well.
2022-12-09 05:31:27 +08:00
|
|
|
STACK_SHRINK(2);
|
2023-03-01 00:49:35 +08:00
|
|
|
next_instr += 4;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(STORE_ATTR_SLOT) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *owner;
|
|
|
|
PyObject *value;
|
2023-09-28 06:27:44 +08:00
|
|
|
// _GUARD_TYPE_VERSION_STORE
|
2023-08-05 00:35:56 +08:00
|
|
|
owner = stack_pointer[-1];
|
2023-09-28 06:27:44 +08:00
|
|
|
{
|
|
|
|
uint32_t type_version = read_u32(&next_instr[1].cache);
|
|
|
|
PyTypeObject *tp = Py_TYPE(owner);
|
|
|
|
assert(type_version != 0);
|
|
|
|
DEOPT_IF(tp->tp_version_tag != type_version, STORE_ATTR);
|
|
|
|
}
|
|
|
|
// _STORE_ATTR_SLOT
|
2023-08-05 00:35:56 +08:00
|
|
|
value = stack_pointer[-2];
|
2023-09-28 06:27:44 +08:00
|
|
|
{
|
|
|
|
uint16_t index = read_u16(&next_instr[3].cache);
|
|
|
|
char *addr = (char *)owner + index;
|
|
|
|
STAT_INC(STORE_ATTR, hit);
|
|
|
|
PyObject *old_value = *(PyObject **)addr;
|
|
|
|
*(PyObject **)addr = value;
|
|
|
|
Py_XDECREF(old_value);
|
|
|
|
Py_DECREF(owner);
|
|
|
|
}
|
GH-98831: Typed stack effects, and more instructions converted (#99764)
Stack effects can now have a type, e.g. `inst(X, (left, right -- jump/uint64_t)) { ... }`.
Instructions converted to the non-legacy format:
* COMPARE_OP
* COMPARE_OP_FLOAT_JUMP
* COMPARE_OP_INT_JUMP
* COMPARE_OP_STR_JUMP
* STORE_ATTR
* DELETE_ATTR
* STORE_GLOBAL
* STORE_ATTR_INSTANCE_VALUE
* STORE_ATTR_WITH_HINT
* STORE_ATTR_SLOT, and complete the store_attr family
* Complete the store_subscr family: STORE_SUBSCR{,DICT,LIST_INT}
(STORE_SUBSCR was alread half converted,
but wasn't using cache effects yet.)
* DELETE_SUBSCR
* PRINT_EXPR
* INTERPRETER_EXIT (a bit weird, ends in return)
* RETURN_VALUE
* GET_AITER (had to restructure it some)
The original had mysterious `SET_TOP(NULL)` before `goto error`.
I assume those just account for `obj` having been decref'ed,
so I got rid of them in favor of the cleanup implied by `ERROR_IF()`.
* LIST_APPEND (a bit unhappy with it)
* SET_ADD (also a bit unhappy with it)
Various other improvements/refactorings as well.
2022-12-09 05:31:27 +08:00
|
|
|
STACK_SHRINK(2);
|
2023-03-01 00:49:35 +08:00
|
|
|
next_instr += 4;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2022-11-10 02:50:09 +08:00
|
|
|
TARGET(COMPARE_OP) {
|
2023-03-24 06:25:09 +08:00
|
|
|
PREDICTED(COMPARE_OP);
|
|
|
|
static_assert(INLINE_CACHE_ENTRIES_COMPARE_OP == 1, "incorrect cache size");
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *right;
|
|
|
|
PyObject *left;
|
GH-98831: Typed stack effects, and more instructions converted (#99764)
Stack effects can now have a type, e.g. `inst(X, (left, right -- jump/uint64_t)) { ... }`.
Instructions converted to the non-legacy format:
* COMPARE_OP
* COMPARE_OP_FLOAT_JUMP
* COMPARE_OP_INT_JUMP
* COMPARE_OP_STR_JUMP
* STORE_ATTR
* DELETE_ATTR
* STORE_GLOBAL
* STORE_ATTR_INSTANCE_VALUE
* STORE_ATTR_WITH_HINT
* STORE_ATTR_SLOT, and complete the store_attr family
* Complete the store_subscr family: STORE_SUBSCR{,DICT,LIST_INT}
(STORE_SUBSCR was alread half converted,
but wasn't using cache effects yet.)
* DELETE_SUBSCR
* PRINT_EXPR
* INTERPRETER_EXIT (a bit weird, ends in return)
* RETURN_VALUE
* GET_AITER (had to restructure it some)
The original had mysterious `SET_TOP(NULL)` before `goto error`.
I assume those just account for `obj` having been decref'ed,
so I got rid of them in favor of the cleanup implied by `ERROR_IF()`.
* LIST_APPEND (a bit unhappy with it)
* SET_ADD (also a bit unhappy with it)
Various other improvements/refactorings as well.
2022-12-09 05:31:27 +08:00
|
|
|
PyObject *res;
|
2023-08-05 00:35:56 +08:00
|
|
|
right = stack_pointer[-1];
|
|
|
|
left = stack_pointer[-2];
|
2023-01-20 02:14:55 +08:00
|
|
|
#if ENABLE_SPECIALIZATION
|
2023-01-16 20:35:21 +08:00
|
|
|
_PyCompareOpCache *cache = (_PyCompareOpCache *)next_instr;
|
|
|
|
if (ADAPTIVE_COUNTER_IS_ZERO(cache->counter)) {
|
|
|
|
next_instr--;
|
2023-03-24 06:25:09 +08:00
|
|
|
_Py_Specialize_CompareOp(left, right, next_instr, oparg);
|
2023-01-16 20:35:21 +08:00
|
|
|
DISPATCH_SAME_OPARG();
|
2022-12-09 07:54:07 +08:00
|
|
|
}
|
2023-03-24 06:25:09 +08:00
|
|
|
STAT_INC(COMPARE_OP, deferred);
|
2023-01-16 20:35:21 +08:00
|
|
|
DECREMENT_ADAPTIVE_COUNTER(cache->counter);
|
2023-01-20 02:14:55 +08:00
|
|
|
#endif /* ENABLE_SPECIALIZATION */
|
2023-06-30 04:49:54 +08:00
|
|
|
assert((oparg >> 5) <= Py_GE);
|
|
|
|
res = PyObject_RichCompare(left, right, oparg >> 5);
|
2023-01-16 20:35:21 +08:00
|
|
|
Py_DECREF(left);
|
|
|
|
Py_DECREF(right);
|
2023-03-24 06:25:09 +08:00
|
|
|
if (res == NULL) goto pop_2_error;
|
2023-06-30 04:49:54 +08:00
|
|
|
if (oparg & 16) {
|
|
|
|
int res_bool = PyObject_IsTrue(res);
|
|
|
|
Py_DECREF(res);
|
|
|
|
if (res_bool < 0) goto pop_2_error;
|
|
|
|
res = res_bool ? Py_True : Py_False;
|
|
|
|
}
|
2023-03-24 06:25:09 +08:00
|
|
|
STACK_SHRINK(1);
|
|
|
|
stack_pointer[-1] = res;
|
|
|
|
next_instr += 1;
|
2022-12-09 07:54:07 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2023-03-24 06:25:09 +08:00
|
|
|
TARGET(COMPARE_OP_FLOAT) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *right;
|
|
|
|
PyObject *left;
|
2023-03-24 06:25:09 +08:00
|
|
|
PyObject *res;
|
2023-08-05 00:35:56 +08:00
|
|
|
right = stack_pointer[-1];
|
|
|
|
left = stack_pointer[-2];
|
2023-03-24 06:25:09 +08:00
|
|
|
DEOPT_IF(!PyFloat_CheckExact(left), COMPARE_OP);
|
|
|
|
DEOPT_IF(!PyFloat_CheckExact(right), COMPARE_OP);
|
|
|
|
STAT_INC(COMPARE_OP, hit);
|
2023-01-16 20:35:21 +08:00
|
|
|
double dleft = PyFloat_AS_DOUBLE(left);
|
|
|
|
double dright = PyFloat_AS_DOUBLE(right);
|
|
|
|
// 1 if NaN, 2 if <, 4 if >, 8 if ==; this matches low four bits of the oparg
|
|
|
|
int sign_ish = COMPARISON_BIT(dleft, dright);
|
|
|
|
_Py_DECREF_SPECIALIZED(left, _PyFloat_ExactDealloc);
|
|
|
|
_Py_DECREF_SPECIALIZED(right, _PyFloat_ExactDealloc);
|
2023-03-24 06:25:09 +08:00
|
|
|
res = (sign_ish & oparg) ? Py_True : Py_False;
|
2023-06-30 04:49:54 +08:00
|
|
|
// It's always a bool, so we don't care about oparg & 16.
|
2023-03-24 06:25:09 +08:00
|
|
|
STACK_SHRINK(1);
|
|
|
|
stack_pointer[-1] = res;
|
|
|
|
next_instr += 1;
|
2022-12-09 07:54:07 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2023-03-24 06:25:09 +08:00
|
|
|
TARGET(COMPARE_OP_INT) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *right;
|
|
|
|
PyObject *left;
|
2023-03-24 06:25:09 +08:00
|
|
|
PyObject *res;
|
2023-08-05 00:35:56 +08:00
|
|
|
right = stack_pointer[-1];
|
|
|
|
left = stack_pointer[-2];
|
2023-03-24 06:25:09 +08:00
|
|
|
DEOPT_IF(!PyLong_CheckExact(left), COMPARE_OP);
|
|
|
|
DEOPT_IF(!PyLong_CheckExact(right), COMPARE_OP);
|
|
|
|
DEOPT_IF(!_PyLong_IsCompact((PyLongObject *)left), COMPARE_OP);
|
|
|
|
DEOPT_IF(!_PyLong_IsCompact((PyLongObject *)right), COMPARE_OP);
|
|
|
|
STAT_INC(COMPARE_OP, hit);
|
2023-03-22 22:49:51 +08:00
|
|
|
assert(_PyLong_DigitCount((PyLongObject *)left) <= 1 &&
|
|
|
|
_PyLong_DigitCount((PyLongObject *)right) <= 1);
|
|
|
|
Py_ssize_t ileft = _PyLong_CompactValue((PyLongObject *)left);
|
|
|
|
Py_ssize_t iright = _PyLong_CompactValue((PyLongObject *)right);
|
2023-01-16 20:35:21 +08:00
|
|
|
// 2 if <, 4 if >, 8 if ==; this matches the low 4 bits of the oparg
|
|
|
|
int sign_ish = COMPARISON_BIT(ileft, iright);
|
|
|
|
_Py_DECREF_SPECIALIZED(left, (destructor)PyObject_Free);
|
|
|
|
_Py_DECREF_SPECIALIZED(right, (destructor)PyObject_Free);
|
2023-03-24 06:25:09 +08:00
|
|
|
res = (sign_ish & oparg) ? Py_True : Py_False;
|
2023-06-30 04:49:54 +08:00
|
|
|
// It's always a bool, so we don't care about oparg & 16.
|
2023-03-24 06:25:09 +08:00
|
|
|
STACK_SHRINK(1);
|
|
|
|
stack_pointer[-1] = res;
|
|
|
|
next_instr += 1;
|
2023-01-16 20:35:21 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2023-03-24 06:25:09 +08:00
|
|
|
TARGET(COMPARE_OP_STR) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *right;
|
|
|
|
PyObject *left;
|
2023-03-24 06:25:09 +08:00
|
|
|
PyObject *res;
|
2023-08-05 00:35:56 +08:00
|
|
|
right = stack_pointer[-1];
|
|
|
|
left = stack_pointer[-2];
|
2023-03-24 06:25:09 +08:00
|
|
|
DEOPT_IF(!PyUnicode_CheckExact(left), COMPARE_OP);
|
|
|
|
DEOPT_IF(!PyUnicode_CheckExact(right), COMPARE_OP);
|
|
|
|
STAT_INC(COMPARE_OP, hit);
|
|
|
|
int eq = _PyUnicode_Equal(left, right);
|
2023-06-30 04:49:54 +08:00
|
|
|
assert((oparg >> 5) == Py_EQ || (oparg >> 5) == Py_NE);
|
2023-01-16 20:35:21 +08:00
|
|
|
_Py_DECREF_SPECIALIZED(left, _PyUnicode_ExactDealloc);
|
|
|
|
_Py_DECREF_SPECIALIZED(right, _PyUnicode_ExactDealloc);
|
2023-03-24 06:25:09 +08:00
|
|
|
assert(eq == 0 || eq == 1);
|
2023-01-16 20:35:21 +08:00
|
|
|
assert((oparg & 0xf) == COMPARISON_NOT_EQUALS || (oparg & 0xf) == COMPARISON_EQUALS);
|
|
|
|
assert(COMPARISON_NOT_EQUALS + 1 == COMPARISON_EQUALS);
|
2023-03-24 06:25:09 +08:00
|
|
|
res = ((COMPARISON_NOT_EQUALS + eq) & oparg) ? Py_True : Py_False;
|
2023-06-30 04:49:54 +08:00
|
|
|
// It's always a bool, so we don't care about oparg & 16.
|
2023-03-24 06:25:09 +08:00
|
|
|
STACK_SHRINK(1);
|
|
|
|
stack_pointer[-1] = res;
|
|
|
|
next_instr += 1;
|
2022-12-09 07:54:07 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2022-11-03 12:31:26 +08:00
|
|
|
TARGET(IS_OP) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *right;
|
|
|
|
PyObject *left;
|
2022-12-28 09:11:03 +08:00
|
|
|
PyObject *b;
|
2023-08-05 00:35:56 +08:00
|
|
|
right = stack_pointer[-1];
|
|
|
|
left = stack_pointer[-2];
|
2022-11-03 12:31:26 +08:00
|
|
|
int res = Py_Is(left, right) ^ oparg;
|
|
|
|
Py_DECREF(left);
|
|
|
|
Py_DECREF(right);
|
2023-05-17 05:36:02 +08:00
|
|
|
b = res ? Py_True : Py_False;
|
2022-12-28 09:11:03 +08:00
|
|
|
STACK_SHRINK(1);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = b;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(CONTAINS_OP) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *right;
|
|
|
|
PyObject *left;
|
2022-12-28 09:11:03 +08:00
|
|
|
PyObject *b;
|
2023-08-05 00:35:56 +08:00
|
|
|
right = stack_pointer[-1];
|
|
|
|
left = stack_pointer[-2];
|
2022-11-03 12:31:26 +08:00
|
|
|
int res = PySequence_Contains(right, left);
|
|
|
|
Py_DECREF(left);
|
|
|
|
Py_DECREF(right);
|
2022-12-28 09:11:03 +08:00
|
|
|
if (res < 0) goto pop_2_error;
|
2023-05-17 05:36:02 +08:00
|
|
|
b = (res ^ oparg) ? Py_True : Py_False;
|
2022-12-28 09:11:03 +08:00
|
|
|
STACK_SHRINK(1);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = b;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(CHECK_EG_MATCH) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *match_type;
|
|
|
|
PyObject *exc_value;
|
2023-01-24 17:43:16 +08:00
|
|
|
PyObject *rest;
|
|
|
|
PyObject *match;
|
2023-08-05 00:35:56 +08:00
|
|
|
match_type = stack_pointer[-1];
|
|
|
|
exc_value = stack_pointer[-2];
|
2023-07-21 04:37:19 +08:00
|
|
|
if (_PyEval_CheckExceptStarTypeValid(tstate, match_type) < 0) {
|
2023-01-24 17:43:16 +08:00
|
|
|
Py_DECREF(exc_value);
|
2022-11-03 12:31:26 +08:00
|
|
|
Py_DECREF(match_type);
|
2023-01-24 17:43:16 +08:00
|
|
|
if (true) goto pop_2_error;
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
|
|
|
|
2023-01-24 17:43:16 +08:00
|
|
|
match = NULL;
|
|
|
|
rest = NULL;
|
2023-07-21 04:37:19 +08:00
|
|
|
int res = _PyEval_ExceptionGroupMatch(exc_value, match_type,
|
|
|
|
&match, &rest);
|
2023-01-24 17:43:16 +08:00
|
|
|
Py_DECREF(exc_value);
|
2022-11-03 12:31:26 +08:00
|
|
|
Py_DECREF(match_type);
|
2023-01-24 17:43:16 +08:00
|
|
|
if (res < 0) goto pop_2_error;
|
2022-11-03 12:31:26 +08:00
|
|
|
|
2023-01-24 17:43:16 +08:00
|
|
|
assert((match == NULL) == (rest == NULL));
|
|
|
|
if (match == NULL) goto pop_2_error;
|
|
|
|
|
|
|
|
if (!Py_IsNone(match)) {
|
2023-04-01 13:01:48 +08:00
|
|
|
PyErr_SetHandledException(match);
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-2] = rest;
|
2023-08-05 00:35:56 +08:00
|
|
|
stack_pointer[-1] = match;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(CHECK_EXC_MATCH) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *right;
|
|
|
|
PyObject *left;
|
2022-12-28 09:11:03 +08:00
|
|
|
PyObject *b;
|
2023-08-05 00:35:56 +08:00
|
|
|
right = stack_pointer[-1];
|
|
|
|
left = stack_pointer[-2];
|
2022-11-03 12:31:26 +08:00
|
|
|
assert(PyExceptionInstance_Check(left));
|
2023-07-21 04:37:19 +08:00
|
|
|
if (_PyEval_CheckExceptTypeValid(tstate, right) < 0) {
|
2022-11-03 12:31:26 +08:00
|
|
|
Py_DECREF(right);
|
2022-12-28 09:11:03 +08:00
|
|
|
if (true) goto pop_1_error;
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
int res = PyErr_GivenExceptionMatches(left, right);
|
|
|
|
Py_DECREF(right);
|
2023-05-17 05:36:02 +08:00
|
|
|
b = res ? Py_True : Py_False;
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = b;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(IMPORT_NAME) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *fromlist;
|
|
|
|
PyObject *level;
|
2022-11-03 12:31:26 +08:00
|
|
|
PyObject *res;
|
2023-08-05 00:35:56 +08:00
|
|
|
fromlist = stack_pointer[-1];
|
|
|
|
level = stack_pointer[-2];
|
2023-06-14 04:42:03 +08:00
|
|
|
PyObject *name = GETITEM(FRAME_CO_NAMES, oparg);
|
2022-11-03 12:31:26 +08:00
|
|
|
res = import_name(tstate, frame, name, fromlist, level);
|
|
|
|
Py_DECREF(level);
|
|
|
|
Py_DECREF(fromlist);
|
2022-12-28 09:11:03 +08:00
|
|
|
if (res == NULL) goto pop_2_error;
|
|
|
|
STACK_SHRINK(1);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = res;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(IMPORT_FROM) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *from;
|
2022-11-03 12:31:26 +08:00
|
|
|
PyObject *res;
|
2023-08-05 00:35:56 +08:00
|
|
|
from = stack_pointer[-1];
|
2023-06-14 04:42:03 +08:00
|
|
|
PyObject *name = GETITEM(FRAME_CO_NAMES, oparg);
|
2022-11-03 12:31:26 +08:00
|
|
|
res = import_from(tstate, from, name);
|
2022-12-28 09:11:03 +08:00
|
|
|
if (res == NULL) goto error;
|
|
|
|
STACK_GROW(1);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = res;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(JUMP_FORWARD) {
|
|
|
|
JUMPBY(oparg);
|
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(JUMP_BACKWARD) {
|
2023-07-04 04:28:27 +08:00
|
|
|
CHECK_EVAL_BREAKER();
|
2023-06-02 18:46:18 +08:00
|
|
|
_Py_CODEUNIT *here = next_instr - 1;
|
|
|
|
assert(oparg <= INSTR_OFFSET());
|
|
|
|
JUMPBY(1-oparg);
|
|
|
|
#if ENABLE_SPECIALIZATION
|
|
|
|
here[1].cache += (1 << OPTIMIZER_BITS_IN_COUNTER);
|
2023-07-21 00:35:39 +08:00
|
|
|
if (here[1].cache > tstate->interp->optimizer_backedge_threshold &&
|
|
|
|
// Double-check that the opcode isn't instrumented or something:
|
2023-09-14 01:26:50 +08:00
|
|
|
here->op.code == JUMP_BACKWARD)
|
2023-07-21 00:35:39 +08:00
|
|
|
{
|
2023-06-02 18:46:18 +08:00
|
|
|
OBJECT_STAT_INC(optimization_attempts);
|
2023-09-14 01:26:50 +08:00
|
|
|
int optimized = _PyOptimizer_BackEdge(frame, here, next_instr, stack_pointer);
|
|
|
|
if (optimized < 0) goto error;
|
|
|
|
if (optimized) {
|
|
|
|
// Rewind and enter the executor:
|
|
|
|
assert(here->op.code == ENTER_EXECUTOR);
|
|
|
|
next_instr = here;
|
2023-06-02 18:46:18 +08:00
|
|
|
}
|
2023-09-14 01:26:50 +08:00
|
|
|
here[1].cache &= ((1 << OPTIMIZER_BITS_IN_COUNTER) - 1);
|
2023-06-02 18:46:18 +08:00
|
|
|
}
|
|
|
|
#endif /* ENABLE_SPECIALIZATION */
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2023-06-02 18:46:18 +08:00
|
|
|
TARGET(ENTER_EXECUTOR) {
|
2023-07-04 04:28:27 +08:00
|
|
|
CHECK_EVAL_BREAKER();
|
|
|
|
|
2023-06-14 20:46:37 +08:00
|
|
|
PyCodeObject *code = _PyFrame_GetCode(frame);
|
2023-06-19 17:32:20 +08:00
|
|
|
_PyExecutorObject *executor = (_PyExecutorObject *)code->co_executors->executors[oparg&255];
|
2023-07-04 04:28:27 +08:00
|
|
|
int original_oparg = executor->vm_data.oparg | (oparg & 0xfffff00);
|
|
|
|
JUMPBY(1-original_oparg);
|
|
|
|
frame->prev_instr = next_instr - 1;
|
2023-06-02 18:46:18 +08:00
|
|
|
Py_INCREF(executor);
|
|
|
|
frame = executor->execute(executor, frame, stack_pointer);
|
|
|
|
if (frame == NULL) {
|
2023-08-17 18:16:03 +08:00
|
|
|
frame = tstate->current_frame;
|
2023-06-27 10:02:57 +08:00
|
|
|
goto resume_with_error;
|
2023-06-02 18:46:18 +08:00
|
|
|
}
|
|
|
|
goto resume_frame;
|
|
|
|
}
|
|
|
|
|
2022-11-03 12:31:26 +08:00
|
|
|
TARGET(POP_JUMP_IF_FALSE) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *cond;
|
|
|
|
cond = stack_pointer[-1];
|
2023-06-30 04:49:54 +08:00
|
|
|
assert(PyBool_Check(cond));
|
2023-09-12 02:20:24 +08:00
|
|
|
int flag = Py_IsFalse(cond);
|
|
|
|
#if ENABLE_SPECIALIZATION
|
|
|
|
next_instr->cache = (next_instr->cache << 1) | flag;
|
|
|
|
#endif
|
|
|
|
JUMPBY(oparg * flag);
|
2023-02-02 03:38:06 +08:00
|
|
|
STACK_SHRINK(1);
|
2023-09-12 02:20:24 +08:00
|
|
|
next_instr += 1;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(POP_JUMP_IF_TRUE) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *cond;
|
|
|
|
cond = stack_pointer[-1];
|
2023-06-30 04:49:54 +08:00
|
|
|
assert(PyBool_Check(cond));
|
2023-09-12 02:20:24 +08:00
|
|
|
int flag = Py_IsTrue(cond);
|
|
|
|
#if ENABLE_SPECIALIZATION
|
|
|
|
next_instr->cache = (next_instr->cache << 1) | flag;
|
|
|
|
#endif
|
|
|
|
JUMPBY(oparg * flag);
|
2023-02-02 03:38:06 +08:00
|
|
|
STACK_SHRINK(1);
|
2023-09-12 02:20:24 +08:00
|
|
|
next_instr += 1;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2023-07-11 18:33:59 +08:00
|
|
|
TARGET(POP_JUMP_IF_NONE) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *value;
|
|
|
|
PyObject *b;
|
|
|
|
PyObject *cond;
|
2023-09-12 06:39:19 +08:00
|
|
|
// _IS_NONE
|
2023-08-05 00:35:56 +08:00
|
|
|
value = stack_pointer[-1];
|
2023-07-11 18:33:59 +08:00
|
|
|
{
|
|
|
|
if (Py_IsNone(value)) {
|
|
|
|
b = Py_True;
|
|
|
|
}
|
|
|
|
else {
|
|
|
|
b = Py_False;
|
|
|
|
Py_DECREF(value);
|
|
|
|
}
|
|
|
|
}
|
2023-08-05 00:35:56 +08:00
|
|
|
// POP_JUMP_IF_TRUE
|
|
|
|
cond = b;
|
2023-07-11 18:33:59 +08:00
|
|
|
{
|
|
|
|
assert(PyBool_Check(cond));
|
2023-09-12 02:20:24 +08:00
|
|
|
int flag = Py_IsTrue(cond);
|
|
|
|
#if ENABLE_SPECIALIZATION
|
|
|
|
next_instr->cache = (next_instr->cache << 1) | flag;
|
|
|
|
#endif
|
|
|
|
JUMPBY(oparg * flag);
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
2023-02-02 03:38:06 +08:00
|
|
|
STACK_SHRINK(1);
|
2023-09-12 02:20:24 +08:00
|
|
|
next_instr += 1;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2023-07-11 18:33:59 +08:00
|
|
|
TARGET(POP_JUMP_IF_NOT_NONE) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *value;
|
|
|
|
PyObject *b;
|
|
|
|
PyObject *cond;
|
2023-09-12 06:39:19 +08:00
|
|
|
// _IS_NONE
|
2023-08-05 00:35:56 +08:00
|
|
|
value = stack_pointer[-1];
|
2023-07-11 18:33:59 +08:00
|
|
|
{
|
|
|
|
if (Py_IsNone(value)) {
|
|
|
|
b = Py_True;
|
|
|
|
}
|
|
|
|
else {
|
|
|
|
b = Py_False;
|
|
|
|
Py_DECREF(value);
|
|
|
|
}
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
2023-08-05 00:35:56 +08:00
|
|
|
// POP_JUMP_IF_FALSE
|
|
|
|
cond = b;
|
2023-07-11 18:33:59 +08:00
|
|
|
{
|
|
|
|
assert(PyBool_Check(cond));
|
2023-09-12 02:20:24 +08:00
|
|
|
int flag = Py_IsFalse(cond);
|
|
|
|
#if ENABLE_SPECIALIZATION
|
|
|
|
next_instr->cache = (next_instr->cache << 1) | flag;
|
|
|
|
#endif
|
|
|
|
JUMPBY(oparg * flag);
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
2023-02-02 03:38:06 +08:00
|
|
|
STACK_SHRINK(1);
|
2023-09-12 02:20:24 +08:00
|
|
|
next_instr += 1;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(JUMP_BACKWARD_NO_INTERRUPT) {
|
|
|
|
/* This bytecode is used in the `yield from` or `await` loop.
|
|
|
|
* If there is an interrupt, we want it handled in the innermost
|
|
|
|
* generator or coroutine, so we deliberately do not check it here.
|
|
|
|
* (see bpo-30039).
|
|
|
|
*/
|
|
|
|
JUMPBY(-oparg);
|
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(GET_LEN) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *obj;
|
2023-01-31 18:23:15 +08:00
|
|
|
PyObject *len_o;
|
2023-08-05 00:35:56 +08:00
|
|
|
obj = stack_pointer[-1];
|
2022-11-03 12:31:26 +08:00
|
|
|
// PUSH(len(TOS))
|
2023-01-31 18:23:15 +08:00
|
|
|
Py_ssize_t len_i = PyObject_Length(obj);
|
|
|
|
if (len_i < 0) goto error;
|
|
|
|
len_o = PyLong_FromSsize_t(len_i);
|
|
|
|
if (len_o == NULL) goto error;
|
|
|
|
STACK_GROW(1);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = len_o;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(MATCH_CLASS) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *names;
|
|
|
|
PyObject *type;
|
|
|
|
PyObject *subject;
|
2023-01-25 06:39:13 +08:00
|
|
|
PyObject *attrs;
|
2023-08-05 00:35:56 +08:00
|
|
|
names = stack_pointer[-1];
|
|
|
|
type = stack_pointer[-2];
|
|
|
|
subject = stack_pointer[-3];
|
2022-11-03 12:31:26 +08:00
|
|
|
// Pop TOS and TOS1. Set TOS to a tuple of attributes on success, or
|
|
|
|
// None on failure.
|
|
|
|
assert(PyTuple_CheckExact(names));
|
2023-07-21 04:37:19 +08:00
|
|
|
attrs = _PyEval_MatchClass(tstate, subject, type, oparg, names);
|
2023-01-25 06:39:13 +08:00
|
|
|
Py_DECREF(subject);
|
2022-11-03 12:31:26 +08:00
|
|
|
Py_DECREF(type);
|
2023-01-25 06:39:13 +08:00
|
|
|
Py_DECREF(names);
|
2022-11-03 12:31:26 +08:00
|
|
|
if (attrs) {
|
2023-01-25 06:39:13 +08:00
|
|
|
assert(PyTuple_CheckExact(attrs)); // Success!
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
|
|
|
else {
|
2023-01-25 06:39:13 +08:00
|
|
|
if (_PyErr_Occurred(tstate)) goto pop_3_error;
|
2023-05-17 05:36:02 +08:00
|
|
|
attrs = Py_None; // Failure!
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
2023-01-25 06:39:13 +08:00
|
|
|
STACK_SHRINK(2);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = attrs;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(MATCH_MAPPING) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *subject;
|
2023-01-25 06:39:13 +08:00
|
|
|
PyObject *res;
|
2023-08-05 00:35:56 +08:00
|
|
|
subject = stack_pointer[-1];
|
2022-11-03 12:31:26 +08:00
|
|
|
int match = Py_TYPE(subject)->tp_flags & Py_TPFLAGS_MAPPING;
|
2023-05-17 05:36:02 +08:00
|
|
|
res = match ? Py_True : Py_False;
|
2023-01-25 06:39:13 +08:00
|
|
|
STACK_GROW(1);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = res;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(MATCH_SEQUENCE) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *subject;
|
2023-01-25 06:39:13 +08:00
|
|
|
PyObject *res;
|
2023-08-05 00:35:56 +08:00
|
|
|
subject = stack_pointer[-1];
|
2022-11-03 12:31:26 +08:00
|
|
|
int match = Py_TYPE(subject)->tp_flags & Py_TPFLAGS_SEQUENCE;
|
2023-05-17 05:36:02 +08:00
|
|
|
res = match ? Py_True : Py_False;
|
2023-01-25 06:39:13 +08:00
|
|
|
STACK_GROW(1);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = res;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(MATCH_KEYS) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *keys;
|
|
|
|
PyObject *subject;
|
2023-01-25 06:39:13 +08:00
|
|
|
PyObject *values_or_none;
|
2023-08-05 00:35:56 +08:00
|
|
|
keys = stack_pointer[-1];
|
|
|
|
subject = stack_pointer[-2];
|
2022-11-03 12:31:26 +08:00
|
|
|
// On successful match, PUSH(values). Otherwise, PUSH(None).
|
2023-07-21 04:37:19 +08:00
|
|
|
values_or_none = _PyEval_MatchKeys(tstate, subject, keys);
|
2023-01-25 06:39:13 +08:00
|
|
|
if (values_or_none == NULL) goto error;
|
|
|
|
STACK_GROW(1);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = values_or_none;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(GET_ITER) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *iterable;
|
2023-01-31 18:23:15 +08:00
|
|
|
PyObject *iter;
|
2023-08-05 00:35:56 +08:00
|
|
|
iterable = stack_pointer[-1];
|
2022-11-03 12:31:26 +08:00
|
|
|
/* before: [obj]; after [getiter(obj)] */
|
2023-01-31 18:23:15 +08:00
|
|
|
iter = PyObject_GetIter(iterable);
|
2022-11-03 12:31:26 +08:00
|
|
|
Py_DECREF(iterable);
|
2023-01-31 18:23:15 +08:00
|
|
|
if (iter == NULL) goto pop_1_error;
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = iter;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(GET_YIELD_FROM_ITER) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *iterable;
|
2022-11-03 12:31:26 +08:00
|
|
|
PyObject *iter;
|
2023-08-05 00:35:56 +08:00
|
|
|
iterable = stack_pointer[-1];
|
2023-02-03 19:30:21 +08:00
|
|
|
/* before: [obj]; after [getiter(obj)] */
|
2022-11-03 12:31:26 +08:00
|
|
|
if (PyCoro_CheckExact(iterable)) {
|
|
|
|
/* `iterable` is a coroutine */
|
2023-06-14 20:46:37 +08:00
|
|
|
if (!(_PyFrame_GetCode(frame)->co_flags & (CO_COROUTINE | CO_ITERABLE_COROUTINE))) {
|
2022-11-03 12:31:26 +08:00
|
|
|
/* and it is used in a 'yield from' expression of a
|
|
|
|
regular generator. */
|
|
|
|
_PyErr_SetString(tstate, PyExc_TypeError,
|
|
|
|
"cannot 'yield from' a coroutine object "
|
|
|
|
"in a non-coroutine generator");
|
|
|
|
goto error;
|
|
|
|
}
|
2023-02-03 19:30:21 +08:00
|
|
|
iter = iterable;
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
2023-02-03 19:30:21 +08:00
|
|
|
else if (PyGen_CheckExact(iterable)) {
|
|
|
|
iter = iterable;
|
|
|
|
}
|
|
|
|
else {
|
2022-11-03 12:31:26 +08:00
|
|
|
/* `iterable` is not a generator. */
|
|
|
|
iter = PyObject_GetIter(iterable);
|
2023-02-03 19:30:21 +08:00
|
|
|
if (iter == NULL) {
|
2022-11-03 12:31:26 +08:00
|
|
|
goto error;
|
2023-02-03 19:30:21 +08:00
|
|
|
}
|
|
|
|
Py_DECREF(iterable);
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = iter;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(FOR_ITER) {
|
|
|
|
PREDICTED(FOR_ITER);
|
2023-02-08 00:28:28 +08:00
|
|
|
static_assert(INLINE_CACHE_ENTRIES_FOR_ITER == 1, "incorrect cache size");
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *iter;
|
2023-02-08 00:28:28 +08:00
|
|
|
PyObject *next;
|
2023-08-05 00:35:56 +08:00
|
|
|
iter = stack_pointer[-1];
|
2023-01-20 02:14:55 +08:00
|
|
|
#if ENABLE_SPECIALIZATION
|
2022-11-10 02:50:09 +08:00
|
|
|
_PyForIterCache *cache = (_PyForIterCache *)next_instr;
|
|
|
|
if (ADAPTIVE_COUNTER_IS_ZERO(cache->counter)) {
|
|
|
|
next_instr--;
|
2023-02-08 00:28:28 +08:00
|
|
|
_Py_Specialize_ForIter(iter, next_instr, oparg);
|
2022-11-10 02:50:09 +08:00
|
|
|
DISPATCH_SAME_OPARG();
|
|
|
|
}
|
|
|
|
STAT_INC(FOR_ITER, deferred);
|
|
|
|
DECREMENT_ADAPTIVE_COUNTER(cache->counter);
|
2023-01-20 02:14:55 +08:00
|
|
|
#endif /* ENABLE_SPECIALIZATION */
|
2023-02-08 00:28:28 +08:00
|
|
|
/* before: [iter]; after: [iter, iter()] *or* [] (and jump over END_FOR.) */
|
|
|
|
next = (*Py_TYPE(iter)->tp_iternext)(iter);
|
|
|
|
if (next == NULL) {
|
2022-11-11 02:50:57 +08:00
|
|
|
if (_PyErr_Occurred(tstate)) {
|
|
|
|
if (!_PyErr_ExceptionMatches(tstate, PyExc_StopIteration)) {
|
|
|
|
goto error;
|
|
|
|
}
|
2023-04-12 19:04:55 +08:00
|
|
|
monitor_raise(tstate, frame, next_instr-1);
|
2022-11-11 02:50:57 +08:00
|
|
|
_PyErr_Clear(tstate);
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
2022-11-11 02:50:57 +08:00
|
|
|
/* iterator ended normally */
|
2023-04-12 19:04:55 +08:00
|
|
|
assert(next_instr[INLINE_CACHE_ENTRIES_FOR_ITER + oparg].op.code == END_FOR ||
|
|
|
|
next_instr[INLINE_CACHE_ENTRIES_FOR_ITER + oparg].op.code == INSTRUMENTED_END_FOR);
|
2022-11-11 02:50:57 +08:00
|
|
|
Py_DECREF(iter);
|
2023-02-08 00:28:28 +08:00
|
|
|
STACK_SHRINK(1);
|
2023-06-15 07:14:22 +08:00
|
|
|
SKIP_OVER(INLINE_CACHE_ENTRIES_FOR_ITER);
|
2023-02-08 00:28:28 +08:00
|
|
|
/* Jump forward oparg, then skip following END_FOR instruction */
|
2023-06-15 07:14:22 +08:00
|
|
|
JUMPBY(oparg + 1);
|
2023-02-08 00:28:28 +08:00
|
|
|
DISPATCH();
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
2023-02-08 00:28:28 +08:00
|
|
|
// Common case: no jump, leave it to the code generator
|
|
|
|
STACK_GROW(1);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = next;
|
|
|
|
next_instr += 1;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2023-04-12 19:04:55 +08:00
|
|
|
TARGET(INSTRUMENTED_FOR_ITER) {
|
|
|
|
_Py_CODEUNIT *here = next_instr-1;
|
|
|
|
_Py_CODEUNIT *target;
|
|
|
|
PyObject *iter = TOP();
|
|
|
|
PyObject *next = (*Py_TYPE(iter)->tp_iternext)(iter);
|
|
|
|
if (next != NULL) {
|
|
|
|
PUSH(next);
|
|
|
|
target = next_instr + INLINE_CACHE_ENTRIES_FOR_ITER;
|
|
|
|
}
|
|
|
|
else {
|
|
|
|
if (_PyErr_Occurred(tstate)) {
|
|
|
|
if (!_PyErr_ExceptionMatches(tstate, PyExc_StopIteration)) {
|
|
|
|
goto error;
|
|
|
|
}
|
|
|
|
monitor_raise(tstate, frame, here);
|
|
|
|
_PyErr_Clear(tstate);
|
|
|
|
}
|
|
|
|
/* iterator ended normally */
|
|
|
|
assert(next_instr[INLINE_CACHE_ENTRIES_FOR_ITER + oparg].op.code == END_FOR ||
|
|
|
|
next_instr[INLINE_CACHE_ENTRIES_FOR_ITER + oparg].op.code == INSTRUMENTED_END_FOR);
|
|
|
|
STACK_SHRINK(1);
|
|
|
|
Py_DECREF(iter);
|
|
|
|
/* Skip END_FOR */
|
|
|
|
target = next_instr + INLINE_CACHE_ENTRIES_FOR_ITER + oparg + 1;
|
|
|
|
}
|
|
|
|
INSTRUMENTED_JUMP(here, target, PY_MONITORING_EVENT_BRANCH);
|
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2022-11-03 12:31:26 +08:00
|
|
|
TARGET(FOR_ITER_LIST) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *iter;
|
|
|
|
PyObject *next;
|
|
|
|
// _ITER_CHECK_LIST
|
|
|
|
iter = stack_pointer[-1];
|
2023-07-14 08:27:35 +08:00
|
|
|
{
|
|
|
|
DEOPT_IF(Py_TYPE(iter) != &PyListIter_Type, FOR_ITER);
|
|
|
|
}
|
2023-08-05 00:35:56 +08:00
|
|
|
// _ITER_JUMP_LIST
|
2023-07-14 08:27:35 +08:00
|
|
|
{
|
|
|
|
_PyListIterObject *it = (_PyListIterObject *)iter;
|
|
|
|
assert(Py_TYPE(iter) == &PyListIter_Type);
|
|
|
|
STAT_INC(FOR_ITER, hit);
|
|
|
|
PyListObject *seq = it->it_seq;
|
|
|
|
if (seq == NULL || it->it_index >= PyList_GET_SIZE(seq)) {
|
|
|
|
if (seq != NULL) {
|
|
|
|
it->it_seq = NULL;
|
|
|
|
Py_DECREF(seq);
|
|
|
|
}
|
|
|
|
Py_DECREF(iter);
|
|
|
|
STACK_SHRINK(1);
|
|
|
|
SKIP_OVER(INLINE_CACHE_ENTRIES_FOR_ITER);
|
|
|
|
/* Jump forward oparg, then skip following END_FOR instruction */
|
|
|
|
JUMPBY(oparg + 1);
|
|
|
|
DISPATCH();
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
2023-07-14 08:27:35 +08:00
|
|
|
}
|
2023-08-05 00:35:56 +08:00
|
|
|
// _ITER_NEXT_LIST
|
2023-07-14 08:27:35 +08:00
|
|
|
{
|
|
|
|
_PyListIterObject *it = (_PyListIterObject *)iter;
|
|
|
|
assert(Py_TYPE(iter) == &PyListIter_Type);
|
|
|
|
PyListObject *seq = it->it_seq;
|
|
|
|
assert(seq);
|
|
|
|
assert(it->it_index < PyList_GET_SIZE(seq));
|
|
|
|
next = Py_NewRef(PyList_GET_ITEM(seq, it->it_index++));
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
2023-07-14 08:27:35 +08:00
|
|
|
STACK_GROW(1);
|
2023-08-05 00:35:56 +08:00
|
|
|
stack_pointer[-1] = next;
|
|
|
|
next_instr += 1;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2022-12-09 18:27:01 +08:00
|
|
|
TARGET(FOR_ITER_TUPLE) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *iter;
|
|
|
|
PyObject *next;
|
|
|
|
// _ITER_CHECK_TUPLE
|
|
|
|
iter = stack_pointer[-1];
|
2023-07-14 08:27:35 +08:00
|
|
|
{
|
|
|
|
DEOPT_IF(Py_TYPE(iter) != &PyTupleIter_Type, FOR_ITER);
|
|
|
|
}
|
2023-08-05 00:35:56 +08:00
|
|
|
// _ITER_JUMP_TUPLE
|
2023-07-14 08:27:35 +08:00
|
|
|
{
|
|
|
|
_PyTupleIterObject *it = (_PyTupleIterObject *)iter;
|
|
|
|
assert(Py_TYPE(iter) == &PyTupleIter_Type);
|
|
|
|
STAT_INC(FOR_ITER, hit);
|
|
|
|
PyTupleObject *seq = it->it_seq;
|
|
|
|
if (seq == NULL || it->it_index >= PyTuple_GET_SIZE(seq)) {
|
|
|
|
if (seq != NULL) {
|
|
|
|
it->it_seq = NULL;
|
|
|
|
Py_DECREF(seq);
|
|
|
|
}
|
|
|
|
Py_DECREF(iter);
|
|
|
|
STACK_SHRINK(1);
|
|
|
|
SKIP_OVER(INLINE_CACHE_ENTRIES_FOR_ITER);
|
|
|
|
/* Jump forward oparg, then skip following END_FOR instruction */
|
|
|
|
JUMPBY(oparg + 1);
|
|
|
|
DISPATCH();
|
2022-12-09 18:27:01 +08:00
|
|
|
}
|
2023-07-14 08:27:35 +08:00
|
|
|
}
|
2023-08-05 00:35:56 +08:00
|
|
|
// _ITER_NEXT_TUPLE
|
2023-07-14 08:27:35 +08:00
|
|
|
{
|
|
|
|
_PyTupleIterObject *it = (_PyTupleIterObject *)iter;
|
|
|
|
assert(Py_TYPE(iter) == &PyTupleIter_Type);
|
|
|
|
PyTupleObject *seq = it->it_seq;
|
|
|
|
assert(seq);
|
|
|
|
assert(it->it_index < PyTuple_GET_SIZE(seq));
|
|
|
|
next = Py_NewRef(PyTuple_GET_ITEM(seq, it->it_index++));
|
2022-12-09 18:27:01 +08:00
|
|
|
}
|
2023-07-14 08:27:35 +08:00
|
|
|
STACK_GROW(1);
|
2023-08-05 00:35:56 +08:00
|
|
|
stack_pointer[-1] = next;
|
|
|
|
next_instr += 1;
|
2022-12-09 18:27:01 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2022-11-03 12:31:26 +08:00
|
|
|
TARGET(FOR_ITER_RANGE) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *iter;
|
|
|
|
PyObject *next;
|
|
|
|
// _ITER_CHECK_RANGE
|
|
|
|
iter = stack_pointer[-1];
|
2023-07-13 01:23:59 +08:00
|
|
|
{
|
|
|
|
_PyRangeIterObject *r = (_PyRangeIterObject *)iter;
|
|
|
|
DEOPT_IF(Py_TYPE(r) != &PyRangeIter_Type, FOR_ITER);
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
2023-08-05 00:35:56 +08:00
|
|
|
// _ITER_JUMP_RANGE
|
2023-07-13 01:23:59 +08:00
|
|
|
{
|
|
|
|
_PyRangeIterObject *r = (_PyRangeIterObject *)iter;
|
|
|
|
assert(Py_TYPE(r) == &PyRangeIter_Type);
|
|
|
|
STAT_INC(FOR_ITER, hit);
|
|
|
|
if (r->len <= 0) {
|
|
|
|
STACK_SHRINK(1);
|
|
|
|
Py_DECREF(r);
|
|
|
|
SKIP_OVER(INLINE_CACHE_ENTRIES_FOR_ITER);
|
|
|
|
// Jump over END_FOR instruction.
|
|
|
|
JUMPBY(oparg + 1);
|
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
}
|
2023-08-05 00:35:56 +08:00
|
|
|
// _ITER_NEXT_RANGE
|
2023-07-13 01:23:59 +08:00
|
|
|
{
|
|
|
|
_PyRangeIterObject *r = (_PyRangeIterObject *)iter;
|
|
|
|
assert(Py_TYPE(r) == &PyRangeIter_Type);
|
|
|
|
assert(r->len > 0);
|
|
|
|
long value = r->start;
|
|
|
|
r->start = value + r->step;
|
|
|
|
r->len--;
|
|
|
|
next = PyLong_FromLong(value);
|
|
|
|
if (next == NULL) goto error;
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
2023-07-13 01:23:59 +08:00
|
|
|
STACK_GROW(1);
|
2023-08-05 00:35:56 +08:00
|
|
|
stack_pointer[-1] = next;
|
|
|
|
next_instr += 1;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2022-11-07 22:49:51 +08:00
|
|
|
TARGET(FOR_ITER_GEN) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *iter;
|
|
|
|
iter = stack_pointer[-1];
|
2023-05-13 06:23:13 +08:00
|
|
|
DEOPT_IF(tstate->interp->eval_frame, FOR_ITER);
|
2023-02-08 00:28:28 +08:00
|
|
|
PyGenObject *gen = (PyGenObject *)iter;
|
2022-11-07 22:49:51 +08:00
|
|
|
DEOPT_IF(Py_TYPE(gen) != &PyGen_Type, FOR_ITER);
|
|
|
|
DEOPT_IF(gen->gi_frame_state >= FRAME_EXECUTING, FOR_ITER);
|
|
|
|
STAT_INC(FOR_ITER, hit);
|
|
|
|
_PyInterpreterFrame *gen_frame = (_PyInterpreterFrame *)gen->gi_iframe;
|
2023-05-17 05:36:02 +08:00
|
|
|
_PyFrame_StackPush(gen_frame, Py_None);
|
2022-11-07 22:49:51 +08:00
|
|
|
gen->gi_frame_state = FRAME_EXECUTING;
|
|
|
|
gen->gi_exc_state.previous_item = tstate->exc_info;
|
|
|
|
tstate->exc_info = &gen->gi_exc_state;
|
2023-06-15 07:14:22 +08:00
|
|
|
SKIP_OVER(INLINE_CACHE_ENTRIES_FOR_ITER);
|
2023-04-13 23:19:07 +08:00
|
|
|
assert(next_instr[oparg].op.code == END_FOR ||
|
|
|
|
next_instr[oparg].op.code == INSTRUMENTED_END_FOR);
|
2023-08-17 07:26:43 +08:00
|
|
|
frame->return_offset = oparg;
|
2022-11-18 03:36:03 +08:00
|
|
|
DISPATCH_INLINED(gen_frame);
|
2022-11-07 22:49:51 +08:00
|
|
|
}
|
|
|
|
|
2022-11-03 12:31:26 +08:00
|
|
|
TARGET(BEFORE_ASYNC_WITH) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *mgr;
|
2023-02-01 02:47:50 +08:00
|
|
|
PyObject *exit;
|
2022-11-03 12:31:26 +08:00
|
|
|
PyObject *res;
|
2023-08-05 00:35:56 +08:00
|
|
|
mgr = stack_pointer[-1];
|
2022-11-03 12:31:26 +08:00
|
|
|
PyObject *enter = _PyObject_LookupSpecial(mgr, &_Py_ID(__aenter__));
|
|
|
|
if (enter == NULL) {
|
|
|
|
if (!_PyErr_Occurred(tstate)) {
|
|
|
|
_PyErr_Format(tstate, PyExc_TypeError,
|
|
|
|
"'%.200s' object does not support the "
|
|
|
|
"asynchronous context manager protocol",
|
|
|
|
Py_TYPE(mgr)->tp_name);
|
|
|
|
}
|
|
|
|
goto error;
|
|
|
|
}
|
2023-02-01 02:47:50 +08:00
|
|
|
exit = _PyObject_LookupSpecial(mgr, &_Py_ID(__aexit__));
|
2022-11-03 12:31:26 +08:00
|
|
|
if (exit == NULL) {
|
|
|
|
if (!_PyErr_Occurred(tstate)) {
|
|
|
|
_PyErr_Format(tstate, PyExc_TypeError,
|
|
|
|
"'%.200s' object does not support the "
|
|
|
|
"asynchronous context manager protocol "
|
|
|
|
"(missed __aexit__ method)",
|
|
|
|
Py_TYPE(mgr)->tp_name);
|
|
|
|
}
|
|
|
|
Py_DECREF(enter);
|
|
|
|
goto error;
|
|
|
|
}
|
|
|
|
Py_DECREF(mgr);
|
|
|
|
res = _PyObject_CallNoArgs(enter);
|
|
|
|
Py_DECREF(enter);
|
2023-02-01 02:47:50 +08:00
|
|
|
if (res == NULL) {
|
|
|
|
Py_DECREF(exit);
|
|
|
|
if (true) goto pop_1_error;
|
|
|
|
}
|
|
|
|
STACK_GROW(1);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-2] = exit;
|
2023-08-05 00:35:56 +08:00
|
|
|
stack_pointer[-1] = res;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(BEFORE_WITH) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *mgr;
|
2023-01-31 18:23:15 +08:00
|
|
|
PyObject *exit;
|
2022-11-03 12:31:26 +08:00
|
|
|
PyObject *res;
|
2023-08-05 00:35:56 +08:00
|
|
|
mgr = stack_pointer[-1];
|
2023-01-31 18:23:15 +08:00
|
|
|
/* pop the context manager, push its __exit__ and the
|
|
|
|
* value returned from calling its __enter__
|
|
|
|
*/
|
2022-11-03 12:31:26 +08:00
|
|
|
PyObject *enter = _PyObject_LookupSpecial(mgr, &_Py_ID(__enter__));
|
|
|
|
if (enter == NULL) {
|
|
|
|
if (!_PyErr_Occurred(tstate)) {
|
|
|
|
_PyErr_Format(tstate, PyExc_TypeError,
|
|
|
|
"'%.200s' object does not support the "
|
|
|
|
"context manager protocol",
|
|
|
|
Py_TYPE(mgr)->tp_name);
|
|
|
|
}
|
|
|
|
goto error;
|
|
|
|
}
|
2023-01-31 18:23:15 +08:00
|
|
|
exit = _PyObject_LookupSpecial(mgr, &_Py_ID(__exit__));
|
2022-11-03 12:31:26 +08:00
|
|
|
if (exit == NULL) {
|
|
|
|
if (!_PyErr_Occurred(tstate)) {
|
|
|
|
_PyErr_Format(tstate, PyExc_TypeError,
|
|
|
|
"'%.200s' object does not support the "
|
|
|
|
"context manager protocol "
|
|
|
|
"(missed __exit__ method)",
|
|
|
|
Py_TYPE(mgr)->tp_name);
|
|
|
|
}
|
|
|
|
Py_DECREF(enter);
|
|
|
|
goto error;
|
|
|
|
}
|
|
|
|
Py_DECREF(mgr);
|
|
|
|
res = _PyObject_CallNoArgs(enter);
|
|
|
|
Py_DECREF(enter);
|
|
|
|
if (res == NULL) {
|
2023-01-31 18:23:15 +08:00
|
|
|
Py_DECREF(exit);
|
|
|
|
if (true) goto pop_1_error;
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
2023-01-31 18:23:15 +08:00
|
|
|
STACK_GROW(1);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-2] = exit;
|
2023-08-05 00:35:56 +08:00
|
|
|
stack_pointer[-1] = res;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(WITH_EXCEPT_START) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *val;
|
|
|
|
PyObject *lasti;
|
|
|
|
PyObject *exit_func;
|
2022-11-23 08:04:57 +08:00
|
|
|
PyObject *res;
|
2023-08-05 00:35:56 +08:00
|
|
|
val = stack_pointer[-1];
|
|
|
|
lasti = stack_pointer[-3];
|
|
|
|
exit_func = stack_pointer[-4];
|
2022-11-03 12:31:26 +08:00
|
|
|
/* At the top of the stack are 4 values:
|
2022-11-23 08:04:57 +08:00
|
|
|
- val: TOP = exc_info()
|
|
|
|
- unused: SECOND = previous exception
|
|
|
|
- lasti: THIRD = lasti of exception in exc_info()
|
|
|
|
- exit_func: FOURTH = the context.__exit__ bound method
|
2022-11-03 12:31:26 +08:00
|
|
|
We call FOURTH(type(TOP), TOP, GetTraceback(TOP)).
|
|
|
|
Then we push the __exit__ return value.
|
|
|
|
*/
|
2022-11-23 08:04:57 +08:00
|
|
|
PyObject *exc, *tb;
|
2022-11-03 12:31:26 +08:00
|
|
|
|
|
|
|
assert(val && PyExceptionInstance_Check(val));
|
|
|
|
exc = PyExceptionInstance_Class(val);
|
|
|
|
tb = PyException_GetTraceback(val);
|
2023-07-27 22:27:11 +08:00
|
|
|
if (tb == NULL) {
|
|
|
|
tb = Py_None;
|
|
|
|
}
|
|
|
|
else {
|
|
|
|
Py_DECREF(tb);
|
|
|
|
}
|
2022-11-23 08:04:57 +08:00
|
|
|
assert(PyLong_Check(lasti));
|
|
|
|
(void)lasti; // Shut up compiler warning if asserts are off
|
2022-11-03 12:31:26 +08:00
|
|
|
PyObject *stack[4] = {NULL, exc, val, tb};
|
|
|
|
res = PyObject_Vectorcall(exit_func, stack + 1,
|
|
|
|
3 | PY_VECTORCALL_ARGUMENTS_OFFSET, NULL);
|
2022-11-23 08:04:57 +08:00
|
|
|
if (res == NULL) goto error;
|
|
|
|
STACK_GROW(1);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = res;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(PUSH_EXC_INFO) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *new_exc;
|
2023-02-02 03:38:06 +08:00
|
|
|
PyObject *prev_exc;
|
2023-08-05 00:35:56 +08:00
|
|
|
new_exc = stack_pointer[-1];
|
2022-11-03 12:31:26 +08:00
|
|
|
_PyErr_StackItem *exc_info = tstate->exc_info;
|
|
|
|
if (exc_info->exc_value != NULL) {
|
2023-02-02 03:38:06 +08:00
|
|
|
prev_exc = exc_info->exc_value;
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
|
|
|
else {
|
2023-05-17 05:36:02 +08:00
|
|
|
prev_exc = Py_None;
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
2023-02-02 03:38:06 +08:00
|
|
|
assert(PyExceptionInstance_Check(new_exc));
|
|
|
|
exc_info->exc_value = Py_NewRef(new_exc);
|
|
|
|
STACK_GROW(1);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-2] = prev_exc;
|
2023-08-05 00:35:56 +08:00
|
|
|
stack_pointer[-1] = new_exc;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(LOAD_ATTR_METHOD_WITH_VALUES) {
|
2023-08-10 02:19:39 +08:00
|
|
|
PyObject *owner;
|
|
|
|
PyObject *attr;
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *self;
|
2023-09-28 06:27:44 +08:00
|
|
|
// _GUARD_TYPE_VERSION
|
2023-08-10 02:19:39 +08:00
|
|
|
owner = stack_pointer[-1];
|
2023-09-28 06:27:44 +08:00
|
|
|
{
|
|
|
|
uint32_t type_version = read_u32(&next_instr[1].cache);
|
|
|
|
PyTypeObject *tp = Py_TYPE(owner);
|
|
|
|
assert(type_version != 0);
|
|
|
|
DEOPT_IF(tp->tp_version_tag != type_version, LOAD_ATTR);
|
|
|
|
}
|
|
|
|
// _GUARD_DORV_VALUES_INST_ATTR_FROM_DICT
|
|
|
|
{
|
2023-09-28 22:31:32 +08:00
|
|
|
assert(Py_TYPE(owner)->tp_flags & Py_TPFLAGS_MANAGED_DICT);
|
2023-09-28 06:27:44 +08:00
|
|
|
PyDictOrValues *dorv = _PyObject_DictOrValuesPointer(owner);
|
|
|
|
DEOPT_IF(!_PyDictOrValues_IsValues(*dorv) &&
|
|
|
|
!_PyObject_MakeInstanceAttributesFromDict(owner, dorv),
|
|
|
|
LOAD_ATTR);
|
|
|
|
}
|
|
|
|
// _GUARD_KEYS_VERSION
|
|
|
|
{
|
|
|
|
uint32_t keys_version = read_u32(&next_instr[3].cache);
|
|
|
|
PyTypeObject *owner_cls = Py_TYPE(owner);
|
|
|
|
PyHeapTypeObject *owner_heap_type = (PyHeapTypeObject *)owner_cls;
|
|
|
|
DEOPT_IF(owner_heap_type->ht_cached_keys->dk_version !=
|
|
|
|
keys_version, LOAD_ATTR);
|
|
|
|
}
|
|
|
|
// _LOAD_ATTR_METHOD_WITH_VALUES
|
|
|
|
{
|
|
|
|
PyObject *descr = read_obj(&next_instr[5].cache);
|
|
|
|
assert(oparg & 1);
|
|
|
|
/* Cached method object */
|
|
|
|
STAT_INC(LOAD_ATTR, hit);
|
|
|
|
assert(descr != NULL);
|
|
|
|
attr = Py_NewRef(descr);
|
|
|
|
assert(_PyType_HasFeature(Py_TYPE(attr), Py_TPFLAGS_METHOD_DESCRIPTOR));
|
|
|
|
self = owner;
|
|
|
|
}
|
2023-07-17 08:09:11 +08:00
|
|
|
STACK_GROW(1);
|
2023-08-10 02:19:39 +08:00
|
|
|
stack_pointer[-2] = attr;
|
|
|
|
stack_pointer[-1] = self;
|
2023-03-01 00:49:35 +08:00
|
|
|
next_instr += 9;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(LOAD_ATTR_METHOD_NO_DICT) {
|
2023-08-10 02:19:39 +08:00
|
|
|
PyObject *owner;
|
|
|
|
PyObject *attr;
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *self;
|
2023-09-28 06:27:44 +08:00
|
|
|
// _GUARD_TYPE_VERSION
|
2023-08-10 02:19:39 +08:00
|
|
|
owner = stack_pointer[-1];
|
2023-09-28 06:27:44 +08:00
|
|
|
{
|
|
|
|
uint32_t type_version = read_u32(&next_instr[1].cache);
|
|
|
|
PyTypeObject *tp = Py_TYPE(owner);
|
|
|
|
assert(type_version != 0);
|
|
|
|
DEOPT_IF(tp->tp_version_tag != type_version, LOAD_ATTR);
|
|
|
|
}
|
|
|
|
// _LOAD_ATTR_METHOD_NO_DICT
|
|
|
|
{
|
|
|
|
PyObject *descr = read_obj(&next_instr[5].cache);
|
|
|
|
assert(oparg & 1);
|
2023-09-28 22:31:32 +08:00
|
|
|
assert(Py_TYPE(owner)->tp_dictoffset == 0);
|
2023-09-28 06:27:44 +08:00
|
|
|
STAT_INC(LOAD_ATTR, hit);
|
|
|
|
assert(descr != NULL);
|
|
|
|
assert(_PyType_HasFeature(Py_TYPE(descr), Py_TPFLAGS_METHOD_DESCRIPTOR));
|
|
|
|
attr = Py_NewRef(descr);
|
|
|
|
self = owner;
|
|
|
|
}
|
2023-07-17 08:09:11 +08:00
|
|
|
STACK_GROW(1);
|
2023-08-10 02:19:39 +08:00
|
|
|
stack_pointer[-2] = attr;
|
|
|
|
stack_pointer[-1] = self;
|
2023-07-10 18:40:35 +08:00
|
|
|
next_instr += 9;
|
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(LOAD_ATTR_NONDESCRIPTOR_WITH_VALUES) {
|
2023-08-10 02:19:39 +08:00
|
|
|
PyObject *owner;
|
|
|
|
PyObject *attr;
|
|
|
|
owner = stack_pointer[-1];
|
2023-07-10 18:40:35 +08:00
|
|
|
uint32_t type_version = read_u32(&next_instr[1].cache);
|
|
|
|
uint32_t keys_version = read_u32(&next_instr[3].cache);
|
|
|
|
PyObject *descr = read_obj(&next_instr[5].cache);
|
|
|
|
assert((oparg & 1) == 0);
|
2023-08-10 02:19:39 +08:00
|
|
|
PyTypeObject *owner_cls = Py_TYPE(owner);
|
2023-07-10 18:40:35 +08:00
|
|
|
assert(type_version != 0);
|
2023-08-10 02:19:39 +08:00
|
|
|
DEOPT_IF(owner_cls->tp_version_tag != type_version, LOAD_ATTR);
|
|
|
|
assert(owner_cls->tp_flags & Py_TPFLAGS_MANAGED_DICT);
|
2023-08-10 03:14:50 +08:00
|
|
|
PyDictOrValues *dorv = _PyObject_DictOrValuesPointer(owner);
|
|
|
|
DEOPT_IF(!_PyDictOrValues_IsValues(*dorv) &&
|
|
|
|
!_PyObject_MakeInstanceAttributesFromDict(owner, dorv),
|
|
|
|
LOAD_ATTR);
|
2023-08-10 02:19:39 +08:00
|
|
|
PyHeapTypeObject *owner_heap_type = (PyHeapTypeObject *)owner_cls;
|
|
|
|
DEOPT_IF(owner_heap_type->ht_cached_keys->dk_version !=
|
2023-07-10 18:40:35 +08:00
|
|
|
keys_version, LOAD_ATTR);
|
|
|
|
STAT_INC(LOAD_ATTR, hit);
|
|
|
|
assert(descr != NULL);
|
2023-08-10 02:19:39 +08:00
|
|
|
Py_DECREF(owner);
|
|
|
|
attr = Py_NewRef(descr);
|
2023-08-13 03:36:46 +08:00
|
|
|
stack_pointer[-1] = attr;
|
2023-07-10 18:40:35 +08:00
|
|
|
next_instr += 9;
|
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(LOAD_ATTR_NONDESCRIPTOR_NO_DICT) {
|
2023-08-10 02:19:39 +08:00
|
|
|
PyObject *owner;
|
|
|
|
PyObject *attr;
|
|
|
|
owner = stack_pointer[-1];
|
2023-07-10 18:40:35 +08:00
|
|
|
uint32_t type_version = read_u32(&next_instr[1].cache);
|
|
|
|
PyObject *descr = read_obj(&next_instr[5].cache);
|
|
|
|
assert((oparg & 1) == 0);
|
2023-08-10 02:19:39 +08:00
|
|
|
PyTypeObject *owner_cls = Py_TYPE(owner);
|
2023-07-10 18:40:35 +08:00
|
|
|
assert(type_version != 0);
|
2023-08-10 02:19:39 +08:00
|
|
|
DEOPT_IF(owner_cls->tp_version_tag != type_version, LOAD_ATTR);
|
|
|
|
assert(owner_cls->tp_dictoffset == 0);
|
2023-07-10 18:40:35 +08:00
|
|
|
STAT_INC(LOAD_ATTR, hit);
|
|
|
|
assert(descr != NULL);
|
2023-08-10 02:19:39 +08:00
|
|
|
Py_DECREF(owner);
|
|
|
|
attr = Py_NewRef(descr);
|
2023-08-13 03:36:46 +08:00
|
|
|
stack_pointer[-1] = attr;
|
2023-03-01 00:49:35 +08:00
|
|
|
next_instr += 9;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(LOAD_ATTR_METHOD_LAZY_DICT) {
|
2023-08-10 02:19:39 +08:00
|
|
|
PyObject *owner;
|
|
|
|
PyObject *attr;
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *self;
|
2023-08-10 02:19:39 +08:00
|
|
|
owner = stack_pointer[-1];
|
2023-02-02 02:56:52 +08:00
|
|
|
uint32_t type_version = read_u32(&next_instr[1].cache);
|
|
|
|
PyObject *descr = read_obj(&next_instr[5].cache);
|
2023-07-10 18:40:35 +08:00
|
|
|
assert(oparg & 1);
|
2023-08-10 02:19:39 +08:00
|
|
|
PyTypeObject *owner_cls = Py_TYPE(owner);
|
|
|
|
DEOPT_IF(owner_cls->tp_version_tag != type_version, LOAD_ATTR);
|
|
|
|
Py_ssize_t dictoffset = owner_cls->tp_dictoffset;
|
2022-11-03 12:31:26 +08:00
|
|
|
assert(dictoffset > 0);
|
2023-08-10 02:19:39 +08:00
|
|
|
PyObject *dict = *(PyObject **)((char *)owner + dictoffset);
|
2022-11-03 12:31:26 +08:00
|
|
|
/* This object has a __dict__, just not yet created */
|
|
|
|
DEOPT_IF(dict != NULL, LOAD_ATTR);
|
|
|
|
STAT_INC(LOAD_ATTR, hit);
|
2023-02-02 02:56:52 +08:00
|
|
|
assert(descr != NULL);
|
|
|
|
assert(_PyType_HasFeature(Py_TYPE(descr), Py_TPFLAGS_METHOD_DESCRIPTOR));
|
2023-08-10 02:19:39 +08:00
|
|
|
attr = Py_NewRef(descr);
|
|
|
|
self = owner;
|
2023-07-17 08:09:11 +08:00
|
|
|
STACK_GROW(1);
|
2023-08-10 02:19:39 +08:00
|
|
|
stack_pointer[-2] = attr;
|
|
|
|
stack_pointer[-1] = self;
|
2023-03-01 00:49:35 +08:00
|
|
|
next_instr += 9;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2023-04-12 19:04:55 +08:00
|
|
|
TARGET(INSTRUMENTED_CALL) {
|
2023-08-10 02:19:39 +08:00
|
|
|
int is_meth = PEEK(oparg + 1) != NULL;
|
2023-04-12 19:04:55 +08:00
|
|
|
int total_args = oparg + is_meth;
|
2023-08-10 02:19:39 +08:00
|
|
|
PyObject *function = PEEK(oparg + 2);
|
2023-04-12 19:04:55 +08:00
|
|
|
PyObject *arg = total_args == 0 ?
|
|
|
|
&_PyInstrumentation_MISSING : PEEK(total_args);
|
|
|
|
int err = _Py_call_instrumentation_2args(
|
|
|
|
tstate, PY_MONITORING_EVENT_CALL,
|
|
|
|
frame, next_instr-1, function, arg);
|
|
|
|
if (err) goto error;
|
|
|
|
_PyCallCache *cache = (_PyCallCache *)next_instr;
|
|
|
|
INCREMENT_ADAPTIVE_COUNTER(cache->counter);
|
|
|
|
GO_TO_INSTRUCTION(CALL);
|
|
|
|
}
|
|
|
|
|
2022-11-03 12:31:26 +08:00
|
|
|
TARGET(CALL) {
|
|
|
|
PREDICTED(CALL);
|
2023-04-06 05:15:49 +08:00
|
|
|
static_assert(INLINE_CACHE_ENTRIES_CALL == 3, "incorrect cache size");
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject **args;
|
2023-08-10 02:19:39 +08:00
|
|
|
PyObject *self_or_null;
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *callable;
|
2023-02-09 03:40:10 +08:00
|
|
|
PyObject *res;
|
2023-08-05 00:35:56 +08:00
|
|
|
args = stack_pointer - oparg;
|
2023-08-10 02:19:39 +08:00
|
|
|
self_or_null = stack_pointer[-1 - oparg];
|
|
|
|
callable = stack_pointer[-2 - oparg];
|
2023-09-14 01:25:45 +08:00
|
|
|
// oparg counts all of the args, but *not* self:
|
2023-02-09 03:40:10 +08:00
|
|
|
int total_args = oparg;
|
2023-08-10 02:19:39 +08:00
|
|
|
if (self_or_null != NULL) {
|
2023-02-09 03:40:10 +08:00
|
|
|
args--;
|
|
|
|
total_args++;
|
|
|
|
}
|
2023-01-20 02:14:55 +08:00
|
|
|
#if ENABLE_SPECIALIZATION
|
2022-11-10 02:50:09 +08:00
|
|
|
_PyCallCache *cache = (_PyCallCache *)next_instr;
|
|
|
|
if (ADAPTIVE_COUNTER_IS_ZERO(cache->counter)) {
|
|
|
|
next_instr--;
|
2023-09-14 01:25:45 +08:00
|
|
|
_Py_Specialize_Call(callable, next_instr, total_args);
|
2022-11-10 02:50:09 +08:00
|
|
|
DISPATCH_SAME_OPARG();
|
|
|
|
}
|
|
|
|
STAT_INC(CALL, deferred);
|
|
|
|
DECREMENT_ADAPTIVE_COUNTER(cache->counter);
|
2023-01-20 02:14:55 +08:00
|
|
|
#endif /* ENABLE_SPECIALIZATION */
|
2023-08-10 02:19:39 +08:00
|
|
|
if (self_or_null == NULL && Py_TYPE(callable) == &PyMethod_Type) {
|
2023-02-09 03:40:10 +08:00
|
|
|
args--;
|
|
|
|
total_args++;
|
|
|
|
PyObject *self = ((PyMethodObject *)callable)->im_self;
|
|
|
|
args[0] = Py_NewRef(self);
|
2023-08-10 02:19:39 +08:00
|
|
|
PyObject *method = ((PyMethodObject *)callable)->im_func;
|
2023-02-09 03:40:10 +08:00
|
|
|
args[-1] = Py_NewRef(method);
|
|
|
|
Py_DECREF(callable);
|
|
|
|
callable = method;
|
|
|
|
}
|
2022-11-03 12:31:26 +08:00
|
|
|
// Check if the call can be inlined or not
|
2023-02-09 03:40:10 +08:00
|
|
|
if (Py_TYPE(callable) == &PyFunction_Type &&
|
2022-11-03 12:31:26 +08:00
|
|
|
tstate->interp->eval_frame == NULL &&
|
2023-02-09 03:40:10 +08:00
|
|
|
((PyFunctionObject *)callable)->vectorcall == _PyFunction_Vectorcall)
|
2022-11-03 12:31:26 +08:00
|
|
|
{
|
2023-02-09 03:40:10 +08:00
|
|
|
int code_flags = ((PyCodeObject*)PyFunction_GET_CODE(callable))->co_flags;
|
|
|
|
PyObject *locals = code_flags & CO_OPTIMIZED ? NULL : Py_NewRef(PyFunction_GET_GLOBALS(callable));
|
2022-11-03 12:31:26 +08:00
|
|
|
_PyInterpreterFrame *new_frame = _PyEvalFramePushAndInit(
|
2023-02-09 03:40:10 +08:00
|
|
|
tstate, (PyFunctionObject *)callable, locals,
|
2023-09-14 01:25:45 +08:00
|
|
|
args, total_args, NULL
|
2022-11-03 12:31:26 +08:00
|
|
|
);
|
2023-02-09 03:40:10 +08:00
|
|
|
// Manipulate stack directly since we leave using DISPATCH_INLINED().
|
|
|
|
STACK_SHRINK(oparg + 2);
|
2022-11-03 12:31:26 +08:00
|
|
|
// The frame has stolen all the arguments from the stack,
|
|
|
|
// so there is no need to clean them up.
|
|
|
|
if (new_frame == NULL) {
|
|
|
|
goto error;
|
|
|
|
}
|
2023-06-15 07:14:22 +08:00
|
|
|
SKIP_OVER(INLINE_CACHE_ENTRIES_CALL);
|
2023-04-13 23:19:07 +08:00
|
|
|
frame->return_offset = 0;
|
2022-11-18 03:36:03 +08:00
|
|
|
DISPATCH_INLINED(new_frame);
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
|
|
|
/* Callable is not a normal Python function */
|
2023-04-12 19:04:55 +08:00
|
|
|
res = PyObject_Vectorcall(
|
|
|
|
callable, args,
|
2023-09-14 01:25:45 +08:00
|
|
|
total_args | PY_VECTORCALL_ARGUMENTS_OFFSET,
|
|
|
|
NULL);
|
2023-04-12 19:04:55 +08:00
|
|
|
if (opcode == INSTRUMENTED_CALL) {
|
|
|
|
PyObject *arg = total_args == 0 ?
|
2023-08-10 02:19:39 +08:00
|
|
|
&_PyInstrumentation_MISSING : args[0];
|
2023-04-12 19:04:55 +08:00
|
|
|
if (res == NULL) {
|
|
|
|
_Py_call_instrumentation_exc2(
|
|
|
|
tstate, PY_MONITORING_EVENT_C_RAISE,
|
|
|
|
frame, next_instr-1, callable, arg);
|
|
|
|
}
|
|
|
|
else {
|
|
|
|
int err = _Py_call_instrumentation_2args(
|
|
|
|
tstate, PY_MONITORING_EVENT_C_RETURN,
|
|
|
|
frame, next_instr-1, callable, arg);
|
|
|
|
if (err < 0) {
|
|
|
|
Py_CLEAR(res);
|
|
|
|
}
|
|
|
|
}
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
|
|
|
assert((res != NULL) ^ (_PyErr_Occurred(tstate) != NULL));
|
2023-02-09 03:40:10 +08:00
|
|
|
Py_DECREF(callable);
|
2022-11-03 12:31:26 +08:00
|
|
|
for (int i = 0; i < total_args; i++) {
|
2023-02-09 03:40:10 +08:00
|
|
|
Py_DECREF(args[i]);
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
2023-02-09 03:40:10 +08:00
|
|
|
if (res == NULL) { STACK_SHRINK(oparg); goto pop_2_error; }
|
|
|
|
STACK_SHRINK(oparg);
|
|
|
|
STACK_SHRINK(1);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = res;
|
2023-04-06 05:15:49 +08:00
|
|
|
next_instr += 3;
|
2022-11-03 12:31:26 +08:00
|
|
|
CHECK_EVAL_BREAKER();
|
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2023-02-09 03:40:10 +08:00
|
|
|
TARGET(CALL_BOUND_METHOD_EXACT_ARGS) {
|
2023-08-10 02:19:39 +08:00
|
|
|
PyObject *null;
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *callable;
|
2023-08-25 08:36:00 +08:00
|
|
|
PyObject *self;
|
|
|
|
PyObject *self_or_null;
|
|
|
|
PyObject *func;
|
|
|
|
PyObject **args;
|
|
|
|
_PyInterpreterFrame *new_frame;
|
|
|
|
// _CHECK_PEP_523
|
|
|
|
{
|
|
|
|
DEOPT_IF(tstate->interp->eval_frame, CALL);
|
|
|
|
}
|
|
|
|
// _CHECK_CALL_BOUND_METHOD_EXACT_ARGS
|
2023-08-10 02:19:39 +08:00
|
|
|
null = stack_pointer[-1 - oparg];
|
|
|
|
callable = stack_pointer[-2 - oparg];
|
2023-08-25 08:36:00 +08:00
|
|
|
{
|
|
|
|
DEOPT_IF(null != NULL, CALL);
|
|
|
|
DEOPT_IF(Py_TYPE(callable) != &PyMethod_Type, CALL);
|
|
|
|
}
|
|
|
|
// _INIT_CALL_BOUND_METHOD_EXACT_ARGS
|
|
|
|
{
|
|
|
|
STAT_INC(CALL, hit);
|
|
|
|
self = Py_NewRef(((PyMethodObject *)callable)->im_self);
|
|
|
|
stack_pointer[-1 - oparg] = self; // Patch stack as it is used by _INIT_CALL_PY_EXACT_ARGS
|
|
|
|
func = Py_NewRef(((PyMethodObject *)callable)->im_func);
|
|
|
|
stack_pointer[-2 - oparg] = func; // This is used by CALL, upon deoptimization
|
|
|
|
Py_DECREF(callable);
|
|
|
|
}
|
|
|
|
// _CHECK_FUNCTION_EXACT_ARGS
|
|
|
|
self_or_null = self;
|
|
|
|
callable = func;
|
|
|
|
{
|
|
|
|
uint32_t func_version = read_u32(&next_instr[1].cache);
|
|
|
|
DEOPT_IF(!PyFunction_Check(callable), CALL);
|
|
|
|
PyFunctionObject *func = (PyFunctionObject *)callable;
|
|
|
|
DEOPT_IF(func->func_version != func_version, CALL);
|
|
|
|
PyCodeObject *code = (PyCodeObject *)func->func_code;
|
|
|
|
DEOPT_IF(code->co_argcount != oparg + (self_or_null != NULL), CALL);
|
|
|
|
}
|
|
|
|
// _CHECK_STACK_SPACE
|
|
|
|
{
|
|
|
|
PyFunctionObject *func = (PyFunctionObject *)callable;
|
|
|
|
PyCodeObject *code = (PyCodeObject *)func->func_code;
|
|
|
|
DEOPT_IF(!_PyThreadState_HasStackSpace(tstate, code->co_framesize), CALL);
|
2023-09-04 18:36:57 +08:00
|
|
|
DEOPT_IF(tstate->py_recursion_remaining <= 1, CALL);
|
2023-08-25 08:36:00 +08:00
|
|
|
}
|
|
|
|
// _INIT_CALL_PY_EXACT_ARGS
|
|
|
|
args = stack_pointer - oparg;
|
|
|
|
{
|
|
|
|
int argcount = oparg;
|
|
|
|
if (self_or_null != NULL) {
|
|
|
|
args--;
|
|
|
|
argcount++;
|
|
|
|
}
|
|
|
|
STAT_INC(CALL, hit);
|
|
|
|
PyFunctionObject *func = (PyFunctionObject *)callable;
|
|
|
|
new_frame = _PyFrame_PushUnchecked(tstate, func, argcount);
|
|
|
|
for (int i = 0; i < argcount; i++) {
|
|
|
|
new_frame->localsplus[i] = args[i];
|
|
|
|
}
|
|
|
|
}
|
2023-09-12 06:39:19 +08:00
|
|
|
// _SAVE_CURRENT_IP
|
2023-08-25 08:36:00 +08:00
|
|
|
next_instr += 3;
|
|
|
|
{
|
|
|
|
#if TIER_ONE
|
|
|
|
frame->prev_instr = next_instr - 1;
|
|
|
|
#endif
|
|
|
|
#if TIER_TWO
|
2023-09-12 06:39:19 +08:00
|
|
|
// Relies on a preceding _SET_IP
|
2023-08-25 08:36:00 +08:00
|
|
|
frame->prev_instr--;
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
// _PUSH_FRAME
|
2023-08-05 00:35:56 +08:00
|
|
|
STACK_SHRINK(oparg);
|
2023-08-25 08:36:00 +08:00
|
|
|
STACK_SHRINK(2);
|
|
|
|
{
|
|
|
|
// Write it out explicitly because it's subtly different.
|
|
|
|
// Eventually this should be the only occurrence of this code.
|
|
|
|
frame->return_offset = 0;
|
|
|
|
assert(tstate->interp->eval_frame == NULL);
|
2023-09-04 18:36:57 +08:00
|
|
|
STORE_SP();
|
2023-08-25 08:36:00 +08:00
|
|
|
new_frame->previous = frame;
|
|
|
|
CALL_STAT_INC(inlined_py_calls);
|
|
|
|
frame = tstate->current_frame = new_frame;
|
2023-09-04 18:36:57 +08:00
|
|
|
tstate->py_recursion_remaining--;
|
|
|
|
LOAD_SP();
|
|
|
|
LOAD_IP();
|
|
|
|
#if LLTRACE && TIER_ONE
|
|
|
|
lltrace = maybe_lltrace_resume_frame(frame, &entry_frame, GLOBALS());
|
|
|
|
if (lltrace < 0) {
|
|
|
|
goto exit_unwind;
|
|
|
|
}
|
|
|
|
#endif
|
2023-08-25 08:36:00 +08:00
|
|
|
}
|
2023-08-31 18:34:52 +08:00
|
|
|
DISPATCH();
|
2023-02-09 03:40:10 +08:00
|
|
|
}
|
|
|
|
|
2022-11-03 12:31:26 +08:00
|
|
|
TARGET(CALL_PY_EXACT_ARGS) {
|
2023-08-10 02:19:39 +08:00
|
|
|
PyObject *self_or_null;
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *callable;
|
2023-08-17 07:26:43 +08:00
|
|
|
PyObject **args;
|
|
|
|
_PyInterpreterFrame *new_frame;
|
|
|
|
// _CHECK_PEP_523
|
|
|
|
{
|
|
|
|
DEOPT_IF(tstate->interp->eval_frame, CALL);
|
|
|
|
}
|
|
|
|
// _CHECK_FUNCTION_EXACT_ARGS
|
|
|
|
self_or_null = stack_pointer[-1 - oparg];
|
|
|
|
callable = stack_pointer[-2 - oparg];
|
|
|
|
{
|
|
|
|
uint32_t func_version = read_u32(&next_instr[1].cache);
|
|
|
|
DEOPT_IF(!PyFunction_Check(callable), CALL);
|
|
|
|
PyFunctionObject *func = (PyFunctionObject *)callable;
|
|
|
|
DEOPT_IF(func->func_version != func_version, CALL);
|
|
|
|
PyCodeObject *code = (PyCodeObject *)func->func_code;
|
|
|
|
DEOPT_IF(code->co_argcount != oparg + (self_or_null != NULL), CALL);
|
|
|
|
}
|
|
|
|
// _CHECK_STACK_SPACE
|
|
|
|
{
|
|
|
|
PyFunctionObject *func = (PyFunctionObject *)callable;
|
|
|
|
PyCodeObject *code = (PyCodeObject *)func->func_code;
|
|
|
|
DEOPT_IF(!_PyThreadState_HasStackSpace(tstate, code->co_framesize), CALL);
|
2023-09-04 18:36:57 +08:00
|
|
|
DEOPT_IF(tstate->py_recursion_remaining <= 1, CALL);
|
2023-08-17 07:26:43 +08:00
|
|
|
}
|
|
|
|
// _INIT_CALL_PY_EXACT_ARGS
|
2023-08-05 00:35:56 +08:00
|
|
|
args = stack_pointer - oparg;
|
2023-08-17 07:26:43 +08:00
|
|
|
{
|
|
|
|
int argcount = oparg;
|
|
|
|
if (self_or_null != NULL) {
|
|
|
|
args--;
|
|
|
|
argcount++;
|
|
|
|
}
|
|
|
|
STAT_INC(CALL, hit);
|
|
|
|
PyFunctionObject *func = (PyFunctionObject *)callable;
|
|
|
|
new_frame = _PyFrame_PushUnchecked(tstate, func, argcount);
|
|
|
|
for (int i = 0; i < argcount; i++) {
|
|
|
|
new_frame->localsplus[i] = args[i];
|
|
|
|
}
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
2023-09-12 06:39:19 +08:00
|
|
|
// _SAVE_CURRENT_IP
|
2023-08-17 07:26:43 +08:00
|
|
|
next_instr += 3;
|
|
|
|
{
|
|
|
|
#if TIER_ONE
|
|
|
|
frame->prev_instr = next_instr - 1;
|
|
|
|
#endif
|
|
|
|
#if TIER_TWO
|
2023-09-12 06:39:19 +08:00
|
|
|
// Relies on a preceding _SET_IP
|
2023-08-17 07:26:43 +08:00
|
|
|
frame->prev_instr--;
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
// _PUSH_FRAME
|
2023-08-05 00:35:56 +08:00
|
|
|
STACK_SHRINK(oparg);
|
2023-08-17 07:26:43 +08:00
|
|
|
STACK_SHRINK(2);
|
|
|
|
{
|
|
|
|
// Write it out explicitly because it's subtly different.
|
|
|
|
// Eventually this should be the only occurrence of this code.
|
|
|
|
frame->return_offset = 0;
|
|
|
|
assert(tstate->interp->eval_frame == NULL);
|
2023-09-04 18:36:57 +08:00
|
|
|
STORE_SP();
|
2023-08-17 07:26:43 +08:00
|
|
|
new_frame->previous = frame;
|
|
|
|
CALL_STAT_INC(inlined_py_calls);
|
2023-08-17 18:16:03 +08:00
|
|
|
frame = tstate->current_frame = new_frame;
|
2023-09-04 18:36:57 +08:00
|
|
|
tstate->py_recursion_remaining--;
|
|
|
|
LOAD_SP();
|
|
|
|
LOAD_IP();
|
|
|
|
#if LLTRACE && TIER_ONE
|
|
|
|
lltrace = maybe_lltrace_resume_frame(frame, &entry_frame, GLOBALS());
|
|
|
|
if (lltrace < 0) {
|
|
|
|
goto exit_unwind;
|
|
|
|
}
|
|
|
|
#endif
|
2023-08-17 07:26:43 +08:00
|
|
|
}
|
2023-08-31 18:34:52 +08:00
|
|
|
DISPATCH();
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(CALL_PY_WITH_DEFAULTS) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject **args;
|
2023-08-10 02:19:39 +08:00
|
|
|
PyObject *self_or_null;
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *callable;
|
|
|
|
args = stack_pointer - oparg;
|
2023-08-10 02:19:39 +08:00
|
|
|
self_or_null = stack_pointer[-1 - oparg];
|
|
|
|
callable = stack_pointer[-2 - oparg];
|
2023-02-09 03:40:10 +08:00
|
|
|
uint32_t func_version = read_u32(&next_instr[1].cache);
|
2022-11-03 12:31:26 +08:00
|
|
|
DEOPT_IF(tstate->interp->eval_frame, CALL);
|
2023-02-09 03:40:10 +08:00
|
|
|
int argcount = oparg;
|
2023-08-10 02:19:39 +08:00
|
|
|
if (self_or_null != NULL) {
|
2023-02-09 03:40:10 +08:00
|
|
|
args--;
|
|
|
|
argcount++;
|
|
|
|
}
|
2022-11-03 12:31:26 +08:00
|
|
|
DEOPT_IF(!PyFunction_Check(callable), CALL);
|
|
|
|
PyFunctionObject *func = (PyFunctionObject *)callable;
|
2023-02-09 03:40:10 +08:00
|
|
|
DEOPT_IF(func->func_version != func_version, CALL);
|
2022-11-03 12:31:26 +08:00
|
|
|
PyCodeObject *code = (PyCodeObject *)func->func_code;
|
2023-04-06 05:15:49 +08:00
|
|
|
assert(func->func_defaults);
|
|
|
|
assert(PyTuple_CheckExact(func->func_defaults));
|
|
|
|
int defcount = (int)PyTuple_GET_SIZE(func->func_defaults);
|
|
|
|
assert(defcount <= code->co_argcount);
|
|
|
|
int min_args = code->co_argcount - defcount;
|
2022-11-03 12:31:26 +08:00
|
|
|
DEOPT_IF(argcount > code->co_argcount, CALL);
|
2023-02-09 03:40:10 +08:00
|
|
|
DEOPT_IF(argcount < min_args, CALL);
|
2022-11-03 12:31:26 +08:00
|
|
|
DEOPT_IF(!_PyThreadState_HasStackSpace(tstate, code->co_framesize), CALL);
|
|
|
|
STAT_INC(CALL, hit);
|
2023-01-06 22:55:56 +08:00
|
|
|
_PyInterpreterFrame *new_frame = _PyFrame_PushUnchecked(tstate, func, code->co_argcount);
|
2022-11-03 12:31:26 +08:00
|
|
|
for (int i = 0; i < argcount; i++) {
|
2023-02-09 03:40:10 +08:00
|
|
|
new_frame->localsplus[i] = args[i];
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
|
|
|
for (int i = argcount; i < code->co_argcount; i++) {
|
2023-02-09 03:40:10 +08:00
|
|
|
PyObject *def = PyTuple_GET_ITEM(func->func_defaults, i - min_args);
|
2022-11-10 18:25:33 +08:00
|
|
|
new_frame->localsplus[i] = Py_NewRef(def);
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
2023-02-09 03:40:10 +08:00
|
|
|
// Manipulate stack and cache directly since we leave using DISPATCH_INLINED().
|
|
|
|
STACK_SHRINK(oparg + 2);
|
2023-06-15 07:14:22 +08:00
|
|
|
SKIP_OVER(INLINE_CACHE_ENTRIES_CALL);
|
2023-04-13 23:19:07 +08:00
|
|
|
frame->return_offset = 0;
|
2022-11-18 03:36:03 +08:00
|
|
|
DISPATCH_INLINED(new_frame);
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
|
|
|
|
2023-09-14 01:25:45 +08:00
|
|
|
TARGET(CALL_TYPE_1) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject **args;
|
|
|
|
PyObject *null;
|
2023-08-10 02:19:39 +08:00
|
|
|
PyObject *callable;
|
2023-02-09 03:40:10 +08:00
|
|
|
PyObject *res;
|
2023-08-05 00:35:56 +08:00
|
|
|
args = stack_pointer - oparg;
|
2023-08-10 02:19:39 +08:00
|
|
|
null = stack_pointer[-1 - oparg];
|
|
|
|
callable = stack_pointer[-2 - oparg];
|
2022-11-03 12:31:26 +08:00
|
|
|
assert(oparg == 1);
|
2023-02-09 03:40:10 +08:00
|
|
|
DEOPT_IF(null != NULL, CALL);
|
|
|
|
PyObject *obj = args[0];
|
2022-11-03 12:31:26 +08:00
|
|
|
DEOPT_IF(callable != (PyObject *)&PyType_Type, CALL);
|
|
|
|
STAT_INC(CALL, hit);
|
2023-02-09 03:40:10 +08:00
|
|
|
res = Py_NewRef(Py_TYPE(obj));
|
2022-11-03 12:31:26 +08:00
|
|
|
Py_DECREF(obj);
|
2023-02-09 03:40:10 +08:00
|
|
|
Py_DECREF(&PyType_Type); // I.e., callable
|
|
|
|
STACK_SHRINK(oparg);
|
|
|
|
STACK_SHRINK(1);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = res;
|
2023-04-06 05:15:49 +08:00
|
|
|
next_instr += 3;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2023-09-14 01:25:45 +08:00
|
|
|
TARGET(CALL_STR_1) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject **args;
|
|
|
|
PyObject *null;
|
2023-08-10 02:19:39 +08:00
|
|
|
PyObject *callable;
|
2023-02-09 03:40:10 +08:00
|
|
|
PyObject *res;
|
2023-08-05 00:35:56 +08:00
|
|
|
args = stack_pointer - oparg;
|
2023-08-10 02:19:39 +08:00
|
|
|
null = stack_pointer[-1 - oparg];
|
|
|
|
callable = stack_pointer[-2 - oparg];
|
2022-11-03 12:31:26 +08:00
|
|
|
assert(oparg == 1);
|
2023-02-09 03:40:10 +08:00
|
|
|
DEOPT_IF(null != NULL, CALL);
|
2022-11-03 12:31:26 +08:00
|
|
|
DEOPT_IF(callable != (PyObject *)&PyUnicode_Type, CALL);
|
|
|
|
STAT_INC(CALL, hit);
|
2023-02-09 03:40:10 +08:00
|
|
|
PyObject *arg = args[0];
|
|
|
|
res = PyObject_Str(arg);
|
2022-11-03 12:31:26 +08:00
|
|
|
Py_DECREF(arg);
|
2023-02-09 03:40:10 +08:00
|
|
|
Py_DECREF(&PyUnicode_Type); // I.e., callable
|
|
|
|
if (res == NULL) { STACK_SHRINK(oparg); goto pop_2_error; }
|
|
|
|
STACK_SHRINK(oparg);
|
|
|
|
STACK_SHRINK(1);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = res;
|
2023-04-06 05:15:49 +08:00
|
|
|
next_instr += 3;
|
2022-11-03 12:31:26 +08:00
|
|
|
CHECK_EVAL_BREAKER();
|
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2023-09-14 01:25:45 +08:00
|
|
|
TARGET(CALL_TUPLE_1) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject **args;
|
|
|
|
PyObject *null;
|
2023-08-10 02:19:39 +08:00
|
|
|
PyObject *callable;
|
2023-02-09 03:40:10 +08:00
|
|
|
PyObject *res;
|
2023-08-05 00:35:56 +08:00
|
|
|
args = stack_pointer - oparg;
|
2023-08-10 02:19:39 +08:00
|
|
|
null = stack_pointer[-1 - oparg];
|
|
|
|
callable = stack_pointer[-2 - oparg];
|
2022-11-03 12:31:26 +08:00
|
|
|
assert(oparg == 1);
|
2023-02-09 03:40:10 +08:00
|
|
|
DEOPT_IF(null != NULL, CALL);
|
2022-11-03 12:31:26 +08:00
|
|
|
DEOPT_IF(callable != (PyObject *)&PyTuple_Type, CALL);
|
|
|
|
STAT_INC(CALL, hit);
|
2023-02-09 03:40:10 +08:00
|
|
|
PyObject *arg = args[0];
|
|
|
|
res = PySequence_Tuple(arg);
|
2022-11-03 12:31:26 +08:00
|
|
|
Py_DECREF(arg);
|
2023-02-09 03:40:10 +08:00
|
|
|
Py_DECREF(&PyTuple_Type); // I.e., tuple
|
|
|
|
if (res == NULL) { STACK_SHRINK(oparg); goto pop_2_error; }
|
|
|
|
STACK_SHRINK(oparg);
|
|
|
|
STACK_SHRINK(1);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = res;
|
2023-04-06 05:15:49 +08:00
|
|
|
next_instr += 3;
|
2022-11-03 12:31:26 +08:00
|
|
|
CHECK_EVAL_BREAKER();
|
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2023-09-14 01:25:45 +08:00
|
|
|
TARGET(CALL_ALLOC_AND_ENTER_INIT) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject **args;
|
|
|
|
PyObject *null;
|
2023-08-10 02:19:39 +08:00
|
|
|
PyObject *callable;
|
2023-08-05 00:35:56 +08:00
|
|
|
args = stack_pointer - oparg;
|
2023-08-10 02:19:39 +08:00
|
|
|
null = stack_pointer[-1 - oparg];
|
|
|
|
callable = stack_pointer[-2 - oparg];
|
2023-06-22 16:48:19 +08:00
|
|
|
/* This instruction does the following:
|
|
|
|
* 1. Creates the object (by calling ``object.__new__``)
|
|
|
|
* 2. Pushes a shim frame to the frame stack (to cleanup after ``__init__``)
|
|
|
|
* 3. Pushes the frame for ``__init__`` to the frame stack
|
|
|
|
* */
|
|
|
|
_PyCallCache *cache = (_PyCallCache *)next_instr;
|
|
|
|
DEOPT_IF(null != NULL, CALL);
|
|
|
|
DEOPT_IF(!PyType_Check(callable), CALL);
|
|
|
|
PyTypeObject *tp = (PyTypeObject *)callable;
|
|
|
|
DEOPT_IF(tp->tp_version_tag != read_u32(cache->func_version), CALL);
|
|
|
|
PyHeapTypeObject *cls = (PyHeapTypeObject *)callable;
|
|
|
|
PyFunctionObject *init = (PyFunctionObject *)cls->_spec_cache.init;
|
|
|
|
PyCodeObject *code = (PyCodeObject *)init->func_code;
|
|
|
|
DEOPT_IF(code->co_argcount != oparg+1, CALL);
|
|
|
|
DEOPT_IF(!_PyThreadState_HasStackSpace(tstate, code->co_framesize + _Py_InitCleanup.co_framesize), CALL);
|
|
|
|
STAT_INC(CALL, hit);
|
|
|
|
PyObject *self = _PyType_NewManagedObject(tp);
|
|
|
|
if (self == NULL) {
|
|
|
|
goto error;
|
|
|
|
}
|
|
|
|
Py_DECREF(tp);
|
|
|
|
_PyInterpreterFrame *shim = _PyFrame_PushTrampolineUnchecked(
|
|
|
|
tstate, (PyCodeObject *)&_Py_InitCleanup, 1, 0);
|
|
|
|
assert(_PyCode_CODE((PyCodeObject *)shim->f_executable)[1].op.code == EXIT_INIT_CHECK);
|
|
|
|
/* Push self onto stack of shim */
|
|
|
|
Py_INCREF(self);
|
|
|
|
shim->localsplus[0] = self;
|
|
|
|
Py_INCREF(init);
|
|
|
|
_PyInterpreterFrame *init_frame = _PyFrame_PushUnchecked(tstate, init, oparg+1);
|
|
|
|
/* Copy self followed by args to __init__ frame */
|
|
|
|
init_frame->localsplus[0] = self;
|
|
|
|
for (int i = 0; i < oparg; i++) {
|
|
|
|
init_frame->localsplus[i+1] = args[i];
|
|
|
|
}
|
|
|
|
SKIP_OVER(INLINE_CACHE_ENTRIES_CALL);
|
|
|
|
frame->prev_instr = next_instr - 1;
|
|
|
|
frame->return_offset = 0;
|
|
|
|
STACK_SHRINK(oparg+2);
|
|
|
|
_PyFrame_SetStackPointer(frame, stack_pointer);
|
|
|
|
/* Link frames */
|
|
|
|
init_frame->previous = shim;
|
|
|
|
shim->previous = frame;
|
2023-08-17 18:16:03 +08:00
|
|
|
frame = tstate->current_frame = init_frame;
|
2023-06-22 16:48:19 +08:00
|
|
|
CALL_STAT_INC(inlined_py_calls);
|
2023-07-07 18:09:26 +08:00
|
|
|
/* Account for pushing the extra frame.
|
|
|
|
* We don't check recursion depth here,
|
|
|
|
* as it will be checked after start_frame */
|
|
|
|
tstate->py_recursion_remaining--;
|
2023-06-22 16:48:19 +08:00
|
|
|
goto start_frame;
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(EXIT_INIT_CHECK) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *should_be_none;
|
|
|
|
should_be_none = stack_pointer[-1];
|
2023-06-22 16:48:19 +08:00
|
|
|
assert(STACK_LEVEL() == 2);
|
|
|
|
if (should_be_none != Py_None) {
|
|
|
|
PyErr_Format(PyExc_TypeError,
|
|
|
|
"__init__() should return None, not '%.200s'",
|
|
|
|
Py_TYPE(should_be_none)->tp_name);
|
|
|
|
goto error;
|
|
|
|
}
|
|
|
|
STACK_SHRINK(1);
|
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2022-11-03 12:31:26 +08:00
|
|
|
TARGET(CALL_BUILTIN_CLASS) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject **args;
|
2023-08-10 02:19:39 +08:00
|
|
|
PyObject *self_or_null;
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *callable;
|
2023-02-09 03:40:10 +08:00
|
|
|
PyObject *res;
|
2023-08-05 00:35:56 +08:00
|
|
|
args = stack_pointer - oparg;
|
2023-08-10 02:19:39 +08:00
|
|
|
self_or_null = stack_pointer[-1 - oparg];
|
|
|
|
callable = stack_pointer[-2 - oparg];
|
2023-02-09 03:40:10 +08:00
|
|
|
int total_args = oparg;
|
2023-08-10 02:19:39 +08:00
|
|
|
if (self_or_null != NULL) {
|
2023-02-09 03:40:10 +08:00
|
|
|
args--;
|
|
|
|
total_args++;
|
|
|
|
}
|
2022-11-03 12:31:26 +08:00
|
|
|
DEOPT_IF(!PyType_Check(callable), CALL);
|
|
|
|
PyTypeObject *tp = (PyTypeObject *)callable;
|
|
|
|
DEOPT_IF(tp->tp_vectorcall == NULL, CALL);
|
|
|
|
STAT_INC(CALL, hit);
|
2023-09-14 01:25:45 +08:00
|
|
|
res = tp->tp_vectorcall((PyObject *)tp, args, total_args, NULL);
|
2022-11-03 12:31:26 +08:00
|
|
|
/* Free the arguments. */
|
|
|
|
for (int i = 0; i < total_args; i++) {
|
2023-02-09 03:40:10 +08:00
|
|
|
Py_DECREF(args[i]);
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
|
|
|
Py_DECREF(tp);
|
2023-02-09 03:40:10 +08:00
|
|
|
if (res == NULL) { STACK_SHRINK(oparg); goto pop_2_error; }
|
|
|
|
STACK_SHRINK(oparg);
|
|
|
|
STACK_SHRINK(1);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = res;
|
2023-04-06 05:15:49 +08:00
|
|
|
next_instr += 3;
|
2022-11-03 12:31:26 +08:00
|
|
|
CHECK_EVAL_BREAKER();
|
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2023-09-14 01:25:45 +08:00
|
|
|
TARGET(CALL_BUILTIN_O) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject **args;
|
2023-08-10 02:19:39 +08:00
|
|
|
PyObject *self_or_null;
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *callable;
|
2023-02-09 03:40:10 +08:00
|
|
|
PyObject *res;
|
2023-08-05 00:35:56 +08:00
|
|
|
args = stack_pointer - oparg;
|
2023-08-10 02:19:39 +08:00
|
|
|
self_or_null = stack_pointer[-1 - oparg];
|
|
|
|
callable = stack_pointer[-2 - oparg];
|
2022-11-03 12:31:26 +08:00
|
|
|
/* Builtin METH_O functions */
|
2023-02-09 03:40:10 +08:00
|
|
|
int total_args = oparg;
|
2023-08-10 02:19:39 +08:00
|
|
|
if (self_or_null != NULL) {
|
2023-02-09 03:40:10 +08:00
|
|
|
args--;
|
|
|
|
total_args++;
|
|
|
|
}
|
2022-11-03 12:31:26 +08:00
|
|
|
DEOPT_IF(total_args != 1, CALL);
|
|
|
|
DEOPT_IF(!PyCFunction_CheckExact(callable), CALL);
|
|
|
|
DEOPT_IF(PyCFunction_GET_FLAGS(callable) != METH_O, CALL);
|
|
|
|
STAT_INC(CALL, hit);
|
|
|
|
PyCFunction cfunc = PyCFunction_GET_FUNCTION(callable);
|
|
|
|
// This is slower but CPython promises to check all non-vectorcall
|
|
|
|
// function calls.
|
|
|
|
if (_Py_EnterRecursiveCallTstate(tstate, " while calling a Python object")) {
|
|
|
|
goto error;
|
|
|
|
}
|
2023-02-09 03:40:10 +08:00
|
|
|
PyObject *arg = args[0];
|
|
|
|
res = _PyCFunction_TrampolineCall(cfunc, PyCFunction_GET_SELF(callable), arg);
|
2022-11-03 12:31:26 +08:00
|
|
|
_Py_LeaveRecursiveCallTstate(tstate);
|
|
|
|
assert((res != NULL) ^ (_PyErr_Occurred(tstate) != NULL));
|
|
|
|
|
|
|
|
Py_DECREF(arg);
|
|
|
|
Py_DECREF(callable);
|
2023-02-09 03:40:10 +08:00
|
|
|
if (res == NULL) { STACK_SHRINK(oparg); goto pop_2_error; }
|
|
|
|
STACK_SHRINK(oparg);
|
|
|
|
STACK_SHRINK(1);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = res;
|
2023-04-06 05:15:49 +08:00
|
|
|
next_instr += 3;
|
2022-11-03 12:31:26 +08:00
|
|
|
CHECK_EVAL_BREAKER();
|
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2023-09-14 01:25:45 +08:00
|
|
|
TARGET(CALL_BUILTIN_FAST) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject **args;
|
2023-08-10 02:19:39 +08:00
|
|
|
PyObject *self_or_null;
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *callable;
|
2023-02-09 03:40:10 +08:00
|
|
|
PyObject *res;
|
2023-08-05 00:35:56 +08:00
|
|
|
args = stack_pointer - oparg;
|
2023-08-10 02:19:39 +08:00
|
|
|
self_or_null = stack_pointer[-1 - oparg];
|
|
|
|
callable = stack_pointer[-2 - oparg];
|
2022-11-03 12:31:26 +08:00
|
|
|
/* Builtin METH_FASTCALL functions, without keywords */
|
2023-02-09 03:40:10 +08:00
|
|
|
int total_args = oparg;
|
2023-08-10 02:19:39 +08:00
|
|
|
if (self_or_null != NULL) {
|
2023-02-09 03:40:10 +08:00
|
|
|
args--;
|
|
|
|
total_args++;
|
|
|
|
}
|
2022-11-03 12:31:26 +08:00
|
|
|
DEOPT_IF(!PyCFunction_CheckExact(callable), CALL);
|
2023-02-09 03:40:10 +08:00
|
|
|
DEOPT_IF(PyCFunction_GET_FLAGS(callable) != METH_FASTCALL, CALL);
|
2022-11-03 12:31:26 +08:00
|
|
|
STAT_INC(CALL, hit);
|
|
|
|
PyCFunction cfunc = PyCFunction_GET_FUNCTION(callable);
|
|
|
|
/* res = func(self, args, nargs) */
|
2023-02-09 03:40:10 +08:00
|
|
|
res = ((_PyCFunctionFast)(void(*)(void))cfunc)(
|
2022-11-03 12:31:26 +08:00
|
|
|
PyCFunction_GET_SELF(callable),
|
2023-02-09 03:40:10 +08:00
|
|
|
args,
|
2022-11-03 12:31:26 +08:00
|
|
|
total_args);
|
|
|
|
assert((res != NULL) ^ (_PyErr_Occurred(tstate) != NULL));
|
|
|
|
|
|
|
|
/* Free the arguments. */
|
|
|
|
for (int i = 0; i < total_args; i++) {
|
2023-02-09 03:40:10 +08:00
|
|
|
Py_DECREF(args[i]);
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
|
|
|
Py_DECREF(callable);
|
2023-02-09 03:40:10 +08:00
|
|
|
if (res == NULL) { STACK_SHRINK(oparg); goto pop_2_error; }
|
2022-11-03 12:31:26 +08:00
|
|
|
/* Not deopting because this doesn't mean our optimization was
|
|
|
|
wrong. `res` can be NULL for valid reasons. Eg. getattr(x,
|
|
|
|
'invalid'). In those cases an exception is set, so we must
|
|
|
|
handle it.
|
|
|
|
*/
|
2023-02-09 03:40:10 +08:00
|
|
|
STACK_SHRINK(oparg);
|
|
|
|
STACK_SHRINK(1);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = res;
|
2023-04-06 05:15:49 +08:00
|
|
|
next_instr += 3;
|
2022-11-03 12:31:26 +08:00
|
|
|
CHECK_EVAL_BREAKER();
|
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(CALL_BUILTIN_FAST_WITH_KEYWORDS) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject **args;
|
2023-08-10 02:19:39 +08:00
|
|
|
PyObject *self_or_null;
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *callable;
|
2023-02-09 03:40:10 +08:00
|
|
|
PyObject *res;
|
2023-08-05 00:35:56 +08:00
|
|
|
args = stack_pointer - oparg;
|
2023-08-10 02:19:39 +08:00
|
|
|
self_or_null = stack_pointer[-1 - oparg];
|
|
|
|
callable = stack_pointer[-2 - oparg];
|
2022-11-03 12:31:26 +08:00
|
|
|
/* Builtin METH_FASTCALL | METH_KEYWORDS functions */
|
2023-02-09 03:40:10 +08:00
|
|
|
int total_args = oparg;
|
2023-08-10 02:19:39 +08:00
|
|
|
if (self_or_null != NULL) {
|
2023-02-09 03:40:10 +08:00
|
|
|
args--;
|
|
|
|
total_args++;
|
|
|
|
}
|
2022-11-03 12:31:26 +08:00
|
|
|
DEOPT_IF(!PyCFunction_CheckExact(callable), CALL);
|
|
|
|
DEOPT_IF(PyCFunction_GET_FLAGS(callable) !=
|
|
|
|
(METH_FASTCALL | METH_KEYWORDS), CALL);
|
|
|
|
STAT_INC(CALL, hit);
|
|
|
|
/* res = func(self, args, nargs, kwnames) */
|
|
|
|
_PyCFunctionFastWithKeywords cfunc =
|
|
|
|
(_PyCFunctionFastWithKeywords)(void(*)(void))
|
|
|
|
PyCFunction_GET_FUNCTION(callable);
|
2023-09-14 01:25:45 +08:00
|
|
|
res = cfunc(PyCFunction_GET_SELF(callable), args, total_args, NULL);
|
2022-11-03 12:31:26 +08:00
|
|
|
assert((res != NULL) ^ (_PyErr_Occurred(tstate) != NULL));
|
|
|
|
|
|
|
|
/* Free the arguments. */
|
|
|
|
for (int i = 0; i < total_args; i++) {
|
2023-02-09 03:40:10 +08:00
|
|
|
Py_DECREF(args[i]);
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
|
|
|
Py_DECREF(callable);
|
2023-02-09 03:40:10 +08:00
|
|
|
if (res == NULL) { STACK_SHRINK(oparg); goto pop_2_error; }
|
|
|
|
STACK_SHRINK(oparg);
|
|
|
|
STACK_SHRINK(1);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = res;
|
2023-04-06 05:15:49 +08:00
|
|
|
next_instr += 3;
|
2022-11-03 12:31:26 +08:00
|
|
|
CHECK_EVAL_BREAKER();
|
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2023-09-14 01:25:45 +08:00
|
|
|
TARGET(CALL_LEN) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject **args;
|
2023-08-10 02:19:39 +08:00
|
|
|
PyObject *self_or_null;
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *callable;
|
2023-02-09 03:40:10 +08:00
|
|
|
PyObject *res;
|
2023-08-05 00:35:56 +08:00
|
|
|
args = stack_pointer - oparg;
|
2023-08-10 02:19:39 +08:00
|
|
|
self_or_null = stack_pointer[-1 - oparg];
|
|
|
|
callable = stack_pointer[-2 - oparg];
|
2022-11-03 12:31:26 +08:00
|
|
|
/* len(o) */
|
2023-02-09 03:40:10 +08:00
|
|
|
int total_args = oparg;
|
2023-08-10 02:19:39 +08:00
|
|
|
if (self_or_null != NULL) {
|
2023-02-09 03:40:10 +08:00
|
|
|
args--;
|
|
|
|
total_args++;
|
|
|
|
}
|
2022-11-03 12:31:26 +08:00
|
|
|
DEOPT_IF(total_args != 1, CALL);
|
2023-08-01 05:18:38 +08:00
|
|
|
PyInterpreterState *interp = tstate->interp;
|
2022-11-03 12:31:26 +08:00
|
|
|
DEOPT_IF(callable != interp->callable_cache.len, CALL);
|
|
|
|
STAT_INC(CALL, hit);
|
2023-02-09 03:40:10 +08:00
|
|
|
PyObject *arg = args[0];
|
2022-11-03 12:31:26 +08:00
|
|
|
Py_ssize_t len_i = PyObject_Length(arg);
|
|
|
|
if (len_i < 0) {
|
|
|
|
goto error;
|
|
|
|
}
|
2023-02-09 03:40:10 +08:00
|
|
|
res = PyLong_FromSsize_t(len_i);
|
2022-11-03 12:31:26 +08:00
|
|
|
assert((res != NULL) ^ (_PyErr_Occurred(tstate) != NULL));
|
|
|
|
|
|
|
|
Py_DECREF(callable);
|
|
|
|
Py_DECREF(arg);
|
2023-02-09 03:40:10 +08:00
|
|
|
if (res == NULL) { STACK_SHRINK(oparg); goto pop_2_error; }
|
|
|
|
STACK_SHRINK(oparg);
|
|
|
|
STACK_SHRINK(1);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = res;
|
2023-04-06 05:15:49 +08:00
|
|
|
next_instr += 3;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2023-09-14 01:25:45 +08:00
|
|
|
TARGET(CALL_ISINSTANCE) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject **args;
|
2023-08-10 02:19:39 +08:00
|
|
|
PyObject *self_or_null;
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *callable;
|
2023-02-09 03:40:10 +08:00
|
|
|
PyObject *res;
|
2023-08-05 00:35:56 +08:00
|
|
|
args = stack_pointer - oparg;
|
2023-08-10 02:19:39 +08:00
|
|
|
self_or_null = stack_pointer[-1 - oparg];
|
|
|
|
callable = stack_pointer[-2 - oparg];
|
2022-11-03 12:31:26 +08:00
|
|
|
/* isinstance(o, o2) */
|
2023-02-09 03:40:10 +08:00
|
|
|
int total_args = oparg;
|
2023-08-10 02:19:39 +08:00
|
|
|
if (self_or_null != NULL) {
|
2023-02-09 03:40:10 +08:00
|
|
|
args--;
|
|
|
|
total_args++;
|
|
|
|
}
|
2022-11-03 12:31:26 +08:00
|
|
|
DEOPT_IF(total_args != 2, CALL);
|
2023-08-01 05:18:38 +08:00
|
|
|
PyInterpreterState *interp = tstate->interp;
|
2022-11-03 12:31:26 +08:00
|
|
|
DEOPT_IF(callable != interp->callable_cache.isinstance, CALL);
|
|
|
|
STAT_INC(CALL, hit);
|
2023-02-09 03:40:10 +08:00
|
|
|
PyObject *cls = args[1];
|
|
|
|
PyObject *inst = args[0];
|
2022-11-03 12:31:26 +08:00
|
|
|
int retval = PyObject_IsInstance(inst, cls);
|
|
|
|
if (retval < 0) {
|
|
|
|
goto error;
|
|
|
|
}
|
2023-02-09 03:40:10 +08:00
|
|
|
res = PyBool_FromLong(retval);
|
2022-11-03 12:31:26 +08:00
|
|
|
assert((res != NULL) ^ (_PyErr_Occurred(tstate) != NULL));
|
|
|
|
|
|
|
|
Py_DECREF(inst);
|
|
|
|
Py_DECREF(cls);
|
|
|
|
Py_DECREF(callable);
|
2023-02-09 03:40:10 +08:00
|
|
|
if (res == NULL) { STACK_SHRINK(oparg); goto pop_2_error; }
|
|
|
|
STACK_SHRINK(oparg);
|
|
|
|
STACK_SHRINK(1);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = res;
|
2023-04-06 05:15:49 +08:00
|
|
|
next_instr += 3;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2023-09-14 01:25:45 +08:00
|
|
|
TARGET(CALL_LIST_APPEND) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject **args;
|
|
|
|
PyObject *self;
|
2023-08-10 02:19:39 +08:00
|
|
|
PyObject *callable;
|
2023-08-05 00:35:56 +08:00
|
|
|
args = stack_pointer - oparg;
|
|
|
|
self = stack_pointer[-1 - oparg];
|
2023-08-10 02:19:39 +08:00
|
|
|
callable = stack_pointer[-2 - oparg];
|
2022-11-03 12:31:26 +08:00
|
|
|
assert(oparg == 1);
|
2023-08-01 05:18:38 +08:00
|
|
|
PyInterpreterState *interp = tstate->interp;
|
2023-08-10 02:19:39 +08:00
|
|
|
DEOPT_IF(callable != interp->callable_cache.list_append, CALL);
|
2023-08-29 01:17:00 +08:00
|
|
|
assert(self != NULL);
|
2023-02-09 03:40:10 +08:00
|
|
|
DEOPT_IF(!PyList_Check(self), CALL);
|
2022-11-03 12:31:26 +08:00
|
|
|
STAT_INC(CALL, hit);
|
2023-02-09 03:40:10 +08:00
|
|
|
if (_PyList_AppendTakeRef((PyListObject *)self, args[0]) < 0) {
|
|
|
|
goto pop_1_error; // Since arg is DECREF'ed already
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
2023-02-09 03:40:10 +08:00
|
|
|
Py_DECREF(self);
|
2023-08-10 02:19:39 +08:00
|
|
|
Py_DECREF(callable);
|
2023-02-09 03:40:10 +08:00
|
|
|
STACK_SHRINK(3);
|
2022-11-11 00:46:56 +08:00
|
|
|
// CALL + POP_TOP
|
2023-06-15 07:14:22 +08:00
|
|
|
SKIP_OVER(INLINE_CACHE_ENTRIES_CALL + 1);
|
2023-02-20 22:56:48 +08:00
|
|
|
assert(next_instr[-1].op.code == POP_TOP);
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2023-09-14 01:25:45 +08:00
|
|
|
TARGET(CALL_METHOD_DESCRIPTOR_O) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject **args;
|
2023-08-10 02:19:39 +08:00
|
|
|
PyObject *self_or_null;
|
|
|
|
PyObject *callable;
|
2023-02-09 03:40:10 +08:00
|
|
|
PyObject *res;
|
2023-08-05 00:35:56 +08:00
|
|
|
args = stack_pointer - oparg;
|
2023-08-10 02:19:39 +08:00
|
|
|
self_or_null = stack_pointer[-1 - oparg];
|
|
|
|
callable = stack_pointer[-2 - oparg];
|
2023-02-09 03:40:10 +08:00
|
|
|
int total_args = oparg;
|
2023-08-10 02:19:39 +08:00
|
|
|
if (self_or_null != NULL) {
|
2023-02-09 03:40:10 +08:00
|
|
|
args--;
|
|
|
|
total_args++;
|
|
|
|
}
|
2023-08-10 02:19:39 +08:00
|
|
|
PyMethodDescrObject *method = (PyMethodDescrObject *)callable;
|
2022-11-03 12:31:26 +08:00
|
|
|
DEOPT_IF(total_args != 2, CALL);
|
2023-08-10 02:19:39 +08:00
|
|
|
DEOPT_IF(!Py_IS_TYPE(method, &PyMethodDescr_Type), CALL);
|
|
|
|
PyMethodDef *meth = method->d_method;
|
2022-11-03 12:31:26 +08:00
|
|
|
DEOPT_IF(meth->ml_flags != METH_O, CALL);
|
2023-02-09 03:40:10 +08:00
|
|
|
PyObject *arg = args[1];
|
|
|
|
PyObject *self = args[0];
|
2023-08-10 02:19:39 +08:00
|
|
|
DEOPT_IF(!Py_IS_TYPE(self, method->d_common.d_type), CALL);
|
2022-11-03 12:31:26 +08:00
|
|
|
STAT_INC(CALL, hit);
|
|
|
|
PyCFunction cfunc = meth->ml_meth;
|
|
|
|
// This is slower but CPython promises to check all non-vectorcall
|
|
|
|
// function calls.
|
|
|
|
if (_Py_EnterRecursiveCallTstate(tstate, " while calling a Python object")) {
|
|
|
|
goto error;
|
|
|
|
}
|
2023-02-09 03:40:10 +08:00
|
|
|
res = _PyCFunction_TrampolineCall(cfunc, self, arg);
|
2022-11-03 12:31:26 +08:00
|
|
|
_Py_LeaveRecursiveCallTstate(tstate);
|
|
|
|
assert((res != NULL) ^ (_PyErr_Occurred(tstate) != NULL));
|
|
|
|
Py_DECREF(self);
|
|
|
|
Py_DECREF(arg);
|
|
|
|
Py_DECREF(callable);
|
2023-02-09 03:40:10 +08:00
|
|
|
if (res == NULL) { STACK_SHRINK(oparg); goto pop_2_error; }
|
|
|
|
STACK_SHRINK(oparg);
|
|
|
|
STACK_SHRINK(1);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = res;
|
2023-04-06 05:15:49 +08:00
|
|
|
next_instr += 3;
|
2022-11-03 12:31:26 +08:00
|
|
|
CHECK_EVAL_BREAKER();
|
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(CALL_METHOD_DESCRIPTOR_FAST_WITH_KEYWORDS) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject **args;
|
2023-08-10 02:19:39 +08:00
|
|
|
PyObject *self_or_null;
|
|
|
|
PyObject *callable;
|
2023-02-09 03:40:10 +08:00
|
|
|
PyObject *res;
|
2023-08-05 00:35:56 +08:00
|
|
|
args = stack_pointer - oparg;
|
2023-08-10 02:19:39 +08:00
|
|
|
self_or_null = stack_pointer[-1 - oparg];
|
|
|
|
callable = stack_pointer[-2 - oparg];
|
2023-02-09 03:40:10 +08:00
|
|
|
int total_args = oparg;
|
2023-08-10 02:19:39 +08:00
|
|
|
if (self_or_null != NULL) {
|
2023-02-09 03:40:10 +08:00
|
|
|
args--;
|
|
|
|
total_args++;
|
|
|
|
}
|
2023-08-10 02:19:39 +08:00
|
|
|
PyMethodDescrObject *method = (PyMethodDescrObject *)callable;
|
|
|
|
DEOPT_IF(!Py_IS_TYPE(method, &PyMethodDescr_Type), CALL);
|
|
|
|
PyMethodDef *meth = method->d_method;
|
2022-11-03 12:31:26 +08:00
|
|
|
DEOPT_IF(meth->ml_flags != (METH_FASTCALL|METH_KEYWORDS), CALL);
|
2023-08-10 02:19:39 +08:00
|
|
|
PyTypeObject *d_type = method->d_common.d_type;
|
2023-02-09 03:40:10 +08:00
|
|
|
PyObject *self = args[0];
|
2022-11-03 12:31:26 +08:00
|
|
|
DEOPT_IF(!Py_IS_TYPE(self, d_type), CALL);
|
|
|
|
STAT_INC(CALL, hit);
|
2023-02-09 03:40:10 +08:00
|
|
|
int nargs = total_args - 1;
|
2022-11-03 12:31:26 +08:00
|
|
|
_PyCFunctionFastWithKeywords cfunc =
|
|
|
|
(_PyCFunctionFastWithKeywords)(void(*)(void))meth->ml_meth;
|
2023-09-14 01:25:45 +08:00
|
|
|
res = cfunc(self, args + 1, nargs, NULL);
|
2022-11-03 12:31:26 +08:00
|
|
|
assert((res != NULL) ^ (_PyErr_Occurred(tstate) != NULL));
|
|
|
|
|
|
|
|
/* Free the arguments. */
|
2023-02-09 03:40:10 +08:00
|
|
|
for (int i = 0; i < total_args; i++) {
|
|
|
|
Py_DECREF(args[i]);
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
|
|
|
Py_DECREF(callable);
|
2023-02-09 03:40:10 +08:00
|
|
|
if (res == NULL) { STACK_SHRINK(oparg); goto pop_2_error; }
|
|
|
|
STACK_SHRINK(oparg);
|
|
|
|
STACK_SHRINK(1);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = res;
|
2023-04-06 05:15:49 +08:00
|
|
|
next_instr += 3;
|
2022-11-03 12:31:26 +08:00
|
|
|
CHECK_EVAL_BREAKER();
|
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2023-09-14 01:25:45 +08:00
|
|
|
TARGET(CALL_METHOD_DESCRIPTOR_NOARGS) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject **args;
|
2023-08-10 02:19:39 +08:00
|
|
|
PyObject *self_or_null;
|
|
|
|
PyObject *callable;
|
2023-02-09 03:40:10 +08:00
|
|
|
PyObject *res;
|
2023-08-05 00:35:56 +08:00
|
|
|
args = stack_pointer - oparg;
|
2023-08-10 02:19:39 +08:00
|
|
|
self_or_null = stack_pointer[-1 - oparg];
|
|
|
|
callable = stack_pointer[-2 - oparg];
|
2022-11-03 12:31:26 +08:00
|
|
|
assert(oparg == 0 || oparg == 1);
|
2023-02-09 03:40:10 +08:00
|
|
|
int total_args = oparg;
|
2023-08-10 02:19:39 +08:00
|
|
|
if (self_or_null != NULL) {
|
2023-02-09 03:40:10 +08:00
|
|
|
args--;
|
|
|
|
total_args++;
|
|
|
|
}
|
2022-11-03 12:31:26 +08:00
|
|
|
DEOPT_IF(total_args != 1, CALL);
|
2023-08-10 02:19:39 +08:00
|
|
|
PyMethodDescrObject *method = (PyMethodDescrObject *)callable;
|
|
|
|
DEOPT_IF(!Py_IS_TYPE(method, &PyMethodDescr_Type), CALL);
|
|
|
|
PyMethodDef *meth = method->d_method;
|
2023-02-09 03:40:10 +08:00
|
|
|
PyObject *self = args[0];
|
2023-08-10 02:19:39 +08:00
|
|
|
DEOPT_IF(!Py_IS_TYPE(self, method->d_common.d_type), CALL);
|
2022-11-03 12:31:26 +08:00
|
|
|
DEOPT_IF(meth->ml_flags != METH_NOARGS, CALL);
|
|
|
|
STAT_INC(CALL, hit);
|
|
|
|
PyCFunction cfunc = meth->ml_meth;
|
|
|
|
// This is slower but CPython promises to check all non-vectorcall
|
|
|
|
// function calls.
|
|
|
|
if (_Py_EnterRecursiveCallTstate(tstate, " while calling a Python object")) {
|
|
|
|
goto error;
|
|
|
|
}
|
2023-02-09 03:40:10 +08:00
|
|
|
res = _PyCFunction_TrampolineCall(cfunc, self, NULL);
|
2022-11-03 12:31:26 +08:00
|
|
|
_Py_LeaveRecursiveCallTstate(tstate);
|
|
|
|
assert((res != NULL) ^ (_PyErr_Occurred(tstate) != NULL));
|
|
|
|
Py_DECREF(self);
|
|
|
|
Py_DECREF(callable);
|
2023-02-09 03:40:10 +08:00
|
|
|
if (res == NULL) { STACK_SHRINK(oparg); goto pop_2_error; }
|
|
|
|
STACK_SHRINK(oparg);
|
|
|
|
STACK_SHRINK(1);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = res;
|
2023-04-06 05:15:49 +08:00
|
|
|
next_instr += 3;
|
2022-11-03 12:31:26 +08:00
|
|
|
CHECK_EVAL_BREAKER();
|
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2023-09-14 01:25:45 +08:00
|
|
|
TARGET(CALL_METHOD_DESCRIPTOR_FAST) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject **args;
|
2023-08-10 02:19:39 +08:00
|
|
|
PyObject *self_or_null;
|
|
|
|
PyObject *callable;
|
2023-02-09 03:40:10 +08:00
|
|
|
PyObject *res;
|
2023-08-05 00:35:56 +08:00
|
|
|
args = stack_pointer - oparg;
|
2023-08-10 02:19:39 +08:00
|
|
|
self_or_null = stack_pointer[-1 - oparg];
|
|
|
|
callable = stack_pointer[-2 - oparg];
|
2023-02-09 03:40:10 +08:00
|
|
|
int total_args = oparg;
|
2023-08-10 02:19:39 +08:00
|
|
|
if (self_or_null != NULL) {
|
2023-02-09 03:40:10 +08:00
|
|
|
args--;
|
|
|
|
total_args++;
|
|
|
|
}
|
2023-08-10 02:19:39 +08:00
|
|
|
PyMethodDescrObject *method = (PyMethodDescrObject *)callable;
|
2022-11-03 12:31:26 +08:00
|
|
|
/* Builtin METH_FASTCALL methods, without keywords */
|
2023-08-10 02:19:39 +08:00
|
|
|
DEOPT_IF(!Py_IS_TYPE(method, &PyMethodDescr_Type), CALL);
|
|
|
|
PyMethodDef *meth = method->d_method;
|
2022-11-03 12:31:26 +08:00
|
|
|
DEOPT_IF(meth->ml_flags != METH_FASTCALL, CALL);
|
2023-02-09 03:40:10 +08:00
|
|
|
PyObject *self = args[0];
|
2023-08-10 02:19:39 +08:00
|
|
|
DEOPT_IF(!Py_IS_TYPE(self, method->d_common.d_type), CALL);
|
2022-11-03 12:31:26 +08:00
|
|
|
STAT_INC(CALL, hit);
|
|
|
|
_PyCFunctionFast cfunc =
|
|
|
|
(_PyCFunctionFast)(void(*)(void))meth->ml_meth;
|
2023-02-09 03:40:10 +08:00
|
|
|
int nargs = total_args - 1;
|
|
|
|
res = cfunc(self, args + 1, nargs);
|
2022-11-03 12:31:26 +08:00
|
|
|
assert((res != NULL) ^ (_PyErr_Occurred(tstate) != NULL));
|
|
|
|
/* Clear the stack of the arguments. */
|
2023-02-09 03:40:10 +08:00
|
|
|
for (int i = 0; i < total_args; i++) {
|
|
|
|
Py_DECREF(args[i]);
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
|
|
|
Py_DECREF(callable);
|
2023-02-09 03:40:10 +08:00
|
|
|
if (res == NULL) { STACK_SHRINK(oparg); goto pop_2_error; }
|
|
|
|
STACK_SHRINK(oparg);
|
|
|
|
STACK_SHRINK(1);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = res;
|
2023-04-06 05:15:49 +08:00
|
|
|
next_instr += 3;
|
2022-11-03 12:31:26 +08:00
|
|
|
CHECK_EVAL_BREAKER();
|
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2023-09-14 01:25:45 +08:00
|
|
|
TARGET(INSTRUMENTED_CALL_KW) {
|
|
|
|
int is_meth = PEEK(oparg + 2) != NULL;
|
|
|
|
int total_args = oparg + is_meth;
|
|
|
|
PyObject *function = PEEK(oparg + 3);
|
|
|
|
PyObject *arg = total_args == 0 ? &_PyInstrumentation_MISSING
|
|
|
|
: PEEK(total_args + 1);
|
|
|
|
int err = _Py_call_instrumentation_2args(
|
|
|
|
tstate, PY_MONITORING_EVENT_CALL,
|
|
|
|
frame, next_instr - 1, function, arg);
|
|
|
|
if (err) goto error;
|
|
|
|
GO_TO_INSTRUCTION(CALL_KW);
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(CALL_KW) {
|
|
|
|
PREDICTED(CALL_KW);
|
|
|
|
PyObject *kwnames;
|
|
|
|
PyObject **args;
|
|
|
|
PyObject *self_or_null;
|
|
|
|
PyObject *callable;
|
|
|
|
PyObject *res;
|
|
|
|
kwnames = stack_pointer[-1];
|
|
|
|
args = stack_pointer - 1 - oparg;
|
|
|
|
self_or_null = stack_pointer[-2 - oparg];
|
|
|
|
callable = stack_pointer[-3 - oparg];
|
|
|
|
// oparg counts all of the args, but *not* self:
|
|
|
|
int total_args = oparg;
|
|
|
|
if (self_or_null != NULL) {
|
|
|
|
args--;
|
|
|
|
total_args++;
|
|
|
|
}
|
|
|
|
if (self_or_null == NULL && Py_TYPE(callable) == &PyMethod_Type) {
|
|
|
|
args--;
|
|
|
|
total_args++;
|
|
|
|
PyObject *self = ((PyMethodObject *)callable)->im_self;
|
|
|
|
args[0] = Py_NewRef(self);
|
|
|
|
PyObject *method = ((PyMethodObject *)callable)->im_func;
|
|
|
|
args[-1] = Py_NewRef(method);
|
|
|
|
Py_DECREF(callable);
|
|
|
|
callable = method;
|
|
|
|
}
|
|
|
|
int positional_args = total_args - (int)PyTuple_GET_SIZE(kwnames);
|
|
|
|
// Check if the call can be inlined or not
|
|
|
|
if (Py_TYPE(callable) == &PyFunction_Type &&
|
|
|
|
tstate->interp->eval_frame == NULL &&
|
|
|
|
((PyFunctionObject *)callable)->vectorcall == _PyFunction_Vectorcall)
|
|
|
|
{
|
|
|
|
int code_flags = ((PyCodeObject*)PyFunction_GET_CODE(callable))->co_flags;
|
|
|
|
PyObject *locals = code_flags & CO_OPTIMIZED ? NULL : Py_NewRef(PyFunction_GET_GLOBALS(callable));
|
|
|
|
_PyInterpreterFrame *new_frame = _PyEvalFramePushAndInit(
|
|
|
|
tstate, (PyFunctionObject *)callable, locals,
|
|
|
|
args, positional_args, kwnames
|
|
|
|
);
|
|
|
|
Py_DECREF(kwnames);
|
|
|
|
// Manipulate stack directly since we leave using DISPATCH_INLINED().
|
|
|
|
STACK_SHRINK(oparg + 3);
|
|
|
|
// The frame has stolen all the arguments from the stack,
|
|
|
|
// so there is no need to clean them up.
|
|
|
|
if (new_frame == NULL) {
|
|
|
|
goto error;
|
|
|
|
}
|
|
|
|
frame->return_offset = 0;
|
|
|
|
DISPATCH_INLINED(new_frame);
|
|
|
|
}
|
|
|
|
/* Callable is not a normal Python function */
|
|
|
|
res = PyObject_Vectorcall(
|
|
|
|
callable, args,
|
|
|
|
positional_args | PY_VECTORCALL_ARGUMENTS_OFFSET,
|
|
|
|
kwnames);
|
|
|
|
if (opcode == INSTRUMENTED_CALL_KW) {
|
|
|
|
PyObject *arg = total_args == 0 ?
|
|
|
|
&_PyInstrumentation_MISSING : args[0];
|
|
|
|
if (res == NULL) {
|
|
|
|
_Py_call_instrumentation_exc2(
|
|
|
|
tstate, PY_MONITORING_EVENT_C_RAISE,
|
|
|
|
frame, next_instr-1, callable, arg);
|
|
|
|
}
|
|
|
|
else {
|
|
|
|
int err = _Py_call_instrumentation_2args(
|
|
|
|
tstate, PY_MONITORING_EVENT_C_RETURN,
|
|
|
|
frame, next_instr-1, callable, arg);
|
|
|
|
if (err < 0) {
|
|
|
|
Py_CLEAR(res);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
Py_DECREF(kwnames);
|
|
|
|
assert((res != NULL) ^ (_PyErr_Occurred(tstate) != NULL));
|
|
|
|
Py_DECREF(callable);
|
|
|
|
for (int i = 0; i < total_args; i++) {
|
|
|
|
Py_DECREF(args[i]);
|
|
|
|
}
|
|
|
|
if (res == NULL) { STACK_SHRINK(oparg); goto pop_3_error; }
|
|
|
|
STACK_SHRINK(oparg);
|
|
|
|
STACK_SHRINK(2);
|
|
|
|
stack_pointer[-1] = res;
|
|
|
|
CHECK_EVAL_BREAKER();
|
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2023-04-12 19:04:55 +08:00
|
|
|
TARGET(INSTRUMENTED_CALL_FUNCTION_EX) {
|
|
|
|
GO_TO_INSTRUCTION(CALL_FUNCTION_EX);
|
|
|
|
}
|
|
|
|
|
2022-11-03 12:31:26 +08:00
|
|
|
TARGET(CALL_FUNCTION_EX) {
|
|
|
|
PREDICTED(CALL_FUNCTION_EX);
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *kwargs = NULL;
|
|
|
|
PyObject *callargs;
|
|
|
|
PyObject *func;
|
2023-02-08 12:03:22 +08:00
|
|
|
PyObject *result;
|
2023-08-05 00:35:56 +08:00
|
|
|
if (oparg & 1) { kwargs = stack_pointer[-(oparg & 1 ? 1 : 0)]; }
|
|
|
|
callargs = stack_pointer[-1 - (oparg & 1 ? 1 : 0)];
|
2023-08-10 02:19:39 +08:00
|
|
|
func = stack_pointer[-3 - (oparg & 1 ? 1 : 0)];
|
2023-04-12 19:04:55 +08:00
|
|
|
// DICT_MERGE is called before this opcode if there are kwargs.
|
|
|
|
// It converts all dict subtypes in kwargs into regular dicts.
|
|
|
|
assert(kwargs == NULL || PyDict_CheckExact(kwargs));
|
2022-11-03 12:31:26 +08:00
|
|
|
if (!PyTuple_CheckExact(callargs)) {
|
|
|
|
if (check_args_iterable(tstate, func, callargs) < 0) {
|
|
|
|
goto error;
|
|
|
|
}
|
2023-02-08 12:03:22 +08:00
|
|
|
PyObject *tuple = PySequence_Tuple(callargs);
|
|
|
|
if (tuple == NULL) {
|
2022-11-03 12:31:26 +08:00
|
|
|
goto error;
|
|
|
|
}
|
2023-02-08 12:03:22 +08:00
|
|
|
Py_SETREF(callargs, tuple);
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
|
|
|
assert(PyTuple_CheckExact(callargs));
|
2023-04-12 19:04:55 +08:00
|
|
|
EVAL_CALL_STAT_INC_IF_FUNCTION(EVAL_CALL_FUNCTION_EX, func);
|
|
|
|
if (opcode == INSTRUMENTED_CALL_FUNCTION_EX &&
|
|
|
|
!PyFunction_Check(func) && !PyMethod_Check(func)
|
|
|
|
) {
|
|
|
|
PyObject *arg = PyTuple_GET_SIZE(callargs) > 0 ?
|
|
|
|
PyTuple_GET_ITEM(callargs, 0) : Py_None;
|
|
|
|
int err = _Py_call_instrumentation_2args(
|
|
|
|
tstate, PY_MONITORING_EVENT_CALL,
|
|
|
|
frame, next_instr-1, func, arg);
|
|
|
|
if (err) goto error;
|
|
|
|
result = PyObject_Call(func, callargs, kwargs);
|
|
|
|
if (result == NULL) {
|
|
|
|
_Py_call_instrumentation_exc2(
|
|
|
|
tstate, PY_MONITORING_EVENT_C_RAISE,
|
|
|
|
frame, next_instr-1, func, arg);
|
|
|
|
}
|
|
|
|
else {
|
|
|
|
int err = _Py_call_instrumentation_2args(
|
|
|
|
tstate, PY_MONITORING_EVENT_C_RETURN,
|
|
|
|
frame, next_instr-1, func, arg);
|
|
|
|
if (err < 0) {
|
|
|
|
Py_CLEAR(result);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
else {
|
2023-04-30 21:08:26 +08:00
|
|
|
if (Py_TYPE(func) == &PyFunction_Type &&
|
|
|
|
tstate->interp->eval_frame == NULL &&
|
|
|
|
((PyFunctionObject *)func)->vectorcall == _PyFunction_Vectorcall) {
|
|
|
|
assert(PyTuple_CheckExact(callargs));
|
|
|
|
Py_ssize_t nargs = PyTuple_GET_SIZE(callargs);
|
|
|
|
int code_flags = ((PyCodeObject *)PyFunction_GET_CODE(func))->co_flags;
|
|
|
|
PyObject *locals = code_flags & CO_OPTIMIZED ? NULL : Py_NewRef(PyFunction_GET_GLOBALS(func));
|
|
|
|
|
|
|
|
_PyInterpreterFrame *new_frame = _PyEvalFramePushAndInit_Ex(tstate,
|
|
|
|
(PyFunctionObject *)func, locals,
|
|
|
|
nargs, callargs, kwargs);
|
|
|
|
// Need to manually shrink the stack since we exit with DISPATCH_INLINED.
|
|
|
|
STACK_SHRINK(oparg + 3);
|
|
|
|
if (new_frame == NULL) {
|
|
|
|
goto error;
|
|
|
|
}
|
|
|
|
frame->return_offset = 0;
|
|
|
|
DISPATCH_INLINED(new_frame);
|
|
|
|
}
|
2023-04-12 19:04:55 +08:00
|
|
|
result = PyObject_Call(func, callargs, kwargs);
|
|
|
|
}
|
2022-11-03 12:31:26 +08:00
|
|
|
Py_DECREF(func);
|
|
|
|
Py_DECREF(callargs);
|
|
|
|
Py_XDECREF(kwargs);
|
2023-08-10 02:19:39 +08:00
|
|
|
assert(PEEK(2 + (oparg & 1)) == NULL);
|
2023-02-08 12:03:22 +08:00
|
|
|
if (result == NULL) { STACK_SHRINK(((oparg & 1) ? 1 : 0)); goto pop_3_error; }
|
|
|
|
STACK_SHRINK(((oparg & 1) ? 1 : 0));
|
|
|
|
STACK_SHRINK(2);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = result;
|
2022-11-03 12:31:26 +08:00
|
|
|
CHECK_EVAL_BREAKER();
|
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(MAKE_FUNCTION) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *codeobj;
|
2023-02-03 22:40:45 +08:00
|
|
|
PyObject *func;
|
2023-08-05 00:35:56 +08:00
|
|
|
codeobj = stack_pointer[-1];
|
2023-02-03 22:40:45 +08:00
|
|
|
|
|
|
|
PyFunctionObject *func_obj = (PyFunctionObject *)
|
2022-11-03 12:31:26 +08:00
|
|
|
PyFunction_New(codeobj, GLOBALS());
|
|
|
|
|
|
|
|
Py_DECREF(codeobj);
|
2023-02-03 22:40:45 +08:00
|
|
|
if (func_obj == NULL) {
|
2022-11-03 12:31:26 +08:00
|
|
|
goto error;
|
|
|
|
}
|
|
|
|
|
2023-08-18 02:29:58 +08:00
|
|
|
_PyFunction_SetVersion(
|
|
|
|
func_obj, ((PyCodeObject *)codeobj)->co_version);
|
2023-02-03 22:40:45 +08:00
|
|
|
func = (PyObject *)func_obj;
|
2023-06-13 16:51:05 +08:00
|
|
|
stack_pointer[-1] = func;
|
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(SET_FUNCTION_ATTRIBUTE) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *func;
|
|
|
|
PyObject *attr;
|
|
|
|
func = stack_pointer[-1];
|
|
|
|
attr = stack_pointer[-2];
|
2023-06-13 16:51:05 +08:00
|
|
|
assert(PyFunction_Check(func));
|
|
|
|
PyFunctionObject *func_obj = (PyFunctionObject *)func;
|
|
|
|
switch(oparg) {
|
|
|
|
case MAKE_FUNCTION_CLOSURE:
|
|
|
|
assert(func_obj->func_closure == NULL);
|
|
|
|
func_obj->func_closure = attr;
|
|
|
|
break;
|
|
|
|
case MAKE_FUNCTION_ANNOTATIONS:
|
|
|
|
assert(func_obj->func_annotations == NULL);
|
|
|
|
func_obj->func_annotations = attr;
|
|
|
|
break;
|
|
|
|
case MAKE_FUNCTION_KWDEFAULTS:
|
|
|
|
assert(PyDict_CheckExact(attr));
|
|
|
|
assert(func_obj->func_kwdefaults == NULL);
|
|
|
|
func_obj->func_kwdefaults = attr;
|
|
|
|
break;
|
|
|
|
case MAKE_FUNCTION_DEFAULTS:
|
|
|
|
assert(PyTuple_CheckExact(attr));
|
|
|
|
assert(func_obj->func_defaults == NULL);
|
|
|
|
func_obj->func_defaults = attr;
|
|
|
|
break;
|
|
|
|
default:
|
|
|
|
Py_UNREACHABLE();
|
|
|
|
}
|
|
|
|
STACK_SHRINK(1);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = func;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(RETURN_GENERATOR) {
|
|
|
|
assert(PyFunction_Check(frame->f_funcobj));
|
|
|
|
PyFunctionObject *func = (PyFunctionObject *)frame->f_funcobj;
|
|
|
|
PyGenObject *gen = (PyGenObject *)_Py_MakeCoro(func);
|
|
|
|
if (gen == NULL) {
|
|
|
|
goto error;
|
|
|
|
}
|
|
|
|
assert(EMPTY());
|
|
|
|
_PyFrame_SetStackPointer(frame, stack_pointer);
|
|
|
|
_PyInterpreterFrame *gen_frame = (_PyInterpreterFrame *)gen->gi_iframe;
|
|
|
|
_PyFrame_Copy(frame, gen_frame);
|
|
|
|
assert(frame->frame_obj == NULL);
|
|
|
|
gen->gi_frame_state = FRAME_CREATED;
|
|
|
|
gen_frame->owner = FRAME_OWNED_BY_GENERATOR;
|
|
|
|
_Py_LeaveRecursiveCallPy(tstate);
|
2022-11-10 20:34:57 +08:00
|
|
|
assert(frame != &entry_frame);
|
|
|
|
_PyInterpreterFrame *prev = frame->previous;
|
|
|
|
_PyThreadState_PopFrame(tstate, frame);
|
2023-08-17 18:16:03 +08:00
|
|
|
frame = tstate->current_frame = prev;
|
2022-11-10 20:34:57 +08:00
|
|
|
_PyFrame_StackPush(frame, (PyObject *)gen);
|
|
|
|
goto resume_frame;
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(BUILD_SLICE) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *step = NULL;
|
|
|
|
PyObject *stop;
|
|
|
|
PyObject *start;
|
2023-02-03 22:40:45 +08:00
|
|
|
PyObject *slice;
|
2023-08-05 00:35:56 +08:00
|
|
|
if (oparg == 3) { step = stack_pointer[-(oparg == 3 ? 1 : 0)]; }
|
|
|
|
stop = stack_pointer[-1 - (oparg == 3 ? 1 : 0)];
|
|
|
|
start = stack_pointer[-2 - (oparg == 3 ? 1 : 0)];
|
2022-11-03 12:31:26 +08:00
|
|
|
slice = PySlice_New(start, stop, step);
|
|
|
|
Py_DECREF(start);
|
|
|
|
Py_DECREF(stop);
|
|
|
|
Py_XDECREF(step);
|
2023-02-03 22:40:45 +08:00
|
|
|
if (slice == NULL) { STACK_SHRINK(((oparg == 3) ? 1 : 0)); goto pop_2_error; }
|
|
|
|
STACK_SHRINK(((oparg == 3) ? 1 : 0));
|
|
|
|
STACK_SHRINK(1);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = slice;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2023-06-14 23:15:08 +08:00
|
|
|
TARGET(CONVERT_VALUE) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *value;
|
2022-11-03 12:31:26 +08:00
|
|
|
PyObject *result;
|
2023-08-05 00:35:56 +08:00
|
|
|
value = stack_pointer[-1];
|
2023-06-14 23:15:08 +08:00
|
|
|
convertion_func_ptr conv_fn;
|
|
|
|
assert(oparg >= FVC_STR && oparg <= FVC_ASCII);
|
|
|
|
conv_fn = CONVERSION_FUNCTIONS[oparg];
|
|
|
|
result = conv_fn(value);
|
|
|
|
Py_DECREF(value);
|
|
|
|
if (result == NULL) goto pop_1_error;
|
|
|
|
stack_pointer[-1] = result;
|
|
|
|
DISPATCH();
|
|
|
|
}
|
2022-11-03 12:31:26 +08:00
|
|
|
|
2023-06-14 23:15:08 +08:00
|
|
|
TARGET(FORMAT_SIMPLE) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *value;
|
2023-06-14 23:15:08 +08:00
|
|
|
PyObject *res;
|
2023-08-05 00:35:56 +08:00
|
|
|
value = stack_pointer[-1];
|
2023-06-14 23:15:08 +08:00
|
|
|
/* If value is a unicode object, then we know the result
|
|
|
|
* of format(value) is value itself. */
|
|
|
|
if (!PyUnicode_CheckExact(value)) {
|
|
|
|
res = PyObject_Format(value, NULL);
|
2022-11-03 12:31:26 +08:00
|
|
|
Py_DECREF(value);
|
2023-06-14 23:15:08 +08:00
|
|
|
if (res == NULL) goto pop_1_error;
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
2023-06-14 23:15:08 +08:00
|
|
|
else {
|
|
|
|
res = value;
|
|
|
|
}
|
|
|
|
stack_pointer[-1] = res;
|
|
|
|
DISPATCH();
|
|
|
|
}
|
2022-11-03 12:31:26 +08:00
|
|
|
|
2023-06-14 23:15:08 +08:00
|
|
|
TARGET(FORMAT_WITH_SPEC) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *fmt_spec;
|
|
|
|
PyObject *value;
|
2023-06-14 23:15:08 +08:00
|
|
|
PyObject *res;
|
2023-08-05 00:35:56 +08:00
|
|
|
fmt_spec = stack_pointer[-1];
|
|
|
|
value = stack_pointer[-2];
|
2023-06-14 23:15:08 +08:00
|
|
|
res = PyObject_Format(value, fmt_spec);
|
2023-03-21 15:47:15 +08:00
|
|
|
Py_DECREF(value);
|
2023-06-14 23:15:08 +08:00
|
|
|
Py_DECREF(fmt_spec);
|
|
|
|
if (res == NULL) goto pop_2_error;
|
|
|
|
STACK_SHRINK(1);
|
|
|
|
stack_pointer[-1] = res;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(COPY) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *bottom;
|
2023-02-07 06:45:18 +08:00
|
|
|
PyObject *top;
|
2023-08-05 00:35:56 +08:00
|
|
|
bottom = stack_pointer[-1 - (oparg-1)];
|
2023-02-07 06:45:18 +08:00
|
|
|
assert(oparg > 0);
|
|
|
|
top = Py_NewRef(bottom);
|
|
|
|
STACK_GROW(1);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = top;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2022-11-10 02:50:09 +08:00
|
|
|
TARGET(BINARY_OP) {
|
|
|
|
PREDICTED(BINARY_OP);
|
2022-11-18 09:06:07 +08:00
|
|
|
static_assert(INLINE_CACHE_ENTRIES_BINARY_OP == 1, "incorrect cache size");
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *rhs;
|
|
|
|
PyObject *lhs;
|
2022-11-18 03:36:57 +08:00
|
|
|
PyObject *res;
|
2023-08-05 00:35:56 +08:00
|
|
|
rhs = stack_pointer[-1];
|
|
|
|
lhs = stack_pointer[-2];
|
2023-01-20 02:14:55 +08:00
|
|
|
#if ENABLE_SPECIALIZATION
|
2022-11-03 12:31:26 +08:00
|
|
|
_PyBinaryOpCache *cache = (_PyBinaryOpCache *)next_instr;
|
2022-11-10 02:50:09 +08:00
|
|
|
if (ADAPTIVE_COUNTER_IS_ZERO(cache->counter)) {
|
2022-11-03 12:31:26 +08:00
|
|
|
next_instr--;
|
2023-07-19 02:42:44 +08:00
|
|
|
_Py_Specialize_BinaryOp(lhs, rhs, next_instr, oparg, LOCALS_ARRAY);
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH_SAME_OPARG();
|
|
|
|
}
|
2022-11-10 02:50:09 +08:00
|
|
|
STAT_INC(BINARY_OP, deferred);
|
|
|
|
DECREMENT_ADAPTIVE_COUNTER(cache->counter);
|
2023-01-20 02:14:55 +08:00
|
|
|
#endif /* ENABLE_SPECIALIZATION */
|
2023-07-21 04:37:19 +08:00
|
|
|
assert(NB_ADD <= oparg);
|
|
|
|
assert(oparg <= NB_INPLACE_XOR);
|
|
|
|
assert(_PyEval_BinaryOps[oparg]);
|
|
|
|
res = _PyEval_BinaryOps[oparg](lhs, rhs);
|
2022-11-18 03:36:57 +08:00
|
|
|
Py_DECREF(lhs);
|
|
|
|
Py_DECREF(rhs);
|
|
|
|
if (res == NULL) goto pop_2_error;
|
|
|
|
STACK_SHRINK(1);
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = res;
|
|
|
|
next_instr += 1;
|
2022-11-18 03:36:57 +08:00
|
|
|
DISPATCH();
|
2022-11-03 12:31:26 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(SWAP) {
|
2023-08-05 00:35:56 +08:00
|
|
|
PyObject *top;
|
|
|
|
PyObject *bottom;
|
|
|
|
top = stack_pointer[-1];
|
|
|
|
bottom = stack_pointer[-2 - (oparg-2)];
|
2023-02-07 06:45:18 +08:00
|
|
|
assert(oparg >= 2);
|
2023-08-05 00:35:56 +08:00
|
|
|
stack_pointer[-2 - (oparg-2)] = top;
|
2023-03-01 00:49:35 +08:00
|
|
|
stack_pointer[-1] = bottom;
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2023-04-12 19:04:55 +08:00
|
|
|
TARGET(INSTRUMENTED_INSTRUCTION) {
|
|
|
|
int next_opcode = _Py_call_instrumentation_instruction(
|
|
|
|
tstate, frame, next_instr-1);
|
|
|
|
if (next_opcode < 0) goto error;
|
|
|
|
next_instr--;
|
|
|
|
if (_PyOpcode_Caches[next_opcode]) {
|
|
|
|
_PyBinaryOpCache *cache = (_PyBinaryOpCache *)(next_instr+1);
|
|
|
|
INCREMENT_ADAPTIVE_COUNTER(cache->counter);
|
|
|
|
}
|
|
|
|
assert(next_opcode > 0 && next_opcode < 256);
|
|
|
|
opcode = next_opcode;
|
|
|
|
DISPATCH_GOTO();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(INSTRUMENTED_JUMP_FORWARD) {
|
|
|
|
INSTRUMENTED_JUMP(next_instr-1, next_instr+oparg, PY_MONITORING_EVENT_JUMP);
|
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(INSTRUMENTED_JUMP_BACKWARD) {
|
|
|
|
CHECK_EVAL_BREAKER();
|
2023-07-04 04:28:27 +08:00
|
|
|
INSTRUMENTED_JUMP(next_instr-1, next_instr+1-oparg, PY_MONITORING_EVENT_JUMP);
|
2023-04-12 19:04:55 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(INSTRUMENTED_POP_JUMP_IF_TRUE) {
|
|
|
|
PyObject *cond = POP();
|
2023-06-30 04:49:54 +08:00
|
|
|
assert(PyBool_Check(cond));
|
|
|
|
_Py_CODEUNIT *here = next_instr - 1;
|
2023-09-12 02:20:24 +08:00
|
|
|
int flag = Py_IsTrue(cond);
|
|
|
|
int offset = flag * oparg;
|
|
|
|
#if ENABLE_SPECIALIZATION
|
|
|
|
next_instr->cache = (next_instr->cache << 1) | flag;
|
|
|
|
#endif
|
2023-04-12 19:04:55 +08:00
|
|
|
INSTRUMENTED_JUMP(here, next_instr + offset, PY_MONITORING_EVENT_BRANCH);
|
2023-09-12 02:20:24 +08:00
|
|
|
next_instr += 1;
|
2023-04-12 19:04:55 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(INSTRUMENTED_POP_JUMP_IF_FALSE) {
|
|
|
|
PyObject *cond = POP();
|
2023-06-30 04:49:54 +08:00
|
|
|
assert(PyBool_Check(cond));
|
|
|
|
_Py_CODEUNIT *here = next_instr - 1;
|
2023-09-12 02:20:24 +08:00
|
|
|
int flag = Py_IsFalse(cond);
|
|
|
|
int offset = flag * oparg;
|
|
|
|
#if ENABLE_SPECIALIZATION
|
|
|
|
next_instr->cache = (next_instr->cache << 1) | flag;
|
|
|
|
#endif
|
2023-04-12 19:04:55 +08:00
|
|
|
INSTRUMENTED_JUMP(here, next_instr + offset, PY_MONITORING_EVENT_BRANCH);
|
2023-09-12 02:20:24 +08:00
|
|
|
next_instr += 1;
|
2023-04-12 19:04:55 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(INSTRUMENTED_POP_JUMP_IF_NONE) {
|
|
|
|
PyObject *value = POP();
|
2023-09-12 02:20:24 +08:00
|
|
|
_Py_CODEUNIT *here = next_instr - 1;
|
|
|
|
int flag = Py_IsNone(value);
|
2023-04-12 19:04:55 +08:00
|
|
|
int offset;
|
2023-09-12 02:20:24 +08:00
|
|
|
if (flag) {
|
2023-04-12 19:04:55 +08:00
|
|
|
offset = oparg;
|
|
|
|
}
|
|
|
|
else {
|
|
|
|
Py_DECREF(value);
|
|
|
|
offset = 0;
|
|
|
|
}
|
2023-09-12 02:20:24 +08:00
|
|
|
#if ENABLE_SPECIALIZATION
|
|
|
|
next_instr->cache = (next_instr->cache << 1) | flag;
|
|
|
|
#endif
|
2023-04-12 19:04:55 +08:00
|
|
|
INSTRUMENTED_JUMP(here, next_instr + offset, PY_MONITORING_EVENT_BRANCH);
|
2023-09-12 02:20:24 +08:00
|
|
|
next_instr += 1;
|
2023-04-12 19:04:55 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(INSTRUMENTED_POP_JUMP_IF_NOT_NONE) {
|
|
|
|
PyObject *value = POP();
|
|
|
|
_Py_CODEUNIT *here = next_instr-1;
|
|
|
|
int offset;
|
2023-09-12 02:20:24 +08:00
|
|
|
int nflag = Py_IsNone(value);
|
|
|
|
if (nflag) {
|
2023-04-12 19:04:55 +08:00
|
|
|
offset = 0;
|
|
|
|
}
|
|
|
|
else {
|
|
|
|
Py_DECREF(value);
|
2023-06-30 04:49:54 +08:00
|
|
|
offset = oparg;
|
2023-04-12 19:04:55 +08:00
|
|
|
}
|
2023-09-12 02:20:24 +08:00
|
|
|
#if ENABLE_SPECIALIZATION
|
|
|
|
next_instr->cache = (next_instr->cache << 1) | !nflag;
|
|
|
|
#endif
|
2023-04-12 19:04:55 +08:00
|
|
|
INSTRUMENTED_JUMP(here, next_instr + offset, PY_MONITORING_EVENT_BRANCH);
|
2023-09-12 02:20:24 +08:00
|
|
|
next_instr += 1;
|
2023-04-12 19:04:55 +08:00
|
|
|
DISPATCH();
|
|
|
|
}
|
|
|
|
|
2022-11-03 12:31:26 +08:00
|
|
|
TARGET(EXTENDED_ARG) {
|
|
|
|
assert(oparg);
|
2023-02-20 22:56:48 +08:00
|
|
|
opcode = next_instr->op.code;
|
|
|
|
oparg = oparg << 8 | next_instr->op.arg;
|
2022-11-10 02:50:09 +08:00
|
|
|
PRE_DISPATCH_GOTO();
|
2022-11-03 12:31:26 +08:00
|
|
|
DISPATCH_GOTO();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(CACHE) {
|
2023-04-12 19:04:55 +08:00
|
|
|
assert(0 && "Executing a cache.");
|
|
|
|
Py_UNREACHABLE();
|
|
|
|
}
|
|
|
|
|
|
|
|
TARGET(RESERVED) {
|
|
|
|
assert(0 && "Executing RESERVED instruction.");
|
2022-11-03 12:31:26 +08:00
|
|
|
Py_UNREACHABLE();
|
|
|
|
}
|