Snort_AIPreproc/mysql.c

172 lines
3.2 KiB
C
Raw Normal View History

/*
* =====================================================================================
*
* Filename: mysql.c
*
* Description: Interface to a MySQL database
*
* Version: 0.1
* Created: 04/09/2010 20:16:47
* Revision: none
* Compiler: gcc
*
* Author: BlackLight (http://0x00.ath.cx), <blacklight@autistici.org>
* Licence: GNU GPL v.3
* Company: DO WHAT YOU WANT CAUSE A PIRATE IS FREE, YOU ARE A PIRATE!
*
* =====================================================================================
*/
2010-09-11 02:12:39 +02:00
#include "spp_ai.h"
#ifdef HAVE_LIBMYSQLCLIENT
#include <mysql/mysql.h>
2010-09-11 02:12:39 +02:00
/** \defgroup mysql Module for the interface with a MySQL DBMS
* @{ */
/***************************/
/* Database descriptors */
PRIVATE MYSQL *db = NULL;
PRIVATE MYSQL *outdb = NULL;
/***************************/
/*************************************************************/
/* Private functions (operating on the database descriptors) */
PRIVATE BOOL
__mysql_is_init ( MYSQL *__DB )
{
return ( __DB != NULL );
}
PRIVATE void*
__mysql_do_init ( MYSQL **__DB, BOOL is_out )
{
if ( __mysql_is_init ( *__DB ) )
return (void*) *__DB;
if ( !( *__DB = (MYSQL*) malloc ( sizeof ( MYSQL ))))
return NULL;
if ( !( mysql_init ( *__DB )))
return NULL;
if ( is_out )
{
if ( !mysql_real_connect ( *__DB, config->outdbhost, config->outdbuser, config->outdbpass, NULL, 0, NULL, 0 ))
return NULL;
if ( mysql_select_db ( *__DB, config->outdbname ))
return NULL;
} else {
if ( !mysql_real_connect ( *__DB, config->dbhost, config->dbuser, config->dbpass, NULL, 0, NULL, 0 ))
return NULL;
if ( mysql_select_db ( *__DB, config->dbname ))
return NULL;
}
return (void*) *__DB;
}
PRIVATE MYSQL_RES*
__mysql_do_query ( MYSQL *__DB, const char *query )
{
MYSQL_RES *res = NULL;
if ( mysql_query ( __DB, query ))
return NULL;
if ( !( res = mysql_store_result ( __DB )))
return NULL;
return res;
}
PRIVATE void
__mysql_do_close ( MYSQL **__DB )
{
if ( *__DB )
mysql_close ( *__DB );
free ( *__DB );
*__DB = NULL;
}
/* End of private functions */
/****************************/
/********************/
/* Public functions */
BOOL
mysql_is_init ()
{
return __mysql_is_init ( db );
}
void*
mysql_do_init ()
{
return __mysql_do_init ( &db, false );
}
MYSQL_RES*
mysql_do_query ( const char *query )
{
return __mysql_do_query ( db, query );
}
unsigned long
mysql_do_escape_string ( char **to, const char *from, unsigned long length )
{
return mysql_real_escape_string ( db, *to, from, length );
}
void
mysql_do_close ()
{
__mysql_do_close ( &db );
}
/* Output database functions */
BOOL
mysql_is_out_init ()
{
return __mysql_is_init ( outdb );
}
void*
mysql_do_out_init ()
{
return __mysql_do_init ( &outdb, true );
}
MYSQL_RES*
mysql_do_out_query ( const char *query )
{
return __mysql_do_query ( outdb, query );
}
unsigned long
mysql_do_out_escape_string ( char **to, const char *from, unsigned long length )
{
return mysql_real_escape_string ( outdb, *to, from, length );
}
void
mysql_do_out_close ()
{
__mysql_do_close ( &outdb );
}
/* End of public functions */
/***************************/
2010-09-11 02:12:39 +02:00
/** @} */
#endif