trafgen: proto: Reference to packet from struct proto_hdr

Using of current_packet() is not possible for dynamically updated fields
so we need to keep the packet index in proto_hdr struct to reference the
correct packet.

Signed-off-by: Vadim Kochan <vadim4j@gmail.com>
Signed-off-by: Tobias Klauser <tklauser@distanz.ch>

1 files changed, 10 insertions, 0 deletions

diff --git a/trafgen_parser.y b/trafgen_parser.y
index 0fe8674..a286e6b 100644
--- a/trafgen_parser.y
+++ b/trafgen_parser.y
@@ -153,6 +153,16 @@ struct packet *current_packet(void)
 	return &packets[packet_last];
 }
 
+uint32_t current_packet_id(void)
+{
+	return packet_last;
+}
+
+struct packet *packet_get(uint32_t id)
+{
+	return &packets[id];
+}
+
 static void set_byte(uint8_t val)
 {
 	struct packet *pkt = &packets[packet_last];