alterwain/Frida
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Changes to be committed:

Browse Source 
modified:   src/server.rs
This commit is contained in:
Michael Wain 2024-08-23 04:24:27 +03:00
parent 3c85631626
commit 58eb0c9b59
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/server.rs
View File

@ -1,9 +1,7 @@
use crossbeam_channel::unbounded; use crossbeam_channel::unbounded;
use tokio::{net::UdpSocket, sync::Mutex}; use tokio::{net::UdpSocket, sync::Mutex};
use x25519_dalek::{PublicKey, StaticSecret}; use x25519_dalek::{PublicKey, StaticSecret};
use core::time;
use std::io::{Read, Write}; use std::io::{Read, Write};
use std::thread;
use base64::prelude::*; use base64::prelude::*;
use log::{error, info}; use log::{error, info};
use std::sync::Arc; use std::sync::Arc;
@ -51,8 +49,11 @@ pub async fn server_mode(server_config: ServerConfiguration) {
let addrs_lcl = addresses.clone(); let addrs_lcl = addresses.clone();
if keepalive_sec > 0 { if keepalive_sec > 0 {
tokio::spawn(async move { tokio::spawn(async move {
let mut now = std::time::Instant::now();
let kps = std::time::Duration::from_secs(keepalive_sec.clone().into());
loop { loop {
thread::sleep(time::Duration::from_secs(keepalive_sec.into())); if now.elapsed() < kps { continue; }
now = std::time::Instant::now();
let mmp = addrs_lcl.lock().await; let mmp = addrs_lcl.lock().await;
mmp.values().for_each(|p| { mmp.values().for_each(|p| {
let _ = send2hnd_cl.send((UDPKeepAlive{}.serialize(), p.addr)); let _ = send2hnd_cl.send((UDPKeepAlive{}.serialize(), p.addr));