wizard go brr

fixes: #33

Author: dmur1

README.md

@@ -259,6 +259,14 @@ Prints a list of cheatsheets for quick reference.
 
 https://github.com/rerrorctf/rctf/blob/main/commands/cheatsheet.go
 
+### wizard
+
+```
+usage: rctf wizard
+```
+
+https://github.com/rerrorctf/rctf/blob/main/commands/wizard.go
+
 ## The .rctf Directory Structure
 
 Certain commands, such as `init`, `add` and `status` will use a hidden directory structure.

commands/add.go

@@ -10,7 +10,6 @@ import (
 	"io"
 	"log"
 	"os"
-	"os/exec"
 	"path/filepath"
 	"rctf/config"
 	"rctf/data"
@@ -61,18 +60,6 @@ func parseFiles(files *data.Files) {
 	}
 }
 
-func runFileCommandOnFile(path string) string {
-	fileOutput := exec.Command("file", path)
-
-	fileOutputResult, err := fileOutput.Output()
-	if err != nil {
-		fmt.Printf("warning: unable to get output from file on %s\n", path)
-		return ""
-	}
-
-	return string(fileOutputResult[len(path)+2 : len(fileOutputResult)-1])
-}
-
 func writeFiles(files *data.Files) {
 	jsonData, err := json.MarshalIndent(files, "", "  ")
 	if err != nil {
@@ -95,7 +82,7 @@ func addFile(srcPath string) {
 
 	_, fileName := filepath.Split(srcPath)
 
-	fileOutput := runFileCommandOnFile(srcPath)
+	fileOutput := util.RunFileCommandOnFile(srcPath)
 
 	content, err := os.ReadFile(srcPath)
 	if err != nil {

commands/wizard.go

@@ -0,0 +1,122 @@
+package commands
+
+import (
+	"fmt"
+	"log"
+	"os"
+	"rctf/config"
+	"rctf/theme"
+	"rctf/util"
+	"strings"
+)
+
+func findInterestingFiles() []string {
+	files, err := os.ReadDir(".")
+	if err != nil {
+		log.Fatalf("💥 " + theme.ColorRed + "error" + theme.ColorReset + ": unable read cwd!\n")
+	}
+
+	interestingFiles := []string{}
+
+	for _, file := range files {
+		if file.IsDir() {
+			continue
+		}
+
+		fileName := file.Name()
+
+		if fileName[0] == '.' {
+			continue
+		}
+
+		interestingFiles = append(interestingFiles, fileName)
+	}
+
+	if len(interestingFiles) < 1 {
+		fmt.Printf("🧙💬 " + theme.ColorGreen + "I don't see any interesting files here..." + theme.ColorReset + "\n")
+		return interestingFiles
+	}
+
+	if len(interestingFiles) > 1 {
+		fmt.Printf("🧙💬 "+theme.ColorGreen+"I see %d interesting files here..."+theme.ColorReset+"\n", len(interestingFiles))
+	} else {
+		fmt.Printf("🧙💬 " + theme.ColorGreen + "I see an interesting file here..." + theme.ColorReset + "\n")
+	}
+
+	for _, file := range interestingFiles {
+		fmt.Printf(" 👀 "+theme.ColorCyan+"%s\n"+theme.ColorReset, file)
+	}
+
+	return interestingFiles
+}
+
+func Wizard(args []string) {
+	if len(args) > 0 {
+		switch args[0] {
+		case "help":
+			fmt.Fprintf(os.Stderr, theme.ColorGreen+"usage"+theme.ColorReset+": rctf "+theme.ColorBlue+"wizard"+theme.ColorReset+"\n")
+			fmt.Fprintf(os.Stderr, "  🧙 do "+theme.ColorPurple+"magic"+theme.ColorReset+" with rctf\n")
+			os.Exit(0)
+		}
+	}
+
+	// create interesting files list
+	interestingFiles := findInterestingFiles()
+
+	// ensure skeleton
+	_, err := os.Stat(config.FolderName)
+	if os.IsNotExist(err) {
+		fmt.Printf("🧙💬 "+theme.ColorGreen+"I see that you don't have a %s.. let me create that for you!"+theme.ColorReset+" 🪄\n", config.FolderName)
+		util.EnsureSkeleton()
+	}
+
+	// init task
+	_, err = os.Stat(config.TaskName)
+	if os.IsNotExist(err) {
+		fmt.Printf("🧙💬 "+theme.ColorGreen+"I see that you don't have a %s.. let me create that for you!"+theme.ColorReset+" 🪄\n", config.TaskName)
+		Init([]string{"flag{.+}"})
+	}
+
+	unzippedAny := false
+
+	// unzip
+	for _, file := range interestingFiles {
+		result := util.RunFileCommandOnFile(file)
+
+		if strings.Contains(result, "Zip archive data") {
+			fmt.Printf("🧙💬 "+theme.ColorGreen+"I see that %s is a zip file.. let me unzip that for you!"+theme.ColorReset+" 🪄\n", file)
+
+			util.UnzipFile(file)
+
+			unzippedAny = true
+		}
+	}
+
+	if unzippedAny {
+		interestingFiles = findInterestingFiles()
+	}
+
+	// add files
+	if len(interestingFiles) > 1 {
+		fmt.Printf("🧙💬 " + theme.ColorGreen + "Let me add those interesting files for you!" + theme.ColorReset + " 🪄\n")
+	} else {
+		fmt.Printf("🧙💬 " + theme.ColorGreen + "Let me add that interesting file for you!" + theme.ColorReset + " 🪄\n")
+	}
+
+	Add(interestingFiles)
+
+	// show status
+	Status([]string{})
+
+	// if binary then pwn
+	for _, file := range interestingFiles {
+		result := util.RunFileCommandOnFile(file)
+
+		if strings.Contains(result, "ELF") {
+			fmt.Printf("🧙💬 "+theme.ColorGreen+"I see that %s is an ELF.. let me pwn that for you!"+theme.ColorReset+" 🪄\n", file)
+
+			Pwn([]string{})
+			break
+		}
+	}
+}

main.go

@@ -12,6 +12,7 @@ import (
 	"rctf/config"
 	"rctf/data"
 	"rctf/theme"
+	"rctf/util"
 )
 
 func parseUserConfig() {
@@ -61,24 +62,6 @@ func parseUserConfig() {
 	}
 }
 
-func ensureDirectory(dirPath string) {
-	if _, err := os.Stat(dirPath); os.IsNotExist(err) {
-		if config.Verbose {
-			fmt.Println("mkdir", dirPath)
-		}
-		err := os.MkdirAll(dirPath, 0755)
-		if err != nil {
-			fmt.Println("error creating directory:", err)
-			os.Exit(1)
-		}
-	}
-}
-
-func ensureSkeleton() {
-	ensureDirectory(config.FolderName)
-	ensureDirectory(config.FilesFolderName)
-}
-
 func main() {
 	flag.BoolVar(&config.Verbose, "v", false, "enable verbose mode")
 
@@ -100,6 +83,7 @@ func main() {
 		fmt.Fprintf(os.Stderr, "  📞 "+theme.ColorBlue+"syscall"+theme.ColorReset+"\n")
 		fmt.Fprintf(os.Stderr, "  📝 "+theme.ColorBlue+"writeup"+theme.ColorReset+"\n")
 		fmt.Fprintf(os.Stderr, "  📚 "+theme.ColorBlue+"cheatsheet"+theme.ColorReset+"\n")
+		fmt.Fprintf(os.Stderr, "  🧙 "+theme.ColorBlue+"wizard"+theme.ColorReset+"\n")
 		fmt.Fprintf(os.Stderr, "\n🚩 https://github.com/rerrorctf/rctf 🚩\n")
 	}
 
@@ -114,10 +98,10 @@ func main() {
 
 	switch flag.Arg(0) {
 	case "init":
-		ensureSkeleton()
+		util.EnsureSkeleton()
 		commands.Init(flag.Args()[1:])
 	case "add":
-		ensureSkeleton()
+		util.EnsureSkeleton()
 		commands.Add(flag.Args()[1:])
 	case "status":
 		commands.Status(flag.Args()[1:])
@@ -126,10 +110,10 @@ func main() {
 	case "docker":
 		commands.Docker(flag.Args()[1:])
 	case "ghidra":
-		ensureSkeleton()
+		util.EnsureSkeleton()
 		commands.Ghidra(flag.Args()[1:])
 	case "ida":
-		ensureSkeleton()
+		util.EnsureSkeleton()
 		commands.Ida(flag.Args()[1:])
 	case "monitor":
 		commands.Monitor(flag.Args()[1:])
@@ -141,6 +125,8 @@ func main() {
 		commands.Writeup(flag.Args()[1:])
 	case "cheatsheet":
 		commands.Cheatsheet(flag.Args()[1:])
+	case "wizard":
+		commands.Wizard(flag.Args()[1:])
 	default:
 		flag.Usage()
 		os.Exit(1)

util/directory.go

@@ -0,0 +1,25 @@
+package util
+
+import (
+	"fmt"
+	"os"
+	"rctf/config"
+)
+
+func EnsureDirectory(dirPath string) {
+	if _, err := os.Stat(dirPath); os.IsNotExist(err) {
+		if config.Verbose {
+			fmt.Println("mkdir", dirPath)
+		}
+		err := os.MkdirAll(dirPath, 0755)
+		if err != nil {
+			fmt.Println("error creating directory:", err)
+			os.Exit(1)
+		}
+	}
+}
+
+func EnsureSkeleton() {
+	EnsureDirectory(config.FolderName)
+	EnsureDirectory(config.FilesFolderName)
+}

util/file.go

@@ -0,0 +1,30 @@
+package util
+
+import (
+	"fmt"
+	"os/exec"
+)
+
+func RunFileCommandOnFile(path string) string {
+	fileOutput := exec.Command("file", path)
+
+	fileOutputResult, err := fileOutput.Output()
+	if err != nil {
+		fmt.Printf("warning: unable to get output from file on %s\n", path)
+		return ""
+	}
+
+	return string(fileOutputResult[len(path)+2 : len(fileOutputResult)-1])
+}
+
+func UnzipFile(path string) {
+	fileOutput := exec.Command("unzip", path)
+
+	fileOutputResult, err := fileOutput.Output()
+	if err != nil {
+		fmt.Printf("warning: unable to unzip file %s\n", path)
+		return
+	}
+
+	fmt.Printf("%s", fileOutputResult)
+}