SMB Server Phase 2: VFS backend build fix + integration test
- Add VfsFile: Send supertrait for Mutex compatibility - Fix SmbServerCommand: struct → Subcommand enum with Start variant - Fix tracing_subscriber::init() → try_init() to avoid panic when logger already initialized - Fix CLI subcommand name: smb-server → smb-start (flatten naming) - Add #[command(name = "smb-start")] for CLI disambiguation - Fix unused variable warnings (smb_fs.rs, smb_server_backend.rs) - Remove unused VfsFile imports (webdav.rs, scp_handler.rs) - Integration test: Docker smbclient verified (list, upload, read)
This commit is contained in:
143
vendor/smb-server/src/handlers/set_info.rs
vendored
Normal file
143
vendor/smb-server/src/handlers/set_info.rs
vendored
Normal file
@@ -0,0 +1,143 @@
|
||||
//! SET_INFO handler.
|
||||
|
||||
use std::sync::Arc;
|
||||
|
||||
use crate::proto::header::Smb2Header;
|
||||
use crate::proto::messages::{InfoType, SetInfoRequest, SetInfoResponse};
|
||||
|
||||
use crate::backend::FileTimes;
|
||||
use crate::conn::state::Connection;
|
||||
use crate::dispatch::HandlerResponse;
|
||||
use crate::handlers::shared::{lookup_open, lookup_session_tree};
|
||||
use crate::info_class as ic;
|
||||
use crate::ntstatus;
|
||||
use crate::path::SmbPath;
|
||||
use crate::server::ServerState;
|
||||
use crate::utils::utf16le_to_units;
|
||||
|
||||
pub async fn handle(
|
||||
_server: &Arc<ServerState>,
|
||||
conn: &Arc<Connection>,
|
||||
hdr: &Smb2Header,
|
||||
body: &[u8],
|
||||
) -> HandlerResponse {
|
||||
let req = match SetInfoRequest::parse(body) {
|
||||
Ok(r) => r,
|
||||
Err(_) => return HandlerResponse::err(ntstatus::STATUS_INVALID_PARAMETER),
|
||||
};
|
||||
let info_type = match InfoType::from_u8(req.info_type) {
|
||||
Some(t) => t,
|
||||
None => return HandlerResponse::err(ntstatus::STATUS_INVALID_INFO_CLASS),
|
||||
};
|
||||
if !matches!(info_type, InfoType::File) {
|
||||
return HandlerResponse::err(ntstatus::STATUS_NOT_SUPPORTED);
|
||||
}
|
||||
|
||||
let tree_arc = match lookup_session_tree(conn, hdr).await {
|
||||
Ok(t) => t,
|
||||
Err(s) => return HandlerResponse::err(s),
|
||||
};
|
||||
let open_arc = match lookup_open(&tree_arc, req.file_id).await {
|
||||
Some(o) => o,
|
||||
None => return HandlerResponse::err(ntstatus::STATUS_FILE_CLOSED),
|
||||
};
|
||||
|
||||
let class = req.file_information_class;
|
||||
let buffer = req.buffer;
|
||||
let backend = {
|
||||
let tree = tree_arc.read().await;
|
||||
tree.share.backend.clone()
|
||||
};
|
||||
|
||||
let result = match class {
|
||||
ic::FILE_BASIC_INFORMATION => {
|
||||
if buffer.len() < 36 {
|
||||
return HandlerResponse::err(ntstatus::STATUS_INFO_LENGTH_MISMATCH);
|
||||
}
|
||||
let creation = u64::from_le_bytes(buffer[0..8].try_into().unwrap());
|
||||
let access = u64::from_le_bytes(buffer[8..16].try_into().unwrap());
|
||||
let write = u64::from_le_bytes(buffer[16..24].try_into().unwrap());
|
||||
let change = u64::from_le_bytes(buffer[24..32].try_into().unwrap());
|
||||
// 0 means "do not change", -1 (u64::MAX) means "do not change" too per spec.
|
||||
let to_some = |v: u64| {
|
||||
if v == 0 || v == u64::MAX {
|
||||
None
|
||||
} else {
|
||||
Some(v)
|
||||
}
|
||||
};
|
||||
let times = FileTimes {
|
||||
creation_time: to_some(creation),
|
||||
last_access_time: to_some(access),
|
||||
last_write_time: to_some(write),
|
||||
change_time: to_some(change),
|
||||
};
|
||||
let open = open_arc.read().await;
|
||||
match open.handle.as_ref() {
|
||||
Some(h) => h.set_times(times).await,
|
||||
None => return HandlerResponse::err(ntstatus::STATUS_FILE_CLOSED),
|
||||
}
|
||||
}
|
||||
ic::FILE_END_OF_FILE_INFORMATION => {
|
||||
if buffer.len() < 8 {
|
||||
return HandlerResponse::err(ntstatus::STATUS_INFO_LENGTH_MISMATCH);
|
||||
}
|
||||
let new_len = u64::from_le_bytes(buffer[0..8].try_into().unwrap());
|
||||
let open = open_arc.read().await;
|
||||
match open.handle.as_ref() {
|
||||
Some(h) => h.truncate(new_len).await,
|
||||
None => return HandlerResponse::err(ntstatus::STATUS_FILE_CLOSED),
|
||||
}
|
||||
}
|
||||
ic::FILE_DISPOSITION_INFORMATION => {
|
||||
if buffer.is_empty() {
|
||||
return HandlerResponse::err(ntstatus::STATUS_INFO_LENGTH_MISMATCH);
|
||||
}
|
||||
let mut open = open_arc.write().await;
|
||||
open.delete_on_close = buffer[0] != 0;
|
||||
Ok(())
|
||||
}
|
||||
ic::FILE_RENAME_INFORMATION => {
|
||||
// FILE_RENAME_INFORMATION layout (MS-FSCC §2.4.37):
|
||||
// ReplaceIfExists (1) | Reserved (7) | RootDirectory (8) | FileNameLength (4) | FileName...
|
||||
if buffer.len() < 20 {
|
||||
return HandlerResponse::err(ntstatus::STATUS_INFO_LENGTH_MISMATCH);
|
||||
}
|
||||
let name_len = u32::from_le_bytes(buffer[16..20].try_into().unwrap()) as usize;
|
||||
if buffer.len() < 20 + name_len {
|
||||
return HandlerResponse::err(ntstatus::STATUS_INFO_LENGTH_MISMATCH);
|
||||
}
|
||||
let name_bytes = &buffer[20..20 + name_len];
|
||||
let units = match utf16le_to_units(name_bytes) {
|
||||
Some(u) => u,
|
||||
None => return HandlerResponse::err(ntstatus::STATUS_OBJECT_NAME_INVALID),
|
||||
};
|
||||
let new_path = match SmbPath::from_utf16(&units) {
|
||||
Ok(p) => p,
|
||||
Err(_) => return HandlerResponse::err(ntstatus::STATUS_OBJECT_NAME_INVALID),
|
||||
};
|
||||
let from = open_arc.read().await.last_path.clone();
|
||||
match backend.rename(&from, &new_path).await {
|
||||
Ok(()) => {
|
||||
open_arc.write().await.last_path = new_path;
|
||||
Ok(())
|
||||
}
|
||||
Err(e) => Err(e),
|
||||
}
|
||||
}
|
||||
ic::FILE_ALLOCATION_INFORMATION => {
|
||||
// We don't preallocate; respond OK.
|
||||
Ok(())
|
||||
}
|
||||
_ => return HandlerResponse::err(ntstatus::STATUS_NOT_SUPPORTED),
|
||||
};
|
||||
|
||||
if let Err(e) = result {
|
||||
return HandlerResponse::err(e.to_nt_status());
|
||||
}
|
||||
let mut buf = Vec::new();
|
||||
SetInfoResponse::default()
|
||||
.write_to(&mut buf)
|
||||
.expect("encode");
|
||||
HandlerResponse::ok(buf)
|
||||
}
|
||||
Reference in New Issue
Block a user