Add a configuration parameter making it possible to avoid listening to the web port

This commit is contained in:
Ola Bini 2020-03-25 19:38:32 +00:00
parent bbb589fb37
commit 4fe03fb233
No known key found for this signature in database
GPG key ID: 6786A150F6A2B28F

View file

@ -1355,6 +1355,12 @@ func (server *Server) Port() int {
return port
}
// ListenWebPort returns true if we should listen to the
// web port, otherwise false
func (server *Server) ListenWebPort() bool {
return !server.cfg.BoolValue("NoWebServer")
}
// WebPort returns the port the web server will listen on when it is
// started.
func (server *Server) WebPort() int {
@ -1396,6 +1402,7 @@ func (server *Server) Start() (err error) {
host := server.HostAddress()
port := server.Port()
webport := server.WebPort()
shouldListenWeb := server.ListenWebPort()
// Setup our UDP listener
server.udpconn, err = net.ListenUDP("udp", &net.UDPAddr{IP: net.ParseIP(host), Port: port})
@ -1434,6 +1441,7 @@ func (server *Server) Start() (err error) {
}
server.tlsl = tls.NewListener(server.tcpl, server.tlscfg)
if shouldListenWeb {
// Create HTTP server and WebSocket "listener"
webaddr := &net.TCPAddr{IP: net.ParseIP(host), Port: webport}
server.webtlscfg = &tls.Config{
@ -1465,6 +1473,10 @@ func (server *Server) Start() (err error) {
}()
server.Printf("Started: listening on %v and %v", server.tcpl.Addr(), server.webwsl.Addr())
} else {
server.Printf("Started: listening on %v", server.tcpl.Addr())
}
server.running = true
// Open a fresh freezer log
@ -1487,10 +1499,17 @@ func (server *Server) Start() (err error) {
// for the servers. Each network goroutine defers a call to
// netwg.Done(). In the Stop() we close all the connections
// and call netwg.Wait() to wait for the goroutines to end.
server.netwg.Add(3)
numWG := 2
if shouldListenWeb {
numWG++
}
server.netwg.Add(numWG)
go server.udpListenLoop()
go server.acceptLoop(server.tlsl)
if shouldListenWeb {
go server.acceptLoop(server.webwsl)
}
// Schedule a server registration update (if needed)
go func() {
@ -1520,6 +1539,7 @@ func (server *Server) Stop() (err error) {
// This does not apply to opened WebSockets, which were forcibly closed when
// all clients were disconnected.
ctx, cancel := context.WithDeadline(context.Background(), time.Now().Add(15*time.Second))
if server.ListenWebPort() {
err = server.webhttp.Shutdown(ctx)
cancel()
if err == context.DeadlineExceeded {
@ -1528,12 +1548,14 @@ func (server *Server) Stop() (err error) {
return err
}
// Close the listeners
err = server.tlsl.Close()
err = server.webwsl.Close()
if err != nil {
return err
}
err = server.webwsl.Close()
}
// Close the listeners
err = server.tlsl.Close()
if err != nil {
return err
}