论坛首页 › 棋牌源码 › 网狐在线更新模块+详细教程【源码】

woaidaima2016 发表于 2017-6-13 12:00:06

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

http://img.woaidaima.com//upload/image/20170509/1494311564869034197.jpg#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;
      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;
      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);
}

代码小生 发表于 2020-5-2 21:11:07

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

查看完整版本: 网狐在线更新模块+详细教程【源码】