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         uint32_t id;
  41         uint8_t *payload;
  42         size_t len;
  43         struct proto_hdr *headers[PROTO_MAX_LAYERS];
  44         size_t headers_count;
  45         struct timespec tstamp;
  46         bool is_created;
  47 };
  48
  49 struct packet_dyn {
  50         struct counter *cnt;
  51         size_t clen;
  52         struct randomizer *rnd;
  53         size_t rlen;
  54         struct csum16 *csum;
  55         size_t slen;
  56         struct proto_field **fields;
  57         size_t flen;
  58 };
  59
  60 static inline bool packet_dyn_has_elems(struct packet_dyn *p)
  61 {
  62         return (p->clen || p->rlen || p->slen);
  63 }
  64
  65 static inline bool packet_dyn_has_only_csums(struct packet_dyn *p)
  66 {
  67         return (p->clen == 0 && p->rlen == 0 && p->slen);
  68 }
  69
  70 static inline bool packet_dyn_has_fields(struct packet_dyn *p)
  71 {
  72         return p->flen;
  73 }
  74
  75 extern void compile_packets_str(char *str, bool verbose, unsigned int cpu);
  76 extern void compile_packets(char *file, bool verbose, unsigned int cpu,
  77                             bool invoke_cpp, char *const cpp_argv[]);
  78 extern void cleanup_packets(void);
  79
  80 extern void set_fill(uint8_t val, size_t len);
  81
  82 extern struct packet *current_packet(void);
  83 extern uint32_t current_packet_id(void);
  84 extern struct packet *packet_get(uint32_t id);
  85 extern struct packet *realloc_packet(void);
  86
  87 #endif /* TRAFGEN_CONF */