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
113
114
115
116
117
118
119
|
/*
* netsniff-ng - the packet sniffing beast
* Copyright 2009, 2010 Daniel Borkmann.
* Subject to the GPL, version 2.
*/
#ifndef RING_H
#define RING_H
/*
* "I love the smell of 10GbE in the morning. Smells like ... victory."
* - W. Richard Stevens, "Secret Teachings of the UNIX Environment"
*/
#include <stdio.h>
#include <stdint.h>
#include <linux/if_packet.h>
#include <linux/socket.h>
#include <linux/sockios.h>
#include <sys/ioctl.h>
#include <string.h>
#include <poll.h>
#include <sys/poll.h>
#include "xutils.h"
#include "built_in.h"
#include "die.h"
struct frame_map {
struct tpacket2_hdr tp_h __aligned_tpacket;
struct sockaddr_ll s_ll __align_tpacket(sizeof(struct tpacket2_hdr));
};
struct block_desc {
uint32_t version;
uint32_t offset_to_priv;
struct tpacket_hdr_v1 h1;
};
struct ring {
struct iovec *frames;
uint8_t *mm_space;
size_t mm_len;
struct sockaddr_ll s_ll;
union {
struct tpacket_req layout;
struct tpacket_req3 layout3;
};
};
static inline void next_rnd_slot(unsigned int *it, struct ring *ring)
{
*it = rand() % ring->layout.tp_frame_nr;
}
static inline unsigned int ring_size(char *ifname, unsigned int size)
{
if (size > 0)
return size;
/*
* Device bitrate in bytes times two as ring size.
* Fallback => ~ 64,00 MB
* 10 MBit => ~ 2,38 MB
* 54 MBit => ~ 12,88 MB
* 100 MBit => ~ 23,84 MB
* 300 MBit => ~ 71,52 MB
* 1.000 MBit => ~ 238,42 MB
* 10.000 MBit => ~ 2.384.18 MB
*/
size = device_bitrate(ifname);
size = (size * 1000000) / 8;
size = size * 2;
if (size == 0)
size = 1 << 26;
return round_up_cacheline(size);
}
static inline unsigned int ring_frame_size(struct ring *ring)
{
return ring->layout.tp_frame_size;
}
static inline void tpacket_hdr_clone(struct tpacket2_hdr *thdrd,
struct tpacket2_hdr *thdrs)
{
thdrd->tp_sec = thdrs->tp_sec;
thdrd->tp_nsec = thdrs->tp_nsec;
thdrd->tp_snaplen = thdrs->tp_snaplen;
thdrd->tp_len = thdrs->tp_len;
}
static inline void prepare_polling(int sock, struct pollfd *pfd)
{
memset(pfd, 0, sizeof(*pfd));
pfd->fd = sock;
pfd->revents = 0;
pfd->events = POLLIN | POLLRDNORM | POLLERR;
}
static inline void __set_sockopt_tpacket(int sock, int val)
{
int ret = setsockopt(sock, SOL_PACKET, PACKET_VERSION, &val, sizeof(val));
if (ret)
panic("Cannot set tpacketv2!\n");
}
static inline void set_sockopt_tpacket_v2(int sock)
{
__set_sockopt_tpacket(sock, TPACKET_V2);
}
static inline void set_sockopt_tpacket_v3(int sock)
{
__set_sockopt_tpacket(sock, TPACKET_V3);
}
#endif /* RING_H */
|
