guess port at -10 to 10

This commit is contained in:
asxalex 2024-11-13 19:24:33 +08:00
parent 966f054a63
commit b51c78bdfb
5 changed files with 43 additions and 18 deletions

2
.cargo/config.toml Normal file
View File

@ -0,0 +1,2 @@
[target.x86_64-unknown-linux-musl]
linker = "x86_64-linux-musl-gcc"

5
.gitignore vendored
View File

@ -4,4 +4,7 @@
.client .client
.output .output
sdlan sdlan
sdlan.exe sdlan.exe
*.bak
*.a
*.o

View File

@ -1,6 +1,9 @@
linux: self:
RUSTFLAGS="-L ." cargo build --features "tap" --release RUSTFLAGS="-L ." cargo build --features "tap" --release
linux:
RUSTFLAGS="-L ." cargo build --features "tap" --release --target x86_64-unknown-linux-musl
linux-tun: linux-tun:
RUSTFLAGS="-L ." cargo build --release RUSTFLAGS="-L ." cargo build --release
@ -12,4 +15,7 @@ pb:
mv src/pb/_.rs src/pb/message.rs mv src/pb/_.rs src/pb/message.rs
libtun-so: libtun-so:
cd src/network && gcc -fPIC -shared -o libtuntap.so tuntap.c && cd - cd src/network && gcc -fPIC -shared -o libtuntap.so tuntap.c && cd -
libtun-a:
cd src/network && gcc -c tuntap.c && ar rcs libtuntap.a tuntap.o && cd -

View File

