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
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182

pub mod auth;
mod content;
mod language;
mod named;
pub mod parse;
pub mod via;
mod write;
pub use self::{
    auth::{AuthHeader, AuthContext, AuthSchema},
    content::ContentType,
    language::Language,
    named::NamedHeader,
    parse::parse_header
};

use crate::core::Method;

/// Wrapper around a Vec<Header> to simplify creating
/// and a list of headers
#[derive(Debug, PartialEq, Clone, Default)]
pub struct Headers(pub Vec<Header>);

impl Headers {
    pub fn new() -> Headers {
        Headers(vec![])
    }

    /// Push a new Header onto the stack.
    pub fn push(&mut self, h: Header) {
        self.0.push(h)
    }

    /// Access the underlying vec iterator.
    pub fn iter(&self) -> impl Iterator<Item = &Header> {
        self.0.iter()
    }

    /// Add the Headers onto the interior Vec<Header>.
    pub fn extend(&mut self, i: Vec<Header>) {
        self.0.extend(i)
    }

    /// Return An Expires header if one is present.
    pub fn expires(&self) -> Option<Header> {
        for h in &self.0 {
            if let Header::Expires(i) = h {
                return Some(Header::Expires(*i));
            }
        }
        None
    }

    /// Return the CSeq header if one is present.
    pub fn cseq(&self) -> Option<Header> {
        for h in &self.0 {
            if let Header::CSeq(a, b) = h {
                return Some(Header::CSeq(*a, *b));
            }
        }
        None
    }

    /// Return the From header if one is present.
    pub fn from(&self) -> Option<Header> {
        for h in &self.0 {
            if let Header::From(a) = h {
                return Some(Header::From(a.clone()));
            }
        }
        None
    }

    /// Return the To header if one is present.
    pub fn to(&self) -> Option<Header> {
        for h in &self.0 {
            if let Header::To(a) = h {
                return Some(Header::To(a.clone()));
            }
        }
        None
    }

    /// Return the Contact header if one is present.
    pub fn contact(&self) -> Option<Header> {
        for h in &self.0 {
            if let Header::Contact(a) = h {
                return Some(Header::Contact(a.clone()));
            }
        }
        None
    }

    /// Return the CoallId header if one is present.
    pub fn call_id(&self) -> Option<Header> {
        for h in &self.0 {
            if let Header::CallId(a) = h {
                return Some(Header::CallId(a.clone()));
            }
        }
        None
    }

    /// Return the Via header if one is present.
    pub fn via(&self) -> Option<Header> {
        for h in &self.0 {
            if let Header::Via(a) = h {
                return Some(Header::Via(a.clone()));
            }
        }
        None
    }

    /// Return XFS Sending Header if one is present.
    pub fn xfs_sending_message(&self) -> Option<Header> {
        for h in &self.0 {
            if let Header::XFsSendingMessage(a) = h {
                return Some(Header::XFsSendingMessage(a.clone()));
            }
        }
        None
    }
}

impl IntoIterator for Headers {
    type IntoIter = ::std::vec::IntoIter<Self::Item>;
    type Item = Header;

    fn into_iter(self) -> Self::IntoIter {
        self.0.into_iter()
    }
}

/// Single SIP Header Representation.
#[derive(Debug, PartialEq, Clone)]
pub enum Header {
    To(NamedHeader),
    Contact(NamedHeader),
    From(NamedHeader),
    ReplyTo(NamedHeader),
    CSeq(u32, Method),
    MaxForwards(u32),
    Expires(u32),
    Accept(Vec<Method>),
    ContentLength(u32),
    Allow(Vec<Method>),
    UserAgent(String),
    CallId(String),
    ContentType(ContentType),
    ContentLanguage(Language),
    ContentEncoding(ContentType),
    AcceptLanguage(Language),
    AcceptEncoding(ContentType),
    AlertInfo(String),
    ErrorInfo(String),
    AuthenticationInfo(String),
    Authorization(auth::AuthHeader),
    CallInfo(String),
    InReplyTo(String),
    ContentDisposition(String),
    Date(String),
    MinExpires(u32),
    MimeVersion(f32),
    Organization(String),
    ProxyAuthenticate(String),
    ProxyAuthorization(String),
    ProxyRequire(String),
    Require(String),
    RetryAfter(String),
    Route(String),
    Subject(String),
    RecordRoute(String),
    Server(String),
    Supported(Vec<String>),
    Timestamp(u32),
    Unsupported(String),
    Warning(String),
    Via(via::ViaHeader),
    Priority(String),
    WwwAuthenticate(auth::AuthHeader),
    XFsSendingMessage(String),
    Other(String, String),
}