vpncloud/src/main.rs

99 lines
2.8 KiB
Rust
Raw Normal View History

2015-11-19 15:34:20 +00:00
#[macro_use] extern crate log;
extern crate time;
extern crate docopt;
extern crate rustc_serialize;
2015-11-20 08:11:54 +00:00
extern crate epoll;
2015-11-19 15:34:20 +00:00
2015-11-19 18:29:42 +00:00
mod util;
2015-11-19 15:34:20 +00:00
mod udpmessage;
mod ethernet;
2015-11-21 15:50:50 +00:00
mod ip;
2015-11-19 15:34:20 +00:00
mod ethcloud;
use time::Duration;
use docopt::Docopt;
2015-11-20 17:40:23 +00:00
use std::hash::{Hash, SipHasher, Hasher};
2015-11-22 15:48:01 +00:00
use ethcloud::{Error, TapCloud};
2015-11-19 15:34:20 +00:00
2015-11-19 18:29:42 +00:00
//TODO: Implement IPv6
2015-11-19 19:51:53 +00:00
//TODO: Encryption
//TODO: Call close
2015-11-19 18:29:42 +00:00
2015-11-19 15:34:20 +00:00
struct SimpleLogger;
impl log::Log for SimpleLogger {
fn enabled(&self, _metadata: &log::LogMetadata) -> bool {
true
}
fn log(&self, record: &log::LogRecord) {
if self.enabled(record.metadata()) {
println!("{} - {}", record.level(), record.args());
}
}
}
static USAGE: &'static str = "
Usage:
2015-11-20 17:09:51 +00:00
ethcloud [options] [-d <device>] [-l <listen>] [-c <connect>...]
2015-11-19 15:34:20 +00:00
Options:
2015-11-19 16:23:36 +00:00
-d <device>, --device <device> Name of the tap device [default: ethcloud%d]
-l <listen>, --listen <listen> Address to listen on [default: 0.0.0.0:3210]
-c <connect>, --connect <connect> List of peers (addr:port) to connect to
2015-11-20 17:09:51 +00:00
--network-id <network_id> Optional token that identifies the network
2015-11-19 21:45:20 +00:00
--peer-timeout <peer_timeout> Peer timeout in seconds [default: 1800]
--mac-timeout <mac_timeout> Mac table entry timeout in seconds [default: 300]
2015-11-19 16:23:36 +00:00
-v, --verbose Log verbosely
-q, --quiet Only print error messages
2015-11-20 13:05:35 +00:00
-h, --help Display the help
2015-11-19 15:34:20 +00:00
";
#[derive(RustcDecodable, Debug)]
struct Args {
2015-11-19 16:23:36 +00:00
flag_device: String,
flag_listen: String,
2015-11-20 17:40:23 +00:00
flag_network_id: Option<String>,
2015-11-19 16:23:36 +00:00
flag_connect: Vec<String>,
flag_peer_timeout: usize,
flag_mac_timeout: usize,
flag_verbose: bool,
flag_quiet: bool
2015-11-19 15:34:20 +00:00
}
fn main() {
let args: Args = Docopt::new(USAGE).and_then(|d| d.decode()).unwrap_or_else(|e| e.exit());
log::set_logger(|max_log_level| {
2015-11-19 16:23:36 +00:00
assert!(!args.flag_verbose || !args.flag_quiet);
2015-11-19 15:34:20 +00:00
if args.flag_verbose {
max_log_level.set(log::LogLevelFilter::Debug);
2015-11-19 16:23:36 +00:00
} else if args.flag_quiet {
max_log_level.set(log::LogLevelFilter::Error);
2015-11-19 15:34:20 +00:00
} else {
max_log_level.set(log::LogLevelFilter::Info);
}
Box::new(SimpleLogger)
}).unwrap();
debug!("Args: {:?}", args);
2015-11-22 15:48:01 +00:00
let mut tapcloud = TapCloud::new_tap_cloud(
2015-11-19 15:34:20 +00:00
&args.flag_device,
args.flag_listen,
2015-11-20 17:40:23 +00:00
args.flag_network_id.map(|name| {
let mut s = SipHasher::new();
name.hash(&mut s);
s.finish()
}),
2015-11-19 15:34:20 +00:00
Duration::seconds(args.flag_mac_timeout as i64),
Duration::seconds(args.flag_peer_timeout as i64)
);
for addr in args.flag_connect {
2015-11-20 11:09:07 +00:00
tapcloud.connect(&addr as &str, true).expect("Failed to send");
2015-11-19 15:34:20 +00:00
}
2015-11-19 19:51:53 +00:00
tapcloud.run()
2015-11-19 15:34:20 +00:00
}