You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
41 lines
1.2 KiB
41 lines
1.2 KiB
use std::net::{TcpListener, TcpStream};
|
|
use std::thread::spawn;
|
|
|
|
use log::*;
|
|
use tungstenite::handshake::HandshakeRole;
|
|
use tungstenite::{accept, Error, HandshakeError, Message, Result};
|
|
|
|
fn must_not_block<Role: HandshakeRole>(err: HandshakeError<Role>) -> Error {
|
|
match err {
|
|
HandshakeError::Interrupted(_) => panic!("Bug: blocking socket would block"),
|
|
HandshakeError::Failure(f) => f,
|
|
}
|
|
}
|
|
|
|
fn handle_client(stream: TcpStream) -> Result<()> {
|
|
let mut socket = accept(stream).map_err(must_not_block)?;
|
|
loop {
|
|
match socket.read_message()? {
|
|
msg @ Message::Text(_) | msg @ Message::Binary(_) => {
|
|
socket.write_message(msg)?;
|
|
}
|
|
Message::Ping(_) | Message::Pong(_) | Message::Close(_) => {}
|
|
}
|
|
}
|
|
}
|
|
|
|
fn main() {
|
|
env_logger::init();
|
|
|
|
let server = TcpListener::bind("127.0.0.1:9002").unwrap();
|
|
|
|
for stream in server.incoming() {
|
|
spawn(move || match stream {
|
|
Ok(stream) => match handle_client(stream) {
|
|
Ok(_) => (),
|
|
Err(e) => warn!("Error in client: {}", e),
|
|
},
|
|
Err(e) => warn!("Error accepting stream: {}", e),
|
|
});
|
|
}
|
|
}
|
|
|