xref: /dpdk/examples/multi_process/simple_mp/main.c (revision dada9ef6edc59015b6674b5a95258787c71401b0)
1 /*-
2  *   BSD LICENSE
3  *
4  *   Copyright(c) 2010-2012 Intel Corporation. All rights reserved.
5  *   All rights reserved.
6  *
7  *   Redistribution and use in source and binary forms, with or without
8  *   modification, are permitted provided that the following conditions
9  *   are met:
10  *
11  *     * Redistributions of source code must retain the above copyright
12  *       notice, this list of conditions and the following disclaimer.
13  *     * Redistributions in binary form must reproduce the above copyright
14  *       notice, this list of conditions and the following disclaimer in
15  *       the documentation and/or other materials provided with the
16  *       distribution.
17  *     * Neither the name of Intel Corporation nor the names of its
18  *       contributors may be used to endorse or promote products derived
19  *       from this software without specific prior written permission.
20  *
21  *   THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
22  *   "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
23  *   LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
24  *   A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
25  *   OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
26  *   SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
27  *   LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
28  *   DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
29  *   THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
30  *   (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
31  *   OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
32  *
33  */
34 
35 /*
36  * This sample application is a simple multi-process application which
37  * demostrates sharing of queues and memory pools between processes, and
38  * using those queues/pools for communication between the processes.
39  *
40  * Application is designed to run with two processes, a primary and a
41  * secondary, and each accepts commands on the commandline, the most
42  * important of which is "send", which just sends a string to the other
43  * process.
44  */
45 
46 #include <stdio.h>
47 #include <string.h>
48 #include <stdint.h>
49 #include <inttypes.h>
50 #include <stdarg.h>
51 #include <errno.h>
52 #include <unistd.h>
53 #include <termios.h>
54 #include <sys/queue.h>
55 
56 #include <rte_common.h>
57 #include <rte_memory.h>
58 #include <rte_memzone.h>
59 #include <rte_launch.h>
60 #include <rte_tailq.h>
61 #include <rte_eal.h>
62 #include <rte_per_lcore.h>
63 #include <rte_lcore.h>
64 #include <rte_debug.h>
65 #include <rte_atomic.h>
66 #include <rte_branch_prediction.h>
67 #include <rte_ring.h>
68 #include <rte_log.h>
69 #include <rte_mempool.h>
70 #include <cmdline_rdline.h>
71 #include <cmdline_parse.h>
72 #include <cmdline_socket.h>
73 #include <cmdline.h>
74 #include "mp_commands.h"
75 
76 #define RTE_LOGTYPE_APP RTE_LOGTYPE_USER1
77 
78 #define SOCKET0 0
79 
80 static const char *_MSG_POOL = "MSG_POOL";
81 static const char *_SEC_2_PRI = "SEC_2_PRI";
82 static const char *_PRI_2_SEC = "PRI_2_SEC";
83 const unsigned string_size = 64;
84 
85 struct rte_ring *send_ring, *recv_ring;
86 struct rte_mempool *message_pool;
87 volatile int quit = 0;
88 
89 static int
90 lcore_recv(__attribute__((unused)) void *arg)
91 {
92 	unsigned lcore_id = rte_lcore_id();
93 
94 	printf("Starting core %u\n", lcore_id);
95 	while (!quit){
96 		void *msg;
97 		if (rte_ring_dequeue(recv_ring, &msg) < 0){
98 			usleep(5);
99 			continue;
100 		}
101 		printf("core %u: Received '%s'\n", lcore_id, (char *)msg);
102 		rte_mempool_put(message_pool, msg);
103 	}
104 
105 	return 0;
106 }
107 
108 int
109 main(int argc, char **argv)
110 {
111 	const unsigned flags = 0;
112 	const unsigned ring_size = 64;
113 	const unsigned pool_size = 1024;
114 	const unsigned pool_cache = 32;
115 	const unsigned priv_data_sz = 0;
116 
117 	int ret;
118 	unsigned lcore_id;
119 
120 	ret = rte_eal_init(argc, argv);
121 	if (ret < 0)
122 		rte_exit(EXIT_FAILURE, "Cannot init EAL\n");
123 
124 	if (rte_eal_process_type() == RTE_PROC_PRIMARY){
125 		send_ring = rte_ring_create(_PRI_2_SEC, ring_size, SOCKET0, flags);
126 		recv_ring = rte_ring_create(_SEC_2_PRI, ring_size, SOCKET0, flags);
127 		message_pool = rte_mempool_create(_MSG_POOL, pool_size,
128 				string_size, pool_cache, priv_data_sz,
129 				NULL, NULL, NULL, NULL,
130 				SOCKET0, flags);
131 	} else {
132 		recv_ring = rte_ring_lookup(_PRI_2_SEC);
133 		send_ring = rte_ring_lookup(_SEC_2_PRI);
134 		message_pool = rte_mempool_lookup(_MSG_POOL);
135 	}
136 	if (send_ring == NULL)
137 		rte_exit(EXIT_FAILURE, "Problem getting sending ring\n");
138 	if (recv_ring == NULL)
139 		rte_exit(EXIT_FAILURE, "Problem getting receiving ring\n");
140 	if (message_pool == NULL)
141 		rte_exit(EXIT_FAILURE, "Problem getting message pool\n");
142 
143 	RTE_LOG(INFO, APP, "Finished Process Init.\n");
144 
145 	/* call lcore_recv() on every slave lcore */
146 	RTE_LCORE_FOREACH_SLAVE(lcore_id) {
147 		rte_eal_remote_launch(lcore_recv, NULL, lcore_id);
148 	}
149 
150 	/* call cmd prompt on master lcore */
151 	struct cmdline *cl = cmdline_stdin_new(simple_mp_ctx, "\nsimple_mp > ");
152 	if (cl == NULL)
153 		rte_exit(EXIT_FAILURE, "Cannot create cmdline instance\n");
154 	cmdline_interact(cl);
155 	cmdline_stdin_exit(cl);
156 
157 	rte_eal_mp_wait_lcore();
158 	return 0;
159 }
160