//+-------------------------------------------------------------------------
//
//  Microsoft Windows
//
//  Copyright (C) Microsoft Corporation, 1998 - 1999
//
//  File:       server.h
//
//--------------------------------------------------------------------------


#ifndef _SERVER_H
#define _SERVER_H

#include "dnsutil.h"

///////////////////////////////////////////////////////////////////////////////
// GLOBAL FUNCTIONS

DNS_STATUS ServerHasCache(LPCWSTR lpszServerName, BOOL* pbRes);
DNS_STATUS ServerHasRootZone(LPCWSTR lpszServerName, BOOL* pbRes);

extern LPCWSTR DNS_EVT_COMMAND_LINE;
extern LPCWSTR MMC_APP;

///////////////////////////////////////////////////////////////////////////////
// FORWARD DECLARATIONS

class CDNSMTContainerNode;
class CDNSServerNode;
class CDNSZoneNode;
class CDNSDomainNode;
class CCathegoryFolderNode;
class CDNSRootHintsNode;
class CDNSServer_TestPropertyPage;
class CDNSQueryFilter;

///////////////////////////////////////////////////////////////////////////////
// defines for server test intervals

#define MIN_SERVER_TEST_INTERVAL		30	// seconds
#define MAX_SERVER_TEST_INTERVAL		3600 // seconds in 1 hour
#define DEFAULT_SERVER_TEST_INTERVAL	60	// seconds


///////////////////////////////////////////////////////////////////////////////
// defines for server sheet messages

#define SHEET_MSG_SERVER_TEST_DATA (1)
#define SHEET_MSG_SELECT_PAGE		(2)

///////////////////////////////////////////////////////////////////////////////
// CZoneInfoHolder : simple memory manager for arrays of zone info handles

class CZoneInfoHolder
{
public:
	CZoneInfoHolder();
	~CZoneInfoHolder();
	BOOL Grow();

	PDNS_ZONE_INFO* m_zoneInfoArray;
	DWORD m_dwArrSize;
	DWORD m_dwZoneCount;

private:
	void AllocateMemory(DWORD dwArrSize);
	void FreeMemory();
};


///////////////////////////////////////////////////////////////////////////////
// CDNSMTContainerNode
// base class from which all the MT nodes derive from

class CDNSMTContainerNode : public CMTContainerNode
{
public:
	// enumeration for node states, to handle icon changes
	typedef enum
	{
		notLoaded = 0, // initial state, valid only if server never contacted
		loading,
		loaded,
		unableToLoad,
		accessDenied
	} nodeStateType;


	CDNSMTContainerNode();

	void SetServerNode(CDNSServerNode* pServerNode)
		{ ASSERT(pServerNode != NULL); m_pServerNode = pServerNode; }
	CDNSServerNode* GetServerNode()
		{ ASSERT( m_pServerNode != NULL); return m_pServerNode; }

  virtual HRESULT OnSetToolbarVerbState(IToolbar* pToolbar, CNodeList* pNodeList);

	virtual int GetImageIndex(BOOL bOpenImage);

  virtual CColumnSet* GetColumnSet() 
  { 
    if (m_pColumnSet == NULL)
    {
      CDNSRootData* pRoot = (CDNSRootData*)GetRootContainer();
      m_pColumnSet = ((CDNSComponentDataObjectBase*)pRoot->GetComponentDataObject())->GetColumnSet(L"---Default Column Set---");
    }
    return m_pColumnSet;
  }
  virtual LPCWSTR GetColumnID() { return GetColumnSet()->GetColumnID(); }
  virtual LPWSTR GetDescriptionBarText();

protected:
	virtual BOOL CanCloseSheets();
	virtual void OnChangeState(CComponentDataObject* pComponentDataObject);
	virtual void OnHaveData(CObjBase* pObj, CComponentDataObject* pComponentDataObject);
  virtual void OnError(DWORD dwErr);

  CColumnSet* m_pColumnSet;
  CString m_szDescriptionBar;
private:
  CDNSServerNode* m_pServerNode;
};




///////////////////////////////////////////////////////////////////////////////
// CDNSQueryObj : general purpose base class

