mirror of
https://github.com/SagerNet/sing-box.git
synced 2024-11-23 17:11:29 +00:00
42 lines
736 B
Go
42 lines
736 B
Go
|
package v2rayquic
|
||
|
|
||
|
import (
|
||
|
"net"
|
||
|
|
||
|
"github.com/sagernet/quic-go"
|
||
|
"github.com/sagernet/sing/common/baderror"
|
||
|
)
|
||
|
|
||
|
type StreamWrapper struct {
|
||
|
Conn quic.Connection
|
||
|
quic.Stream
|
||
|
}
|
||
|
|
||
|
func (s *StreamWrapper) Read(p []byte) (n int, err error) {
|
||
|
n, err = s.Stream.Read(p)
|
||
|
return n, baderror.WrapQUIC(err)
|
||
|
}
|
||
|
|
||
|
func (s *StreamWrapper) Write(p []byte) (n int, err error) {
|
||
|
n, err = s.Stream.Write(p)
|
||
|
return n, baderror.WrapQUIC(err)
|
||
|
}
|
||
|
|
||
|
func (s *StreamWrapper) LocalAddr() net.Addr {
|
||
|
return s.Conn.LocalAddr()
|
||
|
}
|
||
|
|
||
|
func (s *StreamWrapper) RemoteAddr() net.Addr {
|
||
|
return s.Conn.RemoteAddr()
|
||
|
}
|
||
|
|
||
|
func (s *StreamWrapper) Upstream() any {
|
||
|
return s.Stream
|
||
|
}
|
||
|
|
||
|
func (s *StreamWrapper) Close() error {
|
||
|
s.CancelRead(0)
|
||
|
s.Stream.Close()
|
||
|
return nil
|
||
|
}
|