#include "Python.h"

#include "pycore_abstract.h" // _PyIndex_Check()

#include "pycore_call.h" // _PyObject_FastCallDictTstate()

#include "pycore_ceval.h" // _PyEval_SignalAsyncExc()

#include "pycore_code.h"

#include "pycore_function.h"

#include "pycore_long.h" // _PyLong_GetZero()

#include "pycore_object.h" // _PyObject_GC_TRACK()

#include "pycore_moduleobject.h" // PyModuleObject

#include "pycore_opcode.h" // EXTRA_CASES

#include "pycore_pyerrors.h" // _PyErr_Fetch()

#include "pycore_pymem.h" // _PyMem_IsPtrFreed()

#include "pycore_pystate.h" // _PyInterpreterState_GET()

#include "pycore_range.h" // _PyRangeIterObject

#include "pycore_sliceobject.h" // _PyBuildSlice_ConsumeRefs

#include "pycore_sysmodule.h" // _PySys_Audit()

#include "pycore_tuple.h" // _PyTuple_ITEMS()

#include "pycore_emscripten_signal.h" // _Py_CHECK_EMSCRIPTEN_SIGNALS

#include "pycore_dict.h"

#include "dictobject.h"

#include "pycore_frame.h"

#include "opcode.h"

#include "pydtrace.h"

#include "setobject.h"

#include "structmember.h" // struct PyMemberDef, T_OFFSET_EX

void _PyFloat_ExactDealloc(PyObject *);

void _PyUnicode_ExactDealloc(PyObject *);

#define SET_TOP(v) (stack_pointer[-1] = (v))

#define SET_SECOND(v) (stack_pointer[-2] = (v))

#define PEEK(n) (stack_pointer[-(n)])

#define POKE(n, v) (stack_pointer[-(n)] = (v))

#define PUSH(val) (*(stack_pointer++) = (val))

#define POP() (*(--stack_pointer))

#define TOP() PEEK(1)

#define SECOND() PEEK(2)

#define STACK_GROW(n) (stack_pointer += (n))

#define STACK_SHRINK(n) (stack_pointer -= (n))

#define EMPTY() 1

#define STACK_LEVEL() 2

#define GETLOCAL(i) (frame->localsplus[i])

#define SETLOCAL(i, val) \

do { \

PyObject *_tmp = frame->localsplus[i]; \

frame->localsplus[i] = (val); \

Py_XDECREF(_tmp); \

} while (0)

#define DEOPT_IF(cond, instname) ((void)0)

#define ERROR_IF(cond, labelname) ((void)0)

#define JUMPBY(offset) ((void)0)

#define GO_TO_INSTRUCTION(instname) ((void)0)

#define DISPATCH_SAME_OPARG() ((void)0)

#define inst(name, ...) case name:

#define op(name, ...) /* NAME is ignored */

#define macro(name) static int MACRO_##name

#define super(name) static int SUPER_##name

#define family(name, ...) static int family_##name

#define NAME_ERROR_MSG \

"name '%.200s' is not defined"

static PyObject *value, *value1, *value2, *left, *right, *res, *sum, *prod, *sub;

static PyObject *container, *start, *stop, *v, *lhs, *rhs;

static PyObject *list, *tuple, *dict, *owner;

static PyObject *exit_func, *lasti, *val, *retval, *obj, *iter;

static size_t jump;

static _Py_CODEUNIT when_to_jump_mask, invert, counter, index, hint;

static uint32_t type_version;

#define _COMPARE_OP_FLOAT 1003

#define _COMPARE_OP_INT 1004

#define _COMPARE_OP_STR 1005

#define _JUMP_IF 1006

static PyObject *

dummy_func(

)