class CDNSQueryObj : public CQueryObj
{
public:
	CDNSQueryObj(BOOL bAdvancedView, DWORD dwServerVersion)
	{
		m_dwServerVersion = dwServerVersion;
		m_bAdvancedView = bAdvancedView;

    // internal state variables
    m_nObjectCount = 0;
    m_nMaxObjectCount = DNS_QUERY_OBJ_COUNT_MAX;
    m_bGetAll = TRUE;
    m_nFilterOption = DNS_QUERY_FILTER_DISABLED;
	}
	CString m_szServerName;
	DWORD m_dwServerVersion;
	BOOL m_bAdvancedView;

public:
  virtual BOOL AddQueryResult(CObjBase* pObj)
  {
    BOOL bRet = CQueryObj::AddQueryResult(pObj);
    if (bRet)
      m_nObjectCount++;
    return bRet;
  }
  void SetFilterOptions(CDNSQueryFilter* pFilter);
  BOOL TooMuchData();

protected:
  ULONG m_nMaxObjectCount;          // max number of objects in a query
  BOOL m_bGetAll;
  ULONG m_nObjectCount;             // number of objects queried so far

  UINT m_nFilterOption;
  CString m_szFilterString1;
  int m_nFilterStringLen1;       // cached value

  CString m_szFilterString2;
  int m_nFilterStringLen2;       // cached value


protected:
  BOOL MatchName(LPCWSTR lpszName);
};

/////////////////////////////////////////////////////////////////////////
// CCathegoryFolderNode

class CCathegoryFolderQueryObj : public CDNSQueryObj
{
public:
	CCathegoryFolderQueryObj(BOOL bAdvancedView, DWORD dwServerVersion)
		: CDNSQueryObj(bAdvancedView, dwServerVersion)
  {
  }
	virtual BOOL Enumerate();
	typedef enum { unk, cache, fwdAuthoritated, revAuthoritated, domainForwarders } queryType;
	void SetType(queryType type) { m_type = type;}

protected:
	queryType m_type;
	BOOL CanAddZone(PDNS_RPC_ZONE pZoneInfo);
};


class CCathegoryFolderNode : public CDNSMTContainerNode
{
public:
	CCathegoryFolderNode()
	{
		m_type = CCathegoryFolderQueryObj::unk;
	}

	virtual HRESULT OnCommand(long nCommandID, 
                            DATA_OBJECT_TYPES type, 
                            CComponentDataObject* pComponentData,
                            CNodeList* pNodeList);
	virtual void OnDelete(CComponentDataObject*,
                        CNodeList*)
		{ ASSERT(FALSE);}
	virtual BOOL OnSetRefreshVerbState(DATA_OBJECT_TYPES type, 
                                     BOOL* pbHide,
                                     CNodeList* pNodeList);
	virtual LPCWSTR GetString(int nCol)
	{
		return (nCol == 0) ? GetDisplayName() : g_lpszNullString;
	}
	CCathegoryFolderQueryObj::queryType GetType() { return m_type;}

  virtual LPWSTR GetDescriptionBarText();

  virtual CColumnSet* GetColumnSet() 
  { 
    if (m_pColumnSet == NULL)
    {
      CDNSRootData* pRoot = (CDNSRootData*)GetRootContainer();
      m_pColumnSet = ((CDNSComponentDataObjectBase*)pRoot->GetComponentDataObject())->GetColumnSet(L"---Zone Column Set---");
    }
    return m_pColumnSet;
  }

  virtual BOOL CanExpandSync() { return FALSE; }

protected:
	virtual LPCONTEXTMENUITEM2 OnGetContextMenuItemTable()
				{ return CDNSCathegoryFolderHolder::GetContextMenuItem(); }
	virtual BOOL OnAddMenuItem(LPCONTEXTMENUITEM2 pContextMenuItem2,
								             long *pInsertionAllowed);

	virtual CQueryObj* OnCreateQuery();

	CCathegoryFolderQueryObj::queryType m_type;
};

/////////////////////////////////////////////////////////////////////////
// CDNSCacheNode
//
// * contains the root domain "."
// * can delete items underneath, but cannot add or modify

