| /* | |
| --------------------------------------------------------------------- | |
| / Copyright (c) 1996. \ | |
| | The Regents of the University of California. | | |
| | All rights reserved. | | |
| | | | |
| | Permission to use, copy, modify, and distribute this software for | | |
| | any purpose without fee is hereby granted, provided that this en- | | |
| | tire notice is included in all copies of any software which is or | | |
| | includes a copy or modification of this software and in all | | |
| | copies of the supporting documentation for such software. | | |
| | | | |
| | This work was produced at the University of California, Lawrence | | |
| | Livermore National Laboratory under contract no. W-7405-ENG-48 | | |
| | between the U.S. Department of Energy and The Regents of the | | |
| | University of California for the operation of UC LLNL. | | |
| | | | |
| | DISCLAIMER | | |
| | | | |
| | This software was prepared as an account of work sponsored by an | | |
| | agency of the United States Government. Neither the United States | | |
| | Government nor the University of California nor any of their em- | | |
| | ployees, makes any warranty, express or implied, or assumes any | | |
| | liability or responsibility for the accuracy, completeness, or | | |
| | usefulness of any information, apparatus, product, or process | | |
| | disclosed, or represents that its use would not infringe | | |
| | privately-owned rights. Reference herein to any specific commer- | | |
| | cial products, process, or service by trade name, trademark, | | |
| | manufacturer, or otherwise, does not necessarily constitute or | | |
| | imply its endorsement, recommendation, or favoring by the United | | |
| | States Government or the University of California. The views and | | |
| | opinions of authors expressed herein do not necessarily state or | | |
| | reflect those of the United States Government or the University | | |
| | of California, and shall not be used for advertising or product | | |
| \ endorsement purposes. / | |
| --------------------------------------------------------------------- | |
| */ | |
| /* | |
| Floating point exception control module. | |
| This Python module provides bare-bones control over floating point | |
| units from several hardware manufacturers. Specifically, it allows | |
| the user to turn on the generation of SIGFPE whenever any of the | |
| three serious IEEE 754 exceptions (Division by Zero, Overflow, | |
| Invalid Operation) occurs. We currently ignore Underflow and | |
| Inexact Result exceptions, although those could certainly be added | |
| if desired. | |
| The module also establishes a signal handler for SIGFPE during | |
| initialization. This builds on code found in the Python | |
| distribution at Include/pyfpe.h and Python/pyfpe.c. If those files | |
| are not in your Python distribution, find them in a patch at | |
| ftp://icf.llnl.gov/pub/python/busby/patches.961108.tgz. | |
| This module is only useful to you if it happens to include code | |
| specific for your hardware and software environment. If you can | |
| contribute OS-specific code for new platforms, or corrections for | |
| the code provided, it will be greatly appreciated. | |
| ** Version 1.0: September 20, 1996. Lee Busby, LLNL. | |
| */ | |
| #ifdef __cplusplus | |
| extern "C" { | |
| #endif | |
| #include "Python.h" | |
| #include <signal.h> | |
| #if defined(__FreeBSD__) | |
| # include <ieeefp.h> | |
| #elif defined(__VMS) | |
| #define __NEW_STARLET | |
| #include <starlet.h> | |
| #include <ieeedef.h> | |
| #endif | |
| #ifndef WANT_SIGFPE_HANDLER | |
| /* Define locally if they are not defined in Python. This gives only | |
| * the limited control to induce a core dump in case of an exception. | |
| */ | |
| #include <setjmp.h> | |
| static jmp_buf PyFPE_jbuf; | |
| static int PyFPE_counter = 0; | |
| #endif | |
| typedef void Sigfunc(int); | |
| static Sigfunc sigfpe_handler; | |
| static void fpe_reset(Sigfunc *); | |
| static PyObject *fpe_error; | |
| PyMODINIT_FUNC initfpectl(void); | |
| static PyObject *turnon_sigfpe (PyObject *self,PyObject *args); | |
| static PyObject *turnoff_sigfpe (PyObject *self,PyObject *args); | |
| static PyMethodDef fpectl_methods[] = { | |
| {"turnon_sigfpe", (PyCFunction) turnon_sigfpe, METH_VARARGS}, | |
| {"turnoff_sigfpe", (PyCFunction) turnoff_sigfpe, METH_VARARGS}, | |
| {0,0} | |
| }; | |
| static PyObject *turnon_sigfpe(PyObject *self,PyObject *args) | |
| { | |
| /* Do any architecture-specific one-time only initialization here. */ | |
| fpe_reset(sigfpe_handler); | |
| Py_INCREF (Py_None); | |
| return Py_None; | |
| } | |
| static void fpe_reset(Sigfunc *handler) | |
| { | |
| /* Reset the exception handling machinery, and reset the signal | |
| * handler for SIGFPE to the given handler. | |
| */ | |
| /*-- IRIX -----------------------------------------------------------------*/ | |
| #if defined(sgi) | |
| /* See man page on handle_sigfpes -- must link with -lfpe | |
| * My usage doesn't follow the man page exactly. Maybe somebody | |
| * else can explain handle_sigfpes to me.... | |
| * cc -c -I/usr/local/python/include fpectlmodule.c | |
| * ld -shared -o fpectlmodule.so fpectlmodule.o -lfpe | |
| */ | |
| #include <sigfpe.h> | |
| typedef void user_routine (unsigned[5], int[2]); | |
| typedef void abort_routine (unsigned long); | |
| handle_sigfpes(_OFF, 0, | |
| (user_routine *)0, | |
| _TURN_OFF_HANDLER_ON_ERROR, | |
| NULL); | |
| handle_sigfpes(_ON, _EN_OVERFL | _EN_DIVZERO | _EN_INVALID, | |
| (user_routine *)0, | |
| _ABORT_ON_ERROR, | |
| NULL); | |
| PyOS_setsig(SIGFPE, handler); | |
| /*-- SunOS and Solaris ----------------------------------------------------*/ | |
| #elif defined(sun) | |
| /* References: ieee_handler, ieee_sun, ieee_functions, and ieee_flags | |
| man pages (SunOS or Solaris) | |
| cc -c -I/usr/local/python/include fpectlmodule.c | |
| ld -G -o fpectlmodule.so -L/opt/SUNWspro/lib fpectlmodule.o -lsunmath -lm | |
| */ | |
| #include <math.h> | |
| #ifndef _SUNMATH_H | |
| extern void nonstandard_arithmetic(void); | |
| extern int ieee_flags(const char*, const char*, const char*, char **); | |
| extern long ieee_handler(const char*, const char*, sigfpe_handler_type); | |
| #endif | |
| char *mode="exception", *in="all", *out; | |
| (void) nonstandard_arithmetic(); | |
| (void) ieee_flags("clearall",mode,in,&out); | |
| (void) ieee_handler("set","common",(sigfpe_handler_type)handler); | |
| PyOS_setsig(SIGFPE, handler); | |
| /*-- HPUX -----------------------------------------------------------------*/ | |
| #elif defined(__hppa) || defined(hppa) | |
| /* References: fpsetmask man page */ | |
| /* cc -Aa +z -c -I/usr/local/python/include fpectlmodule.c */ | |
| /* ld -b -o fpectlmodule.sl fpectlmodule.o -lm */ | |
| #include <math.h> | |
| fpsetdefaults(); | |
| PyOS_setsig(SIGFPE, handler); | |
| /*-- IBM AIX --------------------------------------------------------------*/ | |
| #elif defined(__AIX) || defined(_AIX) | |
| /* References: fp_trap, fp_enable man pages */ | |
| #include <fptrap.h> | |
| fp_trap(FP_TRAP_SYNC); | |
| fp_enable(TRP_INVALID | TRP_DIV_BY_ZERO | TRP_OVERFLOW); | |
| PyOS_setsig(SIGFPE, handler); | |
| /*-- DEC ALPHA OSF --------------------------------------------------------*/ | |
| #elif defined(__alpha) && defined(__osf__) | |
| /* References: exception_intro, ieee man pages */ | |
| /* cc -c -I/usr/local/python/include fpectlmodule.c */ | |
| /* ld -shared -o fpectlmodule.so fpectlmodule.o */ | |
| #include <machine/fpu.h> | |
| unsigned long fp_control = | |
| IEEE_TRAP_ENABLE_INV | IEEE_TRAP_ENABLE_DZE | IEEE_TRAP_ENABLE_OVF; | |
| ieee_set_fp_control(fp_control); | |
| PyOS_setsig(SIGFPE, handler); | |
| /*-- DEC ALPHA LINUX ------------------------------------------------------*/ | |
| #elif defined(__alpha) && defined(linux) | |
| #include <asm/fpu.h> | |
| unsigned long fp_control = | |
| IEEE_TRAP_ENABLE_INV | IEEE_TRAP_ENABLE_DZE | IEEE_TRAP_ENABLE_OVF; | |
| ieee_set_fp_control(fp_control); | |
| PyOS_setsig(SIGFPE, handler); | |
| /*-- DEC ALPHA VMS --------------------------------------------------------*/ | |
| #elif defined(__ALPHA) && defined(__VMS) | |
| IEEE clrmsk; | |
| IEEE setmsk; | |
| clrmsk.ieee$q_flags = | |
| IEEE$M_TRAP_ENABLE_UNF | IEEE$M_TRAP_ENABLE_INE | | |
| IEEE$M_MAP_UMZ; | |
| setmsk.ieee$q_flags = | |
| IEEE$M_TRAP_ENABLE_INV | IEEE$M_TRAP_ENABLE_DZE | | |
| IEEE$M_TRAP_ENABLE_OVF; | |
| sys$ieee_set_fp_control(&clrmsk, &setmsk, 0); | |
| PyOS_setsig(SIGFPE, handler); | |
| /*-- HP IA64 VMS --------------------------------------------------------*/ | |
| #elif defined(__ia64) && defined(__VMS) | |
| PyOS_setsig(SIGFPE, handler); | |
| /*-- Cray Unicos ----------------------------------------------------------*/ | |
| #elif defined(cray) | |
| /* UNICOS delivers SIGFPE by default, but no matherr */ | |
| #ifdef HAS_LIBMSET | |
| libmset(-1); | |
| #endif | |
| PyOS_setsig(SIGFPE, handler); | |
| /*-- FreeBSD ----------------------------------------------------------------*/ | |
| #elif defined(__FreeBSD__) | |
| fpresetsticky(fpgetsticky()); | |
| fpsetmask(FP_X_INV | FP_X_DZ | FP_X_OFL); | |
| PyOS_setsig(SIGFPE, handler); | |
| /*-- Linux ----------------------------------------------------------------*/ | |
| #elif defined(linux) | |
| #ifdef __GLIBC__ | |
| #include <fpu_control.h> | |
| #else | |
| #include <i386/fpu_control.h> | |
| #endif | |
| #ifdef _FPU_SETCW | |
| { | |
| fpu_control_t cw = 0x1372; | |
| _FPU_SETCW(cw); | |
| } | |
| #else | |
| __setfpucw(0x1372); | |
| #endif | |
| PyOS_setsig(SIGFPE, handler); | |
| /*-- Microsoft Windows, NT ------------------------------------------------*/ | |
| #elif defined(_MSC_VER) | |
| /* Reference: Visual C++ Books Online 4.2, | |
| Run-Time Library Reference, _control87, _controlfp */ | |
| #include <float.h> | |
| unsigned int cw = _EM_INVALID | _EM_ZERODIVIDE | _EM_OVERFLOW; | |
| (void)_controlfp(0, cw); | |
| PyOS_setsig(SIGFPE, handler); | |
| /*-- Give Up --------------------------------------------------------------*/ | |
| #else | |
| fputs("Operation not implemented\n", stderr); | |
| #endif | |
| } | |
| static PyObject *turnoff_sigfpe(PyObject *self,PyObject *args) | |
| { | |
| #ifdef __FreeBSD__ | |
| fpresetsticky(fpgetsticky()); | |
| fpsetmask(0); | |
| #elif defined(__VMS) | |
| IEEE clrmsk; | |
| clrmsk.ieee$q_flags = | |
| IEEE$M_TRAP_ENABLE_UNF | IEEE$M_TRAP_ENABLE_INE | | |
| IEEE$M_MAP_UMZ | IEEE$M_TRAP_ENABLE_INV | | |
| IEEE$M_TRAP_ENABLE_DZE | IEEE$M_TRAP_ENABLE_OVF | | |
| IEEE$M_INHERIT; | |
| sys$ieee_set_fp_control(&clrmsk, 0, 0); | |
| #else | |
| fputs("Operation not implemented\n", stderr); | |
| #endif | |
| Py_INCREF(Py_None); | |
| return Py_None; | |
| } | |
| static void sigfpe_handler(int signo) | |
| { | |
| fpe_reset(sigfpe_handler); | |
| if(PyFPE_counter) { | |
| longjmp(PyFPE_jbuf, 1); | |
| } else { | |
| Py_FatalError("Unprotected floating point exception"); | |
| } | |
| } | |
| PyMODINIT_FUNC initfpectl(void) | |
| { | |
| PyObject *m, *d; | |
| m = Py_InitModule("fpectl", fpectl_methods); | |
| if (m == NULL) | |
| return; | |
| d = PyModule_GetDict(m); | |
| fpe_error = PyErr_NewException("fpectl.error", NULL, NULL); | |
| if (fpe_error != NULL) | |
| PyDict_SetItemString(d, "error", fpe_error); | |
| } | |
| #ifdef __cplusplus | |
| } | |
| #endif |