1 /***************************************************************************************
  2 *文件名稱: HttpDownload MultiThreadDlg.cpp
  3 *文件說明: 測試自己寫的多線程文件下載
  4 *思路說明: 利用最笨的方法  五個線程 每個線程寫入一個文件 然后最后由主線程來組裝成完整
  5 *          的文件
  6 *作    者: 李佳
  7 *完成時間: 09.8.18
  8 **************************************************************************************/
  9 
 10 #include "stdafx.h"
 11 #include "HttpDownload MultiThread.h"
 12 #include "HttpDownload MultiThreadDlg.h"
 13 #include "HttpSocket.h"
 14 #include "MyStruct.h"
 15 #pragma comment(lib , "Http.lib")
 16 
 17 #define  MAX_THREAD_NUM 3        //最大線程數量
 18 
 19 #ifdef _DEBUG
 20 #define new DEBUG_NEW
 21 #endif
 22 DWORD WINAPI DownloadThreadProc(LPVOID pDate);
 23 
 24 CHttpDownloadMultiThreadDlg::CHttpDownloadMultiThreadDlg(CWnd* pParent /*=NULL*/)
 25     : CDialog(CHttpDownloadMultiThreadDlg::IDD, pParent)
 26     , m_strSpeed(_T(""))
 27     , m_strStorgePath(_T(""))
 28     , iCount(0)
 29     , m_strDownloadAddr(_T("http://www.vckbase.com/code/network/internet/testhttp.rar"))
 30     , iGetNum(0)
 31 {
 32     m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
 33 }
 34 
 35 void CHttpDownloadMultiThreadDlg::DoDataExchange(CDataExchange* pDX)
 36 {
 37     CDialog::DoDataExchange(pDX);
 38     DDX_Text(pDX, IDC_EDIT3, m_strSpeed);
 39     DDX_Text(pDX, IDC_EDIT2, m_strStorgePath);
 40     DDX_Text(pDX, IDC_EDIT1, m_strDownloadAddr);
 41     DDX_Control(pDX, IDC_PROGRESS1, m_DownloadProg);
 42 }
 43 
 44 BEGIN_MESSAGE_MAP(CHttpDownloadMultiThreadDlg, CDialog)
 45     ON_WM_SYSCOMMAND()
 46     ON_WM_PAINT()
 47     ON_WM_QUERYDRAGICON()
 48     ON_MESSAGE(WM_USER_UICHANGE , ChangeUI)
 49     //}}AFX_MSG_MAP
 50     ON_BN_CLICKED(IDOK, &CHttpDownloadMultiThreadDlg::OnBnClickedOk)
 51     ON_BN_CLICKED(IDCANCEL, &CHttpDownloadMultiThreadDlg::OnBnClickedCancel)
 52     ON_BN_CLICKED(IDOK2, &CHttpDownloadMultiThreadDlg::OnBnClickedGetFilePath)
 53 END_MESSAGE_MAP()
 54 
 55 
 56 // CHttpDownloadMultiThreadDlg 消息處理程序
 57 
 58 BOOL CHttpDownloadMultiThreadDlg::OnInitDialog()
 59 {
 60     CDialog::OnInitDialog();
 61 
 62     // 將“關于”菜單項添加到系統菜單中。
 63 
 64     // IDM_ABOUTBOX 必須在系統命令范圍內。
 65     ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
 66     ASSERT(IDM_ABOUTBOX < 0xF000);
 67 
 68     CMenu* pSysMenu = GetSystemMenu(FALSE);
 69     if (pSysMenu != NULL)
 70     {
 71         CString strAboutMenu;
 72         strAboutMenu.LoadString(IDS_ABOUTBOX);
 73         if (!strAboutMenu.IsEmpty())
 74         {
 75             pSysMenu->AppendMenu(MF_SEPARATOR);
 76             pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
 77         }
 78     }
 79 
 80     // 設置此對話框的圖標。當應用程序主窗口不是對話框時，框架將自動
 81     //  執行此操作
 82     SetIcon(m_hIcon, TRUE);            // 設置大圖標
 83     SetIcon(m_hIcon, FALSE);        // 設置小圖標
 84 
 85     // TODO: 在此添加額外的初始化代碼
 86     ::InitializeCriticalSection(&cs);
 87     return TRUE;  // 除非將焦點設置到控件，否則返回 TRUE
 88 }
 89 
 90 void CHttpDownloadMultiThreadDlg::OnSysCommand(UINT nID, LPARAM lParam)
 91 {
 92     CDialog::OnSysCommand(nID, lParam);    
 93 }
 94 
 95 void CHttpDownloadMultiThreadDlg::OnPaint()
 96 {
 97     if (IsIconic())
 98     {
 99         CPaintDC dc(this); // 用于繪制的設備上下文
100 
101         SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0);
102 
103         // 使圖標在工作區矩形中居中
104         int cxIcon = GetSystemMetrics(SM_CXICON);
105         int cyIcon = GetSystemMetrics(SM_CYICON);
106         CRect rect;
107         GetClientRect(&rect);
108         int x = (rect.Width() - cxIcon + 1) / 2;
109         int y = (rect.Height() - cyIcon + 1) / 2;
110 
111         // 繪制圖標
112         dc.DrawIcon(x, y, m_hIcon);
113     }
114     else
115     {
116         CDialog::OnPaint();
117     }
118 }
119 
120 //當用戶拖動最小化窗口時系統調用此函數取得光標
121 //顯示。
122 HCURSOR CHttpDownloadMultiThreadDlg::OnQueryDragIcon()
123 {
124     return static_cast<HCURSOR>(m_hIcon);
125 }
126 
127 //確定按鈕
128 void CHttpDownloadMultiThreadDlg::OnBnClickedOk()
129 {
130     if (m_strDownloadAddr == "" || m_strStorgePath == "")
131     {
132         AfxMessageBox("請輸入下載地址或者保存路徑");
133         return;
134     }    
135     //file.Open(m_strStorgePath , CFile::modeCreate | CFile::modeWrite);    
136     int iFileSize = GetFileSize();    //獲取文件大小
137     //file.SetLength(iFileSize);        //創建一個空文件
138     iCount = iFileSize / (4096);    //分段
139     if(iFileSize%(4096))
140         iCount++;    //剩下的一個
141     
142     //本來是準備使用Windows自帶的線程池  后來由于各種問題  中斷了這個想法
143     HANDLE hThread[MAX_THREAD_NUM];
144     MYSTRUCT MyStruct[MAX_THREAD_NUM];
145     int iThreadWork = iFileSize / (MAX_THREAD_NUM - 1);
146     m_DownloadProg.SetRange(0 , iCount);        //進度條    
147     for (int i = 0 ; i < MAX_THREAD_NUM ; i++)    //創建線程
148     {
149         //PMYSTRUCT MyStruct = (PMYSTRUCT)new char[sizeof(MYSTRUCT)];        
150         MyStruct[i].iThreadNum = i;
151         MyStruct[i].strURL = m_strDownloadAddr;
152         MyStruct[i].iStart = iThreadWork * i ;
153         MyStruct[i].iTo = iThreadWork * (i + 1) - 1;
154         if(MyStruct[i].iTo >= iFileSize)
155             MyStruct[i].iTo = iFileSize;
156         MyStruct->Dlg = this;
157         hThread[i] = ::CreateThread(NULL,0,DownloadThreadProc,(PVOID)&MyStruct[i],NULL,NULL);
158     }    
159     WaitForMultipleObjects(MAX_THREAD_NUM , hThread , TRUE , INFINITE);
160     
161     file.Open(m_strStorgePath , CFile::modeCreate | CFile::modeWrite);
162     CFile fileTemp;
163     for(int i = 0 ; i < MAX_THREAD_NUM ; i++)
164     {
165         CString strTemp;
166         int xLen = MyStruct[i].iTo - MyStruct[i].iStart + 1;
167         strTemp.Format("%d" , i);
168         fileTemp.Open(strTemp , CFile::modeRead);
169         //char * pContent = new char[xLen];
170         CHAR pContent[50000];    //簡單起見 直接用了大數組
171         fileTemp.Read(pContent , xLen );
172         file.Write(pContent , xLen);
173         fileTemp.Close();
174 //        delete pContent;
175     }
176     file.Close();
177 }
178 
179 //退出按鈕
180 void CHttpDownloadMultiThreadDlg::OnBnClickedCancel()
181 {    
182     OnCancel();
183 }
184 
185 //獲取存儲地址
186 void CHttpDownloadMultiThreadDlg::OnBnClickedGetFilePath()
187 {
188     UpdateData(TRUE);
189     if(m_strDownloadAddr == "")
190     {
191         AfxMessageBox("請輸入下載地址" , MB_OK);
192         return ;
193     }
194 
195     CString strFileName = m_strDownloadAddr;
196     m_strStorgePath = GetStorgeFolde();
197 
198     while(strFileName.Find("/") != -1)
199     {
200         int i = strFileName.Find("/");
201         strFileName = strFileName.Mid(i +1 , strFileName.GetLength() - i - 1);
202     }
203 
204     m_strStorgePath += strFileName;
205     UpdateData(FALSE);
206 }
207 
208 //獲取存儲的文件夾
209 CString CHttpDownloadMultiThreadDlg::GetStorgeFolde(void)
210 {
211     CString strFold;
212     BROWSEINFO   bi;  
213     TCHAR   buffer[MAX_PATH];  
214     ZeroMemory(buffer,   MAX_PATH);  
215     bi.hwndOwner   =   GetSafeHwnd();  
216     bi.pidlRoot   =   NULL;  
217     bi.pszDisplayName   =   buffer;  
218 
219     bi.lpszTitle   =   _T("選擇一個文件夾");  
220     bi.ulFlags   =   BIF_EDITBOX;  
221     bi.lpfn   =   NULL;  
222     bi.lParam   =   0;  
223     bi.iImage   =   0;  
224 
225     LPITEMIDLIST   pList   =   NULL;  
226     if   ((pList   =   SHBrowseForFolder(&bi))   !=   NULL)  
227     {  
228         TCHAR   path[MAX_PATH];  
229         ZeroMemory(path,   MAX_PATH);  
230         SHGetPathFromIDList(pList,   path);  
231         strFold = path;
232         UpdateData(FALSE);        
233     }    
234     return strFold;
235 }
236 int CHttpDownloadMultiThreadDlg::GetFileSize(void)
237 {
238     CHttpSocket HttpSocket;
239     CString strServer , strObject ;
240     USHORT uPort;
241     DWORD dwServerType;
242     long lLenth;
243     const char * pRequestHeader = NULL;
244 
245     //通過URL獲取相關參數
246     AfxParseURL(m_strDownloadAddr , dwServerType , strServer , strObject , uPort);
247     pRequestHeader = HttpSocket.FormatRequestHeader((LPTSTR)(LPCTSTR)strServer , (LPTSTR)(LPCTSTR)strObject , lLenth );
248     HttpSocket.Socket();
249     HttpSocket.Connect((LPTSTR)(LPCTSTR)strServer );
250     HttpSocket.SendRequest();
251     HttpSocket.SetTimeout(100);
252 
253     char szLength[15];
254     HttpSocket.GetField("Content-Length" , szLength , 15);
255     int iServerState = HttpSocket.GetServerState();
256     return atoi(szLength);    
257 }
258 
259 //更新進度條
260 LRESULT CHttpDownloadMultiThreadDlg::ChangeUI(WPARAM wParam, LPARAM lParam)
261 {    
262     iGetNum++;
263     m_DownloadProg.SetPos(iGetNum);        
264     return 1;
265 }
266 
267 //下載線程
268 DWORD WINAPI DownloadThreadProc(LPVOID pDate)
269 {
270     MYSTRUCT* Mystruct = (PMYSTRUCT)pDate;
271 
272     CHttpSocket HttpSocket;
273     CString strServer , strObject ;
274     USHORT uPort;
275     DWORD dwServerType;
276     long lLenth;
277     const char * pRequestHeader = NULL;
278 
279     //通過URL獲取相關參數
280     AfxParseURL(Mystruct->strURL , dwServerType , strServer , strObject , uPort);
281     pRequestHeader = HttpSocket.FormatRequestHeader((LPTSTR)(LPCTSTR)strServer , (LPTSTR)(LPCTSTR)strObject , lLenth ,NULL , NULL ,
282                                 Mystruct->iStart ,  Mystruct->iTo );
283                                 /*分段傳輸*/
284     HttpSocket.Socket();
285     HttpSocket.Connect((LPTSTR)(LPCTSTR)strServer );
286     HttpSocket.SendRequest();
287     HttpSocket.SetTimeout(100);
288 
289 #ifdef _DEBUG
290     CString strTemp;
291     strTemp.Format("Start :  %d   to   %d  \r\n" , Mystruct->iStart , Mystruct->iTo);
292     TRACE(strTemp);
293 #endif    
294 
295     CFile file;    
296     CString strFormat;
297     strFormat.Format("%d" , Mystruct->iThreadNum);
298     file.Open(strFormat , CFile ::modeCreate | CFile::modeWrite);
299     int iFileSize = Mystruct->iTo - Mystruct->iStart;
300     int iComplete = 0;
301     char szDate[4096];
302     DWORD dwStartTime , dwEndTime;
303     int iCalc = 0;
304     while(iComplete < iFileSize)
305     {        
306         dwStartTime = GetTickCount();
307         int iRecvNum = HttpSocket.Receive(szDate , 4096);
308         dwEndTime = GetTickCount();
309         if (iRecvNum == 0)
310         {
311             ::AfxMessageBox("服務器關閉鏈接");
312             return 0;
313         }
314         if (iRecvNum == -1)
315         {
316             ::AfxMessageBox("接收數據超時");
317             return 0;
318         }    
319     //    int iSpeed = 4096 / (dwStartTime - dwEndTime);
320         file.Write(szDate , iRecvNum);
321         iComplete += iRecvNum;
322 //        strFormat.Format("%d" , iSpeed);
323         //SendMessage(Mystruct->Dlg->m_hWnd  , WM_USER_UICHANGE , 0 , 0);
324     //    Mystruct->Dlg->m_strSpeed = strFormat;
325         /*if (iCalc >3)
326         {
327             Mystruct->Dlg->UpdateData(FALSE);
328             iCalc = 0;
329         }        */
330     }    
331     return 1;
332 }
333 
334 

 1 // HttpDownload MultiThreadDlg.h : 頭文件
 2 //
 3 
 4 #pragma once
 5 #include "afxcmn.h"
 6 #include "afx.h"
 7 
 8 typedef struct _Mystruct
 9 {
10     CString strURL;        //鏈接
11     int iStart ;                //傳送文件的起點
12     int iTo;                //終點
13     int iThreadNum;        //線程標示符    
14     CHttpDownloadMultiThreadDlg * Dlg;
15 }MYSTRUCT , *PMYSTRUCT;
16 // CHttpDownloadMultiThreadDlg 對話框
17 class CHttpDownloadMultiThreadDlg : public CDialog
18 {
19 // 構造
20 public:
21     CHttpDownloadMultiThreadDlg(CWnd* pParent = NULL);    // 標準構造函數
22 
23 // 對話框數據
24     enum { IDD = IDD_HTTPDOWNLOADMULTITHREAD_DIALOG };
25 
26     protected:
27     virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV 支持
28 // 實現
29 protected:
30     HICON m_hIcon;
31 
32     // 生成的消息映射函數
33     virtual BOOL OnInitDialog();
34     afx_msg void OnSysCommand(UINT nID, LPARAM lParam);
35     afx_msg void OnPaint();
36     afx_msg HCURSOR OnQueryDragIcon();
37     DECLARE_MESSAGE_MAP()
38 public:
39     CString m_strSpeed;
40     CString m_strStorgePath;
41     CString m_strDownloadAddr;
42     afx_msg void OnBnClickedOk();
43     afx_msg void OnBnClickedCancel();
44     afx_msg void OnBnClickedGetFilePath();
45     LRESULT ChangeUI (WPARAM , LPARAM);
46     CProgressCtrl m_DownloadProg;
47     CString GetStorgeFolde(void);
48 private:
49     int GetFileSize(void);
50     int iCount;
51     
52 public:
53     CFile file;
54     CRITICAL_SECTION cs;    //臨界區
55     /*typedef struct _Mystruct
56     {
57         CHttpDownloadMultiThreadDlg * Dlg;
58         int iFileDownloadPos;
59     }MYSTRUCT , *PMYSTRUCT;
60     MYSTRUCT Mystruct;*/
61     int iGetNum;
62 };
63 
64