fix race condition

This commit is contained in:
ginuerzh
2023-10-19 23:47:47 +08:00
parent f2fd6554ad
commit cc4310106b
29 changed files with 157 additions and 156 deletions

View File

@ -75,8 +75,8 @@ func (c *udpConn) Read(b []byte) (n int, err error) {
buf := bufpool.Get(dlen) buf := bufpool.Get(dlen)
defer bufpool.Put(buf) defer bufpool.Put(buf)
_, err = io.ReadFull(c.Conn, *buf) _, err = io.ReadFull(c.Conn, buf)
n = copy(b, *buf) n = copy(b, buf)
return return
} }
@ -169,8 +169,8 @@ func (c *bindUDPConn) Read(b []byte) (n int, err error) {
buf := bufpool.Get(dlen) buf := bufpool.Get(dlen)
defer bufpool.Put(buf) defer bufpool.Put(buf)
_, err = io.ReadFull(c.Conn, *buf) _, err = io.ReadFull(c.Conn, buf)
n = copy(b, *buf) n = copy(b, buf)
return return
} }

View File

@ -10,10 +10,6 @@ import (
"github.com/google/uuid" "github.com/google/uuid"
) )
const (
defaultMuxVersion = 1
)
type metadata struct { type metadata struct {
connectTimeout time.Duration connectTimeout time.Duration
noDelay bool noDelay bool
@ -47,9 +43,6 @@ func (c *relayConnector) parseMetadata(md mdata.Metadata) (err error) {
MaxReceiveBuffer: mdutil.GetInt(md, "mux.maxReceiveBuffer"), MaxReceiveBuffer: mdutil.GetInt(md, "mux.maxReceiveBuffer"),
MaxStreamBuffer: mdutil.GetInt(md, "mux.maxStreamBuffer"), MaxStreamBuffer: mdutil.GetInt(md, "mux.maxStreamBuffer"),
} }
if c.md.muxCfg.Version == 0 {
c.md.muxCfg.Version = defaultMuxVersion
}
return return
} }

View File

@ -62,7 +62,7 @@ func (c *socks5Connector) muxBindTCP(ctx context.Context, conn net.Conn, network
return nil, err return nil, err
} }
session, err := mux.ServerSession(conn, nil) session, err := mux.ServerSession(conn, c.md.muxCfg)
if err != nil { if err != nil {
return nil, err return nil, err
} }

View File

@ -36,7 +36,7 @@ func (c *udpRelayConn) ReadFrom(b []byte) (n int, addr net.Addr, err error) {
buf := bufpool.Get(c.bufferSize) buf := bufpool.Get(c.bufferSize)
defer bufpool.Put(buf) defer bufpool.Put(buf)
nn, err := c.udpConn.Read(*buf) nn, err := c.udpConn.Read(buf)
if err != nil { if err != nil {
return return
} }
@ -48,7 +48,7 @@ func (c *udpRelayConn) ReadFrom(b []byte) (n int, addr net.Addr, err error) {
dgram := gosocks5.UDPDatagram{ dgram := gosocks5.UDPDatagram{
Header: &header, Header: &header,
} }
_, err = dgram.ReadFrom(bytes.NewReader((*buf)[:nn])) _, err = dgram.ReadFrom(bytes.NewReader(buf[:nn]))
if err != nil { if err != nil {
return return
} }
@ -81,15 +81,15 @@ func (c *udpRelayConn) WriteTo(b []byte, addr net.Addr) (n int, err error) {
buf := bufpool.Get(c.bufferSize) buf := bufpool.Get(c.bufferSize)
defer bufpool.Put(buf) defer bufpool.Put(buf)
nn, err := dgram.WriteTo(bytes.NewBuffer((*buf)[:0])) nn, err := dgram.WriteTo(bytes.NewBuffer(buf[:0]))
if err != nil { if err != nil {
return return
} }
if nn > int64(len(*buf)) { if nn > int64(len(buf)) {
nn = int64(len(*buf)) nn = int64(len(buf))
} }
_, err = c.udpConn.Write((*buf)[:nn]) _, err = c.udpConn.Write(buf[:nn])
n = len(b) n = len(b)
return return

View File

@ -5,6 +5,7 @@ import (
mdata "github.com/go-gost/core/metadata" mdata "github.com/go-gost/core/metadata"
mdutil "github.com/go-gost/core/metadata/util" mdutil "github.com/go-gost/core/metadata/util"
"github.com/go-gost/x/internal/util/mux"
) )
const ( const (
@ -16,6 +17,7 @@ type metadata struct {
noTLS bool noTLS bool
relay string relay string
udpBufferSize int udpBufferSize int
muxCfg *mux.Config
} }
func (c *socks5Connector) parseMetadata(md mdata.Metadata) (err error) { func (c *socks5Connector) parseMetadata(md mdata.Metadata) (err error) {
@ -34,5 +36,14 @@ func (c *socks5Connector) parseMetadata(md mdata.Metadata) (err error) {
c.md.udpBufferSize = defaultUDPBufferSize c.md.udpBufferSize = defaultUDPBufferSize
} }
c.md.muxCfg = &mux.Config{
Version: mdutil.GetInt(md, "mux.version"),
KeepAliveInterval: mdutil.GetDuration(md, "mux.keepaliveInterval"),
KeepAliveDisabled: mdutil.GetBool(md, "mux.keepaliveDisabled"),
KeepAliveTimeout: mdutil.GetDuration(md, "mux.keepaliveTimeout"),
MaxFrameSize: mdutil.GetInt(md, "mux.maxFrameSize"),
MaxReceiveBuffer: mdutil.GetInt(md, "mux.maxReceiveBuffer"),
MaxStreamBuffer: mdutil.GetInt(md, "mux.maxStreamBuffer"),
}
return return
} }

View File

@ -80,7 +80,7 @@ func (c *ssConnector) Connect(ctx context.Context, conn net.Conn, network, addre
rawaddr := bufpool.Get(512) rawaddr := bufpool.Get(512)
defer bufpool.Put(rawaddr) defer bufpool.Put(rawaddr)
n, err := addr.Encode(*rawaddr) n, err := addr.Encode(rawaddr)
if err != nil { if err != nil {
log.Error("encoding addr: ", err) log.Error("encoding addr: ", err)
return nil, err return nil, err
@ -99,12 +99,12 @@ func (c *ssConnector) Connect(ctx context.Context, conn net.Conn, network, addre
if c.md.noDelay { if c.md.noDelay {
sc = ss.ShadowConn(conn, nil) sc = ss.ShadowConn(conn, nil)
// write the addr at once. // write the addr at once.
if _, err := sc.Write((*rawaddr)[:n]); err != nil { if _, err := sc.Write(rawaddr[:n]); err != nil {
return nil, err return nil, err
} }
} else { } else {
// cache the header // cache the header
sc = ss.ShadowConn(conn, (*rawaddr)[:n]) sc = ss.ShadowConn(conn, rawaddr[:n])
} }
return sc, nil return sc, nil

View File

@ -75,8 +75,8 @@ func (c *udpConn) Read(b []byte) (n int, err error) {
buf := bufpool.Get(dlen) buf := bufpool.Get(dlen)
defer bufpool.Put(buf) defer bufpool.Put(buf)
_, err = io.ReadFull(c.Conn, *buf) _, err = io.ReadFull(c.Conn, buf)
n = copy(b, *buf) n = copy(b, buf)
return return
} }
@ -169,8 +169,8 @@ func (c *bindUDPConn) Read(b []byte) (n int, err error) {
buf := bufpool.Get(dlen) buf := bufpool.Get(dlen)
defer bufpool.Put(buf) defer bufpool.Put(buf)
_, err = io.ReadFull(c.Conn, *buf) _, err = io.ReadFull(c.Conn, buf)
n = copy(b, *buf) n = copy(b, buf)
return return
} }

View File

@ -11,10 +11,6 @@ import (
"github.com/google/uuid" "github.com/google/uuid"
) )
const (
defaultMuxVersion = 2
)
var ( var (
ErrInvalidTunnelID = errors.New("tunnel: invalid tunnel ID") ErrInvalidTunnelID = errors.New("tunnel: invalid tunnel ID")
) )
@ -51,9 +47,6 @@ func (c *tunnelConnector) parseMetadata(md mdata.Metadata) (err error) {
MaxReceiveBuffer: mdutil.GetInt(md, "mux.maxReceiveBuffer"), MaxReceiveBuffer: mdutil.GetInt(md, "mux.maxReceiveBuffer"),
MaxStreamBuffer: mdutil.GetInt(md, "mux.maxStreamBuffer"), MaxStreamBuffer: mdutil.GetInt(md, "mux.maxStreamBuffer"),
} }
if c.md.muxCfg.Version == 0 {
c.md.muxCfg.Version = defaultMuxVersion
}
return return
} }

