geph4-client 4.4.12

Geph client
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48

use std::time::{Duration, SystemTime};

use crate::activity::{self};
use activity::wait_activity;
use anyhow::Context;
use async_net::SocketAddr;
use smol::prelude::*;
use sosistab::Multiplex;

use super::{getsess::get_session, tunnelctx::TunnelCtx};

pub async fn rerouter_loop(
    ctx: TunnelCtx,
    bridge_addr: SocketAddr,
    tunnel_mux: &Multiplex,
) -> anyhow::Result<()> {
    let mut old_addr = bridge_addr;
    // We first request the ID of the other multiplex.
    let other_id = {
        let mut conn = tunnel_mux.open_conn(Some("!id".into())).await?;
        let mut buf = [0u8; 32];
        conn.read_exact(&mut buf).await.context("!id failed")?;
        buf
    };
    loop {
        let start = SystemTime::now();
        wait_activity(Duration::from_secs(300)).await;
        log::debug!("rerouter called after interval of {:?}", start.elapsed());
        let new_sess = get_session(ctx.clone(), Some(old_addr)).await;
        match new_sess {
            Ok(new_sess) => {
                if new_sess.remote_addr() == old_addr {
                    log::trace!("skipping hijack because the best connection was identical")
                } else {
                    old_addr = new_sess.remote_addr();
                    new_sess
                        .hijack(tunnel_mux, other_id)
                        .await
                        .context("hijack failed")?;
                }
            }
            Err(err) => {
                anyhow::bail!("rerouter failed to make new sess: {:?}", err);
            }
        }
        smol::Timer::after(Duration::from_secs(120)).await;
    }
}