mirror of
https://github.com/rofl0r/proxychains-ng.git
synced 2025-09-05 08:06:59 +00:00
af5c6f0c6a
the central dns resolver function proxy_gethostbyname() used to iterate over the gethostent() db (/etc/hosts) on each dns request. since this is not threadsafe, we synchronized access to it previously using mutexes. the parsing of this file is slow, and blocking all threads to do it even moreso. since gethostent_r() is only available on a few platforms, i decided to read the hostent db once and then use a quick in-memory lookup on further usage. + some further refactoring.
56 lines
1.4 KiB
C
56 lines
1.4 KiB
C
#include <stdint.h>
|
|
#include <string.h>
|
|
#include <netdb.h>
|
|
#include <stdlib.h>
|
|
|
|
#include "ip_type.h"
|
|
#include "hash.h"
|
|
#include "stringdump.h"
|
|
#include "hostentdb.h"
|
|
#include "common.h"
|
|
#include "debug.h"
|
|
|
|
#define STEP 16
|
|
static void hdb_add(struct hostent_list* hl, char* host, ip_type ip) {
|
|
if(hl->count +1 > hl->capa) {
|
|
void * nu = realloc(hl->entries, (hl->capa + STEP) * sizeof(struct hostent_entry));
|
|
if(!nu) return;
|
|
hl->entries = nu;
|
|
hl->capa += STEP;
|
|
}
|
|
struct hostent_entry *h = &hl->entries[hl->count];
|
|
h->hash = dalias_hash(host);
|
|
h->ip.as_int = ip.as_int;
|
|
h->str = dumpstring(host, strlen(host) + 1);
|
|
if(h->str) hl->count++;
|
|
}
|
|
|
|
static void hdb_fill(struct hostent_list *hl) {
|
|
struct hostent* hp;
|
|
while((hp = gethostent()))
|
|
if(hp->h_addrtype == AF_INET && hp->h_length == sizeof(in_addr_t)) {
|
|
hdb_add(hl, hp->h_name, (ip_type) { .as_int = *((in_addr_t*)(hp->h_addr_list[0])) });
|
|
}
|
|
}
|
|
|
|
void hdb_init(struct hostent_list *hl) {
|
|
memset(hl, 0, sizeof *hl);
|
|
hdb_fill(hl);
|
|
}
|
|
|
|
ip_type hdb_get(struct hostent_list *hl, char* host) {
|
|
size_t i;
|
|
PFUNC();
|
|
uint32_t hash = dalias_hash(host);
|
|
for(i = 0; i < hl->count; i++) {
|
|
if(hl->entries[i].hash == hash && !strcmp(hl->entries[i].str, host)) {
|
|
#ifdef DEBUG
|
|
char ipbuf[16];
|
|
pc_stringfromipv4(hl->entries[i].ip.octet, ipbuf);
|
|
PDEBUG("got ip %s for hostent entry %s\n", ipbuf, host);
|
|
#endif
|
|
return hl->entries[i].ip;
|
|
}
|
|
}
|
|
return ip_type_invalid;
|
|
} |