ghostream/stream/srt/srt.go

105 lines
2.3 KiB
Go
Raw Normal View History

2020-10-09 20:36:02 +00:00
// Package srt serves a SRT server
2020-09-27 09:14:22 +00:00
package srt
2020-10-02 07:44:31 +00:00
// #include <srt/srt.h>
import "C"
2020-10-02 07:46:15 +00:00
2020-09-27 09:14:22 +00:00
import (
"log"
"net"
"strconv"
2020-10-01 21:31:14 +00:00
"strings"
2020-10-02 07:46:15 +00:00
"github.com/haivision/srtgo"
2020-10-03 15:42:38 +00:00
"gitlab.crans.org/nounous/ghostream/auth"
"gitlab.crans.org/nounous/ghostream/messaging"
)
2020-09-27 09:14:22 +00:00
// Options holds web package configuration
type Options struct {
2020-10-09 20:06:30 +00:00
Enabled bool
2020-09-27 09:14:22 +00:00
ListenAddress string
MaxClients int
}
// Split host and port from listen address
2020-10-09 21:37:08 +00:00
func splitHostPort(hostport string) (string, uint16, error) {
host, portS, err := net.SplitHostPort(hostport)
if err != nil {
2020-10-09 21:37:08 +00:00
return "", 0, err
}
if host == "" {
host = "0.0.0.0"
}
port64, err := strconv.ParseUint(portS, 10, 16)
if err != nil {
2020-10-09 21:37:08 +00:00
return "", 0, err
}
2020-10-09 21:37:08 +00:00
return host, uint16(port64), nil
2020-09-27 09:14:22 +00:00
}
// Serve SRT server
func Serve(streams *messaging.Streams, authBackend auth.Backend, cfg *Options) {
2020-10-09 20:06:30 +00:00
if !cfg.Enabled {
// SRT is not enabled, ignore
return
}
2020-10-04 09:25:55 +00:00
// Start SRT in listening mode
log.Printf("SRT server listening on %s", cfg.ListenAddress)
2020-10-09 21:37:08 +00:00
host, port, err := splitHostPort(cfg.ListenAddress)
if err != nil {
log.Fatalf("Failed to split host and port from %s", cfg.ListenAddress)
}
2020-10-10 11:50:37 +00:00
options := make(map[string]string)
options["blocking"] = "0"
options["transtype"] = "live"
sck := srtgo.NewSrtSocket(host, port, options)
2020-09-30 14:53:15 +00:00
if err := sck.Listen(cfg.MaxClients); err != nil {
2020-10-04 09:25:55 +00:00
log.Fatal("Unable to listen for SRT clients:", err)
2020-09-30 14:53:15 +00:00
}
2020-09-27 09:14:22 +00:00
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-10-10 11:50:37 +00:00
// Something wrong happened
log.Println(err)
2020-10-04 09:45:49 +00:00
continue
2020-09-27 18:43:00 +00:00
}
2020-09-27 20:10:47 +00:00
2020-10-18 09:06:54 +00:00
// FIXME: Flush socket
// Without this, the SRT buffer might get full before reading it
// streamid can be "name:password" for streamer or "name" for viewer
streamID, err := s.GetSockOptString(C.SRTO_STREAMID)
if err != nil {
log.Print("Failed to get socket streamid")
continue
}
split := strings.Split(streamID, ":")
if len(split) > 1 {
// password was provided so it is a streamer
name, password := split[0], split[1]
if authBackend != nil {
// check password
if ok, err := authBackend.Login(name, password); !ok || err != nil {
log.Printf("Failed to authenticate for stream %s", name)
s.Close()
continue
}
}
2020-10-17 10:26:24 +00:00
go handleStreamer(s, streams, name)
2020-10-02 21:35:01 +00:00
} else {
// password was not provided so it is a viewer
name := split[0]
2020-10-17 10:26:24 +00:00
// Send stream
go handleViewer(s, streams, name)
2020-10-02 21:35:01 +00:00
}
}
}