101 lines
2.0 KiB
Go
101 lines
2.0 KiB
Go
|
package gobusterdir
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"fmt"
|
||
|
|
"net/http"
|
||
|
|
|
||
|
|
"github.com/fatih/color"
|
||
|
|
)
|
||
|
|
|
||
|
|
var NoStdout bool
|
||
|
|
|
||
|
|
var (
|
||
|
|
white = color.New(color.FgWhite).FprintfFunc()
|
||
|
|
yellow = color.New(color.FgYellow).FprintfFunc()
|
||
|
|
green = color.New(color.FgGreen).FprintfFunc()
|
||
|
|
blue = color.New(color.FgBlue).FprintfFunc()
|
||
|
|
red = color.New(color.FgRed).FprintfFunc()
|
||
|
|
cyan = color.New(color.FgCyan).FprintfFunc()
|
||
|
|
)
|
||
|
|
|
||
|
|
// Result represents a single result
|
||
|
|
type Result struct {
|
||
|
|
URL string
|
||
|
|
Path string
|
||
|
|
Verbose bool
|
||
|
|
Expanded bool
|
||
|
|
NoStatus bool
|
||
|
|
HideLength bool
|
||
|
|
NoStdout bool
|
||
|
|
Found bool
|
||
|
|
Header http.Header
|
||
|
|
StatusCode int
|
||
|
|
Size int64
|
||
|
|
}
|
||
|
|
|
||
|
|
// ResultToString converts the Result to it's textual representation
|
||
|
|
func (r Result) ResultToString() (string, error) {
|
||
|
|
buf := &bytes.Buffer{}
|
||
|
|
|
||
|
|
// Prefix if we're in verbose mode
|
||
|
|
if r.Verbose {
|
||
|
|
if r.Found {
|
||
|
|
if _, err := fmt.Fprintf(buf, "Found: "); err != nil {
|
||
|
|
return "", err
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
if _, err := fmt.Fprintf(buf, "Missed: "); err != nil {
|
||
|
|
return "", err
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if r.Expanded {
|
||
|
|
if _, err := fmt.Fprintf(buf, "%s", r.URL); err != nil {
|
||
|
|
return "", err
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
if _, err := fmt.Fprintf(buf, "/"); err != nil {
|
||
|
|
return "", err
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if _, err := fmt.Fprintf(buf, "%-20s", r.Path); err != nil {
|
||
|
|
return "", err
|
||
|
|
}
|
||
|
|
|
||
|
|
if !r.NoStatus {
|
||
|
|
color := white
|
||
|
|
if r.StatusCode == 200 {
|
||
|
|
color = green
|
||
|
|
} else if r.StatusCode >= 300 && r.StatusCode < 400 {
|
||
|
|
color = cyan
|
||
|
|
} else if r.StatusCode >= 400 && r.StatusCode < 500 {
|
||
|
|
color = yellow
|
||
|
|
} else if r.StatusCode >= 500 && r.StatusCode < 600 {
|
||
|
|
color = red
|
||
|
|
}
|
||
|
|
|
||
|
|
color(buf, " (Status: %d)", r.StatusCode)
|
||
|
|
}
|
||
|
|
|
||
|
|
if !r.HideLength {
|
||
|
|
if _, err := fmt.Fprintf(buf, " [Size: %d]", r.Size); err != nil {
|
||
|
|
return "", err
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
location := r.Header.Get("Location")
|
||
|
|
if location != "" {
|
||
|
|
blue(buf, " [--> %s]", location)
|
||
|
|
}
|
||
|
|
|
||
|
|
if _, err := fmt.Fprintf(buf, "\n"); err != nil {
|
||
|
|
return "", err
|
||
|
|
}
|
||
|
|
|
||
|
|
s := buf.String()
|
||
|
|
|
||
|
|
return s, nil
|
||
|
|
}
|