src/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 *proto;
  23 };
  24
  25 static inline struct pkt_buff *pkt_alloc(uint8_t *packet, unsigned int len)
  26 {
  27         struct pkt_buff *pkt = xmalloc(sizeof(*pkt));
  28
  29         pkt->head = packet;
  30         pkt->data = packet;
  31         pkt->tail = packet + len;
  32         pkt->size = len;
  33         pkt->proto = 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_proto(struct pkt_buff *pkt, struct hash_table *table,
 103                                  unsigned int key)
 104 {
 105         bug_on(!pkt || !table);
 106
 107         pkt->proto = (struct protocol *) lookup_hash(key, table);
 108         while (pkt->proto && key != pkt->proto->key)
 109                 pkt->proto = pkt->proto->next;
 110 }
 111
 112 #endif /* PKT_BUFF_H */