|
|
// Copyright (c) 1999 Microsoft Corporation
#ifndef __UPGRADE_H_
#define __UPGRADE_H_
#include <wbemidl.h>
class CMultiString;class CString;
#define ToHex(n) ((BYTE) (((n) > 9) ? ((n) - 10 + 'A') : ((n) + '0')))
#define WINMGMT_DBCONVERT_NAME __TEXT("upgrade.imp")
#define MAX_MSG_TITLE_LENGTH 255 //defined by Bakul for localization
#define MAX_MSG_TEXT_LENGTH 1024 //defined by Bakul for localization
#define INTERNAL_DATABASE_VERSION 9
#define LANG_ID_STR_SIZE 5
extern "C" char g_szLangId[LANG_ID_STR_SIZE];
typedef struct { const char* pszMofFilename; //MOF filename
int nInstallType; //Selection of InstallType items
} MofDataTable;
enum InstallType { Core = 1, MUI = 2};
enum MsgType{ MSG_INFO, MSG_WARNING, MSG_ERROR, MSG_NTSETUPERROR};
enum{ no_error, failed, critical_error, out_of_memory};
template<class T>class CDeleteMe{protected: T* m_p;
public: CDeleteMe(T* p = NULL) : m_p(p){} ~CDeleteMe() {delete m_p;}
// overwrites the previous pointer, does NOT delete it
void operator= (T* p) {m_p = p;}};
template<class T>class CVectorDeleteMe{protected: T* m_p; T** m_pp;
public: CVectorDeleteMe(T* p) : m_p(p), m_pp(NULL){} CVectorDeleteMe(T** pp) : m_p(NULL), m_pp(pp){} ~CVectorDeleteMe() {if(m_p) delete [] m_p; else if(m_pp) delete [] *m_pp;}};
typedef void (WINAPI *ESCDOOR_BEFORE_MOF_COMPILATION) ();typedef void (WINAPI *ESCDOOR_AFTER_MOF_COMPILATION) ();
void CallEscapeRouteAfterMofCompilation();void CallEscapeRouteBeforeMofCompilation();void ClearWMISetupRegValue();bool CopyMultiString(CMultiString &mszFrom, CMultiString &mszTo);void DeleteMMFRepository();bool DoConvertRepository();bool DoCoreUpgrade(int nInstallType=1);bool DoWDMNamespaceInit();bool DoesFSRepositoryExist();bool DoesMMFRepositoryExist();bool DoMofLoad(wchar_t* pComponentName, CMultiString& mszSystemMofs);bool EnableESS();bool ExtractPathAndFilename(const char *pszFullPath, CString &path, CString &filename);bool FileExists(const char *pszFilename);char* GetFullFilename(const char *pszFilename, InstallType eInstallType=Core);bool GetMofList(const char* rgpszMofFilename[], CMultiString &mszMofs);bool GetNewMofLists(const char *pszMofList, CMultiString &mszSystemMofs, CMultiString &mszOtherMofs, CString &szMissingMofs);HRESULT GetRepositoryDirectory(wchar_t wszRepositoryDirectory[MAX_PATH+1]);bool GetStandardMofs(CMultiString &mszSystemMofs, int nCurInstallType);bool IsStandardMof(CMultiString &mszStandardMofList, const char* pszMofFile);bool LoadMofList(IWbemContext *, IWinmgmtMofCompiler * pCompiler, const char *mszMofs, CString &szMOFsWhichFailedToLoad, long lOptionFlags = WBEM_FLAG_CONNECT_REPOSITORY_ONLY | WBEM_FLAG_DONT_ADD_TO_LIST, long lClassFlags = WBEM_FLAG_UPDATE_FORCE_MODE);void LogMessage(MsgType msgType, const char *pszMessage);void LogSetupError(const char *pszMessage);void RecordFileVersion();void SetWBEMBuildRegValue();void ShutdownWinMgmt();bool UpgradeAutoRecoveryRegistry(CMultiString &mszSystemMofs, CMultiString &mszExternalMofList, CString &szMissingMofs);bool UpgradeRepository();bool WipeOutAutoRecoveryRegistryEntries();bool WriteBackAutoRecoveryMofs(CMultiString &mszSystemMofs, CMultiString &mszExternalMofList);
HRESULT MoveRepository();
extern HRESULT UpdateServiceSecurity () ;extern HRESULT CheckForServiceSecurity () ;
#ifdef _X86_
bool RemoveOldODBC();#endif
#endif // __UPGRADE_H_
|
