[ceph.git] / ceph / src / spdk / dpdk / test / test / test_sched.c

/* SPDX-License-Identifier: BSD-3-Clause
 * Copyright(c) 2010-2014 Intel Corporation
 */

#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <stdint.h>
#include <unistd.h>

#include "test.h"

#include <rte_cycles.h>
#include <rte_ether.h>
#include <rte_ip.h>
#include <rte_byteorder.h>
#include <rte_sched.h>


#define SUBPORT         0
#define PIPE            1
#define TC              2
#define QUEUE           3

static struct rte_sched_subport_params subport_param[] = {
	{
		.tb_rate = 1250000000,
		.tb_size = 1000000,

		.tc_rate = {1250000000, 1250000000, 1250000000, 1250000000},
		.tc_period = 10,
	},
};

static struct rte_sched_pipe_params pipe_profile[] = {
	{ /* Profile #0 */
		.tb_rate = 305175,
		.tb_size = 1000000,

		.tc_rate = {305175, 305175, 305175, 305175},
		.tc_period = 40,

		.wrr_weights = {1, 1, 1, 1,  1, 1, 1, 1,  1, 1, 1, 1,  1, 1, 1, 1},
	},
};

static struct rte_sched_port_params port_param = {
	.socket = 0, /* computed */
	.rate = 0, /* computed */
	.mtu = 1522,
	.frame_overhead = RTE_SCHED_FRAME_OVERHEAD_DEFAULT,
	.n_subports_per_port = 1,
	.n_pipes_per_subport = 1024,
	.qsize = {32, 32, 32, 32},
	.pipe_profiles = pipe_profile,
	.n_pipe_profiles = 1,
};

#define NB_MBUF          32
#define MBUF_DATA_SZ     (2048 + RTE_PKTMBUF_HEADROOM)
#define MEMPOOL_CACHE_SZ 0
#define SOCKET           0


static struct rte_mempool *
create_mempool(void)
{
	struct rte_mempool * mp;

	mp = rte_mempool_lookup("test_sched");
	if (!mp)
		mp = rte_pktmbuf_pool_create("test_sched", NB_MBUF,
			MEMPOOL_CACHE_SZ, 0, MBUF_DATA_SZ, SOCKET);

	return mp;
}

static void
prepare_pkt(struct rte_mbuf *mbuf)
{
	struct ether_hdr *eth_hdr;
	struct vlan_hdr *vlan1, *vlan2;
	struct ipv4_hdr *ip_hdr;

	/* Simulate a classifier */
	eth_hdr = rte_pktmbuf_mtod(mbuf, struct ether_hdr *);
	vlan1 = (struct vlan_hdr *)(&eth_hdr->ether_type );
	vlan2 = (struct vlan_hdr *)((uintptr_t)&eth_hdr->ether_type + sizeof(struct vlan_hdr));
	eth_hdr = (struct ether_hdr *)((uintptr_t)&eth_hdr->ether_type + 2 *sizeof(struct vlan_hdr));
	ip_hdr = (struct ipv4_hdr *)((uintptr_t)eth_hdr +  sizeof(eth_hdr->ether_type));

	vlan1->vlan_tci = rte_cpu_to_be_16(SUBPORT);
	vlan2->vlan_tci = rte_cpu_to_be_16(PIPE);
	eth_hdr->ether_type =  rte_cpu_to_be_16(ETHER_TYPE_IPv4);
	ip_hdr->dst_addr = IPv4(0,0,TC,QUEUE);


	rte_sched_port_pkt_write(mbuf, SUBPORT, PIPE, TC, QUEUE, e_RTE_METER_YELLOW);

	/* 64 byte packet */
	mbuf->pkt_len  = 60;
	mbuf->data_len = 60;
}


/**
 * test main entrance for library sched
 */
static int
test_sched(void)
{
	struct rte_mempool *mp = NULL;
	struct rte_sched_port *port = NULL;
	uint32_t pipe;
	struct rte_mbuf *in_mbufs[10];
	struct rte_mbuf *out_mbufs[10];
	int i;

	int err;

	mp = create_mempool();
	TEST_ASSERT_NOT_NULL(mp, "Error creating mempool\n");

	port_param.socket = 0;
	port_param.rate = (uint64_t) 10000 * 1000 * 1000 / 8;

	port = rte_sched_port_config(&port_param);
	TEST_ASSERT_NOT_NULL(port, "Error config sched port\n");

	err = rte_sched_subport_config(port, SUBPORT, subport_param);
	TEST_ASSERT_SUCCESS(err, "Error config sched, err=%d\n", err);

	for (pipe = 0; pipe < port_param.n_pipes_per_subport; pipe ++) {
		err = rte_sched_pipe_config(port, SUBPORT, pipe, 0);
		TEST_ASSERT_SUCCESS(err, "Error config sched pipe %u, err=%d\n", pipe, err);
	}

	for (i = 0; i < 10; i++) {
		in_mbufs[i] = rte_pktmbuf_alloc(mp);
		TEST_ASSERT_NOT_NULL(in_mbufs[i], "Packet allocation failed\n");
		prepare_pkt(in_mbufs[i]);
	}


	err = rte_sched_port_enqueue(port, in_mbufs, 10);
	TEST_ASSERT_EQUAL(err, 10, "Wrong enqueue, err=%d\n", err);

	err = rte_sched_port_dequeue(port, out_mbufs, 10);
	TEST_ASSERT_EQUAL(err, 10, "Wrong dequeue, err=%d\n", err);

	for (i = 0; i < 10; i++) {
		enum rte_meter_color color;
		uint32_t subport, traffic_class, queue;

		color = rte_sched_port_pkt_read_color(out_mbufs[i]);
		TEST_ASSERT_EQUAL(color, e_RTE_METER_YELLOW, "Wrong color\n");

		rte_sched_port_pkt_read_tree_path(out_mbufs[i],
				&subport, &pipe, &traffic_class, &queue);

		TEST_ASSERT_EQUAL(subport, SUBPORT, "Wrong subport\n");
		TEST_ASSERT_EQUAL(pipe, PIPE, "Wrong pipe\n");
		TEST_ASSERT_EQUAL(traffic_class, TC, "Wrong traffic_class\n");
		TEST_ASSERT_EQUAL(queue, QUEUE, "Wrong queue\n");

	}


	struct rte_sched_subport_stats subport_stats;
	uint32_t tc_ov;
	rte_sched_subport_read_stats(port, SUBPORT, &subport_stats, &tc_ov);
#if 0
	TEST_ASSERT_EQUAL(subport_stats.n_pkts_tc[TC-1], 10, "Wrong subport stats\n");
#endif
	struct rte_sched_queue_stats queue_stats;
	uint16_t qlen;
	rte_sched_queue_read_stats(port, QUEUE, &queue_stats, &qlen);
#if 0
	TEST_ASSERT_EQUAL(queue_stats.n_pkts, 10, "Wrong queue stats\n");
#endif

	rte_sched_port_free(port);

	return 0;
}

REGISTER_TEST_COMMAND(sched_autotest, test_sched);
Commit	Line	Data
11fdf7f2 TL	1	/* SPDX-License-Identifier: BSD-3-Clause
11fdf7f2 TL	2	* Copyright(c) 2010-2014 Intel Corporation
7c673cae FG	3	*/
	4
	5	#include <stdlib.h>
	6	#include <stdio.h>
	7	#include <string.h>
	8	#include <stdint.h>
	9	#include <unistd.h>
	10
	11	#include "test.h"
	12
	13	#include <rte_cycles.h>
	14	#include <rte_ether.h>
	15	#include <rte_ip.h>
	16	#include <rte_byteorder.h>
	17	#include <rte_sched.h>
	18
	19
	20	#define SUBPORT 0
	21	#define PIPE 1
	22	#define TC 2
	23	#define QUEUE 3
	24
	25	static struct rte_sched_subport_params subport_param[] = {
	26	{
	27	.tb_rate = 1250000000,
	28	.tb_size = 1000000,
	29
	30	.tc_rate = {1250000000, 1250000000, 1250000000, 1250000000},
	31	.tc_period = 10,
	32	},
	33	};
	34
	35	static struct rte_sched_pipe_params pipe_profile[] = {
	36	{ /* Profile #0 */
	37	.tb_rate = 305175,
	38	.tb_size = 1000000,
	39
	40	.tc_rate = {305175, 305175, 305175, 305175},
	41	.tc_period = 40,
	42
	43	.wrr_weights = {1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1},
	44	},
	45	};
	46
	47	static struct rte_sched_port_params port_param = {
	48	.socket = 0, /* computed */
	49	.rate = 0, /* computed */
	50	.mtu = 1522,
	51	.frame_overhead = RTE_SCHED_FRAME_OVERHEAD_DEFAULT,
	52	.n_subports_per_port = 1,
	53	.n_pipes_per_subport = 1024,
	54	.qsize = {32, 32, 32, 32},
	55	.pipe_profiles = pipe_profile,
	56	.n_pipe_profiles = 1,
	57	};
	58
	59	#define NB_MBUF 32
	60	#define MBUF_DATA_SZ (2048 + RTE_PKTMBUF_HEADROOM)
	61	#define MEMPOOL_CACHE_SZ 0
	62	#define SOCKET 0
	63
	64
	65	static struct rte_mempool *
	66	create_mempool(void)
