chiark - git - mdw - ezmlm/blob - sub_pgsql/searchlog.c

   1 /*$Id: searchlog.c,v 1.2 1999/10/07 23:31:01 lindberg Exp $*/
   2 /*$Name: ezmlm-idx-040 $*/
   3 #include "case.h"
   4 #include "scan.h"
   5 #include "stralloc.h"
   6 #include "str.h"
   7 #include "open.h"
   8 #include "datetime.h"
   9 #include "date822fmt.h"
  10 #include "substdio.h"
  11 #include "readwrite.h"
  12 #include "strerr.h"
  13 #include "error.h"
  14 #include "errtxt.h"
  15 #include "subscribe.h"
  16 #include <unistd.h>
  17 #include <libpq-fe.h>
  18
  19 static stralloc line = {0};
  20 static stralloc outline = {0};
  21 static date[DATE822FMT];
  22 static datetime_sec when;
  23 static struct datetime dt;
  24 static substdio ssin;
  25 static char inbuf[256];
  26
  27 static void die_nomem(fatal)
  28 char *fatal;
  29 {
  30   strerr_die2x(111,fatal,ERR_NOMEM);
  31 }
  32
  33 static void lineout(subwrite,fatal)
  34 int subwrite();
  35 char *fatal;
  36 {
  37   (void) scan_ulong(line.s,&when);
  38   datetime_tai(&dt,when);               /* there is always at least a '\n' */
  39   if (!stralloc_copyb(&outline,date,date822fmt(date,&dt) - 1))
  40         die_nomem(fatal);
  41   if (!stralloc_cats(&outline,": ")) die_nomem(fatal);
  42   if (!stralloc_catb(&outline,line.s,line.len - 1)) die_nomem(fatal);
  43   if (subwrite(outline.s,outline.len) == -1)
  44         strerr_die3x(111,fatal,ERR_WRITE,"output");
  45   return;
  46 }
  47
  48 void searchlog(dir,search,subwrite,fatal)
  49 /* opens dir/Log, and outputs via subwrite(s,len) any line that matches   */
  50 /* search. A '_' is search is a wildcard. Any other non-alphanum/'.' char */
  51 /* is replaced by a '_'. mysql version. Falls back on "manual" search of  */
  52 /* local Log if no mysql connect info. */
  53
  54 char *dir;              /* work directory */
  55 char *search;           /* search string */
  56 int subwrite();         /* output fxn */
  57 char *fatal;            /* fatal */
  58 {
  59
  60   register unsigned char x;
  61   register unsigned char y;
  62   register unsigned char *cp;
  63   register unsigned char *cpsearch;
  64   unsigned register char *cps;
  65   unsigned register char ch;
  66   unsigned char *cplast, *cpline;
  67   unsigned int searchlen;
  68   int fd,match;
  69   char *ret;
  70
  71   PGresult *result;
  72   int row_nr;
  73   int length;
  74   char *row;
  75
  76   char *table = (char *) 0;
  77
  78   if (!search) search = "";     /* defensive */
  79   searchlen = str_len(search);
  80   case_lowerb(search,searchlen);
  81   cps = (unsigned char *) search;
  82   while ((ch = *(cps++))) {     /* search is potentially hostile */
  83     if (ch >= 'a' && ch <= 'z') continue;
  84     if (ch >= '0' && ch <= '9') continue;
  85     if (ch == '.' || ch == '_') continue;
  86     *(cps - 1) = '_';           /* will match char specified as well */
  87   }
  88
  89   if ((ret = opensql(dir,&table))) {
  90     if (*ret) strerr_die2x(111,fatal,ret);
  91                                                 /* fallback to local log */
  92   if (!stralloc_copys(&line,dir)) die_nomem(fatal);
  93   if (!stralloc_cats(&line,"/Log")) die_nomem(fatal);
  94   if (!stralloc_0(&line)) die_nomem(fatal);
  95   fd = open_read(line.s);
  96   if (fd == -1)
  97     if (errno != error_noent)
  98         strerr_die4sys(111,fatal,ERR_OPEN,line.s,": ");
  99     else
 100         strerr_die3x(100,fatal,line.s,ERR_NOEXIST);
 101   substdio_fdbuf(&ssin,read,fd,inbuf,sizeof(inbuf));
 102
 103   for (;;) {
 104     if (getln(&ssin,&line,&match,'\n') == -1)
 105       strerr_die2sys(111,fatal,ERR_READ_INPUT);
 106     if (!match) break;
 107     if (!searchlen) {
 108       lineout(subwrite,fatal);
 109     } else {
 110       cpline = (unsigned char *) line.s - 1;
 111       cplast = cpline + line.len - searchlen; /* line has \0 at the end */
 112       while ((cp = ++cpline) <= cplast) {
 113         cpsearch = (unsigned char *) search;
 114         for (;;) {
 115           x = *cpsearch++;
 116           if (!x) break;
 117           y = *cp++ - 'A';
 118           if (y <= (unsigned char) ('Z' - 'A')) y += 'a'; else y += 'A';
 119           if (x != y && x != '_') break;                /* '_' = wildcard */
 120         }
 121         if (!x) {
 122           lineout(subwrite,fatal);
 123           break;
 124         }
 125       }
 126     }
 127   }
 128   close(fd);
 129   } else {
 130
 131 /* SELECT (*) FROM list_slog WHERE fromline LIKE '%search%' OR address   */
 132 /* LIKE '%search%' ORDER BY tai; */
 133 /* The '*' is formatted to look like the output of the non-mysql version */
 134 /* This requires reading the entire table, since search fields are not   */
 135 /* indexed, but this is a rare query and time is not of the essence.     */
 136
 137     if (!stralloc_cats(&line,"SELECT tai::datetime||': '||tai::int8||' '"
 138     "||address||' '||edir||etype||' '||fromline FROM ")) die_nomem(fatal);
 139
 140     if (!stralloc_cats(&line,table)) die_nomem(fatal);
 141     if (!stralloc_cats(&line,"_slog ")) die_nomem(fatal);
 142     if (*search) {      /* We can afford to wait for LIKE '%xx%' */
 143       if (!stralloc_cats(&line,"WHERE fromline LIKE '%")) die_nomem(fatal);
 144       if (!stralloc_cats(&line,search)) die_nomem(fatal);
 145       if (!stralloc_cats(&line,"%' OR address LIKE '%")) die_nomem(fatal);
 146       if (!stralloc_cats(&line,search)) die_nomem(fatal);
 147       if (!stralloc_cats(&line,"%'")) die_nomem(fatal);
 148     }   /* ordering by tai which is an index */
 149     if (!stralloc_cats(&line," ORDER by tai")) die_nomem(fatal);
 150
 151     if (!stralloc_0(&line)) die_nomem(fatal);
 152     result = PQexec(psql,line.s);
 153     if (result == NULL)
 154       strerr_die2x(111,fatal,PQerrorMessage(psql));
 155     if (PQresultStatus(result) != PGRES_TUPLES_OK)
 156       strerr_die2x(111,fatal,PQresultErrorMessage(result));
 157
 158     for(row_nr=0; row_nr<PQntuples(result); row_nr++) {
 159       row = PQgetvalue(result,row_nr,0);
 160       length = PQgetlength(result,row_nr,0);
 161       if (subwrite(row,length) == -1) die_write(fatal);
 162     }
 163     PQclear(result);
 164
 165   }
 166 }