From 1b4998d4e221ae3c52c3b6b389202fbfa1770a9b Mon Sep 17 00:00:00 2001
From: Evan Su <48808396+HACKERALERT@users.noreply.github.com>
Date: Thu, 25 Apr 2024 22:13:41 -0400
Subject: [PATCH] Handle flags after arguments

---
 cli/v2/picocrypt/main.go | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/cli/v2/picocrypt/main.go b/cli/v2/picocrypt/main.go
index 161f055..e74f748 100644
--- a/cli/v2/picocrypt/main.go
+++ b/cli/v2/picocrypt/main.go
@@ -107,6 +107,12 @@ func work() int {
 			}
 		}
 	}
+	for _, v := range flag.Args() {
+		if v == "-p" || v == "-r" || v == "-fix" {
+			fmt.Println("Flags are only accepted before arguments!")
+			return 1
+		}
+	}
 
 	var password, cpassword []byte
 	var err error