assertEquals(0b0101, message.getFlags(), "Flag 0b0100 should be added to existing flags.");

        // Test clearing a flag
        message.setFlag(0b0001, false);
        assertEquals(0b0100, message.getFlags(), "Flag 0b0001 should be cleared.");

        message.setFlag(0b0100, false);
        assertEquals(0, message.getFlags(), "Flag 0b0100 should be cleared.");

    }

    @Override
    public byte[] toByteArray() throws IOException {
        int size = 48;
        int flags = getFlags();
        final String targetName = getTarget();
        final byte[] targetInformationBytes = getTargetInformation();
        byte[] targetBytes = {};

        if (getFlag(NTLMSSP_REQUEST_TARGET)) {
            if (targetName != null && targetName.length() != 0) {

        pos += 8;
        final int skOff = writeSecurityBuffer(type3, 52, sessionKeyBytes);
        pos += 8;

        writeULong(type3, pos, getFlags());
        pos += 4;

        if (getFlag(NTLMSSP_NEGOTIATE_VERSION)) {
            System.arraycopy(NTLMSSP_VERSION, 0, type3, pos, NTLMSSP_VERSION.length);
            pos += NTLMSSP_VERSION.length;
        } else if (micBytes != null) {

    void testSetFlagTrue() {
        msg.setFlag(0x4, true);
        assertTrue(msg.getFlag(0x4), "Bit should be set after setFlag(true)");
    }

    @Test
    @DisplayName("setFlag false clears the bit")
    void testSetFlagFalse() {
        msg.setFlag(0x4, true);
        msg.setFlag(0x4, false);
        assertFalse(msg.getFlag(0x4), "Bit should be cleared after setFlag(false)");
    }

                        || !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)) {

     * clear (<code>false</code>) the specified flag.
     */
    public void setFlag(final int flag, final boolean value) {
        setFlags(value ? getFlags() | flag : getFlags() & (0xffffffff ^ flag));
    }

    static int readULong(final byte[] src, final int index) {
        return src[index] & 0xff | (src[index + 1] & 0xff) << 8 | (src[index + 2] & 0xff) << 16 | (src[index + 3] & 0xff) << 24;
    }

     * @param value
     *            Indicates whether to set (<code>true</code>) or
     *            clear (<code>false</code>) the specified flag.
     */
    public void setFlag(final int flag, final boolean value) {
        setFlags(value ? getFlags() | flag : getFlags() & (0xffffffff ^ flag));
    }

    static int readULong(final byte[] src, final int index) {

     */
    public Type2Message(final Type1Message type1, final byte[] challenge, final String target) {
        this(getDefaultFlags(type1), challenge,
                type1 != null && target == null && type1.getFlag(NTLMSSP_REQUEST_TARGET) ? getDefaultDomain() : target);
    }

    /**
     * Creates a Type-2 message with the specified parameters.
     *
     * @param flags The flags to apply to this message.

            assertTrue(message.isFlagSet(DcerpcConstants.RPC_C_PF_BROADCAST));
        }

        @Test
        @DisplayName("setFlag should add specified flags")
        void testSetFlag() {
            // Test setting a new flag
            message.flags = 0;
            message.setFlag(DcerpcConstants.RPC_C_PF_BROADCAST);
            assertTrue(message.isFlagSet(DcerpcConstants.RPC_C_PF_BROADCAST));

     * @return the packet type
     */
    public int getPtype() {
        return ptype;
    }

    /**
     * Gets the flags of this DCERPC message
     * @return the message flags
     */
    public int getFlags() {
        return flags;
    }

    /**
     * Check if a specific flag is set.
     * @param flag the flag to check
     * @return whether flag is set
     */
    public boolean isFlagSet(final int flag) {