Added ParseCommand (#404)

* added test and parsecommand

Author: lobsterjerusalem

transform/parsing.go

@@ -0,0 +1,28 @@
+package transform
+
+import (
+	"strings"
+
+	"github.com/vulncheck-oss/go-exploit/output"
+)
+
+// Provided a command param that includes args such as "cmd.exe /c whoami" as one might receive from a a user provided flag like
+// -command , this function would return "cmd.exe", "/c whoami", true.
+//
+//	program, args, ok := transform.ParseCommand(command)
+func ParseCommand(command string) (string, string, bool) {
+	sCommand := strings.SplitN(command, " ", 2)
+	if sCommand == nil {
+		output.PrintFrameworkError("Invalid command provided, should be space-separated with program and then args in front. examples: cmd.exe /c someargs or bash -c someargs")
+
+		return "", "", false
+	}
+
+	if len(sCommand) != 2 {
+		output.PrintFrameworkError("Invalid command provided, should be space-separated with program and then args in front. examples: cmd.exe /c someargs or bash -c someargs")
+
+		return "", "", false
+	}
+
+	return sCommand[0], sCommand[1], true
+}

transform/parsing_test.go

@@ -0,0 +1,16 @@
+package transform
+
+import "testing"
+
+func TestParseCommand(t *testing.T) {
+	program, args, ok := ParseCommand("cmd.exe /c calc")
+	if !ok {
+		t.Fatal("Failed to parse command")
+	}
+
+	if program != "cmd.exe" || args != "/c calc" {
+		t.Fatalf("Program or args returned unexpected values: %s %s", program, args)
+	}
+
+	t.Log(program + " " + args)
+}