Added support for writing results as JSON.

README.md

@@ -65,6 +65,7 @@ All funds that are donated to this project will be donated to charity. A full lo
 * `-l` - show the length of the response.
 * `-n` - "no status" mode, disables the output of the result's status code.
 * `-o <file>` - specify a file name to write the output to.
+* `-oj <file>` - specify a file name to write the output as JSON.
 * `-p <proxy url>` - specify a proxy to use for all requests (scheme much match the URL scheme).
 * `-r` - follow redirects.
 * `-s <status codes>` - comma-separated set of the list of status codes to be deemed a "positive" (default: `200,204,301,302,307`).

libgobuster/libgobuster.go

@@ -4,6 +4,7 @@ import (
 	"bufio"
 	"bytes"
 	"context"
+	"encoding/json"
 	"fmt"
 	"net"
 	"os"
@@ -33,6 +34,7 @@ type Gobuster struct {
 	context          context.Context
 	requestsExpected int
 	requestsIssued   int
+	cachedResults    []Result
 	mu               *sync.RWMutex
 	plugin           GobusterPlugin
 	IsWildcard       bool
@@ -79,6 +81,12 @@ func (g *Gobuster) Results() <-chan Result {
 	return g.resultChan
 }
 
+// Returns our results as JSON
+func (g *Gobuster) ResultsAsJson() (string, error) {
+	jsonResults, err := json.Marshal(g.cachedResults)
+	return string(jsonResults), err
+}
+
 // Errors returns a channel of errors
 func (g *Gobuster) Errors() <-chan error {
 	return g.errorChan
@@ -145,13 +153,20 @@ func (g *Gobuster) worker(wordChan <-chan string, wg *sync.WaitGroup) {
 				continue
 			} else {
 				for _, r := range res {
+					g.cacheResult(r)
 					g.resultChan <- r
 				}
 			}
 		}
 	}
 }
 
+func (g *Gobuster) cacheResult(r Result) {
+	g.mu.Lock()
+	g.cachedResults = append(g.cachedResults, r)
+	g.mu.Unlock()
+}
+
 func (g *Gobuster) getWordlist() (*bufio.Scanner, error) {
 	if g.Opts.Wordlist == "-" {
 		// Read directly from stdin

main.go

@@ -94,6 +94,28 @@ func progressWorker(c context.Context, g *libgobuster.Gobuster) {
 	}
 }
 
+func writeJsonFile(g *libgobuster.Gobuster, jsonOutputFile string) {
+	var err error
+	var f *os.File
+	f, err = os.Create(jsonOutputFile)
+
+	if err != nil {
+		fmt.Errorf("[!] Error on creating JSON output file: %v", err)
+	}
+
+	jsonResults, err := g.ResultsAsJson()
+	if err != nil {
+		fmt.Errorf("[!] Error on ecoding results as JSON: %v", err)
+	}
+
+	if f != nil {
+		err = writeToFile(f, jsonResults)
+		if err != nil {
+			fmt.Errorf("[!] Error on writing to JSON output file: %v", err)
+		}
+	}
+}
+
 func writeToFile(f *os.File, output string) error {
 	_, err := f.WriteString(fmt.Sprintf("%s\n", output))
 	if err != nil {
@@ -104,12 +126,14 @@ func writeToFile(f *os.File, output string) error {
 
 func main() {
 	var outputFilename string
+	var outputJsonFilename string
 	o := libgobuster.NewOptions()
 	flag.IntVar(&o.Threads, "t", 10, "Number of concurrent threads")
 	flag.StringVar(&o.Mode, "m", "dir", "Directory/File mode (dir) or DNS mode (dns)")
 	flag.StringVar(&o.Wordlist, "w", "", "Path to the wordlist")
 	flag.StringVar(&o.StatusCodes, "s", "200,204,301,302,307,403", "Positive status codes (dir mode only)")
 	flag.StringVar(&outputFilename, "o", "", "Output file to write results to (defaults to stdout)")
+	flag.StringVar(&outputJsonFilename, "oj", "", "Output file to write json results to")
 	flag.StringVar(&o.URL, "u", "", "The target URL or Domain")
 	flag.StringVar(&o.Cookies, "c", "", "Cookies to use for the requests (dir mode only)")
 	flag.StringVar(&o.Username, "U", "", "Username for Basic Auth (dir mode only)")
@@ -133,6 +157,10 @@ func main() {
 
 	flag.Parse()
 
+	if outputFilename == outputJsonFilename {
+		log.Fatal("[!] Output file name and JSON output file name must differ!")
+	}
+
 	// Prompt for PW if not provided
 	if o.Username != "" && o.Password == "" {
 		fmt.Printf("[?] Auth Password: ")
@@ -207,6 +235,11 @@ func main() {
 		wg.Wait()
 	}
 
+	if len(outputJsonFilename) > 0 {
+		writeJsonFile(gobuster, outputJsonFilename)
+	}
+
+
 	if !o.Quiet {
 		gobuster.ClearProgress()
 		ruler()