forked from M-Labs/artiq
93 lines
2.4 KiB
Rust
93 lines
2.4 KiB
Rust
#![allow(dead_code)]
|
|
|
|
use std::io::{self, Read, Write};
|
|
use std::vec::Vec;
|
|
use std::string::String;
|
|
use byteorder::{ByteOrder, NetworkEndian};
|
|
|
|
// FIXME: replace these with byteorder core io traits once those are in
|
|
pub fn read_u8(reader: &mut Read) -> io::Result<u8> {
|
|
let mut bytes = [0; 1];
|
|
reader.read_exact(&mut bytes)?;
|
|
Ok(bytes[0])
|
|
}
|
|
|
|
pub fn write_u8(writer: &mut Write, value: u8) -> io::Result<()> {
|
|
let bytes = [value; 1];
|
|
writer.write_all(&bytes)
|
|
}
|
|
|
|
pub fn read_bool(reader: &mut Read) -> io::Result<bool> {
|
|
if read_u8(reader)? == 0 {
|
|
Ok(false)
|
|
} else {
|
|
Ok(true)
|
|
}
|
|
}
|
|
|
|
pub fn write_bool(writer: &mut Write, value: bool) -> io::Result<()> {
|
|
if value {
|
|
write_u8(writer, 1)
|
|
} else {
|
|
write_u8(writer, 0)
|
|
}
|
|
}
|
|
|
|
pub fn read_u16(reader: &mut Read) -> io::Result<u16> {
|
|
let mut bytes = [0; 2];
|
|
reader.read_exact(&mut bytes)?;
|
|
Ok(NetworkEndian::read_u16(&bytes))
|
|
}
|
|
|
|
pub fn write_u16(writer: &mut Write, value: u16) -> io::Result<()> {
|
|
let mut bytes = [0; 2];
|
|
NetworkEndian::write_u16(&mut bytes, value);
|
|
writer.write_all(&bytes)
|
|
}
|
|
|
|
pub fn read_u32(reader: &mut Read) -> io::Result<u32> {
|
|
let mut bytes = [0; 4];
|
|
reader.read_exact(&mut bytes)?;
|
|
Ok(NetworkEndian::read_u32(&bytes))
|
|
}
|
|
|
|
pub fn write_u32(writer: &mut Write, value: u32) -> io::Result<()> {
|
|
let mut bytes = [0; 4];
|
|
NetworkEndian::write_u32(&mut bytes, value);
|
|
writer.write_all(&bytes)
|
|
}
|
|
|
|
pub fn read_u64(reader: &mut Read) -> io::Result<u64> {
|
|
let mut bytes = [0; 8];
|
|
reader.read_exact(&mut bytes)?;
|
|
Ok(NetworkEndian::read_u64(&bytes))
|
|
}
|
|
|
|
pub fn write_u64(writer: &mut Write, value: u64) -> io::Result<()> {
|
|
let mut bytes = [0; 8];
|
|
NetworkEndian::write_u64(&mut bytes, value);
|
|
writer.write_all(&bytes)
|
|
}
|
|
|
|
pub fn read_bytes(reader: &mut Read) -> io::Result<Vec<u8>> {
|
|
let length = read_u32(reader)?;
|
|
let mut value = vec![0; length as usize];
|
|
reader.read_exact(&mut value)?;
|
|
Ok(value)
|
|
}
|
|
|
|
pub fn write_bytes(writer: &mut Write, value: &[u8]) -> io::Result<()> {
|
|
write_u32(writer, value.len() as u32)?;
|
|
writer.write_all(value)
|
|
}
|
|
|
|
pub fn read_string(reader: &mut Read) -> io::Result<String> {
|
|
let bytes = read_bytes(reader)?;
|
|
String::from_utf8(bytes)
|
|
.map_err(|_| io::Error::new(io::ErrorKind::InvalidData, "invalid UTF-8"))
|
|
}
|
|
|
|
pub fn write_string(writer: &mut Write, value: &str) -> io::Result<()> {
|
|
write_bytes(writer, value.as_bytes())
|
|
}
|