Return to accmqtt.c CVS log

Up to [ELWIX - Embedded LightWeight unIX -] / mqtt / src

...

    1: #include "global.h"
    2: 
    3: 
    4: extern const char sql_schema[];
    5: 
    6: 
    7: /*
    8:  * mqtt_rtlm_log() Log database connection message
    9:  *
   10:  * @fmt = format string
   11:  * @... = argument list
   12:  * return: none
   13:  */
   14: static void
   15: mqtt_rtlm_log(const char *fmt, ...)
   16: {
   17: 	va_list lst;
   18: 
   19: 	va_start(lst, fmt);
   20: 	vsyslog(LOG_ERR, fmt, lst);
   21: 	va_end(lst);
   22: }
   23: #define MQTT_RTLM_LOG(_sql)	(assert((_sql)), mqtt_rtlm_log("Error:: %s(%d) SQL #%d - %s", \
   24: 					__func__, __LINE__, \
   25: 					sqlite3_errcode((_sql)), sqlite3_errmsg((_sql))))
   26: 
   27: 
   28: /*
   29:  * mqtt_rtlm_open() Open database connection
   30:  *
   31:  * @cfg = loaded config
   32:  * return: NULL error or SQL handle
   33:  */
   34: sqlite3 *
   35: mqtt_rtlm_open(cfg_root_t *cfg)
   36: {
   37: 	sqlite3 *sql = NULL;
   38: 	const char *str = NULL;
   39: 
   40: 	if (!cfg)
   41: 		return NULL;
   42: 
   43: 	sqlite3_config(SQLITE_CONFIG_SERIALIZED);
   44: 	if (!sqlite3_threadsafe())
   45: 		return NULL;
   46: 
   47: 	str = (const char*) cfg_getAttribute(cfg, "mqtt_acc", "name");
   48: 	if (!str) {
   49: 		mqtt_rtlm_log("Error:: Unknown database name ...\n");
   50: 		return NULL;
   51: 	}
   52: 
   53: 	if (sqlite3_open_v2(str, &sql, SQLITE_OPEN_READWRITE | SQLITE_OPEN_CREATE, NULL)) {
   54: 		MQTT_RTLM_LOG(sql);
   55: 		sqlite3_close(sql);
   56: 		return NULL;
   57: 	}
   58: 
   59: 	sqlite3_mutex_enter(sqlite3_db_mutex(sql));
   60: 	if (sqlite3_exec(sql, sql_schema, NULL, NULL, NULL)) {
   61: 		MQTT_RTLM_LOG(sql);
   62: 		sqlite3_mutex_leave(sqlite3_db_mutex(sql));
   63: 		sqlite3_close(sql);
   64: 		return NULL;
   65: 	}
   66: 	sqlite3_mutex_leave(sqlite3_db_mutex(sql));
   67: 	return sql;
   68: }
   69: 
   70: /*
   71:  * mqtt_rtlm_close() Close database connection
   72:  *
   73:  * @sql = SQL handle
   74:  * return: none
   75:  */
   76: void
   77: mqtt_rtlm_close(sqlite3 *sql)
   78: {
   79: 	sqlite3_close(sql);
   80: }
   81: 
   82: /*
   83:  * mqtt_rtlm_login() Verify login account
   84:  *
   85:  * @cfg = loaded config
   86:  * @sql = SQL handle
   87:  * @user = username
   88:  * @pass = password
   89:  * return: -1 error, 0 ALLOW and 1 REJECT
   90:  */
   91: int
   92: mqtt_rtlm_login(cfg_root_t *cfg, sqlite3 *sql, const char *user, const char *pass)
   93: {
   94: 	/* insert into Users values (NULL, "", "", 1, strftime('%s','now')); */
   95: 	int ret = 0;
   96: 	sqlite3_stmt *stmt;
   97: 	char *str, szStmt[BUFSIZ] = { 0 };
   98: 
   99: 	if (!sql)
  100: 		return -1;
  101: 
  102: 	str = (char*) cfg_getAttribute(cfg, "mqtt_acc", "tbl_users");
  103: 	if (!str) {
  104: 		mqtt_rtlm_log("Error:: not found users table name");
  105: 		return -1;
  106: 	}
  107: 	snprintf(szStmt, sizeof szStmt, "SELECT DISTINCT Username, Password, Access FROM %s "
  108: 			"WHERE Username = '%s' AND Password = '%s' AND Access > 0;", str, user, pass);
  109: 
  110: 	sqlite3_mutex_enter(sqlite3_db_mutex(sql));
  111: 	if (sqlite3_prepare_v2(sql, szStmt, strlen(szStmt), &stmt, NULL)) {
  112: 		MQTT_RTLM_LOG(sql);
  113: 		sqlite3_mutex_leave(sqlite3_db_mutex(sql));
  114: 		return -1;
  115: 	}
  116: 	while (sqlite3_step(stmt) == SQLITE_ROW) {
  117: 		if (sqlite3_data_count(stmt) < 1)
  118: 			ret = 0;
  119: 		else
  120: 			ret = 1;
  121: 		break;
  122: 	}
  123: 	sqlite3_finalize(stmt);
  124: 	sqlite3_mutex_leave(sqlite3_db_mutex(sql));
  125: 
  126: 	return ret;
  127: }