*/
    @Override
    public int getSendBufferSize() {
        return this.snd_buf_size;
    }

    /**
     * Gets the negotiated flags2 value.
     *
     * @return negotiated flags2
     */
    public int getNegotiatedFlags2() {
        return this.negotiatedFlags2;
    }

    /**
     * {@inheritDoc}
     *

        this.clientAccessPoint = null;
        this.flags = WITNESS_RESOURCE_STATE_UNKNOWN;
    }

    /**
     * Represents an IP address in a witness notification with associated flags.
     */
    public static class WitnessIPAddress {
        private final InetAddress address;
        private final int flags;

        /** Flag indicating IPv4 address type */

        final String serverAddress;
        final int flags;
        final long timestamp;

        MockRegistration(String registrationId, String shareName, String serverAddress, int flags) {
            this.registrationId = registrationId;
            this.shareName = shareName;
            this.serverAddress = serverAddress;
            this.flags = flags;
            this.timestamp = System.currentTimeMillis();
        }

     * Creates a Type-2 message with the specified parameters.
     *
     * @param flags The flags to apply to this message.
     * @param challenge The challenge from the domain controller/server.
     * @param target The authentication target.
     */
    public Type2Message(final int flags, final byte[] challenge, final String target) {
        setFlags(flags);
        setChallenge(challenge);
        setTarget(target);

        offset += 8;

        // Directory flags (4 bytes)
        int flags = 0;
        if (cacheScope == DirectoryCacheScope.RECURSIVE_TREE) {
            flags |= DIRECTORY_LEASE_FLAG_RECURSIVE;
        }
        if (notificationEnabled) {
            flags |= DIRECTORY_LEASE_FLAG_NOTIFICATIONS;
        }
        SMBUtil.writeInt4(flags, buffer, offset);
        offset += 4;

    @MethodSource("provideFlagCombinations")
    void testShareFlagCombinations(int flags, boolean expectedDfs) throws SMBProtocolDecodingException {
        // Given
        byte[] buffer = createValidResponseBuffer(0, flags, 0, 0);

        // When
        response.readBytesWireFormat(buffer, 0);

        // Then
        assertEquals(flags, response.getShareFlags());
        assertEquals(expectedDfs, response.isShareDfs());

    }

    /**
     * Gets the lease flags for this V2 request
     * @return the lease flags
     */
    public int getLeaseFlags() {
        return leaseFlags;
    }

    /**
     * Sets the lease flags for this V2 request
     * @param leaseFlags the lease flags to set
     */
    public void setLeaseFlags(int leaseFlags) {
        this.leaseFlags = leaseFlags;
    }

class SmbComNegotiate extends ServerMessageBlock {

    private static final String DIALECTS = "\u0002NT LM 0.12\u0000";

    SmbComNegotiate() {
        command = SMB_COM_NEGOTIATE;
        flags2 = DEFAULT_FLAGS2;
    }

    @Override
    int writeParameterWordsWireFormat(final byte[] dst, final int dstIndex) {
        return 0;
    }

    @Override

        assertTrue(req instanceof ServerMessageBlock2);
    }

    @Test
    @DisplayName("Should set notify flags correctly")
    void testSetNotifyFlags() throws Exception {
        // Given
        int flags = Smb2ChangeNotifyRequest.SMB2_WATCH_TREE;

        // When
        request.setNotifyFlags(flags);

        // Then
        Field notifyFlagsField = Smb2ChangeNotifyRequest.class.getDeclaredField("notifyFlags");

    void testGetAndSetFlags() {
        // Test setting and getting flags
        int testFlags = 0x12345678;
        message.setFlags(testFlags);
        assertEquals(testFlags, message.getFlags(), "Flags should be set and retrieved correctly.");
    }

    @Test
    void testGetFlag() {
        // Test getting individual flags
        message.setFlags(0b1010); // Set flags: 8 (0x8) and 2 (0x2)