+++ /dev/null
-/*
- * Copyright (c) 2015 Cisco and/or its affiliates.
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at:
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-#include "vat.h"
-#include "plugin.h"
-#include <signal.h>
-
-vat_main_t vat_main;
-
-void
-vat_suspend (vlib_main_t * vm, f64 interval)
-{
- /* do nothing in the standalone version, just return */
-}
-
-void
-fformat_append_cr (FILE * ofp, const char *fmt, ...)
-{
- va_list va;
-
- va_start (va, fmt);
- (void) va_fformat (ofp, (char *) fmt, &va);
- va_end (va);
- fformat (ofp, "\n");
-}
-
-int
-connect_to_vpe (char *name)
-{
- vat_main_t *vam = &vat_main;
- api_main_t *am = &api_main;
-
- if (vl_client_connect_to_vlib ("/vpe-api", name, 32) < 0)
- return -1;
-
- vam->vl_input_queue = am->shmem_hdr->vl_input_queue;
- vam->my_client_index = am->my_client_index;
-
- return 0;
-}
-
-vlib_main_t vlib_global_main;
-vlib_main_t **vlib_mains;
-void
-vlib_cli_output (struct vlib_main_t *vm, char *fmt, ...)
-{
- clib_warning ("BUG");
-}
-
-
-static u8 *
-format_api_error (u8 * s, va_list * args)
-{
- vat_main_t *vam = va_arg (*args, vat_main_t *);
- i32 error = va_arg (*args, u32);
- uword *p;
-
- p = hash_get (vam->error_string_by_error_number, -error);
-
- if (p)
- s = format (s, "%s", p[0]);
- else
- s = format (s, "%d", error);
- return s;
-}
-
-void
-do_one_file (vat_main_t * vam)
-{
- int rv;
- int (*fp) (vat_main_t * vam);
- int arg_len;
- unformat_input_t _input;
- u8 *cmdp, *argsp;
- uword *p;
- u8 *this_cmd = 0;
-
- vam->input = &_input;
-
- /* Used by the "quit" command handler */
- if (setjmp (vam->jump_buf) != 0)
- return;
-
- vam->jump_buf_set = 1;
-
- while (1)
- {
- if (vam->ifp == stdin)
- {
- if (vam->exec_mode == 0)
- rv = write (1, "vat# ", 5);
- else
- rv = write (1, "exec# ", 6);
- }
-
- _vec_len (vam->inbuf) = 4096;
-
- if (vam->do_exit ||
- fgets ((char *) vam->inbuf, vec_len (vam->inbuf), vam->ifp) == 0)
- break;
-
- vam->input_line_number++;
-
- vec_free (this_cmd);
-
- this_cmd =
- (u8 *) clib_macro_eval (&vam->macro_main, (char *) vam->inbuf,
- 1 /* complain */ );
-
- if (vam->exec_mode == 0)
- {
- /* Split input into cmd + args */
- cmdp = this_cmd;
-
- while (cmdp < (this_cmd + vec_len (this_cmd)))
- {
- if (*cmdp == ' ' || *cmdp == '\t' || *cmdp == '\n')
- {
- cmdp++;
- }
- else
- break;
- }
- argsp = cmdp;
- while (argsp < (this_cmd + vec_len (this_cmd)))
- {
- if (*argsp != ' ' && *argsp != '\t' && *argsp != '\n')
- {
- argsp++;
- }
- else
- break;
- }
- *argsp++ = 0;
- while (argsp < (this_cmd + vec_len (this_cmd)))
- {
- if (*argsp == ' ' || *argsp == '\t' || *argsp == '\n')
- {
- argsp++;
- }
- else
- break;
- }
-
-
- /* Blank input line? */
- if (*cmdp == 0)
- continue;
-
- p = hash_get_mem (vam->function_by_name, cmdp);
- if (p == 0)
- {
- errmsg ("'%s': function not found\n", cmdp);
- continue;
- }
-
- arg_len = strlen ((char *) argsp);
-
- unformat_init_string (vam->input, (char *) argsp, arg_len);
- fp = (void *) p[0];
- }
- else
- {
- unformat_init_string (vam->input, (char *) this_cmd,
- strlen ((char *) this_cmd));
- cmdp = this_cmd;
- fp = exec;
- }
-
- rv = (*fp) (vam);
- if (rv < 0)
- errmsg ("%s error: %U\n", cmdp, format_api_error, vam, rv);
- unformat_free (vam->input);
-
- if (vam->regenerate_interface_table)
- {
- vam->regenerate_interface_table = 0;
- api_sw_interface_dump (vam);
- }
- }
-}
-
-static void
-init_error_string_table (vat_main_t * vam)
-{
-
- vam->error_string_by_error_number = hash_create (0, sizeof (uword));
-
-#define _(n,v,s) hash_set (vam->error_string_by_error_number, -v, s);
- foreach_vnet_api_error;
-#undef _
-
- hash_set (vam->error_string_by_error_number, 99, "Misc");
-}
-
-static i8 *
-eval_current_file (macro_main_t * mm, i32 complain)
-{
- vat_main_t *vam = &vat_main;
- return ((i8 *) format (0, "%s%c", vam->current_file, 0));
-}
-
-static i8 *
-eval_current_line (macro_main_t * mm, i32 complain)
-{
- vat_main_t *vam = &vat_main;
- return ((i8 *) format (0, "%d%c", vam->input_line_number, 0));
-}
-
-static void
-signal_handler (int signum, siginfo_t * si, ucontext_t * uc)
-{
- vat_main_t *vam = &vat_main;
-
- switch (signum)
- {
- /* these (caught) signals cause the application to exit */
- case SIGINT:
- case SIGTERM:
- if (vam->jump_buf_set)
- {
- vam->do_exit = 1;
- return;
- }
-
- /* FALLTHROUGH on purpose */
-
- default:
- break;
- }
-
- _exit (1);
-}
-
-static void
-setup_signal_handlers (void)
-{
- uword i;
- struct sigaction sa;
-
- for (i = 1; i < 32; i++)
- {
- memset (&sa, 0, sizeof (sa));
- sa.sa_sigaction = (void *) signal_handler;
- sa.sa_flags = SA_SIGINFO;
-
- switch (i)
- {
- /* these signals take the default action */
- case SIGABRT:
- case SIGKILL:
- case SIGSTOP:
- case SIGUSR1:
- case SIGUSR2:
- continue;
-
- /* ignore SIGPIPE, SIGCHLD */
- case SIGPIPE:
- case SIGCHLD:
- sa.sa_sigaction = (void *) SIG_IGN;
- break;
-
- /* catch and handle all other signals */
- default:
- break;
- }
-
- if (sigaction (i, &sa, 0) < 0)
- clib_unix_warning ("sigaction %U", format_signal, i);
- }
-}
-
-int
-main (int argc, char **argv)
-{
- vat_main_t *vam = &vat_main;
- unformat_input_t _argv, *a = &_argv;
- u8 **input_files = 0;
- u8 *output_file = 0;
- u8 *chroot_prefix;
- u8 *this_input_file;
- u8 interactive = 1;
- u8 json_output = 0;
- u8 *heap;
- mheap_t *h;
- int i;
-
- clib_mem_init (0, 128 << 20);
-
- heap = clib_mem_get_per_cpu_heap ();
- h = mheap_header (heap);
-
- /* make the main heap thread-safe */
- h->flags |= MHEAP_FLAG_THREAD_SAFE;
-
- clib_macro_init (&vam->macro_main);
- clib_macro_add_builtin (&vam->macro_main, "current_file",
- eval_current_file);
- clib_macro_add_builtin (&vam->macro_main, "current_line",
- eval_current_line);
-
- init_error_string_table (vam);
-
- unformat_init_command_line (a, argv);
-
- while (unformat_check_input (a) != UNFORMAT_END_OF_INPUT)
- {
- if (unformat (a, "in %s", &this_input_file))
- vec_add1 (input_files, this_input_file);
- else if (unformat (a, "out %s", &output_file))
- ;
- else if (unformat (a, "script"))
- interactive = 0;
- else if (unformat (a, "json"))
- json_output = 1;
- else if (unformat (a, "plugin_path %s", (u8 *) & vat_plugin_path))
- vec_add1 (vat_plugin_path, 0);
- else if (unformat (a, "plugin_name_filter %s",
- (u8 *) & vat_plugin_name_filter))
- vec_add1 (vat_plugin_name_filter, 0);
- else if (unformat (a, "chroot prefix %s", &chroot_prefix))
- {
- vl_set_memory_root_path ((char *) chroot_prefix);
- }
- else
- {
- fformat (stderr,
- "%s: usage [in <f1> ... in <fn>] [out <fn>] [script] [json]\n");
- exit (1);
- }
- }
-
- if (output_file)
- vam->ofp = fopen ((char *) output_file, "w");
- else
- vam->ofp = stdout;
-
- if (vam->ofp == NULL)
- {
- fformat (stderr, "Couldn't open output file %s\n",
- output_file ? (char *) output_file : "stdout");
- exit (1);
- }
-
- clib_time_init (&vam->clib_time);
-
- vat_api_hookup (vam);
- vat_plugin_api_reference ();
-
- setup_signal_handlers ();
-
- if (connect_to_vpe ("vpp_api_test") < 0)
- {
- svm_region_exit ();
- fformat (stderr, "Couldn't connect to vpe, exiting...\n");
- exit (1);
- }
-
- vam->json_output = json_output;
-
- if (!json_output)
- {
- api_sw_interface_dump (vam);
- }
-
- vec_validate (vam->inbuf, 4096);
-
- vam->current_file = (u8 *) "plugin-init";
- vat_plugin_init (vam);
-
- for (i = 0; i < vec_len (input_files); i++)
- {
- vam->ifp = fopen ((char *) input_files[i], "r");
- if (vam->ifp == NULL)
- {
- fformat (stderr, "Couldn't open input file %s\n", input_files[i]);
- continue;
- }
- vam->current_file = input_files[i];
- vam->input_line_number = 0;
- do_one_file (vam);
- fclose (vam->ifp);
- }
-
- if (output_file)
- fclose (vam->ofp);
-
- if (interactive)
- {
- vam->ifp = stdin;
- vam->ofp = stdout;
- vam->current_file = (u8 *) "interactive";
- do_one_file (vam);
- fclose (vam->ifp);
- }
-
- vl_client_disconnect_from_vlib ();
- exit (0);
-}
-
-/*
- * fd.io coding-style-patch-verification: ON
- *
- * Local Variables:
- * eval: (c-set-style "gnu")
- * End:
- */
Code Review / vpp.git / blobdiff