vault 11.0.1

A parser for Company of Heroes replay files.
Documentation 
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
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95

use crate::data::Span;
use byteorder::{LittleEndian, ReadBytesExt};
use nom::bytes::complete::{take, take_while};
use nom::combinator::{map, peek, verify};
use nom::error::ParseError;
use nom::multi::many_till;
use nom::number::complete::{le_u16, le_u32};
use nom::{IResult, ToUsize};
use std::io::Cursor;
use std::string::String;

pub fn verify_zero_u16(input: Span) -> IResult<Span, u16> {
    verify(le_u16, |n: &u16| *n == 0)(input)
}

pub fn verify_le_u32<'a>(expected: u32) -> impl FnMut(Span<'a>) -> IResult<Span<'a>, u32> {
    verify(le_u32, move |n: &u32| *n == expected)
}

pub fn parse_utf8_fixed<'a, E, T: ToUsize>(
    len: T,
) -> impl FnMut(Span<'a>) -> IResult<Span<'a>, String, E>
where
    E: ParseError<Span<'a>>,
{
    map(take(len), |s: Span| {
        String::from_utf8_lossy(s.fragment()).into_owned()
    })
}

pub fn parse_utf8_variable<'a, O, E, F>(
    mut f: F,
) -> impl FnMut(Span<'a>) -> IResult<Span<'a>, (O, String), E>
where
    E: ParseError<Span<'a>>,
    F: FnMut(Span<'a>) -> IResult<Span<'a>, O, E>,
    O: ToUsize + Copy,
{
    move |input: Span| {
        let (input, num) = f(input)?;
        let (input, res) = parse_utf8_fixed(num)(input)?;

        Ok((input, (num, res)))
    }
}

fn bytes_to_utf16(bytes: Span) -> String {
    let mut u16_vec = Vec::with_capacity(bytes.len() / 2);
    let mut cursor = Cursor::new(bytes.fragment());

    for _ in 1..=(bytes.len() / 2) {
        let val = cursor.read_u16::<LittleEndian>().unwrap();
        u16_vec.push(val);
    }

    String::from_utf16_lossy(&u16_vec)
}

pub fn parse_utf16_terminated(input: Span) -> IResult<Span, String> {
    map(
        many_till(le_u16, peek(verify(le_u16, |n: &u16| *n == 0))),
        |(u16s, _)| String::from_utf16_lossy(&u16s),
    )(input)
}

pub fn parse_utf16_fixed<'a, E, T>(len: T) -> impl FnMut(Span<'a>) -> IResult<Span<'a>, String, E>
where
    E: ParseError<Span<'a>>,
    T: ToUsize,
{
    let len = len.to_usize();
    let true_len = len * 2;

    map(take(true_len), bytes_to_utf16)
}

pub fn parse_utf16_variable<'a, O, E, F>(
    mut f: F,
) -> impl FnMut(Span<'a>) -> IResult<Span<'a>, (O, String), E>
where
    E: ParseError<Span<'a>>,
    F: FnMut(Span<'a>) -> IResult<Span<'a>, O, E>,
    O: ToUsize + Copy,
{
    move |input: Span| {
        let (input, num) = f(input)?;
        let (input, res) = parse_utf16_fixed(num)(input)?;

        Ok((input, (num, res)))
    }
}

pub fn take_zeroes(input: Span) -> IResult<Span, Span> {
    take_while(|n: u8| n == 0)(input)
}