diff options
Diffstat (limited to 'svm/svmdbtool.c')
| -rw-r--r-- | svm/svmdbtool.c | 326 |
1 files changed, 326 insertions, 0 deletions
diff --git a/svm/svmdbtool.c b/svm/svmdbtool.c new file mode 100644 index 00000000000..c2efbc52a34 --- /dev/null +++ b/svm/svmdbtool.c @@ -0,0 +1,326 @@ +/* + * 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 <stdio.h> +#include <stdlib.h> +#include <sys/types.h> +#include <sys/mman.h> +#include <sys/stat.h> +#include <netinet/in.h> +#include <signal.h> +#include <pthread.h> +#include <unistd.h> +#include <time.h> +#include <fcntl.h> +#include <string.h> +#include <vppinfra/clib.h> +#include <vppinfra/vec.h> +#include <vppinfra/hash.h> +#include <vppinfra/bitmap.h> +#include <vppinfra/fifo.h> +#include <vppinfra/time.h> +#include <vppinfra/mheap.h> +#include <vppinfra/heap.h> +#include <vppinfra/pool.h> +#include <vppinfra/format.h> +#include <vppinfra/serialize.h> +#include "svmdb.h" + +static void get_string (char *chroot_path, u8 *vbl) +{ + svmdb_client_t *c; + char *rv; + + c = svmdb_map_chroot(chroot_path); + + rv = svmdb_local_get_string_variable (c, (char *)vbl); + + fformat(stdout, "%s\n", rv ? rv : "UNSET"); + vec_free(rv); + svmdb_unmap (c); +} + +static void set_string (char *chroot_path, u8 *vbl, u8 *value) +{ + svmdb_client_t *c; + + c = svmdb_map_chroot(chroot_path); + svmdb_local_set_string_variable (c, (char *)vbl, (char *)value); + svmdb_unmap (c); +} + +static void unset_string (char *chroot_path, u8 *vbl) +{ + svmdb_client_t *c; + + c = svmdb_map_chroot(chroot_path); + svmdb_local_unset_string_variable (c, (char *)vbl); + svmdb_unmap (c); +} + +static void dump_strings(char *chroot_path) +{ + svmdb_client_t *c; + + c = svmdb_map_chroot(chroot_path); + svmdb_local_dump_strings (c); + svmdb_unmap (c); +} + +static void test_vlib_vec_rate (char *chroot_path, f64 vr) +{ + svmdb_client_t *c; + f64 *tv = 0; + + vec_add1 (tv, vr); + + c = svmdb_map_chroot(chroot_path); + + svmdb_local_set_vec_variable (c, "vlib_vector_rate", (char *)tv, sizeof (*tv)); + svmdb_unmap (c); + + vec_free(tv); +} + + + +static void test_vec (char *chroot_path, u8 *vbl) +{ + svmdb_client_t *c; + u64 *tv = 0; + int i; + + /* my amp goes to 11 */ + for (i = 0; i < 11; i++) { + vec_add1(tv, i); + } + + c = svmdb_map_chroot(chroot_path); + svmdb_local_set_vec_variable (c, (char *)vbl, (char *)tv, sizeof (tv[0])); + svmdb_unmap (c); + + vec_free(tv); +} + +static void fake_install (char *chroot_path, u8 *add_value) +{ + svmdb_client_t *c; + u8 *v = 0; + u8 **values = 0; + u8 *oldvalue; + u8 *value; + int nitems = 0, i; + serialize_main_t m; + + c = svmdb_map_chroot(chroot_path); + + oldvalue = svmdb_local_get_vec_variable (c, "installed_sw", 1); + if (oldvalue) { + unserialize_open_data (&m, oldvalue, vec_len (oldvalue)); + nitems = unserialize_likely_small_unsigned_integer (&m); + for (i = 0; i < nitems; i++) { + unserialize_cstring (&m, (char **)&value); + vec_add1 (values, value); + } + vec_free (v); + } + nitems++; + value = format (0, "%s%c", add_value, 0); + + vec_add1 (values, value); + + fformat (stdout, "Resulting installed_sw vector:\n"); + + serialize_open_vector (&m, v); + serialize_likely_small_unsigned_integer (&m, vec_len (values)); + for (i = 0; i < vec_len (values); i++) { + fformat (stdout, "%s\n", values[i]); + serialize_cstring (&m, (char *)values[i]); + } + + v = serialize_close_vector (&m); + + svmdb_local_set_vec_variable (c, "installed_sw", v, sizeof (v[0])); + svmdb_unmap (c); + + for (i = 0; i < vec_len (values); i++) + vec_free (values[i]); + vec_free (values); +} + +static void sigaction_handler (int signum, siginfo_t *i, void *notused) +{ + u32 action, opaque; + + action = (u32)(uword) i->si_ptr; + action >>= 28; + opaque = (u32)(uword) i->si_ptr; + opaque &= ~(0xF0000000); + + clib_warning ("signal %d, action %d, opaque %x", + signum, action, opaque); +} + +static void test_reg (char *chroot_path, u8 *vbl) +{ + svmdb_client_t *c; + svmdb_notification_args_t args; + svmdb_notification_args_t *a = &args; + struct sigaction sa; + + memset (&sa, 0, sizeof (sa)); + sa.sa_sigaction = sigaction_handler; + sa.sa_flags = SA_SIGINFO; + if (sigaction (SIGUSR2, &sa, 0) < 0) { + clib_unix_warning ("sigaction"); + return; + } + + memset (a, 0, sizeof (*a)); + + c = svmdb_map_chroot(chroot_path); + + a->add_del = 1 /* add */; + a->nspace = SVMDB_NAMESPACE_STRING; + a->var = (char *) vbl; + a->elsize = 1; + a->signum = SIGUSR2; + a->action = SVMDB_ACTION_GET; + a->opaque = 0x0eadbeef; + + svmdb_local_add_del_notification (c, a); + + (void) svmdb_local_get_string_variable (c, (char *)vbl); + + a->add_del = 0; /* del */ + svmdb_local_add_del_notification (c, a); + + + + svmdb_unmap (c); +} + +static void unset_vec (char *chroot_path, u8 *vbl) +{ + svmdb_client_t *c; + + c = svmdb_map_chroot(chroot_path); + svmdb_local_unset_vec_variable (c, (char *)vbl); + svmdb_unmap (c); +} + +static void dump_vecs(char *chroot_path) +{ + svmdb_client_t *c; + + c = svmdb_map_chroot(chroot_path); + svmdb_local_dump_vecs (c); + svmdb_unmap (c); +} + +static void crash_test(char *chroot_path) +{ + svmdb_client_t *c; + + c = svmdb_map_chroot(chroot_path); + + clib_warning ("Grab region mutex and crash deliberately!"); + c->db_rp->mutex_owner_pid = getpid(); + c->db_rp->mutex_owner_tag = -13; + pthread_mutex_lock(&c->db_rp->mutex); + + abort(); +} + +static void map_with_size (char *chroot_path, uword size) +{ + svmdb_client_t *c; + c = svmdb_map_chroot_size (chroot_path, size); + svmdb_unmap (c); +} + +int main (int argc, char **argv) +{ + unformat_input_t input; + int parsed =0; + u8 *vbl=0, *value=0; + char *chroot_path = 0; + u8 *chroot_path_u8; + uword size; + f64 vr; + + unformat_init_command_line (&input, argv); + + while (unformat_check_input (&input) != UNFORMAT_END_OF_INPUT) { + if (unformat (&input, "get-string %s", &vbl)) { + get_string (chroot_path, vbl); + vec_free(vbl); + parsed++; + } else if (unformat (&input, "set-string %s %s", &vbl, &value)) { + set_string (chroot_path, vbl, value); + vec_free(vbl); + vec_free(value); + parsed++; + } else if (unformat(&input, "unset-string %s", &vbl)) { + unset_string (chroot_path, vbl); + vec_free(vbl); + parsed++; + } else if (unformat(&input, "dump-strings")) { + dump_strings(chroot_path); + parsed++; + } else if (unformat(&input, "unset-vec %s", &vbl)) { + unset_vec (chroot_path, vbl); + vec_free(vbl); + parsed++; + } else if (unformat(&input, "dump-vecs")) { + dump_vecs(chroot_path); + parsed++; + } else if (unformat (&input, "test-vec %s", &vbl)) { + test_vec(chroot_path, vbl); + // vec_free(vbl); + parsed++; + } else if (unformat (&input, "vlib-vec-rate %f", &vr)) { + test_vlib_vec_rate(chroot_path, vr); + parsed++; + } else if (unformat (&input, "test-reg %s", &vbl)) { + test_reg(chroot_path, vbl); + parsed++; + } else if (unformat (&input, "crash-test")) { + crash_test (chroot_path); + } else if (unformat (&input, "chroot %s", &chroot_path_u8)) { + chroot_path = (char *) chroot_path_u8; + } else if (unformat (&input, "fake-install %s", &value)) { + fake_install (chroot_path, value); + parsed++; + } else if (unformat (&input, "size %d", &size)) { + map_with_size (chroot_path, size); + parsed++; + } else { + break; + } + } + + unformat_free (&input); + + if (!parsed) { + fformat(stdout, "%s: get-string <name> | set-string <name> <value>\n", + argv[0]); + fformat(stdout, " unset-string <name> | dump-strings\n"); + fformat(stdout, " test-vec <name> |\n"); + fformat(stdout, " unset-vec <name> | dump-vecs\n"); + } + + exit (0); +} |