//+----------------------------------------------------------------------------
//
// Copyright (C) 1992, Microsoft Corporation.
//
// File: PKT.H
//
// Contents: This module defines the data structures that make up the
// major internal part of the Partition Knowledge Table (PKT).
// The Partition Knowledge Table is used by the Dfs file
// system to resolve a name to a specific partition (volume).
//
// Functions:
//
// History: 1 May 1992 PeterCo Created.
//-----------------------------------------------------------------------------
#ifndef _PKT_
#define _PKT_
//
// Pick up marshalling info for gluons
//
#include "dfsgluon.h"
//
// New Prefix Table Implementation
//
#include "prefix.h"
//
// For determining the status of the Uid
//
#define GuidEqual(g1, g2)\
(RtlCompareMemory((g1), (g2), sizeof(GUID)) == sizeof(GUID))
extern GUID _TheNullGuid;
#define NullGuid( guid )\
(GuidEqual(guid, &_TheNullGuid))
//
// Defines for Create Dispositions (we mimic the io system here).
//
#define PKT_ENTRY_CREATE FILE_CREATE
#define PKT_ENTRY_REPLACE FILE_OPEN
#define PKT_ENTRY_SUPERSEDE FILE_SUPERSEDE
//
// Different kind of referrals that a DC can give out.
//
#define DFS_STORAGE_REFERRAL (0x0001)
#define DFS_REFERRAL_REFERRAL (0x0002)
//
// Types of service that can be supported by a provider.
// A disjunction of any of the following values.
//
#define DFS_SERVICE_TYPE_MASTER (0x0001)
#define DFS_SERVICE_TYPE_READONLY (0x0002)
#define DFS_SERVICE_TYPE_LOCAL (0x0004)
#define DFS_SERVICE_TYPE_REFERRAL (0x0008)
#define DFS_SERVICE_TYPE_ACTIVE (0x0010)
#define DFS_SERVICE_TYPE_DOWN_LEVEL (0x0020)
#define DFS_SERVICE_TYPE_COSTLIER (0x0040)
#define DFS_SERVICE_TYPE_OFFLINE (0x0080)
//
// The status possibilities...
//
#define DFS_SERVICE_STATUS_VERIFIED (0x0001)
#define DFS_SERVICE_STATUS_UNVERIFIED (0x0002)
//
// Types of Partition Knowledge Table Entries.
// Low-order bits in these definitions correspond to volume object
// types as defined in dfsh.idl. High-order bits are specific
// to PKT entries.
//
#define PKT_ENTRY_TYPE_DFS 0x0001 // Entry is to Dfs Aware srv
#define PKT_ENTRY_TYPE_MACHINE 0x0002 // Entry is a machine volume
#define PKT_ENTRY_TYPE_NONDFS 0x0004 // Entry refers to downlevel
#define PKT_ENTRY_TYPE_LEAFONLY 0x0008
#define PKT_ENTRY_TYPE_OUTSIDE_MY_DOM 0x0010 // Entry refers to volume in
// foreign domain
#define PKT_ENTRY_TYPE_SYSVOL 0x0040 // Sysvol
#define PKT_ENTRY_TYPE_REFERRAL_SVC 0x0080 // Entry refers to a DC
#define PKT_ENTRY_TYPE_PERMANENT 0x0100 // Entry cannot be scavenged
#define PKT_ENTRY_TYPE_DELETE_PENDING 0x0200 // Entry has pending delete
#define PKT_ENTRY_TYPE_LOCAL 0x0400 // Entry refers to local vol
#define PKT_ENTRY_TYPE_LOCAL_XPOINT 0x0800 // Entry refers to exit pt
#define PKT_ENTRY_TYPE_OFFLINE 0x2000 // Entry refers to a volume
// that is offline
//
// Type of messages the driver can send to DfsManager / DfsService
//
#define DFS_MSGTYPE_KNOW_INCONSISTENCY 0x0001
#define DFS_MSGTYPE_GET_DOMAIN_REFERRAL 0x0002
#define DFS_MSGTYPE_GET_DC_NAME 0x0003
//
// There is one DFS_MACHINE_ENTRY for every unique DS_MACHINE that the
// Dfs driver knows about. If a particular cairo server services multiple Dfs
// volumes, then multiple DFS_SERVICE structs will point to a single, ref
// counted DFS_MACHINE_ENTRY.
//
typedef struct DFS_MACHINE_ENTRY {
PDS_MACHINE pMachine; // The addressing info is here.
UNICODE_STRING MachineName; // The MachineName (principalName).
ULONG UseCount; // Number of DFS_SVC structs using this
ULONG ConnectionCount;// The number of pkt entries that are
// using this as their active machine
PFILE_OBJECT AuthConn; // Handle to the tree connect with
struct _DFS_CREDENTIALS *Credentials; // these credentials.
UNICODE_PREFIX_TABLE_ENTRY PrefixTableEntry;
} DFS_MACHINE_ENTRY, *PDFS_MACHINE_ENTRY;
//
// Marshalling info for DFS_MACHINE_ENTRY
//
extern MARSHAL_INFO MiMachineEntry;
#define INIT_DFS_MACHINE_ENTRY_MARSHAL_INFO() \
static MARSHAL_TYPE_INFO _MCode_Machine_Entry[] = { \
_MCode_pstruct(DFS_MACHINE_ENTRY, pMachine, &MiDSMachine) \
}; \
MARSHAL_INFO MiMachineEntry = _mkMarshalInfo(DFS_SERVICE, _MCode_Machine_Entry);
//
// A DFS_SERVICE structure is used to describe the provider and
// network address to be contacted for a specific partition of
// the distributed file system.
//
typedef struct _DFS_SERVICE {
ULONG Type; // type of service (see above)
ULONG Capability; // capability of this service
ULONG ProviderId; // identifies which provider
UNICODE_STRING Name; // service name (for authentication)
PFILE_OBJECT ConnFile; // FileObject for tree conn to IPC$ of server
struct _PROVIDER_DEF* pProvider; // pointer to provider definition
UNICODE_STRING Address; // network address
PDFS_MACHINE_ENTRY pMachEntry; // The addressing info is here.
ULONG Cost; // The site based cost of this service.
} DFS_SERVICE, *PDFS_SERVICE;
//
// Marshalling information for DFS_SERVICE
//
// NOTE: ConnFile and pProvider have significance only to the driver and
// are not marshalled.
//
extern MARSHAL_INFO MiService;
#define INIT_DFS_SERVICE_MARSHAL_INFO() \
static MARSHAL_TYPE_INFO _MCode_Service[] = { \
_MCode_ul(DFS_SERVICE, Type), \
_MCode_ul(DFS_SERVICE, Capability), \
_MCode_ul(DFS_SERVICE, ProviderId), \
_MCode_ustr(DFS_SERVICE, Name), \
_MCode_ustr(DFS_SERVICE, Address), \
_MCode_pstruct(DFS_SERVICE, pMachEntry, &MiMachineEntry) \
}; \
MARSHAL_INFO MiService = _mkMarshalInfo(DFS_SERVICE, _MCode_Service);
//
// Structure used in FSCTL_DFS_UPDATE_MACH_ADDRESS
//
typedef struct _DFS_MACHINE_INFO {
UNICODE_STRING MachineName; // Name of Machine (prefix alone).
PDS_MACHINE pMachine; // The new addressing info is here
} DFS_MACHINE_INFO, *PDFS_MACHINE_INFO;
//
// Marshalling info for DFS_MACHINE_INFO
//
extern MARSHAL_INFO MiDfsMachineInfo;
#define INIT_DFS_MACHINE_INFO() \
static MARSHAL_TYPE_INFO _MCode_MachineInfo[] = { \
_MCode_ustr(DFS_MACHINE_INFO, MachineName), \
_MCode_pstruct(DFS_MACHINE_INFO, pMachine, &MiDSMachine) \
}; \
MARSHAL_INFO MiDfsMachineInfo = \
_mkMarshalInfo(DFS_MACHINE_INFO, _MCode_MachineInfo);
//
// How a partition table entry is identified.
//
typedef struct _DFS_PKT_ENTRY_ID {
GUID Uid; // a unique identifier for this partition
UNICODE_STRING Prefix; // The entry path prefix of this partition
UNICODE_STRING ShortPrefix; // The 8.3 form of entry path prefix
} DFS_PKT_ENTRY_ID, *PDFS_PKT_ENTRY_ID;
//
// Marshalling information for DFS_PKT_ENTRY_ID
//
extern MARSHAL_INFO MiPktEntryId;
#define INIT_DFS_PKT_ENTRY_ID_MARSHAL_INFO() \
static MARSHAL_TYPE_INFO _MCode_PktEntryId[] = { \
_MCode_guid(DFS_PKT_ENTRY_ID, Uid), \
_MCode_ustr(DFS_PKT_ENTRY_ID, Prefix) \
}; \
MARSHAL_INFO MiPktEntryId = \
_mkMarshalInfo(DFS_PKT_ENTRY_ID, _MCode_PktEntryId);
//
// The guts of a partition table entry
//
typedef struct _DFS_PKT_ENTRY_INFO {
ULONG ServiceCount; // number of services in list
PDFS_SERVICE ServiceList; // array of servers that support the partition
} DFS_PKT_ENTRY_INFO, *PDFS_PKT_ENTRY_INFO;
typedef struct _DFS_TARGET_INFO_HEADER {
ULONG Type;
LONG UseCount;
ULONG Flags;
} DFS_TARGET_INFO_HEADER, *PDFS_TARGET_INFO_HEADER;
#define TARGET_INFO_DFS 1
#define TARGET_INFO_LMR 2
typedef struct _DFS_TARGET_INFO {
DFS_TARGET_INFO_HEADER DfsHeader;
union {
CREDENTIAL_TARGET_INFORMATIONW TargetInfo;
LMR_QUERY_TARGET_INFO LMRTargetInfo;
};
} DFS_TARGET_INFO, *PDFS_TARGET_INFO;
NTSTATUS
PktGetTargetInfo(
HANDLE IpcHandle,
PUNICODE_STRING pDomainName,
PUNICODE_STRING pShareName,
PDFS_TARGET_INFO *ppTargetInfo );
VOID
PktAcquireTargetInfo(
pTargetInfo);
VOID
PktReleaseTargetInfo(
pTargetInfo);
//
// Marshalling information for DFS_PKT_ENTRY_INFO
//
extern MARSHAL_INFO MiPktEntryInfo;
#define INIT_DFS_PKT_ENTRY_INFO_MARSHAL_INFO() \
static MARSHAL_TYPE_INFO _MCode_PktEntryInfo[] = { \
_MCode_ul(DFS_PKT_ENTRY_INFO, ServiceCount), \
_MCode_pcastruct(DFS_PKT_ENTRY_INFO,ServiceList,ServiceCount,&MiService)\
}; \
MARSHAL_INFO MiPktEntryInfo = \
_mkMarshalInfo(DFS_PKT_ENTRY_INFO, _MCode_PktEntryInfo);
//
// A Partition Knowledge Table Entry (PktEntry) identifies each known
// partition.
//
typedef struct _DFS_PKT_ENTRY {
NODE_TYPE_CODE NodeTypeCode; // node type -> DSFS_NTC_PKT_ENTRY
NODE_BYTE_SIZE NodeByteSize; // node size
LIST_ENTRY Link; // link for PKT EntryList
ULONG Type; // type of partition (see above)
ULONG USN; // Unique Serial Number
DFS_PKT_ENTRY_ID Id; // the Id of this entry
DFS_PKT_ENTRY_INFO Info; // info of this entry
ULONG ExpireTime; // time when partition should be deleted
ULONG TimeToLive; // time to keep this entry in the cache
ULONG UseCount; // # threads (DnrContexts) are looking at it.
ULONG FileOpenCount; // # of files opened via this entry
PDFS_TARGET_INFO pDfsTargetInfo;
PDFS_SERVICE ActiveService; // pointer into info to active service
PDFS_SERVICE LocalService; // pointer to local service (if any)
struct _DFS_PKT_ENTRY *Superior;// this entrys superior (if any)
ULONG SubordinateCount; // number of subordinates (if any)
LIST_ENTRY SubordinateList; // list of subordinates (if any)
LIST_ENTRY SiblingLink; // link to other siblings (if any)
struct _DFS_PKT_ENTRY *ClosestDC; // Link to closest superiorDC in PKT.
LIST_ENTRY ChildList; // Link to subordinate PKT entries (if any)
LIST_ENTRY NextLink; // Link to link up parent's Subord list.
UNICODE_PREFIX_TABLE_ENTRY PrefixTableEntry;// prefix table entry
} DFS_PKT_ENTRY, *PDFS_PKT_ENTRY;
//
// Marshalling information for DFS_PKT_ENTRY
//
// Note that we only marshal the id and the info...no relational info
//
//
extern MARSHAL_INFO MiPktEntry;
#define INIT_DFS_PKT_ENTRY_MARSHAL_INFO() \
static MARSHAL_TYPE_INFO _MCode_PktEntry[] = { \
_MCode_ul(DFS_PKT_ENTRY, Type), \
_MCode_struct(DFS_PKT_ENTRY, Id, &MiPktEntryId), \
_MCode_struct(DFS_PKT_ENTRY, Info, &MiPktEntryInfo) \
}; \
MARSHAL_INFO MiPktEntry = _mkMarshalInfo(DFS_PKT_ENTRY, _MCode_PktEntry);
//
// A special entry table contains a special name and a list of expanded
// names.
//
typedef struct _DFS_EXPANDED_NAME {
UNICODE_STRING ExpandedName; // expanded name itself
GUID Guid; // GUID associated with this name;
} DFS_EXPANDED_NAME, *PDFS_EXPANDED_NAME;
typedef struct _DFS_SPECIAL_ENTRY {
NODE_TYPE_CODE NodeTypeCode; // node type -> DSFS_NTC_SPECIAL_ENTRY
NODE_BYTE_SIZE NodeByteSize; // node size
LIST_ENTRY Link; // link for PKT SpecialEntryList
ULONG USN; // unique serial number
ULONG UseCount; // # threads (DnrContexts) are looking at it.
UNICODE_STRING SpecialName; // Special name itself
ULONG ExpandedCount; // count of expanded names
ULONG Active; // active expanded name
UNICODE_STRING DCName; // DC to go to, to get referrals
PDFS_EXPANDED_NAME ExpandedNames; // the expanded names
BOOLEAN NeedsExpansion; // need to expand this name
BOOLEAN Stale; // Entry has gone stale
BOOLEAN GotDCReferral; // This domain name has already
// been checked for
} DFS_SPECIAL_ENTRY, *PDFS_SPECIAL_ENTRY;
typedef struct _DFS_SPECIAL_TABLE {
LIST_ENTRY SpecialEntryList; // list of special entries in the PKT
ULONG SpecialEntryCount; // number of special entries in the PKT
ULONG TimeToLive; // time when table should be deleted
} DFS_SPECIAL_TABLE, *PDFS_SPECIAL_TABLE;
//
// A Parition Knowledge Table encapsulates all the knowledge that has
// been obtained about partitions in the distributed file system. There
// is only one instance of this struct in the entire system, and is
// part of the DsData structure.
//
typedef struct _DFS_PKT {
NODE_TYPE_CODE NodeTypeCode; // node type -> DSFS_NTC_PKT
NODE_BYTE_SIZE NodeByteSize; // node size...
ERESOURCE Resource; // resource to guard access to Pkt
KSPIN_LOCK UseCountLock; // Use while changing UseCount
ULONG EntryCount; // number of entries in the PKT
ULONG EntryTimeToLive; // time to live for each entry
LIST_ENTRY EntryList; // list of entries in the PKT
UNICODE_STRING DCName; // where to go for expanded referrals
UNICODE_STRING DomainNameFlat; // our domain name (flat)
UNICODE_STRING DomainNameDns; // our domain in dns format
DFS_SPECIAL_TABLE SpecialTable; // special entry table
DFS_PREFIX_TABLE PrefixTable; // prefix table
DFS_PREFIX_TABLE ShortPrefixTable; // prefix table for 8.3 names
UNICODE_PREFIX_TABLE DSMachineTable;// Table for DSMachines
} DFS_PKT, *PDFS_PKT;
#ifndef _UPKT_
//
// PARTITION KNOWLEDGE TABLE PUBLIC INLINE FUNCTIONS
//
#define _GetPkt() (&DfsData.Pkt)
//+-------------------------------------------------------------------------
//
// Function: PktAcquireShared, public inline
//
// Synopsis: PktAcquireShared acquires the partition knowledge table
// for shared access.
//
// Arguments: [WaitOk] - indicates if the call is allowed to wait for
// the PKT to become available or must return immediately
// [Result] - Pointer to boolean that will receive result of
// lock acquisition
//
// Returns: Nothing
//
// Notes: We first check to see if there are any threads waiting to
// update the Pkt. If so, we hold off until that thread has
// finished before we acquire a shared lock on the Pkt.
// Under NT, a thread waiting to acquire a resource exclusive
// does !not! automatically prevent threads from acquiring it
// shared. This is necessary to allow for recursive acquisition
// of resources. So, this event mechanism is required.
//
//--------------------------------------------------------------------------
#define PktAcquireShared( WaitOk, Result ) \
{ \
KeWaitForSingleObject( \
&DfsData.PktWritePending, \
Spare2, \
KernelMode, \
FALSE, \
NULL); \
*(Result) = ExAcquireResourceSharedLite( \
&DfsData.Pkt.Resource, \
WaitOk ); \
}
//+-------------------------------------------------------------------------
//
// Function: PktAcquireExclusive, public inline
//
// Synopsis: PktAcquireExclusive acquires the partition knowledge table
// for exclusive access.
//
// Arguments: [WaitOk] - indicates if the call is allowed to wait for
// the PKT to become available or must return immediately.
// [Result] - Pointer to boolean that will receive result of
// lock acquisition
//
// Returns: Nothing
//
// Notes:
//
//--------------------------------------------------------------------------
#define PktAcquireExclusive( WaitOk, Result ) \
{ \
KeResetEvent(&DfsData.PktWritePending); \
*(Result) = ExAcquireResourceExclusiveLite( \
&DfsData.Pkt.Resource, \
WaitOk ); \
}
//+-------------------------------------------------------------------------
//
// Function: PktRelease, public inline
//
// Synopsis: PktRelease releases the PKT. It can have been acquired
// for exclusive or shared access.
//
// Arguments: None
//
// Returns: Nothing
//
// Notes:
//
//--------------------------------------------------------------------------
#define PktRelease() \
{ \
ExReleaseResourceLite( &DfsData.Pkt.Resource ); \
KeSetEvent( \
&DfsData.PktWritePending, \
0, \
FALSE); \
}
//+----------------------------------------------------------------------------
//
// Function: PktConvertExclusiveToShared, public inline
//
// Synopsis: Converts an exclusive lock on Pkt to a shared lock
//
// Arguments: None
//
// Returns: Nothing
//
//-----------------------------------------------------------------------------
#define PktConvertExclusiveToShared() \
{ \
ExConvertExclusiveToSharedLite( &DfsData.Pkt.Resource ); \
KeSetEvent(&DfsData.PktWritePending, 0, FALSE); \
}
//+----------------------------------------------------------------------------
//
// Function: PKT_LOCKED_FOR_SHARED_ACCESS, public inline
//
// Synopsis: Returns TRUE if Pkt is locked for shared access, FALSE if not
//
// Arguments: None
//
// Returns: TRUE if Pkt is locked for shared access, FALSE otherwise
//
//-----------------------------------------------------------------------------
#define PKT_LOCKED_FOR_SHARED_ACCESS() \
( ExIsResourceAcquiredSharedLite( &DfsData.Pkt.Resource ) )
//+----------------------------------------------------------------------------
//
// Function: PKT_LOCKED_FOR_EXCLUSIVE_ACCESS, public inline
//
// Synopsis: Returns TRUE if Pkt is locked for exclusive access, FALSE if
// not
//
// Arguments: None
//
// Returns: TRUE if Pkt is locked for exclusive access, FALSE otherwise
//
//-----------------------------------------------------------------------------
#define PKT_LOCKED_FOR_EXCLUSIVE_ACCESS() \
( ExIsResourceAcquiredExclusiveLite( &DfsData.Pkt.Resource ) )
//+-------------------------------------------------------------------------
//
// Function: PktInvalidateEntry, public inline
//
// Synopsis: PktInvalidateEntry destroys a PKT Entry. The entry cannot
// be local, and it cannot be an exit point.
//
// Arguments: [Pkt] - pointer to an initialized PKT.
// [Victim] - pointer to the entry to be invalidated.
//
// Returns: [STATUS_SUCCESS] - all is well.
// [DFS_STATUS_LOCAL_ENTRY] - an attempt was made to
// invalidate a local entry, or an entry that is a
// local exit point.
//
// Notes:
//
//--------------------------------------------------------------------------
#define PktInvalidateEntry(p, e) ( \
((e)->Type & (PKT_ENTRY_TYPE_LOCAL|PKT_ENTRY_TYPE_LOCAL_XPOINT)) \
? (DFS_STATUS_LOCAL_ENTRY) \
: (PktEntryDestroy(e, p, (BOOLEAN)TRUE), STATUS_SUCCESS) \
)
//+-------------------------------------------------------------------------
//
// Function: PktFirstEntry, public inline
//
// Synopsis: PktFirstEntry returns the first entry in the list of
// PKT entries.
//
// Arguments: [Pkt] - pointer to an initialized PKT.
//
// Returns: A pointer to the first entry in the PKT, or NULL if the
// PKT is empty.
//
// Notes:
//
//--------------------------------------------------------------------------
#define PktFirstEntry(p) ( \
((p)->EntryList.Flink != &(p)->EntryList) \
? (CONTAINING_RECORD((p)->EntryList.Flink, DFS_PKT_ENTRY, Link)) \
: (NULL) \
)
//+-------------------------------------------------------------------------
//
// Function: PktNextEntry, public inline
//
// Synopsis: PktNextEntry returns the next entry in the list of
// PKT entries.
//
// Arguments: [Pkt] - pointer to an initialized PKT.
// [Entry] - a pointer to the previous entry.
//
// Returns: A pointer to the next entry in the PKT, or NULL if we
// are at the end of the list.
//
// Notes:
//
//--------------------------------------------------------------------------
#define PktNextEntry(p, e) ( \
((e)->Link.Flink != &(p)->EntryList) \
? (CONTAINING_RECORD((e)->Link.Flink, DFS_PKT_ENTRY, Link)) \
: (NULL) \
)
//+-------------------------------------------------------------------------
//
// Function: PktFirstSpecialEntry, public inline
//
// Synopsis: PktFirstSpecialEntry returns the first special entry in the list of
// PKT entries.
//
// Arguments: [Pkt] - pointer to an initialized PKT.
//
// Returns: A pointer to the first special entry in the PKT, or NULL if the
// PKT is empty.
//
// Notes:
//
//--------------------------------------------------------------------------
#define PktFirstSpecialEntry(p) ( \
((p)->SpecialEntryList.Flink != &(p)->SpecialEntryList) \
? (CONTAINING_RECORD((p)->SpecialEntryList.Flink, DFS_SPECIAL_ENTRY, Link)) \
: (NULL) \
)
//+-------------------------------------------------------------------------
//
// Function: PktNextSpecialEntry, public inline
//
// Synopsis: PktNextSpecialEntry returns the next special entry in the list of
// PKT entries.
//
// Arguments: [Pkt] - pointer to an initialized PKT.
// [Entry] - a pointer to the previous special entry.
//
// Returns: A pointer to the next special entry in the PKT, or NULL if we
// are at the end of the list.
//
// Notes:
//
//--------------------------------------------------------------------------
#define PktNextSpecialEntry(p, e) ( \
((e)->Link.Flink != &(p)->SpecialEntryList) \
? (CONTAINING_RECORD((e)->Link.Flink, DFS_SPECIAL_ENTRY, Link)) \
: (NULL) \
)
//+-------------------------------------------------------------------------
//
// Function: PktLinkEntry, public inline
//
// Synopsis: PktLinkEntry links an entry into the list of entries
// in the PKT.
//
// Arguments: [Pkt] - pointer to an initialized PKT.
// [Entry] - a pointer to the entry to be linked in.
//
// Returns: VOID
//
// Notes: Care must be taken to insure that the entry is not already
// linked into the PKT entry list before calling this routine.
// No checking is done to prevent an entry from being linked
// twice...
//
//--------------------------------------------------------------------------
#define PktLinkEntry(p, e) { \
InsertTailList(&(p)->EntryList, &(e)->Link); \
(p)->EntryCount++; \
}
//+-------------------------------------------------------------------------
//
// Function: PktUnlinkEntry, public inline
//
// Synopsis: PktUnlinkEntry unlinks an entry from the list of entries
// in the PKT.
//
// Arguments: [Pkt] - pointer to an initialized PKT.
// [Entry] - a pointer to the entry to be unlinked.
//
// Returns: VOID
//
// Notes: Care must be taken to insure that the entry is in fact
// already linked into the PKT entry list before calling this
// routine. No checking is done to prevent an entry from being
// unlinked twice...
//
//--------------------------------------------------------------------------
#define PktUnlinkEntry(p, e) { \
RemoveEntryList(&(e)->Link); \
(p)->EntryCount--; \
}
#define PktServiceListValidate(e) TRUE
//
// PARTITION KNOWLEDGE TABLE PUBLIC FUNCTIONS (pkt.c)
//
NTSTATUS
PktInitialize(
IN PDFS_PKT Pkt
);
VOID
PktUninitialize(
IN PDFS_PKT Pkt
);
NTSTATUS
PktCreateEntry(
IN PDFS_PKT Pkt,
IN ULONG EntryType,
IN PDFS_PKT_ENTRY_ID PktEntryId,
IN PDFS_PKT_ENTRY_INFO PktEntryInfo OPTIONAL,
IN ULONG CreateDisposition,
IN PDFS_TARGET_INFO pDfsTargetInfo,
OUT PDFS_PKT_ENTRY *ppPktEntry
);
NTSTATUS
PktCreateDomainEntry(
IN PUNICODE_STRING DomainName,
IN PUNICODE_STRING ShareName,
IN BOOLEAN CSCAgentCreate);
NTSTATUS
PktCreateEntryFromReferral(
IN PDFS_PKT Pkt,
IN PUNICODE_STRING ReferralPath,
IN ULONG ReferralSize,
IN PVOID ReferralBuffer,
IN ULONG CreateDisposition,
IN PDFS_TARGET_INFO pDfsTargetInfo,
OUT ULONG *MatchingLength,
OUT ULONG *ReferralType,
OUT PDFS_PKT_ENTRY *ppPktEntry
);
NTSTATUS
PktExpandSpecialEntryFromReferral(
IN PDFS_PKT Pkt,
IN PUNICODE_STRING ReferralPath,
IN ULONG ReferralSize,
IN PVOID ReferralBuffer,
IN PDFS_SPECIAL_ENTRY pSpecialEntry
);
VOID
PktSpecialEntryDestroy(
IN PDFS_SPECIAL_ENTRY pSpecialEntry
);
NTSTATUS
PktCreateSubordinateEntry(
IN PDFS_PKT Pkt,
IN PDFS_PKT_ENTRY Superior,
IN ULONG SubordinateType,
IN PDFS_PKT_ENTRY_ID SubordinateId,
IN PDFS_PKT_ENTRY_INFO SubordinateInfo OPTIONAL,
IN ULONG CreateDisposition,
IN OUT PDFS_PKT_ENTRY *Subordinate
);
PDFS_PKT_ENTRY
PktLookupEntryById(
IN PDFS_PKT Pkt,
IN PDFS_PKT_ENTRY_ID Id
);
PDFS_PKT_ENTRY
PktLookupEntryByPrefix(
IN PDFS_PKT Pkt,
IN PUNICODE_STRING Prefix,
OUT PUNICODE_STRING Remaining
);
PDFS_PKT_ENTRY
PktLookupEntryByShortPrefix(
IN PDFS_PKT Pkt,
IN PUNICODE_STRING Prefix,
OUT PUNICODE_STRING Remaining
);
NTSTATUS
PktEntryModifyPrefix(
IN PDFS_PKT Pkt,
IN PUNICODE_STRING LocalPath,
IN PDFS_PKT_ENTRY Entry
);
PDFS_PKT_ENTRY
PktLookupEntryByUid(
IN PDFS_PKT Pkt,
IN GUID *Uid
);
PDFS_PKT_ENTRY
PktLookupReferralEntry(
IN PDFS_PKT Pkt,
IN PDFS_PKT_ENTRY pEntry
);
PDFS_PKT_ENTRY
PktGetReferralEntryForPath(
PDFS_PKT Pkt,
UNICODE_STRING Path,
ULONG *Type
);
//
// DFS PKT PRIVATE FUNCTIONS (pkt.c)
//
NTSTATUS
PktpOpenDomainService(
IN PDFS_PKT Pkt,
IN PDFS_PKT_ENTRY PktEntry,
IN OUT PHANDLE DomainServiceHandle
);
NTSTATUS
DfsGetMachPktEntry(
UNICODE_STRING Path
);
VOID
RemoveLastComponent(
IN PUNICODE_STRING Prefix,
OUT PUNICODE_STRING newPrefix);
//
// DFS SERVICE PUBLIC FUNCTIONS (pktsup.c)
//
NTSTATUS
PktServiceConstruct(
OUT PDFS_SERVICE Service,
IN ULONG ServiceType,
IN ULONG ServiceCapability,
IN ULONG ServiceStatus,
IN ULONG ServiceProviderId,
IN PUNICODE_STRING ServiceName OPTIONAL,
IN PUNICODE_STRING ServiceAddress OPTIONAL
);
VOID
PktServiceDestroy(
IN PDFS_SERVICE Victim OPTIONAL,
IN BOOLEAN DeallocateAll
);
VOID
DfsDecrementMachEntryCount(
PDFS_MACHINE_ENTRY pMachEntry,
BOOLEAN DeallocateMachine
);
VOID
PktDSMachineDestroy(
IN PDS_MACHINE Victim OPTIONAL,
IN BOOLEAN DeallocateAll
);
VOID
PktDSTransportDestroy(
IN PDS_TRANSPORT Victim OPTIONAL,
IN BOOLEAN DeallocateAll
);
//
// PKT ENTRY ID PUBLIC FUNCTIONS (pktsup.c)
//
NTSTATUS
PktEntryIdConstruct(
OUT PDFS_PKT_ENTRY_ID PktEntryId,
IN GUID *Uid OPTIONAL,
IN UNICODE_STRING *Prefix OPTIONAL
);
VOID
PktEntryIdDestroy(
IN PDFS_PKT_ENTRY_ID Victim OPTIONAL,
IN BOOLEAN DeallocateAll
);
//
// PKT ENTRY ID PUBLIC INLINE FUNCTIONS
//
//+-------------------------------------------------------------------------
//
// Function: PktEntryIdEqual, public inline
//
// Synopsis: PktpEntryIdEqual determines if two entry Ids are equal
// or not.
//
// Arguments: [Id1] - a pointer to an Id to compare.
// [Id2] - a pointer to an Id to compare.
//
// Returns: [TRUE] - if both Ids are equal.
// [FALSE] - if the Ids are not equal.
//
// Notes: The comparison on the Prefix is done case insensitive.
//
//--------------------------------------------------------------------------
#define PktEntryIdEqual(Id1, Id2) ( \
(GuidEqual(&(Id1)->Uid, &(Id2)->Uid)) && \
(RtlEqualUnicodeString(&(Id1)->Prefix, &(Id2)->Prefix, (BOOLEAN)TRUE)) \
)
//
// PKT ENTRY ID PRIVATE INLINE FUNCTIONS
//
//+-------------------------------------------------------------------------
//
// Function: PktpEntryIdMove, private inline
//
// Synopsis: PktpEntryIdMove removes the values from the source Id and
// places them on the destination Id.
//
// Arguments: [DestId] - a pointer to an Id that is to receive the
// sources values.
// [SrcId] - a pointer to an Id that is to be stripped of
// its values.
//
// Returns: VOID
//
// Notes: Any values that are currently on the destination Id are
// overwritten. No memory is freed (either on the source,
// or the destination) by this call.
//
//--------------------------------------------------------------------------
#define PktpEntryIdMove(DestId, SrcId) { \
(*(DestId)) = (*(SrcId)); \
(SrcId)->Prefix.Length = (SrcId)->Prefix.MaximumLength = 0; \
(SrcId)->Prefix.Buffer = NULL; \
(SrcId)->ShortPrefix.Length = (SrcId)->ShortPrefix.MaximumLength = 0; \
(SrcId)->ShortPrefix.Buffer = NULL; \
}
//
// PKT ENTRY INFO PUBLIC FUNCTIONS (pktsup.c)
//
//NTSTATUS
//PktEntryInfoConstruct(
// OUT PDFS_PKT_ENTRY_INFO PktEntryInfo,
// IN PULONG ExpireTime OPTIONAL,
// IN ULONG ServiceCount,
// IN PDFS_SERVICE ServiceList OPTIONAL
// );
VOID
PktEntryInfoDestroy(
IN PDFS_PKT_ENTRY_INFO Victim OPTIONAL,
IN BOOLEAN DeallocateAll
);
//
// PKT ENTRY INFO PRIVATE INLINE FUNCTIONS
//
//
// The following inlines operate on Entry Infos
//
//+-------------------------------------------------------------------------
//
// Function: PktpEntryInfoMove, private inline
//
// Synopsis: PktpEntryInfoMove removes the values from the source Info and
// places them on the destination Info.
//
// Arguments: [DestInfo] - a pointer to an Info that is to receive the
// sources values.
// [SrcInfo] - a pointer to an Info that is to be stripped of
// its values.
//
// Returns: VOID
//
// Notes: Any values that are currently on the destination Info are
// overwritten. No memory is freed (either on the source,
// or the destination) by this call.
//
//--------------------------------------------------------------------------
#define PktpEntryInfoMove(DestInfo, SrcInfo) { \
(*(DestInfo)) = (*(SrcInfo)); \
(SrcInfo)->ServiceCount = 0L; \
(SrcInfo)->ServiceList = NULL; \
}
//
// PKT ENTRY PUBLIC INLINE FUNCTIONS
//
//+-------------------------------------------------------------------------
//
// Function: PktEntryFirstSubordinate, public inline
//
// Synopsis: PktEntryFirstSubordinate returns the first entry in the
// list of subordinates.
//
// Arguments: [Superior] - pointer to a PKT entry.
//
// Returns: A pointer to the first entry in the list of subordinates,
// or NULL if the list is empty.
//
// Notes:
//
//--------------------------------------------------------------------------
#define PktEntryFirstSubordinate(s) ( \
((s)->SubordinateList.Flink != &(s)->SubordinateList) \
? (CONTAINING_RECORD((s)->SubordinateList.Flink, DFS_PKT_ENTRY, \
SiblingLink)) \
: (NULL) \
)
//+-------------------------------------------------------------------------
//
// Function: PktEntryNextSubordinate, public inline
//
// Synopsis: PktEntryNextSubordinate returns the next entry in the
// list of subordinates.
//
// Arguments: [Superior] - pointer to the superior PKT entry (the one
// which we are getting the subordinates for).
// [Subordinate] - pointer to the last subordinate retreived
// via this routine (or PktEntryFirstSubordinate).
//
// Returns: A pointer to the next entry in the list of subordinates,
// or NULL if we've hit the end of the list.
//
// Notes:
//
//--------------------------------------------------------------------------
#define PktEntryNextSubordinate(s, m) ( \
((m)->SiblingLink.Flink != &(s)->SubordinateList) \
? (CONTAINING_RECORD((m)->SiblingLink.Flink,DFS_PKT_ENTRY,SiblingLink))\
: (NULL) \
)
//+-------------------------------------------------------------------------
//
// Function: PktEntryLinkSubordinate, public inline
//
// Synopsis: PktEntryLinkSubordinate links a subordinate to a superior's
// list of subordinates.
//
// Arguments: [Superior] - pointer to the superior PKT entry.
// [Subordinate] - pointer to the subordinate to be linked in.
//
// Returns: VOID
//
// Notes: If the subordinate is part of another superior's list, it
// will be removed from that list as a by-product of being
// put on the specified Superior's list. The Superior pointer
// of the subordinate is adjusted appropriately.
//
// If the superior is a local entry, the subordinate will
// be modified to indicate that it is a local exit point.
//
//--------------------------------------------------------------------------
#define PktEntryLinkSubordinate(sup, sub) { \
while(1) { \
if((sub)->Superior == (sup)) \
break; \
if((sub)->Superior != NULL) \
PktEntryUnlinkSubordinate((sub)->Superior, (sub)); \
InsertTailList(&(sup)->SubordinateList, &(sub)->SiblingLink); \
(sup)->SubordinateCount++; \
(sub)->Superior = (sup); \
if((sup)->Type & PKT_ENTRY_TYPE_LOCAL) \
(sub)->Type |= PKT_ENTRY_TYPE_LOCAL_XPOINT; \
break; \
} \
}
//+-------------------------------------------------------------------------
//
// Function: PktEntryUnlinkSubordinate, public inline
//
// Synopsis: PktEntryUnlinkSubordinate unlinks a subordinate from a
// superior's list of subordinates.
//
// Arguments: [Superior] - pointer to the superior PKT entry.
// [Subordinate] - pointer to the subordinate to be unlinked.
//
// Returns: VOID
//
// Notes: The Superior pointer of the subordinate is NULLed as a
// by-product of this operation.
//
// By default, the subordinate is modified to indicate that
// it is not an exit point (it cannot be an exit point if it
// has no superior).
//
// Milans - We need to turn off the PKT_ENTRY_TYPE_PERMANENT
// bit if the PKT_ENTRY_TYPE_LOCAL bit is not set, and we are
// not the DC. If we decide that a machine can be a server for
// a volume in another domain, then we need to do something
// about the != DS_DC clause.
//
//--------------------------------------------------------------------------
#define PktEntryUnlinkSubordinate(sup, sub) { \
ASSERT((sub)->Superior == (sup)); \
ASSERT((sup)->SubordinateCount > 0); \
RemoveEntryList(&(sub)->SiblingLink); \
(sup)->SubordinateCount--; \
(sub)->Superior = NULL; \
(sub)->Type &= ~PKT_ENTRY_TYPE_LOCAL_XPOINT; \
if ( DfsData.MachineState != DFS_ROOT_SERVER && \
(((sub)->Type & PKT_ENTRY_TYPE_LOCAL) == 0) ) { \
(sub)->Type &= ~PKT_ENTRY_TYPE_PERMANENT; \
} \
}
//
// The following set of inline functions work on the Links maintained in
// the PKT to be able to get to referral entries for interdomain stuff real
// fast. These functions are similar to the above functions.
//
//+-------------------------------------------------------------------------
//
// Function: PktEntryFirstChild, public inline
//
// Synopsis: PktEntryFirstChild returns the first entry in the
// list of child links of a PKT entry.
//
// Arguments: [SuperiorDC] - pointer to a PKT entry.
//
// Returns: A pointer to the first entry in the list of children,
// or NULL if the list is empty.
//
// Notes:
//
//--------------------------------------------------------------------------
#define PktEntryFirstChild(s) ( \
((s)->ChildList.Flink != &(s)->ChildList) \
? (CONTAINING_RECORD((s)->ChildList.Flink,DFS_PKT_ENTRY,NextLink)) \
: (NULL) \
)
//+-------------------------------------------------------------------------
//
// Function: PktEntryNextChild, public inline
//
// Synopsis: PktEntryNextChild returns the next entry in the
// list of children.
//
// Arguments: [Superior] - pointer to the superior PKT entry (the one
// which we are getting the subordinates for).
// [Subordinate] - pointer to the last child retreived
// via this routine (or PktEntryFirstChild).
//
// Returns: A pointer to the next entry in the list of children,
// or NULL if we've hit the end of the list.
//
// Notes:
//
//--------------------------------------------------------------------------
#define PktEntryNextChild(s, m) ( \
((m)->NextLink.Flink != &(s)->ChildList) \
? (CONTAINING_RECORD((m)->NextLink.Flink,DFS_PKT_ENTRY,NextLink)) \
: (NULL) \
)
//+-------------------------------------------------------------------------
//
// Function: PktEntryLinkChild, public inline
//
// Synopsis: PktEntryLinkChild links a child to a closestDC's
// list of children.
//
// Arguments: [Superior] - pointer to the superior PKT entry.
// [Subordinate] - pointer to the subordinate to be linked in.
//
// Returns: VOID
//
// Notes: If the child is part of another superior's list, it
// will be removed from that list as a by-product of being
// put on the specified Superior's list. The Superior pointer
// of the child is adjusted appropriately.
//
//
//--------------------------------------------------------------------------
#define PktEntryLinkChild(sup, sub) { \
while(1) { \
if (sub == sup) { \
(sub)->ClosestDC = NULL; \
break; \
} \
if((sub)->ClosestDC == (sup)) \
break; \
if((sub)->ClosestDC != NULL) \
PktEntryUnlinkChild((sub)->ClosestDC, (sub)); \
InsertTailList(&(sup)->ChildList, &(sub)->NextLink); \
(sub)->ClosestDC = (sup); \
break; \
} \
}
//+-------------------------------------------------------------------------
//
// Function: PktEntryUnlinkChild, public inline
//
// Synopsis: PktEntryUnlinkChild unlinks a child from a
// superior's list of children.
//
// Arguments: [Superior] - pointer to the superior PKT entry.
// [Subordinate] - pointer to the subordinate to be unlinked.
//
// Returns: VOID
//
// Notes: The Superior pointer of the child is NULLed as a
// by-product of this operation.
//
//
//--------------------------------------------------------------------------
#define PktEntryUnlinkChild(sup, sub) { \
ASSERT((sub)->ClosestDC == (sup)); \
RemoveEntryList(&(sub)->NextLink); \
(sub)->ClosestDC = NULL; \
}
//+-------------------------------------------------------------------------
//
// Function: PktEntryUnlinkAndRelinkChild, public inline
//
// Synopsis: PktEntryUnlinkAndRelinkChild unlinks a child from a
// superior's list of children and relinks it to the parent of
// the superior.
//
// Arguments: [Superior] - pointer to the superior PKT entry.
// [Subordinate] - pointer to the subordinate to be unlinked.
//
// Returns: VOID
//
//--------------------------------------------------------------------------
#define PktEntryUnlinkAndRelinkChild(sup, sub) { \
PktEntryUnlinkChild(sup, sub); \
if ((sup)->ClosestDC != NULL) { \
PktEntryLinkChild((sup)->ClosestDC, sub); \
} \
}
//
// PKT ENTRY PUBLIC FUNCTIONS (pktsup.c)
//
NTSTATUS
PktEntryAssemble(
IN OUT PDFS_PKT_ENTRY Entry,
IN PDFS_PKT Pkt,
IN ULONG EntryType,
IN PDFS_PKT_ENTRY_ID EntryId,
IN PDFS_PKT_ENTRY_INFO EntryInfo OPTIONAL,
IN PDFS_TARGET_INFO pDfsTargetInfo
);
NTSTATUS
PktEntryReassemble(
IN OUT PDFS_PKT_ENTRY Entry,
IN PDFS_PKT Pkt OPTIONAL,
IN ULONG EntryType,
IN PDFS_PKT_ENTRY_ID EntryId OPTIONAL,
IN PDFS_PKT_ENTRY_INFO EntryInfo OPTIONAL,
IN PDFS_TARGET_INFO pDfsTargetInfo
);
VOID
PktEntryDestroy(
IN PDFS_PKT_ENTRY Victim OPTIONAL,
IN PDFS_PKT Pkt,
IN BOOLEAN DeallocateAll
);
VOID
PktEntryClearSubordinates(
IN PDFS_PKT_ENTRY PktEntry
);
VOID
PktEntryClearChildren(
IN PDFS_PKT_ENTRY PktEntry
);
NTSTATUS
PktEntryCreateReferral(
IN PDFS_PKT_ENTRY PktEntry,
IN ULONG ServiceTypes,
IN PVOID ReferralBuffer
);
VOID
PktParsePath(
IN PUNICODE_STRING PathName,
OUT PUNICODE_STRING MachineName,
OUT PUNICODE_STRING ShareName,
OUT PUNICODE_STRING Remainder OPTIONAL
);
NTSTATUS
PktExpandSpecialName(
IN PUNICODE_STRING Name,
OUT PDFS_SPECIAL_ENTRY *ppSpecialEntry
);
PDFS_SPECIAL_ENTRY
PktLookupSpecialNameEntry(
PUNICODE_STRING Name
);
NTSTATUS
PktCreateSpecialNameEntry(
PDFS_SPECIAL_ENTRY pSpecialEntry
);
NTSTATUS
PktGetDCName(
ULONG Flags
);
NTSTATUS
PktGetSpecialReferralTable(
PUNICODE_STRING Machine,
BOOLEAN Type
);
NTSTATUS
PktCreateSpecialEntryTableFromReferral(
IN PDFS_PKT Pkt,
IN PUNICODE_STRING ReferralPath,
IN ULONG ReferralSize,
IN PVOID ReferralBuffer,
IN PUNICODE_STRING DCName
);
NTSTATUS
PktEntryFromSpecialEntry(
IN PDFS_SPECIAL_ENTRY pSpecialEntry,
IN PUNICODE_STRING pShareName,
OUT PDFS_PKT_ENTRY *ppPktEntry
);
PDFS_PKT_ENTRY
PktFindEntryByPrefix(
IN PDFS_PKT Pkt,
IN PUNICODE_STRING Prefix
);
#endif // NOT _UPKT_
#endif // _PKT_