{

_PyInterpreterFrame entry_frame;

switch (opcode) {

inst(NOP, (--)) {

}

inst(RESUME, (--)) {

assert(tstate->cframe == &cframe);

assert(frame == cframe.current_frame);

if (_Py_atomic_load_relaxed_int32(eval_breaker) && oparg < 2) {

goto handle_eval_breaker;

}

}

inst(LOAD_CLOSURE, (-- value)) {

/* We keep LOAD_CLOSURE so that the bytecode stays more readable. */

value = GETLOCAL(oparg);

ERROR_IF(value == NULL, unbound_local_error);

Py_INCREF(value);

}

inst(LOAD_FAST_CHECK, (-- value)) {

value = GETLOCAL(oparg);

ERROR_IF(value == NULL, unbound_local_error);

Py_INCREF(value);

}

inst(LOAD_FAST, (-- value)) {

value = GETLOCAL(oparg);

assert(value != NULL);

Py_INCREF(value);

}

inst(LOAD_CONST, (-- value)) {

value = GETITEM(consts, oparg);

Py_INCREF(value);

}

inst(STORE_FAST, (value --)) {

SETLOCAL(oparg, value);

}

super(LOAD_FAST__LOAD_FAST) = LOAD_FAST + LOAD_FAST;

super(LOAD_FAST__LOAD_CONST) = LOAD_FAST + LOAD_CONST;

super(STORE_FAST__LOAD_FAST) = STORE_FAST + LOAD_FAST;

super(STORE_FAST__STORE_FAST) = STORE_FAST + STORE_FAST;

super(LOAD_CONST__LOAD_FAST) = LOAD_CONST + LOAD_FAST;

inst(POP_TOP, (value --)) {

Py_DECREF(value);

}

inst(PUSH_NULL, (-- res)) {

res = NULL;

}

macro(END_FOR) = POP_TOP + POP_TOP;

inst(UNARY_POSITIVE, (value -- res)) {

res = PyNumber_Positive(value);

Py_DECREF(value);

ERROR_IF(res == NULL, error);

}

inst(UNARY_NEGATIVE, (value -- res)) {

res = PyNumber_Negative(value);

Py_DECREF(value);

ERROR_IF(res == NULL, error);

}

inst(UNARY_NOT, (value -- res)) {

int err = PyObject_IsTrue(value);

Py_DECREF(value);

ERROR_IF(err < 0, error);

if (err == 0) {

res = Py_True;

}

else {

res = Py_False;

}

Py_INCREF(res);

}

inst(UNARY_INVERT, (value -- res)) {

res = PyNumber_Invert(value);

Py_DECREF(value);

ERROR_IF(res == NULL, error);

}

family(binary_op, INLINE_CACHE_ENTRIES_BINARY_OP) = {

BINARY_OP,

BINARY_OP_ADD_FLOAT,

BINARY_OP_ADD_INT,

BINARY_OP_ADD_UNICODE,

// BINARY_OP_INPLACE_ADD_UNICODE, // This is an odd duck.

BINARY_OP_MULTIPLY_FLOAT,

BINARY_OP_MULTIPLY_INT,

BINARY_OP_SUBTRACT_FLOAT,

BINARY_OP_SUBTRACT_INT,

};

inst(BINARY_OP_MULTIPLY_INT, (unused/1, left, right -- prod)) {

assert(cframe.use_tracing == 0);

DEOPT_IF(!PyLong_CheckExact(left), BINARY_OP);

DEOPT_IF(!PyLong_CheckExact(right), BINARY_OP);

STAT_INC(BINARY_OP, hit);

prod = _PyLong_Multiply((PyLongObject *)left, (PyLongObject *)right);

_Py_DECREF_SPECIALIZED(right, (destructor)PyObject_Free);

_Py_DECREF_SPECIALIZED(left, (destructor)PyObject_Free);

ERROR_IF(prod == NULL, error);

}

inst(BINARY_OP_MULTIPLY_FLOAT, (unused/1, left, right -- prod)) {

assert(cframe.use_tracing == 0);

DEOPT_IF(!PyFloat_CheckExact(left), BINARY_OP);

DEOPT_IF(!PyFloat_CheckExact(right), BINARY_OP);

STAT_INC(BINARY_OP, hit);

double dprod = ((PyFloatObject *)left)->ob_fval *

((PyFloatObject *)right)->ob_fval;

prod = PyFloat_FromDouble(dprod);

_Py_DECREF_SPECIALIZED(right, _PyFloat_ExactDealloc);

_Py_DECREF_SPECIALIZED(left, _PyFloat_ExactDealloc);

ERROR_IF(prod == NULL, error);

}

inst(BINARY_OP_SUBTRACT_INT, (unused/1, left, right -- sub)) {

assert(cframe.use_tracing == 0);

DEOPT_IF(!PyLong_CheckExact(left), BINARY_OP);

DEOPT_IF(!PyLong_CheckExact(right), BINARY_OP);

STAT_INC(BINARY_OP, hit);

sub = _PyLong_Subtract((PyLongObject *)left, (PyLongObject *)right);

_Py_DECREF_SPECIALIZED(right, (destructor)PyObject_Free);

_Py_DECREF_SPECIALIZED(left, (destructor)PyObject_Free);

ERROR_IF(sub == NULL, error);

}

inst(BINARY_OP_SUBTRACT_FLOAT, (unused/1, left, right -- sub)) {

assert(cframe.use_tracing == 0);

DEOPT_IF(!PyFloat_CheckExact(left), BINARY_OP);

DEOPT_IF(!PyFloat_CheckExact(right), BINARY_OP);

STAT_INC(BINARY_OP, hit);

double dsub = ((PyFloatObject *)left)->ob_fval - ((PyFloatObject *)right)->ob_fval;

sub = PyFloat_FromDouble(dsub);

_Py_DECREF_SPECIALIZED(right, _PyFloat_ExactDealloc);

_Py_DECREF_SPECIALIZED(left, _PyFloat_ExactDealloc);

ERROR_IF(sub == NULL, error);

}

inst(BINARY_OP_ADD_UNICODE, (unused/1, left, right -- res)) {

assert(cframe.use_tracing == 0);

DEOPT_IF(!PyUnicode_CheckExact(left), BINARY_OP);

DEOPT_IF(Py_TYPE(right) != Py_TYPE(left), BINARY_OP);

STAT_INC(BINARY_OP, hit);

res = PyUnicode_Concat(left, right);

_Py_DECREF_SPECIALIZED(left, _PyUnicode_ExactDealloc);

_Py_DECREF_SPECIALIZED(right, _PyUnicode_ExactDealloc);

ERROR_IF(res == NULL, error);

}

// This is a subtle one. It's a super-instruction for

// BINARY_OP_ADD_UNICODE followed by STORE_FAST

// where the store goes into the left argument.

// So the inputs are the same as for all BINARY_OP

// specializations, but there is no output.

// At the end we just skip over the STORE_FAST.

inst(BINARY_OP_INPLACE_ADD_UNICODE, (left, right --)) {

assert(cframe.use_tracing == 0);

DEOPT_IF(!PyUnicode_CheckExact(left), BINARY_OP);

DEOPT_IF(Py_TYPE(right) != Py_TYPE(left), BINARY_OP);

_Py_CODEUNIT true_next = next_instr[INLINE_CACHE_ENTRIES_BINARY_OP];

assert(_Py_OPCODE(true_next) == STORE_FAST ||

_Py_OPCODE(true_next) == STORE_FAST__LOAD_FAST);

PyObject **target_local = &GETLOCAL(_Py_OPARG(true_next));

DEOPT_IF(*target_local != left, BINARY_OP);

STAT_INC(BINARY_OP, hit);

/* Handle `left = left + right` or `left += right` for str.

assert(Py_REFCNT(left) >= 2);

_Py_DECREF_NO_DEALLOC(left);

PyUnicode_Append(target_local, right);

_Py_DECREF_SPECIALIZED(right, _PyUnicode_ExactDealloc);

ERROR_IF(*target_local == NULL, error);

// The STORE_FAST is already done.

JUMPBY(INLINE_CACHE_ENTRIES_BINARY_OP + 1);

}

inst(BINARY_OP_ADD_FLOAT, (unused/1, left, right -- sum)) {

assert(cframe.use_tracing == 0);

DEOPT_IF(!PyFloat_CheckExact(left), BINARY_OP);

DEOPT_IF(Py_TYPE(right) != Py_TYPE(left), BINARY_OP);

STAT_INC(BINARY_OP, hit);

double dsum = ((PyFloatObject *)left)->ob_fval +

((PyFloatObject *)right)->ob_fval;

sum = PyFloat_FromDouble(dsum);

_Py_DECREF_SPECIALIZED(right, _PyFloat_ExactDealloc);

_Py_DECREF_SPECIALIZED(left, _PyFloat_ExactDealloc);

ERROR_IF(sum == NULL, error);

}

inst(BINARY_OP_ADD_INT, (unused/1, left, right -- sum)) {

assert(cframe.use_tracing == 0);

DEOPT_IF(!PyLong_CheckExact(left), BINARY_OP);

DEOPT_IF(Py_TYPE(right) != Py_TYPE(left), BINARY_OP);

STAT_INC(BINARY_OP, hit);

sum = _PyLong_Add((PyLongObject *)left, (PyLongObject *)right);

_Py_DECREF_SPECIALIZED(right, (destructor)PyObject_Free);

_Py_DECREF_SPECIALIZED(left, (destructor)PyObject_Free);

ERROR_IF(sum == NULL, error);

}

family(binary_subscr, INLINE_CACHE_ENTRIES_BINARY_SUBSCR) = {

BINARY_SUBSCR,

BINARY_SUBSCR_DICT,

BINARY_SUBSCR_GETITEM,

BINARY_SUBSCR_LIST_INT,

BINARY_SUBSCR_TUPLE_INT,

};

inst(BINARY_SUBSCR, (unused/4, container, sub -- res)) {

_PyBinarySubscrCache *cache = (_PyBinarySubscrCache *)next_instr;

if (ADAPTIVE_COUNTER_IS_ZERO(cache->counter)) {

assert(cframe.use_tracing == 0);

next_instr--;

_Py_Specialize_BinarySubscr(container, sub, next_instr);

DISPATCH_SAME_OPARG();

}

STAT_INC(BINARY_SUBSCR, deferred);

DECREMENT_ADAPTIVE_COUNTER(cache->counter);

res = PyObject_GetItem(container, sub);

Py_DECREF(container);

Py_DECREF(sub);

ERROR_IF(res == NULL, error);

}

inst(BINARY_SLICE, (container, start, stop -- res)) {

PyObject *slice = _PyBuildSlice_ConsumeRefs(start, stop);

// Can't use ERROR_IF() here, because we haven't

// DECREF'ed container yet, and we still own slice.

if (slice == NULL) {

res = NULL;

}

else {

res = PyObject_GetItem(container, slice);

Py_DECREF(slice);

}

Py_DECREF(container);

ERROR_IF(res == NULL, error);

}

inst(STORE_SLICE, (v, container, start, stop -- )) {

PyObject *slice = _PyBuildSlice_ConsumeRefs(start, stop);

int err;

if (slice == NULL) {

err = 1;

}

else {

err = PyObject_SetItem(container, slice, v);

Py_DECREF(slice);

}

Py_DECREF(v);

Py_DECREF(container);

ERROR_IF(err, error);

}

inst(BINARY_SUBSCR_LIST_INT, (unused/4, list, sub -- res)) {

assert(cframe.use_tracing == 0);

DEOPT_IF(!PyLong_CheckExact(sub), BINARY_SUBSCR);

DEOPT_IF(!PyList_CheckExact(list), BINARY_SUBSCR);

// Deopt unless 0 <= sub < PyList_Size(list)

Py_ssize_t signed_magnitude = Py_SIZE(sub);

DEOPT_IF(((size_t)signed_magnitude) > 1, BINARY_SUBSCR);

assert(((PyLongObject *)_PyLong_GetZero())->ob_digit[0] == 0);

Py_ssize_t index = ((PyLongObject*)sub)->ob_digit[0];

DEOPT_IF(index >= PyList_GET_SIZE(list), BINARY_SUBSCR);

STAT_INC(BINARY_SUBSCR, hit);

res = PyList_GET_ITEM(list, index);

assert(res != NULL);

Py_INCREF(res);

_Py_DECREF_SPECIALIZED(sub, (destructor)PyObject_Free);

Py_DECREF(list);

}

inst(BINARY_SUBSCR_TUPLE_INT, (unused/4, tuple, sub -- res)) {

assert(cframe.use_tracing == 0);

DEOPT_IF(!PyLong_CheckExact(sub), BINARY_SUBSCR);

DEOPT_IF(!PyTuple_CheckExact(tuple), BINARY_SUBSCR);

// Deopt unless 0 <= sub < PyTuple_Size(list)

Py_ssize_t signed_magnitude = Py_SIZE(sub);

DEOPT_IF(((size_t)signed_magnitude) > 1, BINARY_SUBSCR);

assert(((PyLongObject *)_PyLong_GetZero())->ob_digit[0] == 0);

Py_ssize_t index = ((PyLongObject*)sub)->ob_digit[0];

DEOPT_IF(index >= PyTuple_GET_SIZE(tuple), BINARY_SUBSCR);

STAT_INC(BINARY_SUBSCR, hit);

res = PyTuple_GET_ITEM(tuple, index);

assert(res != NULL);

Py_INCREF(res);

_Py_DECREF_SPECIALIZED(sub, (destructor)PyObject_Free);

Py_DECREF(tuple);

}

inst(BINARY_SUBSCR_DICT, (unused/4, dict, sub -- res)) {

assert(cframe.use_tracing == 0);

DEOPT_IF(!PyDict_CheckExact(dict), BINARY_SUBSCR);

STAT_INC(BINARY_SUBSCR, hit);

res = PyDict_GetItemWithError(dict, sub);

if (res == NULL) {

if (!_PyErr_Occurred(tstate)) {

_PyErr_SetKeyError(sub);

}

Py_DECREF(dict);

Py_DECREF(sub);

ERROR_IF(true, error);

}

Py_INCREF(res); // Do this before DECREF'ing dict, sub

Py_DECREF(dict);

Py_DECREF(sub);

}

inst(BINARY_SUBSCR_GETITEM, (unused/1, type_version/2, func_version/1, container, sub -- unused)) {

PyTypeObject *tp = Py_TYPE(container);

DEOPT_IF(tp->tp_version_tag != type_version, BINARY_SUBSCR);

assert(tp->tp_flags & Py_TPFLAGS_HEAPTYPE);

PyObject *cached = ((PyHeapTypeObject *)tp)->_spec_cache.getitem;

assert(PyFunction_Check(cached));

PyFunctionObject *getitem = (PyFunctionObject *)cached;

DEOPT_IF(getitem->func_version != func_version, BINARY_SUBSCR);

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);

_PyInterpreterFrame *new_frame = _PyFrame_PushUnchecked(tstate, getitem);

STACK_SHRINK(2);

new_frame->localsplus[0] = container;

new_frame->localsplus[1] = sub;

for (int i = 2; i < code->co_nlocalsplus; i++) {

new_frame->localsplus[i] = NULL;

}

JUMPBY(INLINE_CACHE_ENTRIES_BINARY_SUBSCR);

DISPATCH_INLINED(new_frame);

}

// Alternative: (list, unused[oparg], v -- list, unused[oparg])

inst(LIST_APPEND, (v --)) {

PyObject *list = PEEK(oparg + 1); // +1 to account for v staying on stack

ERROR_IF(_PyList_AppendTakeRef((PyListObject *)list, v) < 0, error);

PREDICT(JUMP_BACKWARD);

}

// Alternative: (set, unused[oparg], v -- set, unused[oparg])

inst(SET_ADD, (v --)) {

PyObject *set = PEEK(oparg + 1); // +1 to account for v staying on stack

int err = PySet_Add(set, v);

Py_DECREF(v);

ERROR_IF(err, error);

PREDICT(JUMP_BACKWARD);

}

family(store_subscr) = {

STORE_SUBSCR,

STORE_SUBSCR_DICT,

STORE_SUBSCR_LIST_INT,

};

inst(STORE_SUBSCR, (counter/1, v, container, sub -- )) {

if (ADAPTIVE_COUNTER_IS_ZERO(counter)) {

assert(cframe.use_tracing == 0);

next_instr--;

_Py_Specialize_StoreSubscr(container, sub, next_instr);

DISPATCH_SAME_OPARG();

}

STAT_INC(STORE_SUBSCR, deferred);

_PyStoreSubscrCache *cache = (_PyStoreSubscrCache *)next_instr;

DECREMENT_ADAPTIVE_COUNTER(cache->counter);

/* container[sub] = v */

int err = PyObject_SetItem(container, sub, v);

Py_DECREF(v);

Py_DECREF(container);

Py_DECREF(sub);

ERROR_IF(err, error);

}

inst(STORE_SUBSCR_LIST_INT, (unused/1, value, list, sub -- )) {

assert(cframe.use_tracing == 0);

DEOPT_IF(!PyLong_CheckExact(sub), STORE_SUBSCR);

DEOPT_IF(!PyList_CheckExact(list), STORE_SUBSCR);

// Ensure nonnegative, zero-or-one-digit ints.

DEOPT_IF(((size_t)Py_SIZE(sub)) > 1, STORE_SUBSCR);

Py_ssize_t index = ((PyLongObject*)sub)->ob_digit[0];

// 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);

}

inst(STORE_SUBSCR_DICT, (unused/1, value, dict, sub -- )) {

assert(cframe.use_tracing == 0);

DEOPT_IF(!PyDict_CheckExact(dict), STORE_SUBSCR);

STAT_INC(STORE_SUBSCR, hit);

int err = _PyDict_SetItem_Take2((PyDictObject *)dict, sub, value);

Py_DECREF(dict);

ERROR_IF(err, error);

}

inst(DELETE_SUBSCR, (container, sub --)) {

/* del container[sub] */

int err = PyObject_DelItem(container, sub);

Py_DECREF(container);

Py_DECREF(sub);

ERROR_IF(err, error);

}

inst(PRINT_EXPR, (value --)) {

PyObject *hook = _PySys_GetAttr(tstate, &_Py_ID(displayhook));

PyObject *res;

// Can't use ERROR_IF here.

if (hook == NULL) {

_PyErr_SetString(tstate, PyExc_RuntimeError,

"lost sys.displayhook");

Py_DECREF(value);

ERROR_IF(true, error);

}

res = PyObject_CallOneArg(hook, value);

Py_DECREF(value);

ERROR_IF(res == NULL, error);

Py_DECREF(res);

}

// stack effect: (__array[oparg] -- )

inst(RAISE_VARARGS) {

PyObject *cause = NULL, *exc = NULL;

switch (oparg) {

case 2:

cause = POP(); /* cause */

/* fall through */

case 1:

exc = POP(); /* exc */

/* fall through */

case 0:

if (do_raise(tstate, exc, cause)) {

goto exception_unwind;

}

break;

default:

_PyErr_SetString(tstate, PyExc_SystemError,

"bad RAISE_VARARGS oparg");

break;

}

goto error;

}

inst(INTERPRETER_EXIT, (retval --)) {

assert(frame == &entry_frame);

assert(_PyFrame_IsIncomplete(frame));

STACK_SHRINK(1); // Since we're not going to DISPATCH()

assert(EMPTY());

/* Restore previous cframe and return. */

tstate->cframe = cframe.previous;

tstate->cframe->use_tracing = cframe.use_tracing;

assert(tstate->cframe->current_frame == frame->previous);

assert(!_PyErr_Occurred(tstate));

_Py_LeaveRecursiveCallTstate(tstate);

return retval;

}

inst(RETURN_VALUE, (retval --)) {

STACK_SHRINK(1);

assert(EMPTY());

_PyFrame_SetStackPointer(frame, stack_pointer);

TRACE_FUNCTION_EXIT();

DTRACE_FUNCTION_EXIT();

_Py_LeaveRecursiveCallPy(tstate);

assert(frame != &entry_frame);

// GH-99729: We need to unlink the frame *before* clearing it:

_PyInterpreterFrame *dying = frame;

frame = cframe.current_frame = dying->previous;

_PyEvalFrameClearAndPop(tstate, dying);

_PyFrame_StackPush(frame, retval);

goto resume_frame;

}

inst(GET_AITER, (obj -- iter)) {

unaryfunc getter = NULL;

PyTypeObject *type = Py_TYPE(obj);

if (type->tp_as_async != NULL) {

getter = type->tp_as_async->am_aiter;

}

if (getter == NULL) {

_PyErr_Format(tstate, PyExc_TypeError,

"'async for' requires an object with "

"__aiter__ method, got %.100s",

type->tp_name);

Py_DECREF(obj);

ERROR_IF(true, error);

}

iter = (*getter)(obj);

Py_DECREF(obj);

ERROR_IF(iter == NULL, error);

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);

ERROR_IF(true, error);

}

}

