diff options
| -rw-r--r-- | trafgen_conf.h | 2 | ||||
| -rw-r--r-- | trafgen_parser.y | 5 |
2 files changed, 7 insertions, 0 deletions
diff --git a/trafgen_conf.h b/trafgen_conf.h index 887fa3d..50b437b 100644 --- a/trafgen_conf.h +++ b/trafgen_conf.h @@ -62,4 +62,6 @@ extern void cleanup_packets(void); extern void set_fill(uint8_t val, size_t len); +extern struct packet *current_packet(void); + #endif /* TRAFGEN_CONF */ diff --git a/trafgen_parser.y b/trafgen_parser.y index 49ee577..4725f7c 100644 --- a/trafgen_parser.y +++ b/trafgen_parser.y @@ -137,6 +137,11 @@ static void realloc_packet(void) __init_new_csum_slot(&packet_dyn[packetd_last]); } +struct packet *current_packet(void) +{ + return &packets[packet_last]; +} + static void set_byte(uint8_t val) { struct packet *pkt = &packets[packet_last]; |