From 7cfb978cda67dbfab11b871a2244db5306b8168e Mon Sep 17 00:00:00 2001
From: pymq <maksimmerzh@gmail.com>
Date: Wed, 2 Aug 2023 21:09:17 +0300
Subject: [PATCH] cmd/awl-tray: add macos specific root hacks (#107)

* cmd/awl-tray: add macos specific root hacks
---
 cmd/awl-tray/os_hacks_macos.go | 68 ++++++++++++++++++++++++++++++++++
 cmd/awl-tray/os_hacks_other.go |  4 +-
 cmd/awl/main.go                | 12 ++++++
 3 files changed, 82 insertions(+), 2 deletions(-)
 create mode 100644 cmd/awl-tray/os_hacks_macos.go

diff --git a/cmd/awl-tray/os_hacks_macos.go b/cmd/awl-tray/os_hacks_macos.go
new file mode 100644
index 0000000..664101d
--- /dev/null
+++ b/cmd/awl-tray/os_hacks_macos.go
@@ -0,0 +1,68 @@
+//go:build darwin
+
+package main
+
+import (
+	"fmt"
+	"os"
+	"os/exec"
+
+	"github.com/anywherelan/awl/config"
+	"github.com/skratchdot/open-golang/open"
+)
+
+func initOSSpecificHacks() {
+	uid := os.Geteuid()
+	if uid != 0 {
+		fmt.Printf("process is run under non-root uid: %d, ask for root permissions with osascript\n", uid)
+		runItselfWithRoot()
+		return
+	}
+
+	// this is required to allow listening on config.AdminHttpServerIP address
+	//nolint:gosec
+	err := exec.Command("ifconfig", "lo0", "alias", config.AdminHttpServerIP, "up").Run()
+	if err != nil {
+		fmt.Printf("error: `ifconfig lo0 alias %s up`: %v\n", config.AdminHttpServerIP, err)
+	}
+}
+
+func openURL(input string) error {
+	return open.Run(input)
+}
+
+func getRealUserID() (uint32, bool) {
+	return 0, false
+}
+
+func runItselfWithRoot() {
+	// TODO: show pop-up describing that awl needs root for vpn? show it only on first launch?
+
+	executable, err := os.Executable()
+	if err != nil {
+		fmt.Printf("error finding executable path: %v\n", err)
+		executable = os.Args[0]
+	}
+
+	osaScript := fmt.Sprintf("do shell script \"%s\" with administrator privileges", executable)
+
+	//nolint:gosec
+	cmd := exec.Command("osascript", "-e", osaScript)
+	cmd.Stdout = os.Stdout
+	cmd.Stderr = os.Stderr
+
+	err = cmd.Start()
+	if err != nil {
+		fmt.Printf("error executing osascript: %v\n", err)
+		os.Exit(1)
+	}
+
+	err = cmd.Wait()
+	if err != nil {
+		fmt.Printf("error from waiting osascript to finish: %v\n", err)
+		exitCode := cmd.ProcessState.ExitCode()
+		os.Exit(exitCode)
+	}
+
+	os.Exit(0)
+}
diff --git a/cmd/awl-tray/os_hacks_other.go b/cmd/awl-tray/os_hacks_other.go
index 2205165..e2964b2 100644
--- a/cmd/awl-tray/os_hacks_other.go
+++ b/cmd/awl-tray/os_hacks_other.go
@@ -1,5 +1,5 @@
-//go:build !linux
-// +build !linux
+//go:build !linux && !darwin
+// +build !linux,!darwin
 
 package main
 
diff --git a/cmd/awl/main.go b/cmd/awl/main.go
index 1149c79..f38d13d 100644
--- a/cmd/awl/main.go
+++ b/cmd/awl/main.go
@@ -2,7 +2,9 @@ package main
 
 import (
 	"context"
+	"fmt"
 	"os"
+	"os/exec"
 	"os/signal"
 	"syscall"
 	"time"
@@ -17,6 +19,16 @@ import (
 func main() {
 	cli.New(update.AppTypeAwl).Run()
 
+	uid := os.Geteuid()
+	if uid == 0 {
+		// this is required to allow listening on config.AdminHttpServerIP address
+		//nolint:gosec
+		err := exec.Command("ifconfig", "lo0", "alias", config.AdminHttpServerIP, "up").Run()
+		if err != nil {
+			fmt.Printf("error: `ifconfig lo0 alias %s up`: %v\n", config.AdminHttpServerIP, err)
+		}
+	}
+
 	app := awl.New()
 	logger := app.SetupLoggerAndConfig()
 	ctx, ctxCancel := context.WithCancel(context.Background())