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.
110 lines
3.0 KiB
110 lines
3.0 KiB
// Copyright (c) 2021-2022 Laurent Pelecq
|
|
//
|
|
// Licensed under the Apache License, Version 2.0
|
|
// <LICENSE-APACHE or http://www.apache.org/licenses/LICENSE-2.0> or the MIT
|
|
// license <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
|
|
// option. All files in the project carrying such notice may not be copied,
|
|
// modified, or distributed except according to those terms.
|
|
|
|
use std::io::{self, BufRead, BufReader, BufWriter, Write};
|
|
use std::path::Path;
|
|
use std::thread;
|
|
|
|
use std::os::unix::net::UnixListener;
|
|
|
|
/// Server traits
|
|
pub trait Server {
|
|
fn serve(&mut self) -> io::Result<()>;
|
|
}
|
|
|
|
/// Server on a named socket.
|
|
pub struct UnixServer {
|
|
listener: UnixListener,
|
|
communication: Vec<(&'static str, &'static str)>,
|
|
}
|
|
|
|
impl UnixServer {
|
|
/// Create a new server on a named socket.
|
|
///
|
|
/// Argument `communication` is an array of pairs. The first item is a list of strings
|
|
/// the server will receive and the second item is the answer.
|
|
pub fn new<P>(
|
|
socket_path: P,
|
|
communication: &[(&'static str, &'static str)],
|
|
) -> io::Result<Self>
|
|
where
|
|
P: AsRef<Path>,
|
|
{
|
|
let listener = UnixListener::bind(socket_path.as_ref())?;
|
|
Ok(Self {
|
|
listener,
|
|
communication: communication.to_vec(),
|
|
})
|
|
}
|
|
|
|
fn split_lines(lines: &str) -> Vec<String> {
|
|
lines
|
|
.trim_end()
|
|
.split("\r\n")
|
|
.map(|s| format!("{}\r\n", s))
|
|
.collect::<Vec<String>>()
|
|
}
|
|
}
|
|
|
|
impl Server for UnixServer {
|
|
fn serve(&mut self) -> io::Result<()> {
|
|
let (stream, _) = self.listener.accept()?;
|
|
let mut input = BufReader::new(stream.try_clone()?);
|
|
let mut output = BufWriter::new(stream);
|
|
for (questions, answer) in self.communication.iter() {
|
|
for question in Self::split_lines(questions).iter() {
|
|
let mut line = String::new();
|
|
input.read_line(&mut line)?;
|
|
if line != *question {
|
|
return Err(io::Error::new(
|
|
io::ErrorKind::InvalidInput,
|
|
format!("read <{}> instead of <{}>", line, *question),
|
|
));
|
|
}
|
|
}
|
|
output.write_all(answer.as_bytes())?;
|
|
output.flush()?;
|
|
}
|
|
Ok(())
|
|
}
|
|
}
|
|
|
|
/// Run the server in a thread
|
|
pub fn run_server(mut server: Box<dyn Server + Send>) -> thread::JoinHandle<io::Result<()>> {
|
|
thread::spawn(move || -> io::Result<()> {
|
|
server.serve()?;
|
|
Ok(())
|
|
})
|
|
}
|
|
|
|
pub fn run_unix<P>(
|
|
socket_path: P,
|
|
communication: &'static [(&'static str, &'static str)],
|
|
) -> io::Result<thread::JoinHandle<io::Result<()>>>
|
|
where
|
|
P: AsRef<Path>,
|
|
{
|
|
Ok(run_server(Box::new(UnixServer::new(
|
|
&socket_path,
|
|
communication,
|
|
)?)))
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod test {
|
|
|
|
use super::UnixServer;
|
|
|
|
#[test]
|
|
fn test_split_lines() {
|
|
const ONE_LINE: &str = "one line\r\n";
|
|
let one_line = UnixServer::split_lines(ONE_LINE);
|
|
assert_eq!(&[ONE_LINE], one_line.as_slice());
|
|
}
|
|
}
|