77 lines
2.0 KiB
C
77 lines
2.0 KiB
C
#include <stdio.h>
|
|
#include <sys/socket.h>
|
|
#include <netinet/in.h>
|
|
#include <arpa/inet.h>
|
|
|
|
#include <mysql/mysql.h>
|
|
|
|
#include "configfile.h"
|
|
#include "conntrack.h"
|
|
#include "database.h"
|
|
#include "hashtable.h"
|
|
#include "logging.h"
|
|
|
|
static MYSQL *dbh;
|
|
|
|
static void purge_hash_cb(const struct hash_entry *entry, void *privdata)
|
|
{
|
|
int long *now = (long *)privdata;
|
|
char query[256];
|
|
|
|
int len = snprintf(query, sizeof(query),
|
|
"INSERT INTO stats SET timestamp='%lu', srcip='%u', proto='%u', dport='%u', srcbytes='%llu', dstbytes='%llu', count='%u'",
|
|
*now, ntohl(entry->src_ip), entry->protonum, ntohs(entry->dst_port),
|
|
entry->src_bytes, entry->dst_bytes, entry->count);
|
|
|
|
if (mysql_real_query(dbh, query, len +1) != 0)
|
|
log_print(LOG_WARN, "purge_hash_cb: mysql_real_query(): %s", mysql_error(dbh));
|
|
}
|
|
|
|
int database_analyse(void)
|
|
{
|
|
struct hash_table *hash = conntrack_get_hash();
|
|
|
|
if (mysql_ping(dbh) != 0) {
|
|
log_print(LOG_WARN, "database_analyse: mysql_ping(): %s", mysql_error(dbh));
|
|
purge_hash(hash, NULL, NULL);
|
|
return 0;
|
|
}
|
|
|
|
long now = time(NULL);
|
|
purge_hash(hash, purge_hash_cb, (void *)&now);
|
|
return 0;
|
|
}
|
|
|
|
int database_init(void)
|
|
{
|
|
dbh = mysql_init(NULL);
|
|
if (dbh == NULL) {
|
|
log_print(LOG_ERROR, "database_init: mysql_init(): %s", mysql_error(dbh));
|
|
return -1;
|
|
}
|
|
|
|
const char *hostname = config_get_string("mysql", "hostname", NULL);
|
|
const char *username = config_get_string("mysql", "username", NULL);
|
|
const char *password = config_get_string("mysql", "password", NULL);
|
|
MYSQL *ret = mysql_real_connect(dbh, hostname, username, password, NULL, 0, NULL, 0);
|
|
if (ret != dbh) {
|
|
log_print(LOG_ERROR, "database_init: mysql_real_connect(): %s", mysql_error(dbh));
|
|
mysql_close(dbh);
|
|
return -1;
|
|
}
|
|
|
|
const char *database = config_get_string("mysql", "database", NULL);
|
|
if (mysql_select_db(dbh, database) != 0) {
|
|
log_print(LOG_ERROR, "database_init: mysql_select_db(): %s", mysql_error(dbh));
|
|
mysql_close(dbh);
|
|
return -1;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
int database_close(void)
|
|
{
|
|
mysql_close(dbh);
|
|
return 0;
|
|
}
|