+/*
+ * idnlookup - ICMP Domain Name lookup utility for Linux
+ * Should be installed SUID root, even though I don't know if it's secure yet. :-)
+ * Copyright (C) 2005 Fredrik Tolf <fredrik@dolda2000.com>
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+*/
+#include <stdlib.h>
+#include <unistd.h>
+#include <stdio.h>
+#include <string.h>
+#include <errno.h>
+#include <netdb.h>
+#include <sys/socket.h>
+#include <netinet/in.h>
+#include <netinet/ip.h>
+#include <netinet/ip6.h>
+#include <arpa/inet.h>
+#include <sys/types.h>
+#include <sys/poll.h>
+#include <sys/time.h>
+
+struct icmphdr {
+ u_int8_t type;
+ u_int8_t code;
+ u_int16_t checksum;
+};
+
+struct reqhdr {
+ u_int8_t type;
+ u_int8_t code;
+ u_int16_t checksum;
+ u_int16_t id;
+ u_int16_t seq;
+};
+
+struct rephdr {
+ u_int8_t type;
+ u_int8_t code;
+ u_int16_t checksum;
+ u_int16_t id;
+ u_int16_t seq;
+ int32_t ttl;
+
+};
+
+#define ICMP_NAMEREQ 37
+#define ICMP_NAMEREP 38
+
+unsigned char buf[65536];
+
+/* DN decompression not yet implemented, since I don't know where to
+ * begin counting the offset from -- the beginning of the ICMP
+ * payload, or from the beginning of the DN data buffer? */
+void printdn(FILE *f, unsigned char *dnbuf, size_t size, int onlyfirst)
+{
+ unsigned char *p;
+
+ p = dnbuf;
+ while(p - dnbuf < size) {
+ while(*p != 0) {
+ if(*p & 0xc0) {
+ fprintf(stderr, "domain name decompression not implemented, aborting\n");
+ exit(1);
+ }
+ fprintf(f, "%.*s", (int)*p, p + 1);
+ p += 1 + (int)*p;
+ if(*p != 0)
+ fprintf(f, ".");
+ }
+ p++;
+ fprintf(f, "\n");
+ if(onlyfirst)
+ break;
+ }
+}
+
+void cksum(void *hdr, size_t len)
+{
+ struct icmphdr *ih;
+ u_int8_t *cb;
+ int i;
+ int b1, b2;
+
+ ih = (struct icmphdr *)hdr;
+ cb = (u_int8_t *)hdr;
+ ih->checksum = 0;
+ b1 = b2 = 0;
+ for(i = 0; i < (len & ~1); i += 2) {
+ b1 += cb[i];
+ b2 += cb[i + 1];
+ }
+ if(i & 1)
+ b1 += cb[len - 1];
+ while(1) {
+ if(b1 >= 256) {
+ b2 += b1 >> 8;
+ b1 &= 0xff;
+ continue;
+ }
+ if(b2 >= 256) {
+ b1 += b2 >> 8;
+ b2 &= 0xff;
+ continue;
+ }
+ break;
+ }
+ cb = (u_int8_t *)&ih->checksum;
+ cb[0] = ~(u_int8_t)b1;
+ cb[1] = ~(u_int8_t)b2;
+}
+
+void usage(void)
+{
+ fprintf(stderr, "usage: idnlookup [-hTa] [-t timeout] host\n");
+}
+
+int main(int argc, char **argv)
+{
+ int ret;
+ int s, c;
+ int id;
+ int namelen;
+ struct reqhdr req;
+ struct rephdr rep;
+ struct iphdr iphdr;
+ size_t hdrlen;
+ struct addrinfo *ai, *cai, aihint;
+ struct pollfd pfd;
+ struct timeval tvb, tvc;
+ struct sockaddr_storage name;
+ int timeout, dispttl, onlyfirst;
+ int elapsed, timedout, found;
+
+ timeout = 3000;
+ dispttl = 0;
+ onlyfirst = 1;
+ while((c = getopt(argc, argv, "haTt:")) != -1) {
+ switch(c) {
+ case 't':
+ timeout = atoi(optarg);
+ break;
+ case 'a':
+ onlyfirst = 0;
+ break;
+ case 'T':
+ dispttl = 1;
+ break;
+ case 'h':
+ case '?':
+ case ':':
+ default:
+ usage();
+ exit((c == 'h')?0:1);
+ }
+ }
+
+ if(argc - optind < 1) {
+ usage();
+ exit(1);
+ }
+
+ memset(&aihint, 0, sizeof(aihint));
+ aihint.ai_socktype = SOCK_RAW;
+ aihint.ai_protocol = IPPROTO_ICMP;
+ ret = getaddrinfo(argv[optind], NULL, &aihint, &ai);
+
+ for(cai = ai; cai != NULL; cai = cai->ai_next) {
+ if((s = socket(cai->ai_family, SOCK_RAW, IPPROTO_ICMP)) < 0) {
+ perror("could not create raw socket");
+ exit(1);
+ }
+
+ id = random() % 65536;
+ memset(&req, 0, sizeof(req));
+ req.type = ICMP_NAMEREQ;
+ req.id = htons(id);
+ cksum(&req, sizeof(req));
+
+ ret = sendto(s, &req, sizeof(req), 0, cai->ai_addr, cai->ai_addrlen);
+ if(ret < 0) {
+ perror("sendto");
+ exit(1);
+ } else if(ret != sizeof(req)) {
+ fprintf(stderr, "socket would not send entire packet\n");
+ exit(1);
+ }
+
+ timedout = 0;
+ found = 0;
+ gettimeofday(&tvb, NULL);
+ while(1) {
+ pfd.fd = s;
+ pfd.events = POLLIN;
+ gettimeofday(&tvc, NULL);
+ elapsed = ((tvc.tv_sec - tvb.tv_sec) * 1000) + ((tvc.tv_usec - tvb.tv_usec) / 1000);
+ if(elapsed >= timeout) {
+ timedout = 1;
+ break;
+ }
+ ret = poll(&pfd, 1, timeout - elapsed);
+ if(ret < 0) {
+ perror("idnlookup: reading data");
+ exit(1);
+ }
+
+ if(pfd.revents & POLLIN) {
+ namelen = sizeof(name);
+ ret = recvfrom(s, buf, sizeof(buf), 0, (struct sockaddr *)&name, &namelen);
+ if(ret < 0) {
+ perror("idnlookup: receiving data");
+ exit(1);
+ }
+
+ if(name.ss_family != cai->ai_addr->sa_family)
+ continue;
+ if(name.ss_family == AF_INET) {
+ if(memcmp(&(((struct sockaddr_in *)&name)->sin_addr), &(((struct sockaddr_in *)cai->ai_addr)->sin_addr), sizeof(struct in_addr)))
+ continue;
+ if(ret < sizeof(iphdr) + sizeof(rep))
+ continue;
+ hdrlen = sizeof(iphdr);
+ memcpy(&iphdr, buf, sizeof(iphdr));
+ if(iphdr.protocol != IPPROTO_ICMP)
+ continue;
+ } else if(name.ss_family == AF_INET6) {
+ if(memcmp(&(((struct sockaddr_in6 *)&name)->sin6_addr), &(((struct sockaddr_in6 *)cai->ai_addr)->sin6_addr), sizeof(struct in6_addr)))
+ continue;
+ if(ret < sizeof(rep))
+ continue;
+ hdrlen = 0;
+ } else {
+ continue;
+ }
+
+ memcpy(&rep, buf + hdrlen, sizeof(rep));
+ if(rep.type != ICMP_NAMEREP)
+ continue;
+ if((ntohs(rep.id) != id) || (ntohs(rep.seq != 0)))
+ continue;
+
+ found = 1;
+ break;
+ }
+ }
+
+ close(s);
+
+ if(found) {
+ if(dispttl)
+ printf("%i\n", ntohl(rep.ttl));
+ printdn(stdout, buf + hdrlen + sizeof(rep), ret - hdrlen - sizeof(rep), onlyfirst);
+ break;
+ }
+ }
+
+ if(timedout) {
+ fprintf(stderr, "idnlookup: timeout\n");
+ exit(1);
+ }
+
+ return(0);
+}
+
+/*
+ * Local Variables:
+ * compile-command: "gcc -Wall -g -o idnlookup idnlookup.c"
+ * End:
+ */