cClientFTPData.h

Go to the documentation of this file.

//
//
#ifndef _INC_cClientFTPData_H
#define _INC_cClientFTPData_H
#ifndef NO_PRAGMA_ONCE
#pragma once
#endif
 
#include "cClientFTPBase.h"
#include "GrayCore/include/cHeap.h"
#include "GrayCore/include/cFile.h"
#include "GrayCore/include/cFileDir.h"
 
namespace GrayFTP
{
        class cClientFTP;
        typedef cRefPtr<cClientFTP> cClientFTPPtr;
        class cFTPFactory;
        
        class cClientFTPData;
        typedef HRESULT(CALLBACK cClientFTPData::* FTP_DATAFUNC_t)();
 
        enum FTP_DATA_FLAG_TYPE_
        {
                FTP_DATA_FLAG_DATA_TX = 0x0010, 
                FTP_DATA_FLAG_DATA_RX = 0x0020, 
                FTP_DATA_FLAG_VERBOSE = 0x0040,
        };
 
        class cFTPDataParams : public cRefBase
        {
        public:
                // Context info for the current data connection.
                FTP_DATAFUNC_t m_pDataFunc;
                cHeapBlock m_pDataBuffer;               
                WORD m_wDataFlags;                                      
 
                // m_pDataFunc Context specific args.
                cFile m_DataFile;                                       
                cFileFind m_DataDir;                            
                TZ_TYPE m_nTZDisp;                                      
 
        public:
                cFTPDataParams(FTP_DATAFUNC_t pDataFunc, size_t iSize, WORD wDataFlags)
                        : m_pDataFunc(pDataFunc)
                        , m_pDataBuffer(iSize)
                        , m_wDataFlags(wDataFlags)
                {
                        ASSERT(pDataFunc!=nullptr);
                        ASSERT(wDataFlags!=0);
                }
        };
        typedef cRefPtr<cFTPDataParams> cFTPDataParamsPtr;
 
        class cClientFTPData : public cClientFTPBase
        {
 
                typedef cClientFTPBase SUPER_t;
 
        public:
                cClientFTPPtr m_pCC;                    
                cFTPDataParamsPtr m_pDataParams;                        
 
#if 0
                                                                                                        // Current transfer stats.
                DWORD m_dwTransferCurRate;                      
                TIMESYS_t m_timeTransferThrottle;       
#endif
 
        public:
                cClientFTPData(cFTPFactory& rFactory, IUnkObject* pServerConnection, cStreamInput* pStreamInp, cStreamOutput* pStreamOut);
                virtual ~cClientFTPData();
 
                void AssignCmd(cClientFTP* pControl);
 
                static StrLen_t GRAYCALL CmdData_List_DateTime(char *pBuffer, StrLen_t iSizeMax, const cTimeUnits* pSysTime);
                static StrLen_t GRAYCALL CmdData_List_Format(char *pBuffer, StrLen_t iSizeMax, const char* pszName, DWORD dwFileAttributes, STREAM_POS_t nFileSize, const cTimeUnits* pSysTime, bool bVerbose, TZ_TYPE nTZDisp);
                static StrLen_t GRAYCALL CmdData_List_Entry(char *pBuffer, StrLen_t iSizeMax, const cFileFindEntry& fd, bool bVerbose, TZ_TYPE nTZDisp);
 
                HRESULT GRAYCALL CmdData_Retr_SendFile();
                HRESULT GRAYCALL CmdData_Stor_ReceiveFile();
                HRESULT GRAYCALL CmdData_List_SendDir();
                HRESULT GRAYCALL CmdData_List_SendUserGroupRoot();
 
                virtual HRESULT DisposeThis() override;
                virtual HRESULT ProcessStreamPacket() override;
                virtual HRESULT OnTickStream() override;
        };
        typedef cRefPtr<cClientFTPData> cClientFTPDataPtr;
}
 
#endif