"Should throw SMBProtocolDecodingException for invalid structure size");

        assertEquals("Expected structureSize = 2", exception.getMessage());
    }

    @Test
    @DisplayName("Test readBytesWireFormat with zero structure size throws exception")
    void testReadBytesWireFormatZeroStructureSize() {

     */
    @Override
    protected int readBytesWireFormat(final byte[] buffer, final int bufferIndex) throws SMBProtocolDecodingException {
        final int structureSize = SMBUtil.readInt2(buffer, bufferIndex);
        if (structureSize != 4) {
            throw new SMBProtocolDecodingException("Structure size != 4");
        }

        return 4;
    }

     */
    @Override
    protected int readBytesWireFormat(final byte[] buffer, final int bufferIndex) throws SMBProtocolDecodingException {
        final int structureSize = SMBUtil.readInt2(buffer, bufferIndex);
        if (structureSize != 4) {
            throw new SMBProtocolDecodingException("Structure size is not 4");
        }
        return 4;
    }

        final int start = bufferIndex;
        final int structureSize = SMBUtil.readInt2(buffer, bufferIndex);
        if (structureSize != 60) {
            throw new SMBProtocolDecodingException("Expected structureSize = 60");
        }
        this.closeFlags = SMBUtil.readInt2(buffer, bufferIndex + 2);
        bufferIndex += 4;

        int start = bufferIndex;

        // StructureSize (2 bytes) - must be 44
        this.structureSize = SMBUtil.readInt2(buffer, bufferIndex);
        if (this.structureSize != 44) {
            throw new SMBProtocolDecodingException("Invalid lease break structure size: " + this.structureSize);
        }
        bufferIndex += 2;

        // Reserved (2 bytes)

    @Override
    protected int readBytesWireFormat(final byte[] buffer, int bufferIndex) throws SMBProtocolDecodingException {
        final int start = bufferIndex;
        final int structureSize = SMBUtil.readInt2(buffer, bufferIndex);
        if (structureSize != 16) {
            throw new SMBProtocolDecodingException("Structure size is not 16");
        }

        this.shareType = buffer[bufferIndex + 2];
        bufferIndex += 4;

            assertEquals("Expected structureSize = 4", exception.getMessage());
        }

        @ParameterizedTest
        @ValueSource(ints = { 0, 1, 2, 3, 5, 6, 100, 65535 })
        @DisplayName("Should throw exception for various invalid structure sizes")
        void testReadBytesWireFormatVariousInvalidSizes(int structureSize) {
            byte[] buffer = new byte[1024];

    @Override
    protected int readBytesWireFormat(final byte[] buffer, int bufferIndex) throws SMBProtocolDecodingException {
        final int start = bufferIndex;

        final int structureSize = SMBUtil.readInt2(buffer, bufferIndex);
        if (structureSize != 9) {
            throw new SMBProtocolDecodingException("Structure size != 9");
        }

        this.sessionFlags = SMBUtil.readInt2(buffer, bufferIndex + 2);

    @ValueSource(ints = { 0, 1, 2, 4, 8, 15, 17, 32, 64, 65535 })
    void testReadBytesWireFormatInvalidStructureSize(int structureSize) {
        // Given
        byte[] buffer = new byte[256];
        int offset = 0;

        // Write invalid structure size
        SMBUtil.writeInt2(structureSize, buffer, offset);

        // When & Then
        SMBProtocolDecodingException exception =

    void testInvalidStructureSize(int structureSize) {
        // Given
        byte[] buffer = new byte[256];
        SMBUtil.writeInt2(structureSize, buffer, 0);

        // When & Then
        SMBProtocolDecodingException exception =
                assertThrows(SMBProtocolDecodingException.class, () -> response.readBytesWireFormat(buffer, 0));
        assertEquals("Expected structureSize = 9", exception.getMessage());
    }