use std::io::SeekFrom;
use binrw::io::TakeSeekExt;
use binrw::prelude::*;
use modular_bitfield::prelude::*;
use super::FileId;
use smb_dtyp::binrw_util::prelude::*;
use smb_fscc::*;
#[binrw::binrw]
#[derive(Debug)]
pub struct ChangeNotifyRequest {
#[bw(calc = 32)]
#[br(assert(_structure_size == 32))]
_structure_size: u16,
pub flags: NotifyFlags,
pub output_buffer_length: u32,
pub file_id: FileId,
pub completion_filter: NotifyFilter,
#[bw(calc = 0)]
_reserved: u32,
}
#[bitfield]
#[derive(BinWrite, BinRead, Debug, Default, Clone, Copy, PartialEq, Eq)]
#[bw(map = |&x| Self::into_bytes(x))]
#[br(map = Self::from_bytes)]
pub struct NotifyFlags {
pub watch_tree: bool,
#[skip]
__: B15,
}
#[bitfield]
#[derive(BinWrite, BinRead, Debug, Default, Clone, Copy, PartialEq, Eq)]
#[bw(map = |&x| Self::into_bytes(x))]
#[br(map = Self::from_bytes)]
pub struct NotifyFilter {
pub file_name: bool,
pub dir_name: bool,
pub attributes: bool,
pub size: bool,
pub last_write: bool,
pub last_access: bool,
pub creation: bool,
pub ea: bool,
pub security: bool,
pub stream_name: bool,
pub stream_size: bool,
pub stream_write: bool,
#[skip]
__: B20,
}
impl NotifyFilter {
pub fn all() -> Self {
Self::new()
.with_file_name(true)
.with_dir_name(true)
.with_attributes(true)
.with_size(true)
.with_last_write(true)
.with_last_access(true)
.with_creation(true)
.with_ea(true)
.with_security(true)
.with_stream_name(true)
.with_stream_size(true)
.with_stream_write(true)
}
}
#[binrw::binrw]
#[derive(Debug, PartialEq, Eq)]
pub struct ChangeNotifyResponse {
#[bw(calc = 9)]
#[br(assert(_structure_size == 9))]
_structure_size: u16,
#[bw(calc = PosMarker::default())]
_output_buffer_offset: PosMarker<u16>,
#[bw(calc = PosMarker::default())]
_output_buffer_length: PosMarker<u32>,
#[br(seek_before = SeekFrom::Start(_output_buffer_offset.value.into()))]
#[br(map_stream = |s| s.take_seek(_output_buffer_length.value.into()))]
pub buffer: ChainedItemList<FileNotifyInformation, 4>,
}
#[binrw::binrw]
#[derive(Debug, PartialEq, Eq)]
pub struct ServerToClientNotification {
structure_size: u16,
#[bw(calc = 0)]
_reserved: u16,
#[bw(calc = notification.get_type())]
notification_type: NotificationType,
#[br(args(notification_type))]
pub notification: Notification,
}
#[binrw::binrw]
#[derive(Debug, PartialEq, Eq)]
#[brw(repr(u32))]
pub enum NotificationType {
NotifySessionClosed = 0,
}
#[binrw::binrw]
#[derive(Debug, PartialEq, Eq)]
#[br(import(notification_type: NotificationType))]
pub enum Notification {
#[br(pre_assert(notification_type == NotificationType::NotifySessionClosed))]
NotifySessionClosed(NotifySessionClosed),
}
impl Notification {
pub fn get_type(&self) -> NotificationType {
match self {
Notification::NotifySessionClosed(_) => NotificationType::NotifySessionClosed,
}
}
}
#[binrw::binrw]
#[derive(Debug, PartialEq, Eq)]
pub struct NotifySessionClosed {
#[bw(calc = 0)]
_reserved: u32,
}
#[cfg(test)]
mod tests {
use std::io::Cursor;
use crate::*;
use smb_dtyp::guid::Guid;
use super::*;
#[test]
pub fn change_notify_request_write() {
let request = ChangeNotifyRequest {
flags: NotifyFlags::new(),
output_buffer_length: 2048,
file_id: "000005d1-000c-0000-1900-00000c000000"
.parse::<Guid>()
.unwrap()
.into(),
completion_filter: NotifyFilter::new()
.with_file_name(true)
.with_dir_name(true)
.with_attributes(true)
.with_last_write(true),
};
let mut cursor = Cursor::new(Vec::new());
request.write_le(&mut cursor).unwrap();
assert_eq!(
cursor.into_inner(),
[
0x20, 0x0, 0x0, 0x0, 0x0, 0x8, 0x0, 0x0, 0xd1, 0x5, 0x0, 0x0, 0xc, 0x0, 0x0, 0x0,
0x19, 0x0, 0x0, 0x0, 0xc, 0x0, 0x0, 0x0, 0x17, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0
]
);
}
#[test]
pub fn test_change_notify_response_pending_parse() {
let data = [0x9, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0];
let response = ChangeNotifyResponse::read_le(&mut Cursor::new(&data)).unwrap();
assert_eq!(
response,
ChangeNotifyResponse {
buffer: Default::default()
}
);
}
#[test]
pub fn test_change_notify_response_with_data_parse() {
let data = [
0xfe, 0x53, 0x4d, 0x42, 0x40, 0x0, 0x1, 0x0, 0x0, 0x0, 0x0, 0x0, 0xf, 0x0, 0x0, 0x0,
0x33, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x56, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x1,
0x0, 0x0, 0x1, 0x0, 0x0, 0x0, 0x0, 0x25, 0x0, 0x0, 0x0, 0x0, 0x38, 0x0, 0x0, 0x0, 0x0,
0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x9, 0x0, 0x48,
0x0, 0x34, 0x0, 0x0, 0x0, 0x20, 0x0, 0x0, 0x0, 0x4, 0x0, 0x0, 0x0, 0x14, 0x0, 0x0, 0x0,
0x4e, 0x0, 0x65, 0x0, 0x77, 0x0, 0x20, 0x0, 0x66, 0x0, 0x6f, 0x0, 0x6c, 0x0, 0x64, 0x0,
0x65, 0x0, 0x72, 0x0, 0x0, 0x0, 0x0, 0x0, 0x5, 0x0, 0x0, 0x0, 0x8, 0x0, 0x0, 0x0, 0x6a,
0x0, 0x64, 0x0, 0x73, 0x0, 0x61, 0x0,
];
let parsed = decode_content(&data);
let notify_response = parsed.content.to_changenotify().unwrap();
assert_eq!(
notify_response,
ChangeNotifyResponse {
buffer: vec![
FileNotifyInformation {
action: NotifyAction::RenamedOldName,
file_name: "New folder".into()
},
FileNotifyInformation {
action: NotifyAction::RenamedNewName,
file_name: "jdsa".into()
}
]
.into()
}
);
}
#[test]
pub fn test_change_notify_response_azure_files() {
let data = [
0xfe, 0x53, 0x4d, 0x42, 0x40, 0x00, 0x01, 0x00, 0x00, 0x00, 0x00, 0x00, 0x0f, 0x00,
0x00, 0x00, 0x13, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x08, 0x00, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x09, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x5d, 0x00,
0x00, 0x68, 0x6c, 0x30, 0x11, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x09, 0x00, 0x48, 0x00, 0x60, 0x01,
0x00, 0x00, 0x18, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x0c, 0x00, 0x00, 0x00,
0x31, 0x00, 0x31, 0x00, 0x2e, 0x00, 0x74, 0x00, 0x78, 0x00, 0x74, 0x00, 0x28, 0x00,
0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x1c, 0x00, 0x00, 0x00, 0x6b, 0x00, 0x65, 0x00,
0x72, 0x00, 0x6e, 0x00, 0x65, 0x00, 0x6c, 0x00, 0x2e, 0x00, 0x62, 0x00, 0x69, 0x00,
0x6e, 0x00, 0x2e, 0x00, 0x74, 0x00, 0x69, 0x00, 0x6c, 0x00, 0x78, 0x00, 0x00, 0x00,
0x01, 0x00, 0x00, 0x00, 0x6c, 0x00, 0x00, 0x00, 0x65, 0x00, 0x63, 0x00, 0x32, 0x00,
0x2d, 0x00, 0x33, 0x00, 0x2d, 0x00, 0x37, 0x00, 0x30, 0x00, 0x2d, 0x00, 0x32, 0x00,
0x32, 0x00, 0x32, 0x00, 0x2d, 0x00, 0x36, 0x00, 0x39, 0x00, 0x2e, 0x00, 0x65, 0x00,
0x75, 0x00, 0x2d, 0x00, 0x63, 0x00, 0x65, 0x00, 0x6e, 0x00, 0x74, 0x00, 0x72, 0x00,
0x61, 0x00, 0x6c, 0x00, 0x2d, 0x00, 0x31, 0x00, 0x2e, 0x00, 0x63, 0x00, 0x6f, 0x00,
0x6d, 0x00, 0x70, 0x00, 0x75, 0x00, 0x74, 0x00, 0x65, 0x00, 0x2e, 0x00, 0x61, 0x00,
0x6d, 0x00, 0x61, 0x00, 0x7a, 0x00, 0x6f, 0x00, 0x6e, 0x00, 0x61, 0x00, 0x77, 0x00,
0x73, 0x00, 0x2e, 0x00, 0x63, 0x00, 0x6f, 0x00, 0x6d, 0x00, 0x2e, 0x00, 0x72, 0x00,
0x64, 0x00, 0x70, 0x00, 0x80, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x6e, 0x00,
0x00, 0x00, 0x65, 0x00, 0x63, 0x00, 0x32, 0x00, 0x2d, 0x00, 0x31, 0x00, 0x38, 0x00,
0x2d, 0x00, 0x31, 0x00, 0x39, 0x00, 0x38, 0x00, 0x2d, 0x00, 0x35, 0x00, 0x31, 0x00,
0x2d, 0x00, 0x39, 0x00, 0x38, 0x00, 0x2e, 0x00, 0x65, 0x00, 0x75, 0x00, 0x2d, 0x00,
0x63, 0x00, 0x65, 0x00, 0x6e, 0x00, 0x74, 0x00, 0x72, 0x00, 0x61, 0x00, 0x6c, 0x00,
0x2d, 0x00, 0x31, 0x00, 0x2e, 0x00, 0x63, 0x00, 0x6f, 0x00, 0x6d, 0x00, 0x70, 0x00,
0x75, 0x00, 0x74, 0x00, 0x65, 0x00, 0x2e, 0x00, 0x61, 0x00, 0x6d, 0x00, 0x61, 0x00,
0x7a, 0x00, 0x6f, 0x00, 0x6e, 0x00, 0x61, 0x00, 0x77, 0x00, 0x73, 0x00, 0x2e, 0x00,
0x63, 0x00, 0x6f, 0x00, 0x6d, 0x00, 0x2e, 0x00, 0x72, 0x00, 0x64, 0x00, 0x70, 0x00,
0x6f, 0x55, 0x73, 0x61, 0x67, 0x65, 0x00, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00,
0x16, 0x00, 0x00, 0x00, 0x54, 0x00, 0x65, 0x00, 0x73, 0x00, 0x74, 0x00, 0x20, 0x00,
0x44, 0x00, 0x43, 0x00, 0x2e, 0x00, 0x72, 0x00, 0x64, 0x00, 0x70, 0x00, 0x72, 0x6e,
0x65, 0x74, 0x45, 0x67,
];
let parsed = decode_content(&data);
let notify_response = parsed.content.to_changenotify().unwrap();
assert_eq!(
notify_response,
ChangeNotifyResponse {
buffer: vec![
FileNotifyInformation {
action: NotifyAction::Added,
file_name: "11.txt".into()
},
FileNotifyInformation {
action: NotifyAction::Added,
file_name: "kernel.bin.til".into()
},
FileNotifyInformation {
action: NotifyAction::Added,
file_name: "ec2-3-70-222-69.eu-central-1.compute.amazonaws.com.rdp".into()
},
FileNotifyInformation {
action: NotifyAction::Added,
file_name: "ec2-18-198-51-98.eu-central-1.compute.amazonaws.com.rdp".into()
},
FileNotifyInformation {
action: NotifyAction::Added,
file_name: "Test DC.rdp".into()
}
]
.into()
}
);
}
}