mirror of
https://github.com/SagerNet/sing-box.git
synced 2024-11-16 23:53:43 +08:00
66 lines
1.3 KiB
Go
66 lines
1.3 KiB
Go
package hysteria
|
|
|
|
import (
|
|
"net"
|
|
"os"
|
|
"syscall"
|
|
|
|
"github.com/sagernet/quic-go"
|
|
"github.com/sagernet/sing/common"
|
|
)
|
|
|
|
type PacketConnWrapper struct {
|
|
net.PacketConn
|
|
}
|
|
|
|
func (c *PacketConnWrapper) SetReadBuffer(bytes int) error {
|
|
return common.MustCast[*net.UDPConn](c.PacketConn).SetReadBuffer(bytes)
|
|
}
|
|
|
|
func (c *PacketConnWrapper) SetWriteBuffer(bytes int) error {
|
|
return common.MustCast[*net.UDPConn](c.PacketConn).SetWriteBuffer(bytes)
|
|
}
|
|
|
|
func (c *PacketConnWrapper) SyscallConn() (syscall.RawConn, error) {
|
|
return common.MustCast[*net.UDPConn](c.PacketConn).SyscallConn()
|
|
}
|
|
|
|
func (c *PacketConnWrapper) File() (f *os.File, err error) {
|
|
return common.MustCast[*net.UDPConn](c.PacketConn).File()
|
|
}
|
|
|
|
func (c *PacketConnWrapper) Upstream() any {
|
|
return c.PacketConn
|
|
}
|
|
|
|
type StreamWrapper struct {
|
|
Conn quic.Connection
|
|
quic.Stream
|
|
}
|
|
|
|
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) ReaderReplaceable() bool {
|
|
return true
|
|
}
|
|
|
|
func (s *StreamWrapper) WriterReplaceable() bool {
|
|
return true
|
|
}
|
|
|
|
func (s *StreamWrapper) Close() error {
|
|
s.CancelRead(0)
|
|
s.Stream.Close()
|
|
return nil
|
|
}
|