go-gost/x
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

do not recreate tun device when connecting to server failed

Browse Source
This commit is contained in:
ginuerzh
2022-11-11 20:53:34 +08:00
parent 08a937f3a1
commit 81b6efc9b8
3 changed files with 51 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
handler/tun/server.go
View File

@ -6,6 +6,7 @@ import (
"io"
"net"
"net/netip"
"time"
"github.com/go-gost/core/common/bufpool"
"github.com/go-gost/core/logger"
@ -16,16 +17,26 @@ import (
)
func (h *tunHandler) handleServer(ctx context.Context, conn net.Conn, config *tun_util.Config, log logger.Logger) error {
pc, err := net.ListenPacket(conn.LocalAddr().Network(), conn.LocalAddr().String())
if err != nil {
return err
}
defer pc.Close()
for {
err := func() error {
pc, err := net.ListenPacket(conn.LocalAddr().Network(), conn.LocalAddr().String())
if err != nil {
return err
}
defer pc.Close()
return h.transportServer(conn, pc, config, log)
return h.transportServer(conn, pc, config, log)
}()
if err == ErrTun {
return err
}
log.Error(err)
time.Sleep(time.Second)
}
}
func (h *tunHandler) transportServer(tun net.Conn, conn net.PacketConn, config *tun_util.Config, log logger.Logger) error {
func (h *tunHandler) transportServer(tun io.ReadWriter, conn net.PacketConn, config *tun_util.Config, log logger.Logger) error {
tunIP, _, err := net.ParseCIDR(config.Net)
if err != nil {
return err
@ -41,7 +52,7 @@ func (h *tunHandler) transportServer(tun net.Conn, conn net.PacketConn, config *
n, err := tun.Read(*b)
if err != nil {
return err
return ErrTun
}
var src, dst net.IP
@ -181,7 +192,7 @@ func (h *tunHandler) transportServer(tun net.Conn, conn net.PacketConn, config *
}
if _, err := tun.Write((*b)[:n]); err != nil {
return err
return ErrTun
}
return nil
}()