implement http payload for packetgen

src/core/http/http.go

@@ -34,7 +34,6 @@ import (
 	"go.uber.org/zap"
 
 	"github.com/Arriven/db1000n/src/utils"
-	"github.com/Arriven/db1000n/src/utils/metrics"
 	"github.com/Arriven/db1000n/src/utils/templates"
 )
 
@@ -48,7 +47,7 @@ type RequestConfig struct {
 }
 
 // InitRequest is used to populate data from request config to fasthttp.Request
-func InitRequest(c RequestConfig, req *fasthttp.Request) int64 {
+func InitRequest(c RequestConfig, req *fasthttp.Request) {
 	req.SetRequestURI(c.Path)
 	req.Header.SetMethod(c.Method)
 	req.SetBodyString(c.Body)
@@ -62,10 +61,6 @@ func InitRequest(c RequestConfig, req *fasthttp.Request) int64 {
 	for key, value := range c.Cookies {
 		req.Header.SetCookie(key, value)
 	}
-
-	dataSize, _ := req.WriteTo(metrics.NopWriter{})
-
-	return dataSize
 }
 
 type Client interface {

src/core/packetgen/payload.go

@@ -23,11 +23,14 @@
 package packetgen
 
 import (
+	"bytes"
 	"fmt"
 
 	"github.com/google/gopacket"
 	"github.com/google/gopacket/layers"
+	"github.com/valyala/fasthttp"
 
+	"github.com/Arriven/db1000n/src/core/http"
 	"github.com/Arriven/db1000n/src/utils"
 )
 
@@ -45,20 +48,12 @@ func BuildPayload(c LayerConfig) (gopacket.Layer, error) {
 		}
 
 		return gopacket.Payload([]byte(packetConfig.Payload)), nil
+	case "http":
+		return buildHTTPPacket(c.Data)
 	case "icmpv4":
-		var packetConfig ICMPV4PacketConfig
-		if err := utils.Decode(c.Data, &packetConfig); err != nil {
-			return nil, err
-		}
-
-		return buildICMPV4Packet(packetConfig), nil
+		return buildICMPV4Packet(c.Data)
 	case "dns":
-		var packetConfig DNSPacketConfig
-		if err := utils.Decode(c.Data, &packetConfig); err != nil {
-			return nil, err
-		}
-
-		return buildDNSPacket(packetConfig), nil
+		return buildDNSPacket(c.Data)
 	default:
 		return nil, fmt.Errorf("unsupported layer type %s", c.Type)
 	}
@@ -70,12 +65,17 @@ type ICMPV4PacketConfig struct {
 	Seq      uint16
 }
 
-func buildICMPV4Packet(c ICMPV4PacketConfig) *layers.ICMPv4 {
+func buildICMPV4Packet(data map[string]any) (*layers.ICMPv4, error) {
+	var c ICMPV4PacketConfig
+	if err := utils.Decode(data, &c); err != nil {
+		return nil, err
+	}
+
 	return &layers.ICMPv4{
 		TypeCode: layers.ICMPv4TypeCode(c.TypeCode),
 		Id:       c.ID,
 		Seq:      c.Seq,
-	}
+	}, nil
 }
 
 // DNSQuestion wraps a single request (question) within a DNS query.
@@ -103,7 +103,12 @@ type DNSPacketConfig struct {
 	Questions []DNSQuestion
 }
 
-func buildDNSPacket(c DNSPacketConfig) *layers.DNS {
+func buildDNSPacket(data map[string]any) (*layers.DNS, error) {
+	var c DNSPacketConfig
+	if err := utils.Decode(data, &c); err != nil {
+		return nil, err
+	}
+
 	questions := make([]layers.DNSQuestion, 0, len(c.Questions))
 	for _, question := range c.Questions {
 		questions = append(questions, layers.DNSQuestion{Name: []byte(question.Name), Type: question.Type, Class: question.Class})
@@ -122,5 +127,25 @@ func buildDNSPacket(c DNSPacketConfig) *layers.DNS {
 
 		QDCount:   utils.NonNilOrDefault(c.QDCount, uint16(len(c.Questions))),
 		Questions: questions,
+	}, nil
+}
+
+func buildHTTPPacket(data map[string]any) (gopacket.Payload, error) {
+	var c http.RequestConfig
+
+	if err := utils.Decode(data, &c); err != nil {
+		return nil, err
 	}
+
+	var req fasthttp.Request
+
+	http.InitRequest(c, &req)
+
+	var buf bytes.Buffer
+
+	if _, err := req.WriteTo(&buf); err != nil {
+		return nil, err
+	}
+
+	return gopacket.Payload(buf.Bytes()), nil
 }

src/job/http.go

@@ -72,7 +72,9 @@ func singleRequestJob(ctx context.Context, logger *zap.Logger, globalConfig *Glo
 		log.Printf("Sent single http request to %v", requestConfig.Path)
 	}
 
-	dataSize := http.InitRequest(requestConfig, req)
+	http.InitRequest(requestConfig, req)
+
+	dataSize, _ := req.WriteTo(metrics.NopWriter{})
 
 	metrics.Default.Write(metrics.Traffic, uuid.New().String(), uint64(dataSize))
 
@@ -153,7 +155,9 @@ func fastHTTPJob(ctx context.Context, logger *zap.Logger, globalConfig *GlobalCo
 			return nil, fmt.Errorf("error executing request template: %w", err)
 		}
 
-		dataSize := http.InitRequest(requestConfig, req)
+		http.InitRequest(requestConfig, req)
+
+		dataSize, _ := req.WriteTo(metrics.NopWriter{})
 
 		trafficMonitor.Add(uint64(dataSize))