1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25

use std::io::{Error, ErrorKind, Result};
use crate::{Readable, Writable};
use crate::variable_len::{read_variable_u64, write_variable_u64};

pub fn read_u8_vec(source: &mut impl Readable) -> Result<Vec<u8>> {
    let length = read_variable_u64(source)? as usize;
    let mut bytes = vec![0; length];
    source.read_more(&mut bytes)?;
    Ok(bytes)
}
pub fn write_u8_vec(target: &mut impl Writable, message: &[u8]) -> Result<usize> {
    let size = write_variable_u64(target, message.len() as u64)?;
    target.write_more(message)?;
    Ok(size + message.len())
}

pub fn read_string(source: &mut impl Readable) -> Result<String> {
    match String::from_utf8(read_u8_vec(source)?) {
        Ok(s) => Ok(s),
        Err(e) => Err(Error::new(ErrorKind::InvalidData, e.to_string())),
    }
}
pub fn write_string(target: &mut impl Writable, message: &String) -> Result<usize> {
    write_u8_vec(target, message.as_bytes())
}