ghostream/stream/srt/srt.go

96 lines
2.3 KiB
Go
Raw Normal View History

2020-09-27 09:14:22 +00:00
package srt
import (
"log"
"net"
"strconv"
2020-09-27 09:14:22 +00:00
2020-09-27 18:43:00 +00:00
"github.com/haivision/srtgo"
2020-09-27 09:14:22 +00:00
)
// 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)
2020-09-27 09:14:22 +00:00
}
// Serve SRT server
func Serve(cfg *Options) {
2020-09-27 18:43:00 +00:00
options := make(map[string]string)
2020-09-28 14:32:35 +00:00
options["transtype"] = "live"
2020-09-27 18:43:00 +00:00
2020-09-29 18:58:55 +00:00
// 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)
2020-09-30 14:53:15 +00:00
if err := sck.Listen(cfg.MaxClients); err != nil {
log.Fatal("Unable to listen to SRT clients:", err)
}
2020-09-27 09:14:22 +00:00
2020-09-29 18:58:55 +00:00
// FIXME: See srtgo.SocketOptions and value, err := s.GetSockOptString to get parameters
// http://ffmpeg.org/ffmpeg-protocols.html#srt
2020-09-27 18:43:00 +00:00
for {
2020-09-29 18:58:55 +00:00
// Wait for new connection
2020-09-27 20:10:47 +00:00
s, err := sck.Accept()
if err != nil {
2020-09-28 14:32:35 +00:00
log.Println("Error occurred while accepting request:", err)
2020-09-29 18:58:55 +00:00
break // FIXME: should not break here
2020-09-27 18:43:00 +00:00
}
2020-09-27 20:10:47 +00:00
2020-09-29 18:58:55 +00:00
// Create a new buffer
2020-09-28 14:32:35 +00:00
buff := make([]byte, 2048)
2020-09-30 13:07:36 +00:00
// Setup stream forwarding
// FIXME: demo should be replaced by stream name
/* if err := forwarding.RegisterStream("demo"); err != nil {
log.Println("Error occurred during forward stream init:", err)
break
} */
2020-09-28 14:32:35 +00:00
// 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)
break
}
2020-09-29 18:58:55 +00:00
if n == 0 {
// End of stream
log.Printf("Received no bytes, stopping stream.")
break
}
// log.Printf("Received %d bytes", n)
2020-09-28 14:32:35 +00:00
// Send raw packet to other streams
// forwarding.SendPacket("demo", buff[:n])
2020-09-28 14:32:35 +00:00
2020-09-29 18:58:55 +00:00
// TODO: Send to WebRTC
2020-09-30 13:07:36 +00:00
// See https://github.com/ebml-go/webm/blob/master/reader.go
//err := videoTrack.WriteSample(media.Sample{Data: data, Samples: uint32(sampleCount)})
2020-09-28 14:32:35 +00:00
}
/* if err := forwarding.CloseConnection("demo"); err != nil {
log.Printf("Failed to close forward stream: %s", err)
} */
2020-09-27 09:14:22 +00:00
}
}