feat(service): add cross-platform service management support

Implement service installation and management for Windows (sc.exe), Linux (systemd), and macOS (launchd)
Add platform-specific stubs for unsupported platforms
Include service configuration handling and default config generation
Add service installation flags to main application

Author: Tui

cmd/mcp-server/main.go

@@ -3,25 +3,86 @@ package main
 import (
 	"context"
 	"flag"
+	"fmt"
 	"log"
 	"net/http"
 	"os"
 	"time"
 
 	"github.com/ba0f3/MCP-Kali-Server/pkg/executor"
 	"github.com/ba0f3/MCP-Kali-Server/pkg/handlers"
+	"github.com/ba0f3/MCP-Kali-Server/pkg/service"
 	"github.com/modelcontextprotocol/go-sdk/mcp"
 )
 
+// handleServiceInstall installs the server as a system service
+func handleServiceInstall(serviceName, servicePort string) error {
+	config := service.GetDefaultConfig()
+	config.Name = serviceName
+
+	// Set service arguments to use HTTP mode with the specified port
+	config.Args = []string{
+		"-http", servicePort,
+		"-timeout", "900",
+	}
+
+	fmt.Printf("Installing %s service...\n", serviceName)
+	fmt.Printf("Service will listen on port: %s\n", servicePort)
+	fmt.Printf("Executable: %s\n", config.Executable)
+	fmt.Printf("Working directory: %s\n", config.WorkingDir)
+
+	if err := service.InstallService(config); err != nil {
+		return fmt.Errorf("installation failed: %w", err)
+	}
+
+	fmt.Printf("\nService '%s' installed successfully!\n", serviceName)
+	fmt.Println("\nTo start the service:")
+	fmt.Printf("  Windows: sc start %s\n", serviceName)
+	fmt.Printf("  Linux: sudo systemctl start %s\n", serviceName)
+	fmt.Printf("  macOS: sudo launchctl start %s\n", serviceName)
+
+	return nil
+}
+
+// handleServiceUninstall removes the installed service
+func handleServiceUninstall(serviceName string) error {
+	fmt.Printf("Uninstalling %s service...\n", serviceName)
+
+	if err := service.UninstallService(serviceName); err != nil {
+		return fmt.Errorf("uninstallation failed: %w", err)
+	}
+
+	fmt.Printf("Service '%s' uninstalled successfully!\n", serviceName)
+	return nil
+}
 
 func main() {
 	var (
 		debug = flag.Bool("debug", false, "Enable debug logging")
 		httpAddr = flag.String("http", "", "if set, use streamable HTTP at this address, instead of stdin/stdout")
 		timeout = flag.Int("timeout", 900, "Command execution timeout in seconds")
+		installService = flag.Bool("install-service", false, "Install the server as a system service")
+		uninstallService = flag.Bool("uninstall-service", false, "Uninstall the system service")
+		serviceName = flag.String("service-name", "mcp-kali-server", "Name of the service")
+		servicePort = flag.String("service-port", ":8080", "Port for the service to listen on (used with -install-service)")
 	)
 	flag.Parse()
 
+	// Handle service installation/uninstallation
+	if *installService {
+		if err := handleServiceInstall(*serviceName, *servicePort); err != nil {
+			log.Fatalf("Failed to install service: %v", err)
+		}
+		return
+	}
+
+	if *uninstallService {
+		if err := handleServiceUninstall(*serviceName); err != nil {
+			log.Fatalf("Failed to uninstall service: %v", err)
+		}
+		return
+	}
+
 	// Set the global command timeout
 	executor.SetGlobalTimeout(time.Duration(*timeout) * time.Second)
 

pkg/service/service.go

@@ -0,0 +1,60 @@
+package service
+
+import (
+	"fmt"
+	"os"
+	"path/filepath"
+	"runtime"
+)
+
+// ServiceConfig holds the configuration for the service
+type ServiceConfig struct {
+	Name        string
+	DisplayName string
+	Description string
+	Executable  string
+	Args        []string
+	WorkingDir  string
+}
+
+// InstallService installs the application as a system service
+func InstallService(config ServiceConfig) error {
+	switch runtime.GOOS {
+	case "windows":
+		return installWindowsService(config)
+	case "linux":
+		return installLinuxService(config)
+	case "darwin":
+		return installDarwinService(config)
+	default:
+		return fmt.Errorf("unsupported operating system: %s", runtime.GOOS)
+	}
+}
+
+// UninstallService removes the installed service
+func UninstallService(name string) error {
+	switch runtime.GOOS {
+	case "windows":
+		return uninstallWindowsService(name)
+	case "linux":
+		return uninstallLinuxService(name)
+	case "darwin":
+		return uninstallDarwinService(name)
+	default:
+		return fmt.Errorf("unsupported operating system: %s", runtime.GOOS)
+	}
+}
+
+// GetDefaultConfig returns a default service configuration
+func GetDefaultConfig() ServiceConfig {
+	executable, _ := os.Executable()
+	workingDir := filepath.Dir(executable)
+
+	return ServiceConfig{
+		Name:        "mcp-kali-server",
+		DisplayName: "MCP Kali Server",
+		Description: "Model Context Protocol server for Kali Linux security tools",
+		Executable:  executable,
+		WorkingDir:  workingDir,
+	}
+}

pkg/service/service_darwin.go

@@ -0,0 +1,89 @@
+//go:build darwin
+// +build darwin
+
+package service
+
+import (
+	"fmt"
+	"os"
+	"os/exec"
+	"path/filepath"
+	"text/template"
+)
+
+const launchdTemplate = `<?xml version="1.0" encoding="UTF-8"?>
+<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
+<plist version="1.0">
+<dict>
+    <key>Label</key>
+    <string>{{.Name}}</string>
+    <key>ProgramArguments</key>
+    <array>
+        <string>{{.Executable}}</string>
+        {{range .Args}}<string>{{.}}</string>
+        {{end}}
+    </array>
+    <key>WorkingDirectory</key>
+    <string>{{.WorkingDir}}</string>
+    <key>RunAtLoad</key>
+    <true/>
+    <key>KeepAlive</key>
+    <true/>
+    <key>StandardOutPath</key>
+    <string>/var/log/{{.Name}}.log</string>
+    <key>StandardErrorPath</key>
+    <string>/var/log/{{.Name}}.error.log</string>
+</dict>
+</plist>
+`
+
+// installDarwinService installs the service on macOS using launchd
+func installDarwinService(config ServiceConfig) error {
+	// Create plist file
+	plistFile := filepath.Join("/Library/LaunchDaemons", config.Name+".plist")
+
+	file, err := os.Create(plistFile)
+	if err != nil {
+		return fmt.Errorf("failed to create plist file: %v", err)
+	}
+	defer file.Close()
+
+	// Write plist file
+	tmpl, err := template.New("plist").Parse(launchdTemplate)
+	if err != nil {
+		return fmt.Errorf("failed to parse template: %v", err)
+	}
+
+	if err := tmpl.Execute(file, config); err != nil {
+		return fmt.Errorf("failed to write plist file: %v", err)
+	}
+
+	// Set proper permissions
+	if err := os.Chmod(plistFile, 0644); err != nil {
+		return fmt.Errorf("failed to set permissions: %v", err)
+	}
+
+	// Load the service
+	cmd := exec.Command("launchctl", "load", plistFile)
+	if err := cmd.Run(); err != nil {
+		return fmt.Errorf("failed to load service: %v", err)
+	}
+
+	return nil
+}
+
+// uninstallDarwinService removes the service on macOS
+func uninstallDarwinService(name string) error {
+	plistFile := filepath.Join("/Library/LaunchDaemons", name+".plist")
+
+	// Unload the service
+	cmd := exec.Command("launchctl", "unload", plistFile)
+	cmd.Run() // Ignore error if service is not loaded
+
+	// Remove plist file
+	if err := os.Remove(plistFile); err != nil && !os.IsNotExist(err) {
+		return fmt.Errorf("failed to remove plist file: %v", err)
+	}
+
+	return nil
+}

pkg/service/service_darwin_stub.go

@@ -0,0 +1,16 @@
+//go:build !darwin
+// +build !darwin
+
+package service
+
+import "fmt"
+
+// installDarwinService is a stub for non-Darwin platforms
+func installDarwinService(config ServiceConfig) error {
+	return fmt.Errorf("macOS service installation is not supported on this platform")
+}
+
+// uninstallDarwinService is a stub for non-Darwin platforms
+func uninstallDarwinService(name string) error {
+	return fmt.Errorf("macOS service uninstallation is not supported on this platform")
+}

pkg/service/service_linux.go

@@ -0,0 +1,114 @@
+//go:build linux
+// +build linux
+
+package service
+
+import (
+	"fmt"
+	"os"
+	"os/exec"
+	"path/filepath"
+	"text/template"
+)
+
+const systemdTemplate = `[Unit]
+Description={{.Description}}
+After=network.target
+
+[Service]
+Type=simple
+User={{.User}}
+WorkingDirectory={{.WorkingDir}}
+ExecStart={{.Executable}}{{range .Args}} {{.}}{{end}}
+Restart=on-failure
+RestartSec=10
+
+[Install]
+WantedBy=multi-user.target
+`
+
+// installLinuxService installs the service on Linux using systemd
+func installLinuxService(config ServiceConfig) error {
+	// Check if systemd is available
+	if _, err := os.Stat("/usr/bin/systemctl"); os.IsNotExist(err) {
+		return fmt.Errorf("systemd is not available on this system")
+	}
+
+	// Create service file
+	serviceFile := filepath.Join("/etc/systemd/system", config.Name+".service")
+
+	file, err := os.Create(serviceFile)
+	if err != nil {
+		return fmt.Errorf("failed to create service file: %v", err)
+	}
+	defer file.Close()
+
+	// Prepare template data
+	data := struct {
+		Description string
+		User        string
+		WorkingDir  string
+		Executable  string
+		Args        []string
+	}{
+		Description: config.Description,
+		User:        os.Getenv("USER"),
+		WorkingDir:  config.WorkingDir,
+		Executable:  config.Executable,
+		Args:        config.Args,
+	}
+
+	// If running as root, use nobody user
+	if data.User == "root" || data.User == "" {
+		data.User = "nobody"
+	}
+
+	// Write service file
+	tmpl, err := template.New("service").Parse(systemdTemplate)
+	if err != nil {
+		return fmt.Errorf("failed to parse template: %v", err)
+	}
+
+	if err := tmpl.Execute(file, data); err != nil {
+		return fmt.Errorf("failed to write service file: %v", err)
+	}
+
+	// Reload systemd
+	cmd := exec.Command("systemctl", "daemon-reload")
+	if err := cmd.Run(); err != nil {
+		return fmt.Errorf("failed to reload systemd: %v", err)
+	}
+
+	// Enable the service
+	cmd = exec.Command("systemctl", "enable", config.Name)
+	if err := cmd.Run(); err != nil {
+		return fmt.Errorf("failed to enable service: %v", err)
+	}
+
+	return nil
+}
+
+// uninstallLinuxService removes the service on Linux
+func uninstallLinuxService(name string) error {
+	// Stop the service
+	cmd := exec.Command("systemctl", "stop", name)
+	cmd.Run() // Ignore error if service is not running
+
+	// Disable the service
+	cmd = exec.Command("systemctl", "disable", name)
+	cmd.Run() // Ignore error if service is not enabled
+
+	// Remove service file
+	serviceFile := filepath.Join("/etc/systemd/system", name+".service")
+	if err := os.Remove(serviceFile); err != nil && !os.IsNotExist(err) {
+		return fmt.Errorf("failed to remove service file: %v", err)
+	}
+
+	// Reload systemd
+	cmd = exec.Command("systemctl", "daemon-reload")
+	if err := cmd.Run(); err != nil {
+		return fmt.Errorf("failed to reload systemd: %v", err)
+	}
+
+	return nil
+}

pkg/service/service_linux_stub.go

@@ -0,0 +1,16 @@
+//go:build !linux
+// +build !linux
+
+package service
+
+import "fmt"
+
+// installLinuxService is a stub for non-Linux platforms
+func installLinuxService(config ServiceConfig) error {
+	return fmt.Errorf("Linux service installation is not supported on this platform")
+}
+
+// uninstallLinuxService is a stub for non-Linux platforms
+func uninstallLinuxService(name string) error {
+	return fmt.Errorf("Linux service uninstallation is not supported on this platform")
+}