Initial commit
This commit is contained in:
commit
72c84dff1e
1
.dockerignore
Normal file
1
.dockerignore
Normal file
@ -0,0 +1 @@
|
||||
Dockerfile
|
||||
14
Dockerfile
Normal file
14
Dockerfile
Normal file
@ -0,0 +1,14 @@
|
||||
FROM golang:latest as builder
|
||||
WORKDIR /app
|
||||
COPY go.mod /app/go.mod
|
||||
COPY go.sum /app/go.sum
|
||||
RUN go mod download
|
||||
COPY . /app
|
||||
RUN CGO_ENABLED=0 go build -o dogtamer cmd/dogtamer.go
|
||||
|
||||
|
||||
FROM alpine:latest
|
||||
WORKDIR /app
|
||||
COPY --from=builder /app/dogtamer /app/dogtamer
|
||||
EXPOSE 5566
|
||||
CMD ["/app/dogtamer", "serve"]
|
||||
20
README.md
Normal file
20
README.md
Normal file
@ -0,0 +1,20 @@
|
||||
# dogtamer
|
||||
|
||||
A small rtmp relay.
|
||||
|
||||
## Plans
|
||||
|
||||
* Add web-interface for viewing streams
|
||||
* Add tests
|
||||
|
||||
## Usage
|
||||
|
||||
```text
|
||||
$ go build -o dogtamer cmd/dogtamer.go
|
||||
$ ./dogtamer serve
|
||||
```
|
||||
|
||||
|
||||
## TODO
|
||||
|
||||
* Handle client errors better
|
||||
101
cmd/dogtamer.go
Normal file
101
cmd/dogtamer.go
Normal file
@ -0,0 +1,101 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"os"
|
||||
"os/signal"
|
||||
|
||||
"github.com/urfave/cli/v2"
|
||||
"github.uio.no/torjus/dogtamer"
|
||||
"github.uio.no/torjus/dogtamer/server"
|
||||
"go.uber.org/zap"
|
||||
"go.uber.org/zap/zapcore"
|
||||
)
|
||||
|
||||
const (
|
||||
ExitSuccess = 0
|
||||
ExitGenericError = 1
|
||||
)
|
||||
|
||||
func main() {
|
||||
cli.VersionPrinter = func(c *cli.Context) {
|
||||
fmt.Println(dogtamer.GetVersion())
|
||||
}
|
||||
|
||||
app := &cli.App{
|
||||
Name: "dogtamer",
|
||||
Version: dogtamer.Version,
|
||||
Authors: []*cli.Author{
|
||||
{
|
||||
Name: "Torjus Håkestad",
|
||||
Email: "torjus@usit.uio.no",
|
||||
},
|
||||
},
|
||||
Flags: []cli.Flag{
|
||||
&cli.StringFlag{
|
||||
Name: "config",
|
||||
Usage: "Path to config-file",
|
||||
Aliases: []string{"c"},
|
||||
},
|
||||
},
|
||||
Commands: []*cli.Command{
|
||||
{
|
||||
Name: "serve",
|
||||
Usage: "Start dogtamer server",
|
||||
Action: ActionServe,
|
||||
},
|
||||
},
|
||||
}
|
||||
err := app.Run(os.Args)
|
||||
if err != nil {
|
||||
fmt.Printf("Error: %s\n", err)
|
||||
os.Exit(ExitGenericError)
|
||||
}
|
||||
}
|
||||
|
||||
func ActionServe(c *cli.Context) error {
|
||||
|
||||
ctx, cancel := context.WithCancel(context.Background())
|
||||
s := server.NewRTMPServer(ctx, ":5566")
|
||||
logger := setupServerLogger()
|
||||
s.Logger = logger
|
||||
|
||||
// Listen for SIGINT
|
||||
sigChan := make(chan os.Signal, 1)
|
||||
signal.Notify(sigChan, os.Interrupt)
|
||||
go func() {
|
||||
<-sigChan
|
||||
s.Logger.Info("Got shutdown signal.")
|
||||
cancel()
|
||||
}()
|
||||
|
||||
if err := s.Listen(); err != nil {
|
||||
return cli.Exit(err, ExitGenericError)
|
||||
}
|
||||
s.Logger.Info("Server shut down.")
|
||||
return nil
|
||||
}
|
||||
|
||||
func setupServerLogger() *zap.SugaredLogger {
|
||||
logEncoderConfig := zap.NewProductionEncoderConfig()
|
||||
logEncoderConfig.EncodeCaller = zapcore.ShortCallerEncoder
|
||||
logEncoderConfig.EncodeLevel = zapcore.CapitalColorLevelEncoder
|
||||
logEncoderConfig.EncodeTime = zapcore.ISO8601TimeEncoder
|
||||
logEncoderConfig.EncodeDuration = zapcore.StringDurationEncoder
|
||||
|
||||
rootLoggerConfig := &zap.Config{
|
||||
Level: zap.NewAtomicLevelAt(zap.InfoLevel),
|
||||
OutputPaths: []string{"stdout"},
|
||||
ErrorOutputPaths: []string{"stdout"},
|
||||
Encoding: "console",
|
||||
EncoderConfig: logEncoderConfig,
|
||||
}
|
||||
|
||||
rootLogger, err := rootLoggerConfig.Build()
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
return rootLogger.Named("RTMP").WithOptions().Sugar()
|
||||
}
|
||||
15
deploy/dogtamer.service
Normal file
15
deploy/dogtamer.service
Normal file
@ -0,0 +1,15 @@
|
||||
[Unit]
|
||||
Description=Dogtamer Container
|
||||
After=docker.service
|
||||
Requires=docker.service
|
||||
|
||||
[Service]
|
||||
TimeoutStartSec=0
|
||||
Restart=always
|
||||
ExecStartPre=-/usr/bin/docker stop dogtamer
|
||||
ExecStartPre=-/usr/bin/docker rm dogtamer
|
||||
ExecStart=/usr/bin/docker run -p 5566:5566 \
|
||||
--name dogtamer dogtamer:latest
|
||||
|
||||
[Install]
|
||||
WantedBy=multi-user.target
|
||||
14
go.mod
Normal file
14
go.mod
Normal file
@ -0,0 +1,14 @@
|
||||
module github.uio.no/torjus/dogtamer
|
||||
|
||||
go 1.17
|
||||
|
||||
require github.com/urfave/cli/v2 v2.3.0
|
||||
|
||||
require (
|
||||
github.com/cpuguy83/go-md2man/v2 v2.0.1 // indirect
|
||||
github.com/nareix/joy5 v0.0.0-20210317075623-2c912ca30590 // indirect
|
||||
github.com/russross/blackfriday/v2 v2.1.0 // indirect
|
||||
go.uber.org/atomic v1.9.0 // indirect
|
||||
go.uber.org/multierr v1.7.0 // indirect
|
||||
go.uber.org/zap v1.19.0 // indirect
|
||||
)
|
||||
53
go.sum
Normal file
53
go.sum
Normal file
@ -0,0 +1,53 @@
|
||||
github.com/BurntSushi/toml v0.3.1/go.mod h1:xHWCNGjB5oqiDr8zfno3MHue2Ht5sIBksp03qcyfWMU=
|
||||
github.com/benbjohnson/clock v1.1.0/go.mod h1:J11/hYXuz8f4ySSvYwY0FKfm+ezbsZBKZxNJlLklBHA=
|
||||
github.com/cpuguy83/go-md2man/v2 v2.0.0-20190314233015-f79a8a8ca69d/go.mod h1:maD7wRr/U5Z6m/iR4s+kqSMx2CaBsrgA7czyZG/E6dU=
|
||||
github.com/cpuguy83/go-md2man/v2 v2.0.1 h1:r/myEWzV9lfsM1tFLgDyu0atFtJ1fXn261LKYj/3DxU=
|
||||
github.com/cpuguy83/go-md2man/v2 v2.0.1/go.mod h1:tgQtvFlXSQOSOSIRvRPT7W67SCa46tRHOmNcaadrF8o=
|
||||
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||
github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo=
|
||||
github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ=
|
||||
github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI=
|
||||
github.com/nareix/joy5 v0.0.0-20210317075623-2c912ca30590 h1:PnxRU8L8Y2q82vFC2QdNw23Dm2u6WrjecIdpXjiYbXM=
|
||||
github.com/nareix/joy5 v0.0.0-20210317075623-2c912ca30590/go.mod h1:XmAOs6UJXpNXRwKk+KY/nv5kL6xXYXyellk+A1pTlko=
|
||||
github.com/pkg/errors v0.8.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
|
||||
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
||||
github.com/russross/blackfriday/v2 v2.0.1/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM=
|
||||
github.com/russross/blackfriday/v2 v2.1.0 h1:JIOH55/0cWyOuilr9/qlrm0BSXldqnqwMsf35Ld67mk=
|
||||
github.com/russross/blackfriday/v2 v2.1.0/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM=
|
||||
github.com/shurcooL/sanitized_anchor_name v1.0.0/go.mod h1:1NzhyTcUVG4SuEtjjoZeVRXNmyL/1OwPU0+IJeTBvfc=
|
||||
github.com/spf13/cobra v0.0.4-0.20190109003409-7547e83b2d85/go.mod h1:1l0Ry5zgKvJasoi3XT1TypsSe7PqH0Sj9dhYf7v3XqQ=
|
||||
github.com/spf13/pflag v1.0.4-0.20181223182923-24fa6976df40/go.mod h1:DYY7MBk1bdzusC3SYhjObp+wFpr4gzcvqqNjLnInEg4=
|
||||
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
|
||||
github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=
|
||||
github.com/stretchr/testify v1.4.0/go.mod h1:j7eGeouHqKxXV5pUuKE4zz7dFj8WfuZ+81PSLYec5m4=
|
||||
github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
|
||||
github.com/urfave/cli/v2 v2.3.0 h1:qph92Y649prgesehzOrQjdWyxFOp/QVM+6imKHad91M=
|
||||
github.com/urfave/cli/v2 v2.3.0/go.mod h1:LJmUH05zAU44vOAcrfzZQKsZbVcdbOG8rtL3/XcUArI=
|
||||
go.uber.org/atomic v1.7.0/go.mod h1:fEN4uk6kAWBTFdckzkM89CLk9XfWZrxpCo0nPH17wJc=
|
||||
go.uber.org/atomic v1.9.0 h1:ECmE8Bn/WFTYwEW/bpKD3M8VtR/zQVbavAoalC1PYyE=
|
||||
go.uber.org/atomic v1.9.0/go.mod h1:fEN4uk6kAWBTFdckzkM89CLk9XfWZrxpCo0nPH17wJc=
|
||||
go.uber.org/goleak v1.1.10/go.mod h1:8a7PlsEVH3e/a/GLqe5IIrQx6GzcnRmZEufDUTk4A7A=
|
||||
go.uber.org/multierr v1.6.0/go.mod h1:cdWPpRnG4AhwMwsgIHip0KRBQjJy5kYEpYjJxpXp9iU=
|
||||
go.uber.org/multierr v1.7.0 h1:zaiO/rmgFjbmCXdSYJWQcdvOCsthmdaHfr3Gm2Kx4Ec=
|
||||
go.uber.org/multierr v1.7.0/go.mod h1:7EAYxJLBy9rStEaz58O2t4Uvip6FSURkq8/ppBp95ak=
|
||||
go.uber.org/zap v1.19.0 h1:mZQZefskPPCMIBCSEH0v2/iUqqLrYtaeqwD6FUGUnFE=
|
||||
go.uber.org/zap v1.19.0/go.mod h1:xg/QME4nWcxGxrpdeYfq7UvYrLh66cuVKdrbD1XF/NI=
|
||||
golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
|
||||
golang.org/x/lint v0.0.0-20190930215403-16217165b5de/go.mod h1:6SW0HCj/g11FgYtHlgUYUwCkIfeOF89ocIRzGO/8vkc=
|
||||
golang.org/x/net v0.0.0-20190311183353-d8887717615a/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=
|
||||
golang.org/x/net v0.0.0-20190522155817-f3200d17e092/go.mod h1:HSz+uSET+XFnRR8LxR5pz3Of3rY3CfYBVs4xY44aLks=
|
||||
golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=
|
||||
golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||
golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
|
||||
golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
|
||||
golang.org/x/tools v0.0.0-20190311212946-11955173bddd/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs=
|
||||
golang.org/x/tools v0.0.0-20191108193012-7d206e10da11/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=
|
||||
golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
||||
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||
gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||
gopkg.in/yaml.v2 v2.2.2/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
|
||||
gopkg.in/yaml.v2 v2.2.3/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
|
||||
gopkg.in/yaml.v2 v2.2.8/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
|
||||
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||
gopkg.in/yaml.v3 v3.0.0-20210107192922-496545a6307b/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||
339
server/rtmp.go
Normal file
339
server/rtmp.go
Normal file
@ -0,0 +1,339 @@
|
||||
package server
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"io"
|
||||
"net"
|
||||
"regexp"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"github.com/nareix/joy5/av"
|
||||
"github.com/nareix/joy5/format/rtmp"
|
||||
"go.uber.org/zap"
|
||||
)
|
||||
|
||||
type RTMPClient struct {
|
||||
c *rtmp.Conn
|
||||
nc net.Conn
|
||||
server *RTMPServer
|
||||
stream *Stream
|
||||
ctx context.Context
|
||||
ctxCancel context.CancelFunc
|
||||
|
||||
keyframeSent bool
|
||||
|
||||
packetsChan chan av.Packet
|
||||
}
|
||||
|
||||
func NewRTMPClient(rs *RTMPServer, stream *Stream, c *rtmp.Conn, nc net.Conn) *RTMPClient {
|
||||
ctx, cancel := context.WithCancel(stream.ctx)
|
||||
return &RTMPClient{
|
||||
c: c,
|
||||
nc: nc,
|
||||
server: rs,
|
||||
stream: stream,
|
||||
packetsChan: make(chan av.Packet, 10),
|
||||
ctx: ctx,
|
||||
ctxCancel: cancel,
|
||||
}
|
||||
}
|
||||
|
||||
type Stream struct {
|
||||
Name string
|
||||
Clients []*RTMPClient
|
||||
clientsLock sync.Mutex
|
||||
server *RTMPServer
|
||||
nc net.Conn
|
||||
c *rtmp.Conn
|
||||
ctx context.Context
|
||||
ctxCancel context.CancelFunc
|
||||
startTime time.Time
|
||||
|
||||
metadataPacket *av.Packet
|
||||
h264DecoderConfigPacket *av.Packet
|
||||
aacDecoderConfigPacket *av.Packet
|
||||
}
|
||||
|
||||
func (s *Stream) AddClient(client *RTMPClient) {
|
||||
s.clientsLock.Lock()
|
||||
defer s.clientsLock.Unlock()
|
||||
s.Clients = append(s.Clients, client)
|
||||
s.server.Logger.Infow("Added client to stream.", "stream_name", s.Name)
|
||||
}
|
||||
|
||||
func (s *Stream) RemoveClient(client *RTMPClient) {
|
||||
for i, c := range s.Clients {
|
||||
if c == client {
|
||||
s.clientsLock.Lock()
|
||||
defer s.clientsLock.Unlock()
|
||||
copy(s.Clients[i:], s.Clients[i+1:])
|
||||
s.Clients[len(s.Clients)-1] = nil // or the zero value of T
|
||||
s.Clients = s.Clients[:len(s.Clients)-1]
|
||||
continue
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (s *Stream) handleBroadcaster(c *rtmp.Conn, nc net.Conn) {
|
||||
s.nc = nc
|
||||
s.c = c
|
||||
|
||||
logger := s.server.Logger.Desugar()
|
||||
s.startTime = time.Now()
|
||||
|
||||
// Ensure cleanup
|
||||
defer func() {
|
||||
// Ensure all clients have been removed
|
||||
for len(s.Clients) > 0 {
|
||||
time.Sleep(200 * time.Millisecond)
|
||||
}
|
||||
// Once all clients have been removed, remove stream from server
|
||||
s.server.RemoveStream(s)
|
||||
s.server.Logger.Infow("Stream has been successfully stopped.", "stream_name", s.Name, "duration_seconds", time.Since(s.startTime))
|
||||
}()
|
||||
|
||||
for {
|
||||
nc.SetReadDeadline(time.Now().Add(5 * time.Second))
|
||||
|
||||
// Check if stream has been cancelled
|
||||
select {
|
||||
case <-s.ctx.Done():
|
||||
// Stream is cancelled
|
||||
s.server.Logger.Debugw("Stream context has been cancelled.", "stream_name", s.Name)
|
||||
nc.Close()
|
||||
return
|
||||
default:
|
||||
}
|
||||
|
||||
pkt, err := c.ReadPacket()
|
||||
if err != nil {
|
||||
// TODO: Moar better pls
|
||||
if err == io.EOF {
|
||||
// Stream ended
|
||||
nc.Close()
|
||||
// Cancel stream context, to ensure all clients are disconnected
|
||||
s.server.Logger.Infow("Stream disconnected cleanly.", "stream_name", s.Name)
|
||||
s.ctxCancel()
|
||||
}
|
||||
nc.Close()
|
||||
}
|
||||
switch pkt.Type {
|
||||
case av.Metadata:
|
||||
s.server.Logger.Debugw("Got metadata packet for stream.", "stream_name", s.Name)
|
||||
s.metadataPacket = &pkt
|
||||
continue
|
||||
case av.AACDecoderConfig:
|
||||
s.server.Logger.Debugw("Got AAC decoder config packet for stream.", "stream_name", s.Name)
|
||||
s.aacDecoderConfigPacket = &pkt
|
||||
continue
|
||||
case av.H264DecoderConfig:
|
||||
s.server.Logger.Debugw("Got h264 decoder config packet for stream.", "stream_name", s.Name)
|
||||
s.h264DecoderConfigPacket = &pkt
|
||||
continue
|
||||
}
|
||||
|
||||
// TODO: Write packet to buffer, to be used read by client handlers
|
||||
for _, client := range s.Clients {
|
||||
select {
|
||||
case client.packetsChan <- pkt:
|
||||
continue
|
||||
default:
|
||||
logger.Info("Client channel is blocking.")
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (client *RTMPClient) handleClient() {
|
||||
server := client.server
|
||||
stream := client.stream
|
||||
|
||||
// ensure cleanup
|
||||
defer func() {
|
||||
client.nc.Close()
|
||||
stream.RemoveClient(client)
|
||||
server.Logger.Infow("Disconnected viewer.", "client_addr", client.nc.RemoteAddr().String(), "stream_name", stream.Name)
|
||||
}()
|
||||
|
||||
server.Logger.Infow("Viewer connected.", "stream", stream.Name, "remote_addr", client.nc.RemoteAddr().String())
|
||||
// First ensure that metadata, and decoder configs are set
|
||||
// TODO: Ensure stream isnt cancelled before sending metadata and decoder configs
|
||||
for stream.aacDecoderConfigPacket == nil || stream.metadataPacket == nil || stream.h264DecoderConfigPacket == nil {
|
||||
server.Logger.Debugw("Viewer is waiting for stream to receive metadata or decoder config packets.", client.nc.RemoteAddr().String())
|
||||
time.Sleep(time.Second)
|
||||
}
|
||||
if err := client.c.WritePacket(*stream.metadataPacket); err != nil {
|
||||
server.Logger.Info("Error writing metadata-packet to viewer.", "err", err)
|
||||
}
|
||||
if err := client.c.WritePacket(*stream.aacDecoderConfigPacket); err != nil {
|
||||
server.Logger.Info("Error writing aacdecoderconfig-packet to viewer.", "err", err)
|
||||
}
|
||||
if err := client.c.WritePacket(*stream.h264DecoderConfigPacket); err != nil {
|
||||
server.Logger.Info("Error writing aacdecoderconfig-packet to viewer.", "err", err)
|
||||
}
|
||||
server.Logger.Debugw("Wrote metadata and decoder configs to viewer.", "client_addr", client.nc.RemoteAddr().String())
|
||||
|
||||
for {
|
||||
var errCount int
|
||||
select {
|
||||
case <-client.ctx.Done():
|
||||
// Client has been cancelled, stop writing packets
|
||||
server.Logger.Infow("Client context cancelled.", "client_addr", client.nc.RemoteAddr().String(), "stream_name", stream.Name)
|
||||
return
|
||||
default:
|
||||
}
|
||||
pkt := <-client.packetsChan
|
||||
|
||||
// Wait for keyframe before sending video
|
||||
if !pkt.IsKeyFrame && !client.keyframeSent {
|
||||
continue
|
||||
} else {
|
||||
// Assume that the packet will get sent
|
||||
client.keyframeSent = true
|
||||
}
|
||||
server.Logger.Debugw("Got keyframe. Sending video.", "client_addr", client.nc.RemoteAddr().String(), "stream_name", stream.Name)
|
||||
|
||||
client.nc.SetWriteDeadline(time.Now().Add(1 * time.Second))
|
||||
if err := client.c.WritePacket(pkt); err != nil {
|
||||
netErr, ok := err.(*net.OpError)
|
||||
if ok && netErr.Temporary() {
|
||||
// Temporary error, so we should retry?
|
||||
errCount++
|
||||
time.Sleep(1 * time.Second)
|
||||
continue
|
||||
}
|
||||
if errCount > 9 {
|
||||
server.Logger.Warnw("Too many errors from viewer. Disconnecting.", "client_addr", client.nc.RemoteAddr().String(), "stream_name", stream.Name)
|
||||
return
|
||||
}
|
||||
// TODO: Handle client disconnections properly
|
||||
server.Logger.Debugw("Error while writing packet to viewer.", "client_addr", client.nc.RemoteAddr().String(), "error_message", err)
|
||||
return
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
type RTMPServer struct {
|
||||
ListenAddr string
|
||||
Logger *zap.SugaredLogger
|
||||
streamsLock sync.Mutex
|
||||
streams map[string]*Stream
|
||||
ctx context.Context
|
||||
ctxCancel context.CancelFunc
|
||||
}
|
||||
|
||||
func NewRTMPServer(ctx context.Context, addr string) *RTMPServer {
|
||||
serverCtx, cancel := context.WithCancel(ctx)
|
||||
rs := &RTMPServer{
|
||||
ListenAddr: addr,
|
||||
Logger: zap.NewNop().Sugar(),
|
||||
streams: make(map[string]*Stream),
|
||||
ctx: serverCtx,
|
||||
ctxCancel: cancel,
|
||||
}
|
||||
return rs
|
||||
}
|
||||
|
||||
func (rs *RTMPServer) AddStream(stream *Stream) {
|
||||
stream.server = rs
|
||||
rs.streamsLock.Lock()
|
||||
defer rs.streamsLock.Unlock()
|
||||
rs.streams[stream.Name] = stream
|
||||
rs.Logger.Infow("New stream added to server.", "stream_name", stream.Name)
|
||||
}
|
||||
|
||||
func (rs *RTMPServer) RemoveStream(stream *Stream) {
|
||||
rs.streamsLock.Lock()
|
||||
defer rs.streamsLock.Unlock()
|
||||
delete(rs.streams, stream.Name)
|
||||
}
|
||||
|
||||
func (rs *RTMPServer) Listen() error {
|
||||
listener, err := net.Listen("tcp", rs.ListenAddr)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer listener.Close()
|
||||
|
||||
s := rtmp.NewServer()
|
||||
s.HandleConn = rs.handleConn
|
||||
|
||||
rs.Logger.Info("RTMP server listening for connections.")
|
||||
go func() {
|
||||
<-rs.ctx.Done()
|
||||
listener.Close()
|
||||
}()
|
||||
|
||||
for {
|
||||
nc, err := listener.Accept()
|
||||
if err != nil {
|
||||
if errors.Is(err, net.ErrClosed) {
|
||||
// Wait for streams to disconnect
|
||||
for len(rs.streams) > 0 {
|
||||
time.Sleep(200 * time.Millisecond)
|
||||
}
|
||||
rs.Logger.Infow("Server stopped.")
|
||||
return nil
|
||||
}
|
||||
// Connection not closed, wait and try again
|
||||
rs.Logger.Warnw("Error while accepting connection", "error", err)
|
||||
time.Sleep(time.Second)
|
||||
continue
|
||||
}
|
||||
|
||||
go s.HandleNetConn(nc)
|
||||
}
|
||||
}
|
||||
|
||||
func (rs *RTMPServer) handleConn(c *rtmp.Conn, nc net.Conn) {
|
||||
rs.Logger.Debugw("Incoming connection.", "remote_addr", nc.RemoteAddr(), "url", c.URL.String(), "publishing", c.Publishing)
|
||||
viewerPattern := `^/view/(.+)$`
|
||||
viewerRegexp := regexp.MustCompile(viewerPattern)
|
||||
|
||||
viewerMatch := viewerRegexp.FindStringSubmatch(c.URL.Path)
|
||||
if viewerMatch != nil {
|
||||
name := viewerMatch[1]
|
||||
stream, ok := rs.streams[name]
|
||||
if ok {
|
||||
client := NewRTMPClient(rs, stream, c, nc)
|
||||
stream.AddClient(client)
|
||||
client.handleClient()
|
||||
return
|
||||
}
|
||||
|
||||
// No such stream, disconnect viewer
|
||||
nc.Close()
|
||||
rs.Logger.Infow("Closed connection to non-existing stream.", "requested_stream", name)
|
||||
}
|
||||
|
||||
// Get stream name
|
||||
streamPattern := `^/stream/(.+)$`
|
||||
streamRegexp := regexp.MustCompile(streamPattern)
|
||||
match := streamRegexp.FindStringSubmatch(c.URL.Path)
|
||||
if match != nil {
|
||||
name := match[1]
|
||||
|
||||
_, ok := rs.streams[name]
|
||||
if ok {
|
||||
// Stream already exists, disconnect this
|
||||
nc.Close()
|
||||
rs.Logger.Warnw("Connection attempted to stream to stream-key which is already broadcasting.", "stream_key", name)
|
||||
return
|
||||
}
|
||||
|
||||
streamCtx, cancel := context.WithCancel(rs.ctx)
|
||||
|
||||
stream := &Stream{
|
||||
Name: name,
|
||||
ctx: streamCtx,
|
||||
ctxCancel: cancel,
|
||||
}
|
||||
rs.AddStream(stream)
|
||||
stream.handleBroadcaster(c, nc)
|
||||
}
|
||||
|
||||
// Stream URL is invalid, disconnect
|
||||
nc.Close()
|
||||
}
|
||||
33
version.go
Normal file
33
version.go
Normal file
@ -0,0 +1,33 @@
|
||||
package dogtamer
|
||||
|
||||
import "fmt"
|
||||
|
||||
const Version string = "v0.1.2"
|
||||
|
||||
var Build string
|
||||
|
||||
func GetVersion() string {
|
||||
if Build == "" {
|
||||
return fmt.Sprintf("Dogtamer Version %s", Version)
|
||||
} else {
|
||||
return fmt.Sprintf("Dogtamer Version %s (%s)", Version, Build)
|
||||
}
|
||||
}
|
||||
|
||||
func GetShortVersion() string {
|
||||
var s string
|
||||
if Build == "" {
|
||||
s = Version
|
||||
} else {
|
||||
s = fmt.Sprintf("%s-%s", Version, Build)
|
||||
}
|
||||
return s
|
||||
}
|
||||
|
||||
func GetHeaderVersion() string {
|
||||
if Build == "" {
|
||||
return fmt.Sprintf("Dogtamer/%s", Version)
|
||||
} else {
|
||||
return fmt.Sprintf("Dogtamer/%s-%s", Version, Build)
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user