// stack effect: ( -- __0)

inst(GET_ANEXT) {

unaryfunc getter = NULL;

PyObject *next_iter = NULL;

PyObject *awaitable = NULL;

PyObject *aiter = TOP();

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);

}

}

PUSH(awaitable);

PREDICT(LOAD_CONST);

}

// stack effect: ( -- )

inst(GET_AWAITABLE) {

PyObject *iterable = TOP();

PyObject *iter = _PyCoro_GetAwaitableIter(iterable);

if (iter == NULL) {

format_awaitable_error(tstate, Py_TYPE(iterable), oparg);

}

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

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. */

}

}

SET_TOP(iter); /* Even if it's NULL */

if (iter == NULL) {

goto error;

}

PREDICT(LOAD_CONST);

}

// error: SEND stack effect depends on jump flag

inst(SEND) {

assert(frame != &entry_frame);

assert(STACK_LEVEL() >= 2);

PyObject *v = POP();

PyObject *receiver = TOP();

PySendResult gen_status;

PyObject *retval;

if (tstate->c_tracefunc == NULL) {

gen_status = PyIter_Send(receiver, v, &retval);

} else {

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) {

if (tstate->c_tracefunc != NULL

&& _PyErr_ExceptionMatches(tstate, PyExc_StopIteration))

call_exc_trace(tstate->c_tracefunc, tstate->c_traceobj, tstate, frame);

if (_PyGen_FetchStopIterationValue(&retval) == 0) {

gen_status = PYGEN_RETURN;

}

else {

gen_status = PYGEN_ERROR;

}

}

else {

gen_status = PYGEN_NEXT;

}

}

