Refactor environment parsing

Systemd environment variables LISTEN* are unset by default and saved for
future calls

README.md

@@ -2,6 +2,8 @@ Create http server listening on unix sockets or systemd socket activated fds
 
 ## Quick Usage
 
+    go get go.balki.me/anyhttp
+
 Just replace `http.ListenAndServe` with `anyhttp.ListenAndServe`.
 
 ```diff
@@ -49,3 +51,12 @@ Anything else is directly passed to `http.ListenAndServe` as well. Below example
     :http
     :8888
     127.0.0.1:8080
+
+## Documentation
+
+https://pkg.go.dev/go.balki.me/anyhttp
+
+### Related links
+
+  * https://gist.github.com/teknoraver/5ffacb8757330715bcbcc90e6d46ac74#file-unixhttpd-go
+  * https://github.com/coreos/go-systemd/tree/main/activation

anyhttp.go

@@ -10,6 +10,7 @@ import (
 	"os"
 	"strconv"
 	"strings"
+	"sync"
 	"syscall"
 )
 
@@ -39,6 +40,39 @@ func NewUnixSocketConfig(socketPath string) UnixSocketConfig {
 	return usc
 }
 
+type sysdEnvData struct {
+	pid        int
+	fdNames    []string
+	fdNamesStr string
+	numFds     int
+}
+
+var sysdEnvParser = struct {
+	sysdOnce sync.Once
+	data     sysdEnvData
+	err      error
+}{}
+
+func parse() (sysdEnvData, error) {
+	p := &sysdEnvParser
+	p.sysdOnce.Do(func() {
+		p.data.pid, p.err = strconv.Atoi(os.Getenv("LISTEN_PID"))
+		if p.err != nil {
+			p.err = fmt.Errorf("invalid LISTEN_PID, err: %w", p.err)
+			return
+		}
+		p.data.numFds, p.err = strconv.Atoi(os.Getenv("LISTEN_FDS"))
+		if p.err != nil {
+			p.err = fmt.Errorf("invalid LISTEN_FDS, err: %w", p.err)
+			return
+		}
+		p.data.fdNamesStr = os.Getenv("LISTEN_FDNAMES")
+		p.data.fdNames = strings.Split(p.data.fdNamesStr, ":")
+
+	})
+	return p.data, p.err
+}
+
 // SysdConfig has the configuration for the socket activated fd
 type SysdConfig struct {
 	// Integer value starting at 0. Either index or name is required
@@ -54,7 +88,7 @@ type SysdConfig struct {
 // DefaultSysdConfig has the default values for SysdConfig
 var DefaultSysdConfig = SysdConfig{
 	CheckPID: true,
-	UnsetEnv: false,
+	UnsetEnv: true,
 }
 
 // NewSysDConfigWithFDIdx creates SysdConfig with defaults and fdIdx
@@ -112,46 +146,47 @@ func (s *SysdConfig) GetListener() (net.Listener, error) {
 		defer UnsetSystemdListenVars()
 	}
 
-	if s.CheckPID {
+	envData, err := parse()
-		pid, err := strconv.Atoi(os.Getenv("LISTEN_PID"))
-		if err != nil {
-			return nil, err
-		}
-		if pid != os.Getpid() {
-			return nil, fmt.Errorf("fd not for you")
-		}
-	}
-
-	numFds, err := strconv.Atoi(os.Getenv("LISTEN_FDS"))
 	if err != nil {
 		return nil, err
 	}
 
-	fdNames := strings.Split(os.Getenv("LISTEN_FDNAMES"), ":")
+	if s.CheckPID {
+		if envData.pid != os.Getpid() {
+			return nil, fmt.Errorf("unexpected PID, current:%v, LISTEN_PID: %v", os.Getpid(), envData.pid)
+		}
+	}
 
 	if s.FDIndex != nil {
 		idx := *s.FDIndex
-		if idx < 0 || idx >= numFds {
+		if idx < 0 || idx >= envData.numFds {
-			return nil, fmt.Errorf("invalid fd")
+			return nil, fmt.Errorf("invalid fd index, expected between 0 and %v, got: %v", envData.numFds, idx)
 		}
 		fd := StartFD + idx
-		if idx < len(fdNames) {
+		if idx < len(envData.fdNames) {
-			return makeFdListener(fd, fdNames[idx])
+			return makeFdListener(fd, envData.fdNames[idx])
 		}
 		return makeFdListener(fd, fmt.Sprintf("sysdfd_%d", fd))
 	}
 
 	if s.FDName != nil {
-		for idx, name := range fdNames {
+		for idx, name := range envData.fdNames {
 			if name == *s.FDName {
 				fd := StartFD + idx
 				return makeFdListener(fd, name)
 			}
 		}
-		return nil, fmt.Errorf("fdName not found: %q", *s.FDName)
+		return nil, fmt.Errorf("fdName not found: %q, LISTEN_FDNAMES:%q", *s.FDName, envData.fdNamesStr)
 	}
 
-	return nil, fmt.Errorf("neither FDIndex nor FDName set")
+	return nil, errors.New("neither FDIndex nor FDName set")
+}
+
+// UnknownAddress Error is returned when address does not match any known syntax
+type UnknownAddress struct{}
+
+func (u UnknownAddress) Error() string {
+	return "unknown address"
 }
 
 // GetListener gets a unix or systemd socket listener
@@ -164,7 +199,7 @@ func GetListener(addr string) (net.Listener, error) {
 	if strings.HasPrefix(addr, "sysd/fdidx/") {
 		idx, err := strconv.Atoi(strings.TrimPrefix(addr, "sysd/fdidx/"))
 		if err != nil {
-			return nil, err
+			return nil, fmt.Errorf("invalid fdidx, addr:%q err: %w", addr, err)
 		}
 		sysdc := NewSysDConfigWithFDIdx(idx)
 		return sysdc.GetListener()
@@ -175,7 +210,7 @@ func GetListener(addr string) (net.Listener, error) {
 		return sysdc.GetListener()
 	}
 
-	return nil, nil
+	return nil, UnknownAddress{}
 }
 
 // ListenAndServe is the drop-in replacement for `http.ListenAndServe`.
@@ -183,7 +218,7 @@ func GetListener(addr string) (net.Listener, error) {
 func ListenAndServe(addr string, h http.Handler) error {
 
 	listener, err := GetListener(addr)
-	if err != nil {
+	if _, isUnknown := err.(UnknownAddress); err != nil && !isUnknown {
 		return err
 	}
 
@@ -204,7 +239,7 @@ func ListenAndServe(addr string, h http.Handler) error {
 
 // UnsetSystemdListenVars unsets the LISTEN* environment variables so they are not passed to any child processes
 func UnsetSystemdListenVars() {
-	os.Unsetenv("LISTEN_PID")
+	_ = os.Unsetenv("LISTEN_PID")
-	os.Unsetenv("LISTEN_FDS")
+	_ = os.Unsetenv("LISTEN_FDS")
-	os.Unsetenv("LISTEN_FDNAMES")
+	_ = os.Unsetenv("LISTEN_FDNAMES")
 }