Add config oriented URL helper.

Author: terrorbyte

config/config.go

@@ -10,6 +10,7 @@ import (
 	"github.com/vulncheck-oss/go-exploit/c2"
 	"github.com/vulncheck-oss/go-exploit/c2/shelltunnel"
 	"github.com/vulncheck-oss/go-exploit/output"
+	"github.com/vulncheck-oss/go-exploit/protocol"
 )
 
 type ExploitType int
@@ -370,6 +371,13 @@ func (conf *Config) ApplyTemplate(name string) string {
 	return buf.String()
 }
 
+// Generate a URL from a path from the current configuration. This is a
+// way of invoking protocol.GenerateURL for developer ergonomics during
+// exploit development.
+func (conf *Config) GenerateURL(path string) string {
+	return protocol.GenerateURL(conf.Rhost, conf.Rport, conf.SSL, path)
+}
+
 // Disable automatic start of c2 servers. Manually starting is required after
 // this function is called. This is useful when you have an exploit that
 // may have multiple stages and you are guaranteed to not need the C2

config/config_test.go

@@ -1,6 +1,7 @@
 package config_test
 
 import (
+	"strings"
 	"testing"
 
 	"github.com/vulncheck-oss/go-exploit/c2"
@@ -97,3 +98,21 @@ func TestApplyTemplate(t *testing.T) {
 		t.Errorf("'%s' unexpected", s)
 	}
 }
+
+func TestGenerateURL(t *testing.T) {
+	conf := config.NewRemoteExploit(
+		config.ImplementedFeatures{AssetDetection: true, VersionScanning: true, Exploitation: true},
+		config.CodeExecution, []c2.Impl{}, "Apache", []string{"OFBiz"},
+		[]string{"cpe:2.3:a:apache:ofbiz"}, "CVE-2024-45507", "HTTP", 80)
+	conf.Rhost = "127.13.37.1"
+	conf.Rport = 31337
+	conf.SSL = true
+
+	if strings.Compare(conf.GenerateURL("/vulncheck"), `https://127.13.37.1:31337/vulncheck`) != 0 {
+		t.Errorf("GenerateURL did not generate expected HTTPS URL: `%s` != `%s`", conf.GenerateURL("/vulncheck"), `https://127.13.37.1:31337/vulncheck`)
+	}
+	conf.SSL = false
+	if strings.Compare(conf.GenerateURL("/vulncheck"), `http://127.13.37.1:31337/vulncheck`) != 0 {
+		t.Errorf("GenerateURL did not generate expected HTTP URL: `%s` != `%s`", conf.GenerateURL("/vulncheck"), `http://127.13.37.1:31337/vulncheck`)
+	}
+}