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_struct.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 (pkt_len(pkt) && pkt->data + len <= pkt->tail) {
  57                 data = pkt->data;
  58                 pkt->data += len;
  59         }
  60
  61         return data;
  62 }
  63
  64 static inline uint8_t *pkt_peek(struct pkt_buff *pkt)
  65 {
  66         bug_on(!pkt || pkt->head > pkt->data || pkt->data > pkt->tail);
  67
  68         return pkt->data;
  69 }
  70
  71 static inline void pkt_trim(struct pkt_buff *pkt, unsigned int len)
  72 {
  73         bug_on(!pkt || pkt->head > pkt->data || pkt->data > pkt->tail);
  74
  75         if (pkt_len(pkt) && pkt->tail - len >= pkt->data)
  76                 pkt->tail -= len;
  77 }
  78
  79 static inline uint8_t *pkt_pull_tail(struct pkt_buff *pkt, unsigned int len)
  80 {
  81         uint8_t *tail = NULL;
  82
  83         bug_on(!pkt || pkt->head > pkt->data || pkt->data > pkt->tail);
  84
  85         if (pkt_len(pkt) && pkt->tail - len >= pkt->data) {
  86                 tail = pkt->tail;
  87                 pkt->tail -= len;
  88         }
  89
  90         return tail;
  91 }
  92
  93 static inline void pkt_set_proto(struct pkt_buff *pkt, struct hash_table *table,
  94                                  unsigned int key)
  95 {
  96         bug_on(!pkt || !table);
  97
  98         pkt->proto = (struct protocol *) lookup_hash(key, table);
  99         while (pkt->proto && key != pkt->proto->key)
 100                 pkt->proto = pkt->proto->next;
 101 }
 102
 103 #endif /* PKT_BUFF_H */