Diff for /mqtt/src/pubmqtt.c between versions 1.1.2.17 and 1.2.2.2

version 1.1.2.17, 2011/12/16 02:09:20 version 1.2.2.2, 2012/02/01 15:44:40
Line 160  mqtt_rtlm_fini_session(sl_config *cfg, sqlite3 *sql, c Line 160  mqtt_rtlm_fini_session(sl_config *cfg, sqlite3 *sql, c
                 return -1;                  return -1;
         }          }
         snprintf(szStmt, sizeof szStmt, "DELETE FROM %s WHERE ConnID = '%s' AND Username = '%s' "          snprintf(szStmt, sizeof szStmt, "DELETE FROM %s WHERE ConnID = '%s' AND Username = '%s' "
                        "AND RemoteHost LIKE '%s';", str, user, connid, host);                        "AND RemoteHost LIKE '%s';", str, connid, user, host);
   
         if (sqlite3_prepare_v2(sql, szStmt, strlen(szStmt), &stmt, NULL)) {          if (sqlite3_prepare_v2(sql, szStmt, strlen(szStmt), &stmt, NULL)) {
                 MQTT_RTLM_LOG(sql);                  MQTT_RTLM_LOG(sql);
Line 390  mqtt_rtlm_read_topic(sl_config *cfg, sqlite3 *sql, u_s Line 390  mqtt_rtlm_read_topic(sl_config *cfg, sqlite3 *sql, u_s
         /* fill with data */          /* fill with data */
         for (j = 0; j < rowz && sqlite3_step(stmt) == SQLITE_ROW; j++) {          for (j = 0; j < rowz && sqlite3_step(stmt) == SQLITE_ROW; j++) {
                 s[j].sub_ret = (char) sqlite3_column_int(stmt, 0);                  s[j].sub_ret = (char) sqlite3_column_int(stmt, 0);
                s[j].sub_topic._base = (u_char*) strdup((char*) sqlite3_column_text(stmt, 1));                s[j].sub_topic.msg_base = (u_char*) strdup((char*) sqlite3_column_text(stmt, 1));
                s[j].sub_topic._size = strlen((char*) s[j].sub_topic._base);                s[j].sub_topic.msg_len = strlen((char*) s[j].sub_topic.msg_base);
                s[j].sub_value._base = (u_char*) strdup((char*) sqlite3_column_text(stmt, 2));                s[j].sub_value.msg_base = (u_char*) strdup((char*) sqlite3_column_text(stmt, 2));
                s[j].sub_value._size = strlen((char*) s[j].sub_value._base);                s[j].sub_value.msg_len = strlen((char*) s[j].sub_value.msg_base);
         }
 end:
         sqlite3_finalize(stmt);
 
         return s;
 }
 
 /*
  * mqtt_rtlm_write_subscribe() Subscribe topic
  *
  * @cfg = loaded config
  * @sql = SQL handle
  * @msgid = MessageID
  * @topic = topic
  * @user = username
  * @host = hostname
  * @qos = Subscribe QoS
  * return: -1 error, 0 no publish or >0 published ok
  */
 int
 mqtt_rtlm_write_subscribe(sl_config *cfg, sqlite3 *sql, u_short msgid, const char *topic, 
                 const char *user, const char *host, char qos)
 {
         int ret = 0;
         char *str, szStmt[BUFSIZ] = { 0 };
         sqlite3_stmt *stmt;
 
         if (!cfg || !sql || !topic)
                 return -1;
 
         str = (char*) cfg_GetAttribute(cfg, CFG("mqtt_pub"), CFG("tbl_subscribes"));
         if (!str) {
                 mqtt_rtlm_log("Error:: not found subscribes table name");
                 return -1;
         }
         snprintf(szStmt, sizeof szStmt, "INSERT INTO %s (MsgID, QoS, Topic, PubUser, "
                         "PubDate, PubHost) VALUES (%d, %d, '%s', '%s', "
                         "datetime('now', 'localtime'), '%s');", str, 
                         msgid, qos, topic, user, host);
 
         if (sqlite3_prepare_v2(sql, szStmt, strlen(szStmt), &stmt, NULL)) {
                 MQTT_RTLM_LOG(sql);
                 return -1;
         }
         if ((ret = sqlite3_step(stmt)) == SQLITE_DONE)
                 ret = sqlite3_changes(sql);
         else {
                 if (ret > SQLITE_OK && ret < SQLITE_ROW)
                         MQTT_RTLM_LOG(sql);
                 ret = 0;
         }
         sqlite3_finalize(stmt);
 
         return ret;
 }
 
 /*
  * mqtt_rtlm_delete_subscribe() Delete subscribe
  *
  * @cfg = loaded config
  * @sql = SQL handle
  * @topic = topic
  * @user = username
  * @host = hostname
  * @qos = Subscribe QoS if -1 no matter
  * return: -1 error, 0 no changes or >0 deleted rows
  */
 int
 mqtt_rtlm_delete_subscribe(sl_config *cfg, sqlite3 *sql, const char *topic, 
                 const char *user, const char *host, char qos)
 {
         int ret = 0;
         char *str, szStr[STRSIZ] = { 0 }, szStmt[BUFSIZ] = { 0 };
         sqlite3_stmt *stmt;
 
         if (!cfg || !sql || !topic)
                 return -1;
 
         str = (char*) cfg_GetAttribute(cfg, CFG("mqtt_pub"), CFG("tbl_subscribes"));
         if (!str) {
                 mqtt_rtlm_log("Error:: not found subscribes table name");
                 return -1;
         }
         if (qos > -1 && qos < 3)
                 snprintf(szStr, sizeof szStr, "AND QoS = %d", qos);
         snprintf(szStmt, sizeof szStmt, "DELETE FROM %s WHERE Topic LIKE '%s' AND "
                         "PubUser LIKE '%s' AND PubHost LIKE '%s' %s;", str, 
                         topic, user, host, szStr);
 
         if (sqlite3_prepare_v2(sql, szStmt, strlen(szStmt), &stmt, NULL)) {
                 MQTT_RTLM_LOG(sql);
                 return -1;
         }
         if ((ret = sqlite3_step(stmt)) == SQLITE_DONE)
                 ret = sqlite3_changes(sql);
         else {
                 if (ret > SQLITE_OK && ret < SQLITE_ROW)
                         MQTT_RTLM_LOG(sql);
                 ret = 0;
         }
         sqlite3_finalize(stmt);
 
         return ret;
 }
 
 /*
  * mqtt_rtlm_read_subscribe() Get subscribe topic
  *
  * @cfg = loaded config
  * @sql = SQL handle
  * @topic = topic
  * return: NULL error or not found and !=NULL allocated subscribe topics
  */
 mqtt_subscr_t *
 mqtt_rtlm_read_subscribe(sl_config *cfg, sqlite3 *sql, const char *topic)
 {
         int rowz = 0;
         char *str, szStmt[BUFSIZ] = { 0 };
         sqlite3_stmt *stmt;
         register int j;
         mqtt_subscr_t *s = NULL;
 
         if (!cfg || !sql || !topic)
                 return NULL;
 
         str = (char*) cfg_GetAttribute(cfg, CFG("mqtt_pub"), CFG("tbl_subscribes"));
         if (!str) {
                 mqtt_rtlm_log("Error:: not found subscribes table name");
                 return NULL;
         }
         snprintf(szStmt, sizeof szStmt, "SELECT QoS, Topic FROM %s WHERE Topic LIKE '%s';", str, topic);
 
         if (sqlite3_prepare_v2(sql, szStmt, strlen(szStmt), &stmt, NULL)) {
                 MQTT_RTLM_LOG(sql);
                 return NULL;
         }
 
         /* calculate count of rows and allocate subscribe items */
         while (sqlite3_step(stmt) == SQLITE_ROW)
                 rowz++;
         if (!(s = malloc((rowz + 1) * sizeof(mqtt_subscr_t)))) {
                 mqtt_rtlm_log("Error:: System #%d - %s", errno, strerror(errno));
                 goto end;
         } else
                 memset(s, 0, (rowz + 1) * sizeof(mqtt_subscr_t));
         sqlite3_reset(stmt);
 
         /* fill with data */
         for (j = 0; j < rowz && sqlite3_step(stmt) == SQLITE_ROW; j++) {
                 s[j].sub_ret = (char) sqlite3_column_int(stmt, 0);
                 s[j].sub_topic.msg_base = (u_char*) strdup((char*) sqlite3_column_text(stmt, 1));
                 s[j].sub_topic.msg_len = strlen((char*) s[j].sub_topic.msg_base);
                 s[j].sub_value.msg_base = NULL;
                 s[j].sub_value.msg_len = 0;
         }          }
 end:  end:
         sqlite3_finalize(stmt);          sqlite3_finalize(stmt);

Removed from v.1.1.2.17  
changed lines
  Added in v.1.2.2.2


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