2 * netsniff-ng - the packet sniffing beast
3 * By Daniel Borkmann <daniel@netsniff-ng.org>
4 * Copyright 2009, 2010 Daniel Borkmann.
5 * Subject to the GPL, version 2.
12 * "I love the smell of 10GbE in the morning. Smells like ... victory."
13 * - W. Richard Stevens, "Secret Teachings of the UNIX Environment"
18 #include <linux/if_packet.h>
19 #include <linux/socket.h>
20 #include <linux/sockios.h>
21 #include <sys/ioctl.h>
32 # define PACKET_FANOUT 18
33 # define PACKET_FANOUT_POLICY_HASH 0
34 # define PACKET_FANOUT_POLICY_LB 1
35 # define PACKET_FANOUT_POLICY_DEFAULT PACKET_FANOUT_HASH
39 struct tpacket2_hdr tp_h __aligned_tpacket
;
40 struct sockaddr_ll s_ll
__align_tpacket(sizeof(struct tpacket2_hdr
));
44 struct iovec
*frames __cacheline_aligned
;
47 struct tpacket_req layout
;
48 struct sockaddr_ll s_ll
;
51 static inline void next_slot(unsigned int *it
, struct ring
*ring
)
54 atomic_cmp_swp(it
, ring
->layout
.tp_frame_nr
, 0);
57 static inline void next_rnd_slot(unsigned int *it
, struct ring
*ring
)
59 *it
= mt_rand_int32() % ring
->layout
.tp_frame_nr
;
62 #define RING_SIZE_FALLBACK (1 << 26)
64 static inline unsigned int ring_size(char *ifname
, unsigned int size
)
70 * Device bitrate in bytes times two as ring size.
71 * Fallback => ~ 64,00 MB
72 * 10 MBit => ~ 2,38 MB
73 * 54 MBit => ~ 12,88 MB
74 * 100 MBit => ~ 23,84 MB
75 * 300 MBit => ~ 71,52 MB
76 * 1.000 MBit => ~ 238,42 MB
77 * 10.000 MBit => ~ 2.384.18 MB
79 size
= device_bitrate(ifname
);
80 size
= (size
* 1000000) / 8;
83 size
= RING_SIZE_FALLBACK
;
85 return round_up_cacheline(size
);
93 static inline unsigned int ring_frame_size(struct ring
*ring
)
95 return ring
->layout
.tp_frame_size
;
98 static inline void tpacket_hdr_clone(struct tpacket2_hdr
*thdrd
,
99 struct tpacket2_hdr
*thdrs
)
101 thdrd
->tp_sec
= thdrs
->tp_sec
;
102 thdrd
->tp_nsec
= thdrs
->tp_nsec
;
103 thdrd
->tp_snaplen
= thdrs
->tp_snaplen
;
104 thdrd
->tp_len
= thdrs
->tp_len
;
108 # define POLLRDNORM 0x0040
111 # define POLLWRNORM 0x0100
114 # define POLLRDHUP 0x2000
117 #define POLL_NEXT_PKT 0
118 #define POLL_MOVE_OUT 1
120 static inline void prepare_polling(int sock
, struct pollfd
*pfd
)
122 memset(pfd
, 0, sizeof(*pfd
));
125 pfd
->events
= POLLIN
| POLLRDNORM
| POLLERR
;
128 static inline void set_sockopt_fanout(int sock
, unsigned int fanout_id
,
129 unsigned int fanout_type
)
131 unsigned int fanout_arg
= (fanout_id
| (fanout_type
<< 16));
132 int ret
= setsockopt(sock
, SOL_PACKET
, PACKET_FANOUT
, &fanout_arg
,
135 panic("No packet fanout support!\n");
138 static inline void set_sockopt_tpacket(int sock
)
140 int ret
, val
= TPACKET_V2
;
142 ret
= setsockopt(sock
, SOL_PACKET
, PACKET_VERSION
, &val
, sizeof(val
));
144 panic("Cannot set tpacketv2!\n");
147 #if defined(__WITH_HARDWARE_TIMESTAMPING)
148 # include <linux/net_tstamp.h>
150 static inline void set_sockopt_hwtimestamp(int sock
, const char *dev
)
153 struct hwtstamp_config hwconfig
;
156 memset(&hwconfig
, 0, sizeof(hwconfig
));
157 hwconfig
.tx_type
= HWTSTAMP_TX_ON
;
158 hwconfig
.rx_filter
= HWTSTAMP_FILTER_ALL
;
160 memset(&ifr
, 0, sizeof(ifr
));
161 strlcpy(ifr
.ifr_name
, dev
, sizeof(ifr
.ifr_name
));
162 ifr
.ifr_data
= &hwconfig
;
164 ret
= ioctl(sock
, SIOCSHWTSTAMP
, &ifr
);
168 timesource
= SOF_TIMESTAMPING_RAW_HARDWARE
;
170 ret
= setsockopt(sock
, SOL_PACKET
, PACKET_TIMESTAMP
, ×ource
,
173 panic("Cannot set timestamping!\n");
176 static inline void set_sockopt_hwtimestamp(int sock
, const char *dev
)
180 #endif /* defined(__WITH_HARDWARE_TIMESTAMPING) */