--- /dev/null
+/* ulogd_PGSQL.c, Version 1.1
+ *
+ * ulogd output plugin for logging to a PGSQL database
+ *
+ * (C) 2000 by Harald Welte <laforge@gnumonks.org>
+ * This software is distributed under the terms of GNU GPL
+ *
+ * This plugin is based on the MySQL plugin made by Harald Welte.
+ * The support PostgreSQL were made by Jakab Laszlo.
+ *
+ */
+
+#include <stdlib.h>
+#include <string.h>
+#include <ulogd.h>
+#include <libpq-fe.h>
+#include "ulogd.h"
+#include "conffile.h"
+
+#ifdef DEBUG_PQSQL
+#define DEBUGP(x, args...) fprintf(stderr, x, ## args)
+#else
+#define DEBUGP(x, args...)
+#endif
+
+struct _field {
+ char name[ULOGD_MAX_KEYLEN];
+ unsigned int id;
+ struct _field *next;
+};
+
+/* the database handle we are using */
+static PGconn *dbh;
+
+/* a linked list of the fields the table has */
+static struct _field *fields;
+
+/* buffer for our insert statement */
+static char *stmt;
+
+/* pointer to the beginning of the "VALUES" part */
+static char *stmt_val;
+
+/* pointer to current inser position in statement */
+static char *stmt_ins;
+
+/* our configuration directives */
+static config_entry_t db_ce = { NULL, "pgsqldb", CONFIG_TYPE_STRING,
+ CONFIG_OPT_MANDATORY, 0,
+ { } };
+
+static config_entry_t host_ce = { &db_ce, "pgsqlhost", CONFIG_TYPE_STRING,
+ CONFIG_OPT_MANDATORY, 0,
+ { } };
+
+static config_entry_t user_ce = { &host_ce, "pgsqluser", CONFIG_TYPE_STRING,
+ CONFIG_OPT_MANDATORY, 0,
+ { } };
+
+static config_entry_t pass_ce = { &user_ce, "pgsqlpass", CONFIG_TYPE_STRING,
+ CONFIG_OPT_MANDATORY, 0,
+ { } };
+
+static config_entry_t table_ce = { &pass_ce, "pgsqltable", CONFIG_TYPE_STRING,
+ CONFIG_OPT_MANDATORY, 0,
+ { } };
+
+/* is the given string a field in our table? */
+static int is_field(const char *name)
+{
+ struct _field *f;
+
+ for (f = fields; f; f = f->next) {
+ if (!strcmp(f->name, name))
+ return 1;
+ }
+ return 0;
+}
+
+/* our main output function, called by ulogd */
+static int _pgsql_output(ulog_iret_t *result)
+{
+ struct _field *f;
+ ulog_iret_t *res;
+ PGresult *pgres;
+
+ char *tmpstr;
+
+ stmt_ins = stmt_val;
+
+ for (f = fields; f; f = f->next) {
+ res = keyh_getres(f->id);
+
+ if (!res) {
+ ulogd_log(ULOGD_NOTICE,
+ "no result for %s ?!?\n", f->name);
+ }
+
+ if (!res || !IS_VALID((*res))) {
+ /* no result, we have to fake something */
+ sprintf(stmt_ins, "NULL,");
+ stmt_ins = stmt + strlen(stmt);
+ continue;
+ }
+
+ switch (res->type) {
+ case ULOGD_RET_INT8:
+ sprintf(stmt_ins, "%d,", res->value.i8);
+ break;
+ case ULOGD_RET_INT16:
+ sprintf(stmt_ins, "%d,", res->value.i16);
+ break;
+ case ULOGD_RET_INT32:
+ sprintf(stmt_ins, "%d,", res->value.i32);
+ break;
+ case ULOGD_RET_INT64:
+ sprintf(stmt_ins, "%ld,", res->value.i64);
+ break;
+ case ULOGD_RET_UINT8:
+ sprintf(stmt_ins, "%u,", res->value.ui8);
+ break;
+ case ULOGD_RET_UINT16:
+ sprintf(stmt_ins, "%u,", res->value.ui16);
+ break;
+ case ULOGD_RET_IPADDR:
+/* #ifdef IP_AS_STRING
+ *stmt_ins++ = '\'';
+ tmpstr = inet_ntoa(ntohl(res->value.ui32));
+#ifdef OLD_PGSQL
+ pgsql_escape_string(stmt_ins, tmpstr,
+ strlen(tmpstr));
+#else
+ pgsql_real_escape_string(dbh, stmt_ins,
+ tmpstr,
+ strlen(tmpstr));
+#endif /* OLD_PGSQL */
+/*
+ stmt_ins = stmt + strlen(stmt);
+ sprintf(stmt_ins, "',");
+ break;
+#endif /* IP_AS_STRING */
+ /* EVIL: fallthrough when logging IP as
+ * u_int32_t */
+
+ case ULOGD_RET_UINT32:
+ sprintf(stmt_ins, "%u,", res->value.ui32);
+ break;
+ case ULOGD_RET_UINT64:
+ sprintf(stmt_ins, "%lu,", res->value.ui64);
+ break;
+ case ULOGD_RET_BOOL:
+ sprintf(stmt_ins, "1,");
+ break;
+ case ULOGD_RET_STRING:
+ *stmt_ins++ = '\'';
+/* Change by Jaki
+#ifdef OLD_PGSQL
+ pgsql_escape_string(stmt_ins, res->value.ptr,
+ strlen(res->value.ptr));
+#else
+ pgsql_real_escape_string(dbh, stmt_ins,
+ res->value.ptr, strlen(res->value.ptr));
+#endif
+*End Change by Jaki
+*/
+ strcpy(stmt_ins,"\'");
+ strcpy(stmt_ins,res->value.ptr); /* Added by Jaki */
+ stmt_ins = stmt + strlen(stmt);
+ sprintf(stmt_ins, "',");
+ /* sprintf(stmt_ins, "'%s',", res->value.ptr); */
+ break;
+ default:
+ ulogd_log(ULOGD_NOTICE,
+ "unknown type %d for %s\n",
+ res->type, res->key);
+ break;
+ }
+ stmt_ins = stmt + strlen(stmt);
+ }
+ *(stmt_ins - 1) = ')';
+ DEBUGP("stmt=#%s#\n", stmt);
+
+ /* now we have created our statement, insert it */
+ /* Added code by Jaki */
+ pgres = PQexec(dbh, stmt);
+ if(!pgres || PQresultStatus(pgres) != PGRES_COMMAND_OK) {
+ ulogd_log(ULOGD_ERROR, "sql error during insert: %s\n",
+ PQresultErrorMessage(pgres));
+ return 1;
+ }
+
+ return 0;
+}
+
+#define PGSQL_INSERTTEMPL "insert into X (Y) values (Z)"
+#define PGSQL_VALSIZE 100
+
+/* create the static part of our insert statement */
+static int _pgsql_createstmt(void)
+{
+ struct _field *f;
+ unsigned int size;
+ char buf[ULOGD_MAX_KEYLEN];
+ char *underscore;
+
+ if (stmt) {
+ ulogd_log(ULOGD_NOTICE, "createstmt called, but stmt"
+ " already existing\n");
+ return 1;
+ }
+
+ /* caclulate the size for the insert statement */
+ size = strlen(PGSQL_INSERTTEMPL) + strlen(table_ce.u.string);
+
+ for (f = fields; f; f = f->next) {
+ /* we need space for the key and a comma, as well as
+ * enough space for the values */
+ size += strlen(f->name) + 1 + PGSQL_VALSIZE;
+ }
+
+ ulogd_log(ULOGD_DEBUG, "allocating %u bytes for statement\n", size);
+
+ stmt = (char *) malloc(size);
+
+ if (!stmt) {
+ ulogd_log(ULOGD_ERROR, "OOM!\n");
+ return 1;
+ }
+
+ sprintf(stmt, "insert into %s (", table_ce.u.string);
+ stmt_val = stmt + strlen(stmt);
+
+ for (f = fields; f; f = f->next) {
+ strncpy(buf, f->name, ULOGD_MAX_KEYLEN);
+ while (underscore = strchr(buf, '.'))
+ *underscore = '_';
+ sprintf(stmt_val, "%s,", buf);
+ stmt_val = stmt + strlen(stmt);
+ }
+ *(stmt_val - 1) = ')';
+
+ sprintf(stmt_val, " values (");
+ stmt_val = stmt + strlen(stmt);
+
+ ulogd_log(ULOGD_DEBUG, "stmt='%s'\n", stmt);
+
+ return 0;
+}
+
+/* find out which columns the table has */
+static int _pgsql_get_columns(const char *table)
+{
+ PGresult *result;
+ char buf[ULOGD_MAX_KEYLEN];
+ char pgbuf[256];
+ char *underscore;
+ struct _field *f;
+ int id;
+ int intaux;
+ char *sql;
+
+ if (!dbh)
+ return 1;
+
+ strcpy(pgbuf, "SELECT a.attname FROM pg_class c, pg_attribute a WHERE c.relname ='");
+ strncat(pgbuf, table, strlen(table));
+ strcat(pgbuf, "' AND a.attnum>0 AND a.attrelid=c.oid ORDER BY a.attnum");
+ sql = (char *) malloc(strlen(pgbuf));
+ sql = pgbuf;
+ ulogd_log(ULOGD_DEBUG, sql);
+ result = PQexec(dbh, sql);
+ if(!result) {
+ ulogd_log(ULOGD_DEBUG, "\n result false");
+ return 1;
+ }
+ if (PQresultStatus(result) != PGRES_TUPLES_OK) {
+ ulogd_log(ULOGD_DEBUG, "\n pres_command_not_ok");
+ return 1;
+ }
+ for (intaux=0; intaux<PQntuples(result); intaux++) {
+
+ /* replace all underscores with dots */
+ strncpy(buf, PQgetvalue(result, intaux, 0), ULOGD_MAX_KEYLEN);
+ while (underscore = strchr(buf, '_'))
+ *underscore = '.';
+
+ DEBUGP("field '%s' found: ", buf);
+
+ if (!(id = keyh_getid(buf))) {
+ DEBUGP(" no keyid!\n");
+ continue;
+ }
+
+ DEBUGP("keyid %u\n", id);
+
+ /* prepend it to the linked list */
+ f = (struct _field *) malloc(sizeof *f);
+ if (!f) {
+ ulogd_log(ULOGD_ERROR, "OOM!\n");
+ return 1;
+ }
+ strncpy(f->name, buf, ULOGD_MAX_KEYLEN);
+ f->id = id;
+ f->next = fields;
+ fields = f;
+ }
+
+ PQclear(result);
+ return 0;
+}
+
+int exit_nicely(PGconn *conn)
+{
+ PQfinish(conn);
+ return 0;;
+}
+
+/* make connection and select database */
+static int _pgsql_open_db(char *server, char *user, char *pass, char *db)
+{
+ char connstr[80];
+ char * sql;
+
+ strcpy(connstr, " host=");
+ strcat(connstr, server);
+ strcat(connstr, " dbname=");
+ strcat(connstr, db);
+ strcat(connstr, " user=");
+ strcat(connstr, user);
+ strcat(connstr, " password=");
+ strcat(connstr, pass);
+ sql = (char *) malloc(strlen(connstr));
+
+ sql = connstr;
+ dbh = PQconnectdb(sql);
+ if (PQstatus(dbh)!=CONNECTION_OK) {
+ exit_nicely(dbh);
+ return 1;
+ }
+
+
+ return 0;
+}
+
+
+static ulog_output_t _pgsql_plugin = { NULL, "pgsql", &_pgsql_output, NULL };
+
+void _init(void)
+{
+ /* register our configfile options here */
+ config_register_key(&table_ce);
+
+ /* have the opts parsed */
+ config_parse_file(0);
+
+ if (_pgsql_open_db(host_ce.u.string, user_ce.u.string,
+ pass_ce.u.string, db_ce.u.string)) {
+ ulogd_log(ULOGD_ERROR, "can't establish database connection\n");
+ return;
+ }
+
+ /* read the fieldnames to know which values to insert */
+ if (_pgsql_get_columns(table_ce.u.string)) {
+ ulogd_log(ULOGD_ERROR, "unable to get pgsql columns\n");
+ return;
+ }
+ _pgsql_createstmt();
+ register_output(&_pgsql_plugin);
+
+}
projects / thirdparty / ulogd2.git / commitdiff
