X-Git-Url: https://ruderich.org/simon/gitweb/?a=blobdiff_plain;f=src%2Fhookmacros.h;h=469886978e3a6048d8943f92f7a516c6bc7be4ae;hb=fe3f7d4c635b265c58efd5bc420d32135ec7903e;hp=9e1bd75fa6d60721bbf50a5709c14d5c9d38b46b;hpb=ad50ad9c54dea706b45d63dd5fdf1e381158f83f;p=coloredstderr%2Fcoloredstderr.git diff --git a/src/hookmacros.h b/src/hookmacros.h index 9e1bd75..4698869 100644 --- a/src/hookmacros.h +++ b/src/hookmacros.h @@ -26,37 +26,41 @@ #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) \ - _HOOK_PRE_FD_ALONE(type, name, fd) -#define _HOOK_PRE_FD_ALONE(type, name, fd) \ 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_ALONE(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_ALONE(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; @@ -65,11 +69,9 @@ #define HOOK_VOID1(type, name, fd, type1, arg1) \ static type (*real_ ## name)(type1); \ type name(type1 arg1) { \ - int handle; \ - DLSYM_FUNCTION(real_ ## name, #name); \ - _HOOK_PRE_FD_ALONE(type, name, fd) \ + _HOOK_PRE_FD_(type, name, fd) \ real_ ## name(arg1); \ - _HOOK_POST_FD_ALONE(fd) \ + _HOOK_POST_FD_(fd) \ } #define HOOK_FD3(type, name, fd, type1, arg1, type2, arg2, type3, arg3) \