/* Function object implementation */

#include "Python.h"

#include "pycore_code.h" // _PyCode_VerifyStateless()

#include "pycore_dict.h" // _Py_INCREF_DICT()

#include "pycore_function.h" // _PyFunction_Vectorcall

#include "pycore_long.h" // _PyLong_GetOne()

#include "pycore_modsupport.h" // _PyArg_NoKeywords()

#include "pycore_object.h" // _PyObject_GC_UNTRACK()

#include "pycore_object_deferred.h" // _PyObject_SetDeferredRefcount()

#include "pycore_optimizer.h" // _Py_Executors_InvalidateDependency()

#include "pycore_pyerrors.h" // _PyErr_Occurred()

#include "pycore_setobject.h" // _PySet_NextEntry()

#include "pycore_stats.h"

#include "pycore_weakref.h" // FT_CLEAR_WEAKREFS()

static const char *

func_event_name(PyFunction_WatchEvent event) {

switch (event) {

#define CASE(op) \

case PyFunction_EVENT_##op: \

return "PyFunction_EVENT_" #op;

PY_FOREACH_FUNC_EVENT(CASE)

#undef CASE

}

Py_UNREACHABLE();

}

static void

notify_func_watchers(PyInterpreterState *interp, PyFunction_WatchEvent event,

PyFunctionObject *func, PyObject *new_value)

{

uint8_t bits = interp->active_func_watchers;

int i = 0;

while (bits) {

assert(i < FUNC_MAX_WATCHERS);

if (bits & 1) {

PyFunction_WatchCallback cb = interp->func_watchers[i];

// callback must be non-null if the watcher bit is set

assert(cb != NULL);

if (cb(event, func, new_value) < 0) {

PyErr_FormatUnraisable(

"Exception ignored in %s watcher callback for function %U at %p",

func_event_name(event), func->func_qualname, func);

}

}

i++;

bits >>= 1;

}

}

static inline void

handle_func_event(PyFunction_WatchEvent event, PyFunctionObject *func,

PyObject *new_value)

{

assert(Py_REFCNT(func) > 0);

PyInterpreterState *interp = _PyInterpreterState_GET();

assert(interp->_initialized);

if (interp->active_func_watchers) {

notify_func_watchers(interp, event, func, new_value);

}

switch (event) {

case PyFunction_EVENT_MODIFY_CODE:

case PyFunction_EVENT_MODIFY_DEFAULTS:

case PyFunction_EVENT_MODIFY_KWDEFAULTS:

case PyFunction_EVENT_MODIFY_QUALNAME:

#if _Py_TIER2

// Note: we only invalidate JIT code if a function version changes.

// Not when the function is deallocated.

// Function deallocation occurs frequently (think: lambdas),

// so we want to minimize dependency invalidation there.

_Py_Executors_InvalidateDependency(interp, func, 1);

#endif

RARE_EVENT_INTERP_INC(interp, func_modification);

break;

default:

break;

}

}

int

PyFunction_AddWatcher(PyFunction_WatchCallback callback)

{

PyInterpreterState *interp = _PyInterpreterState_GET();

assert(interp->_initialized);

for (int i = 0; i < FUNC_MAX_WATCHERS; i++) {

if (interp->func_watchers[i] == NULL) {

interp->func_watchers[i] = callback;

interp->active_func_watchers |= (1 << i);

return i;

}

}

PyErr_SetString(PyExc_RuntimeError, "no more func watcher IDs available");

return -1;

}

int

PyFunction_ClearWatcher(int watcher_id)

{

PyInterpreterState *interp = _PyInterpreterState_GET();

if (watcher_id < 0 || watcher_id >= FUNC_MAX_WATCHERS) {

PyErr_Format(PyExc_ValueError, "invalid func watcher ID %d",

watcher_id);

return -1;

}

if (!interp->func_watchers[watcher_id]) {

PyErr_Format(PyExc_ValueError, "no func watcher set for ID %d",

watcher_id);

return -1;

}

interp->func_watchers[watcher_id] = NULL;

interp->active_func_watchers &= ~(1 << watcher_id);

return 0;

}

