db_import.c

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

#ifndef MYSQL_USER
#define MYSQL_USER "root"
#endif
#ifndef MYSQL_PASS
#define MYSQL_PASS ""
#endif
#ifndef MYSQL_DBNAME
#define MYSQL_DBNAME "celldb"
#endif

#ifndef KDB_USER
#define KDB_USER "root"
#endif
#ifndef KDB_PASS
#define KDB_PASS ""
#endif
#ifndef KDB_DBNAME
#define KDB_DBNAME "celldb"
#endif

#include <omp.h>
#define __USE_XOPEN 1
#include <time.h>

#include "l3_handler.h"
#include "bit_func.h"
#include "assignment.h"
#include "session.h"
#include "cell_info.h"
#include "process.h"

int explore_session(int id)
{
	MYSQL r_conn, w_conn, *test;
	MYSQL_RES *result;
	MYSQL_ROW row;
	int ret;
	char *cret;
	int mcc, mnc, lac, cid, cracked;
	char query[4096];
	struct session_info *s;
	struct tm tm;

	memset(&r_conn, 0, sizeof(r_conn));

	test = mysql_real_connect(&r_conn, "10.0.0.1", KDB_USER, KDB_PASS, KDB_DBNAME, 3306, 0, 0);
	if (test == 0) {
		printf("Cannot connect to R-database\n");
		return -1;
	}

	snprintf(query, sizeof(query),	"select mcc, mnc, lac, cid, timestamp, success from session"
				 	" where id = %d order by mcc, mnc, lac, cid", id );

	ret = mysql_query(&r_conn, query);
	if (ret != 0) {
		printf("Cannot execute query: %s\n", query);
		return -1;
	}

	result = mysql_use_result(&r_conn);
	if (result == 0) {
		printf("Cannot get query result\n");
		return -1;
	}

	row = mysql_fetch_row(result);
	if (row[0] == 0) {
		printf("Error reading row from result\n");
		return -1;
	}

	mcc = atoi(row[0]);
	mnc = atoi(row[1]);
	lac = atoi(row[2]);
	cid = atoi(row[3]);
	cracked = atoi(row[5]);

	memset(&tm, 0, sizeof(tm));
	cret = strptime(row[4], "%Y-%m-%d %T", &tm);
	if (!cret || cret[0]) {
		printf("Error parsing timestamp at %s\n", cret);
		return -1;
	}

	mysql_free_result(result);

	s = session_create(id, NULL, NULL, mcc, mnc, lac, cid, NULL);
	if (s == NULL) {
		printf("Cannot allocate session structure\n");
		return -1;
	}

	s->timestamp.tv_sec = mktime(&tm);
	s->cracked = cracked;
	s->started = 1;

	s->sql_callback = _s[0].sql_callback;

	snprintf(query, sizeof(query),	"select frameno, channel, uplink, data from session_frame"
					" where session = %d order by frameno, channel, uplink", id);

	ret = mysql_query(&r_conn, query);
	if (ret != 0) {
		printf("Cannot execute query: %s\n", query);
		return -1;
	}

	result = mysql_use_result(&r_conn);
	if (result == 0) {
		printf("Cannot get query result\n");
		return -1;
	}

	while ((row = mysql_fetch_row(result)) && row[0]) {
		int fn = atoi(row[0]);
		int channel = atoi(row[1]);
		int uplink = atoi(row[2]);
		char *data = row[3];
		struct radio_message *m;

		m = (struct radio_message *) malloc(sizeof(struct radio_message));
		if (m == 0)
			return 0;

		memset(m, 0, sizeof(struct radio_message));

		m->rat = RAT_GSM;
		m->domain = DOMAIN_CS;

		switch(channel) {
		case 100:
			m->flags = MSG_SDCCH;
			m->chan_nr = 0x41;
			memcpy(m->msg, data, 23);
			break;
		case 97:
			m->flags = MSG_SACCH;
			m->chan_nr = 0x41;
			memcpy(m->msg, data, 23);
			break;
		default:
			printf("unhandled channel %d in session %d\n", channel, id);
			fflush(stdout);
			free(m);
			continue;
		}
		m->msg_len = 23;
		m->flags |= MSG_DECODED;
		m->bb.fn[0] = fn;
		m->bb.arfcn[0] = (uplink ? ARFCN_UPLINK : 0);

		handle_radio_msg(s, m);
	}

	mysql_free_result(result);

	mysql_close(&r_conn);

	memset(&w_conn, 0, sizeof(w_conn));

	test = mysql_real_connect(&w_conn, "127.0.0.1", MYSQL_USER, MYSQL_PASS, MYSQL_DBNAME, 3306, 0, 0);
	while (test == 0) {
		printf("Cannot connect to W-database\n");
		sleep(1);
		memset(&w_conn, 0, sizeof(w_conn));
		test = mysql_real_connect(&w_conn, "127.0.0.1", MYSQL_USER, MYSQL_PASS, MYSQL_DBNAME, 3306, 0, 0);
	}

	snprintf(query, sizeof(query), "delete from session_info where id = %d", id);

	ret = mysql_query(&w_conn, query);
	if (ret != 0) {
		printf("Cannot execute query: %s\n", query);
		return -1;
	}

	snprintf(query, sizeof(query), "delete from sms_meta where id = %d", id);

	ret = mysql_query(&w_conn, query);
	if (ret != 0) {
		printf("Cannot execute query: %s\n", query);
		return -1;
	}

	/* Write to database */
	session_close(s);

	session_free(s);

	mysql_close(&w_conn);

	return 0;
}

