DeviceSocketBase.h

#pragma once
 
#include "DeviceBase.h"
#include "DeviceSocketDefinitions.h"
#include <winsock2.h>

namespace FLImaging
{
    namespace Devices
    {
        class CDeviceSocketServer;
        class CDeviceSocketPacketDistinguisher;

        class FL_EXPORT CDeviceSocketBase : public CDeviceBase
        {
        public:

            CDeviceSocketBase();

            virtual ~CDeviceSocketBase();

            virtual const CResult SetSocketMode(ESocketMode eSocketMode = ESocketMode_FLProtocol_Passive_Plain);

            virtual ESocketMode GetSocketMode();

            virtual bool IsSecured() const;

            virtual const CResult SetConnectionIPAddress(const char* pU8Address, uint16_t u16Port);

            virtual const CResult SetConnectionDNSAddress(const char* pDNSAddress, uint16_t u16Port);

            const SOCKADDR_IN* GetSockAddrIn() const;

            virtual const CResult Initialize() override;

            virtual const CResult Terminate() override;

            virtual bool IsFLProtocol() const;

            virtual bool IsBlockTransferMode() const;

            virtual bool IsPassiveMode() const;

            virtual const CResult SetBlockSize(int32_t i32BlockSize);

            virtual int32_t GetBlockSize() const;
 

            virtual int64_t GetSocket() const;

            virtual const CResult AssignSocketPacketDistinguisher(CDeviceSocketPacketDistinguisher* pSocketPacketDistinguisher);

            virtual CDeviceSocketPacketDistinguisher* GetSocketPacketDistinguisher() const;

            virtual void ClearSocketPacketDistinguisher();

            const CResult CreateKeyPair(uint32_t u32Bits);

            const CResult GetPublicKey(Base::CFLData& fldModulus, Base::CFLData& fldPublicExponent) const;

            const CResult GetPrivateKey(Base::CFLData& fldModulus, Base::CFLData& fldPublicExponent, Base::CFLData& fldPrivateExponent) const;

            const CResult SetPublicKey(const Base::CFLData& fldModulus, const Base::CFLData& fldPublicExponent);

            const CResult SetPrivateKey(const Base::CFLData& fldModulus, const Base::CFLData& fldPublicExponent, const Base::CFLData& fldPrivateExponent);

            const CResult GetIPAddress(uint8_t arrU8IPAddr[4]) const;

            uint16_t GetPort() const;

            CDeviceSocketBase* GetParent() const;

            const Base::CFLString<char>& GetDNS() const;

            virtual void SetLogging(bool bLogging);

            virtual bool IsLogging() const;
 

            virtual void SetCommLogging(bool bCommLogging);

            virtual bool IsCommLogging() const;

            virtual void SetPacketDataLogging(bool bPacketDataLogging);

            virtual bool IsPacketDataLogging() const;

            virtual const CResult SetPacketDataLoggingSize(int64_t i64PacketDataLoggingSize);

            virtual int64_t GetPacketDataLoggingSize() const;

            const CResult AddSentBytes(int64_t i64SentBytes);

            int64_t GetCumulativeSentBytes() const;

            const CResult ResetCumulativeSentBytes();

            const CResult AddRecvBytes(int64_t i64RecvBytes);

            int64_t GetCumulativeRecvBytes() const;

            const CResult ResetCumulativeRecvBytes();
 
 
            DeclareGetClassType();
 
        protected:
 
            virtual bool CheckDeviceLibraries() const override;
 
        protected:
            virtual void SendOnConnecting();
            virtual void SendOnConnected();
            virtual void SendOnDisconnecting();
            virtual void SendOnDisconnected();
            virtual void SendOnReceived(const CDeviceSocketPacket* pSocketPacket);
            virtual void SendOnSending(const CDeviceSocketPacket* pSocketPacket);
            virtual void SendOnSent(const CDeviceSocketPacket* pSocketPacket);
 
            virtual const CResult ConnectWithTimeout(SOCKET hSocket, const sockaddr_in& sockAddrIn, uint32_t u32TimeoutSec);
            virtual bool IsInternetAvailable(uint32_t u32TimeoutSec = 2);
 
        protected:
            Base::CRSACryptography m_rsaCryptography;
            Base::CFLData m_fldModulus;
            Base::CFLData m_fldPublicExponent;
            Base::CFLData m_fldPrivateExponent;
 
            ESocketMode m_eSocketMode;
            volatile SOCKET m_hSocket;
            volatile bool m_bTrouble;
            int32_t m_i32BlockSize;
            Base::CFLString<char> m_astrDNS;
            CDeviceSocketBase* m_pParent;
 
            bool m_bLogging;
            bool m_bCommLogging;
            bool m_bPacketDataLogging;
 
            int64_t m_i64PacketData;
 
            SOCKADDR_IN m_sockAddr;
 
            static WSADATA m_wsaData;
            static int64_t m_i64WsaInitCount;
 
            int64_t m_i64CumulativeSentBytes;
            int64_t m_i64CumulativeRecvBytes;
 
            CDeviceSocketPacketDistinguisher* m_pSocketPacketDistinguisher;
 
            friend class CDeviceSocketClientManager;
            friend class CDeviceSocketClientASCIIManager;
            friend class CDeviceSocketServerASCII;
            friend class CDeviceSocketClientASCII;
        };
    }
}