install go2rtc on bob

2026-04-04 19:36:14 +02:00
parent f0b56e63d1
commit ccf88187b8
537 changed files with 69213 additions and 0 deletions
@@ -0,0 +1,146 @@
+package dtls
+
+import (
+	"context"
+	"net"
+	"sync"
+	"time"
+
+	"github.com/pion/dtls/v3"
+)
+
+func NewDTLSClient(ctx context.Context, channel uint8, addr net.Addr, writeFn func([]byte, uint8) error, readChan chan []byte, psk []byte) (*dtls.Conn, error) {
+	return dialDTLS(ctx, channel, addr, writeFn, readChan, psk, false)
+}
+
+func NewDTLSServer(ctx context.Context, channel uint8, addr net.Addr, writeFn func([]byte, uint8) error, readChan chan []byte, psk []byte) (*dtls.Conn, error) {
+	return dialDTLS(ctx, channel, addr, writeFn, readChan, psk, true)
+}
+
+func dialDTLS(ctx context.Context, channel uint8, addr net.Addr, writeFn func([]byte, uint8) error, readChan chan []byte, psk []byte, isServer bool) (*dtls.Conn, error) {
+	adapter := &channelAdapter{
+		ctx:      ctx,
+		channel:  channel,
+		addr:     addr,
+		writeFn:  writeFn,
+		readChan: readChan,
+	}
+
+	var conn *dtls.Conn
+	var err error
+
+	if isServer {
+		conn, err = dtls.Server(adapter, addr, buildDTLSConfig(psk, true))
+	} else {
+		conn, err = dtls.Client(adapter, addr, buildDTLSConfig(psk, false))
+	}
+	if err != nil {
+		return nil, err
+	}
+
+	timeout := 5 * time.Second
+	adapter.SetReadDeadline(time.Now().Add(timeout))
+	hsCtx, cancel := context.WithTimeout(ctx, timeout)
+	defer cancel()
+
+	if err := conn.HandshakeContext(hsCtx); err != nil {
+		go conn.Close()
+		return nil, err
+	}
+
+	adapter.SetReadDeadline(time.Time{})
+	return conn, nil
+}
+
+func buildDTLSConfig(psk []byte, isServer bool) *dtls.Config {
+	config := &dtls.Config{
+		PSK: func(hint []byte) ([]byte, error) {
+			return psk, nil
+		},
+		PSKIdentityHint:         []byte("AUTHPWD_admin"),
+		InsecureSkipVerify:      true,
+		InsecureSkipVerifyHello: true,
+		MTU:                     1200,
+		FlightInterval:          300 * time.Millisecond,
+		ExtendedMasterSecret:    dtls.DisableExtendedMasterSecret,
+	}
+
+	if isServer {
+		config.CipherSuites = []dtls.CipherSuiteID{dtls.TLS_PSK_WITH_AES_128_CBC_SHA256}
+	} else {
+		config.CustomCipherSuites = CustomCipherSuites
+	}
+
+	return config
+}
+
+type channelAdapter struct {
+	ctx          context.Context
+	channel      uint8
+	writeFn      func([]byte, uint8) error
+	readChan     chan []byte
+	addr         net.Addr
+	mu           sync.Mutex
+	readDeadline time.Time
+}
+
+func (a *channelAdapter) ReadFrom(p []byte) (n int, addr net.Addr, err error) {
+	a.mu.Lock()
+	deadline := a.readDeadline
+	a.mu.Unlock()
+
+	if !deadline.IsZero() {
+		timeout := time.Until(deadline)
+		if timeout <= 0 {
+			return 0, nil, &timeoutError{}
+		}
+
+		timer := time.NewTimer(timeout)
+		defer timer.Stop()
+
+		select {
+		case data := <-a.readChan:
+			return copy(p, data), a.addr, nil
+		case <-timer.C:
+			return 0, nil, &timeoutError{}
+		case <-a.ctx.Done():
+			return 0, nil, net.ErrClosed
+		}
+	}
+
+	select {
+	case data := <-a.readChan:
+		return copy(p, data), a.addr, nil
+	case <-a.ctx.Done():
+		return 0, nil, net.ErrClosed
+	}
+}
+
+func (a *channelAdapter) WriteTo(p []byte, _ net.Addr) (int, error) {
+	if err := a.writeFn(p, a.channel); err != nil {
+		return 0, err
+	}
+	return len(p), nil
+}
+
+func (a *channelAdapter) Close() error        { return nil }
+func (a *channelAdapter) LocalAddr() net.Addr { return &net.UDPAddr{} }
+func (a *channelAdapter) SetDeadline(t time.Time) error {
+	a.mu.Lock()
+	a.readDeadline = t
+	a.mu.Unlock()
+	return nil
+}
+func (a *channelAdapter) SetReadDeadline(t time.Time) error {
+	a.mu.Lock()
+	a.readDeadline = t
+	a.mu.Unlock()
+	return nil
+}
+func (a *channelAdapter) SetWriteDeadline(time.Time) error { return nil }
+
+type timeoutError struct{}
+
+func (e *timeoutError) Error() string   { return "i/o timeout" }
+func (e *timeoutError) Timeout() bool   { return true }
+func (e *timeoutError) Temporary() bool { return true }