From 17c8a5a5f7d79f44e5a05a96050673be566c4adf Mon Sep 17 00:00:00 2001 From: Edward Rudd Date: Wed, 28 Jul 2004 18:04:44 +0000 Subject: added dbi provider --- mod_log_sql_dbi.c | 263 ++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 263 insertions(+) create mode 100644 mod_log_sql_dbi.c diff --git a/mod_log_sql_dbi.c b/mod_log_sql_dbi.c new file mode 100644 index 0000000..8afd844 --- /dev/null +++ b/mod_log_sql_dbi.c @@ -0,0 +1,263 @@ +/* $Id: mod_log_sql_dbi.c 120 2004-04-17 15:14:12Z urkle@drip.ws $ */ + +#if defined(WITH_APACHE20) +# include "apache20.h" +#elif defined(WITH_APACHE13) +# include "apache13.h" +#else +# error Unsupported Apache version +#endif + + +#ifdef HAVE_CONFIG_H +/* Undefine these to prevent conflicts between Apache ap_config_auto.h and + * my config.h. Only really needed for Apache < 2.0.48, but it can't hurt. + */ +#undef PACKAGE_BUGREPORT +#undef PACKAGE_NAME +#undef PACKAGE_STRING +#undef PACKAGE_TARNAME +#undef PACKAGE_VERSION + +#include "config.h" +#endif + +#include "mod_log_sql.h" + +#include "dbi/dbi.h" + +typedef struct { + dbi_conn conn; +} dbi_conn_rec; + +/* Connect to the MYSQL database */ +static logsql_opendb_ret log_sql_dbi_connect(server_rec *s, logsql_dbconnection *db) +{ + const char *driver = apr_table_get(db->parms,"driver"); + const char *host = apr_table_get(db->parms,"hostname"); + const char *user = apr_table_get(db->parms,"username"); + const char *passwd = apr_table_get(db->parms,"password"); + const char *database = apr_table_get(db->parms,"database"); + const char *s_tcpport = apr_table_get(db->parms,"port"); + unsigned int tcpport = (s_tcpport)?atoi(s_tcpport):0; + const char *socketfile = apr_table_get(db->parms,"socketfile"); + //dbi_result result; + dbi_conn_rec *dblink = db->handle; + if (!dblink) { + dblink = apr_pcalloc(db->p, sizeof(*dblink)); + db->handle = (void *)dblink; + } + + dblink->conn = dbi_conn_new(driver); + + dbi_conn_set_option(dblink->conn, "host", host); + dbi_conn_set_option(dblink->conn, "username", user); + dbi_conn_set_option(dblink->conn, "password", passwd); + dbi_conn_set_option(dblink->conn, "dbname", database); + if (tcpport) { + dbi_conn_set_option_numeric(dblink->conn, "port", tcpport); + } + + if (socketfile && !strcmp(driver,"mysql")) { + dbi_conn_set_option(dblink->conn, "mysql_unix_socket", socketfile); + } + + if (!dbi_conn_connect(dblink->conn)) { + log_error(APLOG_MARK, APLOG_DEBUG, 0, s, + "HOST: '%s' PORT: '%d' DB: '%s' USER: '%s' SOCKET: '%s'", + host, tcpport, database, user, socketfile); + return LOGSQL_OPENDB_SUCCESS; + } else { + const char *error; + dbi_conn_error(dblink->conn, &error); + log_error(APLOG_MARK, APLOG_ERR, 0, s, + "DBI Error: %s", error); + log_error(APLOG_MARK, APLOG_DEBUG, 0, s, + "HOST: '%s' PORT: '%d' DB: '%s' USER: '%s' SOCKET: '%s'", + host, tcpport, database, user, socketfile); + return LOGSQL_OPENDB_FAIL; + } +} + +/* Close the DB link */ +static void log_sql_dbi_close(logsql_dbconnection *db) +{ + dbi_conn_rec *dblink = db->handle; + dbi_conn_close(dblink->conn); + dblink->conn = NULL; +} + +/* Routine to escape the 'dangerous' characters that would otherwise + * corrupt the INSERT string: ', \, and " + */ +static const char *log_sql_dbi_escape(const char *from_str, apr_pool_t *p, + logsql_dbconnection *db) +{ + dbi_conn_rec *dblink = db->handle; + + if (!from_str) + return NULL; + else { + char *to_str = strdup(from_str); + char *retval; + dbi_driver_quote_string(dbi_conn_get_driver(dblink->conn), &to_str); + retval = apr_pstrdup(p, to_str); + free(to_str); + return retval; + } +} + +/* Run a mysql insert query and return a categorized error or success */ +static logsql_query_ret log_sql_dbi_query(request_rec *r,logsql_dbconnection *db, + const char *query) +{ + const char *error; + dbi_result result; + dbi_conn_rec *dblink = db->handle; + + if (!dblink->conn) { + return LOGSQL_QUERY_NOLINK; + } + + /* Run the query */ + if ((result = dbi_conn_query(dblink->conn, query))) { + return LOGSQL_QUERY_SUCCESS; + } + /* Check to see if the error is "nonexistent table" */ + dbi_conn_error(dblink->conn, &error); + log_error(APLOG_MARK, APLOG_ERR, 0, r->server, + "DBI Error: %s", error); +/* if (real_error == ER_NO_SUCH_TABLE) { + log_error(APLOG_MARK,APLOG_ERR,0, r->server,"table does not exist, preserving query"); + return LOGSQL_QUERY_NOTABLE; + }*/ + return LOGSQL_QUERY_FAIL; +} + +/* Create table table_name of type table_type. */ +static logsql_table_ret log_sql_dbi_create(request_rec *r, logsql_dbconnection *db, + logsql_tabletype table_type, const char *table_name) +{ + dbi_result result; + const char *driver = apr_table_get(db->parms,"driver"); + const char *tabletype = apr_table_get(db->parms,"tabletype"); + char *type_suffix = NULL; + + char *create_prefix = "create table if not exists `"; + char *create_suffix = NULL; + char *create_sql; + + dbi_conn_rec *dblink = db->handle; + +/* if (!global_config.createtables) { + return APR_SUCCESS; + }*/ + + switch (table_type) { + case LOGSQL_TABLE_ACCESS: + create_suffix = + "` (id char(19),\ + agent varchar(255),\ + bytes_sent int unsigned,\ + child_pid smallint unsigned,\ + cookie varchar(255),\ + machine_id varchar(25),\ + request_file varchar(255),\ + referer varchar(255),\ + remote_host varchar(50),\ + remote_logname varchar(50),\ + remote_user varchar(50),\ + request_duration smallint unsigned,\ + request_line varchar(255),\ + request_method varchar(10),\ + request_protocol varchar(10),\ + request_time char(28),\ + request_uri varchar(255),\ + request_args varchar(255),\ + server_port smallint unsigned,\ + ssl_cipher varchar(25),\ + ssl_keysize smallint unsigned,\ + ssl_maxkeysize smallint unsigned,\ + status smallint unsigned,\ + time_stamp int unsigned,\ + virtual_host varchar(255))"; + break; + case LOGSQL_TABLE_COOKIES: + case LOGSQL_TABLE_HEADERSIN: + case LOGSQL_TABLE_HEADERSOUT: + case LOGSQL_TABLE_NOTES: + create_suffix = + "` (id char(19),\ + item varchar(80),\ + val varchar(80))"; + break; + } + + if (tabletype && !strcmp(driver,"mysql")) { + type_suffix = apr_pstrcat(r->pool, " TYPE=", + tabletype, NULL); + } + /* Find memory long enough to hold the whole CREATE string + \0 */ + create_sql = apr_pstrcat(r->pool, create_prefix, table_name, create_suffix, + type_suffix, NULL); + + log_error(APLOG_MARK,APLOG_DEBUG,0, r->server,"create string: %s", create_sql); + + if (!dblink) { + return LOGSQL_QUERY_NOLINK; + } + + /* Run the create query */ + if (!(result = dbi_conn_query(dblink, create_sql))) { + const char *error; + dbi_conn_error(dblink->conn, &error); + log_error(APLOG_MARK, APLOG_ERR, 0, r->server, + "DBI Error: %s", error); + return LOGSQL_TABLE_FAIL; + } + + return LOGSQL_TABLE_SUCCESS; +} + +static logsql_dbdriver log_sql_dbi_driver = { + "dbi", + NULL, + log_sql_dbi_connect, /* open DB connection */ + log_sql_dbi_close, /* close DB connection */ + log_sql_dbi_escape, /* escape query */ + log_sql_dbi_query, /* insert query */ + log_sql_dbi_create /* create table */ +}; + +static apr_status_t log_sql_dbi_cleanup(void *data) +{ + dbi_shutdown(); +#if defined(WITH_APACHE20) + return APR_SUCCESS; +#endif +} + +LOGSQL_REGISTER(dbi) { + dbi_driver driver; + const char **driver_list; + int count = 1; + + dbi_initialize(NULL); + + for (driver = dbi_driver_list(NULL); + driver; + driver = dbi_driver_list(driver)) { + count++; + } + driver_list = apr_pcalloc(p, sizeof(char *) * (count)); + count = 0; + for (driver = dbi_driver_list(NULL); + driver; + driver = dbi_driver_list(driver)) { + driver_list[count++] = dbi_driver_get_name(driver); + } + log_sql_dbi_driver.provided_drivers = driver_list; + log_sql_register_driver(p,&log_sql_dbi_driver); + apr_pool_cleanup_register(p, NULL, log_sql_dbi_cleanup, NULL); + LOGSQL_REGISTER_RETURN; +} -- cgit