pep/xmpp-rs-mirror
1
0
Fork 0
mirror of https://gitlab.com/xmpp-rs/xmpp-rs.git synced 2024-07-12 22:21:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

cleaning up connection logic, but it's still quite ugly

Browse source
This commit is contained in:
lumi 2017-02-24 17:36:17 +01:00
parent 158d871df9
commit 79434f2681
1 changed files with 49 additions and 48 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

81
src/client.rs
View file

@ -121,46 +121,21 @@ impl Client {
Ok(())
}
/// Connects using the specified SASL mechanism.
/// Connects and authenticates using the specified SASL mechanism.
pub fn connect<S: SaslMechanism>(&mut self, mechanism: &mut S) -> Result<(), Error> {
// TODO: this is very ugly
loop {
let e = self.transport.read_event().unwrap();
match e {
ReaderEvent::StartElement { .. } => {
break;
},
_ => (),
}
}
let mut did_sasl = false;
loop {
let n = self.transport.read_element().unwrap();
if n.is("features", ns::STREAM) {
if did_sasl {
let mut elem = Element::builder("iq")
.attr("id", "bind")
.attr("type", "set")
.build();
let bind = Element::builder("bind")
.ns(ns::BIND)
.build();
elem.append_child(bind);
self.transport.write_element(&elem)?;
}
else {
self.wait_for_features()?;
let auth = mechanism.initial();
let mut elem = Element::builder("auth")
.ns(ns::SASL)
.attr("mechanism", "PLAIN")
.attr("mechanism", S::name())
.build();
if !auth.is_empty() {
elem.append_text_node(base64::encode(&auth));
}
self.transport.write_element(&elem)?;
}
}
else if n.is("challenge", ns::SASL) {
loop {
let n = self.transport.read_element()?;
if n.is("challenge", ns::SASL) {
let text = n.text();
let challenge = if text == "" {
Vec::new()
@ -178,9 +153,39 @@ impl Client {
self.transport.write_element(&elem)?;
}
else if n.is("success", ns::SASL) {
did_sasl = true;
self.transport.reset_stream();
C2S::init(&mut self.transport, &self.jid.domain, "after_sasl")?;
return self.bind();
}
else if n.is("failure", ns::SASL) {
let msg = n.text();
let inner = if msg == "" { None } else { Some(msg) };
return Err(Error::SaslError(inner));
}
}
}
fn bind(&mut self) -> Result<(), Error> {
self.wait_for_features();
let mut elem = Element::builder("iq")
.attr("id", "bind")
.attr("type", "set")
.build();
let bind = Element::builder("bind")
.ns(ns::BIND)
.build();
elem.append_child(bind);
self.transport.write_element(&elem)?;
loop {
let n = self.transport.read_element()?;
if n.is("iq", ns::CLIENT) && n.has_child("bind", ns::BIND) {
return Ok(());
}
}
}
fn wait_for_features(&mut self) -> Result<Element, Error> {
// TODO: this is very ugly
loop {
let e = self.transport.read_event()?;
match e {
@ -190,14 +195,10 @@ impl Client {
_ => (),
}
}
}
else if n.is("failure", ns::SASL) {
let msg = n.text();
let inner = if msg == "" { None } else { Some(msg) };
return Err(Error::SaslError(inner));
}
else if n.is("iq", ns::CLIENT) && n.has_child("bind", ns::BIND) {
return Ok(());
loop {
let n = self.transport.read_element()?;
if n.is("features", ns::STREAM) {
return Ok(n);
}
}
}