basic structure (flags)

1 files changed, 87 insertions, 0 deletions

diff --git a/src/flags.go b/src/flags.go
new file mode 100644
index 0000000..c96a7d2
--- /dev/null
+++ b/src/flags.go
@@ -0,0 +1,87 @@
+package main
+
+import (
+	"flag"
+)
+
+var (
+	// boolean flags
+	color           bool
+	verbose         bool
+	dryrun          bool
+	followRedirects bool
+
+	// string values
+	proxy    string
+	url      string
+	payload  string
+	wordlist string
+	data     string
+	header   string
+
+	// int values
+	concurrent int
+	delay      int
+
+	// string values defining what request to show or hide
+	hide Hide
+	show Show
+)
+
+func registerFlags() {
+	shortcut := " (shortcut)"
+
+	// Boolean values
+
+	colorUsage := "Output with color"
+	flag.BoolVar(&color, "color", false, colorUsage)
+	flag.BoolVar(&color, "c", false, colorUsage+shortcut)
+
+	verboseUsage := "Verbose output"
+	flag.BoolVar(&verbose, "verbose", false, verboseUsage)
+	flag.BoolVar(&verbose, "v", false, verboseUsage+shortcut)
+
+	dryrunUsage := "Print the results of applying the requests without actually making any HTTP requests"
+	flag.BoolVar(&dryrun, "dryrun", false, dryrunUsage)
+	flag.BoolVar(&dryrun, "D", false, dryrunUsage+shortcut)
+
+	followRedirectsUsage := "Follow HTTP redirects"
+	flag.BoolVar(&followRedirects, "followRedirects", false, followRedirectsUsage)
+	flag.BoolVar(&followRedirects, "L", false, followRedirectsUsage+shortcut)
+
+	// Strings
+
+	proxyUsage := "Proxy to use"
+	flag.StringVar(&proxy, "proxy", "", proxyUsage)
+	flag.StringVar(&proxy, "p", "", proxyUsage+shortcut)
+
+	urlUsage := "url to use"
+	flag.StringVar(&url, "url", "", urlUsage)
+	flag.StringVar(&url, "u", "", urlUsage+shortcut)
+
+	payloadUsage := "payload type"
+	flag.StringVar(&payload, "payload", "file", payloadUsage)
+	flag.StringVar(&payload, "P", "file", payloadUsage+shortcut)
+
+	wordlistUsage := "wordlist type"
+	flag.StringVar(&wordlist, "wordlist", "", wordlistUsage)
+	flag.StringVar(&wordlist, "w", "", wordlistUsage+shortcut)
+
+	headerUsage := "Define the headers to set"
+	flag.StringVar(&header, "header", "", headerUsage)
+	flag.StringVar(&header, "h", "", headerUsage+shortcut)
+
+	// Ints
+
+	concurrentUsage := "Amount of concurrent connections to use"
+	flag.IntVar(&concurrent, "concurrent", 1, concurrentUsage)
+	flag.IntVar(&concurrent, "t", 1, concurrentUsage+shortcut)
+
+	delayUsage := "Amount of delay in between the requests"
+	flag.IntVar(&delay, "delay", 0, delayUsage)
+	flag.IntVar(&delay, "d", 0, delayUsage+shortcut)
+
+	// parse the flags
+
+	flag.Parse()
+}