summaryrefslogtreecommitdiffstats
path: root/src/vat2/main.c
diff options
context:
space:
mode:
Diffstat (limited to 'src/vat2/main.c')
-rw-r--r--src/vat2/main.c227
1 files changed, 227 insertions, 0 deletions
diff --git a/src/vat2/main.c b/src/vat2/main.c
new file mode 100644
index 00000000000..5b042e23503
--- /dev/null
+++ b/src/vat2/main.c
@@ -0,0 +1,227 @@
+/*
+ * Copyright (c) 2020 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 <stdio.h>
+#include <stdbool.h>
+#include <ctype.h>
+#include <vlib/vlib.h>
+#include <vlibapi/api_types.h>
+#include <vppinfra/cJSON.h>
+
+/* VPP API client includes */
+#include <vpp-api/client/vppapiclient.h>
+
+#include <limits.h>
+#include "vat2.h"
+
+uword *function_by_name;
+bool debug = false;
+
+char *vat2_plugin_path;
+static void
+vat2_find_plugin_path ()
+{
+ char *p, path[PATH_MAX];
+ int rv;
+ u8 *s;
+
+ /* find executable path */
+ if ((rv = readlink ("/proc/self/exe", path, PATH_MAX - 1)) == -1)
+ return;
+
+ /* readlink doesn't provide null termination */
+ path[rv] = 0;
+
+ /* strip filename */
+ if ((p = strrchr (path, '/')) == 0)
+ return;
+ *p = 0;
+
+ /* strip bin/ */
+ if ((p = strrchr (path, '/')) == 0)
+ return;
+ *p = 0;
+
+ s = format (0, "%s/lib/" CLIB_TARGET_TRIPLET "/vat2_plugins:"
+ "%s/lib/vat2_plugins", path, path);
+ vec_add1 (s, 0);
+ vat2_plugin_path = (char *) s;
+}
+
+void
+vac_callback (unsigned char *data, int len)
+{
+ u16 result_msg_id = ntohs(*((u16 *)data));
+ DBG("Received something async: %d\n", result_msg_id);
+}
+
+int vat2_load_plugins (char *path, char *filter, int *loaded);
+
+static int
+register_function (void)
+{
+ int loaded;
+
+ vat2_find_plugin_path();
+ DBG("Plugin Path %s\n", vat2_plugin_path);
+ int rv = vat2_load_plugins(vat2_plugin_path, 0, &loaded);
+ DBG("Loaded %u plugins\n", loaded);
+ return rv;
+}
+
+void
+vat2_register_function(char *name, cJSON (*f)(cJSON *))
+{
+ hash_set_mem(function_by_name, name, f);
+}
+
+int main (int argc, char **argv)
+{
+ /* Create a heap of 64MB */
+ clib_mem_init (0, 64 << 20);
+ char *filename = 0;
+ int index;
+ int c;
+ opterr = 0;
+ cJSON *o = 0;
+ uword *p = 0;
+
+ while ((c = getopt (argc, argv, "df:")) != -1) {
+ switch (c) {
+ case 'd':
+ debug = true;
+ break;
+ case 'f':
+ filename = optarg;
+ break;
+ case '?':
+ if (optopt == 'f')
+ fprintf (stderr, "Option -%c requires an argument.\n", optopt);
+ else if (isprint (optopt))
+ fprintf (stderr, "Unknown option `-%c'.\n", optopt);
+ else
+ fprintf (stderr,
+ "Unknown option character `\\x%x'.\n",
+ optopt);
+ return 1;
+ default:
+ abort ();
+ }
+ }
+
+ DBG("debug = %d, filename = %s\n", debug, filename);
+
+ for (index = optind; index < argc; index++)
+ DBG ("Non-option argument %s\n", argv[index]);
+
+ index = optind;
+
+ /* Load plugins */
+ function_by_name = hash_create_string (0, sizeof (uword));
+ int res = register_function();
+ if (res < 0) {
+ fprintf(stderr, "%s: loading plugins failed\n", argv[0]);
+ exit(-1);
+ }
+
+ if (argc > index + 2) {
+ fprintf(stderr, "%s: Too many arguments\n", argv[0]);
+ exit(-1);
+ }
+
+ /* Read JSON from stdin, command line or file */
+ if (argc >= (index + 1)) {
+ p = hash_get_mem (function_by_name, argv[index]);
+ if (p == 0) {
+ fprintf(stderr, "%s: Unknown command: %s\n", argv[0], argv[index]);
+ exit(-1);
+ }
+ }
+
+ if (argc == (index + 2)) {
+ o = cJSON_Parse(argv[index+1]);
+ if (!o) {
+ fprintf(stderr, "%s: Failed parsing JSON input: %s\n", argv[0], cJSON_GetErrorPtr());
+ exit(-1);
+ }
+ }
+
+ if (filename) {
+ if (argc > index + 1) {
+ fprintf(stderr, "%s: Superfluous arguments when filename given\n", argv[0]);
+ exit(-1);
+ }
+
+ FILE *f = fopen(filename, "r");
+ size_t bufsize = 1024;
+ size_t n_read = 0;
+ size_t n;
+
+ if (!f) {
+ fprintf(stderr, "%s: can't open file: %s\n", argv[0], filename);
+ exit(-1);
+ }
+ char *buf = malloc(bufsize);
+ while ((n = fread(buf, 1, bufsize, f))) {
+ n_read += n;
+ if (n == bufsize)
+ buf = realloc(buf, bufsize);
+ }
+ fclose(f);
+ if (n_read) {
+ o = cJSON_Parse(buf);
+ free(buf);
+ if (!o) {
+ fprintf(stderr, "%s: Failed parsing JSON input: %s\n", argv[0], cJSON_GetErrorPtr());
+ exit(-1);
+ }
+ }
+ }
+
+ if (!o) {
+ fprintf(stderr, "%s: Failed parsing JSON input\n", argv[0]);
+ exit(-1);
+ }
+
+ if (vac_connect("vat2", 0, 0, 1024)) {
+ fprintf(stderr, "Failed connecting to VPP\n");
+ exit(-1);
+ }
+ if (!p) {
+ fprintf(stderr, "No such command\n");
+ exit(-1);
+ }
+
+ cJSON * (*fp) (cJSON *);
+ fp = (void *) p[0];
+ cJSON *r = (*fp) (o);
+
+ if (o)
+ cJSON_Delete(o);
+
+ if (r) {
+ char *output = cJSON_Print(r);
+ cJSON_Delete(r);
+ printf("%s\n", output);
+ free(output);
+ } else {
+ fprintf(stderr, "Call failed\n");
+ exit(-1);
+ }
+
+ vac_disconnect();
+ exit (0);
+
+}