1#include "Python.h"
2#include "pycore_symtable.h"      // struct symtable
3
4#include "clinic/symtablemodule.c.h"
5/*[clinic input]
6module _symtable
7[clinic start generated code]*/
8/*[clinic end generated code: output=da39a3ee5e6b4b0d input=f4685845a7100605]*/
9
10
11/*[clinic input]
12_symtable.symtable
13
14    source:    object
15    filename:  object(converter='PyUnicode_FSDecoder')
16    startstr:  str
17    /
18
19Return symbol and scope dictionaries used internally by compiler.
20[clinic start generated code]*/
21
22static PyObject *
23_symtable_symtable_impl(PyObject *module, PyObject *source,
24                        PyObject *filename, const char *startstr)
25/*[clinic end generated code: output=59eb0d5fc7285ac4 input=9dd8a50c0c36a4d7]*/
26{
27    struct symtable *st;
28    PyObject *t;
29    int start;
30    PyCompilerFlags cf = _PyCompilerFlags_INIT;
31    PyObject *source_copy = NULL;
32
33    cf.cf_flags = PyCF_SOURCE_IS_UTF8;
34
35    const char *str = _Py_SourceAsString(source, "symtable", "string or bytes", &cf, &source_copy);
36    if (str == NULL) {
37        return NULL;
38    }
39
40    if (strcmp(startstr, "exec") == 0)
41        start = Py_file_input;
42    else if (strcmp(startstr, "eval") == 0)
43        start = Py_eval_input;
44    else if (strcmp(startstr, "single") == 0)
45        start = Py_single_input;
46    else {
47        PyErr_SetString(PyExc_ValueError,
48           "symtable() arg 3 must be 'exec' or 'eval' or 'single'");
49        Py_DECREF(filename);
50        Py_XDECREF(source_copy);
51        return NULL;
52    }
53    st = _Py_SymtableStringObjectFlags(str, filename, start, &cf);
54    Py_DECREF(filename);
55    Py_XDECREF(source_copy);
56    if (st == NULL) {
57        return NULL;
58    }
59    t = (PyObject *)st->st_top;
60    Py_INCREF(t);
61    _PySymtable_Free(st);
62    return t;
63}
64
65static PyMethodDef symtable_methods[] = {
66    _SYMTABLE_SYMTABLE_METHODDEF
67    {NULL,              NULL}           /* sentinel */
68};
69
70static int
71symtable_init_stentry_type(PyObject *m)
72{
73    return PyType_Ready(&PySTEntry_Type);
74}
75
76static int
77symtable_init_constants(PyObject *m)
78{
79    if (PyModule_AddIntMacro(m, USE) < 0) return -1;
80    if (PyModule_AddIntMacro(m, DEF_GLOBAL) < 0) return -1;
81    if (PyModule_AddIntMacro(m, DEF_NONLOCAL) < 0) return -1;
82    if (PyModule_AddIntMacro(m, DEF_LOCAL) < 0) return -1;
83    if (PyModule_AddIntMacro(m, DEF_PARAM) < 0) return -1;
84    if (PyModule_AddIntMacro(m, DEF_FREE) < 0) return -1;
85    if (PyModule_AddIntMacro(m, DEF_FREE_CLASS) < 0) return -1;
86    if (PyModule_AddIntMacro(m, DEF_IMPORT) < 0) return -1;
87    if (PyModule_AddIntMacro(m, DEF_BOUND) < 0) return -1;
88    if (PyModule_AddIntMacro(m, DEF_ANNOT) < 0) return -1;
89
90    if (PyModule_AddIntConstant(m, "TYPE_FUNCTION", FunctionBlock) < 0)
91        return -1;
92    if (PyModule_AddIntConstant(m, "TYPE_CLASS", ClassBlock) < 0) return -1;
93    if (PyModule_AddIntConstant(m, "TYPE_MODULE", ModuleBlock) < 0)
94        return -1;
95
96    if (PyModule_AddIntMacro(m, LOCAL) < 0) return -1;
97    if (PyModule_AddIntMacro(m, GLOBAL_EXPLICIT) < 0) return -1;
98    if (PyModule_AddIntMacro(m, GLOBAL_IMPLICIT) < 0) return -1;
99    if (PyModule_AddIntMacro(m, FREE) < 0) return -1;
100    if (PyModule_AddIntMacro(m, CELL) < 0) return -1;
101
102    if (PyModule_AddIntConstant(m, "SCOPE_OFF", SCOPE_OFFSET) < 0) return -1;
103    if (PyModule_AddIntMacro(m, SCOPE_MASK) < 0) return -1;
104
105    return 0;
106}
107
108static PyModuleDef_Slot symtable_slots[] = {
109    {Py_mod_exec, symtable_init_stentry_type},
110    {Py_mod_exec, symtable_init_constants},
111    {0, NULL}
112};
113
114static struct PyModuleDef symtablemodule = {
115    PyModuleDef_HEAD_INIT,
116    .m_name = "_symtable",
117    .m_size = 0,
118    .m_methods = symtable_methods,
119    .m_slots = symtable_slots,
120};
121
122PyMODINIT_FUNC
123PyInit__symtable(void)
124{
125    return PyModuleDef_Init(&symtablemodule);
126}
127