From 55f465be474c54edbceb3c6013ae629e0e0323d8 Mon Sep 17 00:00:00 2001
From: guowenxue <guowenxue@gmail.com>
Date: Wed, 31 Jul 2019 16:26:46 +0800
Subject: [PATCH] Add tsl2561 code, and test force fill-in light ok

---
 mqttd/main.c |  390 +++++++++++++++++++++++++++++++++++++++++++++----------
 1 files changed, 318 insertions(+), 72 deletions(-)

diff --git a/mqttd/main.c b/mqttd/main.c
index 7ae6ba3..dd643df 100644
--- a/mqttd/main.c
+++ b/mqttd/main.c
@@ -13,77 +13,128 @@
 #include <stdio.h>
 #include <time.h>
 #include <unistd.h>
+#include <getopt.h>
+#include <libgen.h>
+#include <string.h>
 
 #include <mosquitto.h>
 
 #include "logger.h"
+#include "cJSON.h"
 #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 
-{
-    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> */
-
-    /* Publisher */
-    int           pubQos;
-
-    /* Subscriber */
-    int           subQos;
-} mqtt_ctx_t;  
-
+int g_relay_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)
+{
+    log_nrml("fill-in light timeout and turn light off now\n");
+    turn_relay(OFF);
+    g_relay_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;
-    ctx.pubQos = Qos0;
-    ctx.subQos = Qos2;
+    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}
+    };
 
-    //if( logger_init("mqttd.log", LOG_LEVEL_DEBUG, 1024) < 0 ) 
-    if( logger_init(DBG_LOG_FILE, LOG_LEVEL_DEBUG, 1024) < 0 ) 
+    progname = (char *)basename(argv[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;
@@ -104,10 +155,37 @@
     }
     log_nrml("Start MQTT publisher worker thread ok\n");
 
-
     while( ! g_signal.stop )
     {
         msleep(1000);
+
+        if( ctx.hwconf.infrared_pin )
+        {
+            //log_nrml("infrared dected human incoming\n");
+            if( !g_relay_flag && infrared_detect(1500) )
+            {
+                log_nrml("Someone incoming detected by infrared\n");
+                if( ctx.hwconf.inbreak_beep )
+                {
+                    turn_beep(ctx.hwconf.beep_times);
+                }
+
+                if( ctx.hwconf.fillin_light==FILLIN_LIGHT_ON && !g_relay_flag )
+                {
+                    log_nrml("Force fill-in light on and will be off in [%d] seconds\n", ctx.hwconf.light_intval);
+                    turn_relay(ON);
+                    g_relay_flag = 1;
+                    alarm(ctx.hwconf.light_intval);
+                }
+                else if( ctx.hwconf.fillin_light==FILLIN_LIGHT_AUTO && ctx.hwconf.lux )
+                {
+                    lux = tsl2561_get_lux(); 
+                    log_nrml("TSL2516 lux sensor detect illumination intensity %.3f lux\n", lux);
+                }
+
+            }
+        
+        }
         //log_nrml("Main control thread continue running\n");
     }
 
@@ -118,7 +196,6 @@
     
     return 0;
 } /* ----- End of main() ----- */
-
 
 int check_set_program_running(int daemon)
 {
@@ -148,15 +225,56 @@
     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 = 0.0;
+    int                     retain = 0;
+
+    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); 
+
+    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, 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);
+        }
+    }
+    
+    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 +283,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 +322,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, ctx->subTopic, ctx->subQos);
 }
 
 void sub_disconnect_callback(struct mosquitto *mosq, void *userdata, int result)
@@ -222,17 +332,149 @@
     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, mqtt_ctx_t *ctx)
+{
+    int                    i;
+    char                  *value;
+    cJSON                 *item; 
+    cJSON                 *array;
+    hwconf_t              *hwconf = &ctx->hwconf;
+
+    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, ctx);
+        }
+        else if( cJSON_Array == item->type )
+        {
+            /* RGB colors led control  */
+            if( hwconf->leds && !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);
+                        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);
+                    }
+                }
+            }
+        }
+        else
+        {
+            value = cJSON_Print(item);
+
+            /* light controled by relay */
+            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( hwconf->beep_pin && !strcasecmp(item->string, "buzzer") )
+            {
+                if( strcasestr(value, "on") )
+                {
+                    log_nrml("Turn buzzer on [%d] times\n", hwconf->beep_times);
+                }
+            }
+
+            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)
+    mqtt_ctx_t             *ctx = (mqtt_ctx_t *)userdata;
+
+    cJSON                  *root = NULL;
+    cJSON                  *item;
+    char                   *value;
+
+
+    if ( !message->payloadlen )
     {
-        log_nrml("Subscriber Recv: %s->%s\n", message->topic, message->payload);
+        log_err("%s (null)\n", message->topic);
+        return ;
     }
-    else 
+
+    log_dbg("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 ;
     }
+
+    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 ;
 }
 
 
@@ -249,14 +491,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