smb 0.1.0

A Pure Rust SMB Client implementation
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49

use crate::packets::smb2::*;

use super::ResourceHandle;

pub struct Directory {
    pub handle: ResourceHandle,
    access: DirAccessMask,
}

impl Directory {
    pub fn new(handle: ResourceHandle, access: DirAccessMask) -> Self {
        Directory { handle, access }
    }

    // Query the directory for it's contents.
    pub fn query(
        &mut self,
        pattern: &str,
    ) -> Result<Vec<BothDirectoryInformationItem>, Box<dyn std::error::Error>> {
        if !self.access.file_list_directory() {
            return Err("No directory list permission".into());
        }

        log::debug!("Querying directory {}", self.handle.name());

        let response =
            self.handle
                .send_receive(Content::QueryDirectoryRequest(QueryDirectoryRequest {
                    file_information_class: QueryFileInfoClass::IdBothDirectoryInformation,
                    flags: QueryDirectoryFlags::new().with_restart_scans(true),
                    file_index: 0,
                    file_id: self.handle.file_id(),
                    output_buffer_length: 0x10000,
                    file_name: pattern.into(),
                }))?;

        let content = match response.message.content {
            Content::QueryDirectoryResponse(response) => response,
            _ => panic!("Unexpected response"),
        };
        let result = match QueryDirectoryInfoVector::parse(
            &content.output_buffer,
            QueryFileInfoClass::IdBothDirectoryInformation,
        )? {
            QueryDirectoryInfoVector::IdBothDirectoryInformation(val) => val,
        };
        Ok(result.into())
    }
}