VNFs/DPPD-PROX/input_curses.c

   1 /*
   2 // Copyright (c) 2010-2017 Intel Corporation
   3 //
   4 // Licensed under the Apache License, Version 2.0 (the "License");
   5 // you may not use this file except in compliance with the License.
   6 // You may obtain a copy of the License at
   7 //
   8 //     http://www.apache.org/licenses/LICENSE-2.0
   9 //
  10 // Unless required by applicable law or agreed to in writing, software
  11 // distributed under the License is distributed on an "AS IS" BASIS,
  12 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13 // See the License for the specific language governing permissions and
  14 // limitations under the License.
  15 */
  16
  17 #include <string.h>
  18 #include <errno.h>
  19 #include <stdlib.h>
  20 #include <fcntl.h>
  21 #include <unistd.h>
  22
  23 #include "log.h"
  24 #include "input.h"
  25 #include "display.h"
  26 #include "run.h"
  27 #include "cmd_parser.h"
  28 #include "input_curses.h"
  29 #include "histedit.h"
  30 #include "libedit_autoconf.h"
  31
  32 static EditLine *el;
  33 static History *hist;
  34
  35 static struct input input_curses;
  36 static int tabbed;
  37
  38 static void show_history(struct input *input)
  39 {
  40         HistEvent event;
  41
  42         history(hist, &event, H_LAST);
  43
  44         do {
  45                 plog_info("%s", event.str); /* event.str contains newline */
  46         } while (history(hist, &event, H_PREV) != -1);
  47 }
  48
  49 static int complete(__attribute__((unused)) int ch)
  50 {
  51         const LineInfo *li;
  52         size_t len;
  53         size_t n_match = 0;
  54         char complete_cmd[128] = {0};
  55         int complete_cmd_partial = 0;
  56
  57         li = el_line(el);
  58         for (size_t i = 0; i < cmd_parser_n_cmd(); ++i) {
  59                 len = li->lastchar - li->buffer;
  60                 if (strncmp(cmd_parser_cmd(i), li->buffer, len) == 0) {
  61                         if (n_match) {
  62                                 size_t cur_len = strlen(complete_cmd);
  63                                 for (size_t j = 0; j < cur_len; ++j) {
  64                                         if (complete_cmd[j] != cmd_parser_cmd(i)[j]) {
  65                                                 complete_cmd[j] = 0;
  66                                                 complete_cmd_partial = 1;
  67                                                 break;
  68                                         }
  69                                 }
  70                         }
  71                         else {
  72                                 strcpy(complete_cmd, cmd_parser_cmd(i));
  73                         }
  74
  75                         n_match++;
  76                 }
  77         }
  78
  79         /* Complete only if there are more characters known than
  80            currently entered. */
  81         if (n_match && len < strlen(complete_cmd)) {
  82                 el_deletestr(el, li->cursor - li->buffer);
  83                 el_insertstr(el, complete_cmd);
  84                 if (!complete_cmd_partial)
  85                         el_insertstr(el, " ");
  86
  87                 return CC_REDISPLAY;
  88         }
  89         else if (tabbed) {
  90                 int printed = 0;
  91                 for (size_t i = 0; i < cmd_parser_n_cmd(); ++i) {
  92                         len = li->lastchar - li->buffer;
  93                         if (strncmp(cmd_parser_cmd(i), li->buffer, len) == 0) {
  94                                 plog_info("%-23s", cmd_parser_cmd(i));
  95                                 printed++;
  96                         }
  97                         if (printed == 4) {
  98                                 printed = 0;
  99                                 plog_info("\n");
 100                         }
 101                 }
 102                 if (printed)
 103                         plog_info("\n");
 104         }
 105         else {
 106                 tabbed = 1;
 107         }
 108
 109         return CC_REDISPLAY;
 110 }
 111
 112 /* Returns non-zero if stdin is readable */
 113 static int peek_stdin(void)
 114 {
 115         int tmp;
 116         fd_set in_fd;
 117         struct timeval tv;
 118
 119         tv.tv_sec = 0;
 120         tv.tv_usec = 10000;
 121
 122         FD_ZERO(&in_fd);
 123         FD_SET(fileno(stdin), &in_fd);
 124         tmp = select(fileno(stdin) + 1, &in_fd, NULL, NULL, &tv);
 125         return FD_ISSET(fileno(stdin), &in_fd);
 126 }
 127
 128 #ifdef HAVE_LIBEDIT_EL_RFUNC_T
 129 static int do_get_char(EditLine *e, wchar_t *c)
 130 #else
 131 static int get_char(EditLine *e, char *c)
 132 #endif
 133 {
 134         *c = display_getch();
 135
 136         /* If no characters have been entered, number keys switch the
 137            screen and '0' resets stats. This is provided as a
 138            fall-back in case F-keys do not function. The keys are
 139            intercepted before returning control to libedit. */
 140         if (*c >= '0' && *c <= '9') {
 141                 const LineInfo *li;
 142
 143                 li = el_line(e);
 144                 if (li->lastchar == li->buffer) {
 145                         if (*c >= '1') {
 146                                 display_screen(*c - '0' - 1);
 147                                 return 0;
 148                         }
 149                         else {
 150                                 cmd_parser_parse("reset stats", &input_curses);
 151                                 return 0;
 152                         }
 153                 }
 154         }
 155         if (*c == '=') {
 156                 toggle_display_screen();
 157                 return 0;
 158         }
 159
 160         /* Escape by itself is the first character used for more
 161            complex escape sequences like F-keys. libedit can't be used
 162            to detect both ESC as a unitary key and more complex
 163            sequences starting ESC at the same time. */
 164         if (*c == 27 && !peek_stdin()) {
 165                 quit();
 166                 return 0;
 167         }
 168         else if (*c != 9) {
 169                 tabbed = 0;
 170         }
 171
 172         return 1;
 173 }
 174
 175 #ifdef HAVE_LIBEDIT_EL_RFUNC_T
 176 static el_rfunc_t get_char = &do_get_char;
 177 #endif
 178
 179 static void proc_keyboard(struct input *input)
 180 {
 181         const char *line;
 182         const LineInfo *li;
 183         HistEvent hist_event;
 184         int len;
 185
 186         line = el_gets(el, &len);
 187         li = el_line(el);
 188
 189         if (len == 0 || line == NULL) {
 190                 display_cmd("", 0, 0);
 191                 return;
 192         } else if (len > 0) {
 193                 if (len == 1 && line[0] == '\n') {
 194                         display_print_page();
 195                         el_set(el, EL_UNBUFFERED, 0);
 196                         el_set(el, EL_UNBUFFERED, 1);
 197                         return;
 198                 }
 199                 if (line[len-1] == '\n') {
 200                         if (hist) {
 201                                 history(hist, &hist_event, H_ENTER, line);
 202                         }
 203
 204                         char *line2 = strndup(line, len);
 205                         line2[len - 1] = 0; /* replace \n */
 206                         cmd_parser_parse(line2, input);
 207                         free(line2);
 208
 209                         el_set(el, EL_UNBUFFERED, 0);
 210                         el_set(el, EL_UNBUFFERED, 1);
 211                         display_cmd("", 0, 0);
 212                         return;
 213                 }
 214                 if (line[len-1] == 4) {
 215                         return; /* should quit*/
 216                 }
 217         }
 218         else {
 219                 if (errno) {
 220                        return;
 221                 }
 222                 display_cmd("", 0, 0);
 223                 return;
 224         }
 225         display_cmd(line, len, li->cursor - li->buffer);
 226 }
 227
 228 static int key_f1(__attribute__((unused)) int ch) {display_screen(0); return CC_REDISPLAY;}
 229 static int key_f2(__attribute__((unused)) int ch) {display_screen(1); return CC_REDISPLAY;}
 230 static int key_f3(__attribute__((unused)) int ch) {display_screen(2); return CC_REDISPLAY;}
 231 static int key_f4(__attribute__((unused)) int ch) {display_screen(3); return CC_REDISPLAY;}
 232 static int key_f5(__attribute__((unused)) int ch) {display_screen(4); return CC_REDISPLAY;}
 233 static int key_f6(__attribute__((unused)) int ch) {display_screen(5); return CC_REDISPLAY;}
 234 static int key_f7(__attribute__((unused)) int ch) {display_screen(6); return CC_REDISPLAY;}
 235 static int key_f8(__attribute__((unused)) int ch) {display_screen(7); return CC_REDISPLAY;}
 236 static int key_f9(__attribute__((unused)) int ch) {display_screen(8); return CC_REDISPLAY;}
 237 static int key_f10(__attribute__((unused)) int ch) {display_screen(9); return CC_REDISPLAY;}
 238 static int key_f11(__attribute__((unused)) int ch) {display_screen(10); return CC_REDISPLAY;}
 239 static int key_f12(__attribute__((unused)) int ch) {display_screen(11); return CC_REDISPLAY;}
 240
 241 static int key_page_up(__attribute__((unused)) int ch) {display_page_up(); return CC_REDISPLAY;}
 242 static int key_page_down(__attribute__((unused)) int ch) {display_page_down(); return CC_REDISPLAY;}
 243
 244 static void setup_el(void)
 245 {
 246         int pty;
 247         FILE *dev_pty;
 248         HistEvent hist_event;
 249
 250         /* Open a pseudo-terminal for use in libedit. This is required
 251            since the library checks if it is using a tty. If the file
 252            descriptor does not represent a tty, the library disables
 253            editing. */
 254
 255         pty = posix_openpt(O_RDWR);
 256         /* TODO: On error (posix_openpt() < 0), fall-back to
 257            non-libedit implementation. */
 258         grantpt(pty);
 259         unlockpt(pty);
 260         dev_pty = fdopen(pty, "wr");
 261
 262         el = el_init("", dev_pty, dev_pty, dev_pty);
 263
 264         el_set(el, EL_EDITOR, "emacs");
 265
 266         el_set(el, EL_ADDFN, "complete", "Command completion", complete);
 267
 268         el_set(el, EL_ADDFN, "key_f1", "Switch to screen 1", key_f1);
 269         el_set(el, EL_ADDFN, "key_f2", "Switch to screen 2", key_f2);
 270         el_set(el, EL_ADDFN, "key_f3", "Switch to screen 3", key_f3);
 271         el_set(el, EL_ADDFN, "key_f4", "Switch to screen 4", key_f4);
 272         el_set(el, EL_ADDFN, "key_f5", "Switch to screen 5", key_f5);
 273         el_set(el, EL_ADDFN, "key_f6", "Switch to screen 6", key_f6);
 274         el_set(el, EL_ADDFN, "key_f7", "Switch to screen 7", key_f7);
 275         el_set(el, EL_ADDFN, "key_f8", "Switch to screen 8", key_f8);
 276         el_set(el, EL_ADDFN, "key_f9", "Switch to screen 9", key_f5);
 277         el_set(el, EL_ADDFN, "key_f10", "Switch to screen 10", key_f6);
 278         el_set(el, EL_ADDFN, "key_f11", "Switch to screen 11", key_f7);
 279         el_set(el, EL_ADDFN, "key_f12", "Switch to screen 12", key_f8);
 280
 281         el_set(el, EL_ADDFN, "key_page_up", "Page up", key_page_up);
 282         el_set(el, EL_ADDFN, "key_page_down", "Page down", key_page_down);
 283
 284         el_set(el, EL_BIND, "^I", "complete", NULL);
 285         el_set(el, EL_BIND, "^r", "em-inc-search-prev", NULL);
 286
 287         el_set(el, EL_BIND, "^[[11~", "key_f1", NULL);
 288         el_set(el, EL_BIND, "^[[12~", "key_f2", NULL);
 289         el_set(el, EL_BIND, "^[[13~", "key_f3", NULL);
 290         el_set(el, EL_BIND, "^[[14~", "key_f4", NULL);
 291         el_set(el, EL_BIND, "^[[15~", "key_f5", NULL);
 292         el_set(el, EL_BIND, "^[[17~", "key_f6", NULL);
 293         el_set(el, EL_BIND, "^[[18~", "key_f7", NULL);
 294         el_set(el, EL_BIND, "^[[19~", "key_f8", NULL);
 295         el_set(el, EL_BIND, "^[[20~", "key_f9", NULL);
 296         el_set(el, EL_BIND, "^[[21~", "key_f10", NULL);
 297         el_set(el, EL_BIND, "^[[23~", "key_f11", NULL);
 298         el_set(el, EL_BIND, "^[[24~", "key_f12", NULL);
 299
 300         el_set(el, EL_BIND, "^[OP", "key_f1", NULL);
 301         el_set(el, EL_BIND, "^[OQ", "key_f2", NULL);
 302         el_set(el, EL_BIND, "^[OR", "key_f3", NULL);
 303         el_set(el, EL_BIND, "^[OS", "key_f4", NULL);
 304
 305         el_set(el, EL_BIND, "^[[5~", "key_page_up", NULL);
 306         el_set(el, EL_BIND, "^[[6~", "key_page_down", NULL);
 307
 308         hist = history_init();
 309         if (hist) {
 310                 history(hist, &hist_event, H_SETSIZE, 1000);
 311                 el_set(el, EL_HIST, history, hist);
 312         }
 313         el_set(el, EL_UNBUFFERED, 1);
 314         el_set(el, EL_GETCFN, get_char);
 315 }
 316
 317 void reg_input_curses(void)
 318 {
 319         setup_el();
 320
 321         input_curses.fd = fileno(stdin);
 322         input_curses.proc_input = proc_keyboard;
 323         input_curses.history = show_history;
 324
 325         reg_input(&input_curses);
 326 }
 327
 328 void unreg_input_curses(void)
 329 {
 330         history_end(hist);
 331         el_end(el);
 332
 333         unreg_input(&input_curses);
 334 }