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;

        Smb2LeaseKey key1 = leaseManager.requestLease(path1, Smb2LeaseState.SMB2_LEASE_READ_CACHING);
        Smb2LeaseKey key2 = leaseManager.requestLease(path2, Smb2LeaseState.SMB2_LEASE_WRITE_CACHING);

        assertNotNull(leaseManager.getLease(key1));
        assertNotNull(leaseManager.getLease(key2));

        leaseManager.releaseAll();

        assertNull(leaseManager.getLease(key1));

    /**
     * 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() ? 

        directoryLeaseManager.requestDirectoryLease(directoryPath, DirectoryLeaseState.DIRECTORY_READ_HANDLE,
                DirectoryCacheScope.IMMEDIATE_CHILDREN);

        LeaseManager.LeaseEntry mockLeaseEntry = mock(LeaseManager.LeaseEntry.class);
        when(mockLeaseEntry.hasReadCache()).thenReturn(true);
        when(baseLeaseManager.getLease(leaseKey)).thenReturn(mockLeaseEntry);

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

    @Mock
    private CIFSContext context;

    @Mock
    private Configuration config;

    @Mock
    private SmbFile mockFile1;

    @Mock
    private SmbFile mockFile2;

    private LeaseManager baseLeaseManager;
    private DirectoryLeaseManager directoryLeaseManager;
    private AutoCloseable mocks;

    @BeforeEach
    public void setUp() throws Exception {

├── 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