From 6a924a29cefbea27a101790343137db819b361be Mon Sep 17 00:00:00 2001
From: guowenxue <guowenxue@gmail.com>
Date: Wed, 26 Jun 2019 00:08:35 +0800
Subject: [PATCH] update mqttd to add configure file parser support
---
mqttd/main.c | 248 ++++++++++++++++++++++++++++++++++++------------
1 files changed, 184 insertions(+), 64 deletions(-)
diff --git a/mqttd/main.c b/mqttd/main.c
index 7ae6ba3..982b34a 100644
--- a/mqttd/main.c
+++ b/mqttd/main.c
@@ -17,34 +17,13 @@
#include <mosquitto.h>
#include "logger.h"
+#include "cJSON.h"
#include "proc.h"
#include "hal.h"
+#include "conf.h"
+
#define DAEMON_PIDFILE "/tmp/.mqtt.pid"
-
-#define MQTT_SYS_TOPIC "$Sys/Studio/"
-#define MQTT_SYS_DLINK_TOPIC MQTT_SYS_TOPIC"Downlink"
-#define MQTT_SYS_ULINK_TOPIC MQTT_SYS_TOPIC"Uplink"
-
-enum
-{
- Qos0, /* 发送者只发送一次消息，不进行重试，Broker不会返回确认消息。在Qos0情况下，Broker可能没有接受到消息 */
- Qos1, /* 发送者最少发送一次消息，确保消息到达Broker，Broker需要返回确认消息PUBACK。在Qos1情况下，Broker可能接受到重复消息 */
- Qos2, /* Qos2使用两阶段确认来保证消息的不丢失和不重复。在Qos2情况下，Broker肯定会收到消息，且只收到一次 */
-};
-
-typedef struct mqtt_ctx_s
-{
- char host[128]; /* MQTT broker server name */
- int port; /* MQTT broker listen port */
- int keepalive; /* MQTT broker send PING message to subsciber/publisher keepalive timeout<seconds> */
-
- /* Publisher */
- int pubQos;
-
- /* Subscriber */
- int subQos;
-} mqtt_ctx_t;
int check_set_program_running(int daemon);
@@ -59,20 +38,14 @@
int daemon = 0;
pthread_t tid;
mqtt_ctx_t ctx;
+ char *conf="./etc/mqttd.conf";
+ int debug = 1;
- strcpy(ctx.host, "127.0.0.1");
- ctx.port = 10883;
- ctx.keepalive = 60;
- ctx.pubQos = Qos0;
- ctx.subQos = Qos2;
-
- //if( logger_init("mqttd.log", LOG_LEVEL_DEBUG, 1024) < 0 )
- if( logger_init(DBG_LOG_FILE, LOG_LEVEL_DEBUG, 1024) < 0 )
+ if( mqttd_parser_conf(conf, &ctx, debug)<0 )
{
- fprintf(stderr, "Logger system initialise failure\n");
+ fprintf(stderr, "Parser mqtted configure file failure\n");
return -2;
}
- log_dbg("Logger system initialise ok\n");
#if 0
if( hal_init() < 0 )
@@ -148,15 +121,64 @@
return 0;
}
+
+void pub_connect_callback(struct mosquitto *mosq, void *userdata, int result)
+{
+ mqtt_ctx_t *ctx = (mqtt_ctx_t *)userdata;
+ int rv = 0;
+ char msg[128];
+ float temp = 25.64;
+ float rh = 65.35;
+ char light[] = "on";
+ int retain = 0;
+
+ if( result )
+ {
+ log_err("Publisher connect to broker server[%s:%d] failed, rv=%d\n", ctx->host, ctx->port, result);
+ return ;
+ }
+
+ log_nrml("Publisher connect to broker server[%s:%d] successfully\n", ctx->host, ctx->port);
+
+ memset(msg, 0, sizeof(msg));
+ snprintf(msg, sizeof(msg), "{ \"temp\":\"%.2f\", \"R&H\":\"%.2f%%\" }", temp, rh);
+
+ rv = mosquitto_publish(mosq, NULL, ctx->pubTopic, strlen(msg)+1, msg, ctx->pubQos, retain);
+ if( rv )
+ {
+ log_err("Publisher broadcast message '%s' failure: %d\n", msg, rv);
+ }
+ else
+ {
+ log_nrml("Publisher broadcast message '%s' ok\n", msg);
+ }
+
+ memset(msg, 0, sizeof(msg));
+ snprintf(msg, sizeof(msg), "{ \"light\":\"%s\" }", light);
+
+ rv = mosquitto_publish(mosq, NULL, ctx->pubTopic, strlen(msg)+1, msg, ctx->pubQos, retain);
+ if( rv )
+ {
+ log_err("Publisher broadcast message failure: %d\n", rv);
+ }
+ else
+ {
+ log_nrml("Publisher broadcast message '%s' ok\n", msg);
+ }
+
+ log_nrml("Publisher broadcast over and disconnect broker now\n", ctx->host, ctx->port);
+ mosquitto_disconnect(mosq);
+
+ return ;
+}
+
+
void *mqtt_pub_worker(void *args)
{
mqtt_ctx_t *ctx = (mqtt_ctx_t *)args;
struct mosquitto *mosq;
bool session = true;
- char msg[128];
- float temp = 25.64;
- float rh = 65.35;
- int rv = 0;
+
mosq = mosquitto_new(NULL, session, ctx);
if( !mosq )
@@ -165,40 +187,33 @@
return NULL;
}
+ /* set connnect to broker username and password */
+ if( strlen(ctx->uid)> 0 && strlen(ctx->pwd)> 0 )
+ mosquitto_username_pw_set(mosq, ctx->uid, ctx->pwd);
+
+ /* set callback functions */
+ mosquitto_connect_callback_set(mosq, pub_connect_callback);
+
while( !g_signal.stop )
{
+ /* connect to MQTT broker */
if( mosquitto_connect(mosq, ctx->host, ctx->port, ctx->keepalive) )
{
log_err("Publisher connect to broker[%s:%d] failure: %s\n", ctx->host, ctx->port, strerror(errno));
msleep(1000);
continue;
}
- log_nrml("Publisher connect to broker server[%s:%d] successfully\n", ctx->host, ctx->port);
+ /* -1: use default timeout 1000ms 1: unused */
+ mosquitto_loop_forever(mosq, -1, 1);
- memset(msg, 0, sizeof(msg));
- snprintf(msg, sizeof(msg), "{ \"temp\":\"%.2f\", \"R&H\":\"%.2f%%\" }", temp, rh);
-
-
- rv = mosquitto_publish(mosq, NULL, MQTT_SYS_ULINK_TOPIC, strlen(msg)+1, msg, ctx->pubQos, 0);
- if( rv )
- {
- log_err("Publisher publish message failure: %d\n", rv);
- }
- else
- {
- log_nrml("Publisher publish message '%s' ok\n", msg);
- }
-
-
- mosquitto_disconnect(mosq);
- msleep(3000);
+ /* Publisher broadcast sensors data message interval time, unit seconds */
+ sleep( ctx->interval );
}
mosquitto_destroy(mosq);
return NULL;
}
-
void sub_connect_callback(struct mosquitto *mosq, void *userdata, int result)
{
@@ -211,8 +226,7 @@
}
log_nrml("Subscriber connect to broker server[%s:%d] successfully\n", ctx->host, ctx->port);
- mosquitto_subscribe(mosq, NULL, MQTT_SYS_TOPIC"#", ctx->subQos);
- //mosquitto_subscribe(mosq, NULL, MQTT_SYS_DLINK_TOPIC, 2);
+ mosquitto_subscribe(mosq, NULL, ctx->subTopic, ctx->subQos);
}
void sub_disconnect_callback(struct mosquitto *mosq, void *userdata, int result)
@@ -223,16 +237,118 @@
}
+void proc_json_items(cJSON *root)
+{
+ int i;
+ char *value;
+ cJSON *item;
+ cJSON *array;
+
+ if( !root )
+ {
+ log_err("Invalid input arguments $root\n");
+ return ;
+ }
+
+ for( i=0; i<cJSON_GetArraySize(root); i++ )
+ {
+ item = cJSON_GetArrayItem(root, i);
+ if( !item )
+ break;
+
+ /* if item is cJSON_Object, then recursive call proc_json */
+ if( cJSON_Object == item->type )
+ {
+ proc_json_items(item);
+ }
+ else if( cJSON_Array == item->type )
+ {
+ /* RGB colors led control */
+ if( !strcasecmp(item->string, "led") )
+ {
+ array = cJSON_GetArrayItem(item, 0);
+ if( NULL != array )
+ {
+ cJSON *led_item;
+
+ if( NULL != (led_item=cJSON_GetObjectItem(array , "red")) )
+ {
+ log_nrml("turn red led '%s'\n", led_item->valuestring);
+ }
+
+ if( NULL != (led_item=cJSON_GetObjectItem(array , "green")) )
+ {
+ log_nrml("turn green led '%s'\n", led_item->valuestring);
+ }
+
+ if( NULL != (led_item=cJSON_GetObjectItem(array , "blue")) )
+ {
+ log_nrml("turn blue led '%s'\n", led_item->valuestring);
+ }
+ }
+ }
+ }
+ else
+ {
+ value = cJSON_Print(item);
+
+ /* light controled by relay */
+ if( !strcasecmp(item->string, "light") )
+ {
+ if( strcasestr(value, "on") )
+ {
+ log_nrml("Turn light on\n");
+ }
+ else if( strcasestr(value, "off") )
+ {
+ log_nrml("Turn light off\n");
+ }
+ }
+
+ /* buzzer */
+ if( !strcasecmp(item->string, "buzzer") )
+ {
+ if( strcasestr(value, "on") )
+ {
+ log_nrml("Turn buzzer on\n");
+ }
+ else if( strcasestr(value, "music") )
+ {
+ log_nrml("Turn buzzer play music\n");
+ }
+ }
+
+ free(value); /* must free it, or it will result memory leak */
+ }
+ }
+
+}
+
void sub_message_callback(struct mosquitto *mosq, void *userdata, const struct mosquitto_message *message)
{
- if(message->payloadlen)
+ cJSON *root = NULL;
+ char *out;
+
+
+ if ( !message->payloadlen )
{
- log_nrml("Subscriber Recv: %s->%s\n", message->topic, message->payload);
+ log_err("%s (null)\n", message->topic);
+ return ;
}
- else
+
+ log_nrml("Subscriber receive message: '%s'\n", message->payload);
+
+ root = cJSON_Parse(message->payload);
+ if( !root )
{
- log_nrml("%s (null)\n", message->topic);
+ log_err("cJSON_Parse parser failure: %s\n", cJSON_GetErrorPtr());
+ return ;
}
+
+ proc_json_items(root);
+
+ cJSON_Delete(root); /* must delete it, or it will result memory leak */
+ return ;
}
@@ -249,14 +365,18 @@
return NULL;
}
+ /* set connnect to broker username and password */
+ if( strlen(ctx->uid)> 0 && strlen(ctx->pwd)> 0 )
+ mosquitto_username_pw_set(mosq, ctx->uid, ctx->pwd);
+
/* set callback functions */
mosquitto_connect_callback_set(mosq, sub_connect_callback);
mosquitto_disconnect_callback_set(mosq, sub_disconnect_callback);
mosquitto_message_callback_set(mosq, sub_message_callback);
-
while( !g_signal.stop )
{
+ /* connect to MQTT broker */
if( mosquitto_connect(mosq, ctx->host, ctx->port, ctx->keepalive) )
{
log_err("Subscriber connect to broker[%s:%d] failure: %s\n", ctx->host, ctx->port, strerror(errno));
--
Gitblit v1.9.1