forked from lx/netapp
99 lines
2.5 KiB
Rust
99 lines
2.5 KiB
Rust
use std::io::Write;
|
|
use std::net::SocketAddr;
|
|
|
|
use log::info;
|
|
|
|
use structopt::StructOpt;
|
|
|
|
use sodiumoxide::crypto::auth;
|
|
use sodiumoxide::crypto::sign::ed25519;
|
|
|
|
use netapp::peering::fullmesh::*;
|
|
use netapp::util::*;
|
|
|
|
use netapp::NetApp;
|
|
|
|
#[derive(StructOpt, Debug)]
|
|
#[structopt(name = "netapp")]
|
|
pub struct Opt {
|
|
#[structopt(long = "network-key", short = "n")]
|
|
network_key: Option<String>,
|
|
|
|
#[structopt(long = "private-key", short = "p")]
|
|
private_key: Option<String>,
|
|
|
|
#[structopt(long = "bootstrap-peer", short = "b")]
|
|
bootstrap_peers: Vec<String>,
|
|
|
|
#[structopt(long = "listen-addr", short = "l", default_value = "127.0.0.1:1980")]
|
|
listen_addr: String,
|
|
|
|
#[structopt(long = "public-addr", short = "a")]
|
|
public_addr: Option<String>,
|
|
}
|
|
|
|
#[tokio::main]
|
|
async fn main() {
|
|
env_logger::Builder::new()
|
|
.parse_env("RUST_LOG")
|
|
.format(|buf, record| {
|
|
writeln!(
|
|
buf,
|
|
"{} {} {} {}",
|
|
chrono::Local::now().format("%s%.6f"),
|
|
record.module_path().unwrap_or("_"),
|
|
record.level(),
|
|
record.args()
|
|
)
|
|
})
|
|
.init();
|
|
|
|
let opt = Opt::from_args();
|
|
|
|
let netid = match &opt.network_key {
|
|
Some(k) => auth::Key::from_slice(&hex::decode(k).unwrap()).unwrap(),
|
|
None => auth::gen_key(),
|
|
};
|
|
info!("Network key: {}", hex::encode(&netid));
|
|
|
|
let privkey = match &opt.private_key {
|
|
Some(k) => ed25519::SecretKey::from_slice(&hex::decode(k).unwrap()).unwrap(),
|
|
None => {
|
|
let (_pk, sk) = ed25519::gen_keypair();
|
|
sk
|
|
}
|
|
};
|
|
|
|
info!("Node private key: {}", hex::encode(&privkey));
|
|
info!("Node public key: {}", hex::encode(&privkey.public_key()));
|
|
|
|
let public_addr = opt.public_addr.map(|x| x.parse().unwrap());
|
|
let listen_addr: SocketAddr = opt.listen_addr.parse().unwrap();
|
|
info!("Node public address: {:?}", public_addr);
|
|
info!("Node listen address: {}", listen_addr);
|
|
|
|
let netapp = NetApp::new(0u64, netid.clone(), privkey.clone());
|
|
|
|
let mut bootstrap_peers = vec![];
|
|
for peer in opt.bootstrap_peers.iter() {
|
|
bootstrap_peers.push(parse_peer_addr(peer).expect("Invalid peer address"));
|
|
}
|
|
|
|
let peering = FullMeshPeeringStrategy::new(
|
|
netapp.clone(),
|
|
bootstrap_peers,
|
|
public_addr.map(|a| SocketAddr::new(a, listen_addr.port())),
|
|
);
|
|
|
|
info!("Add more peers to this mesh by running: fullmesh -n {} -l 127.0.0.1:$((1000 + $RANDOM)) -b {}@{}",
|
|
hex::encode(&netid),
|
|
hex::encode(&privkey.public_key()),
|
|
listen_addr);
|
|
|
|
let watch_cancel = netapp::util::watch_ctrl_c();
|
|
|
|
tokio::join!(
|
|
netapp.listen(listen_addr, public_addr, watch_cancel.clone()),
|
|
peering.run(watch_cancel),
|
|
);
|
|
}
|