diff --git a/vendor/smb-server/src/handlers/read.rs b/vendor/smb-server/src/handlers/read.rs index e2d63a7..49578d4 100644 --- a/vendor/smb-server/src/handlers/read.rs +++ b/vendor/smb-server/src/handlers/read.rs @@ -59,6 +59,24 @@ pub async fn handle( } } + // Phase 5: Trigger lease break if lease exists (SMB 3.x) + // READ operation doesn't break WRITE leases (only WRITE/HANDLE operations do) + // But we still check for HANDLE lease conflicts + let lease_notifications = server.lease_manager.break_lease( + crate::oplock::SMB2_LEASE_HANDLE, // READ operation may break HANDLE leases + ).await; + + for lease_notification in lease_notifications { + use crate::proto::framing::encode_frame; + let notification_bytes = lease_notification.write_to_bytes(); + let mut frame = Vec::with_capacity(notification_bytes.len() + 4); + encode_frame(¬ification_bytes, &mut frame); + + if let Some(tx) = conn.notification_tx.read().await.as_ref() { + let _ = tx.send(frame).await; + } + } + let result = { let open = open_arc.read().await; match open.handle.as_ref() { diff --git a/vendor/smb-server/src/handlers/write.rs b/vendor/smb-server/src/handlers/write.rs index 71a3e0f..299601b 100644 --- a/vendor/smb-server/src/handlers/write.rs +++ b/vendor/smb-server/src/handlers/write.rs @@ -75,6 +75,22 @@ pub async fn handle( } } + // Phase 5: Trigger lease break if lease exists (SMB 3.x) + let lease_notifications = server.lease_manager.break_lease( + crate::oplock::SMB2_LEASE_READ, // WRITE operation breaks READ leases + ).await; + + for lease_notification in lease_notifications { + use crate::proto::framing::encode_frame; + let notification_bytes = lease_notification.write_to_bytes(); + let mut frame = Vec::with_capacity(notification_bytes.len() + 4); + encode_frame(¬ification_bytes, &mut frame); + + if let Some(tx) = conn.notification_tx.read().await.as_ref() { + let _ = tx.send(frame).await; + } + } + let result = { let open = open_arc.read().await; match open.handle.as_ref() {