vpp/vpp-api-test/vat/main.c

/*
 * 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"

vat_main_t vat_main;

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;
}

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;
    
    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 (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));
}


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();

    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);
}