/* open fire host and slora report data database */
int open_report_db(void)
{
int rv = -;
char sql[SQL_COMMAND_LEN];
char *errmsg = NULL;
const char *db_file = REPORT_DB_FILE; /* If report database already exist, then we will just open it */
if( ==access(db_file, F_OK) )
{
if( SQLITE_OK != sqlite3_open(db_file, &s_report_db) )
{
log_err("open report database file '%s' failure\n", db_file);
return -;
}
log_nrml("open report database file '%s' ok\n", db_file);
return ;
} pthread_mutex_lock(&s_reportdb_lock); /* If report database file not exist, then we create it now */
if( SQLITE_OK != sqlite3_open(db_file, &s_report_db) )
{
log_err("create report database file '%s' failure\n", db_file);
pthread_mutex_unlock(&s_reportdb_lock);
return -;
}
log_nrml("open report database file '%s' ok\n", db_file); /* Auto increase/decrease */
snprintf(sql, sizeof(sql), "pragma auto_vacuum=2;");
if( SQLITE_OK != sqlite3_exec(s_report_db, sql, NULL, NULL, &errmsg) )
{
log_err("set pragma auto_vacuum=2 failure: %s\n", db_file, errmsg);
rv = -;
goto FAILED;
} /* Create firehost table in the database */
strncpy(sql, "CREATE TABLE firehost(id INTEGER PRIMARY KEY autoincrement, sent BOOL DEFAULT 0, packet BLOB);", sizeof(sql));
if( SQLITE_OK != sqlite3_exec(s_report_db, sql, NULL, NULL, &errmsg) )
{
log_err("create firehost table in report database file '%s' failure: %s\n", db_file, errmsg);
rv = -;
goto FAILED;
} /* Create elora table in the database */
strncpy(sql, "CREATE TABLE elora(id INTEGER PRIMARY KEY autoincrement, sent BOOL DEFAULT 0, packet BLOB);", sizeof(sql));
if( SQLITE_OK != sqlite3_exec(s_report_db, sql, NULL, NULL, &errmsg) )
{
log_err("create elora table in report database file '%s' failure: %s\n", db_file, errmsg);
rv = -;
goto FAILED;
} log_nrml("create report database file '%s' ok\n", db_file);
pthread_mutex_unlock(&s_reportdb_lock);
return ; FAILED:
log_err("Create report database '%s' failure and remove it\n", db_file);
sqlite3_close(s_report_db);
sqlite3_free(errmsg);
unlink(db_file);
pthread_mutex_unlock(&s_reportdb_lock);
return rv;
} int add_reportdb_firehost(firehost_pack_t *pack)
{
int rv;
sqlite3_stmt *stat = NULL; if( !pack )
{
log_err("invalid input arguments\n");
return -;
} if(!s_report_db)
{
log_nrml("report database not opened, try to open it now\n");
if( open_report_db() < )
return -;
} pthread_mutex_lock(&s_reportdb_lock); rv = sqlite3_prepare_v2(s_report_db, "insert into firehost(packet)values(?)", -, &stat, NULL);
if(SQLITE_OK!=rv || !stat)
{
log_err("firehost sqlite3_prepare_v2 failure\n");
rv = -;
goto out;
} cp_log_dump(LOG_LEVEL_NRML, (char *)pack, sizeof(*pack)); if( SQLITE_OK != sqlite3_bind_blob(stat, , pack, sizeof(*pack), NULL) )
{
log_err("firehost sqlite3_bind_blob failure\n");
rv = -;
goto out;
} rv = sqlite3_step(stat);
if( SQLITE_DONE!=rv && SQLITE_ROW!=rv )
{
log_err("firehost sqlite3_step failure\n");
rv = -;
goto out;
} sqlite3_finalize(stat); pthread_mutex_unlock(&s_reportdb_lock);
log_nrml("add new firehost report data into database ok\n"); return ; out:
pthread_mutex_unlock(&s_reportdb_lock);
log_nrml("add new firehost report data into database failure, rv=%d\n", rv);
return rv;
} int mark_reportdb_firehost_sent(int record_id)
{
char sql[SQL_COMMAND_LEN];
char *errmsg = NULL; if(record_id <= )
{
log_err("invalid input arguments\n");
return -;
} if(!s_report_db)
{
log_nrml("report database not opened, try to open it now\n");
if( open_report_db() < )
return -;
} pthread_mutex_lock(&s_reportdb_lock); memset(sql, , sizeof(sql));
snprintf(sql, sizeof(sql), "update firehost set sent=1 where id=%d;", record_id);
if( SQLITE_OK != sqlite3_exec(s_report_db, sql, NULL, NULL, &errmsg) )
{
log_err("mark firehost packet[%d] already sent from database failure: %s\n", record_id, errmsg);
sqlite3_free(errmsg);
pthread_mutex_unlock(&s_reportdb_lock);
return -;
} pthread_mutex_unlock(&s_reportdb_lock);
log_nrml("mark firehost packet[%d] already sent from database ok\n", record_id);
return ;
} int query_reportdb_firehost_unsent(int *record_id, firehost_pack_t *pack)
{
int rv;
sqlite3_stmt *stat = NULL;
const void *blob_ptr; if( !record_id || !pack )
{
log_err("invalid input arguments\n");
return -;
} if(!s_report_db)
{
log_nrml("report database not opened, try to open it now\n");
if( open_report_db() < )
return -;
} pthread_mutex_lock(&s_reportdb_lock); /* Only query the first unsent packet record */
rv = sqlite3_prepare_v2(s_report_db, "select id,packet from firehost where sent=0 limit 0,1;", -, &stat, NULL);
if(SQLITE_OK!=rv || !stat)
{
log_err("firehost sqlite3_prepare_v2 failure\n");
rv = -;
goto out;
} rv = sqlite3_step(stat);
if( SQLITE_DONE!=rv && SQLITE_ROW!=rv )
{
log_err("firehost sqlite3_step failure\n");
rv = -;
goto out;
} /* 2rd argument<0> means first segement is id */
*record_id = sqlite3_column_int(stat, ); /* 2rd argument<1> means first segement is id */
//blob_bytes = sqlite3_column_bytes(stat, 1 );
blob_ptr = sqlite3_column_blob(stat, ); memcpy(pack, blob_ptr, sizeof(*pack)); log_trace("Read firehost packet from database:\n");
cp_log_dump(LOG_LEVEL_TRACE, (char *)pack, sizeof(*pack)); sqlite3_finalize(stat); pthread_mutex_unlock(&s_reportdb_lock);
log_nrml("query new firehost report data into database ok\n");
return ; out:
pthread_mutex_unlock(&s_reportdb_lock);
log_nrml("query new firehost report data into database failure, rv=%d\n", rv);
return rv;
}