public ModelCreateLeaseRequest( CreateOptions_Values createOptions, RequestedOplockLevel_Values requestedOplockLevel_Values, ModelCreateContextRequest leaseContext, LeaseContextType leaseContextType) : base(0) { this.CreateOptions = createOptions; this.RequestedOplockLevel = requestedOplockLevel_Values; this.LeaseContext = leaseContext; this.ContextType = leaseContextType; }
public static void CreateRequest(ConnectTargetType connectTargetType, LeaseContextType leaseContextType, LeaseKeyType leaseKey, uint leaseState, LeaseFlagsValues leaseFlags, ParentLeaseKeyType parentLeaseKey) { Condition.IsTrue(state == ModelState.Connected); Condition.IsNull(request); // Limit these three less important paramters to reduce test case numbers. Combination.NWise(1, connectTargetType, leaseKey, leaseState); // Current leasing only supports to test leasing on the file. So limit the exploration. Condition.In(parentLeaseKey, ParentLeaseKeyType.EmptyParentLeaseKey, ParentLeaseKeyType.ValidParentLeaseKey); Combination.Pairwise(leaseFlags, parentLeaseKey); // If Conenction.Dialect is Smb2.002 or Smb2.1, then do not test Lease V2. Condition.IfThen( negotiateDialect == DialectRevision.Smb21 || negotiateDialect == DialectRevision.Smb2002, leaseContextType == LeaseContextType.LeaseV1); if (leaseContextType == LeaseContextType.LeaseV1) { request = new ModelCreateLeaseRequest( (connectTargetType == ConnectTargetType.ConnectToDirectory) ? CreateOptions_Values.FILE_DIRECTORY_FILE : CreateOptions_Values.FILE_NON_DIRECTORY_FILE, RequestedOplockLevel_Values.OPLOCK_LEVEL_LEASE, new ModelCreateRequestLease { LeaseKey = leaseKey, LeaseState = leaseState, LeaseFlags = leaseFlags }, leaseContextType); } else { request = new ModelCreateLeaseRequest( (connectTargetType == ConnectTargetType.ConnectToDirectory) ? CreateOptions_Values.FILE_DIRECTORY_FILE : CreateOptions_Values.FILE_NON_DIRECTORY_FILE, RequestedOplockLevel_Values.OPLOCK_LEVEL_LEASE, new ModelCreateRequestLeaseV2 { LeaseKey = leaseKey, LeaseState = leaseState, LeaseFlags = leaseFlags, ParentLeaseKey = parentLeaseKey }, leaseContextType); } }
public void CreateRequest(ConnectTargetType connectTargetType, LeaseContextType leaseContextType, LeaseKeyType leaseKey, uint leaseState, LeaseFlagsValues leaseFlags, ParentLeaseKeyType parentLeaseKey) { ValidateLeaseState(leaseState); uint status = 0; #region Fill Contexts LeaseKey = (leaseKey == LeaseKeyType.ValidLeaseKey ? Guid.NewGuid() : Guid.Empty); ParentLeaseKey = (parentLeaseKey == ParentLeaseKeyType.EmptyParentLeaseKey ? Guid.Empty : Guid.NewGuid()); originalClient.IsDirectory = connectTargetType == ConnectTargetType.ConnectToDirectory; switch (leaseContextType) { case LeaseContextType.LeaseV1: testConfig.CheckCreateContext(CreateContextTypeValue.SMB2_CREATE_REQUEST_LEASE); originalClient.CreateContexts = new Smb2CreateContextRequest[] { new Smb2CreateRequestLease { LeaseKey = LeaseKey, LeaseState = (LeaseStateValues)leaseState, LeaseFlags = (uint)leaseFlags, } }; break; case LeaseContextType.LeaseV2: testConfig.CheckCreateContext(CreateContextTypeValue.SMB2_CREATE_REQUEST_LEASE_V2); originalClient.CreateContexts = new Smb2CreateContextRequest[] { new Smb2CreateRequestLeaseV2 { LeaseKey = LeaseKey, LeaseState = (LeaseStateValues)leaseState, LeaseFlags = (uint)leaseFlags, ParentLeaseKey = ParentLeaseKey, } }; break; default: Site.Assume.Fail("Unexpected type: {0}", leaseContextType); break; } #endregion Packet_Header header; CREATE_Response createResponse; Smb2CreateContextResponse[] serverCreateContexts; status = originalClient.Client.Create(1, 64, originalClient.Flags, originalClient.MessageId++, originalClient.SessionId, originalClient.TreeId, originalClient.File, AccessMask.GENERIC_READ | AccessMask.GENERIC_WRITE | AccessMask.DELETE, ShareAccess_Values.FILE_SHARE_READ | ShareAccess_Values.FILE_SHARE_WRITE | ShareAccess_Values.FILE_SHARE_DELETE, (connectTargetType == ConnectTargetType.ConnectToDirectory) ? CreateOptions_Values.FILE_DIRECTORY_FILE : CreateOptions_Values.FILE_NON_DIRECTORY_FILE, CreateDisposition_Values.FILE_OPEN_IF, File_Attributes.NONE, ImpersonationLevel_Values.Impersonation, SecurityFlags_Values.NONE, RequestedOplockLevel_Values.OPLOCK_LEVEL_LEASE, originalClient.CreateContexts, out originalClient.FileId, out serverCreateContexts, out header, out createResponse); #region Handle Create Response // 3.3.5.9.11 Handling the SMB2_CREATE_REQUEST_LEASE_V2 Create Context // If Connection.Dialect does not belong to the SMB 3.x dialect family or if RequestedOplockLevel is not SMB2_OPLOCK_LEVEL_LEASE, the server SHOULD<270> ignore the SMB2_CREATE_REQUEST_LEASE_V2 Create Context request. // <270> Section 3.3.5.9.11: Windows 8, Windows Server 2012, Windows 8.1, // and Windows Server 2012 R2 servers do not ignore the SMB2_CREATE_REQUEST_LEASE_V2 create context when Connection.Dialect is equal to "2.100" or if RequestedOplockLevel is not equal to SMB2_OPLOCK_LEVEL_LEASE. if (!Smb2Utility.IsSmb3xFamily(originalClient.Dialect) && leaseContextType == LeaseContextType.LeaseV2) { Smb2CreateResponseLeaseV2 leaseContext = null; if (serverCreateContexts != null) { foreach (var context in serverCreateContexts) { if (context is Smb2CreateResponseLeaseV2) { leaseContext = context as Smb2CreateResponseLeaseV2; break; } } } if (testConfig.Platform == Platform.NonWindows) { Site.Assert.IsNull(leaseContext, "Create response should not include SMB2_CREATE_RESPONSE_LEASE_V2."); } else { if (originalClient.Dialect == DialectRevision.Smb21 && (testConfig.Platform == Platform.WindowsServer2012 || testConfig.Platform == Platform.WindowsServer2012R2)) { Site.Assert.IsNotNull(leaseContext, "Create response should include SMB2_CREATE_RESPONSE_LEASE_V2 when platform is Windows 8, Windows 2012 and Windows 2012R2 and dialect is equal to \"2.100\"."); } else { Site.Assert.IsNull(leaseContext, "Create response should not include SMB2_CREATE_RESPONSE_LEASE_V2."); } } // Bypass the situation to avoid Windows issues. this.CreateResponse(ModelSmb2Status.STATUS_SUCCESS, ReturnLeaseContextType.ReturnLeaseContextNotIncluded, (uint)LeaseStateValues.SMB2_LEASE_NONE, LeaseFlagsValues.NONE, leasingConfig); } else if (status == Smb2Status.STATUS_SUCCESS) { if (leaseContextType == LeaseContextType.LeaseV1) { if (serverCreateContexts != null) { Smb2CreateResponseLease leaseContext = null; foreach (var context in serverCreateContexts) { if (context is Smb2CreateResponseLease) { leaseContext = context as Smb2CreateResponseLease; break; } } Site.Assert.IsNotNull(leaseContext, "Create response MUST include SMB2_CREATE_RESPONSE_LEASE."); Site.Assert.AreEqual<Guid>(LeaseKey, leaseContext.LeaseKey, "Expect the LeaseKey in the response equals the LeaseKey in the request."); Site.Assert.AreEqual<LeaseFlagsValues>(LeaseFlagsValues.NONE, leaseContext.LeaseFlags, "Expect the lease is not in breaking."); Site.CaptureRequirementIfAreEqual<ulong>(0, leaseContext.LeaseDuration, RequirementCategory.MUST_BE_ZERO.Id, RequirementCategory.MUST_BE_ZERO.Description); originalClient.LeaseState = leaseContext.LeaseState; this.CreateResponse((ModelSmb2Status)status, ReturnLeaseContextType.ReturnLeaseContextIncluded, (uint)leaseContext.LeaseState, leaseContext.LeaseFlags, leasingConfig); } else { this.CreateResponse((ModelSmb2Status)status, ReturnLeaseContextType.ReturnLeaseContextNotIncluded, (uint)LeaseStateValues.SMB2_LEASE_NONE, LeaseFlagsValues.NONE, leasingConfig); } } else if (leaseContextType == LeaseContextType.LeaseV2) { if (serverCreateContexts != null) { Smb2CreateResponseLeaseV2 leaseContext = null; foreach (var context in serverCreateContexts) { if (context is Smb2CreateResponseLeaseV2) { leaseContext = context as Smb2CreateResponseLeaseV2; break; } } Site.Assert.IsNotNull(leaseContext, "Create response MUST include SMB2_CREATE_RESPONSE_LEASE_V2."); Site.Assert.AreEqual<Guid>(LeaseKey, leaseContext.LeaseKey, "Expect the LeaseKey in the response equals the LeaseKey in the request."); Site.Assert.IsTrue((leaseContext.Flags & LeaseFlagsValues.LEASE_FLAG_BREAK_IN_PROGRESS) == 0, "Expect the lease is not in breaking."); Site.CaptureRequirementIfAreEqual<ulong>(0, leaseContext.LeaseDuration, RequirementCategory.MUST_BE_ZERO.Id, RequirementCategory.MUST_BE_ZERO.Description); if ((leaseContext.Flags & LeaseFlagsValues.SMB2_LEASE_FLAG_PARENT_LEASE_KEY_SET) != 0) { Site.Assert.AreEqual<Guid>(ParentLeaseKey, leaseContext.ParentLeaseKey, "Expect the ParentLeaseKey in the response equals the ParentLeaseKey in the request."); } else { Site.Assert.AreEqual<Guid>(Guid.Empty, leaseContext.ParentLeaseKey, "Expect the ParentLeaseKey in the response is empty."); } // Epoch (2 bytes): A 16-bit unsigned integer incremented by the server on a lease state change. Site.Assert.AreEqual<ulong>(1, leaseContext.Epoch, "Expect the Epoch in the response equals 1."); if (testConfig.Platform == Platform.NonWindows) { // Reserved (2 bytes): This field MUST NOT be used and MUST be reserved. The server SHOULD<52> set this to 0, and the client MUST ignore it on receipt. // <52> Section 2.2.14.2.11: Windows 8, Windows Server 2012, Windows 8.1, and Windows Server 2012 R2 set this field to an arbitrary value. Site.CaptureRequirementIfAreEqual<ulong>(0, leaseContext.Reserved, RequirementCategory.MUST_BE_ZERO.Id, RequirementCategory.MUST_BE_ZERO.Description); } originalClient.LeaseState = leaseContext.LeaseState; this.CreateResponse((ModelSmb2Status)status, ReturnLeaseContextType.ReturnLeaseContextIncluded, (uint)leaseContext.LeaseState, leaseContext.Flags, leasingConfig); } else { this.CreateResponse((ModelSmb2Status)status, ReturnLeaseContextType.ReturnLeaseContextNotIncluded, (uint)LeaseStateValues.SMB2_LEASE_NONE, LeaseFlagsValues.NONE, leasingConfig); } } else { Site.Assume.Fail("Unexpected type: {0}", leaseContextType); } } else { this.CreateResponse((ModelSmb2Status)status, ReturnLeaseContextType.ReturnLeaseContextNotIncluded, (uint)LeaseStateValues.SMB2_LEASE_NONE, LeaseFlagsValues.NONE, leasingConfig); } #endregion #region Write Data if (status == Smb2Status.STATUS_SUCCESS && (connectTargetType == ConnectTargetType.ConnectToNonDirectory)) { WRITE_Response writeResponse; byte[] data = Encoding.ASCII.GetBytes(Smb2Utility.CreateRandomString(1)); status = originalClient.Client.Write(1, 1, originalClient.Flags, originalClient.MessageId++, originalClient.SessionId, originalClient.TreeId, 0, originalClient.FileId, Channel_Values.CHANNEL_NONE, WRITE_Request_Flags_Values.None, new byte[0], data, out header, out writeResponse); originalClient.GrantedCredit = header.CreditRequestResponse; Site.Assert.AreEqual(Smb2Status.STATUS_SUCCESS, status, "Write data to the file {0} failed with error code={1}", originalClient.File, Smb2Status.GetStatusCode(status)); FLUSH_Response flushResponse; status = originalClient.Client.Flush(1, 1, originalClient.Flags, originalClient.MessageId++, originalClient.SessionId, originalClient.TreeId, originalClient.FileId, out header, out flushResponse); originalClient.GrantedCredit = header.CreditRequestResponse; Site.Assert.AreEqual(Smb2Status.STATUS_SUCCESS, status, "Flush data to the file {0} failed with error code={1}", originalClient.File, Smb2Status.GetStatusCode(status)); } #endregion if (status == Smb2Status.STATUS_SUCCESS) { clients[(int)OperatorType.SameClientId].CreateContexts = originalClient.CreateContexts; } }