non-k8s: improve karmor profile for host logs

log/log.go

@@ -12,6 +12,7 @@ import (
 	"os/signal"
 	"regexp"
 	"strconv"
+	"strings"
 	"syscall"
 	"time"
 
@@ -138,7 +139,11 @@ func StartObserver(c *k8s.Client, o Options) error {
 	} else {
 		pf, err := utils.InitiatePortForward(c, port, port, matchLabels, targetSvc)
 		if err != nil {
-			return err
+			if strings.Contains(err.Error(), "connection refused") && utils.IsSystemdMode() {
+				// do nothing
+			} else {
+				return err
+			}
 		}
 		gRPC = "localhost:" + strconv.FormatInt(pf.LocalPort, 10)
 	}

probe/probe.go

@@ -9,10 +9,10 @@ import (
 	"context"
 	"encoding/json"
 	"fmt"
+	"github.com/kubearmor/kubearmor-client/utils"
 	"io"
 	"log"
 	"os"
-	"os/exec"
 	"runtime"
 	"sort"
 	"strconv"
@@ -80,7 +80,7 @@ func PrintProbeResult(c *k8s.Client, o Options) error {
 			return errors.New("unsupported environment or cluster not configured correctly")
 		}
 	}
-	if isSystemdMode() {
+	if utils.IsSystemdMode() {
 		kd, err := probeSystemdMode()
 		if err != nil {
 			return err
@@ -566,13 +566,6 @@ func getPostureData(probeData []KubeArmorProbeData) map[string]string {
 	return postureData
 }
 
-// sudo systemctl status kubearmor
-func isSystemdMode() bool {
-	cmd := exec.Command("systemctl", "status", "kubearmor")
-	_, err := cmd.CombinedOutput()
-	return err == nil
-}
-
 func probeSystemdMode() (KubeArmorProbeData, error) {
 	jsonFile, err := os.Open("/tmp/karmorProbeData.cfg")
 	if err != nil {

profile/Client/profileClient.go

@@ -8,6 +8,7 @@ import (
 	"bytes"
 	"encoding/json"
 	"fmt"
+	"github.com/kubearmor/kubearmor-client/utils"
 	"os"
 	"strings"
 	"time"
@@ -103,6 +104,8 @@ func waitForActivity() tea.Cmd {
 var o1 Options
 
 func generateColumns(Operation string) []table.Column {
+	var columns []table.Column
+
 	CountCol := table.NewFlexColumn(ColumnCount, "Count", 1).WithStyle(ColumnStyle).WithFiltered(true)
 
 	Namespace := table.NewFlexColumn(ColumnNamespace, "Namespace", 2).WithStyle(ColumnStyle).WithFiltered(true)
@@ -120,15 +123,22 @@ func generateColumns(Operation string) []table.Column {
 
 	Timestamp := table.NewFlexColumn(ColumnTimestamp, "TimeStamp", 3).WithStyle(ColumnStyle)
 
-	return []table.Column{
-		Namespace,
-		ContainerName,
+	if !utils.IsSystemdMode() {
+		columns = append(columns, []table.Column{
+			Namespace,
+			ContainerName,
+		}...)
+	}
+
+	columns = append(columns, []table.Column{
 		ProcName,
 		Resource,
 		Result,
 		CountCol,
 		Timestamp,
-	}
+	}...)
+
+	return columns
 }
 
 // Init calls initial functions if needed

utils/utils.go

@@ -0,0 +1,10 @@
+package utils
+
+import "os/exec"
+
+// IsSystemdMode checks if kubearmor is running in systemd mode
+func IsSystemdMode() bool {
+	cmd := exec.Command("systemctl", "status", "kubearmor")
+	_, err := cmd.CombinedOutput()
+	return err == nil
+}