Diff for /mqtt/src/accmqtt.c between versions 1.1.2.2 and 1.1.2.5

version 1.1.2.2, 2011/11/23 08:50:03 version 1.1.2.5, 2011/11/28 22:28:08
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_DB_LOG(_sql)      (assert((_sql)), mqtt_rtlm_log("Error:: SQL #%d - %s", \#define MQTT_RTLM_LOG(_sql)      (assert((_sql)), mqtt_rtlm_log("Error:: SQL #%d - %s", \
                                         sqlite3_errcode((_sql)), sqlite3_errmsg((_sql))))                                          sqlite3_errcode((_sql)), sqlite3_errmsg((_sql))))
   
   
 /*  /*
  * mqtt_rtlm_open() Open database connection   * mqtt_rtlm_open() Open database connection
  *   *
 * @csCfgName = config filename * @cfg = loaded config
  * return: NULL error or SQL handle   * return: NULL error or SQL handle
  */   */
 sqlite3 *  sqlite3 *
Line 36  mqtt_rtlm_open(sl_config *cfg) Line 39  mqtt_rtlm_open(sl_config *cfg)
         if (!cfg)          if (!cfg)
                 return NULL;                  return NULL;
   
           sqlite3_config(SQLITE_CONFIG_SERIALIZED);
   
         str = (const char*) cfg_GetAttribute(cfg, CFG("mqtt_acc"), CFG("name"));          str = (const char*) cfg_GetAttribute(cfg, CFG("mqtt_acc"), CFG("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_DB_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 61  void Line 71  void
 mqtt_rtlm_close(sqlite3 *sql)  mqtt_rtlm_close(sqlite3 *sql)
 {  {
         sqlite3_close(sql);          sqlite3_close(sql);
   }
   
   /*
    * mqtt_rtlm_login() Verify login account
    *
    * @cfg = loaded config
    * @sql = SQL handle
    * @user = username
    * @pass = password
    * @... = arguments list
    * return: -1 error, 0 ALLOW and 1 REJECT
    */
   int
   mqtt_rtlm_login(sl_config *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, szStmt[BUFSIZ] = { 0 };
   
           if (!sql)
                   return -1;
   
           str = (char*) cfg_GetAttribute(cfg, CFG("mqtt_acc"), CFG("tbl_users"));
           if (!str) {
                   mqtt_rtlm_log("Error:: not found users table name");
                   return -1;
           }
           snprintf(szStmt, sizeof szStmt, "SELECT DISTINCT Username, Password, Access FROM %s "
                           "WHERE Username = '%s' AND Password = '%s' AND Access > 0;", str, user, pass);
   
           if (sqlite3_prepare_v2(sql, szStmt, strlen(szStmt), &stmt, NULL)) {
                   MQTT_RTLM_LOG(sql);
                   return -1;
           }
           while (sqlite3_step(stmt) == SQLITE_ROW) {
                   if (sqlite3_data_count(stmt) < 1)
                           ret = 0;
                   else
                           ret = 1;
                   break;
           }
           sqlite3_finalize(stmt);
   
           return ret;
 }  }

Removed from v.1.1.2.2  
changed lines
  Added in v.1.1.2.5


FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>