src/vpp-api/client/test.c

   1 /*
   2  *------------------------------------------------------------------
   3  * test.c -- VPP API/Stats tests
   4  *
   5  * Copyright (c) 2016 Cisco and/or its affiliates.
   6  * Licensed under the Apache License, Version 2.0 (the "License");
   7  * you may not use this file except in compliance with the License.
   8  * You may obtain a copy of the License at:
   9  *
  10  *     http://www.apache.org/licenses/LICENSE-2.0
  11  *
  12  * Unless required by applicable law or agreed to in writing, software
  13  * distributed under the License is distributed on an "AS IS" BASIS,
  14  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15  * See the License for the specific language governing permissions and
  16  * limitations under the License.
  17  *------------------------------------------------------------------
  18  */
  19 #include <stdio.h>
  20 #include <stdlib.h>
  21 #include <assert.h>
  22 #include <sys/types.h>
  23 #include <sys/socket.h>
  24 #include <sys/mman.h>
  25 #include <sys/stat.h>
  26 #include <netinet/in.h>
  27 #include <netdb.h>
  28
  29
  30 #include <vnet/vnet.h>
  31 #include <vlib/vlib.h>
  32 #include <vlib/unix/unix.h>
  33 #include <vlibapi/api.h>
  34 #include <vppinfra/time.h>
  35 #include <signal.h>
  36 #include "vppapiclient.h"
  37 #include "stat_client.h"
  38
  39 #include <vlibmemory/vlib.api_enum.h>
  40 #include <vlibmemory/vlib.api_types.h>
  41 #include <vlibmemory/memclnt.api_enum.h>
  42 #include <vlibmemory/memclnt.api_types.h>
  43
  44 #include <vpp/api/vpe.api_enum.h>
  45 #include <vpp/api/vpe.api_types.h>
  46
  47 volatile int sigterm_received = 0;
  48 volatile u32 result_ready;
  49 volatile u16 result_msg_id;
  50
  51 /* M_NOALLOC: construct, but don't yet send a message */
  52
  53 #define M_NOALLOC(T,t)                          \
  54   do {                                          \
  55     result_ready = 0;                           \
  56     clib_memset (mp, 0, sizeof (*mp));          \
  57     mp->_vl_msg_id = ntohs (VL_API_##T);        \
  58     mp->client_index = am->my_client_index;     \
  59   } while(0);
  60
  61
  62
  63 void
  64 wrap_vac_callback (unsigned char *data, int len)
  65 {
  66   result_ready = 1;
  67   result_msg_id = ntohs(*((u16 *)data));
  68 }
  69
  70 static void
  71 test_connect ()
  72 {
  73   int rv = vac_connect("vac_client", NULL, wrap_vac_callback, 32 /* rx queue-length*/);
  74   if (rv != 0) {
  75     printf("Connect failed: %d\n", rv);
  76     exit(rv);
  77   }
  78   printf(".");
  79   vac_disconnect();
  80 }
  81
  82 static void
  83 test_messages (void)
  84 {
  85   api_main_t * am = vlibapi_get_main();
  86   vl_api_show_version_t message;
  87   vl_api_show_version_t *mp;
  88   int async = 1;
  89
  90   int rv = vac_connect("vac_client", NULL, wrap_vac_callback, 32 /* rx queue-length*/);
  91   if (rv != 0) {
  92     printf("Connect failed: %d\n", rv);
  93     exit(rv);
  94   }
  95
  96   double timestamp_start = unix_time_now_nsec() * 1e-6;
  97
  98   /*
  99    * Test vpe_api_write and vpe_api_read to send and recv message for an
 100    * API
 101    */
 102   int i;
 103   long int no_msgs = 10000;
 104   mp = &message;
 105
 106   for (i = 0; i < no_msgs; i++) {
 107     /* Construct the API message */
 108     M_NOALLOC(SHOW_VERSION, show_version);
 109     vac_write((char *)mp, sizeof(*mp));
 110 #ifndef __COVERITY__
 111     /* As given, async is always 1. Shut up Coverity about it */
 112     if (!async)
 113       while (result_ready == 0);
 114 #endif
 115   }
 116   if (async) {
 117     vl_api_control_ping_t control;
 118     vl_api_control_ping_t *mp;
 119     mp = &control;
 120     M_NOALLOC(CONTROL_PING, control_ping);
 121     vac_write((char *)mp, sizeof(*mp));
 122
 123     while (result_msg_id != VL_API_CONTROL_PING_REPLY);
 124   }
 125
 126   double timestamp_end = unix_time_now_nsec() * 1e-6;
 127   printf("\nTook %.2f msec, %.0f msgs/msec \n", (timestamp_end - timestamp_start),
 128          no_msgs/(timestamp_end - timestamp_start));
 129   printf("Exiting...\n");
 130   vac_disconnect();
 131 }
 132
 133 static void
 134 test_stats (void)
 135 {
 136   clib_mem_trace_enable_disable(1);
 137   clib_mem_trace (1);
 138
 139   int rv = stat_segment_connect (STAT_SEGMENT_SOCKET_FILE);
 140   assert(rv == 0);
 141
 142   u32 *dir;
 143   int i, k;
 144   stat_segment_data_t *res;
 145   u8 **pattern = 0;
 146   vec_add1(pattern, (u8 *)"/if/names");
 147   vec_add1(pattern, (u8 *)"/err");
 148
 149   dir = stat_segment_ls ((u8 **)pattern);
 150
 151   res = stat_segment_dump (dir);
 152   for (i = 0; i < vec_len (res); i++) {
 153     switch (res[i].type) {
 154     case STAT_DIR_TYPE_NAME_VECTOR:
 155       if (res[i].name_vector == 0)
 156         continue;
 157       for (k = 0; k < vec_len (res[i].name_vector); k++)
 158         if (res[i].name_vector[k])
 159           fformat (stdout, "[%d]: %s %s\n", k, res[i].name_vector[k],
 160                    res[i].name);
 161       break;
 162     default:
 163       assert(0);
 164     }
 165   }
 166   stat_segment_data_free (res);
 167   stat_segment_disconnect();
 168
 169   vec_free(pattern);
 170   vec_free(dir);
 171
 172   (void) clib_mem_trace_enable_disable (0);
 173   u8 *leak_report = format (0, "%U", format_clib_mem_heap, 0,
 174                             1 /* verbose, i.e. print leaks */ );
 175   printf("%s", leak_report);
 176   vec_free (leak_report);
 177   clib_mem_trace (0);
 178 }
 179
 180 int main (int argc, char ** argv)
 181 {
 182   clib_mem_init (0, 3ULL << 30);
 183   test_stats();
 184
 185   int i;
 186
 187   for (i = 0; i < 1000; i++) {
 188     test_connect();
 189   }
 190   test_messages();
 191   exit (0);
 192 }