Merge pull request #10 from pion/rust-websocket

Add rust websocket

Author: ValorZard

webrtc/main.go

@@ -80,7 +80,7 @@ func main() {
 				log.Println("Channel Open")
 				for i := 10; i < 510; i += 10 {
 					for j := 10; j < 510; j += 10 {
-						message := fmt.Sprintf("%03d,%03d", j, i)
+						message := fmt.Sprintf("%d,%d", j, i)
 						if err := channel.Send([]byte(message)); err != nil {
 							log.Fatal(err)
 						}

websocket/main.go

@@ -23,7 +23,7 @@ func main() {
 		log.Println("Client Connected")
 		for i := 10; i < 510; i += 10 {
 			for j := 10; j < 510; j += 10 {
-				message := fmt.Sprintf("%03d,%03d", j, i)
+				message := fmt.Sprintf("%d,%d", j, i)
 				if err := conn.WriteMessage(websocket.TextMessage, []byte(message)); err != nil {
 					log.Fatal(err)
 				}

websocket/tungstenite-server/src/main.rs

@@ -0,0 +1,79 @@
+use std::net::TcpListener;
+use std::path::Path;
+use std::sync::Arc;
+use std::thread::spawn;
+
+use rustls::pki_types::pem::PemObject;
+use rustls::pki_types::{CertificateDer, PrivateKeyDer};
+use rustls::{ServerConfig, ServerConnection, StreamOwned};
+use tungstenite::accept;
+
+/// A WebSocket echo server over TLS (wss://)
+fn main() {
+    // Use fixed relative paths from this crate to the repo certs
+    let cert_path = Path::new("../../certs/localhost.pem");
+    let key_path = Path::new("../../certs/localhost-key.pem");
+
+    eprintln!("Using certificate: {}", cert_path.display());
+    eprintln!("Using private key: {}", key_path.display());
+
+    let cert = CertificateDer::from_pem_file(cert_path).expect("load certs");
+    let key = PrivateKeyDer::from_pem_file(key_path).expect("load private key");
+
+    // Build rustls server config (no client auth)
+    let config = ServerConfig::builder()
+        .with_no_client_auth()
+        .with_single_cert(vec![cert], key)
+        .expect("invalid cert/key");
+    let config = Arc::new(config);
+
+    let listener = TcpListener::bind("127.0.0.1:8000").expect("bind 127.0.0.1:8000");
+    eprintln!("tungstenite server listening on wss://127.0.0.1:8000");
+
+    for stream in listener.incoming() {
+        match stream {
+            Ok(stream) => {
+                eprintln!(
+                    "incoming TCP connection from {}",
+                    stream.peer_addr().unwrap()
+                );
+                let cfg = Arc::clone(&config);
+                spawn(move || {
+                    // Wrap TCP in a rustls TLS stream.
+                    let conn = match ServerConnection::new(cfg) {
+                        Ok(c) => c,
+                        Err(e) => {
+                            eprintln!("TLS ServerConnection error: {e}");
+                            return;
+                        }
+                    };
+                    let tls_stream = StreamOwned::new(conn, stream);
+
+                    match accept(tls_stream) {
+                        Ok(mut websocket) => loop {
+                            // Send message so we can test the server performance
+                            for i in (10..=510).step_by(10) {
+                                for j in (10..=510).step_by(10) {
+                                    let message = format!("{j},{i}");
+                                    if let Err(e) = websocket.send(message.into()) {
+                                        eprintln!("send error: {e}");
+                                        break;
+                                    }
+                                    std::thread::sleep(std::time::Duration::from_millis(1));
+                                }
+                            }
+                            // break after one batch for demo purposes
+                            eprintln!("Completed one batch of messages, closing connection.");
+                            break;
+                        },
+                        Err(e) => {
+                            eprintln!("websocket handshake failed: {e}");
+                            eprintln!("Hint: Ensure your client connects with wss://localhost:8000 and trusts the local certificate in certs/.");
+                        }
+                    }
+                });
+            }
+            Err(e) => eprintln!("incoming connection error: {e}"),
+        }
+    }
+}