admin/spray
1
0
Fork 0
mirror of https://github.com/chainreactors/spray.git synced 2025-09-15 11:40:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

修复url解析错误的bug

Browse Source
This commit is contained in:
M09Ic 2023-06-03 16:39:35 +08:00
parent 2f131ccd02
commit 976ce55bdc
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
internal/option.go
View File

@ -428,11 +428,12 @@ func (opt *Option) PrepareRunner() (*Runner, error) {
if file != nil { if file != nil {
content, err := ioutil.ReadAll(file) content, err := ioutil.ReadAll(file)
if err != nil { if err != nil {
logs.Log.Error(err.Error()) return nil, err
} }
urls := strings.Split(strings.TrimSpace(string(content)), "\n") urls := strings.Split(strings.TrimSpace(string(content)), "\n")
for _, u := range urls { for _, u := range urls {
if _, err := url.Parse(u); err != nil { u = strings.TrimSpace(u)
if _, err := url.Parse(u); err == nil {
r.Count++ r.Count++
} else if ip := utils.ParseIP(u); ip != nil { } else if ip := utils.ParseIP(u); ip != nil {
r.Count++ r.Count++
@ -443,7 +444,8 @@ func (opt *Option) PrepareRunner() (*Runner, error) {
go func() { go func() {
for _, u := range urls { for _, u := range urls {
if _, err := url.Parse(u); err != nil { u = strings.TrimSpace(u)
if _, err := url.Parse(u); err == nil {
opt.GenerateTasks(tasks, u, ports) opt.GenerateTasks(tasks, u, ports)
} else if ip := utils.ParseIP(u); ip != nil { } else if ip := utils.ParseIP(u); ip != nil {
opt.GenerateTasks(tasks, u, ports) opt.GenerateTasks(tasks, u, ports)