release/src/router/samba3/source/lib/readline.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 2 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, write to the Free Software
  19    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  20 */
  21
  22 #include "includes.h"
  23
  24 #ifdef HAVE_LIBREADLINE
  25 #  ifdef HAVE_READLINE_READLINE_H
  26 #    include <readline/readline.h>
  27 #    ifdef HAVE_READLINE_HISTORY_H
  28 #      include <readline/history.h>
  29 #    endif
  30 #  else
  31 #    ifdef HAVE_READLINE_H
  32 #      include <readline.h>
  33 #      ifdef HAVE_HISTORY_H
  34 #        include <history.h>
  35 #      endif
  36 #    else
  37 #      undef HAVE_LIBREADLINE
  38 #    endif
  39 #  endif
  40 #endif
  41
  42 #ifdef HAVE_NEW_LIBREADLINE
  43 #  define RL_COMPLETION_CAST (rl_completion_func_t *)
  44 #else
  45 /* This type is missing from libreadline<4.0  (approximately) */
  46 #  define RL_COMPLETION_CAST
  47 #endif /* HAVE_NEW_LIBREADLINE */
  48
  49 /****************************************************************************
  50  Display the prompt and wait for input. Call callback() regularly
  51 ****************************************************************************/
  52
  53 static char *smb_readline_replacement(const char *prompt, void (*callback)(void),
  54                                 char **(completion_fn)(const char *text, int start, int end))
  55 {
  56         fd_set fds;
  57         static pstring line;
  58         struct timeval timeout;
  59         int fd = x_fileno(x_stdin);
  60         char *ret;
  61
  62         /* Prompt might be NULL in non-interactive mode. */
  63         if (prompt) {
  64                 x_fprintf(x_stdout, "%s", prompt);
  65                 x_fflush(x_stdout);
  66         }
  67
  68         while (1) {
  69                 timeout.tv_sec = 5;
  70                 timeout.tv_usec = 0;
  71
  72                 if (fd < 0 || fd >= FD_SETSIZE) {
  73                         errno = EBADF;
  74                         break;
  75                 }
  76
  77                 FD_ZERO(&fds);
  78                 FD_SET(fd,&fds);
  79
  80                 if (sys_select_intr(fd+1,&fds,NULL,NULL,&timeout) == 1) {
  81                         ret = x_fgets(line, sizeof(line), x_stdin);
  82                         return ret;
  83                 }
  84                 if (callback)
  85                         callback();
  86         }
  87         return NULL;
  88 }
  89
  90 /****************************************************************************
  91  Display the prompt and wait for input. Call callback() regularly.
  92 ****************************************************************************/
  93
  94 char *smb_readline(const char *prompt, void (*callback)(void),
  95                    char **(completion_fn)(const char *text, int start, int end))
  96 {
  97         char *ret;
  98         BOOL interactive;
  99
 100         interactive = isatty(x_fileno(x_stdin)) || getenv("CLI_FORCE_INTERACTIVE");
 101         if (!interactive) {
 102             return smb_readline_replacement(NULL, callback, completion_fn);
 103         }
 104
 105 #if HAVE_LIBREADLINE
 106
 107         /* Aargh!  Readline does bizzare things with the terminal width
 108         that mucks up expect(1).  Set CLI_NO_READLINE in the environment
 109         to force readline not to be used. */
 110
 111         if (getenv("CLI_NO_READLINE"))
 112                 return smb_readline_replacement(prompt, callback, completion_fn);
 113
 114         if (completion_fn) {
 115                 /* The callback prototype has changed slightly between
 116                 different versions of Readline, so the same function
 117                 works in all of them to date, but we get compiler
 118                 warnings in some.  */
 119                 rl_attempted_completion_function = RL_COMPLETION_CAST completion_fn;
 120         }
 121
 122 #if HAVE_DECL_RL_EVENT_HOOK
 123         if (callback)
 124                 rl_event_hook = (Function *)callback;
 125 #endif
 126         ret = readline(prompt);
 127         if (ret && *ret)
 128                 add_history(ret);
 129
 130 #else
 131         ret = smb_readline_replacement(prompt, callback, completion_fn);
 132 #endif
 133
 134         return ret;
 135 }
 136
 137 /****************************************************************************
 138  * return line buffer text
 139  ****************************************************************************/
 140 const char *smb_readline_get_line_buffer(void)
 141 {
 142 #if defined(HAVE_LIBREADLINE)
 143         return rl_line_buffer;
 144 #else
 145         return NULL;
 146 #endif
 147 }
 148
 149
 150 /****************************************************************************
 151  * set completion append character
 152  ***************************************************************************/
 153 void smb_readline_ca_char(char c)
 154 {
 155 #if defined(HAVE_LIBREADLINE)
 156         rl_completion_append_character = c;
 157 #endif
 158 }
 159
 160 /****************************************************************************
 161 history
 162 ****************************************************************************/
 163 int cmd_history(void)
 164 {
 165 #if defined(HAVE_LIBREADLINE) && defined(HAVE_HISTORY_LIST)
 166         HIST_ENTRY **hlist;
 167         int i;
 168
 169         hlist = history_list();
 170
 171         for (i = 0; hlist && hlist[i]; i++) {
 172                 DEBUG(0, ("%d: %s\n", i, hlist[i]->line));
 173         }
 174 #else
 175         DEBUG(0,("no history without readline support\n"));
 176 #endif
 177
 178         return 0;
 179 }
 180