Implement SMB 3.x Lease support Phase 5
- WRITE handler trigger lease break (READ leases conflict with WRITE) - READ handler trigger lease break (HANDLE leases may conflict) - Send LeaseBreakNotification via notification channel All 229 tests pass.
This commit is contained in:
18
vendor/smb-server/src/handlers/read.rs
vendored
18
vendor/smb-server/src/handlers/read.rs
vendored
@@ -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() {
|
||||
|
||||
16
vendor/smb-server/src/handlers/write.rs
vendored
16
vendor/smb-server/src/handlers/write.rs
vendored
@@ -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() {
|
||||
|
||||
Reference in New Issue
Block a user