libcli/security: fix sid_type_lookup().
[kai/samba.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         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 }