add heartbeat destination broadcasting to transport node #2
3 changed files with 35 additions and 10 deletions
3
Cargo.lock
generated
3
Cargo.lock
generated
|
|
@ -1035,8 +1035,7 @@ checksum = "7a2d987857b319362043e95f5353c0535c1f58eec5336fdfcf626430af7def58"
|
|||
[[package]]
|
||||
name = "reticulum"
|
||||
version = "0.1.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "d838809ab28f9bfa5354b3397bcfc2b4a7a7258e98d9ed2b18a6e39437b00202"
|
||||
source = "git+https://github.com/BeechatNetworkSystemsLtd/Reticulum-rs.git#5012cf2c48044034775207ab7ee2c67d285276bb"
|
||||
dependencies = [
|
||||
"aes",
|
||||
"cbc",
|
||||
|
|
|
|||
|
|
@ -15,7 +15,8 @@ tokio-util = "0.7.15"
|
|||
# Logging
|
||||
log = "0.4.27"
|
||||
env_logger = "0.10"
|
||||
reticulum = "0.1.0"
|
||||
reticulum = { git = "https://github.com/BeechatNetworkSystemsLtd/Reticulum-rs.git" }
|
||||
# reticulum = "0.1.0"
|
||||
clap = { version = "4.5.53", features = ["derive"] }
|
||||
serde = { version = "1.0.228", features = ["derive"] }
|
||||
bincode = { version = "2.0.1", features = ["serde"] }
|
||||
|
|
|
|||
39
src/main.rs
39
src/main.rs
|
|
@ -1,3 +1,6 @@
|
|||
use rand_core::OsRng;
|
||||
use reticulum::destination::DestinationName;
|
||||
use reticulum::identity::PrivateIdentity;
|
||||
use tokio::fs;
|
||||
|
||||
use clap::Parser;
|
||||
|
|
@ -8,6 +11,7 @@ use reticulum::iface::tcp_client::TcpClient;
|
|||
use reticulum::iface::tcp_server::TcpServer;
|
||||
use reticulum::iface::udp::UdpInterface;
|
||||
use reticulum::transport::{Transport, TransportConfig};
|
||||
use tokio::time::{Duration, interval};
|
||||
|
||||
#[derive(Serialize, Deserialize, Debug, Clone)]
|
||||
struct TcpServerConfig {
|
||||
|
|
@ -35,6 +39,8 @@ enum InterfaceConfig {
|
|||
#[derive(Serialize, Deserialize, Debug, Clone)]
|
||||
struct Config {
|
||||
interfaces: Vec<InterfaceConfig>,
|
||||
broadcast: bool,
|
||||
retransmit: bool,
|
||||
}
|
||||
|
||||
#[derive(Parser)]
|
||||
|
|
@ -45,7 +51,7 @@ struct Cli {
|
|||
|
||||
#[tokio::main]
|
||||
async fn main() {
|
||||
env_logger::Builder::from_env(env_logger::Env::default().default_filter_or("trace")).init();
|
||||
env_logger::Builder::from_env(env_logger::Env::default().default_filter_or("info")).init();
|
||||
|
||||
log::info!(">>> TRANSPORT NODE <<<");
|
||||
|
||||
|
|
@ -57,10 +63,18 @@ async fn main() {
|
|||
println!("{config:?}");
|
||||
|
||||
let mut transport_config = TransportConfig::default();
|
||||
transport_config.set_retransmit(false);
|
||||
transport_config.set_broadcast(true);
|
||||
transport_config.set_retransmit(config.retransmit);
|
||||
transport_config.set_broadcast(config.broadcast);
|
||||
// set up the reticulum transport
|
||||
let transport = Transport::new(transport_config);
|
||||
let mut transport = Transport::new(transport_config);
|
||||
|
||||
// set up destination
|
||||
let id = PrivateIdentity::new_from_rand(OsRng);
|
||||
|
||||
let hb_destination = transport
|
||||
.add_destination(id, DestinationName::new("transport node", "heartbeat"))
|
||||
.await;
|
||||
log::info!("dest ID: {}", hb_destination.lock().await.desc.address_hash);
|
||||
|
||||
// set up interfaces
|
||||
for iface_config in config.interfaces {
|
||||
|
|
@ -87,8 +101,19 @@ async fn main() {
|
|||
}
|
||||
}
|
||||
|
||||
// run until exit
|
||||
let _ = tokio::signal::ctrl_c().await;
|
||||
let mut heartbeat_interval = interval(Duration::from_secs(5));
|
||||
|
||||
log::info!("exit");
|
||||
loop {
|
||||
tokio::select! {
|
||||
_ = tokio::signal::ctrl_c() => {
|
||||
break;
|
||||
}
|
||||
_ = heartbeat_interval.tick() => {
|
||||
log::trace!("send heartbeat");
|
||||
transport.send_announce(&hb_destination, None).await;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
log::info!("exit!");
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue