aboutsummaryrefslogtreecommitdiff
path: root/platform/linux-generic/test/pktio_ipc/pktio_ipc2.c
blob: fc3b6833a72fdfc9af4e9be70e648b5b9623bd6f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
/*  SPDX-License-Identifier: BSD-3-Clause
 *  Copyright (c) 2015-2018 Linaro Limited
 *  Copyright (c) 2023 Nokia
 */

/**
 * @file
 *
 * @example pktio_ipc2.c  ODP IPC example application.
 *		This application works in pair with pktio_ipc1 application.
 *		It opens ipc pktio, reads packets and updates magic number.
 *		Also it allocates some packets from internal pool and sends
 *		to ipc pktio.
 */

#include "ipc_common.h"

static int ipc_second_process(int master_pid)
{
	odp_pktio_t ipc_pktio = ODP_PKTIO_INVALID;
	odp_pool_param_t params;
	odp_pool_t pool;
	odp_packet_t pkt_tbl[MAX_PKT_BURST];
	odp_packet_t alloc_pkt;
	int pkts;
	int ret;
	int i;
	odp_time_t start_cycle;
	odp_time_t cycle;
	odp_time_t diff;
	odp_time_t wait;
	uint64_t stat_pkts = 0;
	odp_pktin_queue_t pktin;

	/* Create packet pool */
	memset(&params, 0, sizeof(params));
	params.pkt.seg_len = SHM_PKT_POOL_BUF_SIZE;
	params.pkt.len     = SHM_PKT_POOL_BUF_SIZE;
	params.pkt.num     = SHM_PKT_POOL_SIZE;
	params.type        = ODP_POOL_PACKET;

	pool = odp_pool_create(TEST_IPC_POOL_NAME, &params);
	if (pool == ODP_POOL_INVALID) {
		ODPH_ERR("Error: packet pool create failed.\n");
		exit(EXIT_FAILURE);
	}

	wait = odp_time_local_from_ns(start_time_sec * ODP_TIME_SEC_IN_NS);
	start_cycle = odp_time_local();

	for (;;) {
		/*  exit loop if time specified */
		if (start_time_sec) {
			cycle = odp_time_local();
			diff = odp_time_diff(cycle, start_cycle);
			if (odp_time_cmp(wait, diff) < 0) {
				printf("timeout exit 1, start_time_sec %d\n",
				       start_time_sec);
				goto not_started;
			}
		}

		ipc_pktio = create_pktio(pool, master_pid);
		if (ipc_pktio != ODP_PKTIO_INVALID)
			break;
		if (!master_pid)
			break;

		odp_time_wait_ns(50 * ODP_TIME_MSEC_IN_NS);
	}

	if (ipc_pktio == ODP_PKTIO_INVALID) {
		odp_pool_destroy(pool);
		return -1;
	}

	memset(&pktin, 0, sizeof(pktin)); /* not needed but makes GCC happy */
	if (odp_pktin_queue(ipc_pktio, &pktin, 1) != 1) {
		odp_pool_destroy(pool);
		ODPH_ERR("no input queue\n");
		return -1;
	}

	/* start ipc pktio, i.e. wait until other process connects */
	for (;;) {
		/* 1. exit loop if time specified */
		if (start_time_sec) {
			cycle = odp_time_local();
			diff = odp_time_diff(cycle, start_cycle);
			if (odp_time_cmp(wait, diff) < 0) {
				printf("timeout exit 2, start_time_sec %d\n",
				       start_time_sec);
				goto not_started;
			}
		}

		ret = odp_pktio_start(ipc_pktio);
		if (!ret)
			break;

		/* Reduce polling frequency to once per 50ms */
		odp_time_wait_ns(50 * ODP_TIME_MSEC_IN_NS);
	}

	wait = odp_time_local_from_ns(run_time_sec * ODP_TIME_SEC_IN_NS);
	start_cycle = odp_time_local();
	for (;;) {
		/* exit loop if time specified */
		if (run_time_sec) {
			cycle = odp_time_local();
			diff = odp_time_diff(cycle, start_cycle);
			if (odp_time_cmp(wait, diff) < 0) {
				ODPH_DBG("exit after %d seconds\n",
					 run_time_sec);
				break;
			}
		}

		/* recv some packets and change MAGIC to MAGIC_2 */
		pkts = odp_pktin_recv(pktin, pkt_tbl, MAX_PKT_BURST);
		if (pkts <= 0)
			continue;

		for (i = 0; i < pkts; i++) {
			odp_packet_t pkt = pkt_tbl[i];
			pkt_head_t head;
			size_t off;

			off = odp_packet_l4_offset(pkt);
			if (off ==  ODP_PACKET_OFFSET_INVALID) {
				ODPH_ERR("invalid l4 offset\n");
				for (int j = i; j < pkts; j++)
					odp_packet_free(pkt_tbl[j]);
				break;
			}

			off += ODPH_UDPHDR_LEN;
			ret = odp_packet_copy_to_mem(pkt, off, sizeof(head),
						     &head);
			if (ret)
				ODPH_ABORT("unable copy out head data");

			if (head.magic != TEST_SEQ_MAGIC) {
				ODPH_ERR("Wrong head magic! %x", head.magic);
				for (int j = i; j < pkts; j++)
					odp_packet_free(pkt_tbl[j]);
				break;
			}

			/* Modify magic number in packet */
			head.magic = TEST_SEQ_MAGIC_2;
			ret = odp_packet_copy_from_mem(pkt, off, sizeof(head),
						       &head);
			if (ret)
				ODPH_ABORT("unable to copy in head data");
		}

		/* send all packets back */
		ret = ipc_odp_packet_send_or_free(ipc_pktio, pkt_tbl, i);
		if (ret < 0)
			ODPH_ABORT("can not send packets\n");

		stat_pkts += ret;

		/* alloc packet from local pool, set magic to ALLOC_MAGIC,
		 * and send it.*/
		alloc_pkt = odp_packet_alloc(pool, SHM_PKT_POOL_BUF_SIZE);
		if (alloc_pkt != ODP_PACKET_INVALID) {
			pkt_head_t head;
			size_t off;

			odp_packet_l4_offset_set(alloc_pkt, 30);

			head.magic = TEST_ALLOC_MAGIC;

			off = odp_packet_l4_offset(alloc_pkt);
			off += ODPH_UDPHDR_LEN;
			ret = odp_packet_copy_from_mem(alloc_pkt, off,
						       sizeof(head),
						       &head);
			if (ret)
				ODPH_ABORT("unable to copy in head data");

			pkt_tbl[0] = alloc_pkt;
			ret = ipc_odp_packet_send_or_free(ipc_pktio,
							  pkt_tbl, 1);
			if (ret < 0)
				ODPH_ABORT("can not send packets\n");
			stat_pkts += 1;
		}
	}

	/* cleanup and exit */
	ret = odp_pktio_stop(ipc_pktio);
	if (ret) {
		ODPH_DBG("ipc2: odp_pktio_stop error %d\n", ret);
		return -1;
	}

not_started:
	ret = odp_pktio_close(ipc_pktio);
	if (ret) {
		ODPH_DBG("ipc2: odp_pktio_close error %d\n", ret);
		return -1;
	}

	ret = odp_pool_destroy(pool);
	if (ret)
		ODPH_DBG("ipc2: pool_destroy error %d\n", ret);

	return stat_pkts > 1000 ? 0 : -1;
}

int main(int argc, char *argv[])
{
	odp_instance_t instance;
	int ret;
	cpu_set_t cpu_set;
	odp_cpumask_t mask;
	int cpu;
	pid_t pid;

	/* Parse and store the application arguments */
	parse_args(argc, argv);

	if (odp_init_global(&instance, NULL, NULL)) {
		ODPH_ERR("Error: ODP global init failed.\n");
		exit(EXIT_FAILURE);
	}

	odp_cpumask_default_worker(&mask, 0);
	cpu = odp_cpumask_first(&mask);
	ret = odp_cpumask_next(&mask, cpu);
	if (ret != -1)
		cpu = ret;

	CPU_ZERO(&cpu_set);
	CPU_SET(cpu, &cpu_set);

	pid = getpid();

	if (sched_setaffinity(pid, sizeof(cpu_set_t), &cpu_set)) {
		printf("Set CPU affinity failed to cpu %d.\n", cpu);
		return -1;
	}

	printf("ipc_pktio2 %d run on cpu %d\n", pid, cpu);

	/* Init this thread */
	if (odp_init_local(instance, ODP_THREAD_WORKER)) {
		ODPH_ERR("Error: ODP local init failed.\n");
		exit(EXIT_FAILURE);
	}

	ret = ipc_second_process(master_pid);

	if (odp_term_local()) {
		ODPH_ERR("Error: odp_term_local() failed.\n");
		exit(EXIT_FAILURE);
	}

	if (odp_term_global(instance)) {
		ODPH_ERR("Error: odp_term_global() failed.\n");
		exit(EXIT_FAILURE);
	}

	return ret;
}