#include #include #include #include #include #include #include #include #include /* This is dumb, but glibc doesn't like to do hostname lookups w/o libc.so */ #ifndef DIET union dns_response{ HEADER hdr; u_char buf[PACKETSZ]; } ; static int doQuery(char * query, int queryType, char ** domainName, struct in_addr * ipNum) { int len, ancount, type; u_char * data, * end; char name[MAXDNAME]; union dns_response static_response, *response = &static_response; size_t response_len = sizeof(static_response); /* Give time to finish ethernet negotiation */ _res.retry = 3; do { len = res_search(query, C_IN, queryType, (void*) response, response_len); if (len <= 0) return -1; if (len < response_len) break; if (response != &static_response) free(response); if (len > 0x10000) return -1; response_len = len + 1024; response = malloc(response_len); if (response == NULL) return -1; } while (1); if (len < sizeof(response->hdr)) { if (response != &static_response) free(response); return -1; } if (ntohs(response->hdr.rcode) != NOERROR) { if (response != &static_response) free(response); return -1; } ancount = ntohs(response->hdr.ancount); if (ancount < 1) { if (response != &static_response) free(response); return -1; } data = response->buf + sizeof(HEADER); end = response->buf + len; /* skip the question */ len = dn_skipname(data, end); if (len <= 0) { if (response != &static_response) free(response); return -1; } data += len + QFIXEDSZ; /* parse the answer(s) */ while (--ancount >= 0 && data < end) { /* skip the domain name portion of the RR record */ data += dn_skipname(data, end); /* get RR information */ if (data + 3 * INT16SZ + INT32SZ > end) { if (response != &static_response) free(response); return -1; } GETSHORT(type, data); data += INT16SZ; /* skip class */ data += INT32SZ; /* skip TTL */ GETSHORT(len, data); if (type == T_PTR) { /* we got a pointer */ len = dn_expand(response->buf, end, data, name, sizeof(name)); if (len <= 0) { if (response != &static_response) free(response); return -1; } if (queryType == T_PTR && domainName) { /* we wanted a pointer */ *domainName = malloc(strlen(name) + 1); strcpy(*domainName, name); if (response != &static_response) free(response); return 0; } } else if (type == T_A) { /* we got an address */ if (queryType == T_A && ipNum) { /* we wanted an address */ memcpy(ipNum, data, sizeof(*ipNum)); if (response != &static_response) free(response); return 0; } } /* move ahead to next RR */ data += len; } if (response != &static_response) free(response); return -1; } char * mygethostbyaddr(char * ipnum) { int rc; char * result; char * strbuf; char * chptr; char * splits[4]; int i; _res.retry = 1; strbuf = alloca(strlen(ipnum) + 1); strcpy(strbuf, ipnum); ipnum = alloca(strlen(strbuf) + 20); for (i = 0; i < 4; i++) { chptr = strbuf; while (*chptr && *chptr != '.') chptr++; *chptr = '\0'; if (chptr - strbuf > 3) return NULL; splits[i] = strbuf; strbuf = chptr + 1; } sprintf(ipnum, "%s.%s.%s.%s.in-addr.arpa", splits[3], splits[2], splits[1], splits[0]); rc = doQuery(ipnum, T_PTR, &result, NULL); if (rc) rc = doQuery(ipnum, T_PTR, &result, NULL); if (rc) return NULL; else return result; } int mygethostbyname(char * name, struct in_addr * addr) { return doQuery(name, T_A, NULL, addr); } #else #include #include #include int mygethostbyname(char * host, struct in_addr * address) { struct hostent * hostinfo; hostinfo = gethostbyname(host); if (!hostinfo) return 1; memcpy(address, hostinfo->h_addr_list[0], hostinfo->h_length); return 0; } char * mygethostbyaddr(const char * ipnum) { struct hostent * he; struct in_addr addr; if (!inet_aton(ipnum, &addr)) return NULL; he = gethostbyaddr(&addr, sizeof(struct in_addr), AF_INET); if (he) return he->h_name; else return NULL; } #endif #if 0 int main(int argc, char **argv) { struct in_addr address; fprintf(stderr, "hostname for %s is %s\n", "152.1.2.22", mygethostbyaddr("152.1.2.22")); if (mygethostbyname("www.redhat.com", &address) == 0) { fprintf(stderr, "ip for www.redhat.com is %d.%d.%d.%d\n", (address.s_addr >> 0) & 0xff, (address.s_addr >> 8) & 0xff, (address.s_addr >> 16) & 0xff, (address.s_addr >> 24) & 0xff); } return 0; } #endif