mirror of
				https://github.com/python/cpython.git
				synced 2025-11-04 07:31:38 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			900 lines
		
	
	
	
		
			26 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			900 lines
		
	
	
	
		
			26 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
#include "Python.h"
 | 
						|
#include "frameobject.h"
 | 
						|
 | 
						|
#define MODULE_NAME "_warnings"
 | 
						|
#define DEFAULT_ACTION_NAME "default_action"
 | 
						|
 | 
						|
PyDoc_STRVAR(warnings__doc__,
 | 
						|
MODULE_NAME " provides basic warning filtering support.\n"
 | 
						|
"It is a helper module to speed up interpreter start-up.");
 | 
						|
 | 
						|
/* Both 'filters' and 'onceregistry' can be set in warnings.py;
 | 
						|
   get_warnings_attr() will reset these variables accordingly. */
 | 
						|
static PyObject *_filters;  /* List */
 | 
						|
static PyObject *_once_registry;  /* Dict */
 | 
						|
 | 
						|
 | 
						|
static int
 | 
						|
check_matched(PyObject *obj, PyObject *arg)
 | 
						|
{
 | 
						|
    PyObject *result;
 | 
						|
    int rc;
 | 
						|
 | 
						|
    if (obj == Py_None)
 | 
						|
        return 1;
 | 
						|
    result = PyObject_CallMethod(obj, "match", "O", arg);
 | 
						|
    if (result == NULL)
 | 
						|
        return -1;
 | 
						|
 | 
						|
    rc = PyObject_IsTrue(result);
 | 
						|
    Py_DECREF(result);
 | 
						|
    return rc;
 | 
						|
}
 | 
						|
 | 
						|
/*
 | 
						|
   Returns a new reference.
 | 
						|
   A NULL return value can mean false or an error.
 | 
						|
*/
 | 
						|
static PyObject *
 | 
						|
