forked from quic-go/quic-go
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:
@@ -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) {
|
||||
|
||||
@@ -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())
|
||||
})
|
||||
})
|
||||
|
||||
Reference in New Issue
Block a user