@ValueSource(ints = { 1024, 2048, 4096, 8192, 16384, 32768, 65535 })
    void testMaxDataCountValues(int maxDataCount) throws Exception {
        netShareEnum = new NetShareEnum(realConfig);

        // Set maxDataCount using reflection
        setFieldValue(netShareEnum, "maxDataCount", maxDataCount);

        byte[] dst = new byte[256];

    }

    @ParameterizedTest
    @ValueSource(ints = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 10, 11, 100, 255, 256, 65535 })
    @DisplayName("Test readBytesWireFormat with various invalid structure sizes")
    void testReadBytesWireFormatVariousInvalidSizes(int invalidSize) {
        response = new Smb2QueryDirectoryResponse(mockConfig, (byte) 0x03);
        byte[] buffer = new byte[1024];
        int bufferIndex = 0;

            assertDoesNotThrow(() -> impl.fixupDomain(domain));
        }

        @ParameterizedTest
        @ValueSource(ints = { 0, -1, Integer.MIN_VALUE, Integer.MAX_VALUE })
        @DisplayName("Should handle various values in stripPathConsumed")
        void testStripPathConsumedWithVariousValues(int value) {
            // Test with mock
            doNothing().when(mockReferralData).stripPathConsumed(value);

    }

    @ParameterizedTest
    @ValueSource(ints = { NtStatus.NT_STATUS_SUCCESS, NtStatus.NT_STATUS_NONE_MAPPED, 0x00000107 })
    @DisplayName("resolveSids(DcerpcHandle,...) accepts success/none/some-not-mapped codes")
    void resolveSids_allowsCertainRetvals_noThrow(int ret) throws Exception {
        SIDCacheImpl cache = new SIDCacheImpl(mock(CIFSContext.class));

    })
    void testWriteParametersWireFormatWithDifferentLevels(int infoLevel, int expectedMappedValue) {
        trans2QueryFSInfo = new Trans2QueryFSInformation(config, infoLevel);

        byte[] buffer = new byte[512];
        trans2QueryFSInfo.writeParametersWireFormat(buffer, 0);

        // Check mapped information level
        int actualInfoLevel = SMBUtil.readInt2(buffer, 0);

    void testDeriveKeys_DifferentSessionKeySizes() {
        // Test with various session key sizes
        int[] keySizes = { 8, 16, 24, 32, 64, 128 };
        int dialect = Smb2Constants.SMB2_DIALECT_0311;

        for (int size : keySizes) {
            // Given
            byte[] testKey = new byte[size];
            new SecureRandom().nextBytes(testKey);

            // When

    private static class TestSessionServicePacket extends SessionServicePacket {
        int trailerLength = 10; // Default trailer length for testing
        int trailerBytesRead = 0;

        @Override
        int writeTrailerWireFormat(byte[] dst, int dstIndex) {
            // Simple mock implementation
            for (int i = 0; i < trailerLength && dstIndex + i < dst.length; i++) {
                dst[dstIndex + i] = (byte) i;

        // Add ACEs
        int aceOffset = offset + 28; // After DACL header
        for (int i = 0; i < aceCount; i++) {
            prepareSimpleAce(buffer, aceOffset);
            aceOffset += 32; // Simple ACE size
        }
    }

    private void prepareSecurityDescriptorBufferWithInvalidAceCount(byte[] buffer, int offset) {
        buffer[offset] = 0x01; // revision

        }

        // Test size calculation
        int size = request.size();
        int expectedRawSize = Smb2Constants.SMB2_HEADER_LENGTH + 40 + 64;
        int expectedSize = (expectedRawSize + 7) & ~7;
        assertEquals(expectedSize, size);

        // Test wire format writing
        byte[] buffer = new byte[512];
        int bytesWritten = request.writeBytesWireFormat(buffer, 64);

        assertNotNull(response.getServerGuid());
    }

    @ParameterizedTest
    @DisplayName("Should throw exception for invalid structure size")
    @ValueSource(ints = { 0, 1, 64, 66, 128 })
    void testReadBytesWireFormatInvalidStructureSize(int structureSize) {
        // Given
        byte[] buffer = new byte[256];
        SMBUtil.writeInt2(structureSize, buffer, 0);

        // When & Then