class CDNSCacheNode : public CCathegoryFolderNode
{
public:
	CDNSCacheNode();

  virtual HRESULT OnCommand(long nCommandID, 
                            DATA_OBJECT_TYPES type, 
                            CComponentDataObject* pComponentData,
                            CNodeList* pNodeList);

  void OnClearCache(CComponentDataObject* pComponentData);

  virtual BOOL CanExpandSync() { return FALSE; }

protected:
	virtual LPCONTEXTMENUITEM2 OnGetContextMenuItemTable()
				{ return CDNSCacheMenuHolder::GetContextMenuItem(); }
	virtual BOOL OnAddMenuItem(LPCONTEXTMENUITEM2 pContextMenuItem2,
								             long *pInsertionAllowed);
};

/////////////////////////////////////////////////////////////////////////
// CDNSDomainForwardersNode
//
// * contains the domain forwarders as zones
// * this node will always be hidden
//

class CDNSDomainForwardersNode : public CCathegoryFolderNode
{
public:
	CDNSDomainForwardersNode();

	virtual BOOL OnEnumerate(CComponentDataObject* pComponentData, BOOL bAsync = TRUE);
};


/////////////////////////////////////////////////////////////////////////
// CDNSAuthoritatedZonesNode
//
// * contains autoritated zones, both primary and secondary
// *  have one for FWD and one for REVERSE

class CDNSAuthoritatedZonesNode : public CCathegoryFolderNode
{
public:
	CDNSAuthoritatedZonesNode(BOOL bReverse, UINT nStringID);

	BOOL IsReverse() { return m_bReverse;}

	virtual HRESULT OnCommand(long nCommandID, 
                            DATA_OBJECT_TYPES type, 
                            CComponentDataObject* pComponentData,
                            CNodeList* pNodeList);
	virtual BOOL OnSetRefreshVerbState(DATA_OBJECT_TYPES type, 
                                     BOOL* pbHide,
                                     CNodeList* pNodeList);
  virtual HRESULT OnSetToolbarVerbState(IToolbar* pToolbar, 
                                        CNodeList* pNodeList);

  virtual BOOL CanExpandSync() { return FALSE; }

protected:
	virtual LPCONTEXTMENUITEM2 OnGetContextMenuItemTable()
				{ return CDNSAuthoritatedZonesMenuHolder::GetContextMenuItem(); }
	virtual BOOL OnAddMenuItem(LPCONTEXTMENUITEM2 pContextMenuItem2,
								             long *pInsertionAllowed);

private:
	// command handlers
	HRESULT OnNewZone(CComponentDataObject* pComponentData, CNodeList* pNodeList);

	// data
	BOOL m_bReverse;

  DECLARE_TOOLBAR_MAP()
};

/////////////////////////////////////////////////////////////////////////
// CDNSForwardZonesNode
class CDNSForwardZonesNode : public CDNSAuthoritatedZonesNode
{
public:
	CDNSForwardZonesNode();

  virtual HRESULT OnShow(LPCONSOLE lpConsole);
  virtual HRESULT GetResultViewType(CComponentDataObject* pComponentData, 
                                    LPOLESTR *ppViewType, 
                                    long *pViewOptions);

};

/////////////////////////////////////////////////////////////////////////
// CDNSReverseZonesNode
class CDNSReverseZonesNode : public CDNSAuthoritatedZonesNode
{
public:
	CDNSReverseZonesNode();
	virtual BOOL OnRefresh(CComponentDataObject* pComponentData,
                         CNodeList* pNodeList);
protected:
	virtual void OnHaveData(CObjBase* pObj, CComponentDataObject* pComponentDataObject);

	void ChangeViewOption(BOOL bAdvanced, CComponentDataObject* pComponentDataObject);
  virtual HRESULT OnShow(LPCONSOLE lpConsole);
  virtual HRESULT GetResultViewType(CComponentDataObject* pComponentData, 
                                    LPOLESTR *ppViewType, 
                                    long *pViewOptions);

private:
	// cached pointers to autocreated zones
	CDNSZoneNode* m_p0ZoneNode;
	CDNSZoneNode* m_p127ZoneNode;
	CDNSZoneNode* m_p255ZoneNode;