PyFunctionObject *

_PyFunction_FromConstructor(PyFrameConstructor *constr)

{

PyObject *module;

if (PyDict_GetItemRef(constr->fc_globals, &_Py_ID(__name__), &module) < 0) {

return NULL;

}

PyFunctionObject *op = PyObject_GC_New(PyFunctionObject, &PyFunction_Type);

if (op == NULL) {

Py_XDECREF(module);

return NULL;

}

_Py_INCREF_DICT(constr->fc_globals);

op->func_globals = constr->fc_globals;

_Py_INCREF_BUILTINS(constr->fc_builtins);

op->func_builtins = constr->fc_builtins;

op->func_name = Py_NewRef(constr->fc_name);

op->func_qualname = Py_NewRef(constr->fc_qualname);

_Py_INCREF_CODE((PyCodeObject *)constr->fc_code);

op->func_code = constr->fc_code;

op->func_defaults = Py_XNewRef(constr->fc_defaults);

op->func_kwdefaults = Py_XNewRef(constr->fc_kwdefaults);

op->func_closure = Py_XNewRef(constr->fc_closure);

op->func_doc = Py_NewRef(Py_None);

op->func_dict = NULL;

op->func_weakreflist = NULL;

op->func_module = module;

op->func_annotations = NULL;

op->func_annotate = NULL;

op->func_typeparams = NULL;

op->vectorcall = _PyFunction_Vectorcall;

op->func_version = FUNC_VERSION_UNSET;

// NOTE: functions created via FrameConstructor do not use deferred

// reference counting because they are typically not part of cycles

// nor accessed by multiple threads.

_PyObject_GC_TRACK(op);

handle_func_event(PyFunction_EVENT_CREATE, op, NULL);

return op;

}

PyObject *

PyFunction_NewWithQualName(PyObject *code, PyObject *globals, PyObject *qualname)

{

assert(globals != NULL);

assert(PyAnyDict_Check(globals));

_Py_INCREF_DICT(globals);

PyCodeObject *code_obj = (PyCodeObject *)code;

_Py_INCREF_CODE(code_obj);

assert(code_obj->co_name != NULL);

PyObject *name = Py_NewRef(code_obj->co_name);

if (!qualname) {

qualname = code_obj->co_qualname;

}

assert(qualname != NULL);

Py_INCREF(qualname);

PyObject *consts = code_obj->co_consts;

assert(PyTuple_Check(consts));

PyObject *doc;

if (code_obj->co_flags & CO_HAS_DOCSTRING) {

assert(PyTuple_Size(consts) >= 1);

doc = PyTuple_GetItem(consts, 0);

if (!PyUnicode_Check(doc)) {

doc = Py_None;

}

}

else {

doc = Py_None;

}

Py_INCREF(doc);

// __module__: Use globals['__name__'] if it exists, or NULL.

PyObject *module;

PyObject *builtins = NULL;

if (PyDict_GetItemRef(globals, &_Py_ID(__name__), &module) < 0) {

goto error;

}

builtins = _PyDict_LoadBuiltinsFromGlobals(globals);

if (builtins == NULL) {

goto error;

}

PyFunctionObject *op = PyObject_GC_New(PyFunctionObject, &PyFunction_Type);

if (op == NULL) {

goto error;

}

/* Note: No failures from this point on, since func_dealloc() does not

expect a partially-created object. */

op->func_globals = globals;

op->func_builtins = builtins;

op->func_name = name;

op->func_qualname = qualname;

op->func_code = (PyObject*)code_obj;

op->func_defaults = NULL; // No default positional arguments

op->func_kwdefaults = NULL; // No default keyword arguments

op->func_closure = NULL;

op->func_doc = doc;

op->func_dict = NULL;

op->func_weakreflist = NULL;

op->func_module = module;

op->func_annotations = NULL;

op->func_annotate = NULL;

op->func_typeparams = NULL;

op->vectorcall = _PyFunction_Vectorcall;

op->func_version = FUNC_VERSION_UNSET;

if (((code_obj->co_flags & CO_NESTED) == 0) ||

(code_obj->co_flags & CO_METHOD)) {

// Use deferred reference counting for top-level functions, but not

// nested functions because they are more likely to capture variables,

// which makes prompt deallocation more important.

//

// Nested methods (functions defined in class scope) are also deferred,

// since they will likely be cleaned up by GC anyway.

_PyObject_SetDeferredRefcount((PyObject *)op);

}

_PyObject_GC_TRACK(op);

handle_func_event(PyFunction_EVENT_CREATE, op, NULL);

return (PyObject *)op;

error:

Py_DECREF(globals);

Py_DECREF(code_obj);

Py_DECREF(name);

Py_DECREF(qualname);

Py_DECREF(doc);

Py_XDECREF(module);

Py_XDECREF(builtins);

return NULL;

}

