admin/spray
1
0
Fork 0
mirror of https://github.com/chainreactors/spray.git synced 2025-06-22 02:40:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

adapt spray_dict load

Browse Source
This commit is contained in:
M09Ic 2024-08-26 01:32:11 +08:00
parent 105c426396
commit 06bd9820e7
3 changed files with 16 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
pkg/load.go
View File

@ -55,13 +55,20 @@ func LoadFingers() error {
func LoadTemplates() error { func LoadTemplates() error {
var err error var err error
// load rule // load rule
var data map[string]interface{}
err = json.Unmarshal(LoadConfig("spray_rule"), &data) err = json.Unmarshal(LoadConfig("spray_rule"), &Rules)
if err != nil { if err != nil {
return err return err
} }
for k, v := range data {
Rules[k] = v.(string) // load default words
var dicts map[string]string
err = json.Unmarshal(LoadConfig("spray_dict"), &dicts)
if err != nil {
return err
}
for name, wordlist := range dicts {
Dicts[strings.TrimRight(name, ".txt")] = strings.Split(strings.TrimSpace(wordlist), "\n")
} }
// load mask // load mask
@ -131,7 +138,3 @@ func Load() error {
return nil return nil
} }
func LoadDefaultDict() []string {
return strings.Split(strings.TrimSpace(string(LoadConfig("spray_default"))), "\n")
}

8
pkg/utils.go
View File

@ -32,10 +32,10 @@ var (
EnableAllFingerEngine = false EnableAllFingerEngine = false
) )
var ( var (
Rules map[string]string = make(map[string]string) Rules map[string]string = make(map[string]string)
Dicts map[string][]string = make(map[string][]string)
BadExt = []string{".js", ".css", ".scss", ".,", ".jpeg", ".jpg", ".png", ".gif", ".svg", ".vue", ".ts", ".swf", ".pdf", ".mp4", ".zip", ".rar"} BadExt = []string{".js", ".css", ".scss", ".,", ".jpeg", ".jpg", ".png", ".gif", ".svg", ".vue", ".ts", ".swf", ".pdf", ".mp4", ".zip", ".rar"}
BadURL = []string{";", "}", "\\n", "webpack://", "{", "www.w3.org", ".src", ".url", ".att", ".href", "location.href", "javascript:", "location:", ".createObject", ":location", ".path"} BadURL = []string{";", "}", "\\n", "webpack://", "{", "www.w3.org", ".src", ".url", ".att", ".href", "location.href", "javascript:", "location:", ".createObject", ":location", ".path"}
ContentTypeMap = map[string]string{ ContentTypeMap = map[string]string{
"application/javascript": "js", "application/javascript": "js",

2
templates

@ -1 +1 @@
Subproject commit f2980b8d312c8088f3947d914499e96cfc40d975 Subproject commit 24946e621e3ff07440d43082dc0b60a7f10a0134