forked from bonsai/harakit
48 lines
1023 B
Rust
48 lines
1023 B
Rust
use std::{
|
|
env::args,
|
|
io::{ stdin, stdout, Error, ErrorKind, Read, Write },
|
|
process::ExitCode,
|
|
vec::Vec
|
|
};
|
|
|
|
extern crate sysexits;
|
|
|
|
use sysexits::{ EX_OK, EX_OSERR };
|
|
|
|
fn oserr(s: &str, e: Error) -> ExitCode {
|
|
eprintln!("{}: {}", s, e);
|
|
ExitCode::from(EX_OSERR as u8)
|
|
}
|
|
|
|
fn main() -> ExitCode {
|
|
let argv = args().collect::<Vec<String>>();
|
|
let mut buf: Vec<u8> = Vec::new();
|
|
let mut input = stdin();
|
|
let mut output = stdout().lock();
|
|
|
|
let wordsize: usize = 2;
|
|
let force = false;
|
|
|
|
buf.resize(wordsize, 0);
|
|
|
|
loop {
|
|
match input.read(&mut buf) {
|
|
Ok(0) => break ExitCode::from(EX_OK as u8),
|
|
Ok(v) if v == wordsize => {
|
|
let (left, right) = buf.split_at(v/2);
|
|
if let Err(e) = output.write(&right)
|
|
.and_then(|_| output.write(&left)) {
|
|
break oserr(&argv[0], e)
|
|
}
|
|
},
|
|
Ok(v) => {
|
|
if let Err(e) = output.write(&buf[..v]) {
|
|
break oserr(&argv[0], e)
|
|
}
|
|
},
|
|
Err(e) if e.kind() == ErrorKind::Interrupted && force => continue,
|
|
Err(e) => break oserr(&argv[0], e)
|
|
}
|
|
}
|
|
}
|