73 lines
2.1 KiB
Go
73 lines
2.1 KiB
Go
// Package srt serves a SRT server
|
|
package srt
|
|
|
|
import (
|
|
"log"
|
|
|
|
"github.com/haivision/srtgo"
|
|
)
|
|
|
|
func handleStreamer(s *srtgo.SrtSocket, name string, clientDataChannels map[string][]chan Packet, forwardingChannel, webrtcChannel chan Packet) {
|
|
log.Printf("New SRT streamer for stream %s", name)
|
|
|
|
// Create a new buffer
|
|
// UDP packet cannot be larger than MTU (1500)
|
|
buff := make([]byte, 1500)
|
|
|
|
// Setup stream forwarding
|
|
forwardingChannel <- Packet{StreamName: name, PacketType: "register", Data: nil}
|
|
webrtcChannel <- Packet{StreamName: name, PacketType: "register", Data: nil}
|
|
|
|
// Read RTP packets forever and send them to the WebRTC Client
|
|
for {
|
|
// 5s timeout
|
|
n, err := s.Read(buff, 5000)
|
|
if err != nil {
|
|
log.Println("Error occurred while reading SRT socket:", err)
|
|
break
|
|
}
|
|
|
|
if n == 0 {
|
|
// End of stream
|
|
log.Printf("Received no bytes, stopping stream.")
|
|
break
|
|
}
|
|
|
|
// Send raw packet to other streams
|
|
// Copy data in another buffer to ensure that the data would not be overwritten
|
|
data := make([]byte, n)
|
|
copy(data, buff[:n])
|
|
forwardingChannel <- Packet{StreamName: name, PacketType: "sendData", Data: data}
|
|
webrtcChannel <- Packet{StreamName: name, PacketType: "sendData", Data: data}
|
|
for _, dataChannel := range clientDataChannels[name] {
|
|
dataChannel <- Packet{StreamName: name, PacketType: "sendData", Data: data}
|
|
}
|
|
}
|
|
|
|
forwardingChannel <- Packet{StreamName: name, PacketType: "close", Data: nil}
|
|
webrtcChannel <- Packet{StreamName: name, PacketType: "close", Data: nil}
|
|
}
|
|
|
|
func handleViewer(s *srtgo.SrtSocket, name string, dataChannel chan Packet, dataChannels map[string][]chan Packet) {
|
|
// FIXME Should not pass all dataChannels to one viewer
|
|
|
|
log.Printf("New SRT viewer for stream %s", name)
|
|
|
|
// Receive packets from channel and send them
|
|
for {
|
|
packet := <-dataChannel
|
|
if packet.PacketType == "sendData" {
|
|
_, err := s.Write(packet.Data, 10000)
|
|
if err != nil {
|
|
s.Close()
|
|
for i, channel := range dataChannels[name] {
|
|
if channel == dataChannel {
|
|
dataChannels[name] = append(dataChannels[name][:i], dataChannels[name][i+1:]...)
|
|
}
|
|
}
|
|
return
|
|
}
|
|
}
|
|
}
|
|
}
|