mirrors/krata
1
0
Fork 0
mirror of https://github.com/edera-dev/krata.git synced 2025-08-09 16:11:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

network: optimize proxynat packet reconstruction

Browse Source
This commit is contained in:
Alex Zenla
2024-02-12 17:24:40 +00:00
parent 341907a536
commit 60c9aefdda
3 changed files with 26 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
network/src/proxynat/tcp.rs
View File

@ -203,7 +203,7 @@ impl ProxyTcpHandler {
}
ProxyTcpAcceptSelect::TxIpPacket(payload) => {
let mut buffer: Vec<u8> = Vec::new();
let mut buffer = BytesMut::with_capacity(Ethernet2Header::LEN + payload.len());
let header = Ethernet2Header {
source: context.key.local_mac.0,
destination: context.key.client_mac.0,
@ -212,9 +212,9 @@ impl ProxyTcpHandler {
IpAddress::Ipv6(_) => EtherType::IPV6,
},
};
header.write(&mut buffer)?;
buffer.extend_from_slice(&header.to_bytes());
buffer.extend_from_slice(&payload);
if let Err(error) = context.try_send(buffer.as_slice().into()) {
if let Err(error) = context.try_send(buffer) {
debug!("failed to transmit tcp packet: {}", error);
}
}
@ -376,7 +376,7 @@ impl ProxyTcpHandler {
}
ProxyTcpDataSelect::TxIpPacket(payload) => {
let mut buffer: Vec<u8> = Vec::new();
let mut buffer = BytesMut::with_capacity(Ethernet2Header::LEN + payload.len());
let header = Ethernet2Header {
source: context.key.local_mac.0,
destination: context.key.client_mac.0,
@ -385,9 +385,9 @@ impl ProxyTcpHandler {
IpAddress::Ipv6(_) => EtherType::IPV6,
},
};
header.write(&mut buffer)?;
buffer.extend_from_slice(&header.to_bytes());
buffer.extend_from_slice(&payload);
if let Err(error) = context.try_send(buffer.as_slice().into()) {
if let Err(error) = context.try_send(buffer) {
debug!("failed to transmit tcp packet: {}", error);
}
}
@ -436,7 +436,7 @@ impl ProxyTcpHandler {
}
ProxyTcpFinishSelect::TxIpPacket(payload) => {
let mut buffer: Vec<u8> = Vec::new();
let mut buffer = BytesMut::with_capacity(Ethernet2Header::LEN + payload.len());
let header = Ethernet2Header {
source: context.key.local_mac.0,
destination: context.key.client_mac.0,
@ -445,9 +445,9 @@ impl ProxyTcpHandler {
IpAddress::Ipv6(_) => EtherType::IPV6,
},
};
header.write(&mut buffer)?;
buffer.extend_from_slice(&header.to_bytes());
buffer.extend_from_slice(&payload);
if let Err(error) = context.try_send(buffer.as_slice().into()) {
if let Err(error) = context.try_send(buffer) {
debug!("failed to transmit tcp packet: {}", error);
}
}