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