void testReadUShortOutOfBounds() {
        byte[] a = new byte[] { 0x02 };
        assertThrows(IndexOutOfBoundsException.class, () -\u003e NtlmMessage.readUShort(a, 0));
    }

    @Test
    @DisplayName("readULong throws on too short array")
    void testReadULongOutOfBounds() {
        byte[] a = new byte[] { 0x01, 0x02, 0x03 };

 */
public class SmbComTreeConnectAndXResponse extends AndXServerMessageBlock implements TreeConnectResponse {

    private static final int SMB_SUPPORT_SEARCH_BITS = 0x0001;
    private static final int SMB_SHARE_IS_IN_DFS = 0x0002;

    private boolean supportSearchBits, shareIsInDfs;
    private String service;
    private String nativeFileSystem = "";

    /**
     * Constructs a tree connect AndX response.
     *

        int namedPipeState = 0x03; // NP_NOWAIT | NP_READMODE_MESSAGE
        int readDataAvailable = 100;
        int numberOfMessages = 2;
        int messageLength = 50;
        byte[] testData = { 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08 };

        // Write values to buffer
        SMBUtil.writeInt4(namedPipeState, buffer, bufferIndex);
        SMBUtil.writeInt4(readDataAvailable, buffer, bufferIndex + 4);

        int flags = NegTokenInit.DELEGATION | NegTokenInit.MUTUAL_AUTHENTICATION | NegTokenInit.INTEGRITY;
        byte[] mechToken = new byte[] { 0x01, 0x02, 0x03 };
        byte[] mic = new byte[] { (byte) 0xFE, 0x55 };

        NegTokenInit init = new NegTokenInit(mechs, flags, mechToken, mic);
        byte[] bytes = init.toByteArray();

    /**
     * Timestamp type
     */
    public static final int MsvAvTimestamp = 0x7;

    /**
     * Single host type
     */
    public static final int MsvAvSingleHost = 0x08;

    /**
     * Target name type
     */
    public static final int MsvAvTargetName = 0x09;

    /**
     * Channel bindings type
     */
    public static final int MsvAvChannelBindings = 0x0A;

    /**
     * Read caching lease (R)
     */
    public static final int SMB2_LEASE_READ_CACHING = 0x01;

    /**
     * Handle caching lease (H)
     */
    public static final int SMB2_LEASE_HANDLE_CACHING = 0x02;

    /**
     * Write caching lease (W)
     */
    public static final int SMB2_LEASE_WRITE_CACHING = 0x04;

    /**
     * Read and Handle caching (RH)
     */

        assertNotNull(signingKey, "Should handle large session key");
        assertEquals(16, signingKey.length, "Should still produce 16-byte key");
    }

    @ParameterizedTest
    @ValueSource(ints = { 0x0300, 0x0302, 0x0310 }) // Non-3.1.1 dialects
    @DisplayName("Should use SMB 3.0.x context for non-3.1.1 dialects")
    void testDeriveKeys_NonSMB311Dialects(int dialect) {
        // When

        byte[] testMac = { 0x00, 0x11, 0x22, 0x33, 0x44, 0x55 };
        System.arraycopy(testMac, 0, src, srcIndex + 55, 6);

        // Statistics (6 bytes)
        byte[] stats = { 0x00, 0x01, 0x02, 0x03, 0x04, 0x05 };
        System.arraycopy(stats, 0, src, srcIndex + 61, 6);

        int result = response.readRDataWireFormat(src, srcIndex);
        assertEquals(67, result);

        // Verify parsed data

    void testLeaseStateConstants() {
        assertEquals(0x00, Smb2LeaseState.SMB2_LEASE_NONE);
        assertEquals(0x01, Smb2LeaseState.SMB2_LEASE_READ_CACHING);
        assertEquals(0x02, Smb2LeaseState.SMB2_LEASE_HANDLE_CACHING);
        assertEquals(0x04, Smb2LeaseState.SMB2_LEASE_WRITE_CACHING);
        assertEquals(0x03, Smb2LeaseState.SMB2_LEASE_READ_HANDLE);

            request.setFileId(testFileId);
            request.setData(data, 512, 1024);
            request.setOffset(4096L);
            request.setRemainingBytes(2048);
            request.setWriteFlags(0x02);

            // Calculate expected size
            int expectedSize = ((Smb2Constants.SMB2_HEADER_LENGTH + 48 + 1024 + 7) / 8) * 8;
            assertEquals(expectedSize, request.size());