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