]> The Tcpdump Group git mirrors - libpcap/blob - pcap-dpdk.c
fix typo
[libpcap] / pcap-dpdk.c
1 /*
2 * Copyright (C) 2018 jingle YANG. All rights reserved.
3 *
4 * Redistribution and use in source and binary forms, with or without
5 * modification, are permitted provided that the following conditions
6 * are met:
7 *
8 * 1. Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 * notice, this list of conditions and the following disclaimer in the
12 * documentation and/or other materials provided with the distribution.
13 *
14 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS''AND
15 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24 * SUCH DAMAGE.
25 */
26
27 /*
28 Date: Dec 16, 2018
29
30 Description:
31 1. Pcap-dpdk provides libpcap the ability to use DPDK with the device name as dpdk:{portid}, such as dpdk:0.
32 2. DPDK is a set of libraries and drivers for fast packet processing. (https://www.dpdk.org/)
33 3. The testprogs/capturetest provides 6.4Gbps/800,000 pps on Intel 10-Gigabit X540-AT2 with DPDK 18.11.
34
35 Limitations:
36 1. By default DPDK support is no, unless you explicitly set --enable-dpdk with ./configure or -DDISABLE_DPDK=OFF with cmake.
37 2. Only support link libdpdk.so dynamicly, because the libdpdk.a will not work correctly.
38 3. Only support read operation, and packet injection has not been supported yet.
39
40 Usage:
41 1. compile DPDK as shared library and install.(https://github.com/DPDK/dpdk.git)
42
43 You shall modify the file $RTE_SDK/$RTE_TARGET/.config and set:
44 CONFIG_RTE_BUILD_SHARED_LIB=y
45 By the following command:
46 sed -i 's/CONFIG_RTE_BUILD_SHARED_LIB=n/CONFIG_RTE_BUILD_SHARED_LIB=y/' $RTE_SDK/$RTE_TARGET/.config
47
48 2. launch l2fwd that is one of DPDK examples correctly, and get device information.
49
50 You shall learn how to bind nic with DPDK-compatible driver by $RTE_SDK/usertools/dpdk-devbind.py, such as igb_uio.
51 And enable hugepages by dpdk-setup.sh
52
53 Then launch the l2fwd with dynamic dirver support. For example:
54 $RTE_SDK/examples/l2fwd/$RTE_TARGET/l2fwd -dlibrte_pmd_e1000.so -dlibrte_pmd_ixgbe.so -dlibrte_mempool_ring.so -- -p 0x1
55
56 3. compile libpcap with dpdk options.
57
58 In order to find inlucde and lib automatically, you shall export DPDK envionment variable which are used for compiling DPDK.
59
60 export RTE_SDK={your DPDK base directory}
61 export RTE_TARGET={your target name}
62
63 3.1 with configure
64
65 ./configure --enable-dpdk --with-dpdk-includes=$RTE_SDK/$RTE_TARGET/include --with-dpdk-libraries=$RTE_SDK/$RTE_TARGET/lib && make -s all && make -s testprogs && make install
66
67 3.2 with cmake
68
69 mkdir -p build && cd build && cmake -DDISABLE_DPDK=OFF -DDPDK_INC_DIR=$RTE_SDK/$RTE_TARGET/include -DDPDK_LIB_DIR=$RTE_SDK/$RTE_TARGET/lib" ../ && make -s all && make -s testprogs && make install
70
71 4. link your own program with libpcap, and use DPDK with the device name as dpdk:{portid}, such as dpdk:0.
72 And you shall set DPDK configure options by environment variable DPDK_CFG
73 For example, the testprogs/capturetest could be lanched by:
74
75 env DPDK_CFG="--log-level=debug -l0 -dlibrte_pmd_e1000.so -dlibrte_pmd_ixgbe.so -dlibrte_mempool_ring.so" ./capturetest -i dpdk:0
76 */
77
78 #ifdef HAVE_CONFIG_H
79 #include <config.h>
80 #endif
81
82 #include <ctype.h>
83 #include <errno.h>
84 #include <netdb.h>
85 #include <stdio.h>
86 #include <stdlib.h>
87 #include <string.h>
88 #include <unistd.h>
89 #include <time.h>
90
91 #include <sys/time.h>
92
93 //header for calling dpdk
94 #include <rte_common.h>
95 #include <rte_log.h>
96 #include <rte_malloc.h>
97 #include <rte_memory.h>
98 #include <rte_eal.h>
99 #include <rte_launch.h>
100 #include <rte_atomic.h>
101 #include <rte_cycles.h>
102 #include <rte_lcore.h>
103 #include <rte_per_lcore.h>
104 #include <rte_branch_prediction.h>
105 #include <rte_interrupts.h>
106 #include <rte_random.h>
107 #include <rte_debug.h>
108 #include <rte_ether.h>
109 #include <rte_ethdev.h>
110 #include <rte_mempool.h>
111 #include <rte_mbuf.h>
112 #include <rte_bus.h>
113
114 #include "pcap-int.h"
115 #include "pcap-dpdk.h"
116
117 #define DPDK_DEF_LOG_LEV RTE_LOG_ERR
118 static int is_dpdk_pre_inited=0;
119 #define DPDK_LIB_NAME "libpcap_dpdk"
120 #define DPDK_DESC "Data Plane Development Kit (DPDK) Interface"
121 #define DPDK_ERR_PERM_MSG "permission denied, DPDK needs root permission"
122 #define DPDK_ARGC_MAX 64
123 #define DPDK_CFG_MAX_LEN 1024
124 #define DPDK_DEV_NAME_MAX 32
125 #define DPDK_DEV_DESC_MAX 512
126 #define DPDK_CFG_ENV_NAME "DPDK_CFG"
127 static char dpdk_cfg_buf[DPDK_CFG_MAX_LEN];
128 #define DPDK_MAC_ADDR_SIZE 32
129 #define DPDK_DEF_MAC_ADDR "00:00:00:00:00:00"
130 #define DPDK_PCI_ADDR_SIZE 16
131 #define DPDK_DEF_CFG "--log-level=error -l0 -dlibrte_pmd_e1000.so -dlibrte_pmd_ixgbe.so -dlibrte_mempool_ring.so"
132 #define DPDK_PREFIX "dpdk:"
133 #define DPDK_PORTID_MAX 65535U
134 #define MBUF_POOL_NAME "mbuf_pool"
135 #define DPDK_TX_BUF_NAME "tx_buffer"
136 //The number of elements in the mbuf pool.
137 #define DPDK_NB_MBUFS 8192U
138 #define MEMPOOL_CACHE_SIZE 256
139 #define MAX_PKT_BURST 32
140 // Configurable number of RX/TX ring descriptors
141 #define RTE_TEST_RX_DESC_DEFAULT 1024
142 #define RTE_TEST_TX_DESC_DEFAULT 1024
143
144 static uint16_t nb_rxd = RTE_TEST_RX_DESC_DEFAULT;
145 static uint16_t nb_txd = RTE_TEST_TX_DESC_DEFAULT;
146
147 #define RTE_ETH_PCAP_SNAPLEN ETHER_MAX_JUMBO_FRAME_LEN
148
149 static struct rte_eth_dev_tx_buffer *tx_buffer;
150
151 struct dpdk_ts_helper{
152 struct timeval start_time;
153 uint64_t start_cycles;
154 uint64_t hz;
155 };
156 struct pcap_dpdk{
157 pcap_t * orig;
158 uint16_t portid; // portid of DPDK
159 int must_clear_promisc;
160 int filter_in_userland;
161 uint64_t rx_pkts;
162 uint64_t bpf_drop;
163 struct timeval prev_ts;
164 struct rte_eth_stats prev_stats;
165 struct timeval curr_ts;
166 struct rte_eth_stats curr_stats;
167 uint64_t pps;
168 uint64_t bps;
169 struct rte_mempool * pktmbuf_pool;
170 struct dpdk_ts_helper ts_helper;
171 struct ether_addr eth_addr;
172 char mac_addr[DPDK_MAC_ADDR_SIZE];
173 char pci_addr[DPDK_PCI_ADDR_SIZE];
174 unsigned char pcap_tmp_buf[RTE_ETH_PCAP_SNAPLEN];
175 };
176
177 static struct rte_eth_conf port_conf = {
178 .rxmode = {
179 .split_hdr_size = 0,
180 },
181 .txmode = {
182 .mq_mode = ETH_MQ_TX_NONE,
183 },
184 };
185
186 static int dpdk_init_timer(struct pcap_dpdk *pd){
187 gettimeofday(&(pd->ts_helper.start_time),NULL);
188 pd->ts_helper.start_cycles = rte_get_timer_cycles();
189 pd->ts_helper.hz = rte_get_timer_hz();
190 if (pd->ts_helper.hz == 0){
191 return -1;
192 }
193 return 0;
194 }
195 static inline void calculate_timestamp(struct dpdk_ts_helper *helper,struct timeval *ts)
196 {
197 uint64_t cycles;
198 // delta
199 struct timeval cur_time;
200 cycles = rte_get_timer_cycles() - helper->start_cycles;
201 cur_time.tv_sec = (time_t)(cycles/helper->hz);
202 cur_time.tv_usec = (suseconds_t)((cycles%helper->hz)*1e6/helper->hz);
203 timeradd(&(helper->start_time), &cur_time, ts);
204 }
205
206 static uint32_t dpdk_gather_data(unsigned char *data, int len, struct rte_mbuf *mbuf)
207 {
208 uint32_t total_len = 0;
209 while (mbuf && (total_len+mbuf->data_len) < len ){
210 rte_memcpy(data+total_len, rte_pktmbuf_mtod(mbuf,void *),mbuf->data_len);
211 total_len+=mbuf->data_len;
212 mbuf=mbuf->next;
213 }
214 return total_len;
215 }
216
217 static int pcap_dpdk_dispatch(pcap_t *p, int max_cnt, pcap_handler cb, u_char *cb_arg)
218 {
219 struct pcap_dpdk *pd = (struct pcap_dpdk*)(p->priv);
220 int burst_cnt = 0;
221 int nb_rx=0;
222 struct rte_mbuf *pkts_burst[MAX_PKT_BURST];
223 struct rte_mbuf *m;
224 struct pcap_pkthdr pcap_header;
225 uint16_t portid = pd->portid;
226 // In DPDK, pkt_len is sum of lengths for all segments. And data_len is for one segment
227 uint32_t pkt_len = 0;
228 int caplen = 0;
229 u_char *bp = NULL;
230 int i=0;
231 unsigned int gather_len =0;
232 int pkt_cnt = 0;
233 int is_accepted=0;
234 u_char *large_buffer=NULL;
235
236 pd->rx_pkts = 0;
237 if ( !PACKET_COUNT_IS_UNLIMITED(max_cnt) && max_cnt < MAX_PKT_BURST){
238 burst_cnt = max_cnt;
239 }else{
240 burst_cnt = MAX_PKT_BURST;
241 }
242
243 while( PACKET_COUNT_IS_UNLIMITED(max_cnt) || pkt_cnt < max_cnt){
244 if (p->break_loop){
245 break;
246 }
247 nb_rx = (int)rte_eth_rx_burst(portid, 0, pkts_burst, burst_cnt);
248 pkt_cnt += nb_rx;
249 for ( i = 0; i < nb_rx; i++) {
250 m = pkts_burst[i];
251 calculate_timestamp(&(pd->ts_helper),&(pcap_header.ts));
252 pkt_len = rte_pktmbuf_pkt_len(m);
253 // caplen = min(pkt_len, p->snapshot);
254 // caplen will not be changed, no matter how long the rte_pktmbuf
255 caplen = pkt_len < p->snapshot ? pkt_len: p->snapshot;
256 pcap_header.caplen = caplen;
257 pcap_header.len = pkt_len;
258 // volatile prefetch
259 rte_prefetch0(rte_pktmbuf_mtod(m, void *));
260 bp = NULL;
261 if (m->nb_segs == 1)
262 {
263 bp = rte_pktmbuf_mtod(m, u_char *);
264 }else{
265 // use fast buffer pcap_tmp_buf if pkt_len is small, no need to call malloc and free
266 if ( pkt_len <= ETHER_MAX_JUMBO_FRAME_LEN)
267 {
268 gather_len = dpdk_gather_data(pd->pcap_tmp_buf, RTE_ETH_PCAP_SNAPLEN, m);
269 bp = pd->pcap_tmp_buf;
270 }else{
271 // need call free later
272 large_buffer = (u_char *)malloc(caplen*sizeof(u_char));
273 gather_len = dpdk_gather_data(large_buffer, caplen, m);
274 bp = large_buffer;
275 }
276
277 }
278 if (bp){
279 if (p->fcode.bf_insns==NULL || pcap_filter(p->fcode.bf_insns, bp, pcap_header.len, pcap_header.caplen)){
280 cb(cb_arg, &pcap_header, bp);
281 }else{
282 pd->bpf_drop++;
283 }
284 }
285 //free all pktmbuf
286 rte_pktmbuf_free(m);
287 if (large_buffer){
288 free(large_buffer);
289 large_buffer=NULL;
290 }
291 }
292 }
293 pd->rx_pkts = pkt_cnt;
294 return pd->rx_pkts;
295 }
296
297 static int pcap_dpdk_inject(pcap_t *p, const void *buf _U_, int size _U_)
298 {
299 //not implemented yet
300 pcap_fmt_errmsg_for_errno(p->errbuf, PCAP_ERRBUF_SIZE,
301 errno, "dpdk error: Inject function has not been implemented yet");
302 return PCAP_ERROR;
303 }
304
305 static void pcap_dpdk_close(pcap_t *p)
306 {
307 struct pcap_dpdk *pd = p->priv;
308 if (pd==NULL)
309 {
310 return;
311 }
312 if (pd->must_clear_promisc)
313 {
314 rte_eth_promiscuous_disable(pd->portid);
315 }
316 rte_eth_dev_stop(pd->portid);
317 rte_eth_dev_close(pd->portid);
318 pcap_cleanup_live_common(p);
319 }
320
321 static void nic_stats_display(struct pcap_dpdk *pd)
322 {
323 uint16_t portid = pd->portid;
324 struct rte_eth_stats stats;
325 rte_eth_stats_get(portid, &stats);
326 RTE_LOG(INFO,USER1, "portid:%d, RX-packets: %-10"PRIu64" RX-errors: %-10"PRIu64
327 " RX-bytes: %-10"PRIu64" RX-Imissed: %-10"PRIu64"\n", portid, stats.ipackets, stats.ierrors,
328 stats.ibytes,stats.imissed);
329 RTE_LOG(INFO,USER1, "portid:%d, RX-PPS: %-10"PRIu64" RX-Mbps: %.2lf\n", portid, pd->pps, pd->bps/1e6f );
330 }
331
332 static int pcap_dpdk_stats(pcap_t *p, struct pcap_stat *ps)
333 {
334 struct pcap_dpdk *pd = p->priv;
335 calculate_timestamp(&(pd->ts_helper), &(pd->curr_ts));
336 rte_eth_stats_get(pd->portid,&(pd->curr_stats));
337 if (ps){
338 ps->ps_recv = pd->curr_stats.ipackets;
339 ps->ps_drop = pd->curr_stats.ierrors;
340 ps->ps_drop += pd->bpf_drop;
341 ps->ps_ifdrop = pd->curr_stats.imissed;
342 }
343 uint64_t delta_pkt = pd->curr_stats.ipackets - pd->prev_stats.ipackets;
344 struct timeval delta_tm;
345 timersub(&(pd->curr_ts),&(pd->prev_ts), &delta_tm);
346 uint64_t delta_usec = delta_tm.tv_sec*1e6+delta_tm.tv_usec;
347 uint64_t delta_bit = (pd->curr_stats.ibytes-pd->prev_stats.ibytes)*8;
348 RTE_LOG(DEBUG, USER1, "delta_usec: %-10"PRIu64" delta_pkt: %-10"PRIu64" delta_bit: %-10"PRIu64"\n", delta_usec, delta_pkt, delta_bit);
349 pd->pps = (uint64_t)(delta_pkt*1e6f/delta_usec);
350 pd->bps = (uint64_t)(delta_bit*1e6f/delta_usec);
351 nic_stats_display(pd);
352 pd->prev_stats = pd->curr_stats;
353 pd->prev_ts = pd->curr_ts;
354 return 0;
355 }
356
357 static int pcap_dpdk_setnonblock(pcap_t *p, int fd _U_){
358 pcap_fmt_errmsg_for_errno(p->errbuf, PCAP_ERRBUF_SIZE,
359 errno, "dpdk error: setnonblock not support");
360 return 0;
361 }
362
363 static int pcap_dpdk_getnonblock(pcap_t *p){
364 pcap_fmt_errmsg_for_errno(p->errbuf, PCAP_ERRBUF_SIZE,
365 errno, "dpdk error: getnonblock not support");
366 return 0;
367 }
368 static int check_link_status(uint16_t portid, struct rte_eth_link *plink)
369 {
370 // wait up to 9 seconds to get link status
371 rte_eth_link_get(portid, plink);
372 return plink->link_status == ETH_LINK_UP;
373 }
374 static void eth_addr_str(struct ether_addr *addrp, char* mac_str, int len)
375 {
376 int offset=0;
377 if (addrp == NULL){
378 pcap_snprintf(mac_str, len-1, DPDK_DEF_MAC_ADDR);
379 return;
380 }
381 for (int i=0; i<6; i++)
382 {
383 if (offset >= len)
384 { // buffer overflow
385 return;
386 }
387 if (i==0)
388 {
389 pcap_snprintf(mac_str+offset, len-1-offset, "%02X",addrp->addr_bytes[i]);
390 offset+=2; // FF
391 }else{
392 pcap_snprintf(mac_str+offset, len-1-offset, ":%02X", addrp->addr_bytes[i]);
393 offset+=3; // :FF
394 }
395 }
396 return;
397 }
398 // return portid by device name, otherwise return -1
399 static uint16_t portid_by_device(char * device)
400 {
401 uint16_t ret = DPDK_PORTID_MAX;
402 int len = strlen(device);
403 int prefix_len = strlen(DPDK_PREFIX);
404 unsigned long ret_ul = 0L;
405 char *pEnd;
406 if (len<=prefix_len || strncmp(device, DPDK_PREFIX, prefix_len)) // check prefix dpdk:
407 {
408 return ret;
409 }
410 //check all chars are digital
411 for (int i=prefix_len; device[i]; i++){
412 if (device[i]<'0' || device[i]>'9'){
413 return ret;
414 }
415 }
416 ret_ul = strtoul(&(device[prefix_len]), &pEnd, 10);
417 // too large for portid
418 if (ret_ul >= DPDK_PORTID_MAX){
419 return ret;
420 }
421 ret = (uint16_t)ret_ul;
422 return ret;
423 }
424
425 static int parse_dpdk_cfg(char* dpdk_cfg,char** dargv)
426 {
427 int cnt=0;
428 memset(dargv,0,sizeof(dargv[0])*DPDK_ARGC_MAX);
429 //current process name
430 int skip_space = 1;
431 int i=0;
432 RTE_LOG(INFO, USER1,"dpdk cfg: %s\n",dpdk_cfg);
433 // find first non space char
434 // The last opt is NULL
435 for (i=0;dpdk_cfg[i] && cnt<DPDK_ARGC_MAX-1;i++){
436 if (skip_space && dpdk_cfg[i]!=' '){ // not space
437 skip_space=!skip_space; // skip normal char
438 dargv[cnt++] = dpdk_cfg+i;
439 }
440 if (!skip_space && dpdk_cfg[i]==' '){ // fint a space
441 dpdk_cfg[i]=0x00; // end of this opt
442 skip_space=!skip_space; // skip space char
443 }
444 }
445 dargv[cnt]=NULL;
446 return cnt;
447 }
448
449 // only called once
450 static int dpdk_pre_init(char * ebuf)
451 {
452 int dargv_cnt=0;
453 char *dargv[DPDK_ARGC_MAX];
454 char *ptr_dpdk_cfg = NULL;
455 int ret = PCAP_ERROR;
456 // globale var
457 if (is_dpdk_pre_inited)
458 {
459 // already inited
460 return 0;
461 }
462 // check for root permission
463 if( geteuid() != 0)
464 {
465 RTE_LOG(ERR, USER1, "%s\n", DPDK_ERR_PERM_MSG);
466 pcap_fmt_errmsg_for_errno(ebuf, PCAP_ERRBUF_SIZE,
467 errno, "dpdk error: %s",
468 DPDK_ERR_PERM_MSG);
469 ret = PCAP_ERROR_PERM_DENIED;
470 return ret;
471 }
472 // init EAL
473 ptr_dpdk_cfg = getenv(DPDK_CFG_ENV_NAME);
474 // set default log level to debug
475 rte_log_set_global_level(DPDK_DEF_LOG_LEV);
476 if (ptr_dpdk_cfg == NULL)
477 {
478 RTE_LOG(INFO,USER1,"env $DPDK_CFG is unset, so using default: %s\n",DPDK_DEF_CFG);
479 ptr_dpdk_cfg = DPDK_DEF_CFG;
480 }
481 memset(dpdk_cfg_buf,0,sizeof(dpdk_cfg_buf));
482 snprintf(dpdk_cfg_buf,DPDK_CFG_MAX_LEN-1,"%s %s",DPDK_LIB_NAME,ptr_dpdk_cfg);
483 dargv_cnt = parse_dpdk_cfg(dpdk_cfg_buf,dargv);
484 ret = rte_eal_init(dargv_cnt,dargv);
485 // if init successed, we do not need to do it again later.
486 if (ret == 0){
487 is_dpdk_pre_inited = 1;
488 }
489 return ret;
490 }
491
492 static int pcap_dpdk_activate(pcap_t *p)
493 {
494 struct pcap_dpdk *pd = p->priv;
495 pd->orig = p;
496 int ret = PCAP_ERROR;
497 uint16_t nb_ports=0;
498 uint16_t portid= DPDK_PORTID_MAX;
499 unsigned nb_mbufs = DPDK_NB_MBUFS;
500 struct rte_eth_rxconf rxq_conf;
501 struct rte_eth_txconf txq_conf;
502 struct rte_eth_conf local_port_conf = port_conf;
503 struct rte_eth_dev_info dev_info;
504 int is_port_up = 0;
505 struct rte_eth_link link;
506 do{
507 //init EAL
508 ret = dpdk_pre_init(p->errbuf);
509 if (ret < 0)
510 {
511 pcap_fmt_errmsg_for_errno(p->errbuf, PCAP_ERRBUF_SIZE,
512 errno, "dpdk error: Init failed with device %s",
513 p->opt.device);
514 ret = PCAP_ERROR;
515 break;
516 }
517 ret = dpdk_init_timer(pd);
518 if (ret<0)
519 {
520 pcap_fmt_errmsg_for_errno(p->errbuf, PCAP_ERRBUF_SIZE,
521 errno, "dpdk error: Init timer error with device %s",
522 p->opt.device);
523 ret = PCAP_ERROR;
524 break;
525 }
526
527 nb_ports = rte_eth_dev_count_avail();
528 if (nb_ports == 0)
529 {
530 pcap_fmt_errmsg_for_errno(p->errbuf, PCAP_ERRBUF_SIZE,
531 errno, "dpdk error: No Ethernet ports");
532 ret = PCAP_ERROR;
533 break;
534 }
535
536 portid = portid_by_device(p->opt.device);
537 if (portid == DPDK_PORTID_MAX){
538 pcap_fmt_errmsg_for_errno(p->errbuf, PCAP_ERRBUF_SIZE,
539 errno, "dpdk error: portid is invalid. device %s",
540 p->opt.device);
541 ret = PCAP_ERROR_NO_SUCH_DEVICE;
542 break;
543 }
544
545 pd->portid = portid;
546
547 if (p->snapshot <= 0 || p->snapshot > MAXIMUM_SNAPLEN)
548 {
549 p->snapshot = MAXIMUM_SNAPLEN;
550 }
551 // create the mbuf pool
552 pd->pktmbuf_pool = rte_pktmbuf_pool_create(MBUF_POOL_NAME, nb_mbufs,
553 MEMPOOL_CACHE_SIZE, 0, RTE_MBUF_DEFAULT_BUF_SIZE,
554 rte_socket_id());
555 if (pd->pktmbuf_pool == NULL)
556 {
557 pcap_fmt_errmsg_for_errno(p->errbuf, PCAP_ERRBUF_SIZE,
558 errno, "dpdk error: Cannot init mbuf pool");
559 ret = PCAP_ERROR;
560 break;
561 }
562 // config dev
563 rte_eth_dev_info_get(portid, &dev_info);
564 if (dev_info.tx_offload_capa & DEV_TX_OFFLOAD_MBUF_FAST_FREE)
565 {
566 local_port_conf.txmode.offloads |=DEV_TX_OFFLOAD_MBUF_FAST_FREE;
567 }
568 // only support 1 queue
569 ret = rte_eth_dev_configure(portid, 1, 1, &local_port_conf);
570 if (ret < 0)
571 {
572 pcap_fmt_errmsg_for_errno(p->errbuf, PCAP_ERRBUF_SIZE,
573 errno, "dpdk error: Cannot configure device: err=%d, port=%u",
574 ret, portid);
575 ret = PCAP_ERROR;
576 break;
577 }
578 // adjust rx tx
579 ret = rte_eth_dev_adjust_nb_rx_tx_desc(portid, &nb_rxd, &nb_txd);
580 if (ret < 0)
581 {
582 pcap_fmt_errmsg_for_errno(p->errbuf, PCAP_ERRBUF_SIZE,
583 errno, "dpdk error: Cannot adjust number of descriptors: err=%d, port=%u",
584 ret, portid);
585 ret = PCAP_ERROR;
586 break;
587 }
588 // get MAC addr
589 rte_eth_macaddr_get(portid, &(pd->eth_addr));
590 eth_addr_str(&(pd->eth_addr), pd->mac_addr, DPDK_MAC_ADDR_SIZE-1);
591
592 // init one RX queue
593 rxq_conf = dev_info.default_rxconf;
594 rxq_conf.offloads = local_port_conf.rxmode.offloads;
595 ret = rte_eth_rx_queue_setup(portid, 0, nb_rxd,
596 rte_eth_dev_socket_id(portid),
597 &rxq_conf,
598 pd->pktmbuf_pool);
599 if (ret < 0)
600 {
601 pcap_fmt_errmsg_for_errno(p->errbuf, PCAP_ERRBUF_SIZE,
602 errno, "dpdk error: rte_eth_rx_queue_setup:err=%d, port=%u",
603 ret, portid);
604 ret = PCAP_ERROR;
605 break;
606 }
607
608 // init one TX queue
609 txq_conf = dev_info.default_txconf;
610 txq_conf.offloads = local_port_conf.txmode.offloads;
611 ret = rte_eth_tx_queue_setup(portid, 0, nb_txd,
612 rte_eth_dev_socket_id(portid),
613 &txq_conf);
614 if (ret < 0)
615 {
616 pcap_fmt_errmsg_for_errno(p->errbuf, PCAP_ERRBUF_SIZE,
617 errno, "dpdk error: rte_eth_tx_queue_setup:err=%d, port=%u",
618 ret, portid);
619 ret = PCAP_ERROR;
620 break;
621 }
622 // Initialize TX buffers
623 tx_buffer = rte_zmalloc_socket(DPDK_TX_BUF_NAME,
624 RTE_ETH_TX_BUFFER_SIZE(MAX_PKT_BURST), 0,
625 rte_eth_dev_socket_id(portid));
626 if (tx_buffer == NULL)
627 {
628 pcap_fmt_errmsg_for_errno(p->errbuf, PCAP_ERRBUF_SIZE,
629 errno, "dpdk error: Cannot allocate buffer for tx on port %u", portid);
630 ret = PCAP_ERROR;
631 break;
632 }
633 rte_eth_tx_buffer_init(tx_buffer, MAX_PKT_BURST);
634 // Start device
635 ret = rte_eth_dev_start(portid);
636 if (ret < 0)
637 {
638 pcap_fmt_errmsg_for_errno(p->errbuf, PCAP_ERRBUF_SIZE,
639 errno, "dpdk error: rte_eth_dev_start:err=%d, port=%u",
640 ret, portid);
641 ret = PCAP_ERROR;
642 break;
643 }
644 // set promiscuous mode
645 if (p->opt.promisc){
646 pd->must_clear_promisc=1;
647 rte_eth_promiscuous_enable(portid);
648 }
649 // check link status
650 is_port_up = check_link_status(portid, &link);
651 if (!is_port_up){
652 pcap_fmt_errmsg_for_errno(p->errbuf, PCAP_ERRBUF_SIZE,
653 errno, "dpdk error: link is down, port=%u",portid);
654 ret = PCAP_ERROR_IFACE_NOT_UP;
655 break;
656 }
657 // reset statistics
658 rte_eth_stats_reset(pd->portid);
659 calculate_timestamp(&(pd->ts_helper), &(pd->prev_ts));
660 rte_eth_stats_get(pd->portid,&(pd->prev_stats));
661 // format pcap_t
662 pd->portid = portid;
663 p->fd = pd->portid;
664 if (p->snapshot <=0 || p->snapshot> MAXIMUM_SNAPLEN)
665 {
666 p->snapshot = MAXIMUM_SNAPLEN;
667 }
668 p->linktype = DLT_EN10MB; // Ethernet, the 10MB is historical.
669 p->selectable_fd = p->fd;
670 p->read_op = pcap_dpdk_dispatch;
671 p->inject_op = pcap_dpdk_inject;
672 // DPDK only support filter in userland now
673 pd->filter_in_userland = 1;
674 p->setfilter_op = install_bpf_program;
675 p->setdirection_op = NULL;
676 p->set_datalink_op = NULL;
677 p->getnonblock_op = pcap_dpdk_getnonblock;
678 p->setnonblock_op = pcap_dpdk_setnonblock;
679 p->stats_op = pcap_dpdk_stats;
680 p->cleanup_op = pcap_dpdk_close;
681 p->breakloop_op = pcap_breakloop_common;
682 ret = 0; // OK
683 }while(0);
684
685 if (ret == PCAP_ERROR)
686 {
687 pcap_cleanup_live_common(p);
688 }else{
689 rte_eth_dev_get_name_by_port(portid,pd->pci_addr);
690 RTE_LOG(INFO, USER1,"Port %d device: %s, MAC:%s, PCI:%s\n", portid, p->opt.device, pd->mac_addr, pd->pci_addr);
691 RTE_LOG(INFO, USER1,"Port %d Link Up. Speed %u Mbps - %s\n",
692 portid, link.link_speed,
693 (link.link_duplex == ETH_LINK_FULL_DUPLEX) ?
694 ("full-duplex") : ("half-duplex\n"));
695 }
696 return ret;
697 }
698
699 // device name for dpdk shoud be in the form as dpdk:number, such as dpdk:0
700 pcap_t * pcap_dpdk_create(const char *device, char *ebuf, int *is_ours)
701 {
702 pcap_t *p=NULL;
703 *is_ours = 0;
704
705 *is_ours = !strncmp(device, "dpdk:", 5);
706 if (! *is_ours)
707 return NULL;
708 //memset will happen
709 p = pcap_create_common(ebuf, sizeof(struct pcap_dpdk));
710
711 if (p == NULL)
712 return NULL;
713 p->activate_op = pcap_dpdk_activate;
714 return p;
715 }
716
717 int pcap_dpdk_findalldevs(pcap_if_list_t *devlistp, char *ebuf)
718 {
719 int ret=0;
720 int nb_ports = 0;
721 char dpdk_name[DPDK_DEV_NAME_MAX];
722 char dpdk_desc[DPDK_DEV_DESC_MAX];
723 struct ether_addr eth_addr;
724 char mac_addr[DPDK_MAC_ADDR_SIZE];
725 char pci_addr[DPDK_PCI_ADDR_SIZE];
726 do{
727 ret = dpdk_pre_init(ebuf);
728 if (ret < 0)
729 {
730 pcap_fmt_errmsg_for_errno(ebuf, PCAP_ERRBUF_SIZE,
731 errno, "error: Init failed with device");
732 ret = PCAP_ERROR;
733 break;
734 }
735 nb_ports = rte_eth_dev_count_avail();
736 if (nb_ports == 0)
737 {
738 pcap_fmt_errmsg_for_errno(ebuf, PCAP_ERRBUF_SIZE,
739 errno, "DPDK error: No Ethernet ports");
740 ret = PCAP_ERROR;
741 break;
742 }
743 for (int i=0; i<nb_ports; i++){
744 pcap_snprintf(dpdk_name,DPDK_DEV_NAME_MAX-1,"dpdk:%d",i);
745 // mac addr
746 rte_eth_macaddr_get(i, &eth_addr);
747 eth_addr_str(&eth_addr,mac_addr,DPDK_MAC_ADDR_SIZE);
748 // PCI addr
749 rte_eth_dev_get_name_by_port(i,pci_addr);
750 pcap_snprintf(dpdk_desc,DPDK_DEV_DESC_MAX-1,"%s %s, MAC:%s, PCI:%s", DPDK_DESC, dpdk_name, mac_addr, pci_addr);
751 // continue add all dev, even error happens
752 add_dev(devlistp, dpdk_name, 0, dpdk_desc, ebuf);
753 }
754 }while(0);
755 return ret;
756 }