GuanM/hysteria-dev
1
0
Fork 0
mirror of https://github.com/cmz0228/hysteria-dev.git synced 2025-07-22 19:38:02 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Use core.Transport

Browse Source
This commit is contained in:
Toby
2021-04-27 20:14:43 -07:00
parent 4da73888f4
commit 5ac95d987a
11 changed files with 120 additions and 49 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
pkg/relay/tcp.go
View File

@@ -9,6 +9,7 @@ import (
type TCPRelay struct {
HyClient *core.Client
Transport core.Transport
ListenAddr *net.TCPAddr
Remote string
Timeout time.Duration
@@ -17,14 +18,15 @@ type TCPRelay struct {
ErrorFunc func(addr net.Addr, err error)
}
func NewTCPRelay(hyClient *core.Client, listen, remote string, timeout time.Duration,
func NewTCPRelay(hyClient *core.Client, transport core.Transport, listen, remote string, timeout time.Duration,
connFunc func(addr net.Addr), errorFunc func(addr net.Addr, err error)) (*TCPRelay, error) {
tAddr, err := net.ResolveTCPAddr("tcp", listen)
tAddr, err := transport.LocalResolveTCPAddr(listen)
if err != nil {
return nil, err
}
r := &TCPRelay{
HyClient: hyClient,
Transport: transport,
ListenAddr: tAddr,
Remote: remote,
Timeout: timeout,
@@ -35,7 +37,7 @@ func NewTCPRelay(hyClient *core.Client, listen, remote string, timeout time.Dura
}
func (r *TCPRelay) ListenAndServe() error {
listener, err := net.ListenTCP("tcp", r.ListenAddr)
listener, err := r.Transport.LocalListenTCP(r.ListenAddr)
if err != nil {
return err
}