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  _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  
      65  static PyMethodDef symtable_methods[] = {
      66      _SYMTABLE_SYMTABLE_METHODDEF
      67      {NULL,              NULL}           /* sentinel */
      68  };
      69  
      70  static int
      71  symtable_init_stentry_type(PyObject *m)
      72  {
      73      return PyType_Ready(&PySTEntry_Type);
      74  }
      75  
      76  static int
      77  symtable_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  
     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  PyInit__symtable(void)
     124  {
     125      return PyModuleDef_Init(&symtablemodule);
     126  }