app/test/test_sched.c

   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 *)(&eth_hdr->ether_type);
  93         vlan2 = (struct rte_vlan_hdr *)(
  94                 (uintptr_t)&eth_hdr->ether_type + sizeof(struct rte_vlan_hdr));
  95         eth_hdr = (struct rte_ether_hdr *)(
  96                 (uintptr_t)&eth_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);