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 {
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)