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

chore: better quic close error codes

Browse Source
This commit is contained in:
Toby
2022-10-22 12:17:10 -07:00
parent b1d9ab6c6a
commit c98c7eca4e
3 changed files with 24 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
pkg/core/client.go
View File

@@ -94,18 +94,18 @@ func (c *Client) connect() error {
stream, err := quicConn.OpenStreamSync(ctx)
ctxCancel()
if err != nil {
_ = quicConn.CloseWithError(closeErrorCodeProtocol, "protocol error")
_ = closeErrorProtocol.Send(quicConn)
_ = pktConn.Close()
return err
}
ok, msg, err := c.handleControlStream(quicConn, stream)
if err != nil {
_ = quicConn.CloseWithError(closeErrorCodeProtocol, "protocol error")
_ = closeErrorProtocol.Send(quicConn)
_ = pktConn.Close()
return err
}
if !ok {
_ = quicConn.CloseWithError(closeErrorCodeAuth, "auth error")
_ = closeErrorAuth.Send(quicConn)
_ = pktConn.Close()
return fmt.Errorf("auth error: %s", msg)
}
@@ -296,7 +296,7 @@ func (c *Client) DialUDP() (UDPConn, error) {
func (c *Client) Close() error {
c.reconnectMutex.Lock()
defer c.reconnectMutex.Unlock()
err := c.quicConn.CloseWithError(closeErrorCodeGeneric, "")
err := closeErrorGeneric.Send(c.quicConn)
_ = c.pktConn.Close()
c.closed = true
return err