luckyimnotanigger/tun2proxy
1
0
Fork 0
mirror of https://github.com/tun2proxy/tun2proxy.git synced 2025-04-22 14:59:09 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

IpProtocol values

Browse source
This commit is contained in:
ssrlive 2023-03-23 18:31:09 +08:00
parent 4af656039e
commit 6fe44f8055
3 changed files with 12 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
src/http.rs
View file

@ -4,6 +4,7 @@ use crate::tun2proxy::{
OutgoingDataEvent, OutgoingDirection, TcpProxy, OutgoingDataEvent, OutgoingDirection, TcpProxy,
}; };
use base64::Engine; use base64::Engine;
use smoltcp::wire::IpProtocol;
use std::collections::VecDeque; use std::collections::VecDeque;
use std::net::SocketAddr; use std::net::SocketAddr;
use std::rc::Rc; use std::rc::Rc;
@ -170,7 +171,7 @@ pub struct HttpManager {
impl ConnectionManager for HttpManager { impl ConnectionManager for HttpManager {
fn handles_connection(&self, connection: &Connection) -> bool { fn handles_connection(&self, connection: &Connection) -> bool {
connection.proto == smoltcp::wire::IpProtocol::Tcp.into() connection.proto == IpProtocol::Tcp.into()
} }
fn new_connection( fn new_connection(
@ -178,7 +179,7 @@ impl ConnectionManager for HttpManager {
connection: &Connection, connection: &Connection,
manager: Rc<dyn ConnectionManager>, manager: Rc<dyn ConnectionManager>,
) -> Option<Box<dyn TcpProxy>> { ) -> Option<Box<dyn TcpProxy>> {
if connection.proto != smoltcp::wire::IpProtocol::Tcp.into() { if connection.proto != IpProtocol::Tcp.into() {
return None; return None;
} }
Some(Box::new(HttpConnection::new(connection, manager))) Some(Box::new(HttpConnection::new(connection, manager)))

5
src/socks5.rs
View file

@ -3,6 +3,7 @@ use crate::tun2proxy::{
Connection, ConnectionManager, Credentials, DestinationHost, IncomingDataEvent, Connection, ConnectionManager, Credentials, DestinationHost, IncomingDataEvent,
IncomingDirection, OutgoingDataEvent, OutgoingDirection, TcpProxy, IncomingDirection, OutgoingDataEvent, OutgoingDirection, TcpProxy,
}; };
use smoltcp::wire::IpProtocol;
use std::collections::VecDeque; use std::collections::VecDeque;
use std::net::{IpAddr, SocketAddr}; use std::net::{IpAddr, SocketAddr};
use std::rc::Rc; use std::rc::Rc;
@ -297,7 +298,7 @@ pub struct Socks5Manager {
impl ConnectionManager for Socks5Manager { impl ConnectionManager for Socks5Manager {
fn handles_connection(&self, connection: &Connection) -> bool { fn handles_connection(&self, connection: &Connection) -> bool {
connection.proto == smoltcp::wire::IpProtocol::Tcp.into() connection.proto == IpProtocol::Tcp.into()
} }
fn new_connection( fn new_connection(
@ -305,7 +306,7 @@ impl ConnectionManager for Socks5Manager {
connection: &Connection, connection: &Connection,
manager: Rc<dyn ConnectionManager>, manager: Rc<dyn ConnectionManager>,
) -> Option<Box<dyn TcpProxy>> { ) -> Option<Box<dyn TcpProxy>> {
if connection.proto != smoltcp::wire::IpProtocol::Tcp.into() { if connection.proto != IpProtocol::Tcp.into() {
return None; return None;
} }
Some(Box::new(SocksConnection::new(connection, manager))) Some(Box::new(SocksConnection::new(connection, manager)))

11
src/tun2proxy.rs
View file

@ -10,7 +10,8 @@ use smoltcp::phy::{Device, Medium, RxToken, TunTapInterface, TxToken};
use smoltcp::socket::tcp; use smoltcp::socket::tcp;
use smoltcp::time::Instant; use smoltcp::time::Instant;
use smoltcp::wire::{ use smoltcp::wire::{
IpAddress, IpCidr, Ipv4Address, Ipv4Packet, Ipv6Address, Ipv6Packet, TcpPacket, UdpPacket, IpAddress, IpCidr, IpProtocol, Ipv4Address, Ipv4Packet, Ipv6Address, Ipv6Packet, TcpPacket,
UdpPacket,
}; };
use std::collections::HashMap; use std::collections::HashMap;
use std::convert::From; use std::convert::From;
@ -115,7 +116,7 @@ fn get_transport_info(
transport_offset: usize, transport_offset: usize,
packet: &[u8], packet: &[u8],
) -> Option<((u16, u16), bool, usize, usize)> { ) -> Option<((u16, u16), bool, usize, usize)> {
if proto == smoltcp::wire::IpProtocol::Udp.into() { if proto == IpProtocol::Udp.into() {
match UdpPacket::new_checked(packet) { match UdpPacket::new_checked(packet) {
Ok(result) => Some(( Ok(result) => Some((
(result.src_port(), result.dst_port()), (result.src_port(), result.dst_port()),
@ -125,7 +126,7 @@ fn get_transport_info(
)), )),
Err(_) => None, Err(_) => None,
} }
} else if proto == smoltcp::wire::IpProtocol::Tcp.into() { } else if proto == IpProtocol::Tcp.into() {
match TcpPacket::new_checked(packet) { match TcpPacket::new_checked(packet) {
Ok(result) => Some(( Ok(result) => Some((
(result.src_port(), result.dst_port()), (result.src_port(), result.dst_port()),
@ -380,7 +381,7 @@ impl<'a> TunToProxy<'a> {
if let Some((connection, first_packet, _payload_offset, _payload_size)) = if let Some((connection, first_packet, _payload_offset, _payload_size)) =
connection_tuple(frame) connection_tuple(frame)
{ {
if connection.proto == smoltcp::wire::IpProtocol::Tcp.into() { if connection.proto == IpProtocol::Tcp.into() {
let cm = self.get_connection_manager(&connection); let cm = self.get_connection_manager(&connection);
if cm.is_none() { if cm.is_none() {
return; return;
@ -447,7 +448,7 @@ impl<'a> TunToProxy<'a> {
// The connection handler builds up the connection or encapsulates the data. // The connection handler builds up the connection or encapsulates the data.
// Therefore, we now expect it to write data to the server. // Therefore, we now expect it to write data to the server.
self.write_to_server(&connection); self.write_to_server(&connection);
} else if connection.proto == smoltcp::wire::IpProtocol::Udp.into() { } else if connection.proto == IpProtocol::Udp.into() {
// UDP is not yet supported // UDP is not yet supported
/*if _payload_offset > frame.len() || _payload_offset + _payload_offset > frame.len() { /*if _payload_offset > frame.len() || _payload_offset + _payload_offset > frame.len() {
return; return;