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

version 1.1.2.3, 2011/11/23 14:51:14 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 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_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 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.3  
changed lines
  Added in v.1.1.2.5


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