ModbusTCPServer.h

// -------------------------------------------------------------------------
#ifndef ModbusTCPServer_H_
#define ModbusTCPServer_H_
// -------------------------------------------------------------------------
#include <string>
#include <queue>
#include <cc++/socket.h>
#include "Mutex.h"
#include "Debug.h"
#include "Configuration.h"
#include "PassiveTimer.h"
#include "ModbusTypes.h"
#include "ModbusServer.h"
#include "ModbusTCPSession.h"

// -------------------------------------------------------------------------
class ModbusTCPServer:
    public ModbusServer,
    public ost::TCPSocket
{
    public:
        ModbusTCPServer( ost::InetAddress& ia, int port = 502 );
        virtual ~ModbusTCPServer();

        virtual ModbusRTU::mbErrCode receive( const std::unordered_set<ModbusRTU::ModbusAddr>& vmbaddr, timeout_t msecTimeout ) override;

        virtual bool waitQuery( const std::unordered_set<ModbusRTU::ModbusAddr>& vmbaddr, timeout_t msec = UniSetTimer::WaitUpTime );

        void setMaxSessions( unsigned int num );
        inline unsigned int getMaxSessions()
        {
            return maxSessions;
        }

        void setSessionTimeout( timeout_t msec );
        inline timeout_t getSessionTimeout()
        {
            return sessTimeout;
        }

        unsigned getCountSessions();

        inline void setIgnoreAddrMode( bool st )
        {
            ignoreAddr = st;
        }
        inline bool getIgnoreAddrMode()
        {
            return ignoreAddr;
        }

        void cleanInputStream();
        virtual void cleanupChannel() override
        {
            cleanInputStream();
        }

        virtual void terminate() override;

        // Сбор статистики по соединениям...
        struct SessionInfo
        {
            SessionInfo( const std::string& a, unsigned int ask ): iaddr(a), askCount(ask) {}

            std::string iaddr;
            unsigned int askCount;
        };

        typedef std::list<SessionInfo> Sessions;

        void getSessions( Sessions& lst );

        inline ost::InetAddress getInetAddress(){ return iaddr; }
        inline ost::tpport_t getInetPort(){ return port; }

    protected:

        virtual ModbusRTU::mbErrCode pre_send_request( ModbusRTU::ModbusMessage& request ) override;
        virtual ModbusRTU::mbErrCode post_send_request( ModbusRTU::ModbusMessage& request ) override;

        // realisation (see ModbusServer.h)
        virtual int getNextData( unsigned char* buf, int len ) override;
        virtual void setChannelTimeout( timeout_t msec ) override;
        virtual ModbusRTU::mbErrCode sendData( unsigned char* buf, int len ) override;

        virtual ModbusRTU::mbErrCode tcp_processing( ost::TCPStream& tcp, ModbusTCP::MBAPHeader& mhead );
        void sessionFinished( ModbusTCPSession* s );

        ost::tpport_t port = { 0 };
        ost::TCPStream tcp;
        ost::InetAddress iaddr;
        std::queue<unsigned char> qrecv;
        ModbusTCP::MBAPHeader curQueryHeader;

        typedef std::list<ModbusTCPSession*> SessionList;
        UniSetTypes::uniset_mutex sMutex;
        SessionList slist;

        bool ignoreAddr = { false };

        unsigned int maxSessions = { 5 };
        unsigned int sessCount = { 0 };

        timeout_t sessTimeout = { 10000 }; // msec

    private:

        std::atomic_bool cancelled;
};
// -------------------------------------------------------------------------
#endif // ModbusTCPServer_H_
// -------------------------------------------------------------------------