基于visual c++之windows核心编程代码分析（8）自动下载更新程序

参考链接： C++ wcscat()

我们编写Windows程序，经常需要自动更新，该功能如何实现呢，我们来看代码实现与注释讲解

#include "stdafx.h"

#include "AutoUpdate.h"

#ifdef _DEBUG

#define new DEBUG_NEW

#endif

// 唯一的应用程序对象

CWinApp theApp;

using namespace std;

#define MYVERSION                       5     //定义程序版本。请注意：每次发布，值应该不相同，建议使用自然数递增数列！

DWORD WINAPI UpdateThreadProc(LPVOID lParam)

{

    /*

    *如果有必要，可以使用GetTempPath() 函数获取系统临时文件目录，将用到的文件下载至该目录。本实例中直接使用程序当前目录，

    *并在升级完成时将其删除。

    */

    wchar_t *TempFileName= new wchar_t[255];    //升级文件临时文件名，在该程序中为“AutoUpdate_.exe”

    wmemset(TempFileName,0,255);

    wsprintf(TempFileName,L"%s_.exe",AfxGetApp()->m_pszAppName); //wsprintf对字符串进行格式化处理

    URLDownloadToFile(NULL,_T("http://127.0.0.1/update.txt")/*升级信息配置文件，请根据实际自行修改*/, _T("Update.ini"),0,NULL); 

    FILE *pFile=_wfopen(L"update.ini",L"r");

    if(!pFile)

        return -1;    //如果打开文件失败，则返回。

    char UpdateInfo[10];//变量存储 “update.ini” 中的信息

    memset(UpdateInfo,0,10);

    fgets(UpdateInfo,10,pFile);    //从下载回来的升级配置文件中读取版本号

    fclose(pFile);

    int UpdateVersion=atoi(UpdateInfo);    //atoi 函数返回(const char* _Str)参数转化为 int 的值。

    if(UpdateVersion>MYVERSION)    //判断现有版本是否大于程序自身版本

    {

        //MessageBox(NULL,_T("发现新版"),_T("在线更新"),MB_ICONINFORMATION);

        URLDownloadToFile(NULL,_T("http://127.0.0.1/AutoUpdate.exe")/*升级文件链接,请根据实际自行修改 */,TempFileName,0,NULL);

        PROCESS_INFORMATION PI;

        STARTUPINFO SI;

        memset(&SI, 0, sizeof(SI));

        SI.cb = sizeof(SI);

        if(CreateProcess(TempFileName,L"/install", NULL, NULL, FALSE,NORMAL_PRIORITY_CLASS, NULL, NULL, &SI, &PI))

            //启动升级程序后退出

            ExitProcess(0);

    }

    DeleteFile(_T("update.ini"));    

    return 0;

}

DWORD WINAPI WorkThreadProc(LPVOID lParam)

{

    MessageBox(NULL,L"工作线程已经启动。",L"信息",MB_ICONINFORMATION);

    Sleep(INFINITE);

    return 0;

}

int _tmain(int argc, TCHAR* argv[], TCHAR* envp[])

{

    int nRetCode = 0;

    // 初始化 MFC 并在失败时显示错误

    if (!AfxWinInit(::GetModuleHandle(NULL), NULL, ::GetCommandLine(), 0))

    {

        // TODO: 更改错误代码以符合您的需要

        _tprintf(_T("错误: MFC 初始化失败\n"));

        nRetCode = 1;

    }

    else

    {

        // TODO: 在此处为应用程序的行为编写代码。

        wchar_t StartInfo[255];    //启动参数

        wmemset(StartInfo,0,255);

        wcscat(StartInfo,AfxGetApp()->m_lpCmdLine);

        if(!wcsicmp(StartInfo,_T("/install")))

        {

            //AfxMessageBox(L"命令请求安装");

            wchar_t *AppName        = new wchar_t[255];

            wchar_t *TempFileName        = new wchar_t[255];//升级文件临时文件名，在这里为“AutoUpdate_.exe”

            wchar_t *FinalFileName = new wchar_t[255];    //自身文件名，在该程序中为“AutoUpdate.exe”

            wmemset(AppName,0,255);

            wmemset(TempFileName,0,255);

            wmemset(FinalFileName,0,255);

            wcscat(AppName,AfxGetApp()->m_pszAppName);

            if(AppName[wcslen(AppName)-1]=='_')

            {

                wsprintf(TempFileName,L"%s.exe",AppName); //wsprintf对字符串进行格式化处理

                wcscat(FinalFileName,AppName);

                FinalFileName[wcslen(FinalFileName)-1]='\0';

                wcscat(FinalFileName,_T(".exe"));

                //MessageBox(NULL,TempFileName,L"安装临时文件名（程序自身文件名）",MB_ICONINFORMATION);

                //MessageBox(NULL,FinalFileName,L"目标文件名（最新生成的程序）",MB_ICONINFORMATION);

                CopyFile(TempFileName,FinalFileName,FALSE);

                PROCESS_INFORMATION PI;

                STARTUPINFO SI;

                memset(&SI, 0, sizeof(SI));

                SI.cb = sizeof(SI);

                if(CreateProcess(FinalFileName,L"/finish", NULL, NULL, FALSE,NORMAL_PRIORITY_CLASS, NULL, NULL, &SI, &PI))

                    ExitProcess(0);

            }

            else

            {

                //MessageBox(NULL,L"无效安装文件",L"无效调用",MB_ICONERROR);

            }

        }

        else if(!wcsicmp(StartInfo,_T("/finish")))

        {

            wchar_t *TempFileName    = new wchar_t[255];

            wmemset(TempFileName,0,255);

            wsprintf(TempFileName,L"%s_.exe",AfxGetApp()->m_pszAppName); //wsprintf对字符串进行格式化处理

            //MessageBox(NULL,TempFileName,L"完成更新...删除临时升级文件",MB_ICONINFORMATION);

            DeleteFile(TempFileName);

        }

        /*测试：弹出对话框 显示程序版本*/

        wchar_t tmpMyVer[10];

        wmemset(tmpMyVer,0,10);

        _itow(MYVERSION,tmpMyVer,10/*表示以10进制转换*/);

        MessageBox(NULL,tmpMyVer,_T("程序当前版本："),MB_ICONINFORMATION);

        /*----------------------------*/

        HANDLE hUpdateThread=CreateThread(NULL,0,(LPTHREAD_START_ROUTINE)UpdateThreadProc,NULL,0,0);

        HANDLE hWorkThread=CreateThread(NULL,0,(LPTHREAD_START_ROUTINE)WorkThreadProc,NULL,0,0);

        WaitForSingleObject(hWorkThread,INFINITE);

    }

    return nRetCode;

}