	friend class CDNSServerNode;
};


/////////////////////////////////////////////////////////////////////////
// CDNSServerTestOptions

class CDNSServerTestOptions
{
public:
	CDNSServerTestOptions();
	HRESULT Save(IStream* pStm);
	HRESULT Load(IStream* pStm);
	const CDNSServerTestOptions& operator=(const CDNSServerTestOptions& x);
	BOOL operator==(const CDNSServerTestOptions& x) const;

	BOOL	m_bEnabled;				// polling enabled
	DWORD	m_dwInterval;	// seconds
	// query types
	BOOL	m_bSimpleQuery;
	BOOL	m_bRecursiveQuery;
};

////////////////////////////////////////////////////////////////////////
// CDNSServerTestQueryResult

class CDNSServerTestQueryResult
{
public:
	CDNSServerTestQueryResult()
		{ memset(this, 0x0, sizeof(CDNSServerTestQueryResult));}
	const CDNSServerTestQueryResult& operator=(const CDNSServerTestQueryResult& x)
	{
		memcpy(this, &x, sizeof(CDNSServerTestQueryResult));
		return *this;
	}
	BOOL operator==(const CDNSServerTestQueryResult& x)
	{
		ASSERT(m_serverCookie == x.m_serverCookie); // always compare the same server
		// we do not compare the time
		// we do not compare the force logging flag
		// we want just to compare the query flags and the results
		return (m_dwQueryFlags == x.m_dwQueryFlags) &&
				(m_dwAddressResolutionResult == x.m_dwAddressResolutionResult) &&
				(m_dwPlainQueryResult == x.m_dwPlainQueryResult) &&
				(m_dwRecursiveQueryResult == x.m_dwRecursiveQueryResult);
	}
	BOOL Succeded()
	{
		return (m_dwAddressResolutionResult == 0) &&
				(m_dwPlainQueryResult == 0) &&
				(m_dwRecursiveQueryResult == 0);
	}
	static DWORD Pack(BOOL bPlainQuery, BOOL bRecursiveQuery)
	{
		DWORD dw = 0;
		if (bPlainQuery)
			dw |= 0x1;
		if (bRecursiveQuery)
			dw |= 0x2;
		return dw;
	}
	static void Unpack(DWORD dwQueryFlags, BOOL* pbPlainQuery, BOOL* pbRecursiveQuery)
	{
		*pbPlainQuery = (dwQueryFlags & 0x1);
		*pbRecursiveQuery = (dwQueryFlags & 0x2);
	}
// Data
	MMC_COOKIE  m_serverCookie;
	BOOL		m_bAsyncQuery;
	SYSTEMTIME	m_queryTime;
	DWORD		m_dwQueryFlags;
	DWORD		m_dwAddressResolutionResult;
	DWORD		m_dwPlainQueryResult;
	DWORD		m_dwRecursiveQueryResult;
};

////////////////////////////////////////////////////////////////////////
// CDNSServerTestQueryResultList

class CDNSServerTestQueryResultList : public
		CList< CDNSServerTestQueryResult*, CDNSServerTestQueryResult* >
{
public:
	typedef enum addAction { none = 0, changed, added, addedAndRemoved};

	CDNSServerTestQueryResultList()
	{
		m_nMaxCount = 50;
		ExceptionPropagatingInitializeCriticalSection(&m_cs);
	}
	~CDNSServerTestQueryResultList()
	{
		while (!IsEmpty())
			delete RemoveHead();
		::DeleteCriticalSection(&m_cs);
	}
public:
	void Lock() { ::EnterCriticalSection(&m_cs); }
	void Unlock() { ::LeaveCriticalSection(&m_cs); }

	addAction AddTestQueryResult(CDNSServerTestQueryResult* pTestResult)
	{
		addAction action = none;
		Lock();

		// determine if we have to add
		INT_PTR nCount = GetCount();
		CDNSServerTestQueryResult* pLastQueryResult = NULL;
		if (nCount > 0)
			pLastQueryResult = GetHead();
		if ((pTestResult->m_bAsyncQuery) ||
			(pLastQueryResult == NULL) ||
			!(*pLastQueryResult == *pTestResult))
		{
			// make sure we do not have too many items
			BOOL bRemoveLast = nCount > m_nMaxCount;
			if (bRemoveLast)
			{
				delete RemoveTail();
			}
			// add the item
			AddHead(pTestResult);
			action = bRemoveLast ? addedAndRemoved : added;
		}
		else
		{	
			// just just update the time stamp in the last message we have
			ASSERT(pLastQueryResult != NULL);
			memcpy(&(pLastQueryResult->m_queryTime), &(pTestResult->m_queryTime),
							sizeof(SYSTEMTIME));
			action = changed;
			delete pTestResult; // not added
		}
		Unlock();
		return action;
	}
	BOOL LastQuerySuceeded()
	{
		BOOL bRes = TRUE;
		Lock();
		if (GetCount() > 0)
			bRes = GetHead()->Succeded();
		Unlock();
		return bRes;
	}
	int GetMaxCount()
	{
		return m_nMaxCount;
	}

private:
	CRITICAL_SECTION m_cs;
	int m_nMaxCount;
};



