X-Git-Url: http://www.chiark.greenend.org.uk/ucgi/~ianmdlvl/git?a=blobdiff_plain;f=client%2Fadnslogres.c;h=868734b6bbbcca75fcb938a48755d68d8a3e9ef5;hb=9715e5f4f1bcf1790dd7cab02192efd37bcff73a;hp=bf17ccac1599e449eeb55a3674440e9dfcf71a28;hpb=940356bdf463f1decefb2c8d63dd33d7276d5991;p=adns.git diff --git a/client/adnslogres.c b/client/adnslogres.c index bf17cca..868734b 100644 --- a/client/adnslogres.c +++ b/client/adnslogres.c @@ -4,8 +4,12 @@ */ /* * This file is - * Copyright (C) 1999 Tony Finch + * Copyright (C) 1999 Tony Finch * Copyright (C) 1999 Ian Jackson + * + * It is part of adns, which is + * Copyright (C) 1997-1999 Ian Jackson + * Copyright (C) 1999 Tony Finch * * 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 @@ -52,9 +56,12 @@ static const char * const cvsid = static const char *progname; -static void aargh(const char *msg) { - fprintf(stderr, "%s: %s: %s (%d)\n", progname, msg, - strerror(errno) ? strerror(errno) : "Unknown error", errno); +#define msg(fmt, args...) fprintf(stderr, "%s: " fmt "\n", progname, ##args) + +static void aargh(const char *cause) { + const char *why = strerror(errno); + if (!why) why = "Unknown error"; + msg("%s: %s (%d)", cause, why, errno); exit(1); } @@ -74,13 +81,15 @@ retry: *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) { - ptrs[0]++; + for (i= 1; i < 5; i++) { + ptrs[i]= ptrs[i-1]; + while (isdigit(*ptrs[i]++)); + if ((i == 4 && !isspace(ptrs[i][-1])) || + (i != 4 && ptrs[i][-1] != '.') || + (ptrs[i]-ptrs[i-1] > 4)) { + ptrs[0]= ptrs[i]-1; goto retry; - } else - ptrs[i]++; + } } sprintf(buf, "%.*s.%.*s.%.*s.%.*s.in-addr.arpa.", ptrs[4]-ptrs[3]-1, ptrs[3], @@ -92,12 +101,12 @@ retry: return buf; } -static void printline(char *start, char *addr, char *rest, char *domain) { +static void printline(FILE *outf, char *start, char *addr, char *rest, char *domain) { if (domain) - printf("%.*s%s%s", addr - start, start, domain, rest); + fprintf(outf, "%.*s%s%s", addr - start, start, domain, rest); else - fputs(start, stdout); - if (ferror(stdout)) aargh("write output"); + fputs(start, outf); + if (ferror(outf)) aargh("write output"); } typedef struct logline { @@ -106,33 +115,33 @@ typedef struct logline { adns_query query; } logline; -static logline *readline(adns_state adns, int opts) { +static logline *readline(FILE *inf, adns_state adns, int opts) { static char buf[MAXLINE]; char *str; logline *line; - if (fgets(buf, MAXLINE, stdin)) { + if (fgets(buf, MAXLINE, inf)) { str= malloc(sizeof(*line) + strlen(buf) + 1); if (!str) aargh("malloc"); line= (logline*)str; line->next= NULL; line->start= str+sizeof(logline); strcpy(line->start, buf); - str = ipaddr2domain(line->start, &line->addr, &line->rest); + str= ipaddr2domain(line->start, &line->addr, &line->rest); if (opts & OPT_DEBUG) - fprintf(stderr, "%s: adns_submit %s\n", progname, str); + msg("submitting %.*s -> %s", line->rest-line->addr, line->addr, str); if (adns_submit(adns, str, adns_r_ptr, adns_qf_quoteok_cname|adns_qf_cname_loose, NULL, &line->query)) aargh("adns_submit"); return line; } - if (!feof(stdin)) + if (!feof(inf)) aargh("fgets"); return NULL; } -static void proclog(int opts) { +static void proclog(FILE *inf, FILE *outf, int opts) { int eof, err, len; adns_state adns; adns_answer *answer; @@ -140,9 +149,12 @@ static void proclog(int opts) { errno= adns_init(&adns, (opts & OPT_DEBUG) ? adns_if_debug : 0, 0); if (errno) aargh("adns_init"); - head= tail= readline(adns, opts); + head= tail= readline(inf, adns, opts); len= 1; eof= 0; while (head) { + if (opts & OPT_DEBUG) + msg("%d in queue; checking %.*s", len, + head->rest-head->addr, head->addr); if (eof || len > MAXPENDING) if (opts & OPT_POLL) err= adns_wait_poll(adns, &head->query, &answer, NULL); @@ -150,23 +162,23 @@ static void proclog(int opts) { err= adns_wait(adns, &head->query, &answer, NULL); else err= adns_check(adns, &head->query, &answer, NULL); - if (err != EWOULDBLOCK) { - printline(head->start, head->addr, head->rest, - answer->status == adns_s_ok ? *answer->rrs.str : NULL); - line= head; head= head->next; - free(line); free(answer); - len--; + if (err != EAGAIN) { + printline(outf, head->start, head->addr, head->rest, + answer->status == adns_s_ok ? *answer->rrs.str : NULL); + line= head; head= head->next; + free(line); free(answer); + len--; } if (!eof) { - line= readline(adns, opts); + line= readline(inf, adns, opts); if (!line) eof= 1; else { if (!head) - head = line; + head= line; else - tail->next = line; - tail = line; + tail->next= line; + tail= line; len++; } } @@ -174,30 +186,54 @@ static void proclog(int opts) { adns_finish(adns); } +static void usage(void) { + fprintf(stderr, "usage: %s [-d] [-p] [logfile]\n", progname); + exit(1); +} + int main(int argc, char *argv[]) { int c, opts; + FILE *inf; - progname= *argv; + progname= strrchr(*argv, '/'); + if (progname) + progname++; + else + progname= *argv; opts= 0; - while ((c= getopt(argc, argv, "dp")) != -1) { + while ((c= getopt(argc, argv, "dp")) != -1) switch (c) { case 'd': - opts |= OPT_DEBUG; + opts|= OPT_DEBUG; break; case 'p': - opts |= OPT_POLL; + opts|= OPT_POLL; break; default: - fprintf(stderr, "usage: %s [-d] < logfile\n", progname); - exit(1); + usage(); } - argc-= optind; - argv+= optind; - } - proclog(opts); + argc-= optind; + argv+= optind; + + inf= NULL; + if (argc == 0) + inf= stdin; + else if (argc == 1) + inf= fopen(*argv, "r"); + else + usage(); + + if (!inf) + aargh("couldn't open input"); + + proclog(inf, stdout, opts); + + if (fclose(inf)) + aargh("fclose input"); + if (fclose(stdout)) + aargh("fclose output"); - if (fclose(stdout)) aargh("finish writing output"); return 0; }