xmpp-rs/tokio-xmpp/src/stream_start.rs

76 lines
2.3 KiB
Rust
Raw Normal View History

2020-03-05 00:25:24 +00:00
use futures::{sink::SinkExt, stream::StreamExt};
use std::marker::Unpin;
use tokio::io::{AsyncRead, AsyncWrite};
use tokio_util::codec::Framed;
use xmpp_parsers::{ns, Element, Jid};
2018-12-18 18:04:31 +00:00
use crate::xmpp_codec::{Packet, XMPPCodec};
2018-12-18 17:29:31 +00:00
use crate::xmpp_stream::XMPPStream;
use crate::{Error, ProtocolError};
2020-03-15 23:34:46 +00:00
/// Sends a `<stream:stream>`, then wait for one from the server, and
/// construct an XMPPStream.
2020-03-05 00:25:24 +00:00
pub async fn start<S: AsyncRead + AsyncWrite + Unpin>(
mut stream: Framed<S, XMPPCodec>,
2017-06-13 23:55:56 +00:00
jid: Jid,
2017-07-18 23:02:45 +00:00
ns: String,
2020-03-05 00:25:24 +00:00
) -> Result<XMPPStream<S>, Error> {
let attrs = [
("to".to_owned(), jid.clone().domain()),
("version".to_owned(), "1.0".to_owned()),
("xmlns".to_owned(), ns.clone()),
("xmlns:stream".to_owned(), ns::STREAM.to_owned()),
2020-03-05 00:25:24 +00:00
]
.iter()
.cloned()
.collect();
stream.send(Packet::StreamStart(attrs)).await?;
2020-03-05 00:25:24 +00:00
let stream_attrs;
loop {
match stream.next().await {
Some(Ok(Packet::StreamStart(attrs))) => {
stream_attrs = attrs;
break;
}
Some(Ok(_)) => {}
Some(Err(e)) => return Err(e.into()),
None => return Err(Error::Disconnected),
}
}
2020-03-05 00:25:24 +00:00
let stream_ns = stream_attrs
.get("xmlns")
.ok_or(ProtocolError::NoStreamNamespace)?
.clone();
let stream_id = stream_attrs
.get("id")
.ok_or(ProtocolError::NoStreamId)?
.clone();
let stream = if stream_ns == "jabber:client" && stream_attrs.get("version").is_some() {
let stream_features;
loop {
match stream.next().await {
Some(Ok(Packet::Stanza(stanza))) if stanza.is("features", ns::STREAM) => {
2020-03-05 00:25:24 +00:00
stream_features = stanza;
break;
2018-12-18 18:04:31 +00:00
}
2020-03-05 00:25:24 +00:00
Some(Ok(_)) => {}
Some(Err(e)) => return Err(e.into()),
None => return Err(Error::Disconnected),
}
}
2020-03-05 00:25:24 +00:00
XMPPStream::new(jid, stream, ns, stream_id, stream_features)
} else {
// FIXME: huge hack, shouldnt be an element!
XMPPStream::new(
jid,
stream,
ns,
stream_id.clone(),
Element::builder(stream_id, ns::STREAM).build(),
2020-03-05 00:25:24 +00:00
)
};
Ok(stream)
}