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;
|
|
|
|
|
2017-03-24 09:57:38 -04:00
|
|
|
use error::Error;
|
|
|
|
|
2016-10-06 09:36:23 -04:00
|
|
|
pub fn read8<R>(reader: &mut R) -> Result<u8, io::Error> where R: io::Read {
|
|
|
|
let mut buf: [u8; 1] = unsafe { ::std::mem::uninitialized() };
|
|
|
|
reader.read_exact(&mut buf).and(Ok(buf[0]))
|
|
|
|
}
|
|
|
|
|
|
|
|
pub fn read16<R>(reader: &mut R) -> Result<u16, io::Error> where R: io::Read {
|
|
|
|
let mut buf: [u8; 2] = unsafe { ::std::mem::uninitialized() };
|
|
|
|
try!(reader.read_exact(&mut buf));
|
|
|
|
Ok(((buf[0] as u16) << 8) + buf[1] as u16)
|
|
|
|
}
|
2016-10-08 20:11:10 -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> {
|
|
|
|
const ASCII_0: u8 = 0x30;
|
|
|
|
const ASCII_9: u8 = 0x39;
|
|
|
|
|
|
|
|
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)
|
|
|
|
}
|
|
|
|
|
2016-10-08 20:11:10 -04:00
|
|
|
#[cfg(test)]
|
|
|
|
mod tests {
|
|
|
|
use std::io::Cursor;
|
|
|
|
use std::io::ErrorKind;
|
|
|
|
use std::io::Read;
|
|
|
|
use super::*;
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn read8_len() {
|
|
|
|
let mut reader = Cursor::new([]);
|
|
|
|
assert_err_kind!(read8(&mut reader), ErrorKind::UnexpectedEof);
|
|
|
|
let mut reader = Cursor::new([0x01]);
|
|
|
|
assert_ok!(read8(&mut reader), 0x01);
|
|
|
|
let mut reader = Cursor::new([0x01, 0x02]);
|
|
|
|
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() {
|
|
|
|
let mut reader = Cursor::new([]);
|
|
|
|
assert_err_kind!(read16(&mut reader), ErrorKind::UnexpectedEof);
|
|
|
|
let mut reader = Cursor::new([0x01]);
|
|
|
|
assert_err_kind!(read16(&mut reader), ErrorKind::UnexpectedEof);
|
|
|
|
let mut reader = Cursor::new([0x01, 0x02]);
|
|
|
|
assert_ok!(read16(&mut reader), 0x0102);
|
|
|
|
let mut reader = Cursor::new([0x01, 0x02, 0x03]);
|
|
|
|
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
|
|
|
}
|