From 5b4fd102145118650d963f4ac9ca1123258411d9 Mon Sep 17 00:00:00 2001 From: RafayAhmad7548 Date: Wed, 9 Apr 2025 22:11:17 +0500 Subject: [PATCH] non async code i found out that async code would be better, so saved this code i might see this in future maybe :) --- Cargo.lock | 255 +++++++++++++++++++++++++++++++++++++++++++++++++++- Cargo.toml | 1 + src/main.rs | 113 +++++++++++++++++------ 3 files changed, 341 insertions(+), 28 deletions(-) diff --git a/Cargo.lock b/Cargo.lock index 12050a7..366b242 100644 --- a/Cargo.lock +++ b/Cargo.lock @@ -2,6 +2,42 @@ # It is not intended for manual editing. version = 4 +[[package]] +name = "addr2line" +version = "0.24.2" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "dfbe277e56a376000877090da837660b4427aad530e3028d44e0bffe4f89a1c1" +dependencies = [ + "gimli", +] + +[[package]] +name = "adler2" +version = "2.0.0" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "512761e0bb2578dd7380c6baaa0f4ce03e84f95e960231d1dec8bf4d7d6e2627" + +[[package]] +name = "autocfg" +version = "1.4.0" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "ace50bade8e6234aa140d9a2f552bbee1db4d353f69b8217bc503490fc1a9f26" + +[[package]] +name = "backtrace" +version = "0.3.74" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "8d82cb332cdfaed17ae235a638438ac4d4839913cc2af585c3c6746e8f8bee1a" +dependencies = [ + "addr2line", + "cfg-if", + "libc", + "miniz_oxide", + "object", + "rustc-demangle", + "windows-targets", +] + [[package]] name = "bitflags" version = "2.9.0" @@ -33,6 +69,7 @@ checksum = "baf1de4339761588bc0619e3cbc0120ee582ebb74b53b4efbf79117bd2da40fd" name = "clipsync" version = "0.1.0" dependencies = [ + "tokio-tungstenite", "tungstenite", ] @@ -77,6 +114,38 @@ version = "1.0.7" source = "registry+https://github.com/rust-lang/crates.io-index" checksum = "3f9eec918d3f24069decb9af1554cad7c880e2da24a9afd88aca000531ab82c1" +[[package]] +name = "futures-core" +version = "0.3.31" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "05f29059c0c2090612e8d742178b0580d2dc940c837851ad723096f87af6663e" + +[[package]] +name = "futures-sink" +version = "0.3.31" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "e575fab7d1e0dcb8d0c7bcf9a63ee213816ab51902e6d244a95819acacf1d4f7" + +[[package]] +name = "futures-task" +version = "0.3.31" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "f90f7dce0722e95104fcb095585910c0977252f286e354b5e3bd38902cd99988" + +[[package]] +name = "futures-util" +version = "0.3.31" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "9fa08315bb612088cc391249efdc3bc77536f16c91f6cf495e6fbe85b20a4a81" +dependencies = [ + "futures-core", + "futures-sink", + "futures-task", + "pin-project-lite", + "pin-utils", + "slab", +] + [[package]] name = "generic-array" version = "0.14.7" @@ -96,9 +165,15 @@ dependencies = [ "cfg-if", "libc", "r-efi", - "wasi", + "wasi 0.14.2+wasi-0.2.4", ] +[[package]] +name = "gimli" +version = "0.31.1" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "07e28edb80900c19c28f1072f2e8aeca7fa06b23cd4169cefe1af5aa3260783f" + [[package]] name = "http" version = "1.3.1" @@ -134,6 +209,53 @@ version = "0.4.27" source = "registry+https://github.com/rust-lang/crates.io-index" checksum = "13dc2df351e3202783a1fe0d44375f7295ffb4049267b0f3018346dc122a1d94" +[[package]] +name = "memchr" +version = "2.7.4" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "78ca9ab1a0babb1e7d5695e3530886289c18cf2f87ec19a575a0abdce112e3a3" + +[[package]] +name = "miniz_oxide" +version = "0.8.8" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "3be647b768db090acb35d5ec5db2b0e1f1de11133ca123b9eacf5137868f892a" +dependencies = [ + "adler2", +] + +[[package]] +name = "mio" +version = "1.0.3" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "2886843bf800fba2e3377cff24abf6379b4c4d5c6681eaf9ea5b0d15090450bd" +dependencies = [ + "libc", + "wasi 0.11.0+wasi-snapshot-preview1", + "windows-sys", +] + +[[package]] +name = "object" +version = "0.36.7" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "62948e14d923ea95ea2c7c86c71013138b66525b86bdc08d2dcc262bdb497b87" +dependencies = [ + "memchr", +] + +[[package]] +name = "pin-project-lite" +version = "0.2.16" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "3b3cff922bd51709b605d9ead9aa71031d81447142d828eb4a6eba76fe619f9b" + +[[package]] +name = "pin-utils" +version = "0.1.0" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "8b870d8c151b6f2fb93e84a13146138f05d02ed11c7e7c54f8826aaaf7c9f184" + [[package]] name = "ppv-lite86" version = "0.2.21" @@ -197,6 +319,12 @@ dependencies = [ "getrandom", ] +[[package]] +name = "rustc-demangle" +version = "0.1.24" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "719b953e2095829ee67db738b3bfa9fa368c94900df327b3f07fe6e794d2fe1f" + [[package]] name = "sha1" version = "0.10.6" @@ -208,6 +336,25 @@ dependencies = [ "digest", ] +[[package]] +name = "slab" +version = "0.4.9" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "8f92a496fb766b417c996b9c5e57daf2f7ad3b0bebe1ccfca4856390e3d3bb67" +dependencies = [ + "autocfg", +] + +[[package]] +name = "socket2" +version = "0.5.9" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "4f5fd57c80058a56cf5c777ab8a126398ece8e442983605d280a44ce79d0edef" +dependencies = [ + "libc", + "windows-sys", +] + [[package]] name = "syn" version = "2.0.100" @@ -239,6 +386,33 @@ dependencies = [ "syn", ] +[[package]] +name = "tokio" +version = "1.44.2" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "e6b88822cbe49de4185e3a4cbf8321dd487cf5fe0c5c65695fef6346371e9c48" +dependencies = [ + "backtrace", + "bytes", + "libc", + "mio", + "pin-project-lite", + "socket2", + "windows-sys", +] + +[[package]] +name = "tokio-tungstenite" +version = "0.26.2" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "7a9daff607c6d2bf6c16fd681ccb7eecc83e4e2cdc1ca067ffaadfca5de7f084" +dependencies = [ + "futures-util", + "log", + "tokio", + "tungstenite", +] + [[package]] name = "tungstenite" version = "0.26.2" @@ -280,6 +454,12 @@ version = "0.9.5" source = "registry+https://github.com/rust-lang/crates.io-index" checksum = "0b928f33d975fc6ad9f86c8f283853ad26bdd5b10b7f1542aa2fa15e2289105a" +[[package]] +name = "wasi" +version = "0.11.0+wasi-snapshot-preview1" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "9c8d87e72b64a3b4db28d11ce29237c246188f4f51057d65a7eab63b7987e423" + [[package]] name = "wasi" version = "0.14.2+wasi-0.2.4" @@ -289,6 +469,79 @@ dependencies = [ "wit-bindgen-rt", ] +[[package]] +name = "windows-sys" +version = "0.52.0" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "282be5f36a8ce781fad8c8ae18fa3f9beff57ec1b52cb3de0789201425d9a33d" +dependencies = [ + "windows-targets", +] + +[[package]] +name = "windows-targets" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "9b724f72796e036ab90c1021d4780d4d3d648aca59e491e6b98e725b84e99973" +dependencies = [ + "windows_aarch64_gnullvm", + "windows_aarch64_msvc", + "windows_i686_gnu", + "windows_i686_gnullvm", + "windows_i686_msvc", + "windows_x86_64_gnu", + "windows_x86_64_gnullvm", + "windows_x86_64_msvc", +] + +[[package]] +name = "windows_aarch64_gnullvm" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "32a4622180e7a0ec044bb555404c800bc9fd9ec262ec147edd5989ccd0c02cd3" + +[[package]] +name = "windows_aarch64_msvc" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "09ec2a7bb152e2252b53fa7803150007879548bc709c039df7627cabbd05d469" + +[[package]] +name = "windows_i686_gnu" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "8e9b5ad5ab802e97eb8e295ac6720e509ee4c243f69d781394014ebfe8bbfa0b" + +[[package]] +name = "windows_i686_gnullvm" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "0eee52d38c090b3caa76c563b86c3a4bd71ef1a819287c19d586d7334ae8ed66" + +[[package]] +name = "windows_i686_msvc" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "240948bc05c5e7c6dabba28bf89d89ffce3e303022809e73deaefe4f6ec56c66" + +[[package]] +name = "windows_x86_64_gnu" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "147a5c80aabfbf0c7d901cb5895d1de30ef2907eb21fbbab29ca94c5b08b1a78" + +[[package]] +name = "windows_x86_64_gnullvm" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "24d5b23dc417412679681396f2b49f3de8c1473deb516bd34410872eff51ed0d" + +[[package]] +name = "windows_x86_64_msvc" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "589f6da84c646204747d1270a2a5661ea66ed1cced2631d546fdfb155959f9ec" + [[package]] name = "wit-bindgen-rt" version = "0.39.0" diff --git a/Cargo.toml b/Cargo.toml index dc2f714..20bb145 100644 --- a/Cargo.toml +++ b/Cargo.toml @@ -4,4 +4,5 @@ version = "0.1.0" edition = "2021" [dependencies] +tokio-tungstenite = "0.26.2" tungstenite = "0.26.2" diff --git a/src/main.rs b/src/main.rs index 4c783f8..2e0d809 100644 --- a/src/main.rs +++ b/src/main.rs @@ -1,46 +1,105 @@ -use std::{net::TcpListener, thread::spawn}; - -use tungstenite::{accept, protocol::{frame::coding::CloseCode, CloseFrame}, Message}; +use std::{collections::HashMap, net::TcpListener, sync::{mpsc::{channel, Receiver, Sender}, Arc, Mutex}, thread::spawn}; +use tungstenite::{accept, Message}; fn main() { + let mut connection_count: usize = 0; + let senders = Arc::new(Mutex::new(HashMap::>>::new())); + let server = TcpListener::bind("0.0.0.0:9001").unwrap(); for stream in server.incoming() { + + let t_senders = Arc::clone(&senders); + let (sender, receiver): (Sender>, Receiver>) = channel(); + { + let mut sender_lock = t_senders.lock().unwrap(); + connection_count += 1; + (*sender_lock).insert(connection_count, sender); + } + let websocket = Arc::new(Mutex::new(accept(stream.unwrap()).unwrap())); + + let r_websocket = Arc::clone(&websocket); spawn(move || { - let mut websocket = accept(stream.unwrap()).unwrap(); - println!("Connection Received"); + let id = connection_count; + println!("Connection Received: {}", id); loop { - let res = websocket.read(); - + let res = { + let mut wssocket_lock = r_websocket.lock().unwrap(); + (*wssocket_lock).read() + }; match res { - Ok(Message::Text(txtmsg)) => { - println!("received: {}", txtmsg); - let strmsg = format!("i shall send this back to you {}", txtmsg.to_string()); - websocket.send(Message::text(strmsg)).unwrap(); - - } - Ok(Message::Close(_)) => { - match websocket.send(Message::Close(Some(CloseFrame{ - code: CloseCode::Normal, - reason: "Connection Closed :)".into() - }))) { - Ok(_) => println!("Close frame sent succesfully"), - Err(e) => println!("Failed to send close frame (connection may already be closed): {}", e) - }; - println!("closing connection"); - break; + // Ok(Message::Close(_)) => { + // let mut t_sender_lock = t_senders.lock().unwrap(); + // (*t_sender_lock).remove(&id); + // match (*wssocket_lock).send(Message::Close(Some(CloseFrame{ + // code: CloseCode::Normal, + // reason: "Connection Closed :)".into() + // }))) { + // Ok(_) => println!("Close frame sent succesfully"), + // Err(e) => println!("Failed to send close frame (connection may already be closed): {}", e) + // }; + // println!("closing connection"); + // break; + // } + Ok(msg) => { + let arc_msg = Arc::new(msg); + if arc_msg.is_close() { + let t_sender_lock = t_senders.lock().unwrap(); + let arc_msg_clone = Arc::clone(&arc_msg); + match (*t_sender_lock).get(&id).unwrap().send(arc_msg_clone) { + Ok(_) => println!("close message sent to thread no {}", id), + Err(e) => println!("My Error: {}", e) + } + break; + } + else if arc_msg.is_text() || arc_msg.is_binary() { + let t_sender_lock = t_senders.lock().unwrap(); + for (iter_id, sender) in (*t_sender_lock).iter() { + let arc_msg_clone = Arc::clone(&arc_msg); + if *iter_id != id { + match (*sender).send(arc_msg_clone) { + Ok(_) => println!("message sent to thread no {}", *iter_id), + Err(e) => println!("My Error: {}", e) + } + } + } + } } Err(e) => { + let mut t_sender_lock = t_senders.lock().unwrap(); + (*t_sender_lock).remove(&id); println!("MyError Occured: {}", e); + println!("Connection Closed"); break; } - _ => { - println!("not handled"); - }, - } + } } }); + + let s_websocket = Arc::clone(&websocket); + spawn(move || { + let id = connection_count; + loop { + match receiver.recv() { + Ok(arc_msg) => { + println!("Message received by thread {}", id); + println!("The message is {}", arc_msg); + let mut wssocket_lock = s_websocket.lock().unwrap(); + println!("Lock acquired"); + let msg = (*arc_msg).clone(); + //TODO add match for this + let _ = (*wssocket_lock).send(msg); + println!("message sent to client {}", id); + } + Err(e) => { + println!("Error in receiving from other threads: {}", e); + break; + } + } + } + }); + } }