From 976ce55bdcaf319e67ffb1fb75eb1d190d5228a0 Mon Sep 17 00:00:00 2001 From: M09Ic Date: Sat, 3 Jun 2023 16:39:35 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8Durl=E8=A7=A3=E6=9E=90?= =?UTF-8?q?=E9=94=99=E8=AF=AF=E7=9A=84bug?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- internal/option.go | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/internal/option.go b/internal/option.go index c0a28ca..0539689 100644 --- a/internal/option.go +++ b/internal/option.go @@ -428,11 +428,12 @@ func (opt *Option) PrepareRunner() (*Runner, error) { if file != nil { content, err := ioutil.ReadAll(file) if err != nil { - logs.Log.Error(err.Error()) + return nil, err } urls := strings.Split(strings.TrimSpace(string(content)), "\n") for _, u := range urls { - if _, err := url.Parse(u); err != nil { + u = strings.TrimSpace(u) + if _, err := url.Parse(u); err == nil { r.Count++ } else if ip := utils.ParseIP(u); ip != nil { r.Count++ @@ -443,7 +444,8 @@ func (opt *Option) PrepareRunner() (*Runner, error) { go func() { 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) } else if ip := utils.ParseIP(u); ip != nil { opt.GenerateTasks(tasks, u, ports)