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

修复多个解析输入值的bug

Browse Source
This commit is contained in:
M09Ic 2024-05-30 18:55:02 +08:00
parent cbb8e2f295
commit 4c8c00416b
1 changed files with 9 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
internal/option.go
View File

@ -443,15 +443,16 @@ func (opt *Option) PrepareRunner() (*Runner, error) {
// 根据不同的输入类型生成任务 // 根据不同的输入类型生成任务
if len(opt.URL) == 1 { if len(opt.URL) == 1 {
u, err := url.Parse(opt.URL[0]) //u, err := fixUrl(opt.URL[0])
if err != nil { //if err != nil {
u, _ = url.Parse("http://" + opt.URL[0]) // return nil, err
} //}
go func() { go func() {
opt.GenerateTasks(tasks, u.String(), ports) opt.GenerateTasks(tasks, opt.URL[0], ports)
close(tasks) close(tasks)
}() }()
taskfrom = u.Host parsed, _ := url.Parse(opt.URL[0])
taskfrom = parsed.Host
r.Count = 1 r.Count = 1
} else if len(opt.URL) > 1 { } else if len(opt.URL) > 1 {
go func() { go func() {
@ -771,7 +772,7 @@ func (opt *Option) Validate() error {
func (opt *Option) GenerateTasks(ch chan *Task, u string, ports []string) { func (opt *Option) GenerateTasks(ch chan *Task, u string, ports []string) {
parsed, err := url.Parse(u) parsed, err := url.Parse(u)
if err != nil { if err != nil {
logs.Log.Warn(err.Error()) logs.Log.Warnf("parse %s, %s ", u, err.Error())
return return
} }
@ -784,7 +785,7 @@ func (opt *Option) GenerateTasks(ch chan *Task, u string, ports []string) {
} }
if len(ports) == 0 { if len(ports) == 0 {
ch <- &Task{baseUrl: u} ch <- &Task{baseUrl: parsed.String()}
return return
} }