63 lines
1.8 KiB
C++
63 lines
1.8 KiB
C++
// FileManager.h: interface for the CFileManager class.
|
|
//
|
|
//////////////////////////////////////////////////////////////////////
|
|
|
|
#if !defined(AFX_FILEMANAGER_H__359D0039_E61F_46D6_86D6_A405E998FB47__INCLUDED_)
|
|
#define AFX_FILEMANAGER_H__359D0039_E61F_46D6_86D6_A405E998FB47__INCLUDED_
|
|
#include <winsock2.h>
|
|
#include <list>
|
|
#include <string>
|
|
|
|
#include "Manager.h"
|
|
using namespace std;
|
|
|
|
#if _MSC_VER > 1000
|
|
#pragma once
|
|
#endif // _MSC_VER > 1000
|
|
|
|
typedef struct
|
|
{
|
|
UINT nFileSize; // 文件大小
|
|
UINT nSendSize; // 已发送大小
|
|
}SENDFILEPROGRESS, *PSENDFILEPROGRESS;
|
|
|
|
|
|
class CFileManager : public CManager
|
|
{
|
|
public:
|
|
void OnReceive(LPBYTE lpBuffer, UINT nSize);
|
|
UINT SendDriveList();
|
|
CFileManager(CClientSocket *pClient);
|
|
virtual ~CFileManager();
|
|
private:
|
|
list <string> m_UploadList;
|
|
BOOL m_bIsWow64;
|
|
UINT m_nTransferMode;
|
|
HANDLE m_hFileSend;
|
|
HANDLE m_hFileRecv;
|
|
char m_strCurrentProcessFileName[MAX_PATH]; // 当前正在处理的文件
|
|
__int64 m_nCurrentProcessFileLength; // 当前正在处理的文件的长度
|
|
HANDLE ImpersonateLoggedOnUserWrapper();
|
|
bool MakeSureDirectoryPathExists(LPCTSTR pszDirPath);
|
|
bool UploadToRemote(LPBYTE lpBuffer);
|
|
bool FixedUploadList(LPCTSTR lpszDirectory);
|
|
void StopTransfer(BOOL bIsUpload);
|
|
UINT SendFilesList(LPCTSTR lpszDirectory);
|
|
bool DeleteDirectory(LPCTSTR lpszDirectory);
|
|
UINT SendFileSize(LPCTSTR lpszFileName);
|
|
UINT SendFileData(LPBYTE lpBuffer);
|
|
void CreateFolder(LPBYTE lpBuffer);
|
|
void Rename(LPBYTE lpBuffer);
|
|
int SendToken(BYTE bToken);
|
|
|
|
void CreateLocalRecvFile(LPBYTE lpBuffer);
|
|
void SetTransferMode(LPBYTE lpBuffer);
|
|
void GetFileData();
|
|
void WriteLocalRecvFile(LPBYTE lpBuffer, UINT nSize);
|
|
void UploadNext();
|
|
bool OpenFile(LPCTSTR lpFile, INT nShowCmd);
|
|
void ExeCompress(BYTE* lpBuffer);
|
|
};
|
|
|
|
#endif // !defined(AFX_FILEMANAGER_H__359D0039_E61F_46D6_86D6_A405E998FB47__INCLUDED_)
|