xbnet/src/pipe.rs
John Goerzen e80f891340 cargo fmt
2020-09-20 23:56:11 -05:00

61 lines
1.7 KiB
Rust

/*
Copyright (C) 2019 John Goerzen <jgoerzen@complete.org
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
use crate::ser::*;
use crate::xb::*;
use crate::xbpacket::*;
use crate::xbrx::*;
use bytes::*;
use crossbeam_channel;
use std::io;
use std::io::{Read, Write};
pub fn stdin_processor(
dest: u64,
maxframesize: usize,
sender: crossbeam_channel::Sender<XBTX>,
) -> io::Result<()> {
let stdin = io::stdin();
let mut br = io::BufReader::new(stdin);
let mut buf = vec![0u8; maxframesize - 1];
loop {
let res = br.read(&mut buf)?;
if res == 0 {
// EOF
sender.send(XBTX::Shutdown).unwrap();
return Ok(());
}
sender
.send(XBTX::TXData(
XBDestAddr::U64(dest),
Bytes::copy_from_slice(&buf[0..res]),
))
.unwrap();
}
}
pub fn stdout_processor(xbreframer: &mut XBReframer, ser: &mut XBSerReader) -> io::Result<()> {
let mut stdout = io::stdout();
loop {
let (_fromu64, _fromu16, payload) = xbreframer.rxframe(ser);
stdout.write_all(&payload)?;
stdout.flush()?;
}
}