mmap-loader 0.5.3

A Portable Executable manual map loader
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

use std::{env, net::SocketAddr};

use mmap_loader::server::Server;
use tonic::transport::Identity;

#[tokio::main]
async fn main() -> anyhow::Result<()> {
    let args: Vec<String> = env::args().collect();
    // they must specify both cert and key if they want TLS
    if args.len() > 6 || args.len() == 5 || args.get(1).map(|arg| arg.contains("-help")).unwrap_or(false) {
        eprintln!(
            "Usage: {} <address:ip:0.0.0.0> <port:u16:42220> <cache_path:path:cache.json> <cert_path:opt<path>> <key_path:opt<path>>",
            args[0]
        );
        return Ok(());
    }
    let addr = args
        .get(1)
        .map(|str| str.as_str())
        .unwrap_or("0.0.0.0")
        .parse()?;
    let port = args
        .get(2)
        .map(|str| str.as_str())
        .unwrap_or("443")
        .parse()?;
    let cache_path = args.get(3).map(|str| str.as_str()).unwrap_or("cache.json");
    let tls_identity = match args.len() {
        6 => {
            let cert = tokio::fs::read(&args[4]).await?;
            let key = tokio::fs::read(&args[5]).await?;
            Some(Identity::from_pem(cert, key))
        }
        _ => None,
    };
    let server = Server::new(SocketAddr::new(addr, port), cache_path, tls_identity)?;
    server.run().await?;
    Ok(())
}