xref: /third_party/python/Modules/_opcode.c (revision 7db96d56) 
1#include "Python.h"
2#include "opcode.h"
3#include "internal/pycore_code.h"
4
5/*[clinic input]
6module _opcode
7[clinic start generated code]*/
8/*[clinic end generated code: output=da39a3ee5e6b4b0d input=117442e66eb376e6]*/
9
10#include "clinic/_opcode.c.h"
11
12/*[clinic input]
13
14_opcode.stack_effect -> int
15
16  opcode: int
17  oparg: object = None
18  /
19  *
20  jump: object = None
21
22Compute the stack effect of the opcode.
23[clinic start generated code]*/
24
25static int
26_opcode_stack_effect_impl(PyObject *module, int opcode, PyObject *oparg,
27                          PyObject *jump)
28/*[clinic end generated code: output=64a18f2ead954dbb input=461c9d4a44851898]*/
29{
30    int effect;
31    int oparg_int = 0;
32    int jump_int;
33    if (HAS_ARG(opcode)) {
34        if (oparg == Py_None) {
35            PyErr_SetString(PyExc_ValueError,
36                    "stack_effect: opcode requires oparg but oparg was not specified");
37            return -1;
38        }
39        oparg_int = (int)PyLong_AsLong(oparg);
40        if ((oparg_int == -1) && PyErr_Occurred()) {
41            return -1;
42        }
43    }
44    else if (oparg != Py_None) {
45        PyErr_SetString(PyExc_ValueError,
46                "stack_effect: opcode does not permit oparg but oparg was specified");
47        return -1;
48    }
49    if (jump == Py_None) {
50        jump_int = -1;
51    }
52    else if (jump == Py_True) {
53        jump_int = 1;
54    }
55    else if (jump == Py_False) {
56        jump_int = 0;
57    }
58    else {
59        PyErr_SetString(PyExc_ValueError,
60                "stack_effect: jump must be False, True or None");
61        return -1;
62    }
63    if (IS_ARTIFICIAL(opcode)) {
64        effect = PY_INVALID_STACK_EFFECT;
65    }
66    else {
67        effect = PyCompile_OpcodeStackEffectWithJump(opcode, oparg_int, jump_int);
68    }
69    if (effect == PY_INVALID_STACK_EFFECT) {
70            PyErr_SetString(PyExc_ValueError,
71                    "invalid opcode or oparg");
72            return -1;
73    }
74    return effect;
75}
76
77/*[clinic input]
78
79_opcode.get_specialization_stats
80
81Return the specialization stats
82[clinic start generated code]*/
83
84static PyObject *
85_opcode_get_specialization_stats_impl(PyObject *module)
86/*[clinic end generated code: output=fcbc32fdfbec5c17 input=e1f60db68d8ce5f6]*/
87{
88#ifdef Py_STATS
89    return _Py_GetSpecializationStats();
90#else
91    Py_RETURN_NONE;
92#endif
93}
94
95static PyMethodDef
96opcode_functions[] =  {
97    _OPCODE_STACK_EFFECT_METHODDEF
98    _OPCODE_GET_SPECIALIZATION_STATS_METHODDEF
99    {NULL, NULL, 0, NULL}
100};
101
102static struct PyModuleDef opcodemodule = {
103    PyModuleDef_HEAD_INIT,
104    .m_name = "_opcode",
105    .m_doc = "Opcode support module.",
106    .m_size = 0,
107    .m_methods = opcode_functions
108};
109
110PyMODINIT_FUNC
111PyInit__opcode(void)
112{
113    return PyModuleDef_Init(&opcodemodule);
114}
115