mirror of
https://github.com/status-im/consul.git
synced 2025-01-12 06:44:41 +00:00
9dc7194321
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
67 lines
1.3 KiB
Go
67 lines
1.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 socket
|
|
|
|
import (
|
|
"errors"
|
|
"net"
|
|
"os"
|
|
"syscall"
|
|
)
|
|
|
|
// A Conn represents a raw connection.
|
|
type Conn struct {
|
|
network string
|
|
c syscall.RawConn
|
|
}
|
|
|
|
// NewConn returns a new raw connection.
|
|
func NewConn(c net.Conn) (*Conn, error) {
|
|
var err error
|
|
var cc Conn
|
|
switch c := c.(type) {
|
|
case *net.TCPConn:
|
|
cc.network = "tcp"
|
|
cc.c, err = c.SyscallConn()
|
|
case *net.UDPConn:
|
|
cc.network = "udp"
|
|
cc.c, err = c.SyscallConn()
|
|
case *net.IPConn:
|
|
cc.network = "ip"
|
|
cc.c, err = c.SyscallConn()
|
|
default:
|
|
return nil, errors.New("unknown connection type")
|
|
}
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &cc, nil
|
|
}
|
|
|
|
func (o *Option) get(c *Conn, b []byte) (int, error) {
|
|
var operr error
|
|
var n int
|
|
fn := func(s uintptr) {
|
|
n, operr = getsockopt(s, o.Level, o.Name, b)
|
|
}
|
|
if err := c.c.Control(fn); err != nil {
|
|
return 0, err
|
|
}
|
|
return n, os.NewSyscallError("getsockopt", operr)
|
|
}
|
|
|
|
func (o *Option) set(c *Conn, b []byte) error {
|
|
var operr error
|
|
fn := func(s uintptr) {
|
|
operr = setsockopt(s, o.Level, o.Name, b)
|
|
}
|
|
if err := c.c.Control(fn); err != nil {
|
|
return err
|
|
}
|
|
return os.NewSyscallError("setsockopt", operr)
|
|
}
|