Merge pull request #13 from ICKelin/fix/quic_reconnect

Fix/quic reconnect
ICKelin · Sep 7, 2024 · 99aeb8d · 99aeb8d
2 parents e67277a + 95c6c72
commit 99aeb8d
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/src/gtun/proxy/tproxy_tcp.go b/src/gtun/proxy/tproxy_tcp.go
@@ -141,6 +141,10 @@ func (p *TProxyTCP) doProxy(conn net.Conn) {
 	logs.Debug("%s:%s=>%s:%s", sip, sport, dip, dport)
 	stream, err := sess.OpenStream()
 	if err != nil {
+		// force close to trigger reconnect
+		// quic CAN'T get close state by sess.IsClose()
+		// Close to trigger quic reconnect
+		sess.Close()
 		logs.Error("open stream fail: %v", err)
 		return
 	}

diff --git a/src/gtun/proxy/tproxy_udp.go b/src/gtun/proxy/tproxy_udp.go
@@ -197,6 +197,10 @@ func (p *TProxyUDP) serve(lconn *net.UDPConn) error {
 
 			stream, err := sess.OpenStream()
 			if err != nil {
+				// force close to trigger reconnect
+				// quic CAN'T get close state by sess.IsClose()
+				// Close to trigger quic reconnect
+				sess.Close()
 				logs.Error("open stream fail: %v", err)
 				continue
 			}