update udp listener

This commit is contained in:
ginuerzh
2022-04-03 22:24:47 +08:00
parent eef1b9176f
commit 1c69832f5a
4 changed files with 27 additions and 21 deletions

View File

@ -114,7 +114,7 @@ func (h *dnsHandler) Handle(ctx context.Context, conn net.Conn, opts ...handler.
}).Infof("%s >< %s", conn.RemoteAddr(), conn.LocalAddr())
}()
b := bufpool.Get(4096)
b := bufpool.Get(defaultBufferSize)
defer bufpool.Put(b)
n, err := conn.Read(*b)
@ -165,7 +165,7 @@ func (h *dnsHandler) exchange(ctx context.Context, msg []byte, log logger.Logger
mr = h.lookupHosts(&mq, log)
if mr != nil {
b := bufpool.Get(4096)
b := bufpool.Get(defaultBufferSize)
return mr.PackBuffer(*b)
}
@ -177,7 +177,7 @@ func (h *dnsHandler) exchange(ctx context.Context, msg []byte, log logger.Logger
log.Debugf("exchange message %d (cached): %s", mq.Id, mq.Question[0].String())
mr.Id = mq.Id
b := bufpool.Get(4096)
b := bufpool.Get(defaultBufferSize)
return mr.PackBuffer(*b)
}
@ -188,7 +188,7 @@ func (h *dnsHandler) exchange(ctx context.Context, msg []byte, log logger.Logger
}()
}
b := bufpool.Get(4096)
b := bufpool.Get(defaultBufferSize)
defer bufpool.Put(b)
query, err := mq.PackBuffer(*b)

View File

@ -7,6 +7,10 @@ import (
mdata "github.com/go-gost/core/metadata"
)
const (
defaultBufferSize = 1024
)
type metadata struct {
readTimeout time.Duration
ttl time.Duration