@ -189,7 +189,7 @@ async fn handle_tcp_message(msg: SdlanTcp) {
async fn handle_tcp_command(_edge: &Node, _cmdtype: u8, _cmdprotobuf: &[u8]) {} async fn handle_tcp_command(_edge: &Node, _cmdtype: u8, _cmdprotobuf: &[u8]) {}
async fn handle_tcp_event(edge: &Node, eventtype: EventType, eventprotobuf: &[u8]) { async fn handle_tcp_event(edge: &'static Node, eventtype: EventType, eventprotobuf: &[u8]) {
match eventtype { match eventtype {
EventType::SendRegister => { EventType::SendRegister => {
let Ok(reg) = SdlSendRegisterEvent::decode(eventprotobuf) else { let Ok(reg) = SdlSendRegisterEvent::decode(eventprotobuf) else {
@ -459,7 +459,7 @@ async fn send_stun_request(eee: &Node) {
} }
} }
pub async fn loop_socket_v6(eee: &Node, socket: Arc<Socket>, cancel: CancellationToken) { pub async fn loop_socket_v6(eee: &'static Node, socket: Arc<Socket>, cancel: CancellationToken) {
debug!("loop sock v6"); debug!("loop sock v6");
loop { loop {
tokio::select! { tokio::select! {
@ -487,7 +487,7 @@ pub async fn loop_socket_v6(eee: &Node, socket: Arc<Socket>, cancel: Cancellatio
} }
pub async fn loop_socket_v4( pub async fn loop_socket_v4(
eee: &Node, eee: &'static Node,
socket: &Socket, socket: &Socket,
cancel: CancellationToken, cancel: CancellationToken,
is_multicast_sock: bool, is_multicast_sock: bool,
@ -603,7 +603,7 @@ async fn edge_send_packet_to_net(eee: &Node, data: Vec<u8>) {
} }
} }
pub async fn send_packet_to_net(eee: &Node, dst_mac: Mac, pkt: &[u8], size: u64) { pub async fn send_packet_to_net(eee: &'static Node, dst_mac: Mac, pkt: &[u8], size: u64) {
let (dest_sock, is_p2p) = find_peer_destination(eee, dst_mac).await; let (dest_sock, is_p2p) = find_peer_destination(eee, dst_mac).await;
if is_p2p { if is_p2p {
eee.stats.tx_p2p.fetch_add(size, Ordering::Relaxed); eee.stats.tx_p2p.fetch_add(size, Ordering::Relaxed);
@ -619,7 +619,7 @@ pub async fn send_packet_to_net(eee: &Node, dst_mac: Mac, pkt: &[u8], size: u64)
} }
} }
async fn find_peer_destination(eee: &Node, dst_mac: Mac) -> (SdlanSock, bool) { async fn find_peer_destination(eee: &'static Node, dst_mac: Mac) -> (SdlanSock, bool) {
if is_multi_broadcast(&dst_mac) { if is_multi_broadcast(&dst_mac) {
return ( return (
eee.config.super_nodes[eee.config.super_node_index.load(Ordering::Relaxed) as usize] eee.config.super_nodes[eee.config.super_node_index.load(Ordering::Relaxed) as usize]

View File

@ -29,7 +29,7 @@ use tracing::{debug, error, info};
use super::{EdgePeer, Node}; use super::{EdgePeer, Node};
pub async fn read_and_parse_packet( pub async fn read_and_parse_packet(
eee: &Node, eee: &'static Node,
sock: &Socket, sock: &Socket,
timeout: Option<Duration>, timeout: Option<Duration>,
_is_multicast_sock: bool, _is_multicast_sock: bool,
@ -86,7 +86,7 @@ pub async fn read_and_parse_packet(
Ok(()) Ok(())
} }
pub async fn handle_packet(eee: &Node, addr: SocketAddr, buf: &[u8]) -> Result<()> { pub async fn handle_packet(eee: &'static Node, addr: SocketAddr, buf: &[u8]) -> Result<()> {
if buf.len() < 1 { if buf.len() < 1 {
return Err(SDLanError::NormalError("buf length error")); return Err(SDLanError::NormalError("buf length error"));
} }
@ -228,7 +228,7 @@ pub async fn handle_packet(eee: &Node, addr: SocketAddr, buf: &[u8]) -> Result<(
} }
pub async fn handle_packet_peer_info( pub async fn handle_packet_peer_info(
eee: &Node, eee: &'static Node,
// cmn: Common<'_>, // cmn: Common<'_>,
body: &[u8], body: &[u8],
//sender_sock: &SdlanSock, //sender_sock: &SdlanSock,
@ -254,6 +254,7 @@ pub async fn handle_packet_peer_info(
error!("failed to convert v4"); error!("failed to convert v4");
return Ok(()); return Ok(());
}; };
let remote_nat = v4.nat_type;
let mut v6: [u8; 16] = [0; 16]; let mut v6: [u8; 16] = [0; 16];
let mut v6_port = 0; let mut v6_port = 0;
@ -411,7 +412,7 @@ async fn handle_packet_register_ack(
} }
async fn handle_packet_register( async fn handle_packet_register(
eee: &Node, eee: &'static Node,
// cmn: Common<'_>, // cmn: Common<'_>,
body: &[u8], body: &[u8],
from_sn: bool, from_sn: bool,
@ -513,7 +514,7 @@ async fn handle_packet_packet(
*/ */
pub async fn check_peer_registration_needed( pub async fn check_peer_registration_needed(
eee: &Node, eee: &'static Node,
from_sn: bool, from_sn: bool,
src_mac: Mac, src_mac: Mac,
_v6_info: &Option<V6Info>, _v6_info: &Option<V6Info>,
@ -555,7 +556,7 @@ pub async fn check_peer_registration_needed(
} }
async fn check_known_peer_sock_change( async fn check_known_peer_sock_change(
eee: &Node, eee: &'static Node,
from_sn: bool, from_sn: bool,
mac: Mac, mac: Mac,
// v6_info: &Option<V6Info>, // v6_info: &Option<V6Info>,
@ -592,7 +593,7 @@ async fn check_known_peer_sock_change(
} }
async fn register_with_new_peer( async fn register_with_new_peer(
eee: &Node, eee: &'static Node,
from_sn: bool, from_sn: bool,
mac: Mac, mac: Mac,
v6_info: &Option<V6Info>, v6_info: &Option<V6Info>,
@ -667,11 +668,11 @@ async fn register_with_new_peer(
} }
} }
async fn register_with_local_peers(eee: &Node) { async fn register_with_local_peers(eee: &'static Node) {
send_register(eee, &eee.multicast_sock, BROADCAST_MAC, &None).await; send_register(eee, &eee.multicast_sock, BROADCAST_MAC, &None).await;
} }
async fn send_register(eee: &Node, sock: &SdlanSock, mac: Mac, _v6_info: &Option<V6Info>) { async fn send_register(eee: &'static Node, sock: &SdlanSock, mac: Mac, _v6_info: &Option<V6Info>) {
if !eee.config.allow_p2p { if !eee.config.allow_p2p {
debug!("skipping register as p2p is disabled"); debug!("skipping register as p2p is disabled");
return; return;
@ -706,6 +707,19 @@ async fn send_register(eee: &Node, sock: &SdlanSock, mac: Mac, _v6_info: &Option
) )
.await; .await;
} }
let mut target_sock_v4 = sock.deepcopy();
tokio::spawn(async move {
for i in 1..=10 {
target_sock_v4.port += i;
let _ = send_to_sock(eee, &msg, &target_sock_v4).await;
tokio::time::sleep(Duration::from_millis(500)).await;
}
for i in 1..=10 {
target_sock_v4.port -= i;
let _ = send_to_sock(eee, &msg, &target_sock_v4).await;
tokio::time::sleep(Duration::from_millis(500)).await;
}
});
/* /*
let key = eee.get_header_key(); let key = eee.get_header_key();
if key.len() > 0 { if key.len() > 0 {
@ -985,7 +999,7 @@ fn peer_set_p2p_confirmed(eee: &Node, src_mac: Mac, sender_sock: &SdlanSock) {
eee.known_peers.insert_peer(src_mac, scan); eee.known_peers.insert_peer(src_mac, scan);
} }
pub async fn check_query_peer_info(eee: &Node, mac: Mac) { pub async fn check_query_peer_info(eee: &'static Node, mac: Mac) {
let scan: Arc<EdgePeer>; let scan: Arc<EdgePeer>;
let now = get_current_timestamp(); let now = get_current_timestamp();
match eee.pending_peers.get_peer(&mac) { match eee.pending_peers.get_peer(&mac) {