searchlog.c

Go to the documentation of this file.

#include <unistd.h>
#include "getln.h"
#include "case.h"
#include "scan.h"
#include "stralloc.h"
#include "str.h"
#include "open.h"
#include "datetime.h"
#include "date822fmt.h"
#include "buffer.h"
#include "readwrite.h"
#include "logmsg.h"
#include "error.h"
#include "errtxt.h"
#include "subscribe.h"
#include "idx.h"
#include <mysql.h>

 
extern MYSQL *mysql;
 
extern void die_write(void);
 
static stralloc line = {0};
static stralloc outline = {0};
static char date[DATE822FMT];
static datetime_sec when;
static struct datetime dt;
static buffer bi;
static char inbuf[256];
 
static void lineout(int subwrite())
{
  scan_ulong(line.s,&when);
  datetime_tai(&dt,when);        /* there is always at least a '\n' */
  if (!stralloc_copyb(&outline,date,date822fmt(date,&dt) - 1))
    die_nomem();
  if (!stralloc_cats(&outline,": ")) die_nomem();
  if (!stralloc_catb(&outline,line.s,line.len - 1)) die_nomem();
  if (subwrite(outline.s,outline.len) == -1)
    logmsg(WHO,111,FATAL,B("ERR_WRITE :","output"));
  return;
}

 
void searchlog(const char *dir,char *search,int subwrite())
{
  unsigned char x;
  unsigned char y;
  unsigned char *cp;
  unsigned char *cpsearch;
  unsigned char *cps;
  unsigned char ch;
  unsigned char *cplast, *cpline;
  unsigned int searchlen;
  int fd,match;
  const char *ret;
 
  MYSQL_RES *result;
  MYSQL_ROW row;
  const char *table = (char *) 0;
  const char **ptable = &table;
  unsigned long *lengths;
 
  if (!search) search = (char *)"";    /* defensive */
  searchlen = str_len(search);
  case_lowerb(search,searchlen);
  cps = (unsigned char *) search;
  while ((ch = *(cps++))) {            /* search is potentially hostile */
    if (ch >= 'a' && ch <= 'z') continue;
    if (ch >= '0' && ch <= '9') continue;
    if (ch == '.' || ch == '_') continue;
    *(cps - 1) = '_';                  /* will match char specified as well */
  }
 
  if ((ret = opensql(dir,ptable))) {
    if (*ret) logmsg(WHO,111,FATAL,ret);
 
    /* fallback to local log */
 
    if (!stralloc_copys(&line,dir)) die_nomem();
    if (!stralloc_cats(&line,"/Log")) die_nomem();
    if (!stralloc_0(&line)) die_nomem();
    
    fd = open_read(line.s);
    if (fd == -1)
      if (errno != ENOENT)
        logmsg(WHO,111,FATAL,B("ERR_OPEN :",line.s));
    else
        logmsg(WHO,100,FATAL,line.s,"ERR_NOEXIST");
 
    buffer_init(&bi,buffer_unixread,fd,inbuf,sizeof(inbuf));
 
    for (;;) {
      if (getln(&bi,&line,&match,'\n') == -1) logmsg(WHO,111,FATAL,ERR_READ_INPUT);
      if (!match) break;
      if (!searchlen) {
        lineout(subwrite);
      } else {
        cpline = (unsigned char *) line.s - 1;
        cplast = cpline + line.len - searchlen; /* line has \0 at the end */
        while ((cp = ++cpline) <= cplast) {
          cpsearch = (unsigned char *) search;
          for (;;) {
            x = *cpsearch++;
            if (!x) break;
            y = *cp++ - 'A';
            if (y <= (unsigned char) ('Z' - 'A')) y += 'a'; else y += 'A';
            if (x != y && x != '_') break;       /* '_' = wildcard */
          }
          if (!x) { lineout(subwrite); break; }
      }
    }
  }
  close(fd);
  } else {
 
/* SELECT (*) FROM list_slog WHERE fromline LIKE '%search%' OR address   */
/* LIKE '%search%' ORDER BY tai; */
/* The '*' is formatted to look like the output of the non-mysql version */
/* This requires reading the entire table, since search fields are not   */
/* indexed, but this is a rare query and time is not of the essence.     */
 
    if (!stralloc_cats(&line,"SELECT CONCAT(FROM_UNIXTIME(UNIX_TIMESTAMP(tai)),"
    "'-0000: ',UNIX_TIMESTAMP(tai),' ',edir,etype,' ',address,' ',"
    "fromline) FROM ")) die_nomem();
    if (!stralloc_cats(&line,table)) die_nomem();
    if (!stralloc_cats(&line,"_slog ")) die_nomem();
    if (*search) {                     /* We can afford to wait for LIKE '%xx%' */
      if (!stralloc_cats(&line,"WHERE fromline LIKE '%")) die_nomem();
      if (!stralloc_cats(&line,search)) die_nomem();
      if (!stralloc_cats(&line,"%' OR address LIKE '%")) die_nomem();
      if (!stralloc_cats(&line,search)) die_nomem();
      if (!stralloc_cats(&line,"%'")) die_nomem();
    }                                  /* ordering by tai which is an index */
      if (!stralloc_cats(&line," ORDER by tai")) die_nomem();
 
    if (mysql_real_query(mysql,line.s,line.len)) /* query */
      logmsg(WHO,111,FATAL,mysql_error(mysql));
 
    if (!(result = mysql_use_result(mysql)))
      logmsg(WHO,111,FATAL,mysql_error(mysql));
 
    while ((row = mysql_fetch_row(result))) {
      if (!(lengths = mysql_fetch_lengths(result)))
        logmsg(WHO,111,FATAL,mysql_error(mysql));
      if (subwrite(row[0],lengths[0]) == -1) die_write();
    }
 
    if (!mysql_eof(result))
      logmsg(WHO,111,FATAL,mysql_error(mysql));
 
    mysql_free_result(result);
  }
}