ModbusTCPMaster.h

#ifndef ModbusTCPMaster_H_
#define ModbusTCPMaster_H_
// -------------------------------------------------------------------------
#include <string>
#include <queue>
#include <cc++/socket.h>
#include "ModbusTypes.h"
#include "ModbusClient.h"
#include "UTCPStream.h"
// -------------------------------------------------------------------------
class ModbusTCPMaster:
    public ModbusClient
{
    public:

        ModbusTCPMaster();
        virtual ~ModbusTCPMaster();

        void connect( const std::string& addr, int port );
        void connect( ost::InetAddress addr, int port );
        void disconnect();
        bool isConnection();

        static bool checkConnection( const std::string& ip, int port, int timeout_msec=100 );

        inline void setForceDisconnect( bool s )
        {
            force_disconnect = s;
        }

        void reconnect();
        void cleanInputStream();

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

    protected:

        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 query( ModbusRTU::ModbusAddr addr, ModbusRTU::ModbusMessage& msg, 
                                            ModbusRTU::ModbusMessage& reply, timeout_t timeout ) override;

    private:
        //ost::TCPStream* tcp;
        UTCPStream* tcp;
        ModbusRTU::ModbusData nTransaction;
        std::queue<unsigned char> qrecv;
        PassiveTimer ptTimeout;
        std::string iaddr;
        int port;
        bool force_disconnect;
};
// -------------------------------------------------------------------------
#endif // ModbusTCPMaster_H_
// -------------------------------------------------------------------------