--- mqtt/src/accmqtt.c 2011/11/22 21:12:52 1.1.2.1 +++ mqtt/src/accmqtt.c 2012/07/03 12:46:00 1.4 @@ -1,4 +1,178 @@ +/************************************************************************* +* (C) 2011 AITNET ltd - Sofia/Bulgaria - +* by Michael Pounov +* +* $Author: misho $ +* $Id: accmqtt.c,v 1.4 2012/07/03 12:46:00 misho Exp $ +* +************************************************************************** +The ELWIX and AITNET software is distributed under the following +terms: + +All of the documentation and software included in the ELWIX and AITNET +Releases is copyrighted by ELWIX - Sofia/Bulgaria + +Copyright 2004, 2005, 2006, 2007, 2008, 2009, 2010, 2011, 2012 + by Michael Pounov . All rights reserved. + +Redistribution and use in source and binary forms, with or without +modification, are permitted provided that the following conditions +are met: +1. Redistributions of source code must retain the above copyright + notice, this list of conditions and the following disclaimer. +2. Redistributions in binary form must reproduce the above copyright + notice, this list of conditions and the following disclaimer in the + documentation and/or other materials provided with the distribution. +3. All advertising materials mentioning features or use of this software + must display the following acknowledgement: +This product includes software developed by Michael Pounov +ELWIX - Embedded LightWeight unIX and its contributors. +4. Neither the name of AITNET nor the names of its contributors + may be used to endorse or promote products derived from this software + without specific prior written permission. + +THIS SOFTWARE IS PROVIDED BY AITNET AND CONTRIBUTORS ``AS IS'' AND +ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE +IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE +ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE +FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL +DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS +OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) +HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT +LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY +OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF +SUCH DAMAGE. +*/ #include "global.h" -#include "dbmqtt.h" +extern const char sql_schema[]; + + +/* + * mqtt_rtlm_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:: %s(%d) SQL #%d - %s", \ + __func__, __LINE__, \ + sqlite3_errcode((_sql)), sqlite3_errmsg((_sql)))) + +/* library pre-loaded actions */ +void +_init() +{ + sqlite3_initialize(); +} + +void +_fini() +{ + sqlite3_shutdown(); +} + + +/* + * mqtt_rtlm_open() Open database connection + * + * @cfg = loaded config + * return: NULL error or SQL handle + */ +sqlite3 * +mqtt_rtlm_open(cfg_root_t *cfg) +{ + sqlite3 *sql = NULL; + const char *str = NULL; + + if (!cfg) + return NULL; + + str = (const char*) cfg_getAttribute(cfg, "mqtt_acc", "name"); + if (!str) { + mqtt_rtlm_log("Error:: Unknown database name ...\n"); + return NULL; + } + + if (sqlite3_open_v2(str, &sql, SQLITE_OPEN_READWRITE | SQLITE_OPEN_CREATE, NULL)) { + MQTT_RTLM_LOG(sql); + sqlite3_close(sql); + return NULL; + } + + if (sqlite3_exec(sql, sql_schema, NULL, NULL, 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_login() Verify login account + * + * @cfg = loaded config + * @sql = SQL handle + * @user = username + * @pass = password + * return: -1 error, 0 ALLOW and 1 REJECT + */ +int +mqtt_rtlm_login(cfg_root_t *cfg, sqlite3 *sql, const char *user, const char *pass) +{ + /* insert into Users values (NULL, "", "", 1, strftime('%s','now')); */ + int ret = 0; + sqlite3_stmt *stmt; + char *str, *psStmt; + + if (!sql) + return -1; + + str = (char*) cfg_getAttribute(cfg, "mqtt_acc", "tbl_users"); + if (!str) { + mqtt_rtlm_log("Error:: not found users table name"); + return -1; + } + psStmt = sqlite3_mprintf("SELECT DISTINCT Username, Password, Access FROM %s " + "WHERE Username = '%q' AND Password = '%q' AND Access > 0;", str, user, pass); + + if (sqlite3_prepare_v2(sql, psStmt, strlen(psStmt), &stmt, NULL)) { + MQTT_RTLM_LOG(sql); + sqlite3_free(psStmt); + return -1; + } else + sqlite3_free(psStmt); + while (sqlite3_step(stmt) == SQLITE_ROW) { + if (sqlite3_data_count(stmt) < 1) + ret = 0; + else + ret = 1; + break; + } + sqlite3_finalize(stmt); + + return ret; +}