Akkariin/proxychains-ng
1
0
Fork 0
mirror of https://github.com/rofl0r/proxychains-ng synced 2025-01-23 09:33:00 +08:00
Code Issues Releases Wiki Activity

simplify load_sym code and remove potential aliasing violation.

Browse Source
This commit is contained in:
rofl0r 2012-07-08 00:27:46 +02:00
parent d95ef42d48
commit 2c9c4d9da7
1 changed files with 15 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
src/libproxychains.c
View File

@ -70,49 +70,41 @@ static int init_l = 0;
static inline void get_chain_data(proxy_data * pd, unsigned int *proxy_count, chain_type * ct); static inline void get_chain_data(proxy_data * pd, unsigned int *proxy_count, chain_type * ct);
static void load_sym(void** funcptr, char* symname, void* proxyfunc) { static void* load_sym(char* symname, void* proxyfunc) {
*funcptr = dlsym(RTLD_NEXT, symname); void *funcptr = dlsym(RTLD_NEXT, symname);
if(!(*funcptr)) { if(!funcptr) {
fprintf(stderr, "Cannot load symbol '%s' %s\n", symname, dlerror()); fprintf(stderr, "Cannot load symbol '%s' %s\n", symname, dlerror());
exit(1); exit(1);
} else { } else {
PDEBUG("loaded symbol '%s'" " real addr %p wrapped addr %p\n", symname, (*funcptr), proxyfunc); PDEBUG("loaded symbol '%s'" " real addr %p wrapped addr %p\n", symname, (*funcptr), proxyfunc);
} }
if((*funcptr) == proxyfunc) { if(funcptr == proxyfunc) {
PDEBUG("circular reference detected, aborting!\n"); PDEBUG("circular reference detected, aborting!\n");
abort(); abort();
} }
return funcptr;
} }
#define INIT() init_lib_wrapper(__FUNCTION__) #define INIT() init_lib_wrapper(__FUNCTION__)
#define SETUP_SYM(X) do { true_ ## X = load_sym( # X, X ); } while(0)
static void do_init(void) { static void do_init(void) {
static const struct override_info {
void* funcptr;
char* symname;
void* proxyfunc;
} override_symbols[] = {
#define SYM_ENTRY(X) { .funcptr = &true_ ## X, .symname = # X, .proxyfunc = X,}
SYM_ENTRY(connect),
SYM_ENTRY(gethostbyname),
SYM_ENTRY(getaddrinfo),
SYM_ENTRY(freeaddrinfo),
SYM_ENTRY(gethostbyaddr),
SYM_ENTRY(getnameinfo),
#undef SYM_ENTRY
};
unsigned i;
MUTEX_INIT(&internal_ips_lock, NULL); MUTEX_INIT(&internal_ips_lock, NULL);
/* read the config file */ /* read the config file */
get_chain_data(proxychains_pd, &proxychains_proxy_count, &proxychains_ct); get_chain_data(proxychains_pd, &proxychains_proxy_count, &proxychains_ct);
proxychains_write_log(LOG_PREFIX "DLL init\n"); proxychains_write_log(LOG_PREFIX "DLL init\n");
for (i = 0; i < (sizeof(override_symbols) / sizeof(override_symbols[0])); i++) { SETUP_SYM(connect);
load_sym(override_symbols[i].funcptr, override_symbols[i].symname, override_symbols[i].proxyfunc); SETUP_SYM(gethostbyname);
} SETUP_SYM(getaddrinfo);
SETUP_SYM(freeaddrinfo);
SETUP_SYM(gethostbyaddr);
SETUP_SYM(getnameinfo);
init_l = 1; init_l = 1;
} }