--- mqtt/src/pubmqtt.c 2011/11/22 21:12:52 1.1.2.1 +++ mqtt/src/pubmqtt.c 2011/11/24 13:57:12 1.1.2.4 @@ -1,4 +1,108 @@ #include "global.h" -#include "dbmqtt.h" +/* + * mqtt_db_log() Log database connection message + * + * @fmt = format string + * @... = argument list + * return: none + */ +static void +mqtt_rtlm_log(const char *fmt, ...) +{ + va_list lst; + + va_start(lst, fmt); + vsyslog(LOG_ERR, fmt, lst); + va_end(lst); +} +#define MQTT_RTLM_LOG(_sql) (assert((_sql)), mqtt_rtlm_log("Error:: SQL #%d - %s", \ + sqlite3_errcode((_sql)), sqlite3_errmsg((_sql)))) + + +/* + * mqtt_rtlm_open() Open database connection + * + * @cfg = config filename + * return: NULL error or SQL handle + */ +sqlite3 * +mqtt_rtlm_open(sl_config *cfg) +{ + sqlite3 *sql = NULL; + const char *str = NULL; + + if (!cfg) + return NULL; + + str = (const char*) cfg_GetAttribute(cfg, CFG("mqtt_pub"), CFG("name")); + if (!str) { + mqtt_rtlm_log("Error:: Unknown database name ...\n"); + return NULL; + } + + if (sqlite3_open_v2(str, &sql, SQLITE_OPEN_READWRITE, NULL)) { + MQTT_RTLM_LOG(sql); + sqlite3_close(sql); + return NULL; + } + + return sql; +} + +/* + * mqtt_rtlm_close() Close database connection + * + * @sql = SQL handle + * return: none + */ +void +mqtt_rtlm_close(sqlite3 *sql) +{ + sqlite3_close(sql); +} + +/* + * mqtt_rtlm_init_session() Create session + * + * @cfg = loaded config + * @sql = SQL handle + * @user = username + * @host = hostname + * @port = port + * return: -1 error, 0 session already appears or >0 row changed + */ +int +mqtt_rtlm_init_session(sl_config *cfg, sqlite3 *sql, const char *user, const char *host, u_short port) +{ + int ret = 0; + char *str, szStmt[BUFSIZ] = { 0 }; + sqlite3_stmt *stmt; + + if (!cfg || !sql) + return -1; + + str = (char*) cfg_GetAttribute(cfg, CFG("mqtt_pub"), CFG("tbl_online")); + if (!str) { + mqtt_rtlm_log("Error:: not found topics table name"); + return -1; + } + snprintf(szStmt, sizeof szStmt, "INSERT INTO %s (Username, RemoteHost, RemotePort) " + "VALUES ('%s', '%s', %d);", str, user, host, port); + + if (sqlite3_prepare_v2(sql, szStmt, strlen(szStmt), &stmt, NULL)) { + MQTT_RTLM_LOG(sql); + return -1; + } + if ((ret = sqlite3_step(stmt)) == SQLITE_DONE) + ret = sqlite3_changes(sql); + else { + if (ret > SQLITE_OK && ret < SQLITE_ROW) + MQTT_RTLM_LOG(sql); + ret = 0; + } + sqlite3_finalize(stmt); + + return ret; +}