From 0aa3153f05fe8cd51d76fd5dee9011a7d2296b7d Mon Sep 17 00:00:00 2001
From: guowenxue <guowenxue@gmail.com>
Date: Wed, 26 Jun 2019 00:37:25 +0800
Subject: [PATCH] update mqttd, add command line options parser

---
 mqttd/etc/conf.c     |   28 ++++++++++----
 mqttd/makefile       |    1 
 mqttd/etc/mqttd.conf |    2 
 mqttd/main.c         |   73 ++++++++++++++++++++++++++++++++++--
 4 files changed, 90 insertions(+), 14 deletions(-)

diff --git a/mqttd/etc/conf.c b/mqttd/etc/conf.c
index b66d22a..c78e24d 100644
--- a/mqttd/etc/conf.c
+++ b/mqttd/etc/conf.c
@@ -25,28 +25,42 @@
 
     if( !conf_file )
     { 
-        fprintf(stderr, "\nWARNNING: Use default MQTT configure\n");
 
         /* logger settings */
         strncpy(ctx->logfile, DBG_LOG_FILE, sizeof(ctx->logfile));
         ctx->loglevel = LOG_LEVEL_DEBUG;
         ctx->logsize = 1024;
 
+        if( logger_init(ctx->logfile, ctx->loglevel, ctx->logsize) < 0 ) 
+        {
+            fprintf(stderr, "Logger system initialise failure\n");
+            return -2;
+        } 
+        log_nrml("Logger system initialise ok\n");
+        log_warn("WARNNING: Use default MQTT configure\n");
+
         /* Broker settings */
         strncpy(ctx->host, DEF_BORKER_HOSTNAME, sizeof(ctx->host));
         ctx->port = DEF_BROKER_PORT;
+        log_nrml("Use default broker server [%s:%d]\n", ctx->host, ctx->port);
+
         strncpy(ctx->uid, DEF_BROKER_USERNAME, sizeof(ctx->uid));
         strncpy(ctx->pwd, DEF_BROKER_PASSWD, sizeof(ctx->pwd));
+        log_nrml("Use default broker author by [%s:%s]\n", ctx->uid, ctx->pwd);
+
         ctx->keepalive = DEF_BROKER_KEEPALIVE;
+        log_nrml("Use default broker keepalive timeout [%d] seconds\n", ctx->keepalive);
         
         /* Publisher settings */
         strncpy(ctx->pubTopic, DEF_PUBTOPIC, sizeof(ctx->pubTopic));
         ctx->pubQos = DEF_PUBQOS;
         ctx->interval = DEF_PUBINTERVAL;
+        log_nrml("Use default publisher topic \"%s\" with Qos[%d] interval[%d]\n", ctx->pubTopic, ctx->pubQos, ctx->interval);
 
         /* Subscriber settings */
         strncpy(ctx->subTopic, DEF_SUBTOPIC, sizeof(ctx->subTopic));
         ctx->subQos = DEF_SUBQOS;
+        log_nrml("Use default subscriber topic \"%s\" with Qos[%d]\n", ctx->subTopic, ctx->subQos);
 
         return 0;
     }
@@ -103,7 +117,7 @@
         return -2;
     }
     ctx->port = val;
-    log_nrml("Parser and get broker server [%s:%d]\n", ctx->host, ctx->port);
+    log_nrml("Parser broker server [%s:%d]\n", ctx->host, ctx->port);
 
     str=iniparser_getstring(ini, "broker:username", DEF_BROKER_USERNAME);
     strncpy(ctx->uid, str, sizeof(ctx->uid) );
@@ -111,10 +125,10 @@
     str=iniparser_getstring(ini, "broker:password", DEF_BROKER_PASSWD);
     strncpy(ctx->pwd, str, sizeof(ctx->pwd) );
 
-    log_nrml("Parser and get broker author by [%s:%s]\n", ctx->uid, ctx->pwd);
+    log_nrml("Parser broker author by [%s:%s]\n", ctx->uid, ctx->pwd);
 
     ctx->keepalive = iniparser_getint(ini, "broker:keepalive", DEF_BROKER_KEEPALIVE);
-    log_nrml("Parser and get broker keepalive timeout [%d] seconds\n", ctx->keepalive);
+    log_nrml("Parser broker keepalive timeout [%d] seconds\n", ctx->keepalive);
 
     /*+------------------------------------------------------+
      *|             parser publisher settings                |
@@ -122,11 +136,10 @@
 
     str=iniparser_getstring(ini, "publisher:pubTopic", DEF_PUBTOPIC);
     strncpy(ctx->pubTopic, str, sizeof(ctx->pubTopic) );
-    log_nrml("Parser and get publisher topic \"%s\"\n", ctx->pubTopic);
 
     ctx->pubQos = iniparser_getint(ini, "publisher:pubQos", DEF_PUBQOS);
     ctx->interval = iniparser_getint(ini, "publisher:interval", DEF_PUBINTERVAL);
-    log_nrml("Parser and get publisher Qos[%d] and interval [%d] seconds\n", ctx->pubQos, ctx->interval);
+    log_nrml("Parser publisher topic \"%s\" with Qos[%d] interval[%d]\n", ctx->pubTopic, ctx->pubQos, ctx->interval);
 
     /*+------------------------------------------------------+
      *|             parser subscriber settings               |
@@ -134,10 +147,9 @@
 
     str=iniparser_getstring(ini, "subsciber:subTopic", DEF_SUBTOPIC);
     strncpy(ctx->subTopic, str, sizeof(ctx->subTopic) );
-    log_nrml("Parser and get subscriber topic \"%s\"\n", ctx->subTopic);
 
     ctx->subQos = iniparser_getint(ini, "subsciber:subQos", DEF_SUBQOS);
-    log_nrml("Parser and get subscriber Qos[%d]\n", ctx->subQos);
+    log_nrml("Parser subscriber topic \"%s\" with Qos[%d]\n", ctx->subTopic, ctx->subQos);
 
     return 0;
 }
diff --git a/mqttd/etc/mqttd.conf b/mqttd/etc/mqttd.conf
index 4365594..f531c61 100644
--- a/mqttd/etc/mqttd.conf
+++ b/mqttd/etc/mqttd.conf
@@ -36,7 +36,7 @@
 pubQos=2
 
 # Publisher上报传感器数据的周期
-interval=60
+interval=5
 
 
 [subsciber]
diff --git a/mqttd/main.c b/mqttd/main.c
index 982b34a..df3f10f 100644
--- a/mqttd/main.c
+++ b/mqttd/main.c
@@ -13,6 +13,8 @@
 #include <stdio.h>
 #include <time.h>
 #include <unistd.h>
+#include <getopt.h>
+#include <libgen.h>
 
 #include <mosquitto.h>
 
@@ -23,25 +25,84 @@
 
 #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;
@@ -61,6 +122,8 @@
     if( check_set_program_running(daemon) < 0 ) 
         goto OUT;
 
+    if( !debug )
+
     mosquitto_lib_init();
 
     if( thread_start(&tid, mqtt_sub_worker, &ctx ) < 0 )
diff --git a/mqttd/makefile b/mqttd/makefile
index 932c4c9..0d1d30a 100644
--- a/mqttd/makefile
+++ b/mqttd/makefile
@@ -72,6 +72,7 @@
 
 clean: 
 	@make clean -C hal
+	@make clean -C etc
 	@make clean -C lylib
 	@rm -f version.h 
 	@rm -f *.o $(IMAGE_NAME) 

--
Gitblit v1.9.1