1
0
mirror of https://gitlab.crans.org/nounous/ghostream.git synced 2024-12-23 01:32:19 +00:00
ghostream/stream/srt/srt.go
Yohann D'ANELLO 351870e92a Merge remote-tracking branch 'origin/golang' into multicast
# Conflicts:
#	stream/srt/srt.go
2020-09-29 22:43:37 +02:00

99 lines
2.4 KiB
Go

package srt
import (
"gitlab.crans.org/nounous/ghostream/stream/multicast"
"log"
"net"
"strconv"
"github.com/haivision/srtgo"
"github.com/pion/rtp"
)
// Options holds web package configuration
type Options struct {
ListenAddress string
MaxClients int
}
// Split host and port from listen address
func splitHostPort(hostport string) (string, uint16) {
host, portS, err := net.SplitHostPort(hostport)
if err != nil {
log.Fatalf("Failed to split host and port from %s", hostport)
}
if host == "" {
host = "0.0.0.0"
}
port64, err := strconv.ParseUint(portS, 10, 16)
if err != nil {
log.Fatalf("Port is not a integer: %s", err)
}
return host, uint16(port64)
}
// Serve SRT server
func Serve(cfg *Options) {
options := make(map[string]string)
options["transtype"] = "live"
// Start SRT in listen mode
log.Printf("SRT server listening on %s", cfg.ListenAddress)
host, port := splitHostPort(cfg.ListenAddress)
sck := srtgo.NewSrtSocket(host, uint16(port), options)
sck.Listen(cfg.MaxClients)
// FIXME: See srtgo.SocketOptions and value, err := s.GetSockOptString to get parameters
// http://ffmpeg.org/ffmpeg-protocols.html#srt
for {
// Wait for new connection
s, err := sck.Accept()
if err != nil {
log.Println("Error occurred while accepting request:", err)
break // FIXME: should not break here
}
// Create a new buffer
buff := make([]byte, 2048)
// Setup linked multicasts
multicast.RegisterStream("demo") // FIXME Replace with real stream key
// Read RTP packets forever and send them to the WebRTC Client
for {
n, err := s.Read(buff, 10000)
if err != nil {
log.Println("Error occured while reading SRT socket:", err)
multicast.CloseConnection("demo")
break
}
if n == 0 {
// End of stream
log.Printf("Received no bytes, stopping stream.")
multicast.CloseConnection("demo")
break
}
log.Printf("Received %d bytes", n)
// Send raw packet to other streams
multicast.SendPacket("demo", buff[:n])
// Unmarshal incoming packet
packet := &rtp.Packet{}
if err := packet.Unmarshal(buff[:n]); err != nil {
log.Println("Error occured while unmarshaling SRT:", err)
multicast.CloseConnection("demo")
break
}
// TODO: Send to WebRTC
//payloadType := uint8(22) // FIXME put vp8 payload
//packet.Header.PayloadType = payloadType
//err := videoTrack.WriteRTP(packet)
}
}
}