diff options
-rw-r--r-- | subhook_unix.c | 3 | ||||
-rw-r--r-- | subhook_windows.c | 4 |
2 files changed, 7 insertions, 0 deletions
diff --git a/subhook_unix.c b/subhook_unix.c index 48c6af2..d3616d7 100644 --- a/subhook_unix.c +++ b/subhook_unix.c @@ -52,5 +52,8 @@ void *subhook_alloc_code(size_t size) { } int subhok_free_code(void *address, size_t size) { + if (address == NULL) { + return 0; + } return munmap(address, size); } diff --git a/subhook_windows.c b/subhook_windows.c index b934651..bf6d2e9 100644 --- a/subhook_windows.c +++ b/subhook_windows.c @@ -47,5 +47,9 @@ void *subhook_alloc_code(size_t size) { int subhok_free_code(void *address, size_t size) { (void)size; + + if (address == NULL) { + return 0; + } return !VirtualFree(address, 0, MEM_RELEASE); } |