snapshot of https://github.com/pion/webrtc.git (main, 2026-09-15; blobs > 131072B omitted)

internal/mux/endpoint.go

@@ -0,0 +1,107 @@
+// SPDX-FileCopyrightText: 2026 The Pion community <https://pion.ly>
+// SPDX-License-Identifier: MIT
+
+package mux
+
+import (
+	"errors"
+	"io"
+	"net"
+	"time"
+
+	"github.com/pion/ice/v4"
+	"github.com/pion/transport/v4/packetio"
+)
+
+// Endpoint implements net.Conn. It is used to read muxed packets.
+type Endpoint struct {
+	mux     *Mux
+	buffer  *packetio.Buffer
+	onClose func()
+}
+
+// Close unregisters the endpoint from the Mux.
+func (e *Endpoint) Close() (err error) {
+	if e.onClose != nil {
+		e.onClose()
+	}
+
+	if err = e.close(); err != nil {
+		return err
+	}
+
+	e.mux.RemoveEndpoint(e)
+
+	return nil
+}
+
+func (e *Endpoint) close() error {
+	return e.buffer.Close()
+}
+
+// Read reads a packet of len(p) bytes from the underlying conn
+// that are matched by the associated MuxFunc.
+func (e *Endpoint) Read(p []byte) (int, error) {
+	return e.buffer.Read(p)
+}
+
+// ReadFrom reads a packet of len(p) bytes from the underlying conn
+// that are matched by the associated MuxFunc.
+func (e *Endpoint) ReadFrom(p []byte) (int, net.Addr, error) {
+	i, err := e.Read(p)
+
+	return i, nil, err
+}
+
+// Write writes len(p) bytes to the underlying conn.
+func (e *Endpoint) Write(p []byte) (int, error) {
+	n, err := e.mux.nextConn.Write(p)
+	if errors.Is(err, ice.ErrNoCandidatePairs) {
+		return 0, nil
+	} else if errors.Is(err, ice.ErrClosed) {
+		return 0, io.ErrClosedPipe
+	}
+
+	return n, err
+}
+
+// WriteTo writes len(p) bytes to the underlying conn.
+func (e *Endpoint) WriteTo(p []byte, _ net.Addr) (int, error) {
+	return e.Write(p)
+}
+
+// LocalAddr returns the local network address, if known.
+func (e *Endpoint) LocalAddr() net.Addr {
+	return e.mux.nextConn.LocalAddr()
+}
+
+// RemoteAddr returns the remote network address, if known.
+func (e *Endpoint) RemoteAddr() net.Addr {
+	return e.mux.nextConn.RemoteAddr()
+}
+
+// SetDeadline sets the read and write deadlines on the shared underlying
+// connection. Because the connection is shared, this applies to all endpoints
+// on the mux. Per-endpoint read deadlines can be set with SetReadDeadline.
+func (e *Endpoint) SetDeadline(t time.Time) error {
+	return e.mux.nextConn.SetDeadline(t)
+}
+
+// SetReadDeadline sets the read deadline for this Endpoint's internal
+// packet buffer. This timeout applies only to reads from this Endpoint,
+// not to the shared underlying connection.
+func (e *Endpoint) SetReadDeadline(t time.Time) error {
+	return e.buffer.SetReadDeadline(t)
+}
+
+// SetWriteDeadline sets the write deadline on the shared underlying connection.
+// Because the connection is shared, this applies to all endpoints on the mux.
+func (e *Endpoint) SetWriteDeadline(t time.Time) error {
+	return e.mux.nextConn.SetWriteDeadline(t)
+}
+
+// SetOnClose is a user set callback that
+// will be executed when `Close` is called.
+func (e *Endpoint) SetOnClose(onClose func()) {
+	e.onClose = onClose
+}