Merge branch 'master' of http://master.iot-yun.club:8088/r/raspberrypi

guowenxue

2023-07-10 5e9d03d507aad324a803eb8795e0eed6fb671761

 apue/project_socket/main/client_main.c

New file
@@ -0,0 +1,244 @@
#include <stdio.h>
#include <errno.h>
#include <string.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <stdlib.h>
#include <getopt.h>
#include <time.h>

#include "logger.h"
#include "ds18b20.h"
#include "packet.h"
#include "socket.h"
#include "sqlite_blob.h"
#include "util_proc.h"
#include "util_time.h"


void print_usage(char *progname)
{
    printf("%s usage: \n", progname);
    printf("-i(--ipaddr)  : sepcify server IP address\n");
    printf("-p(--port)    : sepcify server port.\n");
    printf("-I(--interval): sepcify report interval time\n");
    printf("-s(--sn)      : sepcify device serial number\n");
    printf("-d(--debug)   : run as debug mode\n");
    printf("-h(--Help)    : print this help information.\n");

    return ;
}

static int check_sample_time(time_t *last_time, int interval);

int main(int argc, char **argv)
{
    int                  rv = -1;
    int                  port;
    char                *serverip;
    int                  interval = 30; /* default report termperature every 30 seconds */
    int                  sn = 1;        /* default serial number for device ID */
    int                  debug = 0;     /* running in debug mode or not */
    int                  loglevel = LOG_LEVEL_INFO;
    char                *logfile="client.log";

    socket_ctx_t         sock;
    time_t               last_time = 0;
    int                  sample_flag = 0;

    char                 pack_buf[1024];
    int                  pack_bytes;
    pack_info_t          pack_info;
    pack_proc_t          pack_proc = packet_segmented_pack; /* use string packet */

    struct option        opts[] = {
        {"ipaddr", required_argument, NULL, 'i'},
        {"port", required_argument, NULL, 'p'},
        {"interval", no_argument, NULL, 'I'},
        {"debug", no_argument, NULL, 'd'},
        {"sn", required_argument, NULL, 's'},
        {"help", no_argument, NULL, 'h'},
        {NULL, 0, NULL, 0}
    };

    while( (rv=getopt_long(argc, argv, "i:p:dI:s:h", opts, NULL)) != -1 )
    {
        switch(rv)
        {
            case 'i':
                serverip=optarg;
                break;

            case 'p':
                port=atoi(optarg);
                break;

            case 'd':
                debug=1;
                logfile="stdout";
                loglevel=LOG_LEVEL_DEBUG;
                break;

            case 'I':
                interval=atoi(optarg);
                break;

            case 's':
                sn=atoi(optarg);
                break;

            case 'h':
                print_usage(argv[0]);
                return 0;
        }

    }

    if( !serverip || !port )
    {
        print_usage(argv[0]);
        return 0;
    }

    /* set logger to $logfile with level info */
    if( logger_init(logfile, loglevel) < 0 )
    {
        fprintf(stderr, "Initial logger file '%s' failure: %s\n", logfile, strerror(errno));
        return 1;
    }

    if( !debug )
    {
        log_info("set program running on background now.\n");
        daemon(1, 1); /* don't change work path, don't close opened file descriptor */
    }

    install_default_signal();

    log_info("program start running.\n");

    if( database_init("client.db") < 0 )
    {
        return 2;
    }

    socket_init(&sock, serverip, port);

    while( !g_signal.stop )
    {
        /* +----------------------------------+
         * |  check and sample temperature    |
         * +----------------------------------+*/

        sample_flag = 0; /* clear sample flag */

        if( check_sample_time(&last_time, interval) )
        {
            log_debug("start DS18B20 sample termperature\n");

            if( (rv=ds18b20_get_temperature(&pack_info.temper)) < 0 )
            {
                log_error("DS18B20 sample temperature failure, rv=%d\n", rv);
                continue;
            }
            log_info("DS18B20 sample termperature %d.%d oC\n",
                    temper_integer(pack_info.temper), temper_fract(pack_info.temper));

            get_devid(pack_info.devid, DEVID_LEN, sn);
            get_time(&pack_info.sample_time);

            pack_bytes = pack_proc(&pack_info, pack_buf, sizeof(pack_buf));
            sample_flag = 1; /* set sample flag */
        }

        /* +---------------------------------+
         * |  check and do socket connect    |
         * +---------------------------------+*/

        /* start connect to server if not connected */
        if( sock.fd < 0 )
        {
            socket_connect(&sock);
        }

        /* check socket connected or not  */
        if( sock_check_connect(sock.fd) < 0 )
        {
            if( sock.fd > 0 )
            {
                log_error("socket got disconnected, terminate it and reconnect now.\n");
                socket_term(&sock); /* close the soket */
            }
        }

        /* +-------------------------------+
         * |      socket disconnect        |
         * +-------------------------------+*/
        if( sock.fd < 0 )
        {
            if( sample_flag )
            {
                blobdb_push_packet(pack_buf, pack_bytes);
            }

            continue;
        }

        /* +-------------------------------+
         * |      socket connected         |
         * +-------------------------------+*/

        /*  socket send sample packet */
        if( sample_flag )
        {
            log_debug("socket send sample packet bytes[%d]: %s\n", pack_bytes, pack_buf);
            if( socket_send(&sock, pack_buf, pack_bytes) < 0 )
            {
                log_warn("socket send sample packet failure, save it in database now.\n");
                blobdb_push_packet(pack_buf, pack_bytes);
                socket_term(&sock); /* close the soket */
            }
        }

        /*  socket send packet in database  */
        if( !blobdb_pop_packet(pack_buf, sizeof(pack_buf), &pack_bytes) )
        {
            log_debug("socket send database packet bytes[%d]: %s\n", pack_bytes, pack_buf);
            if( socket_send(&sock, pack_buf, pack_bytes) < 0 )
            {
                log_error("socket send database packet failure");
                socket_term(&sock); /* close the soket */
            }
            else
            {
                log_warn("socket send database packet okay, remove it from database now.\n");
                blobdb_del_packet();
            }
        }

        msleep(100);
    }

    socket_term(&sock);
    database_term();

    return 0;
}

int check_sample_time(time_t *last_time, int interval)
{
    int                  need = 0; /* no need sample now */
    time_t               now;

    time(&now);

    if( now >= *last_time+interval )
    {
        need = 1; /* need sample now  */
        *last_time = now;
    }

    return need;
}