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

bitflags! {
    #[derive(Debug)]
    flags Flag: u8 {
        const END_STREAM = 0x1,
        const ACK = 0x1,
        const END_HEADERS = 0x4,
        const PADDED = 0x8,
        const PRIORITY = 0x20
    }
}

impl Flag {
    pub fn new(data: u8) -> Result<Flag, ()> {
        match Flag::from_bits(data) {
            Some(v) => Ok(v),
            None => Err(())
        }
    }

    // Note that ACK and END_STREAM are the same value, but they are only present
    // on different frame types.
    pub fn ack() -> Flag { ACK }
    pub fn end_stream() -> Flag { END_STREAM }
    pub fn end_headers() -> Flag { END_HEADERS }
    pub fn padded() -> Flag { PADDED }
    pub fn priority() -> Flag { PRIORITY }
}

#[cfg(test)]
mod tests {
    use super::Flag;

    const FLAG_EMPTY: u8 = 0x0;
    const FLAG_END_STREAM_OR_ACK: u8 = 0x1;
    const FLAG_END_HEADERS: u8 = 0x4;
    const FLAG_PADDED: u8 = 0x8;

    #[test]
    fn test_flag_empty() {
        assert_eq!(Flag::empty().bits(), FLAG_EMPTY);
    }

    #[test]
    fn test_flag_from_bits() {
        assert_eq!(Flag::from_bits(FLAG_EMPTY).unwrap(), Flag::empty());
        assert_eq!(Flag::from_bits(FLAG_END_HEADERS | FLAG_PADDED).unwrap(),
                   Flag::end_headers() | Flag::padded());
        assert_eq!(Flag::from_bits(FLAG_END_STREAM_OR_ACK | FLAG_PADDED).unwrap(),
                   Flag::end_stream() | Flag::padded());
    }
}