critical bug fix

fagci · Apr 2, 2022 · de6a51e · de6a51e
1 parent 754d069
commit de6a51e
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 18 deletions.
diff --git a/src/services/portscan.go b/src/services/portscan.go
@@ -25,14 +25,12 @@ func (s *PortscanService) ScanAddr(addr net.TCPAddr, ch chan<- HostResult, wg *s
 }
 
 func (s *PortscanService) Check(ip net.IP, ch chan<- HostResult, swg *sync.WaitGroup) {
-    defer swg.Done()
+	defer swg.Done()
 	var wg sync.WaitGroup
-	go func() {
-		for _, port := range s.Ports {
-			addr := net.TCPAddr{IP: ip, Port: port}
-			wg.Add(1)
-			go s.ScanAddr(addr, ch, &wg)
-		}
-		wg.Wait()
-	}()
+	for _, port := range s.Ports {
+		addr := net.TCPAddr{IP: ip, Port: port}
+		wg.Add(1)
+		go s.ScanAddr(addr, ch, &wg)
+	}
+	wg.Wait()
 }
diff --git a/src/services/rtsp.go b/src/services/rtsp.go
@@ -1,13 +1,14 @@
 package services
 
 import (
-	"github.com/fagci/gons/src/protocol"
-	"github.com/fagci/gons/src/utils"
 	"net"
 	"net/url"
 	"strings"
 	"sync"
 	"time"
+
+	"github.com/fagci/gons/src/protocol"
+	"github.com/fagci/gons/src/utils"
 )
 
 type RTSPService struct {
@@ -42,13 +43,11 @@ func (s *RTSPService) ScanAddr(addr net.TCPAddr, ch chan<- HostResult, wg *sync.
 func (s *RTSPService) Check(host net.IP, ch chan<- HostResult, swg *sync.WaitGroup) {
 	defer swg.Done()
 	var wg sync.WaitGroup
-	go func() {
-		for _, port := range s.Ports {
-			addr := net.TCPAddr{IP: host, Port: port}
-			wg.Add(1)
-			go s.ScanAddr(addr, ch, &wg)
-		}
-	}()
+	for _, port := range s.Ports {
+		addr := net.TCPAddr{IP: host, Port: port}
+		wg.Add(1)
+		go s.ScanAddr(addr, ch, &wg)
+	}
 	wg.Wait()
 }