samba: share readline wrappers among all buildsystems.

[ira/wip.git] / libcli / smbreadline / smbreadline.c

diff --git a/libcli/smbreadline/smbreadline.c b/libcli/smbreadline/smbreadline.c
new file mode 100644 (file)
index 0000000..f8441ac
--- /dev/null
+++ b/libcli/smbreadline/smbreadline.c
@@ -0,0 +1,180 @@
+/*
+   Unix SMB/CIFS implementation.
+   Samba readline wrapper implementation
+   Copyright (C) Simo Sorce 2001
+   Copyright (C) Andrew Tridgell 2001
+
+   This program is free software; you can redistribute it and/or modify
+   it under the terms of the GNU General Public License as published by
+   the Free Software Foundation; either version 3 of the License, or
+   (at your option) any later version.
+
+   This program is distributed in the hope that it will be useful,
+   but WITHOUT ANY WARRANTY; without even the implied warranty of
+   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+   GNU General Public License for more details.
+
+   You should have received a copy of the GNU General Public License
+   along with this program.  If not, see <http://www.gnu.org/licenses/>.
+*/
+
+#include "includes.h"
+#include "../lib/util/select.h"
+#include "system/filesys.h"
+#include "system/select.h"
+#include "system/readline.h"
+#include "libcli/smbreadline/smbreadline.h"
+
+#undef malloc
+
+#ifdef HAVE_LIBREADLINE
+#  ifdef HAVE_READLINE_READLINE_H
+#    include <readline/readline.h>
+#    ifdef HAVE_READLINE_HISTORY_H
+#      include <readline/history.h>
+#    endif
+#  else
+#    ifdef HAVE_READLINE_H
+#      include <readline.h>
+#      ifdef HAVE_HISTORY_H
+#        include <history.h>
+#      endif
+#    else
+#      undef HAVE_LIBREADLINE
+#    endif
+#  endif
+#endif
+
+static bool smb_rl_done;
+
+#if HAVE_LIBREADLINE
+/*
+ * MacOS/X does not have rl_done in readline.h, but
+ * readline.so has it
+ */
+extern int rl_done;
+#endif
+
+void smb_readline_done(void)
+{
+       smb_rl_done = true;
+#if HAVE_LIBREADLINE
+       rl_done = 1;
+#endif
+}
+
+/****************************************************************************
+ Display the prompt and wait for input. Call callback() regularly
+****************************************************************************/
+
+static char *smb_readline_replacement(const char *prompt, void (*callback)(void),
+                               char **(completion_fn)(const char *text, int start, int end))
+{
+       fd_set fds;
+       char *line = NULL;
+       struct timeval timeout;
+       int fd = x_fileno(x_stdin);
+       char *ret;
+
+       /* Prompt might be NULL in non-interactive mode. */
+       if (prompt) {
+               x_fprintf(x_stdout, "%s", prompt);
+               x_fflush(x_stdout);
+       }
+
+       line = (char *)malloc(BUFSIZ);
+       if (!line) {
+               return NULL;
+       }
+
+       while (!smb_rl_done) {
+               timeout.tv_sec = 5;
+               timeout.tv_usec = 0;
+
+               FD_ZERO(&fds);
+               FD_SET(fd,&fds);
+
+               if (sys_select_intr(fd+1,&fds,NULL,NULL,&timeout) == 1) {
+                       ret = x_fgets(line, BUFSIZ, x_stdin);
+                       if (ret == 0) {
+                               SAFE_FREE(line);
+                       }
+                       return ret;
+               }
+               if (callback) {
+                       callback();
+               }
+       }
+       SAFE_FREE(line);
+       return NULL;
+}
+
+/****************************************************************************
+ Display the prompt and wait for input. Call callback() regularly.
+****************************************************************************/
+
+char *smb_readline(const char *prompt, void (*callback)(void),
+                  char **(completion_fn)(const char *text, int start, int end))
+{
+       char *ret;
+       bool interactive;
+
+       interactive = isatty(x_fileno(x_stdin)) || getenv("CLI_FORCE_INTERACTIVE");
+       if (!interactive) {
+               return smb_readline_replacement(NULL, callback, completion_fn);
+       }
+
+#if HAVE_LIBREADLINE
+
+       /* Aargh!  Readline does bizzare things with the terminal width
+       that mucks up expect(1).  Set CLI_NO_READLINE in the environment
+       to force readline not to be used. */
+
+       if (getenv("CLI_NO_READLINE"))
+               return smb_readline_replacement(prompt, callback, completion_fn);
+
+       if (completion_fn) {
+               /* The callback prototype has changed slightly between
+               different versions of Readline, so the same function
+               works in all of them to date, but we get compiler
+               warnings in some.  */
+               rl_attempted_completion_function = RL_COMPLETION_CAST completion_fn;
+       }
+
+#if HAVE_DECL_RL_EVENT_HOOK
+       if (callback)
+               rl_event_hook = (Function *)callback;
+#endif
+       ret = readline(prompt);
+       if (ret && *ret)
+               add_history(ret);
+
+#else
+       ret = smb_readline_replacement(prompt, callback, completion_fn);
+#endif
+
+       return ret;
+}
+
+/****************************************************************************
+ * return line buffer text
+ ****************************************************************************/
+const char *smb_readline_get_line_buffer(void)
+{
+#if defined(HAVE_LIBREADLINE)
+       return rl_line_buffer;
+#else
+       return NULL;
+#endif
+}
+
+
+/****************************************************************************
+ * set completion append character
+ ***************************************************************************/
+void smb_readline_ca_char(char c)
+{
+#if defined(HAVE_LIBREADLINE)
+       rl_completion_append_character = c;
+#endif
+}