Update bug/patch counts
[pytest.git] / Python / errors.c
blob7b7105104ad719c5c5d561636c559ca2ffea6014
2 /* Error handling */
4 #include "Python.h"
6 #ifndef __STDC__
7 #ifndef MS_WINDOWS
8 extern char *strerror(int);
9 #endif
10 #endif
12 #ifdef MS_WINDOWS
13 #include "windows.h"
14 #include "winbase.h"
15 #endif
17 #include <ctype.h>
19 #ifdef __cplusplus
20 extern "C" {
21 #endif
24 void
25 PyErr_Restore(PyObject *type, PyObject *value, PyObject *traceback)
27 PyThreadState *tstate = PyThreadState_GET();
28 PyObject *oldtype, *oldvalue, *oldtraceback;
30 if (traceback != NULL && !PyTraceBack_Check(traceback)) {
31 /* XXX Should never happen -- fatal error instead? */
32 /* Well, it could be None. */
33 Py_DECREF(traceback);
34 traceback = NULL;
37 /* Save these in locals to safeguard against recursive
38 invocation through Py_XDECREF */
39 oldtype = tstate->curexc_type;
40 oldvalue = tstate->curexc_value;
41 oldtraceback = tstate->curexc_traceback;
43 tstate->curexc_type = type;
44 tstate->curexc_value = value;
45 tstate->curexc_traceback = traceback;
47 Py_XDECREF(oldtype);
48 Py_XDECREF(oldvalue);
49 Py_XDECREF(oldtraceback);
52 void
53 PyErr_SetObject(PyObject *exception, PyObject *value)
55 Py_XINCREF(exception);
56 Py_XINCREF(value);
57 PyErr_Restore(exception, value, (PyObject *)NULL);
60 void
61 PyErr_SetNone(PyObject *exception)
63 PyErr_SetObject(exception, (PyObject *)NULL);
66 void
67 PyErr_SetString(PyObject *exception, const char *string)
69 PyObject *value = PyString_FromString(string);
70 PyErr_SetObject(exception, value);
71 Py_XDECREF(value);
75 PyObject *
76 PyErr_Occurred(void)
78 PyThreadState *tstate = PyThreadState_GET();
80 return tstate->curexc_type;
84 int
85 PyErr_GivenExceptionMatches(PyObject *err, PyObject *exc)
87 if (err == NULL || exc == NULL) {
88 /* maybe caused by "import exceptions" that failed early on */
89 return 0;
91 if (PyTuple_Check(exc)) {
92 Py_ssize_t i, n;
93 n = PyTuple_Size(exc);
94 for (i = 0; i < n; i++) {
95 /* Test recursively */
96 if (PyErr_GivenExceptionMatches(
97 err, PyTuple_GET_ITEM(exc, i)))
99 return 1;
102 return 0;
104 /* err might be an instance, so check its class. */
105 if (PyExceptionInstance_Check(err))
106 err = PyExceptionInstance_Class(err);
108 if (PyExceptionClass_Check(err) && PyExceptionClass_Check(exc)) {
109 /* problems here!? not sure PyObject_IsSubclass expects to
110 be called with an exception pending... */
111 return PyObject_IsSubclass(err, exc);
114 return err == exc;
119 PyErr_ExceptionMatches(PyObject *exc)
121 return PyErr_GivenExceptionMatches(PyErr_Occurred(), exc);
125 /* Used in many places to normalize a raised exception, including in
126 eval_code2(), do_raise(), and PyErr_Print()
128 void
129 PyErr_NormalizeException(PyObject **exc, PyObject **val, PyObject **tb)
131 PyObject *type = *exc;
132 PyObject *value = *val;
133 PyObject *inclass = NULL;
134 PyObject *initial_tb = NULL;
136 if (type == NULL) {
137 /* There was no exception, so nothing to do. */
138 return;
141 /* If PyErr_SetNone() was used, the value will have been actually
142 set to NULL.
144 if (!value) {
145 value = Py_None;
146 Py_INCREF(value);
149 if (PyExceptionInstance_Check(value))
150 inclass = PyExceptionInstance_Class(value);
152 /* Normalize the exception so that if the type is a class, the
153 value will be an instance.
155 if (PyExceptionClass_Check(type)) {
156 /* if the value was not an instance, or is not an instance
157 whose class is (or is derived from) type, then use the
158 value as an argument to instantiation of the type
159 class.
161 if (!inclass || !PyObject_IsSubclass(inclass, type)) {
162 PyObject *args, *res;
164 if (value == Py_None)
165 args = PyTuple_New(0);
166 else if (PyTuple_Check(value)) {
167 Py_INCREF(value);
168 args = value;
170 else
171 args = PyTuple_Pack(1, value);
173 if (args == NULL)
174 goto finally;
175 res = PyEval_CallObject(type, args);
176 Py_DECREF(args);
177 if (res == NULL)
178 goto finally;
179 Py_DECREF(value);
180 value = res;
182 /* if the class of the instance doesn't exactly match the
183 class of the type, believe the instance
185 else if (inclass != type) {
186 Py_DECREF(type);
187 type = inclass;
188 Py_INCREF(type);
191 *exc = type;
192 *val = value;
193 return;
194 finally:
195 Py_DECREF(type);
196 Py_DECREF(value);
197 /* If the new exception doesn't set a traceback and the old
198 exception had a traceback, use the old traceback for the
199 new exception. It's better than nothing.
201 initial_tb = *tb;
202 PyErr_Fetch(exc, val, tb);
203 if (initial_tb != NULL) {
204 if (*tb == NULL)
205 *tb = initial_tb;
206 else
207 Py_DECREF(initial_tb);
209 /* normalize recursively */
210 PyErr_NormalizeException(exc, val, tb);
214 void
215 PyErr_Fetch(PyObject **p_type, PyObject **p_value, PyObject **p_traceback)
217 PyThreadState *tstate = PyThreadState_GET();
219 *p_type = tstate->curexc_type;
220 *p_value = tstate->curexc_value;
221 *p_traceback = tstate->curexc_traceback;
223 tstate->curexc_type = NULL;
224 tstate->curexc_value = NULL;
225 tstate->curexc_traceback = NULL;
228 void
229 PyErr_Clear(void)
231 PyErr_Restore(NULL, NULL, NULL);
234 /* Convenience functions to set a type error exception and return 0 */
237 PyErr_BadArgument(void)
239 PyErr_SetString(PyExc_TypeError,
240 "bad argument type for built-in operation");
241 return 0;
244 PyObject *
245 PyErr_NoMemory(void)
247 if (PyErr_ExceptionMatches(PyExc_MemoryError))
248 /* already current */
249 return NULL;
251 /* raise the pre-allocated instance if it still exists */
252 if (PyExc_MemoryErrorInst)
253 PyErr_SetObject(PyExc_MemoryError, PyExc_MemoryErrorInst);
254 else
255 /* this will probably fail since there's no memory and hee,
256 hee, we have to instantiate this class
258 PyErr_SetNone(PyExc_MemoryError);
260 return NULL;
263 PyObject *
264 PyErr_SetFromErrnoWithFilenameObject(PyObject *exc, PyObject *filenameObject)
266 PyObject *v;
267 char *s;
268 int i = errno;
269 #ifdef PLAN9
270 char errbuf[ERRMAX];
271 #endif
272 #ifdef MS_WINDOWS
273 char *s_buf = NULL;
274 char s_small_buf[28]; /* Room for "Windows Error 0xFFFFFFFF" */
275 #endif
276 #ifdef EINTR
277 if (i == EINTR && PyErr_CheckSignals())
278 return NULL;
279 #endif
280 #ifdef PLAN9
281 rerrstr(errbuf, sizeof errbuf);
282 s = errbuf;
283 #else
284 if (i == 0)
285 s = "Error"; /* Sometimes errno didn't get set */
286 else
287 #ifndef MS_WINDOWS
288 s = strerror(i);
289 #else
291 /* Note that the Win32 errors do not lineup with the
292 errno error. So if the error is in the MSVC error
293 table, we use it, otherwise we assume it really _is_
294 a Win32 error code
296 if (i > 0 && i < _sys_nerr) {
297 s = _sys_errlist[i];
299 else {
300 int len = FormatMessage(
301 FORMAT_MESSAGE_ALLOCATE_BUFFER |
302 FORMAT_MESSAGE_FROM_SYSTEM |
303 FORMAT_MESSAGE_IGNORE_INSERTS,
304 NULL, /* no message source */
306 MAKELANGID(LANG_NEUTRAL,
307 SUBLANG_DEFAULT),
308 /* Default language */
309 (LPTSTR) &s_buf,
310 0, /* size not used */
311 NULL); /* no args */
312 if (len==0) {
313 /* Only ever seen this in out-of-mem
314 situations */
315 sprintf(s_small_buf, "Windows Error 0x%X", i);
316 s = s_small_buf;
317 s_buf = NULL;
318 } else {
319 s = s_buf;
320 /* remove trailing cr/lf and dots */
321 while (len > 0 && (s[len-1] <= ' ' || s[len-1] == '.'))
322 s[--len] = '\0';
326 #endif /* Unix/Windows */
327 #endif /* PLAN 9*/
328 if (filenameObject != NULL)
329 v = Py_BuildValue("(isO)", i, s, filenameObject);
330 else
331 v = Py_BuildValue("(is)", i, s);
332 if (v != NULL) {
333 PyErr_SetObject(exc, v);
334 Py_DECREF(v);
336 #ifdef MS_WINDOWS
337 LocalFree(s_buf);
338 #endif
339 return NULL;
343 PyObject *
344 PyErr_SetFromErrnoWithFilename(PyObject *exc, char *filename)
346 PyObject *name = filename ? PyString_FromString(filename) : NULL;
347 PyObject *result = PyErr_SetFromErrnoWithFilenameObject(exc, name);
348 Py_XDECREF(name);
349 return result;
352 #ifdef Py_WIN_WIDE_FILENAMES
353 PyObject *
354 PyErr_SetFromErrnoWithUnicodeFilename(PyObject *exc, Py_UNICODE *filename)
356 PyObject *name = filename ?
357 PyUnicode_FromUnicode(filename, wcslen(filename)) :
358 NULL;
359 PyObject *result = PyErr_SetFromErrnoWithFilenameObject(exc, name);
360 Py_XDECREF(name);
361 return result;
363 #endif /* Py_WIN_WIDE_FILENAMES */
365 PyObject *
366 PyErr_SetFromErrno(PyObject *exc)
368 return PyErr_SetFromErrnoWithFilenameObject(exc, NULL);
371 #ifdef MS_WINDOWS
372 /* Windows specific error code handling */
373 PyObject *PyErr_SetExcFromWindowsErrWithFilenameObject(
374 PyObject *exc,
375 int ierr,
376 PyObject *filenameObject)
378 int len;
379 char *s;
380 char *s_buf = NULL; /* Free via LocalFree */
381 char s_small_buf[28]; /* Room for "Windows Error 0xFFFFFFFF" */
382 PyObject *v;
383 DWORD err = (DWORD)ierr;
384 if (err==0) err = GetLastError();
385 len = FormatMessage(
386 /* Error API error */
387 FORMAT_MESSAGE_ALLOCATE_BUFFER |
388 FORMAT_MESSAGE_FROM_SYSTEM |
389 FORMAT_MESSAGE_IGNORE_INSERTS,
390 NULL, /* no message source */
391 err,
392 MAKELANGID(LANG_NEUTRAL,
393 SUBLANG_DEFAULT), /* Default language */
394 (LPTSTR) &s_buf,
395 0, /* size not used */
396 NULL); /* no args */
397 if (len==0) {
398 /* Only seen this in out of mem situations */
399 sprintf(s_small_buf, "Windows Error 0x%X", err);
400 s = s_small_buf;
401 s_buf = NULL;
402 } else {
403 s = s_buf;
404 /* remove trailing cr/lf and dots */
405 while (len > 0 && (s[len-1] <= ' ' || s[len-1] == '.'))
406 s[--len] = '\0';
408 if (filenameObject != NULL)
409 v = Py_BuildValue("(isO)", err, s, filenameObject);
410 else
411 v = Py_BuildValue("(is)", err, s);
412 if (v != NULL) {
413 PyErr_SetObject(exc, v);
414 Py_DECREF(v);
416 LocalFree(s_buf);
417 return NULL;
420 PyObject *PyErr_SetExcFromWindowsErrWithFilename(
421 PyObject *exc,
422 int ierr,
423 const char *filename)
425 PyObject *name = filename ? PyString_FromString(filename) : NULL;
426 PyObject *ret = PyErr_SetExcFromWindowsErrWithFilenameObject(exc,
427 ierr,
428 name);
429 Py_XDECREF(name);
430 return ret;
433 #ifdef Py_WIN_WIDE_FILENAMES
434 PyObject *PyErr_SetExcFromWindowsErrWithUnicodeFilename(
435 PyObject *exc,
436 int ierr,
437 const Py_UNICODE *filename)
439 PyObject *name = filename ?
440 PyUnicode_FromUnicode(filename, wcslen(filename)) :
441 NULL;
442 PyObject *ret = PyErr_SetExcFromWindowsErrWithFilenameObject(exc,
443 ierr,
444 name);
445 Py_XDECREF(name);
446 return ret;
448 #endif /* Py_WIN_WIDE_FILENAMES */
450 PyObject *PyErr_SetExcFromWindowsErr(PyObject *exc, int ierr)
452 return PyErr_SetExcFromWindowsErrWithFilename(exc, ierr, NULL);
455 PyObject *PyErr_SetFromWindowsErr(int ierr)
457 return PyErr_SetExcFromWindowsErrWithFilename(PyExc_WindowsError,
458 ierr, NULL);
460 PyObject *PyErr_SetFromWindowsErrWithFilename(
461 int ierr,
462 const char *filename)
464 PyObject *name = filename ? PyString_FromString(filename) : NULL;
465 PyObject *result = PyErr_SetExcFromWindowsErrWithFilenameObject(
466 PyExc_WindowsError,
467 ierr, name);
468 Py_XDECREF(name);
469 return result;
472 #ifdef Py_WIN_WIDE_FILENAMES
473 PyObject *PyErr_SetFromWindowsErrWithUnicodeFilename(
474 int ierr,
475 const Py_UNICODE *filename)
477 PyObject *name = filename ?
478 PyUnicode_FromUnicode(filename, wcslen(filename)) :
479 NULL;
480 PyObject *result = PyErr_SetExcFromWindowsErrWithFilenameObject(
481 PyExc_WindowsError,
482 ierr, name);
483 Py_XDECREF(name);
484 return result;
486 #endif /* Py_WIN_WIDE_FILENAMES */
487 #endif /* MS_WINDOWS */
489 void
490 _PyErr_BadInternalCall(char *filename, int lineno)
492 PyErr_Format(PyExc_SystemError,
493 "%s:%d: bad argument to internal function",
494 filename, lineno);
497 /* Remove the preprocessor macro for PyErr_BadInternalCall() so that we can
498 export the entry point for existing object code: */
499 #undef PyErr_BadInternalCall
500 void
501 PyErr_BadInternalCall(void)
503 PyErr_Format(PyExc_SystemError,
504 "bad argument to internal function");
506 #define PyErr_BadInternalCall() _PyErr_BadInternalCall(__FILE__, __LINE__)
510 PyObject *
511 PyErr_Format(PyObject *exception, const char *format, ...)
513 va_list vargs;
514 PyObject* string;
516 #ifdef HAVE_STDARG_PROTOTYPES
517 va_start(vargs, format);
518 #else
519 va_start(vargs);
520 #endif
522 string = PyString_FromFormatV(format, vargs);
523 PyErr_SetObject(exception, string);
524 Py_XDECREF(string);
525 va_end(vargs);
526 return NULL;
531 PyObject *
532 PyErr_NewException(char *name, PyObject *base, PyObject *dict)
534 char *dot;
535 PyObject *modulename = NULL;
536 PyObject *classname = NULL;
537 PyObject *mydict = NULL;
538 PyObject *bases = NULL;
539 PyObject *result = NULL;
540 dot = strrchr(name, '.');
541 if (dot == NULL) {
542 PyErr_SetString(PyExc_SystemError,
543 "PyErr_NewException: name must be module.class");
544 return NULL;
546 if (base == NULL)
547 base = PyExc_Exception;
548 if (dict == NULL) {
549 dict = mydict = PyDict_New();
550 if (dict == NULL)
551 goto failure;
553 if (PyDict_GetItemString(dict, "__module__") == NULL) {
554 modulename = PyString_FromStringAndSize(name, (int)(dot-name));
555 if (modulename == NULL)
556 goto failure;
557 if (PyDict_SetItemString(dict, "__module__", modulename) != 0)
558 goto failure;
560 if (PyTuple_Check(base)) {
561 bases = base;
562 /* INCREF as we create a new ref in the else branch */
563 Py_INCREF(bases);
564 } else {
565 bases = PyTuple_Pack(1, base);
566 if (bases == NULL)
567 goto failure;
569 /* Create a real new-style class. */
570 result = PyObject_CallFunction((PyObject *)&PyType_Type, "sOO",
571 dot+1, bases, dict);
572 failure:
573 Py_XDECREF(bases);
574 Py_XDECREF(mydict);
575 Py_XDECREF(classname);
576 Py_XDECREF(modulename);
577 return result;
580 /* Call when an exception has occurred but there is no way for Python
581 to handle it. Examples: exception in __del__ or during GC. */
582 void
583 PyErr_WriteUnraisable(PyObject *obj)
585 PyObject *f, *t, *v, *tb;
586 PyErr_Fetch(&t, &v, &tb);
587 f = PySys_GetObject("stderr");
588 if (f != NULL) {
589 PyFile_WriteString("Exception ", f);
590 if (t) {
591 PyObject* moduleName;
592 char* className = PyExceptionClass_Name(t);
594 if (className != NULL) {
595 char *dot = strrchr(className, '.');
596 if (dot != NULL)
597 className = dot+1;
600 moduleName = PyObject_GetAttrString(t, "__module__");
601 if (moduleName == NULL)
602 PyFile_WriteString("<unknown>", f);
603 else {
604 char* modstr = PyString_AsString(moduleName);
605 if (modstr)
607 PyFile_WriteString(modstr, f);
608 PyFile_WriteString(".", f);
611 if (className == NULL)
612 PyFile_WriteString("<unknown>", f);
613 else
614 PyFile_WriteString(className, f);
615 if (v && v != Py_None) {
616 PyFile_WriteString(": ", f);
617 PyFile_WriteObject(v, f, 0);
619 Py_XDECREF(moduleName);
621 PyFile_WriteString(" in ", f);
622 PyFile_WriteObject(obj, f, 0);
623 PyFile_WriteString(" ignored\n", f);
624 PyErr_Clear(); /* Just in case */
626 Py_XDECREF(t);
627 Py_XDECREF(v);
628 Py_XDECREF(tb);
631 extern PyObject *PyModule_GetWarningsModule(void);
633 /* Function to issue a warning message; may raise an exception. */
635 PyErr_WarnEx(PyObject *category, const char *message, Py_ssize_t stack_level)
637 PyObject *dict, *func = NULL;
638 PyObject *warnings_module = PyModule_GetWarningsModule();
640 if (warnings_module != NULL) {
641 dict = PyModule_GetDict(warnings_module);
642 func = PyDict_GetItemString(dict, "warn");
644 if (func == NULL) {
645 PySys_WriteStderr("warning: %s\n", message);
646 return 0;
648 else {
649 PyObject *res;
651 if (category == NULL)
652 category = PyExc_RuntimeWarning;
653 res = PyObject_CallFunction(func, "sOn",
654 message, category, stack_level);
655 if (res == NULL)
656 return -1;
657 Py_DECREF(res);
658 return 0;
662 /* PyErr_Warn is only for backwards compatability and will be removed.
663 Use PyErr_WarnEx instead. */
665 #undef PyErr_Warn
667 PyAPI_FUNC(int)
668 PyErr_Warn(PyObject *category, char *message)
670 return PyErr_WarnEx(category, message, 1);
673 /* Warning with explicit origin */
675 PyErr_WarnExplicit(PyObject *category, const char *message,
676 const char *filename, int lineno,
677 const char *module, PyObject *registry)
679 PyObject *mod, *dict, *func = NULL;
681 mod = PyImport_ImportModule("warnings");
682 if (mod != NULL) {
683 dict = PyModule_GetDict(mod);
684 func = PyDict_GetItemString(dict, "warn_explicit");
685 Py_DECREF(mod);
687 if (func == NULL) {
688 PySys_WriteStderr("warning: %s\n", message);
689 return 0;
691 else {
692 PyObject *res;
694 if (category == NULL)
695 category = PyExc_RuntimeWarning;
696 if (registry == NULL)
697 registry = Py_None;
698 res = PyObject_CallFunction(func, "sOsizO", message, category,
699 filename, lineno, module, registry);
700 if (res == NULL)
701 return -1;
702 Py_DECREF(res);
703 return 0;
708 /* Set file and line information for the current exception.
709 If the exception is not a SyntaxError, also sets additional attributes
710 to make printing of exceptions believe it is a syntax error. */
712 void
713 PyErr_SyntaxLocation(const char *filename, int lineno)
715 PyObject *exc, *v, *tb, *tmp;
717 /* add attributes for the line number and filename for the error */
718 PyErr_Fetch(&exc, &v, &tb);
719 PyErr_NormalizeException(&exc, &v, &tb);
720 /* XXX check that it is, indeed, a syntax error. It might not
721 * be, though. */
722 tmp = PyInt_FromLong(lineno);
723 if (tmp == NULL)
724 PyErr_Clear();
725 else {
726 if (PyObject_SetAttrString(v, "lineno", tmp))
727 PyErr_Clear();
728 Py_DECREF(tmp);
730 if (filename != NULL) {
731 tmp = PyString_FromString(filename);
732 if (tmp == NULL)
733 PyErr_Clear();
734 else {
735 if (PyObject_SetAttrString(v, "filename", tmp))
736 PyErr_Clear();
737 Py_DECREF(tmp);
740 tmp = PyErr_ProgramText(filename, lineno);
741 if (tmp) {
742 if (PyObject_SetAttrString(v, "text", tmp))
743 PyErr_Clear();
744 Py_DECREF(tmp);
747 if (PyObject_SetAttrString(v, "offset", Py_None)) {
748 PyErr_Clear();
750 if (exc != PyExc_SyntaxError) {
751 if (!PyObject_HasAttrString(v, "msg")) {
752 tmp = PyObject_Str(v);
753 if (tmp) {
754 if (PyObject_SetAttrString(v, "msg", tmp))
755 PyErr_Clear();
756 Py_DECREF(tmp);
757 } else {
758 PyErr_Clear();
761 if (!PyObject_HasAttrString(v, "print_file_and_line")) {
762 if (PyObject_SetAttrString(v, "print_file_and_line",
763 Py_None))
764 PyErr_Clear();
767 PyErr_Restore(exc, v, tb);
770 /* com_fetch_program_text will attempt to load the line of text that
771 the exception refers to. If it fails, it will return NULL but will
772 not set an exception.
774 XXX The functionality of this function is quite similar to the
775 functionality in tb_displayline() in traceback.c.
778 PyObject *
779 PyErr_ProgramText(const char *filename, int lineno)
781 FILE *fp;
782 int i;
783 char linebuf[1000];
785 if (filename == NULL || *filename == '\0' || lineno <= 0)
786 return NULL;
787 fp = fopen(filename, "r" PY_STDIOTEXTMODE);
788 if (fp == NULL)
789 return NULL;
790 for (i = 0; i < lineno; i++) {
791 char *pLastChar = &linebuf[sizeof(linebuf) - 2];
792 do {
793 *pLastChar = '\0';
794 if (Py_UniversalNewlineFgets(linebuf, sizeof linebuf, fp, NULL) == NULL)
795 break;
796 /* fgets read *something*; if it didn't get as
797 far as pLastChar, it must have found a newline
798 or hit the end of the file; if pLastChar is \n,
799 it obviously found a newline; else we haven't
800 yet seen a newline, so must continue */
801 } while (*pLastChar != '\0' && *pLastChar != '\n');
803 fclose(fp);
804 if (i == lineno) {
805 char *p = linebuf;
806 while (*p == ' ' || *p == '\t' || *p == '\014')
807 p++;
808 return PyString_FromString(p);
810 return NULL;
813 #ifdef __cplusplus
815 #endif