网狐在线更新模块+详细教程【源码】

woaidaima2016

#include "StdAfx.h"
#include ".\inetfile.h"

CInetFile::CInetFile(void)
: sLastErr(_T(""))
, m_bProgressMode(false)
, m_sURL(_T(""))
, m_sPath(_T(""))
, m_sServerName(_T(""))
, m_sFileName(_T(""))
, m_hWnd(NULL)
, m_bWantStop(false)
, m_dlFileType(GENERAL_FILE)
, m_nDownloadTime(0)
, m_nAlreadyDown(0)
, m_dwTotalLen(0)
{
    m_bRegister = false;        // 是否注册，如果是组件，要注册
    m_bZipFile = false;            // 是否为ZIP文件，如果是，要解压
    m_bForceOverwrite = false;    // 是否覆盖 如果是新版本，无条件覆盖；如果此值为真，无条件覆盖
    m_bExecute = false;            // 是否执行
    m_sExecutePar = _T("");        // 如果执行，执行参数

    sLastErr = _T("连接服务器失败, 请检查网络或联系管理员");
}

CInetFile::~CInetFile(void)
{
    m_bWantStop = true;
    WaitForStop();
}

bool CInetFile::GetInetFile()
{
    m_nDownloadTime ++;
    WaitForStop();
    m_bWantStop = false;
    return StartThread();
}

CString CInetFile::GetLastError(void)
{
    return sLastErr;
}

void CInetFile::ThrowInetFileException(int nCode)
{
    switch(nCode) {
        case 1:
            sLastErr = _T("URL解析错误, 请选择其他服务器进行尝试");
            break;
        case 2:
            sLastErr = _T("服务器拒绝连接, 请选择其他服务器进行尝试");
            break;
        case 3:
            sLastErr = _T("重定向失败, 请选择其他服务器进行尝试");
            break;
        case 4:
            sLastErr = _T("重定向URL解析错误, 请选择其他服务器进行尝试");
            break;
        case 5:
            sLastErr = _T("不是HTTP连接, 请选择其他服务器进行尝试");
            break;
        case 6:
            sLastErr = _T("取得服务器文件失败, 请选择其他服务器进行尝试");
            break;
        case 7:
            sLastErr = _T("创建本地文件失败");
            break;            
        case 8:
            sLastErr = _T("主动取消连接");
            break;   
        case 9:
            sLastErr = _T("URL不正确");
            break;
        default:
            sLastErr = _T("连接服务器失败, 请检查网络或联系管理员");
            break;
    }
   
    CInetFileException* pEx = new CInetFileException(nCode);
    throw pEx;
}

// 线程函数入口，在此实现更新文件
void CInetFile::ThreadProcMain(void)
{
    //ASSERT(!m_sURL.IsEmpty());
    ASSERT(!m_sPath.IsEmpty());

    int nRetCode = 0;

    CInetFileSession session(_T("AutoUpdate - MFC App"), PRE_CONFIG_INTERNET_ACCESS);
    CHttpConnection* pServer = NULL;
    CHttpFile* pFile = NULL;
    try
    {
        CString strServerName;
        CString strObject;
        INTERNET_PORT nPort;
        DWORD dwServiceType;
        const TCHAR szHeaders[] = _T("Accept: */*\r\nUser-Agent: AutoUpdate_Kevin\r\n");
        DWORD dwHttpRequestFlags = INTERNET_FLAG_RELOAD;

        //if (m_sURL.IsEmpty())
        //    ThrowInetFileException(9);

        if (!AfxParseURL(m_sURL, dwServiceType, strServerName, strObject, nPort) ||
            dwServiceType != INTERNET_SERVICE_HTTP)
        {
            ThrowInetFileException(1);
        }

        if (m_bProgressMode)
        {
            VERIFY(session.EnableStatusCallback(true));
        }

        pServer = session.GetHttpConnection(strServerName, nPort);

        pFile = pServer->OpenRequest(CHttpConnection::HTTP_VERB_GET,
            strObject, NULL, 1, NULL, NULL, dwHttpRequestFlags);
        pFile->AddRequestHeaders(szHeaders);
        pFile->SendRequest();

        DWORD dwRet;
        pFile->QueryInfoStatusCode(dwRet);

        if (dwRet != HTTP_STATUS_OK)
        {
            ThrowInetFileException(6);
        }

        CString strNewLocation;
        pFile->QueryInfo(HTTP_QUERY_RAW_HEADERS_CRLF, strNewLocation);

        CString sTotalLen;
        pFile->QueryInfo(HTTP_QUERY_CONTENT_LENGTH , sTotalLen);
        USES_CONVERSION;
        DWORD dwTotalLen = (DWORD)atoi(T2A(sTotalLen));

        CFile fLocal;
        CString sUpdateFileName = m_sPath;
        if (GENERAL_FILE == m_dlFileType) {
            sUpdateFileName += _T(".update");
        }

        // fixed by sacredfy
        if (!fLocal.Open(sUpdateFileName, CFile::modeCreate | CFile::modeWrite)) {
            // 文件无法打开的情况下，循环创建目录，然后重试
            CString strTmp(_T(""));
            int nPos = 0;
            while(true) {
                nPos = sUpdateFileName.Find(_T('\\'), nPos+1);
                if(-1 == nPos) break;

                strTmp = sUpdateFileName.Left(nPos);
                if(!CreateDirectory(strTmp, NULL)) {
                    // 检查错误参数
                    DWORD dwErr = ::GetLastError();
                    if(ERROR_ALREADY_EXISTS == dwErr) continue;
                    else
                        ThrowInetFileException(7);
                }
            }

            // 再次尝试打开该文件
            if (!fLocal.Open(sUpdateFileName, CFile::modeCreate | CFile::modeWrite))
                ThrowInetFileException(7);
        }

        BYTE szBuf[1024];
        UINT nLen = 0;
        m_nAlreadyDown = 0;

        while ( nLen = pFile->Read(szBuf, sizeof(szBuf)) )
        {
            fLocal.Write(szBuf, nLen);
            m_nAlreadyDown += nLen;
            if (m_hWnd)
                PostMessage(m_hWnd, WM_DOWNLOADFILE, MAKEWPARAM(nLen, m_dlFileType), (LPARAM)this);

            if (m_bWantStop) {
                nRetCode = 8;
                break;
            }
        }

        fLocal.Close();
    }
    catch (CInternetException* pEx)
    {
        TCHAR szErr[1024];
        pEx->GetErrorMessage(szErr, 1024);
        nRetCode = 2;
        pEx->Delete();
    }
    catch (CInetFileException* pEx)
    {
        nRetCode = pEx->m_nErrorCode;
        TRACE("Error: Exiting with CInetFileException(%d)\n", nRetCode);
        pEx->Delete();
    }

    if (pFile != NULL) {
        pFile->Close();
        delete pFile;
    }

    if (pServer != NULL) {
        pServer->Close();
        delete pServer;
    }

    session.Close();

    if (m_hWnd)
        PostMessage(m_hWnd, WM_DOWNLOADFILEEND, MAKEWPARAM(nRetCode, m_dlFileType), (LPARAM)this);
}

代码小生

楼主发帖辛苦了，谢谢楼主分享！