final byte[] challenge = getChallenge();
            byte[] context = getContext();
            final byte[] targetInformation = getTargetInformation();
            int flags = getFlags();
            byte[] target = {};
            if ((flags & NTLMSSP_REQUEST_TARGET) != 0) {
                if (targetName != null && targetName.length() != 0) {

        }

        private void setupReferralDataWithPathConsumed(int consumed) {
            when(mockReferral.getTtl()).thenReturn(300);
            when(mockReferral.getRFlags()).thenReturn(0);
            when(mockReferral.getNode()).thenReturn("\\server\\share\\path");

            String reqPath = "\\\\server\\share\\path\\file";

    public byte[] toByteArray() {
        try {
            final String suppliedDomain = getSuppliedDomain();
            final String suppliedWorkstation = getSuppliedWorkstation();
            int flags = getFlags();
            boolean hostInfo = false;
            byte[] domain = {};
            if (suppliedDomain != null && suppliedDomain.length() != 0) {
                hostInfo = true;

        registration.setFlags(WitnessRegistration.WITNESS_REGISTER_NONE);
        assertEquals(WitnessRegistration.WITNESS_REGISTER_NONE, registration.getFlags());

        registration.setFlags(WitnessRegistration.WITNESS_REGISTER_IP_NOTIFICATION);
        assertEquals(WitnessRegistration.WITNESS_REGISTER_IP_NOTIFICATION, registration.getFlags());
    }

    @DisplayName("Should set and get flags")
    void testFlags() {
        // Given
        int flags = 0x0001; // Encrypted flag

        // When
        transformHeader.setFlags(flags);

        // Then
        assertEquals(flags, transformHeader.getFlags());
    }

    @Test
    @DisplayName("Should set and get session ID")
    void testSessionId() {
        // When

                        || !msg2.getFlag(NtlmFlags.NTLMSSP_NEGOTIATE_EXTENDED_SESSIONSECURITY))) {
                    throw new SmbUnsupportedOperationException("Server does not support extended NTLMv2 key exchange");
                }

                if (!msg2.getFlag(NtlmFlags.NTLMSSP_NEGOTIATE_128)) {

            if ((getFlags() & NTLMSSP_NEGOTIATE_SIGN) == NTLMSSP_NEGOTIATE_SIGN) {
                final HMACT64 hmac = new HMACT64(responseKeyNT);
                hmac.update(ntResponse, 0, 16); // only first 16 bytes of ntResponse
                final byte[] userSessionKey = hmac.digest();

                if ((getFlags() & NTLMSSP_NEGOTIATE_KEY_EXCH) != 0) {

            // When
            Type2Message parsedMessage = new Type2Message(rawMessage);

            // Then
            assertNotNull(parsedMessage);
            assertEquals(originalMessage.getFlags(), parsedMessage.getFlags());
            assertNull(parsedMessage.getChallenge()); // Should be null if all zeros
            assertNull(parsedMessage.getTarget());

        this.suppliedWorkstation = suppliedWorkstation;
    }

    @Override
    public byte[] toByteArray() {
        try {
            int flags = getFlags();
            int size = 8 * 4 + ((flags & NTLMSSP_NEGOTIATE_VERSION) != 0 ? 8 : 0);

            byte[] domain = {};
            final String suppliedDomainString = getSuppliedDomain();

    }

    /**
     * Gets the registration flags.
     *
     * @return the flags
     */
    public int getFlags() {
        return flags;
    }

    /**
     * Sets the registration flags.
     *
     * @param flags the flags
     */
    public void setFlags(int flags) {
        this.flags = flags;
    }