update client and socket.c, add DNS support

GuoWenxue

2022-04-19 ea435789dc2fab30ec16fa765a0b6577fe08166a

 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>

@@ -21,79 +24,123 @@
#include "proc.h"
#include "hal.h"

#include "conf.h"

#define PROG_VERSION               "v1.0.0"
#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 
{
    /* 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 settings */
    int           pubQos;     /* Publisher Qos */
    int           interval;   /* Publish sensor data interval time, unit seconds */

    /* Subscriber settings */
    int           subQos;     /* Subscriber Qos  */

} mqtt_ctx_t;  

#define FLAG_DETECTD        (1<<0)
#define FLAG_RELAYON        (1<<1)
int g_breakin_flag = 0;  /* relay already turn on and wait to off  */

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;
}

void turn_light_off(int signum)
{
    if( g_breakin_flag&FLAG_RELAYON ) 
    {
        log_nrml("fill-in light timeout and turn light off now\n");
        turn_relay(OFF);
    }

    g_breakin_flag = 0;
}

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_file=NULL;
    int                debug = 0;
    int                opt;
    char              *progname=NULL;
    double             lux = 0.0;

    strcpy(ctx.host, "127.0.0.1");
    ctx.port = 10883;
    ctx.keepalive = 60;
    strcpy(ctx.uid, "lingyun");
    strcpy(ctx.pwd, "lingyun-emb");
    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}
    };

    ctx.pubQos = Qos0;
    ctx.interval = 300; 
    ctx.subQos = Qos2;
    progname = (char *)basename(argv[0]);

    //if( logger_init("mqttd.log", LOG_LEVEL_DEBUG, 1024) < 0 ) 
    if( logger_init(DBG_LOG_FILE, LOG_LEVEL_DEBUG, 1024) < 0 ) 
    /* Parser the command line parameters */
    while ((opt = getopt_long(argc, argv, "c:dvh", long_options, NULL)) != -1)
    {
        fprintf(stderr, "Logger system initialise failure\n");
        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;
    }
    log_dbg("Logger system initialise ok\n");

#if 0
    if( hal_init() < 0 )
    if( hal_init(&ctx.hwconf) < 0 )
    {
        log_err("Initialise hardware failure\n");
        return -1;
        return -3;
    }
    log_nrml("HAL initialise ok\n");
#endif
    
    else 
    {
        log_nrml("HAL initialise ok\n");
    }

    install_proc_signal();

    if( ctx.hwconf.fillin_light )
    {
        log_nrml("Install inbreak detected fill-in light off signal\n");
        signal(SIGALRM, turn_light_off);
    }

    if( check_set_program_running(daemon) < 0 ) 
        goto OUT;
@@ -114,10 +161,54 @@
    }
    log_nrml("Start MQTT publisher worker thread ok\n");


    while( ! g_signal.stop )
    {
        msleep(1000);

        if( ctx.hwconf.infrared_pin && !g_breakin_flag )
        {
            //log_nrml("infrared dected human incoming\n");
            if( infrared_detect(1000) )
            {
                log_nrml("Someone incoming detected by infrared\n");
                g_breakin_flag |= FLAG_DETECTD; 
                
                /* setup alarm to turn relay off or clear the flag */
                alarm(ctx.hwconf.light_intval);

                if( ctx.hwconf.inbreak_beep )
                {
                    turn_beep(ctx.hwconf.beep_times);
                }

                if( ctx.hwconf.fillin_light==FILLIN_LIGHT_ON && !g_breakin_flag )
                {
                    log_nrml("Force fill-in light on and will be off in [%d] seconds\n", ctx.hwconf.light_intval);
                    turn_relay(ON);
                    g_breakin_flag |= FLAG_RELAYON;
                }
                else if( ctx.hwconf.fillin_light==FILLIN_LIGHT_AUTO && ctx.hwconf.lux )
                {
                    lux = tsl2561_get_lux(); 
                    log_nrml("TSL2516 detect illumination intensity %.3flux and threshold %.3flux\n", 
                            lux, ctx.hwconf.lux_threshold);

                    /* only lux<0.01 will turn on light */
                    if( lux-0.001 > ctx.hwconf.lux_threshold)
                    {
                        log_nrml("Illumination intensity enuf, no need fill-in light\n");
                    }
                    else
                    {
                        log_nrml("Auto fill-in light on and will be off in [%d] seconds\n", ctx.hwconf.light_intval);
                        turn_relay(ON);
                        g_breakin_flag |= FLAG_RELAYON;
                    }
                }

            }
        
        }
        //log_nrml("Main control thread continue running\n");
    }

