Category

Development Platform
Home > SourceCode/Document > Game Program > Game Server Simulator > View Code
downloadfile.cpp
Package: SwordOnline.rar　[view]
Upload User: dzyhzl
Upload Date: 2019-04-29
Package Size: 56270k
Code Size: 7k
Category:

Game Server Simulator

Development Platform:

C/C++

downloadfile.cpp：Code Content
  1. ////////////////////////////////////////////////////////////////////////////////
  2. //  
  3. //  FileName    :   DownloadFile.cpp
  4. //  Version     :   1.0
  5. //  Creater     :   Linsuyi
  6. //  Date        :   2002-01-31  21:52:16
  7. //  Comment     :   CDownloadFile class source file
  8. //  
  9. ////////////////////////////////////////////////////////////////////////////////
  11. #include "Stdafx.h"
  12. #include "DownloadFile.h"
  14. //////////////////////////////////////////////////////////////////////
  15. // Construction/Destruction
  16. //////////////////////////////////////////////////////////////////////
  18. CDownloadFile::CDownloadFile()
  19. {
  20.     m_nForceDown = false;
  21.     
  22.     m_hNotifyWnd = NULL;
  23.     m_ulNotifyMsg = 0;
  24.     
  25.     m_ulNotifyCode = -1;
  26.     m_nDownProtocol = DOWN_PROTOCOL_NONE;
  28. m_hstartdown = NULL;
  29. m_hdowning = ::CreateEvent(NULL, TRUE, FALSE, NULL);
  30. }
  32. CDownloadFile::~CDownloadFile()
  33. {
  34. ::CloseHandle(m_hdowning);
  35. m_hdowning = NULL;
  36. ASSERT(m_hstartdown == NULL);
  37. }
  39. int GetUrlProtocolType(LPCTSTR lpszUrlPath)
  40. {
  41.     int nResult = DOWN_PROTOCOL_NONE;
  42.     
  43.     int nRetCode = 0;
  44.     CString strUrlPath;
  45.     
  46.     strUrlPath = lpszUrlPath;
  47.     strUrlPath.MakeLower();
  48.     
  49.     strUrlPath.TrimLeft();
  50.     nRetCode = strUrlPath.Find("http://");
  51.     if (0 == nRetCode)
  52.     {
  53.         nResult = DOWN_PROTOCOL_HTTP;
  54.         goto Exit1;
  55.     }
  56.     
  57.     nRetCode = strUrlPath.Find("ftp://");
  58.     if (0 == nRetCode)
  59.     {
  60.         nResult = DOWN_PROTOCOL_FTP;
  61.         goto Exit1;
  62.     }
  63.     
  64.     nRetCode = strUrlPath.Find("file://");
  65.     if (0 == nRetCode)
  66.     {
  67.         nResult = DOWN_PROTOCOL_FILE;
  68.         goto Exit1;
  69.     }
  70.     
  71. Exit1:
  72.     return nResult;
  73. }
  75. int CDownloadFile::SetSavePath(LPCTSTR lpszSavePath)
  76. {
  77.     int nResult = false;
  78.     
  79.     m_strSaveDirPath.Empty();
  80.     
  81.     if (lpszSavePath != NULL)
  82.     {
  83.         m_strSaveDirPath = lpszSavePath;
  84.     }
  85.     nResult = true;
  86.     
  87.     return nResult;
  88. }
  90. void CDownloadFile::SetTimeout(
  91.     ULONG ulSendTimeout,
  92.     ULONG ulRecvTimeout,
  93.     ULONG ulConnTimeout
  94. )
  95. {
  96.     m_FtpDownload.SetTimeout(ulSendTimeout, ulRecvTimeout, ulConnTimeout);
  97.     m_HttpDownload.SetTimeout(ulSendTimeout, ulRecvTimeout, ulConnTimeout);
  98. }
  100. void CDownloadFile::SetProxy(
  101.     int nProxyMode,
  102.     LPCTSTR lpszProxyHost,
  103.     int nProxyPort,
  104.     LPCTSTR lpszProxyUserName,
  105.     LPCTSTR lpszProxyPassword
  106. )
  107. {
  108.     int nProtocolMode = PROXY_MODE_NONE;
  109.     
  110.     switch (nProxyMode)
  111.     {
  112.     case PROXY_MODE_SOCKS4:
  113.         nProtocolMode = FTP_PROXY_SOCKS4;
  114.         break;
  115.     case PROXY_MODE_SOCKS4A:
  116.         nProtocolMode = FTP_PROXY_SOCKS4A;
  117.         break;
  118.     case PROXY_MODE_SOCKS5:
  119.         nProtocolMode = FTP_PROXY_SOCKS5;
  120.         break;
  121.     case PROXY_MODE_USEIE:
  122.         nProtocolMode = FTP_PROXY_USEIE;
  123.         break;
  124.         
  125.     case PROXY_MODE_NONE:
  126.     default:
  127.         nProtocolMode = FTP_PROXY_NONE;
  128.         break;
  129.     }
  130.     m_FtpDownload.SetProxy(
  131.         lpszProxyHost,
  132.         nProxyPort,
  133.         (nProtocolMode != FTP_PROXY_NONE),
  134.         (lpszProxyUserName != NULL),
  135.         lpszProxyUserName,
  136.         lpszProxyPassword,
  137.         nProtocolMode
  138.     );
  139.     
  140.     switch (nProxyMode)
  141.     {
  142.     case PROXY_MODE_HTTPGET:
  143.         nProtocolMode = HTTP_PROXY_HTTPGET;
  144.         break;
  145.     case PROXY_MODE_HTTPCONNECT:
  146.         nProtocolMode = HTTP_PROXY_HTTPCONNECT;
  147.         break;
  148.     case PROXY_MODE_SOCKS4:
  149.         nProtocolMode = HTTP_PROXY_SOCKS4;
  150.         break;
  151.     case PROXY_MODE_SOCKS4A:
  152.         nProtocolMode = HTTP_PROXY_SOCKS4A;
  153.         break;
  154.     case PROXY_MODE_SOCKS5:
  155.         nProtocolMode = HTTP_PROXY_SOCKS5;
  156.         break;
  157.         
  158.     case PROXY_MODE_NONE:
  159.     default:
  160.         nProtocolMode = HTTP_PROXY_NONE;
  161.         break;
  162.     }
  163.     m_HttpDownload.SetProxy(
  164.         lpszProxyHost,
  165.         nProxyPort,
  166.         (nProtocolMode != HTTP_PROXY_NONE),
  167.         (lpszProxyUserName != NULL),
  168.         lpszProxyUserName,
  169.         lpszProxyPassword,
  170.         nProtocolMode
  171.     );
  172. }
  174. int CDownloadFile::StartDownload(LPCTSTR lpszUrlPath, BOOL bEnableResume/*= TRUE*/, LPCTSTR lpszSaveAs/*= NULL*/)
  175. {
  176. if (m_hdowning != NULL)
  177.     {
  178. while (WaitForSingleObject(m_hdowning, 0) == WAIT_OBJECT_0)
  179. Sleep(1);
  180. }
  182.     int nResult = false;
  183.     int nRetCode = false;
  184.     
  185.     m_nForceDown = (bEnableResume == FALSE);
  186.     
  187.     m_strDownUrlPath = lpszUrlPath;
  188.     if (lpszSaveAs != NULL)
  189.     {
  190.         m_strSaveFilePath = lpszSaveAs;
  191.     }
  192.     else
  193.     {
  194.         m_strSaveFilePath = m_strSaveDirPath;
  195.     }    
  196.     
  197.     m_nDownProtocol = GetUrlProtocolType(lpszUrlPath);
  198.     ASSERT(m_nDownProtocol != DOWN_PROTOCOL_NONE);
  199.     
  200. if (IsThreadOK())
  201. {
  202. nRetCode = true;
  203. }
  204. else
  205. {
  206. ASSERT(NULL == m_hstartdown);
  207.     
  208. m_hstartdown = ::CreateEvent(NULL, TRUE, FALSE, NULL);
  209. if (NULL == m_hstartdown)
  210. nRetCode = FALSE;
  211. else
  212. nRetCode = StartThread();
  213. }
  214.     if (!nRetCode)
  215.     {
  216.         ASSERT(false);
  217.         goto Exit0;
  218.     }
  220. SetEvent(m_hstartdown); //green
  221.  
  222.     nResult = true;
  223.     
  224. Exit0:
  225.     return nResult;
  226. }
  228. int CDownloadFile::StopDownload()
  229. {
  230. if (m_hstartdown && WaitForSingleObject(m_hstartdown, 0) == WAIT_OBJECT_0)
  231. {
  232. switch (m_nDownProtocol)
  233. {
  234. case DOWN_PROTOCOL_FTP:
  235. m_FtpDownload.StopDownload();
  236. break;
  237.         
  238. case DOWN_PROTOCOL_HTTP:
  239. m_HttpDownload.StopDownload();
  240. break;
  241.         
  242. default:
  243. ASSERT(false);
  244. }
  246. if (m_hdowning != NULL)
  247. {
  248. while (WaitForSingleObject(m_hdowning, 0) == WAIT_OBJECT_0)
  249. Sleep(1);
  250. }
  251. }
  253. if (m_hstartdown)
  254. CloseHandle(m_hstartdown);
  255. m_hstartdown = NULL;
  257.     int nResult = false;
  258.     
  259. StopThread();
  260.     
  261.     nResult = true;
  262.     
  263.     return nResult;
  264. }
  266. void CDownloadFile::SetNotifyWnd(HWND hNotifyWnd, ULONG ulNotifyMsg)
  267. {
  268.     m_hNotifyWnd = hNotifyWnd;
  269.     m_ulNotifyMsg = ulNotifyMsg;
  270.     
  271.     m_FtpDownload.SetNotifyWnd(m_hNotifyWnd, m_ulNotifyMsg);
  272.     m_HttpDownload.SetNotifyWnd(m_hNotifyWnd, m_ulNotifyMsg);
  273. }
  275. ULONG CDownloadFile::MainExecution()
  276. {
  277. while (1)
  278.     {
  279. if (WaitForSingleObject(m_hStop, 0) == WAIT_OBJECT_0)
  280. break;
  281. Sleep(1);
  282. if (m_hstartdown && WaitForSingleObject(m_hstartdown, 0) == WAIT_OBJECT_0) //开始down
  283. {
  284. ResetEvent(m_hstartdown); //red
  285. DownOneFile();
  286. }
  287.     }
  289. return 0;
  290. }
  292. BOOL CDownloadFile::DownOneFile()
  293. {
  294. SetEvent(m_hdowning);
  295.     BOOL nRetCode = FALSE;
  297. ULONG ulResult = -1;
  298.     m_ulNotifyCode = -1;
  300.     switch (m_nDownProtocol)
  301.     {
  302.     case DOWN_PROTOCOL_HTTP:
  303.         ulResult = (ULONG)m_HttpDownload.Download(
  304.             m_strDownUrlPath,
  305.             m_strSaveFilePath,
  306.             m_nForceDown
  307.         );
  308.         nRetCode = m_HttpDownload.IsUserStop();
  309.         if (!nRetCode)
  310.         {
  311.             m_ulNotifyCode = ulResult;
  312.         }
  313.         break;
  314.         
  315.     case DOWN_PROTOCOL_FTP:
  316.         ulResult = (ULONG)m_FtpDownload.Download(
  317.             m_strDownUrlPath,
  318.             m_strSaveFilePath,
  319.             m_nForceDown
  320.         );
  321.         nRetCode = m_FtpDownload.IsUserStop();
  322.         if (!nRetCode)
  323.         {
  324.             m_ulNotifyCode = ulResult;
  325.         }
  326.         break;
  327.         
  328.     default:
  329.         ASSERT(false);
  330.         break;
  331.     }
  332.     
  333. ResetEvent(m_hdowning);
  335.     if (m_ulNotifyCode != -1)
  336.     {
  337.         ::SendMessage(
  338.             m_hNotifyWnd,
  339.             m_ulNotifyMsg,
  340.             MSG_DOWNLOAD_RESULT,
  341.             m_ulNotifyCode
  342.         );
  343.     }
  345.     return nRetCode;
  346. }