1
Fork 0
mirror of https://github.com/jech/galene.git synced 2024-11-09 02:05:59 +01:00

Split out SDP fragment parsing, add test.

This commit is contained in:
Juliusz Chroboczek 2024-09-30 00:23:36 +02:00
parent 45bbb138c6
commit df274ad6ea
2 changed files with 51 additions and 14 deletions

View file

@ -1,6 +1,7 @@
package webserver
import (
"bufio"
"bytes"
"crypto/aes"
"crypto/cipher"
@ -345,23 +346,25 @@ func whipResourceHandler(w http.ResponseWriter, r *http.Request) {
return
}
body, err := io.ReadAll(http.MaxBytesReader(w, r.Body, sdpLimit))
err = parseSDPFrag(
http.MaxBytesReader(w, r.Body, sdpLimit),
c.GotICECandidate,
)
if err != nil {
httpError(w, err)
log.Printf("WHIP trickle ICE: %v", err)
http.Error(w, "bad request", http.StatusBadRequest)
return
}
w.WriteHeader(http.StatusNoContent)
}
if len(body) < 2 {
http.Error(w, "SDP truncated", http.StatusBadRequest)
return
}
// RFC 8840
lines := bytes.Split(body, []byte{'\n'})
// RFC 8840
func parseSDPFrag(r io.Reader, f func(webrtc.ICECandidateInit) error) error {
scanner := bufio.NewScanner(r)
mLineIndex := -1
var mid, ufrag []byte
for _, l := range lines {
l = bytes.TrimRight(l, " \r")
for scanner.Scan() {
l := scanner.Bytes()
if bytes.HasPrefix(l, []byte("a=ice-ufrag:")) {
ufrag = l[len("a=ice-ufrag:"):]
} else if bytes.HasPrefix(l, []byte("m=")) {
@ -385,12 +388,11 @@ func whipResourceHandler(w http.ResponseWriter, r *http.Request) {
s := string(ufrag)
init.UsernameFragment = &s
}
err := c.GotICECandidate(init)
err := f(init)
if err != nil {
log.Printf("WHIP candidate: %v", err)
}
}
}
w.WriteHeader(http.StatusNoContent)
return
return nil
}

35
webserver/whip_test.go Normal file
View file

@ -0,0 +1,35 @@
package webserver
import (
"strings"
"testing"
"github.com/pion/webrtc/v3"
)
func TestParseSDPFrag(t *testing.T) {
sdp := `a=ice-ufrag:FZ0m
a=ice-pwd:NRT+gj1EhsEwMm9MA7ljzBRy
m=audio 9 UDP/TLS/RTP/SAVPF 0
a=mid:0
a=candidate:2930517337 1 udp 2113937151 1eaafdf1-4127-499f-90d4-8c35ea49d5e6.local 44360 typ host generation 0 ufrag FZ0m network-cost 999
2024/09/30 00:07:41 {candidate:2930517337 1 udp 2113937151 1eaafdf1-4127-499f-90d4-8c35ea49d5e6.local 44360 typ host generation 0 ufrag FZ0m network-cost 999 0xc00062a580 0xc000620288 0xc00062a590}
a=end-of-candidates`
r := strings.NewReader(sdp)
candidates := []webrtc.ICECandidateInit(nil)
err := parseSDPFrag(r, func(c webrtc.ICECandidateInit) error {
candidates = append(candidates, c)
return nil
})
if err != nil {
t.Errorf("parseSDPFrag: %v", err)
}
if len(candidates) != 1 {
t.Errorf("Expected 1, got %v", candidates)
}
if *candidates[0].SDPMLineIndex != 0 ||
*candidates[0].SDPMid != "0" ||
*candidates[0].UsernameFragment != "FZ0m" {
t.Errorf("Got %v", candidates[0])
}
}