pkt_buff.h

   1 /*
   2  * netsniff-ng - the packet sniffing beast
   3  * Copyright (C) 2012 Christoph Jaeger <christoph@netsniff-ng.org>
   4  * Subject to the GPL, version 2.
   5  */
   6
   7 #ifndef PKT_BUFF_H
   8 #define PKT_BUFF_H
   9
  10 #include "hash.h"
  11 #include "built_in.h"
  12 #include "proto.h"
  13 #include "xmalloc.h"
  14
  15 struct pkt_buff {
  16         /* invariant: head <= data <= tail */
  17         uint8_t *head;
  18         uint8_t *data;
  19         uint8_t *tail;
  20
  21         struct protocol *dissector;
  22         uint32_t link_type;
  23         struct sockaddr_ll *sll;
  24 };
  25
  26 static inline struct pkt_buff *pkt_alloc(uint8_t *packet, unsigned int len)
  27 {
  28         struct pkt_buff *pkt = xmalloc(sizeof(*pkt));
  29
  30         pkt->head = packet;
  31         pkt->data = packet;
  32         pkt->tail = packet + len;
  33         pkt->dissector = NULL;
  34
  35         return pkt;
  36 }
  37
  38 static inline void pkt_free(struct pkt_buff *pkt)
  39 {
  40         xfree(pkt);
  41 }
  42
  43 static inline unsigned int pkt_len(struct pkt_buff *pkt)
  44 {
  45         bug_on(!pkt || pkt->data > pkt->tail);
  46
  47         return pkt->tail - pkt->data;
  48 }
  49
  50 static inline uint8_t *pkt_pull(struct pkt_buff *pkt, unsigned int len)
  51 {
  52         uint8_t *data = NULL;
  53
  54         bug_on(!pkt || pkt->head > pkt->data || pkt->data > pkt->tail);
  55
  56         if (len <= pkt_len(pkt)) {
  57                 data = pkt->data;
  58                 pkt->data += len;
  59         }
  60
  61         bug_on(!pkt || pkt->head > pkt->data || pkt->data > pkt->tail);
  62
  63         return data;
  64 }
  65
  66 static inline uint8_t *pkt_peek(struct pkt_buff *pkt)
  67 {
  68         bug_on(!pkt || pkt->head > pkt->data || pkt->data > pkt->tail);
  69
  70         return pkt->data;
  71 }
  72
  73 static inline unsigned int pkt_trim(struct pkt_buff *pkt, unsigned int len)
  74 {
  75         unsigned int ret = 0;
  76
  77         bug_on(!pkt || pkt->head > pkt->data || pkt->data > pkt->tail);
  78
  79         if (len <= pkt_len(pkt))
  80                 ret = len;
  81
  82         pkt->tail -= ret;
  83         bug_on(!pkt || pkt->head > pkt->data || pkt->data > pkt->tail);
  84
  85         return ret;
  86 }
  87
  88 static inline uint8_t *pkt_pull_tail(struct pkt_buff *pkt, unsigned int len)
  89 {
  90         uint8_t *tail = NULL;
  91
  92         bug_on(!pkt || pkt->head > pkt->data || pkt->data > pkt->tail);
  93
  94         if (len <= pkt_len(pkt)) {
  95                 tail = pkt->tail;
  96                 pkt->tail -= len;
  97         }
  98
  99         return tail;
 100 }
 101
 102 static inline void pkt_set_dissector(struct pkt_buff *pkt, struct hash_table *table,
 103                                      unsigned int key)
 104 {
 105         bug_on(!pkt || !table);
 106
 107         pkt->dissector = lookup_hash(key, table);
 108         while (pkt->dissector && key != pkt->dissector->key)
 109                 pkt->dissector = pkt->dissector->next;
 110 }
 111
 112 #endif /* PKT_BUFF_H */