From 1eddc5fcd34ac90df83c1b3ab86db69d3fc5a24a Mon Sep 17 00:00:00 2001 From: M09Ic Date: Sat, 3 Jun 2023 16:39:35 +0800 Subject: [PATCH] fix. url list file parse bug --- 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)