int main(int argc, char **argv)
{
	MYSQL conn, *test;
	MYSQL_RES *result;
	MYSQL_ROW row;
	int ret;
	unsigned i;
	unsigned int row_count;
	char query[128];
	int *session_id;
	int start_id = 0;

	session_init(0, 0, NULL, CALLBACK_MYSQL);
	cell_init(0, 0, CALLBACK_MYSQL);
	auto_reset = 0;
	auto_timestamp = 0;

	if (argc == 2) {
		int s_id = atoi(argv[1]);

		msg_verbose = 1;

		printf("Session %d\n", s_id);

		explore_session(s_id);

		return 0;
	}

	/* Find last process session id */

	memset(&conn, 0, sizeof(conn));

	test = mysql_real_connect(&conn, "127.0.0.1", MYSQL_USER, MYSQL_PASS, MYSQL_DBNAME, 3306, 0, 0);
	if (test == 0) {
		printf("Cannot connect to database\n");
		return -1;
	}

	snprintf(query, sizeof(query), "select max(id) from session_info where id < 8000000;");

	ret = mysql_query(&conn, query);
	if (ret != 0) {
		printf("Cannot execute query: %s\n", query);
		return -1;
	}

	result = mysql_use_result(&conn);
	if (result == 0) {
		printf("Cannot get query result\n");
		return -1;
	}

	row = mysql_fetch_row(result);
	if (!row) {
		printf("Cannot get the starting session ID\n");
		return -1;
	}
	if (row[0]) {
		start_id = atoi(row[0]);
	}

	mysql_free_result(result);

	mysql_close(&conn);

	memset(&conn, 0, sizeof(conn));

	/* Get all the missing session ids */
	test = mysql_real_connect(&conn, "10.0.0.1", KDB_USER, KDB_PASS, KDB_DBNAME, 3306, 0, 0);
	if (test == 0) {
		printf("Cannot connect to database\n");
		return -1;
	}

	snprintf(query, sizeof(query), "select id from session where id > %d order by id;", start_id);

	ret = mysql_query(&conn, query);
	if (ret != 0) {
		printf("Cannot execute query: %s\n", query);
		return -1;
	}

	result = mysql_store_result(&conn);
	if (result == 0) {
		printf("Cannot get query result\n");
		return -1;
	}

	row_count = mysql_num_rows(result);

	printf("Running from %d sessions from ID %d ...\n", row_count, start_id);

	session_id = malloc(row_count*sizeof(*session_id));
	if (!session_id) {
		printf("Cannot allocate memory for session IDs\n");
		return -1;
	}

	for (i = 0; i < row_count; i++) {
		row = mysql_fetch_row(result);
		if (row[0] == 0) {
			printf("Cannot parse session ID\n");
			return -1;
		}
		session_id[i] = atoi(row[0]);
	}

	mysql_free_result(result);
	mysql_close(&conn);

	/* Read and process each session */

	//#pragma omp parallel for num_threads (10)
	for (i = 0; i < row_count; i++) {

		printf("Session %d\n", session_id[i]);

		ret = explore_session(session_id[i]);
		if (ret < 0) {
			printf("Terminating.\n");
			return -1;
		}
	}

	free(session_id);

	cell_and_paging_dump(0, 1, 1);

	return 0;
}