version 1.1.2.1, 2011/11/22 21:12:52
|
version 1.3.2.1, 2012/07/03 12:22:56
|
Line 1
|
Line 1
|
|
/************************************************************************* |
|
* (C) 2011 AITNET ltd - Sofia/Bulgaria - <misho@aitbg.com> |
|
* by Michael Pounov <misho@openbsd-bg.org> |
|
* |
|
* $Author$ |
|
* $Id$ |
|
* |
|
************************************************************************** |
|
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 <info@elwix.org> |
|
|
|
Copyright 2004, 2005, 2006, 2007, 2008, 2009, 2010, 2011, 2012 |
|
by Michael Pounov <misho@elwix.org>. 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 <misho@elwix.org> |
|
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 "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; |
|
} |