unstringify Error type
This commit is contained in:
parent
2e0dd44352
commit
e9d30f16c3
12 changed files with 279 additions and 167 deletions
|
|
@ -28,3 +28,4 @@ xmpp-parsers = "0.11"
|
||||||
idna = "0.1"
|
idna = "0.1"
|
||||||
try_from = "0.2"
|
try_from = "0.2"
|
||||||
quick-xml = "0.12"
|
quick-xml = "0.12"
|
||||||
|
derive-error = "0.0.4"
|
||||||
|
|
|
||||||
|
|
@ -13,9 +13,11 @@ use try_from::TryFrom;
|
||||||
use xmpp_codec::Packet;
|
use xmpp_codec::Packet;
|
||||||
use xmpp_stream::XMPPStream;
|
use xmpp_stream::XMPPStream;
|
||||||
use stream_start::StreamStart;
|
use stream_start::StreamStart;
|
||||||
|
use {Error, AuthError, ProtocolError};
|
||||||
|
|
||||||
const NS_XMPP_SASL: &str = "urn:ietf:params:xml:ns:xmpp-sasl";
|
const NS_XMPP_SASL: &str = "urn:ietf:params:xml:ns:xmpp-sasl";
|
||||||
|
|
||||||
|
|
||||||
pub struct ClientAuth<S: AsyncWrite> {
|
pub struct ClientAuth<S: AsyncWrite> {
|
||||||
state: ClientAuthState<S>,
|
state: ClientAuthState<S>,
|
||||||
mechanism: Box<Mechanism>,
|
mechanism: Box<Mechanism>,
|
||||||
|
|
@ -29,7 +31,7 @@ enum ClientAuthState<S: AsyncWrite> {
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<S: AsyncWrite> ClientAuth<S> {
|
impl<S: AsyncWrite> ClientAuth<S> {
|
||||||
pub fn new(stream: XMPPStream<S>, creds: Credentials) -> Result<Self, String> {
|
pub fn new(stream: XMPPStream<S>, creds: Credentials) -> Result<Self, Error> {
|
||||||
let mechs: Vec<Box<Mechanism>> = vec![
|
let mechs: Vec<Box<Mechanism>> = vec![
|
||||||
Box::new(Scram::<Sha256>::from_credentials(creds.clone()).unwrap()),
|
Box::new(Scram::<Sha256>::from_credentials(creds.clone()).unwrap()),
|
||||||
Box::new(Scram::<Sha1>::from_credentials(creds.clone()).unwrap()),
|
Box::new(Scram::<Sha1>::from_credentials(creds.clone()).unwrap()),
|
||||||
|
|
@ -40,7 +42,7 @@ impl<S: AsyncWrite> ClientAuth<S> {
|
||||||
let mech_names: Vec<String> =
|
let mech_names: Vec<String> =
|
||||||
match stream.stream_features.get_child("mechanisms", NS_XMPP_SASL) {
|
match stream.stream_features.get_child("mechanisms", NS_XMPP_SASL) {
|
||||||
None =>
|
None =>
|
||||||
return Err("No auth mechanisms".to_owned()),
|
return Err(AuthError::NoMechanism.into()),
|
||||||
Some(mechs) =>
|
Some(mechs) =>
|
||||||
mechs.children()
|
mechs.children()
|
||||||
.filter(|child| child.is("mechanism", NS_XMPP_SASL))
|
.filter(|child| child.is("mechanism", NS_XMPP_SASL))
|
||||||
|
|
@ -53,13 +55,18 @@ impl<S: AsyncWrite> ClientAuth<S> {
|
||||||
let name = mech.name().to_owned();
|
let name = mech.name().to_owned();
|
||||||
if mech_names.iter().any(|name1| *name1 == name) {
|
if mech_names.iter().any(|name1| *name1 == name) {
|
||||||
// println!("SASL mechanism selected: {:?}", name);
|
// println!("SASL mechanism selected: {:?}", name);
|
||||||
let initial = mech.initial()?;
|
let initial = match mech.initial() {
|
||||||
|
Ok(initial) => initial,
|
||||||
|
Err(e) => return Err(AuthError::Sasl(e).into()),
|
||||||
|
};
|
||||||
let mut this = ClientAuth {
|
let mut this = ClientAuth {
|
||||||
state: ClientAuthState::Invalid,
|
state: ClientAuthState::Invalid,
|
||||||
mechanism: mech,
|
mechanism: mech,
|
||||||
};
|
};
|
||||||
let mechanism = XMPPMechanism::from_str(&name)
|
let mechanism = match XMPPMechanism::from_str(&name) {
|
||||||
.map_err(|e| format!("{:?}", e))?;
|
Ok(mechanism) => mechanism,
|
||||||
|
Err(e) => return Err(ProtocolError::Parsers(e).into()),
|
||||||
|
};
|
||||||
this.send(
|
this.send(
|
||||||
stream,
|
stream,
|
||||||
Auth {
|
Auth {
|
||||||
|
|
@ -71,7 +78,7 @@ impl<S: AsyncWrite> ClientAuth<S> {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
Err("No supported SASL mechanism available".to_owned())
|
Err(AuthError::NoMechanism.into())
|
||||||
}
|
}
|
||||||
|
|
||||||
fn send<N: Into<Element>>(&mut self, stream: XMPPStream<S>, nonza: N) {
|
fn send<N: Into<Element>>(&mut self, stream: XMPPStream<S>, nonza: N) {
|
||||||
|
|
@ -83,7 +90,7 @@ impl<S: AsyncWrite> ClientAuth<S> {
|
||||||
|
|
||||||
impl<S: AsyncRead + AsyncWrite> Future for ClientAuth<S> {
|
impl<S: AsyncRead + AsyncWrite> Future for ClientAuth<S> {
|
||||||
type Item = XMPPStream<S>;
|
type Item = XMPPStream<S>;
|
||||||
type Error = String;
|
type Error = Error;
|
||||||
|
|
||||||
fn poll(&mut self) -> Poll<Self::Item, Self::Error> {
|
fn poll(&mut self) -> Poll<Self::Item, Self::Error> {
|
||||||
let state = replace(&mut self.state, ClientAuthState::Invalid);
|
let state = replace(&mut self.state, ClientAuthState::Invalid);
|
||||||
|
|
@ -100,13 +107,14 @@ impl<S: AsyncRead + AsyncWrite> Future for ClientAuth<S> {
|
||||||
Ok(Async::NotReady)
|
Ok(Async::NotReady)
|
||||||
},
|
},
|
||||||
Err(e) =>
|
Err(e) =>
|
||||||
Err(format!("{}", e)),
|
Err(e.into()),
|
||||||
},
|
},
|
||||||
ClientAuthState::WaitRecv(mut stream) =>
|
ClientAuthState::WaitRecv(mut stream) =>
|
||||||
match stream.poll() {
|
match stream.poll() {
|
||||||
Ok(Async::Ready(Some(Packet::Stanza(stanza)))) => {
|
Ok(Async::Ready(Some(Packet::Stanza(stanza)))) => {
|
||||||
if let Ok(challenge) = Challenge::try_from(stanza.clone()) {
|
if let Ok(challenge) = Challenge::try_from(stanza.clone()) {
|
||||||
let response = self.mechanism.response(&challenge.data)?;
|
let response = self.mechanism.response(&challenge.data)
|
||||||
|
.map_err(AuthError::Sasl)?;
|
||||||
self.send(stream, Response { data: response });
|
self.send(stream, Response { data: response });
|
||||||
self.poll()
|
self.poll()
|
||||||
} else if let Ok(_) = Success::try_from(stanza.clone()) {
|
} else if let Ok(_) = Success::try_from(stanza.clone()) {
|
||||||
|
|
@ -114,8 +122,7 @@ impl<S: AsyncRead + AsyncWrite> Future for ClientAuth<S> {
|
||||||
self.state = ClientAuthState::Start(start);
|
self.state = ClientAuthState::Start(start);
|
||||||
self.poll()
|
self.poll()
|
||||||
} else if let Ok(failure) = Failure::try_from(stanza) {
|
} else if let Ok(failure) = Failure::try_from(stanza) {
|
||||||
let e = format!("{:?}", failure.defined_condition);
|
Err(AuthError::Fail(failure.defined_condition).into())
|
||||||
Err(e)
|
|
||||||
} else {
|
} else {
|
||||||
Ok(Async::NotReady)
|
Ok(Async::NotReady)
|
||||||
}
|
}
|
||||||
|
|
@ -129,7 +136,7 @@ impl<S: AsyncRead + AsyncWrite> Future for ClientAuth<S> {
|
||||||
Ok(Async::NotReady)
|
Ok(Async::NotReady)
|
||||||
},
|
},
|
||||||
Err(e) =>
|
Err(e) =>
|
||||||
Err(format!("{}", e)),
|
Err(ProtocolError::Parser(e).into())
|
||||||
},
|
},
|
||||||
ClientAuthState::Start(mut start) =>
|
ClientAuthState::Start(mut start) =>
|
||||||
match start.poll() {
|
match start.poll() {
|
||||||
|
|
@ -140,7 +147,7 @@ impl<S: AsyncRead + AsyncWrite> Future for ClientAuth<S> {
|
||||||
Ok(Async::NotReady)
|
Ok(Async::NotReady)
|
||||||
},
|
},
|
||||||
Err(e) =>
|
Err(e) =>
|
||||||
Err(format!("{}", e)),
|
Err(e.into())
|
||||||
},
|
},
|
||||||
ClientAuthState::Invalid =>
|
ClientAuthState::Invalid =>
|
||||||
unreachable!(),
|
unreachable!(),
|
||||||
|
|
|
||||||
|
|
@ -1,5 +1,4 @@
|
||||||
use std::mem::replace;
|
use std::mem::replace;
|
||||||
use std::error::Error;
|
|
||||||
use futures::{Future, Poll, Async, sink, Stream};
|
use futures::{Future, Poll, Async, sink, Stream};
|
||||||
use tokio_io::{AsyncRead, AsyncWrite};
|
use tokio_io::{AsyncRead, AsyncWrite};
|
||||||
use xmpp_parsers::iq::{Iq, IqType};
|
use xmpp_parsers::iq::{Iq, IqType};
|
||||||
|
|
@ -8,6 +7,7 @@ use try_from::TryFrom;
|
||||||
|
|
||||||
use xmpp_codec::Packet;
|
use xmpp_codec::Packet;
|
||||||
use xmpp_stream::XMPPStream;
|
use xmpp_stream::XMPPStream;
|
||||||
|
use {Error, ProtocolError};
|
||||||
|
|
||||||
const NS_XMPP_BIND: &str = "urn:ietf:params:xml:ns:xmpp-bind";
|
const NS_XMPP_BIND: &str = "urn:ietf:params:xml:ns:xmpp-bind";
|
||||||
const BIND_REQ_ID: &str = "resource-bind";
|
const BIND_REQ_ID: &str = "resource-bind";
|
||||||
|
|
@ -42,7 +42,7 @@ impl<S: AsyncWrite> ClientBind<S> {
|
||||||
|
|
||||||
impl<S: AsyncRead + AsyncWrite> Future for ClientBind<S> {
|
impl<S: AsyncRead + AsyncWrite> Future for ClientBind<S> {
|
||||||
type Item = XMPPStream<S>;
|
type Item = XMPPStream<S>;
|
||||||
type Error = String;
|
type Error = Error;
|
||||||
|
|
||||||
fn poll(&mut self) -> Poll<Self::Item, Self::Error> {
|
fn poll(&mut self) -> Poll<Self::Item, Self::Error> {
|
||||||
let state = replace(self, ClientBind::Invalid);
|
let state = replace(self, ClientBind::Invalid);
|
||||||
|
|
@ -61,7 +61,7 @@ impl<S: AsyncRead + AsyncWrite> Future for ClientBind<S> {
|
||||||
Ok(Async::NotReady)
|
Ok(Async::NotReady)
|
||||||
},
|
},
|
||||||
Err(e) =>
|
Err(e) =>
|
||||||
Err(e.description().to_owned()),
|
Err(e.into())
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
ClientBind::WaitRecv(mut stream) => {
|
ClientBind::WaitRecv(mut stream) => {
|
||||||
|
|
@ -80,7 +80,7 @@ impl<S: AsyncRead + AsyncWrite> Future for ClientBind<S> {
|
||||||
Ok(Async::Ready(stream))
|
Ok(Async::Ready(stream))
|
||||||
},
|
},
|
||||||
_ =>
|
_ =>
|
||||||
Err("resource bind response".to_owned()),
|
Err(ProtocolError::InvalidBindResponse.into()),
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
Ok(Async::NotReady)
|
Ok(Async::NotReady)
|
||||||
|
|
@ -96,7 +96,7 @@ impl<S: AsyncRead + AsyncWrite> Future for ClientBind<S> {
|
||||||
Ok(Async::NotReady)
|
Ok(Async::NotReady)
|
||||||
},
|
},
|
||||||
Err(e) =>
|
Err(e) =>
|
||||||
Err(e.description().to_owned()),
|
Err(e.into()),
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
ClientBind::Invalid =>
|
ClientBind::Invalid =>
|
||||||
|
|
|
||||||
|
|
@ -1,11 +1,11 @@
|
||||||
use std::mem::replace;
|
use std::mem::replace;
|
||||||
use std::str::FromStr;
|
use std::str::FromStr;
|
||||||
use std::error::Error;
|
use std::error::Error as StdError;
|
||||||
use tokio_core::reactor::Handle;
|
use tokio_core::reactor::Handle;
|
||||||
use tokio_core::net::TcpStream;
|
use tokio_core::net::TcpStream;
|
||||||
use tokio_io::{AsyncRead, AsyncWrite};
|
use tokio_io::{AsyncRead, AsyncWrite};
|
||||||
use tokio_tls::TlsStream;
|
use tokio_tls::TlsStream;
|
||||||
use futures::{future, Future, Stream, Poll, Async, Sink, StartSend, AsyncSink};
|
use futures::{Future, Stream, Poll, Async, Sink, StartSend, AsyncSink, done};
|
||||||
use minidom::Element;
|
use minidom::Element;
|
||||||
use jid::{Jid, JidParseError};
|
use jid::{Jid, JidParseError};
|
||||||
use sasl::common::{Credentials, ChannelBinding};
|
use sasl::common::{Credentials, ChannelBinding};
|
||||||
|
|
@ -16,6 +16,7 @@ use super::xmpp_stream;
|
||||||
use super::starttls::{NS_XMPP_TLS, StartTlsClient};
|
use super::starttls::{NS_XMPP_TLS, StartTlsClient};
|
||||||
use super::happy_eyeballs::Connecter;
|
use super::happy_eyeballs::Connecter;
|
||||||
use super::event::Event;
|
use super::event::Event;
|
||||||
|
use super::{Error, ProtocolError};
|
||||||
|
|
||||||
mod auth;
|
mod auth;
|
||||||
use self::auth::ClientAuth;
|
use self::auth::ClientAuth;
|
||||||
|
|
@ -35,7 +36,7 @@ const NS_JABBER_CLIENT: &str = "jabber:client";
|
||||||
enum ClientState {
|
enum ClientState {
|
||||||
Invalid,
|
Invalid,
|
||||||
Disconnected,
|
Disconnected,
|
||||||
Connecting(Box<Future<Item=XMPPStream, Error=String>>),
|
Connecting(Box<Future<Item=XMPPStream, Error=Error>>),
|
||||||
Connected(XMPPStream),
|
Connected(XMPPStream),
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
@ -50,47 +51,47 @@ impl Client {
|
||||||
let connect = Self::make_connect(jid.clone(), password.clone(), handle);
|
let connect = Self::make_connect(jid.clone(), password.clone(), handle);
|
||||||
Ok(Client {
|
Ok(Client {
|
||||||
jid,
|
jid,
|
||||||
state: ClientState::Connecting(connect),
|
state: ClientState::Connecting(Box::new(connect)),
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
fn make_connect(jid: Jid, password: String, handle: Handle) -> Box<Future<Item=XMPPStream, Error=String>> {
|
fn make_connect(jid: Jid, password: String, handle: Handle) -> impl Future<Item=XMPPStream, Error=Error> {
|
||||||
let username = jid.node.as_ref().unwrap().to_owned();
|
let username = jid.node.as_ref().unwrap().to_owned();
|
||||||
let jid1 = jid.clone();
|
let jid1 = jid.clone();
|
||||||
let jid2 = jid.clone();
|
let jid2 = jid.clone();
|
||||||
let password = password;
|
let password = password;
|
||||||
let domain = match idna::domain_to_ascii(&jid.domain) {
|
done(idna::domain_to_ascii(&jid.domain))
|
||||||
Ok(domain) =>
|
.map_err(|_| Error::Idna)
|
||||||
domain,
|
.and_then(|domain|
|
||||||
Err(e) =>
|
done(Connecter::from_lookup(handle, &domain, "_xmpp-client._tcp", 5222))
|
||||||
return Box::new(future::err(format!("{:?}", e))),
|
.map_err(Error::Domain)
|
||||||
};
|
)
|
||||||
Box::new(
|
.and_then(|connecter|
|
||||||
Connecter::from_lookup(handle, &domain, "_xmpp-client._tcp", 5222)
|
connecter
|
||||||
.expect("Connector::from_lookup")
|
.map_err(Error::Connection)
|
||||||
.and_then(move |tcp_stream|
|
).and_then(move |tcp_stream|
|
||||||
xmpp_stream::XMPPStream::start(tcp_stream, jid1, NS_JABBER_CLIENT.to_owned())
|
xmpp_stream::XMPPStream::start(tcp_stream, jid1, NS_JABBER_CLIENT.to_owned())
|
||||||
.map_err(|e| format!("{}", e))
|
).and_then(|xmpp_stream| {
|
||||||
).and_then(|xmpp_stream| {
|
if Self::can_starttls(&xmpp_stream) {
|
||||||
if Self::can_starttls(&xmpp_stream) {
|
Ok(Self::starttls(xmpp_stream))
|
||||||
Ok(Self::starttls(xmpp_stream))
|
} else {
|
||||||
} else {
|
Err(Error::Protocol(ProtocolError::NoTls))
|
||||||
Err("No STARTTLS".to_owned())
|
}
|
||||||
}
|
}).and_then(|starttls|
|
||||||
}).and_then(|starttls|
|
// TODO: flatten?
|
||||||
starttls
|
starttls
|
||||||
).and_then(|tls_stream|
|
).and_then(|tls_stream|
|
||||||
XMPPStream::start(tls_stream, jid2, NS_JABBER_CLIENT.to_owned())
|
XMPPStream::start(tls_stream, jid2, NS_JABBER_CLIENT.to_owned())
|
||||||
.map_err(|e| format!("{}", e))
|
).and_then(move |xmpp_stream|
|
||||||
).and_then(move |xmpp_stream| {
|
done(Self::auth(xmpp_stream, username, password))
|
||||||
Self::auth(xmpp_stream, username, password).expect("auth")
|
// TODO: flatten?
|
||||||
}).and_then(|xmpp_stream| {
|
).and_then(|auth| auth)
|
||||||
Self::bind(xmpp_stream)
|
.and_then(|xmpp_stream| {
|
||||||
}).and_then(|xmpp_stream| {
|
Self::bind(xmpp_stream)
|
||||||
// println!("Bound to {}", xmpp_stream.jid);
|
}).and_then(|xmpp_stream| {
|
||||||
Ok(xmpp_stream)
|
// println!("Bound to {}", xmpp_stream.jid);
|
||||||
})
|
Ok(xmpp_stream)
|
||||||
)
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
fn can_starttls<S>(stream: &xmpp_stream::XMPPStream<S>) -> bool {
|
fn can_starttls<S>(stream: &xmpp_stream::XMPPStream<S>) -> bool {
|
||||||
|
|
@ -103,7 +104,7 @@ impl Client {
|
||||||
StartTlsClient::from_stream(stream)
|
StartTlsClient::from_stream(stream)
|
||||||
}
|
}
|
||||||
|
|
||||||
fn auth<S: AsyncRead + AsyncWrite>(stream: xmpp_stream::XMPPStream<S>, username: String, password: String) -> Result<ClientAuth<S>, String> {
|
fn auth<S: AsyncRead + AsyncWrite>(stream: xmpp_stream::XMPPStream<S>, username: String, password: String) -> Result<ClientAuth<S>, Error> {
|
||||||
let creds = Credentials::default()
|
let creds = Credentials::default()
|
||||||
.with_username(username)
|
.with_username(username)
|
||||||
.with_password(password)
|
.with_password(password)
|
||||||
|
|
@ -118,14 +119,14 @@ impl Client {
|
||||||
|
|
||||||
impl Stream for Client {
|
impl Stream for Client {
|
||||||
type Item = Event;
|
type Item = Event;
|
||||||
type Error = String;
|
type Error = Error;
|
||||||
|
|
||||||
fn poll(&mut self) -> Poll<Option<Self::Item>, Self::Error> {
|
fn poll(&mut self) -> Poll<Option<Self::Item>, Self::Error> {
|
||||||
let state = replace(&mut self.state, ClientState::Invalid);
|
let state = replace(&mut self.state, ClientState::Invalid);
|
||||||
|
|
||||||
match state {
|
match state {
|
||||||
ClientState::Invalid =>
|
ClientState::Invalid =>
|
||||||
Err("invalid client state".to_owned()),
|
Err(Error::InvalidState),
|
||||||
ClientState::Disconnected =>
|
ClientState::Disconnected =>
|
||||||
Ok(Async::Ready(None)),
|
Ok(Async::Ready(None)),
|
||||||
ClientState::Connecting(mut connect) => {
|
ClientState::Connecting(mut connect) => {
|
||||||
|
|
@ -148,7 +149,7 @@ impl Stream for Client {
|
||||||
Ok(Async::NotReady) => (),
|
Ok(Async::NotReady) => (),
|
||||||
Ok(Async::Ready(())) => (),
|
Ok(Async::Ready(())) => (),
|
||||||
Err(e) =>
|
Err(e) =>
|
||||||
return Err(e.description().to_owned()),
|
return Err(Error::Io(e)),
|
||||||
};
|
};
|
||||||
|
|
||||||
// Poll stream
|
// Poll stream
|
||||||
|
|
@ -168,7 +169,7 @@ impl Stream for Client {
|
||||||
Ok(Async::NotReady)
|
Ok(Async::NotReady)
|
||||||
},
|
},
|
||||||
Err(e) =>
|
Err(e) =>
|
||||||
Err(e.description().to_owned()),
|
Err(e.into()),
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -5,6 +5,7 @@ use xmpp_parsers::component::Handshake;
|
||||||
|
|
||||||
use xmpp_codec::Packet;
|
use xmpp_codec::Packet;
|
||||||
use xmpp_stream::XMPPStream;
|
use xmpp_stream::XMPPStream;
|
||||||
|
use {Error, AuthError};
|
||||||
|
|
||||||
const NS_JABBER_COMPONENT_ACCEPT: &str = "jabber:component:accept";
|
const NS_JABBER_COMPONENT_ACCEPT: &str = "jabber:component:accept";
|
||||||
|
|
||||||
|
|
@ -19,7 +20,8 @@ enum ComponentAuthState<S: AsyncWrite> {
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<S: AsyncWrite> ComponentAuth<S> {
|
impl<S: AsyncWrite> ComponentAuth<S> {
|
||||||
pub fn new(stream: XMPPStream<S>, password: String) -> Result<Self, String> {
|
// TODO: doesn't have to be a Result<> actually
|
||||||
|
pub fn new(stream: XMPPStream<S>, password: String) -> Result<Self, Error> {
|
||||||
// FIXME: huge hack, shouldn’t be an element!
|
// FIXME: huge hack, shouldn’t be an element!
|
||||||
let sid = stream.stream_features.name().to_owned();
|
let sid = stream.stream_features.name().to_owned();
|
||||||
let mut this = ComponentAuth {
|
let mut this = ComponentAuth {
|
||||||
|
|
@ -42,7 +44,7 @@ impl<S: AsyncWrite> ComponentAuth<S> {
|
||||||
|
|
||||||
impl<S: AsyncRead + AsyncWrite> Future for ComponentAuth<S> {
|
impl<S: AsyncRead + AsyncWrite> Future for ComponentAuth<S> {
|
||||||
type Item = XMPPStream<S>;
|
type Item = XMPPStream<S>;
|
||||||
type Error = String;
|
type Error = Error;
|
||||||
|
|
||||||
fn poll(&mut self) -> Poll<Self::Item, Self::Error> {
|
fn poll(&mut self) -> Poll<Self::Item, Self::Error> {
|
||||||
let state = replace(&mut self.state, ComponentAuthState::Invalid);
|
let state = replace(&mut self.state, ComponentAuthState::Invalid);
|
||||||
|
|
@ -59,7 +61,7 @@ impl<S: AsyncRead + AsyncWrite> Future for ComponentAuth<S> {
|
||||||
Ok(Async::NotReady)
|
Ok(Async::NotReady)
|
||||||
},
|
},
|
||||||
Err(e) =>
|
Err(e) =>
|
||||||
Err(format!("{}", e)),
|
Err(e.into()),
|
||||||
},
|
},
|
||||||
ComponentAuthState::WaitRecv(mut stream) =>
|
ComponentAuthState::WaitRecv(mut stream) =>
|
||||||
match stream.poll() {
|
match stream.poll() {
|
||||||
|
|
@ -72,8 +74,7 @@ impl<S: AsyncRead + AsyncWrite> Future for ComponentAuth<S> {
|
||||||
Ok(Async::Ready(Some(Packet::Stanza(ref stanza))))
|
Ok(Async::Ready(Some(Packet::Stanza(ref stanza))))
|
||||||
if stanza.is("error", "http://etherx.jabber.org/streams") =>
|
if stanza.is("error", "http://etherx.jabber.org/streams") =>
|
||||||
{
|
{
|
||||||
let e = "Authentication failure";
|
Err(AuthError::ComponentFail.into())
|
||||||
Err(e.to_owned())
|
|
||||||
},
|
},
|
||||||
Ok(Async::Ready(event)) => {
|
Ok(Async::Ready(event)) => {
|
||||||
println!("ComponentAuth ignore {:?}", event);
|
println!("ComponentAuth ignore {:?}", event);
|
||||||
|
|
@ -84,7 +85,7 @@ impl<S: AsyncRead + AsyncWrite> Future for ComponentAuth<S> {
|
||||||
Ok(Async::NotReady)
|
Ok(Async::NotReady)
|
||||||
},
|
},
|
||||||
Err(e) =>
|
Err(e) =>
|
||||||
Err(format!("{}", e)),
|
Err(e.into()),
|
||||||
},
|
},
|
||||||
ComponentAuthState::Invalid =>
|
ComponentAuthState::Invalid =>
|
||||||
unreachable!(),
|
unreachable!(),
|
||||||
|
|
|
||||||
|
|
@ -3,11 +3,11 @@
|
||||||
//! allowed to use any user and resource identifiers in their stanzas.
|
//! allowed to use any user and resource identifiers in their stanzas.
|
||||||
use std::mem::replace;
|
use std::mem::replace;
|
||||||
use std::str::FromStr;
|
use std::str::FromStr;
|
||||||
use std::error::Error;
|
use std::error::Error as StdError;
|
||||||
use tokio_core::reactor::Handle;
|
use tokio_core::reactor::Handle;
|
||||||
use tokio_core::net::TcpStream;
|
use tokio_core::net::TcpStream;
|
||||||
use tokio_io::{AsyncRead, AsyncWrite};
|
use tokio_io::{AsyncRead, AsyncWrite};
|
||||||
use futures::{Future, Stream, Poll, Async, Sink, StartSend, AsyncSink};
|
use futures::{Future, Stream, Poll, Async, Sink, StartSend, AsyncSink, done};
|
||||||
use minidom::Element;
|
use minidom::Element;
|
||||||
use jid::{Jid, JidParseError};
|
use jid::{Jid, JidParseError};
|
||||||
|
|
||||||
|
|
@ -15,6 +15,7 @@ use super::xmpp_codec::Packet;
|
||||||
use super::xmpp_stream;
|
use super::xmpp_stream;
|
||||||
use super::happy_eyeballs::Connecter;
|
use super::happy_eyeballs::Connecter;
|
||||||
use super::event::Event;
|
use super::event::Event;
|
||||||
|
use super::Error;
|
||||||
|
|
||||||
mod auth;
|
mod auth;
|
||||||
use self::auth::ComponentAuth;
|
use self::auth::ComponentAuth;
|
||||||
|
|
@ -32,7 +33,7 @@ const NS_JABBER_COMPONENT_ACCEPT: &str = "jabber:component:accept";
|
||||||
enum ComponentState {
|
enum ComponentState {
|
||||||
Invalid,
|
Invalid,
|
||||||
Disconnected,
|
Disconnected,
|
||||||
Connecting(Box<Future<Item=XMPPStream, Error=String>>),
|
Connecting(Box<Future<Item=XMPPStream, Error=Error>>),
|
||||||
Connected(XMPPStream),
|
Connected(XMPPStream),
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
@ -47,43 +48,39 @@ impl Component {
|
||||||
let connect = Self::make_connect(jid.clone(), password, server, port, handle);
|
let connect = Self::make_connect(jid.clone(), password, server, port, handle);
|
||||||
Ok(Component {
|
Ok(Component {
|
||||||
jid,
|
jid,
|
||||||
state: ComponentState::Connecting(connect),
|
state: ComponentState::Connecting(Box::new(connect)),
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
fn make_connect(jid: Jid, password: String, server: &str, port: u16, handle: Handle) -> Box<Future<Item=XMPPStream, Error=String>> {
|
fn make_connect(jid: Jid, password: String, server: &str, port: u16, handle: Handle) -> impl Future<Item=XMPPStream, Error=Error> {
|
||||||
let jid1 = jid.clone();
|
let jid1 = jid.clone();
|
||||||
let password = password;
|
let password = password;
|
||||||
Box::new(
|
done(Connecter::from_lookup(handle, server, "_xmpp-component._tcp", port))
|
||||||
Connecter::from_lookup(handle, server, "_xmpp-component._tcp", port)
|
.map_err(Error::Domain)
|
||||||
.expect("Connector::from_lookup")
|
.and_then(|connecter| connecter
|
||||||
.and_then(move |tcp_stream| {
|
.map_err(Error::Connection)
|
||||||
xmpp_stream::XMPPStream::start(tcp_stream, jid1, NS_JABBER_COMPONENT_ACCEPT.to_owned())
|
).and_then(move |tcp_stream| {
|
||||||
.map_err(|e| format!("{}", e))
|
xmpp_stream::XMPPStream::start(tcp_stream, jid1, NS_JABBER_COMPONENT_ACCEPT.to_owned())
|
||||||
}).and_then(move |xmpp_stream| {
|
}).and_then(move |xmpp_stream| {
|
||||||
Self::auth(xmpp_stream, password).expect("auth")
|
Self::auth(xmpp_stream, password).expect("auth")
|
||||||
}).and_then(|xmpp_stream| {
|
})
|
||||||
// println!("Bound to {}", xmpp_stream.jid);
|
|
||||||
Ok(xmpp_stream)
|
|
||||||
})
|
|
||||||
)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
fn auth<S: AsyncRead + AsyncWrite>(stream: xmpp_stream::XMPPStream<S>, password: String) -> Result<ComponentAuth<S>, String> {
|
fn auth<S: AsyncRead + AsyncWrite>(stream: xmpp_stream::XMPPStream<S>, password: String) -> Result<ComponentAuth<S>, Error> {
|
||||||
ComponentAuth::new(stream, password)
|
ComponentAuth::new(stream, password)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Stream for Component {
|
impl Stream for Component {
|
||||||
type Item = Event;
|
type Item = Event;
|
||||||
type Error = String;
|
type Error = Error;
|
||||||
|
|
||||||
fn poll(&mut self) -> Poll<Option<Self::Item>, Self::Error> {
|
fn poll(&mut self) -> Poll<Option<Self::Item>, Self::Error> {
|
||||||
let state = replace(&mut self.state, ComponentState::Invalid);
|
let state = replace(&mut self.state, ComponentState::Invalid);
|
||||||
|
|
||||||
match state {
|
match state {
|
||||||
ComponentState::Invalid =>
|
ComponentState::Invalid =>
|
||||||
Err("invalid client state".to_owned()),
|
Err(Error::InvalidState),
|
||||||
ComponentState::Disconnected =>
|
ComponentState::Disconnected =>
|
||||||
Ok(Async::Ready(None)),
|
Ok(Async::Ready(None)),
|
||||||
ComponentState::Connecting(mut connect) => {
|
ComponentState::Connecting(mut connect) => {
|
||||||
|
|
@ -106,7 +103,7 @@ impl Stream for Component {
|
||||||
Ok(Async::NotReady) => (),
|
Ok(Async::NotReady) => (),
|
||||||
Ok(Async::Ready(())) => (),
|
Ok(Async::Ready(())) => (),
|
||||||
Err(e) =>
|
Err(e) =>
|
||||||
return Err(e.description().to_owned()),
|
return Err(e.into()),
|
||||||
};
|
};
|
||||||
|
|
||||||
// Poll stream
|
// Poll stream
|
||||||
|
|
@ -129,7 +126,7 @@ impl Stream for Component {
|
||||||
Ok(Async::NotReady)
|
Ok(Async::NotReady)
|
||||||
},
|
},
|
||||||
Err(e) =>
|
Err(e) =>
|
||||||
Err(e.description().to_owned()),
|
Err(e.into()),
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|
|
||||||
94
src/error.rs
Normal file
94
src/error.rs
Normal file
|
|
@ -0,0 +1,94 @@
|
||||||
|
use std::io::Error as IoError;
|
||||||
|
use std::error::Error as StdError;
|
||||||
|
use std::str::Utf8Error;
|
||||||
|
use std::borrow::Cow;
|
||||||
|
use std::fmt;
|
||||||
|
use domain::resolv::error::Error as DNSError;
|
||||||
|
use domain::bits::name::FromStrError;
|
||||||
|
use native_tls::Error as TlsError;
|
||||||
|
use xmpp_parsers::error::Error as ParsersError;
|
||||||
|
use xmpp_parsers::sasl::DefinedCondition as SaslDefinedCondition;
|
||||||
|
|
||||||
|
#[derive(Debug, Error)]
|
||||||
|
pub enum Error {
|
||||||
|
Io(IoError),
|
||||||
|
Connection(ConnecterError),
|
||||||
|
/// DNS label conversion error, no details available from module
|
||||||
|
/// `idna`
|
||||||
|
Idna,
|
||||||
|
Domain(FromStrError),
|
||||||
|
Protocol(ProtocolError),
|
||||||
|
Auth(AuthError),
|
||||||
|
Tls(TlsError),
|
||||||
|
/// Shoud never happen
|
||||||
|
InvalidState,
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Causes for stream parsing errors
|
||||||
|
#[derive(Debug, Error)]
|
||||||
|
pub enum ParserError {
|
||||||
|
/// Encoding error
|
||||||
|
Utf8(Utf8Error),
|
||||||
|
/// XML parse error
|
||||||
|
Parse(ParseError),
|
||||||
|
/// Illegal `</>`
|
||||||
|
ShortTag,
|
||||||
|
/// Required by `impl Decoder`
|
||||||
|
IO(IoError),
|
||||||
|
}
|
||||||
|
|
||||||
|
impl From<ParserError> for Error {
|
||||||
|
fn from(e: ParserError) -> Self {
|
||||||
|
ProtocolError::Parser(e).into()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// XML parse error wrapper type
|
||||||
|
#[derive(Debug)]
|
||||||
|
pub struct ParseError(pub Cow<'static, str>);
|
||||||
|
|
||||||
|
impl StdError for ParseError {
|
||||||
|
fn description(&self) -> &str {
|
||||||
|
self.0.as_ref()
|
||||||
|
}
|
||||||
|
fn cause(&self) -> Option<&StdError> {
|
||||||
|
None
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl fmt::Display for ParseError {
|
||||||
|
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
||||||
|
write!(f, "{}", self.0)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Error)]
|
||||||
|
pub enum ProtocolError {
|
||||||
|
Parser(ParserError),
|
||||||
|
#[error(non_std)]
|
||||||
|
Parsers(ParsersError),
|
||||||
|
NoTls,
|
||||||
|
InvalidBindResponse,
|
||||||
|
NoStreamNamespace,
|
||||||
|
NoStreamId,
|
||||||
|
InvalidToken,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Error)]
|
||||||
|
pub enum AuthError {
|
||||||
|
/// No SASL mechanism available
|
||||||
|
NoMechanism,
|
||||||
|
#[error(no_from, non_std, msg_embedded)]
|
||||||
|
Sasl(String),
|
||||||
|
#[error(non_std)]
|
||||||
|
Fail(SaslDefinedCondition),
|
||||||
|
#[error(no_from)]
|
||||||
|
ComponentFail,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Error)]
|
||||||
|
pub enum ConnecterError {
|
||||||
|
NoSrv,
|
||||||
|
AllFailed,
|
||||||
|
DNS(DNSError),
|
||||||
|
}
|
||||||
|
|
@ -6,7 +6,8 @@ use tokio_core::reactor::Handle;
|
||||||
use tokio_core::net::{TcpStream, TcpStreamNew};
|
use tokio_core::net::{TcpStream, TcpStreamNew};
|
||||||
use domain::resolv::Resolver;
|
use domain::resolv::Resolver;
|
||||||
use domain::resolv::lookup::srv::{lookup_srv, LookupSrv, LookupSrvStream};
|
use domain::resolv::lookup::srv::{lookup_srv, LookupSrv, LookupSrvStream};
|
||||||
use domain::bits::DNameBuf;
|
use domain::bits::name::{DNameBuf, FromStrError};
|
||||||
|
use ConnecterError;
|
||||||
|
|
||||||
pub struct Connecter {
|
pub struct Connecter {
|
||||||
handle: Handle,
|
handle: Handle,
|
||||||
|
|
@ -17,11 +18,9 @@ pub struct Connecter {
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Connecter {
|
impl Connecter {
|
||||||
pub fn from_lookup(handle: Handle, domain: &str, srv: &str, fallback_port: u16) -> Result<Connecter, String> {
|
pub fn from_lookup(handle: Handle, domain: &str, srv: &str, fallback_port: u16) -> Result<Connecter, FromStrError> {
|
||||||
let domain = DNameBuf::from_str(domain)
|
let domain = DNameBuf::from_str(domain)?;
|
||||||
.map_err(|e| format!("{}", e))?;
|
let srv = DNameBuf::from_str(srv)?;
|
||||||
let srv = DNameBuf::from_str(srv)
|
|
||||||
.map_err(|e| format!("{}", e))?;
|
|
||||||
|
|
||||||
let resolver = Resolver::new(&handle);
|
let resolver = Resolver::new(&handle);
|
||||||
let lookup = lookup_srv(resolver.clone(), srv, domain, fallback_port);
|
let lookup = lookup_srv(resolver.clone(), srv, domain, fallback_port);
|
||||||
|
|
@ -38,7 +37,7 @@ impl Connecter {
|
||||||
|
|
||||||
impl Future for Connecter {
|
impl Future for Connecter {
|
||||||
type Item = TcpStream;
|
type Item = TcpStream;
|
||||||
type Error = String;
|
type Error = ConnecterError;
|
||||||
|
|
||||||
fn poll(&mut self) -> Poll<Self::Item, Self::Error> {
|
fn poll(&mut self) -> Poll<Self::Item, Self::Error> {
|
||||||
match self.lookup.as_mut().map(|lookup| lookup.poll()) {
|
match self.lookup.as_mut().map(|lookup| lookup.poll()) {
|
||||||
|
|
@ -49,11 +48,11 @@ impl Future for Connecter {
|
||||||
Some(srvs) =>
|
Some(srvs) =>
|
||||||
self.srvs = Some(srvs.to_stream(self.resolver.clone())),
|
self.srvs = Some(srvs.to_stream(self.resolver.clone())),
|
||||||
None =>
|
None =>
|
||||||
return Err("No SRV records".to_owned()),
|
return Err(ConnecterError::NoSrv),
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
Some(Err(e)) =>
|
Some(Err(e)) =>
|
||||||
return Err(format!("{}", e)),
|
return Err(e.into()),
|
||||||
}
|
}
|
||||||
|
|
||||||
match self.srvs.as_mut().map(|srv| srv.poll()) {
|
match self.srvs.as_mut().map(|srv| srv.poll()) {
|
||||||
|
|
@ -71,7 +70,7 @@ impl Future for Connecter {
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
Some(Err(e)) =>
|
Some(Err(e)) =>
|
||||||
return Err(format!("{}", e)),
|
return Err(e.into()),
|
||||||
}
|
}
|
||||||
|
|
||||||
let mut connected_stream = None;
|
let mut connected_stream = None;
|
||||||
|
|
@ -101,7 +100,7 @@ impl Future for Connecter {
|
||||||
self.srvs.is_none() &&
|
self.srvs.is_none() &&
|
||||||
self.connects.is_empty()
|
self.connects.is_empty()
|
||||||
{
|
{
|
||||||
return Err("All connection attempts failed".to_owned());
|
return Err(ConnecterError::AllFailed);
|
||||||
}
|
}
|
||||||
|
|
||||||
Ok(Async::NotReady)
|
Ok(Async::NotReady)
|
||||||
|
|
|
||||||
|
|
@ -1,4 +1,4 @@
|
||||||
#![deny(unsafe_code, unused, missing_docs)]
|
// #![deny(unsafe_code, unused, missing_docs)]
|
||||||
|
|
||||||
//! XMPP implemeentation with asynchronous I/O using Tokio.
|
//! XMPP implemeentation with asynchronous I/O using Tokio.
|
||||||
|
|
||||||
|
|
@ -18,6 +18,8 @@ extern crate domain;
|
||||||
extern crate idna;
|
extern crate idna;
|
||||||
extern crate xmpp_parsers;
|
extern crate xmpp_parsers;
|
||||||
extern crate try_from;
|
extern crate try_from;
|
||||||
|
#[macro_use]
|
||||||
|
extern crate derive_error;
|
||||||
|
|
||||||
pub mod xmpp_codec;
|
pub mod xmpp_codec;
|
||||||
pub mod xmpp_stream;
|
pub mod xmpp_stream;
|
||||||
|
|
@ -31,3 +33,5 @@ mod client;
|
||||||
pub use client::Client;
|
pub use client::Client;
|
||||||
mod component;
|
mod component;
|
||||||
pub use component::Component;
|
pub use component::Component;
|
||||||
|
mod error;
|
||||||
|
pub use error::{Error, ProtocolError, AuthError, ConnecterError, ParseError, ParserError};
|
||||||
|
|
|
||||||
|
|
@ -10,6 +10,7 @@ use jid::Jid;
|
||||||
|
|
||||||
use xmpp_codec::Packet;
|
use xmpp_codec::Packet;
|
||||||
use xmpp_stream::XMPPStream;
|
use xmpp_stream::XMPPStream;
|
||||||
|
use Error;
|
||||||
|
|
||||||
/// XMPP TLS XML namespace
|
/// XMPP TLS XML namespace
|
||||||
pub const NS_XMPP_TLS: &str = "urn:ietf:params:xml:ns:xmpp-tls";
|
pub const NS_XMPP_TLS: &str = "urn:ietf:params:xml:ns:xmpp-tls";
|
||||||
|
|
@ -48,7 +49,7 @@ impl<S: AsyncRead + AsyncWrite> StartTlsClient<S> {
|
||||||
|
|
||||||
impl<S: AsyncRead + AsyncWrite> Future for StartTlsClient<S> {
|
impl<S: AsyncRead + AsyncWrite> Future for StartTlsClient<S> {
|
||||||
type Item = TlsStream<S>;
|
type Item = TlsStream<S>;
|
||||||
type Error = String;
|
type Error = Error;
|
||||||
|
|
||||||
fn poll(&mut self) -> Poll<Self::Item, Self::Error> {
|
fn poll(&mut self) -> Poll<Self::Item, Self::Error> {
|
||||||
let old_state = replace(&mut self.state, StartTlsClientState::Invalid);
|
let old_state = replace(&mut self.state, StartTlsClientState::Invalid);
|
||||||
|
|
@ -65,7 +66,7 @@ impl<S: AsyncRead + AsyncWrite> Future for StartTlsClient<S> {
|
||||||
Ok(Async::NotReady) =>
|
Ok(Async::NotReady) =>
|
||||||
(StartTlsClientState::SendStartTls(send), Ok(Async::NotReady)),
|
(StartTlsClientState::SendStartTls(send), Ok(Async::NotReady)),
|
||||||
Err(e) =>
|
Err(e) =>
|
||||||
(StartTlsClientState::SendStartTls(send), Err(format!("{}", e))),
|
(StartTlsClientState::SendStartTls(send), Err(e.into())),
|
||||||
},
|
},
|
||||||
StartTlsClientState::AwaitProceed(mut xmpp_stream) =>
|
StartTlsClientState::AwaitProceed(mut xmpp_stream) =>
|
||||||
match xmpp_stream.poll() {
|
match xmpp_stream.poll() {
|
||||||
|
|
@ -87,7 +88,7 @@ impl<S: AsyncRead + AsyncWrite> Future for StartTlsClient<S> {
|
||||||
Ok(_) =>
|
Ok(_) =>
|
||||||
(StartTlsClientState::AwaitProceed(xmpp_stream), Ok(Async::NotReady)),
|
(StartTlsClientState::AwaitProceed(xmpp_stream), Ok(Async::NotReady)),
|
||||||
Err(e) =>
|
Err(e) =>
|
||||||
(StartTlsClientState::AwaitProceed(xmpp_stream), Err(format!("{}", e))),
|
(StartTlsClientState::AwaitProceed(xmpp_stream), Err(Error::Protocol(e.into()))),
|
||||||
},
|
},
|
||||||
StartTlsClientState::StartingTls(mut connect) =>
|
StartTlsClientState::StartingTls(mut connect) =>
|
||||||
match connect.poll() {
|
match connect.poll() {
|
||||||
|
|
@ -96,7 +97,7 @@ impl<S: AsyncRead + AsyncWrite> Future for StartTlsClient<S> {
|
||||||
Ok(Async::NotReady) =>
|
Ok(Async::NotReady) =>
|
||||||
(StartTlsClientState::StartingTls(connect), Ok(Async::NotReady)),
|
(StartTlsClientState::StartingTls(connect), Ok(Async::NotReady)),
|
||||||
Err(e) =>
|
Err(e) =>
|
||||||
(StartTlsClientState::Invalid, Err(format!("{}", e))),
|
(StartTlsClientState::Invalid, Err(e.into())),
|
||||||
},
|
},
|
||||||
StartTlsClientState::Invalid =>
|
StartTlsClientState::Invalid =>
|
||||||
unreachable!(),
|
unreachable!(),
|
||||||
|
|
|
||||||
|
|
@ -1,13 +1,15 @@
|
||||||
use std::mem::replace;
|
use std::mem::replace;
|
||||||
use std::borrow::Cow;
|
// use std::error::Error as StdError;
|
||||||
|
// use std::{fmt, io};
|
||||||
use futures::{Future, Async, Poll, Stream, sink, Sink};
|
use futures::{Future, Async, Poll, Stream, sink, Sink};
|
||||||
use tokio_io::{AsyncRead, AsyncWrite};
|
use tokio_io::{AsyncRead, AsyncWrite};
|
||||||
use tokio_codec::Framed;
|
use tokio_codec::Framed;
|
||||||
use jid::Jid;
|
use jid::Jid;
|
||||||
use minidom::Element;
|
use minidom::Element;
|
||||||
|
|
||||||
use xmpp_codec::{XMPPCodec, Packet, ParserError};
|
use xmpp_codec::{XMPPCodec, Packet};
|
||||||
use xmpp_stream::XMPPStream;
|
use xmpp_stream::XMPPStream;
|
||||||
|
use {Error, ProtocolError};
|
||||||
|
|
||||||
const NS_XMPP_STREAM: &str = "http://etherx.jabber.org/streams";
|
const NS_XMPP_STREAM: &str = "http://etherx.jabber.org/streams";
|
||||||
|
|
||||||
|
|
@ -43,7 +45,7 @@ impl<S: AsyncWrite> StreamStart<S> {
|
||||||
|
|
||||||
impl<S: AsyncRead + AsyncWrite> Future for StreamStart<S> {
|
impl<S: AsyncRead + AsyncWrite> Future for StreamStart<S> {
|
||||||
type Item = XMPPStream<S>;
|
type Item = XMPPStream<S>;
|
||||||
type Error = ParserError;
|
type Error = Error;
|
||||||
|
|
||||||
fn poll(&mut self) -> Poll<Self::Item, Self::Error> {
|
fn poll(&mut self) -> Poll<Self::Item, Self::Error> {
|
||||||
let old_state = replace(&mut self.state, StreamStartState::Invalid);
|
let old_state = replace(&mut self.state, StreamStartState::Invalid);
|
||||||
|
|
@ -67,7 +69,7 @@ impl<S: AsyncRead + AsyncWrite> Future for StreamStart<S> {
|
||||||
let stream_ns = match stream_attrs.get("xmlns") {
|
let stream_ns = match stream_attrs.get("xmlns") {
|
||||||
Some(ns) => ns.clone(),
|
Some(ns) => ns.clone(),
|
||||||
None =>
|
None =>
|
||||||
return Err(ParserError::Parse(Cow::from("Missing stream namespace"))),
|
return Err(ProtocolError::NoStreamNamespace.into()),
|
||||||
};
|
};
|
||||||
if self.ns == "jabber:client" {
|
if self.ns == "jabber:client" {
|
||||||
retry = true;
|
retry = true;
|
||||||
|
|
@ -77,7 +79,7 @@ impl<S: AsyncRead + AsyncWrite> Future for StreamStart<S> {
|
||||||
let id = match stream_attrs.get("id") {
|
let id = match stream_attrs.get("id") {
|
||||||
Some(id) => id.clone(),
|
Some(id) => id.clone(),
|
||||||
None =>
|
None =>
|
||||||
return Err(ParserError::Parse(Cow::from("No stream id"))),
|
return Err(ProtocolError::NoStreamId.into()),
|
||||||
};
|
};
|
||||||
// FIXME: huge hack, shouldn’t be an element!
|
// FIXME: huge hack, shouldn’t be an element!
|
||||||
let stream = XMPPStream::new(self.jid.clone(), stream, self.ns.clone(), Element::builder(id).build());
|
let stream = XMPPStream::new(self.jid.clone(), stream, self.ns.clone(), Element::builder(id).build());
|
||||||
|
|
@ -85,11 +87,11 @@ impl<S: AsyncRead + AsyncWrite> Future for StreamStart<S> {
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
Ok(Async::Ready(_)) =>
|
Ok(Async::Ready(_)) =>
|
||||||
return Err(ParserError::Parse(Cow::from("Invalid XML event received"))),
|
return Err(ProtocolError::InvalidToken.into()),
|
||||||
Ok(Async::NotReady) =>
|
Ok(Async::NotReady) =>
|
||||||
(StreamStartState::RecvStart(stream), Ok(Async::NotReady)),
|
(StreamStartState::RecvStart(stream), Ok(Async::NotReady)),
|
||||||
Err(e) =>
|
Err(e) =>
|
||||||
return Err(e),
|
return Err(ProtocolError::from(e).into()),
|
||||||
},
|
},
|
||||||
StreamStartState::RecvFeatures(mut stream, stream_ns) =>
|
StreamStartState::RecvFeatures(mut stream, stream_ns) =>
|
||||||
match stream.poll() {
|
match stream.poll() {
|
||||||
|
|
@ -103,7 +105,7 @@ impl<S: AsyncRead + AsyncWrite> Future for StreamStart<S> {
|
||||||
Ok(Async::Ready(_)) | Ok(Async::NotReady) =>
|
Ok(Async::Ready(_)) | Ok(Async::NotReady) =>
|
||||||
(StreamStartState::RecvFeatures(stream, stream_ns), Ok(Async::NotReady)),
|
(StreamStartState::RecvFeatures(stream, stream_ns), Ok(Async::NotReady)),
|
||||||
Err(e) =>
|
Err(e) =>
|
||||||
return Err(e),
|
return Err(ProtocolError::from(e).into()),
|
||||||
},
|
},
|
||||||
StreamStartState::Invalid =>
|
StreamStartState::Invalid =>
|
||||||
unreachable!(),
|
unreachable!(),
|
||||||
|
|
@ -117,3 +119,59 @@ impl<S: AsyncRead + AsyncWrite> Future for StreamStart<S> {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// #[derive(Debug)]
|
||||||
|
// pub enum StreamStartError {
|
||||||
|
// MissingStreamNs,
|
||||||
|
// MissingStreamId,
|
||||||
|
// Unexpected,
|
||||||
|
// Parser(ParserError),
|
||||||
|
// IO(io::Error),
|
||||||
|
// }
|
||||||
|
|
||||||
|
// impl From<io::Error> for StreamStartError {
|
||||||
|
// fn from(e: io::Error) -> Self {
|
||||||
|
// StreamStartError::IO(e)
|
||||||
|
// }
|
||||||
|
// }
|
||||||
|
|
||||||
|
// impl From<ParserError> for StreamStartError {
|
||||||
|
// fn from(e: ParserError) -> Self {
|
||||||
|
// match e {
|
||||||
|
// ParserError::IO(e) => StreamStartError::IO(e),
|
||||||
|
// _ => StreamStartError::Parser(e)
|
||||||
|
// }
|
||||||
|
// }
|
||||||
|
// }
|
||||||
|
|
||||||
|
// impl StdError for StreamStartError {
|
||||||
|
// fn description(&self) -> &str {
|
||||||
|
// match *self {
|
||||||
|
// StreamStartError::MissingStreamNs => "Missing stream namespace",
|
||||||
|
// StreamStartError::MissingStreamId => "Missing stream id",
|
||||||
|
// StreamStartError::Unexpected => "Unexpected",
|
||||||
|
// StreamStartError::Parser(ref pe) => pe.description(),
|
||||||
|
// StreamStartError::IO(ref ie) => ie.description(),
|
||||||
|
// }
|
||||||
|
// }
|
||||||
|
|
||||||
|
// fn cause(&self) -> Option<&StdError> {
|
||||||
|
// match *self {
|
||||||
|
// StreamStartError::Parser(ref pe) => pe.cause(),
|
||||||
|
// StreamStartError::IO(ref ie) => ie.cause(),
|
||||||
|
// _ => None,
|
||||||
|
// }
|
||||||
|
// }
|
||||||
|
// }
|
||||||
|
|
||||||
|
// impl fmt::Display for StreamStartError {
|
||||||
|
// fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
||||||
|
// match *self {
|
||||||
|
// StreamStartError::MissingStreamNs => write!(f, "Missing stream namespace"),
|
||||||
|
// StreamStartError::MissingStreamId => write!(f, "Missing stream id"),
|
||||||
|
// StreamStartError::Unexpected => write!(f, "Received unexpected data"),
|
||||||
|
// StreamStartError::Parser(ref pe) => write!(f, "{}", pe),
|
||||||
|
// StreamStartError::IO(ref ie) => write!(f, "{}", ie),
|
||||||
|
// }
|
||||||
|
// }
|
||||||
|
// }
|
||||||
|
|
|
||||||
|
|
@ -6,19 +6,17 @@ use std::iter::FromIterator;
|
||||||
use std::cell::RefCell;
|
use std::cell::RefCell;
|
||||||
use std::rc::Rc;
|
use std::rc::Rc;
|
||||||
use std::fmt::Write;
|
use std::fmt::Write;
|
||||||
use std::str::{from_utf8, Utf8Error};
|
use std::str::from_utf8;
|
||||||
use std::io;
|
use std::io;
|
||||||
use std::collections::HashMap;
|
use std::collections::HashMap;
|
||||||
use std::collections::vec_deque::VecDeque;
|
use std::collections::vec_deque::VecDeque;
|
||||||
use std::error::Error as StdError;
|
|
||||||
use std::fmt;
|
|
||||||
use std::borrow::Cow;
|
|
||||||
use tokio_codec::{Encoder, Decoder};
|
use tokio_codec::{Encoder, Decoder};
|
||||||
use minidom::Element;
|
use minidom::Element;
|
||||||
use xml5ever::tokenizer::{XmlTokenizer, TokenSink, Token, Tag, TagKind};
|
use xml5ever::tokenizer::{XmlTokenizer, TokenSink, Token, Tag, TagKind};
|
||||||
use xml5ever::interface::Attribute;
|
use xml5ever::interface::Attribute;
|
||||||
use bytes::{BytesMut, BufMut};
|
use bytes::{BytesMut, BufMut};
|
||||||
use quick_xml::Writer as EventWriter;
|
use quick_xml::Writer as EventWriter;
|
||||||
|
use {ParserError, ParseError};
|
||||||
|
|
||||||
/// Anything that can be sent or received on an XMPP/XML stream
|
/// Anything that can be sent or received on an XMPP/XML stream
|
||||||
#[derive(Debug)]
|
#[derive(Debug)]
|
||||||
|
|
@ -33,55 +31,6 @@ pub enum Packet {
|
||||||
StreamEnd,
|
StreamEnd,
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Causes for stream parsing errors
|
|
||||||
#[derive(Debug)]
|
|
||||||
pub enum ParserError {
|
|
||||||
/// Encoding error
|
|
||||||
Utf8(Utf8Error),
|
|
||||||
/// XML parse error
|
|
||||||
Parse(Cow<'static, str>),
|
|
||||||
/// Illegal `</>`
|
|
||||||
ShortTag,
|
|
||||||
/// Required by `impl Decoder`
|
|
||||||
IO(io::Error),
|
|
||||||
}
|
|
||||||
|
|
||||||
impl From<io::Error> for ParserError {
|
|
||||||
fn from(e: io::Error) -> Self {
|
|
||||||
ParserError::IO(e)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl StdError for ParserError {
|
|
||||||
fn description(&self) -> &str {
|
|
||||||
match *self {
|
|
||||||
ParserError::Utf8(ref ue) => ue.description(),
|
|
||||||
ParserError::Parse(ref pe) => pe,
|
|
||||||
ParserError::ShortTag => "short tag",
|
|
||||||
ParserError::IO(ref ie) => ie.description(),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn cause(&self) -> Option<&StdError> {
|
|
||||||
match *self {
|
|
||||||
ParserError::Utf8(ref ue) => ue.cause(),
|
|
||||||
ParserError::IO(ref ie) => ie.cause(),
|
|
||||||
_ => None,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl fmt::Display for ParserError {
|
|
||||||
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
|
||||||
match *self {
|
|
||||||
ParserError::Utf8(ref ue) => write!(f, "{}", ue),
|
|
||||||
ParserError::Parse(ref pe) => write!(f, "{}", pe),
|
|
||||||
ParserError::ShortTag => write!(f, "Short tag"),
|
|
||||||
ParserError::IO(ref ie) => write!(f, "{}", ie),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
type QueueItem = Result<Packet, ParserError>;
|
type QueueItem = Result<Packet, ParserError>;
|
||||||
|
|
||||||
/// Parser state
|
/// Parser state
|
||||||
|
|
@ -220,7 +169,7 @@ impl TokenSink for ParserSink {
|
||||||
self.push_queue(Packet::StreamEnd),
|
self.push_queue(Packet::StreamEnd),
|
||||||
Token::ParseError(s) => {
|
Token::ParseError(s) => {
|
||||||
// println!("ParseError: {:?}", s);
|
// println!("ParseError: {:?}", s);
|
||||||
self.push_queue_error(ParserError::Parse(s));
|
self.push_queue_error(ParserError::Parse(ParseError(s)));
|
||||||
},
|
},
|
||||||
_ => (),
|
_ => (),
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue