xref: /dpdk/examples/flow_filtering/main.c (revision 117eaa70584b73eebf6f648cf3ee6f2ab03264a0)
1 /*-
2  *   BSD LICENSE
3  *
4  *   Copyright 2017 Mellanox.
5  *
6  *   Redistribution and use in source and binary forms, with or without
7  *   modification, are permitted provided that the following conditions
8  *   are met:
9  *
10  *     * Redistributions of source code must retain the above copyright
11  *       notice, this list of conditions and the following disclaimer.
12  *     * Redistributions in binary form must reproduce the above copyright
13  *       notice, this list of conditions and the following disclaimer in
14  *       the documentation and/or other materials provided with the
15  *       distribution.
16  *     * Neither the name of Mellanox. nor the names of its
17  *       contributors may be used to endorse or promote products derived
18  *       from this software without specific prior written permission.
19  *
20  *   THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
21  *   "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
22  *   LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
23  *   A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
24  *   OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
25  *   SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
26  *   LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
27  *   DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
28  *   THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
29  *   (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
30  *   OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
31  */
32 
33 #include <stdio.h>
34 #include <stdlib.h>
35 #include <string.h>
36 #include <stdint.h>
37 #include <inttypes.h>
38 #include <sys/types.h>
39 #include <sys/queue.h>
40 #include <netinet/in.h>
41 #include <setjmp.h>
42 #include <stdarg.h>
43 #include <ctype.h>
44 #include <errno.h>
45 #include <getopt.h>
46 #include <signal.h>
47 #include <stdbool.h>
48 
49 #include <rte_eal.h>
50 #include <rte_common.h>
51 #include <rte_malloc.h>
52 #include <rte_ether.h>
53 #include <rte_ethdev.h>
54 #include <rte_mempool.h>
55 #include <rte_mbuf.h>
56 #include <rte_net.h>
57 #include <rte_flow.h>
58 #include <rte_cycles.h>
59 
60 static volatile bool force_quit;
61 
62 static uint16_t port_id;
63 static uint16_t nr_queues = 5;
64 static uint8_t selected_queue = 1;
65 struct rte_mempool *mbuf_pool;
66 struct rte_flow *flow;
67 
68 #define SRC_IP ((0<<24) + (0<<16) + (0<<8) + 0) /* src ip = 0.0.0.0 */
69 #define DEST_IP ((192<<24) + (168<<16) + (1<<8) + 1) /* dest ip = 192.168.1.1 */
70 #define FULL_MASK 0xffffffff /* full mask */
71 #define EMPTY_MASK 0x0 /* empty mask */
72 
73 #include "flow_blocks.c"
74 
75 static inline void
76 print_ether_addr(const char *what, struct ether_addr *eth_addr)
77 {
78 	char buf[ETHER_ADDR_FMT_SIZE];
79 	ether_format_addr(buf, ETHER_ADDR_FMT_SIZE, eth_addr);
80 	printf("%s%s", what, buf);
81 }
82 
83 static void
84 main_loop(void)
85 {
86 	struct rte_mbuf *mbufs[32];
87 	struct ether_hdr *eth_hdr;
88 	struct rte_flow_error error;
89 	uint16_t nb_rx;
90 	uint16_t i;
91 	uint16_t j;
92 
93 	while (!force_quit) {
94 		for (i = 0; i < nr_queues; i++) {
95 			nb_rx = rte_eth_rx_burst(port_id,
96 						i, mbufs, 32);
97 			if (nb_rx) {
98 				for (j = 0; j < nb_rx; j++) {
99 					struct rte_mbuf *m = mbufs[j];
100 
101 					eth_hdr = rte_pktmbuf_mtod(m,
102 							struct ether_hdr *);
103 					print_ether_addr("src=",
104 							&eth_hdr->s_addr);
105 					print_ether_addr(" - dst=",
106 							&eth_hdr->d_addr);
107 					printf(" - queue=0x%x",
108 							(unsigned int)i);
109 					printf("\n");
110 
111 					rte_pktmbuf_free(m);
112 				}
113 			}
114 		}
115 	}
116 
117 	/* closing and releasing resources */
118 	rte_flow_flush(port_id, &error);
119 	rte_eth_dev_stop(port_id);
120 	rte_eth_dev_close(port_id);
121 }
122 
123 #define CHECK_INTERVAL 1000  /* 100ms */
124 #define MAX_REPEAT_TIMES 90  /* 9s (90 * 100ms) in total */
125 
126 static void
127 assert_link_status(void)
128 {
129 	struct rte_eth_link link;
130 	uint8_t rep_cnt = MAX_REPEAT_TIMES;
131 
132 	memset(&link, 0, sizeof(link));
133 	do {
134 		rte_eth_link_get(port_id, &link);
135 		if (link.link_status == ETH_LINK_UP)
136 			break;
137 		rte_delay_ms(CHECK_INTERVAL);
138 	} while (--rep_cnt);
139 
140 	if (link.link_status == ETH_LINK_DOWN)
141 		rte_exit(EXIT_FAILURE, ":: error: link is still down\n");
142 }
143 
144 static void
145 init_port(void)
146 {
147 	int ret;
148 	uint16_t i;
149 	struct rte_eth_conf port_conf = {
150 		.rxmode = {
151 			.split_hdr_size = 0,
152 			.ignore_offload_bitfield = 1,
153 			.offloads = DEV_RX_OFFLOAD_CRC_STRIP,
154 		},
155 		.txmode = {
156 			.offloads =
157 				DEV_TX_OFFLOAD_VLAN_INSERT |
158 				DEV_TX_OFFLOAD_IPV4_CKSUM  |
159 				DEV_TX_OFFLOAD_UDP_CKSUM   |
160 				DEV_TX_OFFLOAD_TCP_CKSUM   |
161 				DEV_TX_OFFLOAD_SCTP_CKSUM  |
162 				DEV_TX_OFFLOAD_TCP_TSO,
163 		},
164 	};
165 	struct rte_eth_txconf txq_conf;
166 	struct rte_eth_rxconf rxq_conf;
167 	struct rte_eth_dev_info dev_info;
168 
169 	printf(":: initializing port: %d\n", port_id);
170 	ret = rte_eth_dev_configure(port_id,
171 				nr_queues, nr_queues, &port_conf);
172 	if (ret < 0) {
173 		rte_exit(EXIT_FAILURE,
174 			":: cannot configure device: err=%d, port=%u\n",
175 			ret, port_id);
176 	}
177 
178 	rte_eth_dev_info_get(port_id, &dev_info);
179 	rxq_conf = dev_info.default_rxconf;
180 	rxq_conf.offloads = port_conf.rxmode.offloads;
181 	/* only set Rx queues: something we care only so far */
182 	for (i = 0; i < nr_queues; i++) {
183 		ret = rte_eth_rx_queue_setup(port_id, i, 512,
184 				     rte_eth_dev_socket_id(port_id),
185 				     &rxq_conf,
186 				     mbuf_pool);
187 		if (ret < 0) {
188 			rte_exit(EXIT_FAILURE,
189 				":: Rx queue setup failed: err=%d, port=%u\n",
190 				ret, port_id);
191 		}
192 	}
193 
194 	txq_conf = dev_info.default_txconf;
195 	txq_conf.offloads = port_conf.txmode.offloads;
196 
197 	for (i = 0; i < nr_queues; i++) {
198 		ret = rte_eth_tx_queue_setup(port_id, i, 512,
199 				rte_eth_dev_socket_id(port_id),
200 				&txq_conf);
201 		if (ret < 0) {
202 			rte_exit(EXIT_FAILURE,
203 				":: Tx queue setup failed: err=%d, port=%u\n",
204 				ret, port_id);
205 		}
206 	}
207 
208 	rte_eth_promiscuous_enable(port_id);
209 	ret = rte_eth_dev_start(port_id);
210 	if (ret < 0) {
211 		rte_exit(EXIT_FAILURE,
212 			"rte_eth_dev_start:err=%d, port=%u\n",
213 			ret, port_id);
214 	}
215 
216 	assert_link_status();
217 
218 	printf(":: initializing port: %d done\n", port_id);
219 }
220 
221 static void
222 signal_handler(int signum)
223 {
224 	if (signum == SIGINT || signum == SIGTERM) {
225 		printf("\n\nSignal %d received, preparing to exit...\n",
226 				signum);
227 		force_quit = true;
228 	}
229 }
230 
231 int
232 main(int argc, char **argv)
233 {
234 	int ret;
235 	uint8_t nr_ports;
236 	struct rte_flow_error error;
237 
238 	ret = rte_eal_init(argc, argv);
239 	if (ret < 0)
240 		rte_exit(EXIT_FAILURE, ":: invalid EAL arguments\n");
241 
242 	force_quit = false;
243 	signal(SIGINT, signal_handler);
244 	signal(SIGTERM, signal_handler);
245 
246 	nr_ports = rte_eth_dev_count();
247 	if (nr_ports == 0)
248 		rte_exit(EXIT_FAILURE, ":: no Ethernet ports found\n");
249 	port_id = 0;
250 	if (nr_ports != 1) {
251 		printf(":: warn: %d ports detected, but we use only one: port %u\n",
252 			nr_ports, port_id);
253 	}
254 	mbuf_pool = rte_pktmbuf_pool_create("mbuf_pool", 4096, 128, 0,
255 					    RTE_MBUF_DEFAULT_BUF_SIZE,
256 					    rte_socket_id());
257 	if (mbuf_pool == NULL)
258 		rte_exit(EXIT_FAILURE, "Cannot init mbuf pool\n");
259 
260 	init_port();
261 
262 	/* create flow for send packet with */
263 	flow = generate_ipv4_flow(port_id, selected_queue,
264 				SRC_IP, EMPTY_MASK,
265 				DEST_IP, FULL_MASK, &error);
266 	if (!flow) {
267 		printf("Flow can't be created %d message: %s\n",
268 			error.type,
269 			error.message ? error.message : "(no stated reason)");
270 		rte_exit(EXIT_FAILURE, "error in creating flow");
271 	}
272 
273 	main_loop();
274 
275 	return 0;
276 }
277