| 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 | } |