ModbusTCPSession.h

// -------------------------------------------------------------------------
#ifndef ModbusTCPSession_H_
#define ModbusTCPSession_H_
// -------------------------------------------------------------------------
#include <string>
#include <queue>
#include <unordered_map>
#include <ev++.h>
#include "ModbusServerSlot.h"
#include "ModbusServer.h"
#include "PassiveTimer.h"
#include "UTCPCore.h"
#include "UTCPStream.h"
// -------------------------------------------------------------------------
namespace uniset
{
// -------------------------------------------------------------------------
class ModbusTCPSession:
    public ModbusServerSlot,
    public ModbusServer
{
    public:

        ModbusTCPSession( const Poco::Net::StreamSocket& s, const std::unordered_set<ModbusRTU::ModbusAddr>& vmbaddr, timeout_t timeout );
        virtual ~ModbusTCPSession();

        void cleanInputStream();

        virtual void cleanupChannel() override;
        virtual void terminate() override;

        typedef sigc::slot<void, const ModbusTCPSession*> FinalSlot;

        void connectFinalSession( FinalSlot sl );

        inline std::string getClientAddress() const
        {
            return caddr;
        }

        void setSessionTimeout( double t );

        // запуск обработки входящих запросов
        void run( ev::loop_ref& loop );

        virtual bool isActive() const override;

        void iowait( timeout_t msec );

    protected:

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

        void callback( ev::io& watcher, int revents );
        void onTimeout( ev::timer& watcher, int revents );
        virtual void readEvent( ev::io& watcher );
        virtual void writeEvent( ev::io& watcher );
        virtual void final();

        virtual size_t getNextData( unsigned char* buf, int len ) override;
        virtual void setChannelTimeout( timeout_t msec );
        virtual ModbusRTU::mbErrCode sendData( unsigned char* buf, int len ) override;
        virtual ModbusRTU::mbErrCode tcp_processing( ModbusRTU::ADUHeader& mhead );
        virtual ModbusRTU::mbErrCode make_adu_header( ModbusRTU::ModbusMessage& request ) override;
        virtual ModbusRTU::mbErrCode post_send_request(ModbusRTU::ModbusMessage& request ) override;

        virtual ModbusRTU::mbErrCode readCoilStatus( ModbusRTU::ReadCoilMessage& query,
                                                     ModbusRTU::ReadCoilRetMessage& reply );

        virtual ModbusRTU::mbErrCode readInputStatus( ModbusRTU::ReadInputStatusMessage& query,
                                                      ModbusRTU::ReadInputStatusRetMessage& reply );

        virtual ModbusRTU::mbErrCode readOutputRegisters( ModbusRTU::ReadOutputMessage& query,
                                                          ModbusRTU::ReadOutputRetMessage& reply );

        virtual ModbusRTU::mbErrCode readInputRegisters( ModbusRTU::ReadInputMessage& query,
                                                         ModbusRTU::ReadInputRetMessage& reply );

        virtual ModbusRTU::mbErrCode forceSingleCoil( ModbusRTU::ForceSingleCoilMessage& query,
                                                      ModbusRTU::ForceSingleCoilRetMessage& reply );

        virtual ModbusRTU::mbErrCode writeOutputSingleRegister( ModbusRTU::WriteSingleOutputMessage& query,
                ModbusRTU::WriteSingleOutputRetMessage& reply );

        virtual ModbusRTU::mbErrCode forceMultipleCoils( ModbusRTU::ForceCoilsMessage& query,
                                                         ModbusRTU::ForceCoilsRetMessage& reply );

        virtual ModbusRTU::mbErrCode writeOutputRegisters( ModbusRTU::WriteOutputMessage& query,
                ModbusRTU::WriteOutputRetMessage& reply );

        virtual ModbusRTU::mbErrCode diagnostics( ModbusRTU::DiagnosticMessage& query,
                                                  ModbusRTU::DiagnosticRetMessage& reply );

        virtual ModbusRTU::mbErrCode read4314( ModbusRTU::MEIMessageRDI& query,
                                               ModbusRTU::MEIMessageRetRDI& reply );

        virtual ModbusRTU::mbErrCode journalCommand( ModbusRTU::JournalCommandMessage& query,
                                                     ModbusRTU::JournalCommandRetMessage& reply );

        virtual ModbusRTU::mbErrCode setDateTime( ModbusRTU::SetDateTimeMessage& query,
                                                  ModbusRTU::SetDateTimeRetMessage& reply );

        virtual ModbusRTU::mbErrCode remoteService( ModbusRTU::RemoteServiceMessage& query,
                                                    ModbusRTU::RemoteServiceRetMessage& reply );

        virtual ModbusRTU::mbErrCode fileTransfer( ModbusRTU::FileTransferMessage& query,
                                                   ModbusRTU::FileTransferRetMessage& reply );

    private:
        std::queue<unsigned char> qrecv;
        std::unordered_set<ModbusRTU::ModbusAddr> vaddr;
        ModbusRTU::ADUHeader curQueryHeader;
        PassiveTimer ptTimeout;
        timeout_t timeout = { 0 };
        ModbusRTU::ModbusMessage buf;

        ev::io  io;
        ev::timer ioTimeout;

        std::shared_ptr<UTCPStream> sock;
        std::queue<UTCPCore::Buffer*> qsend;
        double sessTimeout = { 10.0 };

        bool ignoreAddr = { false };
        std::string peername = { "" };

        std::string caddr = { "" };

        FinalSlot slFin;

        std::atomic_bool cancelled = { false };
        PassiveTimer pt;
        PassiveTimer ptWait;
};
// -------------------------------------------------------------------------
} // end of namespace uniset
// -------------------------------------------------------------------------
#endif // ModbusTCPSession_H_
// -------------------------------------------------------------------------