forked from M-Labs/artiq
1
0
Fork 0

libdrtioaux: always inline read/write helpers.

This shrinks Packet::{read_from,write_to} by 1K.
This commit is contained in:
whitequark 2018-05-13 13:39:14 +00:00
parent 32522be413
commit aa42a69849
1 changed files with 14 additions and 0 deletions

View File

@ -6,17 +6,20 @@ use std::string::String;
use byteorder::{ByteOrder, NetworkEndian}; use byteorder::{ByteOrder, NetworkEndian};
// FIXME: replace these with byteorder core io traits once those are in // FIXME: replace these with byteorder core io traits once those are in
#[inline(always)]
pub fn read_u8(reader: &mut Read) -> io::Result<u8> { pub fn read_u8(reader: &mut Read) -> io::Result<u8> {
let mut bytes = [0; 1]; let mut bytes = [0; 1];
reader.read_exact(&mut bytes)?; reader.read_exact(&mut bytes)?;
Ok(bytes[0]) Ok(bytes[0])
} }
#[inline(always)]
pub fn write_u8(writer: &mut Write, value: u8) -> io::Result<()> { pub fn write_u8(writer: &mut Write, value: u8) -> io::Result<()> {
let bytes = [value; 1]; let bytes = [value; 1];
writer.write_all(&bytes) writer.write_all(&bytes)
} }
#[inline(always)]
pub fn read_bool(reader: &mut Read) -> io::Result<bool> { pub fn read_bool(reader: &mut Read) -> io::Result<bool> {
if read_u8(reader)? == 0 { if read_u8(reader)? == 0 {
Ok(false) Ok(false)
@ -25,6 +28,7 @@ pub fn read_bool(reader: &mut Read) -> io::Result<bool> {
} }
} }
#[inline(always)]
pub fn write_bool(writer: &mut Write, value: bool) -> io::Result<()> { pub fn write_bool(writer: &mut Write, value: bool) -> io::Result<()> {
if value { if value {
write_u8(writer, 1) write_u8(writer, 1)
@ -33,42 +37,49 @@ pub fn write_bool(writer: &mut Write, value: bool) -> io::Result<()> {
} }
} }
#[inline(always)]
pub fn read_u16(reader: &mut Read) -> io::Result<u16> { pub fn read_u16(reader: &mut Read) -> io::Result<u16> {
let mut bytes = [0; 2]; let mut bytes = [0; 2];
reader.read_exact(&mut bytes)?; reader.read_exact(&mut bytes)?;
Ok(NetworkEndian::read_u16(&bytes)) Ok(NetworkEndian::read_u16(&bytes))
} }
#[inline(always)]
pub fn write_u16(writer: &mut Write, value: u16) -> io::Result<()> { pub fn write_u16(writer: &mut Write, value: u16) -> io::Result<()> {
let mut bytes = [0; 2]; let mut bytes = [0; 2];
NetworkEndian::write_u16(&mut bytes, value); NetworkEndian::write_u16(&mut bytes, value);
writer.write_all(&bytes) writer.write_all(&bytes)
} }
#[inline(always)]
pub fn read_u32(reader: &mut Read) -> io::Result<u32> { pub fn read_u32(reader: &mut Read) -> io::Result<u32> {
let mut bytes = [0; 4]; let mut bytes = [0; 4];
reader.read_exact(&mut bytes)?; reader.read_exact(&mut bytes)?;
Ok(NetworkEndian::read_u32(&bytes)) Ok(NetworkEndian::read_u32(&bytes))
} }
#[inline(always)]
pub fn write_u32(writer: &mut Write, value: u32) -> io::Result<()> { pub fn write_u32(writer: &mut Write, value: u32) -> io::Result<()> {
let mut bytes = [0; 4]; let mut bytes = [0; 4];
NetworkEndian::write_u32(&mut bytes, value); NetworkEndian::write_u32(&mut bytes, value);
writer.write_all(&bytes) writer.write_all(&bytes)
} }
#[inline(always)]
pub fn read_u64(reader: &mut Read) -> io::Result<u64> { pub fn read_u64(reader: &mut Read) -> io::Result<u64> {
let mut bytes = [0; 8]; let mut bytes = [0; 8];
reader.read_exact(&mut bytes)?; reader.read_exact(&mut bytes)?;
Ok(NetworkEndian::read_u64(&bytes)) Ok(NetworkEndian::read_u64(&bytes))
} }
#[inline(always)]
pub fn write_u64(writer: &mut Write, value: u64) -> io::Result<()> { pub fn write_u64(writer: &mut Write, value: u64) -> io::Result<()> {
let mut bytes = [0; 8]; let mut bytes = [0; 8];
NetworkEndian::write_u64(&mut bytes, value); NetworkEndian::write_u64(&mut bytes, value);
writer.write_all(&bytes) writer.write_all(&bytes)
} }
#[inline(always)]
pub fn read_bytes(reader: &mut Read) -> io::Result<Vec<u8>> { pub fn read_bytes(reader: &mut Read) -> io::Result<Vec<u8>> {
let length = read_u32(reader)?; let length = read_u32(reader)?;
let mut value = vec![0; length as usize]; let mut value = vec![0; length as usize];
@ -76,17 +87,20 @@ pub fn read_bytes(reader: &mut Read) -> io::Result<Vec<u8>> {
Ok(value) Ok(value)
} }
#[inline(always)]
pub fn write_bytes(writer: &mut Write, value: &[u8]) -> io::Result<()> { pub fn write_bytes(writer: &mut Write, value: &[u8]) -> io::Result<()> {
write_u32(writer, value.len() as u32)?; write_u32(writer, value.len() as u32)?;
writer.write_all(value) writer.write_all(value)
} }
#[inline(always)]
pub fn read_string(reader: &mut Read) -> io::Result<String> { pub fn read_string(reader: &mut Read) -> io::Result<String> {
let bytes = read_bytes(reader)?; let bytes = read_bytes(reader)?;
String::from_utf8(bytes) String::from_utf8(bytes)
.map_err(|_| io::Error::new(io::ErrorKind::InvalidData, "invalid UTF-8")) .map_err(|_| io::Error::new(io::ErrorKind::InvalidData, "invalid UTF-8"))
} }
#[inline(always)]
pub fn write_string(writer: &mut Write, value: &str) -> io::Result<()> { pub fn write_string(writer: &mut Write, value: &str) -> io::Result<()> {
write_bytes(writer, value.as_bytes()) write_bytes(writer, value.as_bytes())
} }