49 lines
1.7 KiB
Rust
49 lines
1.7 KiB
Rust
use futures::stream::StreamExt;
|
|
use std::convert::TryFrom;
|
|
use std::marker::Unpin;
|
|
use tokio::io::{AsyncRead, AsyncWrite};
|
|
use xmpp_parsers::bind::{BindQuery, BindResponse};
|
|
use xmpp_parsers::iq::{Iq, IqType};
|
|
|
|
use crate::xmpp_codec::Packet;
|
|
use crate::xmpp_stream::XMPPStream;
|
|
use crate::{Error, ProtocolError};
|
|
|
|
const BIND_REQ_ID: &str = "resource-bind";
|
|
|
|
pub async fn bind<S: AsyncRead + AsyncWrite + Unpin>(
|
|
mut stream: XMPPStream<S>,
|
|
) -> Result<XMPPStream<S>, Error> {
|
|
if stream.stream_features.can_bind() {
|
|
let resource = stream
|
|
.jid
|
|
.resource()
|
|
.and_then(|resource| Some(resource.to_owned()));
|
|
let iq = Iq::from_set(BIND_REQ_ID, BindQuery::new(resource));
|
|
stream.send_stanza(iq).await?;
|
|
|
|
loop {
|
|
match stream.next().await {
|
|
Some(Ok(Packet::Stanza(stanza))) => match Iq::try_from(stanza) {
|
|
Ok(iq) if iq.id == BIND_REQ_ID => match iq.payload {
|
|
IqType::Result(payload) => {
|
|
payload
|
|
.and_then(|payload| BindResponse::try_from(payload).ok())
|
|
.map(|bind| stream.jid = bind.into());
|
|
return Ok(stream);
|
|
}
|
|
_ => return Err(ProtocolError::InvalidBindResponse.into()),
|
|
},
|
|
_ => {}
|
|
},
|
|
Some(Ok(_)) => {}
|
|
Some(Err(e)) => return Err(e),
|
|
None => return Err(Error::Disconnected),
|
|
}
|
|
}
|
|
} else {
|
|
// No resource binding available,
|
|
// return the (probably // usable) stream immediately
|
|
return Ok(stream);
|
|
}
|
|
}
|