mirrors/uquic
1
0
Fork 0
mirror of https://github.com/refraction-networking/uquic.git synced 2025-04-04 04:37:36 +03:00
Code Issues Projects Releases Packages Wiki Activity

add a QUIC_GO_DISABLE_ECN env to disable ECN support

Browse source
This commit is contained in:
Marten Seemann 2023-09-01 09:18:14 +07:00
parent b73a4de7ea
commit 8df7624c07
2 changed files with 16 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

12
sys_conn_oob.go
View file

@ -8,6 +8,8 @@ import (
"log"
"net"
"net/netip"
"os"
"strconv"
"sync"
"syscall"
"time"
@ -57,6 +59,11 @@ func inspectWriteBuffer(c syscall.RawConn) (int, error) {
return size, serr
}
func isECNDisabled() bool {
disabled, err := strconv.ParseBool(os.Getenv("QUIC_GO_DISABLE_ECN"))
return err == nil && disabled
}
type oobConn struct {
OOBCapablePacketConn
batchConn batchConn
@ -141,7 +148,7 @@ func newConn(c OOBCapablePacketConn, supportsDF bool) (*oobConn, error) {
cap: connCapabilities{
DF: supportsDF,
GSO: isGSOSupported(rawConn),
ECN: true,
ECN: !isECNDisabled(),
},
}
for i := 0; i < batchSize; i++ {
@ -239,6 +246,9 @@ func (c *oobConn) WritePacket(b []byte, addr net.Addr, packetInfoOOB []byte, gso
oob = appendUDPSegmentSizeMsg(oob, gsoSize)
}
if ecn != protocol.ECNUnsupported {
if !c.capabilities().ECN {
panic("tried to send a ECN-marked packet although ECN is disabled")
}
if remoteUDPAddr, ok := addr.(*net.UDPAddr); ok {
if remoteUDPAddr.IP.To4() != nil {
oob = appendIPv4ECNMsg(oob, ecn)