2016-10-06 09:36:23 -04:00
|
|
|
//
|
|
|
|
// Copyright (c) 2016 KAMADA Ken'ichi.
|
|
|
|
// All rights reserved.
|
|
|
|
//
|
|
|
|
// Redistribution and use in source and binary forms, with or without
|
|
|
|
// modification, are permitted provided that the following conditions
|
|
|
|
// are met:
|
|
|
|
// 1. Redistributions of source code must retain the above copyright
|
|
|
|
// notice, this list of conditions and the following disclaimer.
|
|
|
|
// 2. Redistributions in binary form must reproduce the above copyright
|
|
|
|
// notice, this list of conditions and the following disclaimer in the
|
|
|
|
// documentation and/or other materials provided with the distribution.
|
|
|
|
//
|
|
|
|
// THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
|
|
|
|
// ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
|
|
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
|
|
|
// ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
|
|
|
|
// FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
|
|
|
|
// DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
|
|
|
|
// OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
|
|
|
|
// HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
|
|
|
|
// LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
|
|
|
|
// OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
|
|
|
|
// SUCH DAMAGE.
|
|
|
|
//
|
|
|
|
|
|
|
|
use std::io;
|
2021-02-23 08:20:01 -05:00
|
|
|
use std::io::Read as _;
|
2016-10-06 09:36:23 -04:00
|
|
|
|
2019-04-01 08:11:54 -04:00
|
|
|
use crate::error::Error;
|
2017-03-24 09:57:38 -04:00
|
|
|
|
2017-10-01 08:17:17 -04:00
|
|
|
const ASCII_0: u8 = 0x30;
|
|
|
|
const ASCII_9: u8 = 0x39;
|
2017-07-23 08:43:02 -04:00
|
|
|
|
2016-10-06 09:36:23 -04:00
|
|
|
pub fn read8<R>(reader: &mut R) -> Result<u8, io::Error> where R: io::Read {
|
2019-03-16 01:48:16 -04:00
|
|
|
let mut buf = [0u8; 1];
|
2016-10-06 09:36:23 -04:00
|
|
|
reader.read_exact(&mut buf).and(Ok(buf[0]))
|
|
|
|
}
|
|
|
|
|
|
|
|
pub fn read16<R>(reader: &mut R) -> Result<u16, io::Error> where R: io::Read {
|
2019-03-16 01:48:16 -04:00
|
|
|
let mut buf = [0u8; 2];
|
2018-04-15 23:11:08 -04:00
|
|
|
reader.read_exact(&mut buf)?;
|
2019-03-26 10:25:46 -04:00
|
|
|
Ok(u16::from_be_bytes(buf))
|
2016-10-06 09:36:23 -04:00
|
|
|
}
|
2016-10-08 20:11:10 -04:00
|
|
|
|
2020-01-19 06:48:04 -05:00
|
|
|
pub fn read64<R>(reader: &mut R) -> Result<u64, io::Error> where R: io::Read {
|
|
|
|
let mut buf = [0u8; 8];
|
|
|
|
reader.read_exact(&mut buf)?;
|
|
|
|
Ok(u64::from_be_bytes(buf))
|
|
|
|
}
|
|
|
|
|
2020-08-12 10:35:00 -04:00
|
|
|
pub trait BufReadExt {
|
|
|
|
fn discard_exact(&mut self, len: usize) -> io::Result<()>;
|
2021-02-23 08:20:01 -05:00
|
|
|
fn is_eof(&mut self) -> io::Result<bool>;
|
2020-08-12 10:35:00 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
impl<T> BufReadExt for T where T: io::BufRead {
|
|
|
|
fn discard_exact(&mut self, mut len: usize) -> io::Result<()> {
|
|
|
|
while len > 0 {
|
2020-09-06 02:23:36 -04:00
|
|
|
let consume_len = match self.fill_buf() {
|
2021-01-03 21:06:15 -05:00
|
|
|
Ok(buf) if buf.is_empty() =>
|
|
|
|
return Err(io::Error::new(
|
|
|
|
io::ErrorKind::UnexpectedEof, "unexpected EOF")),
|
2020-09-06 02:23:36 -04:00
|
|
|
Ok(buf) => buf.len().min(len),
|
|
|
|
Err(e) if e.kind() == io::ErrorKind::Interrupted => continue,
|
|
|
|
Err(e) => return Err(e),
|
|
|
|
};
|
2020-08-12 10:35:00 -04:00
|
|
|
self.consume(consume_len);
|
|
|
|
len -= consume_len;
|
|
|
|
}
|
|
|
|
Ok(())
|
2020-08-03 09:57:50 -04:00
|
|
|
}
|
2021-02-23 08:20:01 -05:00
|
|
|
|
|
|
|
fn is_eof(&mut self) -> io::Result<bool> {
|
|
|
|
loop {
|
|
|
|
match self.fill_buf() {
|
|
|
|
Ok(buf) => return Ok(buf.is_empty()),
|
|
|
|
Err(e) if e.kind() == io::ErrorKind::Interrupted => continue,
|
|
|
|
Err(e) => return Err(e),
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
pub trait ReadExt {
|
|
|
|
fn read_exact_len(&mut self, buf: &mut Vec<u8>, len: usize)
|
|
|
|
-> io::Result<()>;
|
|
|
|
}
|
|
|
|
|
|
|
|
impl<T> ReadExt for T where T: io::Read {
|
|
|
|
fn read_exact_len(&mut self, buf: &mut Vec<u8>, len: usize)
|
|
|
|
-> io::Result<()> {
|
|
|
|
// Using `vec![0; len]` and `read_exact` is more efficient but
|
|
|
|
// less robust against broken files; a small file can easily
|
|
|
|
// trigger OOM by a huge length value without actual data.
|
|
|
|
// When the fallible allocation feature is stabilized,
|
|
|
|
// we could revisit this.
|
|
|
|
if self.take(len as u64).read_to_end(buf)? != len {
|
|
|
|
return Err(io::Error::new(
|
|
|
|
io::ErrorKind::UnexpectedEof, "unexpected EOF"));
|
|
|
|
}
|
|
|
|
Ok(())
|
|
|
|
}
|
2020-08-03 09:57:50 -04:00
|
|
|
}
|
|
|
|
|
2017-03-24 09:57:38 -04:00
|
|
|
// This function must not be called with more than 4 bytes.
|
|
|
|
pub fn atou16(bytes: &[u8]) -> Result<u16, Error> {
|
|
|
|
if cfg!(debug_assertions) && bytes.len() >= 5 {
|
|
|
|
panic!("atou16 accepts up to 4 bytes");
|
|
|
|
}
|
|
|
|
if bytes.len() == 0 {
|
|
|
|
return Err(Error::InvalidFormat("Not a number"));
|
|
|
|
}
|
|
|
|
let mut n = 0;
|
|
|
|
for &c in bytes {
|
|
|
|
if c < ASCII_0 || ASCII_9 < c {
|
|
|
|
return Err(Error::InvalidFormat("Not a number"));
|
|
|
|
}
|
|
|
|
n = n * 10 + (c - ASCII_0) as u16;
|
|
|
|
}
|
|
|
|
Ok(n)
|
|
|
|
}
|
|
|
|
|
2017-10-01 08:17:17 -04:00
|
|
|
pub fn ctou32(c: u8) -> Result<u32, Error> {
|
|
|
|
if c < ASCII_0 || ASCII_9 < c {
|
|
|
|
return Err(Error::InvalidFormat("Not a number"));
|
|
|
|
}
|
|
|
|
Ok((c - ASCII_0) as u32)
|
|
|
|
}
|
|
|
|
|
2016-10-08 20:11:10 -04:00
|
|
|
#[cfg(test)]
|
|
|
|
mod tests {
|
|
|
|
use std::io::ErrorKind;
|
|
|
|
use std::io::Read;
|
|
|
|
use super::*;
|
|
|
|
|
2021-01-03 21:06:15 -05:00
|
|
|
#[test]
|
|
|
|
fn discard_exact() {
|
|
|
|
let mut buf = b"abc".as_ref();
|
|
|
|
buf.discard_exact(1).unwrap();
|
|
|
|
assert_eq!(buf, b"bc");
|
|
|
|
buf.discard_exact(2).unwrap();
|
|
|
|
assert_eq!(buf, b"");
|
|
|
|
buf.discard_exact(1).unwrap_err();
|
|
|
|
}
|
|
|
|
|
2016-10-08 20:11:10 -04:00
|
|
|
#[test]
|
|
|
|
fn read8_len() {
|
2020-12-02 07:48:40 -05:00
|
|
|
let data = [];
|
|
|
|
assert_err_kind!(read8(&mut &data[..]), ErrorKind::UnexpectedEof);
|
|
|
|
let data = [0x01];
|
|
|
|
assert_ok!(read8(&mut &data[..]), 0x01);
|
|
|
|
let data = [0x01, 0x02];
|
|
|
|
let mut reader = &data[..];
|
2016-10-08 20:11:10 -04:00
|
|
|
let mut buf = Vec::new();
|
|
|
|
assert_ok!(read8(&mut reader), 0x01);
|
|
|
|
assert_ok!(reader.read_to_end(&mut buf), 1);
|
|
|
|
assert_eq!(buf, [0x02]);
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn read16_len() {
|
2020-12-02 07:48:40 -05:00
|
|
|
let data = [];
|
|
|
|
assert_err_kind!(read16(&mut &data[..]), ErrorKind::UnexpectedEof);
|
|
|
|
let data = [0x01];
|
|
|
|
assert_err_kind!(read16(&mut &data[..]), ErrorKind::UnexpectedEof);
|
|
|
|
let data = [0x01, 0x02];
|
|
|
|
assert_ok!(read16(&mut &data[..]), 0x0102);
|
|
|
|
let data = [0x01, 0x02, 0x03];
|
|
|
|
let mut reader = &data[..];
|
2016-10-08 20:11:10 -04:00
|
|
|
let mut buf = Vec::new();
|
|
|
|
assert_ok!(read16(&mut reader), 0x0102);
|
|
|
|
assert_ok!(reader.read_to_end(&mut buf), 1);
|
|
|
|
assert_eq!(buf, [0x03]);
|
|
|
|
}
|
2017-03-24 09:57:38 -04:00
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn atou16_misc() {
|
|
|
|
assert_ok!(atou16(b"0"), 0);
|
|
|
|
assert_ok!(atou16(b"0010"), 10);
|
|
|
|
assert_ok!(atou16(b"9999"), 9999);
|
|
|
|
assert_err_pat!(atou16(b""), Error::InvalidFormat(_));
|
|
|
|
assert_err_pat!(atou16(b"/"), Error::InvalidFormat(_));
|
|
|
|
assert_err_pat!(atou16(b":"), Error::InvalidFormat(_));
|
|
|
|
assert_err_pat!(atou16(b"-1"), Error::InvalidFormat(_));
|
|
|
|
}
|
2016-10-08 20:11:10 -04:00
|
|
|
}
|