ab9dd18bec
See https://github.com/hashicorp/consul/issues/3977 While trying to improve furthermore #3948 (This pull request is still valid since we are not using Compression to compute the result anyway). I saw a strange behaviour of dns library. Basically, msg.Len() and len(msg.Pack()) disagree on Message len. Thus, calculation of DNS response is false consul relies on msg.Len() instead of the result of Pack() This is linked to miekg/dns#453 and a fix has been provided with miekg/dns#454 Would it be possible to upgrade miekg/dns to a more recent function ? Consul might for instance upgrade to a post 1.0 release such as https://github.com/miekg/dns/releases/tag/v1.0.4
120 lines
3.3 KiB
Go
120 lines
3.3 KiB
Go
// Copyright 2017 The Go Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
// +build go1.9
|
|
|
|
package ipv6
|
|
|
|
import (
|
|
"net"
|
|
"runtime"
|
|
"syscall"
|
|
|
|
"golang.org/x/net/internal/socket"
|
|
)
|
|
|
|
// BUG(mikio): On Windows, the ReadBatch and WriteBatch methods of
|
|
// PacketConn are not implemented.
|
|
|
|
// A Message represents an IO message.
|
|
//
|
|
// type Message struct {
|
|
// Buffers [][]byte
|
|
// OOB []byte
|
|
// Addr net.Addr
|
|
// N int
|
|
// NN int
|
|
// Flags int
|
|
// }
|
|
//
|
|
// The Buffers fields represents a list of contiguous buffers, which
|
|
// can be used for vectored IO, for example, putting a header and a
|
|
// payload in each slice.
|
|
// When writing, the Buffers field must contain at least one byte to
|
|
// write.
|
|
// When reading, the Buffers field will always contain a byte to read.
|
|
//
|
|
// The OOB field contains protocol-specific control or miscellaneous
|
|
// ancillary data known as out-of-band data.
|
|
// It can be nil when not required.
|
|
//
|
|
// The Addr field specifies a destination address when writing.
|
|
// It can be nil when the underlying protocol of the endpoint uses
|
|
// connection-oriented communication.
|
|
// After a successful read, it may contain the source address on the
|
|
// received packet.
|
|
//
|
|
// The N field indicates the number of bytes read or written from/to
|
|
// Buffers.
|
|
//
|
|
// The NN field indicates the number of bytes read or written from/to
|
|
// OOB.
|
|
//
|
|
// The Flags field contains protocol-specific information on the
|
|
// received message.
|
|
type Message = socket.Message
|
|
|
|
// ReadBatch reads a batch of messages.
|
|
//
|
|
// The provided flags is a set of platform-dependent flags, such as
|
|
// syscall.MSG_PEEK.
|
|
//
|
|
// On a successful read it returns the number of messages received, up
|
|
// to len(ms).
|
|
//
|
|
// On Linux, a batch read will be optimized.
|
|
// On other platforms, this method will read only a single message.
|
|
func (c *payloadHandler) ReadBatch(ms []Message, flags int) (int, error) {
|
|
if !c.ok() {
|
|
return 0, syscall.EINVAL
|
|
}
|
|
switch runtime.GOOS {
|
|
case "linux":
|
|
n, err := c.RecvMsgs([]socket.Message(ms), flags)
|
|
if err != nil {
|
|
err = &net.OpError{Op: "read", Net: c.PacketConn.LocalAddr().Network(), Source: c.PacketConn.LocalAddr(), Err: err}
|
|
}
|
|
return n, err
|
|
default:
|
|
n := 1
|
|
err := c.RecvMsg(&ms[0], flags)
|
|
if err != nil {
|
|
n = 0
|
|
err = &net.OpError{Op: "read", Net: c.PacketConn.LocalAddr().Network(), Source: c.PacketConn.LocalAddr(), Err: err}
|
|
}
|
|
return n, err
|
|
}
|
|
}
|
|
|
|
// WriteBatch writes a batch of messages.
|
|
//
|
|
// The provided flags is a set of platform-dependent flags, such as
|
|
// syscall.MSG_DONTROUTE.
|
|
//
|
|
// It returns the number of messages written on a successful write.
|
|
//
|
|
// On Linux, a batch write will be optimized.
|
|
// On other platforms, this method will write only a single message.
|
|
func (c *payloadHandler) WriteBatch(ms []Message, flags int) (int, error) {
|
|
if !c.ok() {
|
|
return 0, syscall.EINVAL
|
|
}
|
|
switch runtime.GOOS {
|
|
case "linux":
|
|
n, err := c.SendMsgs([]socket.Message(ms), flags)
|
|
if err != nil {
|
|
err = &net.OpError{Op: "write", Net: c.PacketConn.LocalAddr().Network(), Source: c.PacketConn.LocalAddr(), Err: err}
|
|
}
|
|
return n, err
|
|
default:
|
|
n := 1
|
|
err := c.SendMsg(&ms[0], flags)
|
|
if err != nil {
|
|
n = 0
|
|
err = &net.OpError{Op: "write", Net: c.PacketConn.LocalAddr().Network(), Source: c.PacketConn.LocalAddr(), Err: err}
|
|
}
|
|
return n, err
|
|
}
|
|
}
|