libcli/smbreadline/smbreadline.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    Samba readline wrapper implementation
   4    Copyright (C) Simo Sorce 2001
   5    Copyright (C) Andrew Tridgell 2001
   6
   7    This program is free software; you can redistribute it and/or modify
   8    it under the terms of the GNU General Public License as published by
   9    the Free Software Foundation; either version 3 of the License, or
  10    (at your option) any later version.
  11
  12    This program is distributed in the hope that it will be useful,
  13    but WITHOUT ANY WARRANTY; without even the implied warranty of
  14    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15    GNU General Public License for more details.
  16
  17    You should have received a copy of the GNU General Public License
  18    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  19 */
  20
  21 #include "includes.h"
  22 #include "../lib/util/select.h"
  23 #include "system/filesys.h"
  24 #include "system/select.h"
  25 #include "system/readline.h"
  26 #include "libcli/smbreadline/smbreadline.h"
  27
  28 #undef malloc
  29
  30 #ifdef HAVE_LIBREADLINE
  31 #  ifdef HAVE_READLINE_READLINE_H
  32 #    include <readline/readline.h>
  33 #    ifdef HAVE_READLINE_HISTORY_H
  34 #      include <readline/history.h>
  35 #    endif
  36 #  else
  37 #    ifdef HAVE_READLINE_H
  38 #      include <readline.h>
  39 #      ifdef HAVE_HISTORY_H
  40 #        include <history.h>
  41 #      endif
  42 #    else
  43 #      undef HAVE_LIBREADLINE
  44 #    endif
  45 #  endif
  46 #endif
  47
  48 static bool smb_rl_done;
  49
  50 #if HAVE_LIBREADLINE
  51 /*
  52  * MacOS/X does not have rl_done in readline.h, but
  53  * readline.so has it
  54  */
  55 extern int rl_done;
  56 #endif
  57
  58 void smb_readline_done(void)
  59 {
  60         smb_rl_done = true;
  61 #if HAVE_LIBREADLINE
  62         rl_done = 1;
  63 #endif
  64 }
  65
  66 /****************************************************************************
  67  Display the prompt and wait for input. Call callback() regularly
  68 ****************************************************************************/
  69
  70 static char *smb_readline_replacement(const char *prompt, void (*callback)(void),
  71                                 char **(completion_fn)(const char *text, int start, int end))
  72 {
  73         char *line = NULL;
  74         int fd = x_fileno(x_stdin);
  75         char *ret;
  76
  77         /* Prompt might be NULL in non-interactive mode. */
  78         if (prompt) {
  79                 x_fprintf(x_stdout, "%s", prompt);
  80                 x_fflush(x_stdout);
  81         }
  82
  83         line = (char *)malloc(BUFSIZ);
  84         if (!line) {
  85                 return NULL;
  86         }
  87
  88         while (!smb_rl_done) {
  89                 struct pollfd pfd;
  90
  91                 ZERO_STRUCT(pfd);
  92                 pfd.fd = fd;
  93                 pfd.events = POLLIN|POLLHUP;
  94
  95                 if (sys_poll_intr(&pfd, 1, 5000) == 1) {
  96                         ret = x_fgets(line, BUFSIZ, x_stdin);
  97                         if (ret == 0) {
  98                                 SAFE_FREE(line);
  99                         }
 100                         return ret;
 101                 }
 102                 if (callback) {
 103                         callback();
 104                 }
 105         }
 106         SAFE_FREE(line);
 107         return NULL;
 108 }
 109
 110 /****************************************************************************
 111  Display the prompt and wait for input. Call callback() regularly.
 112 ****************************************************************************/
 113
 114 char *smb_readline(const char *prompt, void (*callback)(void),
 115                    char **(completion_fn)(const char *text, int start, int end))
 116 {
 117         char *ret;
 118         bool interactive;
 119
 120         interactive = isatty(x_fileno(x_stdin)) || getenv("CLI_FORCE_INTERACTIVE");
 121         if (!interactive) {
 122                 return smb_readline_replacement(NULL, callback, completion_fn);
 123         }
 124
 125 #if HAVE_LIBREADLINE
 126
 127         /* Aargh!  Readline does bizzare things with the terminal width
 128         that mucks up expect(1).  Set CLI_NO_READLINE in the environment
 129         to force readline not to be used. */
 130
 131         if (getenv("CLI_NO_READLINE"))
 132                 return smb_readline_replacement(prompt, callback, completion_fn);
 133
 134         if (completion_fn) {
 135                 /* The callback prototype has changed slightly between
 136                 different versions of Readline, so the same function
 137                 works in all of them to date, but we get compiler
 138                 warnings in some.  */
 139                 rl_attempted_completion_function = RL_COMPLETION_CAST completion_fn;
 140         }
 141
 142 #if HAVE_DECL_RL_EVENT_HOOK
 143         if (callback)
 144                 rl_event_hook = (Function *)callback;
 145 #endif
 146         ret = readline(prompt);
 147         if (ret && *ret)
 148                 add_history(ret);
 149
 150 #else
 151         ret = smb_readline_replacement(prompt, callback, completion_fn);
 152 #endif
 153
 154         return ret;
 155 }
 156
 157 /****************************************************************************
 158  * return line buffer text
 159  ****************************************************************************/
 160 const char *smb_readline_get_line_buffer(void)
 161 {
 162 #if defined(HAVE_LIBREADLINE)
 163         return rl_line_buffer;
 164 #else
 165         return NULL;
 166 #endif
 167 }
 168
 169
 170 /****************************************************************************
 171  * set completion append character
 172  ***************************************************************************/
 173 void smb_readline_ca_char(char c)
 174 {
 175 #if defined(HAVE_LIBREADLINE)
 176         rl_completion_append_character = c;
 177 #endif
 178 }