src/trafgen_parser.y

   1 /*
   2  * netsniff-ng - the packet sniffing beast
   3  * By Daniel Borkmann <daniel@netsniff-ng.org>
   4  * Copyright 2012 Daniel Borkmann <dborkma@tik.ee.ethz.ch>,
   5  * Swiss federal institute of technology (ETH Zurich)
   6  * Subject to the GPL, version 2.
   7  */
   8
   9 %{
  10
  11 #include <stdio.h>
  12 #include <stdlib.h>
  13 #include <signal.h>
  14 #include <stdint.h>
  15 #include <errno.h>
  16
  17 #include "xmalloc.h"
  18 #include "trafgen_parser.tab.h"
  19 #include "trafgen_conf.h"
  20 #include "built_in.h"
  21 #include "die.h"
  22 #include "mtrand.h"
  23
  24 #define YYERROR_VERBOSE         0
  25 #define YYDEBUG                 0
  26 #define YYENABLE_NLS            1
  27 #define YYLTYPE_IS_TRIVIAL      1
  28 #define ENABLE_NLS              1
  29
  30 extern FILE *yyin;
  31 extern int yylex(void);
  32 extern void yyerror(const char *);
  33 extern int yylineno;
  34 extern char *yytext;
  35
  36 static struct pktconf *conf = NULL;
  37 static int note_flag = 0;
  38
  39 #define am(x)   ((x)->len - 1)
  40
  41 static void give_note_dynamic(void)
  42 {
  43         if (!note_flag) {
  44                 printf("Note: dynamic elements like drnd, dinc, ddec and "
  45                        "others make trafgen slower!\n");
  46                 note_flag = 1;
  47         }
  48 }
  49
  50 static void dump_conf(struct pktconf *cfg)
  51 {
  52         size_t i, j;
  53
  54         printf("n %lu, gap %lu us, pkts %zu\n", cfg->num, cfg->gap, cfg->len);
  55         if (cfg->len == 0)
  56                 return;
  57         for (i = 0; i < cfg->len; ++i) {
  58                 printf("[%zu] pkt\n", i);
  59                 printf(" len %zu cnts %zu rnds %zu\n", cfg->pkts[i].plen,
  60                        cfg->pkts[i].clen, cfg->pkts[i].rlen);
  61                 printf(" payload ");
  62                 for (j = 0; j < cfg->pkts[i].plen; ++j)
  63                         printf("%02x ", cfg->pkts[i].payload[j]);
  64                 printf("\n");
  65                 for (j = 0; j < cfg->pkts[i].clen; ++j)
  66                         printf(" cnt%zu [%u,%u], inc %u, off %ld type %s\n",
  67                                j, cfg->pkts[i].cnt[j].min,
  68                                cfg->pkts[i].cnt[j].max,
  69                                cfg->pkts[i].cnt[j].inc,
  70                                cfg->pkts[i].cnt[j].off,
  71                                cfg->pkts[i].cnt[j].type == TYPE_INC ?
  72                                "inc" : "dec");
  73                 for (j = 0; j < cfg->pkts[i].rlen; ++j)
  74                         printf(" rnd%zu off %ld\n",
  75                                j, cfg->pkts[i].rnd[j].off);
  76         }
  77 }
  78
  79 static void realloc_packet(void)
  80 {
  81         conf->len++;
  82         conf->pkts = xrealloc(conf->pkts, 1, conf->len * sizeof(*conf->pkts));
  83         fmemset(&conf->pkts[am(conf)], 0, sizeof(conf->pkts[am(conf)]));
  84 }
  85
  86 static void set_byte(uint8_t val)
  87 {
  88         int base;
  89
  90         conf->pkts[am(conf)].plen++;
  91         conf->pkts[am(conf)].payload = xrealloc(conf->pkts[am(conf)].payload,
  92                                                 1, conf->pkts[am(conf)].plen);
  93
  94         base = conf->pkts[am(conf)].plen - 1;
  95         conf->pkts[am(conf)].payload[base] = val;
  96 }
  97
  98 static void set_fill(uint8_t val, size_t len)
  99 {
 100         int i, base;
 101
 102         conf->pkts[am(conf)].plen += len;
 103         conf->pkts[am(conf)].payload = xrealloc(conf->pkts[am(conf)].payload,
 104                                                 1, conf->pkts[am(conf)].plen);
 105
 106         base = conf->pkts[am(conf)].plen - 1;
 107         for (i = 0; i < len; ++i) {
 108                 conf->pkts[am(conf)].payload[base - i] = val;
 109         }
 110 }
 111
 112 static void set_rnd(size_t len)
 113 {
 114         int i, base;
 115
 116         conf->pkts[am(conf)].plen += len;
 117         conf->pkts[am(conf)].payload = xrealloc(conf->pkts[am(conf)].payload,
 118                                                 1, conf->pkts[am(conf)].plen);
 119
 120         base = conf->pkts[am(conf)].plen - 1;
 121         for (i = 0; i < len; ++i) {
 122                 conf->pkts[am(conf)].payload[base - i] = (uint8_t) mt_rand_int32();
 123         }
 124 }
 125
 126 static void set_seqinc(uint8_t start, size_t len, uint8_t stepping)
 127 {
 128         int i, base;
 129
 130         conf->pkts[am(conf)].plen += len;
 131         conf->pkts[am(conf)].payload = xrealloc(conf->pkts[am(conf)].payload,
 132                                                 1, conf->pkts[am(conf)].plen);
 133
 134         base = conf->pkts[am(conf)].plen - 1;
 135         for (i = 0; i < len; ++i) {
 136                 int off = len - 1 - i;
 137                 conf->pkts[am(conf)].payload[base - off] = start;
 138                 start += stepping;
 139         }
 140 }
 141
 142 static void set_seqdec(uint8_t start, size_t len, uint8_t stepping)
 143 {
 144         int i, base;
 145
 146         conf->pkts[am(conf)].plen += len;
 147         conf->pkts[am(conf)].payload = xrealloc(conf->pkts[am(conf)].payload,
 148                                                 1, conf->pkts[am(conf)].plen);
 149
 150         base = conf->pkts[am(conf)].plen - 1;
 151         for (i = 0; i < len; ++i) {
 152                 int off = len - 1 - i;
 153                 conf->pkts[am(conf)].payload[base - off] = start;
 154                 start -= stepping;
 155         }
 156 }
 157
 158 static void set_drnd(void)
 159 {
 160         int base, rnds;
 161         struct randomizer *new;
 162
 163         give_note_dynamic();
 164
 165         conf->pkts[am(conf)].plen++;
 166         conf->pkts[am(conf)].payload = xrealloc(conf->pkts[am(conf)].payload,
 167                                                 1, conf->pkts[am(conf)].plen);
 168
 169         base = conf->pkts[am(conf)].plen - 1;
 170         rnds = ++(conf->pkts[am(conf)].rlen);
 171
 172         conf->pkts[am(conf)].rnd = xrealloc(conf->pkts[am(conf)].rnd,
 173                                             1, rnds * sizeof(struct randomizer));
 174
 175         new = &conf->pkts[am(conf)].rnd[rnds - 1];
 176         new->val = (uint8_t) mt_rand_int32();
 177         new->off = base;
 178 }
 179
 180 static void set_dincdec(uint8_t start, uint8_t stop, uint8_t stepping, int type)
 181 {
 182         int base, cnts;
 183         struct counter *new;
 184
 185         give_note_dynamic();
 186
 187         conf->pkts[am(conf)].plen++;
 188         conf->pkts[am(conf)].payload = xrealloc(conf->pkts[am(conf)].payload,
 189                                                 1, conf->pkts[am(conf)].plen);
 190
 191         base = conf->pkts[am(conf)].plen - 1;
 192         cnts = ++(conf->pkts[am(conf)].clen);
 193
 194         conf->pkts[am(conf)].cnt = xrealloc(conf->pkts[am(conf)].cnt,
 195                                             1, cnts * sizeof(struct counter));
 196
 197         new = &conf->pkts[am(conf)].cnt[cnts - 1];
 198         new->min = start;
 199         new->max = stop;
 200         new->inc = stepping;
 201         new->val = type == TYPE_INC ? start : stop;
 202         new->off = base;
 203         new->type = type;
 204 }
 205
 206 %}
 207
 208 %union {
 209         long int number;
 210 }
 211
 212 %token K_COMMENT K_FILL K_RND K_SEQINC K_SEQDEC K_DRND K_DINC K_DDEC K_WHITE
 213 %token K_NEWL
 214
 215 %token ',' '{' '}' '(' ')' '[' ']'
 216
 217 %token number_hex number_dec number_ascii number_bin number_oct
 218
 219 %type <number> number_hex number_dec number_ascii number_bin number_oct number
 220
 221 %%
 222
 223 packets
 224         : { }
 225         | packets packet { }
 226         | packets inline_comment { }
 227         | packets white { }
 228         ;
 229
 230 inline_comment
 231         : K_COMMENT { }
 232         ;
 233
 234 packet
 235         : '{' delimiter payload delimiter '}' { realloc_packet(); }
 236         ;
 237
 238 payload
 239         : elem { }
 240         | payload elem_delimiter { }
 241         ;
 242
 243 white
 244         : white K_WHITE
 245         | white K_NEWL
 246         | K_WHITE
 247         | K_NEWL
 248         ;
 249
 250 delimiter
 251         : ','
 252         | white
 253         | ',' white
 254         ;
 255
 256 elem_delimiter
 257         : delimiter elem { }
 258         ;
 259
 260 number
 261         : number_dec { $$ = $1; }
 262         | number_hex { $$ = $1; }
 263         | number_ascii { $$ = $1; }
 264         | number_bin { $$ = $1; }
 265         | number_oct { $$ = $1; }
 266         ;
 267
 268 fill
 269         : K_FILL '(' number delimiter number ')'
 270                 { set_fill($3, $5); }
 271         ;
 272
 273 rnd
 274         : K_RND '(' number ')'
 275                 { set_rnd($3); }
 276         ;
 277
 278 seqinc
 279         : K_SEQINC '(' number delimiter number ')'
 280                 { set_seqinc($3, $5, 1); }
 281         | K_SEQINC '(' number delimiter number delimiter number ')'
 282                 { set_seqinc($3, $5, $7); }
 283         ;
 284
 285 seqdec
 286         : K_SEQDEC '(' number delimiter number ')'
 287                 { set_seqdec($3, $5, 1); }
 288         | K_SEQDEC '(' number delimiter number delimiter number ')'
 289                 { set_seqdec($3, $5, $7); }
 290         ;
 291
 292 drnd
 293         : K_DRND '(' ')'
 294                 { set_drnd(); }
 295         ;
 296
 297 dinc
 298         : K_DINC '(' number delimiter number ')'
 299                 { set_dincdec($3, $5, 1, TYPE_INC); }
 300         | K_DINC '(' number delimiter number delimiter number ')'
 301                 { set_dincdec($3, $5, $7, TYPE_INC); }
 302         ;
 303
 304 ddec
 305         : K_DDEC '(' number delimiter number ')'
 306                 { set_dincdec($3, $5, 1, TYPE_DEC); }
 307         | K_DDEC '(' number delimiter number delimiter number ')'
 308                 { set_dincdec($3, $5, $7, TYPE_DEC); }
 309         ;
 310
 311 elem
 312         : number { set_byte((uint8_t) $1); }
 313         | fill { }
 314         | rnd { }
 315         | drnd { }
 316         | seqinc { }
 317         | seqdec { }
 318         | dinc { }
 319         | ddec { }
 320         | inline_comment { }
 321         ;
 322
 323 %%
 324
 325 int compile_packets(char *file, struct pktconf *cfg, int verbose)
 326 {
 327         yyin = fopen(file, "r");
 328         if (!yyin)
 329                 panic("Cannot open file!\n");
 330         if (!cfg)
 331                 panic("No config given!\n");
 332
 333         mt_init_by_seed_time();
 334         conf = cfg;
 335         realloc_packet();
 336
 337         yyparse();
 338         /* hack ... */
 339         conf->len--;
 340
 341         if (verbose)
 342                 dump_conf(cfg);
 343         else {
 344                 int i;
 345                 size_t total_len = 0;
 346
 347                 printf("%zu packets to schedule\n", conf->len);
 348
 349                 for (i = 0; i < conf->len; ++i)
 350                         total_len += conf->pkts[i].plen;
 351                 printf("%zu bytes in total\n", total_len);
 352         }
 353
 354         fclose(yyin);
 355         return 0;
 356 }
 357
 358 void yyerror(const char *err)
 359 {
 360         panic("Syntax error at line %d: '%s'! %s!\n",
 361               yylineno, yytext, err);
 362 }