mirror of
https://github.com/cmz0228/hysteria-dev.git
synced 2025-06-08 13:29:53 +00:00
98 lines
2.2 KiB
Go
98 lines
2.2 KiB
Go
package redirect
|
|
|
|
import (
|
|
"encoding/binary"
|
|
"errors"
|
|
"net"
|
|
"syscall"
|
|
"time"
|
|
|
|
"github.com/apernet/hysteria/core/cs"
|
|
"github.com/apernet/hysteria/core/utils"
|
|
)
|
|
|
|
type TCPRedirect struct {
|
|
HyClient *cs.Client
|
|
ListenAddr *net.TCPAddr
|
|
Timeout time.Duration
|
|
|
|
ConnFunc func(addr, reqAddr net.Addr)
|
|
ErrorFunc func(addr, reqAddr net.Addr, err error)
|
|
}
|
|
|
|
func NewTCPRedirect(hyClient *cs.Client, listen string, timeout time.Duration,
|
|
connFunc func(addr, reqAddr net.Addr),
|
|
errorFunc func(addr, reqAddr net.Addr, err error),
|
|
) (*TCPRedirect, error) {
|
|
tAddr, err := net.ResolveTCPAddr("tcp", listen)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
r := &TCPRedirect{
|
|
HyClient: hyClient,
|
|
ListenAddr: tAddr,
|
|
Timeout: timeout,
|
|
ConnFunc: connFunc,
|
|
ErrorFunc: errorFunc,
|
|
}
|
|
return r, nil
|
|
}
|
|
|
|
func (r *TCPRedirect) ListenAndServe() error {
|
|
listener, err := net.ListenTCP("tcp", r.ListenAddr)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer listener.Close()
|
|
for {
|
|
c, err := listener.Accept()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
go func() {
|
|
defer c.Close()
|
|
dest, err := getDestAddr(c.(*net.TCPConn))
|
|
if err != nil || dest.IP.IsLoopback() {
|
|
// Silently drop the connection if we failed to get the destination address,
|
|
// or if it's a loopback address (not a redirected connection).
|
|
return
|
|
}
|
|
r.ConnFunc(c.RemoteAddr(), dest)
|
|
rc, err := r.HyClient.DialTCP(dest.String())
|
|
if err != nil {
|
|
r.ErrorFunc(c.RemoteAddr(), dest, err)
|
|
return
|
|
}
|
|
defer rc.Close()
|
|
err = utils.PipePairWithTimeout(c, rc, r.Timeout)
|
|
r.ErrorFunc(c.RemoteAddr(), dest, err)
|
|
}()
|
|
}
|
|
}
|
|
|
|
func getDestAddr(conn *net.TCPConn) (*net.TCPAddr, error) {
|
|
rc, err := conn.SyscallConn()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
var addr *sockAddr
|
|
var err2 error
|
|
err = rc.Control(func(fd uintptr) {
|
|
addr, err2 = getOrigDst(fd)
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if err2 != nil {
|
|
return nil, err2
|
|
}
|
|
switch addr.family {
|
|
case syscall.AF_INET:
|
|
return &net.TCPAddr{IP: addr.data[:4], Port: int(binary.BigEndian.Uint16(addr.port[:]))}, nil
|
|
case syscall.AF_INET6:
|
|
return &net.TCPAddr{IP: addr.data[4:20], Port: int(binary.BigEndian.Uint16(addr.port[:]))}, nil
|
|
default:
|
|
return nil, errors.New("unknown address family")
|
|
}
|
|
}
|