gilles/go2rtc
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Move tutk dtls to separate package #2011

Browse Source
This commit is contained in:
Alex X
2026-01-18 08:38:38 +03:00
parent d40f6064d9
commit 514188201a
7 changed files with 39 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files
README.md
+4 -2
View File
@@ -637,11 +637,13 @@ Two-way audio support for Chinese version of [TP-Link cameras](https://www.tp-li
This source allows you to view cameras from the [Xiaomi Mi Home](https://home.mi.com/) ecosystem. This source allows you to view cameras from the [Xiaomi Mi Home](https://home.mi.com/) ecosystem.
*[read more](internal/xiaomi/README.md)*
## Source: Wyze ## Source: Wyze
This source allows you to stream from [Wyze](https://wyze.com/) cameras using native P2P protocol - no docker-wyze-bridge required. Supports H.264/H.265 video, AAC/G.711 audio, and two-way audio. [Read more](https://github.com/AlexxIT/go2rtc/blob/master/internal/wyze/README.md). This source allows you to stream from [Wyze](https://wyze.com/) cameras using native P2P protocol - no `docker-wyze-bridge` required. Supports H.264/H.265 video, AAC/G.711 audio, and two-way audio.
*[read more](internal/xiaomi/README.md)* *[read more](internal/wyze/README.md)*
## Source: GoPro ## Source: GoPro
pkg/tutk/codec.go
+9
View File
@@ -26,6 +26,15 @@ const (
var sampleRates = [9]uint32{8000, 11025, 12000, 16000, 22050, 24000, 32000, 44100, 48000} var sampleRates = [9]uint32{8000, 11025, 12000, 16000, 22050, 24000, 32000, 44100, 48000}
func GetSampleRateIndex(sampleRate uint32) uint8 {
for i, rate := range sampleRates {
if rate == sampleRate {
return uint8(i)
}
}
return 3 // default 16kHz
}
func GetSamplesPerFrame(codecID byte) uint32 { func GetSamplesPerFrame(codecID byte) uint32 {
switch codecID { switch codecID {
case CodecAACRaw, CodecAACADTS, CodecAACLATM, CodecAACAlt: case CodecAACRaw, CodecAACADTS, CodecAACLATM, CodecAACAlt:
pkg/tutk/auth.go → pkg/tutk/dtls/auth.go
+1 -1
View File
@@ -1,4 +1,4 @@
package tutk package dtls
import ( import (
"crypto/sha256" "crypto/sha256"
pkg/tutk/cipher.go → pkg/tutk/dtls/cipher.go
+1 -1
View File
@@ -1,4 +1,4 @@
package tutk package dtls
import ( import (
"crypto/cipher" "crypto/cipher"
pkg/tutk/conn_dtls.go → pkg/tutk/dtls/conn_dtls.go
+19 -24
View File
@@ -1,4 +1,4 @@
package tutk package dtls
import ( import (
"context" "context"
@@ -11,6 +11,7 @@ import (
"sync" "sync"
"time" "time"
"github.com/AlexxIT/go2rtc/pkg/tutk"
"github.com/pion/dtls/v3" "github.com/pion/dtls/v3"
) )
@@ -70,7 +71,7 @@ const (
type DTLSConn struct { type DTLSConn struct {
conn *net.UDPConn conn *net.UDPConn
addr *net.UDPAddr addr *net.UDPAddr
frames *FrameHandler frames *tutk.FrameHandler
err error err error
verbose bool verbose bool
ctx context.Context ctx context.Context
@@ -150,7 +151,7 @@ func DialDTLS(host string, port int, uid, authKey, enr string, verbose bool) (*D
c.clientBuf = make(chan []byte, 64) c.clientBuf = make(chan []byte, 64)
c.serverBuf = make(chan []byte, 64) c.serverBuf = make(chan []byte, 64)
c.rawCmd = make(chan []byte, 16) c.rawCmd = make(chan []byte, 16)
c.frames = NewFrameHandler(c.verbose) c.frames = tutk.NewFrameHandler(c.verbose)
c.wg.Add(1) c.wg.Add(1)
go c.reader() go c.reader()
@@ -167,7 +168,7 @@ func DialDTLS(host string, port int, uid, authKey, enr string, verbose bool) (*D
} }
func (c *DTLSConn) AVClientStart(timeout time.Duration) error { func (c *DTLSConn) AVClientStart(timeout time.Duration) error {
randomID := GenSessionID() randomID := tutk.GenSessionID()
pkt1 := c.msgAVLogin(magicAVLogin1, 570, 0x0001, randomID) pkt1 := c.msgAVLogin(magicAVLogin1, 570, 0x0001, randomID)
pkt2 := c.msgAVLogin(magicAVLogin2, 572, 0x0000, randomID) pkt2 := c.msgAVLogin(magicAVLogin2, 572, 0x0000, randomID)
pkt2[20]++ // pkt2 has randomID incremented by 1 pkt2[20]++ // pkt2 has randomID incremented by 1
@@ -311,7 +312,7 @@ func (c *DTLSConn) AVServStop() error {
return nil return nil
} }
func (c *DTLSConn) AVRecvFrameData() (*Packet, error) { func (c *DTLSConn) AVRecvFrameData() (*tutk.Packet, error) {
select { select {
case pkt, ok := <-c.frames.Recv(): case pkt, ok := <-c.frames.Recv():
if !ok { if !ok {
@@ -351,7 +352,7 @@ func (c *DTLSConn) Write(data []byte) error {
_, err := c.conn.WriteToUDP(data, c.addr) _, err := c.conn.WriteToUDP(data, c.addr)
return err return err
} }
_, err := c.conn.WriteToUDP(TransCodeBlob(data), c.addr) _, err := c.conn.WriteToUDP(tutk.TransCodeBlob(data), c.addr)
return err return err
} }
@@ -397,7 +398,7 @@ func (c *DTLSConn) WriteAndWait(req []byte, ok func(res []byte) bool) ([]byte, e
if c.isCC51 { if c.isCC51 {
res = buf[:n] res = buf[:n]
} else { } else {
res = ReverseTransCodeBlob(buf[:n]) res = tutk.ReverseTransCodeBlob(buf[:n])
} }
if ok(res) { if ok(res) {
@@ -496,9 +497,9 @@ func (c *DTLSConn) Error() error {
} }
func (c *DTLSConn) discovery() error { func (c *DTLSConn) discovery() error {
c.sid = GenSessionID() c.sid = tutk.GenSessionID()
pktIOTC := TransCodeBlob(c.msgDisco(1)) pktIOTC := tutk.TransCodeBlob(c.msgDisco(1))
pktCC51 := c.msgDiscoCC51(0, 0, false) pktCC51 := c.msgDiscoCC51(0, 0, false)
buf := make([]byte, 2048) buf := make([]byte, 2048)
@@ -530,7 +531,7 @@ func (c *DTLSConn) discovery() error {
} }
// IOTC Protocol (Basis) // IOTC Protocol (Basis)
data := ReverseTransCodeBlob(buf[:n]) data := tutk.ReverseTransCodeBlob(buf[:n])
if len(data) >= 16 && binary.LittleEndian.Uint16(data[8:]) == cmdDiscoRes { if len(data) >= 16 && binary.LittleEndian.Uint16(data[8:]) == cmdDiscoRes {
c.addr, c.isCC51 = addr, false c.addr, c.isCC51 = addr, false
return c.discoDone() return c.discoDone()
@@ -638,7 +639,7 @@ func (c *DTLSConn) worker() {
default: default:
channel := data[0] channel := data[0]
if channel == ChannelAudio || channel == ChannelIVideo || channel == ChannelPVideo { if channel == tutk.ChannelAudio || channel == tutk.ChannelIVideo || channel == tutk.ChannelPVideo {
c.frames.Handle(data) c.frames.Handle(data)
} }
} }
@@ -700,7 +701,7 @@ func (c *DTLSConn) reader() {
} }
// IOTC Protocol (Basis) // IOTC Protocol (Basis)
data := ReverseTransCodeBlob(buf[:n]) data := tutk.ReverseTransCodeBlob(buf[:n])
if len(data) < 16 { if len(data) < 16 {
continue continue
} }
@@ -843,8 +844,8 @@ func (c *DTLSConn) msgAudioFrame(payload []byte, timestampUS uint32, codec byte,
b := make([]byte, 36+totalPayload) b := make([]byte, 36+totalPayload)
// Outer header (36 bytes) // Outer header (36 bytes)
b[0] = ChannelAudio // 0x03 b[0] = tutk.ChannelAudio // 0x03
b[1] = FrameTypeStartAlt // 0x09 b[1] = tutk.FrameTypeStartAlt // 0x09
binary.LittleEndian.PutUint16(b[2:], protoVersion) binary.LittleEndian.PutUint16(b[2:], protoVersion)
binary.LittleEndian.PutUint32(b[4:], c.audioSeq) binary.LittleEndian.PutUint32(b[4:], c.audioSeq)
binary.LittleEndian.PutUint32(b[8:], timestampUS) binary.LittleEndian.PutUint32(b[8:], timestampUS)
@@ -855,8 +856,8 @@ func (c *DTLSConn) msgAudioFrame(payload []byte, timestampUS uint32, codec byte,
} }
// Inner header // Inner header
b[16] = ChannelAudio b[16] = tutk.ChannelAudio
b[17] = FrameTypeEndSingle b[17] = tutk.FrameTypeEndSingle
binary.LittleEndian.PutUint16(b[18:], uint16(prevFrame)) binary.LittleEndian.PutUint16(b[18:], uint16(prevFrame))
binary.LittleEndian.PutUint16(b[20:], 0x0001) // pkt_total binary.LittleEndian.PutUint16(b[20:], 0x0001) // pkt_total
binary.LittleEndian.PutUint16(b[22:], 0x0010) // flags binary.LittleEndian.PutUint16(b[22:], 0x0010) // flags
@@ -868,19 +869,13 @@ func (c *DTLSConn) msgAudioFrame(payload []byte, timestampUS uint32, codec byte,
fi[0] = codec // Codec ID (low byte) fi[0] = codec // Codec ID (low byte)
fi[1] = 0 // Codec ID (high byte, unused) fi[1] = 0 // Codec ID (high byte, unused)
// Audio flags: [3:2]=sampleRateIdx [1]=16bit [0]=stereo // Audio flags: [3:2]=sampleRateIdx [1]=16bit [0]=stereo
var srIdx uint8 = 3 // default 16kHz srIdx := tutk.GetSampleRateIndex(sampleRate)
for i, rate := range sampleRates {
if rate == sampleRate {
srIdx = uint8(i)
break
}
}
fi[2] = (srIdx << 2) | 0x02 // 16-bit always set fi[2] = (srIdx << 2) | 0x02 // 16-bit always set
if channels == 2 { if channels == 2 {
fi[2] |= 0x01 fi[2] |= 0x01
} }
fi[4] = 1 // online fi[4] = 1 // online
binary.LittleEndian.PutUint32(fi[12:], (c.audioFrameNo-1)*GetSamplesPerFrame(codec)*1000/sampleRate) binary.LittleEndian.PutUint32(fi[12:], (c.audioFrameNo-1)*tutk.GetSamplesPerFrame(codec)*1000/sampleRate)
return b return b
} }
pkg/tutk/dtls.go → pkg/tutk/dtls/dtls.go
+1 -1
View File
@@ -1,4 +1,4 @@
package tutk package dtls
import ( import (
"context" "context"
pkg/wyze/client.go
+4 -3
View File
@@ -13,6 +13,7 @@ import (
"time" "time"
"github.com/AlexxIT/go2rtc/pkg/tutk" "github.com/AlexxIT/go2rtc/pkg/tutk"
"github.com/AlexxIT/go2rtc/pkg/tutk/dtls"
) )
const ( const (
@@ -49,7 +50,7 @@ const (
) )
type Client struct { type Client struct {
conn *tutk.DTLSConn conn *dtls.DTLSConn
host string host string
uid string uid string
@@ -97,7 +98,7 @@ func Dial(rawURL string) (*Client, error) {
verbose: query.Get("verbose") == "true", verbose: query.Get("verbose") == "true",
} }
c.authKey = string(tutk.CalculateAuthKey(c.enr, c.mac)) c.authKey = string(dtls.CalculateAuthKey(c.enr, c.mac))
if c.verbose { if c.verbose {
fmt.Printf("[Wyze] Connecting to %s (UID: %s)\n", c.host, c.uid) fmt.Printf("[Wyze] Connecting to %s (UID: %s)\n", c.host, c.uid)
@@ -303,7 +304,7 @@ func (c *Client) connect() error {
host = host[:idx] host = host[:idx]
} }
conn, err := tutk.DialDTLS(host, port, c.uid, c.authKey, c.enr, c.verbose) conn, err := dtls.DialDTLS(host, port, c.uid, c.authKey, c.enr, c.verbose)
if err != nil { if err != nil {
return fmt.Errorf("wyze: connect failed: %w", err) return fmt.Errorf("wyze: connect failed: %w", err)
} }