/*

(This is purely internal documentation. There are no public APIs here.)

Function (and code) versions

----------------------------

The Tier 1 specializer generates CALL variants that can be invalidated

by changes to critical function attributes:

- __code__

- __defaults__

- __kwdefaults__

- __closure__

For this purpose function objects have a 32-bit func_version member

that the specializer writes to the specialized instruction's inline

cache and which is checked by a guard on the specialized instructions.

The MAKE_FUNCTION bytecode sets func_version from the code object's

co_version field. The latter is initialized from a counter in the

interpreter state (interp->func_state.next_version) and never changes.

When this counter overflows, it remains zero and the specializer loses

the ability to specialize calls to new functions.

The func_version is reset to zero when any of the critical attributes

is modified; after this point the specializer will no longer specialize

calls to this function, and the guard will always fail.

The function and code version cache

-----------------------------------

The Tier 2 optimizer now has a problem, since it needs to find the

function and code objects given only the version number from the inline

cache. Our solution is to maintain a cache mapping version numbers to

function and code objects. To limit the cache size we could hash

the version number, but for now we simply use it modulo the table size.

There are some corner cases (e.g. generator expressions) where we will

be unable to find the function object in the cache but we can still

find the code object. For this reason the cache stores both the

function object and the code object.

The cache doesn't contain strong references; cache entries are

invalidated whenever the function or code object is deallocated.

Invariants

----------

These should hold at any time except when one of the cache-mutating

functions is running.

- For any slot s at index i:

- s->func == NULL or s->func->func_version % FUNC_VERSION_CACHE_SIZE == i

- s->code == NULL or s->code->co_version % FUNC_VERSION_CACHE_SIZE == i

if s->func != NULL, then s->func->func_code == s->code

*/

#ifndef Py_GIL_DISABLED

static inline struct _func_version_cache_item *

get_cache_item(PyInterpreterState *interp, uint32_t version)

{

return interp->func_state.func_version_cache +

(version % FUNC_VERSION_CACHE_SIZE);

}

#endif

void

_PyFunction_SetVersion(PyFunctionObject *func, uint32_t version)

{

assert(func->func_version == FUNC_VERSION_UNSET);

assert(version >= FUNC_VERSION_FIRST_VALID);

// This should only be called from MAKE_FUNCTION. No code is specialized

// based on the version, so we do not need to stop the world to set it.

func->func_version = version;

#ifndef Py_GIL_DISABLED

PyInterpreterState *interp = _PyInterpreterState_GET();

struct _func_version_cache_item *slot = get_cache_item(interp, version);

slot->func = func;

slot->code = func->func_code;

#endif

}

static void

func_clear_version(PyInterpreterState *interp, PyFunctionObject *func)

{

if (func->func_version < FUNC_VERSION_FIRST_VALID) {

// Version was never set or has already been cleared.

return;

}

#ifndef Py_GIL_DISABLED

struct _func_version_cache_item *slot =

get_cache_item(interp, func->func_version);

if (slot->func == func) {

slot->func = NULL;

// Leave slot->code alone, there may be use for it.

}

#endif

func->func_version = FUNC_VERSION_CLEARED;

}

