Skip to main content

network_mount/
network_mount.rs

1//! Connects to a `doublecrypt-server` over TLS, authenticates with a
2//! key-derived token, mounts a network-backed encrypted filesystem with a
3//! write-back cache, and performs basic operations.
4//!
5//! Run with:
6//!   cargo run --example network_mount -- \
7//!       --addr 127.0.0.1:9100 \
8//!       --server-name localhost \
9//!       --ca certs/ca.pem \
10//!       --master-key 0000000000000000000000000000000000000000000000000000000000000000
11
12use std::path::Path;
13use std::sync::Arc;
14
15use doublecrypt_core::block_store::BlockStore;
16use doublecrypt_core::cached_store::CachedBlockStore;
17use doublecrypt_core::crypto::ChaChaEngine;
18use doublecrypt_core::fs::FilesystemCore;
19use doublecrypt_core::network_store::NetworkBlockStore;
20
21fn main() {
22    let args: Vec<String> = std::env::args().collect();
23
24    let mut addr = "127.0.0.1:9100";
25    let mut server_name = "localhost";
26    let mut ca = "certs/ca.pem";
27    let mut master_key_hex = "";
28    let mut init = false;
29
30    let mut i = 1;
31    while i < args.len() {
32        match args[i].as_str() {
33            "--addr" => {
34                addr = &args[i + 1];
35                i += 2;
36            }
37            "--server-name" => {
38                server_name = &args[i + 1];
39                i += 2;
40            }
41            "--ca" => {
42                ca = &args[i + 1];
43                i += 2;
44            }
45            "--master-key" => {
46                master_key_hex = &args[i + 1];
47                i += 2;
48            }
49            "--init" => {
50                init = true;
51                i += 1;
52            }
53            other => {
54                eprintln!("unknown argument: {other}");
55                std::process::exit(1);
56            }
57        }
58    }
59
60    if master_key_hex.is_empty() || master_key_hex.len() != 64 {
61        eprintln!("--master-key must be a 64-character hex string (32 bytes)");
62        std::process::exit(1);
63    }
64
65    let master_key: Vec<u8> = (0..32)
66        .map(|i| u8::from_str_radix(&master_key_hex[i * 2..i * 2 + 2], 16).unwrap())
67        .collect();
68
69    // ── Connect ─────────────────────────────────────────────
70    println!("Connecting to {addr} (SNI: {server_name})...");
71    let net = NetworkBlockStore::connect(addr, server_name, Path::new(ca), &master_key)
72        .expect("failed to connect to server");
73
74    println!(
75        "Connected: {} blocks × {} bytes ({} MiB)",
76        net.total_blocks(),
77        net.block_size(),
78        net.total_blocks() as usize * net.block_size() / (1024 * 1024)
79    );
80
81    // ── Wrap with cache ─────────────────────────────────────
82    let store = Arc::new(CachedBlockStore::new(net, 1024));
83    let crypto = Arc::new(ChaChaEngine::new(&master_key).expect("invalid master key"));
84    let mut fs = FilesystemCore::new(store.clone(), crypto);
85
86    // ── Mount or init ───────────────────────────────────────
87    if init {
88        println!("Initializing new filesystem...");
89        fs.init_filesystem().expect("init_filesystem failed");
90    } else {
91        println!("Mounting existing filesystem...");
92        fs.open().expect("open failed");
93    }
94
95    // ── Demo operations ─────────────────────────────────────
96    println!("\nCreating file 'hello.txt'...");
97    match fs.create_file("hello.txt") {
98        Ok(()) => {}
99        Err(e) => println!("  (skipped: {e})"),
100    }
101
102    fs.write_file("hello.txt", 0, b"Hello from the network!")
103        .expect("write failed");
104
105    let data = fs.read_file("hello.txt", 0, 4096).expect("read failed");
106    println!("Read back: {:?}", String::from_utf8_lossy(&data));
107
108    println!("\nListing root directory:");
109    for entry in fs.list_directory().expect("list failed") {
110        println!(
111            "  {:?}  {:>10} bytes  {}",
112            entry.kind, entry.size, entry.name
113        );
114    }
115
116    // ── Sync ────────────────────────────────────────────────
117    fs.sync().expect("sync failed");
118    println!("\nAll data synced to server.");
119}