*
 * This class represents negotiate contexts used in SMB 3.1.1
 * for capability negotiation and security enhancements.
 *
 * @author mbechler
 */
public interface NegotiateContextRequest extends Encodable {

    /**
     * Gets the negotiate context type.
     *
     * @return the SMB2 negotiate context type
     */
    int getContextType();

            assertTrue(preauth instanceof NegotiateContextRequest);
            assertTrue(encryption instanceof NegotiateContextRequest);
        }

        @Test
        @DisplayName("Should verify mock implementation")
        void testMockImplementation() {
            NegotiateContextRequest mockRequest = mock(NegotiateContextRequest.class);
            when(mockRequest.getContextType()).thenReturn(0xFF);

 * This negotiate context is used in SMB 3.1.1 to establish
 * pre-authentication integrity protection against downgrade attacks.
 *
 * @author mbechler
 */
public class PreauthIntegrityNegotiateContext implements NegotiateContextRequest, NegotiateContextResponse {

    /**
     * Context type
     */
    public static final int NEGO_CTX_PREAUTH_TYPE = 0x1;

    /**
     * SHA-512
     */

    private final int[] dialects;
    private int capabilities;
    private final byte[] clientGuid = new byte[16];
    private final int securityMode;
    private final NegotiateContextRequest[] negotiateContexts;
    private byte[] preauthSalt;

    /**
     * Constructs an SMB2 negotiate request with the specified configuration and security mode.
     *

 *
 * This negotiate context is used in SMB 3.x to negotiate
 * encryption capabilities and cipher suites.
 *
 * @author mbechler
 */
public class EncryptionNegotiateContext implements NegotiateContextRequest, NegotiateContextResponse {

    /**
     * Context type
     */
    public static final int NEGO_CTX_ENC_TYPE = 0x2;

    /**
     * AES 128 CCM
     */

        when(mockRequest.getCapabilities()).thenReturn(Smb2Constants.SMB2_GLOBAL_CAP_DFS);
        when(mockRequest.getNegotiateContexts())
                .thenReturn(new NegotiateContextRequest[] { createMockPreauthContext(), createMockEncryptionContext() });

        // Setup negotiate contexts
        NegotiateContextResponse[] contexts =

    class InterfaceImplementationTests {

        @Test
        @DisplayName("Should implement NegotiateContextRequest interface")
        void testImplementsNegotiateContextRequest() {
            EncryptionNegotiateContext context = new EncryptionNegotiateContext();

            assertTrue(context instanceof NegotiateContextRequest);
        }

        @Test

        when(mockConfig.isEncryptionEnabled()).thenReturn(true);

        // When
        request = new Smb2NegotiateRequest(mockConfig, 0);

        // Then
        NegotiateContextRequest[] contexts = request.getNegotiateContexts();
        assertNotNull(contexts);
        assertEquals(2, contexts.length);

        // Verify preauth context

 *
 * This context is used during SMB2 negotiation to indicate RDMA transform
 * capabilities when SMB Direct is supported by the client and server.
 */
public class Smb2RdmaTransformCapabilitiesContext implements NegotiateContextRequest, NegotiateContextResponse {

    // Context type
    /** Context ID for RDMA transform capabilities */
    public static final int CONTEXT_ID = Smb2Constants.SMB2_RDMA_TRANSFORM_CAPABILITIES;

 *
 * This context is used during SMB2 negotiation to negotiate compression
 * algorithms for SMB3 data compression support.
 */
public class CompressionNegotiateContext implements NegotiateContextRequest, NegotiateContextResponse {

    private static final Logger log = LoggerFactory.getLogger(CompressionNegotiateContext.class);

    // Context type
    /** Context ID for compression capabilities */