clash/tunnel/connection.go

98 lines
2.1 KiB
Go
Raw Normal View History

package tunnel
import (
"errors"
"io"
2018-09-16 16:15:58 +00:00
"net"
2018-09-21 03:33:29 +00:00
"time"
N "github.com/Dreamacro/clash/common/net"
"github.com/Dreamacro/clash/common/pool"
"github.com/Dreamacro/clash/component/resolver"
C "github.com/Dreamacro/clash/constant"
)
func handleUDPToRemote(packet C.UDPPacket, pc C.PacketConn, metadata *C.Metadata) error {
defer packet.Drop()
// local resolve UDP dns
if !metadata.Resolved() {
ip, err := resolver.ResolveIP(metadata.Host)
if err != nil {
return err
}
metadata.DstIP = ip
}
addr := metadata.UDPAddr()
if addr == nil {
return errors.New("udp addr invalid")
}
if _, err := pc.WriteTo(packet.Data(), addr); err != nil {
return err
}
// reset timeout
pc.SetReadDeadline(time.Now().Add(udpTimeout))
return nil
}
2019-04-23 15:29:36 +00:00
2020-03-10 12:36:24 +00:00
func handleUDPToLocal(packet C.UDPPacket, pc net.PacketConn, key string, fAddr net.Addr) {
2021-11-03 14:26:51 +00:00
buf := pool.Get(pool.UDPBufferSize)
2020-04-24 16:30:40 +00:00
defer pool.Put(buf)
defer natTable.Delete(key)
2019-10-11 12:11:18 +00:00
defer pc.Close()
2019-04-23 15:29:36 +00:00
for {
2020-01-31 06:43:54 +00:00
pc.SetReadDeadline(time.Now().Add(udpTimeout))
n, from, err := pc.ReadFrom(buf)
2019-04-23 15:29:36 +00:00
if err != nil {
return
2019-04-23 15:29:36 +00:00
}
2020-03-10 12:36:24 +00:00
if fAddr != nil {
from = fAddr
}
2020-08-25 14:19:59 +00:00
_, err = packet.WriteBack(buf[:n], from)
if err != nil {
return
2019-04-23 15:29:36 +00:00
}
}
}
2019-04-23 15:29:36 +00:00
func handleSocket(ctx C.ConnContext, outbound net.Conn) {
relay(ctx.Conn(), outbound)
2018-09-16 16:15:58 +00:00
}
// relay copies between left and right bidirectionally.
func relay(leftConn, rightConn net.Conn) {
ch := make(chan error)
2022-03-16 10:17:28 +00:00
tcpKeepAlive(leftConn)
tcpKeepAlive(rightConn)
2018-09-16 16:15:58 +00:00
go func() {
2020-04-24 16:30:40 +00:00
buf := pool.Get(pool.RelayBufferSize)
// Wrapping to avoid using *net.TCPConn.(ReadFrom)
// See also https://github.com/Dreamacro/clash/pull/1209
_, err := io.CopyBuffer(N.WriteOnlyWriter{Writer: leftConn}, N.ReadOnlyReader{Reader: rightConn}, buf)
2020-04-24 16:30:40 +00:00
pool.Put(buf)
2018-09-21 03:33:29 +00:00
leftConn.SetReadDeadline(time.Now())
2018-09-16 16:15:58 +00:00
ch <- err
}()
2020-04-24 16:30:40 +00:00
buf := pool.Get(pool.RelayBufferSize)
io.CopyBuffer(N.WriteOnlyWriter{Writer: rightConn}, N.ReadOnlyReader{Reader: leftConn}, buf)
2020-04-24 16:30:40 +00:00
pool.Put(buf)
2018-09-21 03:33:29 +00:00
rightConn.SetReadDeadline(time.Now())
2018-09-16 16:15:58 +00:00
<-ch
}
2022-03-16 10:17:28 +00:00
func tcpKeepAlive(c net.Conn) {
if tcp, ok := c.(*net.TCPConn); ok {
tcp.SetKeepAlive(true)
}
}