/*********************************************************************************
|
* Copyright: (C) 2019 LingYun IoT System Studio
|
* All rights reserved.
|
*
|
* Filename: main.c
|
* Description: This file
|
*
|
* Version: 1.0.0(29/01/19)
|
* Author: Guo Wenxue <guowenxue@gmail.com>
|
* ChangeLog: 1, Release initial version on "29/01/19 15:34:41"
|
*
|
********************************************************************************/
|
#include <stdio.h>
|
#include <stdlib.h>
|
#include <unistd.h>
|
#include <time.h>
|
#include <getopt.h>
|
#include <libgen.h>
|
#include <string.h>
|
#include <errno.h>
|
|
#include <mosquitto.h>
|
#include <cjson/cJSON.h>
|
|
#include "logger.h"
|
#include "util_proc.h"
|
#include "config.h"
|
#include "tsl2561.h"
|
#include "ds18b20.h"
|
|
#define PROG_VERSION "v1.0.0"
|
#define DAEMON_PIDFILE "/tmp/.lightd.pid"
|
|
int check_timeout(time_t *last_time, int interval);
|
void *thingsboard_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(" -b[daemon ] Running in daemon mode\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)
|
{
|
int daemon = 0;
|
char *conf_file="/etc/lightd.conf";
|
int debug = 0;
|
int opt;
|
char *progname=NULL;
|
pthread_t tid;
|
|
int rv;
|
float lux = 0.0;
|
time_t last_time = 0;
|
iotd_ctx_t ctx;
|
hwinfo_t *hwinfo = &ctx.hwinfo;
|
|
struct option long_options[] = {
|
{"conf", required_argument, NULL, 'c'},
|
{"daemon", no_argument, NULL, 'b'},
|
{"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:bdvh", long_options, NULL)) != -1)
|
{
|
switch (opt)
|
{
|
case 'c': /* Set configure file */
|
conf_file = optarg;
|
break;
|
|
case 'b': /* Set daemon running */
|
daemon = 1;
|
break;
|
|
case 'd': /* Set debug running */
|
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;
|
}
|
|
}
|
|
/* parser configure file */
|
if( parser_conf(conf_file, &ctx, debug)<0 )
|
{
|
fprintf(stderr, "Parser thingsboard configure file failure\n");
|
return -2;
|
}
|
|
/* install signal proc handler */
|
install_default_signal();
|
|
/* check program already running or not, if already running then exit, or set running as daemon */
|
if( check_set_program_running(daemon, DAEMON_PIDFILE) < 0 )
|
goto cleanup;
|
|
/* initial mosquitto library */
|
mosquitto_lib_init();
|
|
/* initial gpio hardware */
|
gpio_init(&hwinfo->gpio);
|
|
/*
|
* +--------------------------------+
|
* | ThingsBoard MQTT Thread |
|
* +--------------------------------+
|
*/
|
if( thread_start(&tid, thingsboard_worker, &ctx.mqtt ) < 0 )
|
{
|
log_error("Start MQTT subsciber worker thread failure\n");
|
goto cleanup;
|
}
|
log_info("Start MQTT subsciber worker thread ok\n");
|
|
/*
|
* +--------------------------------+
|
* | Auto Light Control Thread |
|
* +--------------------------------+
|
*/
|
log_info("Auto light control thread start\n");
|
while( ! g_signal.stop )
|
{
|
/* TSL2561 update lux in every 5 minutes */
|
if(check_timeout(&last_time, 300))
|
{
|
/* The TSL2561 sensor is not present, do not turn on the light */
|
if( !hwinfo->tsl2561 )
|
{
|
log_error("TSL2561 is not present, do not turn on the light\n");
|
lux = hwinfo->lux_threshold + 100.0;
|
goto nextloop;
|
}
|
|
/* The TSL2561 sensor failed, do not turn on the light */
|
if( tsl2561_get_lux(&lux) < 0 )
|
{
|
log_error("TSL2561 sample failed, do not turn on the light\n");
|
lux = hwinfo->lux_threshold + 100.0;
|
goto nextloop;
|
}
|
|
if( lux > hwinfo->lux_threshold )
|
log_info("Lux[%.3f] > Treshold[%.3f], don't auto light\n", lux, hwinfo->lux_threshold);
|
else
|
log_info("Lux[%.3f] <= Treshold[%.3f], need auto light\n", lux, hwinfo->lux_threshold);
|
}
|
|
/* It's bright enough, no lighting is needed */
|
if( lux > hwinfo->lux_threshold )
|
{
|
goto nextloop;
|
}
|
|
/* It's not bright enough, lighting is necessary */
|
rv = infrared_detect();
|
if( rv != 0 )
|
{
|
log_warn("Someone incoming detected by %s infrared\n", rv&FLAG_INFRARED_INDOOR?"indoor":"hallway");
|
|
turn_light("indoor", ON);
|
turn_light("hallway", ON);
|
|
sleep(hwinfo->light_intval);
|
|
turn_light("indoor", OFF);
|
turn_light("hallway", OFF);
|
}
|
|
nextloop:
|
sleep(1);
|
}
|
|
cleanup:
|
mosquitto_lib_cleanup();
|
log_close();
|
|
return 0;
|
}
|
|
/*
|
* +--------------------------------+
|
* | ThingsBoard MQTT Thread |
|
* +--------------------------------+
|
*/
|
|
int check_timeout(time_t *last_time, int interval)
|
{
|
int timeout = 0;
|
time_t now;
|
|
time(&now);
|
|
if( difftime(now, *last_time)>interval )
|
{
|
timeout = 1;
|
*last_time = now;
|
}
|
|
return timeout;
|
}
|
|
/* compatible with MQTT publish callback function */
|
void pub_connect_callback(struct mosquitto *mosq, void *userdata, int result)
|
{
|
mqtt_ctx_t *ctx = (mqtt_ctx_t *)userdata;
|
hwinfo_t *hwinfo = (hwinfo_t *)ctx->userdata;
|
int rv = 0;
|
char msg[128];
|
float temp = 0.0; /* temperature */
|
int retain = 0;
|
static time_t last_time = 0;
|
|
/* publish time is not arrive */
|
if( !check_timeout(&last_time, ctx->interval) )
|
{
|
return ;
|
}
|
|
log_debug("Publish topic '%s'\n", ctx->pubTopic);
|
|
if( hwinfo->ds18b20 )
|
{
|
memset(msg, 0, sizeof(msg));
|
|
log_debug("DS18B20 temperature sensor enabled, start broadcast it\n");
|
|
if( ds18b20_get_temperature(&temp)<0 )
|
return ;
|
|
snprintf(msg, sizeof(msg), "{\"temperature\":%.2f}", temp);
|
rv = mosquitto_publish(mosq, NULL, ctx->pubTopic, strlen(msg), msg, ctx->pubQos, retain);
|
if( rv )
|
{
|
log_error("Publisher broadcast message '%s' failure: %d\n", msg, rv);
|
}
|
else
|
{
|
log_info("Publisher broadcast message '%s' ok\n", msg);
|
}
|
}
|
|
return ;
|
}
|
|
void sub_connect_callback(struct mosquitto *mosq, void *userdata, int result)
|
{
|
mqtt_ctx_t *ctx = (mqtt_ctx_t *)userdata;
|
|
if( result )
|
{
|
log_error("mosquitto connect to broker server failed, rv=%d\n", result);
|
return ;
|
}
|
|
log_info("mosquitto connect to broker server[%s:%d] successfully\n", ctx->host, ctx->port);
|
mosquitto_subscribe(mosq, NULL, ctx->subTopic, ctx->subQos);
|
}
|
|
void sub_disconnect_callback(struct mosquitto *mosq, void *userdata, int result)
|
{
|
mqtt_ctx_t *ctx = (mqtt_ctx_t *)userdata;
|
|
log_warn("mosquitto disconnect to broker server[%s:%d], reason=%d\n", ctx->host, ctx->port, result);
|
}
|
|
void proc_json_items(cJSON *root, mqtt_ctx_t *ctx)
|
{
|
cJSON *item_method;
|
cJSON *item_params;
|
cJSON *item_device;
|
cJSON *item_status;
|
|
char *method;
|
char *device;
|
int status;
|
|
/* 获取 method 字段 */
|
item_method = cJSON_GetObjectItem(root, "method");
|
if (item_method == NULL || !cJSON_IsString(item_method))
|
{
|
log_error("JSON missing method\n");
|
return;
|
}
|
method = item_method->valuestring;
|
|
/* 只处理 setValue 命令 */
|
if (strcmp(method, "setValue") != 0)
|
{
|
log_error("Ignore method: %s\n", method);
|
return;
|
}
|
|
/* 获取 params 对象 */
|
item_params = cJSON_GetObjectItem(root, "params");
|
if (item_params == NULL || !cJSON_IsObject(item_params))
|
{
|
log_error("JSON missing params\n");
|
return;
|
}
|
|
/* 获取 params.device */
|
item_device = cJSON_GetObjectItem(item_params, "device");
|
if (item_device == NULL || !cJSON_IsString(item_device))
|
{
|
log_error("JSON missing params.device\n");
|
return;
|
}
|
device = item_device->valuestring;
|
|
/* 获取 params.status (布尔值) */
|
item_status = cJSON_GetObjectItem(item_params, "status");
|
if (item_status == NULL || !(cJSON_IsBool(item_status)))
|
{
|
log_error("JSON missing params.status\n");
|
return;
|
}
|
status = cJSON_IsTrue(item_status) ? ON : OFF;
|
|
/* 映射 Light 名称并调用控制函数 */
|
if ( strstr(device, "Indoor") || strstr(device, "indoor"))
|
{
|
turn_light("indoor", status);
|
}
|
else if ( strstr(device, "Hallway") || strstr(device, "hallway") )
|
{
|
turn_light("hallway", status);
|
}
|
|
return;
|
}
|
|
void sub_message_callback(struct mosquitto *mosq, void *userdata, const struct mosquitto_message *message)
|
{
|
mqtt_ctx_t *ctx = (mqtt_ctx_t *)userdata;
|
|
cJSON *root = NULL;
|
|
if ( !message->payloadlen )
|
{
|
log_error("%s (null)\n", message->topic);
|
return ;
|
}
|
|
log_info("Subscriber receive message: '%s'\n", message->payload);
|
|
root = cJSON_Parse(message->payload);
|
if( !root )
|
{
|
log_error("cJSON_Parse parser failure: %s\n", cJSON_GetErrorPtr());
|
return ;
|
}
|
|
proc_json_items(root, ctx);
|
|
/* must delete it, or it will result memory leak */
|
cJSON_Delete(root);
|
|
return ;
|
}
|
|
void *thingsboard_worker(void *args)
|
{
|
mqtt_ctx_t *ctx = (mqtt_ctx_t *)args;
|
struct mosquitto *mosq;
|
bool session = true;
|
int rv = 0;
|
|
mosq = mosquitto_new(ctx->devid, session, ctx);
|
if( !mosq )
|
{
|
log_error("mosquitto_new failure\n");
|
return NULL;
|
}
|
|
/* connnect to broker by token or uid/pwd */
|
if( strlen(ctx->token ) > 0)
|
{
|
log_info("Using token authentication: '%s'\n", ctx->token);
|
mosquitto_username_pw_set(mosq, ctx->token, NULL);
|
}
|
else if( strlen(ctx->uid)> 0 && strlen(ctx->pwd)> 0 )
|
{
|
log_info("Using username/password authentication\n");
|
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_error("mosquitto connect to broker[%s:%d] failure: %s\n", ctx->host, ctx->port, strerror(errno));
|
sleep(1);
|
continue;
|
}
|
|
while(!g_signal.stop)
|
{
|
pub_connect_callback(mosq, ctx, MOSQ_ERR_SUCCESS);
|
|
/* MQTT process in nonblock mode, timeout for 1s */
|
if( MOSQ_ERR_SUCCESS != (rv = mosquitto_loop(mosq, 1000, 1)) )
|
{
|
log_warn("MQTT loop error: %s, reconnecting...\n", mosquitto_strerror(rv));
|
break;
|
}
|
}
|
|
mosquitto_disconnect(mosq);
|
sleep(1);
|
}
|
|
mosquitto_destroy(mosq);
|
return NULL;
|
}
|
