使用Win32 Internet API的下载文件

文章来源： http://www.vckbase.com/document/viewdoc/?id=319

1. /*
2. 
   
3. INET.H
4. 
   
5. Header containing InternetGetFile and a wrapper function, InternetDownloadFile which also makes the Internet connection.
6. 
   
7. */
8. 
   
9. #ifndef INET_H
10. #define INET_H
11. #ifndef WIN32_LEAN_AND_MEAN
12. #define WIN32_LEAN_AND_MEAN
13. #include <windows.h>
14. #endif
15. #include <stdio.h>
16. #include <wininet.h>
17. #include <mmsystem.h>
18. 
    
19. enum
20. {
21.     INTERNET_ERROR_OPENURL=1,
22.     INTERNET_ERROR_FILEOPEN,
23.     INTERNET_ERROR_READFILE,
24.     INTERNET_ERROR_OPEN
25. };
26. 
    
27. UINT InternetGetFile (HINTERNET IN hOpen, // Handle from InternetOpen()
28.                  CHAR *szUrl,        // Full URL
29.                  CHAR *szFileName,
30.                  HWND hwndProgress,
31.                  int idStatusText,
32.                  int idProgressBar)
33. {
34.        DWORD dwSize;
35.        CHAR   szHead[] = "Accept: */*/r/n/r/n";
36.        VOID* szTemp[16384];
37.        HINTERNET  hConnect;
38.        FILE * pFile;
39. 
    
40.        if ( !(hConnect = InternetOpenUrlA ( hOpen, szUrl, szHead,
41.              lstrlenA (szHead), INTERNET_FLAG_DONT_CACHE | INTERNET_FLAG_PRAGMA_NOCACHE | INTERNET_FLAG_RELOAD, 0)))
42.        {
43.            return INTERNET_ERROR_OPENURL;
44.        }
45. 
    
46.        if  ( !(pFile = fopen (szFileName, "wb" ) ) )
47.       {
48.           return INTERNET_ERROR_FILEOPEN;
49.       }
50. 
    
51.     DWORD dwByteToRead = 0;
52.     DWORD dwSizeOfRq = 4;
53.     DWORD dwBytes = 0;
54. 
    
55.         if (!HttpQueryInfo(hConnect, HTTP_QUERY_CONTENT_LENGTH | HTTP_QUERY_FLAG_NUMBER, 
56.                       (LPVOID)&dwByteToRead, &dwSizeOfRq, NULL))
57.         {
58.         dwByteToRead = 0;
59.     }
60.     DWORD start;
61.     DWORD end;
62.     DWORD time;
63.     time = 10;
64.     start = timeGetTime();
65.        do
66.        {
67.           // Keep coping in 16 KB chunks, while file has any data left.
68.           // Note: bigger buffer will greatly improve performance.
69.           if (!InternetReadFile (hConnect, szTemp, 16384,  &dwSize) )
70.           {
71.               fclose (pFile);
72.             return INTERNET_ERROR_READFILE;
73.           }
74.           if (!dwSize)
75.               break;  // Condition of dwSize=0 indicate EOF. Stop.
76.           else
77.              fwrite(szTemp, sizeof (char), dwSize , pFile);
78.     dwBytes+=dwSize;
79.     if(dwByteToRead && hwndProgress)
80.     {
81.     SendDlgItemMessageA(hwndProgress, idProgressBar, WM_USER+2, (dwBytes*100)/dwByteToRead, 0);
82.     UpdateWindow(hwndProgress);
83.     }
84.     FLOAT fSpeed = 0;
85.     fSpeed = (float)dwBytes;
86.     fSpeed /= ((float)time)/1000.0f;
87.     fSpeed /= 1024.0f;
88.     if(hwndProgress)
89.     {
90.         char s[260];
91.         sprintf(s, "%d KB / %d KB @ %1.1f KB/s", dwBytes/1024, dwByteToRead/1024, fSpeed);
92.         SetDlgItemTextA(hwndProgress, idStatusText, s);
93.         UpdateWindow(hwndProgress);
94.     }
95.     end = timeGetTime();
96.     time = end - start;
97.     if(time == 0)
98.     time = 10;
99.        }   // do
100.       while (TRUE);
101.       fflush (pFile);
102.       fclose (pFile);
103.       return 0;
104. }
105. 
     
106. int InternetDownloadFile(HWND progressWindow, int idStatusText, int idProgressBar, char *URL, char *FileDest)
107. {
108. DWORD dwFlags;
109. DWORD dwResult = INTERNET_ERROR_OPEN;
110. InternetGetConnectedState(&dwFlags, 0);
111. if(dwFlags & INTERNET_CONNECTION_OFFLINE)//take appropriate steps
112.     return INTERNET_ERROR_OPEN;
113. CHAR strAgent[64];
114. sprintf(strAgent, "Agent%ld", timeGetTime());
115. HINTERNET hOpen;
116. if(!(dwFlags & INTERNET_CONNECTION_PROXY)) //怀疑这个地方有问题
117. //如果本机用代理服务器连接到Internet
118. hOpen = InternetOpenA(strAgent, INTERNET_OPEN_TYPE_PRECONFIG_WITH_NO_AUTOPROXY, NULL, NULL, 0);
119. else
120. //如果本机没有用代理服务器连接到Internet
121. hOpen = InternetOpenA(strAgent, INTERNET_OPEN_TYPE_PRECONFIG, NULL, NULL, 0);
122. if(hOpen)
123. {
124.     dwResult = InternetGetFile(hOpen, URL, FileDest, progressWindow, idStatusText, idProgressBar);
125.     InternetCloseHandle(hOpen);
126. }
127. else return INTERNET_ERROR_OPEN;
128. return dwResult;
129. }
130. 
     
131. #endif /*INET_H*/