public static void ReconnectOpenRequest(
DurableV1ReconnectContext durableV1ReconnectContext,
DurableV2ReconnectContext durableV2ReconnectContext,
OplockLeaseType oplockLeaseType,
LeaseKeyDifferentialType leaseKeyDifferentialType,
ClientIdType clientIdType,
CreateGuidType createGuidType)
{
// If lease does not exist, LeaseKeyDifferentialType and LeaseStateDifferentialType take no effect. So just fix their values.
Condition.IfThen(oplockLeaseType != OplockLeaseType.LeaseV1 && oplockLeaseType != OplockLeaseType.LeaseV2,
leaseKeyDifferentialType == LeaseKeyDifferentialType.SameLeaseKey);
// If there is no durablev2 reconnect context, fix the value of CreateGuid.
Condition.IfThen(durableV2ReconnectContext == DurableV2ReconnectContext.DurableV2ReconnectContextNotExist, createGuidType == CreateGuidType.SameCreateGuid);
// Do not need to test the situation of both V1 and V2 reconnect contexts not existed.
Condition.IsFalse(durableV1ReconnectContext == DurableV1ReconnectContext.DurableV1ReconnectContextNotExist
&& durableV2ReconnectContext == DurableV2ReconnectContext.DurableV2ReconnectContextNotExist);
// Only the server implements the SMB 2.1 or SMB 3.x dialect family supports leasing
Condition.IfThen(NegotiateDialect == DialectRevision.Smb2002, oplockLeaseType != OplockLeaseType.LeaseV1 && oplockLeaseType != OplockLeaseType.LeaseV2);
// Lease V2 is only vaild for SMB 3.x dialect family.
Condition.IfThen(NegotiateDialect == DialectRevision.Smb21, oplockLeaseType != OplockLeaseType.LeaseV2);
// If leasing is not supported, do not test LeaseV1 or LeaseV2.
Condition.IfThen(!Config.IsLeasingSupported,
oplockLeaseType != OplockLeaseType.LeaseV1 && oplockLeaseType != OplockLeaseType.LeaseV2);
// 3.3.5.9.10, 3.3.5.9.12 This section applies only to servers that implement the SMB 3.x dialect family.
Condition.IfThen(!ModelUtility.IsSmb3xFamily(Config.MaxSmbVersionSupported), durableV2ReconnectContext == DurableV2ReconnectContext.DurableV2ReconnectContextNotExist);
// Add restriction to limit the generated cases of model.
Combination.Pairwise(durableV1ReconnectContext, durableV2ReconnectContext, oplockLeaseType, leaseKeyDifferentialType, clientIdType, createGuidType);
Condition.IsNull(Request);
Request = new ModelOpenFileRequest(
DurableV1RequestContext.DurableV1RequestContextNotExist,
DurableV2RequestContext.DurableV2RequestContextNotExist,
durableV1ReconnectContext,
durableV2ReconnectContext,
oplockLeaseType,
leaseKeyDifferentialType == LeaseKeyDifferentialType.SameLeaseKey,
clientIdType == ClientIdType.SameClient,
createGuidType == CreateGuidType.SameCreateGuid);
}
public static void CreateRequest(
ModelDialectRevision maxSmbVersionClientSupported,
ReplayModelShareType shareType,
ReplayModelClientSupportPersistent clientPersistentCapability,
ReplayModelSwitchChannelType switchChannelType,
ReplayModelChannelSequenceType channelSequence,
ReplayModelSetReplayFlag isSetReplayFlag,
ReplayModelDurableHandle modelDurableHandle,
ReplayModelRequestedOplockLevel requestedOplockLevel,
ReplayModelLeaseState leaseState,
ReplayModelFileName fileName,
ReplayModelCreateGuid createGuid,
ReplayModelFileAttributes fileAttributes,
ReplayModelCreateDisposition createDisposition,
ReplayModelLeaseKey leaseKey)
{
Condition.IsTrue(State == ModelState.Connected || State == ModelState.Initialized);
Condition.IsNull(Request);
Combination.NWise(2, maxSmbVersionClientSupported, shareType, clientPersistentCapability, switchChannelType, channelSequence, isSetReplayFlag, modelDurableHandle,
requestedOplockLevel, leaseState, fileName, createGuid, fileAttributes, createDisposition, leaseKey);
Combination.Isolated(channelSequence == ReplayModelChannelSequenceType.InvalidChannelSequence);
Combination.Isolated(channelSequence == ReplayModelChannelSequenceType.ChannelSequenceBoundaryValid);
//No lease for non durable handle
Condition.IfThen(modelDurableHandle == ReplayModelDurableHandle.NormalHandle,
requestedOplockLevel != ReplayModelRequestedOplockLevel.OplockLevelLeaseV1 &&
requestedOplockLevel != ReplayModelRequestedOplockLevel.OplockLevelLeaseV2);
//No lease state or lease key(default) if not requesting lease
Condition.IfThen(
requestedOplockLevel != ReplayModelRequestedOplockLevel.OplockLevelLeaseV1 &&
requestedOplockLevel != ReplayModelRequestedOplockLevel.OplockLevelLeaseV2,
leaseState == ReplayModelLeaseState.LeaseStateIsNone && leaseKey == ReplayModelLeaseKey.DefaultLeaseKey);
//Ensure valid lease state if requesting lease
Condition.IfThen(
requestedOplockLevel == ReplayModelRequestedOplockLevel.OplockLevelLeaseV1 ||
requestedOplockLevel == ReplayModelRequestedOplockLevel.OplockLevelLeaseV2,
leaseState != ReplayModelLeaseState.LeaseStateIsNone);
if (State == ModelState.Initialized)
{
Condition.IsTrue(switchChannelType == ReplayModelSwitchChannelType.MainChannel);
DialectRevision negotiateDialect = ModelHelper.DetermineNegotiateDialect(maxSmbVersionClientSupported,
Config.MaxSmbVersionSupported);
MainChannel = new ModelReplayChannel(negotiateDialect, shareType,
clientPersistentCapability == ReplayModelClientSupportPersistent.ClientSupportPersistent);
}
else // There's already an open exist
{
//Open eixsts
Condition.IsTrue(Open != null);
// Ensure same dialect
Condition.IsTrue(ModelUtility.GetDialectRevision(maxSmbVersionClientSupported) == MainChannel.Connection_NegotiateDialect);
// Ensure connect to same share
Condition.IsTrue(shareType == MainChannel.Connection_Session_TreeConnect_Share_IsCA);
switch (switchChannelType)
{
case ReplayModelSwitchChannelType.MainChannel:
{
Condition.IsTrue((clientPersistentCapability ==
ReplayModelClientSupportPersistent.ClientSupportPersistent) ==
MainChannel.Connection_ClientCapabilities_SupportPersistent);
break;
}
case ReplayModelSwitchChannelType.ReconnectMainChannel:
{
MainChannel.Connection_ClientCapabilities_SupportPersistent =
(clientPersistentCapability == ReplayModelClientSupportPersistent.ClientSupportPersistent);
break;
}
case ReplayModelSwitchChannelType.AlternativeChannelWithDisconnectMainChannel:
case ReplayModelSwitchChannelType.AlternativeChannelWithMainChannel:
case ReplayModelSwitchChannelType.MainChannelWithAlternativeChannel:
{
AlternativeChannel = new ModelReplayChannel(MainChannel.Connection_NegotiateDialect, shareType,
clientPersistentCapability == ReplayModelClientSupportPersistent.ClientSupportPersistent);
break;
}
}
#region Handle state changes when there's connection lost or reconnection
if (switchChannelType == ReplayModelSwitchChannelType.AlternativeChannelWithDisconnectMainChannel ||
switchChannelType == ReplayModelSwitchChannelType.ReconnectMainChannel)
{
bool isPreserved = false;
ModelHelper.Log(
LogType.Requirement,
"3.3.7.1: The server MUST iterate over the Session.OpenTable and determine whether each Open is to be preserved for reconnect." +
" If any of the following conditions is satisfied, it indicates that the Open is to be preserved for reconnect");
ModelHelper.Log(
LogType.TestInfo,
"Open.OplockLevel is {0}, Open.OplockState is {1}, Open.IsDurable {2}", Open.OplockLevel,
Open.OplockStateIsHeld ? "Held" : "not Held", Open.IsDurable);
if (Open.Lease != null)
{
ModelHelper.Log(
LogType.TestInfo,
"LeaseState is {0}", Open.Lease.LeaseState);
}
if (switchChannelType == ReplayModelSwitchChannelType.AlternativeChannelWithDisconnectMainChannel)
{
ModelHelper.Log(
LogType.TestInfo,
"TD does not states the open is preserved when multi channels exist.");
isPreserved = true;
}
else if (Open.OplockLevel == ReplayModelRequestedOplockLevel.OplockLevelBatch &&
Open.OplockStateIsHeld &&
Open.IsDurable) // Open.OplockLevel is equal to SMB2_OPLOCK_LEVEL_BATCH
{
ModelHelper.Log(
LogType.Requirement,
"Open.OplockLevel is equal to SMB2_OPLOCK_LEVEL_BATCH and Open.OplockState is equal to Held, and Open.IsDurable is TRUE");
isPreserved = true;
ModelHelper.Log(
LogType.TestInfo,
"Open is to be preserved for reconnect");
}
else if (((Open.OplockLevel == ReplayModelRequestedOplockLevel.OplockLevelLeaseV1 ||
Open.OplockLevel == ReplayModelRequestedOplockLevel.OplockLevelLeaseV2) &&
Open.Lease != null &&
Open.Lease.LeaseState == ReplayModelLeaseState.LeaseStateIncludeH &&
Open.OplockStateIsHeld &&
Open.IsDurable)) // Open.OplockLevel is equal to SMB2_OPLOCK_LEVEL_LEASE and contains SMB2_LEASE_HANDLE_CACHING
{
ModelHelper.Log(
LogType.Requirement,
"Open.OplockLevel is equal to SMB2_OPLOCK_LEVEL_LEASE, Lease.LeaseState contains SMB2_LEASE_HANDLE_CACHING," +
" Open.OplockState is equal to Held, and Open.IsDurable is TRUE.");
isPreserved = true;
ModelHelper.Log(
LogType.TestInfo,
"Open is to be preserved for reconnect");
}
// "The server supports leasing and Open.IsResilient is TRUE" covered by resilient model
if (isPreserved)
{
ModelHelper.Log(
LogType.Requirement,
"If the Open is to be preserved for reconnect, perform the following actions:");
ModelHelper.Log(
LogType.Requirement,
"Set Open.Connection to NULL");
//Skip requirement for resilient handle which covered by resilient model
Open.ConnectionIsNotNull = false;
}
else
{
ModelHelper.Log(
LogType.Requirement,
"If the Open is not to be preserved for reconnect, the server MUST close the Open as specified in section 3.3.4.17");
LastOpen = Open;
Open = null;
ModelHelper.Log(
LogType.TestInfo,
"Open is set to NULL");
}
}
#endregion
}
ModelReplayChannel channel = GetChannel(switchChannelType);
ModelReplayCreateRequest replayCreateRequest =
new ModelReplayCreateRequest(channel, switchChannelType, channelSequence, modelDurableHandle, requestedOplockLevel, fileName,
createGuid, fileAttributes, createDisposition, leaseState, isSetReplayFlag, leaseKey);
Request = replayCreateRequest;
State = ModelState.Connected;
}
public static void OpenRequest(
ModelDialectRevision clientMaxDialect,
PersistentBitType persistentBit,
CAShareType connectToCAShare,
OplockLeaseType oplockLeaseType,
DurableV1RequestContext durableV1RequestContext,
DurableV2RequestContext durableV2RequestContext,
DurableV1ReconnectContext durableV1ReconnectContext,
DurableV2ReconnectContext durableV2ReconnectContext)
{
Condition.IsNull(Request);
Condition.IsNull(Open);
// CAShare, Persistent Handle and Durable Handle V2 are only applied for SMB 3.x family.
Condition.IfThen(!ModelUtility.IsSmb3xFamily(Config.MaxSmbVersionSupported), connectToCAShare == CAShareType.NonCAShare);
Condition.IfThen(!ModelUtility.IsSmb3xFamily(Config.MaxSmbVersionSupported), persistentBit == PersistentBitType.PersistentBitNotSet);
Condition.IfThen(!ModelUtility.IsSmb3xFamily(Config.MaxSmbVersionSupported), durableV2RequestContext == DurableV2RequestContext.DurableV2RequestContextNotExist);
Condition.IfThen(!ModelUtility.IsSmb3xFamily(Config.MaxSmbVersionSupported), durableV2ReconnectContext == DurableV2ReconnectContext.DurableV2ReconnectContextNotExist);
Condition.IfThen(!ModelUtility.IsSmb3xFamily(Config.MaxSmbVersionSupported), oplockLeaseType != OplockLeaseType.LeaseV2);
// If leasing is not supported, do not test LeaseV1 or LeaseV2.
Condition.IfThen(!Config.IsLeasingSupported, oplockLeaseType != OplockLeaseType.LeaseV1 && oplockLeaseType != OplockLeaseType.LeaseV2);
Combination.Pairwise(
clientMaxDialect,
persistentBit,
connectToCAShare,
oplockLeaseType,
durableV1RequestContext,
durableV2RequestContext,
durableV1ReconnectContext,
durableV2ReconnectContext);
NegotiateDialect = ModelHelper.DetermineNegotiateDialect(clientMaxDialect, Config.MaxSmbVersionSupported);
Share_IsCA = (connectToCAShare == CAShareType.CAShare);
if (ModelUtility.IsSmb3xFamily(NegotiateDialect)
&& persistentBit == PersistentBitType.PersistentBitSet
&& Config.IsPersistentHandleSupported)
{
ModelHelper.Log(LogType.Requirement,
"3.3.5.4: The Capabilities field MUST be set to a combination of zero or more of the following bit values, as specified in section 2.2.4:");
ModelHelper.Log(LogType.Requirement,
"\tSMB2_GLOBAL_CAP_PERSISTENT_HANDLES if Connection.Dialect belongs to the SMB 3.x dialect family, " +
"SMB2_GLOBAL_CAP_PERSISTENT_HANDLES is set in the Capabilities field of the request, and the server supports persistent handles.");
ModelHelper.Log(LogType.TestInfo, "All the above conditions are met. So SMB2_DHANDLE_FLAG_PERSISTENT bit is set in Connection.ServerCapabilities.");
ServerCapabilities_PersistentBitSet = true;
}
Request = new ModelOpenFileRequest(
durableV1RequestContext,
durableV2RequestContext,
durableV1ReconnectContext,
durableV2ReconnectContext,
oplockLeaseType,
false,
false,
false);
}
public static void FileOperationRequest(
ReplayModelSwitchChannelType switchChannelType,
ModelDialectRevision maxSmbVersionClientSupported,
ReplayModelRequestCommand requestCommand,
ReplayModelChannelSequenceType channelSequence,
ReplayModelSetReplayFlag isSetReplayFlag,
ReplayModelRequestCommandParameters requestCommandParameters)
{
Condition.IsNull(Request);
Condition.IfThen(ModelRequestCommand != ReplayModelRequestCommand.NoRequest, ModelRequestCommand == requestCommand);
Condition.IsTrue(requestCommand != ReplayModelRequestCommand.Create && requestCommand != ReplayModelRequestCommand.NoRequest);
Condition.IsTrue(switchChannelType != ReplayModelSwitchChannelType.ReconnectMainChannel);
Combination.NWise(2, switchChannelType, maxSmbVersionClientSupported, requestCommand, channelSequence, isSetReplayFlag,
requestCommandParameters);
ModelReplayChannel channel = null;
if (State == ModelState.Initialized)
{
Condition.IsTrue(switchChannelType == ReplayModelSwitchChannelType.MainChannel);
Condition.IsTrue(requestCommandParameters == ReplayModelRequestCommandParameters.DefaultParameters);
MainChannel =
new ModelReplayChannel(ModelHelper.DetermineNegotiateDialect(maxSmbVersionClientSupported,
Config.MaxSmbVersionSupported));
Open = new ReplayOpen();
channel = MainChannel;
}
else
{
Condition.IsNotNull(MainChannel);
Condition.IsNotNull(Open);
Condition.IsTrue(maxSmbVersionClientSupported == ModelUtility.GetModelDialectRevision(MainChannel.Connection_NegotiateDialect));
if (switchChannelType == ReplayModelSwitchChannelType.MainChannel)
{
channel = MainChannel;
}
else
{
AlternativeChannel = new ModelReplayChannel(MainChannel.Connection_NegotiateDialect);
channel = AlternativeChannel;
}
}
ModelReplayFileOperationRequest operationRequest = new ModelReplayFileOperationRequest(channel, switchChannelType,
maxSmbVersionClientSupported, requestCommand, channelSequence, isSetReplayFlag, requestCommandParameters);
Request = operationRequest;
State = ModelState.Connected;
}