vpp/vpp-api/api_main.c

   1 #include "vat.h"
   2
   3 vat_main_t vat_main;
   4
   5 void
   6 vat_suspend (vlib_main_t * vm, f64 interval)
   7 {
   8   vlib_process_suspend (vm, interval);
   9 }
  10
  11 static u8 *
  12 format_api_error (u8 * s, va_list * args)
  13 {
  14   vat_main_t *vam = va_arg (*args, vat_main_t *);
  15   i32 error = va_arg (*args, u32);
  16   uword *p;
  17
  18   p = hash_get (vam->error_string_by_error_number, -error);
  19
  20   if (p)
  21     s = format (s, "%s", p[0]);
  22   else
  23     s = format (s, "%d", error);
  24   return s;
  25 }
  26
  27
  28 static void
  29 init_error_string_table (vat_main_t * vam)
  30 {
  31
  32   vam->error_string_by_error_number = hash_create (0, sizeof (uword));
  33
  34 #define _(n,v,s) hash_set (vam->error_string_by_error_number, -v, s);
  35   foreach_vnet_api_error;
  36 #undef _
  37
  38   hash_set (vam->error_string_by_error_number, 99, "Misc");
  39 }
  40
  41 static clib_error_t *
  42 api_main_init (vlib_main_t * vm)
  43 {
  44   vat_main_t *vam = &vat_main;
  45
  46   vam->vlib_main = vm;
  47   vam->my_client_index = (u32) ~ 0;
  48   init_error_string_table (vam);
  49   vat_api_hookup (vam);
  50   return 0;
  51 }
  52
  53 VLIB_INIT_FUNCTION (api_main_init);
  54
  55 static clib_error_t *
  56 api_command_fn (vlib_main_t * vm,
  57                 unformat_input_t * input, vlib_cli_command_t * cmd)
  58 {
  59   vat_main_t *vam = &vat_main;
  60   unformat_input_t _input;
  61   uword c;
  62   u8 *cmdp, *argsp, *this_cmd;
  63   uword *p;
  64   u32 arg_len;
  65   int rv;
  66   int (*fp) (vat_main_t *);
  67   api_main_t *am = &api_main;
  68
  69   vam->vl_input_queue = am->shmem_hdr->vl_input_queue;
  70
  71   vec_reset_length (vam->inbuf);
  72   vam->input = &_input;
  73
  74   while (((c = unformat_get_input (input)) != '\n') &&
  75          (c != UNFORMAT_END_OF_INPUT))
  76     vec_add1 (vam->inbuf, c);
  77
  78   /* Add 1 octet's worth of extra space in case there are no args... */
  79   vec_add1 (vam->inbuf, 0);
  80
  81   /*$$$$ reinstall macro evaluator */
  82
  83   /* Split input into cmd + args */
  84   this_cmd = cmdp = vam->inbuf;
  85
  86   while (cmdp < (this_cmd + vec_len (this_cmd)))
  87     {
  88       if (*cmdp == ' ' || *cmdp == '\t' || *cmdp == '\n')
  89         {
  90           cmdp++;
  91         }
  92       else
  93         break;
  94     }
  95
  96   argsp = cmdp;
  97   while (argsp < (this_cmd + vec_len (this_cmd)))
  98     {
  99       if (*argsp != ' ' && *argsp != '\t' && *argsp != '\n')
 100         {
 101           argsp++;
 102         }
 103       else
 104         break;
 105     }
 106   *argsp++ = 0;
 107
 108   while (argsp < (this_cmd + vec_len (this_cmd)))
 109     {
 110       if (*argsp == ' ' || *argsp == '\t' || *argsp == '\n')
 111         {
 112           argsp++;
 113         }
 114       else
 115         break;
 116     }
 117
 118   /* Blank input line? */
 119   if (*cmdp == 0)
 120     return 0;
 121
 122   p = hash_get_mem (vam->function_by_name, cmdp);
 123   if (p == 0)
 124     {
 125       return clib_error_return (0, "'%s': function not found\n", cmdp);
 126     }
 127
 128   arg_len = strlen ((char *) argsp);
 129
 130   unformat_init_string (vam->input, (char *) argsp, arg_len);
 131   fp = (void *) p[0];
 132
 133   rv = (*fp) (vam);
 134
 135   if (rv < 0)
 136     {
 137       unformat_free (vam->input);
 138       return clib_error_return (0,
 139                                 "%s error: %U\n", cmdp,
 140                                 format_api_error, vam, rv);
 141
 142       if (vam->regenerate_interface_table)
 143         {
 144           vam->regenerate_interface_table = 0;
 145           api_sw_interface_dump (vam);
 146         }
 147     }
 148   unformat_free (vam->input);
 149   return 0;
 150 }
 151
 152 /* *INDENT-OFF* */
 153 VLIB_CLI_COMMAND (api_command, static) =
 154 {
 155   .path = "binary-api",
 156   .short_help = "binary-api <name> [<args>]",
 157   .function = api_command_fn,
 158 };
 159 /* *INDENT-ON* */
 160
 161 void
 162 api_cli_output (void *notused, const char *fmt, ...)
 163 {
 164   va_list va;
 165   vat_main_t *vam = &vat_main;
 166   vlib_main_t *vm = vam->vlib_main;
 167   vlib_process_t *cp = vlib_get_current_process (vm);
 168   u8 *s;
 169
 170   va_start (va, fmt);
 171   s = va_format (0, fmt, &va);
 172   va_end (va);
 173
 174   /* Terminate with \n if not present. */
 175   if (vec_len (s) > 0 && s[vec_len (s) - 1] != '\n')
 176     vec_add1 (s, '\n');
 177
 178   if ((!cp) || (!cp->output_function))
 179     fformat (stdout, "%v", s);
 180   else
 181     cp->output_function (cp->output_function_arg, s, vec_len (s));
 182
 183   vec_free (s);
 184 }
 185
 186 /*
 187  * fd.io coding-style-patch-verification: ON
 188  *
 189  * Local Variables:
 190  * eval: (c-set-style "gnu")
 191  * End:
 192  */