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