ECN support

connsocket.go

@@ -0,0 +1,95 @@
+// SPDX-FileCopyrightText: 2025 The Pion community <https://pion.ly>
+// SPDX-License-Identifier: MIT
+
+package transport
+
+import (
+	"net"
+	"time"
+)
+
+type NetConnSocket interface {
+	net.Conn
+
+	ReadWithAttributes(p []byte, attr *PacketAttributes) (n int, err error)
+}
+
+type PacketConnSocket interface {
+	net.PacketConn
+
+	ReadFromWithAttributes(p []byte, attr *PacketAttributes) (n int, addr net.Addr, err error)
+}
+
+// NetConnToNetConnSocket wraps a net.Conn and implements the PacketStream interface by delegating
+// calls to the underlying connection. ReadWithAttributes delegates to Read and
+// ignores the provided PacketAttributes.
+type NetConnToNetConnSocket struct {
+	conn net.Conn
+}
+
+// NewNetConnToNetConnSocket returns a new Proxy that wraps the provided net.Conn.
+func NewNetConnToNetConnSocket(conn net.Conn) *NetConnToNetConnSocket {
+	return &NetConnToNetConnSocket{conn: conn}
+}
+
+// ReadWithAttributes reads from the underlying connection and ignores attributes.
+func (p *NetConnToNetConnSocket) ReadWithAttributes(b []byte, _ *PacketAttributes) (int, error) {
+	return p.conn.Read(b)
+}
+
+// Delegate net.Conn methods to the underlying connection.
+func (p *NetConnToNetConnSocket) Read(b []byte) (int, error)        { return p.conn.Read(b) }
+func (p *NetConnToNetConnSocket) Write(b []byte) (int, error)       { return p.conn.Write(b) }
+func (p *NetConnToNetConnSocket) Close() error                      { return p.conn.Close() }
+func (p *NetConnToNetConnSocket) LocalAddr() net.Addr               { return p.conn.LocalAddr() }
+func (p *NetConnToNetConnSocket) RemoteAddr() net.Addr              { return p.conn.RemoteAddr() }
+func (p *NetConnToNetConnSocket) SetDeadline(t time.Time) error     { return p.conn.SetDeadline(t) }
+func (p *NetConnToNetConnSocket) SetReadDeadline(t time.Time) error { return p.conn.SetReadDeadline(t) }
+func (p *NetConnToNetConnSocket) SetWriteDeadline(t time.Time) error {
+	return p.conn.SetWriteDeadline(t)
+}
+
+type PacketConnToPacketConnSocket struct {
+	conn net.PacketConn
+}
+
+func NewPacketConnToPacketConnSocket(conn net.PacketConn) *PacketConnToPacketConnSocket {
+	return &PacketConnToPacketConnSocket{
+		conn: conn,
+	}
+}
+
+func (u *PacketConnToPacketConnSocket) ReadFromWithAttributes(
+	p []byte, _ *PacketAttributes) (n int, addr net.Addr, err error) {
+	return u.conn.ReadFrom(p)
+}
+
+func (u *PacketConnToPacketConnSocket) ReadFrom(p []byte) (n int, addr net.Addr, err error) {
+	return u.conn.ReadFrom(p)
+}
+
+func (u *PacketConnToPacketConnSocket) WriteTo(b []byte, addr net.Addr) (int, error) {
+	n, err := u.conn.WriteTo(b, addr)
+
+	return n, err
+}
+
+func (u *PacketConnToPacketConnSocket) Close() error {
+	return u.conn.Close()
+}
+
+func (u *PacketConnToPacketConnSocket) LocalAddr() net.Addr {
+	return u.conn.LocalAddr()
+}
+
+func (u *PacketConnToPacketConnSocket) SetDeadline(t time.Time) error {
+	return u.conn.SetDeadline(t)
+}
+
+func (u *PacketConnToPacketConnSocket) SetReadDeadline(t time.Time) error {
+	return u.conn.SetReadDeadline(t)
+}
+
+func (u *PacketConnToPacketConnSocket) SetWriteDeadline(t time.Time) error {
+	return u.conn.SetWriteDeadline(t)
+}

packetattributes.go

@@ -0,0 +1,57 @@
+// SPDX-FileCopyrightText: 2025 The Pion community <https://pion.ly>
+// SPDX-License-Identifier: MIT
+
+package transport
+
+const attrMaxLen = 1
+
+type ECN uint8
+
+type PacketAttributesBuffer interface {
+	// for serializing
+	Marshal() []byte
+
+	// for de-serializing. The bytes will be copied into the returned buffer
+	GetBuffer() []byte
+}
+
+type PacketAttributes struct {
+	buffer [attrMaxLen]byte
+}
+
+func NewPacketAttributes() *PacketAttributes {
+	p := &PacketAttributes{}
+	p.Reset()
+
+	return p
+}
+
+func (p *PacketAttributes) Reset() {
+	p.WithECN(0)
+}
+
+func (p *PacketAttributes) GetECN() ECN {
+	return ECN(p.buffer[0])
+}
+
+func (p *PacketAttributes) WithECN(e ECN) *PacketAttributes {
+	p.buffer[0] = byte(e)
+
+	return p
+}
+
+// Marshal returns the internal buffer as-is.
+func (p *PacketAttributes) Marshal() []byte {
+	return p.buffer[:]
+}
+
+func (p *PacketAttributes) GetBuffer() []byte {
+	return p.buffer[:]
+}
+
+func (p *PacketAttributes) Clone() *PacketAttributes {
+	clone := &PacketAttributes{}
+	clone.buffer[0] = p.buffer[0]
+
+	return clone
+}