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         fd_set fds;
  74         char *line = NULL;
  75         struct timeval timeout;
  76         int fd = x_fileno(x_stdin);
  77         char *ret;
  78
  79         /* Prompt might be NULL in non-interactive mode. */
  80         if (prompt) {
  81                 x_fprintf(x_stdout, "%s", prompt);
  82                 x_fflush(x_stdout);
  83         }
  84
  85         line = (char *)malloc(BUFSIZ);
  86         if (!line) {
  87                 return NULL;
  88         }
  89
  90         while (!smb_rl_done) {
  91                 timeout.tv_sec = 5;
  92                 timeout.tv_usec = 0;
  93
  94                 FD_ZERO(&fds);
  95                 FD_SET(fd,&fds);
  96
  97                 if (sys_select_intr(fd+1,&fds,NULL,NULL,&timeout) == 1) {
  98                         ret = x_fgets(line, BUFSIZ, x_stdin);
  99                         if (ret == 0) {
 100                                 SAFE_FREE(line);
 101                         }
 102                         return ret;
 103                 }
 104                 if (callback) {
 105                         callback();
 106                 }
 107         }
 108         SAFE_FREE(line);
 109         return NULL;
 110 }
 111
 112 /****************************************************************************
 113  Display the prompt and wait for input. Call callback() regularly.
 114 ****************************************************************************/
 115
 116 char *smb_readline(const char *prompt, void (*callback)(void),
 117                    char **(completion_fn)(const char *text, int start, int end))
 118 {
 119         char *ret;
 120         bool interactive;
 121
 122         interactive = isatty(x_fileno(x_stdin)) || getenv("CLI_FORCE_INTERACTIVE");
 123         if (!interactive) {
 124                 return smb_readline_replacement(NULL, callback, completion_fn);
 125         }
 126
 127 #if HAVE_LIBREADLINE
 128
 129         /* Aargh!  Readline does bizzare things with the terminal width
 130         that mucks up expect(1).  Set CLI_NO_READLINE in the environment
 131         to force readline not to be used. */
 132
 133         if (getenv("CLI_NO_READLINE"))
 134                 return smb_readline_replacement(prompt, callback, completion_fn);
 135
 136         if (completion_fn) {
 137                 /* The callback prototype has changed slightly between
 138                 different versions of Readline, so the same function
 139                 works in all of them to date, but we get compiler
 140                 warnings in some.  */
 141                 rl_attempted_completion_function = RL_COMPLETION_CAST completion_fn;
 142         }
 143
 144 #if HAVE_DECL_RL_EVENT_HOOK
 145         if (callback)
 146                 rl_event_hook = (Function *)callback;
 147 #endif
 148         ret = readline(prompt);
 149         if (ret && *ret)
 150                 add_history(ret);
 151
 152 #else
 153         ret = smb_readline_replacement(prompt, callback, completion_fn);
 154 #endif
 155
 156         return ret;
 157 }
 158
 159 /****************************************************************************
 160  * return line buffer text
 161  ****************************************************************************/
 162 const char *smb_readline_get_line_buffer(void)
 163 {
 164 #if defined(HAVE_LIBREADLINE)
 165         return rl_line_buffer;
 166 #else
 167         return NULL;
 168 #endif
 169 }
 170
 171
 172 /****************************************************************************
 173  * set completion append character
 174  ***************************************************************************/
 175 void smb_readline_ca_char(char c)
 176 {
 177 #if defined(HAVE_LIBREADLINE)
 178         rl_completion_append_character = c;
 179 #endif
 180 }