pavao 0.2.16

Rust client library for SMB 2/3 based on libsmbclient.
Documentation 
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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82

use std::path::PathBuf;

use argh::FromArgs;
use pavao::{SmbClient, SmbCredentials, SmbDirentInfo, SmbDirentType, SmbOptions};

#[derive(FromArgs)]
#[argh(description = "
where positional can be: [smb://address[:port]]

Please, report issues to <https://github.com/veeso/pavao>
Please, consider supporting the author <https://ko-fi.com/veeso>")]
struct Args {
    #[argh(option, short = 'P', description = "specify password")]
    password: Option<String>,
    #[argh(option, short = 'u', description = "specify username")]
    username: String,
    #[argh(option, short = 'w', description = "specify workgroup")]
    workgroup: String,
    #[argh(option, short = 's', description = "specify share")]
    share: String,
    #[argh(positional, description = "smb://address[:port]")]
    server: String,
}

fn main() {
    assert!(env_logger::builder().try_init().is_ok());
    let args: Args = argh::from_env();
    let password = match args.password {
        Some(p) => p,
        None => read_secret_from_tty("Password: ").ok().unwrap(),
    };
    //setup server
    let client = SmbClient::new(
        SmbCredentials::default()
            .server(args.server)
            .share(args.share)
            .password(password)
            .username(args.username)
            .workgroup(args.workgroup),
        SmbOptions::default().one_share_per_server(true),
    )
    .unwrap();
    treeplus(&client, "/DCIM", 0);
}

/// Read a secret from tty with customisable prompt
fn read_secret_from_tty(prompt: &str) -> std::io::Result<String> {
    rpassword::prompt_password(prompt)
}

fn treeplus(client: &SmbClient, uri: &str, depth: usize) {
    let vec = client.list_dirplus(uri).unwrap();
    for entityplus in vec.into_iter() {
        let entityplus_uri = entityplus_uri(&entityplus, uri);
        print_entry_plus(&entityplus, depth);
        // if is dir, iter directory
        if entityplus.get_type() == SmbDirentType::Dir {
            treeplus(client, entityplus_uri.as_str(), depth + 1)
        }
    }
}

fn entityplus_uri(entity: &SmbDirentInfo, path: &str) -> String {
    let mut p = PathBuf::from(path);
    p.push(PathBuf::from(entity.name()));
    p.as_path().to_string_lossy().to_string()
}

fn print_entry_plus(entityplus: &SmbDirentInfo, depth: usize) {
    println!(
        "{}{:32}\t{}\t{:x}\t{:?}",
        fmt_depth(depth),
        entityplus.name(),
        entityplus.size,
        entityplus.attrs,
        entityplus.get_type(),
    )
}

fn fmt_depth(depth: usize) -> String {
    " ".repeat(depth * 4)
}