main-snet-streamtestclient.cpp

Go to the documentation of this file.

#include "Config.h"
#include <assert.h>
#include <iostream>
#include <boost/asio.hpp>
#include <boost/program_options.hpp>
#include <boost/regex.hpp>
#include "HdlcdClient.h"
#include "StreamTestEntity.h"

int main(int argc, char* argv[]) {
    try {
        // Declare the supported options.
        boost::program_options::options_description l_Description("Allowed options");
        l_Description.add_options()
            ("help,h",        "produce this help message")
            ("version,v",     "show version information")
            ("connect,c",     boost::program_options::value<std::string>(),
                              "syntax: SerialPort@IPAddess:PortNbr\n"
                              "  linux:   /dev/ttyUSB0@localhost:5001\n"
                              "  windows: //./COM1@example.com:5001")
            ("destination,d", boost::program_options::value<std::string>(),
                              "the destination address (SSA) as a 16bit hex value, e.g., as 0x3FF4 or as 3FF4")
            ("source,s",      boost::program_options::value<std::string>()->default_value("0x4000"),
                              "the OPTIONAL source address (SSA) as a 16bit hex value, e.g., as 0xFFFE or FFFE")
        ;

        // Parse the command line
        boost::program_options::variables_map l_VariablesMap;
        boost::program_options::store(boost::program_options::parse_command_line(argc, argv, l_Description), l_VariablesMap);
        boost::program_options::notify(l_VariablesMap);
        if (l_VariablesMap.count("version")) {
            std::cerr << "s-net(r) stream test client version " << SNET_TOOLS_VERSION_MAJOR << "." << SNET_TOOLS_VERSION_MINOR
                      << " built with hdlcd-devel version " << HDLCD_DEVEL_VERSION_MAJOR << "." << HDLCD_DEVEL_VERSION_MINOR
                      << " and snet-devel version " << SNET_DEVEL_VERSION_MAJOR << "." << SNET_DEVEL_VERSION_MINOR << std::endl;
        } // if

        if (l_VariablesMap.count("help")) {
            std::cout << l_Description << std::endl;
            std::cout << "The s-net(r) stream test client for the HDLC Daemon is Copyright (C) 2016, and GNU GPL'd, by Florian Evers." << std::endl;
            std::cout << "Bug reports, feedback, admiration, abuse, etc, to: https://github.com/Strunzdesign/snet-tools" << std::endl;
            return 1;
        } // if

        if (!l_VariablesMap.count("connect")) {
            std::cout << "snet-streamtestclient: you have to specify one device to connect to" << std::endl;
            std::cout << "snet-streamtestclient: Use --help for more information." << std::endl;
            return 1;
        } // if

        uint16_t l_DstSSA = 0x3FF4;
        if (!l_VariablesMap.count("destination")) {
            std::cout << "snet-streamtestclient: you have to specify the destination address (SSA)" << std::endl;
            std::cout << "snet-streamtestclient: Use --help for more information." << std::endl;
            return 1;
        } else {
            // Convert the provided hexadecimal SSA
            std::istringstream l_Converter(l_VariablesMap["destination"].as<std::string>());
            l_Converter >> std::hex >> l_DstSSA;
        } // else

        uint16_t l_SrcSSA;
        if (!l_VariablesMap.count("source")) {
            assert(false);
        } else {
            // Convert the provided hexadecimal SSA
            std::istringstream l_Converter(l_VariablesMap["source"].as<std::string>());
            l_Converter >> std::hex >> l_SrcSSA;
        } // else

        // Install signal handlers
        boost::asio::io_service l_IoService;
        boost::asio::signal_set l_Signals(l_IoService);
        l_Signals.add(SIGINT);
        l_Signals.add(SIGTERM);
        l_Signals.async_wait([&l_IoService](boost::system::error_code, int){ l_IoService.stop(); });
        
        // Parse the destination specifier
        static boost::regex s_RegEx("^(.*?)@(.*?):(.*?)$");
        boost::smatch l_Match;
        if (boost::regex_match(l_VariablesMap["connect"].as<std::string>(), l_Match, s_RegEx)) {
            // Resolve destination
            boost::asio::ip::tcp::resolver l_Resolver(l_IoService);
            auto l_EndpointIterator = l_Resolver.resolve({ l_Match[2], l_Match[3] });
            
            // Prepare the HDLCd client entity: 0x01 = Payload RX and TX, Ctrl RX and TX
            HdlcdClient l_HdlcdClient(l_IoService, l_Match[1], HdlcdSessionDescriptor(SESSION_TYPE_TRX_ALL, SESSION_FLAGS_DELIVER_RCVD));
            l_HdlcdClient.SetOnClosedCallback([&l_IoService](){ l_IoService.stop(); });

            // Prepare input then connect
            StreamTestEntity l_StreamTestEntity(l_HdlcdClient, l_SrcSSA, l_DstSSA);
            l_HdlcdClient.AsyncConnect(l_EndpointIterator, [&l_StreamTestEntity, &l_Signals](bool a_bSuccess) {
                if (a_bSuccess) {
                    l_StreamTestEntity.Start();
                } else {
                    std::cout << "Failed to connect to the HDLC Daemon!" << std::endl;
                    l_Signals.cancel();
                } // else
            }); // AsyncConnect

            // Start event processing
            l_IoService.run();
        } else {
            throw boost::program_options::validation_error(boost::program_options::validation_error::invalid_option_value);
        } // else
    } catch (std::exception& a_Error) {
        std::cerr << "Exception: " << a_Error.what() << "\n";
    } // catch
    
    return 0;
}