/////////////////////////////////////////////////////////////////////////
// CDNSServerNode

class CDNSServerQueryObj : public CDNSQueryObj
{
public:
	CDNSServerQueryObj(BOOL bAdvancedView, DWORD dwServerVersion)
			: CDNSQueryObj(bAdvancedView, dwServerVersion){}
	virtual BOOL Enumerate();
private:
};


class CDNSServerNode : public CDNSMTContainerNode
{
public:
	CDNSServerNode(LPCTSTR lpszName, BOOL bIsLocalServer = FALSE);
	virtual ~CDNSServerNode();

	virtual void SetDisplayName(LPCWSTR lpszDisplayName);
  void SetLocalServer(BOOL bLocalServer) { m_bLocalServer = bLocalServer; }
  BOOL IsLocalServer() { return m_bLocalServer; }

	// node info
	DECLARE_NODE_GUID()
	virtual HRESULT GetDataHere(CLIPFORMAT cf, LPSTGMEDIUM lpMedium,
			CDataObject* pDataObject);
	virtual HRESULT GetData(CLIPFORMAT cf, LPSTGMEDIUM lpMedium,
			CDataObject* pDataObject);
  HRESULT RetrieveEventViewerLogs(LPSTGMEDIUM lpMedium, CDataObject* pDataObject);

  virtual CColumnSet* GetColumnSet() 
  { 
    if (m_pColumnSet == NULL)
    {
      CDNSRootData* pRoot = (CDNSRootData*)GetRootContainer();
      m_pColumnSet = ((CDNSComponentDataObjectBase*)pRoot->GetComponentDataObject())->GetColumnSet(L"---Server Column Set---");
    }
    return m_pColumnSet;
  }

	virtual HRESULT OnCommand(long nCommandID, 
                            DATA_OBJECT_TYPES type, 
                            CComponentDataObject* pComponentData,
                            CNodeList* pNodeList);
	virtual void OnDelete(CComponentDataObject* pComponentData,
                        CNodeList* pNodeList);
	virtual BOOL OnRefresh(CComponentDataObject* pComponentData,
                         CNodeList* pNodeList);

	virtual LPCWSTR GetString(int nCol)
	{
		return (nCol == 0) ? GetDisplayName() : g_lpszNullString;
	}

	virtual int GetImageIndex(BOOL bOpenImage);

	virtual BOOL HasPropertyPages(DATA_OBJECT_TYPES type, 
                                BOOL* pbHideVerb,
                                CNodeList* pNodeList);
	virtual HRESULT CreatePropertyPages(LPPROPERTYSHEETCALLBACK lpProvider, 
                                      LONG_PTR handle,
                                      CNodeList* pNodeList);
   virtual HRESULT CreatePropertyPagesHelper(LPPROPERTYSHEETCALLBACK lpProvider, 
									                  LONG_PTR handle, 
                                             long nStartProppage);

  virtual void DecrementSheetLockCount();