@@ -128,7 +219,6 @@
    
    return 0;
} /* ----- End of main() ----- */


int check_set_program_running(int daemon)
{
@@ -164,9 +254,9 @@
    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; /* temperature */
    float                   rh = 0.0;   /* relative humidity */
    int                     retain = 0;

    if( result )
    {
@@ -176,32 +266,47 @@
    
    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( ctx->hwconf.ds18b20 )
    {
        memset(msg, 0, sizeof(msg));
        
        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, 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);
        rv = mosquitto_publish(mosq, NULL, ctx->pubTopic, strlen(msg), 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);
    if( ctx->hwconf.sht2x )
    {
        memset(msg, 0, sizeof(msg));
        
        if( sht2x_get_temp_humidity(&temp, &rh) < 0 )
            snprintf(msg, sizeof(msg), "{ \"id\":%s, \"temp\":\"error\", \"RH\":\"error\" }", ctx->id);
        else 
            snprintf(msg, sizeof(msg), "{ \"id\":%s, \"temp\":\"%.2f\", \"RH\":\"%.2f\" }", ctx->id, 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 failure: %d\n", rv);
    }
    else 
    {
        log_nrml("Publisher broadcast message '%s' ok\n", msg);
        rv = mosquitto_publish(mosq, NULL, ctx->pubTopic, strlen(msg), 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);
        }
    }

    
    log_nrml("Publisher broadcast over and disconnect broker now\n", ctx->host, ctx->port); 
    mosquitto_disconnect(mosq);

@@ -262,7 +367,7 @@
    }
    
    log_nrml("Subscriber connect to broker server[%s:%d] successfully\n", ctx->host, ctx->port); 
    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)
@@ -272,13 +377,21 @@
    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)
void proc_json_items(cJSON *root, mqtt_ctx_t *ctx)
{
    int                    i;
    char                  *value;
    cJSON                 *item; 
    cJSON                 *array;
    hwconf_t              *hwconf = &ctx->hwconf;

    if( !root )
    {
@@ -295,12 +408,12 @@
        /* if item is cJSON_Object, then recursive call proc_json */
        if( cJSON_Object == item->type )
        {
            proc_json_items(item);
            proc_json_items(item, ctx);
        }
        else if( cJSON_Array == item->type )
        {
            /* RGB colors led control  */
            if( !strcasecmp(item->string, "led") )
            if( hwconf->leds && !strcasecmp(item->string, "led") )
            {
                array = cJSON_GetArrayItem(item, 0);
                if( NULL != array )
@@ -310,16 +423,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);
                    }
                }
            }
@@ -329,28 +445,34 @@
            value = cJSON_Print(item);

            /* light controled by relay */
            if( !strcasecmp(item->string, "light") )
            if( hwconf->relay_pin && !strcasecmp(item->string, "light") )
            {
                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);
                }
            }

            /* buzzer */
            if( !strcasecmp(item->string, "buzzer") )
            if( hwconf->beep_pin && !strcasecmp(item->string, "buzzer") )
            {
                if( strcasestr(value, "on") )
                {
                    log_nrml("Turn buzzer on\n");
          turn_passive_beep(ON, BEEP_FREQ);
                }
                else if( strcasestr(value, "music") )
      else if( strcasestr(value, "off") )
                {
                    log_nrml("Turn buzzer play music\n");
          turn_passive_beep(OFF, BEEP_FREQ);
                }
      else if( strcasestr(value, "music") )
                { 
          play_little_star();
                }
            }

@@ -362,8 +484,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 )
@@ -372,7 +497,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 )
@@ -381,8 +506,26 @@
        return ;
    }

    proc_json_items(root); 
    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, ctx); 

OUT:
    cJSON_Delete(root); /* must delete it, or it will result memory leak */
    return ;
}