Py_DECREF(v);

if (gen_status == PYGEN_ERROR) {

assert(retval == NULL);

goto error;

}

if (gen_status == PYGEN_RETURN) {

assert(retval != NULL);

Py_DECREF(receiver);

SET_TOP(retval);

JUMPBY(oparg);

}

else {

assert(gen_status == PYGEN_NEXT);

assert(retval != NULL);

PUSH(retval);

}

}

// stack effect: ( -- )

inst(ASYNC_GEN_WRAP) {

PyObject *v = TOP();

assert(frame->f_code->co_flags & CO_ASYNC_GENERATOR);

PyObject *w = _PyAsyncGenValueWrapperNew(v);

if (w == NULL) {

goto error;

}

SET_TOP(w);

Py_DECREF(v);

}

// stack effect: ( -- )

inst(YIELD_VALUE) {

// 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.

assert(oparg == STACK_LEVEL());

assert(frame != &entry_frame);

PyObject *retval = POP();

PyGenObject *gen = _PyFrame_GetGenerator(frame);

gen->gi_frame_state = FRAME_SUSPENDED;

_PyFrame_SetStackPointer(frame, stack_pointer);

TRACE_FUNCTION_EXIT();

DTRACE_FUNCTION_EXIT();

tstate->exc_info = gen->gi_exc_state.previous_item;

gen->gi_exc_state.previous_item = NULL;

_Py_LeaveRecursiveCallPy(tstate);

_PyInterpreterFrame *gen_frame = frame;

frame = cframe.current_frame = frame->previous;

gen_frame->previous = NULL;

frame->prev_instr -= frame->yield_offset;

_PyFrame_StackPush(frame, retval);

goto resume_frame;

}