  virtual BOOL CanExpandSync() { return TRUE; }

protected:
	virtual LPCONTEXTMENUITEM2 OnGetContextMenuItemTable()
				{ return CDNSServerMenuHolder::GetContextMenuItem(); }
	virtual BOOL OnAddMenuItem(LPCONTEXTMENUITEM2 pContextMenuItem2,
								             long *pInsertionAllowed);
	virtual BOOL OnSetDeleteVerbState(DATA_OBJECT_TYPES type, 
                                    BOOL* pbHide,
                                    CNodeList* pNodeList);
	virtual BOOL OnSetRefreshVerbState(DATA_OBJECT_TYPES type, 
                                     BOOL* pbHide,
                                     CNodeList* pNodeList);
  virtual HRESULT OnSetToolbarVerbState(IToolbar* pToolbar, 
                                        CNodeList* pNodeList);

  virtual HRESULT GetResultViewType(CComponentDataObject* pComponentData, 
                                    LPOLESTR *ppViewType, 
                                    long *pViewOptions);
  virtual HRESULT OnShow(LPCONSOLE lpConsole);

	virtual CQueryObj* OnCreateQuery();
	virtual void OnHaveData(CObjBase* pObj, CComponentDataObject* pComponentDataObject);

#ifdef USE_NDNC
  BOOL ContainsDefaultNDNCs();
#endif

  virtual void SetRecordAging();
  virtual void ScavengeRecords();

private:
	// folders mainipulation
	//CCathegoryFolderNode* GetCathegoryFolder(CCathegoryFolderQueryObj::queryType type);

	CDNSAuthoritatedZonesNode* GetAuthoritatedZoneFolder(BOOL bFwd);

	// command handlers
	HRESULT OnNewZone(CComponentDataObject* pComponentData, CNodeList* pNodeList);
	void OnUpdateDataFiles(CComponentDataObject* pComponentData);
  void OnClearCache(CComponentDataObject* pComponentData);
  void OnConfigureServer(CComponentDataObject* pComponentData);
  void OnLaunchNSLookup();

#ifdef USE_NDNC
  void OnCreateNDNC();
#endif

public:
	void ChangeViewOption(BOOL bAdvanced, CComponentDataObject* pComponentData);
	
	// serialization from/to stream
	static HRESULT CreateFromStream(IStream* pStm, CDNSServerNode** ppServerNode);
	HRESULT SaveToStream(IStream* pStm);

	// DNS specific helpers
#ifdef USE_NDNC
	DNS_STATUS CreatePrimaryZone(LPCTSTR lpszName, 
                               LPCTSTR lpszFileName, 
                               BOOL bLoadExisting,
							                 BOOL bFwd, 
                               BOOL bDSIntegrated, 
                               UINT nDynamicUpdate,
							                 CComponentDataObject* pComponentData,
                               ReplicationType replType,
                               PCWSTR pszPartitionName);
	DNS_STATUS CreateStubZone(LPCTSTR lpszName, 
                            LPCTSTR lpszFileName,
									          BOOL bLoadExisting, 
                            BOOL bDSIntegrated, 
                            BOOL bFwd,
									          DWORD* ipMastersArray, 
                            int nIPMastersCount,
                            BOOL bLocalListOfMasters,
									          CComponentDataObject* pComponentData,
                            ReplicationType replType,
                            PCWSTR pszPartitionName);
#else
	DNS_STATUS CreatePrimaryZone(LPCTSTR lpszName, LPCTSTR lpszFileName, BOOL bLoadExisting,
							BOOL bFwd, BOOL bDSIntegrated, UINT nDynamicUpdate,
							CComponentDataObject* pComponentData);
	DNS_STATUS CreateStubZone(LPCTSTR lpszName, LPCTSTR lpszFileName,
									BOOL bLoadExisting, BOOL bDSIntegrated, BOOL bFwd,
									DWORD* ipMastersArray, int nIPMastersCount,
                            BOOL bLocalListOfMasters,
									CComponentDataObject* pComponentData);
#endif // USE_NDNC

	DNS_STATUS CreateSecondaryZone(LPCTSTR lpszName, LPCTSTR lpszFileName,
									BOOL bLoadExisting, BOOL bFwd,
									DWORD* ipMastersArray, int nIPMastersCount,
									CComponentDataObject* pComponentData);
  DNS_STATUS CreateForwarderZone(LPCTSTR lpszName, 
				                         DWORD* ipMastersArray, 
                                 int nIPMastersCount,
                                 DWORD dwTimeout,
                                 DWORD fSlave,
                                 CComponentDataObject* pComponentData);

