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( ModbusRTU::ModbusAddr mbaddr, timeout_t msecTimeout ) override;

        virtual bool waitQuery( ModbusRTU::ModbusAddr mbaddr, 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 );

    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;
        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;

        unsigned int maxSessions;
        unsigned int sessCount;

        timeout_t sessTimeout;

    private:

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