use end_point::{ Client, Server };
+mod utils;
mod crypto;
mod packet;
mod end_point;
let args = os::args();
if args.iter().any(|a| a.as_slice() == "--help" || a.as_slice() == "-h") {
- return Help
+ return Mode::Help
}
if args.len() <= 1 {
- ServerAlone
+ Mode::ServerAlone
} else {
match args[1].as_slice() {
- "genkey" => GenKey,
- "tests" => Tests,
- "oracle-weak" => OracleWeak,
- "oracle-fixed" => OracleFixed,
- _ => ServerAlone,
+ "genkey" => Mode::GenKey,
+ "tests" => Mode::Tests,
+ "oracle-weak" => Mode::OracleWeak,
+ "oracle-fixed" => Mode::OracleFixed,
+ _ => Mode::ServerAlone,
}
}
}
let mode = mode();
match mode {
- Help => print_usage(),
- GenKey =>
+ Mode::Help => print_usage(),
+ Mode::GenKey =>
match crypto::generate_key(256 / 8) {
Ok(key) => println!("key: {}", key),
Err(e) => println!("Unable to generate a key. Error: {}", e)
let address = "::1";
println!("Starting server on [{}]:{}...", address, PORT);
- match Server::new(address, PORT, match mode { OracleFixed => packet::Fixed, _ => packet::Weak }) {
+ match Server::new(address, PORT, match mode { Mode::OracleFixed => packet::Variant::Fixed, _ => packet::Variant::Weak }) {
Ok(mut server) => {
println!("Server started");
match mode {
- Tests => Client::start_tests(address, PORT, packet::Weak),
- OracleWeak => do_oracle_attack(address, packet::Weak),
- OracleFixed => do_oracle_attack(address, packet::Fixed),
+ Mode::Tests => Client::start_tests(address, PORT, packet::Variant::Weak),
+ Mode::OracleWeak => do_oracle_attack(address, packet::Variant::Weak),
+ Mode::OracleFixed => do_oracle_attack(address, packet::Variant::Fixed),
_ => {
println!("Press any key to quit");
io::stdin().read_line().ok().expect("Failed to read line");