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