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