update sing-box support

common/rate/conn.go

@@ -0,0 +1,56 @@
+package rate
+
+import (
+	"net"
+
+	"github.com/juju/ratelimit"
+	"github.com/sagernet/sing/common/buf"
+	M "github.com/sagernet/sing/common/metadata"
+	"github.com/sagernet/sing/common/network"
+)
+
+func NewConnRateLimiter(c net.Conn, l *ratelimit.Bucket) *Conn {
+	return &Conn{
+		Conn:    c,
+		limiter: l,
+	}
+}
+
+type Conn struct {
+	net.Conn
+	limiter *ratelimit.Bucket
+}
+
+func (c *Conn) Read(b []byte) (n int, err error) {
+	c.limiter.Wait(int64(len(b)))
+	return c.Conn.Read(b)
+}
+
+func (c *Conn) Write(b []byte) (n int, err error) {
+	c.limiter.Wait(int64(len(b)))
+	return c.Conn.Write(b)
+}
+
+type PacketConnCounter struct {
+	network.PacketConn
+	limiter *ratelimit.Bucket
+}
+
+func NewPacketConnCounter(conn network.PacketConn, l *ratelimit.Bucket) network.PacketConn {
+	return &PacketConnCounter{
+		PacketConn: conn,
+		limiter:    l,
+	}
+}
+
+func (p *PacketConnCounter) ReadPacket(buff *buf.Buffer) (destination M.Socksaddr, err error) {
+	pLen := buff.Len()
+	destination, err = p.ReadPacket(buff)
+	p.limiter.Wait(int64(buff.Len() - pLen))
+	return
+}
+
+func (p *PacketConnCounter) WritePacket(buff *buf.Buffer, destination M.Socksaddr) (err error) {
+	p.limiter.Wait(int64(buff.Len()))
+	return p.PacketConn.WritePacket(buff, destination)
+}