app/test/test_sched.c

   1 /*-
   2  *   BSD LICENSE
   3  *
   4  *   Copyright(c) 2010-2013 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 #include <stdlib.h>
  35 #include <stdio.h>
  36 #include <string.h>
  37 #include <stdint.h>
  38 #include <unistd.h>
  39
  40 #include <cmdline_parse.h>
  41
  42 #include "test.h"
  43
  44 #if defined(RTE_LIBRTE_SCHED) && defined(RTE_ARCH_X86_64)
  45
  46 #include <rte_cycles.h>
  47 #include <rte_ether.h>
  48 #include <rte_ip.h>
  49 #include <rte_byteorder.h>
  50 #include <rte_sched.h>
  51
  52
  53 #define VERIFY(exp,fmt,args...)                                         \
  54                 if (!(exp)) {                                               \
  55                         printf(fmt, ##args);                                    \
  56                         return -1;                                              \
  57                 }
  58
  59
  60 #define SUBPORT         0
  61 #define PIPE            1
  62 #define TC                      2
  63 #define QUEUE           3
  64
  65 static struct rte_sched_subport_params subport_param[] = {
  66         {
  67                 .tb_rate = 1250000000,
  68                 .tb_size = 1000000,
  69
  70                 .tc_rate = {1250000000, 1250000000, 1250000000, 1250000000},
  71                 .tc_period = 10,
  72         },
  73 };
  74
  75 static struct rte_sched_pipe_params pipe_profile[] = {
  76         { /* Profile #0 */
  77                 .tb_rate = 305175,
  78                 .tb_size = 1000000,
  79
  80                 .tc_rate = {305175, 305175, 305175, 305175},
  81                 .tc_period = 40,
  82
  83                 .wrr_weights = {1, 1, 1, 1,  1, 1, 1, 1,  1, 1, 1, 1,  1, 1, 1, 1},
  84         },
  85 };
  86
  87 static struct rte_sched_port_params port_param = {
  88         .name = "port_0",
  89         .socket = 0, /* computed */
  90         .rate = 0, /* computed */
  91         .frame_overhead = RTE_SCHED_FRAME_OVERHEAD_DEFAULT,
  92         .n_subports_per_port = 1,
  93         .n_pipes_per_subport = 4096,
  94         .qsize = {64, 64, 64, 64},
  95         .pipe_profiles = pipe_profile,
  96         .n_pipe_profiles = 1,
  97 };
  98
  99 #define NB_MBUF          32
 100 #define MAX_PACKET_SZ    2048
 101 #define MBUF_SZ (MAX_PACKET_SZ + sizeof(struct rte_mbuf) + RTE_PKTMBUF_HEADROOM)
 102 #define PKT_BURST_SZ     32
 103 #define MEMPOOL_CACHE_SZ PKT_BURST_SZ
 104 #define SOCKET           0
 105
 106
 107 static struct rte_mempool *
 108 create_mempool(void)
 109 {
 110         struct rte_mempool * mp;
 111
 112         mp = rte_mempool_lookup("test_sched");
 113         if (!mp)
 114                 mp = rte_mempool_create("test_sched",
 115                                 NB_MBUF,
 116                                 MBUF_SZ,
 117                                 MEMPOOL_CACHE_SZ,
 118                                 sizeof(struct rte_pktmbuf_pool_private),
 119                                 rte_pktmbuf_pool_init,
 120                                 NULL,
 121                                 rte_pktmbuf_init,
 122                                 NULL,
 123                                 SOCKET,
 124                                 0);
 125
 126         return mp;
 127 }
 128
 129 static void
 130 prepare_pkt(struct rte_mbuf *mbuf)
 131 {
 132         struct ether_hdr *eth_hdr;
 133         struct vlan_hdr *vlan1, *vlan2;
 134         struct ipv4_hdr *ip_hdr;
 135
 136         /* Simulate a classifier */
 137         eth_hdr = rte_pktmbuf_mtod(mbuf, struct ether_hdr *);
 138         vlan1 = (struct vlan_hdr *)(&eth_hdr->ether_type );
 139         vlan2 = (struct vlan_hdr *)((uintptr_t)&eth_hdr->ether_type + sizeof(struct vlan_hdr));
 140         eth_hdr = (struct ether_hdr *)((uintptr_t)&eth_hdr->ether_type + 2 *sizeof(struct vlan_hdr));
 141         ip_hdr = (struct ipv4_hdr *)((uintptr_t)eth_hdr +  sizeof(eth_hdr->ether_type));
 142
 143         vlan1->vlan_tci = rte_cpu_to_be_16(SUBPORT);
 144         vlan2->vlan_tci = rte_cpu_to_be_16(PIPE);
 145         eth_hdr->ether_type =  rte_cpu_to_be_16(ETHER_TYPE_IPv4);
 146         ip_hdr->dst_addr = IPv4(0,0,TC,QUEUE);
 147
 148
 149         rte_sched_port_pkt_write(mbuf, SUBPORT, PIPE, TC, QUEUE, e_RTE_METER_YELLOW);
 150
 151         /* 64 byte packet */
 152         mbuf->pkt.pkt_len  = 60;
 153         mbuf->pkt.data_len = 60;
 154 }
 155
 156
 157 /**
 158  * test main entrance for library sched
 159  */
 160 int
 161 test_sched(void)
 162 {
 163         struct rte_mempool *mp = NULL;
 164         struct rte_sched_port *port = NULL;
 165         uint32_t pipe;
 166         struct rte_mbuf *in_mbufs[10];
 167         struct rte_mbuf *out_mbufs[10];
 168         int i;
 169
 170         int err;
 171
 172         mp = create_mempool();
 173
 174         port_param.socket = 0;
 175         port_param.rate = (uint64_t) 10000 * 1000 * 1000 / 8;
 176         port_param.name = "port_0";
 177
 178         port = rte_sched_port_config(&port_param);
 179         VERIFY(port != NULL, "Error config sched port\n");
 180
 181
 182         err = rte_sched_subport_config(port, SUBPORT, subport_param);
 183         VERIFY(err == 0, "Error config sched, err=%d\n", err);
 184
 185         for (pipe = 0; pipe < port_param.n_pipes_per_subport; pipe ++) {
 186                 err = rte_sched_pipe_config(port, SUBPORT, pipe, 0);
 187                 VERIFY(err == 0, "Error config sched pipe %u, err=%d\n", pipe, err);
 188         }
 189
 190         for (i = 0; i < 10; i++) {
 191                 in_mbufs[i] = rte_pktmbuf_alloc(mp);
 192                 prepare_pkt(in_mbufs[i]);
 193         }
 194
 195
 196         err = rte_sched_port_enqueue(port, in_mbufs, 10);
 197         VERIFY(err == 10, "Wrong enqueue, err=%d\n", err);
 198
 199         err = rte_sched_port_dequeue(port, out_mbufs, 10);
 200         VERIFY(err == 10, "Wrong dequeue, err=%d\n", err);
 201
 202         for (i = 0; i < 10; i++) {
 203                 enum rte_meter_color color;
 204                 uint32_t subport, traffic_class, queue;
 205
 206                 color = rte_sched_port_pkt_read_color(out_mbufs[i]);
 207                 VERIFY(color == e_RTE_METER_YELLOW, "Wrong color\n");
 208
 209                 rte_sched_port_pkt_read_tree_path(out_mbufs[i],
 210                                 &subport, &pipe, &traffic_class, &queue);
 211
 212                 VERIFY(subport == SUBPORT, "Wrong subport\n");
 213                 VERIFY(pipe == PIPE, "Wrong pipe\n");
 214                 VERIFY(traffic_class == TC, "Wrong traffic_class\n");
 215                 VERIFY(queue == QUEUE, "Wrong queue\n");
 216
 217         }
 218
 219
 220         struct rte_sched_subport_stats subport_stats;
 221         uint32_t tc_ov;
 222         rte_sched_subport_read_stats(port, SUBPORT, &subport_stats, &tc_ov);
 223         //VERIFY(subport_stats.n_pkts_tc[TC-1] == 10, "Wrong subport stats\n");
 224
 225         struct rte_sched_queue_stats queue_stats;
 226         uint16_t qlen;
 227         rte_sched_queue_read_stats(port, QUEUE, &queue_stats, &qlen);
 228         //VERIFY(queue_stats.n_pkts == 10, "Wrong queue stats\n");
 229
 230         rte_sched_port_free(port);
 231
 232         return 0;
 233 }
 234
 235 #else /* RTE_LIBRTE_SCHED */
 236
 237 int
 238 test_sched(void)
 239 {
 240         printf("The Scheduler library is not included in this build\n");
 241         return 0;
 242 }
 243 #endif /* RTE_LIBRTE_SCHED */