// Testing invalid protocol ID during decode instead
        byte[] invalidBuffer = new byte[52];
        // Write invalid protocol ID
        invalidBuffer[0] = 0x00;
        invalidBuffer[1] = 0x00;
        invalidBuffer[2] = 0x00;
        invalidBuffer[3] = 0x00;

        assertThrows(IllegalArgumentException.class, () -> {
            Smb2TransformHeader.decode(invalidBuffer, 0);
        });
    }

    @Test

        int written = trans2QueryFSInfo.writeSetupWireFormat(buffer, 0);

        // Should write 2 bytes: subcommand and 0x00
        assertEquals(2, written);
        assertEquals(SmbComTransaction.TRANS2_QUERY_FS_INFORMATION, buffer[0]);
        assertEquals(0x00, buffer[1]);
    }

    @Test
    @DisplayName("Test writeSetupWireFormat with offset")
    void testWriteSetupWireFormatWithOffset() {

        int written = trans2SetFileInfo.writeSetupWireFormat(buffer, 0);

        // Should write 2 bytes: subcommand and 0x00
        assertEquals(2, written);
        assertEquals(SmbComTransaction.TRANS2_SET_FILE_INFORMATION, buffer[0]);
        assertEquals(0x00, buffer[1]);
    }

    @Test
    @DisplayName("Test writeSetupWireFormat with offset")
    void testWriteSetupWireFormatWithOffset() {

            dstIndex++;
            dst[dstIndex++] = (byte) 0x00; // Reserved1
            SMBUtil.writeInt2(this.tflags, dst, dstIndex);
            dstIndex += 2;
            SMBUtil.writeInt4(this.timeout, dst, dstIndex);
            dstIndex += 4;
            dst[dstIndex++] = (byte) 0x00; // Reserved2
            dst[dstIndex++] = (byte) 0x00;
        }
        SMBUtil.writeInt2(this.parameterCount, dst, dstIndex);

 */
@ExtendWith(MockitoExtension.class)
class SocketInputStreamTest {

    /**
     * Creates a NetBIOS session message header.
     * The header format is:
     * - Byte 0: Message type (0x00 for SESSION_MESSAGE)
     * - Byte 1: High bit of 17-bit length field
     * - Bytes 2-3: Lower 16 bits of length field
     */
    private static byte[] messageHeader(int length) {
        byte[] h = new byte[4];

        this.maxDataCount = 0xFFFF;
        this.maxSetupCount = (byte) 0x00;
        this.setupCount = 2;
        this.name = "\\PIPE\\";
    }

    @Override
    protected int writeSetupWireFormat(final byte[] dst, int dstIndex) {
        dst[dstIndex] = this.getSubCommand();
        dstIndex++;
        dst[dstIndex++] = (byte) 0x00;
        SMBUtil.writeInt2(this.pipeFid, dst, dstIndex);
        dstIndex += 2;

        buffer[bufferIndex++] = 1;
        buffer[bufferIndex++] = 0;
        // maxBufferSize
        buffer[bufferIndex++] = (byte) 0x00;
        buffer[bufferIndex++] = (byte) 0x40;
        buffer[bufferIndex++] = 0;
        buffer[bufferIndex++] = 0;
        // maxRawSize
        buffer[bufferIndex++] = (byte) 0x00;
        buffer[bufferIndex++] = (byte) 0x10;
        buffer[bufferIndex++] = 0;
        buffer[bufferIndex++] = 0;

        assertEquals(35000, SmbConstants.DEFAULT_CONN_TIMEOUT);
    }

    @Test
    @DisplayName("Should define SMB flags constants")
    void testSmbFlags() {
        assertEquals(0x00, SmbConstants.FLAGS_NONE);
        assertEquals(0x01, SmbConstants.FLAGS_LOCK_AND_READ_WRITE_AND_UNLOCK);
        assertEquals(0x02, SmbConstants.FLAGS_RECEIVE_BUFFER_POSTED);

        // per bouncycastle
        // <li>1: K(i) := PRF( KI, [i]_2 || Label || 0x00 || Context || [L]_2 ) with the counter at the very beginning
        // of the fixedInputData (The default implementation has this format)</li>
        // with the parameters
        // <li>1. KDFCounterParameters(ki, null, "Label || 0x00 || Context || [L]_2]", 8);

        // all fixed inputs go into the suffix:
        // + label

        Arrays.fill(destData, (byte) 0x00); // Clear destData
        ndrBuffer.readOctetArray(destData, 1, 3); // Read 0x10, 0x11, 0x12 into destData[1-3]

        assertEquals(3, ndrBuffer.getIndex());
        assertEquals(0x00, destData[0]);
        assertEquals(0x10, destData[1]);
        assertEquals(0x11, destData[2]);
        assertEquals(0x12, destData[3]);
        assertEquals(0x00, destData[4]);
    }

    @Test