cpython/Objects/funcobject.c

345 lines
8.6 KiB
C
Raw Normal View History

1991-02-19 12:39:46 +00:00
1990-10-14 12:07:46 +00:00
/* Function object implementation */
1997-05-02 03:12:38 +00:00
#include "Python.h"
#include "compile.h"
1990-12-20 15:06:42 +00:00
#include "structmember.h"
1990-10-14 12:07:46 +00:00
1997-05-02 03:12:38 +00:00
PyObject *
2000-07-09 06:03:25 +00:00
PyFunction_New(PyObject *code, PyObject *globals)
1990-10-14 12:07:46 +00:00
{
1997-05-02 03:12:38 +00:00
PyFunctionObject *op = PyObject_NEW(PyFunctionObject,
&PyFunction_Type);
1990-10-14 12:07:46 +00:00
if (op != NULL) {
1997-05-02 03:12:38 +00:00
PyObject *doc;
PyObject *consts;
op->func_weakreflist = NULL;
1997-05-02 03:12:38 +00:00
Py_INCREF(code);
op->func_code = code;
1997-05-02 03:12:38 +00:00
Py_INCREF(globals);
1990-10-14 12:07:46 +00:00
op->func_globals = globals;
1997-05-02 03:12:38 +00:00
op->func_name = ((PyCodeObject *)code)->co_name;
Py_INCREF(op->func_name);
op->func_defaults = NULL; /* No default arguments */
PEP 227 implementation The majority of the changes are in the compiler. The mainloop changes primarily to implement the new opcodes and to pass a function's closure to eval_code2(). Frames and functions got new slots to hold the closure. Include/compile.h Add co_freevars and co_cellvars slots to code objects. Update PyCode_New() to take freevars and cellvars as arguments Include/funcobject.h Add func_closure slot to function objects. Add GetClosure()/SetClosure() functions (and corresponding macros) for getting at the closure. Include/frameobject.h PyFrame_New() now takes a closure. Include/opcode.h Add four new opcodes: MAKE_CLOSURE, LOAD_CLOSURE, LOAD_DEREF, STORE_DEREF. Remove comment about old requirement for opcodes to fit in 7 bits. compile.c Implement changes to code objects for co_freevars and co_cellvars. Modify symbol table to use st_cur_name (string object for the name of the current scope) and st_cur_children (list of nested blocks). Also define st_nested, which might more properly be called st_cur_nested. Add several DEF_XXX flags to track def-use information for free variables. New or modified functions of note: com_make_closure(struct compiling *, PyCodeObject *) Emit LOAD_CLOSURE opcodes as needed to pass cells for free variables into nested scope. com_addop_varname(struct compiling *, int, char *) Emits opcodes for LOAD_DEREF and STORE_DEREF. get_ref_type(struct compiling *, char *name) Return NAME_CLOSURE if ref type is FREE or CELL symtable_load_symbols(struct compiling *) Decides what variables are cell or free based on def-use info. Can now raise SyntaxError if nested scopes are mixed with exec or from blah import *. make_scope_info(PyObject *, PyObject *, int, int) Helper functions for symtable scope stack. symtable_update_free_vars(struct symtable *) After a code block has been analyzed, it must check each of its children for free variables that are not defined in the block. If a variable is free in a child and not defined in the parent, then it is defined by block the enclosing the current one or it is a global. This does the right logic. symtable_add_use() is now a macro for symtable_add_def() symtable_assign(struct symtable *, node *) Use goto instead of for (;;) Fixed bug in symtable where name of keyword argument in function call was treated as assignment in the scope of the call site. Ex: def f(): g(a=2) # a was considered a local of f ceval.c eval_code2() now take one more argument, a closure. Implement LOAD_CLOSURE, LOAD_DEREF, STORE_DEREF, MAKE_CLOSURE> Also: When name error occurs for global variable, report that the name was global in the error mesage. Objects/frameobject.c Initialize f_closure to be a tuple containing space for cellvars and freevars. f_closure is NULL if neither are present. Objects/funcobject.c Add support for func_closure. Python/import.c Change the magic number. Python/marshal.c Track changes to code objects.
2001-01-25 20:06:59 +00:00
op->func_closure = NULL;
1997-05-02 03:12:38 +00:00
consts = ((PyCodeObject *)code)->co_consts;
if (PyTuple_Size(consts) >= 1) {
doc = PyTuple_GetItem(consts, 0);
if (!PyString_Check(doc) && !PyUnicode_Check(doc))
1997-05-02 03:12:38 +00:00
doc = Py_None;
}
else
1997-05-02 03:12:38 +00:00
doc = Py_None;
Py_INCREF(doc);
op->func_doc = doc;
op->func_dict = NULL;
1990-10-14 12:07:46 +00:00
}
else
return NULL;
PyObject_GC_Init(op);
1997-05-02 03:12:38 +00:00
return (PyObject *)op;
1990-10-14 12:07:46 +00:00
}
1997-05-02 03:12:38 +00:00
PyObject *
2000-07-09 06:03:25 +00:00
PyFunction_GetCode(PyObject *op)
1990-10-14 12:07:46 +00:00
{
1997-05-02 03:12:38 +00:00
if (!PyFunction_Check(op)) {
PyErr_BadInternalCall();
1990-10-14 12:07:46 +00:00
return NULL;
}
1997-05-02 03:12:38 +00:00
return ((PyFunctionObject *) op) -> func_code;
1990-10-14 12:07:46 +00:00
}
1997-05-02 03:12:38 +00:00
PyObject *
2000-07-09 06:03:25 +00:00
PyFunction_GetGlobals(PyObject *op)
1990-10-14 12:07:46 +00:00
{
1997-05-02 03:12:38 +00:00
if (!PyFunction_Check(op)) {
PyErr_BadInternalCall();
1990-10-14 12:07:46 +00:00
return NULL;
}
1997-05-02 03:12:38 +00:00
return ((PyFunctionObject *) op) -> func_globals;
1990-10-14 12:07:46 +00:00
}
1997-05-02 03:12:38 +00:00
PyObject *
2000-07-09 06:03:25 +00:00
PyFunction_GetDefaults(PyObject *op)
1994-08-30 08:27:36 +00:00
{
1997-05-02 03:12:38 +00:00
if (!PyFunction_Check(op)) {
PyErr_BadInternalCall();
1994-08-30 08:27:36 +00:00
return NULL;
}
1997-05-02 03:12:38 +00:00
return ((PyFunctionObject *) op) -> func_defaults;
1994-08-30 08:27:36 +00:00
}
int
2000-07-09 06:03:25 +00:00
PyFunction_SetDefaults(PyObject *op, PyObject *defaults)
1994-08-30 08:27:36 +00:00
{
1997-05-02 03:12:38 +00:00
if (!PyFunction_Check(op)) {
PyErr_BadInternalCall();
1994-08-30 08:27:36 +00:00
return -1;
}
1997-05-02 03:12:38 +00:00
if (defaults == Py_None)
defaults = NULL;
1998-04-10 22:16:39 +00:00
else if (PyTuple_Check(defaults)) {
1997-05-02 03:12:38 +00:00
Py_XINCREF(defaults);
1998-04-10 22:16:39 +00:00
}
1994-08-30 08:27:36 +00:00
else {
1997-05-02 03:12:38 +00:00
PyErr_SetString(PyExc_SystemError, "non-tuple default args");
1994-08-30 08:27:36 +00:00
return -1;
}
1997-05-02 03:12:38 +00:00
Py_XDECREF(((PyFunctionObject *) op) -> func_defaults);
((PyFunctionObject *) op) -> func_defaults = defaults;
1994-08-30 08:27:36 +00:00
return 0;
}
PEP 227 implementation The majority of the changes are in the compiler. The mainloop changes primarily to implement the new opcodes and to pass a function's closure to eval_code2(). Frames and functions got new slots to hold the closure. Include/compile.h Add co_freevars and co_cellvars slots to code objects. Update PyCode_New() to take freevars and cellvars as arguments Include/funcobject.h Add func_closure slot to function objects. Add GetClosure()/SetClosure() functions (and corresponding macros) for getting at the closure. Include/frameobject.h PyFrame_New() now takes a closure. Include/opcode.h Add four new opcodes: MAKE_CLOSURE, LOAD_CLOSURE, LOAD_DEREF, STORE_DEREF. Remove comment about old requirement for opcodes to fit in 7 bits. compile.c Implement changes to code objects for co_freevars and co_cellvars. Modify symbol table to use st_cur_name (string object for the name of the current scope) and st_cur_children (list of nested blocks). Also define st_nested, which might more properly be called st_cur_nested. Add several DEF_XXX flags to track def-use information for free variables. New or modified functions of note: com_make_closure(struct compiling *, PyCodeObject *) Emit LOAD_CLOSURE opcodes as needed to pass cells for free variables into nested scope. com_addop_varname(struct compiling *, int, char *) Emits opcodes for LOAD_DEREF and STORE_DEREF. get_ref_type(struct compiling *, char *name) Return NAME_CLOSURE if ref type is FREE or CELL symtable_load_symbols(struct compiling *) Decides what variables are cell or free based on def-use info. Can now raise SyntaxError if nested scopes are mixed with exec or from blah import *. make_scope_info(PyObject *, PyObject *, int, int) Helper functions for symtable scope stack. symtable_update_free_vars(struct symtable *) After a code block has been analyzed, it must check each of its children for free variables that are not defined in the block. If a variable is free in a child and not defined in the parent, then it is defined by block the enclosing the current one or it is a global. This does the right logic. symtable_add_use() is now a macro for symtable_add_def() symtable_assign(struct symtable *, node *) Use goto instead of for (;;) Fixed bug in symtable where name of keyword argument in function call was treated as assignment in the scope of the call site. Ex: def f(): g(a=2) # a was considered a local of f ceval.c eval_code2() now take one more argument, a closure. Implement LOAD_CLOSURE, LOAD_DEREF, STORE_DEREF, MAKE_CLOSURE> Also: When name error occurs for global variable, report that the name was global in the error mesage. Objects/frameobject.c Initialize f_closure to be a tuple containing space for cellvars and freevars. f_closure is NULL if neither are present. Objects/funcobject.c Add support for func_closure. Python/import.c Change the magic number. Python/marshal.c Track changes to code objects.
2001-01-25 20:06:59 +00:00
PyObject *
PyFunction_GetClosure(PyObject *op)
{
if (!PyFunction_Check(op)) {
PyErr_BadInternalCall();
return NULL;
}
return ((PyFunctionObject *) op) -> func_closure;
}
int
PyFunction_SetClosure(PyObject *op, PyObject *closure)
{
if (!PyFunction_Check(op)) {
PyErr_BadInternalCall();
return -1;
}
if (closure == Py_None)
closure = NULL;
else if (PyTuple_Check(closure)) {
Py_XINCREF(closure);
}
else {
PyErr_SetString(PyExc_SystemError, "non-tuple closure");
return -1;
}
Py_XDECREF(((PyFunctionObject *) op) -> func_closure);
((PyFunctionObject *) op) -> func_closure = closure;
return 0;
}
1990-10-14 12:07:46 +00:00
/* Methods */
1997-05-02 03:12:38 +00:00
#define OFF(x) offsetof(PyFunctionObject, x)
1990-12-20 15:06:42 +00:00
static struct memberlist func_memberlist[] = {
{"func_code", T_OBJECT, OFF(func_code)},
{"func_globals", T_OBJECT, OFF(func_globals), READONLY},
{"func_name", T_OBJECT, OFF(func_name), READONLY},
{"__name__", T_OBJECT, OFF(func_name), READONLY},
{"func_closure", T_OBJECT, OFF(func_closure), READONLY},
{"func_defaults", T_OBJECT, OFF(func_defaults)},
{"func_doc", T_OBJECT, OFF(func_doc)},
{"__doc__", T_OBJECT, OFF(func_doc)},
{"func_dict", T_OBJECT, OFF(func_dict)},
{"__dict__", T_OBJECT, OFF(func_dict)},
{NULL} /* Sentinel */
1990-12-20 15:06:42 +00:00
};
1997-05-02 03:12:38 +00:00
static PyObject *
func_getattro(PyFunctionObject *op, PyObject *name)
1990-12-20 15:06:42 +00:00
{
PyObject *rtn;
char *sname = PyString_AsString(name);
if (sname[0] != '_' && PyEval_GetRestricted()) {
1997-05-02 03:12:38 +00:00
PyErr_SetString(PyExc_RuntimeError,
1995-01-10 10:39:49 +00:00
"function attributes not accessible in restricted mode");
return NULL;
}
/* no API for PyMember_HasAttr() */
rtn = PyMember_Get((char *)op, func_memberlist, sname);
if (rtn == NULL && PyErr_ExceptionMatches(PyExc_AttributeError)) {
PyErr_Clear();
if (op->func_dict != NULL) {
rtn = PyDict_GetItem(op->func_dict, name);
Py_XINCREF(rtn);
}
if (rtn == NULL)
PyErr_SetObject(PyExc_AttributeError, name);
}
return rtn;
1990-12-20 15:06:42 +00:00
}
static int
func_setattro(PyFunctionObject *op, PyObject *name, PyObject *value)
{
int rtn;
char *sname = PyString_AsString(name);
if (PyEval_GetRestricted()) {
PyErr_SetString(PyExc_RuntimeError,
"function attributes not settable in restricted mode");
return -1;
}
if (strcmp(sname, "func_code") == 0) {
/* not legal to del f.func_code or to set it to anything
* other than a code object.
*/
if (value == NULL || !PyCode_Check(value)) {
PyErr_SetString(
PyExc_TypeError,
"func_code must be set to a code object");
return -1;
}
}
else if (strcmp(sname, "func_defaults") == 0) {
/* legal to del f.func_defaults. Can only set
* func_defaults to NULL or a tuple.
*/
if (value == Py_None)
value = NULL;
if (value != NULL && !PyTuple_Check(value)) {
PyErr_SetString(
PyExc_TypeError,
"func_defaults must be set to a tuple object");
return -1;
}
}
else if (!strcmp(sname, "func_dict") || !strcmp(sname, "__dict__")) {
/* legal to del f.func_dict. Can only set func_dict to
* NULL or a dictionary.
*/
if (value == Py_None)
value = NULL;
if (value != NULL && !PyDict_Check(value)) {
PyErr_SetString(
PyExc_TypeError,
"func_dict must be set to a dict object");
return -1;
}
}
rtn = PyMember_Set((char *)op, func_memberlist, sname, value);
if (rtn < 0 && PyErr_ExceptionMatches(PyExc_AttributeError)) {
PyErr_Clear();
if (op->func_dict == NULL) {
/* don't create the dict if we're deleting an
* attribute. In that case, we know we'll get an
* AttributeError.
*/
if (value == NULL) {
PyErr_SetString(PyExc_AttributeError, sname);
return -1;
}
op->func_dict = PyDict_New();
if (op->func_dict == NULL)
return -1;
}
if (value == NULL)
rtn = PyDict_DelItem(op->func_dict, name);
else
rtn = PyDict_SetItem(op->func_dict, name, value);
/* transform KeyError into AttributeError */
if (rtn < 0 && PyErr_ExceptionMatches(PyExc_KeyError))
PyErr_SetString(PyExc_AttributeError, sname);
}
return rtn;
}
1990-10-14 12:07:46 +00:00
static void
2000-07-09 06:03:25 +00:00
func_dealloc(PyFunctionObject *op)
1990-10-14 12:07:46 +00:00
{
PyObject_ClearWeakRefs((PyObject *) op);
PyObject_GC_Fini(op);
1997-05-02 03:12:38 +00:00
Py_DECREF(op->func_code);
Py_DECREF(op->func_globals);
Py_DECREF(op->func_name);
Py_XDECREF(op->func_defaults);
Py_XDECREF(op->func_doc);
Py_XDECREF(op->func_dict);
Py_XDECREF(op->func_closure);
2000-07-01 01:00:38 +00:00
op = (PyFunctionObject *) PyObject_AS_GC(op);
PyObject_DEL(op);
1990-10-14 12:07:46 +00:00
}
1997-05-02 03:12:38 +00:00
static PyObject*
2000-07-09 06:03:25 +00:00
func_repr(PyFunctionObject *op)
{
char buf[140];
1997-05-02 03:12:38 +00:00
if (op->func_name == Py_None)
sprintf(buf, "<anonymous function at %p>", op);
else
sprintf(buf, "<function %.100s at %p>",
1997-05-02 03:12:38 +00:00
PyString_AsString(op->func_name),
op);
1997-05-02 03:12:38 +00:00
return PyString_FromString(buf);
}
static int
func_traverse(PyFunctionObject *f, visitproc visit, void *arg)
{
int err;
if (f->func_code) {
err = visit(f->func_code, arg);
if (err)
return err;
}
if (f->func_globals) {
err = visit(f->func_globals, arg);
if (err)
return err;
}
if (f->func_defaults) {
err = visit(f->func_defaults, arg);
if (err)
return err;
}
if (f->func_doc) {
err = visit(f->func_doc, arg);
if (err)
return err;
}
if (f->func_name) {
err = visit(f->func_name, arg);
if (err)
return err;
}
if (f->func_dict) {
err = visit(f->func_dict, arg);
if (err)
return err;
}
if (f->func_closure) {
err = visit(f->func_closure, arg);
if (err)
return err;
}
return 0;
}
1997-05-02 03:12:38 +00:00
PyTypeObject PyFunction_Type = {
PyObject_HEAD_INIT(&PyType_Type)
1990-10-14 12:07:46 +00:00
0,
"function",
sizeof(PyFunctionObject) + PyGC_HEAD_SIZE,
1990-10-14 12:07:46 +00:00
0,
(destructor)func_dealloc, /* tp_dealloc */
0, /* tp_print */
0, /* tp_getattr */
0, /* tp_setattr */
0, /* tp_compare */
(reprfunc)func_repr, /* tp_repr */
0, /* tp_as_number */
0, /* tp_as_sequence */
0, /* tp_as_mapping */
0, /* tp_hash */
0, /* tp_call */
0, /* tp_str */
(getattrofunc)func_getattro, /* tp_getattro */
(setattrofunc)func_setattro, /* tp_setattro */
0, /* tp_as_buffer */
Py_TPFLAGS_DEFAULT | Py_TPFLAGS_GC, /* tp_flags */
0, /* tp_doc */
(traverseproc)func_traverse, /* tp_traverse */
0, /* tp_clear */
0, /* tp_richcompare */
offsetof(PyFunctionObject, func_weakreflist), /* tp_weaklistoffset */
1990-10-14 12:07:46 +00:00
};