-
-
Save marciol/29fe2dd4ac9a29ae4152469776738061 to your computer and use it in GitHub Desktop.
A UNIX socket server written in Rust
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
// A UNIX socket server written in pure Rust | |
use std::io::Read; | |
use std::os::unix::net::{UnixListener, UnixStream}; | |
use std::path::Path; | |
fn main() { | |
let socket = Path::new("/tmp/echo.sock"); | |
if socket.exists() { | |
std::fs::remove_file(socket).unwrap(); | |
} | |
let listener = UnixListener::bind(socket).unwrap(); | |
println!("Listening on socket {:?}", socket); | |
println!("===> Connect using netcat: `nc -U {:?}`", socket); | |
println!("===> Connect using curl: `curl -s --unix-socket {:?} http://localhost/`", socket); | |
println!("===> Press Ctrl+C to exit\n",); | |
for stream in listener.incoming() { | |
match stream { | |
Ok(stream) => { | |
handle_stream(stream); | |
} | |
Err(err) => { | |
println!("Error while accepting connection: {:?}", err); | |
break; | |
} | |
} | |
} | |
} | |
fn handle_stream(mut stream: UnixStream) { | |
let mut buf = [0; 512]; | |
loop { | |
match stream.read(&mut buf) { | |
Ok(0) => break, | |
Ok(n) => { | |
let data = String::from_utf8(buf[0..n].to_vec()).unwrap(); | |
println!("{}", data); | |
break; | |
} | |
Err(_) => break, | |
} | |
} | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment