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         unsigned int  size;
  21
  22         struct protocol *dissector;
  23         uint32_t link_type;
  24         struct sockaddr_ll *sll;
  25 };
  26
  27 static inline struct pkt_buff *pkt_alloc(uint8_t *packet, unsigned int len)
  28 {
  29         struct pkt_buff *pkt = xmalloc(sizeof(*pkt));
  30
  31         pkt->head = packet;
  32         pkt->data = packet;
  33         pkt->tail = packet + len;
  34         pkt->size = len;
  35         pkt->dissector = NULL;
  36
  37         return pkt;
  38 }
  39
  40 static inline void pkt_free(struct pkt_buff *pkt)
  41 {
  42         xfree(pkt);
  43 }
  44
  45 static inline unsigned int pkt_len(struct pkt_buff *pkt)
  46 {
  47         bug_on(!pkt || pkt->data > pkt->tail);
  48
  49         return pkt->tail - pkt->data;
  50 }
  51
  52 static inline uint8_t *pkt_pull(struct pkt_buff *pkt, unsigned int len)
  53 {
  54         uint8_t *data = NULL;
  55
  56         bug_on(!pkt || pkt->head > pkt->data || pkt->data > pkt->tail);
  57
  58         if (len <= pkt_len(pkt)) {
  59                 data = pkt->data;
  60                 pkt->data += len;
  61         }
  62
  63         bug_on(!pkt || pkt->head > pkt->data || pkt->data > pkt->tail);
  64
  65         return data;
  66 }
  67
  68 static inline uint8_t *pkt_peek(struct pkt_buff *pkt)
  69 {
  70         bug_on(!pkt || pkt->head > pkt->data || pkt->data > pkt->tail);
  71
  72         return pkt->data;
  73 }
  74
  75 static inline unsigned int pkt_trim(struct pkt_buff *pkt, unsigned int len)
  76 {
  77         unsigned int ret = 0;
  78
  79         bug_on(!pkt || pkt->head > pkt->data || pkt->data > pkt->tail);
  80
  81         if (len <= pkt_len(pkt))
  82                 ret = len;
  83
  84         pkt->tail -= ret;
  85         bug_on(!pkt || pkt->head > pkt->data || pkt->data > pkt->tail);
  86
  87         return ret;
  88 }
  89
  90 static inline uint8_t *pkt_pull_tail(struct pkt_buff *pkt, unsigned int len)
  91 {
  92         uint8_t *tail = NULL;
  93
  94         bug_on(!pkt || pkt->head > pkt->data || pkt->data > pkt->tail);
  95
  96         if (len <= pkt_len(pkt)) {
  97                 tail = pkt->tail;
  98                 pkt->tail -= len;
  99         }
 100
 101         return tail;
 102 }
 103
 104 static inline void pkt_set_dissector(struct pkt_buff *pkt, struct hash_table *table,
 105                                      unsigned int key)
 106 {
 107         bug_on(!pkt || !table);
 108
 109         pkt->dissector = lookup_hash(key, table);
 110         while (pkt->dissector && key != pkt->dissector->key)
 111                 pkt->dissector = pkt->dissector->next;
 112 }
 113
 114 #endif /* PKT_BUFF_H */