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

use nom::{branch::alt, bytes::complete::tag_no_case, combinator::map, error::ParseError, IResult};
use serde::{Deserialize, Serialize};

use std::fmt;

/// SIP protocol transport.
#[derive(Debug, PartialEq, Clone, Copy, Serialize, Deserialize)]
pub enum Transport {
    Udp,
    Tcp,
}

impl Default for Transport {
    fn default() -> Transport {
        Transport::Udp
    }
}

impl fmt::Display for Transport {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        match self {
            Transport::Udp => write!(f, "UDP"),
            Transport::Tcp => write!(f, "TCP"),
        }
    }
}

/// Parse a SIP message transport protocol.
pub fn parse_transport<'a, E: ParseError<&'a [u8]>>(
    input: &'a [u8],
) -> IResult<&'a [u8], Transport, E> {
    alt::<_, _, E, _>((
        map(tag_no_case::<_, _, E>("TCP"), |_| Transport::Tcp),
        map(tag_no_case::<_, _, E>("UDP"), |_| Transport::Udp),
    ))(input)
}