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

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

    /**
     * {@inheritDoc}
     *

        try {
            int flags = getFlags();
            int size = 8 * 4 + ((flags & NTLMSSP_NEGOTIATE_VERSION) != 0 ? 8 : 0);

            byte[] domain = {};
            final String suppliedDomainString = getSuppliedDomain();
            if ((flags & NTLMSSP_NEGOTIATE_VERSION) == 0 && suppliedDomainString != null && suppliedDomainString.length() != 0) {
                flags |= NTLMSSP_NEGOTIATE_OEM_DOMAIN_SUPPLIED;

        assertEquals((byte) 0x34, dst[1]);
        assertEquals((byte) (0x80 | (NameServicePacket.QUERY << 3) | 0x04 | 0x02 | 0x01), dst[2]); // flags1
        assertEquals((byte) (0x80 | 0x10 | NameServicePacket.FMT_ERR), dst[3]); // flags2
        assertEquals((byte) 0x00, dst[4]); // questionCount
        assertEquals((byte) 0x01, dst[5]);
        assertEquals((byte) 0x00, dst[6]); // answerCount

        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();
        }

        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 */

     * 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);

    @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());

        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;

    }

    /**
     * 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;
    }

        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");