From c0f90386900f0f767c7d0569e0f758d305af3d09 Mon Sep 17 00:00:00 2001 From: metamuffin Date: Wed, 17 Aug 2022 17:49:31 +0200 Subject: Revert "Revert "modularize interfaces"" This reverts commit c10e88e42b26637e3c48ed781ced1382fb20fa26. --- karld/src/interface/unix.rs | 77 +++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 77 insertions(+) create mode 100644 karld/src/interface/unix.rs (limited to 'karld/src/interface/unix.rs') diff --git a/karld/src/interface/unix.rs b/karld/src/interface/unix.rs new file mode 100644 index 0000000..8d038d4 --- /dev/null +++ b/karld/src/interface/unix.rs @@ -0,0 +1,77 @@ +use crate::{handle_packet, CLIENT_ID_COUNTER}; +use karlcommon::{socket_path, version, ClientboundPacket, ProtoError, ServerboundPacket}; +use log::{debug, error, info, warn}; +use std::io; +use std::io::{BufRead, BufReader, ErrorKind, Write}; +use std::os::unix::net::{UnixListener, UnixStream}; +use std::thread; + +pub fn run() { + if socket_path().exists() { + info!("remove old socket"); + std::fs::remove_file(socket_path()).unwrap(); + } + let listener = UnixListener::bind(socket_path()).unwrap(); + info!("listening."); + + loop { + let (stream, addr) = listener.accept().unwrap(); + let id = CLIENT_ID_COUNTER.fetch_add(1, std::sync::atomic::Ordering::Relaxed); + thread::spawn(move || { + info!("client connected: {:?}", addr); + if let Err(err) = handle_connection(id, stream) { + warn!("client dropped: {:?} ({})", addr, err); + } else { + info!("client dropped: {:?}", addr); + } + }); + } +} + +fn handle_connection(id: u32, mut stream: UnixStream) -> io::Result<()> { + let mut reader = BufReader::new(stream.try_clone()?); + let (responder, responses) = crossbeam_channel::unbounded(); + responder + .send(ClientboundPacket::Handshake { + version: version!(), + }) + .unwrap(); + thread::spawn(move || { + for m in responses { + debug!("{id} -> {m:?}"); + match stream + .write_fmt(format_args!("{}\n", serde_json::to_string(&m).unwrap())) + .map_err(|e| e.kind()) + { + Ok(_) => (), + Err(ErrorKind::BrokenPipe) => break, + Err(e) => error!("network error: {:?}", e), + } + } + }); + { + let mut buf = String::new(); + loop { + if reader.read_line(&mut buf)? == 0 { + break Ok(()); + }; + match serde_json::from_str::(buf.as_str()) { + Ok(packet) => { + debug!("{id} <- {packet:?}"); + handle_packet(id, packet, responder.clone()); + } + Err(err) => { + warn!("client error: {:?}", &err); + responder + .send(ClientboundPacket::Error(ProtoError::FormatError(format!( + "{}", + &err + )))) + .map_err(|_| io::Error::from(ErrorKind::InvalidInput))? + } + } + + buf.clear(); + } + } +} -- cgit v1.2.3-70-g09d2