only use proxy for tcp in packetgen job

arriven · Mar 31, 2022 · 0d22de2 · 0d22de2
1 parent 9ef2c3b
commit 0d22de2
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/examples/config/advanced/packetgen-udp.yaml b/examples/config/advanced/packetgen-udp.yaml
@@ -0,0 +1,13 @@
+jobs:
+  - type: packetgen
+    args:
+      connection:
+        type: net
+        args:
+          protocol: "udp"
+          address: "localhost:1234"
+      packet:
+        application:
+          type: raw
+          data:
+            payload: "test"
diff --git a/src/job/packetgen.go b/src/job/packetgen.go
@@ -117,7 +117,7 @@ func parsePacketgenArgs(ctx context.Context, logger *zap.Logger, globalConfig *G
 		return nil, fmt.Errorf("error parsing packet: %w", err)
 	}
 
-	if globalConfig.ProxyURLs != "" {
+	if globalConfig.ProxyURLs != "" && jobConfig.Connection.Args["protocol"] == "tcp" {
 		jobConfig.Connection.Args["proxy_urls"] = templates.ParseAndExecute(logger, globalConfig.ProxyURLs, ctx)
 	}