1 files changed, 51 insertions, 0 deletions
diff --git a/xdp-trafficgen/xdp-trafficgen.h b/xdp-trafficgen/xdp-trafficgen.h
new file mode 100644
index 0000000..f1e841b
--- /dev/null
+++ b/xdp-trafficgen/xdp-trafficgen.h
@@ -0,0 +1,51 @@
+/* SPDX-License-Identifier: GPL-2.0 */
+
+#ifndef XDP_TRAFFICGEN_H
+#define XDP_TRAFFICGEN_H
+
+#include <linux/bpf.h>
+#include <linux/if_ether.h>
+#include <linux/ipv6.h>
+
+struct tcp_flowkey {
+  struct in6_addr src_ip;
+  struct in6_addr dst_ip;
+  __u16 dst_port;
+  __u16 src_port;
+};
+
+#define FLOW_STATE_NEW 1
+#define FLOW_STATE_RUNNING 2
+#define FLOW_STATE_DONE 3
+
+struct tcp_flowstate {
+  struct bpf_spin_lock lock;
+  __u8 dst_mac[ETH_ALEN];
+  __u8 src_mac[ETH_ALEN];
+  __u64 last_progress;
+  __u64 retransmits;
+  __u32 flow_state;
+  __u32 seq;     /* our last sent seqno */
+  __u32 ack_seq; /* last seqno that got acked */
+  __u32 rcv_seq; /* receiver's seqno (our ACK seq) */
+  __u32 dupack;
+  __u32 last_print;
+  __u32 highest_seq;
+  __u16 window;
+  __u8 wscale;
+};
+
+struct trafficgen_config {
+  int ifindex_out;
+  __u16 port_start;
+  __u16 port_range;
+};
+
+struct trafficgen_state {
+  struct tcp_flowkey flow_key;
+  __u16 next_port;
+};
+
+
+
+#endif