aler9
5 years ago
committed by
Alessandro Ros
21 changed files with 1139 additions and 227 deletions
@ -0,0 +1,543 @@ |
|||||||
|
package core |
||||||
|
|
||||||
|
import ( |
||||||
|
"context" |
||||||
|
"encoding/json" |
||||||
|
"net" |
||||||
|
"net/http" |
||||||
|
"reflect" |
||||||
|
"sync" |
||||||
|
"time" |
||||||
|
|
||||||
|
"github.com/gin-gonic/gin" |
||||||
|
|
||||||
|
"github.com/aler9/rtsp-simple-server/internal/conf" |
||||||
|
"github.com/aler9/rtsp-simple-server/internal/logger" |
||||||
|
) |
||||||
|
|
||||||
|
func fillStruct(dest interface{}, source interface{}) { |
||||||
|
rvsource := reflect.ValueOf(source) |
||||||
|
rvdest := reflect.ValueOf(dest) |
||||||
|
nf := rvsource.NumField() |
||||||
|
for i := 0; i < nf; i++ { |
||||||
|
fnew := rvsource.Field(i) |
||||||
|
if !fnew.IsNil() { |
||||||
|
f := rvdest.Elem().FieldByName(rvsource.Type().Field(i).Name) |
||||||
|
if f.Kind() == reflect.Ptr { |
||||||
|
f.Set(fnew) |
||||||
|
} else { |
||||||
|
f.Set(fnew.Elem()) |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
func cloneStruct(dest interface{}, source interface{}) { |
||||||
|
enc, _ := json.Marshal(dest) |
||||||
|
_ = json.Unmarshal(enc, source) |
||||||
|
} |
||||||
|
|
||||||
|
func loadConfData(ctx *gin.Context) (interface{}, error) { |
||||||
|
var in struct { |
||||||
|
// general
|
||||||
|
LogLevel *string `json:"logLevel"` |
||||||
|
LogDestinations *[]string `json:"logDestinations"` |
||||||
|
LogFile *string `json:"logFile"` |
||||||
|
ReadTimeout *time.Duration `json:"readTimeout"` |
||||||
|
WriteTimeout *time.Duration `json:"writeTimeout"` |
||||||
|
ReadBufferCount *int `json:"readBufferCount"` |
||||||
|
API *bool `json:"api"` |
||||||
|
APIAddress *string `json:"apiAddress"` |
||||||
|
Metrics *bool `json:"metrics"` |
||||||
|
MetricsAddress *string `json:"metricsAddress"` |
||||||
|
PPROF *bool `json:"pprof"` |
||||||
|
PPROFAddress *string `json:"pprofAddress"` |
||||||
|
RunOnConnect *string `json:"runOnConnect"` |
||||||
|
RunOnConnectRestart *bool `json:"runOnConnectRestart"` |
||||||
|
|
||||||
|
// rtsp
|
||||||
|
RTSPDisable *bool `json:"rtspDisable"` |
||||||
|
Protocols *[]string `json:"protocols"` |
||||||
|
Encryption *string `json:"encryption"` |
||||||
|
RTSPAddress *string `json:"rtspAddress"` |
||||||
|
RTSPSAddress *string `json:"rtspsAddress"` |
||||||
|
RTPAddress *string `json:"rtpAddress"` |
||||||
|
RTCPAddress *string `json:"rtcpAddress"` |
||||||
|
MulticastIPRange *string `json:"multicastIPRange"` |
||||||
|
MulticastRTPPort *int `json:"multicastRTPPort"` |
||||||
|
MulticastRTCPPort *int `json:"multicastRTCPPort"` |
||||||
|
ServerKey *string `json:"serverKey"` |
||||||
|
ServerCert *string `json:"serverCert"` |
||||||
|
AuthMethods *[]string `json:"authMethods"` |
||||||
|
ReadBufferSize *int `json:"readBufferSize"` |
||||||
|
|
||||||
|
// rtmp
|
||||||
|
RTMPDisable *bool `json:"rtmpDisable"` |
||||||
|
RTMPAddress *string `json:"rtmpAddress"` |
||||||
|
|
||||||
|
// hls
|
||||||
|
HLSDisable *bool `json:"hlsDisable"` |
||||||
|
HLSAddress *string `json:"hlsAddress"` |
||||||
|
HLSAlwaysRemux *bool `json:"hlsAlwaysRemux"` |
||||||
|
HLSSegmentCount *int `json:"hlsSegmentCount"` |
||||||
|
HLSSegmentDuration *time.Duration `json:"hlsSegmentDuration"` |
||||||
|
HLSAllowOrigin *string `json:"hlsAllowOrigin"` |
||||||
|
} |
||||||
|
err := json.NewDecoder(ctx.Request.Body).Decode(&in) |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
|
||||||
|
return in, err |
||||||
|
} |
||||||
|
|
||||||
|
func loadConfPathData(ctx *gin.Context) (interface{}, error) { |
||||||
|
var in struct { |
||||||
|
// source
|
||||||
|
Source *string `json:"source"` |
||||||
|
SourceProtocol *string `json:"sourceProtocol"` |
||||||
|
SourceAnyPortEnable *bool `json:"sourceAnyPortEnable"` |
||||||
|
SourceFingerprint *string `json:"sourceFingerprint"` |
||||||
|
SourceOnDemand *bool `json:"sourceOnDemand"` |
||||||
|
SourceOnDemandStartTimeout *time.Duration `json:"sourceOnDemandStartTimeout"` |
||||||
|
SourceOnDemandCloseAfter *time.Duration `json:"sourceOnDemandCloseAfter"` |
||||||
|
SourceRedirect *string `json:"sourceRedirect"` |
||||||
|
DisablePublisherOverride *bool `json:"disablePublisherOverride"` |
||||||
|
Fallback *string `json:"fallback"` |
||||||
|
|
||||||
|
// authentication
|
||||||
|
PublishUser *string `json:"publishUser"` |
||||||
|
PublishPass *string `json:"publishPass"` |
||||||
|
PublishIPs *[]string `json:"publishIPs"` |
||||||
|
ReadUser *string `json:"readUser"` |
||||||
|
ReadPass *string `json:"readPass"` |
||||||
|
ReadIPs *[]string `json:"readIPs"` |
||||||
|
|
||||||
|
// custom commands
|
||||||
|
RunOnInit *string `json:"runOnInit"` |
||||||
|
RunOnInitRestart *bool `json:"runOnInitRestart"` |
||||||
|
RunOnDemand *string `json:"runOnDemand"` |
||||||
|
RunOnDemandRestart *bool `json:"runOnDemandRestart"` |
||||||
|
RunOnDemandStartTimeout *time.Duration `json:"runOnDemandStartTimeout"` |
||||||
|
RunOnDemandCloseAfter *time.Duration `json:"runOnDemandCloseAfter"` |
||||||
|
RunOnPublish *string `json:"runOnPublish"` |
||||||
|
RunOnPublishRestart *bool `json:"runOnPublishRestart"` |
||||||
|
RunOnRead *string `json:"runOnRead"` |
||||||
|
RunOnReadRestart *bool `json:"runOnReadRestart"` |
||||||
|
} |
||||||
|
err := json.NewDecoder(ctx.Request.Body).Decode(&in) |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
|
||||||
|
return in, err |
||||||
|
} |
||||||
|
|
||||||
|
type apiPathsItem struct { |
||||||
|
Name string `json:"name"` |
||||||
|
ConfName string `json:"confName"` |
||||||
|
Conf *conf.PathConf `json:"conf"` |
||||||
|
Source interface{} `json:"source"` |
||||||
|
SourceReady bool `json:"sourceReady"` |
||||||
|
Readers []interface{} `json:"readers"` |
||||||
|
} |
||||||
|
|
||||||
|
type apiPathsListData struct { |
||||||
|
Items []apiPathsItem `json:"items"` |
||||||
|
} |
||||||
|
|
||||||
|
type apiPathsListRes1 struct { |
||||||
|
Paths map[string]*path |
||||||
|
Err error |
||||||
|
} |
||||||
|
|
||||||
|
type apiPathsListReq1 struct { |
||||||
|
Res chan apiPathsListRes1 |
||||||
|
} |
||||||
|
|
||||||
|
type apiPathsListRes2 struct { |
||||||
|
Err error |
||||||
|
} |
||||||
|
|
||||||
|
type apiPathsListReq2 struct { |
||||||
|
Data *apiPathsListData |
||||||
|
Res chan apiPathsListRes2 |
||||||
|
} |
||||||
|
|
||||||
|
type apiRTSPSessionsItem struct { |
||||||
|
ID string `json:"id"` |
||||||
|
RemoteAddr string `json:"remoteAddr"` |
||||||
|
} |
||||||
|
|
||||||
|
type apiRTSPSessionsListData struct { |
||||||
|
Items []apiRTSPSessionsItem `json:"items"` |
||||||
|
} |
||||||
|
|
||||||
|
type apiRTSPSessionsListRes struct { |
||||||
|
Err error |
||||||
|
} |
||||||
|
|
||||||
|
type apiRTSPSessionsListReq struct { |
||||||
|
Data *apiRTSPSessionsListData |
||||||
|
} |
||||||
|
|
||||||
|
type apiRTSPSessionsKickRes struct { |
||||||
|
Err error |
||||||
|
} |
||||||
|
|
||||||
|
type apiRTSPSessionsKickReq struct { |
||||||
|
ID string |
||||||
|
} |
||||||
|
|
||||||
|
type apiRTMPConnsListItem struct { |
||||||
|
ID string `json:"id"` |
||||||
|
RemoteAddr string `json:"remoteAddr"` |
||||||
|
} |
||||||
|
|
||||||
|
type apiRTMPConnsListData struct { |
||||||
|
Items []apiRTMPConnsListItem `json:"items"` |
||||||
|
} |
||||||
|
|
||||||
|
type apiRTMPConnsListRes struct { |
||||||
|
Err error |
||||||
|
} |
||||||
|
|
||||||
|
type apiRTMPConnsListReq struct { |
||||||
|
Data *apiRTMPConnsListData |
||||||
|
Res chan apiRTMPConnsListRes |
||||||
|
} |
||||||
|
|
||||||
|
type apiRTMPConnsKickRes struct { |
||||||
|
Err error |
||||||
|
} |
||||||
|
|
||||||
|
type apiRTMPConnsKickReq struct { |
||||||
|
ID string |
||||||
|
Res chan apiRTMPConnsKickRes |
||||||
|
} |
||||||
|
|
||||||
|
type apiParent interface { |
||||||
|
Log(logger.Level, string, ...interface{}) |
||||||
|
OnAPIConfigSet(conf *conf.Conf) |
||||||
|
} |
||||||
|
|
||||||
|
type api struct { |
||||||
|
conf *conf.Conf |
||||||
|
pathManager *pathManager |
||||||
|
rtspServer *rtspServer |
||||||
|
rtspsServer *rtspServer |
||||||
|
rtmpServer *rtmpServer |
||||||
|
parent apiParent |
||||||
|
|
||||||
|
mutex sync.Mutex |
||||||
|
s *http.Server |
||||||
|
} |
||||||
|
|
||||||
|
func newAPI( |
||||||
|
address string, |
||||||
|
conf *conf.Conf, |
||||||
|
pathManager *pathManager, |
||||||
|
rtspServer *rtspServer, |
||||||
|
rtspsServer *rtspServer, |
||||||
|
rtmpServer *rtmpServer, |
||||||
|
parent apiParent, |
||||||
|
) (*api, error) { |
||||||
|
ln, err := net.Listen("tcp", address) |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
|
||||||
|
a := &api{ |
||||||
|
conf: conf, |
||||||
|
pathManager: pathManager, |
||||||
|
rtspServer: rtspServer, |
||||||
|
rtspsServer: rtspsServer, |
||||||
|
rtmpServer: rtmpServer, |
||||||
|
parent: parent, |
||||||
|
} |
||||||
|
|
||||||
|
gin.SetMode(gin.ReleaseMode) |
||||||
|
router := gin.New() |
||||||
|
|
||||||
|
router.GET("/config/get", a.onConfigGet) |
||||||
|
router.POST("/config/set", a.onConfigSet) |
||||||
|
router.POST("/config/paths/add/:name", a.onConfigPathsAdd) |
||||||
|
router.POST("/config/paths/edit/:name", a.onConfigPathsEdit) |
||||||
|
router.POST("/config/paths/delete/:name", a.onConfigPathsDelete) |
||||||
|
router.GET("/paths/list", a.onPathsList) |
||||||
|
router.GET("/rtspsessions/list", a.onRTSPSessionsList) |
||||||
|
router.POST("/rtspsessions/kick/:id", a.onRTSPSessionsKick) |
||||||
|
router.GET("/rtmpconns/list", a.onRTMPConnsList) |
||||||
|
router.POST("/rtmpconns/kick/:id", a.onRTMPConnsKick) |
||||||
|
|
||||||
|
a.s = &http.Server{ |
||||||
|
Handler: router, |
||||||
|
} |
||||||
|
|
||||||
|
go a.s.Serve(ln) |
||||||
|
|
||||||
|
a.log(logger.Info, "listener opened on "+address) |
||||||
|
|
||||||
|
return a, nil |
||||||
|
} |
||||||
|
|
||||||
|
func (a *api) close() { |
||||||
|
a.s.Shutdown(context.Background()) |
||||||
|
a.log(logger.Info, "closed") |
||||||
|
} |
||||||
|
|
||||||
|
// Log is the main logging function.
|
||||||
|
func (a *api) log(level logger.Level, format string, args ...interface{}) { |
||||||
|
a.parent.Log(level, "[API] "+format, args...) |
||||||
|
} |
||||||
|
|
||||||
|
func (a *api) onConfigGet(ctx *gin.Context) { |
||||||
|
a.mutex.Lock() |
||||||
|
c := a.conf |
||||||
|
a.mutex.Unlock() |
||||||
|
|
||||||
|
ctx.JSON(http.StatusOK, c) |
||||||
|
} |
||||||
|
|
||||||
|
func (a *api) onConfigSet(ctx *gin.Context) { |
||||||
|
in, err := loadConfData(ctx) |
||||||
|
if err != nil { |
||||||
|
ctx.AbortWithStatus(http.StatusBadRequest) |
||||||
|
return |
||||||
|
} |
||||||
|
|
||||||
|
a.mutex.Lock() |
||||||
|
var newConf conf.Conf |
||||||
|
cloneStruct(a.conf, &newConf) |
||||||
|
a.mutex.Unlock() |
||||||
|
|
||||||
|
fillStruct(&newConf, in) |
||||||
|
|
||||||
|
err = newConf.CheckAndFillMissing() |
||||||
|
if err != nil { |
||||||
|
ctx.AbortWithStatus(http.StatusBadRequest) |
||||||
|
return |
||||||
|
} |
||||||
|
|
||||||
|
// since reloading the configuration can cause the shutdown of the API,
|
||||||
|
// call it in a goroutine
|
||||||
|
go a.parent.OnAPIConfigSet(&newConf) |
||||||
|
|
||||||
|
ctx.Status(http.StatusOK) |
||||||
|
} |
||||||
|
|
||||||
|
func (a *api) onConfigPathsAdd(ctx *gin.Context) { |
||||||
|
in, err := loadConfPathData(ctx) |
||||||
|
if err != nil { |
||||||
|
ctx.AbortWithStatus(http.StatusBadRequest) |
||||||
|
return |
||||||
|
} |
||||||
|
|
||||||
|
name := ctx.Param("name") |
||||||
|
|
||||||
|
a.mutex.Lock() |
||||||
|
var newConf conf.Conf |
||||||
|
cloneStruct(a.conf, &newConf) |
||||||
|
a.mutex.Unlock() |
||||||
|
|
||||||
|
if _, ok := newConf.Paths[name]; ok { |
||||||
|
ctx.AbortWithStatus(http.StatusBadRequest) |
||||||
|
return |
||||||
|
} |
||||||
|
newConfPath := &conf.PathConf{} |
||||||
|
|
||||||
|
fillStruct(newConfPath, in) |
||||||
|
newConf.Paths[name] = newConfPath |
||||||
|
|
||||||
|
err = newConf.CheckAndFillMissing() |
||||||
|
if err != nil { |
||||||
|
ctx.AbortWithStatus(http.StatusBadRequest) |
||||||
|
return |
||||||
|
} |
||||||
|
|
||||||
|
// since reloading the configuration can cause the shutdown of the API,
|
||||||
|
// call it in a goroutine
|
||||||
|
go a.parent.OnAPIConfigSet(&newConf) |
||||||
|
|
||||||
|
ctx.Status(http.StatusOK) |
||||||
|
} |
||||||
|
|
||||||
|
func (a *api) onConfigPathsEdit(ctx *gin.Context) { |
||||||
|
in, err := loadConfPathData(ctx) |
||||||
|
if err != nil { |
||||||
|
ctx.AbortWithStatus(http.StatusBadRequest) |
||||||
|
return |
||||||
|
} |
||||||
|
|
||||||
|
name := ctx.Param("name") |
||||||
|
|
||||||
|
a.mutex.Lock() |
||||||
|
var newConf conf.Conf |
||||||
|
cloneStruct(a.conf, &newConf) |
||||||
|
a.mutex.Unlock() |
||||||
|
|
||||||
|
newConfPath, ok := newConf.Paths[name] |
||||||
|
if !ok { |
||||||
|
ctx.AbortWithStatus(http.StatusBadRequest) |
||||||
|
return |
||||||
|
} |
||||||
|
|
||||||
|
fillStruct(newConfPath, in) |
||||||
|
|
||||||
|
err = newConf.CheckAndFillMissing() |
||||||
|
if err != nil { |
||||||
|
ctx.AbortWithStatus(http.StatusBadRequest) |
||||||
|
return |
||||||
|
} |
||||||
|
|
||||||
|
// since reloading the configuration can cause the shutdown of the API,
|
||||||
|
// call it in a goroutine
|
||||||
|
go a.parent.OnAPIConfigSet(&newConf) |
||||||
|
|
||||||
|
ctx.Status(http.StatusOK) |
||||||
|
} |
||||||
|
|
||||||
|
func (a *api) onConfigPathsDelete(ctx *gin.Context) { |
||||||
|
name := ctx.Param("name") |
||||||
|
|
||||||
|
a.mutex.Lock() |
||||||
|
var newConf conf.Conf |
||||||
|
cloneStruct(a.conf, &newConf) |
||||||
|
a.mutex.Unlock() |
||||||
|
|
||||||
|
if _, ok := newConf.Paths[name]; !ok { |
||||||
|
ctx.AbortWithStatus(http.StatusBadRequest) |
||||||
|
return |
||||||
|
} |
||||||
|
|
||||||
|
delete(newConf.Paths, name) |
||||||
|
|
||||||
|
err := newConf.CheckAndFillMissing() |
||||||
|
if err != nil { |
||||||
|
ctx.AbortWithStatus(http.StatusBadRequest) |
||||||
|
return |
||||||
|
} |
||||||
|
|
||||||
|
// since reloading the configuration can cause the shutdown of the API,
|
||||||
|
// call it in a goroutine
|
||||||
|
go a.parent.OnAPIConfigSet(&newConf) |
||||||
|
|
||||||
|
ctx.Status(http.StatusOK) |
||||||
|
} |
||||||
|
|
||||||
|
func (a *api) onPathsList(ctx *gin.Context) { |
||||||
|
data := apiPathsListData{ |
||||||
|
Items: []apiPathsItem{}, |
||||||
|
} |
||||||
|
|
||||||
|
res := a.pathManager.OnAPIPathsList(apiPathsListReq1{}) |
||||||
|
if res.Err != nil { |
||||||
|
ctx.AbortWithStatus(http.StatusInternalServerError) |
||||||
|
return |
||||||
|
} |
||||||
|
|
||||||
|
for _, pa := range res.Paths { |
||||||
|
pa.OnAPIPathsList(apiPathsListReq2{Data: &data}) |
||||||
|
} |
||||||
|
|
||||||
|
ctx.JSON(http.StatusOK, data) |
||||||
|
} |
||||||
|
|
||||||
|
func (a *api) onRTSPSessionsList(ctx *gin.Context) { |
||||||
|
if a.rtspServer == nil && a.rtspsServer == nil { |
||||||
|
ctx.AbortWithStatus(http.StatusNotFound) |
||||||
|
return |
||||||
|
} |
||||||
|
|
||||||
|
data := apiRTSPSessionsListData{ |
||||||
|
Items: []apiRTSPSessionsItem{}, |
||||||
|
} |
||||||
|
|
||||||
|
if a.rtspServer != nil { |
||||||
|
res := a.rtspServer.OnAPIRTSPSessionsList(apiRTSPSessionsListReq{Data: &data}) |
||||||
|
if res.Err != nil { |
||||||
|
ctx.AbortWithStatus(http.StatusInternalServerError) |
||||||
|
return |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
if a.rtspsServer != nil { |
||||||
|
res := a.rtspsServer.OnAPIRTSPSessionsList(apiRTSPSessionsListReq{Data: &data}) |
||||||
|
if res.Err != nil { |
||||||
|
ctx.AbortWithStatus(http.StatusInternalServerError) |
||||||
|
return |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
ctx.JSON(http.StatusOK, data) |
||||||
|
} |
||||||
|
|
||||||
|
func (a *api) onRTSPSessionsKick(ctx *gin.Context) { |
||||||
|
if a.rtspServer == nil && a.rtspsServer == nil { |
||||||
|
ctx.AbortWithStatus(http.StatusNotFound) |
||||||
|
return |
||||||
|
} |
||||||
|
|
||||||
|
id := ctx.Param("id") |
||||||
|
|
||||||
|
if a.rtspServer != nil { |
||||||
|
res := a.rtspServer.OnAPIRTSPSessionsKick(apiRTSPSessionsKickReq{ID: id}) |
||||||
|
if res.Err == nil { |
||||||
|
ctx.Status(http.StatusOK) |
||||||
|
return |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
if a.rtspsServer != nil { |
||||||
|
res := a.rtspsServer.OnAPIRTSPSessionsKick(apiRTSPSessionsKickReq{ID: id}) |
||||||
|
if res.Err != nil { |
||||||
|
ctx.Status(http.StatusOK) |
||||||
|
return |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
ctx.AbortWithStatus(http.StatusNotFound) |
||||||
|
} |
||||||
|
|
||||||
|
func (a *api) onRTMPConnsList(ctx *gin.Context) { |
||||||
|
if a.rtmpServer == nil { |
||||||
|
ctx.AbortWithStatus(http.StatusNotFound) |
||||||
|
return |
||||||
|
} |
||||||
|
|
||||||
|
data := apiRTMPConnsListData{ |
||||||
|
Items: []apiRTMPConnsListItem{}, |
||||||
|
} |
||||||
|
|
||||||
|
res := a.rtmpServer.OnAPIRTMPConnsList(apiRTMPConnsListReq{Data: &data}) |
||||||
|
if res.Err != nil { |
||||||
|
ctx.AbortWithStatus(http.StatusInternalServerError) |
||||||
|
return |
||||||
|
} |
||||||
|
|
||||||
|
ctx.JSON(http.StatusOK, data) |
||||||
|
} |
||||||
|
|
||||||
|
// OnConfReload is called by core.
|
||||||
|
func (a *api) OnConfReload(conf *conf.Conf) { |
||||||
|
a.mutex.Lock() |
||||||
|
defer a.mutex.Unlock() |
||||||
|
a.conf = conf |
||||||
|
} |
||||||
|
|
||||||
|
func (a *api) onRTMPConnsKick(ctx *gin.Context) { |
||||||
|
if a.rtmpServer == nil { |
||||||
|
ctx.AbortWithStatus(http.StatusNotFound) |
||||||
|
return |
||||||
|
} |
||||||
|
|
||||||
|
id := ctx.Param("id") |
||||||
|
|
||||||
|
res := a.rtmpServer.OnAPIRTMPConnsKick(apiRTMPConnsKickReq{ID: id}) |
||||||
|
if res.Err != nil { |
||||||
|
ctx.AbortWithStatus(http.StatusNotFound) |
||||||
|
return |
||||||
|
} |
||||||
|
|
||||||
|
ctx.Status(http.StatusOK) |
||||||
|
} |
||||||
Loading…
Reference in new issue