diff options
| -rw-r--r-- | trafgen_lexer.l | 15 | ||||
| -rw-r--r-- | trafgen_parser.y | 51 |
2 files changed, 66 insertions, 0 deletions
diff --git a/trafgen_lexer.l b/trafgen_lexer.l index 5467c70..eeb252f 100644 --- a/trafgen_lexer.l +++ b/trafgen_lexer.l @@ -119,8 +119,23 @@ ip_addr ([0-9]+\.[0-9]+\.[0-9]+\.[0-9]+) "htype" { return K_HTYPE; } "ptype" { return K_PTYPE; } +"ihl" { return K_IHL; } +"ver"|"version" { return K_VER; } +"ttl" { return K_TTL; } +"dscp" { return K_DSCP; } +"ecn" { return K_ECN; } +"tos" { return K_TOS; } +"len"|"length" { return K_LEN; } +"id" { return K_ID; } +"flags" { return K_FLAGS; } +"frag" { return K_FRAG; } +"csum" { return K_CSUM; } +"df" { return K_DF; } +"mf" { return K_MF; } + "eth" { return K_ETH; } "arp" { return K_ARP; } +"ip4"|"ipv4" { return K_IP4; } [ ]*"-"[ ]* { return '-'; } [ ]*"+"[ ]* { return '+'; } diff --git a/trafgen_parser.y b/trafgen_parser.y index 16f9025..ebc4054 100644 --- a/trafgen_parser.y +++ b/trafgen_parser.y @@ -25,6 +25,7 @@ #include "trafgen_conf.h" #include "trafgen_proto.h" #include "trafgen_l2.h" +#include "trafgen_l3.h" #include "built_in.h" #include "die.h" #include "str.h" @@ -350,9 +351,11 @@ 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_ETH %token K_ARP +%token K_IP4 %token ',' '{' '}' '(' ')' '[' ']' ':' '-' '+' '*' '/' '%' '&' '|' '<' '>' '^' @@ -574,6 +577,7 @@ ddec proto : eth_proto { } | arp_proto { } + | ip4_proto { } ; eth_proto @@ -637,6 +641,53 @@ arp : K_ARP { proto_add(PROTO_ARP); } ; +ip4_proto + : ip4 '(' ip4_param_list ')' { } + ; + +ip4_param_list + : { } + | ip4_field { } + | ip4_field delimiter ip4_param_list { } + ; + +ip4_field + : K_VER skip_white '=' skip_white number + { proto_field_set_u8(hdr, IP4_VER, $5); } + | K_IHL skip_white '=' skip_white number + { proto_field_set_u8(hdr, IP4_IHL, $5); } + | K_DADDR skip_white '=' skip_white ip_addr + { proto_field_set_u32(hdr, IP4_DADDR, $5.s_addr); } + | K_SADDR skip_white '=' skip_white ip_addr + { proto_field_set_u32(hdr, IP4_SADDR, $5.s_addr); } + | K_PROT skip_white '=' skip_white number + { proto_field_set_u8(hdr, IP4_PROTO, $5); } + | K_TTL skip_white '=' skip_white number + { proto_field_set_u8(hdr, IP4_TTL, $5); } + | K_DSCP skip_white '=' skip_white number + { proto_field_set_u8(hdr, IP4_DSCP, $5); } + | K_ECN skip_white '=' skip_white number + { proto_field_set_u8(hdr, IP4_ECN, $5); } + | K_TOS skip_white '=' skip_white number + { proto_field_set_u8(hdr, IP4_TOS, $5); } + | K_LEN skip_white '=' skip_white number + { proto_field_set_be16(hdr, IP4_LEN, $5); } + | K_ID skip_white '=' skip_white number + { proto_field_set_be16(hdr, IP4_ID, $5); } + | K_FLAGS skip_white '=' skip_white number + { proto_field_set_be16(hdr, IP4_FLAGS, $5); } + | K_DF { proto_field_set_be16(hdr, IP4_DF, 1); } + | K_MF { proto_field_set_be16(hdr, IP4_MF, 1); } + | K_FRAG skip_white '=' skip_white number + { proto_field_set_be16(hdr, IP4_FRAG_OFFS, $5); } + | K_CSUM skip_white '=' skip_white number + { proto_field_set_be16(hdr, IP4_CSUM, $5); } + ; + +ip4 + : K_IP4 { proto_add(PROTO_IP4); } + ; + %% static void finalize_packet(void) |