	DNS_STATUS EnumZoneInfo(CZoneInfoHolder* pZoneInfoHolder);

	static DNS_STATUS EnumZoneInfo(LPCTSTR lpszServerName, CZoneInfoHolder* pZoneInfoHolder);

  DNS_STATUS ClearCache();

	// name used for RPC calls
	LPCWSTR GetRPCName();

	// server info access functions
	BOOL HasServerInfo() { ASSERT(m_pServInfoEx != NULL); return m_pServInfoEx->HasData();}
	void AttachServerInfo(CDNSServerInfoEx* pNewInfo);

	BOOL CanUseADS();
	DWORD GetVersion();
	BYTE GetMajorVersion(){ return DNS_SRV_MAJOR_VERSION(GetVersion());}
	BYTE GetMinorVersion(){ return DNS_SRV_MINOR_VERSION(GetVersion());}
	WORD GetBuildNumber() { return DNS_SRV_BUILD_NUMBER(GetVersion());}

   DWORD GetForestVersion();
   DWORD GetDomainVersion();

	void CreateDsServerLdapPath(CString& sz);
  void CreateDsZoneLdapPath(CDNSZoneNode* pZoneNode, CString& sz);
	void CreateDsZoneName(CDNSZoneNode* pZoneNode, CString& sz);
	void CreateDsNodeLdapPath(CDNSZoneNode* pZoneNode, CDNSDomainNode* pDomainNode, CString& sz);
  void CreateLdapPathFromX500Name(LPCWSTR lpszX500Name, CString& szLdapPath);

  //
  // Server property accessors
  //
	DWORD      GetNameCheckFlag();
	DNS_STATUS ResetNameCheckFlag(DWORD dwNameCheckFlag);

	DWORD      GetLogLevelFlag();
	DNS_STATUS ResetLogLevelFlag(DWORD dwLogLevel);

  DWORD      GetEventLogLevelFlag();
  DNS_STATUS ResetEventLogLevelFlag(DWORD dwEventLogLevel);

  PIP_ARRAY  GetDebugLogFilterList();
  DNS_STATUS ResetDebugLogFilterList(PIP_ARRAY pIPArray);

  PCWSTR     GetDebugLogFileName();
  DNS_STATUS ResetDebugLogFileName(PCWSTR pszLogFileName);

  DWORD      GetDebugLogFileMaxSize();
  DNS_STATUS ResetDebugLogFileMaxSize(DWORD dwMaxSize);

  BOOL       DoesRecursion();

	void       GetAdvancedOptions(BOOL* bOptionsArray);
	DNS_STATUS ResetAdvancedOptions(BOOL* bOptionsArray, DNS_STATUS* dwRegKeyOptionsErrorArr);

  UCHAR      GetBootMethod();
  DNS_STATUS ResetBootMethod(UCHAR fBootMethod);

  BOOL       IsServerConfigured();
  DNS_STATUS SetServerConfigured();

  BOOL       GetScavengingState();

  DWORD      GetScavengingInterval();
  DNS_STATUS ResetScavengingInterval(DWORD dwScavengingInterval);

  DWORD      GetDefaultRefreshInterval();
  DNS_STATUS ResetDefaultRefreshInterval(DWORD dwRefreshInterval);

  DWORD      GetDefaultNoRefreshInterval();
  DNS_STATUS ResetDefaultNoRefreshInterval(DWORD dwNoRefreshInterval);

  DWORD      GetDefaultScavengingState();
  DNS_STATUS ResetDefaultScavengingState(DWORD bScavengingState);

#ifdef USE_NDNC
  PCSTR     GetDomainName();
  PCSTR     GetForestName();
#endif

  DNS_STATUS ResetListenAddresses(DWORD cAddrCount, PIP_ADDRESS pipAddrs);
	void GetListenAddressesInfo(DWORD* pcAddrCount, PIP_ADDRESS* ppipAddrs);
	void GetServerAddressesInfo(DWORD* pcAddrCount, PIP_ADDRESS* ppipAddrs);

