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 |
|
---|
23 | #ifdef HAVE_LIBREADLINE
|
---|
24 | # ifdef HAVE_READLINE_READLINE_H
|
---|
25 | # include <readline/readline.h>
|
---|
26 | # ifdef HAVE_READLINE_HISTORY_H
|
---|
27 | # include <readline/history.h>
|
---|
28 | # endif
|
---|
29 | # else
|
---|
30 | # ifdef HAVE_READLINE_H
|
---|
31 | # include <readline.h>
|
---|
32 | # ifdef HAVE_HISTORY_H
|
---|
33 | # include <history.h>
|
---|
34 | # endif
|
---|
35 | # else
|
---|
36 | # undef HAVE_LIBREADLINE
|
---|
37 | # endif
|
---|
38 | # endif
|
---|
39 | #endif
|
---|
40 |
|
---|
41 | #ifdef HAVE_NEW_LIBREADLINE
|
---|
42 | # define RL_COMPLETION_CAST (rl_completion_func_t *)
|
---|
43 | #else
|
---|
44 | /* This type is missing from libreadline<4.0 (approximately) */
|
---|
45 | # define RL_COMPLETION_CAST
|
---|
46 | #endif /* HAVE_NEW_LIBREADLINE */
|
---|
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 *)SMB_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 | if (fd < 0 || fd >= FD_SETSIZE) {
|
---|
95 | errno = EBADF;
|
---|
96 | break;
|
---|
97 | }
|
---|
98 |
|
---|
99 | FD_ZERO(&fds);
|
---|
100 | FD_SET(fd,&fds);
|
---|
101 |
|
---|
102 | if (sys_select_intr(fd+1,&fds,NULL,NULL,&timeout) == 1) {
|
---|
103 | ret = x_fgets(line, BUFSIZ, x_stdin);
|
---|
104 | if (ret == 0) {
|
---|
105 | SAFE_FREE(line);
|
---|
106 | }
|
---|
107 | return ret;
|
---|
108 | }
|
---|
109 | #ifdef __OS2__
|
---|
110 | #include <conio.h>
|
---|
111 | {
|
---|
112 | char * cp = line;
|
---|
113 | char c;
|
---|
114 | int flag = 1;
|
---|
115 | while (flag && cp < line + sizeof(line) - 1)
|
---|
116 | {
|
---|
117 | c = getch();
|
---|
118 | switch (c)
|
---|
119 | {
|
---|
120 | case 8:
|
---|
121 | {
|
---|
122 | if (cp > line)
|
---|
123 | {
|
---|
124 | cp--;
|
---|
125 | putchar(8);
|
---|
126 | putchar(32);
|
---|
127 | putchar(8);
|
---|
128 | }
|
---|
129 | } break;
|
---|
130 | case 0: break;
|
---|
131 | case 10: // generally should not be
|
---|
132 | case 13:
|
---|
133 | {
|
---|
134 | flag = 0;
|
---|
135 | } break;
|
---|
136 | case 27:
|
---|
137 | {
|
---|
138 | for (; cp > line; cp--)
|
---|
139 | {
|
---|
140 | putchar(8);
|
---|
141 | putchar(32);
|
---|
142 | putchar(8);
|
---|
143 | }
|
---|
144 | } break;
|
---|
145 | default:
|
---|
146 | {
|
---|
147 | *cp++ = c;
|
---|
148 | putchar(c);
|
---|
149 | }
|
---|
150 | }
|
---|
151 | }
|
---|
152 | *cp = 0;
|
---|
153 | printf("\n");
|
---|
154 | return line;
|
---|
155 | }
|
---|
156 | #endif /* __OS2__ */
|
---|
157 | if (callback) {
|
---|
158 | callback();
|
---|
159 | }
|
---|
160 | }
|
---|
161 | SAFE_FREE(line);
|
---|
162 | return NULL;
|
---|
163 | }
|
---|
164 |
|
---|
165 | /****************************************************************************
|
---|
166 | Display the prompt and wait for input. Call callback() regularly.
|
---|
167 | ****************************************************************************/
|
---|
168 |
|
---|
169 | char *smb_readline(const char *prompt, void (*callback)(void),
|
---|
170 | char **(completion_fn)(const char *text, int start, int end))
|
---|
171 | {
|
---|
172 | char *ret;
|
---|
173 | bool interactive;
|
---|
174 |
|
---|
175 | interactive = isatty(x_fileno(x_stdin)) || getenv("CLI_FORCE_INTERACTIVE");
|
---|
176 | if (!interactive) {
|
---|
177 | return smb_readline_replacement(NULL, callback, completion_fn);
|
---|
178 | }
|
---|
179 |
|
---|
180 | #if HAVE_LIBREADLINE
|
---|
181 |
|
---|
182 | /* Aargh! Readline does bizzare things with the terminal width
|
---|
183 | that mucks up expect(1). Set CLI_NO_READLINE in the environment
|
---|
184 | to force readline not to be used. */
|
---|
185 |
|
---|
186 | if (getenv("CLI_NO_READLINE"))
|
---|
187 | return smb_readline_replacement(prompt, callback, completion_fn);
|
---|
188 |
|
---|
189 | if (completion_fn) {
|
---|
190 | /* The callback prototype has changed slightly between
|
---|
191 | different versions of Readline, so the same function
|
---|
192 | works in all of them to date, but we get compiler
|
---|
193 | warnings in some. */
|
---|
194 | rl_attempted_completion_function = RL_COMPLETION_CAST completion_fn;
|
---|
195 | }
|
---|
196 |
|
---|
197 | #if HAVE_DECL_RL_EVENT_HOOK
|
---|
198 | if (callback)
|
---|
199 | rl_event_hook = (Function *)callback;
|
---|
200 | #endif
|
---|
201 | ret = readline(prompt);
|
---|
202 | if (ret && *ret)
|
---|
203 | add_history(ret);
|
---|
204 |
|
---|
205 | #else
|
---|
206 | ret = smb_readline_replacement(prompt, callback, completion_fn);
|
---|
207 | #endif
|
---|
208 |
|
---|
209 | return ret;
|
---|
210 | }
|
---|
211 |
|
---|
212 | /****************************************************************************
|
---|
213 | * return line buffer text
|
---|
214 | ****************************************************************************/
|
---|
215 | const char *smb_readline_get_line_buffer(void)
|
---|
216 | {
|
---|
217 | #if defined(HAVE_LIBREADLINE)
|
---|
218 | return rl_line_buffer;
|
---|
219 | #else
|
---|
220 | return NULL;
|
---|
221 | #endif
|
---|
222 | }
|
---|
223 |
|
---|
224 |
|
---|
225 | /****************************************************************************
|
---|
226 | * set completion append character
|
---|
227 | ***************************************************************************/
|
---|
228 | void smb_readline_ca_char(char c)
|
---|
229 | {
|
---|
230 | #if defined(HAVE_LIBREADLINE)
|
---|
231 | rl_completion_append_character = c;
|
---|
232 | #endif
|
---|
233 | }
|
---|
234 |
|
---|
235 | /****************************************************************************
|
---|
236 | history
|
---|
237 | ****************************************************************************/
|
---|
238 | int cmd_history(void)
|
---|
239 | {
|
---|
240 | #if defined(HAVE_LIBREADLINE) && defined(HAVE_HISTORY_LIST)
|
---|
241 | HIST_ENTRY **hlist;
|
---|
242 | int i;
|
---|
243 |
|
---|
244 | hlist = history_list();
|
---|
245 |
|
---|
246 | for (i = 0; hlist && hlist[i]; i++) {
|
---|
247 | DEBUG(0, ("%d: %s\n", i, hlist[i]->line));
|
---|
248 | }
|
---|
249 | #else
|
---|
250 | DEBUG(0,("no history without readline support\n"));
|
---|
251 | #endif
|
---|
252 |
|
---|
253 | return 0;
|
---|
254 | }
|
---|