void

_PyFunction_ClearCodeByVersion(uint32_t version)

{

#ifndef Py_GIL_DISABLED

PyInterpreterState *interp = _PyInterpreterState_GET();

struct _func_version_cache_item *slot = get_cache_item(interp, version);

if (slot->code) {

assert(PyCode_Check(slot->code));

PyCodeObject *code = (PyCodeObject *)slot->code;

if (code->co_version == version) {

slot->code = NULL;

slot->func = NULL;

}

}

#endif

}

uint32_t

_PyFunction_GetVersionForCurrentState(PyFunctionObject *func)

{

return func->func_version;

}

PyObject *

PyFunction_New(PyObject *code, PyObject *globals)

{

return PyFunction_NewWithQualName(code, globals, NULL);

}

PyObject *

PyFunction_GetCode(PyObject *op)

{

if (!PyFunction_Check(op)) {

PyErr_BadInternalCall();

return NULL;

}

return ((PyFunctionObject *) op) -> func_code;

}

PyObject *

PyFunction_GetGlobals(PyObject *op)

{

if (!PyFunction_Check(op)) {

PyErr_BadInternalCall();

return NULL;

}

return ((PyFunctionObject *) op) -> func_globals;

}

PyObject *

PyFunction_GetModule(PyObject *op)

{

if (!PyFunction_Check(op)) {

PyErr_BadInternalCall();

return NULL;

}

return ((PyFunctionObject *) op) -> func_module;

}

PyObject *

PyFunction_GetDefaults(PyObject *op)

{

if (!PyFunction_Check(op)) {

PyErr_BadInternalCall();

return NULL;

}

return ((PyFunctionObject *) op) -> func_defaults;

}

int

PyFunction_SetDefaults(PyObject *op, PyObject *defaults)

{

if (!PyFunction_Check(op)) {

PyErr_BadInternalCall();

return -1;

}

if (defaults == Py_None)

defaults = NULL;

else if (defaults && PyTuple_Check(defaults)) {

Py_INCREF(defaults);

}

else {

PyErr_SetString(PyExc_SystemError, "non-tuple default args");

return -1;

}

PyFunctionObject *func = (PyFunctionObject *)op;

handle_func_event(PyFunction_EVENT_MODIFY_DEFAULTS, func, defaults);

PyInterpreterState *interp = _PyInterpreterState_GET();

_PyEval_StopTheWorld(interp);

func_clear_version(interp, func);

PyObject *old_defaults = func->func_defaults;

func->func_defaults = defaults;

_PyEval_StartTheWorld(interp);

Py_XDECREF(old_defaults);

return 0;

}

void

PyFunction_SetVectorcall(PyFunctionObject *func, vectorcallfunc vectorcall)

{

assert(func != NULL);

PyInterpreterState *interp = _PyInterpreterState_GET();

_PyEval_StopTheWorld(interp);

func_clear_version(interp, func);

func->vectorcall = vectorcall;

_PyEval_StartTheWorld(interp);

}

PyObject *

PyFunction_GetKwDefaults(PyObject *op)

{

if (!PyFunction_Check(op)) {

PyErr_BadInternalCall();

return NULL;

}

return ((PyFunctionObject *) op) -> func_kwdefaults;

}

int

PyFunction_SetKwDefaults(PyObject *op, PyObject *defaults)

{

if (!PyFunction_Check(op)) {

PyErr_BadInternalCall();

return -1;

}

if (defaults == Py_None)

defaults = NULL;

else if (defaults && PyDict_Check(defaults)) {

Py_INCREF(defaults);

}

else {

PyErr_SetString(PyExc_SystemError,

"non-dict keyword only default args");

return -1;

}

PyFunctionObject *func = (PyFunctionObject *)op;

handle_func_event(PyFunction_EVENT_MODIFY_KWDEFAULTS, func, defaults);

PyInterpreterState *interp = _PyInterpreterState_GET();

_PyEval_StopTheWorld(interp);

func_clear_version(interp, func);

PyObject *old_kwdefaults = func->func_kwdefaults;

func->func_kwdefaults = defaults;

_PyEval_StartTheWorld(interp);

Py_XDECREF(old_kwdefaults);

return 0;

}

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_INCREF(closure);

}

