mp4san 0.3.0

An MP4 file sanitizer.
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

#![allow(missing_docs)]

use std::mem::size_of;

use bytes::Buf;
use bytes::BufMut;

use crate::error::Result;

use super::error::WhileParsingType;
use super::ParseError;

pub trait Mpeg4Int: Clone + Copy + Sized {
    fn parse<B: Buf>(buf: B) -> Result<Self, ParseError>;
    fn put_buf<B: BufMut>(&self, buf: B);
}

pub trait Mpeg4IntReaderExt: Buf {
    fn get<T: Mpeg4Int>(&mut self) -> Result<T, ParseError> {
        T::parse(self)
    }
}

pub trait Mpeg4IntWriterExt: BufMut {
    fn put_mp4int<T: Mpeg4Int>(&mut self, value: T) {
        value.put_buf(self)
    }
}

macro_rules! mpeg4_int {
    ($($ty:ty => ($get_fun:ident, $put_fun:ident)),+ $(,)?) => {
        $(impl Mpeg4Int for $ty {
            fn parse<B: Buf>(mut buf: B) -> Result<Self, ParseError> {
                if buf.remaining() < size_of::<Self>() {
                    use crate::parse::error::WhileParsingType;
                    bail_attach!(ParseError::TruncatedBox, WhileParsingType(stringify!($ty)));
                }
                Ok(buf.$get_fun())
            }

            fn put_buf<B: BufMut>(&self, mut buf: B) {
                buf.$put_fun(*self)
            }
        })+
    };
}

mpeg4_int! {
    u8 => (get_u8, put_u8),
    u16 => (get_u16, put_u16),
    u32 => (get_u32, put_u32),
    u64 => (get_u64, put_u64),
    i8 => (get_i8, put_i8),
    i16 => (get_i16, put_i16),
    i32 => (get_i32, put_i32),
    i64 => (get_i64, put_i64),
}

impl<T: Mpeg4Int, const N: usize> Mpeg4Int for [T; N] {
    fn parse<B: Buf>(mut buf: B) -> Result<Self, ParseError> {
        ensure_attach!(
            buf.remaining() >= size_of::<Self>(),
            ParseError::TruncatedBox,
            WhileParsingType::new::<Self>(),
        );
        Ok([(); N].map(|()| T::parse(&mut buf).unwrap_or_else(|_| unreachable!())))
    }

    fn put_buf<B: BufMut>(&self, mut buf: B) {
        for value in self {
            buf.put_mp4int(*value);
        }
    }
}

impl<T: Buf> Mpeg4IntReaderExt for T {}

impl<T: BufMut> Mpeg4IntWriterExt for T {}