--- mqtt/src/pubmqtt.c	2011/11/23 08:50:03	1.1.2.2
+++ mqtt/src/pubmqtt.c	2011/11/24 15:34:18	1.1.2.6
@@ -17,14 +17,14 @@ mqtt_rtlm_log(const char *fmt, ...)
 	vsyslog(LOG_ERR, fmt, 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))))
 
 
 /*
  * mqtt_rtlm_open() Open database connection
  *
- * @csCfgName = config filename
+ * @cfg = config filename
  * return: NULL error or SQL handle
  */
 sqlite3 *
@@ -43,7 +43,7 @@ mqtt_rtlm_open(sl_config *cfg)
 	}
 
 	if (sqlite3_open_v2(str, &sql, SQLITE_OPEN_READWRITE, NULL)) {
-		MQTT_DB_LOG(sql);
+		MQTT_RTLM_LOG(sql);
 		sqlite3_close(sql);
 		return NULL;
 	}
@@ -61,4 +61,129 @@ void
 mqtt_rtlm_close(sqlite3 *sql)
 {
 	sqlite3_close(sql);
+}
+
+/*
+ * mqtt_rtlm_init_session() Create session
+ *
+ * @cfg = loaded config
+ * @sql = SQL handle
+ * @user = username
+ * @host = hostname
+ * @port = port
+ * return: -1 error, 0 session already appears or >0 row changed
+ */
+int
+mqtt_rtlm_init_session(sl_config *cfg, sqlite3 *sql, const char *user, const char *host, u_short port)
+{
+	int ret = 0;
+	char *str, szStmt[BUFSIZ] = { 0 };
+	sqlite3_stmt *stmt;
+
+	if (!cfg || !sql)
+		return -1;
+
+	str = (char*) cfg_GetAttribute(cfg, CFG("mqtt_pub"), CFG("tbl_online"));
+	if (!str) {
+		mqtt_rtlm_log("Error:: not found topics table name");
+		return -1;
+	}
+	snprintf(szStmt, sizeof szStmt, "INSERT INTO %s (Username, RemoteHost, RemotePort) "
+		       "VALUES ('%s', '%s', %d);", str, user, host, port);
+
+	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_fini_session() Delete session(s)
+ *
+ * @cfg = loaded config
+ * @sql = SQL handle
+ * @user = username
+ * @host = hostname
+ * return: -1 error, 0 session already appears or >0 row changed
+ */
+int
+mqtt_rtlm_fini_session(sl_config *cfg, sqlite3 *sql, const char *user, const char *host)
+{
+	int ret = 0;
+	char *str, szStmt[BUFSIZ] = { 0 };
+	sqlite3_stmt *stmt;
+
+	if (!cfg || !sql)
+		return -1;
+
+	str = (char*) cfg_GetAttribute(cfg, CFG("mqtt_pub"), CFG("tbl_online"));
+	if (!str) {
+		mqtt_rtlm_log("Error:: not found topics table name");
+		return -1;
+	}
+	snprintf(szStmt, sizeof szStmt, "DELETE FROM %s WHERE Username = '%s' AND RemoteHost LIKE '%s';", 
+			str, 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_chk_session() Check session(s)
+ *
+ * @cfg = loaded config
+ * @sql = SQL handle
+ * @user = username
+ * @host = hostname
+ * return: -1 error, 0 not logged or >0 logged found rows
+ */
+int
+mqtt_rtlm_chk_session(sl_config *cfg, sqlite3 *sql, const char *user, const char *host)
+{
+	int ret = 0;
+	char *str, szStmt[BUFSIZ] = { 0 };
+	sqlite3_stmt *stmt;
+
+	if (!cfg || !sql)
+		return -1;
+
+	str = (char*) cfg_GetAttribute(cfg, CFG("mqtt_pub"), CFG("tbl_online"));
+	if (!str) {
+		mqtt_rtlm_log("Error:: not found topics table name");
+		return -1;
+	}
+	snprintf(szStmt, sizeof szStmt, "SELECT RemoteHost, RemotePort FROM %s WHERE "
+			"Username = '%s' AND RemoteHost LIKE '%s';", str, user, host);
+
+	if (sqlite3_prepare_v2(sql, szStmt, strlen(szStmt), &stmt, NULL)) {
+		MQTT_RTLM_LOG(sql);
+		return -1;
+	}
+	while (sqlite3_step(stmt) == SQLITE_ROW);
+	ret = sqlite3_changes(sql);
+	sqlite3_finalize(stmt);
+
+	return ret;
 }