开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 游戏 > 模拟服务器 > 查看源码
ProcessIndex.cpp
资源名称：SwordOnline.rar　[点击查看]
上传用户：dzyhzl
上传日期：2019-04-29
资源大小：56270k
文件大小：34k
源码类别：

模拟服务器

开发平台：

C/C++

ProcessIndex.cpp：源码内容
  1. //////////////////////////////////////////////////////////////////////////////////////
  2. //
  3. //  FileName    :   ProcessIndex.cpp
  4. //  Version     :   1.0
  5. //  Creater     :   Cheng Bitao
  6. //  Date        :   2002-6-25 10:06:47
  7. //  Comment     :   
  8. //
  9. //////////////////////////////////////////////////////////////////////////////////////
  11. // ProcessIndex.cpp: implementation of the CProcessIndex class.
  12. //
  13. //////////////////////////////////////////////////////////////////////
  14. #include "stdafx.h"
  15. #include "ProcessIndex.h"
  16. #include "DataDefine.h"
  17. #include "PublicFun.h"
  18. #include "GetVersion.h"
  19. #include "GetFilesVersion.h"
  20. #include "Global.h"
  21. #include "KAVStrTranslate.h"
  22. #include "AtlBase.h"
  23. #include "Resource.h"
  24. #include "UpdateData.h"
  25. #include "SourceDef.h"
  27. extern KPATH_TABLE g_PathTable;
  28. extern KUPDATE_DATA g_UpdateData;
  30. //-------------------------------------------------------------------
  31. // comment function
  32. //-------------------------------------------------------------------
  35. int CheckSectionIsUpdateItem(const char cszFileName[], const char cszItemName[])
  36. {
  37.     DWORD dwRetCode     = 0;
  38.     int nRetCode        = 0;
  39.     char szUpdateFileFlag[10];
  43.     dwRetCode = ::GetPrivateProfileString(cszItemName, "UpdateFileFlag", "", szUpdateFileFlag, 10, cszFileName);
  44.     if (!dwRetCode)
  45.         return 0;
  47.     nRetCode = stricmp(szUpdateFileFlag, "FALSE");
  48.     if (!nRetCode)
  49.         return 0;
  50.     nRetCode = stricmp(szUpdateFileFlag, "0");
  51.     if (!nRetCode)
  52.         return 0;
  54.     nRetCode = stricmp(szUpdateFileFlag, "TRUE");
  55.     if (!nRetCode)
  56.         return 1;
  57.     nRetCode = stricmp(szUpdateFileFlag, "1");
  58.     if (!nRetCode)
  59.         return 1;
  61.     return 0;
  62. }
  64. int GetActuallyPath(char szPath[], const char cszBasePath[])
  65. {
  66.     int nRetCode = 0;
  67.     char *pszPos    = NULL;
  68.     char szTemp[MAX_PATH] = {0};
  70.     if (!(*szPath))
  71.     {
  72.         strcpy(szPath, cszBasePath);
  73.         goto Exit0;
  74.     }
  75.     
  76.     nRetCode = stricmp(szPath,"%PROGRAMDIR%");
  77.     if (!nRetCode)
  78.     {
  79.         strcpy(szPath, g_PathTable.szUpdateDestPath);
  80.         goto Exit0;
  81.     }
  82.     
  83.     nRetCode = stricmp(szPath, "%TEMPDIR%");
  84.     if (!nRetCode)
  85.     {
  86.         strcpy(szPath, g_PathTable.szTempPath);
  87.         goto Exit0;
  88.     }
  89.     
  90.     nRetCode = stricmp(szPath, "%SYSTEMDIR%"); 
  91.     if (!nRetCode) 
  92.     {
  93.         strcpy(szPath, g_PathTable.szSystemPath);
  94.         goto Exit0;
  95.     }
  96.     
  97.     nRetCode = stricmp(szPath, "%WINDIR%");
  98.     if (!nRetCode)
  99.     {
  100.         strcpy(szPath, g_PathTable.szWindowsPath);
  101.         goto Exit0;
  102.     }
  103.     
  104.     pszPos = strstr(szPath, "%PROGRAMDIR\");
  105.     if (pszPos)
  106.     {
  107.         strcpy(szTemp, szPath);
  108.         strcpy(szPath, g_PathTable.szUpdateDestPath);
  109.         pszPos = strchr(szTemp, '\');
  110.         pszPos++;
  111.         strcat(szPath, pszPos);
  112.         nRetCode = strlen(szPath);
  113.         szPath[nRetCode - 1] = '';
  114.         goto Exit0;
  115.     }  
  116.     
  117.     if (!(
  118.         ((*szPath = '\') || (*szPath = '/')) ||           //Maybe '/' in Unix
  119.         (
  120. (strlen(szPath) > 2) && (*(szPath + 1) == ':') && 
  121. ((*(szPath + 2) == '\')  || (*(szPath + 2) == '/'))
  122.         )
  123. ))              // if is sub-dir
  124.     {
  125.         char szTemp[MAX_PATH] = {0};
  126.         strcpy(szTemp, szPath);
  127.         strcpy(szPath, cszBasePath);
  128.         strcat(szPath, szTemp);
  129.     } 
  131. Exit0:
  132.     AddPathChar(szPath);
  133.     return true;
  134. }
  136. ////////////////////////////////////////////////////////////////////////////////////////////////////////////
  139. int CProcessIndex::GetItemInfo(
  140.    PKUPDATE_ITEM pUpdateItem, 
  141.    int nUpdateMethod, char szLocalPath[]
  142.    )
  143. {
  144.     int     nResult             = false;
  145.     DWORD   dwRetCode           = 0;
  146.     int     nRetCode            = 0;
  147.     char    szValue[50]         = {0};
  148.     char    szBuffer[1024]      = {0};
  149.     int     nBufferLen          = 1024;
  150.     char    szCommentName[20]   = {0};
  151.     int     nComment            = 0;
  152. int     nLen     = 0;
  153.     char    szDownloadDir[MAX_PATH] = {0};   
  155.     
  156.     KAV_PROCESS_ERROR(pUpdateItem);
  158.     dwRetCode = ::GetPrivateProfileString(pUpdateItem->szItemName, "FileName", "", pUpdateItem->szFileName, MAX_PATH, m_szTxtFileName); 
  159.     KAV_PROCESS_ERROR(dwRetCode);
  160.     
  161.     pUpdateItem->nFileSize = (int)::GetPrivateProfileInt(pUpdateItem->szItemName, "FileSize", 0, m_szTxtFileName);
  162.     KAV_PROCESS_ERROR(pUpdateItem->nFileSize);
  164.     pUpdateItem->KCheckVersionMethod = (KCHECKVERSIONMETHOD)::GetPrivateProfileInt(pUpdateItem->szItemName, "CheckVersionMethod", 0, m_szTxtFileName);
  165.     
  166.     dwRetCode = ::GetPrivateProfileString(pUpdateItem->szItemName, "SNMask", "", pUpdateItem->szSNMask, defSNMaskSize, m_szTxtFileName);
  168.     // --------------- Process Version As File Date------------------
  169.     dwRetCode = ::GetPrivateProfileString(pUpdateItem->szItemName, "Version", "", szValue, 50, m_szTxtFileName);
  170.     KAV_PROCESS_ERROR(dwRetCode);
  172. strcpy(pUpdateItem->szFileDate, szValue);
  173.     dwRetCode = ::GetPrivateProfileString(pUpdateItem->szItemName, "CRC", "", pUpdateItem->szCRC, 20, m_szTxtFileName);
  174.     KAV_PROCESS_ERROR(dwRetCode);
  176.     if ((pUpdateItem->KCheckVersionMethod == enumCVM_TIME) ||
  177. (pUpdateItem->KCheckVersionMethod == enumCVM_DIRECT) ||
  178.         (pUpdateItem->KCheckVersionMethod == enumCVM_CRC)
  179. )
  180.     {
  181.         pUpdateItem->uMajorVersion = 0;
  182.         pUpdateItem->uMinorVersion = 0;        
  183.     }
  184.     else
  185.     {
  186.         nRetCode = GetVersionFromString(szValue, &(pUpdateItem->uMajorVersion), &(pUpdateItem->uMinorVersion));
  188.         KAV_PROCESS_ERROR(nRetCode);
  189.     }
  191.     // --------------- End of Process Version -----------------------
  193.     dwRetCode = ::GetPrivateProfileString(pUpdateItem->szItemName, "Reboot", "", szValue, 50, m_szTxtFileName);
  194.     KAV_PROCESS_ERROR(dwRetCode);
  195.     if (!stricmp(szValue,  "FALSE") || !stricmp(szValue, "0"))
  196.         pUpdateItem->nReboot = 0;
  197.     else 
  198.         pUpdateItem->nReboot = 1;
  200.     // Update Relative flag
  201.     dwRetCode = ::GetPrivateProfileString(pUpdateItem->szItemName, "UpdateRelative", "", szValue, 50, m_szTxtFileName);
  202.     if (!stricmp(szValue,  "TRUE") || !stricmp(szValue, "1"))
  203.         pUpdateItem->nUpdateRelative = 1;
  204.     else 
  205.         pUpdateItem->nUpdateRelative = 0;
  207.     pUpdateItem->nBelong = ::GetPrivateProfileInt(pUpdateItem->szItemName, "Belong", -1, m_szTxtFileName);
  209.     dwRetCode = ::GetPrivateProfileString(pUpdateItem->szItemName, "SharedNeedReboot", "0", szValue, 50, m_szTxtFileName);
  210.     if (!stricmp(szValue,  "TRUE") || !stricmp(szValue, "1"))
  211.         pUpdateItem->nSharedNeedReboot = 1;
  212.     else 
  213.         pUpdateItem->nSharedNeedReboot = 0;
  215.     dwRetCode = ::GetPrivateProfileString(pUpdateItem->szItemName, "RemotePath", "", pUpdateItem->szRemotePath, MAX_PATH, m_szTxtFileName); 
  216.     KAV_PROCESS_ERROR(dwRetCode);
  217.  //   if (strcmp(pUpdateItem->szRemotePath, ".") == 0)
  218.     if (pUpdateItem->szRemotePath[0] == '.' && (pUpdateItem->szRemotePath[1] == 0 || pUpdateItem->szRemotePath[1] == '/'))
  219.       *(pUpdateItem->szRemotePath) = '';
  220.     
  221.     // --------------- Process LocalPath -----------------------
  222.     dwRetCode = ::GetPrivateProfileString(pUpdateItem->szItemName, "LocalPath", "%PROGRAMDIR%", pUpdateItem->szLocalPath, MAX_PATH, m_szTxtFileName);
  223.     KAV_PROCESS_ERROR(dwRetCode);
  224.     nRetCode = GetActuallyPath(pUpdateItem->szLocalPath, g_PathTable.szUpdateDestPath);
  225.     // --------------- End of Process LocalPath -----------------------
  227.     if (nUpdateMethod == defUPDATE_METHOD_INTERNET)
  228.     {
  229.         strcpy(pUpdateItem->szDownloadTmpFileName, g_PathTable.szDownloadDestPath);
  230.         strcat(pUpdateItem->szDownloadTmpFileName, pUpdateItem->szRemotePath);
  231. nLen = strlen(pUpdateItem->szDownloadTmpFileName) - 1;
  232. for (int i = 0; i <= nLen; i++)
  233.         {
  234.             if (pUpdateItem->szDownloadTmpFileName[i] == '/')
  235.                 pUpdateItem->szDownloadTmpFileName[i] = '\';
  236.         }     
  238.         AddPathChar(pUpdateItem->szDownloadTmpFileName);        
  239.     }
  240.     else
  241.     {
  242.         AddPathChar(szLocalPath);
  243.         strcpy(pUpdateItem->szDownloadTmpFileName, szLocalPath); 
  244. strcat(pUpdateItem->szDownloadTmpFileName, pUpdateItem->szRemotePath);
  245. nLen = strlen(pUpdateItem->szDownloadTmpFileName) - 1;
  246.         for (int i = 0; i <= nLen; i++)
  247.         {
  248.             if (pUpdateItem->szDownloadTmpFileName[i] == '/')
  249.                 pUpdateItem->szDownloadTmpFileName[i] = '\';
  250.         }
  251.         AddPathChar(pUpdateItem->szDownloadTmpFileName);
  252.     }
  253.     //MkDirEx(pUpdateItem->szDownloadTmpFileName);
  254.     strcat(pUpdateItem->szDownloadTmpFileName, pUpdateItem->szFileName);
  256.     dwRetCode = ::GetPrivateProfileString(pUpdateItem->szItemName, "Description", "", pUpdateItem->szDescription, 200, m_szTxtFileName);
  257.     dwRetCode = ::GetPrivateProfileString(pUpdateItem->szItemName, "ApplyVersion", "", pUpdateItem->szApplyVersion, defApplyVersionSize, m_szTxtFileName);
  258.     
  259.     if (nUpdateMethod == defUPDATE_METHOD_INTERNET)
  260.     {
  261.         nRetCode = CheckFileCRC(atoi(pUpdateItem->szCRC), pUpdateItem->szDownloadTmpFileName);
  262.         if (nRetCode)
  263.             pUpdateItem->DownloadStatus = enumDOWNLOADSTATUS_DOWNLOADED;
  264.         else
  265.             pUpdateItem->DownloadStatus = enumDOWNLOADSTATUS_QUEUE;
  266.     }
  268.     pUpdateItem->strReadme = "";
  269.     for (nComment = 1; ; nComment++)
  270.     {
  271.         sprintf(szCommentName, "Comment%d", nComment);
  272.         dwRetCode = ::GetPrivateProfileString(pUpdateItem->szItemName, szCommentName, "", szBuffer, nBufferLen + 1, m_szTxtFileName); 
  273.         if (*szBuffer == '')
  274.             break;
  276.         pUpdateItem->strReadme += (CString)szBuffer + "rn";
  277.     }
  279.     if (!pUpdateItem->strReadme.IsEmpty())
  280.     {
  281.         if (
  282.             (IsSignFileName(pUpdateItem->szFileName)) || 
  283.             (IsUpdFileName(pUpdateItem->szFileName))  || 
  284.             (stricmp(pUpdateItem->szFileName, "KAExtend.DAT") == 0)
  285. )
  286.             m_sVirusNames += pUpdateItem->strReadme + "rn";
  287.         else
  288.             m_sReadme += pUpdateItem->strReadme;
  289.     }
  291.     dwRetCode = ::GetPrivateProfileString(pUpdateItem->szItemName, "CallBy", "", pUpdateItem->szCallBy, MAX_PATH, m_szTxtFileName);
  293.     *szValue = '';
  294.     dwRetCode = ::GetPrivateProfileString(pUpdateItem->szItemName, "FileType", "Program", szValue, 50, m_szTxtFileName);
  295.     pUpdateItem->nItemType = -1;
  296.     dwRetCode = ::GetPrivateProfileString(pUpdateItem->szItemName, "NeedDownload", "1", szValue, 50, m_szTxtFileName);
  297.     if (!stricmp(szValue,  "TRUE") || !stricmp(szValue, "1"))
  298.         pUpdateItem->bNeedDownload = 1;
  299.     else 
  300.         pUpdateItem->bNeedDownload = 0; 
  302. dwRetCode = ::GetPrivateProfileString(pUpdateItem->szItemName, "UpdateFileMethod", "Copy", pUpdateItem->szUpdateFileMethod, MAX_PATH, m_szTxtFileName);
  304.     pUpdateItem->bNeedUpdate = false;
  305.     nResult = true;
  306. Exit0:
  307.     return nResult;
  308. }
  311. // Return -1 if Version < szDate, 
  312. // Return 0  if Version == szDate,
  313. // Return 1  if Version > szDate
  314. int _CheckVersionByRes(const char cszFileName[], char szDate[])
  315. {
  316.     IMAGE_DOS_HEADER DOSHeader;
  317.     IMAGE_VXD_HEADER VXDHeader;
  318.     CFile File;
  319.     int nResult = 0;
  320.     
  321.     if (File.Open(cszFileName, CFile::modeRead | CFile::shareDenyNone, NULL))
  322.     {
  323.         UINT uReadLen = 0;
  324.         uReadLen = File.Read(&DOSHeader, sizeof(DOSHeader));
  325.         if ((uReadLen == sizeof(DOSHeader)) && (IMAGE_DOS_SIGNATURE == DOSHeader.e_magic))
  326.         {
  327.             File.Seek(DOSHeader.e_lfanew, CFile::begin);
  328.             uReadLen = File.Read(&VXDHeader, sizeof(VXDHeader));
  329.             if ((uReadLen == sizeof(VXDHeader)) && (IMAGE_OS2_SIGNATURE_LE == VXDHeader.e32_magic))
  330.             {
  331.                 char szBuf[1024];
  332.                 File.Seek(VXDHeader.e32_winresoff, CFile::begin);
  333.                 uReadLen = File.Read(szBuf, VXDHeader.e32_winreslen);
  334.                 if (uReadLen == VXDHeader.e32_winreslen)
  335.                 {
  336.                     unsigned i = 0;
  337.                     for (; i < uReadLen; i++)
  338.                     {
  339.                         if (stricmp(szBuf + i, "FILEVERSION"))
  340.                             continue;
  342.                         char *pszVersion = szBuf + i + 11;   // 2001, 2, 23, 27
  343.                         while (*pszVersion == 0)
  344.                             pszVersion++;
  345.                         
  346.                         i = 0;
  347.                         int j = 0;
  348.                         
  349.                         while (*pszVersion)        // cszDate = "2001.2.23,27"
  350.                         {
  351.                             if (szDate[j] == '')
  352.                                 break;
  353.                             
  354.                             i = _StrToInt(pszVersion);
  355.                             
  356.                             while (*pszVersion != ',')
  357.                             {
  358.                                 if (*pszVersion == 0)
  359.                                     break;
  360.                                 pszVersion++;
  361.                             }
  362.                             
  363.                             while ((*pszVersion == 0x20) || (*pszVersion == ','))
  364.                                 pszVersion++;
  365.                             
  366.                             if (i < (unsigned)_StrToInt(szDate + j))
  367.                             {
  368.                                 nResult = -1;
  369.                                 break;
  370.                             }
  371.                             else if (i > (unsigned)_StrToInt(szDate + j))
  372.                             {
  373.                                 nResult = 1;
  374.                                 break;
  375.                             }
  376.                             
  377.                             
  378.                             while ((szDate[j] != '.') && (szDate[j] != ''))
  379.                                 j++;
  380.                             if (szDate[j] == '.')
  381.                                 j++;
  382.                             else
  383.                                 break;
  384.                         }
  385.                         
  386.                         if ((*pszVersion == '') && (szDate[j] == ''))
  387.                             nResult = 0;
  388.                         
  389.                         break;
  390.                     }
  391.                 }
  392.             }
  393.         }
  394.         
  395.         File.Close();
  396.     }
  397.     
  398.     return nResult;
  399. }   
  402. //////////////////////////////////////////////////////////////////////
  403. // Construction/Destruction
  404. //////////////////////////////////////////////////////////////////////
  406. CProcessIndex::CProcessIndex()
  407. {
  408.     m_szFileName[0]           = '';
  409.     m_szTxtFileName[0]        = '';
  410.     m_bInitFlag               = false; 
  411.     
  412. m_bVersionError           = true;
  413.     m_nVersion_ValueCount     = 0;  
  414. m_nMajorVersionValueCount = 0;
  415. m_sWebDownAddress         = "";
  416.     
  417. m_nMaxDATFileNum          = 0;
  418.     m_nMaxUPDFileNum          = 0;
  419.     m_sVirusNames             = "";
  420.     m_sReadme                 = "";
  421.     m_sAnnounce               = "";
  422.     m_bNeedUpdateSelfFirst    = false;
  424.     m_pUpdateItemList         = NULL;
  426.     m_pModuleExitInfoList     = NULL;    
  427.     
  428.     m_bFileError              = false;
  430. m_szCallBy[0]   = '';
  431. }
  433. CProcessIndex::~CProcessIndex()
  434. {
  435.     Uninit();
  436.     
  437.     m_szFileName[0]         = '';
  438.     m_szTxtFileName[0]      = '';
  439.     m_bInitFlag             = false; 
  440.     m_bVersionError         = true;
  441.     m_nVersion_ValueCount   = 0;     
  442.     m_nMaxDATFileNum        = 0;
  443.     m_nMaxUPDFileNum        = 0;
  444.     m_sVirusNames           = "";
  445.     m_sReadme               = "";
  446.     m_sAnnounce             = "";
  447.     m_bNeedUpdateSelfFirst  = false;
  448.     
  449.     m_pUpdateItemList       = NULL;    
  450. }
  452. int CProcessIndex::Init(const char cszFileName[], const char cszTxtFileName[])
  453. {
  454.     int nRetCode    = 0;
  455.     int nResult     = false;
  456.     
  457.     strcpy(m_szFileName, cszFileName);
  458.     strcpy(m_szTxtFileName, cszTxtFileName);
  459.     
  460.     FreeUpdateItemList();
  461.     FreeModuleExitInfoList();
  462.     
  463.     nRetCode = FileExists(m_szFileName);
  464.     KAV_PROCESS_ERROR(nRetCode);
  465.     
  466.     // UnCompress index.dat to index.txt
  467.     nRetCode = UnCompressFile(m_szFileName, m_szTxtFileName);
  468.     KAV_PROCESS_ERROR(nRetCode);
  469.     
  470.     m_bInitFlag = true;
  472.     nResult = true;
  473. Exit0:
  474.     return nResult;    
  475. }
  477. int CProcessIndex::Uninit()
  478. {
  479.     ::DeleteFile(m_szTxtFileName);
  480.     FreeUpdateItemList();
  481.     FreeModuleExitInfoList();
  483.     return true;
  484. }
  486. int CProcessIndex::FreeUpdateItemList()
  487. {
  488.     PKUPDATE_ITEM pUpdateItem = NULL;
  490.     while (m_pUpdateItemList)
  491.     {
  492.         pUpdateItem = m_pUpdateItemList->pNext;      
  493.         KAV_DELETE(m_pUpdateItemList);         
  494.         m_pUpdateItemList = pUpdateItem;
  495.     }
  496.     return true;
  497. }
  499. int CProcessIndex::FreeModuleExitInfoList()
  500. {
  501.     PKModuleExitInfoList pModuleExitInfoList = NULL;
  503.     while (m_pModuleExitInfoList)
  504.     {
  505.         pModuleExitInfoList = m_pModuleExitInfoList->pNext;      
  506.         KAV_DELETE(m_pModuleExitInfoList);         
  507.         m_pModuleExitInfoList = pModuleExitInfoList;
  508.     }
  509.     return true;
  510. }
  512. int CProcessIndex::InitModuleExitInfo()
  513. {
  514.     int nResult = false;
  515.     int i;
  516.     char szKeyName[MAX_PATH] = {0};
  517.     char szFileName[MAX_PATH] = {0};
  518. char szReleaseFileName[MAX_PATH] = {0};
  519.     PKModuleExitInfoList pModuleExitInfoList = NULL;
  521.     KAV_PROCESS_ERROR(m_bInitFlag);
  523.     for (i = 0; ; i++)
  524.     {
  525.         sprintf(szKeyName, "FileName%d", i);
  526.         ::GetPrivateProfileString("ExitType", szKeyName, "", szFileName, MAX_PATH, m_szTxtFileName);
  527.         if (szFileName[0] == '')
  528.             break;
  530.         pModuleExitInfoList = new KModuleExitInfoList;
  531.         if (!pModuleExitInfoList)
  532.             goto Exit0;
  534.         pModuleExitInfoList->pNext = NULL;
  535.         strcpy(pModuleExitInfoList->MouleExitInfo.szFileName, szFileName);
  537. sprintf(szKeyName, "ReleaseFileName%d", i);
  538.         ::GetPrivateProfileString("ExitType", szKeyName, "", szReleaseFileName, MAX_PATH, m_szTxtFileName);
  539.         if (szReleaseFileName[0] == '')
  540.             strcpy(szReleaseFileName, szFileName);
  542. strcpy(pModuleExitInfoList->MouleExitInfo.szReleaseFileName, szReleaseFileName);
  544.         sprintf(szKeyName, "ExitType%d", i);
  545.         pModuleExitInfoList->MouleExitInfo.nExitType = ::GetPrivateProfileInt("ExitType", szKeyName, 0, m_szTxtFileName);
  547. pModuleExitInfoList->MouleExitInfo.uID = 0;
  548. pModuleExitInfoList->MouleExitInfo.szStartParam[0] = '';
  549. pModuleExitInfoList->MouleExitInfo.szStopParam[0] = '';
  550.         
  551.         switch (pModuleExitInfoList->MouleExitInfo.nExitType)
  552.         {
  553.         case defEXIT_TYPE_DIRECTORY:
  554.         case defEXIT_TYPE_IPC:
  555. case defEXIT_TYPE_RPC:
  556.             sprintf(szKeyName, "StopParam%d", i);
  557.             ::GetPrivateProfileString("ExitType", szKeyName, "", pModuleExitInfoList->MouleExitInfo.szStopParam, MAX_PATH, m_szTxtFileName);    
  558.             sprintf(szKeyName, "StartParam%d", i);
  559.             ::GetPrivateProfileString("ExitType", szKeyName, "", pModuleExitInfoList->MouleExitInfo.szStartParam, MAX_PATH, m_szTxtFileName);    
  561.             if ((pModuleExitInfoList->MouleExitInfo.nExitType == defEXIT_TYPE_IPC) ||
  562. (pModuleExitInfoList->MouleExitInfo.nExitType == defEXIT_TYPE_RPC)
  563. )
  564.             {
  565.                 sprintf(szKeyName, "UID%d", i);
  566.                 pModuleExitInfoList->MouleExitInfo.uID = ::GetPrivateProfileInt("ExitType", szKeyName, 0, m_szTxtFileName);    
  567.             }
  569.             break;
  571.         case defEXIT_TYPE_SERVICE:
  572.             break;
  573.         default:
  574.             break;
  575.         }
  577.         if (!m_pModuleExitInfoList)
  578.         {
  579.             m_pModuleExitInfoList = pModuleExitInfoList;
  580.         }
  581.         else
  582.         {
  583.             pModuleExitInfoList->pNext = m_pModuleExitInfoList;
  584.             m_pModuleExitInfoList = pModuleExitInfoList;
  585.         }
  587.         pModuleExitInfoList = NULL;
  588.     }
  589.     // For old index 
  590.     if (0 == i)
  591.     {
  592.         pModuleExitInfoList = new KModuleExitInfoList;
  593.         if (!pModuleExitInfoList)
  594.             goto Exit0;
  596.         strcpy(pModuleExitInfoList->MouleExitInfo.szFileName, "KAV9X.EXE");
  597.         pModuleExitInfoList->MouleExitInfo.nExitType = 1;
  598.         pModuleExitInfoList->MouleExitInfo.uID = 1;
  599.         strcpy(pModuleExitInfoList->MouleExitInfo.szStopParam, "/quit");
  600. pModuleExitInfoList->MouleExitInfo.szStartParam[0] = '';
  601.         pModuleExitInfoList->pNext = NULL;
  603.         m_pModuleExitInfoList = pModuleExitInfoList;
  605.         pModuleExitInfoList = new KModuleExitInfoList;
  606.         if (!pModuleExitInfoList)
  607.             goto Exit0;
  609.         strcpy(pModuleExitInfoList->MouleExitInfo.szFileName, "Watcher.EXE");
  610.         pModuleExitInfoList->MouleExitInfo.nExitType = 1;
  611.         pModuleExitInfoList->MouleExitInfo.uID = 2;
  612.         strcpy(pModuleExitInfoList->MouleExitInfo.szStopParam, "/quit");
  613. pModuleExitInfoList->MouleExitInfo.szStartParam[0] = '';
  614. pModuleExitInfoList->pNext = NULL;  
  615.         
  616.         m_pModuleExitInfoList->pNext = pModuleExitInfoList;
  617.     }
  619.     nResult = true;
  620. Exit0:
  621.     return nResult;
  622. }
  625. int CProcessIndex::AddItemInfoToList(
  626.  const char cszSectionName[], int nUpdateMethod,
  627.  char szLocalPath[],
  628.  int nMainVersion
  629.  )
  630. {
  631.     int nResult                 = false;
  632.     int nRetCode                = 0;
  633.     PKUPDATE_ITEM pUpdateItem   = NULL;
  634.     char szTemp[MAX_PATH]       = {0};
  635.     char *pszFileNamePos        = NULL;
  636.     CString sMsg;
  637.     CString sFormat;
  638.     
  640.     nRetCode = CheckSectionIsUpdateItem(m_szTxtFileName, cszSectionName);
  641.     if (nRetCode)
  642.     {
  645.         pUpdateItem = new KUPDATE_ITEM;
  646.         KAV_PROCESS_ERROR(pUpdateItem);
  647.         
  648.         pUpdateItem->pNext = NULL;
  649.         pUpdateItem->nDownTryTimes = 0;
  650.         
  651.         strcpy(pUpdateItem->szItemName, cszSectionName);
  652.         
  653.         nRetCode = GetItemInfo(pUpdateItem, nUpdateMethod, szLocalPath);
  654.         KAV_PROCESS_ERROR(nRetCode);
  656.         nRetCode = CheckIsNeedUpdate(pUpdateItem, nMainVersion);
  657.         if (nRetCode)
  658.         {
  659.             if (nUpdateMethod == defUPDATE_METHOD_INTERNET)
  660.             {
  661.                 strcpy(szTemp, pUpdateItem->szDownloadTmpFileName);
  662.                 pszFileNamePos = strrchr(szTemp, '\');
  663.                 if (pszFileNamePos)
  664.                     *pszFileNamePos = '';
  666.                 MkDirEx(szTemp);
  667.             }
  668.             else
  669.             {
  670.                 nRetCode = CheckFileCRC(atoi(pUpdateItem->szCRC), pUpdateItem->szDownloadTmpFileName);
  671.                 if (nRetCode)
  672.                     pUpdateItem->DownloadStatus = enumDOWNLOADSTATUS_DOWNLOADED;
  673.                 else 
  674.                 {
  675.                     pUpdateItem->DownloadStatus = enumDOWNLOADSTATUS_ERROR;
  676.                     if (!m_bFileError)
  677.                         m_bFileError = pUpdateItem->bNeedDownload;
  679.                     sFormat = defIDS_FILE_ERROR_FORMAT;
  680.                     sMsg.Format(sFormat, pUpdateItem->szDownloadTmpFileName);
  681.                     g_UpdateData.SaveLog.WriteLogString(sMsg, true);  
  682.                 }
  683.             }
  685.             if (!m_pUpdateItemList)
  686.             {
  687.                 m_pUpdateItemList = pUpdateItem;
  688.             }
  689.             else
  690.             {
  691.                 pUpdateItem->pNext = m_pUpdateItemList;
  692.                 m_pUpdateItemList = pUpdateItem;                
  693.             }
  695. StrAddStr(m_szCallBy, pUpdateItem->szCallBy);
  697.             // Set bNeedUpdateSelfFirst flag
  698.             if ((pUpdateItem->nUpdateRelative) && pUpdateItem->bNeedUpdate)
  699.                 m_bNeedUpdateSelfFirst = true;
  701.             // Get max KAVXXXXX.Dat file Number
  702.             nRetCode = GetSignFileNumber(pUpdateItem->szFileName);
  703.             if (IsSignFileName(pUpdateItem->szFileName))
  704.             {
  705.                 if (nRetCode > m_nMaxDATFileNum)
  706.                     m_nMaxDATFileNum = nRetCode;
  707.             }
  708.             else 
  709.             {
  710.                 if (nRetCode > m_nMaxUPDFileNum)
  711.                     m_nMaxUPDFileNum = nRetCode;
  712.             }
  713.         }
  714.         else
  715.         {
  716.             KAV_DELETE(pUpdateItem);
  717.         }        
  719. }
  720.     nResult = true;
  721. Exit0:
  722.     if (!nResult)
  723. {
  724.         KAV_DELETE(pUpdateItem);
  725. }
  726.     return nResult;
  728.  
  729. }
  730.  
  731. int CProcessIndex::AnalysisUpdateFiles
  732. (int nMethod, 
  733.   char szLocalPath[],
  734.   int nMainVersion
  735.  )
  736.  
  737.  
  738. {
  739. int nResult                     = false;
  740. int nRetCode                    = 0;
  741. char szSectionNames[10 * 1024]  = {0};
  742. int  nSectionNamesSize          = 1024 * 10;
  743. char *pszSectionName            = NULL;
  744. int nSectionNameLen             = 0;
  745. DWORD dwRetCode                 = 0;
  746. char szTemp[20]                 = {0};
  747. PKUPDATE_ITEM pUpdateItem       = NULL;
  748. PKUPDATE_ITEM pPreUpdateItem    = NULL;
  749. int nBufferLen             = 1024;
  750. char szBuffer[1024]             = {0};
  751. int nFlag                       = 0;
  752. int nItemNumber                 = 0;
  753. int i                           = 0;
  755. KAV_PROCESS_ERROR(m_bInitFlag);
  757. // Free item list at first
  758. FreeUpdateItemList();
  759. FreeModuleExitInfoList();
  761. m_nMaxDATFileNum = 0;
  762. m_nMaxUPDFileNum = 0;
  764. // Get Comment information from index.txt
  765. m_sAnnounce = "";
  767. // Get message for user
  768. ::GetPrivateProfileString(
  769. "COMMENT", 
  770. "Readme", 
  771. "", 
  772. szBuffer, 
  773. nBufferLen, 
  774. m_szTxtFileName
  775. );
  777. if (*szBuffer != '')
  778. {
  779. m_sAnnounce = szBuffer;
  780. m_sAnnounce.Replace("#", "rn");
  781. }
  783. // Initialize Module exit info
  784. nRetCode = InitModuleExitInfo();
  785. KAV_PROCESS_ERROR(nRetCode);
  787. // Initialize Version string and version value
  788. nRetCode = InitVersionInfo();
  789. KAV_PROCESS_ERROR(nRetCode);
  791. if(nRetCode == defVersionNotenough)
  792. {
  793. nResult = defVersionNotenough;
  794. ::GetPrivateProfileString(
  795. "Version", 
  796. "WebDown", 
  797. "http://www.jxonline.net", 
  798. url, 
  799. MAX_PATH, 
  800. m_szTxtFileName
  801. );
  802. goto Exit0;
  803. }
  804. else if(nRetCode == defVersionMore) {
  805. nResult = defVersionMore;
  806. goto Exit0;
  807. }
  808. // Get Update file information from index.txt
  809. dwRetCode = ::GetPrivateProfileSectionNames(szSectionNames, nSectionNamesSize, m_szTxtFileName);
  810. KAV_PROCESS_ERROR(dwRetCode);
  812. pszSectionName = szSectionNames;
  813. while (*pszSectionName)
  814. {
  815. // Process one update file item, add to update file list
  816. nRetCode = AddItemInfoToList(pszSectionName, nMethod, szLocalPath, nMainVersion);
  818. // Get next Item name
  819. nSectionNameLen = strlen(pszSectionName);
  820. pszSectionName += nSectionNameLen + 1;
  821. }
  823. // Update Update.EXE first
  824. // Delete the not relative file with update.exe
  825. if (m_bNeedUpdateSelfFirst)
  826. {                             
  827. pUpdateItem     = m_pUpdateItemList;                    
  828. pPreUpdateItem  = pUpdateItem;
  829. while (pUpdateItem)
  830. {
  831. if (!pUpdateItem->nUpdateRelative)
  832. {
  833. if (pUpdateItem == pPreUpdateItem)
  834. {
  835. m_pUpdateItemList = m_pUpdateItemList->pNext;
  836. pPreUpdateItem = m_pUpdateItemList;
  837. pUpdateItem->pNext = NULL;
  838. delete pUpdateItem;
  839. pUpdateItem = m_pUpdateItemList;
  840. }
  841. else 
  842. {
  843. pPreUpdateItem->pNext = pUpdateItem->pNext;                    
  844. pUpdateItem->pNext = NULL;
  845. delete pUpdateItem;
  846. pUpdateItem = pPreUpdateItem->pNext;
  847. }
  849. continue;
  850. }        
  852. if (pUpdateItem != m_pUpdateItemList)
  853. pPreUpdateItem = pPreUpdateItem->pNext;
  854. pUpdateItem = pUpdateItem->pNext;            
  855. }
  857. nResult = true;
  858. goto Exit0;
  859. }
  861. nResult = true;
  862. Exit0:
  863. if (!nResult)
  864. {
  865. FreeUpdateItemList();    
  866. FreeModuleExitInfoList();
  867. }
  870. return nResult;
  872. }
  874. int CProcessIndex::GetExitType(PKModuleExitInfo pModuleExitInfo)
  875. {
  876.     int nResult = false;
  877.     PKModuleExitInfoList pModuleExitInfoList = NULL;
  880.     KAV_PROCESS_ERROR(m_bInitFlag);
  881.     KAV_PROCESS_ERROR(pModuleExitInfo);
  883. pModuleExitInfo->nExitType = -1;
  884. pModuleExitInfo->szReleaseFileName[0] = '';
  885. pModuleExitInfo->szStartParam[0] = '';
  886. pModuleExitInfo->szStopParam[0] = '';
  887. pModuleExitInfo->uID = 0;
  889.     pModuleExitInfoList = m_pModuleExitInfoList;
  890.     while(pModuleExitInfoList)
  891.     {
  892.         if (stricmp(pModuleExitInfo->szFileName, pModuleExitInfoList->MouleExitInfo.szFileName) == 0)
  893.         {
  894.             sprintf(pModuleExitInfo->szReleaseFileName, "%s%s", g_PathTable.szUpdateDestPath, pModuleExitInfoList->MouleExitInfo.szReleaseFileName);
  895.             pModuleExitInfo->nExitType = pModuleExitInfoList->MouleExitInfo.nExitType;
  896.             pModuleExitInfo->uID = pModuleExitInfoList->MouleExitInfo.uID;
  897.             strcpy(pModuleExitInfo->szStartParam, pModuleExitInfoList->MouleExitInfo.szStartParam);
  898.             strcpy(pModuleExitInfo->szStopParam, pModuleExitInfoList->MouleExitInfo.szStopParam);
  899.             break;
  900.         }
  901.         pModuleExitInfoList = pModuleExitInfoList->pNext;
  902.     }
  904.     if (!pModuleExitInfoList)
  905.         goto Exit0;
  907.     nResult = true;
  908. Exit0:
  909.     return nResult;
  910. }
  912. bool CProcessIndex::IsVersionError()
  913. {
  914.     return m_bVersionError;
  915. }
  917. bool CProcessIndex::IsNeedUpdateSelf()
  918. {
  919.     return m_bNeedUpdateSelfFirst;
  920. }
  922. bool CProcessIndex::IsNotUpdateItem()
  923. {
  924.     return (m_pUpdateItemList == NULL);
  925. }
  927. int CProcessIndex::InitVersionInfo()
  928. {
  929.     if (!m_bInitFlag)
  930. {
  931.         return false;
  932. }
  933. m_nVersion_ValueCount = ::GetPrivateProfileInt
  934. (
  935. "Version", 
  936. "Version", 
  937. 0,  
  938. m_szTxtFileName
  939.         );
  941.     m_nMajorVersionValueCount = ::GetPrivateProfileInt
  942. (
  943. "Version", 
  944. "MajorVersion", 
  945. 0,  
  946. m_szTxtFileName
  947.         );
  950. CString strVersion = g_PathTable.szModulePath;
  951.     int nVersionNumber = ::GetPrivateProfileInt
  952. (
  953. "Version",
  954. "MajorVersion",
  955. 0,
  956. strVersion + "Version.cfg"
  957. );
  958.     
  961. char szWebAddressBuffer[MAX_PATH];
  962. ::GetPrivateProfileString("Version", "WebDown", "http://www.jxonline.net", szWebAddressBuffer, MAX_PATH, m_szTxtFileName);
  963.     m_sWebDownAddress = szWebAddressBuffer;
  966. if(nVersionNumber < m_nMajorVersionValueCount)
  967. {
  968. return defVersionNotenough;
  969. }
  971. if(nVersionNumber > m_nMajorVersionValueCount)
  972. {
  973. return defVersionMore;
  974. }
  976.     return true;
  977. }
  979. int CProcessIndex::CheckIsNeedUpdate(
  980.  PKUPDATE_ITEM pUpdateItem,
  981.  int nMainVersion
  982.  )
  983. {
  984.     int nSignMaxNum     = 0;  
  985.     int nRetCode            = 0;
  986.     int nCurrentVersion     = 0;
  987.     char *pszPos            = NULL;
  988.     int nCurrentNumber      = 0;
  989.     char szLocalFullFileName[MAX_PATH] = {0};
  990.     DWORD dwMajorVersion    = 0;
  991.     DWORD dwMinorVersion    = 0;
  992.     int nSignFileFlag       = 0;
  993.     
  994.     KAV_PROCESS_ERROR(pUpdateItem);
  996.     pUpdateItem->bNeedUpdate = false;
  998. // Belone Version error
  999.     if (pUpdateItem->nBelong >= 0 && pUpdateItem->nBelong != nMainVersion)
  1000. goto Exit0;
  1002.     m_bVersionError = FALSE;
  1004.     // Data file, such as readme.txt or KAVDX.txt
  1005.     if (pUpdateItem->KCheckVersionMethod == enumCVM_DIRECT)
  1006.     {
  1007.         pUpdateItem->bNeedUpdate = true;
  1008.         goto Exit0;
  1009.     }
  1011.     strcpy(szLocalFullFileName, pUpdateItem->szLocalPath);
  1012.     strcat(szLocalFullFileName, pUpdateItem->szFileName);
  1013. nRetCode = FileExists(szLocalFullFileName);
  1014.     if (!nRetCode)
  1015.     {
  1016.         pUpdateItem->bNeedUpdate = true;
  1017.         goto Exit0;
  1018.     }
  1020.     if (pUpdateItem->KCheckVersionMethod == enumCVM_CRC)
  1021.     {
  1022.         pUpdateItem->bNeedUpdate = !CheckFileCRC(atoi(pUpdateItem->szCRC), szLocalFullFileName);   // CRC        
  1023.         goto Exit0;
  1024.     }
  1026.     if (pUpdateItem->KCheckVersionMethod == enumCVM_TIME)
  1027.     {
  1028.         nRetCode = CheckFileDate(szLocalFullFileName, pUpdateItem->szFileDate);
  1029.         if (nRetCode)
  1030.         {
  1031.             pUpdateItem->bNeedUpdate = true;
  1032.             goto Exit0;
  1033.         }
  1034.         
  1035.         goto Exit1;
  1036.     }
  1037.     
  1038.     if ((pUpdateItem->KCheckVersionMethod == enumCVM_SIGN) &&
  1039.         (IsSignFileName(pUpdateItem->szFileName) || IsUpdFileName(pUpdateItem->szFileName)) 
  1040. )
  1041.     {
  1042.         nRetCode = GetLocalSignVersion(&dwMajorVersion, &dwMinorVersion); 
  1043.         nSignFileFlag = 1;
  1044.     }
  1045.     else
  1046.     {
  1047.         // Such as KAEngine.dat, KAEPlats KAEngine.dat, KAEPlat.dll, KAV9X.EXE......
  1048.         
  1049.         nRetCode = GetKAVFileVersion(szLocalFullFileName, pUpdateItem->KCheckVersionMethod, &dwMajorVersion, &dwMinorVersion);
  1050.         if (!nRetCode)
  1051. {
  1052. nRetCode = _CheckVersionByRes(szLocalFullFileName, pUpdateItem->szFileDate);
  1053.             if (nRetCode < 0)
  1054.                 pUpdateItem->bNeedUpdate = true;
  1055.             else if (nRetCode == 0)
  1056.                 goto Exit1;
  1058.             goto Exit0;
  1059. }
  1060.     }
  1062.     pUpdateItem->bNeedUpdate = (
  1063.         (pUpdateItem->uMajorVersion > dwMajorVersion) ||
  1064.         (
  1065. (pUpdateItem->uMajorVersion == dwMajorVersion) && 
  1066. (pUpdateItem->uMinorVersion > dwMinorVersion)
  1067.         ) 
  1068. );
  1070.     if (pUpdateItem->bNeedUpdate || 
  1071.         !((pUpdateItem->uMajorVersion == dwMajorVersion) && 
  1072. (pUpdateItem->uMinorVersion == dwMinorVersion))
  1073. )
  1074.         goto Exit0;
  1076. Exit1:
  1077.     // if is the same version but the file was destroy then update this file
  1078.     if (nSignFileFlag) 
  1079.     {
  1080. nRetCode = CheckExistSameVersionFile(pUpdateItem->szFileName);
  1081. if (nRetCode == 0)
  1082. pUpdateItem->bNeedUpdate = true;
  1083. else if (nRetCode == 1)
  1084. pUpdateItem->bNeedUpdate = false;
  1085. else
  1086. pUpdateItem->bNeedUpdate = !CheckFileCRC(atoi(pUpdateItem->szCRC), szLocalFullFileName);
  1087.     }
  1088.     else
  1089.         pUpdateItem->bNeedUpdate = !CheckFileCRC(atoi(pUpdateItem->szCRC), szLocalFullFileName);
  1091. Exit0:
  1092.     return pUpdateItem->bNeedUpdate;
  1093. }
  1095. CString CProcessIndex::GetVirusNames()
  1096. {
  1097.     return m_sVirusNames;
  1098. }
  1100. CString CProcessIndex::GetReadme()
  1101. {
  1102.     return m_sReadme;
  1103. }
  1105. CString CProcessIndex::GetAnnounce()
  1106. {
  1107.     return m_sAnnounce;
  1108. }
  1110. int CProcessIndex::GetHost(char szHostName[], int *pnNameSize, char szHostURL[], int *pnURLSize)
  1111. {
  1112.     int nResult = false;    
  1113.     
  1114.     if (!pnNameSize || !pnURLSize)
  1115.         goto Exit0;
  1119.     nResult = true;
  1120. Exit0:
  1121.     return nResult;
  1122. }
  1124. int CProcessIndex::GetDubaVersion()
  1125. {
  1126.     int nResult = -1;
  1128.     if (!m_bInitFlag)  
  1129. return nResult;
  1131. nResult = ::GetPrivateProfileInt("COMMENT", "Version", 0, m_szTxtFileName);
  1133. return nResult;
  1134. }
  1136. int CProcessIndex::GetPath(char szPath[], int nSize)
  1137. {
  1138. if (!m_bInitFlag)  
  1139. return false;
  1141. ::GetPrivateProfileString("COMMENT", "PathName", "", szPath, nSize, m_szTxtFileName);
  1143. return true;    
  1144. }