forked from quic-go/quic-go
eb6bdfdfc1
When the server is listening on multiple interfaces or interfaces with multiple IPs, the outgoing datagrams are sometime delivered with the wrong source IP address. In order to fix that, each quic connection needs to extract the destination IP (and optionally interface id) of the received datagrams, and set it as source IP (and interface) on the sent datagrams. On most platforms, this can be done using ancillary data with recvmsg() and sendmsg(). Some of the machinery for this is already there for ECN, this change extends it to read the destination IP info and write it to the outgoing packets. Fix #1736
69 lines
1.2 KiB
Go
69 lines
1.2 KiB
Go
package quic
|
|
|
|
import (
|
|
"net"
|
|
)
|
|
|
|
// A sendConn allows sending using a simple Write() on a non-connected packet conn.
|
|
type sendConn interface {
|
|
Write([]byte) error
|
|
Close() error
|
|
LocalAddr() net.Addr
|
|
RemoteAddr() net.Addr
|
|
}
|
|
|
|
type sconn struct {
|
|
connection
|
|
|
|
remoteAddr net.Addr
|
|
info *packetInfo
|
|
}
|
|
|
|
var _ sendConn = &sconn{}
|
|
|
|
func newSendConn(c connection, remote net.Addr, info *packetInfo) sendConn {
|
|
return &sconn{connection: c, remoteAddr: remote, info: info}
|
|
}
|
|
|
|
func (c *sconn) Write(p []byte) error {
|
|
_, err := c.WritePacket(p, c.remoteAddr, c.info)
|
|
return err
|
|
}
|
|
|
|
func (c *sconn) RemoteAddr() net.Addr {
|
|
return c.remoteAddr
|
|
}
|
|
|
|
func (c *sconn) LocalAddr() net.Addr {
|
|
addr := c.connection.LocalAddr()
|
|
if c.info != nil {
|
|
if udpAddr, ok := addr.(*net.UDPAddr); ok {
|
|
addrCopy := *udpAddr
|
|
addrCopy.IP = c.info.addr
|
|
addr = &addrCopy
|
|
}
|
|
}
|
|
return addr
|
|
}
|
|
|
|
type spconn struct {
|
|
net.PacketConn
|
|
|
|
remoteAddr net.Addr
|
|
}
|
|
|
|
var _ sendConn = &spconn{}
|
|
|
|
func newSendPconn(c net.PacketConn, remote net.Addr) sendConn {
|
|
return &spconn{PacketConn: c, remoteAddr: remote}
|
|
}
|
|
|
|
func (c *spconn) Write(p []byte) error {
|
|
_, err := c.WriteTo(p, c.remoteAddr)
|
|
return err
|
|
}
|
|
|
|
func (c *spconn) RemoteAddr() net.Addr {
|
|
return c.remoteAddr
|
|
}
|