return output;
    }

    private static byte[] getHmac(byte[] data, byte[] key) throws GeneralSecurityException {
        Key macKey = new SecretKeySpec(key.clone(), KerberosConstants.HMAC_ALGORITHM);
        Mac mac = Mac.getInstance(KerberosConstants.HMAC_ALGORITHM);
        mac.init(macKey);
        return mac.doFinal(data);
    }

    /**
     * Returns the user's Kerberos realm.
     *

        // When
        byte[] encKey = Smb3KeyDerivation.deriveEncryptionKey(dialect, sessionKey, preauthIntegrity);

        // Then
        assertNotNull(encKey, "Encryption key should not be null");
        assertEquals(16, encKey.length, "Encryption key should be 16 bytes");
        assertFalse(Arrays.equals(sessionKey, encKey), "Encryption key should be different from session key");
    }

        // Given
        byte[] encKey = new byte[16];
        byte[] decKey = new byte[16];
        new SecureRandom().nextBytes(encKey);
        new SecureRandom().nextBytes(decKey);

        // When using try-with-resources
        assertDoesNotThrow(() -> {
            try (Smb2EncryptionContext context = new Smb2EncryptionContext(1, DialectVersion.SMB311, encKey, decKey)) {

        try {
            int expectedKeyLength = getKeyLength();
            byte[] encKey = getEncryptionKey();
            byte[] decKey = getDecryptionKey();

            try {
                boolean valid =
                        encKey != null && encKey.length == expectedKeyLength && decKey != null && decKey.length == expectedKeyLength;
                return valid;
            } finally {

    void testConstructorWithConfigOnly() {
        SessionRequestPacket packet = new SessionRequestPacket(mockConfig);

        assertNotNull(packet);
        // The packet should have calledName and callingName initialized but empty
        byte[] dst = new byte[256];
        int written = packet.writeTrailerWireFormat(dst, 0);
        assertTrue(written > 0);
    }

    @Test

        }
        byte[] packet = new byte[totalLen];
        System.arraycopy(header, 0, packet, 0, header.length);
        System.arraycopy(body, 0, packet, SMB2_HEADER_LENGTH, body.length);
        if (ctxOffsetFromHeader != null && ctxBytes != null) {
            System.arraycopy(ctxBytes, 0, packet, ctxOffsetFromHeader, ctxBytes.length);
        }
        return packet;
    }

    @Test

        packet.nameTrnId = 0x1234;
        packet.isResponse = true;
        packet.opCode = NameServicePacket.QUERY;
        packet.isAuthAnswer = true;
        packet.isTruncated = true;
        packet.isRecurDesired = true;
        packet.isRecurAvailable = true;
        packet.isBroadcast = true;
        packet.resultCode = NameServicePacket.FMT_ERR;
        packet.questionCount = 1;
        packet.answerCount = 2;

        int totalRead = packet.readWireFormat(bais, data, 0);

        assertEquals(14, totalRead); // 4 header + 10 trailer
        assertEquals(0x85, packet.type);
        assertEquals(10, packet.length);
    }

    @Test
    @DisplayName("writeHeaderWireFormat should write correct header")
    void testWriteHeaderWireFormat() {
        packet.type = SessionServicePacket.SESSION_REQUEST;

        byte[] packet = new byte[header.length + body.length];
        System.arraycopy(header, 0, packet, 0, header.length);
        System.arraycopy(body, 0, packet, header.length, body.length);

        BaseConfiguration config = new BaseConfiguration(true);
        Smb2IoctlResponse resp = new Smb2IoctlResponse(config);

        int read = resp.decode(packet, 0);

        void testSourceKeyAllMax() {
            // Given
            byte[] maxKey = new byte[SOURCE_KEY_SIZE];
            Arrays.fill(maxKey, (byte) 0xFF);
            SrvCopychunk chunk = new SrvCopychunk(Long.MAX_VALUE, Long.MAX_VALUE, Integer.MAX_VALUE);
            SrvCopychunkCopy copy = new SrvCopychunkCopy(maxKey, chunk);
            byte[] buffer = new byte[100];

            // When