cFTPFactory.h

Go to the documentation of this file.

//
//
#ifndef _INC_cFTPFactory_H
#define _INC_cFTPFactory_H
#ifndef NO_PRAGMA_ONCE
#pragma once
#endif
 
#include "GrayFTPInt.h"
#include "cFTPCfg.h"
#include "Client/cClientFTP.h"
#include "GrayCore/include/cAppImpl.h"
#ifdef USE_UPNP
#include "GrayLib/include/WinAPI/cUPnPNAT.h"
#include "GrayLib/include/WinAPI/cComInit.h"
#endif
 
namespace GrayFTP
{
        class cClientFTP;
        class cFTPFactory;
 
        class cFTPFactoryData : public IProtocolFactory
        {
        public:
                cFTPFactory&  m_rFactory;       
                cClientFTP* m_pClient;          
        public:
                cFTPFactoryData(cFTPFactory& rFactory, cClientFTP* pClient = nullptr)
                        : m_rFactory(rFactory)
                        , m_pClient(pClient)
                {
                }
                virtual const char* get_ProtocolName() const override
                {
                        return "FTPData";
                }
                virtual HRESULT TestProtocol(const BYTE* pszPrefixData, size_t iPrefixLen) const override
                {
                        return S_OK;
                }
                virtual cProtocolStreamPtr CreateProtocolStream(IUnkObject* pServerConnection, cStreamInput* pStreamInp, cStreamOutput* pStreamOut) override;
        };
 
        class cFTPFactory : public IProtocolFactory     // cNetServer protocol for command/control connection.
        {
 
        public:
                cNetServer& m_rNetServer;
                cFTPCfg& m_rCfg;
                cNetAddrName m_AddrMe;                          
 
                cFTPFactoryData m_ProtocolData;         
 
                cNetServerPortPtr m_pSocketCtrl;                
                cNetServerPortPtr m_pSocketDataPasv;    
 
                cArrayRef<cClientFTP> m_aConnectCmd;    
 
        protected:
                virtual const char* get_ProtocolName() const override
                {
                        // IProtocolFactory
                        return "FTP";
                }
                virtual HRESULT TestProtocol(const BYTE* pszPrefixData, size_t iPrefixLen) const override
                {
                        return S_OK;
                }
                virtual cProtocolStreamPtr CreateProtocolStream(IUnkObject* pServerConnection, cStreamInput* pStreamInp, cStreamOutput* pStreamOut) override;
 
        public:
                cFTPFactory(cNetServer& rNetServer, cFTPCfg& rCfg);
                virtual ~cFTPFactory();
 
                // Event types. LOG_ATTR_MASK_t
                StrLen_t _cdecl LogEventF(LOG_ATTR_MASK_t dwAttrMask, const LOGCHAR_t* pszFormat, ...);
 
                HRESULT CheckAddressBeforeAccept(const cNetAddress& AddrNew) const;
                bool OnClientClose(cNetServerConnection* pClient);
                HRESULT OpenDPasvListen(OUT cNetServerPortPtr& pDPasv);
 
                bool OpenFactory();
                void CloseFactory();
        };
} 
#endif