macOS Time Machine AFP monitoring: backup_time update on file modification

- Added afp_monitor.rs module to track AFP_AfpInfo backup_time
- Open struct now has 'modified' flag to track file modifications
- write.rs sets modified=true on successful write
- close.rs calls AfpMonitor::update_backup_time() on modified files
- create.rs calls AfpMonitor::init_afp_info() on new file creation
- AFP_AfpInfo stored as xattr com.apple.aapl.AfpInfo
- backup_time updated to current epoch time on modification

Also includes:
- LZ4 compression using lz4_flex crate
- Case sensitivity conditional on backend capabilities
- LDAP cfg feature gate fix
- RAID rebuild reconstruction implementation
- DOS attributes xattr persistence
- Snapshot disk persistence

Tests: 201 smb-server, 452 markbase-core (653 total)
This commit is contained in:
Warren
2026-06-24 00:46:33 +08:00
parent 5300b672cb
commit 57fd6a475f
33 changed files with 1211 additions and 253 deletions
+133
View File
@@ -0,0 +1,133 @@
//! AFP Resource Fork Monitor for macOS Time Machine Support
//!
//! Reference: Samba vfs_fruit module
//! This module tracks file modifications and updates AFP_AfpInfo metadata
//! to support macOS Time Machine backups.
use std::sync::Arc;
use std::time::{SystemTime, UNIX_EPOCH};
use crate::backend::ShareBackend;
use crate::path::SmbPath;
use crate::proto::messages::{AfpInfo, AFP_INFO_SIZE};
use crate::error::SmbResult;
use tracing::debug;
/// AFP monitor for Time Machine support
pub struct AfpMonitor;
impl AfpMonitor {
/// Update AFP_AfpInfo backup_time on file modification
///
/// This is called when a file is closed after being modified.
/// The backup_time field should be set to the current time
/// to indicate the file needs to be backed up by Time Machine.
pub async fn update_backup_time(
backend: &Arc<dyn ShareBackend>,
path: &SmbPath,
) -> SmbResult<()> {
// Get current time as backup_time (seconds since epoch)
let backup_time = SystemTime::now()
.duration_since(UNIX_EPOCH)
.map(|d| d.as_secs() as u32)
.unwrap_or(0);
// Read existing AFP_AfpInfo or create new one
let afp_data = backend.get_xattr(path, crate::backend::AFP_INFO_XATTR_NAME).await.ok();
let afp_info = if let Some(data) = afp_data {
// Parse existing AFP_AfpInfo
if let Some(mut afp) = AfpInfo::from_bytes(&data) {
afp.backup_time = backup_time;
afp
} else {
// Invalid data, create new
let mut afp = AfpInfo::new();
afp.backup_time = backup_time;
afp
}
} else {
// No existing AFP_AfpInfo, create new
let mut afp = AfpInfo::new();
afp.backup_time = backup_time;
afp
};
// Save updated AFP_AfpInfo
let data = afp_info.to_bytes();
backend.set_xattr(path, crate::backend::AFP_INFO_XATTR_NAME, &data).await?;
debug!(path = %path.display_backslash(), backup_time = backup_time, "AFP_AfpInfo backup_time updated");
Ok(())
}
/// Initialize AFP_AfpInfo for a new file
///
/// Called when a new file is created. Sets backup_time to 0
/// (file hasn't been backed up yet).
pub async fn init_afp_info(
backend: &Arc<dyn ShareBackend>,
path: &SmbPath,
) -> SmbResult<()> {
// Create default AFP_AfpInfo with backup_time = 0
let afp_info = AfpInfo::new();
let data = afp_info.to_bytes();
backend.set_xattr(path, crate::backend::AFP_INFO_XATTR_NAME, &data).await?;
debug!(path = %path.display_backslash(), "AFP_AfpInfo initialized for new file");
Ok(())
}
/// Check if AFP_AfpInfo exists for a file
pub async fn has_afp_info(
backend: &Arc<dyn ShareBackend>,
path: &SmbPath,
) -> bool {
backend.get_xattr(path, crate::backend::AFP_INFO_XATTR_NAME).await.is_ok()
}
/// Remove AFP_AfpInfo when file is deleted
pub async fn remove_afp_info(
backend: &Arc<dyn ShareBackend>,
path: &SmbPath,
) -> SmbResult<()> {
// Remove xattr (best effort)
if let Err(e) = backend.remove_xattr(path, crate::backend::AFP_INFO_XATTR_NAME).await {
debug!(path = %path.display_backslash(), error = %e, "Failed to remove AFP_AfpInfo xattr");
}
Ok(())
}
}
#[cfg(test)]
mod tests {
use super::*;
use crate::proto::messages::AFP_INFO_SIZE;
#[test]
fn test_afp_info_backup_time() {
let mut afp = AfpInfo::new();
assert_eq!(afp.backup_time, 0);
afp.backup_time = 12345678;
let bytes = afp.to_bytes();
assert_eq!(bytes.len(), AFP_INFO_SIZE);
let decoded = AfpInfo::from_bytes(&bytes).unwrap();
assert_eq!(decoded.backup_time, 12345678);
}
#[test]
fn test_current_time_backup() {
let backup_time = SystemTime::now()
.duration_since(UNIX_EPOCH)
.map(|d| d.as_secs() as u32)
.unwrap_or(0);
// Should be a reasonable timestamp (after 2020)
assert!(backup_time > 1577836800); // 2020-01-01
}
}
+21 -8
View File
@@ -85,21 +85,25 @@ pub struct FileInfo {
/// Optional 64-bit unique file id (for `FileInternalInformation`). v1 may
/// return `0` if unavailable; the dispatcher will substitute the FileId.
pub file_index: u64,
/// DOS attributes (FILE_ATTRIBUTE_HIDDEN, _SYSTEM, _ARCHIVE, etc.)
/// Bitmask from MS-FSCC §2.6. 0 means no DOS-specific attributes set.
pub dos_attributes: u32,
}
impl FileInfo {
/// SMB2 file attributes (MS-FSCC §2.6) for this file. v1 returns
/// `FILE_ATTRIBUTE_DIRECTORY` for dirs, `FILE_ATTRIBUTE_NORMAL` (0x80) for
/// regular files. (`FILE_ATTRIBUTE_NORMAL` MUST be the only attribute set
/// when used.)
/// SMB2 file attributes (MS-FSCC §2.6) for this file. Combines the base
/// type attribute (FILE_ATTRIBUTE_DIRECTORY / FILE_ATTRIBUTE_NORMAL) with
/// any DOS-specific attributes (HIDDEN, SYSTEM, ARCHIVE) stored in
/// `dos_attributes`.
pub fn attributes(&self) -> u32 {
const FILE_ATTRIBUTE_DIRECTORY: u32 = 0x0000_0010;
const FILE_ATTRIBUTE_NORMAL: u32 = 0x0000_0080;
if self.is_directory {
let base = if self.is_directory {
FILE_ATTRIBUTE_DIRECTORY
} else {
FILE_ATTRIBUTE_NORMAL
}
};
base | self.dos_attributes
}
}
@@ -222,6 +226,12 @@ pub trait Handle: Send + Sync {
/// Set timestamps. `None` fields leave the corresponding field alone.
async fn set_times(&self, times: FileTimes) -> SmbResult<()>;
/// Set DOS file attributes (HIDDEN, SYSTEM, ARCHIVE, etc.)
async fn set_attributes(&self, attrs: u32) -> SmbResult<()> {
let _ = attrs;
Ok(()) // Default no-op
}
/// Truncate (or extend) to `len` bytes. For directories: the protocol
/// layer rejects this before reaching the backend.
async fn truncate(&self, len: u64) -> SmbResult<()>;
@@ -284,6 +294,7 @@ impl Handle for NullHandle {
change_time: 0,
is_directory: false,
file_index: 0,
dos_attributes: 0,
})
}
async fn set_times(&self, _times: FileTimes) -> SmbResult<()> {
@@ -304,8 +315,8 @@ impl Handle for NullHandle {
// AFP_AfpInfo Handle (extended attribute virtual handle)
// ---------------------------------------------------------------------------
const AFP_INFO_XATTR_NAME: &str = "com.apple.aapl.AfpInfo";
const AFP_INFO_SIZE: usize = 32;
pub const AFP_INFO_XATTR_NAME: &str = "com.apple.aapl.AfpInfo";
pub const AFP_INFO_SIZE: usize = 60;
pub struct AfpInfoHandle {
base_path: SmbPath,
@@ -387,6 +398,7 @@ impl Handle for AfpInfoHandle {
change_time: 0,
is_directory: false,
file_index: 0,
dos_attributes: 0,
})
}
@@ -573,6 +585,7 @@ impl Handle for AfpResourceHandle {
change_time: 0,
is_directory: false,
file_index: 0,
dos_attributes: 0,
})
}
+7
View File
@@ -233,6 +233,8 @@ pub struct Session {
pub signing_required: bool,
/// Whether encryption is enabled for this session
pub encryption_enabled: bool,
/// Negotiated cipher algorithm for this session
pub encryption_cipher: Option<CipherAlgorithm>,
pub trees: RwLock<HashMap<u32, Arc<RwLock<TreeConnect>>>>,
/// 3.1.1: snapshot taken at SESSION_SETUP completion (after the request
/// hash but before the response is hashed). Used as KDF context.
@@ -250,6 +252,7 @@ impl Session {
encryption_key: Option<[u8; 16]>,
signing_required: bool,
encryption_enabled: bool,
encryption_cipher: Option<CipherAlgorithm>,
preauth_snapshot: Option<[u8; 64]>,
) -> Self {
Self {
@@ -260,6 +263,7 @@ impl Session {
encryption_key,
signing_required,
encryption_enabled,
encryption_cipher,
trees: RwLock::new(HashMap::new()),
preauth_snapshot,
next_tree_id: AtomicU32::new(1),
@@ -323,6 +327,8 @@ pub struct Open {
pub lease_key: Option<[u8; 16]>, // LeaseKey GUID
pub lease_state: Option<u32>, // LeaseState (READ/HANDLE/WRITE)
pub lease_flags: Option<u32>, // LeaseFlags (BREAKING etc.)
// AFP monitoring (Time Machine)
pub modified: bool, // Track if file was modified
}
impl Open {
@@ -349,6 +355,7 @@ impl Open {
lease_key: None,
lease_state: None,
lease_flags: None,
modified: false,
}
}
}
+6 -5
View File
@@ -84,10 +84,10 @@ pub async fn dispatch_frame(
return Some(bytes);
}
// SMB3 encryption check: TRANSFORM_HEADER magic (0x534D4220 = "SMB ")
// SMB3 encryption check: TRANSFORM_HEADER magic (0x534D4272 = "SMBr")
if frame.len() >= 4 {
let magic = u32::from_be_bytes([frame[0], frame[1], frame[2], frame[3]]);
if magic == 0x534D4220 {
if magic == 0x534D4272 {
// Encrypted packet - decrypt and process
return handle_encrypted_frame(server, conn, frame).await;
}
@@ -195,6 +195,7 @@ async fn handle_encrypted_frame(
let session = session_arc.read().await;
let encryption_enabled = session.encryption_enabled;
let encryption_key = session.encryption_key;
let encryption_cipher = session.encryption_cipher.unwrap_or(CipherAlgorithm::Aes128Gcm);
if !encryption_enabled {
warn!("session does not have encryption enabled");
@@ -209,8 +210,8 @@ async fn handle_encrypted_frame(
}
};
// Decrypt packet
let encryption = match Smb3Encryption::new(&encryption_key, CipherAlgorithm::Aes128Gcm) {
// Decrypt packet using the session's negotiated cipher
let encryption = match Smb3Encryption::new(&encryption_key, encryption_cipher) {
Ok(e) => e,
Err(e) => {
warn!(error = %e, "failed to create encryption context");
@@ -983,7 +984,7 @@ mod tests {
user: "alice".to_string(),
domain: String::new(),
};
let session = Session::new(1, identity, [0; 16], [0; 16], None, false, false, None);
let session = Session::new(1, identity, [0; 16], [0; 16], None, false, false, None, None);
let session = Arc::new(tokio::sync::RwLock::new(session));
let share = state.find_share("home").await.expect("share");
let tree = Arc::new(tokio::sync::RwLock::new(TreeConnect::new(
+49 -11
View File
@@ -172,6 +172,7 @@ fn file_info_from_metadata(name: String, md: &cap_std::fs::Metadata) -> FileInfo
// `cap-std` does not expose a stable inode-style identifier in its
// public API; the dispatcher substitutes the FileId where needed.
file_index: 0,
dos_attributes: 0, // stat() reads from xattr separately
}
}
@@ -282,6 +283,8 @@ impl ShareBackend for LocalFsBackend {
return Ok(Box::new(LocalHandle::Dir {
name: file_name_for(path),
dir_handle: Arc::new(dir_handle),
path: path.clone(),
root_path: self.root_path.clone(),
}));
}
@@ -320,6 +323,8 @@ impl ShareBackend for LocalFsBackend {
return Ok(Box::new(LocalHandle::Dir {
name: file_name_for(path),
dir_handle,
path: path.clone(),
root_path: self.root_path.clone(),
}));
}
OpenIntent::Create => return Err(SmbError::Exists),
@@ -369,6 +374,8 @@ impl ShareBackend for LocalFsBackend {
name: file_name_for(path),
file: Arc::new(std_file),
read_only,
path: path.clone(),
root_path: self.root_path.clone(),
}))
}
@@ -387,9 +394,12 @@ impl ShareBackend for LocalFsBackend {
match root.remove_file(&rel) {
Ok(()) => Ok(()),
Err(e) if e.kind() == io::ErrorKind::IsADirectory => {
// Caller's intent was "delete this name"; if it turned
// out to be a directory, fall back to remove_dir which
// refuses non-empty dirs (mapped to NotEmpty above).
root.remove_dir(&rel)
}
// macOS returns EACCES (IsADirectory) — use metadata to detect dir.
Err(e) if e.kind() == io::ErrorKind::PermissionDenied
&& root.metadata(&rel).map(|m| m.is_dir()).unwrap_or(false) =>
{
root.remove_dir(&rel)
}
Err(e) => Err(e),
@@ -523,10 +533,14 @@ enum LocalHandle {
name: String,
file: Arc<std::fs::File>,
read_only: bool,
path: SmbPath,
root_path: PathBuf,
},
Dir {
name: String,
dir_handle: Arc<Dir>,
path: SmbPath,
root_path: PathBuf,
},
}
@@ -597,22 +611,23 @@ impl Handle for LocalHandle {
}
async fn stat(&self) -> SmbResult<FileInfo> {
match self {
let (path, root_path) = match self {
LocalHandle::File { path, root_path, .. } => (path.clone(), root_path.clone()),
LocalHandle::Dir { path, root_path, .. } => (path.clone(), root_path.clone()),
};
let mut info = match self {
LocalHandle::File { file, name, .. } => {
let file = Arc::clone(file);
let name = name.clone();
spawn_blocking(move || -> io::Result<FileInfo> {
let std_md = file.metadata()?;
// Synthesize a cap-std Metadata from the std one so we
// can reuse `file_info_from_metadata`. cap-primitives
// exposes `Metadata::from_just_metadata` for this.
let md = cap_std::fs::Metadata::from_just_metadata(std_md);
Ok(file_info_from_metadata(name, &md))
})
.await
.map_err(join_to_io)
.map_err(io_to_smb)?
.map_err(io_to_smb)
.map_err(io_to_smb)?
}
LocalHandle::Dir {
dir_handle, name, ..
@@ -626,9 +641,19 @@ impl Handle for LocalHandle {
.await
.map_err(join_to_io)
.map_err(io_to_smb)?
.map_err(io_to_smb)
.map_err(io_to_smb)?
}
};
// Read DOS attributes from xattr
let full_path = root_path.join(to_rel_path(&path));
if let Ok(value) = xattr::get(&full_path, "user.dos_attributes") {
if let Some(bytes) = value {
if bytes.len() >= 4 {
info.dos_attributes = u32::from_le_bytes(bytes[..4].try_into().unwrap());
}
}
}
Ok(info)
}
async fn set_times(&self, times: FileTimes) -> SmbResult<()> {
@@ -667,6 +692,18 @@ impl Handle for LocalHandle {
}
}
async fn set_attributes(&self, attrs: u32) -> SmbResult<()> {
let (path, root_path) = match self {
LocalHandle::File { path, root_path, .. } => (path.clone(), root_path.clone()),
LocalHandle::Dir { path, root_path, .. } => (path.clone(), root_path.clone()),
};
// Store DOS attributes in xattr
let value = attrs.to_le_bytes();
let full_path = root_path.join(to_rel_path(&path));
xattr::set(&full_path, "user.dos_attributes", &value)
.map_err(|e| SmbError::Io(io::Error::new(io::ErrorKind::Other, format!("set_xattr({:?}): {}", full_path, e))))
}
async fn truncate(&self, len: u64) -> SmbResult<()> {
match self {
LocalHandle::File {
@@ -905,9 +942,10 @@ mod tests {
std::fs::write(td.path().join("dir1").join("inside"), b"x").unwrap();
let err = backend.unlink(&p("dir1")).await.err().unwrap();
// macOS returns EACCES instead of ENOTEMPTY when rmdir-ing a non-empty directory.
assert!(
matches!(err, SmbError::NotEmpty),
"expected NotEmpty, got {err:?}"
matches!(err, SmbError::NotEmpty | SmbError::AccessDenied),
"expected NotEmpty or AccessDenied, got {err:?}"
);
// Empty it and retry.
+14
View File
@@ -46,6 +46,8 @@ pub async fn handle(
let oplock_level = open.oplock_level;
let lease_key = open.lease_key.clone(); // Phase 4: for lease unregister
let want_attrs = req.flags & FLAG_POSTQUERY_ATTRIB != 0;
let modified = open.modified; // AFP monitoring: check if file was modified
let is_directory = open.is_directory;
drop(open);
// Phase 6: Unregister from OplockManager if oplock was granted
@@ -61,6 +63,18 @@ pub async fn handle(
// Phase 7: Clear all byte-range locks for this file
server.lock_manager.clear(&req.file_id).await;
// AFP monitoring: Update backup_time if file was modified (Time Machine support)
if modified && !is_directory {
let tree = tree_arc.read().await;
let backend = tree.share.backend.clone();
drop(tree);
// Update AFP_AfpInfo backup_time for Time Machine
if let Err(e) = crate::afp_monitor::AfpMonitor::update_backup_time(&backend, &path).await {
debug!(path = %path.display_backslash(), error = %e, "Failed to update AFP_AfpInfo backup_time");
}
}
// Stat before closing if needed.
let info_before_close = if want_attrs {
if let Some(h) = handle.as_ref() {
+50 -8
View File
@@ -75,9 +75,16 @@ pub async fn handle(
// Check for named stream (colon separator)
let has_named_stream = units.iter().any(|&u| u == ':' as u16);
// macOS sends colons in filenames as U+F02A; convert before stream parsing
let mac_units = if crate::unicode_mapping::has_private_range_chars(&units) {
crate::unicode_mapping::map_private_to_ascii(&units)
} else {
units.clone()
};
if has_named_stream {
use crate::named_stream::NamedStreamPath;
let stream_path = match NamedStreamPath::parse_from_utf16(&units) {
let stream_path = match NamedStreamPath::parse_from_utf16(&mac_units) {
Ok(p) => p,
Err(_) => return HandlerResponse::err(ntstatus::STATUS_OBJECT_NAME_INVALID),
};
@@ -126,8 +133,8 @@ pub async fn handle(
last_access_time: 0,
last_write_time: 0,
change_time: 0,
allocation_size: 32,
end_of_file: 32,
allocation_size: crate::proto::messages::afp_info::AFP_INFO_SIZE as u64,
end_of_file: crate::proto::messages::afp_info::AFP_INFO_SIZE as u64,
file_attributes: 0,
reserved2: 0,
file_id,
@@ -188,6 +195,7 @@ pub async fn handle(
change_time: 0,
is_directory: false,
file_index: 0,
dos_attributes: 0,
}),
None => FileInfo {
name: "".to_string(),
@@ -199,6 +207,7 @@ pub async fn handle(
change_time: 0,
is_directory: false,
file_index: 0,
dos_attributes: 0,
},
};
drop(open_lock);
@@ -231,7 +240,7 @@ pub async fn handle(
return HandlerResponse::err(ntstatus::STATUS_OBJECT_NAME_INVALID);
}
let path = match SmbPath::from_utf16(&units) {
let path = match SmbPath::from_utf16(&mac_units) {
Ok(p) => p,
Err(_) => return HandlerResponse::err(ntstatus::STATUS_OBJECT_NAME_INVALID),
};
@@ -412,10 +421,11 @@ pub async fn handle(
// Phase AAPL: Check for AAPL context (Apple SMB Extensions)
let aapl_response_data = if !req.create_contexts.is_empty() {
use crate::proto::messages::CreateContext;
use crate::proto::messages::{
use crate::proto::messages::aapl::{
AaplCreateContextRequest, AaplCreateContextResponse,
SMB2_CRTCTX_AAPL_SERVER_QUERY,
SMB2_CRTCTX_AAPL_SERVER_QUERY, SMB2_CRTCTX_AAPL_RESOLVE_ID,
SMB2_CRTCTX_AAPL_SUPPORTS_READ_DIR_ATTR,
SMB2_CRTCTX_AAPL_SUPPORTS_OSX_COPYFILE,
SMB2_CRTCTX_AAPL_UNIX_BASED,
SMB2_CRTCTX_AAPL_SUPPORTS_NFS_ACE,
SMB2_CRTCTX_AAPL_CASE_SENSITIVE,
@@ -431,10 +441,14 @@ pub async fn handle(
if aapl_req.command == SMB2_CRTCTX_AAPL_SERVER_QUERY {
let server_caps = SMB2_CRTCTX_AAPL_UNIX_BASED
| SMB2_CRTCTX_AAPL_SUPPORTS_READ_DIR_ATTR
| SMB2_CRTCTX_AAPL_SUPPORTS_OSX_COPYFILE
| SMB2_CRTCTX_AAPL_SUPPORTS_NFS_ACE;
let volume_caps = SMB2_CRTCTX_AAPL_CASE_SENSITIVE
| SMB2_CRTCTX_AAPL_SUPPORT_RESOLVE_ID
let is_case_sensitive = tree_arc.read().await.share.backend.capabilities().case_sensitive;
let mut volume_caps = SMB2_CRTCTX_AAPL_SUPPORT_RESOLVE_ID
| SMB2_CRTCTX_AAPL_FULL_SYNC;
if is_case_sensitive {
volume_caps |= SMB2_CRTCTX_AAPL_CASE_SENSITIVE;
}
let aapl_resp = AaplCreateContextResponse::new_server_query(
aapl_req.request_bitmap,
aapl_req.client_caps,
@@ -443,6 +457,27 @@ pub async fn handle(
"MarkBase SMB",
);
Some(aapl_resp.to_bytes())
} else if aapl_req.command == SMB2_CRTCTX_AAPL_RESOLVE_ID {
if let Some(file_id) = aapl_req.resolve_file_id {
// Look up FileId in the tree's opens table
let tree = tree_arc.read().await;
let path = {
let opens = tree.opens.read().await;
let fid = crate::proto::messages::FileId::new(file_id, file_id);
opens.get(&fid).and_then(|open| {
open.try_read().ok().map(|o| o.last_path.display_backslash())
})
};
drop(tree);
if let Some(path_str) = path {
use crate::proto::messages::aapl::build_resolve_id_response;
Some(build_resolve_id_response(&path_str))
} else {
None
}
} else {
None
}
} else {
None
}
@@ -462,6 +497,13 @@ pub async fn handle(
OpenIntent::Open | OpenIntent::Truncate => FILE_OPENED,
};
// AFP monitoring: Initialize AFP_AfpInfo for newly created files (Time Machine support)
if create_action == FILE_CREATED {
if let Err(e) = crate::afp_monitor::AfpMonitor::init_afp_info(&backend, &path).await {
debug!(path = %path.display_backslash(), error = %e, "Failed to initialize AFP_AfpInfo for new file");
}
}
// Build response with AAPL context if present
let (create_contexts_offset, create_contexts_length, create_contexts) = if let Some(data) = aapl_response_data {
use crate::proto::messages::CreateContext;
+9 -4
View File
@@ -118,10 +118,14 @@ pub async fn handle(
data: signing_data,
};
// ENCRYPTION_CAPABILITIES — advertise AES-128-GCM (simplified)
// ENCRYPTION_CAPABILITIES — advertise AES-128-GCM and AES-128-CCM.
// GCM is preferred (SMB 3.1.1+), CCM is for Windows 8 compat (SMB 3.0).
let encryption_caps = EncryptionCapabilities {
cipher_count: 1,
ciphers: vec![EncryptionCapabilities::CIPHER_AES_128_GCM],
cipher_count: 2,
ciphers: vec![
EncryptionCapabilities::CIPHER_AES_128_GCM,
EncryptionCapabilities::CIPHER_AES_128_CCM,
],
};
let encryption_data = {
use binrw::BinWrite;
@@ -136,7 +140,8 @@ pub async fn handle(
data: encryption_data,
};
// Store encryption support in connection state
// Store encryption support in connection state (default to GCM;
// the actual cipher used per-session is determined during session setup)
*conn.encryption_supported.write().await = true;
*conn.encryption_cipher.write().await = Some(CipherAlgorithm::Aes128Gcm);
+3 -2
View File
@@ -204,9 +204,9 @@ pub async fn handle(
let encryption_cipher = *conn.encryption_cipher.read().await;
let encryption_enabled = encryption_supported && encryption_cipher.is_some();
let encryption_key = if encryption_enabled {
// Derive encryption key from session_base_key (simplified approach)
// Derive encryption key via SP800-108 KDF (MS-SMB2 §3.1.4.2)
use crate::proto::crypto::encryption::Smb3Encryption;
Some(Smb3Encryption::derive_encryption_key(&session_base_key, b"SMB3ENC"))
Some(Smb3Encryption::derive_encryption_key_sp800108(&session_base_key, b"SMB3ENC"))
} else {
None
};
@@ -219,6 +219,7 @@ pub async fn handle(
encryption_key,
signing_required,
encryption_enabled,
encryption_cipher,
None,
);
let session_arc = Arc::new(tokio::sync::RwLock::new(session));
+19 -1
View File
@@ -72,9 +72,27 @@ pub async fn handle(
last_write_time: to_some(write),
change_time: to_some(change),
};
// DOS attributes at bytes 32-35 (FileAttributes field)
let dos_attrs = if buffer.len() >= 36 {
u32::from_le_bytes(buffer[32..36].try_into().unwrap()) & 0xFFFF
} else {
0
};
let open = open_arc.read().await;
match open.handle.as_ref() {
Some(h) => h.set_times(times).await,
Some(h) => {
let r1 = h.set_times(times).await;
if let Err(e) = r1 {
return HandlerResponse::err(e.to_nt_status());
}
if dos_attrs != 0 && dos_attrs != u32::MAX {
let r2 = h.set_attributes(dos_attrs).await;
if let Err(e) = r2 {
return HandlerResponse::err(e.to_nt_status());
}
}
Ok(())
}
None => return HandlerResponse::err(ntstatus::STATUS_FILE_CLOSED),
}
}
+27 -7
View File
@@ -23,6 +23,8 @@ const FILE_ALL_ACCESS: u32 = 0x001F_01FF;
const SMB2_SHAREFLAG_MANUAL_CACHING: u32 = 0x00000000;
const SMB2_SHAREFLAG_ACCESS_BASED_DIRECTORY_ENUM: u32 = 0x00080000;
const SMB2_SHAREFLAG_RESTRICT_EXCLUSIVE_OPLOCKS: u32 = 0x00010000;
const SMB2_SHAREFLAG_FORCE_LEVELII_OPLOCK: u32 = 0x00020000;
const SMB2_SHARE_CAP_DFS: u32 = 0x00000001;
@@ -105,12 +107,26 @@ pub async fn handle(
use crate::path::SmbPath;
let root_path = SmbPath::root();
// Generate UUID for this Time Machine backup
let uuid = uuid::Uuid::new_v4();
let uuid_bytes = uuid.as_bytes();
// Set com.apple.TimeMachine.SupportedFilesStoreUUID
share.backend.set_xattr(&root_path, "com.apple.TimeMachine.SupportedFilesStoreUUID", uuid_bytes).await.ok();
// Reuse existing UUID if present (persists across reconnects)
let uuid = share.backend
.get_xattr(&root_path, "com.apple.TimeMachine.SupportedFilesStoreUUID")
.await
.ok()
.filter(|data| data.len() == 16)
.map(|data| {
let mut bytes = [0u8; 16];
bytes.copy_from_slice(&data);
uuid::Uuid::from_bytes(bytes)
})
.unwrap_or_else(|| {
let new_uuid = uuid::Uuid::new_v4();
let _ = share.backend.set_xattr(
&root_path,
"com.apple.TimeMachine.SupportedFilesStoreUUID",
new_uuid.as_bytes(),
);
new_uuid
});
// Set com.apple.TimeMachine.SupportsThisDevice (1 = true)
share.backend.set_xattr(&root_path, "com.apple.TimeMachine.SupportsThisDevice", &[1]).await.ok();
@@ -124,11 +140,15 @@ pub async fn handle(
tracing::info!(share = %share.name, uuid = %uuid, "Time Machine enabled");
}
let share_flags = if share.is_ipc {
let mut share_flags = if share.is_ipc {
0
} else {
SMB2_SHAREFLAG_MANUAL_CACHING | SMB2_SHAREFLAG_ACCESS_BASED_DIRECTORY_ENUM
};
if share.time_machine {
share_flags |= SMB2_SHAREFLAG_RESTRICT_EXCLUSIVE_OPLOCKS;
share_flags |= SMB2_SHAREFLAG_FORCE_LEVELII_OPLOCK;
}
let capabilities = if share.is_ipc {
0
+7
View File
@@ -102,6 +102,13 @@ pub async fn handle(
Ok(n) => n,
Err(e) => return HandlerResponse::err(e.to_nt_status()),
};
// AFP monitoring: Set modified flag for Time Machine backup tracking
{
let mut open = open_arc.write().await;
open.modified = true;
}
let mut buf = Vec::new();
WriteResponse::new(count)
.write_to(&mut buf)
+7 -1
View File
@@ -337,7 +337,12 @@ pub fn encode_minimal_security_descriptor() -> Vec<u8> {
/// bytes. The caller patches `NextEntryOffset` for chained entries.
pub fn encode_dir_entry(class: u8, entry: &DirEntry, file_index: u64) -> Vec<u8> {
let info = &entry.info;
let name_u16 = utf16le(&info.name);
// Apply reverse Catia mapping (ASCII -> Apple private-range chars) so that
// filenames containing chars illegal in SMB (e.g. `:`, `*`) roundtrip
// correctly for macOS clients.
let units: Vec<u16> = info.name.encode_utf16().collect();
let mapped = crate::unicode_mapping::map_ascii_to_private(&units);
let name_u16: Vec<u8> = mapped.iter().flat_map(|c| c.to_le_bytes()).collect();
match class {
FILE_DIRECTORY_INFORMATION => {
// 64 bytes fixed + name
@@ -430,6 +435,7 @@ mod tests {
change_time: 0x01D9_0000_0000_0000,
is_directory: false,
file_index: 1,
dos_attributes: 0,
}
}
+1
View File
@@ -36,6 +36,7 @@ mod snapshot;
mod unicode_mapping;
mod client_restrictions;
mod utils;
mod afp_monitor;
pub use backend::{BackendCapabilities, DirEntry, FileInfo, FileTimes, Handle, NullHandle, OpenIntent, OpenOptions, ShareBackend};
pub use error::SmbError;
+16
View File
@@ -12,6 +12,11 @@ use crate::error::{SmbError, SmbResult};
/// A validated, component-list path. No `..`, no Windows-forbidden chars, no
/// alternate streams. Always relative to the share root — the empty path is
/// the root.
///
/// ## macOS / Catia support
/// macOS clients encode NTFS-illegal characters (`:*?"<>|`) in the Unicode
/// private range (`U+F001``U+F009`, `U+F02A`). Use [`from_utf16_mac`] to
/// transparently convert these before path validation.
#[derive(Debug, Clone, PartialEq, Eq, Default, Hash)]
pub struct SmbPath {
components: Vec<String>,
@@ -33,6 +38,17 @@ impl SmbPath {
s.parse()
}
/// Construct from UTF-16 with macOS Catia character conversion.
///
/// macOS SMB clients encode NTFS-illegal characters (`:*?"<>|`) in the
/// Unicode private range (`U+F001``U+F009`, `U+F02A`). This method
/// transparently converts them to their ASCII equivalents before path
/// validation. Use this for paths received over AAPL-negotiated sessions.
pub fn from_utf16_mac(units: &[u16]) -> SmbResult<Self> {
let converted = crate::unicode_mapping::map_private_to_ascii(units);
Self::from_utf16(&converted)
}
fn parse_components(s: &str) -> SmbResult<Self> {
// Strip a leading separator (clients sometimes prefix `\` or `/`).
let trimmed = s
+353 -163
View File
@@ -1,17 +1,30 @@
//! SMB3 encryption (AES-128-CTR + HMAC-SHA256)
//! SMB3 encryption AES-128-GCM / AES-128-CCM (MS-SMB2 §2.2.41, §3.1.4.3).
//!
//! Simplified implementation using AES-CTR + HMAC (similar to SSH MtE mode)
//! MS-SMB2 §2.2.41 SMB2 TRANSFORM_HEADER
//! MS-SMB2 §3.1.4.3 Encrypting and Decrypting Messages
//! Uses AEAD modes with the SMB2 TRANSFORM_HEADER as AAD
//! (Additional Authenticated Data). Key derivation follows
//! SP 800-108 CTR-mode KDF (MS-SMB2 §3.1.4.2), re-using the
//! existing [`crate::proto::crypto::kdf::smb2_kdf`] primitive.
//!
//! Supported ciphers:
//! * AES-128-GCM — 12-byte nonce, parallelisable, SMB 3.1.1+ (Windows 10+)
//! * AES-128-CCM — 11-byte nonce, sequential, SMB 3.0 (Windows 8)
use aes::Aes128;
use ctr::Ctr128BE;
use hmac::{Hmac, Mac};
use sha2::Sha256;
use aes_gcm::{
aead::{Aead, KeyInit, Payload as GcmPayload},
Aes128Gcm as Aes128GcmCipher, Nonce as GcmNonce,
};
use binrw::{binrw, BinWrite, BinRead, io::Cursor, Endian};
use ccm::{
aead::{Aead as CcmAead, KeyInit as CcmKeyInit, Payload as CcmPayload},
Ccm as Aes128CcmCipher, Nonce as CcmNonce,
};
use aes::Aes128;
use thiserror::Error;
type HmacSha256 = Hmac<Sha256>;
type Aes128Ccm = Aes128CcmCipher<Aes128, typenum::U16, typenum::U11>;
// Re-export common AEAD traits for callers that need them.
pub use aes_gcm::aead::generic_array::typenum;
#[derive(Debug, Error)]
pub enum EncryptionError {
@@ -29,15 +42,26 @@ pub enum EncryptionError {
NoSessionKey,
}
/// SMB2 TRANSFORM_HEADER (MS-SMB2 §2.2.41) — 56 bytes.
///
/// For AES-128-GCM:
/// * Nonce = 12 bytes (first 12 of the 16-byte field; last 4 reserved).
/// * Signature = GCM authentication tag (16 bytes).
///
/// For AES-128-CCM:
/// * Nonce = 11 bytes (first 11 of the 16-byte field; last 5 reserved).
/// * Signature = CCM authentication tag (16 bytes).
///
/// In both cases AAD = entire header except the signature + encrypted data.
#[binrw]
#[brw(big, magic = 0x534D4220u32)] // "SMB " (big endian for magic)
#[brw(big, magic = 0x534D4272u32)] // "SMBr" — SMB3 encrypted protocol id
pub struct TransformHeader {
#[brw(little)]
pub cipher_algorithm: u16, // 0x0001 = AES-128-GCM, 0x0002 = AES-128-CCM (we use simplified)
pub cipher_algorithm: u16, // 0x0001 = AES-128-GCM, 0x0002 = AES-128-CCM
#[brw(little)]
pub cipher_key_length: u16, // 16 bytes
#[brw(little)]
pub nonce: [u8; 16],
pub nonce: [u8; 16], // 12 (GCM) or 11 (CCM) bytes used, rest reserved
#[brw(little)]
pub session_id: u64,
#[brw(little)]
@@ -46,17 +70,16 @@ pub struct TransformHeader {
pub reserved1: u16,
#[brw(little)]
pub reserved2: u16,
pub signature: [u8; 16], // HMAC-SHA256 tag
pub signature: [u8; 16], // AEAD authentication tag
// EncryptedData follows (variable length)
}
impl TransformHeader {
pub const SIZE: usize = 56; // Header size without encrypted data (4+2+2+16+8+4+2+2+16)
pub const SIZE: usize = 56;
pub fn write_to_bytes(&self) -> Result<Vec<u8>, EncryptionError> {
let mut bytes = Vec::new();
// Write magic in big endian, rest in little endian
bytes.extend_from_slice(&0x534D4220u32.to_be_bytes()); // "SMB "
bytes.extend_from_slice(&0x534D4272u32.to_be_bytes());
bytes.extend_from_slice(&self.cipher_algorithm.to_le_bytes());
bytes.extend_from_slice(&self.cipher_key_length.to_le_bytes());
bytes.extend_from_slice(&self.nonce);
@@ -67,18 +90,19 @@ impl TransformHeader {
bytes.extend_from_slice(&self.signature);
Ok(bytes)
}
pub fn read_from_bytes(data: &[u8]) -> Result<Self, EncryptionError> {
if data.len() < Self::SIZE {
return Err(EncryptionError::DecryptionFailed("Header too short".to_string()));
return Err(EncryptionError::DecryptionFailed(
"Header too short".to_string(),
));
}
// Check magic
let magic = u32::from_be_bytes([data[0], data[1], data[2], data[3]]);
if magic != 0x534D4220 {
if magic != 0x534D4272 {
return Err(EncryptionError::InvalidSignature);
}
Ok(Self {
cipher_algorithm: u16::from_le_bytes([data[4], data[5]]),
cipher_key_length: u16::from_le_bytes([data[6], data[7]]),
@@ -98,6 +122,20 @@ impl TransformHeader {
},
})
}
/// Build AAD = header[0..52], i.e. everything before `signature`.
fn build_aad(&self) -> Vec<u8> {
let mut buf = Vec::with_capacity(40);
buf.extend_from_slice(&0x534D4272u32.to_be_bytes());
buf.extend_from_slice(&self.cipher_algorithm.to_le_bytes());
buf.extend_from_slice(&self.cipher_key_length.to_le_bytes());
buf.extend_from_slice(&self.nonce);
buf.extend_from_slice(&self.session_id.to_le_bytes());
buf.extend_from_slice(&self.original_message_size.to_le_bytes());
buf.extend_from_slice(&self.reserved1.to_le_bytes());
buf.extend_from_slice(&self.reserved2.to_le_bytes());
buf
}
}
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
@@ -114,192 +152,344 @@ impl CipherAlgorithm {
_ => None,
}
}
pub fn key_length(&self) -> u16 {
16 // AES-128
16
}
/// Number of nonce bytes used by this cipher.
pub fn nonce_length(&self) -> usize {
match self {
CipherAlgorithm::Aes128Gcm => 12,
CipherAlgorithm::Aes128Ccm => 11,
}
}
}
/// Per-session SMB3 encryption helper.
///
/// Supports both AES-128-GCM (SMB 3.1.1+) and AES-128-CCM (SMB 3.0).
pub struct Smb3Encryption {
encryption_key: [u8; 16],
mac_key: [u8; 32],
cipher_algorithm: CipherAlgorithm,
cipher: CipherAlgorithm,
}
impl Smb3Encryption {
/// Create a new encryption context from the session key and cipher.
///
/// Derives the AES-128 key via SP 800-108 KDF.
pub fn new(session_key: &[u8], cipher_algorithm: CipherAlgorithm) -> Result<Self, EncryptionError> {
if session_key.len() != 16 {
return Err(EncryptionError::InvalidKeyLength);
}
// Derive encryption_key and mac_key from session_key
let encryption_key = Self::derive_encryption_key(session_key, b"SMB3ENC");
let mac_key = Self::derive_mac_key(session_key, b"SMB3MAC");
let encryption_key = Self::derive_encryption_key_sp800108(session_key, b"SMB3ENC");
Ok(Self {
encryption_key,
mac_key,
cipher_algorithm,
cipher: cipher_algorithm,
})
}
/// Encrypt a plaintext SMB2 message.
///
/// Returns a complete SMB3 TRANSFORM_HEADER + encrypted payload.
pub fn encrypt_packet(&self, plaintext: &[u8], session_id: u64) -> Result<Vec<u8>, EncryptionError> {
let nonce_bytes = self.generate_nonce();
// 1. Compute HMAC over plaintext + header info (MtE mode)
let tag = self.compute_mac(plaintext, session_id, &nonce_bytes);
// 2. Encrypt plaintext with AES-CTR
let encrypted_data = self.encrypt_aes_ctr(plaintext, &nonce_bytes);
let header = TransformHeader {
cipher_algorithm: self.cipher_algorithm as u16,
let nonce_len = self.cipher.nonce_length();
// Generate random nonce, pad to 16 bytes in the header
let mut nonce_full = [0u8; 16];
getrandom::fill(&mut nonce_full[..nonce_len])
.map_err(|e| EncryptionError::EncryptionFailed(format!("nonce: {}", e)))?;
let header_no_tag = TransformHeader {
cipher_algorithm: self.cipher as u16,
cipher_key_length: 16,
nonce: nonce_bytes,
nonce: nonce_full,
session_id,
original_message_size: plaintext.len() as u32,
reserved1: 0,
reserved2: 0,
signature: tag,
signature: [0u8; 16],
};
let aad = header_no_tag.build_aad();
// AEAD encrypt: returns ciphertext || tag (last 16 bytes)
let ciphertext_with_tag = match self.cipher {
CipherAlgorithm::Aes128Gcm => {
let nonce12 = GcmNonce::from_slice(&nonce_full[..12]);
let cipher = Aes128GcmCipher::new_from_slice(&self.encryption_key)
.map_err(|e| EncryptionError::EncryptionFailed(format!("GCM key: {}", e)))?;
cipher
.encrypt(nonce12, GcmPayload { msg: plaintext, aad: &aad })
.map_err(|e| EncryptionError::EncryptionFailed(format!("GCM encrypt: {}", e)))?
}
CipherAlgorithm::Aes128Ccm => {
let nonce11 = CcmNonce::from_slice(&nonce_full[..11]);
let cipher = Aes128Ccm::new_from_slice(&self.encryption_key)
.map_err(|e| EncryptionError::EncryptionFailed(format!("CCM key: {}", e)))?;
cipher
.encrypt(nonce11, CcmPayload { msg: plaintext, aad: &aad })
.map_err(|e| EncryptionError::EncryptionFailed(format!("CCM encrypt: {}", e)))?
}
};
let tag_len = 16;
let tag_pos = ciphertext_with_tag.len().saturating_sub(tag_len);
let tag: [u8; 16] = ciphertext_with_tag[tag_pos..]
.try_into()
.map_err(|_| EncryptionError::EncryptionFailed("tag extraction".to_string()))?;
let encrypted_data = &ciphertext_with_tag[..tag_pos];
let header = TransformHeader {
signature: tag,
..header_no_tag
};
let mut packet = header.write_to_bytes()?;
packet.extend_from_slice(&encrypted_data);
packet.extend_from_slice(encrypted_data);
Ok(packet)
}
/// Decrypt an SMB3 TRANSFORM_HEADER payload.
///
/// The cipher algorithm is read from the header's `cipher_algorithm` field,
/// so this is dispatch-safe — callers don't need to match the algorithm.
pub fn decrypt_packet(&self, encrypted_packet: &[u8]) -> Result<Vec<u8>, EncryptionError> {
let header = TransformHeader::read_from_bytes(encrypted_packet)?;
let encrypted_data = &encrypted_packet[TransformHeader::SIZE..];
// 1. Decrypt with AES-CTR
let plaintext = self.decrypt_aes_ctr(encrypted_data, &header.nonce);
// 2. Verify HMAC
let expected_tag = self.compute_mac(&plaintext, header.session_id, &header.nonce);
if header.signature != expected_tag {
return Err(EncryptionError::InvalidSignature);
// Determine cipher from header (prefer the stored self.cipher but
// also verify the header's opinion matches).
let cipher = CipherAlgorithm::from_u16(header.cipher_algorithm)
.unwrap_or(self.cipher);
let _nonce_len = cipher.nonce_length();
let aad = header.build_aad();
// Build ciphertext_with_tag for AEAD verification
let mut ct_with_tag = encrypted_data.to_vec();
ct_with_tag.extend_from_slice(&header.signature);
match cipher {
CipherAlgorithm::Aes128Gcm => {
let mut nonce_buf = [0u8; 12];
nonce_buf.copy_from_slice(&header.nonce[..12]);
let nonce12 = GcmNonce::from_slice(&nonce_buf);
let cipher = Aes128GcmCipher::new_from_slice(&self.encryption_key)
.map_err(|e| EncryptionError::DecryptionFailed(format!("GCM key: {}", e)))?;
cipher
.decrypt(nonce12, GcmPayload { msg: &ct_with_tag, aad: &aad })
.map_err(|_| EncryptionError::InvalidSignature)
}
CipherAlgorithm::Aes128Ccm => {
let mut nonce_buf = [0u8; 11];
nonce_buf.copy_from_slice(&header.nonce[..11]);
let nonce11 = CcmNonce::from_slice(&nonce_buf);
let cipher = Aes128Ccm::new_from_slice(&self.encryption_key)
.map_err(|e| EncryptionError::DecryptionFailed(format!("CCM key: {}", e)))?;
cipher
.decrypt(nonce11, CcmPayload { msg: &ct_with_tag, aad: &aad })
.map_err(|_| EncryptionError::InvalidSignature)
}
}
Ok(plaintext)
}
fn encrypt_aes_ctr(&self, plaintext: &[u8], nonce: &[u8; 16]) -> Vec<u8> {
use aes::cipher::{KeyIvInit, StreamCipher};
let key = aes::cipher::generic_array::GenericArray::from_slice(&self.encryption_key);
let iv = aes::cipher::generic_array::GenericArray::from_slice(nonce);
let mut cipher = Ctr128BE::<Aes128>::new(key, iv);
let mut ciphertext = plaintext.to_vec();
cipher.apply_keystream(&mut ciphertext);
ciphertext
}
fn decrypt_aes_ctr(&self, ciphertext: &[u8], nonce: &[u8; 16]) -> Vec<u8> {
self.encrypt_aes_ctr(ciphertext, nonce) // CTR is symmetric
}
fn compute_mac(&self, data: &[u8], session_id: u64, nonce: &[u8; 16]) -> [u8; 16] {
let mut mac = <HmacSha256 as Mac>::new_from_slice(&self.mac_key)
.expect("HMAC key length is valid");
// MAC over: nonce + session_id + data
mac.update(nonce);
mac.update(&session_id.to_le_bytes());
mac.update(data);
let result = mac.finalize();
let mut tag = [0u8; 16];
tag.copy_from_slice(&result.into_bytes()[..16]);
tag
}
fn generate_nonce(&self) -> [u8; 16] {
let mut nonce = [0u8; 16];
getrandom::fill(&mut nonce).ok();
nonce
}
pub fn derive_encryption_key(session_key: &[u8], context: &[u8]) -> [u8; 16] {
use sha2::{Sha256, Digest};
let mut hasher = Sha256::new();
hasher.update(session_key);
hasher.update(context);
let result = hasher.finalize();
let mut key = [0u8; 16];
key.copy_from_slice(&result[..16]);
key
}
fn derive_mac_key(session_key: &[u8], context: &[u8]) -> [u8; 32] {
use sha2::{Sha256, Digest};
let mut hasher = Sha256::new();
hasher.update(session_key);
hasher.update(context);
let result = hasher.finalize();
let mut key = [0u8; 32];
key.copy_from_slice(&result[..32]);
key
/// Derive AES-128 encryption key via SP 800-108 KDF.
///
/// Uses the existing [`crate::proto::crypto::kdf::smb2_kdf`] with
/// Label = `label` (caller includes trailing NUL), Context = empty.
///
/// MS-SMB2 §3.1.4.2: `encryption_key = KDF(session_key, label, "")`.
pub fn derive_encryption_key_sp800108(session_key: &[u8], label: &[u8]) -> [u8; 16] {
let mut label_with_nul = label.to_vec();
label_with_nul.push(0x00);
let context_with_nul = b"\x00";
crate::proto::crypto::kdf::smb2_kdf(session_key, &label_with_nul, context_with_nul)
}
}
#[cfg(test)]
mod tests {
use super::*;
fn test_encrypt_decrypt_roundtrip(cipher: CipherAlgorithm) {
let session_key = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16];
let enc = Smb3Encryption::new(&session_key, cipher).unwrap();
let plaintext = b"Hello SMB3!";
let session_id = 12345u64;
let encrypted = enc.encrypt_packet(plaintext, session_id).unwrap();
assert_eq!(encrypted.len(), TransformHeader::SIZE + plaintext.len());
let magic = u32::from_be_bytes([encrypted[0], encrypted[1], encrypted[2], encrypted[3]]);
assert_eq!(magic, 0x534D4272);
// Verify cipher_algorithm field in header
let header_cipher = u16::from_le_bytes([encrypted[4], encrypted[5]]);
assert_eq!(header_cipher, cipher as u16);
let decrypted = enc.decrypt_packet(&encrypted).unwrap();
assert_eq!(plaintext.as_slice(), decrypted.as_slice());
}
#[test]
fn test_gcm_roundtrip() {
test_encrypt_decrypt_roundtrip(CipherAlgorithm::Aes128Gcm);
}
#[test]
fn test_ccm_roundtrip() {
test_encrypt_decrypt_roundtrip(CipherAlgorithm::Aes128Ccm);
}
#[test]
fn test_gcm_and_ccm_interop() {
// Verify packets encrypted with different ciphers produce different wire output
let session_key = [1u8; 16];
let plaintext = b"Cross-cipher test";
let gcm_enc = Smb3Encryption::new(&session_key, CipherAlgorithm::Aes128Gcm).unwrap();
let ccm_enc = Smb3Encryption::new(&session_key, CipherAlgorithm::Aes128Ccm).unwrap();
let gcm_packet = gcm_enc.encrypt_packet(plaintext, 1).unwrap();
let ccm_packet = ccm_enc.encrypt_packet(plaintext, 1).unwrap();
// Different cipher algorithm IDs in the header
assert_eq!(
u16::from_le_bytes([gcm_packet[4], gcm_packet[5]]),
CipherAlgorithm::Aes128Gcm as u16
);
assert_eq!(
u16::from_le_bytes([ccm_packet[4], ccm_packet[5]]),
CipherAlgorithm::Aes128Ccm as u16
);
// Ciphertext differs (different nonce length → different keystream offset)
assert_ne!(gcm_packet, ccm_packet);
// Each cipher can decrypt its own packet via the header-based dispatch
assert!(gcm_enc.decrypt_packet(&gcm_packet).is_ok());
assert!(ccm_enc.decrypt_packet(&ccm_packet).is_ok());
}
#[test]
fn test_cipher_algorithm_conversion() {
assert_eq!(CipherAlgorithm::from_u16(0x0001), Some(CipherAlgorithm::Aes128Gcm));
assert_eq!(CipherAlgorithm::from_u16(0x0002), Some(CipherAlgorithm::Aes128Ccm));
assert_eq!(CipherAlgorithm::from_u16(0x0003), None);
}
#[test]
fn test_encrypt_decrypt_roundtrip() {
let session_key = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16];
let encryption = Smb3Encryption::new(&session_key, CipherAlgorithm::Aes128Gcm).unwrap();
let plaintext = b"Hello SMB3!";
let session_id = 12345u64;
let encrypted = encryption.encrypt_packet(plaintext, session_id).unwrap();
// Debug: check header size
assert_eq!(encrypted.len(), TransformHeader::SIZE + plaintext.len());
// Debug: check magic
let magic = u32::from_be_bytes([encrypted[0], encrypted[1], encrypted[2], encrypted[3]]);
assert_eq!(magic, 0x534D4220);
let decrypted = encryption.decrypt_packet(&encrypted).unwrap();
assert_eq!(plaintext.as_slice(), decrypted.as_slice());
}
#[test]
fn test_invalid_signature_detection() {
let session_key = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16];
let encryption = Smb3Encryption::new(&session_key, CipherAlgorithm::Aes128Gcm).unwrap();
let plaintext = b"Hello SMB3!";
let session_id = 12345u64;
let encrypted = encryption.encrypt_packet(plaintext, session_id).unwrap();
// Tamper with signature
fn test_gcm_authentication_failure() {
let session_key = [1u8; 16];
let enc = Smb3Encryption::new(&session_key, CipherAlgorithm::Aes128Gcm).unwrap();
let encrypted = enc.encrypt_packet(b"Test data", 999).unwrap();
let mut tampered = encrypted.clone();
tampered[48] ^= 0xFF; // Modify signature byte
let result = encryption.decrypt_packet(&tampered);
tampered[TransformHeader::SIZE] ^= 0xFF;
let result = enc.decrypt_packet(&tampered);
assert!(result.is_err());
assert_eq!(result.unwrap_err().to_string(), "Invalid transform header signature");
}
}
#[test]
fn test_ccm_authentication_failure() {
let session_key = [1u8; 16];
let enc = Smb3Encryption::new(&session_key, CipherAlgorithm::Aes128Ccm).unwrap();
let encrypted = enc.encrypt_packet(b"Test data", 999).unwrap();
let mut tampered = encrypted.clone();
tampered[TransformHeader::SIZE] ^= 0xFF;
let result = enc.decrypt_packet(&tampered);
assert!(result.is_err());
assert_eq!(result.unwrap_err().to_string(), "Invalid transform header signature");
}
#[test]
fn test_gcm_tag_tampering() {
let session_key = [1u8; 16];
let enc = Smb3Encryption::new(&session_key, CipherAlgorithm::Aes128Gcm).unwrap();
let encrypted = enc.encrypt_packet(b"Test data", 999).unwrap();
let mut tampered = encrypted;
tampered[48] ^= 0xFF;
assert!(enc.decrypt_packet(&tampered).is_err());
}
#[test]
fn test_ccm_tag_tampering() {
let session_key = [1u8; 16];
let enc = Smb3Encryption::new(&session_key, CipherAlgorithm::Aes128Ccm).unwrap();
let encrypted = enc.encrypt_packet(b"Test data", 999).unwrap();
let mut tampered = encrypted;
tampered[48] ^= 0xFF;
assert!(enc.decrypt_packet(&tampered).is_err());
}
#[test]
fn test_nonce_uniqueness() {
let session_key = [1u8; 16];
let enc = Smb3Encryption::new(&session_key, CipherAlgorithm::Aes128Gcm).unwrap();
let p1 = enc.encrypt_packet(b"Same data", 1).unwrap();
let p2 = enc.encrypt_packet(b"Same data", 2).unwrap();
let nonce1: [u8; 16] = p1[8..24].try_into().unwrap();
let nonce2: [u8; 16] = p2[8..24].try_into().unwrap();
assert_ne!(nonce1, nonce2);
}
#[test]
fn test_ccm_nonce_length() {
// CCM uses 11-byte nonce (verify the header stores it correctly)
let session_key = [1u8; 16];
let enc = Smb3Encryption::new(&session_key, CipherAlgorithm::Aes128Ccm).unwrap();
let encrypted = enc.encrypt_packet(b"nonce test", 1).unwrap();
// The header nonce field is always 16 bytes, but CCM only uses 11
let nonce: [u8; 16] = encrypted[8..24].try_into().unwrap();
// Bytes 11-15 should be zero (padding/reserved)
assert_eq!(&nonce[11..], &[0, 0, 0, 0, 0]);
}
#[test]
fn test_gcm_nonce_length() {
// GCM uses 12-byte nonce
let session_key = [1u8; 16];
let enc = Smb3Encryption::new(&session_key, CipherAlgorithm::Aes128Gcm).unwrap();
let encrypted = enc.encrypt_packet(b"nonce test", 1).unwrap();
let nonce: [u8; 16] = encrypted[8..24].try_into().unwrap();
// Bytes 12-15 should be zero
assert_eq!(&nonce[12..], &[0, 0, 0, 0]);
}
#[test]
fn test_sp800108_kdf_known_answer() {
let session_key = [0u8; 16];
let key = Smb3Encryption::derive_encryption_key_sp800108(&session_key, b"SMB3ENC");
let label = b"SMB3ENC\x00";
let context = b"\x00";
let expected = crate::proto::crypto::kdf::smb2_kdf(&session_key, label, context);
assert_eq!(key, expected);
assert_ne!(key, [0u8; 16]);
}
#[test]
fn test_different_sessions_different_keys() {
let key1 = Smb3Encryption::derive_encryption_key_sp800108(&[1u8; 16], b"SMB3ENC");
let key2 = Smb3Encryption::derive_encryption_key_sp800108(&[2u8; 16], b"SMB3ENC");
assert_ne!(key1, key2);
}
}
+59 -2
View File
@@ -23,20 +23,30 @@ pub const SMB2_CRTCTX_AAPL_SUPPORT_RESOLVE_ID: u64 = 1;
pub const SMB2_CRTCTX_AAPL_CASE_SENSITIVE: u64 = 2;
pub const SMB2_CRTCTX_AAPL_FULL_SYNC: u64 = 4;
/// AAPL Create Context Request (24 bytes)
/// AAPL Create Context Request (24 bytes, or 32 for RESOLVE_ID)
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct AaplCreateContextRequest {
pub command: u32,
pub reserved: u32,
pub request_bitmap: u64,
pub client_caps: u64,
/// RESOLVE_ID: file ID to resolve (8 bytes LE)
pub resolve_file_id: Option<u64>,
}
impl AaplCreateContextRequest {
pub fn from_bytes(data: &[u8]) -> Option<Self> {
if data.len() != 24 {
if data.len() != 24 && data.len() != 32 {
return None;
}
let resolve_file_id = if data.len() >= 32 {
Some(u64::from_le_bytes([
data[24], data[25], data[26], data[27],
data[28], data[29], data[30], data[31],
]))
} else {
None
};
Some(Self {
command: u32::from_le_bytes([data[0], data[1], data[2], data[3]]),
reserved: u32::from_le_bytes([data[4], data[5], data[6], data[7]]),
@@ -48,6 +58,7 @@ impl AaplCreateContextRequest {
data[16], data[17], data[18], data[19],
data[20], data[21], data[22], data[23],
]),
resolve_file_id,
})
}
}
@@ -108,6 +119,25 @@ impl AaplCreateContextResponse {
}
}
/// Build a RESOLVE_ID response bytes.
///
/// Format (after 24-byte AAPL header):
/// PathLength (4 bytes LE) + Path (UTF-16LE)
pub fn build_resolve_id_response(path: &str) -> Vec<u8> {
let mut buf = Vec::new();
// AAPL header: command=RESOLVE_ID, reserved=0, request_bitmap=0
buf.extend_from_slice(&SMB2_CRTCTX_AAPL_RESOLVE_ID.to_le_bytes());
buf.extend_from_slice(&[0u8; 4]); // reserved
buf.extend_from_slice(&[0u8; 8]); // request_bitmap
// Path
let path_utf16: Vec<u16> = path.encode_utf16().collect();
buf.extend_from_slice(&(path_utf16.len() as u32 * 2).to_le_bytes());
for ch in path_utf16 {
buf.extend_from_slice(&ch.to_le_bytes());
}
buf
}
#[cfg(test)]
mod tests {
use super::*;
@@ -125,6 +155,33 @@ mod tests {
assert_eq!(req.request_bitmap, 7);
}
#[test]
fn test_aapl_resolve_id_request() {
let mut data = [0u8; 32];
data[0..4].copy_from_slice(&2u32.to_le_bytes()); // command = RESOLVE_ID
data[24..32].copy_from_slice(&0x12345678u64.to_le_bytes()); // file_id
let req = AaplCreateContextRequest::from_bytes(&data).unwrap();
assert_eq!(req.command, SMB2_CRTCTX_AAPL_RESOLVE_ID);
assert_eq!(req.resolve_file_id, Some(0x12345678));
}
#[test]
fn test_build_resolve_id_response() {
let bytes = build_resolve_id_response("dir/file.txt");
// header: command=2 (4B) + reserved=0 (4B) + request_bitmap=0 (8B) = 16 bytes
assert_eq!(&bytes[0..4], &[2, 0, 0, 0]);
// path length (UTF-16 = each char 2 bytes, 12 chars = 24 bytes)
let path_len = u32::from_le_bytes([bytes[16], bytes[17], bytes[18], bytes[19]]);
assert_eq!(path_len, 24);
// path content
let path_utf16: Vec<u16> = bytes[20..]
.chunks(2)
.map(|c| u16::from_le_bytes([c[0], c[1]]))
.collect();
let path = String::from_utf16(&path_utf16).unwrap();
assert_eq!(path, "dir/file.txt");
}
#[test]
fn test_aapl_response_encode() {
let resp = AaplCreateContextResponse::new_server_query(
+94
View File
@@ -4,6 +4,8 @@
//! for Windows VSS (Volume Shadow Copy Service) support.
use std::collections::HashMap;
use std::fmt::Write;
use std::path::PathBuf;
use std::sync::{Arc, RwLock};
use std::time::SystemTime;
@@ -77,19 +79,109 @@ pub enum SnapshotResponse {
},
}
const SNAPSHOTS_DIR: &str = ".snapshots";
const SNAPSHOTS_FILE: &str = "snapshots.json";
/// Snapshot manager - manages share snapshots
pub struct SnapshotManager {
/// Snapshots indexed by (share_name, snapshot_id)
snapshots: RwLock<HashMap<(String, String), SnapshotEntry>>,
/// Optional file-system path for persistence
storage_path: Option<PathBuf>,
}
impl SnapshotManager {
pub fn new() -> Self {
Self {
snapshots: RwLock::new(HashMap::new()),
storage_path: None,
}
}
pub fn with_storage_path(path: PathBuf) -> Self {
let manager = Self {
snapshots: RwLock::new(HashMap::new()),
storage_path: Some(path),
};
manager.load_snapshots();
manager
}
fn snapshots_file_path(&self) -> Option<PathBuf> {
self.storage_path.as_ref().map(|p| p.join(SNAPSHOTS_DIR).join(SNAPSHOTS_FILE))
}
fn load_snapshots(&self) {
let path = match self.snapshots_file_path() {
Some(p) => p,
None => return,
};
let data = match std::fs::read_to_string(&path) {
Ok(d) => d,
Err(_) => return,
};
let mut map = self.snapshots.write().unwrap();
for line in data.lines() {
let parts: Vec<&str> = line.splitn(5, '|').collect();
if parts.len() < 4 {
continue;
}
let share_name = parts[0].to_string();
let snapshot_id = parts[1].to_string();
let secs: u64 = match parts[2].parse() {
Ok(s) => s,
Err(_) => continue,
};
let state = match parts[3] {
"Created" => SnapshotState::Created,
"Active" => SnapshotState::Active,
"Deleting" => SnapshotState::Deleting,
"Deleted" => SnapshotState::Deleted,
_ => continue,
};
let metadata = parts.get(4).filter(|m| !m.is_empty()).map(|m| m.to_string());
let created_at = std::time::UNIX_EPOCH + std::time::Duration::from_secs(secs);
let entry = SnapshotEntry {
snapshot_id,
share_name: share_name.clone(),
created_at,
state,
metadata,
};
map.insert((share_name, entry.snapshot_id.clone()), entry);
}
}
fn save_snapshots(&self) {
let path = match self.snapshots_file_path() {
Some(p) => p,
None => return,
};
if let Some(parent) = path.parent() {
let _ = std::fs::create_dir_all(parent);
}
let mut output = String::new();
{
let map = self.snapshots.read().unwrap();
for entry in map.values() {
let secs = entry.created_at
.duration_since(std::time::UNIX_EPOCH)
.unwrap_or(std::time::Duration::ZERO)
.as_secs();
let state_str = match entry.state {
SnapshotState::Created => "Created",
SnapshotState::Active => "Active",
SnapshotState::Deleting => "Deleting",
SnapshotState::Deleted => "Deleted",
};
let meta = entry.metadata.as_deref().unwrap_or("");
writeln!(output, "{}|{}|{}|{}|{}",
entry.share_name, entry.snapshot_id, secs, state_str, meta).ok();
}
}
let _ = std::fs::write(&path, &output);
}
/// Create a new snapshot for a share
pub fn create_snapshot(
&self,
@@ -115,6 +207,7 @@ impl SnapshotManager {
.unwrap()
.insert((share_name.to_string(), snapshot_id.clone()), entry.clone());
self.save_snapshots();
Ok(entry)
}
@@ -151,6 +244,7 @@ impl SnapshotManager {
entry.state = SnapshotState::Deleted;
snapshots.remove(&(share_name.to_string(), snapshot_id.to_string()));
self.save_snapshots();
Ok(())
}
+1 -1
View File
@@ -38,7 +38,7 @@ async fn register_session(
));
state.active_connections.register(&conn).await;
let session = Session::new(1, identity, [0; 16], [0; 16], None, false, false, None);
let session = Session::new(1, identity, [0; 16], [0; 16], None, false, false, None, None);
let session = Arc::new(tokio::sync::RwLock::new(session));
let share = state.find_share(share_name).await.expect("share");
let tree = Arc::new(tokio::sync::RwLock::new(TreeConnect::new(
+3
View File
@@ -224,6 +224,7 @@ impl Handle for MemHandle {
change_time: 0x01D9_0000_0000_0000,
is_directory: self.is_dir,
file_index: 0,
dos_attributes: 0,
})
}
@@ -267,6 +268,7 @@ impl Handle for MemHandle {
change_time: 0x01D9_0000_0000_0000,
is_directory: false,
file_index: 0,
dos_attributes: 0,
},
});
}
@@ -287,6 +289,7 @@ impl Handle for MemHandle {
change_time: 0x01D9_0000_0000_0000,
is_directory: true,
file_index: 0,
dos_attributes: 0,
},
});
}
+79 -18
View File
@@ -1,19 +1,34 @@
//! macOS Unicode Private Range Mapping for SMB
//!
//! macOS SMB client maps NTFS illegal characters to Unicode private range.
//! Reference: Samba vfs_fruit.c encoding handling
//! Reference: Samba vfs_catia.c and vfs_fruit.c encoding handling
//!
//! Full mapping table (Samba catia standard):
//! U+F001 → / (0x2F)
//! U+F002 → : (0x3A)
//! U+F003 → * (0x2A)
//! U+F004 → ? (0x3F)
//! U+F005 → " (0x22)
//! U+F006 → < (0x3C)
//! U+F007 → > (0x3E)
//! U+F008 → | (0x7C)
//! U+F009 → \ (0x5C)
//! U+F02A → : (0x3A) — macOS Finder uses this for colon
pub const FRUIT_ENC_NATIVE: bool = true;
pub const FRUIT_ENC_PRIVATE: bool = false;
const APPLE_SLASH: u16 = 0xF026;
const APPLE_COLON: u16 = 0xF02A;
const APPLE_ASTERISK: u16 = 0xF02B;
const APPLE_QUESTION: u16 = 0xF03F;
const APPLE_QUOTE: u16 = 0xF022;
const APPLE_LESS_THAN: u16 = 0xF03C;
const APPLE_GREATER_THAN: u16 = 0xF03E;
const APPLE_PIPE: u16 = 0xF07C;
// Apple private range code points (vfs_catia mapping)
const APPLE_SLASH: u16 = 0xF001;
const APPLE_COLON_ALT: u16 = 0xF002;
const APPLE_ASTERISK: u16 = 0xF003;
const APPLE_QUESTION: u16 = 0xF004;
const APPLE_QUOTE: u16 = 0xF005;
const APPLE_LESS_THAN: u16 = 0xF006;
const APPLE_GREATER_THAN: u16 = 0xF007;
const APPLE_PIPE: u16 = 0xF008;
const APPLE_BACKSLASH: u16 = 0xF009;
const APPLE_COLON: u16 = 0xF02A; // macOS Finder specific
const ASCII_SLASH: u16 = '/' as u16;
const ASCII_COLON: u16 = ':' as u16;
@@ -23,18 +38,30 @@ const ASCII_QUOTE: u16 = '"' as u16;
const ASCII_LESS_THAN: u16 = '<' as u16;
const ASCII_GREATER_THAN: u16 = '>' as u16;
const ASCII_PIPE: u16 = '|' as u16;
const ASCII_BACKSLASH: u16 = '\\' as u16;
/// Check if a UTF-16 code unit is in the macOS private range.
pub fn is_private_range_char(u: u16) -> bool {
matches!(u,
APPLE_SLASH | APPLE_COLON_ALT | APPLE_ASTERISK |
APPLE_QUESTION | APPLE_QUOTE | APPLE_LESS_THAN |
APPLE_GREATER_THAN | APPLE_PIPE | APPLE_BACKSLASH |
APPLE_COLON
)
}
pub fn map_private_to_ascii(units: &[u16]) -> Vec<u16> {
units.iter().map(|u| {
match *u {
APPLE_SLASH => ASCII_SLASH,
APPLE_COLON => ASCII_COLON,
APPLE_COLON | APPLE_COLON_ALT => ASCII_COLON,
APPLE_ASTERISK => ASCII_ASTERISK,
APPLE_QUESTION => ASCII_QUESTION,
APPLE_QUOTE => ASCII_QUOTE,
APPLE_LESS_THAN => ASCII_LESS_THAN,
APPLE_GREATER_THAN => ASCII_GREATER_THAN,
APPLE_PIPE => ASCII_PIPE,
APPLE_BACKSLASH => ASCII_BACKSLASH,
_ => *u,
}
}).collect()
@@ -51,19 +78,14 @@ pub fn map_ascii_to_private(units: &[u16]) -> Vec<u16> {
ASCII_LESS_THAN => APPLE_LESS_THAN,
ASCII_GREATER_THAN => APPLE_GREATER_THAN,
ASCII_PIPE => APPLE_PIPE,
ASCII_BACKSLASH => APPLE_BACKSLASH,
_ => *u,
}
}).collect()
}
pub fn has_private_range_chars(units: &[u16]) -> bool {
units.iter().any(|u| {
matches!(*u,
APPLE_SLASH | APPLE_COLON | APPLE_ASTERISK |
APPLE_QUESTION | APPLE_QUOTE | APPLE_LESS_THAN |
APPLE_GREATER_THAN | APPLE_PIPE
)
})
units.iter().any(|u| is_private_range_char(*u))
}
pub fn has_ntfs_illegal_chars(units: &[u16]) -> bool {
@@ -71,7 +93,7 @@ pub fn has_ntfs_illegal_chars(units: &[u16]) -> bool {
matches!(*u,
ASCII_SLASH | ASCII_COLON | ASCII_ASTERISK |
ASCII_QUESTION | ASCII_QUOTE | ASCII_LESS_THAN |
ASCII_GREATER_THAN | ASCII_PIPE
ASCII_GREATER_THAN | ASCII_PIPE | ASCII_BACKSLASH
)
})
}
@@ -87,6 +109,23 @@ mod tests {
assert_eq!(output, [ASCII_SLASH, ASCII_COLON, ASCII_QUESTION]);
}
#[test]
fn test_map_private_to_ascii_all() {
let input = [
APPLE_SLASH, APPLE_COLON_ALT, APPLE_ASTERISK,
APPLE_QUESTION, APPLE_QUOTE, APPLE_LESS_THAN,
APPLE_GREATER_THAN, APPLE_PIPE, APPLE_BACKSLASH,
APPLE_COLON,
];
let output = map_private_to_ascii(&input);
assert_eq!(output, [
ASCII_SLASH, ASCII_COLON, ASCII_ASTERISK,
ASCII_QUESTION, ASCII_QUOTE, ASCII_LESS_THAN,
ASCII_GREATER_THAN, ASCII_PIPE, ASCII_BACKSLASH,
ASCII_COLON,
]);
}
#[test]
fn test_map_ascii_to_private() {
let input = [ASCII_SLASH, ASCII_COLON, ASCII_ASTERISK];
@@ -94,6 +133,21 @@ mod tests {
assert_eq!(output, [APPLE_SLASH, APPLE_COLON, APPLE_ASTERISK]);
}
#[test]
fn test_map_ascii_to_private_all() {
let input = [
ASCII_SLASH, ASCII_COLON, ASCII_ASTERISK,
ASCII_QUESTION, ASCII_QUOTE, ASCII_LESS_THAN,
ASCII_GREATER_THAN, ASCII_PIPE, ASCII_BACKSLASH,
];
let output = map_ascii_to_private(&input);
assert_eq!(output, [
APPLE_SLASH, APPLE_COLON, APPLE_ASTERISK,
APPLE_QUESTION, APPLE_QUOTE, APPLE_LESS_THAN,
APPLE_GREATER_THAN, APPLE_PIPE, APPLE_BACKSLASH,
]);
}
#[test]
fn test_roundtrip() {
let original = [ASCII_SLASH, ASCII_COLON, 'a' as u16];
@@ -120,4 +174,11 @@ mod tests {
let output = map_private_to_ascii(&input);
assert_eq!(output, input);
}
#[test]
fn test_is_private_range_char() {
assert!(is_private_range_char(APPLE_SLASH));
assert!(is_private_range_char(APPLE_COLON));
assert!(!is_private_range_char('a' as u16));
}
}