else {

PyErr_Format(PyExc_SystemError,

"expected tuple for closure, got '%.100s'",

Py_TYPE(closure)->tp_name);

return -1;

}

PyFunctionObject *func = (PyFunctionObject *)op;

PyInterpreterState *interp = _PyInterpreterState_GET();

_PyEval_StopTheWorld(interp);

func_clear_version(interp, func);

PyObject *old_closure = func->func_closure;

func->func_closure = closure;

_PyEval_StartTheWorld(interp);

Py_XDECREF(old_closure);

return 0;

}

static PyObject *

func_get_annotation_dict(PyFunctionObject *op)

{

if (op->func_annotations == NULL) {

if (op->func_annotate == NULL || !PyCallable_Check(op->func_annotate)) {

Py_RETURN_NONE;

}

PyObject *one = _PyLong_GetOne();

PyObject *ann_dict = _PyObject_CallOneArg(op->func_annotate, one);

if (ann_dict == NULL) {

return NULL;

}

if (!PyDict_Check(ann_dict)) {

PyErr_Format(PyExc_TypeError,

"__annotate__() must return a dict, not %T",

ann_dict);

Py_DECREF(ann_dict);

return NULL;

}

Py_XSETREF(op->func_annotations, ann_dict);

return ann_dict;

}

if (PyTuple_CheckExact(op->func_annotations)) {

PyObject *ann_tuple = op->func_annotations;

PyObject *ann_dict = PyDict_New();

if (ann_dict == NULL) {

return NULL;

}

assert(PyTuple_GET_SIZE(ann_tuple) % 2 == 0);

for (Py_ssize_t i = 0; i < PyTuple_GET_SIZE(ann_tuple); i += 2) {

int err = PyDict_SetItem(ann_dict,

PyTuple_GET_ITEM(ann_tuple, i),

PyTuple_GET_ITEM(ann_tuple, i + 1));

if (err < 0) {

Py_DECREF(ann_dict);

return NULL;

}

}

Py_SETREF(op->func_annotations, ann_dict);

}

assert(PyDict_Check(op->func_annotations));

return op->func_annotations;

}

PyObject *

PyFunction_GetAnnotations(PyObject *op)

{

if (!PyFunction_Check(op)) {

PyErr_BadInternalCall();

return NULL;

}

return func_get_annotation_dict((PyFunctionObject *)op);

}

int

PyFunction_SetAnnotations(PyObject *op, PyObject *annotations)

{

if (!PyFunction_Check(op)) {

PyErr_BadInternalCall();

return -1;

}

if (annotations == Py_None)

annotations = NULL;

else if (annotations && PyDict_Check(annotations)) {

Py_INCREF(annotations);

}

else {

PyErr_SetString(PyExc_SystemError,

"non-dict annotations");

return -1;

}

PyFunctionObject *func = (PyFunctionObject *)op;

PyInterpreterState *interp = _PyInterpreterState_GET();

_PyEval_StopTheWorld(interp);

PyObject *old_annotations = func->func_annotations;

func->func_annotations = annotations;

PyObject *old_annotate = func->func_annotate;

func->func_annotate = NULL;

_PyEval_StartTheWorld(interp);

Py_XDECREF(old_annotations);

Py_XDECREF(old_annotate);

return 0;

}

/* Methods */

#define OFF(x) offsetof(PyFunctionObject, x)

static PyMemberDef func_memberlist[] = {

{"__closure__", _Py_T_OBJECT, OFF(func_closure), Py_READONLY},

{"__globals__", _Py_T_OBJECT, OFF(func_globals), Py_READONLY},

{"__builtins__", _Py_T_OBJECT, OFF(func_builtins), Py_READONLY},

{NULL} /* Sentinel */

};

