优化线程总控 2022-07-19 11:45:1658202310

GhostTroops · Jul 19, 2022 · e0cb025 · e0cb025
1 parent b0c5924
commit e0cb025
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -31,7 +31,7 @@
   * Redis
   * FTP
   * Mongodb
-  * SMB
+  * SMB,同时检测 MS17-010、SmbGhost
   * Telnet
   * Snmp
   * Wap-wsp（Elasticsearch）

diff --git a/brute/bypass403.go b/brute/bypass403.go
@@ -2,6 +2,7 @@ package brute
 
 import (
 	"context"
+	"github.com/hktalent/scan4all/lib"
 	"net/http"
 	"regexp"
 	"strings"
@@ -62,7 +63,7 @@ func constructEndpointPayloads(domain, path string) []string {
 }
 
 func PenetrateEndpoint(wg *sync.WaitGroup, url string, rst chan Result403, header ...string) {
-	ctx, cancel := context.WithTimeout(context.Background(), 20*time.Second)
+	ctx, cancel := context.WithTimeout(lib.Ctx_global, 20*time.Second)
 	defer func() {
 		cancel()
 		wg.Done()

diff --git a/pkg/hydra/smb/smb.go b/pkg/hydra/smb/smb.go
@@ -3,13 +3,14 @@ package smb
 import (
 	"context"
 	"errors"
+	"github.com/hktalent/scan4all/lib"
 	"github.com/stacktitan/smb/smb"
 	"time"
 )
 
 func Check(Host, Username, Domain, Password string, Port int) (bool, error) {
 	status := make(chan error)
-	ctx, cancel := context.WithTimeout(context.Background(), time.Second*5)
+	ctx, cancel := context.WithTimeout(lib.Ctx_global, time.Second*5)
 	defer cancel()
 	options := smb.Options{
 		Host:        Host,