View File

@ -141,17 +141,17 @@ func (h *dnsHandler) Handle(ctx context.Context, conn net.Conn, opts ...handler.
b := bufpool.Get(h.md.bufferSize) b := bufpool.Get(h.md.bufferSize)
defer bufpool.Put(b) defer bufpool.Put(b)
n, err := conn.Read(*b) n, err := conn.Read(b)
if err != nil { if err != nil {
log.Error(err) log.Error(err)
return err return err
} }
reply, err := h.request(ctx, (*b)[:n], log) reply, err := h.request(ctx, b[:n], log)
if err != nil { if err != nil {
return err return err
} }
defer bufpool.Put(&reply) defer bufpool.Put(reply)
if _, err = conn.Write(reply); err != nil { if _, err = conn.Write(reply); err != nil {
log.Error(err) log.Error(err)
@ -203,14 +203,14 @@ func (h *dnsHandler) request(ctx context.Context, msg []byte, log logger.Logger)
log.Debug("bypass: ", mq.Question[0].Name) log.Debug("bypass: ", mq.Question[0].Name)
mr = (&dns.Msg{}).SetReply(&mq) mr = (&dns.Msg{}).SetReply(&mq)
b := bufpool.Get(h.md.bufferSize) b := bufpool.Get(h.md.bufferSize)
return mr.PackBuffer(*b) return mr.PackBuffer(b)
} }
} }
mr = h.lookupHosts(ctx, &mq, log) mr = h.lookupHosts(ctx, &mq, log)
if mr != nil { if mr != nil {
b := bufpool.Get(h.md.bufferSize) b := bufpool.Get(h.md.bufferSize)
return mr.PackBuffer(*b) return mr.PackBuffer(b)
} }
// only cache for single question message. // only cache for single question message.
@ -222,14 +222,14 @@ func (h *dnsHandler) request(ctx context.Context, msg []byte, log logger.Logger)
if int32(ttl.Seconds()) > 0 { if int32(ttl.Seconds()) > 0 {
log.Debugf("message %d (cached): %s", mq.Id, mq.Question[0].String()) log.Debugf("message %d (cached): %s", mq.Id, mq.Question[0].String())
b := bufpool.Get(h.md.bufferSize) b := bufpool.Get(h.md.bufferSize)
return mr.PackBuffer(*b) return mr.PackBuffer(b)
} }
} }
} }
if mr != nil && h.md.async { if mr != nil && h.md.async {
b := bufpool.Get(h.md.bufferSize) b := bufpool.Get(h.md.bufferSize)
reply, err := mr.PackBuffer(*b) reply, err := mr.PackBuffer(b)
if err != nil { if err != nil {
return nil, err return nil, err
} }
@ -248,7 +248,7 @@ func (h *dnsHandler) exchange(ctx context.Context, mq *dns.Msg) ([]byte, error)
b := bufpool.Get(h.md.bufferSize) b := bufpool.Get(h.md.bufferSize)
defer bufpool.Put(b) defer bufpool.Put(b)
query, err := mq.PackBuffer(*b) query, err := mq.PackBuffer(b)
if err != nil { if err != nil {
return nil, err return nil, err
} }

View File

@ -70,7 +70,7 @@ func (h *socks5Handler) muxBindLocal(ctx context.Context, conn net.Conn, network
func (h *socks5Handler) serveMuxBind(ctx context.Context, conn net.Conn, ln net.Listener, log logger.Logger) error { func (h *socks5Handler) serveMuxBind(ctx context.Context, conn net.Conn, ln net.Listener, log logger.Logger) error {
// Upgrade connection to multiplex stream. // Upgrade connection to multiplex stream.
session, err := mux.ClientSession(conn, nil) session, err := mux.ClientSession(conn, h.md.muxCfg)
if err != nil { if err != nil {
log.Error(err) log.Error(err)
return err return err

View File

@ -6,6 +6,7 @@ import (
mdata "github.com/go-gost/core/metadata" mdata "github.com/go-gost/core/metadata"
mdutil "github.com/go-gost/core/metadata/util" mdutil "github.com/go-gost/core/metadata/util"
"github.com/go-gost/x/internal/util/mux"
) )
type metadata struct { type metadata struct {
@ -16,6 +17,7 @@ type metadata struct {
udpBufferSize int udpBufferSize int
compatibilityMode bool compatibilityMode bool
hash string hash string
muxCfg *mux.Config
} }
func (h *socks5Handler) parseMetadata(md mdata.Metadata) (err error) { func (h *socks5Handler) parseMetadata(md mdata.Metadata) (err error) {
@ -43,5 +45,15 @@ func (h *socks5Handler) parseMetadata(md mdata.Metadata) (err error) {
h.md.compatibilityMode = mdutil.GetBool(md, compatibilityMode) h.md.compatibilityMode = mdutil.GetBool(md, compatibilityMode)
h.md.hash = mdutil.GetString(md, hash) h.md.hash = mdutil.GetString(md, hash)
h.md.muxCfg = &mux.Config{
Version: mdutil.GetInt(md, "mux.version"),
KeepAliveInterval: mdutil.GetDuration(md, "mux.keepaliveInterval"),
KeepAliveDisabled: mdutil.GetBool(md, "mux.keepaliveDisabled"),
KeepAliveTimeout: mdutil.GetDuration(md, "mux.keepaliveTimeout"),
MaxFrameSize: mdutil.GetInt(md, "mux.maxFrameSize"),
MaxReceiveBuffer: mdutil.GetInt(md, "mux.maxReceiveBuffer"),
MaxStreamBuffer: mdutil.GetInt(md, "mux.maxStreamBuffer"),
}
return nil return nil
} }

View File

@ -130,7 +130,7 @@ func (h *ssuHandler) relayPacket(pc1, pc2 net.PacketConn, log logger.Logger) (er
b := bufpool.Get(bufSize) b := bufpool.Get(bufSize)
defer bufpool.Put(b) defer bufpool.Put(b)
n, addr, err := pc1.ReadFrom(*b) n, addr, err := pc1.ReadFrom(b)
if err != nil { if err != nil {
return err return err
} }
@ -140,7 +140,7 @@ func (h *ssuHandler) relayPacket(pc1, pc2 net.PacketConn, log logger.Logger) (er
return nil return nil
} }
if _, err = pc2.WriteTo((*b)[:n], addr); err != nil { if _, err = pc2.WriteTo(b[:n], addr); err != nil {
return err return err
} }
@ -162,7 +162,7 @@ func (h *ssuHandler) relayPacket(pc1, pc2 net.PacketConn, log logger.Logger) (er
b := bufpool.Get(bufSize) b := bufpool.Get(bufSize)
defer bufpool.Put(b) defer bufpool.Put(b)
n, raddr, err := pc2.ReadFrom(*b) n, raddr, err := pc2.ReadFrom(b)
if err != nil { if err != nil {
return err return err
} }
@ -172,7 +172,7 @@ func (h *ssuHandler) relayPacket(pc1, pc2 net.PacketConn, log logger.Logger) (er
return nil return nil
} }
if _, err = pc1.WriteTo((*b)[:n], raddr); err != nil { if _, err = pc1.WriteTo(b[:n], raddr); err != nil {
return err return err
} }

View File

@ -196,7 +196,7 @@ func (h *tapHandler) transport(tap net.Conn, conn net.PacketConn, raddr net.Addr
b := bufpool.Get(h.md.bufferSize) b := bufpool.Get(h.md.bufferSize)
defer bufpool.Put(b) defer bufpool.Put(b)
n, err := tap.Read(*b) n, err := tap.Read(b)
if err != nil { if err != nil {
select { select {
case h.exit <- struct{}{}: case h.exit <- struct{}{}:
@ -208,22 +208,22 @@ func (h *tapHandler) transport(tap net.Conn, conn net.PacketConn, raddr net.Addr
return nil return nil
} }
src := waterutil.MACSource((*b)[:n]) src := waterutil.MACSource(b[:n])
dst := waterutil.MACDestination((*b)[:n]) dst := waterutil.MACDestination(b[:n])
eType := etherType(waterutil.MACEthertype((*b)[:n])) eType := etherType(waterutil.MACEthertype(b[:n]))
log.Debugf("%s >> %s %s %d", src, dst, eType, n) log.Debugf("%s >> %s %s %d", src, dst, eType, n)
// client side, deliver frame directly. // client side, deliver frame directly.
if raddr != nil { if raddr != nil {
_, err := conn.WriteTo((*b)[:n], raddr) _, err := conn.WriteTo(b[:n], raddr)
return err return err
} }
// server side, broadcast. // server side, broadcast.
if waterutil.IsBroadcast(dst) { if waterutil.IsBroadcast(dst) {
go h.routes.Range(func(k, v any) bool { go h.routes.Range(func(k, v any) bool {
conn.WriteTo((*b)[:n], v.(net.Addr)) conn.WriteTo(b[:n], v.(net.Addr))
return true return true
}) })
return nil return nil
@ -238,7 +238,7 @@ func (h *tapHandler) transport(tap net.Conn, conn net.PacketConn, raddr net.Addr
return nil return nil
} }
if _, err := conn.WriteTo((*b)[:n], addr); err != nil { if _, err := conn.WriteTo(b[:n], addr); err != nil {
return err return err
} }
@ -258,7 +258,7 @@ func (h *tapHandler) transport(tap net.Conn, conn net.PacketConn, raddr net.Addr
b := bufpool.Get(h.md.bufferSize) b := bufpool.Get(h.md.bufferSize)
defer bufpool.Put(b) defer bufpool.Put(b)
n, addr, err := conn.ReadFrom(*b) n, addr, err := conn.ReadFrom(b)
if err != nil && if err != nil &&
err != shadowaead.ErrShortPacket { err != shadowaead.ErrShortPacket {
return err return err
@ -267,15 +267,15 @@ func (h *tapHandler) transport(tap net.Conn, conn net.PacketConn, raddr net.Addr
return nil return nil
} }
src := waterutil.MACSource((*b)[:n]) src := waterutil.MACSource(b[:n])
dst := waterutil.MACDestination((*b)[:n]) dst := waterutil.MACDestination(b[:n])
eType := etherType(waterutil.MACEthertype((*b)[:n])) eType := etherType(waterutil.MACEthertype(b[:n]))
log.Debugf("%s >> %s %s %d", src, dst, eType, n) log.Debugf("%s >> %s %s %d", src, dst, eType, n)
// client side, deliver frame to tap device. // client side, deliver frame to tap device.
if raddr != nil { if raddr != nil {
_, err := tap.Write((*b)[:n]) _, err := tap.Write(b[:n])
return err return err
} }
@ -294,7 +294,7 @@ func (h *tapHandler) transport(tap net.Conn, conn net.PacketConn, raddr net.Addr
if waterutil.IsBroadcast(dst) { if waterutil.IsBroadcast(dst) {
go h.routes.Range(func(k, v any) bool { go h.routes.Range(func(k, v any) bool {
if k.(tapRouteKey) != rkey { if k.(tapRouteKey) != rkey {
conn.WriteTo((*b)[:n], v.(net.Addr)) conn.WriteTo(b[:n], v.(net.Addr))
} }
return true return true
}) })
@ -302,11 +302,11 @@ func (h *tapHandler) transport(tap net.Conn, conn net.PacketConn, raddr net.Addr
if v, ok := h.routes.Load(hwAddrToTapRouteKey(dst)); ok { if v, ok := h.routes.Load(hwAddrToTapRouteKey(dst)); ok {
log.Debugf("find route: %s -> %s", dst, v) log.Debugf("find route: %s -> %s", dst, v)
_, err := conn.WriteTo((*b)[:n], v.(net.Addr)) _, err := conn.WriteTo(b[:n], v.(net.Addr))
return err return err
} }
if _, err := tap.Write((*b)[:n]); err != nil { if _, err := tap.Write(b[:n]); err != nil {
select { select {
case h.exit <- struct{}{}: case h.exit <- struct{}{}:
default: default:

View File

@ -62,14 +62,14 @@ func (h *tunHandler) keepAlive(ctx context.Context, conn net.Conn, ips []net.IP)
keepAliveData := bufpool.Get(keepAliveHeaderLength + len(ips)*net.IPv6len) keepAliveData := bufpool.Get(keepAliveHeaderLength + len(ips)*net.IPv6len)
defer bufpool.Put(keepAliveData) defer bufpool.Put(keepAliveData)
copy((*keepAliveData)[:4], magicHeader) // magic header copy(keepAliveData[:4], magicHeader) // magic header
copy((*keepAliveData)[4:20], []byte(h.md.passphrase)) copy(keepAliveData[4:20], []byte(h.md.passphrase))
pos := 20 pos := 20
for _, ip := range ips { for _, ip := range ips {
copy((*keepAliveData)[pos:pos+net.IPv6len], ip.To16()) copy(keepAliveData[pos:pos+net.IPv6len], ip.To16())
pos += net.IPv6len pos += net.IPv6len
} }
if _, err := conn.Write((*keepAliveData)); err != nil { if _, err := conn.Write(keepAliveData); err != nil {
return return
} }
@ -84,7 +84,7 @@ func (h *tunHandler) keepAlive(ctx context.Context, conn net.Conn, ips []net.IP)
for { for {
select { select {
case <-ticker.C: case <-ticker.C:
if _, err := conn.Write((*keepAliveData)); err != nil { if _, err := conn.Write(keepAliveData); err != nil {
return return
} }
h.options.Logger.Debugf("keepalive sended") h.options.Logger.Debugf("keepalive sended")
@ -103,23 +103,23 @@ func (h *tunHandler) transportClient(tun io.ReadWriter, conn net.Conn, log logge
b := bufpool.Get(h.md.bufferSize) b := bufpool.Get(h.md.bufferSize)
defer bufpool.Put(b) defer bufpool.Put(b)
n, err := tun.Read(*b) n, err := tun.Read(b)
if err != nil { if err != nil {
return ErrTun return ErrTun
} }
if waterutil.IsIPv4((*b)[:n]) { if waterutil.IsIPv4(b[:n]) {
header, err := ipv4.ParseHeader((*b)[:n]) header, err := ipv4.ParseHeader(b[:n])
if err != nil { if err != nil {
log.Warn(err) log.Warn(err)
return nil return nil
} }
log.Tracef("%s >> %s %-4s %d/%-4d %-4x %d", log.Tracef("%s >> %s %-4s %d/%-4d %-4x %d",
header.Src, header.Dst, ipProtocol(waterutil.IPv4Protocol((*b)[:n])), header.Src, header.Dst, ipProtocol(waterutil.IPv4Protocol(b[:n])),
header.Len, header.TotalLen, header.ID, header.Flags) header.Len, header.TotalLen, header.ID, header.Flags)
} else if waterutil.IsIPv6((*b)[:n]) { } else if waterutil.IsIPv6(b[:n]) {
header, err := ipv6.ParseHeader((*b)[:n]) header, err := ipv6.ParseHeader(b[:n])
if err != nil { if err != nil {
log.Warn(err) log.Warn(err)
return nil return nil
@ -134,7 +134,7 @@ func (h *tunHandler) transportClient(tun io.ReadWriter, conn net.Conn, log logge
return nil return nil
} }
_, err = conn.Write((*b)[:n]) _, err = conn.Write(b[:n])
return err return err
}() }()
@ -151,13 +151,13 @@ func (h *tunHandler) transportClient(tun io.ReadWriter, conn net.Conn, log logge
b := bufpool.Get(h.md.bufferSize) b := bufpool.Get(h.md.bufferSize)
defer bufpool.Put(b) defer bufpool.Put(b)
n, err := conn.Read(*b) n, err := conn.Read(b)
if err != nil { if err != nil {
return err return err
} }
if n == keepAliveHeaderLength && bytes.Equal((*b)[:4], magicHeader) { if n == keepAliveHeaderLength && bytes.Equal(b[:4], magicHeader) {
ip := net.IP((*b)[4:20]) ip := net.IP(b[4:20])
log.Debugf("keepalive received at %v", ip) log.Debugf("keepalive received at %v", ip)
if h.md.keepAlivePeriod > 0 { if h.md.keepAlivePeriod > 0 {
@ -166,18 +166,18 @@ func (h *tunHandler) transportClient(tun io.ReadWriter, conn net.Conn, log logge
return nil return nil
} }
if waterutil.IsIPv4((*b)[:n]) { if waterutil.IsIPv4(b[:n]) {
header, err := ipv4.ParseHeader((*b)[:n]) header, err := ipv4.ParseHeader(b[:n])
if err != nil { if err != nil {
log.Warn(err) log.Warn(err)
return nil return nil
} }
log.Tracef("%s >> %s %-4s %d/%-4d %-4x %d", log.Tracef("%s >> %s %-4s %d/%-4d %-4x %d",
header.Src, header.Dst, ipProtocol(waterutil.IPv4Protocol((*b)[:n])), header.Src, header.Dst, ipProtocol(waterutil.IPv4Protocol(b[:n])),
header.Len, header.TotalLen, header.ID, header.Flags) header.Len, header.TotalLen, header.ID, header.Flags)
} else if waterutil.IsIPv6((*b)[:n]) { } else if waterutil.IsIPv6(b[:n]) {
header, err := ipv6.ParseHeader((*b)[:n]) header, err := ipv6.ParseHeader(b[:n])
if err != nil { if err != nil {
log.Warn(err) log.Warn(err)
return nil return nil
@ -192,7 +192,7 @@ func (h *tunHandler) transportClient(tun io.ReadWriter, conn net.Conn, log logge
return nil return nil
} }
if _, err = tun.Write((*b)[:n]); err != nil { if _, err = tun.Write(b[:n]); err != nil {
return ErrTun return ErrTun
} }
return nil return nil

View File

@ -45,7 +45,7 @@ func (h *tunHandler) transportServer(ctx context.Context, tun io.ReadWriter, con
b := bufpool.Get(h.md.bufferSize) b := bufpool.Get(h.md.bufferSize)
defer bufpool.Put(b) defer bufpool.Put(b)
n, err := tun.Read(*b) n, err := tun.Read(b)
if err != nil { if err != nil {
return ErrTun return ErrTun
} }
@ -54,8 +54,8 @@ func (h *tunHandler) transportServer(ctx context.Context, tun io.ReadWriter, con
} }
var src, dst net.IP var src, dst net.IP
if waterutil.IsIPv4((*b)[:n]) { if waterutil.IsIPv4(b[:n]) {
header, err := ipv4.ParseHeader((*b)[:n]) header, err := ipv4.ParseHeader(b[:n])
if err != nil { if err != nil {
log.Warnf("parse ipv4 packet header: %v", err) log.Warnf("parse ipv4 packet header: %v", err)
return nil return nil
@ -63,10 +63,10 @@ func (h *tunHandler) transportServer(ctx context.Context, tun io.ReadWriter, con
src, dst = header.Src, header.Dst src, dst = header.Src, header.Dst
log.Tracef("%s >> %s %-4s %d/%-4d %-4x %d", log.Tracef("%s >> %s %-4s %d/%-4d %-4x %d",
src, dst, ipProtocol(waterutil.IPv4Protocol((*b)[:n])), src, dst, ipProtocol(waterutil.IPv4Protocol(b[:n])),
header.Len, header.TotalLen, header.ID, header.Flags) header.Len, header.TotalLen, header.ID, header.Flags)
} else if waterutil.IsIPv6((*b)[:n]) { } else if waterutil.IsIPv6(b[:n]) {
header, err := ipv6.ParseHeader((*b)[:n]) header, err := ipv6.ParseHeader(b[:n])
if err != nil { if err != nil {
log.Warnf("parse ipv6 packet header: %v", err) log.Warnf("parse ipv6 packet header: %v", err)
return nil return nil
@ -90,7 +90,7 @@ func (h *tunHandler) transportServer(ctx context.Context, tun io.ReadWriter, con
log.Debugf("find route: %s -> %s", dst, addr) log.Debugf("find route: %s -> %s", dst, addr)
if _, err := conn.WriteTo((*b)[:n], addr); err != nil { if _, err := conn.WriteTo(b[:n], addr); err != nil {
return err return err
} }
return nil return nil
@ -109,16 +109,16 @@ func (h *tunHandler) transportServer(ctx context.Context, tun io.ReadWriter, con
b := bufpool.Get(h.md.bufferSize) b := bufpool.Get(h.md.bufferSize)
defer bufpool.Put(b) defer bufpool.Put(b)
n, addr, err := conn.ReadFrom(*b) n, addr, err := conn.ReadFrom(b)
if err != nil { if err != nil {
return err return err
} }
if n == 0 { if n == 0 {
return nil return nil
} }
if n > keepAliveHeaderLength && bytes.Equal((*b)[:4], magicHeader) { if n > keepAliveHeaderLength && bytes.Equal(b[:4], magicHeader) {
var peerIPs []net.IP var peerIPs []net.IP
data := (*b)[keepAliveHeaderLength:n] data := b[keepAliveHeaderLength:n]
if len(data)%net.IPv6len == 0 { if len(data)%net.IPv6len == 0 {
for len(data) > 0 { for len(data) > 0 {
peerIPs = append(peerIPs, net.IP(data[:net.IPv6len])) peerIPs = append(peerIPs, net.IP(data[:net.IPv6len]))
@ -139,7 +139,7 @@ func (h *tunHandler) transportServer(ctx context.Context, tun io.ReadWriter, con
if auther := h.options.Auther; auther != nil { if auther := h.options.Auther; auther != nil {
ok := true ok := true
key := bytes.TrimRight((*b)[4:20], "\x00") key := bytes.TrimRight(b[4:20], "\x00")
for _, ip := range peerIPs { for _, ip := range peerIPs {
if _, ok = auther.Authenticate(ctx, ip.String(), string(key)); !ok { if _, ok = auther.Authenticate(ctx, ip.String(), string(key)); !ok {
break break
@ -175,8 +175,8 @@ func (h *tunHandler) transportServer(ctx context.Context, tun io.ReadWriter, con
} }
var src, dst net.IP var src, dst net.IP
if waterutil.IsIPv4((*b)[:n]) { if waterutil.IsIPv4(b[:n]) {
header, err := ipv4.ParseHeader((*b)[:n]) header, err := ipv4.ParseHeader(b[:n])
if err != nil { if err != nil {
log.Warnf("parse ipv4 packet header: %v", err) log.Warnf("parse ipv4 packet header: %v", err)
return nil return nil
@ -184,10 +184,10 @@ func (h *tunHandler) transportServer(ctx context.Context, tun io.ReadWriter, con
src, dst = header.Src, header.Dst src, dst = header.Src, header.Dst
log.Tracef("%s >> %s %-4s %d/%-4d %-4x %d", log.Tracef("%s >> %s %-4s %d/%-4d %-4x %d",
src, dst, ipProtocol(waterutil.IPv4Protocol((*b)[:n])), src, dst, ipProtocol(waterutil.IPv4Protocol(b[:n])),
header.Len, header.TotalLen, header.ID, header.Flags) header.Len, header.TotalLen, header.ID, header.Flags)
} else if waterutil.IsIPv6((*b)[:n]) { } else if waterutil.IsIPv6(b[:n]) {
header, err := ipv6.ParseHeader((*b)[:n]) header, err := ipv6.ParseHeader(b[:n])
if err != nil { if err != nil {
log.Warnf("parse ipv6 packet header: %v", err) log.Warnf("parse ipv6 packet header: %v", err)
return nil return nil
@ -206,11 +206,11 @@ func (h *tunHandler) transportServer(ctx context.Context, tun io.ReadWriter, con
if addr := h.findRouteFor(dst, config.Routes...); addr != nil { if addr := h.findRouteFor(dst, config.Routes...); addr != nil {
log.Debugf("find route: %s -> %s", dst, addr) log.Debugf("find route: %s -> %s", dst, addr)
_, err := conn.WriteTo((*b)[:n], addr) _, err := conn.WriteTo(b[:n], addr)
return err return err
} }
if _, err := tun.Write((*b)[:n]); err != nil { if _, err := tun.Write(b[:n]); err != nil {
return ErrTun return ErrTun
} }
return nil return nil

View File

@ -14,10 +14,6 @@ import (
"github.com/go-gost/x/registry" "github.com/go-gost/x/registry"
) )
const (
defaultMuxVersion = 2
)
type metadata struct { type metadata struct {
readTimeout time.Duration readTimeout time.Duration
noDelay bool noDelay bool
@ -69,9 +65,6 @@ func (h *tunnelHandler) parseMetadata(md mdata.Metadata) (err error) {
MaxReceiveBuffer: mdutil.GetInt(md, "mux.maxReceiveBuffer"), MaxReceiveBuffer: mdutil.GetInt(md, "mux.maxReceiveBuffer"),
MaxStreamBuffer: mdutil.GetInt(md, "mux.maxStreamBuffer"), MaxStreamBuffer: mdutil.GetInt(md, "mux.maxStreamBuffer"),
} }
if h.md.muxCfg.Version == 0 {
h.md.muxCfg.Version = defaultMuxVersion
}
return return
} }

View File

@ -33,7 +33,7 @@ func CopyBuffer(dst io.Writer, src io.Reader, bufSize int) error {
buf := bufpool.Get(bufSize) buf := bufpool.Get(bufSize)
defer bufpool.Put(buf) defer bufpool.Put(buf)
_, err := io.CopyBuffer(dst, src, *buf) _, err := io.CopyBuffer(dst, src, buf)
return err return err
} }

View File

@ -53,7 +53,7 @@ func (r *Relay) Run() (err error) {
b := bufpool.Get(bufSize) b := bufpool.Get(bufSize)
defer bufpool.Put(b) defer bufpool.Put(b)
n, raddr, err := r.pc1.ReadFrom(*b) n, raddr, err := r.pc1.ReadFrom(b)
if err != nil { if err != nil {
return err return err
} }
@ -65,7 +65,7 @@ func (r *Relay) Run() (err error) {
return nil return nil
} }
if _, err := r.pc2.WriteTo((*b)[:n], raddr); err != nil { if _, err := r.pc2.WriteTo(b[:n], raddr); err != nil {
return err return err
} }
@ -91,7 +91,7 @@ func (r *Relay) Run() (err error) {
b := bufpool.Get(bufSize) b := bufpool.Get(bufSize)
defer bufpool.Put(b) defer bufpool.Put(b)
n, raddr, err := r.pc2.ReadFrom(*b) n, raddr, err := r.pc2.ReadFrom(b)
if err != nil { if err != nil {
return err return err
} }
@ -103,7 +103,7 @@ func (r *Relay) Run() (err error) {
return nil return nil
} }
if _, err := r.pc1.WriteTo((*b)[:n], raddr); err != nil { if _, err := r.pc1.WriteTo(b[:n], raddr); err != nil {
return err return err
} }

View File

@ -39,13 +39,13 @@ func (c *dtlsConn) Read(p []byte) (n int, err error) {
buf := bufpool.Get(bufferSize) buf := bufpool.Get(bufferSize)
defer bufpool.Put(buf) defer bufpool.Put(buf)
nn, err := c.Conn.Read(*buf) nn, err := c.Conn.Read(buf)
if err != nil { if err != nil {
return 0, err return 0, err
} }
n = copy(p, (*buf)[:nn]) n = copy(p, buf[:nn])
c.rbuf.Write((*buf)[n:nn]) c.rbuf.Write(buf[n:nn])
return return
} }

View File

@ -96,12 +96,12 @@ func (c *clientConn) ReadFrom(b []byte) (n int, addr net.Addr, err error) {
defer bufpool.Put(buf) defer bufpool.Put(buf)
for { for {
n, addr, err = c.PacketConn.ReadFrom(*buf) n, addr, err = c.PacketConn.ReadFrom(buf)
if err != nil { if err != nil {
return return
} }
m, err := icmp.ParseMessage(1, (*buf)[:n]) m, err := icmp.ParseMessage(1, buf[:n])
if err != nil { if err != nil {
// logger.Default().Error("icmp: parse message %v", err) // logger.Default().Error("icmp: parse message %v", err)
return 0, addr, err return 0, addr, err
@ -155,7 +155,7 @@ func (c *clientConn) WriteTo(b []byte, addr net.Addr) (n int, err error) {
msg := message{ msg := message{
data: b, data: b,
} }
nn, err := msg.Encode(*buf) nn, err := msg.Encode(buf)
if err != nil { if err != nil {
return return
} }
@ -163,7 +163,7 @@ func (c *clientConn) WriteTo(b []byte, addr net.Addr) (n int, err error) {
echo := icmp.Echo{ echo := icmp.Echo{
ID: c.id, ID: c.id,
Seq: int(atomic.AddUint32(&c.seq, 1)), Seq: int(atomic.AddUint32(&c.seq, 1)),
Data: (*buf)[:nn], Data: buf[:nn],
} }
m := icmp.Message{ m := icmp.Message{
Type: ipv4.ICMPTypeEcho, Type: ipv4.ICMPTypeEcho,
@ -195,12 +195,12 @@ func (c *serverConn) ReadFrom(b []byte) (n int, addr net.Addr, err error) {
defer bufpool.Put(buf) defer bufpool.Put(buf)
for { for {
n, addr, err = c.PacketConn.ReadFrom(*buf) n, addr, err = c.PacketConn.ReadFrom(buf)
if err != nil { if err != nil {
return return
} }
m, err := icmp.ParseMessage(1, (*buf)[:n]) m, err := icmp.ParseMessage(1, buf[:n])
if err != nil { if err != nil {
// logger.Default().Error("icmp: parse message %v", err) // logger.Default().Error("icmp: parse message %v", err)
return 0, addr, err return 0, addr, err
@ -260,7 +260,7 @@ func (c *serverConn) WriteTo(b []byte, addr net.Addr) (n int, err error) {
flags: FlagAck, flags: FlagAck,
data: b, data: b,
} }
nn, err := msg.Encode(*buf) nn, err := msg.Encode(buf)
if err != nil { if err != nil {
return return
} }
@ -268,7 +268,7 @@ func (c *serverConn) WriteTo(b []byte, addr net.Addr) (n int, err error) {
echo := icmp.Echo{ echo := icmp.Echo{
ID: id, ID: id,
Seq: int(atomic.LoadUint32(&c.seqs[id-1])), Seq: int(atomic.LoadUint32(&c.seqs[id-1])),
Data: (*buf)[:nn], Data: buf[:nn],
} }
m := icmp.Message{ m := icmp.Message{
Type: ipv4.ICMPTypeEchoReply, Type: ipv4.ICMPTypeEchoReply,

View File

@ -7,6 +7,10 @@ import (
smux "github.com/xtaci/smux" smux "github.com/xtaci/smux"
) )
const (
defaultVersion = 2
)
type Config struct { type Config struct {
// SMUX Protocol version, support 1,2 // SMUX Protocol version, support 1,2
Version int Version int
@ -36,6 +40,8 @@ type Config struct {
func convertConfig(cfg *Config) *smux.Config { func convertConfig(cfg *Config) *smux.Config {
smuxCfg := smux.DefaultConfig() smuxCfg := smux.DefaultConfig()
smuxCfg.Version = defaultVersion
if cfg == nil { if cfg == nil {
return smuxCfg return smuxCfg
} }

View File

@ -375,10 +375,10 @@ func (s *Server) handlePull(w http.ResponseWriter, r *http.Request) {
for { for {
conn.SetReadDeadline(time.Now().Add(s.options.readTimeout)) conn.SetReadDeadline(time.Now().Add(s.options.readTimeout))
n, err := conn.Read(*b) n, err := conn.Read(b)
if n > 0 { if n > 0 {
bw := bufio.NewWriter(w) bw := bufio.NewWriter(w)
bw.WriteString(base64.StdEncoding.EncodeToString((*b)[:n])) bw.WriteString(base64.StdEncoding.EncodeToString(b[:n]))
bw.WriteString("\n") bw.WriteString("\n")
if err := bw.Flush(); err != nil { if err := bw.Flush(); err != nil {
return return
@ -389,8 +389,8 @@ func (s *Server) handlePull(w http.ResponseWriter, r *http.Request) {
} }
if err != nil { if err != nil {
if errors.Is(err, os.ErrDeadlineExceeded) { if errors.Is(err, os.ErrDeadlineExceeded) {
(*b)[0] = '\n' // no data b[0] = '\n' // no data
w.Write((*b)[:1]) w.Write(b[:1])
} else if errors.Is(err, io.EOF) { } else if errors.Is(err, io.EOF) {
// server connection closed // server connection closed
} else { } else {

View File

@ -110,7 +110,7 @@ func (c *udpConn) ReadFrom(b []byte) (n int, addr net.Addr, err error) {
rbuf := bufpool.Get(c.bufferSize) rbuf := bufpool.Get(c.bufferSize)
defer bufpool.Put(rbuf) defer bufpool.Put(rbuf)
n, c.raddr, err = c.PacketConn.ReadFrom(*rbuf) n, c.raddr, err = c.PacketConn.ReadFrom(rbuf)
if err != nil { if err != nil {
return return
} }
@ -119,11 +119,11 @@ func (c *udpConn) ReadFrom(b []byte) (n int, addr net.Addr, err error) {
header := gosocks5.UDPHeader{ header := gosocks5.UDPHeader{
Addr: &socksAddr, Addr: &socksAddr,
} }
hlen, err := header.ReadFrom(bytes.NewReader((*rbuf)[:n])) hlen, err := header.ReadFrom(bytes.NewReader(rbuf[:n]))
if err != nil { if err != nil {
return return
} }
n = copy(b, (*rbuf)[hlen:n]) n = copy(b, rbuf[hlen:n])
addr, err = net.ResolveUDPAddr("udp", socksAddr.String()) addr, err = net.ResolveUDPAddr("udp", socksAddr.String())
return return
@ -151,7 +151,7 @@ func (c *udpConn) WriteTo(b []byte, addr net.Addr) (n int, err error) {
Data: b, Data: b,
} }
buf := bytes.NewBuffer((*wbuf)[:0]) buf := bytes.NewBuffer(wbuf[:0])
_, err = dgram.WriteTo(buf) _, err = dgram.WriteTo(buf)
if err != nil { if err != nil {
return return

View File

@ -110,7 +110,7 @@ func (c *udpConn) ReadFrom(b []byte) (n int, addr net.Addr, err error) {
rbuf := bufpool.Get(c.bufferSize) rbuf := bufpool.Get(c.bufferSize)
defer bufpool.Put(rbuf) defer bufpool.Put(rbuf)
n, c.raddr, err = c.PacketConn.ReadFrom(*rbuf) n, c.raddr, err = c.PacketConn.ReadFrom(rbuf)
if err != nil { if err != nil {
return return
} }
@ -119,11 +119,11 @@ func (c *udpConn) ReadFrom(b []byte) (n int, addr net.Addr, err error) {
header := gosocks5.UDPHeader{ header := gosocks5.UDPHeader{
Addr: &socksAddr, Addr: &socksAddr,
} }
hlen, err := header.ReadFrom(bytes.NewReader((*rbuf)[:n])) hlen, err := header.ReadFrom(bytes.NewReader(rbuf[:n]))
if err != nil { if err != nil {
return return
} }
n = copy(b, (*rbuf)[hlen:n]) n = copy(b, rbuf[hlen:n])
addr, err = net.ResolveUDPAddr("udp", socksAddr.String()) addr, err = net.ResolveUDPAddr("udp", socksAddr.String())
return return
@ -151,7 +151,7 @@ func (c *udpConn) WriteTo(b []byte, addr net.Addr) (n int, err error) {
Data: b, Data: b,
} }
buf := bytes.NewBuffer((*wbuf)[:0]) buf := bytes.NewBuffer(wbuf[:0])
_, err = dgram.WriteTo(buf) _, err = dgram.WriteTo(buf)
if err != nil { if err != nil {
return return

View File

@ -45,18 +45,18 @@ func (c *UDPConn) ReadFrom(b []byte) (n int, addr net.Addr, err error) {
rbuf := bufpool.Get(c.bufferSize) rbuf := bufpool.Get(c.bufferSize)
defer bufpool.Put(rbuf) defer bufpool.Put(rbuf)
n, _, err = c.PacketConn.ReadFrom(*rbuf) n, _, err = c.PacketConn.ReadFrom(rbuf)
if err != nil { if err != nil {
return return
} }
saddr := gosocks5.Addr{} saddr := gosocks5.Addr{}
addrLen, err := saddr.ReadFrom(bytes.NewReader((*rbuf)[:n])) addrLen, err := saddr.ReadFrom(bytes.NewReader(rbuf[:n]))
if err != nil { if err != nil {
return return
} }
n = copy(b, (*rbuf)[addrLen:n]) n = copy(b, rbuf[addrLen:n])
addr, err = net.ResolveUDPAddr("udp", saddr.String()) addr, err = net.ResolveUDPAddr("udp", saddr.String())
return return
@ -76,13 +76,13 @@ func (c *UDPConn) WriteTo(b []byte, addr net.Addr) (n int, err error) {
return return
} }
addrLen, err := socksAddr.Encode(*wbuf) addrLen, err := socksAddr.Encode(wbuf)
if err != nil { if err != nil {
return return
} }
n = copy((*wbuf)[addrLen:], b) n = copy(wbuf[addrLen:], b)
_, err = c.PacketConn.WriteTo((*wbuf)[:addrLen+n], c.raddr) _, err = c.PacketConn.WriteTo(wbuf[:addrLen+n], c.raddr)
return return
} }

View File

@ -41,13 +41,6 @@ func (l *mwsListener) parseMetadata(md mdata.Metadata) (err error) {
readBufferSize = "readBufferSize" readBufferSize = "readBufferSize"
writeBufferSize = "writeBufferSize" writeBufferSize = "writeBufferSize"
enableCompression = "enableCompression" enableCompression = "enableCompression"
muxKeepAliveDisabled = "muxKeepAliveDisabled"
muxKeepAliveInterval = "muxKeepAliveInterval"
muxKeepAliveTimeout = "muxKeepAliveTimeout"
muxMaxFrameSize = "muxMaxFrameSize"
muxMaxReceiveBuffer = "muxMaxReceiveBuffer"
muxMaxStreamBuffer = "muxMaxStreamBuffer"
) )
l.md.path = mdutil.GetString(md, path) l.md.path = mdutil.GetString(md, path)

View File

@ -23,7 +23,7 @@ func (c *redirConn) Read(b []byte) (n int, err error) {
c.once.Do(func() { c.once.Do(func() {
n = copy(b, c.buf) n = copy(b, c.buf)
bufpool.Put(&c.buf) bufpool.Put(c.buf)
}) })
if n == 0 { if n == 0 {

View File

@ -45,7 +45,7 @@ func (l *redirectListener) listenUDP(addr string) (*net.UDPConn, error) {
func (l *redirectListener) accept() (conn net.Conn, err error) { func (l *redirectListener) accept() (conn net.Conn, err error) {
b := bufpool.Get(l.md.readBufferSize) b := bufpool.Get(l.md.readBufferSize)
n, raddr, dstAddr, err := readFromUDP(l.ln, *b) n, raddr, dstAddr, err := readFromUDP(l.ln, b)
if err != nil { if err != nil {
l.logger.Error(err) l.logger.Error(err)
return return
@ -65,7 +65,7 @@ func (l *redirectListener) accept() (conn net.Conn, err error) {
conn = &redirConn{ conn = &redirConn{
Conn: c, Conn: c,
buf: (*b)[:n], buf: b[:n],
ttl: l.md.ttl, ttl: l.md.ttl,
} }
return return
@ -81,12 +81,12 @@ func readFromUDP(conn *net.UDPConn, b []byte) (n int, remoteAddr *net.UDPAddr, d
oob := bufpool.Get(1024) oob := bufpool.Get(1024)
defer bufpool.Put(oob) defer bufpool.Put(oob)
n, oobn, _, remoteAddr, err := conn.ReadMsgUDP(b, *oob) n, oobn, _, remoteAddr, err := conn.ReadMsgUDP(b, oob)
if err != nil { if err != nil {
return 0, nil, nil, err return 0, nil, nil, err
} }
msgs, err := unix.ParseSocketControlMessage((*oob)[:oobn]) msgs, err := unix.ParseSocketControlMessage(oob[:oobn])
if err != nil { if err != nil {
return 0, nil, nil, fmt.Errorf("parsing socket control message: %s", err) return 0, nil, nil, fmt.Errorf("parsing socket control message: %s", err)
} }

View File

@ -24,7 +24,7 @@ func (d *tunDevice) Read(p []byte) (n int, err error) {
b := bufpool.Get(rbuf) b := bufpool.Get(rbuf)
defer bufpool.Put(b) defer bufpool.Put(b)
n, err = d.dev.Read(*b, tunOffsetBytes) n, err = d.dev.Read(b, tunOffsetBytes)
if n <= tunOffsetBytes || err != nil { if n <= tunOffsetBytes || err != nil {
d.dev.Flush() d.dev.Flush()
if n <= tunOffsetBytes { if n <= tunOffsetBytes {
@ -33,7 +33,7 @@ func (d *tunDevice) Read(p []byte) (n int, err error) {
return return
} }
n = copy(p, (*b)[tunOffsetBytes:tunOffsetBytes+n]) n = copy(p, b[tunOffsetBytes:tunOffsetBytes+n])
return return
} }
@ -41,8 +41,8 @@ func (d *tunDevice) Write(p []byte) (n int, err error) {
b := bufpool.Get(tunOffsetBytes + len(p)) b := bufpool.Get(tunOffsetBytes + len(p))
defer bufpool.Put(b) defer bufpool.Put(b)
copy((*b)[tunOffsetBytes:], p) copy(b[tunOffsetBytes:], p)
return d.dev.Write(*b, tunOffsetBytes) return d.dev.Write(b, tunOffsetBytes)
} }
func (d *tunDevice) Close() error { func (d *tunDevice) Close() error {