apps/go2rtc
1
0
Fork 0
mirror of https://github.com/AlexxIT/go2rtc.git synced 2025-10-05 00:12:48 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add support unix socket for api module

Browse Source
This commit is contained in:
Alex X
2023-11-12 17:25:14 +03:00
parent 5d3953a948
commit 28830a697d
1 changed files with 65 additions and 57 deletions
Download Patch File Download Diff File Expand all files Collapse all files

74
internal/api/api.go
View File

@@ -10,6 +10,7 @@ import (
"strconv" "strconv"
"strings" "strings"
"sync" "sync"
"syscall"
"github.com/AlexxIT/go2rtc/internal/app" "github.com/AlexxIT/go2rtc/internal/app"
"github.com/AlexxIT/go2rtc/pkg/shell" "github.com/AlexxIT/go2rtc/pkg/shell"
@@ -28,6 +29,7 @@ func Init() {
TLSListen string `yaml:"tls_listen"` TLSListen string `yaml:"tls_listen"`
TLSCert string `yaml:"tls_cert"` TLSCert string `yaml:"tls_cert"`
TLSKey string `yaml:"tls_key"` TLSKey string `yaml:"tls_key"`
UnixListen string `yaml:"unix_listen"`
} `yaml:"api"` } `yaml:"api"`
} }
@@ -37,7 +39,7 @@ func Init() {
// load config from YAML // load config from YAML
app.LoadConfig(&cfg) app.LoadConfig(&cfg)
if cfg.Mod.Listen == "" { if cfg.Mod.Listen == "" && cfg.Mod.UnixListen == "" && cfg.Mod.TLSListen == "" {
return return
} }
@@ -51,16 +53,6 @@ func Init() {
HandleFunc("api/exit", exitHandler) HandleFunc("api/exit", exitHandler)
HandleFunc("api/restart", restartHandler) HandleFunc("api/restart", restartHandler)
// ensure we can listen without errors
var err error
ln, err = net.Listen("tcp", cfg.Mod.Listen)
if err != nil {
log.Fatal().Err(err).Msg("[api] listen")
return
}
log.Info().Str("addr", cfg.Mod.Listen).Msg("[api] listen")
Handler = http.DefaultServeMux // 4th Handler = http.DefaultServeMux // 4th
if cfg.Mod.Origin == "*" { if cfg.Mod.Origin == "*" {
@@ -75,50 +67,66 @@ func Init() {
Handler = middlewareLog(Handler) // 1st Handler = middlewareLog(Handler) // 1st
} }
go func() { if cfg.Mod.Listen != "" {
s := http.Server{} go listen("tcp", cfg.Mod.Listen)
s.Handler = Handler }
if err = s.Serve(ln); err != nil {
log.Fatal().Err(err).Msg("[api] serve") if cfg.Mod.UnixListen != "" {
_ = syscall.Unlink(cfg.Mod.UnixListen)
go listen("unix", cfg.Mod.UnixListen)
} }
}()
// Initialize the HTTPS server // Initialize the HTTPS server
if cfg.Mod.TLSListen != "" && cfg.Mod.TLSCert != "" && cfg.Mod.TLSKey != "" { if cfg.Mod.TLSListen != "" && cfg.Mod.TLSCert != "" && cfg.Mod.TLSKey != "" {
go tlsListen("tcp", cfg.Mod.TLSListen, cfg.Mod.TLSCert, cfg.Mod.TLSKey)
}
}
func listen(network, address string) {
ln, err := net.Listen(network, address)
if err != nil {
log.Error().Err(err).Msg("[api] listen")
return
}
log.Info().Str("addr", address).Msg("[api] listen")
server := http.Server{Handler: Handler}
if err = server.Serve(ln); err != nil {
log.Fatal().Err(err).Msg("[api] serve")
}
}
func tlsListen(network, address, certFile, keyFile string) {
var cert tls.Certificate var cert tls.Certificate
if strings.IndexByte(cfg.Mod.TLSCert, '\n') < 0 && strings.IndexByte(cfg.Mod.TLSKey, '\n') < 0 { var err error
if strings.IndexByte(certFile, '\n') < 0 && strings.IndexByte(keyFile, '\n') < 0 {
// check if file path // check if file path
cert, err = tls.LoadX509KeyPair(cfg.Mod.TLSCert, cfg.Mod.TLSKey) cert, err = tls.LoadX509KeyPair(certFile, keyFile)
} else { } else {
// if text file content // if text file content
cert, err = tls.X509KeyPair([]byte(cfg.Mod.TLSCert), []byte(cfg.Mod.TLSKey)) cert, err = tls.X509KeyPair([]byte(certFile), []byte(keyFile))
} }
if err != nil { if err != nil {
log.Error().Err(err).Caller().Send() log.Error().Err(err).Caller().Send()
return return
} }
tlsListener, err := net.Listen("tcp", cfg.Mod.TLSListen) ln, err := net.Listen(network, address)
if err != nil { if err != nil {
log.Fatal().Err(err).Caller().Send() log.Error().Err(err).Msg("[api] tls listen")
return return
} }
log.Info().Str("addr", cfg.Mod.TLSListen).Msg("[api] tls listen") log.Info().Str("addr", address).Msg("[api] tls listen")
tlsServer := &http.Server{ server := &http.Server{
Handler: Handler, Handler: Handler,
TLSConfig: &tls.Config{ TLSConfig: &tls.Config{Certificates: []tls.Certificate{cert}},
Certificates: []tls.Certificate{cert},
},
} }
if err = server.ServeTLS(ln, "", ""); err != nil {
go func() {
if err := tlsServer.ServeTLS(tlsListener, "", ""); err != nil {
log.Fatal().Err(err).Msg("[api] tls serve") log.Fatal().Err(err).Msg("[api] tls serve")
} }
}()
}
} }
func Port() int { func Port() int {
@@ -187,7 +195,7 @@ func middlewareLog(next http.Handler) http.Handler {
func middlewareAuth(username, password string, next http.Handler) http.Handler { func middlewareAuth(username, password string, next http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
if !strings.HasPrefix(r.RemoteAddr, "127.") && !strings.HasPrefix(r.RemoteAddr, "[::1]") { if !strings.HasPrefix(r.RemoteAddr, "127.") && !strings.HasPrefix(r.RemoteAddr, "[::1]") && r.RemoteAddr != "@" {
user, pass, ok := r.BasicAuth() user, pass, ok := r.BasicAuth()
if !ok || user != username || pass != password { if !ok || user != username || pass != password {
w.Header().Set("Www-Authenticate", `Basic realm="go2rtc"`) w.Header().Set("Www-Authenticate", `Basic realm="go2rtc"`)