Update mqttd program, add cJSON parser and test ok with memory leak

guowenxue

2019-06-25 985548471892c5f2a7430c5f7b4c03fe39f0134f

Update mqttd program, add cJSON parser and test ok with memory leak

 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 sub_message_callback(struct mosquitto *mosq, void *userdata, const struct mosquitto_message *message)
void proc_json_items(cJSON *root)
{
    if(message->payloadlen)
    int                    i;
    char                  *value;
    cJSON                 *item; 
    cJSON                 *array;

    if( !root )
    {
        log_nrml("Subscriber Recv: %s->%s\n", message->topic, message->payload);
        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 
    {
        log_nrml("%s (null)\n", message->topic);
            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)
{
    cJSON                 *root = NULL;
    char                  *out;


    if ( !message->payloadlen )
    {
        log_err("%s (null)\n", message->topic);
        return ;
    }

    log_nrml("Subscriber receive message: '%s'\n", message->payload);

    root = cJSON_Parse(message->payload);
    if( !root )
    {
        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)); 

 mqttd/makefile

@@ -37,10 +37,11 @@
export CFLAGS
export LDFLAGS


CFLAGS+=-Ihal -Ilylib
LDFLAGS+=-L hal -lhal -Llylib -llylib

LDFLAGS+=-lmosquitto
LDFLAGS+=-lmosquitto -lpthread -lm

SRCFILES = $(wildcard *.c)
IMAGE_NAME=$(shell basename ${PWD})