summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--trafgen_lexer.l4
-rw-r--r--trafgen_parser.y29
2 files changed, 33 insertions, 0 deletions
diff --git a/trafgen_lexer.l b/trafgen_lexer.l
index eeb252f..56011a3 100644
--- a/trafgen_lexer.l
+++ b/trafgen_lexer.l
@@ -133,9 +133,13 @@ ip_addr ([0-9]+\.[0-9]+\.[0-9]+\.[0-9]+)
"df" { return K_DF; }
"mf" { return K_MF; }
+"sp"|"sport" { return K_SPORT; }
+"dp"|"dport" { return K_DPORT; }
+
"eth" { return K_ETH; }
"arp" { return K_ARP; }
"ip4"|"ipv4" { return K_IP4; }
+"udp" { return K_UDP; }
[ ]*"-"[ ]* { return '-'; }
[ ]*"+"[ ]* { return '+'; }
diff --git a/trafgen_parser.y b/trafgen_parser.y
index ebc4054..9ec09b1 100644
--- a/trafgen_parser.y
+++ b/trafgen_parser.y
@@ -26,6 +26,7 @@
#include "trafgen_proto.h"
#include "trafgen_l2.h"
#include "trafgen_l3.h"
+#include "trafgen_l4.h"
#include "built_in.h"
#include "die.h"
#include "str.h"
@@ -352,10 +353,12 @@ static void proto_add(enum proto_id pid)
%token K_DADDR K_SADDR K_PROT
%token K_OPER K_SHA K_SPA K_THA K_TPA K_REQUEST K_REPLY K_PTYPE K_HTYPE
%token K_TTL K_DSCP K_ECN K_TOS K_LEN K_ID K_FLAGS K_FRAG K_IHL K_VER K_CSUM K_DF K_MF
+%token K_SPORT K_DPORT
%token K_ETH
%token K_ARP
%token K_IP4
+%token K_UDP
%token ',' '{' '}' '(' ')' '[' ']' ':' '-' '+' '*' '/' '%' '&' '|' '<' '>' '^'
@@ -578,6 +581,7 @@ proto
: eth_proto { }
| arp_proto { }
| ip4_proto { }
+ | udp_proto { }
;
eth_proto
@@ -688,6 +692,31 @@ ip4
: K_IP4 { proto_add(PROTO_IP4); }
;
+udp_proto
+ : udp '(' udp_param_list ')' { }
+ ;
+
+udp_param_list
+ : { }
+ | udp_field { }
+ | udp_field delimiter udp_param_list { }
+ ;
+
+udp_field
+ : K_SPORT skip_white '=' skip_white number
+ { proto_field_set_be16(hdr, UDP_SPORT, $5); }
+ | K_DPORT skip_white '=' skip_white number
+ { proto_field_set_be16(hdr, UDP_DPORT, $5); }
+ | K_LEN skip_white '=' skip_white number
+ { proto_field_set_be16(hdr, UDP_LEN, $5); }
+ | K_CSUM skip_white '=' skip_white number
+ { proto_field_set_be16(hdr, UDP_CSUM, $5); }
+ ;
+
+udp
+ : K_UDP { proto_add(PROTO_UDP); }
+ ;
+
%%
static void finalize_packet(void)