fortify: root check before command handling

Signed-off-by: Ophestra Umiker <cat@ophivana.moe>

1 files changed, 6 insertions, 7 deletions

diff --git a/main.go b/main.go
index b8bb0916..1e3e2ee3 100644
--- a/main.go
+++ b/main.go
@@ -35,6 +35,11 @@ func main() {
 		// not fatal: this program runs as the privileged user
 	}
 
+	if os.Geteuid() == 0 {
+		fmsg.Fatal("this program must not run as root")
+		panic("unreachable")
+	}
+
 	flag.CommandLine.Usage = func() {
 		fmt.Println()
 		fmt.Println("Usage:\tfortify [-v] COMMAND [OPTIONS]")
@@ -54,19 +59,13 @@ func main() {
 			_, _ = fmt.Fprintf(w, "\t%s\t%s\n", c[0], c[1])
 		}
 		if err := w.Flush(); err != nil {
-			fmsg.Fatalf("cannot print help: %v", err)
+			fmt.Printf("fortify: cannot write command list: %v\n", err)
 		}
 		fmt.Println()
 	}
 	flag.Parse()
 	fmsg.SetVerbose(flagVerbose)
 
-	// root check
-	if os.Geteuid() == 0 {
-		fmsg.Fatal("this program must not run as root")
-		panic("unreachable")
-	}
-
 	args := flag.Args()
 	if len(args) == 0 {
 		flag.CommandLine.Usage()