/*[clinic input]

class function "PyFunctionObject *" "&PyFunction_Type"

[clinic start generated code]*/

/*[clinic end generated code: output=da39a3ee5e6b4b0d input=70af9c90aa2e71b0]*/

#include "clinic/funcobject.c.h"

static PyObject *

func_get_code(PyObject *self, void *Py_UNUSED(ignored))

{

PyFunctionObject *op = _PyFunction_CAST(self);

if (PySys_Audit("object.__getattr__", "Os", op, "__code__") < 0) {

return NULL;

}

return Py_NewRef(op->func_code);

}

static int

func_set_code(PyObject *self, PyObject *value, void *Py_UNUSED(ignored))

{

PyFunctionObject *op = _PyFunction_CAST(self);

/* 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,

"__code__ must be set to a code object");

return -1;

}

if (PySys_Audit("object.__setattr__", "OsO",

op, "__code__", value) < 0) {

return -1;

}

int nfree = ((PyCodeObject *)value)->co_nfreevars;

Py_ssize_t nclosure = (op->func_closure == NULL ? 0 :

PyTuple_GET_SIZE(op->func_closure));

if (nclosure != nfree) {

PyErr_Format(PyExc_ValueError,

"%U() requires a code object with %zd free vars,"

" not %d",

op->func_name,

nclosure, nfree);

return -1;

}

PyObject *func_code = PyFunction_GET_CODE(op);

int old_flags = ((PyCodeObject *)func_code)->co_flags;

int new_flags = ((PyCodeObject *)value)->co_flags;

int mask = CO_GENERATOR | CO_COROUTINE | CO_ASYNC_GENERATOR;

if ((old_flags & mask) != (new_flags & mask)) {

if (PyErr_Warn(PyExc_DeprecationWarning,

"Assigning a code object of non-matching type is deprecated "

"(e.g., from a generator to a plain function)") < 0)

{

return -1;

}

}

handle_func_event(PyFunction_EVENT_MODIFY_CODE, op, value);

PyInterpreterState *interp = _PyInterpreterState_GET();

_PyEval_StopTheWorld(interp);

func_clear_version(interp, op);

PyObject *old_code = op->func_code;

op->func_code = Py_NewRef(value);

_PyEval_StartTheWorld(interp);

Py_XDECREF(old_code);

return 0;

}

static PyObject *

func_get_name(PyObject *self, void *Py_UNUSED(ignored))

{

PyFunctionObject *op = _PyFunction_CAST(self);

return Py_NewRef(op->func_name);

}

static int

func_set_name(PyObject *self, PyObject *value, void *Py_UNUSED(ignored))

{

PyFunctionObject *op = _PyFunction_CAST(self);

/* Not legal to del f.func_name or to set it to anything

* other than a string object. */

if (value == NULL || !PyUnicode_Check(value)) {

PyErr_SetString(PyExc_TypeError,

"__name__ must be set to a string object");

return -1;

}

PyInterpreterState *interp = _PyInterpreterState_GET();

_PyEval_StopTheWorld(interp);

PyObject *old_name = op->func_name;

op->func_name = Py_NewRef(value);

_PyEval_StartTheWorld(interp);

Py_XDECREF(old_name);

return 0;

}

static PyObject *

func_get_qualname(PyObject *self, void *Py_UNUSED(ignored))

{

PyFunctionObject *op = _PyFunction_CAST(self);

return Py_NewRef(op->func_qualname);

}

static int

func_set_qualname(PyObject *self, PyObject *value, void *Py_UNUSED(ignored))

