style: refine code style

This commit is contained in:
AkinoKaede 2023-05-28 22:43:08 +08:00 committed by Shelikhoo
parent d1c3007d30
commit c324d3af70
No known key found for this signature in database
GPG Key ID: C4D5E79D22B25316
3 changed files with 8 additions and 8 deletions

View File

@ -55,12 +55,12 @@ func HandleTCP(handle func(tun_net.TCPConn)) StackOption {
// TODO: set sockopt
tcpConn := &tcpConn{
conn := &tcpConn{
TCPConn: gonet.NewTCPConn(wg, linkedEndpoint),
id: r.ID(),
}
tcpQueue <- tcpConn
handle(conn)
})
s.SetTransportProtocolHandler(tcp.ProtocolNumber, tcpForwarder.HandlePacket)

View File

@ -46,12 +46,12 @@ func HandleUDP(handle func(tun_net.UDPConn)) StackOption {
return
}
udpConn := &udpConn{
conn := &udpConn{
UDPConn: gonet.NewUDPConn(s, wg, linkedEndpoint),
id: r.ID(),
}
handle(udpConn)
handle(conn)
})
s.SetTransportProtocolHandler(gvisor_udp.ProtocolNumber, udpForwarder.HandlePacket)
return nil

View File

@ -42,21 +42,21 @@ func SetSpoofing(id tcpip.NICID, enable bool) StackOption {
func AddProtocolAddress(id tcpip.NICID, ips []*routercommon.CIDR) StackOption {
return func(s *stack.Stack) error {
for _, ip := range ips {
tcpIpAddr := tcpip.AddrFrom4Slice(ip.Ip)
tcpIPAddr := tcpip.AddrFrom4Slice(ip.Ip)
protocolAddress := tcpip.ProtocolAddress{
AddressWithPrefix: tcpip.AddressWithPrefix{
Address: tcpIpAddr,
Address: tcpIPAddr,
PrefixLen: int(ip.Prefix),
},
}
switch tcpIpAddr.Len() {
switch tcpIPAddr.Len() {
case 4:
protocolAddress.Protocol = ipv4.ProtocolNumber
case 16:
protocolAddress.Protocol = ipv6.ProtocolNumber
default:
return newError("invalid IP address length:", tcpIpAddr.Len())
return newError("invalid IP address length:", tcpIPAddr.Len())
}
if err := s.AddProtocolAddress(id, protocolAddress, stack.AddressProperties{}); err != nil {