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