mirror of https://github.com/tongzx/nt5src
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
783 lines
36 KiB
783 lines
36 KiB
/*++ BUILD Version: 0009 // Increment this if a change has global effects
|
|
Copyright (c) 1987-1993 Microsoft Corporation
|
|
|
|
Module Name:
|
|
|
|
smbcedbp.h
|
|
|
|
Abstract:
|
|
|
|
This is the include file that defines all constants and types for
|
|
implementing the SMB mini redirector connection engine.
|
|
|
|
This module contains all the implementation details of the connection engine
|
|
data structures and should be included only by the implementation modules.
|
|
|
|
Revision History:
|
|
|
|
Balan Sethu Raman (SethuR) 06-Mar-95 Created
|
|
|
|
Notes:
|
|
|
|
--*/
|
|
|
|
#ifndef _SMBCEDBP_H_
|
|
#define _SMBCEDBP_H_
|
|
|
|
//
|
|
// There is reliance on the fact that SMBCEDB_OT_SENTINEL is the last entry in the
|
|
// enumerated type and the types have a range of values from 0. Please ensure that
|
|
// this is always true.
|
|
//
|
|
typedef struct _REFERENCE_RECORD_ {
|
|
PVOID FileName;
|
|
ULONG FileLine;
|
|
} REFERENCE_RECORD,*PREFERENCE_RECORD;
|
|
|
|
#define REFERENCE_RECORD_SIZE 20
|
|
|
|
typedef enum _SMBCEDB_OBJECT_TYPE {
|
|
SMBCEDB_OT_SERVER,
|
|
SMBCEDB_OT_NETROOT,
|
|
SMBCEDB_OT_SESSION,
|
|
SMBCEDB_OT_REQUEST,
|
|
SMBCEDB_OT_VNETROOTCONTEXT,
|
|
SMBCEDB_OT_SENTINEL,
|
|
SMBCEDB_OT_TRANSPORT
|
|
} SMBCEDB_OBJECT_TYPE, *PSMBCEDB_OBJECT_TYPE;
|
|
|
|
typedef enum _SMBCEDB_OBJECT_STATE_ {
|
|
SMBCEDB_ACTIVE, // the instance is in use
|
|
SMBCEDB_INVALID, // the instance has been invalidated/disconnected.
|
|
SMBCEDB_MARKED_FOR_DELETION, // the instance has been marked for deletion.
|
|
SMBCEDB_RECYCLE, // the instance is available for recycling
|
|
SMBCEDB_START_CONSTRUCTION, // Initiate construction.
|
|
SMBCEDB_CONSTRUCTION_IN_PROGRESS, // the instance construction is in progress
|
|
SMBCEDB_DESTRUCTION_IN_PROGRESS, // the instance destruction is in progress
|
|
SMBCEDB_RECOVER // the instance need to be recovered
|
|
} SMBCEDB_OBJECT_STATE, *PSMBCEDB_OBJECT_STATE;
|
|
|
|
typedef struct _SMBCE_OBJECT_HEADER_ {
|
|
union {
|
|
struct {
|
|
UCHAR ObjectType; // type of the object
|
|
UCHAR ObjectCategory; // Node type for debugging
|
|
};
|
|
USHORT NodeType;
|
|
};
|
|
UCHAR Flags; // flags associated with the object, This is implementation dependent
|
|
UCHAR Reserved; // padding
|
|
LONG SwizzleCount; // Number of swizzled references to this object
|
|
LONG State; // State of the object
|
|
} SMBCE_OBJECT_HEADER, *PSMBCE_OBJECT_HEADER;
|
|
|
|
typedef struct _SMBCE_SERVERS_LIST_ {
|
|
LIST_ENTRY ListHead;
|
|
} SMBCEDB_SERVERS, *PSMBCEDB_SERVERS;
|
|
|
|
typedef struct _SMBCEDB_SESSIONS_ {
|
|
LIST_ENTRY ListHead;
|
|
LIST_ENTRY DefaultSessionList;
|
|
} SMBCEDB_SESSIONS, *PSMBCEDB_SESSIONS;
|
|
|
|
typedef struct _SMBCEDB_NET_ROOTS_ {
|
|
LIST_ENTRY ListHead;
|
|
} SMBCEDB_NET_ROOTS, *PSMBCEDB_NET_ROOTS;
|
|
|
|
typedef struct _MRX_SMB_V_NET_ROOT_CONTEXTS {
|
|
LIST_ENTRY ListHead;
|
|
} SMBCE_V_NET_ROOT_CONTEXTS, *PSMBCE_V_NET_ROOT_CONTEXTS;
|
|
|
|
typedef struct _SMBCEDB_REQUESTS_ {
|
|
LIST_ENTRY ListHead;
|
|
SMB_MPX_ID NextRequestId;
|
|
} SMBCEDB_REQUESTS, *PSMBCEDB_REQUESTS;
|
|
|
|
typedef enum _SMBCEDB_SERVER_TYPE_ {
|
|
SMBCEDB_MAILSLOT_SERVER = 1,
|
|
SMBCEDB_FILE_SERVER = 2
|
|
} SMBCEDB_SERVER_TYPE, *PSMBCEDB_SERVER_TYPE;
|
|
|
|
//
|
|
// The SMBCEDB_SERVER_ENTRY is the data structure which encapsulates all the information
|
|
// w.r.t a remote server for the connection engine. This information includes the dialect
|
|
// details as well as the operational data structures required to communicate with the server.
|
|
//
|
|
// All the dialect related details are further encapsulated in SMBCE_SERVER while the operational
|
|
// data structures constitute the remaining parts of the server entry. A pointer to a
|
|
// SMBCEDB_SERVER_ENTRY instance is associated with every SRV_CALL that has been hooked
|
|
// onto this mini redirector by the wrapper. It is stored in the Context field of MRX_SRV_CALL.
|
|
//
|
|
// The operational information associated with a server entry includes the Transport related
|
|
// information, a collection of requests and a mechanism for associating MID's ( See SMB
|
|
// protocol spec.) and a mechanism for posting to threads ( WORK_QUEUE_ITEM ).
|
|
//
|
|
|
|
typedef struct _SMBCEDB_SERVER_ENTRY {
|
|
SMBCE_OBJECT_HEADER Header; // struct header.
|
|
LIST_ENTRY ServersList; // list of server instances.
|
|
PMRX_SRV_CALL pRdbssSrvCall;
|
|
UNICODE_STRING Name; // the server name.
|
|
UNICODE_STRING DomainName; // the server domain name.
|
|
SMBCEDB_SESSIONS Sessions; // the sessions associated with the server
|
|
SMBCEDB_NET_ROOTS NetRoots; // the net roots associated with the server.
|
|
SMBCE_V_NET_ROOT_CONTEXTS VNetRootContexts; // the V_NET_ROOT contexts
|
|
LIST_ENTRY ActiveExchanges; // list of exchanges active for this server
|
|
LIST_ENTRY ExpiredExchanges; // exchanges that have been timed out
|
|
RX_WORK_QUEUE_ITEM WorkQueueItem; // work queue item for posting
|
|
NTSTATUS ServerStatus; // the status of the server as determined by negotiate response
|
|
struct _SMBCE_TRANSPORT_ *PreferredTransport;
|
|
LONG TransportSpecifiedByUser; // ture if the connection is established on the tranport
|
|
// with the name specified
|
|
struct SMBCE_SERVER_TRANSPORT *pTransport;
|
|
struct SMBCE_SERVER_TRANSPORT *pMailSlotTransport;
|
|
|
|
SMBCEDB_REQUESTS MidAssignmentRequests;
|
|
SMBCEDB_REQUESTS OutstandingRequests;
|
|
PMID_ATLAS pMidAtlas;
|
|
struct _SMB_EXCHANGE *pNegotiateExchange;
|
|
SMBCE_SERVER Server; // the server data structure.
|
|
UNICODE_STRING DfsRootName;
|
|
UNICODE_STRING DnsName;
|
|
PVOID ConstructionContext; // debug only
|
|
KEVENT MailSlotTransportRundownEvent;
|
|
KEVENT TransportRundownEvent;
|
|
BOOLEAN IsTransportDereferenced; // prevent transport from being dereferenced more than once
|
|
BOOLEAN NegotiateInProgress; // a negotiate is in progress for this server
|
|
BOOLEAN SecuritySignaturesActive; // process the security signature if it is active
|
|
BOOLEAN SecuritySignaturesEnabled; // true if the security signature is required by either the
|
|
// client or server, and both have the capability.
|
|
BOOLEAN ExtSessionSetupInProgress; // a probe server is in progress for security signature
|
|
BOOLEAN ResumeRequestsInProgress;
|
|
RX_WORK_QUEUE_ITEM WorkQueueItemForResume; // work queue item for posting resume requests
|
|
SMBCEDB_REQUESTS SecuritySignatureSyncRequests; // requests waiting on extended session setup for security signature
|
|
REFERENCE_RECORD ReferenceRecord[REFERENCE_RECORD_SIZE]; // debug only
|
|
RX_WORK_QUEUE_ITEM WorkQueueItemForDisconnect; // work queue item for posting dereference server entry requests
|
|
BOOLEAN DisconnectWorkItemOutstanding; // is the disconnect work item in the queue?
|
|
RX_CONNECTION_ID ConnectionId;
|
|
} SMBCEDB_SERVER_ENTRY, *PSMBCEDB_SERVER_ENTRY;
|
|
|
|
// The SMBCEDB_NET_ROOT_ENTRY encapsulates all the information associated with a particular
|
|
// TREE_CONNECT ( Net use ) made on a server. As with the server entry this data structure
|
|
// encapsulates the dialect oriented details as well as the opertaional information
|
|
// associated with handling the requests on a net root.
|
|
//
|
|
// The dialect specific information is encapsulated in the SMBCE_NET_ROOT data structure. A
|
|
// pointer to an instance of this data structure is associated with every MRX_NET_ROOT call
|
|
// associated with a MRX_SRV_CALL hooked to this mini redirector.
|
|
|
|
// ********** code.improvement ******* The Name cache control structs should be replaced
|
|
// ************************************ with pointers to alloced structs so their size remains the
|
|
// ************************************ province of the wrapper.
|
|
//
|
|
typedef struct _SMBCEDB_NET_ROOT_ENTRY {
|
|
SMBCE_OBJECT_HEADER Header; // the struct header
|
|
LIST_ENTRY NetRootsList; // the list of net roots asssociated with a server
|
|
PMRX_NET_ROOT pRdbssNetRoot; // the associated net root ( purely as a debug aid )
|
|
PSMBCEDB_SERVER_ENTRY pServerEntry; // the associated server entry
|
|
struct _SMB_EXCHANGE *pExchange; // the exchange which is responsible for construction
|
|
SMBCEDB_REQUESTS Requests; // the pending requests for this net root
|
|
UNICODE_STRING Name;
|
|
ACCESS_MASK MaximalAccessRights;
|
|
ACCESS_MASK GuestMaximalAccessRights;
|
|
SMBCE_NET_ROOT NetRoot; // the net root data structure.
|
|
NAME_CACHE_CONTROL NameCacheCtlGFABasic; // The basic file information name cache control.
|
|
NAME_CACHE_CONTROL NameCacheCtlGFAStandard; // The standard file information name cache control.
|
|
NAME_CACHE_CONTROL NameCacheCtlGFAInternal; // The internal file information name cache control.
|
|
NAME_CACHE_CONTROL NameCacheCtlFNF; // The File not found name cache control.
|
|
PFILE_FS_VOLUME_INFORMATION VolumeInfo; // The FS Volume Information cache.
|
|
LONG VolumeInfoLength;
|
|
LARGE_INTEGER VolumeInfoExpiryTime;
|
|
BOOLEAN IsRemoteBoot;
|
|
} SMBCEDB_NET_ROOT_ENTRY, *PSMBCEDB_NET_ROOT_ENTRY;
|
|
|
|
// The SMBCEDB_SESSION_ENTRY encapsulates all the information associated with a session
|
|
// established to a remote machine. The session encapsulates all the security information.
|
|
// The dialect specific details are encapsulated in teh SMBCE_SESSION data structure. The
|
|
// SMBCE_SESSION data structure is available in many flavours depending on the security
|
|
// package used. Currently there is support for handling LSA and KERBEROS sessions.
|
|
//
|
|
// A pointer to an instance of this data structure is associated with every MRX_V_NET_ROOT
|
|
// data structure hooked to this mini redirector by the wrapper.
|
|
|
|
typedef struct _SMBCEDB_SESSION_ENTRY {
|
|
SMBCE_OBJECT_HEADER Header; // the struct header
|
|
LIST_ENTRY SessionsList; // the list of sessions associated with the server
|
|
LIST_ENTRY DefaultSessionLink; // the list of explicit credentials for this server
|
|
PSMBCEDB_SERVER_ENTRY pServerEntry; // the associated server entry
|
|
struct _SMB_EXCHANGE *pExchange; // the exchange which is responsible for construction
|
|
SMBCEDB_REQUESTS Requests; // pending requests
|
|
LIST_ENTRY SerializationList; // session construction serialization
|
|
PKEVENT pSerializationEvent;
|
|
ULONG SessionVCNumber; // the VC number to be packaged with session setup
|
|
SMBCE_SESSION Session; // the Session
|
|
PUNICODE_STRING pNetRootName; // for share level security only
|
|
BOOLEAN SessionRecoverInProgress;
|
|
} SMBCEDB_SESSION_ENTRY, *PSMBCEDB_SESSION_ENTRY;
|
|
|
|
//
|
|
// The wrapper exposes three data structures for manipulating and describing
|
|
// name spaces set up on remote servers, Viz., MRX_SRV_CALL, MRX_NET_ROOT and
|
|
// MRX_V_NET_ROOT. The SRV_CALL corresponds to a remote server, the MRX_NET_ROOT
|
|
// corresponds to a share on that machine and V_NET_ROOT encapsulates
|
|
// the notion of a view of a MRX_NET_ROOT ( share in SMB terminology)
|
|
//
|
|
// The mapping between the wrapper level data structures and the SMB notion
|
|
// of SMBCEDB_SERVER_ENTRY, SMBCEDB_SESSION_ENTRY and SMBCEDB_NET_ROOT_ENTRY
|
|
// is not one to one in all cases.
|
|
//
|
|
// It is one to one between MRX_SRV_CALL and SMBCEDB_SERVER_ENTRY. It is for this
|
|
// reason that a pointer to SMBCEDB_SERVER_ENTRY is stored in the context field
|
|
// of the MRX_SRV_CALL instance.
|
|
//
|
|
// SMBCEDB_SESSION_ENTRY has a one to one mapping with the set of credentials
|
|
// supplied to establish a connection to a server. Having established a session
|
|
// one can have access to all the shares available on the server.
|
|
//
|
|
// SMBCEDB_NET_ROOT_ENTRY has a one to one mapping with a share on a given
|
|
// server. Since this closely corresponds to the wrappers interpretation of
|
|
// MRX_NET_ROOT a pointer to SMBCEDB_NET_ROOT_ENTRY is stored as part of the
|
|
// MRX_NET_ROOT instance.
|
|
//
|
|
// The context associated with every MRX_V_NET_ROOT instance is a pointer to
|
|
// an instance of SMBCE_V_NET_ROOT_CONTEXT. This encapsulates the associated session
|
|
// entry, the net root entry and the relevant book keeping information.
|
|
//
|
|
// The bookkeeping information is the UID/TID used in the SMB protocol, a
|
|
// reference count and a LIST_ENTRY to thread the instance into the appropriate
|
|
// list.
|
|
//
|
|
|
|
#define SMBCE_V_NET_ROOT_CONTEXT_FLAG_VALID_TID (0x1)
|
|
#define SMBCE_V_NET_ROOT_CONTEXT_CSCAGENT_INSTANCE (0x2)
|
|
|
|
typedef struct _SMBCE_V_NET_ROOT_CONTEXT {
|
|
SMBCE_OBJECT_HEADER Header;
|
|
|
|
PMRX_V_NET_ROOT pRdbssVNetRoot; // the associated VNetRoot ( purely as a debug aid)
|
|
struct _SMB_EXCHANGE *pExchange; // the exchange which is responsible for construction
|
|
SMBCEDB_REQUESTS Requests;
|
|
|
|
LIST_ENTRY ListEntry;
|
|
LARGE_INTEGER ExpireTime;
|
|
|
|
struct _SMBCEDB_SERVER_ENTRY *pServerEntry;
|
|
struct _SMBCEDB_SESSION_ENTRY *pSessionEntry;
|
|
struct _SMBCEDB_NET_ROOT_ENTRY *pNetRootEntry;
|
|
|
|
USHORT Flags;
|
|
SMB_TREE_ID TreeId;
|
|
|
|
REFERENCE_RECORD ReferenceRecord[REFERENCE_RECORD_SIZE]; // debug only
|
|
} SMBCE_V_NET_ROOT_CONTEXT, *PSMBCE_V_NET_ROOT_CONTEXT;
|
|
|
|
//
|
|
// An SMBCEDB_REQUEST_ENTRY encapsulates an action being processed by the SMBCE connection
|
|
// engine. The requests come in vairous flavours and each of these flavours is associated
|
|
// with the appropriate context required for resumption. In order to provide better memory
|
|
// management mechanisms the REQUEST_ENTRY encapsulates a union of the requests of various
|
|
// flavours. Each SERVER_ENTRY in the connection engine is associated with a list or
|
|
// request entries. In order to hide the abstraction of a list which does not scale well to
|
|
// the case of GATEWAY redirectors a set of routines are provided to manipulate the
|
|
// collection of requests. They provide a mechanism for intializing the collection of requests,
|
|
// adding a request, deleting a request and enumeratiung requests in a collection.
|
|
//
|
|
// Special mechanisms are built in to handle batching of operations. Each operation on the
|
|
// collection of requests come in two flavours, a vanila version and a lite version. In the
|
|
// lite version it is assumed that the appropriate concurrency control action has been taken
|
|
//
|
|
// One common scenario that is often encountered in processing the requests is invocation
|
|
// of a specific function on the requests in the collection. As an example if a disconnect
|
|
// request is received on a server entry then all the outstanding requests must be resumed
|
|
// with the appropriate error. Since these indications can potentially occur at DPC levels in
|
|
// NT it is not desirable to manipulate the collection while holding onto a spinlock, nor is
|
|
// it desirable to repeatedly release and accquire the spin lock. A special operation is
|
|
// provided for transferring the requests enmasse from one collection to another and resetting
|
|
// the original. With the help of this operation it is sufficient to hold the spinlock only
|
|
// for the duration of the transfer. The remainder of the processing can be done on the newly
|
|
// created collection.
|
|
//
|
|
//
|
|
// NT Specific Implementation Note:
|
|
//
|
|
// On NT the transport indications are at DPC level, therefore it is required to protect
|
|
// the manipulation of the requests data structure with a spinlock.
|
|
//
|
|
//
|
|
|
|
typedef struct _SMBCEDB_REQUEST_ENTRY_ {
|
|
SMBCE_OBJECT_HEADER Header; // the struct header
|
|
LIST_ENTRY RequestsList; // the next request for the VC.
|
|
union {
|
|
SMBCE_GENERIC_REQUEST GenericRequest;
|
|
SMBCE_REQUEST Request; // the next request.
|
|
SMBCE_COPY_DATA_REQUEST CopyDataRequest;
|
|
SMBCE_RECONNECT_REQUEST ReconnectRequest;
|
|
SMBCE_MID_REQUEST MidRequest;
|
|
};
|
|
} SMBCEDB_REQUEST_ENTRY, *PSMBCEDB_REQUEST_ENTRY;
|
|
|
|
#define SmbCeInitializeRequests(pRequests) \
|
|
InitializeListHead(&(pRequests)->ListHead); \
|
|
(pRequests)->NextRequestId = 0
|
|
|
|
#define SmbCeAddRequestEntry(pRequestList,pRequestEntry) \
|
|
SmbCeAcquireSpinLock(); \
|
|
InsertTailList(&(pRequestList)->ListHead,&(pRequestEntry)->RequestsList); \
|
|
SmbCeReleaseSpinLock()
|
|
|
|
#define SmbCeAddRequestEntryLite(pRequestList,pRequestEntry) \
|
|
InsertTailList(&(pRequestList)->ListHead,&(pRequestEntry)->RequestsList)
|
|
|
|
#define SmbCeRemoveRequestEntry(pRequests,pEntry) \
|
|
SmbCeAcquireSpinLock(); \
|
|
RemoveEntryList(&(pEntry)->RequestsList); \
|
|
SmbCeReleaseSpinLock()
|
|
|
|
#define SmbCeRemoveRequestEntryLite(pRequests,pEntry) \
|
|
RemoveEntryList(&(pEntry)->RequestsList)
|
|
|
|
#define SmbCeGetFirstRequestEntry(pRequestList) \
|
|
(IsListEmpty(&(pRequestList)->ListHead) \
|
|
? NULL \
|
|
: (PSMBCEDB_REQUEST_ENTRY) \
|
|
(CONTAINING_RECORD((pRequestList)->ListHead.Flink, \
|
|
SMBCEDB_REQUEST_ENTRY, \
|
|
RequestsList)))
|
|
|
|
#define SmbCeGetNextRequestEntry(pRequestList,pRequestEntry) \
|
|
(((pRequestEntry)->RequestsList.Flink == &(pRequestList)->ListHead) \
|
|
? NULL \
|
|
: (PSMBCEDB_REQUEST_ENTRY) \
|
|
(CONTAINING_RECORD((pRequestEntry)->RequestsList.Flink, \
|
|
SMBCEDB_REQUEST_ENTRY, \
|
|
RequestsList)))
|
|
|
|
#define SmbCeTransferRequests(pDestination,pSource) \
|
|
if (IsListEmpty(&(pSource)->ListHead)) { \
|
|
SmbCeInitializeRequests((pDestination)); \
|
|
} else { \
|
|
*(pDestination) = *(pSource); \
|
|
(pDestination)->ListHead.Flink->Blink = &(pDestination)->ListHead; \
|
|
(pDestination)->ListHead.Blink->Flink = &(pDestination)->ListHead; \
|
|
SmbCeInitializeRequests((pSource)); \
|
|
}
|
|
|
|
|
|
// Much along the lines of a collection of request a collection of all server entries is
|
|
// maintained as part of the connection engine. The following operations are supported on
|
|
// the colection of server entries
|
|
// 1) adding a server entry to the collection
|
|
// 2) removing a server entry from the colection
|
|
// 3) enumerating the entries in the collection
|
|
//
|
|
// As in the case of the collection of requests all these operations come in two flavours
|
|
// the vanila version in which concurrency control is enforced and the lite version in
|
|
// which the concurrency control is left to the user's discretion.
|
|
|
|
#define SmbCeAddServerEntry(pServerEntry) \
|
|
SmbCeAcquireSpinLock(); \
|
|
InsertTailList(&s_DbServers.ListHead,&pServerEntry->ServersList); \
|
|
SmbCeReleaseSpinLock()
|
|
|
|
#define SmbCeAddServerEntryLite(pServerEntry) \
|
|
InsertTailList(&s_DbServers.ListHead,&pServerEntry->ServersList)
|
|
|
|
#define SmbCeRemoveServerEntry(pServerEntry) \
|
|
SmbCeAcquireSpinLock(); \
|
|
RemoveEntryList(&(pServerEntry)->ServersList); \
|
|
SmbCeReleaseSpinLock()
|
|
|
|
#define SmbCeRemoveServerEntryLite(pServerEntry) \
|
|
RemoveEntryList(&(pServerEntry)->ServersList)
|
|
|
|
#define SmbCeGetFirstServerEntry() \
|
|
(IsListEmpty(&s_DbServers.ListHead) \
|
|
? NULL \
|
|
: (PSMBCEDB_SERVER_ENTRY) \
|
|
(CONTAINING_RECORD(s_DbServers.ListHead.Flink, \
|
|
SMBCEDB_SERVER_ENTRY, \
|
|
ServersList)))
|
|
|
|
#define SmbCeGetNextServerEntry(pServerEntry) \
|
|
(((pServerEntry)->ServersList.Flink == &s_DbServers.ListHead) \
|
|
? NULL \
|
|
: (PSMBCEDB_SERVER_ENTRY) \
|
|
(CONTAINING_RECORD((pServerEntry)->ServersList.Flink, \
|
|
SMBCEDB_SERVER_ENTRY, \
|
|
ServersList)))
|
|
|
|
|
|
// Since the mapping between V_NET_ROOT's in the RDBSS and the session entries in the mini
|
|
// redirector is a many to one mapping a collection of session entries is maintained as part
|
|
// of each server entry. The following operations are supported on the collection of session
|
|
// entries
|
|
// 1) adding a session entry to the collection
|
|
// 2) removing a session entry from the colection
|
|
// 3) enumerating the entries in the collection
|
|
//
|
|
// As in the case of the collection of requests all these operations come in two flavours
|
|
// the vanila version in which concurrency control is enforced and the lite version in
|
|
// which the concurrency control is left to the user's discretion.
|
|
//
|
|
// In addition two more methods are specified for retrieving the default session entry and
|
|
// setting the default session entry for any given server.
|
|
|
|
#define SmbCeAddSessionEntry(pServerEntry,pSessionEntry) \
|
|
SmbCeAcquireSpinLock(); \
|
|
InsertTailList(&(pServerEntry)->Sessions.ListHead,&(pSessionEntry)->SessionsList); \
|
|
SmbCeReleaseSpinLock()
|
|
|
|
|
|
#define SmbCeAddSessionEntryLite(pServerEntry,pSessionEntry) \
|
|
InsertTailList(&(pServerEntry)->Sessions.ListHead,&(pSessionEntry)->SessionsList)
|
|
|
|
#define SmbCeRemoveSessionEntry(pServerEntry,pSessionEntry) \
|
|
SmbCeAcquireSpinLock(); \
|
|
if ((pSessionEntry)->DefaultSessionLink.Flink != NULL) { \
|
|
RemoveEntryList(&(pSessionEntry)->DefaultSessionLink); \
|
|
pSessionEntry->DefaultSessionLink.Flink = NULL; \
|
|
pSessionEntry->DefaultSessionLink.Blink = NULL; \
|
|
}; \
|
|
RemoveEntryList(&(pSessionEntry)->SessionsList); \
|
|
SmbCeReleaseSpinLock()
|
|
|
|
#define SmbCeRemoveSessionEntryLite(pServerEntry,pSessionEntry) \
|
|
ASSERT( SmbCeSpinLockAcquired() ); \
|
|
if ((pSessionEntry)->DefaultSessionLink.Flink != NULL) { \
|
|
RemoveEntryList(&(pSessionEntry)->DefaultSessionLink); \
|
|
pSessionEntry->DefaultSessionLink.Flink = NULL; \
|
|
pSessionEntry->DefaultSessionLink.Blink = NULL; \
|
|
}; \
|
|
RemoveEntryList(&(pSessionEntry)->SessionsList);
|
|
|
|
|
|
#define SmbCeGetFirstSessionEntry(pServerEntry) \
|
|
(IsListEmpty(&(pServerEntry)->Sessions.ListHead) \
|
|
? NULL \
|
|
: (PSMBCEDB_SESSION_ENTRY) \
|
|
(CONTAINING_RECORD((pServerEntry)->Sessions.ListHead.Flink, \
|
|
SMBCEDB_SESSION_ENTRY, \
|
|
SessionsList)))
|
|
|
|
#define SmbCeGetNextSessionEntry(pServerEntry,pSessionEntry) \
|
|
(((pSessionEntry)->SessionsList.Flink == \
|
|
&(pServerEntry)->Sessions.ListHead) \
|
|
? NULL \
|
|
: (PSMBCEDB_SESSION_ENTRY) \
|
|
(CONTAINING_RECORD((pSessionEntry)->SessionsList.Flink, \
|
|
SMBCEDB_SESSION_ENTRY, \
|
|
SessionsList)))
|
|
|
|
#define SmbCeSetDefaultSessionEntry(pServerEntry,pSessionEntry) \
|
|
SmbCeAcquireSpinLock(); \
|
|
if ((pSessionEntry)->DefaultSessionLink.Flink == NULL) { \
|
|
ASSERT( pSessionEntry->DefaultSessionLink.Blink == NULL ); \
|
|
InsertHeadList(&(pServerEntry)->Sessions.DefaultSessionList,&(pSessionEntry)->DefaultSessionLink); \
|
|
}; \
|
|
SmbCeReleaseSpinLock()
|
|
|
|
extern PSMBCEDB_SESSION_ENTRY
|
|
SmbCeGetDefaultSessionEntry(
|
|
PSMBCEDB_SERVER_ENTRY pServerEntry,
|
|
ULONG SessionId,
|
|
PLUID pLogonId
|
|
);
|
|
|
|
VOID
|
|
SmbCeRemoveDefaultSessionEntry(
|
|
PSMBCEDB_SESSION_ENTRY pDefaultSessionEntry
|
|
);
|
|
|
|
// In order to encapsulate the notion of reconnects and to provide for hot reconnects,
|
|
// i.e., reconnection attempts in which the saved state in the server/client prior to
|
|
// a transport level disconnect can be reused it is required to mark each net root
|
|
// entry associated with a server as invalid on receipt of a transport level disconnect.
|
|
//
|
|
// Therefore an abstraction of a collection of net root entries is provided and is associated
|
|
// with each server entry.
|
|
//
|
|
// The following operations are supported on the collection of net root entries
|
|
// 1) adding a net root entry to the collection
|
|
// 2) removing a net root entry from the colection
|
|
// 3) enumerating the entries in the collection
|
|
//
|
|
// As in the case of the collection of requests all these operations come in two flavours
|
|
// the vanila version in which concurrency control is enforced and the lite version in
|
|
// which the concurrency control is left to the user's discretion.
|
|
//
|
|
|
|
|
|
#define SmbCeAddNetRootEntry(pServerEntry,pNetRootEntry) \
|
|
SmbCeAcquireSpinLock(); \
|
|
InsertTailList(&(pServerEntry)->NetRoots.ListHead,&(pNetRootEntry)->NetRootsList); \
|
|
SmbCeReleaseSpinLock()
|
|
|
|
|
|
#define SmbCeAddNetRootEntryLite(pServerEntry,pNetRootEntry) \
|
|
InsertTailList(&(pServerEntry)->NetRoots.ListHead,&(pNetRootEntry)->NetRootsList)
|
|
|
|
#define SmbCeRemoveNetRootEntry(pServerEntry,pNetRootEntry) \
|
|
SmbCeAcquireSpinLock(); \
|
|
RemoveEntryList(&(pNetRootEntry)->NetRootsList); \
|
|
SmbCeReleaseSpinLock()
|
|
|
|
#define SmbCeRemoveNetRootEntryLite(pServerEntry,pNetRootEntry) \
|
|
RemoveEntryList(&(pNetRootEntry)->NetRootsList)
|
|
|
|
|
|
#define SmbCeGetFirstNetRootEntry(pServerEntry) \
|
|
(IsListEmpty(&(pServerEntry)->NetRoots.ListHead) \
|
|
? NULL \
|
|
: (PSMBCEDB_NET_ROOT_ENTRY) \
|
|
(CONTAINING_RECORD((pServerEntry)->NetRoots.ListHead.Flink, \
|
|
SMBCEDB_NET_ROOT_ENTRY, \
|
|
NetRootsList)))
|
|
|
|
#define SmbCeGetNextNetRootEntry(pServerEntry,pNetRootEntry) \
|
|
(((pNetRootEntry)->NetRootsList.Flink == \
|
|
&(pServerEntry)->NetRoots.ListHead) \
|
|
? NULL \
|
|
: (PSMBCEDB_NET_ROOT_ENTRY) \
|
|
(CONTAINING_RECORD((pNetRootEntry)->NetRootsList.Flink, \
|
|
SMBCEDB_NET_ROOT_ENTRY, \
|
|
NetRootsList)))
|
|
|
|
|
|
// Macros to manipulate the collection of SMBCE_V_NET_ROOT_CONTEXT instances.
|
|
|
|
#define SmbCeAddVNetRootContext(pVNetRootContexts,pVNetRootContext) \
|
|
SmbCeAcquireSpinLock(); \
|
|
InsertTailList(&(pVNetRootContexts)->ListHead,&(pVNetRootContext)->ListEntry); \
|
|
SmbCeReleaseSpinLock()
|
|
|
|
|
|
#define SmbCeAddVNetRootContextLite(pVNetRootContexts,pVNetRootContext) \
|
|
InsertTailList(&(pVNetRootContexts)->ListHead,&(pVNetRootContext)->ListEntry)
|
|
|
|
#define SmbCeRemoveVNetRootContext(pVNetRootContexts,pVNetRootContext) \
|
|
SmbCeAcquireSpinLock(); \
|
|
RemoveEntryList(&(pVNetRootContext)->ListEntry); \
|
|
SmbCeReleaseSpinLock()
|
|
|
|
#define SmbCeRemoveVNetRootContextLite(pVNetRootContexts,pVNetRootContext) \
|
|
RemoveEntryList(&(pVNetRootContext)->ListEntry)
|
|
|
|
|
|
#define SmbCeGetFirstVNetRootContext(pVNetRootContexts) \
|
|
(IsListEmpty(&((pVNetRootContexts)->ListHead)) \
|
|
? NULL \
|
|
: (PSMBCE_V_NET_ROOT_CONTEXT) \
|
|
(CONTAINING_RECORD((pVNetRootContexts)->ListHead.Flink, \
|
|
SMBCE_V_NET_ROOT_CONTEXT, \
|
|
ListEntry)))
|
|
|
|
#define SmbCeGetNextVNetRootContext(pVNetRootContexts,pVNetRootContext) \
|
|
(((pVNetRootContext)->ListEntry.Flink == \
|
|
&(pVNetRootContexts)->ListHead) \
|
|
? NULL \
|
|
: (PSMBCE_V_NET_ROOT_CONTEXT) \
|
|
(CONTAINING_RECORD((pVNetRootContext)->ListEntry.Flink, \
|
|
SMBCE_V_NET_ROOT_CONTEXT, \
|
|
ListEntry)))
|
|
|
|
|
|
//
|
|
// SmbCe database initialization
|
|
//
|
|
|
|
extern NTSTATUS
|
|
SmbCeDbInit();
|
|
|
|
extern VOID
|
|
SmbCeDbTearDown();
|
|
|
|
//
|
|
// Object allocation and deallocation
|
|
//
|
|
|
|
extern PSMBCE_OBJECT_HEADER
|
|
SmbCeDbAllocateObject(
|
|
SMBCEDB_OBJECT_TYPE ObjectType);
|
|
|
|
extern VOID
|
|
SmbCeDbFreeObject(
|
|
PVOID pObject);
|
|
|
|
//
|
|
// Object destruction
|
|
//
|
|
|
|
extern VOID
|
|
SmbCeTearDownServerEntry(PSMBCEDB_SERVER_ENTRY pServerEntry);
|
|
|
|
extern VOID
|
|
SmbCeTearDownNetRootEntry(PSMBCEDB_NET_ROOT_ENTRY pNetRootEntry);
|
|
|
|
extern VOID
|
|
SmbCeTearDownSessionEntry(PSMBCEDB_SESSION_ENTRY pSessionEntry);
|
|
|
|
extern VOID
|
|
SmbCeTearDownRequestEntry(PSMBCEDB_REQUEST_ENTRY pRequestEntry);
|
|
|
|
//
|
|
// The routines for mapping a MID with an exchange and for associating an exchange with
|
|
// a MID
|
|
//
|
|
|
|
extern NTSTATUS
|
|
SmbCeAssociateExchangeWithMid(
|
|
PSMBCEDB_SERVER_ENTRY pServerEntry,
|
|
struct _SMB_EXCHANGE *pExchange);
|
|
|
|
extern struct _SMB_EXCHANGE *
|
|
SmbCeMapMidToExchange(
|
|
PSMBCEDB_SERVER_ENTRY pServerEntry,
|
|
SMB_MPX_ID Mid);
|
|
|
|
extern NTSTATUS
|
|
SmbCeDissociateMidFromExchange(
|
|
PSMBCEDB_SERVER_ENTRY pServerEntry,
|
|
struct _SMB_EXCHANGE *pExchange);
|
|
|
|
extern struct _SMB_EXCHANGE *
|
|
SmbCeGetExchangeAssociatedWithBuffer(
|
|
PSMBCEDB_SERVER_ENTRY pServerEntry,
|
|
PVOID pBuffer);
|
|
|
|
extern NTSTATUS
|
|
SmbCeAssociateBufferWithExchange(
|
|
PSMBCEDB_SERVER_ENTRY pServerEntry,
|
|
struct _SMB_EXCHANGE * pExchange,
|
|
PVOID pBuffer);
|
|
|
|
extern VOID
|
|
SmbCePurgeBuffersAssociatedWithExchange(
|
|
PSMBCEDB_SERVER_ENTRY pServerEntry,
|
|
struct _SMB_EXCHANGE * pExchange);
|
|
|
|
extern NTSTATUS
|
|
SmbCepDiscardMidAssociatedWithExchange(
|
|
struct _SMB_EXCHANGE * pExchange);
|
|
|
|
extern VOID
|
|
SmbCeResumeDiscardedMidAssignmentRequests(
|
|
PSMBCEDB_REQUESTS pMidRequests,
|
|
NTSTATUS ResumptionStatus);
|
|
|
|
//
|
|
// Routines for handling transport disconnects/invalidation.
|
|
//
|
|
|
|
extern VOID
|
|
SmbCeTransportDisconnectIndicated(
|
|
PSMBCEDB_SERVER_ENTRY pServerEntry);
|
|
|
|
|
|
extern VOID
|
|
SmbCeResumeAllOutstandingRequestsOnError(
|
|
PSMBCEDB_SERVER_ENTRY pServerEntry);
|
|
|
|
extern VOID
|
|
SmbCeHandleTransportInvalidation(
|
|
struct _SMBCE_TRANSPORT_ *pTransport);
|
|
|
|
extern VOID
|
|
SmbCeFinalizeAllExchangesForNetRoot(
|
|
PMRX_NET_ROOT pNetRoot);
|
|
|
|
extern NTSTATUS
|
|
MRxSmbCheckForLoopBack(
|
|
IN PSMBCEDB_SERVER_ENTRY pServerEntry);
|
|
|
|
//
|
|
// Resource acquistion/release
|
|
//
|
|
|
|
PVOID SmbCeDbResourceAcquireFile;
|
|
ULONG SmbCeDbResourceAcquireLine;
|
|
|
|
#define SmbCeAcquireResource() \
|
|
KeEnterCriticalRegion(); \
|
|
ExAcquireResourceExclusiveLite(&s_SmbCeDbResource,TRUE);\
|
|
SmbCeDbResourceAcquireFile = __FILE__;\
|
|
SmbCeDbResourceAcquireLine = __LINE__
|
|
|
|
#define SmbCeReleaseResource() \
|
|
SmbCeDbResourceAcquireFile = NULL;\
|
|
SmbCeDbResourceAcquireLine = 0;\
|
|
ExReleaseResourceLite(&s_SmbCeDbResource);\
|
|
KeLeaveCriticalRegion()
|
|
|
|
#define SmbCeIsResourceOwned() ExIsResourceAcquiredExclusiveLite(&s_SmbCeDbResource)
|
|
|
|
#define SmbCeAcquireSpinLock() \
|
|
KeAcquireSpinLock(&s_SmbCeDbSpinLock,&s_SmbCeDbSpinLockSavedIrql); \
|
|
s_SmbCeDbSpinLockAcquired = TRUE
|
|
|
|
#define SmbCeReleaseSpinLock() \
|
|
s_SmbCeDbSpinLockAcquired = FALSE; \
|
|
KeReleaseSpinLock(&s_SmbCeDbSpinLock,s_SmbCeDbSpinLockSavedIrql)
|
|
|
|
#define SmbCeSpinLockAcquired() \
|
|
(s_SmbCeDbSpinLockAcquired == TRUE)
|
|
|
|
#define SmbCeAcquireSecuritySignatureResource() \
|
|
ExAcquireResourceExclusiveLite(&s_SmbSecuritySignatureResource,TRUE)
|
|
|
|
#define SmbCeReleaseSecuritySignatureResource() \
|
|
ExReleaseResourceLite(&s_SmbSecuritySignatureResource)
|
|
|
|
//INLINE BOOLEAN SmbCeDbIsEntryInUse(PSMBCE_OBJECT_HEADER pHeader)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
This routine determines if a SmbCe database entry is in use.
|
|
|
|
Arguments:
|
|
|
|
pHeader - the entry header
|
|
|
|
Return Value:
|
|
|
|
TRUE if the entry is in use otherwise FALSE
|
|
|
|
--*/
|
|
|
|
#define SmbCeIsEntryInUse(pHeader) \
|
|
(((PSMBCE_OBJECT_HEADER)(pHeader))->State == SMBCEDB_ACTIVE || \
|
|
((PSMBCE_OBJECT_HEADER)(pHeader))->State == SMBCEDB_INVALID || \
|
|
((PSMBCE_OBJECT_HEADER)(pHeader))->State == SMBCEDB_CONSTRUCTION_IN_PROGRESS)
|
|
|
|
|
|
#define SmbCeSetServerType(pServerEntry,ServerType) \
|
|
(pServerEntry)->Header.Flags = (UCHAR)(ServerType)
|
|
|
|
#define SmbCeGetServerType(pServerEntry) \
|
|
((SMBCEDB_SERVER_TYPE)(pServerEntry)->Header.Flags)
|
|
|
|
|
|
//
|
|
// Static variable declarations that constitute the SmbCe database.
|
|
//
|
|
|
|
extern SMBCEDB_SERVERS s_DbServers;
|
|
|
|
//
|
|
// Currently there is only one resource for synchronizing the access to all the
|
|
// entities in the connection engine database. It is possible to customize it
|
|
// subsequently since the acquistion/release methods take the type of the object
|
|
// as a parameter.
|
|
//
|
|
|
|
extern ERESOURCE s_SmbCeDbResource;
|
|
extern RX_SPIN_LOCK s_SmbCeDbSpinLock;
|
|
extern KIRQL s_SmbCeDbSpinLockSavedIrql;
|
|
extern BOOLEAN s_SmbCeDbSpinLockAcquired;
|
|
|
|
#endif // _SMBCEDBP_H_
|
|
|
|
|
|
|
|
|