X-Git-Url: http://git.euphorik.ch/?p=crypto_lab1.git;a=blobdiff_plain;f=src%2Fmain.rs;h=5f2fdf93df305e1e64b90692ee699c8ce30cf831;hp=cec395706daab70b756a49362d6e5b4f1fc81d3b;hb=8226b0eced022af21512d87902bf6e252afe1a67;hpb=3a33f82b92400ffabdc5fd7bdcbdc9f888277418

diff --git a/src/main.rs b/src/main.rs
index cec3957..5f2fdf9 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,14 +1,16 @@
 #![feature(macro_rules)]
 
 extern crate openssl;
+extern crate serialize;
 
 use std::io;
 use std::os;
 
+use end_point::{ Client, Server };
+
 mod crypto;
-mod command;
-mod client;
-mod server;
+mod packet;
+mod end_point;
 
 /*
 TODO
@@ -19,7 +21,7 @@ TODO
 const PORT: u16 = 4221;
 
 fn print_usage() {
-   println!("{} <genkey> | ...", os::args()[0]);
+   println!("{} <genkey> | <tests> | ...", os::args()[0]);
 }
 
 fn main() {   
@@ -33,26 +35,17 @@ fn main() {
          Err(e) => println!("Unable to generate a key. Error: {}", e)
       }
    } else {
-      match server::Server::new(PORT) {
+      println!("Starting server..., Press any key to quit");
+      
+      match Server::new("::1", PORT) {
          Ok(mut server) => {
             println!("Server started");
-            
-            match client::Client::new("127.0.0.1", PORT) {
-               Ok(mut client) => {
-                  client.start_tests();
-                  client.close();
-               },
-               Err(e) => { 
-                  println!("Unable to create a client. Error: {}", e);
-                  return
-               }
-            }         
-            
-            println!("Press any key to quit");
+            Client::start_tests("::1", PORT);            
             io::stdin().read_line().ok().expect("Failed to read line");         
             server.close().ok().expect("Failed to close the server");
          },
-         Err(e) => println!("Unable to create a new server. Error: {}", e)
+         Err(e) =>
+            println!("Unable to create a new server. Error: {}", e)
       }
    }
 }