selftest:Samba3: fix signature for check_or_start()
[ira/wip.git] / 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 }