forked from C-Sto/recursebuster
-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.go
80 lines (72 loc) · 12.5 KB
/
main.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
package main
import (
"flag"
"os"
"runtime"
"github.com/c-sto/recursebuster/cmd/cli"
"github.com/c-sto/recursebuster/pkg/recursebuster"
"github.com/fatih/color"
)
const version = "1.6.13"
func main() {
if runtime.GOOS == "windows" { //lol goos
//can't use color.Error, because *nix etc don't have that for some reason :(
recursebuster.InitLogger(color.Output, color.Output, color.Output, color.Output, color.Output, color.Output, color.Output, color.Output, color.Output, color.Output)
} else {
recursebuster.InitLogger(os.Stdout, os.Stdout, os.Stdout, os.Stdout, os.Stdout, os.Stdout, os.Stdout, os.Stdout, os.Stdout, os.Stderr)
}
//the state should probably change per different host. eventually
globalState := recursebuster.State{}.Init()
globalState.Hosts.Init() //**
globalState.Cfg.Version = version //**
totesTested := uint64(0)
globalState.TotalTested = &totesTested
flag.BoolVar(&globalState.Cfg.ShowAll, "all", false, "Show, and write the result of all checks") //Test written
flag.BoolVar(&globalState.Cfg.AppendDir, "appendslash", false, "Append a / to all directory bruteforce requests (like extension, but slash instead of .yourthing)") //Test Written
flag.BoolVar(&globalState.Cfg.Ajax, "ajax", false, "Add the X-Requested-With: XMLHttpRequest header to all requests") //Test Written
flag.StringVar(&globalState.Cfg.Auth, "auth", "", "Basic auth. Supply this with the base64 encoded portion to be placed after the word 'Basic' in the Authorization header.") //Test Written
flag.StringVar(&globalState.Cfg.BadResponses, "bad", "404", "Responses to consider 'bad' or 'not found'. Comma-separated. This works the opposite way of gobuster!") //Test Written
flag.StringVar(&globalState.Cfg.GoodResponses, "good", "", "Whitelist of response codes to consider good. If this flag is used, ONLY the provided codes will be considered 'good'.")
flag.Var(&globalState.Cfg.BadHeader, "badheader", "Check for presence of this header. If an prefix match is found, the response is considered bad.Supply as key:value. Can specify multiple - eg '-badheader Location:cats -badheader X-ATT-DeviceId:XXXXX'") //Test Written
flag.StringVar(&globalState.Cfg.BadBod, "badbod", "", "Body value to look for to consider the response 'bad'.")
flag.StringVar(&globalState.Cfg.BodyContent, "body", "", "File containing content to send in the body of the request. Content-length header will be set accordingly. Note: HEAD requests will/should fail (server will reply with a 400). Set the '-nohead' option to prevent HEAD being sent before a GET.") //Test Written
flag.StringVar(&globalState.Cfg.BlacklistLocation, "blacklist", "", "Blacklist of prefixes to not check. Will not check on exact matches.") //Test Written
flag.StringVar(&globalState.Cfg.Canary, "canary", "", "Custom value to use to check for wildcards") //Test Written
flag.BoolVar(&globalState.Cfg.CleanOutput, "clean", false, "Output clean URLs to the output file for easy loading into other tools and whatnot.") //todo: add test
flag.StringVar(&globalState.Cfg.Cookies, "cookies", "", "Any cookies to include with requests. This is smashed into the cookies header, so copy straight from burp I guess? (-cookies 'cookie1=cookie1content; cookie2=1; cookie3=cookie3lol?!?;'.") //Test Written
flag.BoolVar(&globalState.Cfg.Debug, "debug", false, "Enable debugging") //todo: add test
flag.StringVar(&globalState.Cfg.Extensions, "ext", "", "Extensions to append to checks. Multiple extensions can be specified, comma separate them. (-ext 'csv,exe,aspx')") //Test Written
flag.Var(&globalState.Cfg.Headers, "headers", "Additional headers to include with request. Supply as key:value. Can specify multiple - eg '-headers X-Forwarded-For:127.0.01 -headers X-ATT-DeviceId:XXXXX'") //Test Written
flag.BoolVar(&globalState.Cfg.HTTPS, "https", false, "Use HTTPS instead of HTTP.") //todo: write test
flag.StringVar(&globalState.Cfg.InputList, "iL", "", "File to use as an input list of URL's to start from") //todo: write test
flag.BoolVar(&globalState.Cfg.SSLIgnore, "k", false, "Ignore SSL check") //todo: write test
flag.BoolVar(&globalState.Cfg.ShowLen, "len", false, "Show, and write the length of the response") //todo: write test
flag.BoolVar(&globalState.Cfg.NoBase, "nobase", false, "Don't perform a request to the base URL") //todo: write test
flag.BoolVar(&globalState.Cfg.NoGet, "noget", false, "Do not perform a GET request (only use HEAD request/response)") //Test Written
flag.BoolVar(&globalState.Cfg.NoEncode, "nodencode", false, "Don't encode non-url safe words in the wordlist")
flag.BoolVar(&globalState.Cfg.NoHead, "nohead", false, "Don't optimize GET requests with a HEAD (only send the GET)") //Test Written
flag.BoolVar(&globalState.Cfg.NoRecursion, "norecursion", false, "Disable recursion, just work on the specified directory. Also disables spider function.") //Test Written
flag.BoolVar(&globalState.Cfg.NoSpider, "nospider", false, "Don't search the page body for links and directories to add to the spider queue.") //Test Written
flag.BoolVar(&globalState.Cfg.NoStatus, "nostatus", false, "Don't print status info (for if it messes with the terminal)") //todo: write test
flag.BoolVar(&globalState.Cfg.NoStartStop, "nostartstop", false, "Don't show start/stop info messages") //todo: write test
flag.BoolVar(&globalState.Cfg.NoWildcardChecks, "nowildcard", false, "Don't perform wildcard checks for soft 404 detection (or in plain english, don't do soft404)") //Test Written
flag.BoolVar(&globalState.Cfg.NoUI, "noui", false, "Don't use sexy ui") //todo: write test
flag.StringVar(&globalState.Cfg.Localpath, "o", "."+string(os.PathSeparator)+"busted.txt", "Local file to dump into") //todo: write test
flag.StringVar(&globalState.Cfg.Methods, "methods", "GET", "Methods to use for checks. Multiple methods can be specified, comma separate them. Requests will be sent with an empty body (unless body is specified)") //Test Written
flag.StringVar(&globalState.Cfg.ProxyAddr, "proxy", "", "Proxy configuration options in the form ip:port eg: 127.0.0.1:9050. Note! If you want this to work with burp/use it with a HTTP proxy, specify as http://ip:port") //todo: write test
flag.Float64Var(&globalState.Cfg.Ratio404, "ratio", 0.95, "Similarity ratio to the 404 canary page.") //todo: write test
flag.BoolVar(&globalState.Cfg.NoRobots, "norobots", false, "Don't query and add robots.txt values to checks")
flag.BoolVar(&globalState.Cfg.FollowRedirects, "redirect", false, "Follow redirects") //todo: write test
flag.BoolVar(&globalState.Cfg.BurpMode, "sitemap", false, "Send 'good' requests to the configured proxy. Requires the proxy flag to be set. ***NOTE: with this option, the proxy is ONLY used for good requests - all other requests go out as normal!***") //todo: write test
flag.IntVar(&globalState.Cfg.Threads, "t", 1, "Number of concurrent threads") //todo: write test
flag.IntVar(&globalState.Cfg.Timeout, "timeout", 20, "Timeout (seconds) for HTTP/TCP connections") //todo: write test
flag.StringVar(&globalState.Cfg.URL, "u", "", "Url to spider") //todo: write test
flag.StringVar(&globalState.Cfg.Agent, "ua", "RecurseBuster/"+version, "User agent to use when sending requests.") //todo: write test
flag.IntVar(&globalState.Cfg.VerboseLevel, "v", 0, "Verbosity level for output messages.") //todo: write test
flag.StringVar(&globalState.Cfg.Vhost, "vhost", "", "Vhost to send") //test written
flag.BoolVar(&globalState.Cfg.ShowVersion, "version", false, "Show version number and exit") //todo: write test
flag.StringVar(&globalState.Cfg.Wordlist, "w", "", "Wordlist to use for bruteforce. Blank for spider only") //todo: write test
flag.StringVar(&globalState.Cfg.WhitelistLocation, "whitelist", "", "Whitelist of domains to include in brute-force") //todo: write test
flag.Parse()
cli.Run(globalState)
}