src/vlib/unix/main.c

   1 /*
   2  * Copyright (c) 2015 Cisco and/or its affiliates.
   3  * Licensed under the Apache License, Version 2.0 (the "License");
   4  * you may not use this file except in compliance with the License.
   5  * You may obtain a copy of the License at:
   6  *
   7  *     http://www.apache.org/licenses/LICENSE-2.0
   8  *
   9  * Unless required by applicable law or agreed to in writing, software
  10  * distributed under the License is distributed on an "AS IS" BASIS,
  11  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12  * See the License for the specific language governing permissions and
  13  * limitations under the License.
  14  */
  15 /*
  16  * main.c: Unix main routine
  17  *
  18  * Copyright (c) 2008 Eliot Dresselhaus
  19  *
  20  * Permission is hereby granted, free of charge, to any person obtaining
  21  * a copy of this software and associated documentation files (the
  22  * "Software"), to deal in the Software without restriction, including
  23  * without limitation the rights to use, copy, modify, merge, publish,
  24  * distribute, sublicense, and/or sell copies of the Software, and to
  25  * permit persons to whom the Software is furnished to do so, subject to
  26  * the following conditions:
  27  *
  28  * The above copyright notice and this permission notice shall be
  29  * included in all copies or substantial portions of the Software.
  30  *
  31  *  THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  32  *  EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  33  *  MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  34  *  NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
  35  *  LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
  36  *  OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
  37  *  WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  38  */
  39 #include <vlib/vlib.h>
  40 #include <vlib/unix/unix.h>
  41 #include <vlib/unix/plugin.h>
  42
  43 #include <signal.h>
  44 #include <sys/ucontext.h>
  45 #include <syslog.h>
  46 #include <sys/types.h>
  47 #include <sys/stat.h>
  48 #include <fcntl.h>
  49 #include <sys/time.h>
  50 #include <sys/resource.h>
  51 #include <unistd.h>
  52
  53 /** Default CLI pager limit is not configured in startup.conf */
  54 #define UNIX_CLI_DEFAULT_PAGER_LIMIT 100000
  55
  56 /** Default CLI history depth if not configured in startup.conf */
  57 #define UNIX_CLI_DEFAULT_HISTORY 50
  58
  59 char *vlib_default_runtime_dir __attribute__ ((weak));
  60 char *vlib_default_runtime_dir = "vlib";
  61
  62 unix_main_t unix_main;
  63
  64 static clib_error_t *
  65 unix_main_init (vlib_main_t * vm)
  66 {
  67   unix_main_t *um = &unix_main;
  68   um->vlib_main = vm;
  69   return vlib_call_init_function (vm, unix_input_init);
  70 }
  71
  72 VLIB_INIT_FUNCTION (unix_main_init);
  73
  74 static void
  75 unix_signal_handler (int signum, siginfo_t * si, ucontext_t * uc)
  76 {
  77   uword fatal;
  78   u8 *msg = 0;
  79
  80   msg = format (msg, "received signal %U, PC %U",
  81                 format_signal, signum, format_ucontext_pc, uc);
  82
  83   if (signum == SIGSEGV)
  84     msg = format (msg, ", faulting address %p", si->si_addr);
  85
  86   switch (signum)
  87     {
  88       /* these (caught) signals cause the application to exit */
  89     case SIGTERM:
  90       if (unix_main.vlib_main->main_loop_exit_set)
  91         {
  92           syslog (LOG_ERR | LOG_DAEMON, "received SIGTERM, exiting...");
  93
  94           clib_longjmp (&unix_main.vlib_main->main_loop_exit,
  95                         VLIB_MAIN_LOOP_EXIT_CLI);
  96         }
  97       /* fall through */
  98     case SIGQUIT:
  99     case SIGINT:
 100     case SIGILL:
 101     case SIGBUS:
 102     case SIGSEGV:
 103     case SIGHUP:
 104     case SIGFPE:
 105       fatal = 1;
 106       break;
 107
 108       /* by default, print a message and continue */
 109     default:
 110       fatal = 0;
 111       break;
 112     }
 113
 114   /* Null terminate. */
 115   vec_add1 (msg, 0);
 116
 117   if (fatal)
 118     {
 119       syslog (LOG_ERR | LOG_DAEMON, "%s", msg);
 120       os_exit (1);
 121     }
 122   else
 123     clib_warning ("%s", msg);
 124
 125   vec_free (msg);
 126 }
 127
 128 static clib_error_t *
 129 setup_signal_handlers (unix_main_t * um)
 130 {
 131   uword i;
 132   struct sigaction sa;
 133
 134   for (i = 1; i < 32; i++)
 135     {
 136       memset (&sa, 0, sizeof (sa));
 137       sa.sa_sigaction = (void *) unix_signal_handler;
 138       sa.sa_flags = SA_SIGINFO;
 139
 140       switch (i)
 141         {
 142           /* these signals take the default action */
 143         case SIGABRT:
 144         case SIGKILL:
 145         case SIGSTOP:
 146         case SIGUSR1:
 147         case SIGUSR2:
 148           continue;
 149
 150           /* ignore SIGPIPE, SIGCHLD */
 151         case SIGPIPE:
 152         case SIGCHLD:
 153           sa.sa_sigaction = (void *) SIG_IGN;
 154           break;
 155
 156           /* catch and handle all other signals */
 157         default:
 158           break;
 159         }
 160
 161       if (sigaction (i, &sa, 0) < 0)
 162         return clib_error_return_unix (0, "sigaction %U", format_signal, i);
 163     }
 164
 165   return 0;
 166 }
 167
 168 static void
 169 unix_error_handler (void *arg, u8 * msg, int msg_len)
 170 {
 171   unix_main_t *um = arg;
 172
 173   /* Echo to stderr when interactive. */
 174   if (um->flags & UNIX_FLAG_INTERACTIVE)
 175     {
 176       CLIB_UNUSED (int r) = write (2, msg, msg_len);
 177     }
 178   else
 179     {
 180       char save = msg[msg_len - 1];
 181
 182       /* Null Terminate. */
 183       msg[msg_len - 1] = 0;
 184
 185       syslog (LOG_ERR | LOG_DAEMON, "%s", msg);
 186
 187       msg[msg_len - 1] = save;
 188     }
 189 }
 190
 191 void
 192 vlib_unix_error_report (vlib_main_t * vm, clib_error_t * error)
 193 {
 194   unix_main_t *um = &unix_main;
 195
 196   if (um->flags & UNIX_FLAG_INTERACTIVE || error == 0)
 197     return;
 198
 199   {
 200     char save;
 201     u8 *msg;
 202     u32 msg_len;
 203
 204     msg = error->what;
 205     msg_len = vec_len (msg);
 206
 207     /* Null Terminate. */
 208     save = msg[msg_len - 1];
 209     msg[msg_len - 1] = 0;
 210
 211     syslog (LOG_ERR | LOG_DAEMON, "%s", msg);
 212
 213     msg[msg_len - 1] = save;
 214   }
 215 }
 216
 217 static uword
 218 startup_config_process (vlib_main_t * vm,
 219                         vlib_node_runtime_t * rt, vlib_frame_t * f)
 220 {
 221   unix_main_t *um = &unix_main;
 222   u8 *buf = 0;
 223   uword l, n = 1;
 224
 225   vlib_process_suspend (vm, 2.0);
 226
 227   while (um->unix_config_complete == 0)
 228     vlib_process_suspend (vm, 0.1);
 229
 230   if (um->startup_config_filename)
 231     {
 232       unformat_input_t sub_input;
 233       int fd;
 234       struct stat s;
 235       char *fn = (char *) um->startup_config_filename;
 236
 237       fd = open (fn, O_RDONLY);
 238       if (fd < 0)
 239         {
 240           clib_warning ("failed to open `%s'", fn);
 241           return 0;
 242         }
 243
 244       if (fstat (fd, &s) < 0)
 245         {
 246           clib_warning ("failed to stat `%s'", fn);
 247         bail:
 248           close (fd);
 249           return 0;
 250         }
 251
 252       if (!(S_ISREG (s.st_mode) || S_ISLNK (s.st_mode)))
 253         {
 254           clib_warning ("not a regular file: `%s'", fn);
 255           goto bail;
 256         }
 257
 258       while (n > 0)
 259         {
 260           l = vec_len (buf);
 261           vec_resize (buf, 4096);
 262           n = read (fd, buf + l, 4096);
 263           if (n > 0)
 264             {
 265               _vec_len (buf) = l + n;
 266               if (n < 4096)
 267                 break;
 268             }
 269           else
 270             break;
 271         }
 272       if (um->log_fd && vec_len (buf))
 273         {
 274           u8 *lv = 0;
 275           lv = format (lv, "%U: ***** Startup Config *****\n%v",
 276                        format_timeval, 0 /* current bat-time */ ,
 277                        0 /* current bat-format */ ,
 278                        buf);
 279           {
 280             int rv __attribute__ ((unused)) =
 281               write (um->log_fd, lv, vec_len (lv));
 282           }
 283           vec_reset_length (lv);
 284           lv = format (lv, "%U: ***** End Startup Config *****\n",
 285                        format_timeval, 0 /* current bat-time */ ,
 286                        0 /* current bat-format */ );
 287           {
 288             int rv __attribute__ ((unused)) =
 289               write (um->log_fd, lv, vec_len (lv));
 290           }
 291           vec_free (lv);
 292         }
 293
 294       if (vec_len (buf))
 295         {
 296           unformat_init_vector (&sub_input, buf);
 297           vlib_cli_input (vm, &sub_input, 0, 0);
 298           /* frees buf for us */
 299           unformat_free (&sub_input);
 300         }
 301       close (fd);
 302     }
 303   return 0;
 304 }
 305
 306 /* *INDENT-OFF* */
 307 VLIB_REGISTER_NODE (startup_config_node,static) = {
 308     .function = startup_config_process,
 309     .type = VLIB_NODE_TYPE_PROCESS,
 310     .name = "startup-config-process",
 311 };
 312 /* *INDENT-ON* */
 313
 314 static clib_error_t *
 315 unix_config (vlib_main_t * vm, unformat_input_t * input)
 316 {
 317   unix_main_t *um = &unix_main;
 318   clib_error_t *error = 0;
 319   gid_t gid;
 320
 321   /* Defaults */
 322   um->cli_pager_buffer_limit = UNIX_CLI_DEFAULT_PAGER_LIMIT;
 323   um->cli_history_limit = UNIX_CLI_DEFAULT_HISTORY;
 324
 325   while (unformat_check_input (input) != UNFORMAT_END_OF_INPUT)
 326     {
 327       char *cli_prompt;
 328       if (unformat (input, "interactive"))
 329         um->flags |= UNIX_FLAG_INTERACTIVE;
 330       else if (unformat (input, "nodaemon"))
 331         um->flags |= UNIX_FLAG_NODAEMON;
 332       else if (unformat (input, "cli-prompt %s", &cli_prompt))
 333         vlib_unix_cli_set_prompt (cli_prompt);
 334       else
 335         if (unformat (input, "cli-listen %s", &um->cli_listen_socket.config))
 336         ;
 337       else if (unformat (input, "runtime-dir %s", &um->runtime_dir))
 338         ;
 339       else if (unformat (input, "cli-line-mode"))
 340         um->cli_line_mode = 1;
 341       else if (unformat (input, "cli-no-banner"))
 342         um->cli_no_banner = 1;
 343       else if (unformat (input, "cli-no-pager"))
 344         um->cli_no_pager = 1;
 345       else if (unformat (input, "cli-pager-buffer-limit %d",
 346                          &um->cli_pager_buffer_limit))
 347         ;
 348       else
 349         if (unformat (input, "cli-history-limit %d", &um->cli_history_limit))
 350         ;
 351       else if (unformat (input, "coredump-size"))
 352         {
 353           uword coredump_size = 0;
 354           if (unformat (input, "unlimited"))
 355             {
 356               coredump_size = RLIM_INFINITY;
 357             }
 358           else
 359             if (!unformat (input, "%U", unformat_memory_size, &coredump_size))
 360             {
 361               return clib_error_return (0,
 362                                         "invalid coredump-size parameter `%U'",
 363                                         format_unformat_error, input);
 364             }
 365           const struct rlimit new_limit = { coredump_size, coredump_size };
 366           if (0 != setrlimit (RLIMIT_CORE, &new_limit))
 367             {
 368               clib_unix_warning ("prlimit() failed");
 369             }
 370         }
 371       else if (unformat (input, "full-coredump"))
 372         {
 373           int fd;
 374
 375           fd = open ("/proc/self/coredump_filter", O_WRONLY);
 376           if (fd >= 0)
 377             {
 378               if (write (fd, "0x6f\n", 5) != 5)
 379                 clib_unix_warning ("coredump filter write failed!");
 380               close (fd);
 381             }
 382           else
 383             clib_unix_warning ("couldn't open /proc/self/coredump_filter");
 384         }
 385       else if (unformat (input, "startup-config %s",
 386                          &um->startup_config_filename))
 387         ;
 388       else if (unformat (input, "exec %s", &um->startup_config_filename))
 389         ;
 390       else if (unformat (input, "log %s", &um->log_filename))
 391         {
 392           um->log_fd = open ((char *) um->log_filename,
 393                              O_CREAT | O_WRONLY | O_APPEND, 0644);
 394           if (um->log_fd < 0)
 395             {
 396               clib_warning ("couldn't open log '%s'\n", um->log_filename);
 397               um->log_fd = 0;
 398             }
 399           else
 400             {
 401               u8 *lv = 0;
 402               lv = format (0, "%U: ***** Start: PID %d *****\n",
 403                            format_timeval, 0 /* current bat-time */ ,
 404                            0 /* current bat-format */ ,
 405                            getpid ());
 406               {
 407                 int rv __attribute__ ((unused)) =
 408                   write (um->log_fd, lv, vec_len (lv));
 409               }
 410               vec_free (lv);
 411             }
 412         }
 413       else if (unformat (input, "gid %U", unformat_unix_gid, &gid))
 414         {
 415           if (setegid (gid) == -1)
 416             return clib_error_return_unix (0, "setegid");
 417         }
 418       else
 419         return clib_error_return (0, "unknown input `%U'",
 420                                   format_unformat_error, input);
 421     }
 422
 423   error = setup_signal_handlers (um);
 424   if (error)
 425     return error;
 426
 427   if (!(um->flags & UNIX_FLAG_INTERACTIVE))
 428     {
 429       openlog (vm->name, LOG_CONS | LOG_PERROR | LOG_PID, LOG_DAEMON);
 430       clib_error_register_handler (unix_error_handler, um);
 431
 432       if (!(um->flags & UNIX_FLAG_NODAEMON) && daemon ( /* chdir to / */ 0,
 433                                                        /* stdin/stdout/stderr -> /dev/null */
 434                                                        0) < 0)
 435         clib_error_return (0, "daemon () fails");
 436     }
 437   um->unix_config_complete = 1;
 438
 439   if (um->runtime_dir == 0)
 440     {
 441       uid_t uid = geteuid ();
 442       if (uid == 00)
 443         um->runtime_dir = format (0, "/run/%s%c",
 444                                   vlib_default_runtime_dir, 0);
 445       else
 446         um->runtime_dir = format (0, "/run/user/%u/%s%c", uid,
 447                                   vlib_default_runtime_dir, 0);
 448     }
 449
 450
 451   return 0;
 452 }
 453
 454 /* unix { ... } configuration. */
 455 /*?
 456  *
 457  * @cfgcmd{interactive}
 458  * Attach CLI to stdin/out and provide a debugging command line interface.
 459  * Implies @c nodaemon.
 460  *
 461  * @cfgcmd{nodaemon}
 462  * Do not fork or background the VPP process. Typically used when invoking
 463  * VPP applications from a process monitor.
 464  *
 465  * @cfgcmd{exec, &lt;filename&gt;}
 466  * @par <code>startup-config &lt;filename&gt;</code>
 467  * Read startup operational configuration from @c filename.
 468  * The contents of the file will be performed as though entered at the CLI.
 469  * The two keywords are aliases for the same function; if both are specified,
 470  * only the last will have an effect.
 471  *
 472  * @cfgcmd{log, &lt;filename&gt;}
 473  * Logs the startup configuration and all subsequent CLI commands in
 474  * @c filename.
 475  * Very useful in situations where folks don't remember or can't be bothered
 476  * to include CLI commands in bug reports.
 477  *
 478  * @cfgcmd{full-coredump}
 479  * Ask the Linux kernel to dump all memory-mapped address regions, instead
 480  * of just text+data+bss.
 481  *
 482  * @cfgcmd{runtime-dir}
 483  * Define directory where VPP is going to store all runtime files.
 484  * Default is /run/vpp.
 485  *
 486  * @cfgcmd{cli-listen, &lt;address:port&gt;}
 487  * Bind the CLI to listen at the address and port given. @clocalhost
 488  * on TCP port @c 5002, given as <tt>cli-listen localhost:5002</tt>,
 489  * is typical.
 490  *
 491  * @cfgcmd{cli-line-mode}
 492  * Disable character-by-character I/O on stdin. Useful when combined with,
 493  * for example, <tt>emacs M-x gud-gdb</tt>.
 494  *
 495  * @cfgcmd{cli-prompt, &lt;string&gt;}
 496  * Configure the CLI prompt to be @c string.
 497  *
 498  * @cfgcmd{cli-history-limit, &lt;nn&gt;}
 499  * Limit commmand history to @c nn  lines. A value of @c 0
 500  * disables command history. Default value: @c 50
 501  *
 502  * @cfgcmd{cli-no-banner}
 503  * Disable the login banner on stdin and Telnet connections.
 504  *
 505  * @cfgcmd{cli-no-pager}
 506  * Disable the output pager.
 507  *
 508  * @cfgcmd{cli-pager-buffer-limit, &lt;nn&gt;}
 509  * Limit pager buffer to @c nn lines of output.
 510  * A value of @c 0 disables the pager. Default value: @c 100000
 511 ?*/
 512 VLIB_EARLY_CONFIG_FUNCTION (unix_config, "unix");
 513
 514 static clib_error_t *
 515 unix_exit (vlib_main_t * vm)
 516 {
 517   /* Close syslog connection. */
 518   closelog ();
 519   return 0;
 520 }
 521
 522 VLIB_MAIN_LOOP_EXIT_FUNCTION (unix_exit);
 523
 524 u8 **vlib_thread_stacks;
 525
 526 static uword
 527 thread0 (uword arg)
 528 {
 529   vlib_main_t *vm = (vlib_main_t *) arg;
 530   unformat_input_t input;
 531   int i;
 532
 533   unformat_init_command_line (&input, (char **) vm->argv);
 534   i = vlib_main (vm, &input);
 535   unformat_free (&input);
 536
 537   return i;
 538 }
 539
 540 u8 *
 541 vlib_thread_stack_init (uword thread_index)
 542 {
 543   vec_validate (vlib_thread_stacks, thread_index);
 544   vlib_thread_stacks[thread_index] = clib_mem_alloc_aligned
 545     (VLIB_THREAD_STACK_SIZE, VLIB_THREAD_STACK_SIZE);
 546
 547   /*
 548    * Disallow writes to the bottom page of the stack, to
 549    * catch stack overflows.
 550    */
 551   if (mprotect (vlib_thread_stacks[thread_index],
 552                 clib_mem_get_page_size (), PROT_READ) < 0)
 553     clib_unix_warning ("thread stack");
 554   return vlib_thread_stacks[thread_index];
 555 }
 556
 557 int
 558 vlib_unix_main (int argc, char *argv[])
 559 {
 560   vlib_main_t *vm = &vlib_global_main;  /* one and only time for this! */
 561   unformat_input_t input;
 562   clib_error_t *e;
 563   int i;
 564
 565   vm->argv = (u8 **) argv;
 566   vm->name = argv[0];
 567   vm->heap_base = clib_mem_get_heap ();
 568   ASSERT (vm->heap_base);
 569
 570   unformat_init_command_line (&input, (char **) vm->argv);
 571   if ((e = vlib_plugin_config (vm, &input)))
 572     {
 573       clib_error_report (e);
 574       return 1;
 575     }
 576   unformat_free (&input);
 577
 578   i = vlib_plugin_early_init (vm);
 579   if (i)
 580     return i;
 581
 582   unformat_init_command_line (&input, (char **) vm->argv);
 583   if (vm->init_functions_called == 0)
 584     vm->init_functions_called = hash_create (0, /* value bytes */ 0);
 585   e = vlib_call_all_config_functions (vm, &input, 1 /* early */ );
 586   if (e != 0)
 587     {
 588       clib_error_report (e);
 589       return 1;
 590     }
 591   unformat_free (&input);
 592
 593   vlib_thread_stack_init (0);
 594
 595   __os_thread_index = 0;
 596
 597   i = clib_calljmp (thread0, (uword) vm,
 598                     (void *) (vlib_thread_stacks[0] +
 599                               VLIB_THREAD_STACK_SIZE));
 600   return i;
 601 }
 602
 603 /*
 604  * fd.io coding-style-patch-verification: ON
 605  *
 606  * Local Variables:
 607  * eval: (c-set-style "gnu")
 608  * End:
 609  */