Exclude certain types from c2 spin up

vulncheck-oss · Jul 29, 2024 · bd3fc01 · bd3fc01
1 parent a454499
commit bd3fc01
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/framework.go b/framework.go
@@ -66,6 +66,7 @@ import (
 	"sync"
 	"time"
 
+	"github.com/vulncheck-oss/go-exploit"
 	"github.com/vulncheck-oss/go-exploit/c2"
 	"github.com/vulncheck-oss/go-exploit/cli"
 	"github.com/vulncheck-oss/go-exploit/config"
@@ -275,7 +276,8 @@ func parseCommandLine(conf *config.Config) bool {
 }
 
 func startC2Server(conf *config.Config) bool {
-	if conf.DoExploit && !conf.ThirdPartyC2Server && conf.Bport == 0 {
+	if conf.DoExploit && !conf.ThirdPartyC2Server && conf.Bport == 0 &&
+		(conf.ExType != exploit.InformationDisclosure && conf.ExType != exploit.Webshell) {
 		c2Impl, success := c2.GetInstance(conf.C2Type)
 		if !success || c2Impl == nil {
 			return false