interactive.c

   1 /*
   2  * Copyright (C) 2011-2012 Andre Noll <maan@systemlinux.org>
   3  *
   4  * Licensed under the GPL v2. For licencing details see COPYING.
   5  */
   6
   7 /** \file interactive.c Readline abstraction for interactive sessions. */
   8
   9 #include <regex.h>
  10 #include <stdbool.h>
  11 #include <curses.h>
  12 #include <readline/readline.h>
  13 #include <readline/history.h>
  14 #include <sys/ioctl.h>
  15 #include <assert.h>
  16 #include <signal.h>
  17
  18 #include "para.h"
  19 #include "fd.h"
  20 #include "buffer_tree.h"
  21 #include "list.h"
  22 #include "sched.h"
  23 #include "interactive.h"
  24 #include "string.h"
  25 #include "error.h"
  26
  27 struct i9e_private {
  28         struct i9e_client_info *ici;
  29         FILE *stderr_stream;
  30         char empty_line[1000];
  31         struct task task;
  32         struct btr_node *stdout_btrn;
  33         bool line_handler_running;
  34         bool input_eof;
  35         bool caught_sigint;
  36 };
  37 static struct i9e_private i9e_private, *i9ep = &i9e_private;
  38
  39 static bool is_prefix(const char *partial, const char *full, size_t len)
  40 {
  41         if (len == 0)
  42                 len = strlen(partial);
  43         return !strncmp(partial, full, len);
  44 }
  45
  46 /*
  47  * Generator function for command completion. STATE lets us know whether
  48  * to start from scratch; without any state (i.e. STATE == 0), then we
  49  * start at the top of the list.
  50  */
  51 static char *command_generator(const char *text, int state)
  52 {
  53         static int list_index, len;
  54         const char *name;
  55         struct i9e_client_info *ici = i9ep->ici;
  56
  57         rl_attempted_completion_over = 1; /* disable filename completion */
  58         /*
  59          * If this is a new word to complete, initialize now. This includes
  60          * saving the length of TEXT for efficiency, and initializing the index
  61          * variable to 0.
  62          */
  63         if (state == 0) {
  64                 list_index = 0;
  65                 len = strlen(text);
  66         }
  67         /* Return the next name which partially matches from the command list. */
  68         while ((name = ici->completers[list_index].name)) {
  69                 list_index++;
  70                 if (is_prefix(text, name, len))
  71                         return para_strdup(name);
  72         }
  73         return NULL; /* no names matched */
  74 }
  75
  76 static void reset_completion_result(struct i9e_completion_result *cr)
  77 {
  78         cr->dont_append_space = false;
  79         cr->filename_completion_desired = false;
  80         cr->matches = NULL;
  81 }
  82
  83 static void create_matches(struct i9e_completion_info *ci,
  84                 struct i9e_completer *completers,
  85                 struct i9e_completion_result *cr)
  86 {
  87         int i, ret;
  88
  89         reset_completion_result(cr);
  90
  91         ret = create_argv(ci->buffer, " ", &ci->argv);
  92         if (ret < 0 || !ci->argv[0])
  93                 return;
  94
  95         ci->argc = ret;
  96         ci->word_num = compute_word_num(ci->buffer, " ", ci->point);
  97         for (i = 0; completers[i].name; i++) {
  98                 if (strcmp(completers[i].name, ci->argv[0]) != 0)
  99                         continue;
 100                 completers[i].completer(ci, cr);
 101                 break;
 102         }
 103         PARA_DEBUG_LOG("current word: %d (%s)\n", ci->word_num,
 104                 ci->argv[ci->word_num]);
 105         if (cr->matches)
 106                 for (i = 0; cr->matches[i]; i++)
 107                         PARA_DEBUG_LOG("match %d: %s\n", i, cr->matches[i]);
 108 }
 109
 110 static char *completion_generator(const char *word, int state)
 111 {
 112         static int list_index;
 113         static char **argv, **matches;
 114         struct i9e_completer *completers = i9ep->ici->completers;
 115         struct i9e_completion_info ci = {
 116                 .word = (char *)word,
 117                 .point = rl_point,
 118                 .buffer = rl_line_buffer,
 119         };
 120         struct i9e_completion_result cr = {.matches = NULL};
 121
 122         if (state != 0)
 123                 goto out;
 124         /* clean up previous matches and set defaults */
 125         free(matches);
 126         matches = NULL;
 127         free_argv(argv);
 128         argv = NULL;
 129         list_index = 0;
 130         rl_completion_append_character = ' ';
 131         rl_completion_suppress_append = false;
 132         rl_attempted_completion_over = true;
 133
 134         create_matches(&ci, completers, &cr);
 135
 136         matches = cr.matches;
 137         argv = ci.argv;
 138         rl_completion_suppress_append = cr.dont_append_space;
 139         rl_attempted_completion_over = !cr.filename_completion_desired;
 140 out:
 141         if (!matches)
 142                 return NULL;
 143         return matches[list_index++];
 144 }
 145
 146 /*
 147  * Attempt to complete on the contents of TEXT. START and END bound the
 148  * region of rl_line_buffer that contains the word to complete.  TEXT is
 149  * the word to complete.  We can use the entire contents of rl_line_buffer
 150  * in case we want to do some simple parsing. Return the array of matches,
 151  * or NULL if there aren't any.
 152  */
 153 static char **i9e_completer(const char *text, int start, __a_unused int end)
 154 {
 155         struct i9e_client_info *ici = i9ep->ici;
 156
 157         if (!ici->completers)
 158                 return NULL;
 159         /* Complete on command names if this is the first word in the line. */
 160         if (start == 0)
 161                 return rl_completion_matches(text, command_generator);
 162         return rl_completion_matches(text, completion_generator);
 163 }
 164
 165 /**
 166  * Prepare writing to stdout.
 167  *
 168  * \param producer The buffer tree node which produces output.
 169  *
 170  * The i9e subsystem maintains a buffer tree node which may be attached to
 171  * another node which generates output (a "producer"). When attached, the i9e
 172  * buffer tree node copies the buffers generated by the producer to stdout.
 173  *
 174  * This function attaches the i9e input queue to an output queue of \a
 175  * producer.
 176  *
 177  * \return Standard.
 178  */
 179 void i9e_attach_to_stdout(struct btr_node *producer)
 180 {
 181         assert(!i9ep->stdout_btrn);
 182         i9ep->stdout_btrn = btr_new_node(&(struct btr_node_description)
 183                 EMBRACE(.name = "interactive_stdout", .parent = producer));
 184 }
 185
 186 /**
 187  * Reset the terminal and save the in-memory command line history.
 188  *
 189  * This should be called before the caller exits.
 190  */
 191 void i9e_close(void)
 192 {
 193         char *hf = i9ep->ici->history_file;
 194
 195         rl_deprep_terminal();
 196         fprintf(i9ep->stderr_stream, "\n");
 197         if (hf)
 198                 write_history(hf);
 199         fclose(i9ep->stderr_stream);
 200 }
 201
 202 static void wipe_bottom_line(void)
 203 {
 204         fprintf(i9ep->stderr_stream, "\r%s\r", i9ep->empty_line);
 205 }
 206
 207 static void clear_bottom_line(void)
 208 {
 209         int point;
 210         char *text;
 211
 212         if (rl_point == 0 && rl_end == 0)
 213                 return wipe_bottom_line();
 214         /*
 215          * We might have a multi-line input that needs to be wiped here, so the
 216          * simple printf("\r<space>\r") is insufficient. To workaround this, we
 217          * remove the whole line, redisplay and restore the killed text.
 218          */
 219         point = rl_point;
 220         text = rl_copy_text(0, rl_end);
 221         rl_kill_full_line(0, 0);
 222         rl_redisplay();
 223         wipe_bottom_line(); /* wipe out the prompt */
 224         rl_insert_text(text);
 225         rl_point = point;
 226 }
 227
 228 static bool input_available(void)
 229 {
 230         fd_set rfds;
 231         struct timeval tv = {0, 0};
 232         int ret;
 233
 234         FD_ZERO(&rfds);
 235         FD_SET(i9ep->ici->fds[0], &rfds);
 236         ret = para_select(1, &rfds, NULL, &tv);
 237         return ret > 0;
 238 }
 239
 240 static void i9e_line_handler(char *line)
 241 {
 242         int ret;
 243
 244         i9ep->line_handler_running = true;
 245         ret = i9ep->ici->line_handler(line);
 246         i9ep->line_handler_running = false;
 247         if (ret < 0)
 248                 PARA_WARNING_LOG("%s\n", para_strerror(-ret));
 249         rl_set_prompt("");
 250         if (line) {
 251                 if (!*line)
 252                         rl_set_prompt(i9ep->ici->prompt);
 253                 else
 254                         add_history(line);
 255                 free(line);
 256         } else {
 257                 rl_set_prompt("");
 258                 i9ep->input_eof = true;
 259         }
 260 }
 261
 262 static void i9e_input(void)
 263 {
 264         do {
 265                 rl_callback_read_char();
 266         } while (input_available());
 267 }
 268
 269 static void i9e_post_select(struct sched *s, struct task *t)
 270 {
 271         int ret;
 272         struct btr_node *btrn = i9ep->stdout_btrn;
 273         struct i9e_client_info *ici = i9ep->ici;
 274         char *buf;
 275         size_t sz;
 276
 277         if (i9ep->input_eof) {
 278                 t->error = -E_I9E_EOF;
 279                 return;
 280         }
 281         if (!btrn) {
 282                 i9ep->caught_sigint = false;
 283                 if (FD_ISSET(ici->fds[0], &s->rfds))
 284                         i9e_input();
 285                 return;
 286         }
 287         if (i9ep->caught_sigint)
 288                 goto rm_btrn;
 289         ret = btr_node_status(i9ep->stdout_btrn, 0, BTR_NT_LEAF);
 290         if (ret < 0)
 291                 goto rm_btrn;
 292         sz = btr_next_buffer(btrn, &buf);
 293         if (sz == 0)
 294                 goto out;
 295         ret = xwrite(ici->fds[1], buf, sz);
 296         if (ret < 0)
 297                 goto rm_btrn;
 298         btr_consume(btrn, ret);
 299         goto out;
 300 rm_btrn:
 301         btr_remove_node(btrn);
 302         btr_free_node(btrn);
 303         i9ep->stdout_btrn = NULL;
 304         rl_set_prompt(i9ep->ici->prompt);
 305         rl_forced_update_display();
 306 out:
 307         t->error = 0;
 308 }
 309
 310 static void i9e_pre_select(struct sched *s, __a_unused struct task *t)
 311 {
 312         int ret;
 313
 314         if (i9ep->input_eof || i9ep->caught_sigint) {
 315                 sched_min_delay(s);
 316                 return;
 317         }
 318         if (i9ep->stdout_btrn) {
 319                 ret = btr_node_status(i9ep->stdout_btrn, 0, BTR_NT_LEAF);
 320                 if (ret < 0) {
 321                         sched_min_delay(s);
 322                         return;
 323                 }
 324                 if (ret > 0)
 325                         para_fd_set(i9ep->ici->fds[1], &s->wfds, &s->max_fileno);
 326         }
 327         /*
 328          * fd[0] might have been reset to blocking mode if our job was moved to
 329          * the background due to CTRL-Z or SIGSTOP, so set the fd back to
 330          * nonblocking mode.
 331          */
 332         ret = mark_fd_nonblocking(i9ep->ici->fds[0]);
 333         if (ret < 0)
 334                 PARA_WARNING_LOG("set to nonblock failed: (fd0 %d, %s)\n",
 335                         i9ep->ici->fds[0], para_strerror(-ret));
 336         para_fd_set(i9ep->ici->fds[0], &s->rfds, &s->max_fileno);
 337 }
 338
 339 static void update_winsize(void)
 340 {
 341         struct winsize w;
 342         int ret = ioctl(i9ep->ici->fds[2], TIOCGWINSZ, (char *)&w);
 343         int num_columns = 80;
 344
 345         if (ret >= 0) {
 346                 assert(w.ws_col < sizeof(i9ep->empty_line));
 347                 num_columns = w.ws_col;
 348         }
 349         memset(i9ep->empty_line, ' ', num_columns);
 350         i9ep->empty_line[num_columns] = '\0';
 351 }
 352
 353 /**
 354  * Register the i9e task and initialize readline.
 355  *
 356  * \param ici The i9e configuration parameters set by the caller.
 357  * \param s The scheduler instance to add the i9e task to.
 358  *
 359  * The caller must allocate and initialize the structure \a ici points to.
 360  *
 361  * \return Standard.
 362  * \sa \ref register_task().
 363  */
 364 int i9e_open(struct i9e_client_info *ici, struct sched *s)
 365 {
 366         int ret;
 367
 368         if (!isatty(ici->fds[0]))
 369                 return -E_I9E_SETUPTERM;
 370         ret = mark_fd_nonblocking(ici->fds[0]);
 371         if (ret < 0)
 372                 return ret;
 373         ret = mark_fd_nonblocking(ici->fds[1]);
 374         if (ret < 0)
 375                 return ret;
 376         i9ep->task.pre_select = i9e_pre_select;
 377         i9ep->task.post_select = i9e_post_select;
 378         sprintf(i9ep->task.status, "i9e");
 379         register_task(s, &i9ep->task);
 380         rl_readline_name = "para_i9e";
 381         rl_basic_word_break_characters = " ";
 382         rl_attempted_completion_function = i9e_completer;
 383         i9ep->ici = ici;
 384         i9ep->stderr_stream = fdopen(ici->fds[2], "w");
 385         setvbuf(i9ep->stderr_stream, NULL, _IONBF, 0);
 386
 387         if (ici->history_file)
 388                 read_history(ici->history_file);
 389         update_winsize();
 390         rl_callback_handler_install(i9ep->ici->prompt, i9e_line_handler);
 391         return 1;
 392 }
 393
 394 static void reset_line_state(void)
 395 {
 396         if (i9ep->line_handler_running)
 397                 return;
 398         rl_on_new_line();
 399         rl_reset_line_state();
 400         rl_forced_update_display();
 401 }
 402
 403 /**
 404  * The log function of the i9e subsystem.
 405  *
 406  * \param ll Severity log level.
 407  * \param fmt Printf-like format string.
 408  *
 409  * This clears the bottom line of the terminal if necessary and writes the
 410  * string given by \a fmt to fd[2], where fd[] is the array provided earlier in
 411  * \ref i9e_open().
 412  */
 413 __printf_2_3 void i9e_log(int ll, const char* fmt,...)
 414 {
 415         va_list argp;
 416
 417         if (ll < i9ep->ici->loglevel)
 418                 return;
 419         if (i9ep->line_handler_running == false)
 420                 clear_bottom_line();
 421         va_start(argp, fmt);
 422         vfprintf(i9ep->stderr_stream, fmt, argp);
 423         va_end(argp);
 424         reset_line_state();
 425 }
 426
 427 /**
 428  * Tell i9e that the caller received a signal.
 429  *
 430  * \param sig_num The number of the signal received.
 431  *
 432  * Currently the function only cares about \p SIGINT, but this may change.
 433  */
 434 void i9e_signal_dispatch(int sig_num)
 435 {
 436         if (sig_num == SIGINT) {
 437                 fprintf(i9ep->stderr_stream, "\n");
 438                 rl_replace_line ("", false /* clear_undo */);
 439                 reset_line_state();
 440                 i9ep->caught_sigint = true;
 441         }
 442 }
 443
 444 /**
 445  * Wrapper for select(2) which does not restart on interrupts.
 446  *
 447  * \param n \sa \ref para_select().
 448  * \param readfds \sa \ref para_select().
 449  * \param writefds \sa \ref para_select().
 450  * \param timeout_tv \sa \ref para_select().
 451  *
 452  * \return \sa \ref para_select().
 453  *
 454  * The only difference between this function and \ref para_select() is that
 455  * \ref i9e_select() returns zero if the select call returned \p EINTR.
 456  */
 457 int i9e_select(int n, fd_set *readfds, fd_set *writefds,
 458                 struct timeval *timeout_tv)
 459 {
 460         int ret = select(n, readfds, writefds, NULL, timeout_tv);
 461
 462         if (ret < 0) {
 463                 if (errno == EINTR)
 464                         ret = 0;
 465                 else
 466                         ret = -ERRNO_TO_PARA_ERROR(errno);
 467         }
 468         return ret;
 469 }
 470
 471 /**
 472  * Return the possible completions for a given word.
 473  *
 474  * \param word The word to complete.
 475  * \param string_list All possible words in this context.
 476  * \param result String list is returned here.
 477  *
 478  * This function never fails. If no completion was found, a string list of
 479  * length zero is returned. In any case, the result must be freed by the caller
 480  * using \ref free_argv().
 481  *
 482  * This function is independent of readline and may be called before
 483  * i9e_open().
 484  *
 485  * \return The number of possible completions.
 486  */
 487 int i9e_extract_completions(const char *word, char **string_list,
 488                 char ***result)
 489 {
 490         char **matches = para_malloc(sizeof(char *));
 491         int match_count = 0, matches_len = 1;
 492         char **p;
 493         int len = strlen(word);
 494
 495         for (p = string_list; *p; p++) {
 496                 if (!is_prefix(word, *p, len))
 497                         continue;
 498                 match_count++;
 499                 if (match_count >= matches_len) {
 500                         matches_len *= 2;
 501                         matches = para_realloc(matches,
 502                                 matches_len * sizeof(char *));
 503                 }
 504                 matches[match_count - 1] = para_strdup(*p);
 505         }
 506         matches[match_count] = NULL;
 507         *result = matches;
 508         return match_count;
 509 }
 510
 511 /**
 512  * Return the list of partially matching words.
 513  *
 514  * \param word The command to complete.
 515  * \param completers The array containing all command names.
 516  *
 517  * This is similar to \ref i9e_extract_completions(), but completes on the
 518  * command names in \a completers.
 519  *
 520  * \return See \ref i9e_extract_completions().
 521  */
 522 char **i9e_complete_commands(const char *word, struct i9e_completer *completers)
 523 {
 524         char **matches;
 525         const char *cmd;
 526         int i, match_count, len = strlen(word);
 527
 528         /*
 529          * In contrast to completing against an arbitrary string list, here we
 530          * know all possible completions and expect that there will not be many
 531          * of them. So it should be OK to iterate twice over all commands which
 532          * simplifies the code a bit.
 533          */
 534         for (i = 0, match_count = 0; (cmd = completers[i].name); i++) {
 535                 if (is_prefix(word, cmd, len))
 536                         match_count++;
 537         }
 538         matches = para_malloc((match_count + 1) * sizeof(*matches));
 539         for (i = 0, match_count = 0; (cmd = completers[i].name); i++)
 540                 if (is_prefix(word, cmd, len))
 541                         matches[match_count++] = para_strdup(cmd);
 542         matches[match_count] = NULL;
 543         return matches;
 544 }
 545
 546 /**
 547  * Complete according to the given options.
 548  *
 549  * \param opts All available options.
 550  * \param ci Information which was passed to the completer.
 551  * \param cr Result pointer.
 552  *
 553  * This convenience helper can be used to complete an option. The array of all
 554  * possible options is passed as the first argument. Flags, i.e. options
 555  * without an argument, are expected to be listed as strings of type "-X" in \a
 556  * opts while options which require an argument should be passed with a
 557  * trailing "=" character like "-X=".
 558  *
 559  * If the word can be uniquely completed to a flag option, an additional space
 560  * character is appended to the output. For non-flag options no space character
 561  * is appended.
 562  */
 563 void i9e_complete_option(char **opts, struct i9e_completion_info *ci,
 564                 struct i9e_completion_result *cr)
 565 {
 566         int num_matches;
 567
 568         num_matches = i9e_extract_completions(ci->word, opts, &cr->matches);
 569         if (num_matches == 1) {
 570                 char *opt = cr->matches[0];
 571                 char c = opt[strlen(opt) - 1];
 572                 if (c == '=')
 573                         cr->dont_append_space = true;
 574         }
 575 }
 576
 577 /**
 578  * Print possible completions to stdout.
 579  *
 580  * \param completers The array of completion functions.
 581  *
 582  * At the end of the output a line starting with "-o=", followed by the
 583  * (possibly empty) list of completion options is printed. Currently, the only
 584  * two completion options are "nospace" and "filenames". The former indicates
 585  * that no space should be appended even for a unique match while the latter
 586  * indicates that usual filename completion should be performed in addition to
 587  * the previously printed options.
 588  *
 589  * \return Standard.
 590  */
 591 int i9e_print_completions(struct i9e_completer *completers)
 592 {
 593         struct i9e_completion_result cr;
 594         struct i9e_completion_info ci;
 595         char *buf;
 596         const char *end, *p;
 597         int i, n, ret;
 598
 599         reset_completion_result(&cr);
 600         buf = getenv("COMP_POINT");
 601         ci.point = buf? atoi(buf) : 0;
 602         ci.buffer = para_strdup(getenv("COMP_LINE"));
 603
 604         ci.argc = create_argv(ci.buffer, " ", &ci.argv);
 605         ci.word_num = compute_word_num(ci.buffer, " ", ci.point);
 606
 607         end = ci.buffer + ci.point;
 608         for (p = end; p > ci.buffer && *p != ' '; p--)
 609                 ; /* nothing */
 610         if (*p == ' ')
 611                 p++;
 612
 613         n = end - p + 1;
 614         ci.word = para_malloc(n + 1);
 615         strncpy(ci.word, p, n);
 616         ci.word[n] = '\0';
 617
 618         PARA_DEBUG_LOG("line: %s, point: %d (%c), wordnum: %d, word: %s\n",
 619                 ci.buffer, ci.point, ci.buffer[ci.point], ci.word_num, ci.word);
 620         if (ci.word_num == 0)
 621                 cr.matches = i9e_complete_commands(ci.word, completers);
 622         else
 623                 create_matches(&ci, completers, &cr);
 624         ret = 0;
 625         if (cr.matches && cr.matches[0]) {
 626                 for (i = 0; cr.matches[i]; i++)
 627                         printf("%s\n", cr.matches[i]);
 628                 ret = 1;
 629         }
 630         printf("-o=");
 631         if (cr.dont_append_space)
 632                 printf("nospace");
 633         if (cr.filename_completion_desired)
 634                 printf(",filenames");
 635         printf("\n");
 636         free_argv(cr.matches);
 637         free_argv(ci.argv);
 638         free(ci.buffer);
 639         free(ci.word);
 640         return ret;
 641 }