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 result = {
|
||||||
let open = open_arc.read().await;
|
let open = open_arc.read().await;
|
||||||
match open.handle.as_ref() {
|
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 result = {
|
||||||
let open = open_arc.read().await;
|
let open = open_arc.read().await;
|
||||||
match open.handle.as_ref() {
|
match open.handle.as_ref() {
|
||||||
|
|||||||
Reference in New Issue
Block a user