	DNS_STATUS ResetForwarders(DWORD cForwardersCount, PIP_ADDRESS pipForwarders, DWORD dwForwardTimeout, DWORD fSlave);
	void GetForwardersInfo(DWORD* pcForwardersCount, PIP_ADDRESS* ppipForwarders, DWORD* pdwForwardTimeout, DWORD* pfSlave);

	// Root Hints management API's
	BOOL HasRootHints() { return m_pRootHintsNode != NULL;}
	CDNSRootHintsNode* GetRootHints();
  void AttachRootHints(CDNSRootHintsNode* pNewRootHints);

	// testing options
	void GetTestOptions(CDNSServerTestOptions* pOptions);
	void ResetTestOptions(CDNSServerTestOptions* pOptions);

	BOOL IsTestEnabled() { return m_testOptions.m_bEnabled;}
	DWORD GetTestInterval() { return m_testOptions.m_dwInterval;}
	BOOL IsTestSimpleQueryEnabled() { return m_testOptions.m_bSimpleQuery;}
	BOOL IsRecursiveQueryEnabled() { return m_testOptions.m_bRecursiveQuery;}

	// test result manipulation
	void AddTestQueryResult(CDNSServerTestQueryResult* pTestResult,
							CComponentDataObject* pComponentData);

  void SetProppageStart(int nStartPage) { m_nStartProppage = nStartPage; }

  CDNSDomainForwardersNode* GetDomainForwardersNode() { return m_pDomainForwardersFolderNode; }

private:
	DNS_STATUS WriteDirtyZones();
	CDNSZoneNode* GetNewZoneNode();

	// server info manipulation
	CDNSServerInfoEx* m_pServInfoEx;

	void FreeServInfo();		// free memory
	DNS_STATUS GetServInfo();	// read info from server

	// root hints info
	CDNSRootHintsNode*		m_pRootHintsNode;

	void FreeRootHints();

	// server test query info
	CDNSServerTestOptions			m_testOptions;
	DWORD							m_dwTestTime;
	CDNSServerTestQueryResultList	m_testResultList;
	BOOL							m_bTestQueryPending;
  BOOL              m_bShowMessages;
  BOOL              m_bPrevQuerySuccess;

  int m_nStartProppage;
  BOOL  m_bLocalServer;

	// cached pointer to cache folder
	CDNSCacheNode*					  m_pCacheFolderNode;
	CDNSForwardZonesNode*			m_pFwdZonesFolderNode;
	CDNSReverseZonesNode*			m_pRevZonesFolderNode;
  CDNSDomainForwardersNode* m_pDomainForwardersFolderNode;

	friend class CDNSRootData;
	friend class CDNSServer_TestPropertyPage;

  DECLARE_TOOLBAR_MAP()
};


/////////////////////////////////////////////////////////////////////////
// CDNSServerOptionNode
// 
// represents the root hints in the UI as leaf node of the server node
// It serves no other function but to give access to the Root Hints page
// of the server properties

class CDNSServerOptionNode : public CLeafNode
{
public:
   CDNSServerOptionNode(UINT nDisplayNameID, UINT startPageCode);

   virtual void OnDelete(CComponentDataObject*, CNodeList*) { ASSERT(false); } // no delete allowed

   virtual LPCWSTR GetString(int nCol);
   virtual int     GetImageIndex(BOOL bOpenImage);

   virtual HRESULT OnSetToolbarVerbState(IToolbar* pToolbar, CNodeList*);

   virtual BOOL HasPropertyPages(DATA_OBJECT_TYPES type, 
                                 BOOL* pbHideVerb,
                                 CNodeList* pNodeList);
   virtual HRESULT CreatePropertyPages(LPPROPERTYSHEETCALLBACK lpProvider, 
                                       LONG_PTR handle,
                                       CNodeList* pNodeList);

   virtual BOOL DelegatesPPToContainer() { return TRUE; }

   virtual void ShowPageForNode(CComponentDataObject* pComponentDataObject);

private:
   UINT m_nStartPageCode;
};




#endif // _SERVER_H