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