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