// stack effect: (__0 -- )

inst(POP_EXCEPT) {

_PyErr_StackItem *exc_info = tstate->exc_info;

PyObject *value = exc_info->exc_value;

exc_info->exc_value = POP();

Py_XDECREF(value);

}

// stack effect: (__0 -- )

inst(RERAISE) {

if (oparg) {

PyObject *lasti = PEEK(oparg + 1);

if (PyLong_Check(lasti)) {

frame->prev_instr = _PyCode_CODE(frame->f_code) + PyLong_AsLong(lasti);

assert(!_PyErr_Occurred(tstate));

}

else {

assert(PyLong_Check(lasti));

_PyErr_SetString(tstate, PyExc_SystemError, "lasti is not an int");

goto error;

}

}

PyObject *val = POP();

assert(val && PyExceptionInstance_Check(val));

PyObject *exc = Py_NewRef(PyExceptionInstance_Class(val));

PyObject *tb = PyException_GetTraceback(val);

_PyErr_Restore(tstate, exc, val, tb);

goto exception_unwind;

}

// stack effect: (__0 -- )

inst(PREP_RERAISE_STAR) {

PyObject *excs = POP();

assert(PyList_Check(excs));

PyObject *orig = POP();

PyObject *val = _PyExc_PrepReraiseStar(orig, excs);

Py_DECREF(excs);

Py_DECREF(orig);

if (val == NULL) {

goto error;

}

PUSH(val);

}

