trafgen_conf.h

   1 #ifndef TRAFGEN_CONF
   2 #define TRAFGEN_CONF
   3
   4 #include <stdint.h>
   5 #include <stdio.h>
   6 #include <sys/time.h>
   7 #include <sys/types.h>
   8
   9 #include "trafgen_proto.h"
  10
  11 #define PROTO_MAX_LAYERS        16
  12
  13 #define TYPE_INC        0
  14 #define TYPE_DEC        1
  15
  16 enum csum {
  17         CSUM_IP,
  18         CSUM_UDP,
  19         CSUM_TCP,
  20         CSUM_UDP6,
  21         CSUM_TCP6,
  22 };
  23
  24 struct counter {
  25         int type;
  26         uint8_t min, max, inc, val;
  27         off_t off;
  28 };
  29
  30 struct randomizer {
  31         off_t off;
  32 };
  33
  34 struct csum16 {
  35         off_t off, from, to;
  36         enum csum which;
  37 };
  38
  39 struct packet {
  40         uint8_t *payload;
  41         size_t len;
  42         struct proto_hdr *headers[PROTO_MAX_LAYERS];
  43         size_t headers_count;
  44         struct timespec tstamp;
  45 };
  46
  47 struct packet_dyn {
  48         struct counter *cnt;
  49         size_t clen;
  50         struct randomizer *rnd;
  51         size_t rlen;
  52         struct csum16 *csum;
  53         size_t slen;
  54         struct proto_field **fields;
  55         size_t flen;
  56 };
  57
  58 static inline bool packet_dyn_has_elems(struct packet_dyn *p)
  59 {
  60         return (p->clen || p->rlen || p->slen);
  61 }
  62
  63 static inline bool packet_dyn_has_only_csums(struct packet_dyn *p)
  64 {
  65         return (p->clen == 0 && p->rlen == 0 && p->slen);
  66 }
  67
  68 static inline bool packet_dyn_has_fields(struct packet_dyn *p)
  69 {
  70         return p->flen;
  71 }
  72
  73 extern void compile_packets_str(char *str, bool verbose, unsigned int cpu);
  74 extern void compile_packets(char *file, bool verbose, unsigned int cpu,
  75                             bool invoke_cpp, char *const cpp_argv[]);
  76 extern void cleanup_packets(void);
  77
  78 extern void set_fill(uint8_t val, size_t len);
  79
  80 extern struct packet *current_packet(void);
  81 extern uint32_t current_packet_id(void);
  82 extern struct packet *packet_get(uint32_t id);
  83 extern void realloc_packet(void);
  84
  85 #endif /* TRAFGEN_CONF */