get_warnings_attr(const char *attr)
 | 
						|
{
 | 
						|
    static PyObject *warnings_str = NULL;
 | 
						|
    PyObject *all_modules;
 | 
						|
    PyObject *warnings_module;
 | 
						|
    int result;
 | 
						|
 | 
						|
    if (warnings_str == NULL) {
 | 
						|
        warnings_str = PyBytes_InternFromString("warnings");
 | 
						|
        if (warnings_str == NULL)
 | 
						|
            return NULL;
 | 
						|
    }
 | 
						|
 | 
						|
    all_modules = PyImport_GetModuleDict();
 | 
						|
    result = PyDict_Contains(all_modules, warnings_str);
 | 
						|
    if (result == -1 || result == 0)
 | 
						|
        return NULL;
 | 
						|
 | 
						|
    warnings_module = PyDict_GetItem(all_modules, warnings_str);
 | 
						|
    if (!PyObject_HasAttrString(warnings_module, attr))
 | 
						|
            return NULL;
 | 
						|
    return PyObject_GetAttrString(warnings_module, attr);
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
static PyObject *
 | 
						|
get_once_registry(void)
 | 
						|
{
 | 
						|
    PyObject *registry;
 | 
						|
 | 
						|
    registry = get_warnings_attr("onceregistry");
 | 
						|
    if (registry == NULL) {
 | 
						|
        if (PyErr_Occurred())
 | 
						|
            return NULL;
 | 
						|
        return _once_registry;
 | 
						|
    }
 | 
						|
    Py_DECREF(_once_registry);
 | 
						|
    _once_registry = registry;
 | 
						|
    return registry;
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/* The item is a borrowed reference. */
 | 
						|
static const char *
 | 
						|
get_filter(PyObject *category, PyObject *text, Py_ssize_t lineno,
 | 
						|
           PyObject *module, PyObject **item)
 | 
						|
{
 | 
						|
    PyObject *action, *m, *d;
 | 
						|
    Py_ssize_t i;
 | 
						|
    PyObject *warnings_filters;
 | 
						|
 | 
						|
    warnings_filters = get_warnings_attr("filters");
 | 
						|
    if (warnings_filters == NULL) {
 | 
						|
        if (PyErr_Occurred())
 | 
						|
            return NULL;
 | 
						|
    }
 | 
						|
    else {
 | 
						|
        Py_DECREF(_filters);
 | 
						|
        _filters = warnings_filters;
 | 
						|
    }
 | 
						|
 | 
						|
    if (!PyList_Check(_filters)) {
 | 
						|
        PyErr_SetString(PyExc_ValueError,
 | 
						|
                        MODULE_NAME ".filters must be a list");
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
 | 
						|
    /* _filters could change while we are iterating over it. */
 | 
						|
    for (i = 0; i < PyList_GET_SIZE(_filters); i++) {
 | 
						|
        PyObject *tmp_item, *action, *msg, *cat, *mod, *ln_obj;
 | 
						|
        Py_ssize_t ln;
 | 
						|
        int is_subclass, good_msg, good_mod;
 | 
						|
 | 
						|
        tmp_item = *item = PyList_GET_ITEM(_filters, i);
 | 
						|
        if (PyTuple_Size(tmp_item) != 5) {
 | 
						|
            PyErr_Format(PyExc_ValueError,
 | 
						|
                         MODULE_NAME ".filters item %zd isn't a 5-tuple", i);
 | 
						|
            return NULL;
 | 
						|
        }
 | 
						|
 | 
						|
        /* Python code: action, msg, cat, mod, ln = item */
 | 
						|
        action = PyTuple_GET_ITEM(tmp_item, 0);
 | 
						|
        msg = PyTuple_GET_ITEM(tmp_item, 1);
 | 
						|
        cat = PyTuple_GET_ITEM(tmp_item, 2);
 | 
						|
        mod = PyTuple_GET_ITEM(tmp_item, 3);
 | 
						|
        ln_obj = PyTuple_GET_ITEM(tmp_item, 4);
 | 
						|
 | 
						|
        good_msg = check_matched(msg, text);
 | 
						|
        good_mod = check_matched(mod, module);
 | 
						|
        is_subclass = PyObject_IsSubclass(category, cat);
 | 
						|
        ln = PyInt_AsSsize_t(ln_obj);
 | 
						|
        if (good_msg == -1 || good_mod == -1 || is_subclass == -1 ||
 | 
						|
            (ln == -1 && PyErr_Occurred()))
 | 
						|
            return NULL;
 | 
						|
 | 
						|
        if (good_msg && is_subclass && good_mod && (ln == 0 || lineno == ln))
 | 
						|
            return PyBytes_AsString(action);
 | 
						|
    }
 | 
						|
 | 
						|
    m = PyImport_ImportModule(MODULE_NAME);
 | 
						|
    if (m == NULL)
 | 
						|
        return NULL;
 | 
						|
    d = PyModule_GetDict(m);
 | 
						|
    Py_DECREF(m);
 | 
						|
    if (d == NULL)
 | 
						|
        return NULL;
 | 
						|
    action = PyDict_GetItemString(d, DEFAULT_ACTION_NAME);
 | 
						|
    if (action != NULL)
 | 
						|
        return PyBytes_AsString(action);
 | 
						|
 | 
						|
    PyErr_SetString(PyExc_ValueError,
 | 
						|
                    MODULE_NAME "." DEFAULT_ACTION_NAME " not found");
 | 
						|
    return NULL;
 | 
						|
}
 | 
						|
 | 
						|
static int
 | 
						|
already_warned(PyObject *registry, PyObject *key, int should_set)
 | 
						|
{
 | 
						|
    PyObject *already_warned;
 | 
						|
 | 
						|
    if (key == NULL)
 | 
						|
        return -1;
 | 
						|
 | 
						|
    already_warned = PyDict_GetItem(registry, key);
 | 
						|
    if (already_warned != NULL) {
 | 
						|
        int rc = PyObject_IsTrue(already_warned);
 | 
						|
        if (rc != 0)
 | 
						|
            return rc;
 | 
						|
    }
 | 
						|
 | 
						|
    /* This warning wasn't found in the registry, set it. */
 | 
						|
    if (should_set)
 | 
						|
        return PyDict_SetItem(registry, key, Py_True);
 | 
						|
    return 0;
 | 
						|
}
 | 
						|
 | 
						|
/* New reference. */
 | 
						|
static PyObject *
 | 
						|
normalize_module(PyObject *filename)
 | 
						|
{
 | 
						|
    PyObject *module;
 | 
						|
    const char *mod_str;
 | 
						|
    Py_ssize_t len;
 | 
						|
 | 
						|
    int rc = PyObject_IsTrue(filename);
 | 
						|
    if (rc == -1)
 | 
						|
        return NULL;
 | 
						|
    else if (rc == 0)
 | 
						|
        return PyBytes_FromString("<unknown>");
 | 
						|
 | 
						|
    mod_str = PyBytes_AsString(filename);
 | 
						|
    if (mod_str == NULL)
 | 
						|
	    return NULL;
 | 
						|
    len = PyBytes_Size(filename);
 | 
						|
    if (len < 0)
 | 
						|
        return NULL;
 | 
						|
    if (len >= 3 &&
 | 
						|
	strncmp(mod_str + (len - 3), ".py", 3) == 0) {
 | 
						|
        module = PyBytes_FromStringAndSize(mod_str, len-3);
 | 
						|
    }
 | 
						|
    else {
 | 
						|
        module = filename;
 | 
						|
        Py_INCREF(module);
 | 
						|
    }
 | 
						|
    return module;
 | 
						|
}
 | 
						|
 | 
						|
static int
 | 
						|
update_registry(PyObject *registry, PyObject *text, PyObject *category,
 | 
						|
                int add_zero)
 | 
						|
{
 | 
						|
    PyObject *altkey, *zero = NULL;
 | 
						|
    int rc;
 | 
						|
 | 
						|
    if (add_zero) {
 | 
						|
        zero = PyInt_FromLong(0);
 | 
						|
        if (zero == NULL)
 | 
						|
            return -1;
 | 
						|
        altkey = PyTuple_Pack(3, text, category, zero);
 | 
						|
    }
 | 
						|
    else
 | 
						|
        altkey = PyTuple_Pack(2, text, category);
 | 
						|
 | 
						|
    rc = already_warned(registry, altkey, 1);
 | 
						|
    Py_XDECREF(zero);
 | 
						|
    Py_XDECREF(altkey);
 | 
						|
    return rc;
 | 
						|
}
 | 
						|
 | 
						|
static void
 | 
						|
show_warning(PyObject *filename, int lineno, PyObject *text, PyObject
 | 
						|
                *category, PyObject *sourceline)
 | 
						|
{
 | 
						|
    PyObject *f_stderr;
 | 
						|
    PyObject *name;
 | 
						|
    char lineno_str[128];
 | 
						|
 | 
						|
    PyOS_snprintf(lineno_str, sizeof(lineno_str), ":%d: ", lineno);
 | 
						|
 | 
						|
    name = PyObject_GetAttrString(category, "__name__");
 | 
						|
    if (name == NULL)  /* XXX Can an object lack a '__name__' attribute? */
 | 
						|
	    return;
 | 
						|
 | 
						|
    f_stderr = PySys_GetObject("stderr");
 | 
						|
    if (f_stderr == NULL) {
 | 
						|
        fprintf(stderr, "lost sys.stderr\n");
 | 
						|
        Py_DECREF(name);
 | 
						|
        return;
 | 
						|
    }
 | 
						|
 | 
						|
    /* Print "filename:lineno: category: text\n" */
 | 
						|
    PyFile_WriteObject(filename, f_stderr, Py_PRINT_RAW);
 | 
						|
    PyFile_WriteString(lineno_str, f_stderr);
 | 
						|
    PyFile_WriteObject(name, f_stderr, Py_PRINT_RAW);
 | 
						|
    PyFile_WriteString(": ", f_stderr);
 | 
						|
    PyFile_WriteObject(text, f_stderr, Py_PRINT_RAW);
 | 
						|
    PyFile_WriteString("\n", f_stderr);
 | 
						|
    Py_XDECREF(name);
 | 
						|
 | 
						|
    /* Print "  source_line\n" */
 | 
						|
    PyFile_WriteString("  ", f_stderr);
 | 
						|
    if (sourceline) {
 | 
						|
        char *source_line_str = PyBytes_AS_STRING(sourceline);
 | 
						|
        while (*source_line_str == ' ' || *source_line_str == '\t' ||
 | 
						|
                *source_line_str == '\014')
 | 
						|
            source_line_str++;
 | 
						|
 | 
						|
        PyFile_WriteString(source_line_str, f_stderr);
 | 
						|
        PyFile_WriteString("\n", f_stderr);
 | 
						|
    }
 | 
						|
    else
 | 
						|
        Py_DisplaySourceLine(f_stderr, PyBytes_AS_STRING(filename), lineno);
 | 
						|
    PyErr_Clear();
 | 
						|
}
 | 
						|
 | 
						|
static PyObject *
 | 
						|
warn_explicit(PyObject *category, PyObject *message,
 | 
						|
              PyObject *filename, int lineno,
 | 
						|
              PyObject *module, PyObject *registry, PyObject *sourceline)
 | 
						|
{
 | 
						|
    PyObject *key = NULL, *text = NULL, *result = NULL, *lineno_obj = NULL;
 | 
						|
    PyObject *item = Py_None;
 | 
						|
    const char *action;
 | 
						|
    int rc;
 | 
						|
 | 
						|
    /* Normalize module. */
 | 
						|
    if (module == NULL) {
 | 
						|
        module = normalize_module(filename);
 | 
						|
        if (module == NULL)
 | 
						|
            return NULL;
 | 
						|
    }
 | 
						|
    else
 | 
						|
        Py_INCREF(module);
 | 
						|
 | 
						|
    /* Normalize message. */
 | 
						|
    Py_INCREF(message);  /* DECREF'ed in cleanup. */
 | 
						|
    rc = PyObject_IsInstance(message, PyExc_Warning);
 | 
						|
    if (rc == -1) {
 | 
						|
        goto cleanup;
 | 
						|
    }
 | 
						|
    if (rc == 1) {
 | 
						|
        text = PyObject_Str(message);
 | 
						|
        category = (PyObject*)message->ob_type;
 | 
						|
    }
 | 
						|
    else {
 | 
						|
        text = message;
 | 
						|
        message = PyObject_CallFunction(category, "O", message);
 | 
						|
    }
 | 
						|
 | 
						|
    lineno_obj = PyInt_FromLong(lineno);
 | 
						|
    if (lineno_obj == NULL)
 | 
						|
        goto cleanup;
 | 
						|
 | 
						|
    /* Create key. */
 | 
						|
    key = PyTuple_Pack(3, text, category, lineno_obj);
 | 
						|
    if (key == NULL)
 | 
						|
        goto cleanup;
 | 
						|
 | 
						|
    if (registry != NULL) {
 | 
						|
        rc = already_warned(registry, key, 0);
 | 
						|
        if (rc == -1)
 | 
						|
            goto cleanup;
 | 
						|
	else if (rc == 1)
 | 
						|
            goto return_none;
 | 
						|
        /* Else this warning hasn't been generated before. */
 | 
						|
    }
 | 
						|
 | 
						|
    action = get_filter(category, text, lineno, module, &item);
 | 
						|
    if (action == NULL)
 | 
						|
        goto cleanup;
 | 
						|
 | 
						|
    if (strcmp(action, "error") == 0) {
 | 
						|
        PyErr_SetObject(category, message);
 | 
						|
        goto cleanup;
 | 
						|
    }
 | 
						|
 | 
						|
    /* Store in the registry that we've been here, *except* when the action
 | 
						|
       is "always". */
 | 
						|
    rc = 0;
 | 
						|
    if (strcmp(action, "always") != 0) {
 | 
						|
        if (registry != NULL && PyDict_SetItem(registry, key, Py_True) < 0)
 | 
						|
            goto cleanup;
 | 
						|
        else if (strcmp(action, "ignore") == 0)
 | 
						|
            goto return_none;
 | 
						|
        else if (strcmp(action, "once") == 0) {
 | 
						|
            if (registry == NULL) {
 | 
						|
                registry = get_once_registry();
 | 
						|
                if (registry == NULL)
 | 
						|
                    goto cleanup;
 | 
						|
            }
 | 
						|
            /* _once_registry[(text, category)] = 1 */
 | 
						|
            rc = update_registry(registry, text, category, 0);
 | 
						|
        }
 | 
						|
        else if (strcmp(action, "module") == 0) {
 | 
						|
            /* registry[(text, category, 0)] = 1 */
 | 
						|
            if (registry != NULL)
 | 
						|
                rc = update_registry(registry, text, category, 0);
 | 
						|
        }
 | 
						|
        else if (strcmp(action, "default") != 0) {
 | 
						|
            PyObject *to_str = PyObject_Str(item);
 | 
						|
            const char *err_str = "???";
 | 
						|
 | 
						|
            if (to_str != NULL)
 | 
						|
                err_str = PyBytes_AS_STRING(to_str);
 | 
						|
            PyErr_Format(PyExc_RuntimeError,
 | 
						|
                        "Unrecognized action (%s) in warnings.filters:\n %s",
 | 
						|
                        action, err_str);
 | 
						|
            Py_XDECREF(to_str);
 | 
						|
            goto cleanup;
 | 
						|
        }
 | 
						|
    }
 | 
						|
 | 
						|
    if (rc == 1)  // Already warned for this module. */
 | 
						|
        goto return_none;
 | 
						|
    if (rc == 0) {
 | 
						|
        PyObject *show_fxn = get_warnings_attr("showwarning");
 | 
						|
        if (show_fxn == NULL) {
 | 
						|
            if (PyErr_Occurred())
 | 
						|
                goto cleanup;
 | 
						|
            show_warning(filename, lineno, text, category, sourceline);
 | 
						|
        }
 | 
						|
        else {
 | 
						|
            const char *msg = "functions overriding warnings.showwarning() "
 | 
						|
                                "must support the 'line' argument";
 | 
						|
            const char *text_char = PyBytes_AS_STRING(text);
 | 
						|
 | 
						|
            if (strcmp(msg, text_char) == 0) {
 | 
						|
                /* Prevent infinite recursion by using built-in implementation
 | 
						|
                   of showwarning(). */
 | 
						|
                show_warning(filename, lineno, text, category, sourceline);
 | 
						|
            }
 | 
						|
            else {
 | 
						|
                PyObject *check_fxn;
 | 
						|
                PyObject *defaults;
 | 
						|
                PyObject *res;
 | 
						|
 | 
						|
                if (PyMethod_Check(show_fxn))
 | 
						|
                    check_fxn = PyMethod_Function(show_fxn);
 | 
						|
                else if (PyFunction_Check(show_fxn))
 | 
						|
                    check_fxn = show_fxn;
 | 
						|
                else {
 | 
						|
                    PyErr_SetString(PyExc_TypeError,
 | 
						|
                                    "warnings.showwarning() must be set to a "
 | 
						|
                                    "function or method");
 | 
						|
                    Py_DECREF(show_fxn);
 | 
						|
                    goto cleanup;
 | 
						|
                }
 | 
						|
 | 
						|
                defaults = PyFunction_GetDefaults(check_fxn);
 | 
						|
                /* A proper implementation of warnings.showwarning() should
 | 
						|
                    have at least two default arguments. */
 | 
						|
                if ((defaults == NULL) || (PyTuple_Size(defaults) < 2)) {
 | 
						|
                    if (PyErr_WarnEx(PyExc_DeprecationWarning, msg, 1) < 0) {
 | 
						|
                        Py_DECREF(show_fxn);
 | 
						|
                        goto cleanup;
 | 
						|
                    }
 | 
						|
                }
 | 
						|
                res = PyObject_CallFunctionObjArgs(show_fxn, message, category,
 | 
						|
                                                    filename, lineno_obj,
 | 
						|
                                                    NULL);
 | 
						|
                Py_DECREF(show_fxn);
 | 
						|
                Py_XDECREF(res);
 | 
						|
                if (res == NULL)
 | 
						|
                    goto cleanup;
 | 
						|
            }
 | 
						|
        }
 | 
						|
    }
 | 
						|
    else /* if (rc == -1) */
 | 
						|
        goto cleanup;
 | 
						|
 | 
						|
 return_none:
 | 
						|
    result = Py_None;
 | 
						|
    Py_INCREF(result);
 | 
						|
 | 
						|
 cleanup:
 | 
						|
    Py_XDECREF(key);
 | 
						|
    Py_XDECREF(text);
 | 
						|
    Py_XDECREF(lineno_obj);
 | 
						|
    Py_DECREF(module);
 | 
						|
    Py_DECREF(message);
 | 
						|
    return result;  /* Py_None or NULL. */
 | 
						|
}
 | 
						|
 | 
						|
/* filename, module, and registry are new refs, globals is borrowed */
 | 
						|
/* Returns 0 on error (no new refs), 1 on success */
 | 
						|
static int
 | 
						|
setup_context(Py_ssize_t stack_level, PyObject **filename, int *lineno,
 | 
						|
              PyObject **module, PyObject **registry)
 | 
						|
{
 | 
						|
    PyObject *globals;
 | 
						|
 | 
						|
    /* Setup globals and lineno. */
 | 
						|
    PyFrameObject *f = PyThreadState_GET()->frame;
 | 
						|
    while (--stack_level > 0 && f != NULL)
 | 
						|
        f = f->f_back;
 | 
						|
 | 
						|
    if (f == NULL) {
 | 
						|
        globals = PyThreadState_Get()->interp->sysdict;
 | 
						|
        *lineno = 1;
 | 
						|
    }
 | 
						|
    else {
 | 
						|
        globals = f->f_globals;
 | 
						|
        *lineno = PyCode_Addr2Line(f->f_code, f->f_lasti);
 | 
						|
    }
 | 
						|
 | 
						|
    *module = NULL;
 | 
						|
 | 
						|
    /* Setup registry. */
 | 
						|
    assert(globals != NULL);
 | 
						|
    assert(PyDict_Check(globals));
 | 
						|
    *registry = PyDict_GetItemString(globals, "__warningregistry__");
 | 
						|
    if (*registry == NULL) {
 | 
						|
        int rc;
 | 
						|
 | 
						|
        *registry = PyDict_New();
 | 
						|
        if (*registry == NULL)
 | 
						|
            return 0;
 | 
						|
 | 
						|
         rc = PyDict_SetItemString(globals, "__warningregistry__", *registry);
 | 
						|
         if (rc < 0)
 | 
						|
            goto handle_error;
 | 
						|
    }
 | 
						|
    else
 | 
						|
        Py_INCREF(*registry);
 | 
						|
 | 
						|
    /* Setup module. */
 | 
						|
    *module = PyDict_GetItemString(globals, "__name__");
 | 
						|
    if (*module == NULL) {
 | 
						|
        *module = PyBytes_FromString("<string>");
 | 
						|
        if (*module == NULL)
 | 
						|
            goto handle_error;
 | 
						|
    }
 | 
						|
    else
 | 
						|
        Py_INCREF(*module);
 | 
						|
 | 
						|
    /* Setup filename. */
 | 
						|
    *filename = PyDict_GetItemString(globals, "__file__");
 | 
						|
    if (*filename != NULL) {
 | 
						|
	    Py_ssize_t len = PyBytes_Size(*filename);
 | 
						|
        const char *file_str = PyBytes_AsString(*filename);
 | 
						|
	    if (file_str == NULL || (len < 0 && PyErr_Occurred()))
 | 
						|
            goto handle_error;
 | 
						|
 | 
						|
        /* if filename.lower().endswith((".pyc", ".pyo")): */
 | 
						|
        if (len >= 4 &&
 | 
						|
            file_str[len-4] == '.' &&
 | 
						|
            tolower(file_str[len-3]) == 'p' &&
 | 
						|
            tolower(file_str[len-2]) == 'y' &&
 | 
						|
            (tolower(file_str[len-1]) == 'c' ||
 | 
						|
                tolower(file_str[len-1]) == 'o'))
 | 
						|
        {
 | 
						|
            *filename = PyBytes_FromStringAndSize(file_str, len-1);
 | 
						|
	        if (*filename == NULL)
 | 
						|
		        goto handle_error;
 | 
						|
	    }
 | 
						|
	    else
 | 
						|
            Py_INCREF(*filename);
 | 
						|
    }
 | 
						|
    else {
 | 
						|
        const char *module_str = PyBytes_AsString(*module);
 | 
						|
        if (module_str && strcmp(module_str, "__main__") == 0) {
 | 
						|
            PyObject *argv = PySys_GetObject("argv");
 | 
						|
            if (argv != NULL && PyList_Size(argv) > 0) {
 | 
						|
                int is_true;
 | 
						|
                *filename = PyList_GetItem(argv, 0);
 | 
						|
                Py_INCREF(*filename);
 | 
						|
                /* If sys.argv[0] is false, then use '__main__'. */
 | 
						|
                is_true = PyObject_IsTrue(*filename);
 | 
						|
                if (is_true < 0) {
 | 
						|
                    Py_DECREF(*filename);
 | 
						|
                    goto handle_error;
 | 
						|
                }
 | 
						|
                else if (!is_true) {
 | 
						|
                    Py_DECREF(*filename);
 | 
						|
                    *filename = PyBytes_FromString("__main__");
 | 
						|
                    if (*filename == NULL)
 | 
						|
                        goto handle_error;
 | 
						|
                }
 | 
						|
            }
 | 
						|
            else {
 | 
						|
                /* embedded interpreters don't have sys.argv, see bug #839151 */
 | 
						|
                *filename = PyBytes_FromString("__main__");
 | 
						|
	            if (*filename == NULL)
 | 
						|
	                goto handle_error;
 | 
						|
            }
 | 
						|
        }
 | 
						|
        if (*filename == NULL) {
 | 
						|
            *filename = *module;
 | 
						|
            Py_INCREF(*filename);
 | 
						|
        }
 | 
						|
    }
 | 
						|
 | 
						|
    return 1;
 | 
						|
 | 
						|
 handle_error:
 | 
						|
    /* filename not XDECREF'ed here as there is no way to jump here with a
 | 
						|
       dangling reference. */
 | 
						|
    Py_XDECREF(*registry);
 | 
						|
    Py_XDECREF(*module);
 | 
						|
    return 0;
 | 
						|
}
 | 
						|
 | 
						|
static PyObject *
 | 
						|
get_category(PyObject *message, PyObject *category)
 | 
						|
{
 | 
						|
    int rc;
 | 
						|
 | 
						|
    /* Get category. */
 | 
						|
    rc = PyObject_IsInstance(message, PyExc_Warning);
 | 
						|
    if (rc == -1)
 | 
						|
        return NULL;
 | 
						|
 | 
						|
    if (rc == 1)
 | 
						|
        category = (PyObject*)message->ob_type;
 | 
						|
    else if (category == NULL)
 | 
						|
        category = PyExc_UserWarning;
 | 
						|
 | 
						|
    /* Validate category. */
 | 
						|
    rc = PyObject_IsSubclass(category, PyExc_Warning);
 | 
						|
    if (rc == -1)
 | 
						|
        return NULL;
 | 
						|
    if (rc == 0) {
 | 
						|
        PyErr_SetString(PyExc_ValueError,
 | 
						|
                        "category is not a subclass of Warning");
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
 | 
						|
    return category;
 | 
						|
}
 | 
						|
 | 
						|
static PyObject *
 | 
						|
do_warn(PyObject *message, PyObject *category, Py_ssize_t stack_level)
 | 
						|
{
 | 
						|
    PyObject *filename, *module, *registry, *res;
 | 
						|
    int lineno;
 | 
						|
 | 
						|
    if (!setup_context(stack_level, &filename, &lineno, &module, ®istry))
 | 
						|
        return NULL;
 | 
						|
 | 
						|
    res = warn_explicit(category, message, filename, lineno, module, registry,
 | 
						|
                        NULL);
 | 
						|
    Py_DECREF(filename);
 | 
						|
    Py_DECREF(registry);
 | 
						|
    Py_DECREF(module);
 | 
						|
    return res;
 | 
						|
}
 | 
						|
 | 
						|
static PyObject *
 | 
						|
warnings_warn(PyObject *self, PyObject *args, PyObject *kwds)
 | 
						|
{
 | 
						|
    static char *kw_list[] = { "message", "category", "stacklevel", 0 };
 | 
						|
    PyObject *message, *category = NULL;
 | 
						|
    Py_ssize_t stack_level = 1;
 | 
						|
 | 
						|
    if (!PyArg_ParseTupleAndKeywords(args, kwds, "O|On:warn", kw_list,
 | 
						|
                                     &message, &category, &stack_level))
 | 
						|
        return NULL;
 | 
						|
 | 
						|
    category = get_category(message, category);
 | 
						|
    if (category == NULL)
 | 
						|
        return NULL;
 | 
						|
    return do_warn(message, category, stack_level);
 | 
						|
}
 | 
						|
 | 
						|
static PyObject *
 | 
						|
warnings_warn_explicit(PyObject *self, PyObject *args, PyObject *kwds)
 | 
						|
{
 | 
						|
    static char *kwd_list[] = {"message", "category", "filename", "lineno",
 | 
						|
                                "module", "registry", "module_globals", 0};
 | 
						|
    PyObject *message;
 | 
						|
    PyObject *category;
 | 
						|
    PyObject *filename;
 | 
						|
    int lineno;
 | 
						|
    PyObject *module = NULL;
 | 
						|
    PyObject *registry = NULL;
 | 
						|
    PyObject *module_globals = NULL;
 | 
						|
 | 
						|
    if (!PyArg_ParseTupleAndKeywords(args, kwds, "OOOi|OOO:warn_explicit",
 | 
						|
                kwd_list, &message, &category, &filename, &lineno, &module,
 | 
						|
                ®istry, &module_globals))
 | 
						|
        return NULL;
 | 
						|
 | 
						|
    if (module_globals) {
 | 
						|
        static PyObject *get_source_name = NULL;
 | 
						|
        static PyObject *splitlines_name = NULL;
 | 
						|
        PyObject *loader;
 | 
						|
        PyObject *module_name;
 | 
						|
        PyObject *source;
 | 
						|
        PyObject *source_list;
 | 
						|
        PyObject *source_line;
 | 
						|
        PyObject *returned;
 | 
						|
 | 
						|
        if (get_source_name == NULL) {
 | 
						|
            get_source_name = PyBytes_InternFromString("get_source");
 | 
						|
            if (!get_source_name)
 | 
						|
                return NULL;
 | 
						|
        }
 | 
						|
        if (splitlines_name == NULL) {
 | 
						|
            splitlines_name = PyBytes_InternFromString("splitlines");
 | 
						|
            if (!splitlines_name)
 | 
						|
                return NULL;
 | 
						|
        }
 | 
						|
 | 
						|
        /* Check/get the requisite pieces needed for the loader. */
 | 
						|
        loader = PyDict_GetItemString(module_globals, "__loader__");
 | 
						|
        module_name = PyDict_GetItemString(module_globals, "__name__");
 | 
						|
 | 
						|
        if (loader == NULL || module_name == NULL)
 | 
						|
            goto standard_call;
 | 
						|
 | 
						|
        /* Make sure the loader implements the optional get_source() method. */
 | 
						|
        if (!PyObject_HasAttrString(loader, "get_source"))
 | 
						|
                goto standard_call;
 | 
						|
        /* Call get_source() to get the source code. */
 | 
						|
        source = PyObject_CallMethodObjArgs(loader, get_source_name,
 | 
						|
                                                module_name, NULL);
 | 
						|
        if (!source)
 | 
						|
            return NULL;
 | 
						|
        else if (source == Py_None) {
 | 
						|
            Py_DECREF(Py_None);
 | 
						|
            goto standard_call;
 | 
						|
        }
 | 
						|
 | 
						|
        /* Split the source into lines. */
 | 
						|
        source_list = PyObject_CallMethodObjArgs(source, splitlines_name,
 | 
						|
                                                    NULL);
 | 
						|
        Py_DECREF(source);
 | 
						|
        if (!source_list)
 | 
						|
            return NULL;
 | 
						|
 | 
						|
        /* Get the source line. */
 | 
						|
        source_line = PyList_GetItem(source_list, lineno-1);
 | 
						|
        if (!source_line) {
 | 
						|
            Py_DECREF(source_list);
 | 
						|
            return NULL;
 | 
						|
        }
 | 
						|
 | 
						|
        /* Handle the warning. */
 | 
						|
        returned = warn_explicit(category, message, filename, lineno, module,
 | 
						|
                            registry, source_line);
 | 
						|
        Py_DECREF(source_list);
 | 
						|
        return returned;
 | 
						|
    }
 | 
						|
 | 
						|
 standard_call:
 | 
						|
    return warn_explicit(category, message, filename, lineno, module,
 | 
						|
                                registry, NULL);
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/* Function to issue a warning message; may raise an exception. */
 | 
						|
int
 | 
						|
PyErr_WarnEx(PyObject *category, const char *text, Py_ssize_t stack_level)
 | 
						|
{
 | 
						|
    PyObject *res;
 | 
						|
    PyObject *message = PyBytes_FromString(text);
 | 
						|
    if (message == NULL)
 | 
						|
        return -1;
 | 
						|
 | 
						|
    if (category == NULL)
 | 
						|
        category = PyExc_RuntimeWarning;
 | 
						|
 | 
						|
    res = do_warn(message, category, stack_level);
 | 
						|
    Py_DECREF(message);
 | 
						|
    if (res == NULL)
 | 
						|
        return -1;
 | 
						|
    Py_DECREF(res);
 | 
						|
 | 
						|
    return 0;
 | 
						|
}
 | 
						|
 | 
						|
/* PyErr_Warn is only for backwards compatability and will be removed.
 | 
						|
   Use PyErr_WarnEx instead. */
 | 
						|
 | 
						|
#undef PyErr_Warn
 | 
						|
 | 
						|
PyAPI_FUNC(int)
 | 
						|
PyErr_Warn(PyObject *category, char *text)
 | 
						|
{
 | 
						|
    return PyErr_WarnEx(category, text, 1);
 | 
						|
}
 | 
						|
 | 
						|
/* Warning with explicit origin */
 | 
						|
int
 | 
						|
PyErr_WarnExplicit(PyObject *category, const char *text,
 | 
						|
                   const char *filename_str, int lineno,
 | 
						|
                   const char *module_str, PyObject *registry)
 | 
						|
{
 | 
						|
    PyObject *res;
 | 
						|
    PyObject *message = PyBytes_FromString(text);
 | 
						|
    PyObject *filename = PyBytes_FromString(filename_str);
 | 
						|
    PyObject *module = NULL;
 | 
						|
    int ret = -1;
 | 
						|
 | 
						|
    if (message == NULL || filename == NULL)
 | 
						|
        goto exit;
 | 
						|
    if (module_str != NULL) {
 | 
						|
        module = PyBytes_FromString(module_str);
 | 
						|
            if (module == NULL)
 | 
						|
                goto exit;
 | 
						|
    }
 | 
						|
 | 
						|
    if (category == NULL)
 | 
						|
        category = PyExc_RuntimeWarning;
 | 
						|
    res = warn_explicit(category, message, filename, lineno, module, registry,
 | 
						|
                        NULL);
 | 
						|
    if (res == NULL)
 | 
						|
        goto exit;
 | 
						|
    Py_DECREF(res);
 | 
						|
    ret = 0;
 | 
						|
 | 
						|
 exit:
 | 
						|
    Py_XDECREF(message);
 | 
						|
    Py_XDECREF(module);
 | 
						|
    Py_XDECREF(filename);
 | 
						|
    return ret;
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
PyDoc_STRVAR(warn_doc,
 | 
						|
"Issue a warning, or maybe ignore it or raise an exception.");
 | 
						|
 | 
						|
PyDoc_STRVAR(warn_explicit_doc,
 | 
						|
"Low-level inferface to warnings functionality.");
 | 
						|
 | 
						|
static PyMethodDef warnings_functions[] = {
 | 
						|
    {"warn", (PyCFunction)warnings_warn, METH_VARARGS | METH_KEYWORDS,
 | 
						|
        warn_doc},
 | 
						|
    {"warn_explicit", (PyCFunction)warnings_warn_explicit,
 | 
						|
        METH_VARARGS | METH_KEYWORDS, warn_explicit_doc},
 | 
						|
    // XXX(brett.cannon): add showwarning?
 | 
						|
    // XXX(brett.cannon): Reasonable to add formatwarning?
 | 
						|
    {NULL, NULL}	        /* sentinel */
 | 
						|
};
 | 
						|
 | 
						|
 | 
						|
static PyObject *
 | 
						|
create_filter(PyObject *category, const char *action)
 | 
						|
{
 | 
						|
    static PyObject *ignore_str = NULL;
 | 
						|
    static PyObject *error_str = NULL;
 | 
						|
    static PyObject *default_str = NULL;
 | 
						|
    PyObject *action_obj = NULL;
 | 
						|
    PyObject *lineno, *result;
 | 
						|
 | 
						|
    if (!strcmp(action, "ignore")) {
 | 
						|
        if (ignore_str == NULL) {
 | 
						|
            ignore_str = PyBytes_InternFromString("ignore");
 | 
						|
            if (ignore_str == NULL)
 | 
						|
                return NULL;
 | 
						|
        }
 | 
						|
        action_obj = ignore_str;
 | 
						|
    }
 | 
						|
    else if (!strcmp(action, "error")) {
 | 
						|
        if (error_str == NULL) {
 | 
						|
            error_str = PyBytes_InternFromString("error");
 | 
						|
            if (error_str == NULL)
 | 
						|
                return NULL;
 | 
						|
        }
 | 
						|
        action_obj = error_str;
 | 
						|
    }
 | 
						|
    else if (!strcmp(action, "default")) {
 | 
						|
        if (default_str == NULL) {
 | 
						|
            default_str = PyBytes_InternFromString("default");
 | 
						|
            if (default_str == NULL)
 | 
						|
                return NULL;
 | 
						|
        }
 | 
						|
        action_obj = default_str;
 | 
						|
    }
 | 
						|
    else {
 | 
						|
        Py_FatalError("unknown action");
 | 
						|
    }
 | 
						|
 | 
						|
    /* This assumes the line number is zero for now. */
 | 
						|
    lineno = PyInt_FromLong(0);
 | 
						|
    if (lineno == NULL)
 | 
						|
        return NULL;
 | 
						|
    result = PyTuple_Pack(5, action_obj, Py_None, category, Py_None, lineno);
 | 
						|
    Py_DECREF(lineno);
 | 
						|
    return result;
 | 
						|
}
 | 
						|
 | 
						|
static PyObject *
 | 
						|
init_filters(void)
 | 
						|
{
 | 
						|
    PyObject *filters = PyList_New(3);
 | 
						|
    const char *bytes_action;
 | 
						|
    if (filters == NULL)
 | 
						|
        return NULL;
 | 
						|
 | 
						|
    PyList_SET_ITEM(filters, 0,
 | 
						|
                    create_filter(PyExc_PendingDeprecationWarning, "ignore"));
 | 
						|
    PyList_SET_ITEM(filters, 1, create_filter(PyExc_ImportWarning, "ignore"));
 | 
						|
    if (Py_BytesWarningFlag > 1)
 | 
						|
        bytes_action = "error";
 | 
						|
    else if (Py_BytesWarningFlag)
 | 
						|
        bytes_action = "default";
 | 
						|
    else
 | 
						|
        bytes_action = "ignore";
 | 
						|
    PyList_SET_ITEM(filters, 2, create_filter(PyExc_BytesWarning,
 | 
						|
                    bytes_action));
 | 
						|
 | 
						|
    if (PyList_GET_ITEM(filters, 0) == NULL ||
 | 
						|
        PyList_GET_ITEM(filters, 1) == NULL ||
 | 
						|
        PyList_GET_ITEM(filters, 2) == NULL) {
 | 
						|
        Py_DECREF(filters);
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
 | 
						|
    return filters;
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
PyMODINIT_FUNC
 | 
						|
_PyWarnings_Init(void)
 | 
						|
{
 | 
						|
    PyObject *m, *default_action;
 | 
						|
 | 
						|
    m = Py_InitModule3(MODULE_NAME, warnings_functions, warnings__doc__);
 | 
						|
    if (m == NULL)
 | 
						|
        return;
 | 
						|
 | 
						|
    _filters = init_filters();
 | 
						|
    if (_filters == NULL)
 | 
						|
        return;
 | 
						|
    Py_INCREF(_filters);
 | 
						|
    if (PyModule_AddObject(m, "filters", _filters) < 0)
 | 
						|
        return;
 | 
						|
 | 
						|
    _once_registry = PyDict_New();
 | 
						|
    if (_once_registry == NULL)
 | 
						|
        return;
 | 
						|
    Py_INCREF(_once_registry);
 | 
						|
    if (PyModule_AddObject(m, "once_registry", _once_registry) < 0)
 | 
						|
        return;
 | 
						|
 | 
						|
    default_action = PyBytes_InternFromString("default");
 | 
						|
    if (default_action == NULL)
 | 
						|
        return;
 | 
						|
    if (PyModule_AddObject(m, DEFAULT_ACTION_NAME, default_action) < 0)
 | 
						|
        return;
 | 
						|
}
 |