Merge pull request #3178 from lucas-clemente/fix-windows-ipv6-dontfragment

fix DONT_FRAGMENT error when using a IPv6 connection on Windows
This commit is contained in:
Marten Seemann
2021-05-11 12:46:20 -07:00
committed by GitHub
2 changed files with 19 additions and 13 deletions

View File

@@ -13,24 +13,20 @@ import (
const IP_DONTFRAGMENT = 14
func newConn(c net.PacketConn) (connection, error) {
conn, ok := c.(interface {
SyscallConn() (syscall.RawConn, error)
})
if !ok {
return nil, errors.New("doesn't have a SyscallConn")
}
rawConn, err := conn.SyscallConn()
func newConn(c OOBCapablePacketConn) (connection, error) {
rawConn, err := c.SyscallConn()
if err != nil {
return nil, fmt.Errorf("couldn't get syscall.RawConn: %w", err)
}
var serr error
if err := rawConn.Control(func(fd uintptr) {
serr = windows.SetsockoptInt(windows.Handle(fd), windows.IPPROTO_IP, IP_DONTFRAGMENT, 1)
// This should succeed if the connection is a IPv4 or a dual-stack connection.
// It will fail for IPv6 connections.
// TODO: properly handle error.
_ = windows.SetsockoptInt(windows.Handle(fd), windows.IPPROTO_IP, IP_DONTFRAGMENT, 1)
}); err != nil {
return nil, err
}
return &basicConn{PacketConn: c}, serr
return &basicConn{PacketConn: c}, nil
}
func inspectReadBuffer(c net.PacketConn) (int, error) {

View File

@@ -10,13 +10,23 @@ import (
)
var _ = Describe("Windows Conn Test", func() {
It("try newConn", func() {
It("works on IPv4", func() {
addr, err := net.ResolveUDPAddr("udp4", "localhost:0")
Expect(err).ToNot(HaveOccurred())
udpConn, err := net.ListenUDP("udp4", addr)
Expect(err).ToNot(HaveOccurred())
conn, err := newConn(udpConn)
Expect(err).ToNot(HaveOccurred())
_ = conn.Close()
Expect(conn.Close()).To(Succeed())
})
It("works on IPv6", func() {
addr, err := net.ResolveUDPAddr("udp6", "[::1]:0")
Expect(err).ToNot(HaveOccurred())
udpConn, err := net.ListenUDP("udp6", addr)
Expect(err).ToNot(HaveOccurred())
conn, err := newConn(udpConn)
Expect(err).ToNot(HaveOccurred())
Expect(conn.Close()).To(Succeed())
})
})