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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > 网格计算 > 查看源码
FtpWorker.cpp
资源名称：fmailstable.rar　[点击查看]
上传用户：maryhy001
上传日期：2007-05-02
资源大小：2317k
文件大小：9k
源码类别：

网格计算

开发平台：

Visual C++

FtpWorker.cpp：源码内容
  2. #include "stdAfx.h"
  3. #include "FtpWorker.h"
  5. CFtpWorker::CFtpWorker()
  6. {
  7. m_hMsgWnd = NULL;
  8. m_bIsConnected = FALSE;
  10. m_pFtpConnection = NULL;
  11. m_nConnectionTimeout = 10000;
  13. m_hEventKill = CreateEvent(NULL, FALSE, FALSE, NULL);
  15. m_poInternetSession = new CInternetSession("trfFtpWorker/1.0");
  16. m_poInternetSession->SetOption(INTERNET_OPTION_CONNECT_TIMEOUT, m_nConnectionTimeout);
  17. m_poInternetSession->SetOption(INTERNET_OPTION_RECEIVE_TIMEOUT, m_nConnectionTimeout);
  18. m_poInternetSession->SetOption(INTERNET_OPTION_SEND_TIMEOUT, m_nConnectionTimeout);
  19. }
  21. CFtpWorker::CFtpWorker(CFtpSite &oFtpSite)
  22. {
  23. m_hMsgWnd = NULL;
  24. m_oFptSite = oFtpSite;
  25. m_bIsConnected = FALSE;
  27. m_pFtpConnection = NULL;
  28. m_nConnectionTimeout = 10000;
  30. m_hEventKill = CreateEvent(NULL, FALSE, FALSE, NULL);
  32. m_poInternetSession = new CInternetSession("trfFtpWorker/1.0");
  33. m_poInternetSession->SetOption(INTERNET_OPTION_CONNECT_TIMEOUT, m_nConnectionTimeout);
  34. m_poInternetSession->SetOption(INTERNET_OPTION_RECEIVE_TIMEOUT, m_nConnectionTimeout);
  35. m_poInternetSession->SetOption(INTERNET_OPTION_SEND_TIMEOUT, m_nConnectionTimeout);
  36. }
  38. CFtpWorker::~CFtpWorker()
  39. {
  40. CloseHandle(this->m_hEventKill);
  41. if(NULL != m_poInternetSession)
  42. {
  43. delete m_poInternetSession;
  44. }
  45. this->Disconnect();
  46. }
  48. BOOL CFtpWorker::IsConnected(void)
  49. {
  50. return m_bIsConnected;
  51. }
  53. CFtpSite &CFtpWorker::GetFtpSiteInfo(void) const
  54. {
  55. return (CFtpSite&)m_oFptSite;
  56. }
  58. void CFtpWorker::SetFtpSiteInfo(CFtpSite &oFtpSite)
  59. {
  60. this->m_oFptSite = oFtpSite;
  61. }
  63. BOOL CFtpWorker::Connect(void)
  64. {
  65. try
  66. {
  67. m_pFtpConnection = m_poInternetSession->GetFtpConnection(m_oFptSite.m_sHost, 
  68. m_oFptSite.m_sUser, m_oFptSite.m_sPassword, 
  69. m_oFptSite.m_nPort, m_oFptSite.m_bPASVMode);
  71. return (m_bIsConnected = (NULL != m_pFtpConnection));
  72. }
  73. catch (CInternetException *ex)
  74. {
  75. ex->Delete();
  76. m_pFtpConnection = NULL;
  77. m_bIsConnected = FALSE;
  79. return FALSE;
  80. }
  81. return FALSE;
  82. }
  84. void CFtpWorker::Disconnect(void)
  85. {
  86. if(NULL != m_pFtpConnection)
  87. {
  88. m_pFtpConnection->Close();
  89. delete m_pFtpConnection;
  90. m_pFtpConnection = NULL;
  91. m_bIsConnected = FALSE;
  92. }
  93. }
  95. BOOL CFtpWorker::GetCurrentDir(CString strDirName)
  96. {
  97. try
  98. {
  99. strDirName.Empty();
  100. return m_pFtpConnection->GetCurrentDirectory(strDirName);
  101. }
  102. catch (CInternetException *ex)
  103. {
  104. ex->Delete();
  105. return FALSE;
  106. }
  107. return FALSE;
  108. }
  110. BOOL CFtpWorker::ChangeCurrentDir(LPCSTR lpszToDir)
  111. {
  112. try
  113. {
  114. if(NULL == lpszToDir) return FALSE;
  115. return m_pFtpConnection->SetCurrentDirectory(lpszToDir);
  116. }
  117. catch (CInternetException *ex)
  118. {
  119. ex->Delete();
  120. return FALSE;
  121. }
  122. return FALSE;
  123. }
  125. BOOL CFtpWorker::GetCurSubDirectorys(CStringList &oSubDirNames, LPCSTR lpszSubDirMask)
  126. {
  127. CFtpFileFind oFtpFileFinder(this->m_pFtpConnection);
  128. try
  129. {
  130. CString strFindMask = (NULL == lpszSubDirMask) ? "*" : lpszSubDirMask;
  132. BOOL bContinue = oFtpFileFinder.FindFile(strFindMask.GetBuffer(0));
  134. for(oSubDirNames.RemoveAll(); bContinue; )
  135. {
  136. bContinue = oFtpFileFinder.FindNextFile();
  138. if(oFtpFileFinder.IsDots()) continue;
  140. if(!oFtpFileFinder.IsDirectory()) continue;
  142. oSubDirNames.AddTail(oFtpFileFinder.GetFileName());
  143. }
  144. oFtpFileFinder.Close();
  145. return TRUE;
  146. }
  147. catch(CInternetException *ex)
  148. {
  149. ex->Delete();
  150. oFtpFileFinder.Close();
  151. return FALSE;
  152. }
  153. return FALSE;
  154. }
  156. BOOL CFtpWorker::GetFileNamesOfCurDir(CList<LPREMOTEFILEINFO,LPREMOTEFILEINFO&> &oFileInfos, LPCSTR lpszFileNameMask)
  157. {
  158. CFtpFileFind oFtpFileFinder(this->m_pFtpConnection);
  159. try
  160. {
  161. CString strFindMask = (NULL == lpszFileNameMask) ? "*" : lpszFileNameMask;
  163. BOOL bContinue = oFtpFileFinder.FindFile(strFindMask.GetBuffer(0));
  165. for(; oFileInfos.GetCount() > 0; )
  166. {
  167. REMOTEFILEINFO *pitem = (REMOTEFILEINFO*)oFileInfos.GetHead();
  168. if(NULL != pitem) delete pitem;
  169. oFileInfos.RemoveHead();
  170. }
  171. for(; bContinue; )
  172. {
  173. bContinue = oFtpFileFinder.FindNextFile();
  175. if(oFtpFileFinder.IsDots()) continue;
  177. if(oFtpFileFinder.IsDirectory()) continue;
  179. REMOTEFILEINFO *lpRemoteFile = new REMOTEFILEINFO;
  180. memset(lpRemoteFile, 0x0, sizeof(REMOTEFILEINFO));
  182. lpRemoteFile->dwFilesize = oFtpFileFinder.GetLength();
  183. _snprintf(lpRemoteFile->szRemoteFile, MAX_PATH, "%s", oFtpFileFinder.GetFileName());
  185. oFileInfos.AddTail(lpRemoteFile);
  186. }
  187. oFtpFileFinder.Close();
  188. return TRUE;
  189. }
  190. catch(CInternetException *ex)
  191. {
  192. ex->Delete();
  193. oFtpFileFinder.Close();
  194. return FALSE;
  195. }
  196. return FALSE;
  197. }
  199. void CFtpWorker::SetMsgWnd(HWND hwnd)
  200. {
  201. this->m_hMsgWnd = hwnd;
  202. }
  204. BOOL CFtpWorker::DownloadFile(LPREMOTEFILEINFO lpRemoteFile, LPCSTR lpszNewFilename)
  205. {
  206. DWORD dwExitCode = 0x0000;
  207. LPDOWNLOADARGS lpParameters = new DOWNLOADARGS;
  209. lpParameters->lpAddition = this;
  210. memcpy(&(lpParameters->rfRemoteFile), lpRemoteFile, sizeof(REMOTEFILEINFO));
  211. _snprintf(lpParameters->szLocalFilename, MAX_PATH, "%s", lpszNewFilename);
  213. // create dummy window as parent for the progress dialog
  214. if (!m_wndDummy.CreateEx(0, AfxRegisterWndClass(0), "trfAgent Download Dummy Window",
  215.  WS_OVERLAPPED, 0, 0, 0, 0, NULL, NULL))
  216. {
  217. delete lpParameters;
  218. return FALSE;
  219. }
  220. // Create the progress dialog box.
  221. if (!m_ProgressDlg.Create(&m_wndDummy, m_hEventKill))
  222. {
  223. delete lpParameters;
  224. return FALSE;
  225. }
  226. WaitForProgressDialog();
  227. this->SetMsgWnd(m_ProgressDlg.GetSafeHwnd());
  229. m_ProgressDlg.m_oFromToHint.Format("%s -> %s", lpRemoteFile->szRemoteFile, lpszNewFilename);
  230. m_ProgressDlg.UpdateData(FALSE);
  232. DWORD dwthreadid = 0x0000;
  233. HANDLE hDownThread = CreateThread(NULL, 0, Routine_Download, lpParameters, 0, &dwthreadid);
  235. DWORD dwWaitRet = WaitForSingleObject(hDownThread, 0);
  237. for(; TRUE;)
  238. {
  239. dwWaitRet = WaitForSingleObject(hDownThread, 0);
  240. if(dwWaitRet == WAIT_OBJECT_0) break;
  241. DoEvents();
  242. Sleep(10);
  243. }
  244. GetExitCodeThread(hDownThread, &dwExitCode);
  245. CloseHandle(hDownThread);
  247. delete lpParameters;
  248. m_ProgressDlg.DestroyWindow();
  249. m_wndDummy.DestroyWindow();
  251. return (dwExitCode == 0);
  252. }
  254. DWORD CFtpWorker::Routine_Download(LPVOID lpParameters)
  255. {
  256. char  lpBuffer[MAX_BUFSIZE];
  257. DWORD dwLeftSize = 0x0000, dwReaded = 0x0000, dwWriten = 0x0000;
  259. HANDLE hLocalFile = INVALID_HANDLE_VALUE;
  260. CInternetFile *pfile = NULL;
  262. //get parameters.
  263. LPDOWNLOADARGS lpDownParams = (LPDOWNLOADARGS)lpParameters;
  264. if(NULL == lpDownParams) return 1;
  266. CFtpWorker *pthis = (CFtpWorker*)lpDownParams->lpAddition;
  267. if(NULL == pthis) return 1;
  269. try
  270. {
  271. //open remote file.
  272. for(; (WaitForSingleObject(pthis->m_hEventKill, 0) == WAIT_TIMEOUT); )
  273. {
  274. dwLeftSize = lpDownParams->rfRemoteFile.dwFilesize;
  276. pfile = pthis->m_pFtpConnection->OpenFile(
  277. lpDownParams->rfRemoteFile.szRemoteFile);
  278. if(NULL == pfile) return 1;
  280. //create local file.
  282. hLocalFile = CreateFile(lpDownParams->szLocalFilename, GENERIC_READ | GENERIC_WRITE, 
  283. FILE_SHARE_READ, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
  284. if(INVALID_HANDLE_VALUE == hLocalFile)
  285. {
  286. //close remote file.
  287. pfile->Close();
  288. return 1;
  289. }
  290. //start read and write actions.
  292. if(pthis->m_hMsgWnd)
  293. {
  294. SendMessage(pthis->m_hMsgWnd, UWM_SETPPROGRESS, (WPARAM)0x0000, dwLeftSize);
  295. }
  296. BOOL bReadFinished = FALSE;
  297. for(; (WaitForSingleObject(pthis->m_hEventKill, 0) == WAIT_TIMEOUT) && dwLeftSize > 0; )
  298. {
  299. //read contents from remote file.
  300. memset(lpBuffer, 0x0, sizeof(lpBuffer));
  301. dwReaded = pfile->Read(lpBuffer, MAX_BUFSIZE);
  303. //write remote file's contents into local file.
  304. WriteFile(hLocalFile, lpBuffer, dwReaded, &dwWriten, NULL);
  305. if(pthis->m_hMsgWnd)
  306. {
  307. PostMessage(pthis->m_hMsgWnd, UWM_SETPPROGRESS, (WPARAM)0x0001, dwReaded);
  308. }
  309. dwLeftSize -= dwReaded;
  311. //if arrived end of file...
  312. if(dwReaded < MAX_BUFSIZE)
  313. {
  314. bReadFinished = TRUE;
  315. break;
  316. }
  317. }
  318. CloseHandle(hLocalFile);
  319. pfile->Close();
  320. if(pthis->m_hMsgWnd)
  321. {
  322. SendMessage(pthis->m_hMsgWnd, UWM_SETPPROGRESS, (WPARAM)0x0002, 0);
  323. }
  324. return bReadFinished ? 0 : 1;
  325. }
  326. }
  327. catch (CInternetException *ex)
  328. {
  329. ex->Delete();
  330. if(NULL != pfile) pfile->Close();
  331. return 1;
  332. }
  333. return 0L;
  334. }
  336. /********************************************************************/
  337. /* */
  338. /* Function name : WaitForProgressDialog */
  339. /* Description   : If the thread has just started, it may not have */
  340. /*    had time yet to initialize the dialog window. */
  341. /* */
  342. /********************************************************************/
  343. void CFtpWorker::WaitForProgressDialog(void)
  344. {
  345. if(m_ProgressDlg.m_hWnd == NULL)
  346. {
  347. while(m_ProgressDlg.m_hWnd == NULL)
  348. {
  349. DoEvents();
  350. }
  351. }
  352. if(!::IsWindow(m_ProgressDlg.m_hWnd))
  353. {
  354. while(!::IsWindow(m_ProgressDlg.m_hWnd))
  355. {
  356. DoEvents();
  357. }
  358. }
  359. }