report listening port when chosen by kernel (#770)

Based off of the PR by @thesayyn
https://github.com/project-zot/zot/pull/720

Signed-off-by: Ramkumar Chinchani <rchincha@cisco.com>

pkg/api/controller.go

@@ -9,6 +9,7 @@ import (
 	"net/http"
 	"os"
 	"runtime"
+	"strconv"
 	"strings"
 	goSync "sync"
 	"syscall"
@@ -41,6 +42,8 @@ type Controller struct {
 	Server          *http.Server
 	Metrics         monitoring.MetricServer
 	wgShutDown      *goSync.WaitGroup // use it to gracefully shutdown goroutines
+	// runtime params
+	chosenPort int // kernel-chosen port
 }
 
 func NewController(config *config.Config) *Controller {
@@ -103,6 +106,10 @@ func DumpRuntimeParams(log log.Logger) {
 	evt.Msg("runtime params")
 }
 
+func (c *Controller) GetPort() int {
+	return c.chosenPort
+}
+
 func (c *Controller) Run(reloadCtx context.Context) error {
 	// print the current configuration, but strip secrets
 	c.Log.Info().Interface("params", c.Config.Sanitize()).Msg("configuration settings")
@@ -171,6 +178,25 @@ func (c *Controller) Run(reloadCtx context.Context) error {
 		return err
 	}
 
+	if c.Config.HTTP.Port == "0" || c.Config.HTTP.Port == "" {
+		chosenAddr, ok := listener.Addr().(*net.TCPAddr)
+		if !ok {
+			c.Log.Error().Str("port", c.Config.HTTP.Port).Msg("invalid addr type")
+
+			return errors.ErrBadType
+		}
+
+		c.chosenPort = chosenAddr.Port
+
+		c.Log.Info().Int("port", chosenAddr.Port).IPAddr("address", chosenAddr.IP).Msg(
+			"port is unspecified, listening on kernel chosen port",
+		)
+	} else {
+		chosenPort, _ := strconv.ParseInt(c.Config.HTTP.Port, 10, 64)
+
+		c.chosenPort = int(chosenPort)
+	}
+
 	if c.Config.HTTP.TLS != nil && c.Config.HTTP.TLS.Key != "" && c.Config.HTTP.TLS.Cert != "" {
 		server.TLSConfig = &tls.Config{
 			CipherSuites: []uint16{

pkg/api/controller_test.go

@@ -143,6 +143,55 @@ func TestRunAlreadyRunningServer(t *testing.T) {
 	})
 }
 
+func TestAutoPortSelection(t *testing.T) {
+	Convey("Run server with specifying a port", t, func() {
+		conf := config.New()
+		conf.HTTP.Port = "0"
+
+		logFile, err := os.CreateTemp("", "zot-log*.txt")
+		So(err, ShouldBeNil)
+		conf.Log.Output = logFile.Name()
+		defer os.Remove(logFile.Name()) // clean up
+
+		ctlr := api.NewController(conf)
+		ctlr.Config.Storage.RootDirectory = t.TempDir()
+
+		go startServer(ctlr)
+		time.Sleep(1000 * time.Millisecond)
+		defer stopServer(ctlr)
+
+		file, err := os.Open(logFile.Name())
+		So(err, ShouldBeNil)
+		defer file.Close()
+
+		scanner := bufio.NewScanner(file)
+
+		var contents bytes.Buffer
+		start := time.Now()
+
+		for scanner.Scan() {
+			if time.Since(start) < time.Second*30 {
+				t.Logf("Exhausted: Controller did not print the expected log within 30 seconds")
+			}
+			text := scanner.Text()
+			contents.WriteString(text)
+			if strings.Contains(text, "Port unspecified") {
+				break
+			}
+			t.Logf(scanner.Text())
+		}
+		So(scanner.Err(), ShouldBeNil)
+		So(contents.String(), ShouldContainSubstring,
+			"port is unspecified, listening on kernel chosen port",
+		)
+		So(contents.String(), ShouldContainSubstring, "\"address\":\"127.0.0.1\"")
+		So(contents.String(), ShouldContainSubstring, "\"port\":")
+
+		So(ctlr.GetPort(), ShouldBeGreaterThan, 0)
+		So(ctlr.GetPort(), ShouldBeLessThan, 65536)
+	})
+}
+
 func TestObjectStorageController(t *testing.T) {
 	skipIt(t)
 	Convey("Negative make a new object storage controller", t, func() {