Skip to content

Commit

Permalink
Refactor loadPayloads to use embedded default wordlist
Browse files Browse the repository at this point in the history
CX330Blake committed Dec 11, 2024
1 parent 3106175 commit a5cbcaf
Showing 2 changed files with 20,155 additions and 20,153 deletions.
40,296 changes: 20,148 additions & 20,148 deletions default.txt → assets/default.txt

Large diffs are not rendered by default.

12 changes: 7 additions & 5 deletions main.go
Original file line number Diff line number Diff line change
@@ -6,19 +6,25 @@ import (
"io"
"log"
"os"
"strings"

"github.com/CX330Blake/letsgo/pkg/greet"
"github.com/CX330Blake/letsgo/pkg/letsgo"
"github.com/fatih/color"

_ "embed"
)

//go:embed assets/default.txt
var defaultWordlist string

// Load payload
func loadPayloads(filePath string) ([]string, bool) {
useDefault := false
file, err := os.Open(filePath)
if err != nil {
file, _ = os.Open("./default.txt")
useDefault = true
return strings.Split(defaultWordlist, "\n"), useDefault
// return nil, fmt.Errorf("cannot load the wordlist: %v", err)
}
defer file.Close()
@@ -29,10 +35,6 @@ func loadPayloads(filePath string) ([]string, bool) {
payloads = append(payloads, scanner.Text())
}

// if err := scanner.Err(); err != nil {
// return nil, useDefault
// }

return payloads, useDefault
}

0 comments on commit a5cbcaf

Please sign in to comment.