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
;
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);
58 static inline void next_slot_prerd(unsigned int *it
, struct ring
*ring
)
61 atomic_cmp_swp(it
, ring
->layout
.tp_frame_nr
, 0);
62 prefetch_rd_hi(ring
->frames
[*it
].iov_base
);
65 static inline void next_slot_prewr(unsigned int *it
, struct ring
*ring
)
68 atomic_cmp_swp(it
, ring
->layout
.tp_frame_nr
, 0);
69 prefetch_wr_hi(ring
->frames
[*it
].iov_base
);
72 # define next_slot_prerd next_slot
73 # define next_slot_prewr next_slot
76 static inline void next_rnd_slot(unsigned int *it
, struct ring
*ring
)
78 *it
= mt_rand_int32() % ring
->layout
.tp_frame_nr
;
81 #define RING_SIZE_FALLBACK (1 << 26)
83 static inline unsigned int ring_size(char *ifname
, unsigned int size
)
89 * Device bitrate in bytes times two as ring size.
90 * Fallback => ~ 64,00 MB
91 * 10 MBit => ~ 2,38 MB
92 * 54 MBit => ~ 12,88 MB
93 * 100 MBit => ~ 23,84 MB
94 * 300 MBit => ~ 71,52 MB
95 * 1.000 MBit => ~ 238,42 MB
96 * 10.000 MBit => ~ 2.384.18 MB
98 size
= device_bitrate(ifname
);
99 size
= (size
* 1000000) / 8;
102 size
= RING_SIZE_FALLBACK
;
104 return round_up_cacheline(size
);
112 static inline unsigned int ring_frame_size(struct ring
*ring
)
114 return ring
->layout
.tp_frame_size
;
117 static inline void tpacket_hdr_clone(struct tpacket2_hdr
*thdrd
,
118 struct tpacket2_hdr
*thdrs
)
120 thdrd
->tp_sec
= thdrs
->tp_sec
;
121 thdrd
->tp_nsec
= thdrs
->tp_nsec
;
122 thdrd
->tp_snaplen
= thdrs
->tp_snaplen
;
123 thdrd
->tp_len
= thdrs
->tp_len
;
127 # define POLLRDNORM 0x0040
130 # define POLLWRNORM 0x0100
133 # define POLLRDHUP 0x2000
136 #define POLL_NEXT_PKT 0
137 #define POLL_MOVE_OUT 1
139 static inline void prepare_polling(int sock
, struct pollfd
*pfd
)
141 memset(pfd
, 0, sizeof(*pfd
));
144 pfd
->events
= POLLIN
| POLLRDNORM
| POLLERR
;
147 static inline void set_sockopt_fanout(int sock
, unsigned int fanout_id
,
148 unsigned int fanout_type
)
150 unsigned int fanout_arg
= (fanout_id
| (fanout_type
<< 16));
151 int ret
= setsockopt(sock
, SOL_PACKET
, PACKET_FANOUT
, &fanout_arg
,
154 panic("No packet fanout support!\n");
157 static inline void set_sockopt_tpacket(int sock
)
159 int ret
, val
= TPACKET_V2
;
161 ret
= setsockopt(sock
, SOL_PACKET
, PACKET_VERSION
, &val
, sizeof(val
));
163 panic("Cannot set tpacketv2!\n");
166 #if defined(__WITH_HARDWARE_TIMESTAMPING)
167 # include <linux/net_tstamp.h>
169 static inline void set_sockopt_hwtimestamp(int sock
, const char *dev
)
172 struct hwtstamp_config hwconfig
;
175 memset(&hwconfig
, 0, sizeof(hwconfig
));
176 hwconfig
.tx_type
= HWTSTAMP_TX_ON
;
177 hwconfig
.rx_filter
= HWTSTAMP_FILTER_ALL
;
179 memset(&ifr
, 0, sizeof(ifr
));
180 strlcpy(ifr
.ifr_name
, dev
, sizeof(ifr
.ifr_name
));
181 ifr
.ifr_data
= &hwconfig
;
183 ret
= ioctl(sock
, SIOCSHWTSTAMP
, &ifr
);
187 timesource
= SOF_TIMESTAMPING_RAW_HARDWARE
;
189 ret
= setsockopt(sock
, SOL_PACKET
, PACKET_TIMESTAMP
, ×ource
,
192 panic("Cannot set timestamping!\n");
195 static inline void set_sockopt_hwtimestamp(int sock
, const char *dev
)
199 #endif /* defined(__WITH_HARDWARE_TIMESTAMPING) */