SerialPortMultiplexer/main.cpp

#include <iostream>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/un.h>
#include <string>
#include <cstdlib>
#include <vector>
#include <thread>
#include <signal.h>

#include "serial_io.h"
#include "Socket.h"

bool stop = false;

void intHandler(int dummy) 
{
    stop = true;
}

static void printUsage()
{
    std::cout<<"usage mulitplexer [option]\n\
 Available options:\n\
 -h, --help       print this help\n\
 -p, --serialport serial port device to use\n\
 -P, --port       tcp port to use\n\
 -b, --baud       set baud rate with termios id\n\
 -r, --rates      list Available baud rates\n";
}

static void printRates()
{
std::cout<<"Rates:\n\
    B50      0000001\n\
    B75      0000002\n\
    B110     0000003\n\
    B134     0000004\n\
    B150     0000005\n\
    B200     0000006\n\
    B300     0000007\n\
    B600     0000010\n\
    B1200    0000011\n\
    B1800    0000012\n\
    B2400    0000013\n\
    B4800    0000014\n\
    B9600    0000015\n\
    B19200   0000016\n\
    B38400   0000017\n\
    B57600   0010001\n\
    B115200  0010002\n\
    B500000  0010005\n\
    B1000000 0010010\n\
    B1152000 0010011\n\
    B1500000 0010012\n\
    B2000000 0010013\n";
}

class Config
{
public:
    std::string portFileName = "/dev/ttyUSB0";
    unsigned short port = 6856;
    int baud = 0000017;
    bool ignMissingSerialport = false;
};

static int parseCmdArgs(int argc, char** argv, Config *config)
{
    for (int i = 1; i < argc; i++)
    {
        if (std::string(argv[i]) == "--help" || std::string(argv[i]) == "-h")
        {
            printUsage();
            return -1;
        }
        else if (std::string(argv[i]) == "--serialport" || std::string(argv[i]) == "-p")
        {
            if(argc > i) config->portFileName = argv[i+1];
            else return -1;
        }
        else if (std::string(argv[i]) == "--port" || std::string(argv[i]) == "-P")
        {
            if(argc > i) config->port = atoi(argv[i+1]);
            else return -1;
        }
        else if (std::string(argv[i]) == "--baud" || std::string(argv[i]) == "-b")
        {
            if(argc > i) config->baud = atoi(argv[i+1]);
            else return -1;
        }
        else if (std::string(argv[i]) == "-d")
        {
            config->ignMissingSerialport=true;
        }
        
        else if (std::string(argv[i]) == "-r" || std::string(argv[i]) == "--rates")
        {
            printRates();
            return -1;
        }
    }
    return 0;
}

void acceptThreadFunction( TCPServerSocket* servSock,  std::vector<TCPSocket*>* clientSockets )
{
    while(!stop) 
    {   
        TCPSocket* newSock = servSock->accept();
        if(newSock != nullptr)
        {
            clientSockets->push_back(newSock);       // Wait for a client to connect
            clientSockets->back()->send("UVOS serial port multiplexer v0.1\n", sizeof("uvos serial port multiplexer v0.1\n")-1);
            std::cout<<"got client\n";
        }
        std::this_thread::sleep_for(std::chrono::milliseconds(20));
    }
}

int main(int argc, char* argv[])
{
    Config config;
    
    if(parseCmdArgs(argc, argv, &config) != 0) return -1;
    
    std::cout<<"UVOS serial mulitplexer v0.2\n";
    
    std::cout<<"Using serial port: "<<config.portFileName<<" at "<<config.baud<<" baud\n";
    int serial = serialport_init(config.portFileName.c_str(), config.baud);
    if(serial == -1)
    {
        if(config.ignMissingSerialport) std::cout<<"Continuing in demo mode\n";
        else return 1;
    }

    std::vector<TCPSocket*> clientSockets;
    
    std::cout<<"opening TCP socket on port "<<config.port<<'\n';
    TCPServerSocket servSock(config.port, 5, true);     // Server Socket object
    servSock.setBlocking(false);
    
    std::thread acceptThread(acceptThreadFunction, &servSock, &clientSockets);
    
    char buffer[256];
    
    signal(SIGINT, intHandler);
    //signal(SIGTERM, intHandler);
    
    std::cout<<"starting loop\n";
    
    while(!stop)
    {
        int readlen = sRead(serial, buffer, 256);
        //std::cout<<clientSockets.size()<<std::endl;
        for(unsigned int i = 0; i < clientSockets.size(); i++)
        {
            try
            {

                if(readlen > 0) 
                {
                    std::cout<<"sending: ";
                    for( int j = 0; j < readlen; j++  )std::cout<<buffer[j];
                    clientSockets[i]->send(buffer, readlen);
                    std::cout<<std::endl;
                }
                
                char inBuffer[256];
                int reclen = clientSockets[i]->recv(inBuffer, 256);
                if( reclen > 0 ) 
                {
                    std::cout<<"rec "<<reclen<<'\n';
                    for( int j = 0; j < reclen; j++  )std::cout<<inBuffer[j];
                    sWrite(serial, inBuffer, reclen);
                }
                else if(reclen == 0)
                {
                    std::cout<<"client "<<i<<" disconnected"<<'\n';
                    clientSockets[i]->cleanUp();
                    clientSockets.erase(clientSockets.begin()+i);
                    i--;
                    if(i < 0) i=0;
                }
            }
            catch(SocketException e)
            {
                std::cout<<e.what()<<std::endl;
                clientSockets[i]->cleanUp();
                clientSockets.erase(clientSockets.begin()+i);
                i--;
                if(i < 0) i=0;
            }
        }
        std::this_thread::sleep_for(std::chrono::milliseconds(20));
    }
    
    acceptThread.join();
    for(unsigned int i = 0; i < clientSockets.size(); i++) clientSockets[i]->cleanUp();
    servSock.cleanUp();
    return 0;
}