开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > 网络截获/分析 > 查看源码
FileManageDlg.cpp
资源名称：Visualhk.rar　[点击查看]
上传用户：cjw5120
上传日期：2022-05-11
资源大小：5032k
文件大小：10k
源码类别：

网络截获/分析

开发平台：

Visual C++

FileManageDlg.cpp：源码内容
  1. // FileManageDlg.cpp : implementation file
  2. //
  4. #include "stdafx.h"
  5. #include "MyClient.h"
  6. #include "FileManageDlg.h"
  7. #include "TcpTran.h"
  9. #include "comm.h"
  11. #include "shellapi.h"
  14. //#include <direct.h>
  15. #define DIR 0
  16. #define FILE 1
  17. #define ID_FILETREE 96367
  19. HTREEITEM m_hItem;
  22. #ifdef _DEBUG
  23. #define new DEBUG_NEW
  24. #undef THIS_FILE
  25. static char THIS_FILE[] = __FILE__;
  26. #endif
  29. CWinThread *pDriverInfoThread = NULL;
  30. CWinThread *pListDirThread              = NULL;
  31. //CWinThread *pSendFileThread             = NULL;
  33. /////////////////////////////////////////////////////////////////////////////
  34. // CFileManageDlg dialog
  35. CFileManageDlg *m_filemanageDlg = NULL;
  39. //DriverInfo线程
  40. UINT DriverInfoThread(LPVOID lparam)
  41. {
  42. CTcpTran m_tcptran;
  43. CFileManageDlg *pDlg=(CFileManageDlg *)lparam;
  45. if(pDlg->ClientSocket!=INVALID_SOCKET)
  46. {  
  47.     int ret;
  48. DRIVER driver;
  50. COMMAND m_command;
  51. m_command.wCmd = CMD_FILE_MANAGE;
  53. ret=m_tcptran.mysend(pDlg->ClientSocket,(char *)&m_command,sizeof(COMMAND),0,60);//获得盘符信息
  55.     pDlg->m_tree.DeleteAllItems();
  56.     pDlg->m_list.DeleteAllItems();
  57. while(ret>0)
  58. {   HTREEITEM hit;
  59. ret=m_tcptran.myrecv(pDlg->ClientSocket,(char *)&driver,sizeof(DRIVER),0,60,NULL,false);//接收目标数据
  60. if(ret<=0||driver.end)
  61. break;
  62.             //表示接收到数据
  63. if(driver.drivertype==DRIVE_FIXED)
  64. {
  65.             hit=pDlg->m_tree.InsertItem(driver.driver,0,0,TVI_ROOT,TVI_LAST);
  66. pDlg->m_tree.InsertItem(NULL,hit,TVI_LAST);
  67. }
  68. if(driver.drivertype==DRIVE_CDROM)
  69. {
  70.             hit=pDlg->m_tree.InsertItem(driver.driver,1,1,TVI_ROOT,TVI_LAST);
  71. }
  72. if(driver.drivertype==DRIVE_REMOVABLE)
  73. {
  74.             hit=pDlg->m_tree.InsertItem(driver.driver,2,2,TVI_ROOT,TVI_LAST);
  75. pDlg->m_tree.InsertItem(NULL,hit,TVI_LAST);
  76. }
  77. }
  78. }
  79. pDriverInfoThread = NULL;
  80. return 0;
  81. }
  84. UINT ListDirThread(LPVOID lparam)
  85. {
  86. CTcpTran m_tcptran;
  87. CFileManageDlg *pDlg=(CFileManageDlg *)lparam;
  88.     pDlg->DeleteTreeChildItem(pDlg->TreeNode);
  89. pDlg->m_list.DeleteAllItems();
  91. if(pDlg->ClientSocket!=INVALID_SOCKET)
  92. {
  93. pDlg->m_list.SetRedraw(FALSE);
  94.         pDlg->m_list.DrawSearch(TRUE);
  95. COMMAND m_command;
  96. memset(&m_command,0,sizeof(COMMAND));
  97. m_command.wCmd = CMD_FILE_GETSUBFILE;
  99.         strcpy(m_command.szCurDir , pDlg->chrPath );
  101. m_tcptran.mysend(pDlg->ClientSocket,(char *)&m_command,sizeof(COMMAND),0,60);//枚举文件
  102. FILEINFO fileinfo;
  103.     do
  104. {
  105.    m_tcptran.myrecv(pDlg->ClientSocket,(char *)&fileinfo,sizeof(FILEINFO),0,60,NULL,false);//接收目标数据
  106.    if(fileinfo.next ==0)
  107.   break;
  109.    HTREEITEM hit;
  110.    if(fileinfo.isdirectory ==1)
  111.    {
  112.     if(fileinfo.filename [0]=='.')
  113.     continue;
  114.     hit=pDlg->m_tree.InsertItem(fileinfo.filename,3,4,pDlg->TreeNode,TVI_LAST);
  115.     pDlg->m_tree.InsertItem(NULL,hit,TVI_LAST);
  116.    }
  117.    else
  118.    {
  119.    int iIcon;
  120.    CString icon;
  121.    icon.Format("%s",fileinfo.filename);
  122.    pDlg->GetFileIcon(icon,&iIcon);
  123.    int n =pDlg->m_list.GetItemCount();
  124.            pDlg->m_list.InsertItem(n,"",iIcon);
  125.        CString kb;
  126.    kb.Format("%dK",fileinfo.filesize);
  127.            pDlg->m_list.SetItemText(n,0,fileinfo.filename);
  128.    pDlg->m_list.SetItemText(n,1,kb);
  129.    pDlg->m_list.SetItemText(n,2,fileinfo.time);
  130.    }
  131. }while(fileinfo.invalidir!=1);
  132.         pDlg->PostMessage(STOP_MESSAGE);
  133. pDlg->m_list.SetRedraw(TRUE);
  134. pDlg->m_list.Invalidate();
  135. }
  136. pListDirThread = NULL;
  137. return 0;
  138. }
  140. CFileManageDlg::CFileManageDlg(CWnd* pParent /*=NULL*/)
  141. : CDialog(CFileManageDlg::IDD, pParent)
  142. {
  143. //{{AFX_DATA_INIT(CFileManageDlg)
  144. // NOTE: the ClassWizard will add member initialization here
  145. //}}AFX_DATA_INIT
  146. m_filemanageDlg = this ;
  147. }
  150. void CFileManageDlg::DoDataExchange(CDataExchange* pDX)
  151. {
  152. CDialog::DoDataExchange(pDX);
  153. //{{AFX_DATA_MAP(CFileManageDlg)
  154. DDX_Control(pDX, IDC_TREE, m_tree);
  155. DDX_Control(pDX, IDC_LIST, m_list);
  156. //}}AFX_DATA_MAP
  157. }
  160. BEGIN_MESSAGE_MAP(CFileManageDlg, CDialog)
  161. //{{AFX_MSG_MAP(CFileManageDlg)
  162. ON_MESSAGE(BEGIN_MESSAGE, OnBeginMsg)
  163. ON_MESSAGE(STOP_MESSAGE, OnStopMsg)
  164. ON_NOTIFY(NM_CLICK, IDC_TREE, OnClickTreeDir)
  165. ON_NOTIFY(NM_RCLICK, IDC_LIST, OnRclickList)
  166. ON_COMMAND(ID_MENUITEM_FILE_DEL, OnMenuitemFileDel)
  167. //}}AFX_MSG_MAP
  168. END_MESSAGE_MAP()
  170. /////////////////////////////////////////////////////////////////////////////
  171. // CFileManageDlg message handlers
  174. BOOL CFileManageDlg::OnInitDialog()
  175. {
  176. CDialog::OnInitDialog();
  178. m_ImageList1.Create(16,16,ILC_COLOR16|ILC_MASK,2,2);
  179. HICON hIcon0 = ::LoadIcon (::AfxGetInstanceHandle(), MAKEINTRESOURCE(IDI_FILE1));
  180. m_ImageList1.Add(hIcon0);
  181. HICON hIcon1 = ::LoadIcon (::AfxGetInstanceHandle(), MAKEINTRESOURCE(IDI_FILE2));
  182. m_ImageList1.Add(hIcon1);
  183. HICON hIcon2 = ::LoadIcon (::AfxGetInstanceHandle(), MAKEINTRESOURCE(IDI_FILE3));
  184. m_ImageList1.Add(hIcon2);
  185. HICON hIcon3 = ::LoadIcon (::AfxGetInstanceHandle(), MAKEINTRESOURCE(IDI_FILE4));
  186. m_ImageList1.Add(hIcon3);
  187. HICON hIcon4 = ::LoadIcon (::AfxGetInstanceHandle(), MAKEINTRESOURCE(IDI_FILE5));
  188. m_ImageList1.Add(hIcon4);
  189. m_tree.SetImageList(&m_ImageList1,TVSIL_NORMAL);//添加图标
  191. // TODO: Add extra initialization here
  192. LONG lStyle = m_list.SendMessage(LVM_GETEXTENDEDLISTVIEWSTYLE);
  193. lStyle |= LVS_EX_FULLROWSELECT | LVS_EX_GRIDLINES | LVS_EX_HEADERDRAGDROP;
  194. m_list.SendMessage(LVM_SETEXTENDEDLISTVIEWSTYLE, 0,(LPARAM)lStyle);
  195. LV_COLUMN lvc;
  197. lvc.mask = LVCF_TEXT | LVCF_SUBITEM | LVCF_WIDTH /*| LVCF_FMT*/;
  199. m_list.DeleteAllItems();
  200.   m_list.InsertColumn(0,"文件名",LVCFMT_LEFT,130);  
  201.   m_list.InsertColumn(1,"文件大小",LVCFMT_LEFT,90);
  202.   m_list.InsertColumn(2,"修改时间",LVCFMT_LEFT,120);
  205. //备注：字符串结尾设置的问题 初始化时要更新
  207. //创建ImageList
  208.     HIMAGELIST hImageList;
  209. //定义文件信息变量
  210. SHFILEINFO shFi;
  211. //获取文件图标信息
  212. char cSysDir[100],dir[4]={0};
  213.     GetSystemDirectory(cSysDir,100);
  214. strncpy(dir,cSysDir,3);
  216.     //得到系统盘符
  217. hImageList = (HIMAGELIST)SHGetFileInfo(dir,0,&shFi,sizeof(shFi),
  218. SHGFI_SYSICONINDEX|SHGFI_SMALLICON);
  219. //将文件图象列表存入m_ctImageList
  220. m_ImageList2.m_hImageList = hImageList;
  221. m_list.SetImageList(&m_ImageList2,LVSIL_SMALL);//添加图标
  224. pDriverInfoThread = AfxBeginThread(DriverInfoThread, (LPVOID)this);
  226. return TRUE;
  227. }
  232. void CFileManageDlg::GetItPath(HTREEITEM hitem, char *path)
  233. {
  234. char tpath[MAX_PATH]="";
  235. char chrtmp[MAX_PATH]="";
  236. do{
  237. strcpy(chrtmp,m_tree.GetItemText(hitem));
  238. strcat(chrtmp,"\");
  239. strcat(chrtmp,tpath);
  240. strcpy(tpath,chrtmp);
  241. hitem=m_tree.GetParentItem(hitem);
  242. }while(hitem!=0);
  243. strcat(tpath,"*");
  244. strcpy(path,tpath);
  246. }
  248. void CFileManageDlg::DeleteTreeChildItem(HTREEITEM hit)
  249. {
  250. HTREEITEM hchild;
  251. hchild=m_tree.GetChildItem(hit);
  252. while(hchild!=0)
  253. {
  254. m_tree.DeleteItem(hchild);
  255. hchild=m_tree.GetChildItem(hit);
  256. }
  258. }
  260. void CFileManageDlg::GetFileIcon(CString &fileName, int *iIcon)
  261. {
  262. CString str=fileName;
  263. SHFILEINFO shfi;
  264. memset(&shfi,0,sizeof(shfi));
  265. if(!SHGetFileInfo(str,FILE_ATTRIBUTE_NORMAL,&shfi,sizeof(shfi),SHGFI_ICON|SHGFI_SMALLICON|SHGFI_USEFILEATTRIBUTES))
  266. {
  267. AfxMessageBox("@@@@");
  268. return;
  269. }
  270. *iIcon=shfi.iIcon;
  271. DestroyIcon(shfi.hIcon);
  272. return ;
  274. }
  276. void CFileManageDlg::OnBeginMsg(WPARAM wParam, LPARAM lParam)
  277. {
  278.      m_list.DrawSearch(TRUE);
  279. }
  281. void CFileManageDlg::OnStopMsg(WPARAM wParam, LPARAM lParam)
  282. {
  283.     m_list.DrawSearch(FALSE);
  284. }
  289. void CFileManageDlg::OnClickTreeDir(NMHDR* pNMHDR, LRESULT* pResult) 
  290. {
  291. // TODO: Add your control notification handler code here
  292. POINT ptMouse;
  293. GetCursorPos(&ptMouse);
  294. m_tree.ScreenToClient(&ptMouse);
  295. HTREEITEM TreeSel=m_tree.HitTest(ptMouse,0);
  296. if(!TreeSel)
  297. return;
  298. GetItPath(TreeSel,chrPath);//得到当前展开的路径
  299. TreeNode=TreeSel;
  300. CString str;
  301. str.Format("%s",chrPath);
  302. str=str.Left(str.ReverseFind('*'));
  303. GetDlgItem(IDC_CURRENT)->SetWindowText(str);
  305. if(pDriverInfoThread != NULL)
  306. {
  307.   ::TerminateThread(pDriverInfoThread->m_hThread, 0);
  308.   ::WaitForSingleObject(pDriverInfoThread->m_hThread, INFINITE);
  309.   CloseHandle(pDriverInfoThread->m_hThread);
  310.   pDriverInfoThread = NULL;
  311. }
  313.     if(pListDirThread != NULL)
  314. {
  315.   ::TerminateThread(pListDirThread->m_hThread, 0);
  316.   ::WaitForSingleObject(pListDirThread->m_hThread, INFINITE);
  317.   CloseHandle(pListDirThread->m_hThread);
  318.   pListDirThread = NULL;
  319. }
  321.     pListDirThread = AfxBeginThread(ListDirThread, (LPVOID)this);
  322. PostMessage(BEGIN_MESSAGE);
  324. *pResult = 0;
  325. }
  327. void CFileManageDlg::OnRclickList(NMHDR* pNMHDR, LRESULT* pResult) 
  328. {
  329. // TODO: Add your control notification handler code here
  330. POINT ptMouse;
  331. POINT ptClientMouse;
  333. GetCursorPos(&ptMouse);
  335. ptClientMouse=ptMouse;
  336. m_list.ScreenToClient(&ptClientMouse);
  337. int ListSel=m_list.HitTest(ptClientMouse,0);
  338. if(ListSel==-1)
  339. return;
  341. UpdateData(TRUE);
  342. CMenu menu;
  343. VERIFY(menu.LoadMenu(IDR_MENU_FILE));
  344. CMenu* pPopup = menu.GetSubMenu(0);
  345. ASSERT(pPopup != NULL);
  346. pPopup->TrackPopupMenu(TPM_LEFTALIGN | TPM_RIGHTBUTTON, ptMouse.x, ptMouse.y, this);
  347. *pResult = 0;
  348. }
  350. void CFileManageDlg::OnMenuitemFileDel() 
  351. {
  352.     char FilePath[MAX_PATH];
  353. CString path;
  354. GetDlgItem(IDC_CURRENT)->GetWindowText(path);
  355. strcpy(FilePath,path);
  356. POSITION pos =m_list.GetFirstSelectedItemPosition();
  357. int i=m_list.GetNextSelectedItem(pos);
  358. if(i>=0)//选择某项
  359. strcat(FilePath,m_list.GetItemText(i,0));
  360. // TODO: Add your command handler code here
  361.     CTcpTran m_tcptran;
  362. COMMAND m_command;
  363.     m_command.wCmd = CMD_FILE_DEL;
  364. strcpy(m_command.szCurDir,FilePath); 
  365. m_tcptran.mysend(ClientSocket,(char *)&m_command,sizeof(COMMAND),0,60);
  367. if(pDriverInfoThread != NULL)
  368. {
  369.   ::TerminateThread(pDriverInfoThread->m_hThread, 0);
  370.   ::WaitForSingleObject(pDriverInfoThread->m_hThread, INFINITE);
  371.   CloseHandle(pDriverInfoThread->m_hThread);
  372.   pDriverInfoThread = NULL;
  373. }
  374.    if(pListDirThread != NULL)
  375. {
  376.   ::TerminateThread(pListDirThread->m_hThread, 0);
  377.   ::WaitForSingleObject(pListDirThread->m_hThread, INFINITE);
  378.   CloseHandle(pListDirThread->m_hThread);
  379.   pListDirThread = NULL;
  380. }
  382. pListDirThread = AfxBeginThread(ListDirThread, (LPVOID)this);
  384. }