private Smb2LeaseKey leaseKey;
private int leaseState;
private LeaseManager leaseManager;

protected void doConnect() throws IOException {
    // ... existing connection logic ...
    
    if (context.getConfig().isUseLeases() && tree.getSession().supports(SMB3_0)) {
        // Request lease when opening file
        leaseManager = tree.getSession().getLeaseManager();
        int requestedState = isDirectory() ? 

     */
    @Override
    public boolean isWitnessServiceDiscovery() {
        return this.delegate.isWitnessServiceDiscovery();
    }

    @Override
    public boolean isUseLeases() {
        return this.delegate.isUseLeases();
    }

    @Override
    public long getPersistentHandleTimeout() {
        return this.delegate.getPersistentHandleTimeout();
    }

     *
     * @return true if witness is enabled
     */
    boolean isUseWitness();

    /**
     * @return whether to use SMB3 leases for client-side caching
     */
    boolean isUseLeases();

    /**
     * @return timeout for persistent handles in milliseconds
     */
    long getPersistentHandleTimeout();

    /**
     * Gets the witness heartbeat timeout in milliseconds.
     *

    }

    @Override
    public boolean isWitnessServiceDiscovery() {
        return this.witnessServiceDiscovery;
    }

    @Override
    public boolean isUseLeases() {
        return this.useLeases;
    }

    @Override
    public long getPersistentHandleTimeout() {
        return this.persistentHandleTimeout;
    }

                SmbSessionImpl session = h.getSession();

                // Enable lease support if available, with fallback for compatibility
                boolean leasesAdded = false;
                if (config.isUseLeases() && h.isSMB3()) {
                    try {
                        LeaseManager leaseManager = session.getLeaseManager();
                        if (leaseManager != null) {