2 d3e1ab0c 2022-11-24 op * Copyright (c) 2022 Omar Polo <op@omarpolo.com>
4 d3e1ab0c 2022-11-24 op * Permission to use, copy, modify, and distribute this software for any
5 d3e1ab0c 2022-11-24 op * purpose with or without fee is hereby granted, provided that the above
6 d3e1ab0c 2022-11-24 op * copyright notice and this permission notice appear in all copies.
8 d3e1ab0c 2022-11-24 op * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
9 d3e1ab0c 2022-11-24 op * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
10 d3e1ab0c 2022-11-24 op * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
11 d3e1ab0c 2022-11-24 op * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
12 d3e1ab0c 2022-11-24 op * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
13 d3e1ab0c 2022-11-24 op * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
14 d3e1ab0c 2022-11-24 op * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
17 d3e1ab0c 2022-11-24 op #include "compat.h"
19 d3e1ab0c 2022-11-24 op #if !HAVE_READLINE
21 d3e1ab0c 2022-11-24 op #include <stdio.h>
22 d3e1ab0c 2022-11-24 op #include <string.h>
25 d3e1ab0c 2022-11-24 op readline(const char *prompt)
27 d3e1ab0c 2022-11-24 op char *ch, *line = NULL;
28 d3e1ab0c 2022-11-24 op size_t linesize = 0;
29 d3e1ab0c 2022-11-24 op ssize_t linelen;
31 d3e1ab0c 2022-11-24 op printf("%s", prompt);
32 d3e1ab0c 2022-11-24 op fflush(stdout);
34 d3e1ab0c 2022-11-24 op linelen = getline(&line, &linesize, stdin);
35 d3e1ab0c 2022-11-24 op if (linelen == -1)
38 d3e1ab0c 2022-11-24 op if ((ch = strchr(line, '\n')) != NULL)
44 d3e1ab0c 2022-11-24 op add_history(const char *line)
50 d3e1ab0c 2022-11-24 op compl_setup(void)
55 d3e1ab0c 2022-11-24 op #else /* HAVE_READLINE */
57 d3e1ab0c 2022-11-24 op #include <ctype.h>
58 d3e1ab0c 2022-11-24 op #include <limits.h>
59 d3e1ab0c 2022-11-24 op #include <stdio.h>
60 d3e1ab0c 2022-11-24 op #include <stdlib.h>
61 d3e1ab0c 2022-11-24 op #include <string.h>
63 d3e1ab0c 2022-11-24 op #include <readline/readline.h>
64 d3e1ab0c 2022-11-24 op #include <readline/history.h>
66 d3e1ab0c 2022-11-24 op #include "kami.h"
67 d3e1ab0c 2022-11-24 op #include "kamiftp.h"
69 d3e1ab0c 2022-11-24 op struct compl_state {
72 d3e1ab0c 2022-11-24 op char **entries;
75 d3e1ab0c 2022-11-24 op static struct compl_state compl_state;
76 d3e1ab0c 2022-11-24 op static char compl_prfx[PATH_MAX];
79 d3e1ab0c 2022-11-24 op compl_state_reset(void)
83 d3e1ab0c 2022-11-24 op for (i = 0; i < compl_state.len; ++i)
84 d3e1ab0c 2022-11-24 op free(compl_state.entries[i]);
85 d3e1ab0c 2022-11-24 op free(compl_state.entries);
87 e3644194 2022-12-21 op memset(&compl_state, 0, sizeof(compl_state));
91 d3e1ab0c 2022-11-24 op compl_add_entry(const struct np_stat *st)
93 d3e1ab0c 2022-11-24 op const char *sufx = "";
97 d3e1ab0c 2022-11-24 op if (compl_state.len == compl_state.size) {
98 d3e1ab0c 2022-11-24 op size_t newsz = compl_state.size * 1.5;
101 e3644194 2022-12-21 op if (newsz == 0)
104 e3644194 2022-12-21 op /* one for the NULL entry at the end */
105 d3e1ab0c 2022-11-24 op t = recallocarray(compl_state.entries, compl_state.size,
106 e3644194 2022-12-21 op newsz + 1, sizeof(char *));
107 d3e1ab0c 2022-11-24 op if (t == NULL)
109 d3e1ab0c 2022-11-24 op compl_state.entries = t;
110 d3e1ab0c 2022-11-24 op compl_state.size = newsz;
113 d3e1ab0c 2022-11-24 op if (st->qid.type & QTDIR)
116 d3e1ab0c 2022-11-24 op if (asprintf(&dup, "%s%s%s", compl_prfx, st->name, sufx) == -1)
118 d3e1ab0c 2022-11-24 op compl_state.entries[compl_state.len++] = dup;
123 d3e1ab0c 2022-11-24 op cleanword(char *buf, int brkspc)
126 d3e1ab0c 2022-11-24 op int escape, quote;
128 d3e1ab0c 2022-11-24 op while (brkspc && isspace((unsigned char)*buf))
129 d3e1ab0c 2022-11-24 op memmove(buf, buf + 1, strlen(buf));
131 d3e1ab0c 2022-11-24 op escape = quote = 0;
132 d3e1ab0c 2022-11-24 op for (cmd = buf; *cmd != '\0'; ++cmd) {
133 d3e1ab0c 2022-11-24 op if (escape) {
137 d3e1ab0c 2022-11-24 op if (*cmd == '\\')
139 d3e1ab0c 2022-11-24 op if (*cmd == quote) {
143 d3e1ab0c 2022-11-24 op if (*cmd == '\'' || *cmd == '"') {
144 d3e1ab0c 2022-11-24 op quote = *cmd;
149 d3e1ab0c 2022-11-24 op if (brkspc && isspace((unsigned char)*cmd))
154 d3e1ab0c 2022-11-24 op memmove(cmd, cmd + 1, strlen(cmd));
161 d3e1ab0c 2022-11-24 op tellcmd(char *buf)
165 d3e1ab0c 2022-11-24 op cleanword(buf, 1);
166 d3e1ab0c 2022-11-24 op for (i = 0; i < nitems(cmds); ++i) {
167 d3e1ab0c 2022-11-24 op if (!strcmp(cmds[i].name, buf))
168 d3e1ab0c 2022-11-24 op return cmds[i].cmdtype;
171 d3e1ab0c 2022-11-24 op return CMD_UNKNOWN;
175 d3e1ab0c 2022-11-24 op tell_argno(const char *cmd, int *cmdtype)
177 d3e1ab0c 2022-11-24 op char cmd0[64]; /* plenty of space */
178 d3e1ab0c 2022-11-24 op const char *start = cmd;
179 d3e1ab0c 2022-11-24 op int escape, quote;
180 d3e1ab0c 2022-11-24 op int argno = 0;
182 d3e1ab0c 2022-11-24 op *cmdtype = CMD_UNKNOWN;
184 d3e1ab0c 2022-11-24 op /* find which argument needs to be completed */
185 d3e1ab0c 2022-11-24 op while (*cmd) {
186 d3e1ab0c 2022-11-24 op while (isspace((unsigned char)*cmd))
188 d3e1ab0c 2022-11-24 op if (*cmd == '\0')
191 d3e1ab0c 2022-11-24 op escape = quote = 0;
192 d3e1ab0c 2022-11-24 op for (; *cmd; ++cmd) {
193 d3e1ab0c 2022-11-24 op if (escape) {
197 d3e1ab0c 2022-11-24 op if (*cmd == '\\') {
201 d3e1ab0c 2022-11-24 op if (*cmd == quote) {
205 d3e1ab0c 2022-11-24 op if (*cmd == '\'' || *cmd == '\"') {
206 d3e1ab0c 2022-11-24 op quote = *cmd;
211 d3e1ab0c 2022-11-24 op if (isspace((unsigned char)*cmd))
214 d3e1ab0c 2022-11-24 op if (isspace((unsigned char)*cmd))
217 d3e1ab0c 2022-11-24 op if (argno == 1 && strlcpy(cmd0, start, sizeof(cmd0)) <
218 d3e1ab0c 2022-11-24 op sizeof(cmd0))
219 d3e1ab0c 2022-11-24 op *cmdtype = tellcmd(cmd0);
222 d3e1ab0c 2022-11-24 op return argno;
225 d3e1ab0c 2022-11-24 op static char *
226 d3e1ab0c 2022-11-24 op ftp_cmdname_generator(const char *text, int state)
228 d3e1ab0c 2022-11-24 op static size_t i, len;
229 d3e1ab0c 2022-11-24 op struct cmd *cmd;
231 d3e1ab0c 2022-11-24 op if (state == 0) {
233 d3e1ab0c 2022-11-24 op len = strlen(text);
236 d3e1ab0c 2022-11-24 op while (i < nitems(cmds)) {
237 d3e1ab0c 2022-11-24 op cmd = &cmds[i++];
238 d3e1ab0c 2022-11-24 op if (strncmp(text, cmd->name, len) == 0)
239 d3e1ab0c 2022-11-24 op return strdup(cmd->name);
245 d3e1ab0c 2022-11-24 op static char *
246 d3e1ab0c 2022-11-24 op ftp_bool_generator(const char *text, int state)
248 d3e1ab0c 2022-11-24 op static const char *toks[] = { "on", "off" };
249 d3e1ab0c 2022-11-24 op static size_t i, len;
250 d3e1ab0c 2022-11-24 op const char *tok;
252 d3e1ab0c 2022-11-24 op if (state == 0) {
254 d3e1ab0c 2022-11-24 op len = strlen(text);
257 d3e1ab0c 2022-11-24 op while ((tok = toks[i++]) != NULL) {
258 d3e1ab0c 2022-11-24 op if (strncmp(text, tok, len) == 0)
259 d3e1ab0c 2022-11-24 op return strdup(tok);
264 d3e1ab0c 2022-11-24 op static char *
265 d3e1ab0c 2022-11-24 op ftp_dirent_generator(const char *text, int state)
267 d3e1ab0c 2022-11-24 op static size_t i, len;
268 d3e1ab0c 2022-11-24 op const char *entry;
270 d3e1ab0c 2022-11-24 op if (state == 0) {
272 d3e1ab0c 2022-11-24 op len = strlen(text);
275 d3e1ab0c 2022-11-24 op while (i < compl_state.len) {
276 d3e1ab0c 2022-11-24 op entry = compl_state.entries[i++];
277 d3e1ab0c 2022-11-24 op if (strncmp(text, entry, len) == 0)
278 d3e1ab0c 2022-11-24 op return strdup(entry);
283 d3e1ab0c 2022-11-24 op static char **
284 d3e1ab0c 2022-11-24 op ftp_remote_files(const char *text, int start, int end)
286 d3e1ab0c 2022-11-24 op const char *dir;
287 d3e1ab0c 2022-11-24 op char t[PATH_MAX];
290 d3e1ab0c 2022-11-24 op strlcpy(t, text, sizeof(t));
291 d3e1ab0c 2022-11-24 op cleanword(t, 0);
293 d3e1ab0c 2022-11-24 op if (!strcmp(t, "..")) {
295 d3e1ab0c 2022-11-24 op if ((cs = calloc(2, sizeof(*cs))) == NULL)
297 d3e1ab0c 2022-11-24 op cs[0] = strdup("../");
302 d3e1ab0c 2022-11-24 op if (!strncmp(s, "./", 2)) {
304 d3e1ab0c 2022-11-24 op while (*s == '/')
308 d3e1ab0c 2022-11-24 op if ((e = strrchr(s, '/')) != NULL)
312 d3e1ab0c 2022-11-24 op if (!strcmp(dir, "."))
313 d3e1ab0c 2022-11-24 op strlcpy(compl_prfx, "", sizeof(compl_prfx));
315 d3e1ab0c 2022-11-24 op strlcpy(compl_prfx, dir, sizeof(compl_prfx));
317 d3e1ab0c 2022-11-24 op compl_state_reset();
318 d3e1ab0c 2022-11-24 op if (dir_listing(dir, compl_add_entry, 0) == -1)
320 d3e1ab0c 2022-11-24 op return rl_completion_matches(text, ftp_dirent_generator);
323 d3e1ab0c 2022-11-24 op static char **
324 d3e1ab0c 2022-11-24 op ftp_completion(const char *text, int start, int end)
326 d3e1ab0c 2022-11-24 op int argno, cmdtype;
329 d3e1ab0c 2022-11-24 op /* don't fall back on the default completion system by default */
330 d3e1ab0c 2022-11-24 op rl_attempted_completion_over = 1;
332 d3e1ab0c 2022-11-24 op if ((line = rl_copy_text(0, start)) == NULL)
335 d3e1ab0c 2022-11-24 op argno = tell_argno(line, &cmdtype);
337 d3e1ab0c 2022-11-24 op if (argno == 0)
338 d3e1ab0c 2022-11-24 op return rl_completion_matches(text, ftp_cmdname_generator);
340 d3e1ab0c 2022-11-24 op switch (cmdtype) {
341 d3e1ab0c 2022-11-24 op case CMD_BELL:
342 d3e1ab0c 2022-11-24 op case CMD_HEXDUMP:
343 d3e1ab0c 2022-11-24 op case CMD_VERBOSE:
344 d3e1ab0c 2022-11-24 op if (argno != 1)
346 d3e1ab0c 2022-11-24 op return rl_completion_matches(text, ftp_bool_generator);
348 d3e1ab0c 2022-11-24 op case CMD_BYE:
349 d3e1ab0c 2022-11-24 op case CMD_LPWD:
350 d3e1ab0c 2022-11-24 op /* no args */
354 d3e1ab0c 2022-11-24 op case CMD_EDIT:
356 d3e1ab0c 2022-11-24 op case CMD_PAGE:
357 d3e1ab0c 2022-11-24 op if (argno != 1)
359 d3e1ab0c 2022-11-24 op /* fallthrough */
361 d3e1ab0c 2022-11-24 op return ftp_remote_files(text, start, end);
363 d3e1ab0c 2022-11-24 op case CMD_GET:
364 d3e1ab0c 2022-11-24 op if (argno > 2)
366 d3e1ab0c 2022-11-24 op if (argno == 2)
367 d3e1ab0c 2022-11-24 op return ftp_remote_files(text, start, end);
368 d3e1ab0c 2022-11-24 op /* try local */
369 d3e1ab0c 2022-11-24 op rl_attempted_completion_over = 0;
372 d3e1ab0c 2022-11-24 op case CMD_LCD:
373 d3e1ab0c 2022-11-24 op if (argno != 1)
375 d3e1ab0c 2022-11-24 op /* try local */
376 d3e1ab0c 2022-11-24 op rl_attempted_completion_over = 0;
379 d3e1ab0c 2022-11-24 op case CMD_PIPE:
380 d3e1ab0c 2022-11-24 op if (argno > 2)
382 d3e1ab0c 2022-11-24 op if (argno == 1)
383 d3e1ab0c 2022-11-24 op return ftp_remote_files(text, start, end);
384 d3e1ab0c 2022-11-24 op /* try local */
385 d3e1ab0c 2022-11-24 op rl_attempted_completion_over = 0;
388 d3e1ab0c 2022-11-24 op case CMD_PUT:
389 d3e1ab0c 2022-11-24 op if (argno > 2)
391 d3e1ab0c 2022-11-24 op if (argno == 1) {
392 d3e1ab0c 2022-11-24 op /* try local */
393 d3e1ab0c 2022-11-24 op rl_attempted_completion_over = 0;
396 d3e1ab0c 2022-11-24 op return ftp_remote_files(text, start, end);
398 d3e1ab0c 2022-11-24 op case CMD_RENAME:
399 d3e1ab0c 2022-11-24 op if (argno > 2)
401 d3e1ab0c 2022-11-24 op return ftp_remote_files(text, start, end);
408 d3e1ab0c 2022-11-24 op ftp_quoted(char *line, int index)
410 d3e1ab0c 2022-11-24 op if (index > 0 && line[index - 1] == '\\')
411 d3e1ab0c 2022-11-24 op return !ftp_quoted(line, index - 1);
416 d3e1ab0c 2022-11-24 op compl_setup(void)
418 d3e1ab0c 2022-11-24 op rl_attempted_completion_function = ftp_completion;
419 d3e1ab0c 2022-11-24 op rl_completer_word_break_characters = "\t ";
420 d3e1ab0c 2022-11-24 op rl_completer_quote_characters = "\"'";
421 d3e1ab0c 2022-11-24 op rl_char_is_quoted_p = ftp_quoted;