src/plugins/mactime/mactime_test.c

   1 /*
   2  * mactime.c - skeleton vpp-api-test plug-in
   3  *
   4  * Copyright (c) <current-year> <your-organization>
   5  * Licensed under the Apache License, Version 2.0 (the "License");
   6  * you may not use this file except in compliance with the License.
   7  * You may obtain a copy of the License at:
   8  *
   9  *     http://www.apache.org/licenses/LICENSE-2.0
  10  *
  11  * Unless required by applicable law or agreed to in writing, software
  12  * distributed under the License is distributed on an "AS IS" BASIS,
  13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14  * See the License for the specific language governing permissions and
  15  * limitations under the License.
  16  */
  17 #include <vat/vat.h>
  18 #include <vlibapi/api.h>
  19 #include <vlibmemory/api.h>
  20 #include <vppinfra/error.h>
  21 #include <vppinfra/time_range.h>
  22 #include <vnet/ethernet/ethernet.h>
  23
  24 uword vat_unformat_sw_if_index (unformat_input_t * input, va_list * args);
  25
  26 /* Declare message IDs */
  27 #include <mactime/mactime_msg_enum.h>
  28
  29 /* define message structures */
  30 #define vl_typedefs
  31 #include <mactime/mactime_all_api_h.h>
  32 #undef vl_typedefs
  33
  34 /* declare message handlers for each api */
  35
  36 #define vl_endianfun            /* define message structures */
  37 #include <mactime/mactime_all_api_h.h>
  38 #undef vl_endianfun
  39
  40 /* instantiate all the print functions we know about */
  41 #define vl_print(handle, ...)
  42 #define vl_printfun
  43 #include <mactime/mactime_all_api_h.h>
  44 #undef vl_printfun
  45
  46 /* Get the API version number. */
  47 #define vl_api_version(n,v) static u32 api_version=(v);
  48 #include <mactime/mactime_all_api_h.h>
  49 #undef vl_api_version
  50
  51 typedef struct
  52 {
  53   /* API message ID base */
  54   u16 msg_id_base;
  55   vat_main_t *vat_main;
  56 } mactime_test_main_t;
  57
  58 mactime_test_main_t mactime_test_main;
  59
  60 #define __plugin_msg_base mactime_test_main.msg_id_base
  61 #include <vlibapi/vat_helper_macros.h>
  62
  63 #define foreach_standard_reply_retval_handler   \
  64 _(mactime_enable_disable_reply)                 \
  65 _(mactime_add_del_range_reply)
  66
  67 #define _(n)                                            \
  68     static void vl_api_##n##_t_handler                  \
  69     (vl_api_##n##_t * mp)                               \
  70     {                                                   \
  71         vat_main_t * vam = mactime_test_main.vat_main;   \
  72         i32 retval = ntohl(mp->retval);                 \
  73         if (vam->async_mode) {                          \
  74             vam->async_errors += (retval < 0);          \
  75         } else {                                        \
  76             vam->retval = retval;                       \
  77             vam->result_ready = 1;                      \
  78         }                                               \
  79     }
  80 foreach_standard_reply_retval_handler;
  81 #undef _
  82
  83 /*
  84  * Table of message reply handlers, must include boilerplate handlers
  85  * we just generated
  86  */
  87 #define foreach_vpe_api_reply_msg                               \
  88 _(MACTIME_ENABLE_DISABLE_REPLY, mactime_enable_disable_reply)   \
  89 _(MACTIME_ADD_DEL_RANGE_REPLY, mactime_add_del_range_reply)
  90
  91 static int
  92 api_mactime_enable_disable (vat_main_t * vam)
  93 {
  94   unformat_input_t *i = vam->input;
  95   int enable_disable = 1;
  96   u32 sw_if_index = ~0;
  97   vl_api_mactime_enable_disable_t *mp;
  98   int ret;
  99
 100   /* Parse args required to build the message */
 101   while (unformat_check_input (i) != UNFORMAT_END_OF_INPUT)
 102     {
 103       if (unformat (i, "%U", unformat_sw_if_index, vam, &sw_if_index))
 104         ;
 105       else if (unformat (i, "sw_if_index %d", &sw_if_index))
 106         ;
 107       else if (unformat (i, "disable"))
 108         enable_disable = 0;
 109       else
 110         break;
 111     }
 112
 113   if (sw_if_index == ~0)
 114     {
 115       errmsg ("missing interface name / explicit sw_if_index number \n");
 116       return -99;
 117     }
 118
 119   /* Construct the API message */
 120   M (MACTIME_ENABLE_DISABLE, mp);
 121   mp->sw_if_index = ntohl (sw_if_index);
 122   mp->enable_disable = enable_disable;
 123
 124   /* send it... */
 125   S (mp);
 126
 127   /* Wait for a reply... */
 128   W (ret);
 129   return ret;
 130 }
 131
 132 /* These two ought to be in a library somewhere but they aren't */
 133 static uword
 134 my_unformat_mac_address (unformat_input_t * input, va_list * args)
 135 {
 136   u8 *a = va_arg (*args, u8 *);
 137   return unformat (input, "%x:%x:%x:%x:%x:%x", &a[0], &a[1], &a[2], &a[3],
 138                    &a[4], &a[5]);
 139 }
 140
 141 static u8 *
 142 my_format_mac_address (u8 * s, va_list * args)
 143 {
 144   u8 *a = va_arg (*args, u8 *);
 145   return format (s, "%02x:%02x:%02x:%02x:%02x:%02x",
 146                  a[0], a[1], a[2], a[3], a[4], a[5]);
 147 }
 148
 149 static int
 150 api_mactime_add_del_range (vat_main_t * vam)
 151 {
 152   unformat_input_t *i = vam->input;
 153   vl_api_mactime_add_del_range_t *mp;
 154   u8 mac_address[8];
 155   u8 *device_name = 0;
 156   clib_timebase_range_t *rp = 0;
 157   int name_set = 0;
 158   int mac_set = 0;
 159   u8 is_add = 1;
 160   u8 allow = 0;
 161   u8 drop = 0;
 162   u8 no_udp_10001 = 0;
 163   u64 data_quota = 0;
 164   int ret;
 165   int ii;
 166
 167   /* Parse args required to build the message */
 168   while (unformat_check_input (i) != UNFORMAT_END_OF_INPUT)
 169     {
 170       if (unformat (i, "name %s", &device_name))
 171         {
 172           vec_add1 (device_name, 0);
 173           name_set = 1;
 174         }
 175       else if (unformat (i, "allow-range %U",
 176                          unformat_clib_timebase_range_vector, &rp))
 177         allow = 1;
 178       else if (unformat (i, "drop-range %U",
 179                          unformat_clib_timebase_range_vector, &rp))
 180         drop = 1;
 181       else if (unformat (i, "allow-static"))
 182         allow = 1;
 183       else if (unformat (i, "drop-static"))
 184         drop = 1;
 185       else if (unformat (i, "no-udp-10001"))
 186         no_udp_10001 = 1;
 187       else if (unformat (i, "mac %U", my_unformat_mac_address, mac_address))
 188         mac_set = 1;
 189       else if (unformat (i, "del"))
 190         is_add = 0;
 191       else if (unformat (i, "data-quota %lldM", &data_quota))
 192         data_quota <<= 20;
 193       else if (unformat (i, "data-quota %lldG", &data_quota))
 194         data_quota <<= 30;
 195       else
 196         break;
 197     }
 198
 199   /* Sanity checks */
 200   if (mac_set == 0)
 201     {
 202       vec_free (rp);
 203       vec_free (device_name);
 204       errmsg ("mac address required, not set\n");
 205       return -99;
 206     }
 207
 208   /* allow-range / drop-range parse errors cause this condition */
 209   if (is_add && allow == 0 && drop == 0)
 210     {
 211       vec_free (rp);
 212       vec_free (device_name);
 213       errmsg ("neither allow nor drop set, parse error...\n");
 214     }
 215
 216   /* Unlikely, but check anyhow */
 217   if (vec_len (device_name) > ARRAY_LEN (mp->device_name))
 218     {
 219       vec_free (rp);
 220       vec_free (device_name);
 221       errmsg ("device name too long, max %d\n", ARRAY_LEN (mp->device_name));
 222       return -99;
 223     }
 224
 225   /* Cough up a device name if none set */
 226   if (name_set == 0)
 227     {
 228       device_name = format (0, "mac %U%c", my_format_mac_address,
 229                             mac_address, 0);
 230     }
 231
 232   /* Construct the API message */
 233   M2 (MACTIME_ADD_DEL_RANGE, mp, sizeof (rp[0]) * vec_len (rp));
 234   mp->is_add = is_add;
 235   mp->drop = drop;
 236   mp->allow = allow;
 237   mp->no_udp_10001 = no_udp_10001;
 238   mp->data_quota = clib_host_to_net_u64 (data_quota);
 239   memcpy (mp->mac_address, mac_address, sizeof (mp->mac_address));
 240   memcpy (mp->device_name, device_name, vec_len (device_name));
 241   mp->count = clib_host_to_net_u32 (vec_len (rp));
 242
 243   for (ii = 0; ii < vec_len (rp); ii++)
 244     {
 245       mp->ranges[ii].start = rp[ii].start;
 246       mp->ranges[ii].end = rp[ii].end;
 247     }
 248
 249   vec_free (rp);
 250   vec_free (device_name);
 251
 252   /* send it... */
 253   S (mp);
 254
 255   /* Wait for a reply... */
 256   W (ret);
 257   return ret;
 258 }
 259
 260 /*
 261  * List of messages that the api test plugin sends,
 262  * and that the data plane plugin processes
 263  */
 264 #define foreach_vpe_api_msg                     \
 265 _(mactime_enable_disable, "<intfc> [disable]")  \
 266 _(mactime_add_del_range,                        \
 267   "name <devname> mac <mac-addr> allow drop\n"  \
 268   "allow-range Mon - Fri 9:00 - 17:00")
 269
 270 static void
 271 mactime_api_hookup (vat_main_t * vam)
 272 {
 273   mactime_test_main_t *sm = &mactime_test_main;
 274   /* Hook up handlers for replies from the data plane plug-in */
 275 #define _(N,n)                                                  \
 276     vl_msg_api_set_handlers((VL_API_##N + sm->msg_id_base),     \
 277                            #n,                                  \
 278                            vl_api_##n##_t_handler,              \
 279                            vl_noop_handler,                     \
 280                            vl_api_##n##_t_endian,               \
 281                            vl_api_##n##_t_print,                \
 282                            sizeof(vl_api_##n##_t), 1);
 283   foreach_vpe_api_reply_msg;
 284 #undef _
 285
 286   /* API messages we can send */
 287 #define _(n,h) hash_set_mem (vam->function_by_name, #n, api_##n);
 288   foreach_vpe_api_msg;
 289 #undef _
 290
 291   /* Help strings */
 292 #define _(n,h) hash_set_mem (vam->help_by_name, #n, h);
 293   foreach_vpe_api_msg;
 294 #undef _
 295 }
 296
 297 clib_error_t *
 298 vat_plugin_register (vat_main_t * vam)
 299 {
 300   mactime_test_main_t *sm = &mactime_test_main;
 301   u8 *name;
 302
 303   sm->vat_main = vam;
 304
 305   /* Ask the vpp engine for the first assigned message-id */
 306   name = format (0, "mactime_%08x%c", api_version, 0);
 307   sm->msg_id_base = vl_client_get_first_plugin_msg_id ((char *) name);
 308
 309   if (sm->msg_id_base != (u16) ~ 0)
 310     mactime_api_hookup (vam);
 311
 312   vec_free (name);
 313
 314   return 0;
 315 }
 316
 317 /*
 318  * fd.io coding-style-patch-verification: ON
 319  *
 320  * Local Variables:
 321  * eval: (c-set-style "gnu")
 322  * End:
 323  */