chiark / gitweb /
Support for reverse queries.
[adns.git] / client / adnslogres.c
index 7aa46f10166c25596a677ceb44d07c6bcb561c39..c2ca2effead5eb0b66337760e3c14400ed34e741 100644 (file)
@@ -3,7 +3,13 @@
  * - a replacement for the Apache logresolve program using adns
  */
 /*
- *  This file is Copyright (C) 1999 Tony Finch <fanf@demon.net> <dot@dotat.at>
+ *  This file is
+ *   Copyright (C) 1999 Tony Finch <dot@dotat.at>
+ *   Copyright (C) 1999 Ian Jackson <ian@davenant.greenend.org.uk>
+ *
+ *  It is part of adns, which is
+ *    Copyright (C) 1997-1999 Ian Jackson <ian@davenant.greenend.org.uk>
+ *    Copyright (C) 1999 Tony Finch <dot@dotat.at>
  *  
  *  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
  *  
  *  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. 
+ *  Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
+ *
+ * This version was originally supplied by Tony Finch, but has been
+ * modified by Ian Jackson as it was incorporated into adns.
  */
 
 static const char * const cvsid =
@@ -26,6 +35,7 @@ static const char * const cvsid =
 #include <sys/types.h>
 #include <sys/time.h>
 
+#include <unistd.h>
 #include <string.h>
 #include <stdlib.h>
 #include <stdio.h>
@@ -40,6 +50,10 @@ static const char * const cvsid =
 /* maximum length of a line */
 #define MAXLINE 1024
 
+/* option flags */
+#define OPT_DEBUG 1
+#define OPT_POLL 2
+
 static const char *progname;
 
 static void aargh(const char *msg) {
@@ -51,19 +65,25 @@ static void aargh(const char *msg) {
 /*
  * Parse the IP address and convert to a reverse domain name.
  */
-static void ipaddr2domain(char *start, char **addr, char **rest, char **domain) {
+static char *ipaddr2domain(char *start, char **addr, char **rest) {
   static char buf[30]; /* "123.123.123.123.in-addr.arpa.\0" */
   char *ptrs[5];
   int i;
 
-  for (ptrs[0]= start; !isdigit(*ptrs[0]); ptrs[0]++)
-    if (!*ptrs[0])
-      goto invalid;
+  ptrs[0]= start;
+retry:
+  while (!isdigit(*ptrs[0]))
+    if (!*ptrs[0]++) {
+      strcpy(buf, "invalid.");
+      *addr= *rest= NULL;
+      return buf;
+    }
   for (i= 1; i < 5; i ++) {
     ptrs[i]= strchr(ptrs[i-1], (i == 4) ? ' ' : '.');
-    if (!ptrs[i] || ptrs[i]-ptrs[i-1] > 3)
-      goto invalid;
-    else
+    if (!ptrs[i] || ptrs[i]-ptrs[i-1] > 3) {
+      ptrs[0]++;
+      goto retry;
+    } else
       ptrs[i]++;
   }
   sprintf(buf, "%.*s.%.*s.%.*s.%.*s.in-addr.arpa.",
@@ -73,13 +93,7 @@ static void ipaddr2domain(char *start, char **addr, char **rest, char **domain)
          ptrs[1]-ptrs[0]-1, ptrs[0]);
   *addr= ptrs[0];
   *rest= ptrs[4]-1;
-  *domain= buf;
-  return;
-invalid:
-  strcpy(buf, "invalid.");
-  *addr= *rest= NULL;
-  *domain= buf;
-  return;
+  return buf;
 }
 
 static void printline(char *start, char *addr, char *rest, char *domain) {
@@ -87,7 +101,7 @@ static void printline(char *start, char *addr, char *rest, char *domain) {
     printf("%.*s%s%s", addr - start, start, domain, rest);
   else
     fputs(start, stdout);
-  /* XXX: error checking */
+  if (ferror(stdout)) aargh("write output");
 }
 
 typedef struct logline {
@@ -96,7 +110,7 @@ typedef struct logline {
   adns_query query;
 } logline;
 
-static logline *readline(adns_state adns) {
+static logline *readline(adns_state adns, int opts) {
   static char buf[MAXLINE];
   char *str;
   logline *line;
@@ -108,7 +122,9 @@ static logline *readline(adns_state adns) {
     line->next= NULL;
     line->start= str+sizeof(logline);
     strcpy(line->start, buf);
-    ipaddr2domain(line->start, &line->addr, &line->rest, &str);
+    str = ipaddr2domain(line->start, &line->addr, &line->rest);
+    if (opts & OPT_DEBUG)
+       fprintf(stderr, "%s: adns_submit %s\n", progname, str);
     if (adns_submit(adns, str, adns_r_ptr,
                    adns_qf_quoteok_cname|adns_qf_cname_loose,
                    NULL, &line->query))
@@ -120,19 +136,22 @@ static logline *readline(adns_state adns) {
   return NULL;
 }
        
-static void proclog(void) {
+static void proclog(int opts) {
   int eof, err, len;
   adns_state adns;
   adns_answer *answer;
   logline *head, *tail, *line;
 
-  errno= adns_init(&adns, 0, 0);
+  errno= adns_init(&adns, (opts & OPT_DEBUG) ? adns_if_debug : 0, 0);
   if (errno) aargh("adns_init");
-  head= tail= readline(adns);
+  head= tail= readline(adns, opts);
   len= 1; eof= 0;
   while (head) {
     if (eof || len > MAXPENDING)
-      err= adns_wait(adns, &head->query, &answer, NULL);
+      if (opts & OPT_POLL)
+       err= adns_wait_poll(adns, &head->query, &answer, NULL);
+      else
+       err= adns_wait(adns, &head->query, &answer, NULL);
     else
       err= adns_check(adns, &head->query, &answer, NULL);
     if (err != EWOULDBLOCK) {
@@ -143,7 +162,7 @@ static void proclog(void) {
        len--;
     }
     if (!eof) {
-      line= readline(adns);
+      line= readline(adns, opts);
       if (!line)
        eof= 1;
       else {
@@ -159,9 +178,30 @@ static void proclog(void) {
   adns_finish(adns);
 }
 
-int main(int argc, char *argv[])
-{
+int main(int argc, char *argv[]) {
+  int c, opts;
+
   progname= *argv;
-  proclog();
+  opts= 0;
+
+  while ((c= getopt(argc, argv, "dp")) != -1) {
+    switch (c) {
+    case 'd':
+      opts |= OPT_DEBUG;
+      break;
+    case 'p':
+      opts |= OPT_POLL;
+      break;
+    default:
+      fprintf(stderr, "usage: %s [-d] < logfile\n", progname);
+      exit(1);
+    }
+    argc-= optind;
+    argv+= optind;
+  }
+
+  proclog(opts);
+
+  if (fclose(stdout)) aargh("finish writing output");
   return 0;
 }