Jean-Paul Calderone | fd59312 | 2013-03-03 10:21:53 -0800 | [diff] [blame] | 1 | import sys |
Jean-Paul Calderone | 68a6f8f | 2013-03-01 17:56:22 -0800 | [diff] [blame] | 2 | |
| 3 | import traceback |
| 4 | |
| 5 | from cffi import api as _api |
Alex Gaynor | c3fac9d | 2016-01-16 12:15:09 -0500 | [diff] [blame^] | 6 | |
| 7 | |
| 8 | assert sys.modules.get('ssl') is None |
| 9 | assert sys.modules.get('_hashlib') is None |
| 10 | sys.modules['ssl'] = None |
| 11 | sys.modules['_hashlib'] = None |
| 12 | |
| 13 | |
Jean-Paul Calderone | 68a6f8f | 2013-03-01 17:56:22 -0800 | [diff] [blame] | 14 | _ffi = _api.FFI() |
| 15 | _ffi.cdef( |
| 16 | """ |
| 17 | void *malloc(size_t size); |
| 18 | void free(void *ptr); |
| 19 | void *realloc(void *ptr, size_t size); |
| 20 | |
| 21 | int CRYPTO_set_mem_functions(void *(*m)(size_t),void *(*r)(void *,size_t), void (*f)(void *)); |
| 22 | |
| 23 | int backtrace(void **buffer, int size); |
| 24 | char **backtrace_symbols(void *const *buffer, int size); |
| 25 | void backtrace_symbols_fd(void *const *buffer, int size, int fd); |
Hynek Schlawack | 7b3aa90 | 2015-10-16 21:06:20 +0200 | [diff] [blame] | 26 | """) # noqa |
Jean-Paul Calderone | 68a6f8f | 2013-03-01 17:56:22 -0800 | [diff] [blame] | 27 | _api = _ffi.verify( |
| 28 | """ |
| 29 | #include <openssl/crypto.h> |
| 30 | #include <stdlib.h> |
| 31 | #include <execinfo.h> |
| 32 | """, libraries=["crypto"]) |
| 33 | C = _ffi.dlopen(None) |
| 34 | |
Jean-Paul Calderone | fd59312 | 2013-03-03 10:21:53 -0800 | [diff] [blame] | 35 | verbose = False |
| 36 | |
Hynek Schlawack | 7b3aa90 | 2015-10-16 21:06:20 +0200 | [diff] [blame] | 37 | |
Jean-Paul Calderone | fd59312 | 2013-03-03 10:21:53 -0800 | [diff] [blame] | 38 | def log(s): |
| 39 | if verbose: |
Jean-Paul Calderone | 558b78a | 2014-01-10 14:08:50 -0500 | [diff] [blame] | 40 | print(s) |
Jean-Paul Calderone | 68a6f8f | 2013-03-01 17:56:22 -0800 | [diff] [blame] | 41 | |
Hynek Schlawack | 7b3aa90 | 2015-10-16 21:06:20 +0200 | [diff] [blame] | 42 | |
Jean-Paul Calderone | 68a6f8f | 2013-03-01 17:56:22 -0800 | [diff] [blame] | 43 | def _backtrace(): |
| 44 | buf = _ffi.new("void*[]", 64) |
| 45 | result = _api.backtrace(buf, len(buf)) |
| 46 | strings = _api.backtrace_symbols(buf, result) |
| 47 | stack = [_ffi.string(strings[i]) for i in range(result)] |
| 48 | C.free(strings) |
| 49 | return stack |
| 50 | |
| 51 | |
Jean-Paul Calderone | fd59312 | 2013-03-03 10:21:53 -0800 | [diff] [blame] | 52 | @_ffi.callback("void*(*)(size_t)") |
Jean-Paul Calderone | ede8b71 | 2014-01-01 12:36:46 -0500 | [diff] [blame] | 53 | def malloc(n): |
Jean-Paul Calderone | 68a6f8f | 2013-03-01 17:56:22 -0800 | [diff] [blame] | 54 | memory = C.malloc(n) |
| 55 | python_stack = traceback.extract_stack(limit=3) |
| 56 | c_stack = _backtrace() |
Jean-Paul Calderone | c2e8b41 | 2013-03-02 16:27:55 -0800 | [diff] [blame] | 57 | heap[memory] = [(n, python_stack, c_stack)] |
Jean-Paul Calderone | fd59312 | 2013-03-03 10:21:53 -0800 | [diff] [blame] | 58 | log("malloc(%d) -> %s" % (n, memory)) |
Jean-Paul Calderone | 68a6f8f | 2013-03-01 17:56:22 -0800 | [diff] [blame] | 59 | return memory |
Jean-Paul Calderone | 68a6f8f | 2013-03-01 17:56:22 -0800 | [diff] [blame] | 60 | |
| 61 | |
| 62 | @_ffi.callback("void*(*)(void*, size_t)") |
Jean-Paul Calderone | ede8b71 | 2014-01-01 12:36:46 -0500 | [diff] [blame] | 63 | def realloc(p, n): |
Jean-Paul Calderone | 68a6f8f | 2013-03-01 17:56:22 -0800 | [diff] [blame] | 64 | memory = C.realloc(p, n) |
| 65 | old = heap.pop(p) |
| 66 | |
| 67 | python_stack = traceback.extract_stack(limit=3) |
| 68 | c_stack = _backtrace() |
| 69 | |
Jean-Paul Calderone | c2e8b41 | 2013-03-02 16:27:55 -0800 | [diff] [blame] | 70 | old.append((n, python_stack, c_stack)) |
Jean-Paul Calderone | 68a6f8f | 2013-03-01 17:56:22 -0800 | [diff] [blame] | 71 | heap[memory] = old |
Jean-Paul Calderone | fd59312 | 2013-03-03 10:21:53 -0800 | [diff] [blame] | 72 | log("realloc(0x%x, %d) -> %s" % (int(_ffi.cast("int", p)), n, memory)) |
Jean-Paul Calderone | 68a6f8f | 2013-03-01 17:56:22 -0800 | [diff] [blame] | 73 | return memory |
| 74 | |
| 75 | |
| 76 | @_ffi.callback("void(*)(void*)") |
Jean-Paul Calderone | ede8b71 | 2014-01-01 12:36:46 -0500 | [diff] [blame] | 77 | def free(p): |
Jean-Paul Calderone | 68a6f8f | 2013-03-01 17:56:22 -0800 | [diff] [blame] | 78 | if p != _ffi.NULL: |
| 79 | C.free(p) |
| 80 | del heap[p] |
Jean-Paul Calderone | fd59312 | 2013-03-03 10:21:53 -0800 | [diff] [blame] | 81 | log("free(0x%x)" % (int(_ffi.cast("int", p)),)) |
Jean-Paul Calderone | 68a6f8f | 2013-03-01 17:56:22 -0800 | [diff] [blame] | 82 | |
| 83 | |
Jean-Paul Calderone | ede8b71 | 2014-01-01 12:36:46 -0500 | [diff] [blame] | 84 | if _api.CRYPTO_set_mem_functions(malloc, realloc, free): |
Jean-Paul Calderone | 558b78a | 2014-01-10 14:08:50 -0500 | [diff] [blame] | 85 | log('Enabled memory debugging') |
Jean-Paul Calderone | 8e4a663 | 2013-03-03 17:30:27 -0800 | [diff] [blame] | 86 | heap = {} |
Jean-Paul Calderone | 68a6f8f | 2013-03-01 17:56:22 -0800 | [diff] [blame] | 87 | else: |
Jean-Paul Calderone | 558b78a | 2014-01-10 14:08:50 -0500 | [diff] [blame] | 88 | log('Failed to enable memory debugging') |
Jean-Paul Calderone | 8e4a663 | 2013-03-03 17:30:27 -0800 | [diff] [blame] | 89 | heap = None |