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 occured 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 } } } }