VC++完整的MFC自动更新程序实现，包含文件下载和安装逻辑

源码资源网

VC++完整的MFC自动更新程序实现，包含文件下载和安装逻辑
AutoUpdater.cpp

1. 
   
2. #include "stdafx.h"
   
3. #include "AutoUpdater.h"
   
4. #include <wininet.h>
   
5. #include <zip.h>
   
6. #include <shlwapi.h>
   
7. 
   
8. BOOL CAutoUpdater::DownloadUpdate(const CString& strUrl) {
   
9.     HINTERNET hInternet = InternetOpen(_T("Updater"), INTERNET_OPEN_TYPE_DIRECT, NULL, NULL, 0);
   
10.     HINTERNET hFile = InternetOpenUrl(hInternet, strUrl, NULL, 0, 0, 0);
    
11.    
    
12.     // 创建临时目录
    
13.     TCHAR tempPath[MAX_PATH];
    
14.     GetTempPath(MAX_PATH, tempPath);
    
15.     m_strTempPath = tempPath + CString(_T("update_pkg.zip"));
    
16.    
    
17.     // 下载文件
    
18.     HANDLE hLocalFile = CreateFile(m_strTempPath, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
    
19.    
    
20.     DWORD bytesRead;
    
21.     BYTE buffer[4096];
    
22.     while(InternetReadFile(hFile, buffer, sizeof(buffer), &bytesRead) && bytesRead > 0) {
    
23.         DWORD bytesWritten;
    
24.         WriteFile(hLocalFile, buffer, bytesRead, &bytesWritten, NULL);
    
25.     }
    
26.    
    
27.     CloseHandle(hLocalFile);
    
28.     InternetCloseHandle(hFile);
    
29.     InternetCloseHandle(hInternet);
    
30.    
    
31.     return PathFileExists(m_strTempPath);
    
32. }
    
33. 
    
34. BOOL CAutoUpdater::InstallUpdate() {
    
35.     // 解压更新包
    
36.     CString strExtractPath = m_strTempPath.Left(m_strTempPath.ReverseFind('\\')) + _T("\\update\");
    
37.     CreateDirectory(strExtractPath, NULL);
    
38.    
    
39.     // 使用minizip解压
    
40.     unzFile zipfile = unzOpen(CT2A(m_strTempPath));
    
41.     if(!zipfile) return FALSE;
    
42.    
    
43.     unz_global_info global_info;
    
44.     unzGetGlobalInfo(zipfile, &global_info);
    
45.    
    
46.     for(int i=0; i<global_info.number_entry; ++i) {
    
47.         char filename[256];
    
48.         unz_file_info file_info;
    
49.         unzGetCurrentFileInfo(zipfile, &file_info, filename, sizeof(filename), NULL, 0, NULL, 0);
    
50.         
    
51.         if(unzOpenCurrentFile(zipfile) == UNZ_OK) {
    
52.             CString strDestPath = strExtractPath + filename;
    
53.             HANDLE hFile = CreateFile(strDestPath, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
    
54.             
    
55.             BYTE buffer[4096];
    
56.             int bytes;
    
57.             while((bytes = unzReadCurrentFile(zipfile, buffer, sizeof(buffer))) > 0) {
    
58.                 DWORD written;
    
59.                 WriteFile(hFile, buffer, bytes, &written, NULL);
    
60.             }
    
61.             
    
62.             CloseHandle(hFile);
    
63.             unzCloseCurrentFile(zipfile);
    
64.         }
    
65.         unzGoToNextFile(zipfile);
    
66.     }
    
67.     unzClose(zipfile);
    
68.    
    
69.     // 替换旧文件
    
70.     WIN32_FIND_DATA findData;
    
71.     HANDLE hFind = FindFirstFile(strExtractPath + _T("*.*"), &findData);
    
72.     if(hFind != INVALID_HANDLE_VALUE) {
    
73.         do {
    
74.             if(!(findData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)) {
    
75.                 CString strSrc = strExtractPath + findData.cFileName;
    
76.                 CString strDest;
    
77.                 GetModuleFileName(NULL, strDest.GetBuffer(MAX_PATH), MAX_PATH);
    
78.                 strDest.ReleaseBuffer();
    
79.                 strDest = strDest.Left(strDest.ReverseFind('\\') + 1) + findData.cFileName;
    
80.                
    
81.                 MoveFileEx(strSrc, strDest, MOVEFILE_REPLACE_EXISTING);
    
82.             }
    
83.         } while(FindNextFile(hFind, &findData));
    
84.         FindClose(hFind);
    
85.     }
    
86.    
    
87.     return TRUE;
    
88. }
    

复制代码

代码说明：

• 下载功能使用WinINet API实现HTTP文件下载
• 安装过程包含ZIP解压和文件替换
• 需要链接wininet.lib和minizip库
• 实际使用需添加错误处理和进度回调
• 建议添加文件校验和数字签名验证
  

注意事项：

• 需要准备minizip库处理ZIP解压
• 文件替换可能需要管理员权限
• 生产环境应使用HTTPS确保下载安全
• 建议添加回滚机制防止更新失败