#ifndef _HELPER_H_
#define _HELPER_H_

#include "wmrgexp.h"
#include "ocmanage.h"
#include "setupapi.h"

#if DBG == 1
	#include <stdio.h>
	#include <stdarg.h>
	#define DECLARE_INFOLEVEL(comp) extern unsigned long comp##InfoLevel = DEF_INFOLEVEL;
	#define DECLARE_DEBUG(comp) \
	    extern unsigned long comp##InfoLevel; \
	    _inline void \
	    comp##InlineDebugOut(unsigned long fDebugMask, TCHAR *pszfmt, ...) \
	    { \
	        if (comp##InfoLevel & fDebugMask) \
	        { \
	            TCHAR acsString[1000];\
	            va_list va; \
	            va_start(va, pszfmt);\
	            _vstprintf(acsString, pszfmt, va); \
	            va_end(va);\
	            OutputDebugString(acsString);\
	        } \
	    }\
	    _inline void \
	    comp##InlineDebugOut( TCHAR *pszfmt, ...) \
	    { \
	        if ( TRUE ) \
	        { \
	            TCHAR acsString[1000];\
	            va_list va; \
	            va_start(va, pszfmt);\
	            _vstprintf(acsString, pszfmt, va); \
	            va_end(va);\
	            OutputDebugString(acsString);\
	        } \
	    }
#else  // DBG == 0
	#define DECLARE_DEBUG(comp)
	#define DECLARE_INFOLEVEL(comp)
#endif // DBG == 0

DECLARE_DEBUG(iis);
#define iisDebugOut(x) iisDebugOut2 x
#define iisDebugOutSafeParams(x) iisDebugOutSafeParams2 x
#define iisDebugOutSafe(x) iisDebugOutSafe2 x

#define DO_IT_FOR_W3SVC_ANONYMOUSUSER    1
#define DO_IT_FOR_MSFTPSVC_ANONYMOUSUSER 2
#define DO_IT_FOR_W3SVC_WAMUSER          4

// External globals from other .cpp files
class  MyLogFile;
extern MyLogFile g_MyLogFile;
class  CInitApp;
extern CInitApp* g_pTheApp;
extern HANDLE g_MyModuleHandle;

// structs
typedef struct _CLUSTER_SVC_INFO_FILL_STRUCT
{
    LPTSTR szTheClusterName;
    LPWSTR pszTheServiceType;
    CString * csTheReturnServiceResName;
    DWORD dwReturnStatus;
} CLUSTER_SVC_INFO_FILL_STRUCT;

typedef struct _ScriptMapNode {
    TCHAR szExt[32];
    TCHAR szProcessor[_MAX_PATH];
    DWORD dwFlags;
    TCHAR szMethods[_MAX_PATH];
    struct _ScriptMapNode *prev, *next;
} ScriptMapNode;

//
// Functions
//
void iisDebugOut2(int iLogType, TCHAR *pszfmt, ...);
void iisDebugOutSafeParams2(int iLogType, TCHAR *pszfmt, ...);
void iisDebugOutSafe2(int iLogType, TCHAR *pszfmt);
void iisDebugOut_Start(TCHAR *pszString, int iLogType = LOG_TYPE_TRACE_WIN32_API);
void iisDebugOut_Start1(TCHAR *pszString1, TCHAR *pszString2, int iLogType = LOG_TYPE_TRACE_WIN32_API);
void iisDebugOut_Start1(TCHAR *pszString1, CString pszString2, int iLogType = LOG_TYPE_TRACE_WIN32_API);
void iisDebugOut_End(TCHAR *pszString, int iLogType = LOG_TYPE_TRACE_WIN32_API);
void iisDebugOut_End1(TCHAR *pszString1, TCHAR *pszString2, int iLogType = LOG_TYPE_TRACE_WIN32_API);
void iisDebugOut_End1(TCHAR *pszString1, CString pszString2, int iLogType = LOG_TYPE_TRACE_WIN32_API);
void ProgressBarTextStack_Push(CString csText);
void ProgressBarTextStack_Push(LPTSTR szText);
void ProgressBarTextStack_Pop(void);
void ProgressBarTextStack_Set(int iStringID);
void ProgressBarTextStack_Set(int iStringID, const CString& csFileName);
void ProgressBarTextStack_Set(int iStringID, const CString& csString1, const CString& csString2);
void ProgressBarTextStack_Set(LPCTSTR szProgressTextString);
void ProgressBarTextStack_Inst_Set( int ServiceNameID, int iInstanceNum);
void ProgressBarTextStack_InstVRoot_Set( int ServiceNameID, int iInstanceNum, CString csVRName);
void ProgressBarTextStack_InstInProc_Set( int ServiceNameID, int iInstanceNum, CString csVRName);
void ListOfWarnings_Add(TCHAR * szEntry);
void ListOfWarnings_Display(void);
HRESULT GetLastSectionToBeCalled(void);
INT  Register_iis_common();
INT  Unregister_iis_common();
INT  Unregister_old_asp();
INT  Register_iis_core();
INT  Unregister_iis_core();
INT  Register_iis_inetmgr();
INT  Unregister_iis_inetmgr();
INT  Register_iis_pwmgr();
INT  Unregister_iis_pwmgr();
INT  Register_iis_www();
INT  Unregister_iis_www();
INT  Register_iis_doc();
INT  Register_iis_ftp();
INT  Unregister_iis_ftp();
LPWSTR MakeWideStrFromAnsi(LPSTR psz);
int  GetMultiStrSize(LPTSTR p);
BOOL IsValidNumber(LPCTSTR szValue);
int  GetRandomNum(void);
void SetRebootFlag(void);
BOOL RunProgram( LPCTSTR pszProgram, LPTSTR CmdLine, BOOL fMinimized , DWORD dwWaitTimeInSeconds, BOOL fCreateNewConsole);
void HandleSpecificErrors(DWORD iTheErrorCode, DWORD dwFormatReturn, CString csMsg, TCHAR pMsg[], CString *);
BOOL GetDataFromMetabase(LPCTSTR szPath, int nID, LPBYTE Buffer, int BufSize);
void AddOLEAUTRegKey();
DWORD RegisterOLEControl(LPCTSTR lpszOcxFile, BOOL fAction);
void lodctr(LPCTSTR lpszIniFile);
void unlodctr(LPCTSTR lpszDriver);
INT  InstallPerformance(CString nlsRegPerf,CString nlsDll,CString nlsOpen,CString nlsClose,CString nlsCollect );
INT  AddEventLog(BOOL fSystem, CString nlsService, CString nlsMsgFile, DWORD dwType);
INT  RemoveEventLog(BOOL fSystem, CString nlsService );
INT  InstallAgent( CString nlsName, CString nlsPath );
INT  RemoveAgent( CString nlsServiceName );
void InstallMimeMap();
int  CreateInProc(LPCTSTR lpszPath, int iUseOOPPool);
void CreateInProc_Wrap(LPCTSTR lpszPath, int iUseOOPPool);
void DeleteInProc(LPCTSTR lpszKeyPath);
void SetAppFriendlyName(LPCTSTR szKeyPath);
void SetInProc(LPCTSTR szKeyPath);
void AddCustomError(IN DWORD dwCustErr, IN INT intSubCode, IN LPCTSTR szErrorString, IN LPCTSTR szKeyPath, IN BOOL fOverwriteExisting );
ScriptMapNode *AllocNewScriptMapNode(LPTSTR szExt, LPTSTR szProcessor, DWORD dwFlags, LPTSTR szMethods);
void InsertScriptMapList(ScriptMapNode *pList, ScriptMapNode *p, BOOL fReplace);
void FreeScriptMapList(ScriptMapNode *pList);
void GetScriptMapListFromRegistry(ScriptMapNode *pList);
void GetScriptMapListFromMetabase(ScriptMapNode *pList, int iUpgradeType);
void WriteScriptMapListToMetabase(ScriptMapNode *pList, LPTSTR szKeyPath, DWORD dwFlags);
DWORD CallProcedureInDll_wrap(LPCTSTR lpszDLLFile, LPCTSTR lpszProcedureToCall, BOOL bDisplayMsgOnErrFlag, BOOL bInitOleFlag, BOOL iFunctionPrototypeFlag);
DWORD CallProcedureInDll(LPCTSTR lpszDLLFile, LPCTSTR lpszProcedureToCall, BOOL bDisplayMsgOnErrFlag, BOOL bInitOleFlag, BOOL iFunctionPrototypeFlag);
void  GetDebugLevelFromInf(IN HINF hInfFileHandle);
int  IsThisStringInThisCStringList(CStringList &strList, LPCTSTR szStringToLookFor);
int  KillProcess_Wrap(LPCTSTR lpFullPathOrJustFileName);
int  ProcessSection(IN HINF hFile, IN LPCTSTR szTheSection);
void uiCenterDialog( HWND hwndDlg );
int  AfterRemoveAll_SaveMetabase(void);
int  iOleInitialize(void);
void iOleUnInitialize(int iBalanceOLE);
BOOL SetupSetDirectoryId_Wrapper(HINF InfHandle,DWORD Id,LPCTSTR Directory);
BOOL SetupSetStringId_Wrapper(HINF InfHandle,DWORD Id,LPCTSTR TheString);
void LogImportantFiles(void);
HRESULT FTestForOutstandingCoInits(void);
void DisplayStringForMetabaseID(DWORD dwMetabaseID);
void ReturnStringForMetabaseID(DWORD dwMetabaseID, LPTSTR lpReturnString);
void SetErrorFlag(char *szFileName, int iLineNumber);
DWORD FillStrListWithListOfSections(IN HINF hFile, CStringList &strList, IN LPCTSTR szSection);
void MesssageBoxErrors_IIS(void);
void MesssageBoxErrors_MTS(int iMtsThingWeWereDoing, DWORD dwErrorCode);
void ShowIfModuleUsedForGroupOfSections(IN HINF hFile, int iUnlockThem);
int ReadGlobalsFromInf(HINF InfHandle);
int CheckIfPlatformMatchesInf(HINF InfHandle);
int CheckSpecificBuildinInf(HINF InfHandle);
int CheckForOldGopher(HINF InfHandle);
void SetOCGlobalPrivateData(void);
BOOL GetJavaTLD(LPTSTR lpszDir);
void SetDIRIDforThisInf(HINF InfHandle,BOOL bOCInit);
void ShowStateOfTheseServices(IN HINF hFile);
int  GetScriptMapAllInclusionVerbs(CString &csTheVerbList);
void DumpScriptMapList();
int  GetSectionNameToDo(IN HINF hFile, TSTR *pstrSectionName);
void CustomWWWRoot(LPCTSTR szWWWRoot);
void CustomFTPRoot(LPCTSTR szFTPRoot);
void AdvanceProgressBarTickGauge(int iTicks = 1);
int  IsMetabaseCorrupt(void);
int ReadUserConfigurable(HINF InfHandle);
int ReverseExpandEnvironmentStrings(LPTSTR szOriginalDir,LPTSTR szNewlyMungedDir);
BOOL SetupFindFirstLine_Wrapped(IN HINF InfHandle,IN LPCTSTR Section,IN LPCTSTR Key,OPTIONAL INFCONTEXT *Context);
INT IsThisOnNotStopList(IN HINF hFile, CString csInputName, BOOL bServiceFlag);
HRESULT MofCompile(TCHAR * szPathMofFile);
DWORD   DoesEntryPointExist(LPCTSTR lpszDLLFile, LPCTSTR lpszProcedure);
void    CreateDummyMetabaseBin(void);
BOOL    RunningAsAdministrator();
LPTSTR  CreatePassword(int iSize);
void    CreatePasswordOld(TCHAR *pszPassword,int iSize);
void StopAllServicesRegardless(int iShowErrorsFlag);
LPTSTR SafeCopy( LPTSTR szDestination, LPTSTR szSource, DWORD dwSize);
LPTSTR SafeCat( LPTSTR szDestination, LPTSTR szSource, DWORD dwSize);
BOOL DoesTheInstallDrivePreserveAcls( LPBOOL pbPreservesAcls );
BOOL UpdateAnonymousUsers( HINF InfHandle );

#endif // _HELPER_H_