added IsDockerContainer config var and used it in sanitycheck to allow running as root in docker container. Added getter and setter acordingly.

Author: JacksonTheMaster

server.go

@@ -22,9 +22,7 @@ package main
 
 import (
 	"embed"
-	"os"
 	"sync"
-	"time"
 
 	"github.com/JacksonTheMaster/StationeersServerUI/v5/src/cli"
 	"github.com/JacksonTheMaster/StationeersServerUI/v5/src/core/loader"
@@ -39,11 +37,8 @@ var v1uiFS embed.FS
 func main() {
 	var wg sync.WaitGroup
 	logger.ConfigureConsole()
-	if err := loader.SanityCheck(); err != nil {
-		logger.Main.Error("Sanity check failed, exiting in 10 secconds: " + err.Error())
-		time.Sleep(10 * time.Second)
-		os.Exit(1)
-	}
+	loader.SanityCheck(&wg)
+	wg.Wait()
 	logger.Main.Debug("Initializing resources...")
 	loader.InitVirtFS(v1uiFS)
 	logger.Install.Info("Starting setup...")

src/config/setters.go

@@ -54,6 +54,14 @@ func SetSkipSteamCMD(value bool) error {
 	return nil
 }
 
+func SetIsDockerContainer(value bool) error {
+	ConfigMu.Lock()
+	defer ConfigMu.Unlock()
+
+	IsDockerContainer = value
+	return nil
+}
+
 // ALL SETTERS BELOW THIS LINE ARE UNUSED AT THE MOMENT
 // ALL SETTERS BELOW THIS LINE ARE UNUSED AT THE MOMENT
 // ALL SETTERS BELOW THIS LINE ARE UNUSED AT THE MOMENT

src/config/vars.go

@@ -61,9 +61,15 @@ var (
 	LanguageSetting          string
 	AutoStartServerOnStartup bool
 	SSUIIdentifier           string
-	CurrentBranchBuildID     string // ONLY RUNTIME
-	ExtractedGameVersion     string // ONLY RUNTIME
-	SkipSteamCMD             bool   // ONLY RUNTIME
+)
+
+// Runtime only variables
+
+var (
+	CurrentBranchBuildID string // ONLY RUNTIME
+	ExtractedGameVersion string // ONLY RUNTIME
+	SkipSteamCMD         bool   // ONLY RUNTIME
+	IsDockerContainer    bool   // ONLY RUNTIME
 )
 
 // Discord integration

src/core/loader/helpers.go

@@ -1,10 +1,9 @@
 package loader
 
 import (
+	"bufio"
 	"fmt"
 	"os"
-	"path/filepath"
-	"runtime"
 	"strings"
 
 	"github.com/JacksonTheMaster/StationeersServerUI/v5/src/config"
@@ -165,59 +164,35 @@ func PrintConfigDetails(logLevel ...string) {
 	logger.Config.Debug("=======================================")
 }
 
-func SanityCheck() error {
-
-	if runtime.GOOS == "windows" {
-		return nil
-	}
-
-	// Check if running as root (UID 0)
-	if os.Geteuid() == 0 {
-		return fmt.Errorf("root: SSUI should not be run as root")
+func IsInsideContainer() bool {
+	// Check .dockerenv file (Docker-specific)
+	if _, err := os.Stat("/.dockerenv"); err == nil {
+		config.SetIsDockerContainer(true)
+		return true
 	}
+	// Check cgroup (works for Docker and other container runtimes)
+	return isContainerFromCGroup()
+}
 
-	// Get the current executable path from /proc/self/exe
-	exePath, err := os.Readlink("/proc/self/exe")
+func isContainerFromCGroup() bool {
+	file, err := os.Open("/proc/1/cgroup")
 	if err != nil {
-		return err
-	}
-	// Get the directory path of the executable
-	dirPath := filepath.Dir(exePath)
-	// Change the working directory to the executable's directory
-	cwd, err := os.Getwd()
-	if err != nil {
-		return err
-	}
-
-	if cwd != dirPath && !strings.Contains(dirPath, "/tmp") {
-		err = os.Chdir(dirPath)
-		if err != nil {
-			return err
+		return false
+	}
+	defer file.Close()
+
+	scanner := bufio.NewScanner(file)
+	for scanner.Scan() {
+		line := scanner.Text()
+		// Check for various container runtime indicators
+		if strings.Contains(line, "docker") ||
+			strings.Contains(line, "containerd") ||
+			strings.Contains(line, "kubepods") ||
+			strings.Contains(line, "crio") ||
+			strings.Contains(line, "libpod") {
+			config.SetIsDockerContainer(true)
+			return true
 		}
 	}
-
-	// Check if current working directory is writable
-	workDir, err := os.Getwd()
-	if err != nil {
-		return fmt.Errorf("failed to get working directory: %w", err)
-	}
-
-	// Try to create a temporary file to test write permissions
-	testFile := filepath.Join(workDir, ".write_test")
-	if err := os.WriteFile(testFile, []byte("test"), 0600); err != nil {
-		return fmt.Errorf("cannot write to working directory %s: %w", workDir, err)
-	}
-	// Clean up test file
-	if err := os.Remove(testFile); err != nil {
-		return fmt.Errorf("failed to clean up sanity check writetest file: %w", err)
-	}
-
-	// Check if steamcmd package is installed  (requires further testing, disabled for now)
-	//cmd := exec.Command("dpkg-query", "-W", "-f='${Status}'", "steamcmd")
-	//output, err := cmd.CombinedOutput()
-	//if err == nil && strings.Contains(string(output), "install ok installed") {
-	//	return fmt.Errorf("steamcmd package is installed")
-	//}
-
-	return nil
+	return false
 }

src/core/loader/loader.go

@@ -3,7 +3,9 @@ package loader
 
 import (
 	"embed"
+	"os"
 	"sync"
+	"time"
 
 	"github.com/JacksonTheMaster/StationeersServerUI/v5/src/config"
 	"github.com/JacksonTheMaster/StationeersServerUI/v5/src/discordbot"
@@ -98,3 +100,13 @@ func ReloadAppInfoPoller() {
 func InitVirtFS(v1uiFS embed.FS) {
 	config.SetV1UIFS(v1uiFS)
 }
+
+func SanityCheck(wg *sync.WaitGroup) {
+	defer wg.Done()
+	err := runSanityCheck()
+	if err != nil {
+		logger.Main.Error("Sanity check failed, exiting in 10 secconds: " + err.Error())
+		time.Sleep(10 * time.Second)
+		os.Exit(1)
+	}
+}

src/core/loader/sanitycheck.go

@@ -0,0 +1,69 @@
+package loader
+
+import (
+	"fmt"
+	"os"
+	"path/filepath"
+	"runtime"
+	"strings"
+)
+
+func runSanityCheck() error {
+
+	if runtime.GOOS == "windows" {
+		return nil
+	}
+
+	// Check if running as root (UID 0)
+	if os.Geteuid() == 0 {
+		// Check if running inside a container
+		if !IsInsideContainer() {
+			return fmt.Errorf("root: SSUI should not be run as root")
+		}
+	}
+
+	// Get the current executable path from /proc/self/exe
+	exePath, err := os.Readlink("/proc/self/exe")
+	if err != nil {
+		return err
+	}
+	// Get the directory path of the executable
+	dirPath := filepath.Dir(exePath)
+	// Change the working directory to the executable's directory
+	cwd, err := os.Getwd()
+	if err != nil {
+		return err
+	}
+
+	if cwd != dirPath && !strings.Contains(dirPath, "/tmp") {
+		err = os.Chdir(dirPath)
+		if err != nil {
+			return err
+		}
+	}
+
+	// Check if current working directory is writable
+	workDir, err := os.Getwd()
+	if err != nil {
+		return fmt.Errorf("failed to get working directory: %w", err)
+	}
+
+	// Try to create a temporary file to test write permissions
+	testFile := filepath.Join(workDir, ".write_test")
+	if err := os.WriteFile(testFile, []byte("test"), 0600); err != nil {
+		return fmt.Errorf("cannot write to working directory, please make sure your user has write permissions in %s: %w", workDir, err)
+	}
+	// Clean up test file
+	if err := os.Remove(testFile); err != nil {
+		return fmt.Errorf("failed to clean up sanity check writetest file: %w", err)
+	}
+
+	// Check if steamcmd package is installed  (requires further testing, disabled for now)
+	//cmd := exec.Command("dpkg-query", "-W", "-f='${Status}'", "steamcmd")
+	//output, err := cmd.CombinedOutput()
+	//if err == nil && strings.Contains(string(output), "install ok installed") {
+	//	return fmt.Errorf("steamcmd package is installed")
+	//}
+
+	return nil
+}