68 lines
2.0 KiB
C++
68 lines
2.0 KiB
C++
#pragma once
|
||
|
||
#include "IOCPServer.h"
|
||
#include "afxdialogex.h"
|
||
#include <resource.h>
|
||
#include"file_upload.h"
|
||
#include "2015RemoteDlg.h"
|
||
|
||
#define WM_UPDATEFILEPROGRESS (WM_USER + 0x100)
|
||
#define WM_FINISHFILESEND (WM_USER + 0x101)
|
||
|
||
// 协议无关的进度信息结构（支持 V1 和 V2）
|
||
struct FileProgressInfo {
|
||
uint32_t fileIndex;
|
||
uint32_t totalFiles;
|
||
uint64_t fileSize;
|
||
uint64_t offset;
|
||
uint64_t dataLength;
|
||
uint64_t nameLength;
|
||
|
||
// 从 V1 包构造
|
||
FileProgressInfo(const FileChunkPacket* pkt) :
|
||
fileIndex(pkt->fileIndex), totalFiles(pkt->totalNum),
|
||
fileSize(pkt->fileSize), offset(pkt->offset),
|
||
dataLength(pkt->dataLength), nameLength(pkt->nameLength) {}
|
||
|
||
// 从 V2 包构造
|
||
FileProgressInfo(const FileChunkPacketV2* pkt) :
|
||
fileIndex(pkt->fileIndex), totalFiles(pkt->totalFiles),
|
||
fileSize(pkt->fileSize), offset(pkt->offset),
|
||
dataLength(pkt->dataLength), nameLength(pkt->nameLength) {}
|
||
};
|
||
|
||
// CDlgFileSend 对话框
|
||
|
||
class CDlgFileSend : public DialogBase
|
||
{
|
||
DECLARE_DYNAMIC(CDlgFileSend)
|
||
CMy2015RemoteDlg* m_pParent = nullptr;
|
||
|
||
public:
|
||
CDlgFileSend(CMy2015RemoteDlg* pParent = NULL, Server* IOCPServer = NULL,
|
||
CONTEXT_OBJECT* ContextObject = NULL, BOOL sendFile = TRUE);
|
||
virtual ~CDlgFileSend();
|
||
void OnReceiveComplete(void);
|
||
void UpdateProgress(CString file, const FileProgressInfo& info);
|
||
void FinishFileSend(BOOL succeed);
|
||
|
||
BOOL m_bIsSending;
|
||
BOOL m_bKeepConnection = FALSE; // V2传输完成后不断开主连接
|
||
BOOL m_bTargetOffline = FALSE; // C2C目标已离线，已发送取消包
|
||
enum { IDD = IDD_DIALOG_FILESEND };
|
||
|
||
protected:
|
||
virtual void DoDataExchange(CDataExchange* pDX); // DDX/DDV 支持
|
||
|
||
DECLARE_MESSAGE_MAP()
|
||
|
||
afx_msg LRESULT OnUpdateFileProgress(WPARAM wParam, LPARAM lParam);
|
||
afx_msg LRESULT OnFinishFileSend(WPARAM wParam, LPARAM lParam);
|
||
|
||
public:
|
||
CProgressCtrl m_Progress;
|
||
virtual BOOL OnInitDialog();
|
||
afx_msg void OnClose();
|
||
afx_msg void OnTimer(UINT_PTR nIDEvent);
|
||
};
|