67	{
68	struct rte_mempool * mp;
69
70	mp = rte_mempool_lookup("test_sched");
71	if (!mp)
72	mp = rte_pktmbuf_pool_create("test_sched", NB_MBUF,
73	MEMPOOL_CACHE_SZ, 0, MBUF_DATA_SZ, SOCKET);
74
75	return mp;
76	}
77
78	static void
79	prepare_pkt(struct rte_mbuf *mbuf)
80	{
81	struct ether_hdr *eth_hdr;
82	struct vlan_hdr vlan1, vlan2;
83	struct ipv4_hdr *ip_hdr;
84
85	/* Simulate a classifier */
86	eth_hdr = rte_pktmbuf_mtod(mbuf, struct ether_hdr *);
87	vlan1 = (struct vlan_hdr *)(&eth_hdr->ether_type );
88	vlan2 = (struct vlan_hdr *)((uintptr_t)&eth_hdr->ether_type + sizeof(struct vlan_hdr));
89	eth_hdr = (struct ether_hdr )((uintptr_t)&eth_hdr->ether_type + 2 sizeof(struct vlan_hdr));
90	ip_hdr = (struct ipv4_hdr *)((uintptr_t)eth_hdr + sizeof(eth_hdr->ether_type));
91
92	vlan1->vlan_tci = rte_cpu_to_be_16(SUBPORT);
93	vlan2->vlan_tci = rte_cpu_to_be_16(PIPE);
94	eth_hdr->ether_type = rte_cpu_to_be_16(ETHER_TYPE_IPv4);
95	ip_hdr->dst_addr = IPv4(0,0,TC,QUEUE);
96
97
98	rte_sched_port_pkt_write(mbuf, SUBPORT, PIPE, TC, QUEUE, e_RTE_METER_YELLOW);
99
100	/* 64 byte packet */
101	mbuf->pkt_len = 60;
102	mbuf->data_len = 60;
103	}
104
105
106	/**
107	* test main entrance for library sched
108	*/
109	static int
110	test_sched(void)
111	{
112	struct rte_mempool *mp = NULL;
113	struct rte_sched_port *port = NULL;
114	uint32_t pipe;
115	struct rte_mbuf *in_mbufs[10];
116	struct rte_mbuf *out_mbufs[10];
117	int i;
118
119	int err;
120
121	mp = create_mempool();
122	TEST_ASSERT_NOT_NULL(mp, "Error creating mempool\n");
123
124	port_param.socket = 0;
125	port_param.rate = (uint64_t) 10000 * 1000 * 1000 / 8;
126
127	port = rte_sched_port_config(&port_param);
128	TEST_ASSERT_NOT_NULL(port, "Error config sched port\n");
129
130	err = rte_sched_subport_config(port, SUBPORT, subport_param);
131	TEST_ASSERT_SUCCESS(err, "Error config sched, err=%d\n", err);
132
133	for (pipe = 0; pipe < port_param.n_pipes_per_subport; pipe ++) {
134	err = rte_sched_pipe_config(port, SUBPORT, pipe, 0);
135	TEST_ASSERT_SUCCESS(err, "Error config sched pipe %u, err=%d\n", pipe, err);
136	}
137
138	for (i = 0; i < 10; i++) {
139	in_mbufs[i] = rte_pktmbuf_alloc(mp);
140	TEST_ASSERT_NOT_NULL(in_mbufs[i], "Packet allocation failed\n");
141	prepare_pkt(in_mbufs[i]);
142	}
143
144
145	err = rte_sched_port_enqueue(port, in_mbufs, 10);
146	TEST_ASSERT_EQUAL(err, 10, "Wrong enqueue, err=%d\n", err);
147
148	err = rte_sched_port_dequeue(port, out_mbufs, 10);
149	TEST_ASSERT_EQUAL(err, 10, "Wrong dequeue, err=%d\n", err);
150
151	for (i = 0; i < 10; i++) {
152	enum rte_meter_color color;
153	uint32_t subport, traffic_class, queue;
154
155	color = rte_sched_port_pkt_read_color(out_mbufs[i]);
156	TEST_ASSERT_EQUAL(color, e_RTE_METER_YELLOW, "Wrong color\n");
157
158	rte_sched_port_pkt_read_tree_path(out_mbufs[i],
159	&subport, &pipe, &traffic_class, &queue);
160
161	TEST_ASSERT_EQUAL(subport, SUBPORT, "Wrong subport\n");
162	TEST_ASSERT_EQUAL(pipe, PIPE, "Wrong pipe\n");
163	TEST_ASSERT_EQUAL(traffic_class, TC, "Wrong traffic_class\n");
164	TEST_ASSERT_EQUAL(queue, QUEUE, "Wrong queue\n");
165
166	}
167
168
169	struct rte_sched_subport_stats subport_stats;
170	uint32_t tc_ov;
171	rte_sched_subport_read_stats(port, SUBPORT, &subport_stats, &tc_ov);
172	#if 0
173	TEST_ASSERT_EQUAL(subport_stats.n_pkts_tc[TC-1], 10, "Wrong subport stats\n");
174	#endif
175	struct rte_sched_queue_stats queue_stats;
176	uint16_t qlen;
177	rte_sched_queue_read_stats(port, QUEUE, &queue_stats, &qlen);
178	#if 0
179	TEST_ASSERT_EQUAL(queue_stats.n_pkts, 10, "Wrong queue stats\n");
180	#endif
181
182	rte_sched_port_free(port);
183
184	return 0;
185	}
186
187	REGISTER_TEST_COMMAND(sched_autotest, test_sched);