// stack effect: (__0, __1 -- )

inst(END_ASYNC_FOR) {

PyObject *val = POP();

assert(val && PyExceptionInstance_Check(val));

if (PyErr_GivenExceptionMatches(val, PyExc_StopAsyncIteration)) {

Py_DECREF(val);

Py_DECREF(POP());

}

else {

PyObject *exc = Py_NewRef(PyExceptionInstance_Class(val));

PyObject *tb = PyException_GetTraceback(val);

_PyErr_Restore(tstate, exc, val, tb);

goto exception_unwind;

}

}

// stack effect: (__0, __1 -- )

inst(CLEANUP_THROW) {

assert(throwflag);

PyObject *exc_value = TOP();

assert(exc_value && PyExceptionInstance_Check(exc_value));

if (PyErr_GivenExceptionMatches(exc_value, PyExc_StopIteration)) {

PyObject *value = ((PyStopIterationObject *)exc_value)->value;

Py_INCREF(value);

Py_DECREF(POP()); // The StopIteration.

Py_DECREF(POP()); // The last sent value.

Py_DECREF(POP()); // The delegated sub-iterator.

PUSH(value);

}

else {

PyObject *exc_type = Py_NewRef(Py_TYPE(exc_value));

PyObject *exc_traceback = PyException_GetTraceback(exc_value);

_PyErr_Restore(tstate, exc_type, Py_NewRef(exc_value), exc_traceback);

goto exception_unwind;

}

}

