Add stubs for pvalloc

`pvalloc` is a `glibc` extension as is not always available in the
system. To simplify the test suite, add stubs for `pvalloc` if is not
defined and skip the tests that involve this allocator if is not
available.

Signed-off-by: Pablo Galindo <pablogsal@gmail.com>

src/memray/_memray/alloc.h

@@ -0,0 +1,12 @@
+#include <malloc.h>
+#include <stdlib.h>
+
+#ifndef __GLIBC__
+extern "C" {
+void*
+pvalloc(size_t size)
+{
+    return NULL;
+}
+}
+#endif

src/memray/_memray/alloc.pxd

@@ -1,11 +1,9 @@
-cdef extern from "<stdlib.h>" nogil:
+cdef extern from "alloc.h" nogil:
     void *calloc (size_t count, size_t eltsize)
     void free (void *ptr)
     void *malloc (size_t size)
     int posix_memalign(void** memptr, size_t alignment, size_t size)
     void *realloc (void *ptr, size_t newsize)
     void* valloc(size_t size)
-
-cdef extern from "<malloc.h>" nogil:
     void* memalign(size_t alignment, size_t size)
     void* pvalloc(size_t size)

src/memray/_memray_test_utils.pyx

@@ -52,31 +52,38 @@ cdef class MemoryAllocator:
     @cython.profile(True)
     def malloc(self, size_t size):
         self.ptr = malloc(size)
+        return self.ptr != NULL
 
     @cython.profile(True)
     def calloc(self, size_t size):
         self.ptr = calloc(1, size)
+        return self.ptr != NULL
 
     @cython.profile(True)
     def realloc(self, size_t size):
         self.ptr = malloc(1)
         self.ptr = realloc(self.ptr, size)
+        return self.ptr != NULL
 
     @cython.profile(True)
     def posix_memalign(self, size_t size):
         posix_memalign(&self.ptr, sizeof(void*), size)
+        return self.ptr != NULL
 
     @cython.profile(True)
     def memalign(self, size_t size):
         self.ptr = memalign(sizeof(void*), size)
+        return self.ptr != NULL
 
     @cython.profile(True)
     def valloc(self, size_t size):
         self.ptr = valloc(size)
+        return self.ptr != NULL
 
     @cython.profile(True)
     def pvalloc(self, size_t size):
         self.ptr = pvalloc(size)
+        return self.ptr != NULL
 
     @cython.profile(True)
     def run_in_pthread(self, callback):

tests/integration/multithreaded_extension/testext.cpp

@@ -5,6 +5,9 @@
 #include <pthread.h>
 #include <malloc.h>
 
+#pragma GCC push_options
+#pragma GCC optimize ("O0")
+
 const int NUM_THREADS = 100;
 const int NUM_BUFFERS = 100;
 pthread_t threads[NUM_THREADS];
@@ -74,7 +77,7 @@ run_valloc_at_exit(PyObject*, PyObject*)
     Py_RETURN_NONE;
 }
 
-
+#pragma GCC pop_options
 
 static PyMethodDef methods[] = {
         {"run", run, METH_NOARGS, "Run a bunch of threads"},

tests/integration/test_tracing.py

@@ -141,14 +141,14 @@ def test_cython_traceback(tmpdir):
 
     traceback = list(alloc1.stack_trace())
     assert traceback[-3:] == [
-        ("valloc", ANY, 74),
-        ("_cython_nested_allocation", ANY, 92),
+        ("valloc", ANY, 79),
+        ("_cython_nested_allocation", ANY, 99),
         ("test_cython_traceback", ANY, 132),
     ]
 
     traceback = list(alloc2.stack_trace())
     assert traceback[-3:] == [
-        ("_cython_nested_allocation", ANY, 92),
+        ("_cython_nested_allocation", ANY, 99),
         ("test_cython_traceback", ANY, 132),
     ]
 

tests/integration/test_tracking.py

@@ -47,8 +47,12 @@ def test_simple_allocation_tracking(allocator_func, allocator_type, tmp_path):
 
     # WHEN
     with Tracker(output):
-        getattr(allocator, allocator_func)(1234)
-        allocator.free()
+        res = getattr(allocator, allocator_func)(1234)
+        if res:
+            allocator.free()
+
+    if not res:
+        pytest.skip("Allocator {allocator_func} not supported in this platform")
 
     # THEN
     allocations = list(FileReader(output).get_allocation_records())
@@ -188,8 +192,12 @@ def test_simple_allocation_tracking(self, tmp_path, allocator_func, allocator_ty
 
         # WHEN
         with Tracker(output):
-            getattr(allocator, allocator_func)(1234)
-            allocator.free()
+            res = getattr(allocator, allocator_func)(1234)
+            if res:
+                allocator.free()
+
+        if not res:
+            pytest.skip("Allocator {allocator_func} not supported in this platform")
 
         # THEN
         peak_allocations_unfiltered = FileReader(