X-Git-Url: https://ruderich.org/simon/gitweb/?a=blobdiff_plain;f=src%2Fhookmacros.h;h=6f65d1e8e84ae86cd8974aea29fba3487201edd7;hb=30876b212cf0d143fcb7098b8968eb4f6c67c5a8;hp=9e1bd75fa6d60721bbf50a5709c14d5c9d38b46b;hpb=ad50ad9c54dea706b45d63dd5fdf1e381158f83f;p=coloredstderr%2Fcoloredstderr.git diff --git a/src/hookmacros.h b/src/hookmacros.h index 9e1bd75..6f65d1e 100644 --- a/src/hookmacros.h +++ b/src/hookmacros.h @@ -26,36 +26,40 @@ #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); \ - } + } \ + 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); \ - } + } \ + 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; \ + saved_errno = errno; \ handle_fd_post(fd, handle); \ 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; \ + saved_errno = errno; \ handle_file_post(file, handle); \ errno = saved_errno; \ } \ @@ -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) \