/*
*  (c) 2000 venglin / buffer0verfl0w security (www.b0f.com)
*
*  syslogd-to-MySQL wrapper v0.1 prebeta (FreeBSD)
*
*  1. compile it: cc -O6 -Wall -pipe -m486 -L/usr/local/lib/mysql \
*                 -I/usr/local/include -o /usr/local/sbin/sqlsyslogd \
*                 sqlsyslogd.c -lmysqlclient
*
*  2. create table in some database:
*     CREATE TABLE logs (
*       id int(10) DEFAULT '0' NOT NULL auto_increment,
*       timestamp varchar(16),
*       host varchar(255),
*       prog varchar(255),
*       mesg text,
*       PRIMARY KEY (id)
*     );
*
*  3. put mysql password into /etc/sqlsyslogd.conf
*
*  4. add "*.* |/usr/local/sbin/sqlsyslogd [params]"
*     to /etc/syslog.conf and rehash syslogd.
*
* $Log: sqlsyslogd.c,v $
* Revision 1.4  2000/04/19 08:55:13  venglin
* Small changes.
*
* Revision 1.3  2000/04/19 08:50:05  venglin
* Fixes.
*
* Revision 1.2  2000/04/18 19:53:40  venglin
* Fixes
*
* Revision 1.1  2000/04/18 19:51:25  venglin
* Initial revision
*
*/

#include <mysql/mysql.h>
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <signal.h>
#include <stdlib.h>

#define BUFSIZE    1024
#define CONF    "/etc/sqlsyslogd.conf"
#define CR    13
#define LF    10

MYSQL db;

static char rcsid[] = "$Id: sqlsyslogd.c,v 1.4 2000/04/19 08:55:13 venglin Exp $";

void usage(av0)
char *av0;
{
  fprintf(stderr, "usage: %s [-h hostname] <-u username> [-p]"
        " <-t table> [database]\n\n", av0);
  exit(0);
}  

void cleanup(x)
int x;
{
  mysql_close(&db);
  exit(0);
}

char *password(void)
{
  FILE *fp;
  static char passwd[BUFSIZE/16];
  char *p;

  if ((fp=fopen(CONF, "r")) == NULL)
    return NULL;

  fgets(passwd, sizeof(passwd), fp);

  if (p = index(passwd, CR))
    *p = '\0';
  if (p = index(passwd, LF))
    *p = '\0';

  return passwd;
}
  

int main(argc, argv)
int argc;
char **argv;
{
  extern char *optarg;
  extern int optind;
        int ch;
  char buf[BUFSIZE], querybuf[BUFSIZE+100];
  char *loghost, *host, *user, *passwd, *av0, *table, *logprog, *logmesg;

  av0 = argv[0];
  loghost = host = user = passwd = table = logprog = logmesg = NULL;

  while ((ch = getopt(argc, argv, "h:u:pt:")) != -1)
    switch((char)ch)
    {
      case 'h':
        host = optarg;
        break;

      case 'u':
        user = optarg;
        break;

      case 'p':
        passwd = password();
        break;

      case 't':
        table = optarg;
        break;

      case '?':
      default:
          (void)usage(av0);
    }

  argc -= optind;
  argv += optind;

  if (!user || !table)
    (void)usage(av0);

  if (argc < 1)
    (void)usage(av0);

  mysql_init(&db);

  if (!mysql_real_connect(&db, host, user, passwd, *argv, 0, NULL, 0))
  {
    fprintf(stderr, "failed to connect to database: %s\n",
      mysql_error(&db));
    exit(1);
  }

  signal(SIGHUP, cleanup);
  signal(SIGINT, cleanup);
  signal(SIGQUIT, cleanup);
  signal(SIGTERM, cleanup);
  signal(SIGSEGV, cleanup);
  signal(SIGBUS, cleanup);

  while(fgets(buf, sizeof(buf), stdin))
  {
    if (strlen(buf) > 18)
    {
      loghost = strtok(buf + 16, " ");
      logprog = strtok(NULL, ":");
      logmesg = buf + 16 + strlen(loghost) +
        strlen(logprog) + 3;

      if (loghost && logprog && logmesg)
      {
        snprintf(querybuf, sizeof(querybuf),
          "INSERT INTO %s (timestamp, host, "
          "prog, mesg) VALUES ('%.15s', '%s', "
          "'%s', '%s')", table, buf,
          loghost, logprog, logmesg);

        if (mysql_query(&db, querybuf))
          fprintf(stderr, "failed to run "
            "query: %s\n",
            mysql_error(&db));
      }
    }
  }

  mysql_close(&db);

  exit(0);
}