version 1.1.2.4, 2011/11/24 01:13:27
|
version 1.2.2.6, 2012/06/26 08:05:58
|
Line 1
|
Line 1
|
#include "global.h" |
#include "global.h" |
|
|
|
|
|
extern const char sql_schema[]; |
|
|
|
|
/* |
/* |
* mqtt_rtlm_log() Log database connection message |
* mqtt_rtlm_log() Log database connection message |
* |
* |
Line 17 mqtt_rtlm_log(const char *fmt, ...)
|
Line 20 mqtt_rtlm_log(const char *fmt, ...)
|
vsyslog(LOG_ERR, fmt, lst); |
vsyslog(LOG_ERR, fmt, lst); |
va_end(lst); |
va_end(lst); |
} |
} |
#define MQTT_RTLM_LOG(_sql) (assert((_sql)), mqtt_rtlm_log("Error:: SQL #%d - %s", \ | #define MQTT_RTLM_LOG(_sql) (assert((_sql)), mqtt_rtlm_log("Error:: %s(%d) SQL #%d - %s", \ |
| __func__, __LINE__, \ |
sqlite3_errcode((_sql)), sqlite3_errmsg((_sql)))) |
sqlite3_errcode((_sql)), sqlite3_errmsg((_sql)))) |
|
|
|
/* library pre-loaded actions */ |
|
void |
|
_init() |
|
{ |
|
sqlite3_initialize(); |
|
} |
|
|
|
void |
|
_fini() |
|
{ |
|
sqlite3_shutdown(); |
|
} |
|
|
|
|
/* |
/* |
* mqtt_rtlm_open() Open database connection |
* mqtt_rtlm_open() Open database connection |
* |
* |
Line 28 mqtt_rtlm_log(const char *fmt, ...)
|
Line 45 mqtt_rtlm_log(const char *fmt, ...)
|
* return: NULL error or SQL handle |
* return: NULL error or SQL handle |
*/ |
*/ |
sqlite3 * |
sqlite3 * |
mqtt_rtlm_open(sl_config *cfg) | mqtt_rtlm_open(cfg_root_t *cfg) |
{ |
{ |
sqlite3 *sql = NULL; |
sqlite3 *sql = NULL; |
const char *str = NULL; |
const char *str = NULL; |
Line 36 mqtt_rtlm_open(sl_config *cfg)
|
Line 53 mqtt_rtlm_open(sl_config *cfg)
|
if (!cfg) |
if (!cfg) |
return NULL; |
return NULL; |
|
|
str = (const char*) cfg_GetAttribute(cfg, CFG("mqtt_acc"), CFG("name")); | str = (const char*) cfg_getAttribute(cfg, "mqtt_acc", "name"); |
if (!str) { |
if (!str) { |
mqtt_rtlm_log("Error:: Unknown database name ...\n"); |
mqtt_rtlm_log("Error:: Unknown database name ...\n"); |
return NULL; |
return NULL; |
} |
} |
|
|
if (sqlite3_open_v2(str, &sql, SQLITE_OPEN_READONLY, NULL)) { | if (sqlite3_open_v2(str, &sql, SQLITE_OPEN_READWRITE | SQLITE_OPEN_CREATE, NULL)) { |
MQTT_RTLM_LOG(sql); |
MQTT_RTLM_LOG(sql); |
sqlite3_close(sql); |
sqlite3_close(sql); |
return NULL; |
return NULL; |
} |
} |
|
|
|
if (sqlite3_exec(sql, sql_schema, NULL, NULL, NULL)) { |
|
MQTT_RTLM_LOG(sql); |
|
sqlite3_close(sql); |
|
return NULL; |
|
} |
|
|
return sql; |
return sql; |
} |
} |
|
|
Line 70 mqtt_rtlm_close(sqlite3 *sql)
|
Line 93 mqtt_rtlm_close(sqlite3 *sql)
|
* @sql = SQL handle |
* @sql = SQL handle |
* @user = username |
* @user = username |
* @pass = password |
* @pass = password |
* @... = arguments list |
|
* return: -1 error, 0 ALLOW and 1 REJECT |
* return: -1 error, 0 ALLOW and 1 REJECT |
*/ |
*/ |
int |
int |
mqtt_rtlm_login(sl_config *cfg, sqlite3 *sql, const char *user, const char *pass, ...) | mqtt_rtlm_login(cfg_root_t *cfg, sqlite3 *sql, const char *user, const char *pass) |
{ |
{ |
/* insert into Users values (NULL, "", "", 1, strftime('%s','now')); */ |
/* insert into Users values (NULL, "", "", 1, strftime('%s','now')); */ |
int ret = 0; |
int ret = 0; |
sqlite3_stmt *stmt; |
sqlite3_stmt *stmt; |
char *str, szStmt[BUFSIZ] = { 0 }; | char *str, *psStmt; |
|
|
if (!sql) |
if (!sql) |
return -1; |
return -1; |
|
|
str = (char*) cfg_GetAttribute(cfg, CFG("mqtt_acc"), CFG("tbl_users")); | str = (char*) cfg_getAttribute(cfg, "mqtt_acc", "tbl_users"); |
if (!str) { |
if (!str) { |
mqtt_rtlm_log("Error:: not found users table name"); |
mqtt_rtlm_log("Error:: not found users table name"); |
return -1; |
return -1; |
} |
} |
snprintf(szStmt, sizeof szStmt, "SELECT DISTINCT Username, Password, Access FROM %s " | psStmt = sqlite3_mprintf("SELECT DISTINCT Username, Password, Access FROM %s " |
"WHERE Username = '%s' AND Password = '%s' AND Access > 0;", str, user, pass); | "WHERE Username = '%q' AND Password = '%q' AND Access > 0;", str, user, pass); |
|
|
if (sqlite3_prepare_v2(sql, szStmt, strlen(szStmt), &stmt, NULL)) { | if (sqlite3_prepare_v2(sql, psStmt, strlen(psStmt), &stmt, NULL)) { |
MQTT_RTLM_LOG(sql); |
MQTT_RTLM_LOG(sql); |
|
sqlite3_free(psStmt); |
return -1; |
return -1; |
} | } else |
| sqlite3_free(psStmt); |
while (sqlite3_step(stmt) == SQLITE_ROW) { |
while (sqlite3_step(stmt) == SQLITE_ROW) { |
if (sqlite3_data_count(stmt) < 1) |
if (sqlite3_data_count(stmt) < 1) |
ret = 0; |
ret = 0; |