luckyimnotanigger/tun2proxy
1
0
Fork 0
mirror of https://github.com/tun2proxy/tun2proxy.git synced 2025-06-09 16:17:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

refine code

Browse source
This commit is contained in:
ssrlive 2023-03-27 22:26:57 +08:00 • committed by B. Blechschmidt
parent 94d70d11f4
commit eac87c4f9f
2 changed files with 4 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/tun2proxy.rs
View file

@ -389,9 +389,10 @@ impl<'a> TunToProxy<'a> {
if resolved_conn.proto == IpProtocol::Tcp { if resolved_conn.proto == IpProtocol::Tcp {
let cm = self.get_connection_manager(&resolved_conn); let cm = self.get_connection_manager(&resolved_conn);
if cm.is_none() { if cm.is_none() {
log::trace!("no connect manager");
return Ok(()); return Ok(());
} }
let server = cm.ok_or("no connect manager")?.get_server(); let server = cm.unwrap().get_server();
if first_packet { if first_packet {
for manager in self.connection_managers.iter_mut() { for manager in self.connection_managers.iter_mut() {
if let Some(handler) = if let Some(handler) =
@ -581,6 +582,7 @@ impl<'a> TunToProxy<'a> {
let e = "connection not found"; let e = "connection not found";
let conn_ref = self.token_to_connection.get(&event.token()); let conn_ref = self.token_to_connection.get(&event.token());
if conn_ref.is_none() { if conn_ref.is_none() {
log::trace!("{e}");
return Ok(()); return Ok(());
} }
let connection = conn_ref.unwrap().clone(); let connection = conn_ref.unwrap().clone();

3
src/virtdevice.rs
View file

@ -11,8 +11,7 @@ pub struct VirtualTunDevice {
impl VirtualTunDevice { impl VirtualTunDevice {
pub fn inject_packet(&mut self, buffer: &[u8]) { pub fn inject_packet(&mut self, buffer: &[u8]) {
let vec = Vec::from(buffer); self.inbuf.push(buffer.to_vec());
self.inbuf.push(vec);
} }
pub fn exfiltrate_packet(&mut self) -> Option<Vec<u8>> { pub fn exfiltrate_packet(&mut self) -> Option<Vec<u8>> {