83 lines
2.4 KiB
Rust
83 lines
2.4 KiB
Rust
// Copyright (c) 2024 Jonas Schäfer <jonas@zombofant.net>
|
|
//
|
|
// This Source Code Form is subject to the terms of the Mozilla Public
|
|
// License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
|
|
//! Keep a connection alive
|
|
//!
|
|
//! This example demonstrates that tokio_xmpp will keep a connection alive
|
|
//! as good as it can, transparently reconnecting on interruptions of the TCP
|
|
//! stream.
|
|
|
|
use core::str::FromStr;
|
|
use core::time::Duration;
|
|
use std::env::args;
|
|
use std::process::exit;
|
|
|
|
use rand::{thread_rng, Rng};
|
|
|
|
use futures::StreamExt;
|
|
|
|
use tokio_xmpp::{
|
|
connect::{DnsConfig, StartTlsServerConnector},
|
|
parsers::{
|
|
iq::Iq,
|
|
jid::{BareJid, Jid},
|
|
ping,
|
|
},
|
|
stanzastream::StanzaStream,
|
|
xmlstream::Timeouts,
|
|
};
|
|
|
|
#[tokio::main]
|
|
async fn main() {
|
|
env_logger::init();
|
|
|
|
let args: Vec<String> = args().collect();
|
|
if args.len() != 3 {
|
|
println!("Usage: {} <jid> <password>", args[0]);
|
|
exit(1);
|
|
}
|
|
let jid = BareJid::from_str(&args[1]).expect(&format!("Invalid JID: {}", &args[1]));
|
|
let password = &args[2];
|
|
|
|
let mut timeouts = Timeouts::tight();
|
|
timeouts.read_timeout = Duration::new(5, 0);
|
|
|
|
let mut stream = StanzaStream::new_c2s(
|
|
StartTlsServerConnector::from(DnsConfig::UseSrv {
|
|
host: jid.domain().as_str().to_owned(),
|
|
srv: "_xmpp-client._tcp".to_owned(),
|
|
fallback_port: 5222,
|
|
}),
|
|
jid.clone().into(),
|
|
password.clone(),
|
|
timeouts,
|
|
16,
|
|
);
|
|
let domain: Jid = jid.domain().to_owned().into();
|
|
let mut ping_timer = tokio::time::interval(Duration::new(5, 0));
|
|
let mut ping_ctr: u64 = thread_rng().gen();
|
|
let signal = tokio::signal::ctrl_c();
|
|
tokio::pin!(signal);
|
|
loop {
|
|
tokio::select! {
|
|
_ = &mut signal => {
|
|
log::info!("Ctrl+C pressed, shutting down cleanly.");
|
|
break;
|
|
}
|
|
_ = ping_timer.tick() => {
|
|
log::info!("sending ping for fun & profit");
|
|
ping_ctr = ping_ctr.wrapping_add(1);
|
|
let mut iq = Iq::from_get(format!("ping-{}", ping_ctr), ping::Ping);
|
|
iq.to = Some(domain.clone());
|
|
stream.send(Box::new(iq.into())).await;
|
|
}
|
|
ev = stream.next() => {
|
|
log::info!("{:?}", ev);
|
|
}
|
|
}
|
|
}
|
|
stream.close().await;
|
|
}
|