{

PyFunctionObject *op = _PyFunction_CAST(self);

/* Not legal to del f.__qualname__ or to set it to anything

* other than a string object. */

if (value == NULL || !PyUnicode_Check(value)) {

PyErr_SetString(PyExc_TypeError,

"__qualname__ must be set to a string object");

return -1;

}

handle_func_event(PyFunction_EVENT_MODIFY_QUALNAME, (PyFunctionObject *) op, value);

PyInterpreterState *interp = _PyInterpreterState_GET();

_PyEval_StopTheWorld(interp);

PyObject *old_qualname = op->func_qualname;

op->func_qualname = Py_NewRef(value);

_PyEval_StartTheWorld(interp);

Py_XDECREF(old_qualname);

return 0;

}

static PyObject *

func_get_doc(PyObject *self, void *Py_UNUSED(ignored))

{

PyFunctionObject *op = _PyFunction_CAST(self);

PyObject *doc = op->func_doc;

if (doc == NULL) {

doc = Py_None;

}

return Py_NewRef(doc);

}

static int

func_set_doc(PyObject *self, PyObject *value, void *Py_UNUSED(ignored))

{

/* Legal to del f.__doc__ or to set it to any object. */

PyFunctionObject *op = _PyFunction_CAST(self);

PyInterpreterState *interp = _PyInterpreterState_GET();

_PyEval_StopTheWorld(interp);

PyObject *old_doc = op->func_doc;

op->func_doc = Py_XNewRef(value);

_PyEval_StartTheWorld(interp);

Py_XDECREF(old_doc);

return 0;

}

static PyObject *

func_get_module(PyObject *self, void *Py_UNUSED(ignored))

{

PyFunctionObject *op = _PyFunction_CAST(self);

PyObject *module = op->func_module;

if (module == NULL) {

module = Py_None;

}

return Py_NewRef(module);

}

static int

func_set_module(PyObject *self, PyObject *value, void *Py_UNUSED(ignored))

{

/* Legal to del f.__module__ or to set it to any object. */

PyFunctionObject *op = _PyFunction_CAST(self);

PyInterpreterState *interp = _PyInterpreterState_GET();

_PyEval_StopTheWorld(interp);

PyObject *old_module = op->func_module;

op->func_module = Py_XNewRef(value);

_PyEval_StartTheWorld(interp);

Py_XDECREF(old_module);

return 0;

}

static PyObject *

func_get_defaults(PyObject *self, void *Py_UNUSED(ignored))

{

PyFunctionObject *op = _PyFunction_CAST(self);

if (PySys_Audit("object.__getattr__", "Os", op, "__defaults__") < 0) {

return NULL;

}

if (op->func_defaults == NULL) {

Py_RETURN_NONE;

}

return Py_NewRef(op->func_defaults);

}

static int

func_set_defaults(PyObject *self, PyObject *value, void *Py_UNUSED(ignored))

{

/* Legal to del f.func_defaults.

* Can only set func_defaults to NULL or a tuple. */

PyFunctionObject *op = _PyFunction_CAST(self);

if (value == Py_None)

value = NULL;

if (value != NULL && !PyTuple_Check(value)) {

PyErr_SetString(PyExc_TypeError,

"__defaults__ must be set to a tuple object");

return -1;

}

if (value) {

if (PySys_Audit("object.__setattr__", "OsO",

op, "__defaults__", value) < 0) {

return -1;

}

} else if (PySys_Audit("object.__delattr__", "Os",

op, "__defaults__") < 0) {

return -1;

}

handle_func_event(PyFunction_EVENT_MODIFY_DEFAULTS, op, value);

PyInterpreterState *interp = _PyInterpreterState_GET();

_PyEval_StopTheWorld(interp);

func_clear_version(interp, op);

PyObject *old_defaults = op->func_defaults;

op->func_defaults = Py_XNewRef(value);

_PyEval_StartTheWorld(interp);

Py_XDECREF(old_defaults);

return 0;

}

static PyObject *

func_get_kwdefaults(PyObject *self, void *Py_UNUSED(ignored))

{

PyFunctionObject *op = _PyFunction_CAST(self);

if (PySys_Audit("object.__getattr__", "Os",

op, "__kwdefaults__") < 0) {

return NULL;

}

if (op->func_kwdefaults == NULL) {

Py_RETURN_NONE;

}

return Py_NewRef(op->func_kwdefaults);

}