inst(STOPITERATION_ERROR) {

assert(frame->owner == FRAME_OWNED_BY_GENERATOR);

PyObject *exc = TOP();

assert(PyExceptionInstance_Check(exc));

const char *msg = NULL;

if (PyErr_GivenExceptionMatches(exc, PyExc_StopIteration)) {

msg = "generator raised StopIteration";

if (frame->f_code->co_flags & CO_ASYNC_GENERATOR) {

msg = "async generator raised StopIteration";

}

else if (frame->f_code->co_flags & CO_COROUTINE) {

msg = "coroutine raised StopIteration";

}

}

else if ((frame->f_code->co_flags & CO_ASYNC_GENERATOR) &&

PyErr_GivenExceptionMatches(exc, PyExc_StopAsyncIteration))

{

/* code in `gen` raised a StopAsyncIteration error:

msg = "async generator raised StopAsyncIteration";

}

if (msg != NULL) {

PyObject *message = _PyUnicode_FromASCII(msg, strlen(msg));

if (message == NULL) {

goto error;

}

PyObject *error = PyObject_CallOneArg(PyExc_RuntimeError, message);

if (error == NULL) {

Py_DECREF(message);

goto error;

}

assert(PyExceptionInstance_Check(error));

SET_TOP(error);

PyException_SetCause(error, Py_NewRef(exc));

// Steal exc reference, rather than Py_NewRef+Py_DECREF

PyException_SetContext(error, exc);

Py_DECREF(message);

}

}

// stack effect: ( -- __0)

inst(LOAD_ASSERTION_ERROR) {

PyObject *value = PyExc_AssertionError;

PUSH(Py_NewRef(value));

}

// stack effect: ( -- __0)

inst(LOAD_BUILD_CLASS) {

PyObject *bc;

if (PyDict_CheckExact(BUILTINS())) {

bc = _PyDict_GetItemWithError(BUILTINS(),

&_Py_ID(__build_class__));

if (bc == NULL) {

if (!_PyErr_Occurred(tstate)) {

_PyErr_SetString(tstate, PyExc_NameError,

"__build_class__ not found");

}

goto error;

}

Py_INCREF(bc);

}

else {

bc = PyObject_GetItem(BUILTINS(), &_Py_ID(__build_class__));

if (bc == NULL) {

if (_PyErr_ExceptionMatches(tstate, PyExc_KeyError))

_PyErr_SetString(tstate, PyExc_NameError,

"__build_class__ not found");

goto error;

}

}

PUSH(bc);

}

// stack effect: (__0 -- )

inst(STORE_NAME) {

PyObject *name = GETITEM(names, oparg);

PyObject *v = POP();

PyObject *ns = LOCALS();

int err;

if (ns == NULL) {

_PyErr_Format(tstate, PyExc_SystemError,

"no locals found when storing %R", name);

Py_DECREF(v);

goto error;

}

if (PyDict_CheckExact(ns))

err = PyDict_SetItem(ns, name, v);

else

err = PyObject_SetItem(ns, name, v);

Py_DECREF(v);

if (err != 0)

goto error;

}

inst(DELETE_NAME, (--)) {

PyObject *name = GETITEM(names, oparg);