GNU gcc/ld - wrapping a call to symbol with caller and callee defined in the same object file

Question

to clarify, my question refers to wrapping/intercepting calls from one function/symbol to another function/symbol when the caller and the callee are defined in the same compilation unit with the GCC compiler and linker.

I have a situation resembling the following:

/* foo.c */ void foo(void) {   /* ... some stuff */   bar(); }  void bar(void) {   /* ... some other stuff */ } 

I would like to wrap calls to these functions, and I can do that (to a point) with ld's --wrap option (and then I implement __wrap_foo and __wrap_bar which in turn call __real_foo and __real_bar as expected by the result of ld's --wrap option).

gcc -Wl,--wrap=foo -Wl,--wrap=bar ... 

The problem I'm having is that this only takes effect for references to foo and bar from outside of this compilation unit (and resolved at link time). That is, calls to foo and bar from other functions within foo.c do not get wrapped.

I tried using objcopy --redefine-sym, but that only renames the symbols and their references.

I would like to replace calls to foo and bar (within foo.o) to __wrap_foo and __wrap_bar (just as they get resolved in other object files by the linker's --wrap option) BEFORE I pass the *.o files to the linker's --wrap options, and without having to modify foo.c's source code.

That way, the wrapping/interception takes place for all calls to foo and bar, and not just the ones taking place outside of foo.o.

Is this possible?

Answer 1

You have to weaken and globalize the symbol using objcopy.

-W symbolname --weaken-symbol=symbolname     Make symbol symbolname weak. This option may be given more than once. --globalize-symbol=symbolname     Give symbol symbolname global scoping so that it is visible outside of the file in which it is defined. This option may be given more than once. 

This worked for me

bar.c:

#include <stdio.h> int foo(){   printf("Wrap-FU\n"); } 

foo.c:

#include <stdio.h>  void foo(){ printf("foo\n"); }  int main(){ printf("main\n"); foo(); } 

Compile it

$ gcc -c foo.c bar.c  

Weaken the foo symbol and make it global, so it's available for linker again.

$ objcopy foo.o --globalize-symbol=foo --weaken-symbol=foo foo2.o 

Now you can link your new obj with the wrap from bar.c

$ gcc -o nowrap foo.o #for reference $ gcc -o wrapme foo2.o bar.o 

Test

$ ./nowrap  main foo 

And the wrapped one:

$ ./wrapme  main Wrap-FU 

Answer 2

You can use __attribute__((weak)) before the implementation of the callee in order to let someone reimplement it without GCC yelling about multiple definitons.

For example suppose you want to mock the world function in the following hello.c code unit. You can prepend the attribute in order to be able to override it.

#include "hello.h" #include <stdio.h>  __attribute__((weak)) void world(void) {     printf("world from lib\n"); }  void hello(void) {     printf("hello\n");     world(); } 

And you can then override it in another unit file. Very useful for unit testing/mocking:

#include <stdio.h> #include "hello.h"  /* overrides */ void world(void) {     printf("world from main.c"\n); }  void main(void) {     hello();     return 0; }