community
/
dns
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge branch 'majek/packwithbuffer' of https://github.com/majek/dns into bufpack

This commit is contained in:
Miek Gieben 2014-01-12 10:17:35 +00:00
parent be9467466c d18d87b37d
commit 7f8964662e
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
msg.go
View File

@ -1290,6 +1290,11 @@ func (h *MsgHdr) String() string {
// Pack packs a Msg: it is converted to to wire format.
// If the dns.Compress is true the message will be in compressed wire format.
func (dns *Msg) Pack() (msg []byte, err error) {
return dns.PackBuffer(nil)
}
// PackWithBuffer packs a Msg, reusing a given buffer if possible to reduce memory allocations
func (dns *Msg) PackBuffer(buf []byte) (msg []byte, err error) {
var dh Header
var compression map[string]int
if dns.Compress {
@ -1335,7 +1340,11 @@ func (dns *Msg) Pack() (msg []byte, err error) {
dh.Nscount = uint16(len(ns))
dh.Arcount = uint16(len(extra))
msg = make([]byte, dns.packLen()+1)
msg = buf
if packLen := dns.packLen(); len(msg) <= packLen {
msg = make([]byte, packLen+1)
}
// Pack it in: header and then the pieces.
off := 0
off, err = packStructCompress(&dh, msg, off, compression, dns.Compress)