Add a JingleIBB parser.

This commit is contained in:
Emmanuel Gil Peyrot 2017-04-22 19:15:48 +01:00
parent 22ec4b1913
commit ff5be32a0e
3 changed files with 107 additions and 0 deletions

101
src/jingle_ibb.rs Normal file
View file

@ -0,0 +1,101 @@
use std::str::FromStr;
use minidom::Element;
use error::Error;
use ns;
use ibb::Stanza;
#[derive(Debug, Clone)]
pub struct Transport {
block_size: u16,
sid: String,
stanza: Stanza,
}
fn optional_attr<T: FromStr>(root: &Element, attr: &str) -> Option<T> {
root.attr(attr)
.and_then(|value| value.parse().ok())
}
fn required_attr<T: FromStr>(root: &Element, attr: &str, err: Error) -> Result<T, Error> {
optional_attr(root, attr).ok_or(err)
}
pub fn parse_jingle_ibb(root: &Element) -> Result<Transport, Error> {
if root.is("transport", ns::JINGLE_IBB) {
for _ in root.children() {
return Err(Error::ParseError("Unknown child in JingleIBB element."));
}
let block_size = required_attr(root, "block-size", Error::ParseError("Required attribute 'block-size' missing in JingleIBB element."))?;
let sid = required_attr(root, "sid", Error::ParseError("Required attribute 'sid' missing in JingleIBB element."))?;
let stanza = root.attr("stanza")
.unwrap_or("iq")
.parse()?;
Ok(Transport {
block_size: block_size,
sid: sid,
stanza: stanza
})
} else {
Err(Error::ParseError("This is not an JingleIBB element."))
}
}
#[cfg(test)]
mod tests {
use minidom::Element;
use error::Error;
use ibb;
use jingle_ibb;
#[test]
fn test_simple() {
let elem: Element = "<transport xmlns='urn:xmpp:jingle:transports:ibb:1' block-size='3' sid='coucou'/>".parse().unwrap();
let transport = jingle_ibb::parse_jingle_ibb(&elem).unwrap();
assert_eq!(transport.block_size, 3);
assert_eq!(transport.sid, "coucou");
assert_eq!(transport.stanza, ibb::Stanza::Iq);
}
#[test]
fn test_invalid() {
let elem: Element = "<transport xmlns='urn:xmpp:jingle:transports:ibb:1'/>".parse().unwrap();
let error = jingle_ibb::parse_jingle_ibb(&elem).unwrap_err();
let message = match error {
Error::ParseError(string) => string,
_ => panic!(),
};
assert_eq!(message, "Required attribute 'block-size' missing in JingleIBB element.");
// TODO: maybe make a better error message here.
let elem: Element = "<transport xmlns='urn:xmpp:jingle:transports:ibb:1' block-size='-5'/>".parse().unwrap();
let error = jingle_ibb::parse_jingle_ibb(&elem).unwrap_err();
let message = match error {
Error::ParseError(string) => string,
_ => panic!(),
};
assert_eq!(message, "Required attribute 'block-size' missing in JingleIBB element.");
let elem: Element = "<transport xmlns='urn:xmpp:jingle:transports:ibb:1' block-size='128'/>".parse().unwrap();
let error = jingle_ibb::parse_jingle_ibb(&elem).unwrap_err();
let message = match error {
Error::ParseError(string) => string,
_ => panic!(),
};
assert_eq!(message, "Required attribute 'sid' missing in JingleIBB element.");
}
#[test]
fn test_invalid_stanza() {
let elem: Element = "<transport xmlns='urn:xmpp:jingle:transports:ibb:1' block-size='128' sid='coucou' stanza='fdsq'/>".parse().unwrap();
let error = jingle_ibb::parse_jingle_ibb(&elem).unwrap_err();
let message = match error {
Error::ParseError(string) => string,
_ => panic!(),
};
assert_eq!(message, "Invalid 'stanza' attribute.");
}
}

View file

@ -52,6 +52,9 @@ pub mod attention;
/// XEP-0234: Jingle File Transfer
pub mod jingle_ft;
/// XEP-0261: Jingle In-Band Bytestreams Transport Method
pub mod jingle_ibb;
/// XEP-0300: Use of Cryptographic Hash Functions in XMPP
pub mod hashes;

View file

@ -36,6 +36,9 @@ pub const JINGLE_FT: &'static str = "urn:xmpp:jingle:apps:file-transfer:5";
/// XEP-0234: Jingle File Transfer
pub const JINGLE_FT_ERROR: &'static str = "urn:xmpp:jingle:apps:file-transfer:errors:0";
/// XEP-0261: Jingle In-Band Bytestreams Transport Method
pub const JINGLE_IBB: &'static str = "urn:xmpp:jingle:transports:ibb:1";
/// XEP-0300: Use of Cryptographic Hash Functions in XMPP
pub const HASHES: &'static str = "urn:xmpp:hashes:2";