This commit is contained in:
TenderIronh
2025-12-10 16:47:16 +08:00
parent d3e8ee2a32
commit 47220fe38b
5 changed files with 70 additions and 38 deletions
+25 -14
View File
@@ -16,14 +16,14 @@ import (
"github.com/quic-go/quic-go"
)
// quic.DialContext do not support version 44,disable it
var quicVersion []quic.VersionNumber
// quic.Dial do not support version 44, disable it
var quicVersion []quic.Version
type underlayQUIC struct {
listener quic.Listener
listener *quic.Listener
writeMtx *sync.Mutex
quic.Stream
quic.Connection
*quic.Stream
*quic.Conn
}
func (conn *underlayQUIC) Protocol() string {
@@ -47,8 +47,17 @@ func (conn *underlayQUIC) WriteMessage(mainType uint16, subType uint16, packet i
}
func (conn *underlayQUIC) Close() error {
conn.Stream.CancelRead(1)
conn.Connection.CloseWithError(0, "")
// CancelRead expects a StreamErrorCode; using 1 as before (application-defined)
if conn.Stream != nil {
conn.Stream.CancelRead(1)
// close send-side of stream
_ = conn.Stream.Close()
}
if conn.Conn != nil {
// CloseWithError expects an ApplicationErrorCode and a description.
// 0 is zero-value; keep behavior similar to old CloseWithError(0,"")
_ = conn.Conn.CloseWithError(0, "")
}
conn.CloseListener()
return nil
}
@@ -60,7 +69,7 @@ func (conn *underlayQUIC) WUnlock() {
}
func (conn *underlayQUIC) CloseListener() {
if conn.listener != nil {
conn.listener.Close()
_ = conn.listener.Close()
}
}
@@ -76,7 +85,7 @@ func (conn *underlayQUIC) Accept() error {
return err
}
conn.Stream = stream
conn.Connection = sess
conn.Conn = sess
return nil
}
@@ -96,23 +105,25 @@ func listenQuic(addr string, idleTimeout time.Duration) (*underlayQUIC, error) {
return ul, nil
}
func dialQuic(conn *net.UDPConn, remoteAddr *net.UDPAddr, timeout time.Duration) (*underlayQUIC, error) {
func dialQuic(pconn *net.UDPConn, remoteAddr *net.UDPAddr, timeout time.Duration) (*underlayQUIC, error) {
tlsConf := &tls.Config{
InsecureSkipVerify: true,
NextProtos: []string{"openp2pv1"},
}
ctx, cancel := context.WithTimeout(context.Background(), timeout)
defer cancel()
Connection, err := quic.DialContext(ctx, conn, remoteAddr, conn.LocalAddr().String(), tlsConf,
// New API: quic.Dial(ctx, packetConn, remoteAddr, tlsConf, quicConfig)
connection, err := quic.Dial(ctx, pconn, remoteAddr, tlsConf,
&quic.Config{Versions: quicVersion, MaxIdleTimeout: TunnelIdleTimeout, DisablePathMTUDiscovery: true})
if err != nil {
return nil, fmt.Errorf("quic.DialContext error:%s", err)
return nil, fmt.Errorf("quic.Dial error:%s", err)
}
stream, err := Connection.OpenStreamSync(context.Background())
stream, err := connection.OpenStreamSync(context.Background())
if err != nil {
return nil, fmt.Errorf("OpenStreamSync error:%s", err)
}
qConn := &underlayQUIC{nil, &sync.Mutex{}, stream, Connection}
qConn := &underlayQUIC{nil, &sync.Mutex{}, stream, connection}
return qConn, nil
}