pep/xmpp-rs
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

jingle_s5b: Strengthen type safety for ids.

Browse source
This commit is contained in:
Emmanuel Gil Peyrot 2017-06-25 21:38:58 +01:00
parent f428ee070d
commit 3ea0c45337
1 changed files with 43 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

54
src/jingle_s5b.rs
View file

@ -20,9 +20,46 @@ generate_attribute!(Type, "type", {
Tunnel => "tunnel", Tunnel => "tunnel",
}, Default = Direct); }, Default = Direct);
generate_attribute!(Mode, "mode", {
Tcp => "tcp",
Udp => "udp",
}, Default = Tcp);
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
pub struct CandidateId(String);
impl FromStr for CandidateId {
type Err = Error;
fn from_str(s: &str) -> Result<CandidateId, Error> {
Ok(CandidateId(String::from(s)))
}
}
impl IntoAttributeValue for CandidateId {
fn into_attribute_value(self) -> Option<String> {
return Some(self.0);
}
}
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
pub struct StreamId(String);
impl FromStr for StreamId {
type Err = Error;
fn from_str(s: &str) -> Result<StreamId, Error> {
Ok(StreamId(String::from(s)))
}
}
impl IntoAttributeValue for StreamId {
fn into_attribute_value(self) -> Option<String> {
return Some(self.0);
}
}
#[derive(Debug, Clone)] #[derive(Debug, Clone)]
pub struct Candidate { pub struct Candidate {
pub cid: String, pub cid: CandidateId,
pub host: String, pub host: String,
pub jid: String, pub jid: String,
pub port: Option<u16>, pub port: Option<u16>,
@ -44,11 +81,6 @@ impl Into<Element> for Candidate {
} }
} }
generate_attribute!(Mode, "mode", {
Tcp => "tcp",
Udp => "udp",
}, Default = Tcp);
#[derive(Debug, Clone)] #[derive(Debug, Clone)]
pub enum TransportPayload { pub enum TransportPayload {
Activated(String), Activated(String),
@ -61,7 +93,7 @@ pub enum TransportPayload {
#[derive(Debug, Clone)] #[derive(Debug, Clone)]
pub struct Transport { pub struct Transport {
pub sid: String, pub sid: StreamId,
pub dstaddr: Option<String>, pub dstaddr: Option<String>,
pub mode: Mode, pub mode: Mode,
pub payload: TransportPayload, pub payload: TransportPayload,
@ -181,7 +213,7 @@ mod tests {
fn test_simple() { fn test_simple() {
let elem: Element = "<transport xmlns='urn:xmpp:jingle:transports:s5b:1' sid='coucou'/>".parse().unwrap(); let elem: Element = "<transport xmlns='urn:xmpp:jingle:transports:s5b:1' sid='coucou'/>".parse().unwrap();
let transport = Transport::try_from(elem).unwrap(); let transport = Transport::try_from(elem).unwrap();
assert_eq!(transport.sid, "coucou"); assert_eq!(transport.sid, StreamId(String::from("coucou")));
assert_eq!(transport.dstaddr, None); assert_eq!(transport.dstaddr, None);
assert_eq!(transport.mode, Mode::Tcp); assert_eq!(transport.mode, Mode::Tcp);
match transport.payload { match transport.payload {
@ -194,7 +226,7 @@ mod tests {
fn test_serialise_activated() { fn test_serialise_activated() {
let elem: Element = "<transport xmlns='urn:xmpp:jingle:transports:s5b:1' sid='coucou'><activated cid='coucou'/></transport>".parse().unwrap(); let elem: Element = "<transport xmlns='urn:xmpp:jingle:transports:s5b:1' sid='coucou'><activated cid='coucou'/></transport>".parse().unwrap();
let transport = Transport { let transport = Transport {
sid: String::from("coucou"), sid: StreamId(String::from("coucou")),
dstaddr: None, dstaddr: None,
mode: Mode::Tcp, mode: Mode::Tcp,
payload: TransportPayload::Activated(String::from("coucou")), payload: TransportPayload::Activated(String::from("coucou")),
@ -207,11 +239,11 @@ mod tests {
fn test_serialise_candidate() { fn test_serialise_candidate() {
let elem: Element = "<transport xmlns='urn:xmpp:jingle:transports:s5b:1' sid='coucou'><candidate cid='coucou' host='coucou' jid='coucou@coucou' priority='0'/></transport>".parse().unwrap(); let elem: Element = "<transport xmlns='urn:xmpp:jingle:transports:s5b:1' sid='coucou'><candidate cid='coucou' host='coucou' jid='coucou@coucou' priority='0'/></transport>".parse().unwrap();
let transport = Transport { let transport = Transport {
sid: String::from("coucou"), sid: StreamId(String::from("coucou")),
dstaddr: None, dstaddr: None,
mode: Mode::Tcp, mode: Mode::Tcp,
payload: TransportPayload::Candidates(vec!(Candidate { payload: TransportPayload::Candidates(vec!(Candidate {
cid: String::from("coucou"), cid: CandidateId(String::from("coucou")),
host: String::from("coucou"), host: String::from("coucou"),
jid: String::from("coucou@coucou"), jid: String::from("coucou@coucou"),
port: None, port: None,