everything is patched (i think)

Author: ZeroIntensity

README.md

@@ -35,7 +35,7 @@ free(ptr)
 -   Fully type safe
 -   Pythonic pointer API
 -   Bindings for the entire C standard library
--   Segfaults
+-   Undefined behavior
 
 ### Why does this exist?
 

src/mod.c

@@ -6,6 +6,10 @@
 #include <stdio.h>
 #include <frameobject.h> // needed to get members of PyFrameObject
 #define GETOBJ PyObject* obj; if (!PyArg_ParseTuple(args, "O", &obj)) return NULL
+#define INIT_HANDLER(handle, msg) if (signal(SIGSEGV, handle) == SIG_ERR) { \
+        PyErr_SetString(PyExc_ImportError, msg); \
+        return NULL; \
+    }
 static jmp_buf buf;
 
 static PyObject* add_ref(PyObject* self, PyObject* args) {
@@ -41,10 +45,14 @@ static PyObject* force_set_attr(PyObject* self, PyObject* args) {
     Py_RETURN_NONE;
 }
 
-void handler(int signum) {
+static void sigsegv_handler(int signum) {
     longjmp(buf, 1);
 }
 
+static void sigiot_handler(int signum) {
+    longjmp(buf, 2);
+}
+
 static PyObject* handle(PyObject* self, PyObject* args) {
     PyObject* func;
     PyObject* params = NULL;
@@ -63,21 +71,31 @@ static PyObject* handle(PyObject* self, PyObject* args) {
 
     if (!params) params = PyTuple_New(0);
     if (!kwargs) kwargs = PyDict_New();
+    int val = setjmp(buf);
 
-    if (setjmp(buf)) {
+    if (val) {
         PyFrameObject* frame = PyEval_GetFrame();
-        PyCodeObject* code = frame->f_code;
-        Py_INCREF(code);
+        PyObject* name;
+        PyCodeObject* code = NULL;
+
+        if (frame) {
+            code = frame->f_code;
+            Py_INCREF(code);
+            name = code->co_name;
+        } else {
+            name = PyObject_GetAttrString(func, "__name__");
+        }
 
         // this is basically a copy of PyFrame_GetCode, which is only available on 3.9+
 
         PyErr_Format(
             PyExc_RuntimeError,
-            "segment violation occured during execution of %S",
-            code->co_name
+            "%s occured during execution of %S",
+            val == 1 ? "segment violation" : "python aborted",
+            name
         );
 
-        Py_DECREF(code);
+        if (code) Py_DECREF(code);
         return NULL;
     }
 
@@ -105,10 +123,14 @@ static struct PyModuleDef module = {
 };
 
 PyMODINIT_FUNC PyInit__pointers(void) {
-    if (signal(SIGSEGV, handler) == SIG_ERR) {
-        PyErr_SetString(PyExc_RuntimeError, "failed to setup SIGSEGV handler");
-        return NULL;
-    }
+    INIT_HANDLER(
+        sigiot_handler,
+        "cant load _pointers: failed to setup SIGIOT handler"
+    );
+    INIT_HANDLER(
+        sigsegv_handler,
+        "cant load _pointers: failed to setup SIGSEGV handler"
+    );
 
     return PyModule_Create(&module);
 }

src/pointers/__init__.py

@@ -18,8 +18,9 @@
 from .custom_binding import binding, binds
 from .decay import decay, decay_annotated, decay_wrapped
 from .exceptions import (
-    AllocationError, DereferenceError, FreedMemoryError,
-    InvalidBindingParameter, InvalidSizeError
+    Aborted, AllocationError, DereferenceError, FreedMemoryError,
+    InvalidBindingParameter, InvalidSizeError, NullPointerError,
+    SegmentViolation
 )
 from .magic import _
 from .malloc import AllocatedPointer, free, malloc, realloc

src/pointers/base_pointers.py

@@ -1,11 +1,9 @@
 import ctypes
-import faulthandler
 import sys
 import warnings
 import weakref
 from abc import ABC, abstractmethod
 from contextlib import suppress
-from io import UnsupportedOperation
 from typing import (
     Any, Generic, Iterator, Optional, Tuple, Type, TypeVar, Union
 )
@@ -33,11 +31,6 @@
 T = TypeVar("T")
 A = TypeVar("A", bound="BasicPointer")
 
-with suppress(
-    UnsupportedOperation
-):  # in case its running in idle or something like that
-    faulthandler.enable()
-
 
 class BasicPointer(ABC):
     """Base class representing a pointer with no operations."""

src/pointers/c_pointer.py

@@ -39,7 +39,6 @@ def _as_parameter_(self) -> ctypes.c_void_p:
 
     @handle
     def dereference(self) -> Optional[int]:
-        """Dereference the pointer."""
         deref = ctypes.c_void_p.from_address(self.ensure())
         return deref.value
 

src/pointers/constants.py

@@ -9,7 +9,7 @@
 
 from _pointers import handle as _handle
 
-from .exceptions import SegmentViolation
+from .exceptions import Aborted, SegmentViolation
 
 with suppress(
     UnsupportedOperation
@@ -56,15 +56,23 @@ def wrapper(*args: P.args, **kwargs: P.kwargs) -> T:
             return call
         except (RuntimeError, OSError) as e:
             msg = str(e)
-
-            if not msg.startswith("segment violation") and not msg.startswith(
-                "access violation"
-            ):
+            segv = any(
+                {
+                    msg.startswith("segment violation"),
+                    msg.startswith("access violation"),
+                }
+            )
+
+            aborted = msg.startswith(
+                "python aborted",
+            )
+
+            if (not segv) and (not aborted):
                 raise
 
             with suppress(UnsupportedOperation):
                 faulthandler.enable()
 
-            raise SegmentViolation(str(e)) from None
+            raise (SegmentViolation if segv else Aborted)(str(e)) from None
 
     return wrapper

src/pointers/exceptions.py

@@ -7,6 +7,7 @@
     "NullPointerError",
     "InvalidVersionError",
     "SegmentViolation",
+    "Aborted",
 )
 
 
@@ -56,3 +57,9 @@ class SegmentViolation(Exception):
     """SIGSEGV was sent to Python."""
 
     pass
+
+
+class Aborted(Exception):
+    """SIGIOT/SIGABRT was sent to Python."""
+
+    pass

tests/test_pointer.py

@@ -2,10 +2,10 @@
 
 from ward import raises, test
 
-from _pointers import handle
-from pointers import NULL, InvalidSizeError, Pointer
+from _pointers import add_ref
+from pointers import NULL, InvalidSizeError, Pointer, SegmentViolation
 from pointers import _ as m
-from pointers import strlen, to_c_ptr, to_ptr
+from pointers import handle, to_c_ptr, to_ptr
 from pointers.exceptions import NullPointerError
 
 
@@ -37,14 +37,14 @@ def _():
 
 @test("movement")
 def _():
-    a = "teststring"
+    a = 763723
     ptr = to_ptr(a)
-    ptr <<= "hi"
+    ptr <<= 2
 
-    assert a == "hi"
+    assert a == 2
 
     with raises(InvalidSizeError):
-        ptr <<= "hello world"
+        ptr <<= 758347580937450893
 
 
 @test("assignment with tracked types")
@@ -78,7 +78,7 @@ def _():
     with raises(NullPointerError):
         print(~to_ptr(NULL))
 
-    ptr2 = to_ptr(NULL)
+    ptr2: Pointer[int] = to_ptr(NULL)
     ptr2 >>= 1
 
     ptr2 >>= NULL
@@ -93,8 +93,9 @@ def _():
 
 @test("segfault handler")
 def _():
+    @handle
     def segfault():
         ctypes.string_at(0)
 
-    with raises(RuntimeError):
-        handle(segfault)
+    with raises(SegmentViolation):
+        segfault()