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,38 @@
+// SPDX-FileCopyrightText: 2025 The Pion community <https://pion.ly>
+// SPDX-License-Identifier: MIT
+
+package transport
+
+const MaxAttributesLen = 1024
+
+type PacketAttributes struct {
+	Buffer      []byte
+	BytesCopied int
+}
+
+func NewPacketAttributesWithLen(length int) *PacketAttributes {
+	buff := make([]byte, length)
+
+	return &PacketAttributes{
+		Buffer:      buff,
+		BytesCopied: 0,
+	}
+}
+
+func (p *PacketAttributes) Clone() *PacketAttributes {
+	b := make([]byte, p.BytesCopied)
+	copy(b, p.Buffer)
+	return &PacketAttributes{
+		Buffer:      b,
+		BytesCopied: p.BytesCopied,
+	}
+}
+
+// Returns the read buffer. Just like when calling a read on a socket we have n, err := conn.Read(buf)
+// and the read bytes are in buf[:n], we should use this method after calling the ReadWithAttributes method.
+func (p *PacketAttributes) GetReadPacketAttributes() *PacketAttributes {
+	return &PacketAttributes{
+		Buffer:      p.Buffer[:p.BytesCopied],
+		BytesCopied: p.BytesCopied,
+	}
+}