allegiance/lobby/AutoUpdate.cpp

/*-------------------------------------------------------------------------
  AutoUpdate.cpp
  
  Management of the AutoUpdate system
  
  Owner: 
  
  Copyright 1986-2000 Microsoft Corporation, All Rights Reserved
 *-----------------------------------------------------------------------*/


#include "pch.h"



class CAutoUpdateImpl :
    public IAutoUpdate
{

public:

    CAutoUpdateImpl() :
      m_crcFileList(0),
      m_nFileListSize(0)
    {
    }


    /////////////////////////////////////////////////////////////////////////

    void    LoadSettings(HKEY hk)
    {
        bool m_bAutoUpdateActive;

        m_bAutoUpdateActive = _Module.ReadFromRegistry(hk, true, "FTPArtServer", m_szFTPServer, NULL);
        m_bAutoUpdateActive = _Module.ReadFromRegistry(hk, true, "FTPArtRoot", m_szFTPInitialDir, NULL) && m_bAutoUpdateActive;
        _Module.ReadFromRegistry(hk, true, "FTPAccount", m_szFTPAccount, (DWORD)"Anonymous", false);
        _Module.ReadFromRegistry(hk, true, "FTPPW", m_szFTPPassword, (DWORD)"Unknown", false);

        if (!m_bAutoUpdateActive)
        {
          _Module.LogEvent(EVENTLOG_ERROR_TYPE, LE_BadAutoUpdateConfig);
          g_pAutoUpdate = NULL;
          delete this;
        }
    }

    /////////////////////////////////////////////////////////////////////////

    void   LoadCRC(char * szFileName) 
    {
        char szErrorMsg[200+MAX_PATH];

        HANDLE hFile = CreateFile(szFileName, 
                                  GENERIC_READ, 
                                  FILE_SHARE_READ, 
                                  NULL, 
                                  OPEN_EXISTING, 
                                  FILE_ATTRIBUTE_NORMAL, 
                                  NULL);

        if (hFile == INVALID_HANDLE_VALUE)
        {
           sprintf(szErrorMsg, "Failed open file (%s); Error Code: %d ", szFileName, GetLastError());
        }
        else
        {
            m_nFileListSize = GetFileSize(hFile, NULL);

            char szErrorMsgBuffer[100+MAX_PATH];

            m_crcFileList = FileCRC(hFile, szErrorMsgBuffer);

            if (m_crcFileList == 0)
            {
                sprintf(szErrorMsg, "Error while determining CRC; %s", szErrorMsgBuffer);
            }
            else
            {
                ::CloseHandle(hFile);
                return;
            }
    
        }

        _Module.LogEvent(EVENTLOG_ERROR_TYPE, LE_AutoUpdateDeactivated, szErrorMsg);
        g_pAutoUpdate = NULL;
        delete this;
    }

    /////////////////////////////////////////////////////////////////////////

    char *  GetFTPServer()
    {
        return m_szFTPServer;
    }

    /////////////////////////////////////////////////////////////////////////

    char *  GetFTPInitialDir()
    {
        return m_szFTPInitialDir;
    }

    /////////////////////////////////////////////////////////////////////////

    char *  GetFTPAccount()
    {
        return m_szFTPAccount;
    }

    /////////////////////////////////////////////////////////////////////////

    char *  GetFTPPassword()    
    {
        return m_szFTPPassword;
    }

    /////////////////////////////////////////////////////////////////////////

    int     GetFileListCRC()    
    {
        return m_crcFileList;
    }

    /////////////////////////////////////////////////////////////////////////

    unsigned GetFileListSize()
    {
        return m_nFileListSize;
    }

    /////////////////////////////////////////////////////////////////////////

private:

  char              m_szFTPServer[MAX_PATH+1];
  char              m_szFTPInitialDir[MAX_PATH+1];
  char              m_szFTPAccount[MAX_PATH+1];
  char              m_szFTPPassword[MAX_PATH+1];
  int               m_crcFileList;
  unsigned          m_nFileListSize;
};

IAutoUpdate * g_pAutoUpdate = NULL; // NULL if AutoUpdate is disabled


void CreateAutoUpdate(HKEY hk, char * szFileName)
{
    g_pAutoUpdate = new CAutoUpdateImpl();

    g_pAutoUpdate->LoadSettings(hk);

    if (g_pAutoUpdate) // if loading settings failed, g_pAutoUpdate is NULL
    {
        g_pAutoUpdate->LoadCRC(szFileName);
    }
}