black-internet-days-ahead/sing-box
1
0
Fork 0
mirror of https://github.com/SagerNet/sing-box.git synced 2024-11-09 18:43:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add retry for mux stream open

Browse source
This commit is contained in:
世界 2022-07-31 18:08:24 +08:00
parent d06fd03dd8
commit e014220508
No known key found for this signature in database
GPG key ID: CD109927C34A63C4
1 changed files with 17 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
common/mux/client.go
View file

@ -75,20 +75,30 @@ func (c *Client) ListenPacket(ctx context.Context, destination M.Socksaddr) (net
if err != nil {
return nil, err
}
// return bufio.NewUnbindPacketConn(&ClientPacketConn{ExtendedConn: bufio.NewExtendedConn(stream), destination: destination}), nil
return &ClientPacketAddrConn{ExtendedConn: bufio.NewExtendedConn(stream), destination: destination}, nil
}
func (c *Client) openStream() (net.Conn, error) {
session, err := c.offer()
var (
session *yamux.Session
stream *yamux.Stream
err error
)
for attempts := 0; attempts < 2; attempts++ {
session, err = c.offer()
if err != nil {
continue
}
stream, err = session.OpenStream()
if err != nil {
continue
}
break
}
if err != nil {
return nil, err
}
conn, err := session.Open()
if err != nil {
return nil, err
}
return &wrapStream{conn}, nil
return &wrapStream{stream}, nil
}
func (c *Client) offer() (*yamux.Session, error) {