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);
57 static inline void next_slot_prerd(unsigned int *it
, struct ring
*ring
)
60 atomic_cmp_swp(it
, ring
->layout
.tp_frame_nr
, 0);
61 prefetch_rd_hi(ring
->frames
[*it
].iov_base
);
64 static inline void next_slot_prewr(unsigned int *it
, struct ring
*ring
)
67 atomic_cmp_swp(it
, ring
->layout
.tp_frame_nr
, 0);
68 prefetch_wr_hi(ring
->frames
[*it
].iov_base
);
71 static inline void next_rnd_slot(unsigned int *it
, struct ring
*ring
)
73 *it
= mt_rand_int32() % ring
->layout
.tp_frame_nr
;
76 #define RING_SIZE_FALLBACK (1 << 26)
78 static inline unsigned int ring_size(char *ifname
, unsigned int size
)
84 * Device bitrate in bytes times two as ring size.
85 * Fallback => ~ 64,00 MB
86 * 10 MBit => ~ 2,38 MB
87 * 54 MBit => ~ 12,88 MB
88 * 100 MBit => ~ 23,84 MB
89 * 300 MBit => ~ 71,52 MB
90 * 1.000 MBit => ~ 238,42 MB
91 * 10.000 MBit => ~ 2.384.18 MB
93 size
= device_bitrate(ifname
);
94 size
= (size
* 1000000) / 8;
97 size
= RING_SIZE_FALLBACK
;
98 return round_up_cacheline(size
);
106 static inline unsigned int ring_frame_size(struct ring
*ring
)
108 return ring
->layout
.tp_frame_size
;
111 static inline void tpacket_hdr_clone(struct tpacket2_hdr
*thdrd
,
112 struct tpacket2_hdr
*thdrs
)
114 thdrd
->tp_sec
= thdrs
->tp_sec
;
115 thdrd
->tp_nsec
= thdrs
->tp_nsec
;
116 thdrd
->tp_snaplen
= thdrs
->tp_snaplen
;
117 thdrd
->tp_len
= thdrs
->tp_len
;
121 # define POLLRDNORM 0x0040
124 # define POLLWRNORM 0x0100
127 # define POLLRDHUP 0x2000
130 #define POLL_NEXT_PKT 0
131 #define POLL_MOVE_OUT 1
133 static inline void prepare_polling(int sock
, struct pollfd
*pfd
)
135 memset(pfd
, 0, sizeof(*pfd
));
138 pfd
->events
= POLLIN
| POLLRDNORM
| POLLERR
;
141 static inline void set_sockopt_fanout(int sock
, unsigned int fanout_id
,
142 unsigned int fanout_type
)
144 unsigned int fanout_arg
= (fanout_id
| (fanout_type
<< 16));
145 int ret
= setsockopt(sock
, SOL_PACKET
, PACKET_FANOUT
, &fanout_arg
,
148 panic("No packet fanout support!\n");
151 static inline void set_sockopt_tpacket(int sock
)
153 int ret
, val
= TPACKET_V2
;
155 ret
= setsockopt(sock
, SOL_PACKET
, PACKET_VERSION
, &val
, sizeof(val
));
157 panic("Cannot set tpacketv2!\n");
160 #if defined(__WITH_HARDWARE_TIMESTAMPING)
161 # include <linux/net_tstamp.h>
163 static inline void set_sockopt_hwtimestamp(int sock
, const char *dev
)
166 struct hwtstamp_config hwconfig
;
169 memset(&hwconfig
, 0, sizeof(hwconfig
));
170 hwconfig
.tx_type
= HWTSTAMP_TX_ON
;
171 hwconfig
.rx_filter
= HWTSTAMP_FILTER_ALL
;
173 memset(&ifr
, 0, sizeof(ifr
));
174 strlcpy(ifr
.ifr_name
, dev
, sizeof(ifr
.ifr_name
));
175 ifr
.ifr_data
= &hwconfig
;
177 ret
= ioctl(sock
, SIOCSHWTSTAMP
, &ifr
);
181 timesource
= SOF_TIMESTAMPING_RAW_HARDWARE
;
183 ret
= setsockopt(sock
, SOL_PACKET
, PACKET_TIMESTAMP
, ×ource
,
186 panic("Cannot set timestamping!\n");
189 static inline void set_sockopt_hwtimestamp(int sock
, const char *dev
)
193 #endif /* defined(__WITH_HARDWARE_TIMESTAMPING) */