GuanM/hysteria-dev
1
0
Fork 0
mirror of https://github.com/cmz0228/hysteria-dev.git synced 2025-06-08 05:19:53 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Rename the original relay to TCPRelay

Browse Source
This commit is contained in:
Toby 2021-04-21 16:38:08 -07:00
parent 51abb06911
commit d8b198ca31
2 changed files with 9 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
cmd/client.go
View File

@ -183,7 +183,7 @@ func client(config *clientConfig) {
if len(config.Relay.Listen) > 0 {
go func() {
rl, err := relay.NewRelay(client, config.Relay.Listen, config.Relay.Remote,
rl, err := relay.NewTCPRelay(client, config.Relay.Listen, config.Relay.Remote,
time.Duration(config.Relay.Timeout)*time.Second,
func(addr net.Addr) {
logrus.WithFields(logrus.Fields{

19
pkg/relay/relay.go → pkg/relay/tcp.go
View File

@ -7,7 +7,7 @@ import (
"time"
)
type Relay struct {
type TCPRelay struct {
HyClient *core.Client
ListenAddr *net.TCPAddr
Remote string
@ -15,17 +15,15 @@ type Relay struct {
ConnFunc func(addr net.Addr)
ErrorFunc func(addr net.Addr, err error)
tcpListener *net.TCPListener
}
func NewRelay(hyClient *core.Client, listen, remote string, timeout time.Duration,
connFunc func(addr net.Addr), errorFunc func(addr net.Addr, err error)) (*Relay, error) {
func NewTCPRelay(hyClient *core.Client, 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)
if err != nil {
return nil, err
}
r := &Relay{
r := &TCPRelay{
HyClient: hyClient,
ListenAddr: tAddr,
Remote: remote,
@ -36,15 +34,14 @@ func NewRelay(hyClient *core.Client, listen, remote string, timeout time.Duratio
return r, nil
}
func (r *Relay) ListenAndServe() error {
var err error
r.tcpListener, err = net.ListenTCP("tcp", r.ListenAddr)
func (r *TCPRelay) ListenAndServe() error {
listener, err := net.ListenTCP("tcp", r.ListenAddr)
if err != nil {
return err
}
defer r.tcpListener.Close()
defer listener.Close()
for {
c, err := r.tcpListener.AcceptTCP()
c, err := listener.AcceptTCP()
if err != nil {
return err
}