import jcifs.CIFSContext;
import jcifs.smb.SmbFile;

/**
 * SMB2/SMB3 Lease Manager
 *
 * Manages lease state for SMB2/SMB3 connections
 */
public class LeaseManager {

    private static final Logger log = LoggerFactory.getLogger(LeaseManager.class);

    private final ConcurrentHashMap<Smb2LeaseKey, LeaseEntry> leases;
    private final ConcurrentHashMap<String, Smb2LeaseKey> pathToLease;

    /**
     * Create a new directory lease manager
     *
     * @param context CIFS context
     * @param leaseManager base lease manager
     */
    public DirectoryLeaseManager(CIFSContext context, LeaseManager leaseManager) {
        this.context = context;
        this.baseLeaseManager = leaseManager;
        this.directoryCache = new ConcurrentHashMap<>();
        this.leaseToPath = new ConcurrentHashMap<>();

    
    // Change notification integration
    private final DirectoryChangeNotifier changeNotifier;
    
    public DirectoryLeaseManager(CIFSContext context, LeaseManager leaseManager) {
        this.context = context;
        this.baseLeaseManager = leaseManager;
        this.directoryCache = new ConcurrentHashMap<>();
        this.leaseToPath = new ConcurrentHashMap<>();
        this.scheduler = Executors.newScheduledThreadPool(1);

// In SmbFile.java
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() ? 

    private static final long MAX_RETRY_DELAY = 30000;
    private static final int MAX_BACKOFF_SHIFT = 3; // Maximum 8 seconds (2^3 * 1000ms)

    private final DirectoryLeaseManager leaseManager;
    private final ConcurrentHashMap<String, ChangeNotificationHandle> activeWatchers;
    private final ConcurrentHashMap<String, Integer> failureCounts;

    /**
     * Directory change types
     */

                    }
                }

                if (this.leaseManager != null) {
                    try {
                        this.leaseManager.shutdown();
                        this.leaseManager = null;
                    } catch (Exception e) {
                        log.debug("Error shutting down lease manager during emergency cleanup", e);

                    try {
                        LeaseManager leaseManager = session.getLeaseManager();
                        if (leaseManager != null) {
                            // Request appropriate lease state based on file type and access
                            int requestedState = determineLeaseState(access);

                            Smb2LeaseKey leaseKey = leaseManager.requestLease(uncPath, requestedState);

├── Smb2LeaseContext.java        - Lease context for create requests
├── Smb2LeaseKey.java            - 16-byte lease key management
├── Smb2LeaseState.java          - Lease state flags (R, H, W)
└── LeaseManager.java            - Central lease management
```

#### 1.2 Implementation Tasks
- [ ] Define lease state constants (SMB2_LEASE_READ, SMB2_LEASE_HANDLE, SMB2_LEASE_WRITE)
- [ ] Implement lease key generation and management