From 985548471892c5f2a7430c5f7b4c03fe39f0134f Mon Sep 17 00:00:00 2001
From: guowenxue <guowenxue@gmail.com>
Date: Tue, 25 Jun 2019 21:55:18 +0800
Subject: [PATCH] Update mqttd program, add cJSON parser and test ok with memory leak

---
 mqttd/main.c |  224 +++++++++++++++++++++++++++++++++++++++++++++++--------
 1 files changed, 190 insertions(+), 34 deletions(-)

diff --git a/mqttd/main.c b/mqttd/main.c
index 7ae6ba3..f8868d2 100644
--- a/mqttd/main.c
+++ b/mqttd/main.c
@@ -17,6 +17,7 @@
 #include <mosquitto.h>
 
 #include "logger.h"
+#include "cJSON.h"
 #include "proc.h"
 #include "hal.h"
 
@@ -35,15 +36,20 @@
 
 typedef struct mqtt_ctx_s 
 {
+    /* Broker settings  */
     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> */
+    char          uid[64];    /* username */
+    char          pwd[64];    /* password */
 
-    /* Publisher */
-    int           pubQos;
+    /* Publisher settings */
+    int           pubQos;     /* Publisher Qos */
+    int           interval;   /* Publish sensor data interval time, unit seconds */
 
-    /* Subscriber */
-    int           subQos;
+    /* Subscriber settings */
+    int           subQos;     /* Subscriber Qos  */
+
 } mqtt_ctx_t;  
 
 
@@ -63,7 +69,11 @@
     strcpy(ctx.host, "127.0.0.1");
     ctx.port = 10883;
     ctx.keepalive = 60;
+    strcpy(ctx.uid, "lingyun");
+    strcpy(ctx.pwd, "lingyun-emb");
+
     ctx.pubQos = Qos0;
+    ctx.interval = 300; 
     ctx.subQos = Qos2;
 
     //if( logger_init("mqttd.log", LOG_LEVEL_DEBUG, 1024) < 0 ) 
@@ -148,15 +158,63 @@
     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";
+
+    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, MQTT_SYS_ULINK_TOPIC, strlen(msg)+1, msg, ctx->pubQos, 0);
+    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, MQTT_SYS_ULINK_TOPIC, strlen(msg)+1, msg, ctx->pubQos, 0);
+    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 +223,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 +262,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, MQTT_SYS_DLINK_TOPIC, 2);
 }
 
 void sub_disconnect_callback(struct mosquitto *mosq, void *userdata, int result)
@@ -223,16 +273,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 +401,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