From 4c8c00416bdc3ae4bb00850667ca642bc94e079b Mon Sep 17 00:00:00 2001 From: M09Ic Date: Thu, 30 May 2024 18:55:02 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E5=A4=9A=E4=B8=AA=E8=A7=A3?= =?UTF-8?q?=E6=9E=90=E8=BE=93=E5=85=A5=E5=80=BC=E7=9A=84bug?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- internal/option.go | 17 +++++++++-------- 1 file changed, 9 insertions(+), 8 deletions(-) diff --git a/internal/option.go b/internal/option.go index fc35556..b374d3a 100644 --- a/internal/option.go +++ b/internal/option.go @@ -443,15 +443,16 @@ func (opt *Option) PrepareRunner() (*Runner, error) { // 根据不同的输入类型生成任务 if len(opt.URL) == 1 { - u, err := url.Parse(opt.URL[0]) - if err != nil { - u, _ = url.Parse("http://" + opt.URL[0]) - } + //u, err := fixUrl(opt.URL[0]) + //if err != nil { + // return nil, err + //} go func() { - opt.GenerateTasks(tasks, u.String(), ports) + opt.GenerateTasks(tasks, opt.URL[0], ports) close(tasks) }() - taskfrom = u.Host + parsed, _ := url.Parse(opt.URL[0]) + taskfrom = parsed.Host r.Count = 1 } else if len(opt.URL) > 1 { go func() { @@ -771,7 +772,7 @@ func (opt *Option) Validate() error { func (opt *Option) GenerateTasks(ch chan *Task, u string, ports []string) { parsed, err := url.Parse(u) if err != nil { - logs.Log.Warn(err.Error()) + logs.Log.Warnf("parse %s, %s ", u, err.Error()) return } @@ -784,7 +785,7 @@ func (opt *Option) GenerateTasks(ch chan *Task, u string, ports []string) { } if len(ports) == 0 { - ch <- &Task{baseUrl: u} + ch <- &Task{baseUrl: parsed.String()} return }