summaryrefslogtreecommitdiff
path: root/pkt_buff.h
blob: f86c01175a0ab3a2d3eac8730a87fa3e9909e9c4 (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 = (struct protocol *) lookup_hash(key, table);
	while (pkt->proto && key != pkt->proto->key)
		pkt->proto = pkt->proto->next;
}

#endif /* PKT_BUFF_H */