change way to check for root

seriousm4x · Jan 29, 2023 · a182f7b · a182f7b
1 parent 752e61c
commit a182f7b
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/backend/pb/handlers.go b/backend/pb/handlers.go
@@ -68,8 +68,11 @@ func HandlerScan(c echo.Context) error {
 		}
 	} else {
 		// check for root on anything else
-		sudoUID := os.Getenv("SUDO_UID")
-		if sudoUID == "" {
+		uid, err := exec.Command("id", "-u").Output()
+		if err != nil {
+			return err
+		}
+		if string(uid) != "0" {
 			err := errors.New("network scan requires upsnap to be run as root")
 			return apis.NewBadRequestError(err.Error(), nil)
 		}