1383 lines
47 KiB
C
1383 lines
47 KiB
C
//+----------------------------------------------------------------------------
|
||
//
|
||
// 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_
|
||
|