X-Git-Url: https://ruderich.org/simon/gitweb/?a=blobdiff_plain;f=src%2Fhookmacros.h;h=469886978e3a6048d8943f92f7a516c6bc7be4ae;hb=fe3f7d4c635b265c58efd5bc420d32135ec7903e;hp=ffbc6b6de457a7b45aa2a31c285fb03fdff3d873;hpb=6d15f6c9902c1ac879ce0dee201ff8c6c66afaf1;p=coloredstderr%2Fcoloredstderr.git

diff --git a/src/hookmacros.h b/src/hookmacros.h
index ffbc6b6..4698869 100644
--- a/src/hookmacros.h
+++ b/src/hookmacros.h
@@ -26,38 +26,54 @@
 
 #define _HOOK_PRE(type, name) \
         int handle; \
-        type result; \
+        int saved_errno = errno; \
         DLSYM_FUNCTION(real_ ## name, #name);
 #define _HOOK_PRE_FD(type, name, fd) \
+        type result; \
+        _HOOK_PRE_FD_(type, name, fd)
+#define _HOOK_PRE_FD_(type, name, fd) \
         _HOOK_PRE(type, name) \
         handle = check_handle_fd(fd); \
         if (handle) { \
-            handle_fd_pre(fd, handle); \
-        }
+            handle_fd_pre(fd); \
+        } \
+        errno = saved_errno;
 #define _HOOK_PRE_FILE(type, name, file) \
+        type result; \
         _HOOK_PRE(type, name) \
         handle = check_handle_fd(fileno(file)); \
         if (handle) { \
-            handle_file_pre(file, handle); \
-        }
+            handle_file_pre(file); \
+        } \
+        errno = saved_errno;
 /* Save and restore the errno to make sure we return the errno of the original
  * function call. */
-#define _HOOK_POST_FD(fd) \
+#define _HOOK_POST_FD_(fd) \
         if (handle) { \
-            int saved_errno = errno; \
-            handle_fd_post(fd, handle); \
+            saved_errno = errno; \
+            handle_fd_post(fd); \
             errno = saved_errno; \
-        } \
+        }
+#define _HOOK_POST_FD(fd) \
+        _HOOK_POST_FD_(fd) \
         return result;
 #define _HOOK_POST_FILE(file) \
         if (handle) { \
-            int saved_errno = errno; \
-            handle_file_post(file, handle); \
+            saved_errno = errno; \
+            handle_file_post(file); \
             errno = saved_errno; \
         } \
         return result;
 
 
+#define HOOK_VOID1(type, name, fd, type1, arg1) \
+    static type (*real_ ## name)(type1); \
+    type name(type1 arg1) { \
+        _HOOK_PRE_FD_(type, name, fd) \
+        real_ ## name(arg1); \
+        _HOOK_POST_FD_(fd) \
+    }
+
 #define HOOK_FD3(type, name, fd, type1, arg1, type2, arg2, type3, arg3) \
     static type (*real_ ## name)(type1, type2, type3); \
     type name(type1 arg1, type2 arg2, type3 arg3) { \