feat: add TelemetryURL support to ApiClient and configuration

- Introduced TelemetryURL in the ApiClient struct to allow separate telemetry endpoint usage.
- Updated the agent to utilize TelemetryURL for sending DNS records and network connections.
- Enhanced configuration handling to initialize TelemetryURL from the config file, defaulting to APIURL if not specified.

Author: rohan-stepsecurity

agent.go

@@ -67,7 +67,7 @@ func Run(ctx context.Context, configFilePath string, hostDNSServer DNSServer,
 		return err
 	}
 
-	apiclient := &ApiClient{Client: &http.Client{Timeout: 3 * time.Second}, APIURL: config.APIURL, DisableTelemetry: config.DisableTelemetry, EgressPolicy: config.EgressPolicy, OneTimeKey: config.OneTimeKey}
+	apiclient := &ApiClient{Client: &http.Client{Timeout: 3 * time.Second}, APIURL: config.APIURL, TelemetryURL: config.TelemetryURL, DisableTelemetry: config.DisableTelemetry, EgressPolicy: config.EgressPolicy, OneTimeKey: config.OneTimeKey}
 
 	config.OneTimeKey = ""
 	// TODO: pass in an iowriter/ use log library

apiclient.go

@@ -41,6 +41,7 @@ type NetworkConnection struct {
 type ApiClient struct {
 	Client           *http.Client
 	APIURL           string
+	TelemetryURL     string
 	DisableTelemetry bool
 	EgressPolicy     string
 	OneTimeKey       string
@@ -57,7 +58,7 @@ func (apiclient *ApiClient) sendDNSRecord(correlationId, repo, domainName, ipAdd
 		dnsRecord.ResolvedIPAddress = ipAddress
 		dnsRecord.TimeStamp = time.Now().UTC()
 
-		url := fmt.Sprintf("%s/github/%s/actions/jobs/%s/dns", apiclient.APIURL, repo, correlationId)
+		url := fmt.Sprintf("%s/github/%s/actions/jobs/%s/dns", apiclient.TelemetryURL, repo, correlationId)
 
 		return apiclient.sendApiRequest("POST", url, dnsRecord)
 	}
@@ -76,7 +77,7 @@ func (apiclient *ApiClient) sendNetConnection(correlationId, repo, ipAddress, po
 		networkConnection.TimeStamp = timestamp
 		networkConnection.Tool = tool
 
-		url := fmt.Sprintf("%s/github/%s/actions/jobs/%s/networkconnection", apiclient.APIURL, repo, correlationId)
+		url := fmt.Sprintf("%s/github/%s/actions/jobs/%s/networkconnection", apiclient.TelemetryURL, repo, correlationId)
 
 		return apiclient.sendApiRequest("POST", url, networkConnection)
 	}

config.go

@@ -16,6 +16,7 @@ type config struct {
 	RunId                    string
 	WorkingDirectory         string
 	APIURL                   string
+	TelemetryURL             string
 	OneTimeKey               string
 	Endpoints                map[string][]Endpoint
 	EgressPolicy             string
@@ -37,6 +38,7 @@ type configFile struct {
 	RunId                    string `json:"run_id"`
 	WorkingDirectory         string `json:"working_directory"`
 	APIURL                   string `json:"api_url"`
+	TelemetryURL             string `json:"telemetry_url"`
 	OneTimeKey               string `json:"one_time_key"`
 	AllowedEndpoints         string `json:"allowed_endpoints"`
 	EgressPolicy             string `json:"egress_policy"`
@@ -65,6 +67,10 @@ func (c *config) init(configFilePath string) error {
 	c.RunId = configFile.RunId
 	c.WorkingDirectory = configFile.WorkingDirectory
 	c.APIURL = configFile.APIURL
+	c.TelemetryURL = configFile.TelemetryURL
+	if c.TelemetryURL == "" {
+		c.TelemetryURL = c.APIURL
+	}
 	c.Endpoints = parseEndpoints(configFile.AllowedEndpoints)
 	c.EgressPolicy = configFile.EgressPolicy
 	c.DisableTelemetry = configFile.DisableTelemetry