yeyinlaoying/vnt
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

修复不加密不能打洞的问题

Browse Source
This commit is contained in:
lubeilin
2023-08-27 17:51:42 +08:00
parent e611d69a25
commit 2e507f1a9d
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
vnt/src/handle/punch_handler.rs
+4 -4
View File
@@ -113,8 +113,8 @@ async fn start_punch_(
if count > 2 { if count > 2 {
break; break;
} }
let buf = punch_packet(client_cipher, current_device.virtual_ip(), &nat_info, info.virtual_ip)?; let packet = punch_packet(client_cipher, current_device.virtual_ip(), &nat_info, info.virtual_ip)?;
let _ = sender.send_main(&buf, current_device.connect_server).await; let _ = sender.send_main(packet.buffer(), current_device.connect_server).await;
} }
tokio::time::sleep(sleep_time).await; tokio::time::sleep(sleep_time).await;
Ok(()) Ok(())
@@ -125,7 +125,7 @@ pub fn punch_packet(
virtual_ip: Ipv4Addr, virtual_ip: Ipv4Addr,
nat_info: &NatInfo, nat_info: &NatInfo,
dest: Ipv4Addr, dest: Ipv4Addr,
) -> crate::Result<Vec<u8>> { ) -> crate::Result<NetPacket<Vec<u8>>> {
let mut punch_reply = PunchInfo::new(); let mut punch_reply = PunchInfo::new();
punch_reply.reply = false; punch_reply.reply = false;
punch_reply.public_ip_list = nat_info punch_reply.public_ip_list = nat_info
@@ -148,5 +148,5 @@ pub fn punch_packet(
net_packet.set_destination(dest); net_packet.set_destination(dest);
net_packet.set_payload(&bytes)?; net_packet.set_payload(&bytes)?;
client_cipher.encrypt_ipv4(&mut net_packet)?; client_cipher.encrypt_ipv4(&mut net_packet)?;
Ok(net_packet.into_buffer()) Ok(net_packet)
} }