trafgen_proto.h

   1 #ifndef TRAFGEN_PROTO_H
   2 #define TRAFGEN_PROTO_H
   3
   4 #include <stddef.h>
   5 #include <stdint.h>
   6 #include <stdbool.h>
   7
   8 enum proto_id {
   9         PROTO_NONE = 0,
  10         PROTO_ETH,
  11         PROTO_VLAN,
  12         PROTO_ARP,
  13         PROTO_MPLS,
  14         PROTO_IP4,
  15         PROTO_ICMP4,
  16         PROTO_IP6,
  17         PROTO_ICMP6,
  18         PROTO_UDP,
  19         PROTO_TCP,
  20         __PROTO_MAX,
  21 };
  22
  23 enum proto_layer {
  24         PROTO_L0, /* invalid layer */
  25         PROTO_L2,
  26         PROTO_L3,
  27         PROTO_L4,
  28 };
  29
  30 struct proto_field;
  31 struct proto_hdr;
  32
  33 struct proto_ops {
  34         enum proto_id id;
  35         enum proto_layer layer;
  36
  37         void (*header_init)(struct proto_hdr *hdr);
  38         void (*header_finish)(struct proto_hdr *hdr);
  39         void (*field_changed)(struct proto_field *field);
  40         void (*packet_finish)(struct proto_hdr *hdr);
  41         void (*packet_update)(struct proto_hdr *hdr);
  42         void (*set_next_proto)(struct proto_hdr *hdr, enum proto_id pid);
  43 };
  44
  45 struct proto_hdr {
  46         const struct proto_ops *ops;
  47         uint16_t pkt_offset;
  48         uint32_t pkt_id;
  49         struct proto_field *fields;
  50         size_t fields_count;
  51         size_t len;
  52 };
  53
  54 struct proto_field_func {
  55         void (*update_field)(struct proto_field *field);
  56 };
  57
  58 struct proto_field {
  59         uint32_t id;
  60         size_t len;
  61         uint32_t shift;
  62         uint32_t mask;
  63         /* might be negative (e.g. VLAN TPID field) */
  64         int16_t offset;
  65
  66         struct proto_field_func func;
  67         bool is_set;
  68         uint16_t pkt_offset;
  69         struct proto_hdr *hdr;
  70 };
  71
  72 extern void protos_init(const char *dev);
  73 extern void proto_ops_register(const struct proto_ops *ops);
  74
  75 extern struct proto_hdr *proto_header_push(enum proto_id pid);
  76 extern void proto_header_finish(struct proto_hdr *hdr);
  77 extern void proto_packet_finish(void);
  78 extern void proto_packet_update(uint32_t idx);
  79
  80 extern struct proto_hdr *proto_lower_default_add(struct proto_hdr *hdr,
  81                                                  enum proto_id pid);
  82
  83 extern struct proto_hdr *proto_lower_header(struct proto_hdr *hdr);
  84 extern uint8_t *proto_header_ptr(struct proto_hdr *hdr);
  85
  86 extern void proto_header_fields_add(struct proto_hdr *hdr,
  87                                     const struct proto_field *fields,
  88                                     size_t count);
  89
  90 extern bool proto_field_is_set(struct proto_hdr *hdr, uint32_t fid);
  91 extern void proto_field_set_bytes(struct proto_hdr *hdr, uint32_t fid,
  92                                   const uint8_t *bytes);
  93 extern void proto_field_set_u8(struct proto_hdr *hdr, uint32_t fid, uint8_t val);
  94 extern uint8_t proto_field_get_u8(struct proto_hdr *hdr, uint32_t fid);
  95 extern void proto_field_set_u16(struct proto_hdr *hdr, uint32_t fid, uint16_t val);
  96 extern uint16_t proto_field_get_u16(struct proto_hdr *hdr, uint32_t fid);
  97 extern void proto_field_set_u32(struct proto_hdr *hdr, uint32_t fid, uint32_t val);
  98 extern uint32_t proto_field_get_u32(struct proto_hdr *hdr, uint32_t fid);
  99
 100 extern void proto_field_set_default_bytes(struct proto_hdr *hdr, uint32_t fid,
 101                                           const uint8_t *bytes);
 102 extern void proto_field_set_default_u8(struct proto_hdr *hdr, uint32_t fid,
 103                                        uint8_t val);
 104 extern void proto_field_set_default_u16(struct proto_hdr *hdr, uint32_t fid,
 105                                         uint16_t val);
 106 extern void proto_field_set_default_u32(struct proto_hdr *hdr, uint32_t fid,
 107                                         uint32_t val);
 108
 109 extern void proto_field_set_be16(struct proto_hdr *hdr, uint32_t fid, uint16_t val);
 110 extern void proto_field_set_be32(struct proto_hdr *hdr, uint32_t fid, uint32_t val);
 111
 112 extern void proto_field_set_default_be16(struct proto_hdr *hdr, uint32_t fid,
 113                                          uint16_t val);
 114 extern void proto_field_set_default_be32(struct proto_hdr *hdr, uint32_t fid,
 115                                          uint32_t val);
 116
 117 extern void proto_field_set_dev_mac(struct proto_hdr *hdr, uint32_t fid);
 118 extern void proto_field_set_default_dev_mac(struct proto_hdr *hdr, uint32_t fid);
 119
 120 extern void proto_field_set_dev_ipv4(struct proto_hdr *hdr, uint32_t fid);
 121 extern void proto_field_set_default_dev_ipv4(struct proto_hdr *hdr, uint32_t fid);
 122
 123 extern void proto_field_set_dev_ipv6(struct proto_hdr *hdr, uint32_t fid);
 124 extern void proto_field_set_default_dev_ipv6(struct proto_hdr *hdr, uint32_t fid);
 125
 126 extern void proto_field_dyn_apply(struct proto_field *field);
 127
 128 extern void proto_field_func_add(struct proto_hdr *hdr, uint32_t fid,
 129                                  struct proto_field_func *func);
 130
 131 #endif /* TRAFGEN_PROTO_H */