//+---------------------------------------------------------------------------- // // 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_