src/vlib/unix/cli.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  * cli.c: Unix stdin/socket CLI.
  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 /**
  40  * @file
  41  * @brief Unix stdin/socket command line interface.
  42  * Provides a command line interface so humans can interact with VPP.
  43  * This is predominantly a debugging and testing mechanism.
  44  */
  45 /*? %%clicmd:group_label Command line session %% ?*/
  46 /*? %%syscfg:group_label Command line session %% ?*/
  47
  48 #include <vlib/vlib.h>
  49 #include <vlib/unix/unix.h>
  50
  51 #include <ctype.h>
  52 #include <fcntl.h>
  53 #include <sys/stat.h>
  54 #include <termios.h>
  55 #include <signal.h>
  56 #include <unistd.h>
  57 #include <arpa/telnet.h>
  58 #include <sys/ioctl.h>
  59 #include <sys/types.h>
  60 #include <unistd.h>
  61 #include <limits.h>
  62 #include <netinet/tcp.h>
  63 #include <math.h>
  64 #include <vppinfra/macros.h>
  65
  66 /** ANSI escape code. */
  67 #define ESC "\x1b"
  68
  69 /** ANSI Control Sequence Introducer. */
  70 #define CSI ESC "["
  71
  72 /** ANSI clear screen. */
  73 #define ANSI_CLEAR      CSI "2J" CSI "1;1H"
  74 /** ANSI reset color settings. */
  75 #define ANSI_RESET      CSI "0m"
  76 /** ANSI Start bold text. */
  77 #define ANSI_BOLD       CSI "1m"
  78 /** ANSI Stop bold text. */
  79 #define ANSI_DIM        CSI "2m"
  80 /** ANSI Start dark red text. */
  81 #define ANSI_DRED       ANSI_DIM CSI "31m"
  82 /** ANSI Start bright red text. */
  83 #define ANSI_BRED       ANSI_BOLD CSI "31m"
  84 /** ANSI clear line cursor is on. */
  85 #define ANSI_CLEARLINE  CSI "2K"
  86 /** ANSI scroll screen down one line. */
  87 #define ANSI_SCROLLDN   CSI "1T"
  88 /** ANSI save cursor position. */
  89 #define ANSI_SAVECURSOR CSI "s"
  90 /** ANSI restore cursor position if previously saved. */
  91 #define ANSI_RESTCURSOR CSI "u"
  92
  93 /** Maximum depth into a byte stream from which to compile a Telnet
  94  * protocol message. This is a safety measure. */
  95 #define UNIX_CLI_MAX_DEPTH_TELNET 32
  96
  97 /** Maximum terminal width we will accept */
  98 #define UNIX_CLI_MAX_TERMINAL_WIDTH 512
  99 /** Maximum terminal height we will accept */
 100 #define UNIX_CLI_MAX_TERMINAL_HEIGHT 512
 101 /** Default terminal height */
 102 #define UNIX_CLI_DEFAULT_TERMINAL_HEIGHT 24
 103 /** Default terminal width */
 104 #define UNIX_CLI_DEFAULT_TERMINAL_WIDTH 80
 105
 106 /** A CLI banner line. */
 107 typedef struct
 108 {
 109   u8 *line;     /**< The line to print. */
 110   u32 length;   /**< The length of the line without terminating NUL. */
 111 } unix_cli_banner_t;
 112
 113 #define _(a) { .line = (u8 *)(a), .length = sizeof(a) - 1 }
 114 /** Plain welcome banner. */
 115 static unix_cli_banner_t unix_cli_banner[] = {
 116   _("    _______    _        _   _____  ___ \n"),
 117   _(" __/ __/ _ \\  (_)__    | | / / _ \\/ _ \\\n"),
 118   _(" _/ _// // / / / _ \\   | |/ / ___/ ___/\n"),
 119   _(" /_/ /____(_)_/\\___/   |___/_/  /_/    \n"),
 120   _("\n")
 121 };
 122
 123 /** ANSI color welcome banner. */
 124 static unix_cli_banner_t unix_cli_banner_color[] = {
 125   _(ANSI_BRED "    _______    _     " ANSI_RESET "   _   _____  ___ \n"),
 126   _(ANSI_BRED " __/ __/ _ \\  (_)__ " ANSI_RESET "   | | / / _ \\/ _ \\\n"),
 127   _(ANSI_BRED " _/ _// // / / / _ \\" ANSI_RESET "   | |/ / ___/ ___/\n"),
 128   _(ANSI_BRED " /_/ /____(_)_/\\___/" ANSI_RESET "   |___/_/  /_/    \n"),
 129   _("\n")
 130 };
 131
 132 #undef _
 133
 134 /** Pager line index */
 135 typedef struct
 136 {
 137   /** Index into pager_vector */
 138   u32 line;
 139
 140   /** Offset of the string in the line */
 141   u32 offset;
 142
 143   /** Length of the string in the line */
 144   u32 length;
 145 } unix_cli_pager_index_t;
 146
 147
 148 /** Unix CLI session. */
 149 typedef struct
 150 {
 151   /** The file index held by unix.c */
 152   u32 clib_file_index;
 153
 154   /** Vector of output pending write to file descriptor. */
 155   u8 *output_vector;
 156
 157   /** Vector of input saved by Unix input node to be processed by
 158      CLI process. */
 159   u8 *input_vector;
 160
 161   /** This session has command history. */
 162   u8 has_history;
 163   /** Array of vectors of commands in the history. */
 164   u8 **command_history;
 165   /** The command currently pointed at by the history cursor. */
 166   u8 *current_command;
 167   /** How far from the end of the history array the user has browsed. */
 168   i32 excursion;
 169
 170   /** Maximum number of history entries this session will store. */
 171   u32 history_limit;
 172
 173   /** Current command line counter */
 174   u32 command_number;
 175
 176   /** The string being searched for in the history. */
 177   u8 *search_key;
 178   /** If non-zero then the CLI is searching in the history array.
 179    * - @c -1 means search backwards.
 180    * - @c 1 means search forwards.
 181    */
 182   int search_mode;
 183
 184   /** Position of the insert cursor on the current input line */
 185   u32 cursor;
 186
 187   /** Line mode or char mode */
 188   u8 line_mode;
 189
 190   /** Set if the CRLF mode wants CR + LF */
 191   u8 crlf_mode;
 192
 193   /** Can we do ANSI output? */
 194   u8 ansi_capable;
 195
 196   /** Has the session started? */
 197   u8 started;
 198
 199   /** Disable the pager? */
 200   u8 no_pager;
 201
 202   /** Whether the session is interactive or not.
 203    * Controls things like initial banner, the CLI prompt etc.  */
 204   u8 is_interactive;
 205
 206   /** Whether the session is attached to a socket. */
 207   u8 is_socket;
 208
 209   /** If EPIPE has been detected, prevent further write-related
 210    * activity on the descriptor.
 211    */
 212   u8 has_epipe;
 213
 214   /** Pager buffer */
 215   u8 **pager_vector;
 216
 217   /** Index of line fragments in the pager buffer */
 218   unix_cli_pager_index_t *pager_index;
 219
 220   /** Line number of top of page */
 221   u32 pager_start;
 222
 223   /** Terminal width */
 224   u32 width;
 225
 226   /** Terminal height */
 227   u32 height;
 228
 229   /** Process node identifier */
 230   u32 process_node_index;
 231
 232   /** The current direction of cursor travel.
 233    *  This is important since when advancing left-to-right, at the
 234    *  right hand edge of the console the terminal typically defers
 235    *  wrapping the cursor to the next line until a character is
 236    *  actually displayed.
 237    *  This messes up our heuristic for whether to use ANSI to return
 238    *  the cursor to the end of the line and instead we have to
 239    *  nudge the cursor to the next line.
 240    *  A Value of @c 0 means we're advancing left-to-right; @c 1 means
 241    *  the opposite.
 242    */
 243   u8 cursor_direction;
 244
 245   /** Macro tables for this session */
 246   clib_macro_main_t macro_main;
 247 } unix_cli_file_t;
 248
 249 /** Resets the pager buffer and other data.
 250  * @param f The CLI session whose pager needs to be reset.
 251  */
 252 always_inline void
 253 unix_cli_pager_reset (unix_cli_file_t * f)
 254 {
 255   u8 **p;
 256
 257   f->pager_start = 0;
 258
 259   vec_free (f->pager_index);
 260   f->pager_index = 0;
 261
 262   vec_foreach (p, f->pager_vector)
 263   {
 264     vec_free (*p);
 265   }
 266   vec_free (f->pager_vector);
 267   f->pager_vector = 0;
 268 }
 269
 270 /** Release storage used by a CLI session.
 271  * @param f The CLI session whose storage needs to be released.
 272  */
 273 always_inline void
 274 unix_cli_file_free (unix_cli_file_t * f)
 275 {
 276   vec_free (f->output_vector);
 277   vec_free (f->input_vector);
 278   unix_cli_pager_reset (f);
 279 }
 280
 281 /** CLI actions */
 282 typedef enum
 283 {
 284   UNIX_CLI_PARSE_ACTION_NOACTION = 0,   /**< No action */
 285   UNIX_CLI_PARSE_ACTION_CRLF,           /**< Carriage return, newline or enter */
 286   UNIX_CLI_PARSE_ACTION_TAB,            /**< Tab key */
 287   UNIX_CLI_PARSE_ACTION_ERASE,          /**< Erase cursor left */
 288   UNIX_CLI_PARSE_ACTION_ERASERIGHT,     /**< Erase cursor right */
 289   UNIX_CLI_PARSE_ACTION_UP,             /**< Up arrow */
 290   UNIX_CLI_PARSE_ACTION_DOWN,           /**< Down arrow */
 291   UNIX_CLI_PARSE_ACTION_LEFT,           /**< Left arrow */
 292   UNIX_CLI_PARSE_ACTION_RIGHT,          /**< Right arrow */
 293   UNIX_CLI_PARSE_ACTION_HOME,           /**< Home key (jump to start of line) */
 294   UNIX_CLI_PARSE_ACTION_END,            /**< End key (jump to end of line) */
 295   UNIX_CLI_PARSE_ACTION_WORDLEFT,       /**< Jump cursor to start of left word */
 296   UNIX_CLI_PARSE_ACTION_WORDRIGHT,      /**< Jump cursor to start of right word */
 297   UNIX_CLI_PARSE_ACTION_ERASELINELEFT,  /**< Erase line to left of cursor */
 298   UNIX_CLI_PARSE_ACTION_ERASELINERIGHT, /**< Erase line to right & including cursor */
 299   UNIX_CLI_PARSE_ACTION_ERASEWORDLEFT,  /**< Erase word left */
 300   UNIX_CLI_PARSE_ACTION_CLEAR,          /**< Clear the terminal */
 301   UNIX_CLI_PARSE_ACTION_REVSEARCH,      /**< Search backwards in command history */
 302   UNIX_CLI_PARSE_ACTION_FWDSEARCH,      /**< Search forwards in command history */
 303   UNIX_CLI_PARSE_ACTION_YANK,           /**< Undo last erase action */
 304   UNIX_CLI_PARSE_ACTION_TELNETIAC,      /**< Telnet control code */
 305
 306   UNIX_CLI_PARSE_ACTION_PAGER_CRLF,     /**< Enter pressed (CR, CRLF, LF, etc) */
 307   UNIX_CLI_PARSE_ACTION_PAGER_QUIT,     /**< Exit the pager session */
 308   UNIX_CLI_PARSE_ACTION_PAGER_NEXT,     /**< Scroll to next page */
 309   UNIX_CLI_PARSE_ACTION_PAGER_DN,       /**< Scroll to next line */
 310   UNIX_CLI_PARSE_ACTION_PAGER_UP,       /**< Scroll to previous line */
 311   UNIX_CLI_PARSE_ACTION_PAGER_TOP,      /**< Scroll to first line */
 312   UNIX_CLI_PARSE_ACTION_PAGER_BOTTOM,   /**< Scroll to last line */
 313   UNIX_CLI_PARSE_ACTION_PAGER_PGDN,     /**< Scroll to next page */
 314   UNIX_CLI_PARSE_ACTION_PAGER_PGUP,     /**< Scroll to previous page */
 315   UNIX_CLI_PARSE_ACTION_PAGER_REDRAW,   /**< Clear and redraw the page on the terminal */
 316   UNIX_CLI_PARSE_ACTION_PAGER_SEARCH,   /**< Search the pager buffer */
 317
 318   UNIX_CLI_PARSE_ACTION_PARTIALMATCH,   /**< Action parser found a partial match */
 319   UNIX_CLI_PARSE_ACTION_NOMATCH         /**< Action parser did not find any match */
 320 } unix_cli_parse_action_t;
 321
 322 /** @brief Mapping of input buffer strings to action values.
 323  * @note This won't work as a hash since we need to be able to do
 324  *       partial matches on the string.
 325  */
 326 typedef struct
 327 {
 328   u8 *input;                        /**< Input string to match. */
 329   u32 len;                          /**< Length of input without final NUL. */
 330   unix_cli_parse_action_t action;   /**< Action to take when matched. */
 331 } unix_cli_parse_actions_t;
 332
 333 /** @brief Given a capital ASCII letter character return a @c NUL terminated
 334  * string with the control code for that letter.
 335  *
 336  * @param c An ASCII character.
 337  * @return A @c NUL terminated string of type @c u8[].
 338  *
 339  * @par Example
 340  *     @c CTL('A') returns <code>{ 0x01, 0x00 }</code> as a @c u8[].
 341  */
 342 #define CTL(c) (u8[]){ (c) - '@', 0 }
 343
 344 #define _(a,b) { .input = (u8 *)(a), .len = sizeof(a) - 1, .action = (b) }
 345 /**
 346  * Patterns to match on a CLI input stream.
 347  * @showinitializer
 348  */
 349 static unix_cli_parse_actions_t unix_cli_parse_strings[] = {
 350   /* Line handling */
 351   _("\r\n", UNIX_CLI_PARSE_ACTION_CRLF),        /* Must be before '\r' */
 352   _("\n", UNIX_CLI_PARSE_ACTION_CRLF),
 353   _("\r\0", UNIX_CLI_PARSE_ACTION_CRLF),        /* Telnet does this */
 354   _("\r", UNIX_CLI_PARSE_ACTION_CRLF),
 355
 356   /* Unix shell control codes */
 357   _(CTL ('B'), UNIX_CLI_PARSE_ACTION_LEFT),
 358   _(CTL ('F'), UNIX_CLI_PARSE_ACTION_RIGHT),
 359   _(CTL ('P'), UNIX_CLI_PARSE_ACTION_UP),
 360   _(CTL ('N'), UNIX_CLI_PARSE_ACTION_DOWN),
 361   _(CTL ('A'), UNIX_CLI_PARSE_ACTION_HOME),
 362   _(CTL ('E'), UNIX_CLI_PARSE_ACTION_END),
 363   _(CTL ('D'), UNIX_CLI_PARSE_ACTION_ERASERIGHT),
 364   _(CTL ('U'), UNIX_CLI_PARSE_ACTION_ERASELINELEFT),
 365   _(CTL ('K'), UNIX_CLI_PARSE_ACTION_ERASELINERIGHT),
 366   _(CTL ('W'), UNIX_CLI_PARSE_ACTION_ERASEWORDLEFT),
 367   _(CTL ('Y'), UNIX_CLI_PARSE_ACTION_YANK),
 368   _(CTL ('L'), UNIX_CLI_PARSE_ACTION_CLEAR),
 369   _(ESC "b", UNIX_CLI_PARSE_ACTION_WORDLEFT),   /* Alt-B */
 370   _(ESC "f", UNIX_CLI_PARSE_ACTION_WORDRIGHT),  /* Alt-F */
 371   _("\b", UNIX_CLI_PARSE_ACTION_ERASE), /* ^H */
 372   _("\x7f", UNIX_CLI_PARSE_ACTION_ERASE),       /* Backspace */
 373   _("\t", UNIX_CLI_PARSE_ACTION_TAB),   /* ^I */
 374
 375   /* VT100 Normal mode - Broadest support */
 376   _(CSI "A", UNIX_CLI_PARSE_ACTION_UP),
 377   _(CSI "B", UNIX_CLI_PARSE_ACTION_DOWN),
 378   _(CSI "C", UNIX_CLI_PARSE_ACTION_RIGHT),
 379   _(CSI "D", UNIX_CLI_PARSE_ACTION_LEFT),
 380   _(CSI "H", UNIX_CLI_PARSE_ACTION_HOME),
 381   _(CSI "F", UNIX_CLI_PARSE_ACTION_END),
 382   _(CSI "3~", UNIX_CLI_PARSE_ACTION_ERASERIGHT),        /* Delete */
 383   _(CSI "1;5D", UNIX_CLI_PARSE_ACTION_WORDLEFT),        /* C-Left */
 384   _(CSI "1;5C", UNIX_CLI_PARSE_ACTION_WORDRIGHT),       /* C-Right */
 385
 386   /* VT100 Application mode - Some Gnome Terminal functions use these */
 387   _(ESC "OA", UNIX_CLI_PARSE_ACTION_UP),
 388   _(ESC "OB", UNIX_CLI_PARSE_ACTION_DOWN),
 389   _(ESC "OC", UNIX_CLI_PARSE_ACTION_RIGHT),
 390   _(ESC "OD", UNIX_CLI_PARSE_ACTION_LEFT),
 391   _(ESC "OH", UNIX_CLI_PARSE_ACTION_HOME),
 392   _(ESC "OF", UNIX_CLI_PARSE_ACTION_END),
 393
 394   /* ANSI X3.41-1974 - sent by Microsoft Telnet and PuTTY */
 395   _(CSI "1~", UNIX_CLI_PARSE_ACTION_HOME),
 396   _(CSI "4~", UNIX_CLI_PARSE_ACTION_END),
 397
 398   /* Emacs-ish history search */
 399   _(CTL ('S'), UNIX_CLI_PARSE_ACTION_FWDSEARCH),
 400   _(CTL ('R'), UNIX_CLI_PARSE_ACTION_REVSEARCH),
 401
 402   /* Other protocol things */
 403   _("\xff", UNIX_CLI_PARSE_ACTION_TELNETIAC),   /* IAC */
 404   _("\0", UNIX_CLI_PARSE_ACTION_NOACTION),      /* NUL */
 405   _(NULL, UNIX_CLI_PARSE_ACTION_NOMATCH)
 406 };
 407
 408 /**
 409  * Patterns to match when a CLI session is in the pager.
 410  * @showinitializer
 411  */
 412 static unix_cli_parse_actions_t unix_cli_parse_pager[] = {
 413   /* Line handling */
 414   _("\r\n", UNIX_CLI_PARSE_ACTION_PAGER_CRLF),  /* Must be before '\r' */
 415   _("\n", UNIX_CLI_PARSE_ACTION_PAGER_CRLF),
 416   _("\r\0", UNIX_CLI_PARSE_ACTION_PAGER_CRLF),  /* Telnet does this */
 417   _("\r", UNIX_CLI_PARSE_ACTION_PAGER_CRLF),
 418
 419   /* Pager commands */
 420   _(" ", UNIX_CLI_PARSE_ACTION_PAGER_NEXT),
 421   _("q", UNIX_CLI_PARSE_ACTION_PAGER_QUIT),
 422   _(CTL ('L'), UNIX_CLI_PARSE_ACTION_PAGER_REDRAW),
 423   _(CTL ('R'), UNIX_CLI_PARSE_ACTION_PAGER_REDRAW),
 424   _("/", UNIX_CLI_PARSE_ACTION_PAGER_SEARCH),
 425
 426   /* VT100 */
 427   _(CSI "A", UNIX_CLI_PARSE_ACTION_PAGER_UP),
 428   _(CSI "B", UNIX_CLI_PARSE_ACTION_PAGER_DN),
 429   _(CSI "H", UNIX_CLI_PARSE_ACTION_PAGER_TOP),
 430   _(CSI "F", UNIX_CLI_PARSE_ACTION_PAGER_BOTTOM),
 431
 432   /* VT100 Application mode */
 433   _(ESC "OA", UNIX_CLI_PARSE_ACTION_PAGER_UP),
 434   _(ESC "OB", UNIX_CLI_PARSE_ACTION_PAGER_DN),
 435   _(ESC "OH", UNIX_CLI_PARSE_ACTION_PAGER_TOP),
 436   _(ESC "OF", UNIX_CLI_PARSE_ACTION_PAGER_BOTTOM),
 437
 438   /* ANSI X3.41-1974 */
 439   _(CSI "1~", UNIX_CLI_PARSE_ACTION_PAGER_TOP),
 440   _(CSI "4~", UNIX_CLI_PARSE_ACTION_PAGER_BOTTOM),
 441   _(CSI "5~", UNIX_CLI_PARSE_ACTION_PAGER_PGUP),
 442   _(CSI "6~", UNIX_CLI_PARSE_ACTION_PAGER_PGDN),
 443
 444   /* Other protocol things */
 445   _("\xff", UNIX_CLI_PARSE_ACTION_TELNETIAC),   /* IAC */
 446   _("\0", UNIX_CLI_PARSE_ACTION_NOACTION),      /* NUL */
 447   _(NULL, UNIX_CLI_PARSE_ACTION_NOMATCH)
 448 };
 449
 450 #undef _
 451
 452 /** CLI session telnet negotiation timer events. */
 453 typedef enum
 454 {
 455   UNIX_CLI_NEW_SESSION_EVENT_ADD, /**< Add a CLI session to the new session list */
 456 } unix_cli_timeout_event_type_t;
 457
 458 /** Each new session is stored on a list with a deadline after which
 459  * a prompt is issued, in case the session TELNET negotiation fails to
 460  * complete. */
 461 typedef struct
 462 {
 463   uword cf_index; /**< Session index of the new session. */
 464   f64 deadline;   /**< Deadline after which the new session must have a prompt. */
 465 } unix_cli_new_session_t;
 466
 467 /** CLI global state. */
 468 typedef struct
 469 {
 470   /** Prompt string for CLI. */
 471   u8 *cli_prompt;
 472
 473   /** Vec pool of CLI sessions. */
 474   unix_cli_file_t *cli_file_pool;
 475
 476   /** Vec pool of unused session indices. */
 477   u32 *unused_cli_process_node_indices;
 478
 479   /** The session index of the stdin cli */
 480   u32 stdin_cli_file_index;
 481
 482   /** File pool index of current input. */
 483   u32 current_input_file_index;
 484
 485   /** New session process node identifier */
 486   u32 new_session_process_node_index;
 487
 488   /** List of new sessions */
 489   unix_cli_new_session_t *new_sessions;
 490
 491   /** system default macro table */
 492   clib_macro_main_t macro_main;
 493
 494 } unix_cli_main_t;
 495
 496 /** CLI global state */
 497 static unix_cli_main_t unix_cli_main;
 498
 499 /** Return the macro main / tables we should use for this session
 500  */
 501 static clib_macro_main_t *
 502 get_macro_main (void)
 503 {
 504   unix_cli_main_t *cm = &unix_cli_main;
 505   vlib_main_t *vm = vlib_get_main ();
 506   vlib_process_t *cp = vlib_get_current_process (vm);
 507   unix_cli_file_t *cf;
 508
 509   if (pool_is_free_index (cm->cli_file_pool, cp->output_function_arg))
 510     return (&cm->macro_main);
 511
 512   cf = pool_elt_at_index (cm->cli_file_pool, cp->output_function_arg);
 513
 514   return (&cf->macro_main);
 515 }
 516
 517 /**
 518  * @brief Search for a byte sequence in the action list.
 519  *
 520  * Searches the @ref unix_cli_parse_actions_t list in @a a for a match with
 521  * the bytes in @a input of maximum length @a ilen bytes.
 522  * When a match is made @a *matched indicates how many bytes were matched.
 523  * Returns a value from the enum @ref unix_cli_parse_action_t to indicate
 524  * whether no match was found, a partial match was found or a complete
 525  * match was found and what action, if any, should be taken.
 526  *
 527  * @param[in]  a        Actions list to search within.
 528  * @param[in]  input    String fragment to search for.
 529  * @param[in]  ilen     Length of the string in 'input'.
 530  * @param[out] matched  Pointer to an integer that will contain the number
 531  *                      of bytes matched when a complete match is found.
 532  *
 533  * @return Action from @ref unix_cli_parse_action_t that the string fragment
 534  *         matches.
 535  *         @ref UNIX_CLI_PARSE_ACTION_PARTIALMATCH is returned when the
 536  *         whole input string matches the start of at least one action.
 537  *         @ref UNIX_CLI_PARSE_ACTION_NOMATCH is returned when there is no
 538  *         match at all.
 539  */
 540 static unix_cli_parse_action_t
 541 unix_cli_match_action (unix_cli_parse_actions_t * a,
 542                        u8 * input, u32 ilen, i32 * matched)
 543 {
 544   u8 partial = 0;
 545
 546   while (a->input)
 547     {
 548       if (ilen >= a->len)
 549         {
 550           /* see if the start of the input buffer exactly matches the current
 551            * action string. */
 552           if (memcmp (input, a->input, a->len) == 0)
 553             {
 554               *matched = a->len;
 555               return a->action;
 556             }
 557         }
 558       else
 559         {
 560           /* if the first ilen characters match, flag this as a partial -
 561            * meaning keep collecting bytes in case of a future match */
 562           if (memcmp (input, a->input, ilen) == 0)
 563             partial = 1;
 564         }
 565
 566       /* check next action */
 567       a++;
 568     }
 569
 570   return partial ?
 571     UNIX_CLI_PARSE_ACTION_PARTIALMATCH : UNIX_CLI_PARSE_ACTION_NOMATCH;
 572 }
 573
 574
 575 /** Add bytes to the output vector and then flagg the I/O system that bytes
 576  * are available to be sent.
 577  */
 578 static void
 579 unix_cli_add_pending_output (clib_file_t * uf,
 580                              unix_cli_file_t * cf,
 581                              u8 * buffer, uword buffer_bytes)
 582 {
 583   clib_file_main_t *fm = &file_main;
 584
 585   vec_add (cf->output_vector, buffer, buffer_bytes);
 586   if (vec_len (cf->output_vector) > 0)
 587     {
 588       int skip_update = 0 != (uf->flags & UNIX_FILE_DATA_AVAILABLE_TO_WRITE);
 589       uf->flags |= UNIX_FILE_DATA_AVAILABLE_TO_WRITE;
 590       if (!skip_update)
 591         fm->file_update (uf, UNIX_FILE_UPDATE_MODIFY);
 592     }
 593 }
 594
 595 /** Delete all bytes from the output vector and flag the I/O system
 596  * that no more bytes are available to be sent.
 597  */
 598 static void
 599 unix_cli_del_pending_output (clib_file_t * uf,
 600                              unix_cli_file_t * cf, uword n_bytes)
 601 {
 602   clib_file_main_t *fm = &file_main;
 603
 604   vec_delete (cf->output_vector, n_bytes, 0);
 605   if (vec_len (cf->output_vector) <= 0)
 606     {
 607       int skip_update = 0 == (uf->flags & UNIX_FILE_DATA_AVAILABLE_TO_WRITE);
 608       uf->flags &= ~UNIX_FILE_DATA_AVAILABLE_TO_WRITE;
 609       if (!skip_update)
 610         fm->file_update (uf, UNIX_FILE_UPDATE_MODIFY);
 611     }
 612 }
 613
 614 /** @brief A bit like strchr with a buffer length limit.
 615  * Search a buffer for the first instance of a character up to the limit of
 616  * the buffer length. If found then return the position of that character.
 617  *
 618  * The key departure from strchr is that if the character is not found then
 619  * return the buffer length.
 620  *
 621  * @param chr The byte value to search for.
 622  * @param str The buffer in which to search for the value.
 623  * @param len The depth into the buffer to search.
 624  *
 625  * @return The index of the first occurrence of \c chr. If \c chr is not
 626  *          found then \c len instead.
 627  */
 628 always_inline word
 629 unix_vlib_findchr (u8 chr, u8 * str, word len)
 630 {
 631   word i = 0;
 632   for (i = 0; i < len; i++, str++)
 633     {
 634       if (*str == chr)
 635         return i;
 636     }
 637   return len;
 638 }
 639
 640 /** @brief Send a buffer to the CLI stream if possible, enqueue it otherwise.
 641  * Attempts to write given buffer to the file descriptor of the given
 642  * Unix CLI session. If that session already has data in the output buffer
 643  * or if the write attempt tells us to try again later then the given buffer
 644  * is appended to the pending output buffer instead.
 645  *
 646  * This is typically called only from \c unix_vlib_cli_output_cooked since
 647  * that is where CRLF handling occurs or from places where we explicitly do
 648  * not want cooked handling.
 649  *
 650  * @param cf Unix CLI session of the desired stream to write to.
 651  * @param uf The Unix file structure of the desired stream to write to.
 652  * @param buffer Pointer to the buffer that needs to be written.
 653  * @param buffer_bytes The number of bytes from \c buffer to write.
 654  */
 655 static void
 656 unix_vlib_cli_output_raw (unix_cli_file_t * cf,
 657                           clib_file_t * uf, u8 * buffer, uword buffer_bytes)
 658 {
 659   int n = 0;
 660
 661   if (cf->has_epipe)            /* don't try writing anything */
 662     return;
 663
 664   if (vec_len (cf->output_vector) == 0)
 665     {
 666       if (cf->is_socket)
 667         /* If it's a socket we use MSG_NOSIGNAL to prevent SIGPIPE */
 668         n = send (uf->file_descriptor, buffer, buffer_bytes, MSG_NOSIGNAL);
 669       else
 670         n = write (uf->file_descriptor, buffer, buffer_bytes);
 671     }
 672
 673   if (n < 0 && errno != EAGAIN)
 674     {
 675       if (errno == EPIPE)
 676         {
 677           /* connection closed on us */
 678           unix_main_t *um = &unix_main;
 679           cf->has_epipe = 1;
 680           vlib_process_signal_event (um->vlib_main, cf->process_node_index,
 681                                      UNIX_CLI_PROCESS_EVENT_QUIT,
 682                                      uf->private_data);
 683         }
 684       else
 685         {
 686           clib_unix_warning ("write");
 687         }
 688     }
 689   else if ((word) n < (word) buffer_bytes)
 690     {
 691       /* We got EAGAIN or we already have stuff in the buffer;
 692        * queue up whatever didn't get sent for later. */
 693       if (n < 0)
 694         n = 0;
 695       unix_cli_add_pending_output (uf, cf, buffer + n, buffer_bytes - n);
 696     }
 697 }
 698
 699 /** @brief Process a buffer for CRLF handling before outputting it to the CLI.
 700  *
 701  * @param cf Unix CLI session of the desired stream to write to.
 702  * @param uf The Unix file structure of the desired stream to write to.
 703  * @param buffer Pointer to the buffer that needs to be written.
 704  * @param buffer_bytes The number of bytes from \c buffer to write.
 705  */
 706 static void
 707 unix_vlib_cli_output_cooked (unix_cli_file_t * cf,
 708                              clib_file_t * uf,
 709                              u8 * buffer, uword buffer_bytes)
 710 {
 711   word end = 0, start = 0;
 712
 713   while (end < buffer_bytes)
 714     {
 715       if (cf->crlf_mode)
 716         {
 717           /* iterate the line on \n's so we can insert a \r before it */
 718           end = unix_vlib_findchr ('\n',
 719                                    buffer + start,
 720                                    buffer_bytes - start) + start;
 721         }
 722       else
 723         {
 724           /* otherwise just send the whole buffer */
 725           end = buffer_bytes;
 726         }
 727
 728       unix_vlib_cli_output_raw (cf, uf, buffer + start, end - start);
 729
 730       if (cf->crlf_mode)
 731         {
 732           if (end < buffer_bytes)
 733             {
 734               unix_vlib_cli_output_raw (cf, uf, (u8 *) "\r\n", 2);
 735               end++;            /* skip the \n that we already sent */
 736             }
 737           start = end;
 738         }
 739     }
 740
 741   /* Use the last character to determine the last direction of the cursor. */
 742   if (buffer_bytes > 0)
 743     cf->cursor_direction = (buffer[buffer_bytes - 1] == (u8) '\b');
 744 }
 745
 746 /** @brief Moves the terminal cursor one character to the left, with
 747  * special handling when it reaches the left edge of the terminal window.
 748  *
 749  * Ordinarily we can simply send a '\b' to move the cursor left, however
 750  * most terminals will not reverse-wrap to the end of the previous line
 751  * if the cursor is in the left-most column. To counter this we must
 752  * check the cursor position + prompt length modulo terminal width and
 753  * if available use some other means, such as ANSI terminal escape
 754  * sequences, to move the cursor.
 755  *
 756  * @param cf Unix CLI session of the desired stream to write to.
 757  * @param uf The Unix file structure of the desired stream to write to.
 758  */
 759 static void
 760 unix_vlib_cli_output_cursor_left (unix_cli_file_t * cf, clib_file_t * uf)
 761 {
 762   unix_cli_main_t *cm = &unix_cli_main;
 763   static u8 *ansi = 0;          /* assumes no reentry */
 764   u32 position;
 765
 766   if (!cf->is_interactive || !cf->ansi_capable || !cf->width)
 767     {
 768       /* No special handling for dumb terminals */
 769       unix_vlib_cli_output_cooked (cf, uf, (u8 *) "\b", 1);
 770       return;
 771     }
 772
 773   position = ((u32) vec_len (cm->cli_prompt) + cf->cursor) % cf->width;
 774
 775   if (position != 0)
 776     {
 777       /* No special handling required if we're not at the left edge */
 778       unix_vlib_cli_output_cooked (cf, uf, (u8 *) "\b", 1);
 779       return;
 780     }
 781
 782   if (!cf->cursor_direction)
 783     {
 784       /* Special handling for when we are at the left edge but
 785        * the cursor was going left-to-right, but in this situation
 786        * xterm-like terminals actually hide the cursor off the right
 787        * edge. A \b here seems to jump one char too many, so let's
 788        * force the cursor onto the next line instead.
 789        */
 790       if (cf->cursor < vec_len (cf->current_command))
 791         unix_vlib_cli_output_cooked (cf, uf, &cf->current_command[cf->cursor],
 792                                      1);
 793       else
 794         unix_vlib_cli_output_cooked (cf, uf, (u8 *) " ", 1);
 795       unix_vlib_cli_output_cooked (cf, uf, (u8 *) "\r", 1);
 796     }
 797
 798   /* Relocate the cursor at the right hand edge one line above */
 799   ansi = format (ansi, CSI "A" CSI "%dC", cf->width - 1);
 800   unix_vlib_cli_output_cooked (cf, uf, ansi, vec_len (ansi));
 801   vec_reset_length (ansi);      /* keep the vec around for next time */
 802   cf->cursor_direction = 1;     /* going backwards now */
 803 }
 804
 805 /** @brief Output the CLI prompt */
 806 static void
 807 unix_cli_cli_prompt (unix_cli_file_t * cf, clib_file_t * uf)
 808 {
 809   unix_cli_main_t *cm = &unix_cli_main;
 810
 811   if (cf->is_interactive)       /* Only interactive sessions get a prompt */
 812     unix_vlib_cli_output_raw (cf, uf, cm->cli_prompt,
 813                               vec_len (cm->cli_prompt));
 814 }
 815
 816 /** @brief Output a pager prompt and show number of buffered lines */
 817 static void
 818 unix_cli_pager_prompt (unix_cli_file_t * cf, clib_file_t * uf)
 819 {
 820   u8 *prompt;
 821   u32 h;
 822
 823   h = cf->pager_start + (cf->height - 1);
 824   if (h > vec_len (cf->pager_index))
 825     h = vec_len (cf->pager_index);
 826
 827   prompt = format (0, "\r%s-- more -- (%d-%d/%d)%s",
 828                    cf->ansi_capable ? ANSI_BOLD : "",
 829                    cf->pager_start + 1,
 830                    h,
 831                    vec_len (cf->pager_index),
 832                    cf->ansi_capable ? ANSI_RESET : "");
 833
 834   unix_vlib_cli_output_cooked (cf, uf, prompt, vec_len (prompt));
 835
 836   vec_free (prompt);
 837 }
 838
 839 /** @brief Output a pager "skipping" message */
 840 static void
 841 unix_cli_pager_message (unix_cli_file_t * cf, clib_file_t * uf,
 842                         char *message, char *postfix)
 843 {
 844   u8 *prompt;
 845
 846   prompt = format (0, "\r%s-- %s --%s%s",
 847                    cf->ansi_capable ? ANSI_BOLD : "",
 848                    message, cf->ansi_capable ? ANSI_RESET : "", postfix);
 849
 850   unix_vlib_cli_output_cooked (cf, uf, prompt, vec_len (prompt));
 851
 852   vec_free (prompt);
 853 }
 854
 855 /** @brief Erase the printed pager prompt */
 856 static void
 857 unix_cli_pager_prompt_erase (unix_cli_file_t * cf, clib_file_t * uf)
 858 {
 859   if (cf->ansi_capable)
 860     {
 861       unix_vlib_cli_output_cooked (cf, uf, (u8 *) "\r", 1);
 862       unix_vlib_cli_output_cooked (cf, uf,
 863                                    (u8 *) ANSI_CLEARLINE,
 864                                    sizeof (ANSI_CLEARLINE) - 1);
 865     }
 866   else
 867     {
 868       int i;
 869
 870       unix_vlib_cli_output_cooked (cf, uf, (u8 *) "\r", 1);
 871       for (i = 0; i < cf->width - 1; i++)
 872         unix_vlib_cli_output_cooked (cf, uf, (u8 *) " ", 1);
 873       unix_vlib_cli_output_cooked (cf, uf, (u8 *) "\r", 1);
 874     }
 875 }
 876
 877 /** @brief Uses an ANSI escape sequence to move the cursor */
 878 static void
 879 unix_cli_ansi_cursor (unix_cli_file_t * cf, clib_file_t * uf, u16 x, u16 y)
 880 {
 881   u8 *str;
 882
 883   str = format (0, "%s%d;%dH", CSI, y, x);
 884
 885   unix_vlib_cli_output_cooked (cf, uf, str, vec_len (str));
 886
 887   vec_free (str);
 888 }
 889
 890 /** Redraw the currently displayed page of text.
 891  * @param cf CLI session to redraw the pager buffer of.
 892  * @param uf Unix file of the CLI session.
 893  */
 894 static void
 895 unix_cli_pager_redraw (unix_cli_file_t * cf, clib_file_t * uf)
 896 {
 897   unix_cli_pager_index_t *pi = NULL;
 898   u8 *line = NULL;
 899   word i;
 900
 901   /* No active pager? Do nothing. */
 902   if (!vec_len (cf->pager_index))
 903     return;
 904
 905   if (cf->ansi_capable)
 906     {
 907       /* If we have ANSI, send the clear screen sequence */
 908       unix_vlib_cli_output_cooked (cf, uf,
 909                                    (u8 *) ANSI_CLEAR,
 910                                    sizeof (ANSI_CLEAR) - 1);
 911     }
 912   else
 913     {
 914       /* Otherwise make sure we're on a blank line */
 915       unix_cli_pager_prompt_erase (cf, uf);
 916     }
 917
 918   /* (Re-)send the current page of content */
 919   for (i = 0; i < cf->height - 1 &&
 920        i + cf->pager_start < vec_len (cf->pager_index); i++)
 921     {
 922       pi = &cf->pager_index[cf->pager_start + i];
 923       line = cf->pager_vector[pi->line] + pi->offset;
 924
 925       unix_vlib_cli_output_cooked (cf, uf, line, pi->length);
 926     }
 927   /* if the last line didn't end in newline, add a newline */
 928   if (pi && line[pi->length - 1] != '\n')
 929     unix_vlib_cli_output_cooked (cf, uf, (u8 *) "\n", 1);
 930
 931   unix_cli_pager_prompt (cf, uf);
 932 }
 933
 934 /** @brief Process and add a line to the pager index.
 935  * In normal operation this function will take the given character string
 936  * found in @c line and with length @c len_or_index and iterates the over the
 937  * contents, adding each line of text discovered within it to the
 938  * pager index. Lines are identified by newlines ("<code>\\n</code>") and by
 939  * strings longer than the width of the terminal.
 940  *
 941  * If instead @c line is @c NULL then @c len_or_index is taken to mean the
 942  * index of an existing line in the pager buffer; this simply means that the
 943  * input line does not need to be cloned since we already have it. This is
 944  * typical if we are reindexing the pager buffer.
 945  *
 946  * @param cf           The CLI session whose pager we are adding to.
 947  * @param line         The string of text to be indexed into the pager buffer.
 948  *                     If @c line is @c NULL then the mode of operation
 949  *                     changes slightly; see the description above.
 950  * @param len_or_index If @c line is a pointer to a string then this parameter
 951  *                     indicates the length of that string; Otherwise this
 952  *                     value provides the index in the pager buffer of an
 953  *                     existing string to be indexed.
 954  */
 955 static void
 956 unix_cli_pager_add_line (unix_cli_file_t * cf, u8 * line, word len_or_index)
 957 {
 958   u8 *p = NULL;
 959   word i, j, k;
 960   word line_index, len;
 961   u32 width = cf->width;
 962   unix_cli_pager_index_t *pi;
 963
 964   if (line == NULL)
 965     {
 966       /* Use a line already in the pager buffer */
 967       line_index = len_or_index;
 968       if (cf->pager_vector != NULL)
 969         p = cf->pager_vector[line_index];
 970       len = vec_len (p);
 971     }
 972   else
 973     {
 974       len = len_or_index;
 975       /* Add a copy of the raw string to the pager buffer */
 976       p = vec_new (u8, len);
 977       clib_memcpy (p, line, len);
 978
 979       /* store in pager buffer */
 980       line_index = vec_len (cf->pager_vector);
 981       vec_add1 (cf->pager_vector, p);
 982     }
 983
 984   i = 0;
 985   while (i < len)
 986     {
 987       /* Find the next line, or run to terminal width, or run to EOL */
 988       int l = len - i;
 989       j = unix_vlib_findchr ((u8) '\n', p, l < width ? l : width);
 990
 991       if (j < l && p[j] == '\n')        /* incl \n */
 992         j++;
 993
 994       /* Add the line to the index */
 995       k = vec_len (cf->pager_index);
 996       vec_validate (cf->pager_index, k);
 997       pi = &cf->pager_index[k];
 998
 999       pi->line = line_index;
1000       pi->offset = i;
1001       pi->length = j;
1002
1003       i += j;
1004       p += j;
1005     }
1006 }
1007
1008 /** @brief Reindex entire pager buffer.
1009  * Resets the current pager index and then re-adds the lines in the pager
1010  * buffer to the index.
1011  *
1012  * Additionally this function attempts to retain the current page start
1013  * line offset by searching for the same top-of-screen line in the new index.
1014  *
1015  * @param cf The CLI session whose pager buffer should be reindexed.
1016  */
1017 static void
1018 unix_cli_pager_reindex (unix_cli_file_t * cf)
1019 {
1020   word i, old_line, old_offset;
1021   unix_cli_pager_index_t *pi;
1022
1023   /* If there is nothing in the pager buffer then make sure the index
1024    * is empty and move on.
1025    */
1026   if (cf->pager_vector == 0)
1027     {
1028       vec_reset_length (cf->pager_index);
1029       return;
1030     }
1031
1032   /* Retain a pointer to the current page start line so we can
1033    * find it later
1034    */
1035   pi = &cf->pager_index[cf->pager_start];
1036   old_line = pi->line;
1037   old_offset = pi->offset;
1038
1039   /* Re-add the buffered lines to the index */
1040   vec_reset_length (cf->pager_index);
1041   vec_foreach_index (i, cf->pager_vector)
1042   {
1043     unix_cli_pager_add_line (cf, NULL, i);
1044   }
1045
1046   /* Attempt to re-locate the previously stored page start line */
1047   vec_foreach_index (i, cf->pager_index)
1048   {
1049     pi = &cf->pager_index[i];
1050
1051     if (pi->line == old_line &&
1052         (pi->offset <= old_offset || pi->offset + pi->length > old_offset))
1053       {
1054         /* Found it! */
1055         cf->pager_start = i;
1056         break;
1057       }
1058   }
1059
1060   /* In case the start line was not found (rare), ensure the pager start
1061    * index is within bounds
1062    */
1063   if (cf->pager_start >= vec_len (cf->pager_index))
1064     {
1065       if (!cf->height || vec_len (cf->pager_index) < (cf->height - 1))
1066         cf->pager_start = 0;
1067       else
1068         cf->pager_start = vec_len (cf->pager_index) - (cf->height - 1);
1069     }
1070 }
1071
1072 /** VLIB CLI output function.
1073  *
1074  * If the terminal has a pager configured then this function takes care
1075  * of collating output into the pager buffer; ensuring only the first page
1076  * is displayed and any lines in excess of the first page are buffered.
1077  *
1078  * If the maximum number of index lines in the buffer is exceeded then the
1079  * pager is cancelled and the contents of the current buffer are sent to the
1080  * terminal.
1081  *
1082  * If there is no pager configured then the output is sent directly to the
1083  * terminal.
1084  *
1085  * @param cli_file_index Index of the CLI session where this output is
1086  *                       directed.
1087  * @param buffer         String of printabe bytes to be output.
1088  * @param buffer_bytes   The number of bytes in @c buffer to be output.
1089  */
1090 static void
1091 unix_vlib_cli_output (uword cli_file_index, u8 * buffer, uword buffer_bytes)
1092 {
1093   unix_main_t *um = &unix_main;
1094   clib_file_main_t *fm = &file_main;
1095   unix_cli_main_t *cm = &unix_cli_main;
1096   unix_cli_file_t *cf;
1097   clib_file_t *uf;
1098
1099   cf = pool_elt_at_index (cm->cli_file_pool, cli_file_index);
1100   uf = pool_elt_at_index (fm->file_pool, cf->clib_file_index);
1101
1102   if (cf->no_pager || um->cli_pager_buffer_limit == 0 || cf->height == 0)
1103     {
1104       unix_vlib_cli_output_cooked (cf, uf, buffer, buffer_bytes);
1105     }
1106   else
1107     {
1108       word row = vec_len (cf->pager_index);
1109       u8 *line;
1110       unix_cli_pager_index_t *pi;
1111
1112       /* Index and add the output lines to the pager buffer. */
1113       unix_cli_pager_add_line (cf, buffer, buffer_bytes);
1114
1115       /* Now iterate what was added to display the lines.
1116        * If we reach the bottom of the page, display a prompt.
1117        */
1118       while (row < vec_len (cf->pager_index))
1119         {
1120           if (row < cf->height - 1)
1121             {
1122               /* output this line */
1123               pi = &cf->pager_index[row];
1124               line = cf->pager_vector[pi->line] + pi->offset;
1125               unix_vlib_cli_output_cooked (cf, uf, line, pi->length);
1126
1127               /* if the last line didn't end in newline, and we're at the
1128                * bottom of the page, add a newline */
1129               if (line[pi->length - 1] != '\n' && row == cf->height - 2)
1130                 unix_vlib_cli_output_cooked (cf, uf, (u8 *) "\n", 1);
1131             }
1132           else
1133             {
1134               /* Display the pager prompt every 10 lines */
1135               if (!(row % 10))
1136                 unix_cli_pager_prompt (cf, uf);
1137             }
1138           row++;
1139         }
1140
1141       /* Check if we went over the pager buffer limit */
1142       if (vec_len (cf->pager_index) > um->cli_pager_buffer_limit)
1143         {
1144           /* Stop using the pager for the remainder of this CLI command */
1145           cf->no_pager = 2;
1146
1147           /* If we likely printed the prompt, erase it */
1148           if (vec_len (cf->pager_index) > cf->height - 1)
1149             unix_cli_pager_prompt_erase (cf, uf);
1150
1151           /* Dump out the contents of the buffer */
1152           for (row = cf->pager_start + (cf->height - 1);
1153                row < vec_len (cf->pager_index); row++)
1154             {
1155               pi = &cf->pager_index[row];
1156               line = cf->pager_vector[pi->line] + pi->offset;
1157               unix_vlib_cli_output_cooked (cf, uf, line, pi->length);
1158             }
1159
1160           unix_cli_pager_reset (cf);
1161         }
1162     }
1163 }
1164
1165 /** Identify whether a terminal type is ANSI capable.
1166  *
1167  * Compares the string given in @c term with a list of terminal types known
1168  * to support ANSI escape sequences.
1169  *
1170  * This list contains, for example, @c xterm, @c screen and @c ansi.
1171  *
1172  * @param term A string with a terminal type in it.
1173  * @param len The length of the string in @c term.
1174  *
1175  * @return @c 1 if the terminal type is recognized as supporting ANSI
1176  *         terminal sequences; @c 0 otherwise.
1177  */
1178 static u8
1179 unix_cli_terminal_type_ansi (u8 * term, uword len)
1180 {
1181   /* This may later be better done as a hash of some sort. */
1182 #define _(a) do { \
1183     if (strncasecmp(a, (char *)term, (size_t)len) == 0) return 1; \
1184   } while(0)
1185
1186   _("xterm");
1187   _("xterm-color");
1188   _("xterm-256color");          /* iTerm on Mac */
1189   _("screen");
1190   _("screen-256color");         /* Screen and tmux */
1191   _("ansi");                    /* Microsoft Telnet */
1192 #undef _
1193
1194   return 0;
1195 }
1196
1197 /** Identify whether a terminal type is non-interactive.
1198  *
1199  * Compares the string given in @c term with a list of terminal types known
1200  * to be non-interactive, as send by tools such as @c vppctl .
1201  *
1202  * This list contains, for example, @c vppctl.
1203  *
1204  * @param term A string with a terminal type in it.
1205  * @param len The length of the string in @c term.
1206  *
1207  * @return @c 1 if the terminal type is recognized as being non-interactive;
1208  *         @c 0 otherwise.
1209  */
1210 static u8
1211 unix_cli_terminal_type_noninteractive (u8 * term, uword len)
1212 {
1213   /* This may later be better done as a hash of some sort. */
1214 #define _(a) do { \
1215     if (strncasecmp(a, (char *)term, (size_t)len) == 0) return 1; \
1216   } while(0)
1217
1218   _("vppctl");
1219 #undef _
1220
1221   return 0;
1222 }
1223
1224 /** Set a session to be non-interactive. */
1225 static void
1226 unix_cli_set_session_noninteractive (unix_cli_file_t * cf)
1227 {
1228   /* Non-interactive sessions don't get these */
1229   cf->is_interactive = 0;
1230   cf->no_pager = 1;
1231   cf->history_limit = 0;
1232   cf->has_history = 0;
1233   cf->line_mode = 1;
1234 }
1235
1236 /** @brief Emit initial welcome banner and prompt on a connection. */
1237 static void
1238 unix_cli_file_welcome (unix_cli_main_t * cm, unix_cli_file_t * cf)
1239 {
1240   unix_main_t *um = &unix_main;
1241   clib_file_main_t *fm = &file_main;
1242   clib_file_t *uf = pool_elt_at_index (fm->file_pool, cf->clib_file_index);
1243   unix_cli_banner_t *banner;
1244   int i, len;
1245
1246   /* Mark the session as started if we get here */
1247   cf->started = 1;
1248
1249   if (!(cf->is_interactive))    /* No banner for non-interactive sessions */
1250     return;
1251
1252   /*
1253    * Put the first bytes directly into the buffer so that further output is
1254    * queued until everything is ready. (oterwise initial prompt can appear
1255    * mid way through VPP initialization)
1256    */
1257   unix_cli_add_pending_output (uf, cf, (u8 *) "\r", 1);
1258
1259   if (!um->cli_no_banner && (um->flags & UNIX_FLAG_NOBANNER) == 0)
1260     {
1261       if (cf->ansi_capable && (um->flags & UNIX_FLAG_NOCOLOR) == 0)
1262         {
1263           banner = unix_cli_banner_color;
1264           len = ARRAY_LEN (unix_cli_banner_color);
1265         }
1266       else
1267         {
1268           banner = unix_cli_banner;
1269           len = ARRAY_LEN (unix_cli_banner);
1270         }
1271
1272       for (i = 0; i < len; i++)
1273         {
1274           unix_vlib_cli_output_cooked (cf, uf,
1275                                        banner[i].line, banner[i].length);
1276         }
1277     }
1278
1279   /* Prompt. */
1280   unix_cli_cli_prompt (cf, uf);
1281
1282 }
1283
1284 /**
1285  * @brief A failsafe manager that ensures CLI sessions issue an initial
1286  * prompt if TELNET negotiation fails.
1287  */
1288 static uword
1289 unix_cli_new_session_process (vlib_main_t * vm, vlib_node_runtime_t * rt,
1290                               vlib_frame_t * f)
1291 {
1292   unix_cli_main_t *cm = &unix_cli_main;
1293   uword event_type, *event_data = 0;
1294   f64 wait = 10.0;
1295
1296   while (1)
1297     {
1298       if (vec_len (cm->new_sessions) > 0)
1299         wait = vlib_process_wait_for_event_or_clock (vm, wait);
1300       else
1301         vlib_process_wait_for_event (vm);
1302
1303       event_type = vlib_process_get_events (vm, &event_data);
1304
1305       switch (event_type)
1306         {
1307         case ~0:                /* no events => timeout */
1308           break;
1309
1310         case UNIX_CLI_NEW_SESSION_EVENT_ADD:
1311           {
1312             /* Add an identifier to the new session list */
1313             unix_cli_new_session_t ns;
1314
1315             ns.cf_index = event_data[0];
1316             ns.deadline = vlib_time_now (vm) + 1.0;
1317
1318             vec_add1 (cm->new_sessions, ns);
1319
1320             if (wait > 0.1)
1321               wait = 0.1;       /* force a re-evaluation soon, but not too soon */
1322           }
1323           break;
1324
1325         default:
1326           clib_warning ("BUG: unknown event type 0x%wx", event_type);
1327           break;
1328         }
1329
1330       vec_reset_length (event_data);
1331
1332       if (vlib_process_suspend_time_is_zero (wait))
1333         {
1334           /* Scan the list looking for expired deadlines.
1335            * Emit needed prompts and remove from the list.
1336            * While scanning, look for the nearest new deadline
1337            * for the next iteration.
1338            * Since the list is ordered with newest sessions first
1339            * we can make assumptions about expired sessions being
1340            * contiguous at the beginning and the next deadline is the
1341            * next entry on the list, if any.
1342            */
1343           f64 now = vlib_time_now (vm);
1344           unix_cli_new_session_t *nsp;
1345           word index = 0;
1346
1347           wait = INFINITY;
1348
1349           vec_foreach (nsp, cm->new_sessions)
1350           {
1351             if (vlib_process_suspend_time_is_zero (nsp->deadline - now))
1352               {
1353                 /* Deadline reached */
1354                 unix_cli_file_t *cf;
1355
1356                 /* Mark the highwater */
1357                 index++;
1358
1359                 /* Check the connection didn't close already */
1360                 if (pool_is_free_index (cm->cli_file_pool, nsp->cf_index))
1361                   continue;
1362
1363                 cf = pool_elt_at_index (cm->cli_file_pool, nsp->cf_index);
1364
1365                 if (!cf->started)
1366                   unix_cli_file_welcome (cm, cf);
1367               }
1368             else
1369               {
1370                 wait = nsp->deadline - now;
1371                 break;
1372               }
1373           }
1374
1375           if (index)
1376             {
1377               /* We have sessions to remove */
1378               vec_delete (cm->new_sessions, index, 0);
1379             }
1380         }
1381     }
1382
1383   return 0;
1384 }
1385
1386
1387 /** @brief A mostly no-op Telnet state machine.
1388  * Process Telnet command bytes in a way that ensures we're mostly
1389  * transparent to the Telnet protocol. That is, it's mostly a no-op.
1390  *
1391  * @return -1 if we need more bytes, otherwise a positive integer number of
1392  *          bytes to consume from the input_vector, not including the initial
1393  *          IAC byte.
1394  */
1395 static i32
1396 unix_cli_process_telnet (unix_main_t * um,
1397                          unix_cli_file_t * cf,
1398                          clib_file_t * uf, u8 * input_vector, uword len)
1399 {
1400   /* Input_vector starts at IAC byte.
1401    * See if we have a complete message; if not, return -1 so we wait for more.
1402    * if we have a complete message, consume those bytes from the vector.
1403    */
1404   i32 consume = 0;
1405
1406   if (len == 1)
1407     return -1;                  /* want more bytes */
1408
1409   switch (input_vector[1])
1410     {
1411     case IAC:
1412       /* two IAC's in a row means to pass through 0xff.
1413        * since that makes no sense here, just consume it.
1414        */
1415       consume = 1;
1416       break;
1417
1418     case WILL:
1419     case WONT:
1420     case DO:
1421     case DONT:
1422       /* Expect 3 bytes */
1423       if (len < 3)
1424         return -1;              /* want more bytes */
1425
1426       consume = 2;
1427       break;
1428
1429     case SB:
1430       {
1431         /* Sub option - search ahead for IAC SE to end it */
1432         i32 i;
1433         for (i = 3; i < len && i < UNIX_CLI_MAX_DEPTH_TELNET; i++)
1434           {
1435             if (input_vector[i - 1] == IAC && input_vector[i] == SE)
1436               {
1437                 /* We have a complete message; see if we care about it */
1438                 switch (input_vector[2])
1439                   {
1440                   case TELOPT_TTYPE:
1441                     if (input_vector[3] != 0)
1442                       break;
1443                     {
1444                       /* See if the the terminal type is recognized */
1445                       u8 *term = input_vector + 4;
1446                       uword len = i - 5;
1447
1448                       /* See if the terminal type is ANSI capable */
1449                       cf->ansi_capable =
1450                         unix_cli_terminal_type_ansi (term, len);
1451
1452                       /* See if the terminal type indicates non-interactive */
1453                       if (unix_cli_terminal_type_noninteractive (term, len))
1454                         unix_cli_set_session_noninteractive (cf);
1455                     }
1456
1457                     /* If session not started, we can release the pause */
1458                     if (!cf->started)
1459                       /* Send the welcome banner and initial prompt */
1460                       unix_cli_file_welcome (&unix_cli_main, cf);
1461                     break;
1462
1463                   case TELOPT_NAWS:
1464                     /* Window size */
1465                     if (i != 8) /* check message is correct size */
1466                       break;
1467
1468                     cf->width =
1469                       clib_net_to_host_u16 (*((u16 *) (input_vector + 3)));
1470                     if (cf->width > UNIX_CLI_MAX_TERMINAL_WIDTH)
1471                       cf->width = UNIX_CLI_MAX_TERMINAL_WIDTH;
1472                     if (cf->width == 0)
1473                       cf->width = UNIX_CLI_DEFAULT_TERMINAL_WIDTH;
1474
1475                     cf->height =
1476                       clib_net_to_host_u16 (*((u16 *) (input_vector + 5)));
1477                     if (cf->height > UNIX_CLI_MAX_TERMINAL_HEIGHT)
1478                       cf->height = UNIX_CLI_MAX_TERMINAL_HEIGHT;
1479                     if (cf->height == 0)
1480                       cf->height = UNIX_CLI_DEFAULT_TERMINAL_HEIGHT;
1481
1482                     /* reindex pager buffer */
1483                     unix_cli_pager_reindex (cf);
1484                     /* redraw page */
1485                     unix_cli_pager_redraw (cf, uf);
1486                     break;
1487
1488                   default:
1489                     break;
1490                   }
1491                 /* Consume it all */
1492                 consume = i;
1493                 break;
1494               }
1495           }
1496
1497         if (i == UNIX_CLI_MAX_DEPTH_TELNET)
1498           consume = 1;          /* hit max search depth, advance one byte */
1499
1500         if (consume == 0)
1501           return -1;            /* want more bytes */
1502
1503         break;
1504       }
1505
1506     case GA:
1507     case EL:
1508     case EC:
1509     case AO:
1510     case IP:
1511     case BREAK:
1512     case DM:
1513     case NOP:
1514     case SE:
1515     case EOR:
1516     case ABORT:
1517     case SUSP:
1518     case xEOF:
1519       /* Simple one-byte messages */
1520       consume = 1;
1521       break;
1522
1523     case AYT:
1524       /* Are You There - trigger a visible response */
1525       consume = 1;
1526       unix_vlib_cli_output_cooked (cf, uf, (u8 *) "fd.io VPP\n", 10);
1527       break;
1528
1529     default:
1530       /* Unknown command! Eat the IAC byte */
1531       break;
1532     }
1533
1534   return consume;
1535 }
1536
1537 /** @brief Process actionable input.
1538  * Based on the \c action process the input; this typically involves
1539  * searching the command history or editing the current command line.
1540  */
1541 static int
1542 unix_cli_line_process_one (unix_cli_main_t * cm,
1543                            unix_main_t * um,
1544                            unix_cli_file_t * cf,
1545                            clib_file_t * uf,
1546                            u8 input, unix_cli_parse_action_t action)
1547 {
1548   u8 *prev;
1549   u8 *save = 0;
1550   u8 **possible_commands;
1551   int j, delta;
1552
1553   switch (action)
1554     {
1555     case UNIX_CLI_PARSE_ACTION_NOACTION:
1556       break;
1557
1558     case UNIX_CLI_PARSE_ACTION_REVSEARCH:
1559     case UNIX_CLI_PARSE_ACTION_FWDSEARCH:
1560       if (!cf->has_history || !cf->history_limit)
1561         break;
1562       if (cf->search_mode == 0)
1563         {
1564           /* Erase the current command (if any) */
1565           for (; cf->cursor > 0; cf->cursor--)
1566             {
1567               unix_vlib_cli_output_cursor_left (cf, uf);
1568               unix_vlib_cli_output_cooked (cf, uf, (u8 *) " ", 1);
1569               unix_vlib_cli_output_cursor_left (cf, uf);
1570             }
1571
1572           vec_reset_length (cf->search_key);
1573           vec_reset_length (cf->current_command);
1574
1575           if (action == UNIX_CLI_PARSE_ACTION_REVSEARCH)
1576             cf->search_mode = -1;
1577           else
1578             cf->search_mode = 1;
1579         }
1580       else
1581         {
1582           if (action == UNIX_CLI_PARSE_ACTION_REVSEARCH)
1583             cf->search_mode = -1;
1584           else
1585             cf->search_mode = 1;
1586
1587           cf->excursion += cf->search_mode;
1588           goto search_again;
1589         }
1590       break;
1591
1592     case UNIX_CLI_PARSE_ACTION_ERASELINELEFT:
1593       /* Erase the command from the cursor to the start */
1594
1595       j = cf->cursor;
1596       /* Shimmy backwards to the new end of line position */
1597       delta = vec_len (cf->current_command) - cf->cursor;
1598       for (; cf->cursor > delta; cf->cursor--)
1599         unix_vlib_cli_output_cursor_left (cf, uf);
1600       /* Zap from here to the end of what is currently displayed */
1601       for (; cf->cursor < vec_len (cf->current_command); cf->cursor++)
1602         unix_vlib_cli_output_cooked (cf, uf, (u8 *) " ", 1);
1603       /* Get back to the start of the line */
1604       for (; cf->cursor > 0; cf->cursor--)
1605         unix_vlib_cli_output_cursor_left (cf, uf);
1606
1607       /* Delete the desired text from the command */
1608       memmove (cf->current_command, cf->current_command + j, delta);
1609       _vec_len (cf->current_command) = delta;
1610
1611       /* Print the new contents */
1612       unix_vlib_cli_output_cooked (cf, uf, cf->current_command, delta);
1613       cf->cursor = delta;       /* for backspace tracking */
1614
1615       /* Shimmy back to the start */
1616       for (; cf->cursor > 0; cf->cursor--)
1617         unix_vlib_cli_output_cursor_left (cf, uf);
1618
1619       cf->search_mode = 0;
1620       break;
1621
1622     case UNIX_CLI_PARSE_ACTION_ERASELINERIGHT:
1623       /* Erase the command from the cursor to the end */
1624
1625       j = cf->cursor;
1626       /* Zap from cursor to end of what is currently displayed */
1627       for (; cf->cursor < (vec_len (cf->current_command)); cf->cursor++)
1628         unix_vlib_cli_output_cooked (cf, uf, (u8 *) " ", 1);
1629       /* Get back to where we were */
1630       for (; cf->cursor > j; cf->cursor--)
1631         unix_vlib_cli_output_cursor_left (cf, uf);
1632
1633       /* Truncate the line at the cursor */
1634       _vec_len (cf->current_command) = cf->cursor;
1635
1636       cf->search_mode = 0;
1637       break;
1638
1639     case UNIX_CLI_PARSE_ACTION_ERASEWORDLEFT:
1640       /* calculate num of caracter to be erased */
1641       delta = 0;
1642       while (cf->cursor > delta
1643              && cf->current_command[cf->cursor - delta - 1] == ' ')
1644         delta++;
1645       while (cf->cursor > delta
1646              && cf->current_command[cf->cursor - delta - 1] != ' ')
1647         delta++;
1648
1649       if (vec_len (cf->current_command))
1650         {
1651           if (cf->cursor > 0)
1652             {
1653               /* move cursor left delta times */
1654               for (j = delta; j > 0; j--, cf->cursor--)
1655                 unix_vlib_cli_output_cursor_left (cf, uf);
1656               save = cf->current_command + cf->cursor;
1657
1658               /* redraw remainder of line */
1659               memmove (cf->current_command + cf->cursor,
1660                        cf->current_command + cf->cursor + delta,
1661                        _vec_len (cf->current_command) - cf->cursor - delta);
1662               unix_vlib_cli_output_cooked (cf, uf,
1663                                            cf->current_command + cf->cursor,
1664                                            _vec_len (cf->current_command) -
1665                                            cf->cursor);
1666               cf->cursor += _vec_len (cf->current_command) - cf->cursor;
1667
1668               /* print delta amount of blank spaces,
1669                * then finally fix the cursor position */
1670               for (j = delta; j > 0; j--, cf->cursor--)
1671                 unix_vlib_cli_output_cursor_left (cf, uf);
1672               for (j = delta; j > 0; j--, cf->cursor++)
1673                 unix_vlib_cli_output_cooked (cf, uf, (u8 *) " ", 1);
1674               for (; (cf->current_command + cf->cursor) > save; cf->cursor--)
1675                 unix_vlib_cli_output_cursor_left (cf, uf);
1676               _vec_len (cf->current_command) -= delta;
1677             }
1678         }
1679       cf->search_mode = 0;
1680       cf->excursion = 0;
1681       vec_reset_length (cf->search_key);
1682       break;
1683
1684     case UNIX_CLI_PARSE_ACTION_LEFT:
1685       if (cf->cursor > 0)
1686         {
1687           unix_vlib_cli_output_cursor_left (cf, uf);
1688           cf->cursor--;
1689         }
1690
1691       cf->search_mode = 0;
1692       break;
1693
1694     case UNIX_CLI_PARSE_ACTION_RIGHT:
1695       if (cf->cursor < vec_len (cf->current_command))
1696         {
1697           /* have to emit the character under the cursor */
1698           unix_vlib_cli_output_cooked (cf, uf,
1699                                        cf->current_command + cf->cursor, 1);
1700           cf->cursor++;
1701         }
1702
1703       cf->search_mode = 0;
1704       break;
1705
1706     case UNIX_CLI_PARSE_ACTION_UP:
1707     case UNIX_CLI_PARSE_ACTION_DOWN:
1708       if (!cf->has_history || !cf->history_limit)
1709         break;
1710       cf->search_mode = 0;
1711       /* Erase the command */
1712       for (; cf->cursor < vec_len (cf->current_command); cf->cursor++)
1713         unix_vlib_cli_output_cooked (cf, uf, (u8 *) " ", 1);
1714       for (; cf->cursor > 0; cf->cursor--)
1715         {
1716           unix_vlib_cli_output_cursor_left (cf, uf);
1717           unix_vlib_cli_output_cooked (cf, uf, (u8 *) " ", 1);
1718           unix_vlib_cli_output_cursor_left (cf, uf);
1719         }
1720       vec_reset_length (cf->current_command);
1721       if (vec_len (cf->command_history))
1722         {
1723           if (action == UNIX_CLI_PARSE_ACTION_UP)
1724             delta = -1;
1725           else
1726             delta = 1;
1727
1728           cf->excursion += delta;
1729
1730           if (cf->excursion == vec_len (cf->command_history))
1731             {
1732               /* down-arrowed to last entry - want a blank line */
1733               _vec_len (cf->current_command) = 0;
1734             }
1735           else if (cf->excursion < 0)
1736             {
1737               /* up-arrowed over the start to the end, want a blank line */
1738               cf->excursion = vec_len (cf->command_history);
1739               _vec_len (cf->current_command) = 0;
1740             }
1741           else
1742             {
1743               if (cf->excursion > (i32) vec_len (cf->command_history) - 1)
1744                 /* down-arrowed past end - wrap to start */
1745                 cf->excursion = 0;
1746
1747               /* Print the command at the current position */
1748               prev = cf->command_history[cf->excursion];
1749               vec_validate (cf->current_command, vec_len (prev) - 1);
1750
1751               clib_memcpy (cf->current_command, prev, vec_len (prev));
1752               _vec_len (cf->current_command) = vec_len (prev);
1753               unix_vlib_cli_output_cooked (cf, uf, cf->current_command,
1754                                            vec_len (cf->current_command));
1755             }
1756         }
1757       cf->cursor = vec_len (cf->current_command);
1758       break;
1759
1760     case UNIX_CLI_PARSE_ACTION_HOME:
1761       if (vec_len (cf->current_command) && cf->cursor > 0)
1762         {
1763           for (; cf->cursor > 0; cf->cursor--)
1764             unix_vlib_cli_output_cursor_left (cf, uf);
1765         }
1766
1767       cf->search_mode = 0;
1768       break;
1769
1770     case UNIX_CLI_PARSE_ACTION_END:
1771       if (vec_len (cf->current_command) &&
1772           cf->cursor < vec_len (cf->current_command))
1773         {
1774           unix_vlib_cli_output_cooked (cf, uf,
1775                                        cf->current_command + cf->cursor,
1776                                        vec_len (cf->current_command) -
1777                                        cf->cursor);
1778           cf->cursor = vec_len (cf->current_command);
1779         }
1780
1781       cf->search_mode = 0;
1782       break;
1783
1784     case UNIX_CLI_PARSE_ACTION_WORDLEFT:
1785       if (vec_len (cf->current_command) && cf->cursor > 0)
1786         {
1787           unix_vlib_cli_output_cursor_left (cf, uf);
1788           cf->cursor--;
1789
1790           while (cf->cursor && isspace (cf->current_command[cf->cursor]))
1791             {
1792               unix_vlib_cli_output_cursor_left (cf, uf);
1793               cf->cursor--;
1794             }
1795           while (cf->cursor && !isspace (cf->current_command[cf->cursor]))
1796             {
1797               if (isspace (cf->current_command[cf->cursor - 1]))
1798                 break;
1799               unix_vlib_cli_output_cursor_left (cf, uf);
1800               cf->cursor--;
1801             }
1802
1803         }
1804
1805       cf->search_mode = 0;
1806       break;
1807
1808     case UNIX_CLI_PARSE_ACTION_WORDRIGHT:
1809       if (vec_len (cf->current_command) &&
1810           cf->cursor < vec_len (cf->current_command))
1811         {
1812           int e = vec_len (cf->current_command);
1813           j = cf->cursor;
1814           while (j < e && !isspace (cf->current_command[j]))
1815             j++;
1816           while (j < e && isspace (cf->current_command[j]))
1817             j++;
1818           unix_vlib_cli_output_cooked (cf, uf,
1819                                        cf->current_command + cf->cursor,
1820                                        j - cf->cursor);
1821           cf->cursor = j;
1822         }
1823
1824       cf->search_mode = 0;
1825       break;
1826
1827
1828     case UNIX_CLI_PARSE_ACTION_ERASE:
1829       if (vec_len (cf->current_command))
1830         {
1831           if (cf->cursor == vec_len (cf->current_command))
1832             {
1833               unix_vlib_cli_output_cursor_left (cf, uf);
1834               cf->cursor--;
1835               unix_vlib_cli_output_cooked (cf, uf, (u8 *) " ", 1);
1836               cf->cursor++;
1837               unix_vlib_cli_output_cursor_left (cf, uf);
1838               cf->cursor--;
1839               _vec_len (cf->current_command)--;
1840             }
1841           else if (cf->cursor > 0)
1842             {
1843               /* shift everything at & to the right of the cursor left by 1 */
1844               j = vec_len (cf->current_command) - cf->cursor;
1845               memmove (cf->current_command + cf->cursor - 1,
1846                        cf->current_command + cf->cursor, j);
1847               _vec_len (cf->current_command)--;
1848
1849               /* redraw the rest of the line */
1850               unix_vlib_cli_output_cursor_left (cf, uf);
1851               cf->cursor--;
1852               unix_vlib_cli_output_cooked (cf, uf,
1853                                            cf->current_command + cf->cursor,
1854                                            j);
1855               cf->cursor += j;
1856               /* erase last char */
1857               unix_vlib_cli_output_cooked (cf, uf, (u8 *) " ", 1);
1858               cf->cursor++;
1859
1860               /* and shift the terminal cursor back where it should be */
1861               j += 2;           /* account for old string length and offset position */
1862               while (--j)
1863                 {
1864                   unix_vlib_cli_output_cursor_left (cf, uf);
1865                   cf->cursor--;
1866                 }
1867             }
1868         }
1869       cf->search_mode = 0;
1870       cf->excursion = 0;
1871       vec_reset_length (cf->search_key);
1872       break;
1873
1874     case UNIX_CLI_PARSE_ACTION_ERASERIGHT:
1875       if (vec_len (cf->current_command))
1876         {
1877           if (cf->cursor < vec_len (cf->current_command))
1878             {
1879               /* shift everything to the right of the cursor left by 1 */
1880               j = vec_len (cf->current_command) - cf->cursor - 1;
1881               memmove (cf->current_command + cf->cursor,
1882                        cf->current_command + cf->cursor + 1, j);
1883               _vec_len (cf->current_command)--;
1884               /* redraw the rest of the line */
1885               unix_vlib_cli_output_cooked (cf, uf,
1886                                            cf->current_command + cf->cursor,
1887                                            j);
1888               cf->cursor += j;
1889               unix_vlib_cli_output_cooked (cf, uf, (u8 *) " ", 1);
1890               cf->cursor++;
1891               unix_vlib_cli_output_cursor_left (cf, uf);
1892               cf->cursor--;
1893               /* and shift the terminal cursor back where it should be */
1894               if (j)
1895                 {
1896                   unix_vlib_cli_output_cursor_left (cf, uf);
1897                   cf->cursor--;
1898                   while (--j)
1899                     {
1900                       unix_vlib_cli_output_cursor_left (cf, uf);
1901                       cf->cursor--;
1902                     }
1903                 }
1904             }
1905         }
1906       else if (input == 'D' - '@')
1907         {
1908           /* ^D with no command entered = quit */
1909           unix_vlib_cli_output_cooked (cf, uf, (u8 *) "quit\n", 5);
1910           vlib_process_signal_event (um->vlib_main,
1911                                      vlib_current_process (um->vlib_main),
1912                                      UNIX_CLI_PROCESS_EVENT_QUIT,
1913                                      cf - cm->cli_file_pool);
1914         }
1915       cf->search_mode = 0;
1916       cf->excursion = 0;
1917       vec_reset_length (cf->search_key);
1918       break;
1919
1920     case UNIX_CLI_PARSE_ACTION_CLEAR:
1921       /* If we're in ANSI mode, clear the screen.
1922        * Then redraw the prompt and any existing command input, then put
1923        * the cursor back where it was in that line.
1924        */
1925       if (cf->ansi_capable)
1926         unix_vlib_cli_output_cooked (cf, uf,
1927                                      (u8 *) ANSI_CLEAR,
1928                                      sizeof (ANSI_CLEAR) - 1);
1929       else
1930         unix_vlib_cli_output_cooked (cf, uf, (u8 *) "\n", 1);
1931
1932       unix_vlib_cli_output_raw (cf, uf,
1933                                 cm->cli_prompt, vec_len (cm->cli_prompt));
1934       unix_vlib_cli_output_cooked (cf, uf,
1935                                    cf->current_command,
1936                                    vec_len (cf->current_command));
1937       j = cf->cursor;
1938       cf->cursor = vec_len (cf->current_command);
1939       for (; cf->cursor > j; cf->cursor--)
1940         unix_vlib_cli_output_cursor_left (cf, uf);
1941
1942       break;
1943
1944     case UNIX_CLI_PARSE_ACTION_TAB:
1945       if (cf->cursor < vec_len (cf->current_command))
1946         {
1947           /* if we are in the middle of a line, complete only if
1948            * the cursor points to whitespace */
1949           if (isspace (cf->current_command[cf->cursor]))
1950             {
1951               /* save and clear any input that is after the cursor */
1952               vec_resize (save, vec_len (cf->current_command) - cf->cursor);
1953               clib_memcpy (save, cf->current_command + cf->cursor,
1954                            vec_len (cf->current_command) - cf->cursor);
1955               _vec_len (cf->current_command) = cf->cursor;
1956             }
1957           else
1958             {
1959               unix_vlib_cli_output_raw (cf, uf, (u8 *) "\a", 1);
1960               break;
1961             }
1962         }
1963       possible_commands =
1964         vlib_cli_get_possible_completions (cf->current_command);
1965       if (vec_len (possible_commands) == 1)
1966         {
1967           u8 *completed = possible_commands[0];
1968           j = cf->cursor;
1969
1970           /* find the last word of current_command */
1971           while (j >= 1 && !isspace (cf->current_command[j - 1]))
1972             {
1973               unix_vlib_cli_output_cursor_left (cf, uf);
1974               cf->cursor--;
1975               j--;
1976             }
1977           _vec_len (cf->current_command) = j;
1978
1979           /* replace it with the newly expanded command */
1980           vec_append (cf->current_command, completed);
1981
1982           /* echo to the terminal */
1983           unix_vlib_cli_output_cooked (cf, uf, completed,
1984                                        vec_len (completed));
1985
1986           /* add one trailing space if needed */
1987           if (vec_len (save) == 0)
1988             {
1989               vec_add1 (cf->current_command, ' ');
1990               unix_vlib_cli_output_cooked (cf, uf, (u8 *) " ", 1);
1991             }
1992
1993           cf->cursor = vec_len (cf->current_command);
1994
1995         }
1996       else if (vec_len (possible_commands) >= 2)
1997         {
1998           u8 **possible_command;
1999           uword max_command_len = 0, min_command_len = ~0;
2000           u32 i;
2001
2002           vec_foreach (possible_command, possible_commands)
2003           {
2004             if (vec_len (*possible_command) > max_command_len)
2005               max_command_len = vec_len (*possible_command);
2006             if (vec_len (*possible_command) < min_command_len)
2007               min_command_len = vec_len (*possible_command);
2008           }
2009
2010           unix_vlib_cli_output_cooked (cf, uf, (u8 *) "\n", 1);
2011
2012           i = 0;
2013           vec_foreach (possible_command, possible_commands)
2014           {
2015             if (i + max_command_len >= cf->width)
2016               {
2017                 unix_vlib_cli_output_cooked (cf, uf, (u8 *) "\n", 1);
2018                 i = 0;
2019               }
2020             unix_vlib_cli_output_cooked (cf, uf, *possible_command,
2021                                          vec_len (*possible_command));
2022             for (j = vec_len (*possible_command); j < max_command_len + 2;
2023                  j++)
2024               {
2025                 unix_vlib_cli_output_cooked (cf, uf, (u8 *) " ", 1);
2026               }
2027             i += max_command_len + 2;
2028           }
2029
2030           unix_vlib_cli_output_cooked (cf, uf, (u8 *) "\n", 1);
2031
2032           /* rewrite prompt */
2033           unix_cli_cli_prompt (cf, uf);
2034           unix_vlib_cli_output_cooked (cf, uf, cf->current_command,
2035                                        vec_len (cf->current_command));
2036
2037           /* count length of last word */
2038           j = cf->cursor;
2039           i = 0;
2040           while (j >= 1 && !isspace (cf->current_command[j - 1]))
2041             {
2042               j--;
2043               i++;
2044             }
2045
2046           /* determine smallest common command */
2047           for (; i < min_command_len; i++)
2048             {
2049               u8 common = '\0';
2050               int stop = 0;
2051
2052               vec_foreach (possible_command, possible_commands)
2053               {
2054                 if (common == '\0')
2055                   {
2056                     common = (*possible_command)[i];
2057                   }
2058                 else if (common != (*possible_command)[i])
2059                   {
2060                     stop = 1;
2061                     break;
2062                   }
2063               }
2064
2065               if (!stop)
2066                 {
2067                   vec_add1 (cf->current_command, common);
2068                   cf->cursor++;
2069                   unix_vlib_cli_output_cooked (cf, uf, (u8 *) & common, 1);
2070                 }
2071               else
2072                 {
2073                   break;
2074                 }
2075             }
2076         }
2077       else
2078         {
2079           unix_vlib_cli_output_raw (cf, uf, (u8 *) "\a", 1);
2080         }
2081
2082       if (vec_len (save) > 0)
2083         {
2084           /* restore remaining input if tab was hit in the middle of a line */
2085           unix_vlib_cli_output_cooked (cf, uf, save, vec_len (save));
2086           cf->cursor += vec_len (save);
2087           for (j = 0; j < vec_len (save); j++, cf->cursor--)
2088             unix_vlib_cli_output_cursor_left (cf, uf);
2089           vec_append (cf->current_command, save);
2090           vec_free (save);
2091         }
2092       vec_free (possible_commands);
2093
2094       break;
2095     case UNIX_CLI_PARSE_ACTION_YANK:
2096       /* TODO */
2097       break;
2098
2099
2100     case UNIX_CLI_PARSE_ACTION_PAGER_QUIT:
2101     pager_quit:
2102       unix_cli_pager_prompt_erase (cf, uf);
2103       unix_cli_pager_reset (cf);
2104       unix_cli_cli_prompt (cf, uf);
2105       break;
2106
2107     case UNIX_CLI_PARSE_ACTION_PAGER_NEXT:
2108     case UNIX_CLI_PARSE_ACTION_PAGER_PGDN:
2109       /* show next page of the buffer */
2110       if (cf->height + cf->pager_start <= vec_len (cf->pager_index))
2111         {
2112           u8 *line = NULL;
2113           unix_cli_pager_index_t *pi = NULL;
2114
2115           int m = cf->pager_start + (cf->height - 1);
2116           unix_cli_pager_prompt_erase (cf, uf);
2117           for (j = m;
2118                j < vec_len (cf->pager_index) && cf->pager_start < m;
2119                j++, cf->pager_start++)
2120             {
2121               pi = &cf->pager_index[j];
2122               line = cf->pager_vector[pi->line] + pi->offset;
2123               unix_vlib_cli_output_cooked (cf, uf, line, pi->length);
2124             }
2125           /* if the last line didn't end in newline, add a newline */
2126           if (pi && line[pi->length - 1] != '\n')
2127             unix_vlib_cli_output_cooked (cf, uf, (u8 *) "\n", 1);
2128           unix_cli_pager_prompt (cf, uf);
2129         }
2130       else
2131         {
2132           if (action == UNIX_CLI_PARSE_ACTION_PAGER_NEXT)
2133             /* no more in buffer, exit, but only if it was <space> */
2134             goto pager_quit;
2135         }
2136       break;
2137
2138     case UNIX_CLI_PARSE_ACTION_PAGER_DN:
2139     case UNIX_CLI_PARSE_ACTION_PAGER_CRLF:
2140       /* display the next line of the buffer */
2141       if (cf->height + cf->pager_start <= vec_len (cf->pager_index))
2142         {
2143           u8 *line;
2144           unix_cli_pager_index_t *pi;
2145
2146           unix_cli_pager_prompt_erase (cf, uf);
2147           pi = &cf->pager_index[cf->pager_start + (cf->height - 1)];
2148           line = cf->pager_vector[pi->line] + pi->offset;
2149           unix_vlib_cli_output_cooked (cf, uf, line, pi->length);
2150           cf->pager_start++;
2151           /* if the last line didn't end in newline, add a newline */
2152           if (line[pi->length - 1] != '\n')
2153             unix_vlib_cli_output_cooked (cf, uf, (u8 *) "\n", 1);
2154           unix_cli_pager_prompt (cf, uf);
2155         }
2156       else
2157         {
2158           if (action == UNIX_CLI_PARSE_ACTION_PAGER_CRLF)
2159             /* no more in buffer, exit, but only if it was <enter> */
2160             goto pager_quit;
2161         }
2162
2163       break;
2164
2165     case UNIX_CLI_PARSE_ACTION_PAGER_UP:
2166       /* scroll the page back one line */
2167       if (cf->pager_start > 0)
2168         {
2169           u8 *line = NULL;
2170           unix_cli_pager_index_t *pi = NULL;
2171
2172           cf->pager_start--;
2173           if (cf->ansi_capable)
2174             {
2175               pi = &cf->pager_index[cf->pager_start];
2176               line = cf->pager_vector[pi->line] + pi->offset;
2177               unix_cli_pager_prompt_erase (cf, uf);
2178               unix_vlib_cli_output_cooked (cf, uf, (u8 *) ANSI_SCROLLDN,
2179                                            sizeof (ANSI_SCROLLDN) - 1);
2180               unix_vlib_cli_output_cooked (cf, uf, (u8 *) ANSI_SAVECURSOR,
2181                                            sizeof (ANSI_SAVECURSOR) - 1);
2182               unix_cli_ansi_cursor (cf, uf, 1, 1);
2183               unix_vlib_cli_output_cooked (cf, uf, (u8 *) ANSI_CLEARLINE,
2184                                            sizeof (ANSI_CLEARLINE) - 1);
2185               unix_vlib_cli_output_cooked (cf, uf, line, pi->length);
2186               unix_vlib_cli_output_cooked (cf, uf, (u8 *) ANSI_RESTCURSOR,
2187                                            sizeof (ANSI_RESTCURSOR) - 1);
2188               unix_cli_pager_prompt_erase (cf, uf);
2189               unix_cli_pager_prompt (cf, uf);
2190             }
2191           else
2192             {
2193               int m = cf->pager_start + (cf->height - 1);
2194               unix_cli_pager_prompt_erase (cf, uf);
2195               for (j = cf->pager_start;
2196                    j < vec_len (cf->pager_index) && j < m; j++)
2197                 {
2198                   pi = &cf->pager_index[j];
2199                   line = cf->pager_vector[pi->line] + pi->offset;
2200                   unix_vlib_cli_output_cooked (cf, uf, line, pi->length);
2201                 }
2202               /* if the last line didn't end in newline, add a newline */
2203               if (pi && line[pi->length - 1] != '\n')
2204                 unix_vlib_cli_output_cooked (cf, uf, (u8 *) "\n", 1);
2205               unix_cli_pager_prompt (cf, uf);
2206             }
2207         }
2208       break;
2209
2210     case UNIX_CLI_PARSE_ACTION_PAGER_TOP:
2211       /* back to the first page of the buffer */
2212       if (cf->pager_start > 0)
2213         {
2214           u8 *line = NULL;
2215           unix_cli_pager_index_t *pi = NULL;
2216
2217           cf->pager_start = 0;
2218           int m = cf->pager_start + (cf->height - 1);
2219           unix_cli_pager_prompt_erase (cf, uf);
2220           for (j = cf->pager_start; j < vec_len (cf->pager_index) && j < m;
2221                j++)
2222             {
2223               pi = &cf->pager_index[j];
2224               line = cf->pager_vector[pi->line] + pi->offset;
2225               unix_vlib_cli_output_cooked (cf, uf, line, pi->length);
2226             }
2227           /* if the last line didn't end in newline, add a newline */
2228           if (pi && line[pi->length - 1] != '\n')
2229             unix_vlib_cli_output_cooked (cf, uf, (u8 *) "\n", 1);
2230           unix_cli_pager_prompt (cf, uf);
2231         }
2232       break;
2233
2234     case UNIX_CLI_PARSE_ACTION_PAGER_BOTTOM:
2235       /* skip to the last page of the buffer */
2236       if (cf->pager_start < vec_len (cf->pager_index) - (cf->height - 1))
2237         {
2238           u8 *line = NULL;
2239           unix_cli_pager_index_t *pi = NULL;
2240
2241           cf->pager_start = vec_len (cf->pager_index) - (cf->height - 1);
2242           unix_cli_pager_prompt_erase (cf, uf);
2243           unix_cli_pager_message (cf, uf, "skipping", "\n");
2244           for (j = cf->pager_start; j < vec_len (cf->pager_index); j++)
2245             {
2246               pi = &cf->pager_index[j];
2247               line = cf->pager_vector[pi->line] + pi->offset;
2248               unix_vlib_cli_output_cooked (cf, uf, line, pi->length);
2249             }
2250           /* if the last line didn't end in newline, add a newline */
2251           if (pi && line[pi->length - 1] != '\n')
2252             unix_vlib_cli_output_cooked (cf, uf, (u8 *) "\n", 1);
2253           unix_cli_pager_prompt (cf, uf);
2254         }
2255       break;
2256
2257     case UNIX_CLI_PARSE_ACTION_PAGER_PGUP:
2258       /* wander back one page in the buffer */
2259       if (cf->pager_start > 0)
2260         {
2261           u8 *line = NULL;
2262           unix_cli_pager_index_t *pi = NULL;
2263           int m;
2264
2265           if (cf->pager_start >= cf->height)
2266             cf->pager_start -= cf->height - 1;
2267           else
2268             cf->pager_start = 0;
2269           m = cf->pager_start + cf->height - 1;
2270           unix_cli_pager_prompt_erase (cf, uf);
2271           for (j = cf->pager_start; j < vec_len (cf->pager_index) && j < m;
2272                j++)
2273             {
2274               pi = &cf->pager_index[j];
2275               line = cf->pager_vector[pi->line] + pi->offset;
2276               unix_vlib_cli_output_cooked (cf, uf, line, pi->length);
2277             }
2278           /* if the last line didn't end in newline, add a newline */
2279           if (pi && line[pi->length - 1] != '\n')
2280             unix_vlib_cli_output_cooked (cf, uf, (u8 *) "\n", 1);
2281           unix_cli_pager_prompt (cf, uf);
2282         }
2283       break;
2284
2285     case UNIX_CLI_PARSE_ACTION_PAGER_REDRAW:
2286       /* Redraw the current pager screen */
2287       unix_cli_pager_redraw (cf, uf);
2288       break;
2289
2290     case UNIX_CLI_PARSE_ACTION_PAGER_SEARCH:
2291       /* search forwards in the buffer */
2292       break;
2293
2294
2295     case UNIX_CLI_PARSE_ACTION_CRLF:
2296     crlf:
2297       unix_vlib_cli_output_cooked (cf, uf, (u8 *) "\n", 1);
2298
2299       if (cf->has_history && cf->history_limit)
2300         {
2301           if (cf->command_history
2302               && vec_len (cf->command_history) >= cf->history_limit)
2303             {
2304               vec_free (cf->command_history[0]);
2305               vec_delete (cf->command_history, 1, 0);
2306             }
2307           /* Don't add blank lines to the cmd history */
2308           if (vec_len (cf->current_command))
2309             {
2310               /* Don't duplicate the previous command */
2311               j = vec_len (cf->command_history);
2312               if (j == 0 ||
2313                   (vec_len (cf->current_command) !=
2314                    vec_len (cf->command_history[j - 1])
2315                    || memcmp (cf->current_command, cf->command_history[j - 1],
2316                               vec_len (cf->current_command)) != 0))
2317                 {
2318                   /* copy the command to the history */
2319                   u8 *c = 0;
2320                   vec_append (c, cf->current_command);
2321                   vec_add1 (cf->command_history, c);
2322                   cf->command_number++;
2323                 }
2324             }
2325           cf->excursion = vec_len (cf->command_history);
2326         }
2327
2328       cf->search_mode = 0;
2329       vec_reset_length (cf->search_key);
2330       cf->cursor = 0;
2331
2332       return 0;
2333
2334     case UNIX_CLI_PARSE_ACTION_PARTIALMATCH:
2335     case UNIX_CLI_PARSE_ACTION_NOMATCH:
2336       if (vec_len (cf->pager_index))
2337         {
2338           /* no-op for now */
2339         }
2340       else if (cf->has_history && cf->search_mode != 0 && isprint (input))
2341         {
2342           int k, limit, offset;
2343           u8 *item;
2344
2345           vec_add1 (cf->search_key, input);
2346
2347         search_again:
2348           for (j = 0; j < vec_len (cf->command_history); j++)
2349             {
2350               if (cf->excursion > (i32) vec_len (cf->command_history) - 1)
2351                 cf->excursion = 0;
2352               else if (cf->excursion < 0)
2353                 cf->excursion = vec_len (cf->command_history) - 1;
2354
2355               item = cf->command_history[cf->excursion];
2356
2357               limit = (vec_len (cf->search_key) > vec_len (item)) ?
2358                 vec_len (item) : vec_len (cf->search_key);
2359
2360               for (offset = 0; offset <= vec_len (item) - limit; offset++)
2361                 {
2362                   for (k = 0; k < limit; k++)
2363                     {
2364                       if (item[k + offset] != cf->search_key[k])
2365                         goto next_offset;
2366                     }
2367                   goto found_at_offset;
2368
2369                 next_offset:
2370                   ;
2371                 }
2372               goto next;
2373
2374             found_at_offset:
2375               for (; cf->cursor > 0; cf->cursor--)
2376                 {
2377                   unix_vlib_cli_output_cursor_left (cf, uf);
2378                   unix_vlib_cli_output_cooked (cf, uf, (u8 *) " ", 1);
2379                   unix_vlib_cli_output_cursor_left (cf, uf);
2380                 }
2381
2382               vec_validate (cf->current_command, vec_len (item) - 1);
2383               clib_memcpy (cf->current_command, item, vec_len (item));
2384               _vec_len (cf->current_command) = vec_len (item);
2385
2386               unix_vlib_cli_output_cooked (cf, uf, cf->current_command,
2387                                            vec_len (cf->current_command));
2388               cf->cursor = vec_len (cf->current_command);
2389               goto found;
2390
2391             next:
2392               cf->excursion += cf->search_mode;
2393             }
2394
2395           unix_vlib_cli_output_cooked (cf, uf, (u8 *) "\nNo match...", 12);
2396           vec_reset_length (cf->search_key);
2397           vec_reset_length (cf->current_command);
2398           cf->search_mode = 0;
2399           cf->cursor = 0;
2400           goto crlf;
2401         }
2402       else if (isprint (input)) /* skip any errant control codes */
2403         {
2404           if (cf->cursor == vec_len (cf->current_command))
2405             {
2406               /* Append to end */
2407               vec_add1 (cf->current_command, input);
2408               cf->cursor++;
2409
2410               /* Echo the character back to the client */
2411               unix_vlib_cli_output_cooked (cf, uf, &input, 1);
2412             }
2413           else
2414             {
2415               /* Insert at cursor: resize +1 byte, move everything over */
2416               j = vec_len (cf->current_command) - cf->cursor;
2417               vec_add1 (cf->current_command, (u8) 'A');
2418               memmove (cf->current_command + cf->cursor + 1,
2419                        cf->current_command + cf->cursor, j);
2420               cf->current_command[cf->cursor] = input;
2421               /* Redraw the line */
2422               j++;
2423               unix_vlib_cli_output_cooked (cf, uf,
2424                                            cf->current_command + cf->cursor,
2425                                            j);
2426               cf->cursor += j;
2427               j--;
2428               /* Put terminal cursor back */
2429               for (; j > 0; j--, cf->cursor--)
2430                 unix_vlib_cli_output_cursor_left (cf, uf);
2431             }
2432         }
2433       else
2434         {
2435           /* no-op - not printable or otherwise not actionable */
2436         }
2437
2438     found:
2439
2440       break;
2441
2442     case UNIX_CLI_PARSE_ACTION_TELNETIAC:
2443       break;
2444     }
2445   return 1;
2446 }
2447
2448 /** @brief Process input bytes on a stream to provide line editing and
2449  * command history in the CLI. */
2450 static int
2451 unix_cli_line_edit (unix_cli_main_t * cm, unix_main_t * um,
2452                     clib_file_main_t * fm, unix_cli_file_t * cf)
2453 {
2454   clib_file_t *uf = pool_elt_at_index (fm->file_pool, cf->clib_file_index);
2455   int i;
2456
2457   for (i = 0; i < vec_len (cf->input_vector); i++)
2458     {
2459       unix_cli_parse_action_t action;
2460       i32 matched = 0;
2461       unix_cli_parse_actions_t *a;
2462
2463       /* If we're in the pager mode, search the pager actions */
2464       a =
2465         vec_len (cf->pager_index) ? unix_cli_parse_pager :
2466         unix_cli_parse_strings;
2467
2468       /* See if the input buffer is some sort of control code */
2469       action = unix_cli_match_action (a, &cf->input_vector[i],
2470                                       vec_len (cf->input_vector) - i,
2471                                       &matched);
2472
2473       switch (action)
2474         {
2475         case UNIX_CLI_PARSE_ACTION_PARTIALMATCH:
2476           if (i)
2477             {
2478               /* There was a partial match which means we need more bytes
2479                * than the input buffer currently has.
2480                * Since the bytes before here have been processed, shift
2481                * the remaining contents to the start of the input buffer.
2482                */
2483               vec_delete (cf->input_vector, i, 0);
2484             }
2485           return 1;             /* wait for more */
2486
2487         case UNIX_CLI_PARSE_ACTION_TELNETIAC:
2488           /* process telnet options */
2489           matched = unix_cli_process_telnet (um, cf, uf,
2490                                              cf->input_vector + i,
2491                                              vec_len (cf->input_vector) - i);
2492           if (matched < 0)
2493             {
2494               /* There was a partial match which means we need more bytes
2495                * than the input buffer currently has.
2496                */
2497               if (i)
2498                 {
2499                   /*
2500                    * Since the bytes before here have been processed, shift
2501                    * the remaining contents to the start of the input buffer.
2502                    */
2503                   vec_delete (cf->input_vector, i, 0);
2504                 }
2505               return 1;         /* wait for more */
2506             }
2507           break;
2508
2509         default:
2510           /* If telnet option processing switched us to line mode, get us
2511            * out of here!
2512            */
2513           if (cf->line_mode)
2514             {
2515               vec_delete (cf->input_vector, i, 0);
2516               vec_free (cf->current_command);
2517               cf->current_command = cf->input_vector;
2518               return 0;
2519             }
2520
2521           /* process the action */
2522           if (!unix_cli_line_process_one (cm, um, cf, uf,
2523                                           cf->input_vector[i], action))
2524             {
2525               /* CRLF found. Consume the bytes from the input_vector */
2526               vec_delete (cf->input_vector, i + matched, 0);
2527               /* And tell our caller to execute cf->input_command */
2528               return 0;
2529             }
2530         }
2531
2532       i += matched;
2533     }
2534
2535   vec_reset_length (cf->input_vector);
2536   return 1;
2537 }
2538
2539 /** @brief Process input to a CLI session. */
2540 static void
2541 unix_cli_process_input (unix_cli_main_t * cm, uword cli_file_index)
2542 {
2543   unix_main_t *um = &unix_main;
2544   clib_file_main_t *fm = &file_main;
2545   clib_file_t *uf;
2546   unix_cli_file_t *cf = pool_elt_at_index (cm->cli_file_pool, cli_file_index);
2547   unformat_input_t input;
2548   int vlib_parse_eval (u8 *);
2549
2550   cm->current_input_file_index = cli_file_index;
2551
2552 more:
2553   /* Try vlibplex first.  Someday... */
2554   if (0 && vlib_parse_eval (cf->input_vector) == 0)
2555     goto done;
2556
2557
2558   if (cf->line_mode)
2559     {
2560       /* just treat whatever we got as a complete line of input */
2561       cf->current_command = cf->input_vector;
2562     }
2563   else
2564     {
2565       /* Line edit, echo, etc. */
2566       if (unix_cli_line_edit (cm, um, fm, cf))
2567         /* want more input */
2568         return;
2569     }
2570
2571   if (um->log_fd)
2572     {
2573       static u8 *lv;
2574       vec_reset_length (lv);
2575       lv = format (lv, "%U[%d]: %v",
2576                    format_timeval, 0 /* current bat-time */ ,
2577                    0 /* current bat-format */ ,
2578                    cli_file_index, cf->current_command);
2579       if ((vec_len (cf->current_command) > 0) &&
2580           (cf->current_command[vec_len (cf->current_command) - 1] != '\n'))
2581         lv = format (lv, "\n");
2582       int rv __attribute__ ((unused)) = write (um->log_fd, lv, vec_len (lv));
2583     }
2584
2585   /* Run the command through the macro processor */
2586   if (vec_len (cf->current_command))
2587     {
2588       u8 *expanded;
2589       vec_validate (cf->current_command, vec_len (cf->current_command));
2590       cf->current_command[vec_len (cf->current_command) - 1] = 0;
2591       /* The macro expander expects proper C-strings, not vectors */
2592       expanded = (u8 *) clib_macro_eval (&cf->macro_main,
2593                                          (i8 *) cf->current_command,
2594                                          1 /* complain */ ,
2595                                          0 /* level */ ,
2596                                          8 /* max_level */ );
2597       /* Macro processor NULL terminates the return */
2598       _vec_len (expanded) -= 1;
2599       vec_reset_length (cf->current_command);
2600       vec_append (cf->current_command, expanded);
2601       vec_free (expanded);
2602     }
2603
2604   /* Build an unformat structure around our command */
2605   unformat_init_vector (&input, cf->current_command);
2606
2607   /* Remove leading white space from input. */
2608   (void) unformat (&input, "");
2609
2610   cf->pager_start = 0;          /* start a new pager session */
2611
2612   if (unformat_check_input (&input) != UNFORMAT_END_OF_INPUT)
2613     vlib_cli_input (um->vlib_main, &input, unix_vlib_cli_output,
2614                     cli_file_index);
2615
2616   /* Zero buffer since otherwise unformat_free will call vec_free on it. */
2617   input.buffer = 0;
2618
2619   unformat_free (&input);
2620
2621   /* Re-fetch pointer since pool may have moved. */
2622   cf = pool_elt_at_index (cm->cli_file_pool, cli_file_index);
2623   uf = pool_elt_at_index (fm->file_pool, cf->clib_file_index);
2624
2625 done:
2626   /* reset vector; we'll re-use it later  */
2627   if (cf->line_mode)
2628     {
2629       vec_reset_length (cf->input_vector);
2630       cf->current_command = 0;
2631     }
2632   else
2633     {
2634       vec_reset_length (cf->current_command);
2635     }
2636
2637   if (cf->no_pager == 2)
2638     {
2639       /* Pager was programmatically disabled */
2640       unix_cli_pager_message (cf, uf, "pager buffer overflowed", "\n");
2641       cf->no_pager = um->cli_no_pager;
2642     }
2643
2644   if (vec_len (cf->pager_index) == 0
2645       || vec_len (cf->pager_index) < cf->height)
2646     {
2647       /* There was no need for the pager */
2648       unix_cli_pager_reset (cf);
2649
2650       /* Prompt. */
2651       unix_cli_cli_prompt (cf, uf);
2652     }
2653   else
2654     {
2655       /* Display the pager prompt */
2656       unix_cli_pager_prompt (cf, uf);
2657     }
2658
2659   /* Any residual data in the input vector? */
2660   if (vec_len (cf->input_vector))
2661     goto more;
2662
2663   /* For non-interactive sessions send a NUL byte.
2664    * Specifically this is because vppctl needs to see some traffic in
2665    * order to move on to closing the session. Commands with no output
2666    * would thus cause vppctl to hang indefinitely in non-interactive mode
2667    * since there is also no prompt sent after the command completes.
2668    */
2669   if (!cf->is_interactive)
2670     unix_vlib_cli_output_raw (cf, uf, (u8 *) "\0", 1);
2671 }
2672
2673 /** Destroy a CLI session.
2674  * @note If we destroy the @c stdin session this additionally signals
2675  *       the shutdown of VPP.
2676  */
2677 static void
2678 unix_cli_kill (unix_cli_main_t * cm, uword cli_file_index)
2679 {
2680   unix_main_t *um = &unix_main;
2681   clib_file_main_t *fm = &file_main;
2682   unix_cli_file_t *cf;
2683   clib_file_t *uf;
2684   int i;
2685
2686   /* Validate cli_file_index */
2687   if (pool_is_free_index (cm->cli_file_pool, cli_file_index))
2688     return;
2689
2690   vec_foreach_index (i, cm->new_sessions)
2691     {
2692       unix_cli_new_session_t *ns = vec_elt_at_index (cm->new_sessions, i);
2693
2694       if (ns->cf_index == cli_file_index)
2695         {
2696           vec_del1 (cm->new_sessions, i);
2697           break;
2698         }
2699     }
2700
2701   cf = pool_elt_at_index (cm->cli_file_pool, cli_file_index);
2702   uf = pool_elt_at_index (fm->file_pool, cf->clib_file_index);
2703
2704   /* Quit/EOF on stdin means quit program. */
2705   if (uf->file_descriptor == STDIN_FILENO)
2706     clib_longjmp (&um->vlib_main->main_loop_exit, VLIB_MAIN_LOOP_EXIT_CLI);
2707
2708   vec_free (cf->current_command);
2709   vec_free (cf->search_key);
2710
2711   for (i = 0; i < vec_len (cf->command_history); i++)
2712     vec_free (cf->command_history[i]);
2713
2714   vec_free (cf->command_history);
2715   vec_free (cf->input_vector);
2716
2717   clib_file_del (fm, uf);
2718
2719   unix_cli_file_free (cf);
2720   clib_macro_free (&cf->macro_main);
2721   pool_put (cm->cli_file_pool, cf);
2722 }
2723
2724 /** Handle system events. */
2725 static uword
2726 unix_cli_process (vlib_main_t * vm,
2727                   vlib_node_runtime_t * rt, vlib_frame_t * f)
2728 {
2729   unix_cli_main_t *cm = &unix_cli_main;
2730   uword i, *data = 0;
2731
2732   while (1)
2733     {
2734       unix_cli_process_event_type_t event_type;
2735       vlib_process_wait_for_event (vm);
2736       event_type = vlib_process_get_events (vm, &data);
2737
2738       switch (event_type)
2739         {
2740         case UNIX_CLI_PROCESS_EVENT_READ_READY:
2741           for (i = 0; i < vec_len (data); i++)
2742             unix_cli_process_input (cm, data[i]);
2743           break;
2744
2745         case UNIX_CLI_PROCESS_EVENT_QUIT:
2746           /* Kill this process. */
2747           for (i = 0; i < vec_len (data); i++)
2748             unix_cli_kill (cm, data[i]);
2749           goto done;
2750         }
2751
2752       if (data)
2753         _vec_len (data) = 0;
2754     }
2755
2756 done:
2757   vec_free (data);
2758
2759   vlib_node_set_state (vm, rt->node_index, VLIB_NODE_STATE_DISABLED);
2760
2761   /* Add node index so we can re-use this process later. */
2762   vec_add1 (cm->unused_cli_process_node_indices, rt->node_index);
2763
2764   return 0;
2765 }
2766
2767 /** Called when a CLI session file descriptor can be written to without
2768  * blocking. */
2769 static clib_error_t *
2770 unix_cli_write_ready (clib_file_t * uf)
2771 {
2772   unix_cli_main_t *cm = &unix_cli_main;
2773   unix_cli_file_t *cf;
2774   int n;
2775
2776   cf = pool_elt_at_index (cm->cli_file_pool, uf->private_data);
2777
2778   /* Flush output vector. */
2779   if (cf->is_socket)
2780     /* If it's a socket we use MSG_NOSIGNAL to prevent SIGPIPE */
2781     n = send (uf->file_descriptor,
2782               cf->output_vector, vec_len (cf->output_vector), MSG_NOSIGNAL);
2783   else
2784     n = write (uf->file_descriptor,
2785                cf->output_vector, vec_len (cf->output_vector));
2786
2787   if (n < 0 && errno != EAGAIN)
2788     {
2789       if (errno == EPIPE)
2790         {
2791           /* connection closed on us */
2792           unix_main_t *um = &unix_main;
2793           cf->has_epipe = 1;
2794           vlib_process_signal_event (um->vlib_main, cf->process_node_index,
2795                                      UNIX_CLI_PROCESS_EVENT_QUIT,
2796                                      uf->private_data);
2797         }
2798       else
2799         {
2800           return clib_error_return_unix (0, "write");
2801         }
2802     }
2803
2804   else if (n > 0)
2805     unix_cli_del_pending_output (uf, cf, n);
2806
2807   return /* no error */ 0;
2808 }
2809
2810 /** Called when a CLI session file descriptor has data to be read. */
2811 static clib_error_t *
2812 unix_cli_read_ready (clib_file_t * uf)
2813 {
2814   unix_main_t *um = &unix_main;
2815   unix_cli_main_t *cm = &unix_cli_main;
2816   unix_cli_file_t *cf;
2817   uword l;
2818   int n, n_read, n_try;
2819
2820   cf = pool_elt_at_index (cm->cli_file_pool, uf->private_data);
2821
2822   n = n_try = 4096;
2823   while (n == n_try)
2824     {
2825       l = vec_len (cf->input_vector);
2826       vec_resize (cf->input_vector, l + n_try);
2827
2828       n = read (uf->file_descriptor, cf->input_vector + l, n_try);
2829
2830       /* Error? */
2831       if (n < 0 && errno != EAGAIN)
2832         return clib_error_return_unix (0, "read");
2833
2834       n_read = n < 0 ? 0 : n;
2835       _vec_len (cf->input_vector) = l + n_read;
2836     }
2837
2838   if (!(n < 0))
2839     vlib_process_signal_event (um->vlib_main,
2840                                cf->process_node_index,
2841                                (n_read == 0
2842                                 ? UNIX_CLI_PROCESS_EVENT_QUIT
2843                                 : UNIX_CLI_PROCESS_EVENT_READ_READY),
2844                                /* event data */ uf->private_data);
2845
2846   return /* no error */ 0;
2847 }
2848
2849 /** Called when a CLI session file descriptor has an error condition. */
2850 static clib_error_t *
2851 unix_cli_error_detected (clib_file_t * uf)
2852 {
2853   unix_main_t *um = &unix_main;
2854   unix_cli_main_t *cm = &unix_cli_main;
2855   unix_cli_file_t *cf;
2856
2857   cf = pool_elt_at_index (cm->cli_file_pool, uf->private_data);
2858   cf->has_epipe = 1;            /* prevent writes while the close is pending */
2859   vlib_process_signal_event (um->vlib_main,
2860                              cf->process_node_index,
2861                              UNIX_CLI_PROCESS_EVENT_QUIT,
2862                              /* event data */ uf->private_data);
2863
2864   return /* no error */ 0;
2865 }
2866
2867 /** Store a new CLI session.
2868  * @param name The name of the session.
2869  * @param fd   The file descriptor for the session I/O.
2870  * @return The session ID.
2871  */
2872 static u32
2873 unix_cli_file_add (unix_cli_main_t * cm, char *name, int fd)
2874 {
2875   unix_main_t *um = &unix_main;
2876   clib_file_main_t *fm = &file_main;
2877   vlib_node_main_t *nm = &vlib_get_main ()->node_main;
2878   unix_cli_file_t *cf;
2879   clib_file_t template = { 0 };
2880   vlib_main_t *vm = um->vlib_main;
2881   vlib_node_t *n = 0;
2882   u8 *file_desc = 0;
2883
2884   file_desc = format (0, "%s", name);
2885
2886   name = (char *) format (0, "unix-cli-%s", name);
2887
2888   if (vec_len (cm->unused_cli_process_node_indices) > 0)
2889     {
2890       uword l = vec_len (cm->unused_cli_process_node_indices);
2891       int i;
2892       vlib_main_t *this_vlib_main;
2893       u8 *old_name = 0;
2894
2895       /*
2896        * Nodes are bulk-copied, so node name pointers are shared.
2897        * Find the cli node in all graph replicas, and give all of them
2898        * the same new name.
2899        * Then, throw away the old shared name-vector.
2900        */
2901       for (i = 0; i < vlib_get_n_threads (); i++)
2902         {
2903           this_vlib_main = vlib_get_main_by_index (i);
2904           if (this_vlib_main == 0)
2905             continue;
2906           n = vlib_get_node (this_vlib_main,
2907                              cm->unused_cli_process_node_indices[l - 1]);
2908           old_name = n->name;
2909           n->name = (u8 *) name;
2910         }
2911       ASSERT (old_name);
2912       hash_unset (nm->node_by_name, old_name);
2913       hash_set (nm->node_by_name, name, n->index);
2914       vec_free (old_name);
2915
2916       vlib_node_set_state (vm, n->index, VLIB_NODE_STATE_POLLING);
2917       _vec_len (cm->unused_cli_process_node_indices) = l - 1;
2918     }
2919   else
2920     {
2921       static vlib_node_registration_t r = {
2922         .function = unix_cli_process,
2923         .type = VLIB_NODE_TYPE_PROCESS,
2924         .process_log2_n_stack_bytes = 18,
2925       };
2926
2927       r.name = name;
2928
2929       vlib_worker_thread_barrier_sync (vm);
2930
2931       vlib_register_node (vm, &r);
2932       vec_free (name);
2933
2934       n = vlib_get_node (vm, r.index);
2935       vlib_worker_thread_node_runtime_update ();
2936       vlib_worker_thread_barrier_release (vm);
2937     }
2938
2939   pool_get (cm->cli_file_pool, cf);
2940   clib_memset (cf, 0, sizeof (*cf));
2941   clib_macro_init (&cf->macro_main);
2942
2943   template.read_function = unix_cli_read_ready;
2944   template.write_function = unix_cli_write_ready;
2945   template.error_function = unix_cli_error_detected;
2946   template.file_descriptor = fd;
2947   template.private_data = cf - cm->cli_file_pool;
2948   template.description = file_desc;
2949
2950   cf->process_node_index = n->index;
2951   cf->clib_file_index = clib_file_add (fm, &template);
2952   cf->output_vector = 0;
2953   cf->input_vector = 0;
2954   vec_validate (cf->current_command, 0);
2955   _vec_len (cf->current_command) = 0;
2956
2957   vlib_start_process (vm, n->runtime_index);
2958
2959   vlib_process_t *p = vlib_get_process_from_node (vm, n);
2960   p->output_function = unix_vlib_cli_output;
2961   p->output_function_arg = cf - cm->cli_file_pool;
2962
2963   return cf - cm->cli_file_pool;
2964 }
2965
2966 /** Telnet listening socket has a new connection. */
2967 static clib_error_t *
2968 unix_cli_listen_read_ready (clib_file_t * uf)
2969 {
2970   unix_main_t *um = &unix_main;
2971   clib_file_main_t *fm = &file_main;
2972   unix_cli_main_t *cm = &unix_cli_main;
2973   clib_socket_t *s = &um->cli_listen_socket;
2974   clib_socket_t client;
2975   char *client_name;
2976   clib_error_t *error;
2977   unix_cli_file_t *cf;
2978   u32 cf_index;
2979   int one;
2980
2981   error = clib_socket_accept (s, &client);
2982   if (error)
2983     return error;
2984
2985   /* Disable Nagle, ignore any errors doing so eg on PF_LOCAL socket */
2986   one = 1;
2987   (void) setsockopt (client.fd, IPPROTO_TCP, TCP_NODELAY,
2988                      (void *) &one, sizeof (one));
2989
2990   client_name = (char *) format (0, "%U%c", format_sockaddr, &client.peer, 0);
2991
2992   cf_index = unix_cli_file_add (cm, client_name, client.fd);
2993   cf = pool_elt_at_index (cm->cli_file_pool, cf_index);
2994   cf->is_socket = 1;
2995
2996   /* No longer need CLIB version of socket. */
2997   clib_socket_free (&client);
2998   vec_free (client_name);
2999
3000   /* if we're supposed to run telnet session in character mode (default) */
3001   if (um->cli_line_mode == 0)
3002     {
3003       /*
3004        * Set telnet client character mode, echo on, suppress "go-ahead".
3005        * Technically these should be negotiated, but this works.
3006        */
3007       u8 charmode_option[] = {
3008         IAC, WONT, TELOPT_LINEMODE,     /* server will do char-by-char */
3009         IAC, DONT, TELOPT_LINEMODE,     /* client should do char-by-char */
3010         IAC, WILL, TELOPT_SGA,  /* server willl supress GA */
3011         IAC, DO, TELOPT_SGA,    /* client should supress Go Ahead */
3012         IAC, WILL, TELOPT_ECHO, /* server will do echo */
3013         IAC, DONT, TELOPT_ECHO, /* client should not echo */
3014         IAC, DO, TELOPT_TTYPE,  /* client should tell us its term type */
3015         IAC, SB, TELOPT_TTYPE, 1, IAC, SE,      /* now tell me ttype */
3016         IAC, DO, TELOPT_NAWS,   /* client should tell us its window sz */
3017         IAC, SB, TELOPT_NAWS, 1, IAC, SE,       /* now tell me window size */
3018       };
3019
3020       /* Enable history on this CLI */
3021       cf->history_limit = um->cli_history_limit;
3022       cf->has_history = cf->history_limit != 0;
3023
3024       /* This is an interactive session until we decide otherwise */
3025       cf->is_interactive = 1;
3026
3027       /* Make sure this session is in line mode */
3028       cf->line_mode = 0;
3029
3030       /* We need CRLF */
3031       cf->crlf_mode = 1;
3032
3033       /* Setup the pager */
3034       cf->no_pager = um->cli_no_pager;
3035
3036       /* Default terminal dimensions, should the terminal
3037        * fail to provide any.
3038        */
3039       cf->width = UNIX_CLI_DEFAULT_TERMINAL_WIDTH;
3040       cf->height = UNIX_CLI_DEFAULT_TERMINAL_HEIGHT;
3041
3042       /* Send the telnet options */
3043       uf = pool_elt_at_index (fm->file_pool, cf->clib_file_index);
3044       unix_vlib_cli_output_raw (cf, uf, charmode_option,
3045                                 ARRAY_LEN (charmode_option));
3046
3047       if (cm->new_session_process_node_index == ~0)
3048         {
3049           /* Create thw new session deadline process */
3050           cm->new_session_process_node_index =
3051             vlib_process_create (um->vlib_main, "unix-cli-new-session",
3052                                  unix_cli_new_session_process,
3053                                  16 /* log2_n_stack_bytes */ );
3054         }
3055
3056       /* In case the client doesn't negotiate terminal type, register
3057        * our session with a process that will emit the prompt if
3058        * a deadline passes */
3059       vlib_process_signal_event (um->vlib_main,
3060                                  cm->new_session_process_node_index,
3061                                  UNIX_CLI_NEW_SESSION_EVENT_ADD, cf_index);
3062
3063     }
3064
3065   return error;
3066 }
3067
3068 /** The system terminal has informed us that the window size
3069  * has changed.
3070  */
3071 static void
3072 unix_cli_resize_interrupt (int signum)
3073 {
3074   clib_file_main_t *fm = &file_main;
3075   unix_cli_main_t *cm = &unix_cli_main;
3076   unix_cli_file_t *cf = pool_elt_at_index (cm->cli_file_pool,
3077                                            cm->stdin_cli_file_index);
3078   clib_file_t *uf = pool_elt_at_index (fm->file_pool, cf->clib_file_index);
3079   struct winsize ws;
3080   (void) signum;
3081
3082   /* Terminal resized, fetch the new size */
3083   if (ioctl (STDIN_FILENO, TIOCGWINSZ, &ws) < 0)
3084     {
3085       /* "Should never happen..." */
3086       clib_unix_warning ("TIOCGWINSZ");
3087       /* We can't trust ws.XXX... */
3088       return;
3089     }
3090
3091   cf->width = ws.ws_col;
3092   if (cf->width > UNIX_CLI_MAX_TERMINAL_WIDTH)
3093     cf->width = UNIX_CLI_MAX_TERMINAL_WIDTH;
3094   if (cf->width == 0)
3095     cf->width = UNIX_CLI_DEFAULT_TERMINAL_WIDTH;
3096
3097   cf->height = ws.ws_row;
3098   if (cf->height > UNIX_CLI_MAX_TERMINAL_HEIGHT)
3099     cf->height = UNIX_CLI_MAX_TERMINAL_HEIGHT;
3100   if (cf->height == 0)
3101     cf->height = UNIX_CLI_DEFAULT_TERMINAL_HEIGHT;
3102
3103   /* Reindex the pager buffer */
3104   unix_cli_pager_reindex (cf);
3105
3106   /* Redraw the page */
3107   unix_cli_pager_redraw (cf, uf);
3108 }
3109
3110 /** Handle configuration directives in the @em unix section. */
3111 static clib_error_t *
3112 unix_cli_config (vlib_main_t * vm, unformat_input_t * input)
3113 {
3114   unix_main_t *um = &unix_main;
3115   clib_file_main_t *fm = &file_main;
3116   unix_cli_main_t *cm = &unix_cli_main;
3117   int flags;
3118   clib_error_t *error = 0;
3119   unix_cli_file_t *cf;
3120   u32 cf_index;
3121   struct termios tio;
3122   struct sigaction sa;
3123   struct winsize ws;
3124   u8 *term;
3125
3126   /* We depend on unix flags being set. */
3127   if ((error = vlib_call_config_function (vm, unix_config)))
3128     return error;
3129
3130   if (um->flags & UNIX_FLAG_INTERACTIVE)
3131     {
3132       /* Set stdin to be non-blocking. */
3133       if ((flags = fcntl (STDIN_FILENO, F_GETFL, 0)) < 0)
3134         flags = 0;
3135       (void) fcntl (STDIN_FILENO, F_SETFL, flags | O_NONBLOCK);
3136
3137       cf_index = unix_cli_file_add (cm, "stdin", STDIN_FILENO);
3138       cf = pool_elt_at_index (cm->cli_file_pool, cf_index);
3139       cm->stdin_cli_file_index = cf_index;
3140
3141       /* If stdin is a tty and we are using chacracter mode, enable
3142        * history on the CLI and set the tty line discipline accordingly. */
3143       if (isatty (STDIN_FILENO) && um->cli_line_mode == 0)
3144         {
3145           /* Capture terminal resize events */
3146           clib_memset (&sa, 0, sizeof (sa));
3147           sa.sa_handler = unix_cli_resize_interrupt;
3148           if (sigaction (SIGWINCH, &sa, 0) < 0)
3149             clib_panic ("sigaction");
3150
3151           /* Retrieve the current terminal size */
3152           if (ioctl (STDIN_FILENO, TIOCGWINSZ, &ws) == 0)
3153             {
3154               cf->width = ws.ws_col;
3155               cf->height = ws.ws_row;
3156             }
3157
3158           if (cf->width == 0 || cf->height == 0)
3159             {
3160               /*
3161                * We have a tty, but no size. Use defaults.
3162                * vpp "unix interactive" inside emacs + gdb ends up here.
3163                */
3164               cf->width = UNIX_CLI_DEFAULT_TERMINAL_WIDTH;
3165               cf->height = UNIX_CLI_DEFAULT_TERMINAL_HEIGHT;
3166             }
3167
3168           /* Setup the history */
3169           cf->history_limit = um->cli_history_limit;
3170           cf->has_history = cf->history_limit != 0;
3171
3172           /* Setup the pager */
3173           cf->no_pager = um->cli_no_pager;
3174
3175           /* This is an interactive session until we decide otherwise */
3176           cf->is_interactive = 1;
3177
3178           /* We're going to be in char by char mode */
3179           cf->line_mode = 0;
3180
3181           /* Save the original tty state so we can restore it later */
3182           tcgetattr (STDIN_FILENO, &um->tio_stdin);
3183           um->tio_isset = 1;
3184
3185           /* Tweak the tty settings */
3186           tio = um->tio_stdin;
3187           /* echo off, canonical mode off, ext'd input processing off */
3188           tio.c_lflag &= ~(ECHO | ICANON | IEXTEN);
3189           /* disable XON/XOFF, so ^S invokes the history search */
3190           tio.c_iflag &= ~(IXON | IXOFF);
3191           tio.c_cc[VMIN] = 1;   /* 1 byte at a time */
3192           tio.c_cc[VTIME] = 0;  /* no timer */
3193           tio.c_cc[VSTOP] = _POSIX_VDISABLE;    /* not ^S */
3194           tio.c_cc[VSTART] = _POSIX_VDISABLE;   /* not ^Q */
3195           tcsetattr (STDIN_FILENO, TCSAFLUSH, &tio);
3196
3197           /* See if we can do ANSI/VT100 output */
3198           term = (u8 *) getenv ("TERM");
3199           if (term != NULL)
3200             {
3201               int len = strlen ((char *) term);
3202               cf->ansi_capable = unix_cli_terminal_type_ansi (term, len);
3203               if (unix_cli_terminal_type_noninteractive (term, len))
3204                 unix_cli_set_session_noninteractive (cf);
3205             }
3206         }
3207       else
3208         {
3209           /* No tty, so make sure the session doesn't have tty-like features */
3210           unix_cli_set_session_noninteractive (cf);
3211         }
3212
3213       /* Send banner and initial prompt */
3214       unix_cli_file_welcome (cm, cf);
3215     }
3216
3217   /* If we have socket config, LISTEN, otherwise, don't */
3218   clib_socket_t *s = &um->cli_listen_socket;
3219   if (s->config && s->config[0] != 0)
3220     {
3221       /* CLI listen. */
3222       clib_file_t template = { 0 };
3223
3224       /* mkdir of file socketu, only under /run  */
3225       if (strncmp (s->config, "/run", 4) == 0)
3226         {
3227           u8 *tmp = format (0, "%s", s->config);
3228           int i = vec_len (tmp);
3229           while (i && tmp[--i] != '/')
3230             ;
3231
3232           tmp[i] = '\0';
3233
3234           if (i)
3235             vlib_unix_recursive_mkdir ((char *) tmp);
3236           vec_free (tmp);
3237         }
3238
3239       s->flags = CLIB_SOCKET_F_IS_SERVER |      /* listen, don't connect */
3240         CLIB_SOCKET_F_ALLOW_GROUP_WRITE;        /* PF_LOCAL socket only */
3241       error = clib_socket_init (s);
3242
3243       if (error)
3244         return error;
3245
3246       template.read_function = unix_cli_listen_read_ready;
3247       template.file_descriptor = s->fd;
3248       template.description = format (0, "cli listener %s", s->config);
3249
3250       clib_file_add (fm, &template);
3251     }
3252
3253   /* Set CLI prompt. */
3254   if (!cm->cli_prompt)
3255     cm->cli_prompt = format (0, "VLIB: ");
3256
3257   return 0;
3258 }
3259
3260 /*?
3261  * This module has no configurable parameters.
3262 ?*/
3263 VLIB_CONFIG_FUNCTION (unix_cli_config, "unix-cli");
3264
3265 /** Called when VPP is shutting down, this restores the system
3266  * terminal state if previously saved.
3267  */
3268 static clib_error_t *
3269 unix_cli_exit (vlib_main_t * vm)
3270 {
3271   unix_main_t *um = &unix_main;
3272
3273   /* If stdin is a tty and we saved the tty state, reset the tty state */
3274   if (isatty (STDIN_FILENO) && um->tio_isset)
3275     tcsetattr (STDIN_FILENO, TCSAFLUSH, &um->tio_stdin);
3276
3277   return 0;
3278 }
3279
3280 VLIB_MAIN_LOOP_EXIT_FUNCTION (unix_cli_exit);
3281
3282 /** Set the CLI prompt.
3283  * @param prompt The C string to set the prompt to.
3284  * @note This setting is global; it impacts all current
3285  *       and future CLI sessions.
3286  */
3287 void
3288 vlib_unix_cli_set_prompt (char *prompt)
3289 {
3290   char *fmt = (prompt[strlen (prompt) - 1] == ' ') ? "%s" : "%s ";
3291   unix_cli_main_t *cm = &unix_cli_main;
3292   if (cm->cli_prompt)
3293     vec_free (cm->cli_prompt);
3294   cm->cli_prompt = format (0, fmt, prompt);
3295 }
3296
3297 static unix_cli_file_t *
3298 unix_cli_file_if_exists (unix_cli_main_t * cm)
3299 {
3300   if (!cm->cli_file_pool)
3301     return 0;
3302   return pool_elt_at_index (cm->cli_file_pool, cm->current_input_file_index);
3303 }
3304
3305 static unix_cli_file_t *
3306 unix_cli_file_if_interactive (unix_cli_main_t * cm)
3307 {
3308   unix_cli_file_t *cf;
3309   if ((cf = unix_cli_file_if_exists (cm)) && !cf->is_interactive)
3310     return 0;
3311   return cf;
3312 }
3313
3314 /** CLI command to quit the terminal session.
3315  * @note If this is a stdin session then this will
3316  *       shutdown VPP also.
3317  */
3318 static clib_error_t *
3319 unix_cli_quit (vlib_main_t * vm,
3320                unformat_input_t * input, vlib_cli_command_t * cmd)
3321 {
3322   unix_cli_main_t *cm = &unix_cli_main;
3323   unix_cli_file_t *cf;
3324
3325   if (!(cf = unix_cli_file_if_exists (cm)))
3326     return clib_error_return (0, "invalid session");
3327
3328   /* Cosmetic: suppress the final prompt from appearing before we die */
3329   cf->is_interactive = 0;
3330   cf->started = 1;
3331
3332   vlib_process_signal_event (vm,
3333                              vlib_current_process (vm),
3334                              UNIX_CLI_PROCESS_EVENT_QUIT,
3335                              cm->current_input_file_index);
3336   return 0;
3337 }
3338
3339 /*?
3340  * Terminates the current CLI session.
3341  *
3342  * If VPP is running in @em interactive mode and this is the console session
3343  * (that is, the session on @c stdin) then this will also terminate VPP.
3344 ?*/
3345 /* *INDENT-OFF* */
3346 VLIB_CLI_COMMAND (unix_cli_quit_command, static) = {
3347   .path = "quit",
3348   .short_help = "Exit CLI",
3349   .function = unix_cli_quit,
3350 };
3351 /* *INDENT-ON* */
3352
3353 /* *INDENT-OFF* */
3354 VLIB_CLI_COMMAND (unix_cli_q_command, static) = {
3355   .path = "q",
3356   .short_help = "Exit CLI",
3357   .function = unix_cli_quit,
3358 };
3359 /* *INDENT-ON* */
3360
3361 /** CLI command to execute a VPP command script. */
3362 static clib_error_t *
3363 unix_cli_exec (vlib_main_t * vm,
3364                unformat_input_t * input, vlib_cli_command_t * cmd)
3365 {
3366   char *file_name;
3367   int fd;
3368   unformat_input_t sub_input;
3369   clib_error_t *error;
3370   unix_cli_main_t *cm = &unix_cli_main;
3371   unix_cli_file_t *cf;
3372   u8 *file_data = 0;
3373   file_name = 0;
3374   fd = -1;
3375   error = 0;
3376   struct stat s;
3377
3378
3379   if (!unformat (input, "%s", &file_name))
3380     {
3381       error = clib_error_return (0, "expecting file name, got `%U'",
3382                                  format_unformat_error, input);
3383       goto done;
3384     }
3385
3386   fd = open (file_name, O_RDONLY);
3387   if (fd < 0)
3388     {
3389       error = clib_error_return_unix (0, "failed to open `%s'", file_name);
3390       goto done;
3391     }
3392
3393   /* Make sure its a regular file. */
3394   if (fstat (fd, &s) < 0)
3395     {
3396       error = clib_error_return_unix (0, "failed to stat `%s'", file_name);
3397       goto done;
3398     }
3399
3400   if (!(S_ISREG (s.st_mode) || S_ISLNK (s.st_mode)))
3401     {
3402       error = clib_error_return (0, "not a regular file `%s'", file_name);
3403       goto done;
3404     }
3405
3406   /* Read the file */
3407   vec_validate (file_data, s.st_size);
3408
3409   if (read (fd, file_data, s.st_size) != s.st_size)
3410     {
3411       error = clib_error_return_unix (0, "Failed to read %d bytes from '%s'",
3412                                       s.st_size, file_name);
3413       vec_free (file_data);
3414       goto done;
3415     }
3416
3417   /* The macro expander expects a c string... */
3418   vec_add1 (file_data, 0);
3419
3420   unformat_init_vector (&sub_input, file_data);
3421
3422   /* Run the file contents through the macro processor */
3423   if (vec_len (sub_input.buffer) > 1)
3424     {
3425       u8 *expanded;
3426       clib_macro_main_t *mm = 0;
3427
3428       /* Initial config process? Use the global macro table. */
3429       if (pool_is_free_index
3430           (cm->cli_file_pool, cm->current_input_file_index))
3431         mm = &cm->macro_main;
3432       else
3433         {
3434           /* Otherwise, use the per-cli-process macro table */
3435           cf = pool_elt_at_index (cm->cli_file_pool,
3436                                   cm->current_input_file_index);
3437           mm = &cf->macro_main;
3438         }
3439
3440       expanded = (u8 *) clib_macro_eval (mm,
3441                                          (i8 *) sub_input.buffer,
3442                                          1 /* complain */ ,
3443                                          0 /* level */ ,
3444                                          8 /* max_level */ );
3445       /* Macro processor NULL terminates the return */
3446       _vec_len (expanded) -= 1;
3447       vec_reset_length (sub_input.buffer);
3448       vec_append (sub_input.buffer, expanded);
3449       vec_free (expanded);
3450     }
3451
3452   vlib_cli_input (vm, &sub_input, 0, 0);
3453   unformat_free (&sub_input);
3454
3455 done:
3456   if (fd >= 0)
3457     close (fd);
3458   vec_free (file_name);
3459
3460   return error;
3461 }
3462
3463 /*?
3464  * Executes a sequence of CLI commands which are read from a file. If
3465  * a command is unrecognised or otherwise invalid then the usual CLI
3466  * feedback will be generated, however execution of subsequent commands
3467  * from the file will continue.
3468  *
3469  * The VPP code is indifferent to the file location. However, if SELinux
3470  * is enabled, then the file needs to have an SELinux label the VPP
3471  * process is allowed to access. For example, if a file is created in
3472  * '<em>/usr/share/vpp/</em>', it will be allowed. However, files manually
3473  * created in '/tmp/' or '/home/<user>/' will not be accessible by the VPP
3474  * process when SELinux is enabled.
3475  *
3476  * @cliexpar
3477  * Sample file:
3478  * @clistart
3479  * <b><em>$ cat /usr/share/vpp/scripts/gigup.txt</em></b>
3480  * set interface state GigabitEthernet0/8/0 up
3481  * set interface state GigabitEthernet0/9/0 up
3482  * @cliend
3483  * Example of how to execute a set of CLI commands from a file:
3484  * @cliexcmd{exec /usr/share/vpp/scripts/gigup.txt}
3485 ?*/
3486 /* *INDENT-OFF* */
3487 VLIB_CLI_COMMAND (cli_exec, static) = {
3488   .path = "exec",
3489   .short_help = "exec <filename>",
3490   .function = unix_cli_exec,
3491   .is_mp_safe = 1,
3492 };
3493 /* *INDENT-ON* */
3494
3495 /** CLI command to show various unix error statistics. */
3496 static clib_error_t *
3497 unix_show_errors (vlib_main_t * vm,
3498                   unformat_input_t * input, vlib_cli_command_t * cmd)
3499 {
3500   unix_main_t *um = &unix_main;
3501   clib_error_t *error = 0;
3502   int i, n_errors_to_show;
3503   unix_error_history_t *unix_errors = 0;
3504
3505   n_errors_to_show = 1 << 30;
3506
3507   if (unformat_check_input (input) != UNFORMAT_END_OF_INPUT)
3508     {
3509       if (!unformat (input, "%d", &n_errors_to_show))
3510         {
3511           error =
3512             clib_error_return (0,
3513                                "expecting integer number of errors to show, got `%U'",
3514                                format_unformat_error, input);
3515           goto done;
3516         }
3517     }
3518
3519   n_errors_to_show =
3520     clib_min (ARRAY_LEN (um->error_history), n_errors_to_show);
3521
3522   i =
3523     um->error_history_index >
3524     0 ? um->error_history_index - 1 : ARRAY_LEN (um->error_history) - 1;
3525
3526   while (n_errors_to_show > 0)
3527     {
3528       unix_error_history_t *eh = um->error_history + i;
3529
3530       if (!eh->error)
3531         break;
3532
3533       vec_add1 (unix_errors, eh[0]);
3534       n_errors_to_show -= 1;
3535       if (i == 0)
3536         i = ARRAY_LEN (um->error_history) - 1;
3537       else
3538         i--;
3539     }
3540
3541   if (vec_len (unix_errors) == 0)
3542     vlib_cli_output (vm, "no Unix errors so far");
3543   else
3544     {
3545       vlib_cli_output (vm, "%Ld total errors seen", um->n_total_errors);
3546       for (i = vec_len (unix_errors) - 1; i >= 0; i--)
3547         {
3548           unix_error_history_t *eh = vec_elt_at_index (unix_errors, i);
3549           vlib_cli_output (vm, "%U: %U",
3550                            format_time_interval, "h:m:s:u", eh->time,
3551                            format_clib_error, eh->error);
3552         }
3553       vlib_cli_output (vm, "%U: time now",
3554                        format_time_interval, "h:m:s:u", vlib_time_now (vm));
3555     }
3556
3557 done:
3558   vec_free (unix_errors);
3559   return error;
3560 }
3561
3562 /* *INDENT-OFF* */
3563 VLIB_CLI_COMMAND (cli_unix_show_errors, static) = {
3564   .path = "show unix errors",
3565   .short_help = "Show Unix system call error history",
3566   .function = unix_show_errors,
3567 };
3568 /* *INDENT-ON* */
3569
3570 /** CLI command to show various unix error statistics. */
3571 static clib_error_t *
3572 unix_show_files (vlib_main_t * vm,
3573                  unformat_input_t * input, vlib_cli_command_t * cmd)
3574 {
3575   clib_error_t *error = 0;
3576   clib_file_main_t *fm = &file_main;
3577   clib_file_t *f;
3578   char path[PATH_MAX];
3579   u8 *s = 0;
3580
3581   vlib_cli_output (vm, "%3s %6s %12s %12s %12s %-32s %s", "FD", "Thread",
3582                    "Read", "Write", "Error", "File Name", "Description");
3583
3584   /* *INDENT-OFF* */
3585   pool_foreach (f, fm->file_pool)
3586    {
3587       int rv;
3588       s = format (s, "/proc/self/fd/%d%c", f->file_descriptor, 0);
3589       rv = readlink((char *) s, path, PATH_MAX - 1);
3590
3591       path[rv < 0 ? 0 : rv] = 0;
3592
3593       vlib_cli_output (vm, "%3d %6d %12d %12d %12d %-32s %v",
3594                        f->file_descriptor, f->polling_thread_index,
3595                        f->read_events, f->write_events, f->error_events,
3596                        path, f->description);
3597       vec_reset_length (s);
3598     }
3599   /* *INDENT-ON* */
3600   vec_free (s);
3601
3602   return error;
3603 }
3604
3605 /* *INDENT-OFF* */
3606 VLIB_CLI_COMMAND (cli_unix_show_files, static) = {
3607   .path = "show unix files",
3608   .short_help = "Show Unix files in use",
3609   .function = unix_show_files,
3610 };
3611 /* *INDENT-ON* */
3612
3613 /** CLI command to show session command history. */
3614 static clib_error_t *
3615 unix_cli_show_history (vlib_main_t * vm,
3616                        unformat_input_t * input, vlib_cli_command_t * cmd)
3617 {
3618   unix_cli_main_t *cm = &unix_cli_main;
3619   unix_cli_file_t *cf;
3620   int i, j;
3621
3622   if (!(cf = unix_cli_file_if_interactive (cm)))
3623     return clib_error_return (0, "invalid for non-interactive sessions");
3624
3625   if (cf->has_history && cf->history_limit)
3626     {
3627       i = 1 + cf->command_number - vec_len (cf->command_history);
3628       for (j = 0; j < vec_len (cf->command_history); j++)
3629         vlib_cli_output (vm, "%d  %v\n", i + j, cf->command_history[j]);
3630     }
3631   else
3632     {
3633       vlib_cli_output (vm, "History not enabled.\n");
3634     }
3635
3636   return 0;
3637 }
3638
3639 /*?
3640  * Displays the command history for the current session, if any.
3641 ?*/
3642 /* *INDENT-OFF* */
3643 VLIB_CLI_COMMAND (cli_unix_cli_show_history, static) = {
3644   .path = "history",
3645   .short_help = "Show current session command history",
3646   .function = unix_cli_show_history,
3647 };
3648 /* *INDENT-ON* */
3649
3650 /** CLI command to show terminal status. */
3651 static clib_error_t *
3652 unix_cli_show_terminal (vlib_main_t * vm,
3653                         unformat_input_t * input, vlib_cli_command_t * cmd)
3654 {
3655   unix_main_t *um = &unix_main;
3656   unix_cli_main_t *cm = &unix_cli_main;
3657   unix_cli_file_t *cf;
3658   vlib_node_t *n;
3659
3660   if (!(cf = unix_cli_file_if_exists (cm)))
3661     return clib_error_return (0, "invalid session");
3662
3663   n = vlib_get_node (vm, cf->process_node_index);
3664
3665   vlib_cli_output (vm, "Terminal name:   %v\n", n->name);
3666   vlib_cli_output (vm, "Terminal mode:   %s\n", cf->line_mode ?
3667                    "line-by-line" : "char-by-char");
3668   vlib_cli_output (vm, "Terminal width:  %d\n", cf->width);
3669   vlib_cli_output (vm, "Terminal height: %d\n", cf->height);
3670   vlib_cli_output (vm, "ANSI capable:    %s\n",
3671                    cf->ansi_capable ? "yes" : "no");
3672   vlib_cli_output (vm, "Interactive:     %s\n",
3673                    cf->is_interactive ? "yes" : "no");
3674   vlib_cli_output (vm, "History enabled: %s%s\n",
3675                    cf->has_history ? "yes" : "no", !cf->has_history
3676                    || cf->history_limit ? "" :
3677                    " (disabled by history limit)");
3678   if (cf->has_history)
3679     vlib_cli_output (vm, "History limit:   %d\n", cf->history_limit);
3680   vlib_cli_output (vm, "Pager enabled:   %s%s%s\n",
3681                    cf->no_pager ? "no" : "yes",
3682                    cf->no_pager
3683                    || cf->height ? "" : " (disabled by terminal height)",
3684                    cf->no_pager
3685                    || um->cli_pager_buffer_limit ? "" :
3686                    " (disabled by buffer limit)");
3687   if (!cf->no_pager)
3688     vlib_cli_output (vm, "Pager limit:     %d\n", um->cli_pager_buffer_limit);
3689   vlib_cli_output (vm, "CRLF mode:       %s\n",
3690                    cf->crlf_mode ? "CR+LF" : "LF");
3691
3692   return 0;
3693 }
3694
3695 /*?
3696  * Displays various information about the state of the current terminal
3697  * session.
3698  *
3699  * @cliexpar
3700  * @cliexstart{show terminal}
3701  * Terminal name:   unix-cli-stdin
3702  * Terminal mode:   char-by-char
3703  * Terminal width:  123
3704  * Terminal height: 48
3705  * ANSI capable:    yes
3706  * Interactive:     yes
3707  * History enabled: yes
3708  * History limit:   50
3709  * Pager enabled:   yes
3710  * Pager limit:     100000
3711  * CRLF mode:       LF
3712  * @cliexend
3713 ?*/
3714 /* *INDENT-OFF* */
3715 VLIB_CLI_COMMAND (cli_unix_cli_show_terminal, static) = {
3716   .path = "show terminal",
3717   .short_help = "Show current session terminal settings",
3718   .function = unix_cli_show_terminal,
3719 };
3720 /* *INDENT-ON* */
3721
3722 /** CLI command to display a list of CLI sessions. */
3723 static clib_error_t *
3724 unix_cli_show_cli_sessions (vlib_main_t * vm,
3725                             unformat_input_t * input,
3726                             vlib_cli_command_t * cmd)
3727 {
3728   unix_cli_main_t *cm = &unix_cli_main;
3729   clib_file_main_t *fm = &file_main;
3730   unix_cli_file_t *cf;
3731   clib_file_t *uf;
3732   vlib_node_t *n;
3733
3734   vlib_cli_output (vm, "%-5s %-5s %-20s %s", "PNI", "FD", "Name", "Flags");
3735
3736 #define fl(x, y) ( (x) ? toupper((y)) : tolower((y)) )
3737   /* *INDENT-OFF* */
3738   pool_foreach (cf, cm->cli_file_pool)  {
3739     uf = pool_elt_at_index (fm->file_pool, cf->clib_file_index);
3740     n = vlib_get_node (vm, cf->process_node_index);
3741     vlib_cli_output (vm,
3742                      "%-5d %-5d %-20v %c%c%c%c%c\n",
3743                      cf->process_node_index,
3744                      uf->file_descriptor,
3745                      n->name,
3746                      fl (cf->is_interactive, 'i'),
3747                      fl (cf->is_socket, 's'),
3748                      fl (cf->line_mode, 'l'),
3749                      fl (cf->has_epipe, 'p'),
3750                      fl (cf->ansi_capable, 'a'));
3751   }
3752   /* *INDENT-ON* */
3753 #undef fl
3754
3755   return 0;
3756 }
3757
3758 /*?
3759  * Displays a summary of all the current CLI sessions.
3760  *
3761  * Typically used to diagnose connection issues with the CLI
3762  * socket.
3763  *
3764  * @cliexpar
3765  * @cliexstart{show cli-sessions}
3766  * PNI   FD    Name                 Flags
3767  * 343   0     unix-cli-stdin       IslpA
3768  * 344   7     unix-cli-local:20    ISlpA
3769  * 346   8     unix-cli-local:21    iSLpa
3770  * @cliexend
3771
3772  * In this example we have the debug console of the running process
3773  * on stdin/out, we have an interactive socket session and we also
3774  * have a non-interactive socket session.
3775  *
3776  * Fields:
3777  *
3778  * - @em PNI: Process node index.
3779  * - @em FD: Unix file descriptor.
3780  * - @em Name: Name of the session.
3781  * - @em Flags: Various flags that describe the state of the session.
3782  *
3783  * @em Flags have the following meanings; lower-case typically negates
3784  * upper-case:
3785  *
3786  * - @em I Interactive session.
3787  * - @em S Connected by socket.
3788  * - @em s Not a socket, likely stdin.
3789  * - @em L Line-by-line mode.
3790  * - @em l Char-by-char mode.
3791  * - @em P EPIPE detected on connection; it will close soon.
3792  * - @em A ANSI-capable terminal.
3793 ?*/
3794 /* *INDENT-OFF* */
3795 VLIB_CLI_COMMAND (cli_unix_cli_show_cli_sessions, static) = {
3796   .path = "show cli-sessions",
3797   .short_help = "Show current CLI sessions",
3798   .function = unix_cli_show_cli_sessions,
3799 };
3800 /* *INDENT-ON* */
3801
3802 /** CLI command to set terminal pager settings. */
3803 static clib_error_t *
3804 unix_cli_set_terminal_pager (vlib_main_t * vm,
3805                              unformat_input_t * input,
3806                              vlib_cli_command_t * cmd)
3807 {
3808   unix_main_t *um = &unix_main;
3809   unix_cli_main_t *cm = &unix_cli_main;
3810   unix_cli_file_t *cf;
3811   unformat_input_t _line_input, *line_input = &_line_input;
3812   clib_error_t *error = 0;
3813
3814   if (!(cf = unix_cli_file_if_interactive (cm)))
3815     return clib_error_return (0, "invalid for non-interactive sessions");
3816
3817   if (!unformat_user (input, unformat_line_input, line_input))
3818     return 0;
3819
3820   while (unformat_check_input (line_input) != UNFORMAT_END_OF_INPUT)
3821     {
3822       if (unformat (line_input, "on"))
3823         cf->no_pager = 0;
3824       else if (unformat (line_input, "off"))
3825         cf->no_pager = 1;
3826       else if (unformat (line_input, "limit %u", &um->cli_pager_buffer_limit))
3827         vlib_cli_output (vm,
3828                          "Pager limit set to %u lines; note, this is global.\n",
3829                          um->cli_pager_buffer_limit);
3830       else
3831         {
3832           error = clib_error_return (0, "unknown parameter: `%U`",
3833                                      format_unformat_error, line_input);
3834           goto done;
3835         }
3836     }
3837
3838 done:
3839   unformat_free (line_input);
3840
3841   return error;
3842 }
3843
3844 /*?
3845  * Enables or disables the terminal pager for this session. Generally
3846  * this defaults to enabled.
3847  *
3848  * Additionally allows the pager buffer size to be set; though note that
3849  * this value is set globally and not per session.
3850 ?*/
3851 /* *INDENT-OFF* */
3852 VLIB_CLI_COMMAND (cli_unix_cli_set_terminal_pager, static) = {
3853   .path = "set terminal pager",
3854   .short_help = "set terminal pager [on|off] [limit <lines>]",
3855   .function = unix_cli_set_terminal_pager,
3856 };
3857 /* *INDENT-ON* */
3858
3859 /** CLI command to set terminal history settings. */
3860 static clib_error_t *
3861 unix_cli_set_terminal_history (vlib_main_t * vm,
3862                                unformat_input_t * input,
3863                                vlib_cli_command_t * cmd)
3864 {
3865   unix_cli_main_t *cm = &unix_cli_main;
3866   unix_cli_file_t *cf;
3867   unformat_input_t _line_input, *line_input = &_line_input;
3868   u32 limit;
3869   clib_error_t *error = 0;
3870
3871   if (!(cf = unix_cli_file_if_interactive (cm)))
3872     return clib_error_return (0, "invalid for non-interactive sessions");
3873
3874   if (!unformat_user (input, unformat_line_input, line_input))
3875     return 0;
3876
3877   while (unformat_check_input (line_input) != UNFORMAT_END_OF_INPUT)
3878     {
3879       if (unformat (line_input, "on"))
3880         cf->has_history = 1;
3881       else if (unformat (line_input, "off"))
3882         cf->has_history = 0;
3883       else if (unformat (line_input, "limit %u", &cf->history_limit))
3884         ;
3885       else
3886         {
3887           error = clib_error_return (0, "unknown parameter: `%U`",
3888                                      format_unformat_error, line_input);
3889           goto done;
3890         }
3891
3892     }
3893
3894   /* If we reduced history size, or turned it off, purge the history */
3895   limit = cf->has_history ? cf->history_limit : 0;
3896   if (limit < vec_len (cf->command_history))
3897     {
3898       u32 i;
3899
3900       /* How many items to remove from the start of history */
3901       limit = vec_len (cf->command_history) - limit;
3902
3903       for (i = 0; i < limit; i++)
3904         vec_free (cf->command_history[i]);
3905
3906       vec_delete (cf->command_history, limit, 0);
3907     }
3908
3909 done:
3910   unformat_free (line_input);
3911
3912   return error;
3913 }
3914
3915 /*?
3916  * Enables or disables the command history function of the current
3917  * terminal. Generally this defaults to enabled.
3918  *
3919  * This command also allows the maximum size of the history buffer for
3920  * this session to be altered.
3921 ?*/
3922 /* *INDENT-OFF* */
3923 VLIB_CLI_COMMAND (cli_unix_cli_set_terminal_history, static) = {
3924   .path = "set terminal history",
3925   .short_help = "set terminal history [on|off] [limit <lines>]",
3926   .function = unix_cli_set_terminal_history,
3927 };
3928 /* *INDENT-ON* */
3929
3930 /** CLI command to set terminal ANSI settings. */
3931 static clib_error_t *
3932 unix_cli_set_terminal_ansi (vlib_main_t * vm,
3933                             unformat_input_t * input,
3934                             vlib_cli_command_t * cmd)
3935 {
3936   unix_cli_main_t *cm = &unix_cli_main;
3937   unix_cli_file_t *cf;
3938
3939   if (!(cf = unix_cli_file_if_interactive (cm)))
3940     return clib_error_return (0, "invalid for non-interactive sessions");
3941
3942   if (unformat (input, "on"))
3943     cf->ansi_capable = 1;
3944   else if (unformat (input, "off"))
3945     cf->ansi_capable = 0;
3946   else
3947     return clib_error_return (0, "unknown parameter: `%U`",
3948                               format_unformat_error, input);
3949
3950   return 0;
3951 }
3952
3953 /*?
3954  * Enables or disables the use of ANSI control sequences by this terminal.
3955  * The default will vary based on terminal detection at the start of the
3956  * session.
3957  *
3958  * ANSI control sequences are used in a small number of places to provide,
3959  * for example, color text output and to control the cursor in the pager.
3960 ?*/
3961 /* *INDENT-OFF* */
3962 VLIB_CLI_COMMAND (cli_unix_cli_set_terminal_ansi, static) = {
3963   .path = "set terminal ansi",
3964   .short_help = "set terminal ansi [on|off]",
3965   .function = unix_cli_set_terminal_ansi,
3966 };
3967 /* *INDENT-ON* */
3968
3969
3970 #define MAX_CLI_WAIT 86400
3971 /** CLI command to wait <sec> seconds. Useful for exec script. */
3972 static clib_error_t *
3973 unix_wait_cmd (vlib_main_t * vm,
3974                unformat_input_t * input, vlib_cli_command_t * cmd)
3975 {
3976   unformat_input_t _line_input, *line_input = &_line_input;
3977   f64 sec = 1.0;
3978
3979   if (!unformat_user (input, unformat_line_input, line_input))
3980     return 0;
3981
3982   while (unformat_check_input (line_input) != UNFORMAT_END_OF_INPUT)
3983     {
3984       if (unformat (line_input, "%f", &sec))
3985         ;
3986       else
3987         return clib_error_return (0, "unknown parameter: `%U`",
3988                                   format_unformat_error, input);
3989     }
3990
3991   if (sec <= 0 || sec > MAX_CLI_WAIT || floor (sec * 1000) / 1000 != sec)
3992     return clib_error_return (0,
3993                               "<sec> must be a positive value and less than 86400 (one day) with no more than msec precision.");
3994
3995   vlib_process_wait_for_event_or_clock (vm, sec);
3996   vlib_cli_output (vm, "waited %.3f sec.", sec);
3997
3998   unformat_free (line_input);
3999   return 0;
4000 }
4001 /* *INDENT-OFF* */
4002 VLIB_CLI_COMMAND (cli_unix_wait_cmd, static) = {
4003   .path = "wait",
4004   .short_help = "wait <sec>",
4005   .function = unix_wait_cmd,
4006 };
4007 /* *INDENT-ON* */
4008
4009 static clib_error_t *
4010 echo_cmd (vlib_main_t * vm,
4011           unformat_input_t * input, vlib_cli_command_t * cmd)
4012 {
4013   unformat_input_t _line_input, *line_input = &_line_input;
4014
4015   /* Get a line of input. */
4016   if (!unformat_user (input, unformat_line_input, line_input))
4017     {
4018       vlib_cli_output (vm, "");
4019       return 0;
4020     }
4021
4022   vlib_cli_output (vm, "%v", line_input->buffer);
4023
4024   unformat_free (line_input);
4025   return 0;
4026 }
4027
4028 /* *INDENT-OFF* */
4029 VLIB_CLI_COMMAND (cli_unix_echo_cmd, static) = {
4030   .path = "echo",
4031   .short_help = "echo <rest-of-line>",
4032   .function = echo_cmd,
4033 };
4034 /* *INDENT-ON* */
4035
4036 static clib_error_t *
4037 define_cmd_fn (vlib_main_t * vm,
4038                unformat_input_t * input, vlib_cli_command_t * cmd)
4039 {
4040   u8 *macro_name;
4041   unformat_input_t _line_input, *line_input = &_line_input;
4042   clib_macro_main_t *mm = get_macro_main ();
4043   clib_error_t *error;
4044
4045   if (!unformat (input, "%s", &macro_name))
4046     return clib_error_return (0, "missing variable name...");
4047
4048   /* Remove white space */
4049   (void) unformat (input, "");
4050
4051   /* Get a line of input. */
4052   if (!unformat_user (input, unformat_line_input, line_input))
4053     {
4054       error = clib_error_return (0, "missing value for '%s'...", macro_name);
4055       vec_free (macro_name);
4056       return error;
4057     }
4058   /* the macro expander expects c-strings, not vectors... */
4059   vec_add1 (line_input->buffer, 0);
4060   clib_macro_set_value (mm, (char *) macro_name, (char *) line_input->buffer);
4061   vec_free (macro_name);
4062   unformat_free (line_input);
4063   return 0;
4064 }
4065
4066 /* *INDENT-OFF* */
4067 VLIB_CLI_COMMAND (define_cmd, static) = {
4068   .path = "define",
4069   .short_help = "define <variable-name> <value>",
4070   .function = define_cmd_fn,
4071 };
4072
4073 /* *INDENT-ON* */
4074
4075 static clib_error_t *
4076 undefine_cmd_fn (vlib_main_t * vm,
4077                  unformat_input_t * input, vlib_cli_command_t * cmd)
4078 {
4079   u8 *macro_name;
4080   clib_macro_main_t *mm = get_macro_main ();
4081
4082   if (!unformat (input, "%s", &macro_name))
4083     return clib_error_return (0, "missing variable name...");
4084
4085   if (clib_macro_unset (mm, (char *) macro_name))
4086     vlib_cli_output (vm, "%s wasn't set...", macro_name);
4087
4088   vec_free (macro_name);
4089   return 0;
4090 }
4091
4092 /* *INDENT-OFF* */
4093 VLIB_CLI_COMMAND (undefine_cmd, static) = {
4094   .path = "undefine",
4095   .short_help = "undefine <variable-name>",
4096   .function = undefine_cmd_fn,
4097 };
4098 /* *INDENT-ON* */
4099
4100 static clib_error_t *
4101 show_macro_cmd_fn (vlib_main_t * vm,
4102                    unformat_input_t * input, vlib_cli_command_t * cmd)
4103 {
4104   clib_macro_main_t *mm = get_macro_main ();
4105   int evaluate = 1;
4106
4107   if (unformat (input, "noevaluate %=", &evaluate, 0))
4108     ;
4109   else if (unformat (input, "noeval %=", &evaluate, 0))
4110     ;
4111
4112   vlib_cli_output (vm, "%U", format_clib_macro_main, mm, evaluate);
4113   return 0;
4114 }
4115
4116 /* *INDENT-OFF* */
4117 VLIB_CLI_COMMAND (show_macro, static) = {
4118   .path = "show macro",
4119   .short_help = "show macro [noevaluate]",
4120   .function = show_macro_cmd_fn,
4121 };
4122 /* *INDENT-ON* */
4123
4124 static clib_error_t *
4125 unix_cli_init (vlib_main_t * vm)
4126 {
4127   unix_cli_main_t *cm = &unix_cli_main;
4128
4129   /* Breadcrumb to indicate the new session process
4130    * has not been started */
4131   cm->new_session_process_node_index = ~0;
4132   clib_macro_init (&cm->macro_main);
4133
4134   return 0;
4135 }
4136
4137 VLIB_INIT_FUNCTION (unix_cli_init);
4138
4139 /*
4140  * fd.io coding-style-patch-verification: ON
4141  *
4142  * Local Variables:
4143  * eval: (c-set-style "gnu")
4144  * End:
4145  */