Add whitespace checks to audit analyze

Whitespace in a leafnode server name, JetStream domain, etc. could cause issues.

As discussed in nats-io/nats-server#5676.

Signed-off-by: reubenninan <reuben@nats.io>

Author: ReubenMathew

internal/audit/checks.go

@@ -156,6 +156,21 @@ func GetDefaultChecks() []Check {
 			Description: "Verify that all nodes part of the meta group agree on the meta cluster leader",
 			fun:         checkMetaClusterLeader,
 		},
+		{
+			Name:        "Whitespace in leafnode server names",
+			Description: "Verify that no leafnode contains whitespace in its name",
+			fun:         checkLeafnodeServerNamesForWhitespace,
+		},
+		{
+			Name:        "Whitespace in JetStream domains",
+			Description: "Verify that no JetStream server is configured with whitespace in its domain",
+			fun:         checkJetStreamDomainsForWhitespace,
+		},
+		{
+			Name:        "Whitespace in cluster name",
+			Description: "Verify that no cluster name contains whitespace",
+			fun:         checkClusterNamesForWhitespace,
+		},
 	}
 }
 

internal/audit/cluster_checks.go

@@ -16,11 +16,15 @@ package audit
 import (
 	"errors"
 	"fmt"
+	"reflect"
+	"sort"
+	"strings"
+	"unicode"
+
 	"github.com/dustin/go-humanize"
 	"github.com/nats-io/nats-server/v2/server"
 	"github.com/nats-io/natscli/internal/archive"
-	"reflect"
-	"sort"
+	"golang.org/x/exp/maps"
 )
 
 // checkClusterMemoryUsageOutliers creates a parametrized check to verify the memory usage of any given node in a
@@ -214,3 +218,57 @@ func makeCheckClusterHighHAAssets(haAssetsThreshold int) checkFunc {
 		return Pass, nil
 	}
 }
+
+func checkClusterNamesForWhitespace(reader *archive.Reader, examples *ExamplesCollection) (Outcome, error) {
+
+	for _, clusterName := range reader.GetClusterNames() {
+		if strings.ContainsFunc(clusterName, unicode.IsSpace) {
+			examples.add("Cluster: %s", clusterName)
+		}
+	}
+
+	if examples.Count() > 0 {
+		logCritical("Found %d clusters with names containing whitespace", examples.Count())
+		return Fail, nil
+	}
+
+	return Pass, nil
+}
+
+func checkLeafnodeServerNamesForWhitespace(r *archive.Reader, examples *ExamplesCollection) (Outcome, error) {
+
+	for _, clusterName := range r.GetClusterNames() {
+		clusterTag := archive.TagCluster(clusterName)
+
+		leafnodesWithWhitespace := map[string]struct{}{}
+
+		for _, serverName := range r.GetClusterServerNames(clusterName) {
+			serverTag := archive.TagServer(serverName)
+
+			var serverLeafz server.Leafz
+			err := r.Load(&serverLeafz, clusterTag, serverTag, archive.TagServerLeafs())
+			if err != nil {
+				logWarning("Artifact 'LEAFZ' is missing for server %s", serverName)
+				continue
+			}
+
+			for _, leaf := range serverLeafz.Leafs {
+				// check if leafnode name contains whitespace
+				if strings.ContainsFunc(leaf.Name, unicode.IsSpace) {
+					leafnodesWithWhitespace[leaf.Name] = struct{}{}
+				}
+			}
+		}
+
+		if len(leafnodesWithWhitespace) > 0 {
+			examples.add("Cluster %s: %v", clusterName, maps.Keys(leafnodesWithWhitespace))
+		}
+	}
+
+	if examples.Count() > 0 {
+		logCritical("Found %d clusters with leafnode names containing whitespace", examples.Count())
+		return Fail, nil
+	}
+
+	return Pass, nil
+}

internal/audit/server_checks.go

@@ -16,6 +16,9 @@ package audit
 import (
 	"errors"
 	"fmt"
+	"strings"
+	"unicode"
+
 	"github.com/dustin/go-humanize"
 	"github.com/nats-io/nats-server/v2/server"
 	"github.com/nats-io/natscli/internal/archive"
@@ -230,3 +233,33 @@ func makeCheckServerResourceLimits(memoryUsageThreshold, storeUsageThreshold flo
 		return Pass, nil
 	}
 }
+
+func checkJetStreamDomainsForWhitespace(r *archive.Reader, examples *ExamplesCollection) (Outcome, error) {
+
+	for _, clusterName := range r.GetClusterNames() {
+		clusterTag := archive.TagCluster(clusterName)
+
+		for _, serverName := range r.GetClusterServerNames(clusterName) {
+			serverTag := archive.TagServer(serverName)
+
+			var serverJsz server.JSInfo
+			err := r.Load(&serverJsz, clusterTag, serverTag, archive.TagServerJetStream())
+			if err != nil {
+				logWarning("Artifact 'JSZ' is missing for server %s", serverName)
+				continue
+			}
+
+			// check if jetstream domain contains whitespace
+			if strings.ContainsFunc(serverJsz.Config.Domain, unicode.IsSpace) {
+				examples.add("Cluster %s Server %s Domain %s", clusterName, serverName, serverJsz.Config.Domain)
+			}
+		}
+	}
+
+	if examples.Count() > 0 {
+		logCritical("Found %d servers with JetStream domains containing whitespace", examples.Count())
+		return Fail, nil
+	}
+
+	return Pass, nil
+}