static int

func_set_kwdefaults(PyObject *self, PyObject *value, void *Py_UNUSED(ignored))

{

PyFunctionObject *op = _PyFunction_CAST(self);

if (value == Py_None)

value = NULL;

/* Legal to del f.func_kwdefaults.

* Can only set func_kwdefaults to NULL or a dict. */

if (value != NULL && !PyDict_Check(value)) {

PyErr_SetString(PyExc_TypeError,

"__kwdefaults__ must be set to a dict object");

return -1;

}

if (value) {

if (PySys_Audit("object.__setattr__", "OsO",

op, "__kwdefaults__", value) < 0) {

return -1;

}

} else if (PySys_Audit("object.__delattr__", "Os",

op, "__kwdefaults__") < 0) {

return -1;

}

handle_func_event(PyFunction_EVENT_MODIFY_KWDEFAULTS, op, value);

PyInterpreterState *interp = _PyInterpreterState_GET();

_PyEval_StopTheWorld(interp);

func_clear_version(interp, op);

PyObject *old_kwdefaults = op->func_kwdefaults;

op->func_kwdefaults = Py_XNewRef(value);

_PyEval_StartTheWorld(interp);

Py_XDECREF(old_kwdefaults);

return 0;

}

/*[clinic input]

@critical_section

@getter

function.__annotate__

Get the code object for a function.

[clinic start generated code]*/

static PyObject *

function___annotate___get_impl(PyFunctionObject *self)

/*[clinic end generated code: output=5ec7219ff2bda9e6 input=7f3db11e3c3329f3]*/

{

if (self->func_annotate == NULL) {

Py_RETURN_NONE;

}

return Py_NewRef(self->func_annotate);

}

/*[clinic input]

@critical_section

@setter

function.__annotate__

[clinic start generated code]*/

static int

function___annotate___set_impl(PyFunctionObject *self, PyObject *value)

/*[clinic end generated code: output=05b7dfc07ada66cd input=eb6225e358d97448]*/

{

if (value == NULL) {

PyErr_SetString(PyExc_TypeError,

"__annotate__ cannot be deleted");

return -1;

}

if (Py_IsNone(value)) {

Py_XSETREF(self->func_annotate, Py_NewRef(value));

return 0;

}

else if (PyCallable_Check(value)) {

Py_XSETREF(self->func_annotate, Py_NewRef(value));

Py_CLEAR(self->func_annotations);

return 0;

}

else {

PyErr_SetString(PyExc_TypeError,

"__annotate__ must be callable or None");

return -1;

}

}

/*[clinic input]

@critical_section

@getter

function.__annotations__

Dict of annotations in a function object.

[clinic start generated code]*/

static PyObject *

function___annotations___get_impl(PyFunctionObject *self)

/*[clinic end generated code: output=a4cf4c884c934cbb input=92643d7186c1ad0c]*/

{

PyObject *d = NULL;

if (self->func_annotations == NULL &&

(self->func_annotate == NULL || !PyCallable_Check(self->func_annotate))) {

self->func_annotations = PyDict_New();

if (self->func_annotations == NULL)

return NULL;

}

d = func_get_annotation_dict(self);

return Py_XNewRef(d);

}

/*[clinic input]

@critical_section

@setter

function.__annotations__

[clinic start generated code]*/

static int

function___annotations___set_impl(PyFunctionObject *self, PyObject *value)

/*[clinic end generated code: output=a61795d4a95eede4 input=5302641f686f0463]*/

{

if (value == Py_None)

value = NULL;

/* Legal to del f.func_annotations.

* Can only set func_annotations to NULL (through C api)

* or a dict. */

if (value != NULL && !PyDict_Check(value)) {

PyErr_SetString(PyExc_TypeError,

"__annotations__ must be set to a dict object");

return -1;

}

Py_XSETREF(self->func_annotations, Py_XNewRef(value));