diff options
Diffstat (limited to 'vpp-api-test/vat')
-rw-r--r-- | vpp-api-test/vat/restart.c | 234 |
1 files changed, 234 insertions, 0 deletions
diff --git a/vpp-api-test/vat/restart.c b/vpp-api-test/vat/restart.c new file mode 100644 index 00000000..d94ddf89 --- /dev/null +++ b/vpp-api-test/vat/restart.c @@ -0,0 +1,234 @@ +/* + * 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 <time.h> +#include <sys/types.h> +#include <signal.h> +#include <sys/stat.h> +#include <unistd.h> + +#include <vppinfra/clib.h> +#include <vppinfra/vec.h> +#include <vppinfra/hash.h> +#include <svmdb.h> +#include <vppinfra/format.h> +#include <vppinfra/error.h> +#include <vppinfra/time.h> +#include <vppinfra/macros.h> + +int restart_main_fn (unformat_input_t * i) +{ + int verbose = 0; + int old_pid; + int wait; + u8 * chroot_path = 0; + svmdb_client_t * svmdb_client; + volatile pid_t *pidp; + struct stat statb; + ino_t old_inode; + int sleeps; + + struct timespec _req, *req = &_req; + struct timespec _rem, *rem = &_rem; + + if (geteuid()) + clib_error ("vpp_restart: must be root..."); + + while (unformat_check_input (i) != UNFORMAT_END_OF_INPUT) + { + if (unformat (i, "verbose") || unformat (i, "v")) + verbose = 1; + else if (unformat (i, "chroot %s", &chroot_path)) + ; + else + { + clib_error ("unknown input `%U'", format_unformat_error, i); + return 1; + } + } + + /* + * Step 1: look up the current VPP pid in the shared-memory database + */ + svmdb_client = svmdb_map_chroot ((char *) chroot_path); + + pidp = svmdb_local_get_variable_reference (svmdb_client, + SVMDB_NAMESPACE_VEC, + "vpp_pid"); + if (pidp == 0) + { + clib_error ("'vpp_pid' svm variable not found, vpp has never run?"); + return 2; + } + + /* Spin for up to 10 seconds for vpp to start */ + for (wait = 0; wait < 1000; wait++) + { + req->tv_sec = 0; + req->tv_nsec = 10000*1000; /* 10 ms */ + while (nanosleep(req, rem) < 0) + *req = *rem; + + if (*pidp) + goto found2; + } + + clib_error ("VPP not runnning..."); + return 3; + + found2: + + old_pid = *pidp; + + /* + * Step 2: sanity check the pid we discovered + */ + if (verbose) + fformat(stdout, "Sanity check current vpp pid %d\n", old_pid); + + if (kill (old_pid, 0) < 0) + { + svmdb_unmap (svmdb_client); + clib_error ("vpp current pid %d not running...", old_pid); + return 2; + } + + if (verbose) + fformat(stdout, "Sanity check vpp pid %d OK\n", old_pid); + + /* + * Step 3: figure out the current vpp <--> client shared-VM file + * inode number + */ + if (stat("/dev/shm/vpe-api", &statb) < 0) + { + clib_unix_error ("stat fail"); + return 4; + } + + old_inode = statb.st_ino; + + if (verbose) + fformat(stdout, "Old inode %u\n", old_inode); + + /* Note: restart wipes out the shared VM database*/ + svmdb_unmap (svmdb_client); + + /* + * Step 4: send SIGTERM to vpp. + * systemd et al. will restart vpp after wiping out the shared-VM + * database and (crucially) the shared API messaging segment + */ + + if (kill (old_pid, SIGTERM) < 0) + { + clib_unix_error ("SIGTERM fail"); + return 3; + } + + sleeps = 0; + + /* + * Step 5: wait up to 15 seconds for a new incarnation of + * the shared-VM API segment to appear. + */ + for (wait = 0; wait < 150; wait++) + { + if ((stat("/dev/shm/vpe-api", &statb) < 0) + || statb.st_ino == old_inode) + { + req->tv_sec = 0; + req->tv_nsec = 100000*1000; /* 100 ms */ + while (nanosleep(req, rem) < 0) + *req = *rem; + sleeps++; + } + else + goto new_inode; + } + + clib_error ("Timeout waiting for new inode to appear..."); + return 5; + + new_inode: + if (verbose && sleeps > 0) + fformat(stdout, "Inode sleeps %d\n", sleeps); + + if (verbose) + fformat(stdout, "New inode %u\n", statb.st_ino); + + /* + * Step 6: remap the SVM database + */ + svmdb_client = svmdb_map_chroot ((char *) chroot_path); + + pidp = svmdb_local_get_variable_reference (svmdb_client, + SVMDB_NAMESPACE_VEC, + "vpp_pid"); + if (pidp == 0) + { + clib_error ("post_restart: 'vpp_pid' svm variable not found," + "vpp did not restart?"); + return 2; + } + + sleeps = 0; + + /* + * Step 7: wait for vpp to publish its new PID + */ + + /* Spin for up to 15 seconds */ + for (wait = 0; wait < 150; wait++) + { + if (*pidp && (*pidp != old_pid)) + goto restarted; + req->tv_sec = 0; + req->tv_nsec = 100000*1000; /* 100 ms */ + while (nanosleep(req, rem) < 0) + *req = *rem; + sleeps++; + } + + clib_error ("Timeout waiting for vpp to publish pid after restart..."); + return 4; + + restarted: + + /* Done... */ + + if (verbose && sleeps) + fformat(stdout, "pid sleeps %d\n", sleeps); + + if (verbose) + fformat (stdout, "New PID %d... Restarted...\n", *pidp); + + svmdb_unmap (svmdb_client); + return 0; +} + +int main (int argc, char **argv) +{ + unformat_input_t i; + int ret; + + clib_mem_init (0, 64ULL<<20); + + unformat_init_command_line (&i, argv); + ret = restart_main_fn (&i); + unformat_free (&i); + return ret; +} |