From 5335b4c76b099c37760475c1a1e5093ec920323d Mon Sep 17 00:00:00 2001
From: Phuntsok Drak-pa <drakpa@drakpa.fr>
Date: Tue, 20 Feb 2018 15:09:03 +0100
Subject: [PATCH] First version with separated clients and a server

---
 .gitignore  |   4 ++
 Cargo.toml  |   6 ++
 src/main.rs | 177 ++++++++++++++++++++++++++++++++++++++++++++++++++++
 3 files changed, 187 insertions(+)
 create mode 100644 .gitignore
 create mode 100644 Cargo.toml
 create mode 100644 src/main.rs

diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..e6a502f
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1,4 @@
+
+/target
+**/*.rs.bk
+*.lock
diff --git a/Cargo.toml b/Cargo.toml
new file mode 100644
index 0000000..e4201bc
--- /dev/null
+++ b/Cargo.toml
@@ -0,0 +1,6 @@
+[package]
+name = "chat-reseau"
+version = "0.1.0"
+authors = ["Phuntsok Drak-pa <drakpa@drakpa.fr>"]
+
+[dependencies]
diff --git a/src/main.rs b/src/main.rs
new file mode 100644
index 0000000..8cf7aea
--- /dev/null
+++ b/src/main.rs
@@ -0,0 +1,177 @@
+use std::env;
+use std::net::{TcpListener, TcpStream};
+use std::thread;
+#[allow(unused_imports)]
+use std::io::{stdin, stdout, Read, Write};
+#[allow(unused_imports)]
+use std::sync::{Arc, Mutex};
+
+///////////////////////////////////////////////////////////////////////////////
+//                                                                           //
+//                                   Client                                  //
+//                                                                           //
+///////////////////////////////////////////////////////////////////////////////
+
+fn get_entry() -> String {
+    let mut buf = String::new();
+
+    stdin().read_line(&mut buf).unwrap();
+    buf.replace("\n", "").replace("\r", "")
+}
+
+fn exchange_with_server(mut stream: TcpStream) {
+    let stdout = std::io::stdout();
+    let mut io = stdout.lock();
+    let buf = &mut [0; 3];
+
+    println!("Enter `quit` when you want to leave");
+    loop {
+        write!(io, "> ").unwrap();
+        io.flush().unwrap();
+        match &*get_entry() {
+            "quit" => {
+                println!("bye!");
+                return;
+            }
+            "exit" => {
+                println!("bye!");
+                return;
+            }
+            line => {
+                write!(stream, "{}\n", line).unwrap();
+                match stream.read(buf) {
+                    Ok(received) => {
+                        if received < 1 {
+                            println!("Perte de la connexion avec le serveur");
+                            return;
+                        }
+                    }
+                    Err(_) => {
+                        println!("Perte de la connexion avec le serveur");
+                        return;
+                    }
+                }
+                // println!("Réponse du serveur : {:?}", buf);
+                let reponse = String::from_utf8(buf.to_vec()).unwrap();
+                println!("Réponse du serveur : {}", reponse);
+            }
+        }
+    }
+}
+
+fn client(server_address: String) {
+    println!("Tentative de connexion a serveur...");
+    match TcpStream::connect(server_address) {
+        Ok(stream) => {
+            println!("Connexion au serveur réussie !");
+            exchange_with_server(stream);
+        }
+        Err(e) => {
+            println!("La connection au serveur a échoué : {}", e);
+        }
+    }
+}
+
+///////////////////////////////////////////////////////////////////////////////
+//                                                                           //
+//                                   Server                                  //
+//                                                                           //
+///////////////////////////////////////////////////////////////////////////////
+
+fn handle_client(mut stream: &TcpStream, adresse: &str) {
+    let mut msg: Vec<u8> = Vec::new();
+    loop {
+        let buf = &mut [0; 10];
+
+        match stream.read(buf) {
+            Ok(received) => {
+                // si on a reçu 0 octet, ça veut dire que le client s'est déconnecté
+                if received < 1 {
+                    println!("Client disconnected {}", adresse);
+                    return;
+                }
+                let mut x = 0;
+
+                for c in buf {
+                    // si on a dépassé le nombre d'octets reçus, inutile de continuer
+                    if x >= received {
+                        break;
+                    }
+                    x += 1;
+                    if *c == '\n' as u8 {
+                        println!(
+                            "message reçu {} : {}",
+                            adresse,
+                            // on convertit maintenant notre buffer en String
+                            String::from_utf8(msg).unwrap()
+                        );
+
+                        stream.write(b"ok\n").unwrap();
+
+                        msg = Vec::new();
+                    } else {
+                        msg.push(*c);
+                    }
+                }
+            }
+            Err(_) => {
+                println!("Client disconnected {}", adresse);
+                return;
+            }
+        }
+    }
+}
+
+fn serveur(port: String) {
+    println!("Port: {}", port);
+    let mut serv = String::from("127.0.0.1:");
+    serv.push_str(&port);
+    let listener = TcpListener::bind(serv.to_string()).unwrap();
+
+    println!("En attente d’un client...");
+
+    // Multi-client ///////////////////////////////////////////////////////////
+    for stream in listener.incoming() {
+        match stream {
+            Ok(stream) => {
+                let adresse = match stream.peer_addr() {
+                    Ok(addr) => format!("[adresse : {}]", addr),
+                    Err(_) => "inconnue".to_owned(),
+                };
+
+                println!("Nouveau client {}", adresse);
+                thread::spawn(move || handle_client(&stream, &*adresse));
+            }
+            Err(e) => {
+                println!("La connexion du client a échoué : {}", e);
+            }
+        }
+        println!("En attente d’un autre client...");
+    }
+}
+
+fn main() {
+    let args: Vec<String> = env::args().collect();
+    if args.len() == 2 {
+        ///////////////////////////////////////////////////////////////////////
+        //                           Server opened                           //
+        ///////////////////////////////////////////////////////////////////////
+        println!("Opening server on port {}", args[1]);
+        serveur(args[1].clone());
+    } else if args.len() == 3 {
+        ///////////////////////////////////////////////////////////////////////
+        //                           Client opened                           //
+        ///////////////////////////////////////////////////////////////////////
+        println!("Client connecting on server {}:{}", args[1], args[2]);
+        let mut serv = if args[1] == String::from("localhost") {
+            String::from("127.0.0.1")
+        } else {
+            args[1].clone()
+        };
+        serv.push(':');
+        serv.push_str(&args[2]);
+        client(serv);
+    } else {
+        println!("Usage: {} [server ip] port", args[0]);
+    }
+}