From a976efe67c5f936eb553a0d61b78a311da7224d9 Mon Sep 17 00:00:00 2001
From: comex
Date: Sat, 28 Feb 2015 21:54:23 -0500
Subject: Add extra argument to substitute_hook_functions and interpose_imports
 for use with unhooking.

I think we'll need to explicitly record trampoline locations to avoid
the possibility of stomping on someone else's stuff if the function was
re-patched...

Also, document substitute_hook_functions.
---
 lib/hook-functions.c | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

(limited to 'lib/hook-functions.c')

diff --git a/lib/hook-functions.c b/lib/hook-functions.c
index 6d53a0d..2a8d618 100644
--- a/lib/hook-functions.c
+++ b/lib/hook-functions.c
@@ -127,11 +127,16 @@ end:
 
 EXPORT
 int substitute_hook_functions(const struct substitute_function_hook *hooks,
-                              size_t nhooks, int options) {
+                              size_t nhooks,
+                              struct substitute_function_hook_record **recordp,
+                              int options) {
     bool thread_safe = !(options & SUBSTITUTE_NO_THREAD_SAFETY);
     if (thread_safe && !pthread_main_np())
         return SUBSTITUTE_ERR_NOT_ON_MAIN_THREAD;
 
+    if (recordp)
+        *recordp = NULL;
+
     struct execmem_foreign_write *fws;
     struct hook_internal *his = malloc(nhooks * sizeof(*his) +
                                        nhooks + sizeof(*fws));
-- 
cgit v1.2.3