guowenxue
2019-06-26 57b8d8022cf8444cf59cd20a25cfee491ec88799
mqttd/main.c
@@ -13,6 +13,9 @@
#include <stdio.h>
#include <time.h>
#include <unistd.h>
#include <getopt.h>
#include <libgen.h>
#include <string.h>
#include <mosquitto.h>
@@ -23,43 +26,102 @@
#include "conf.h"
#define PROG_VERSION               "v1.0.0"
#define DAEMON_PIDFILE             "/tmp/.mqtt.pid"
int check_set_program_running(int daemon);
void *mqtt_sub_worker(void *args);
void *mqtt_pub_worker(void *args);
static void program_usage(char *progname)
{
    printf("Usage: %s [OPTION]...\n", progname);
    printf(" %s is LingYun studio MQTT daemon program running on RaspberryPi\n", progname);
    printf("\nMandatory arguments to long options are mandatory for short options too:\n");
    printf(" -d[debug   ]  Running in debug mode\n");
    printf(" -c[conf    ]  Specify configure file\n");
    printf(" -h[help    ]  Display this help information\n");
    printf(" -v[version ]  Display the program version\n");
    printf("\n%s version %s\n", progname, PROG_VERSION);
    return;
}
int main (int argc, char **argv)
{
    float              temp;
    float              rh;
    int                daemon = 0;
    int                daemon = 1;
    pthread_t          tid;
    mqtt_ctx_t         ctx;
    char               *conf="./etc/mqttd.conf";
    int                debug = 1;
    char               *conf_file=NULL;
    int                debug = 0;
    int                opt;
    char              *progname=NULL;
    if( mqttd_parser_conf(conf, &ctx, debug)<0 )
    struct option long_options[] = {
        {"conf", required_argument, NULL, 'c'},
        {"debug", no_argument, NULL, 'd'},
        {"version", no_argument, NULL, 'v'},
        {"help", no_argument, NULL, 'h'},
        {NULL, 0, NULL, 0}
    };
    progname = (char *)basename(argv[0]);
    /* Parser the command line parameters */
    while ((opt = getopt_long(argc, argv, "c:dvh", long_options, NULL)) != -1)
    {
        switch (opt)
        {
            case 'c': /* Set configure file */
                conf_file = optarg;
                break;
            case 'd': /* Set debug running */
                daemon = 0;
                debug = 1;
                break;
            case 'v':  /* Get software version */
                printf("%s version %s\n", progname, PROG_VERSION);
                return 0;
            case 'h':  /* Get help information */
                program_usage(progname);
                return 0;
            default:
                break;
        }
    }
    if( !conf_file )
        debug = 1;
    if( mqttd_parser_conf(conf_file, &ctx, debug)<0 )
    {
        fprintf(stderr, "Parser mqtted configure file failure\n");
        return -2;
    }
#if 0
    if( hal_init() < 0 )
    {
        log_err("Initialise hardware failure\n");
        return -1;
    }
    log_nrml("HAL initialise ok\n");
#endif
    
    install_proc_signal();
    if( check_set_program_running(daemon) < 0 ) 
        goto OUT;
    if( !debug )
    mosquitto_lib_init();
@@ -127,9 +189,7 @@
    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";
    float                   temp = 0.0;
    int                     retain = 0;
    if( result )
@@ -141,25 +201,15 @@
    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);
    if( ds18b20_get_temperature(&temp) < 0 )
        snprintf(msg, sizeof(msg), "{ \"id\":%s, \"temp\":\"error\" }", ctx->id);
    else
        snprintf(msg, sizeof(msg), "{ \"id\":%s, \"temp\":\"%.2f\" }", ctx->id, temp);
    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 
    {
@@ -236,6 +286,13 @@
    log_warn("Subscriber disconnect to broker server[%s:%d], reason=%d\n", ctx->host, ctx->port, result); 
}
static inline void mqtt_turn_led(int which, char *cmd)
{
    if( strcasestr(cmd, "on") )
        turn_led(which, ON);
    else if( strcasestr(cmd, "off") )
        turn_led(which, OFF);
}
void proc_json_items(cJSON *root)
{
@@ -274,16 +331,19 @@
                    if( NULL != (led_item=cJSON_GetObjectItem(array , "red")) )
                    {
                        log_nrml("turn red led '%s'\n", led_item->valuestring);
                        mqtt_turn_led(LED_R, led_item->valuestring);
                    }
                    if( NULL != (led_item=cJSON_GetObjectItem(array , "green")) )
                    {
                        log_nrml("turn green led '%s'\n", led_item->valuestring);
                        mqtt_turn_led(LED_G, led_item->valuestring);
                    }
                    if( NULL != (led_item=cJSON_GetObjectItem(array , "blue")) )
                    {
                        log_nrml("turn blue led '%s'\n", led_item->valuestring);
                        mqtt_turn_led(LED_B, led_item->valuestring);
                    }
                }
            }
@@ -298,10 +358,12 @@
                if( strcasestr(value, "on") )
                {
                    log_nrml("Turn light on\n");
                    turn_relay(ON);
                }
                else if( strcasestr(value, "off") )
                {
                    log_nrml("Turn light off\n");
                    turn_relay(OFF);
                }
            }
@@ -326,8 +388,11 @@
void sub_message_callback(struct mosquitto *mosq, void *userdata, const struct mosquitto_message *message)
{
    cJSON                 *root = NULL;
    char                  *out;
    mqtt_ctx_t             *ctx = (mqtt_ctx_t *)userdata;
    cJSON                  *root = NULL;
    cJSON                  *item;
    char                   *value;
    if ( !message->payloadlen )
@@ -336,7 +401,7 @@
        return ;
    }
    log_nrml("Subscriber receive message: '%s'\n", message->payload);
    log_dbg("Subscriber receive message: '%s'\n", message->payload);
    root = cJSON_Parse(message->payload);
    if( !root )
@@ -345,8 +410,26 @@
        return ;
    }
    item = cJSON_GetObjectItem(root, "id");
    if( !item )
    {
        log_err("cJSON_Parse get ID failure: %s\n", cJSON_GetErrorPtr());
        goto OUT;
    }
    value = cJSON_PrintUnformatted(item);
    if( strcasecmp(value, ctx->id) )
    {
        free(value);
        goto OUT;
    }
    free(value);
    log_nrml("Subscriber receive message: '%s'\n", message->payload);
    proc_json_items(root); 
OUT:
    cJSON_Delete(root); /* must delete it, or it will result memory leak */
    return ;
}