0295abb917
* ignore the compiled binary * added possibility to read wordlist from standard input with -w - * Update README.md Co-Authored-By: Joona Hoikkala <joohoi@users.noreply.github.com> * Update main.go Co-Authored-By: Joona Hoikkala <joohoi@users.noreply.github.com> * updated changelog about the wordlist standard input mode * Update README.md Co-Authored-By: Joona Hoikkala <joohoi@users.noreply.github.com>
111 lines
2.2 KiB
Go
111 lines
2.2 KiB
Go
package input
|
|
|
|
import (
|
|
"bufio"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/ffuf/ffuf/pkg/ffuf"
|
|
)
|
|
|
|
type WordlistInput struct {
|
|
config *ffuf.Config
|
|
data [][]byte
|
|
position int
|
|
}
|
|
|
|
func NewWordlistInput(conf *ffuf.Config) (*WordlistInput, error) {
|
|
var wl WordlistInput
|
|
wl.config = conf
|
|
wl.position = -1
|
|
var valid bool
|
|
var err error
|
|
// stdin?
|
|
if conf.Wordlist == "-" {
|
|
// yes
|
|
valid = true
|
|
} else {
|
|
// no
|
|
valid, err = wl.validFile(conf.Wordlist)
|
|
}
|
|
if err != nil {
|
|
return &wl, err
|
|
}
|
|
if valid {
|
|
err = wl.readFile(conf.Wordlist)
|
|
}
|
|
return &wl, err
|
|
}
|
|
|
|
//Next will increment the cursor position, and return a boolean telling if there's words left in the list
|
|
func (w *WordlistInput) Next() bool {
|
|
w.position++
|
|
if w.position >= len(w.data) {
|
|
return false
|
|
}
|
|
return true
|
|
}
|
|
|
|
//Value returns the value from wordlist at current cursor position
|
|
func (w *WordlistInput) Value() []byte {
|
|
return w.data[w.position]
|
|
}
|
|
|
|
//Total returns the size of wordlist
|
|
func (w *WordlistInput) Total() int {
|
|
return len(w.data)
|
|
}
|
|
|
|
//validFile checks that the wordlist file exists and can be read
|
|
func (w *WordlistInput) validFile(path string) (bool, error) {
|
|
_, err := os.Stat(path)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
f, err := os.Open(path)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
f.Close()
|
|
return true, nil
|
|
}
|
|
|
|
//readFile reads the file line by line to a byte slice
|
|
func (w *WordlistInput) readFile(path string) error {
|
|
var file *os.File
|
|
var err error
|
|
if path == "-" {
|
|
file = os.Stdin
|
|
} else {
|
|
file, err = os.Open(path)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
defer file.Close()
|
|
|
|
var data [][]byte
|
|
reader := bufio.NewScanner(file)
|
|
for reader.Scan() {
|
|
if w.config.DirSearchCompat && len(w.config.Extensions) > 0 {
|
|
if strings.Index(reader.Text(), "%EXT%") != -1 {
|
|
for _, ext := range w.config.Extensions {
|
|
contnt := strings.Replace(reader.Text(), "%EXT%", ext, -1)
|
|
data = append(data, []byte(contnt))
|
|
}
|
|
} else {
|
|
data = append(data, []byte(reader.Text()))
|
|
}
|
|
} else {
|
|
data = append(data, []byte(reader.Text()))
|
|
if len(w.config.Extensions) > 0 {
|
|
for _, ext := range w.config.Extensions {
|
|
data = append(data, []byte(reader.Text()+ext))
|
|
}
|
|
}
|
|
}
|
|
}
|
|
w.data = data
|
|
return reader.Err()
|
|
}
|