Branch data     Line data    Source code

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