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 65b37bc8 2022-12-21 op #include <libgen.h>
59 d3e1ab0c 2022-11-24 op #include <limits.h>
60 d3e1ab0c 2022-11-24 op #include <stdio.h>
61 d3e1ab0c 2022-11-24 op #include <stdlib.h>
62 d3e1ab0c 2022-11-24 op #include <string.h>
64 d3e1ab0c 2022-11-24 op #include <readline/readline.h>
65 d3e1ab0c 2022-11-24 op #include <readline/history.h>
67 d3e1ab0c 2022-11-24 op #include "kami.h"
68 d3e1ab0c 2022-11-24 op #include "kamiftp.h"
70 d3e1ab0c 2022-11-24 op struct compl_state {
73 d3e1ab0c 2022-11-24 op char **entries;
76 d3e1ab0c 2022-11-24 op static struct compl_state compl_state;
77 d3e1ab0c 2022-11-24 op static char compl_prfx[PATH_MAX];
80 d3e1ab0c 2022-11-24 op compl_state_reset(void)
84 d3e1ab0c 2022-11-24 op for (i = 0; i < compl_state.len; ++i)
85 d3e1ab0c 2022-11-24 op free(compl_state.entries[i]);
86 d3e1ab0c 2022-11-24 op free(compl_state.entries);
88 e3644194 2022-12-21 op memset(&compl_state, 0, sizeof(compl_state));
92 d3e1ab0c 2022-11-24 op compl_add_entry(const struct np_stat *st)
94 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 65b37bc8 2022-12-21 op static inline void
284 65b37bc8 2022-12-21 op set_compl_prfx(const char *path)
286 65b37bc8 2022-12-21 op if (!strcmp(path, "."))
287 65b37bc8 2022-12-21 op strlcpy(compl_prfx, "", sizeof(compl_prfx));
289 65b37bc8 2022-12-21 op strlcpy(compl_prfx, path, sizeof(compl_prfx));
292 65b37bc8 2022-12-21 op static inline void
293 65b37bc8 2022-12-21 op inplace_dirname(char *path, size_t len)
297 65b37bc8 2022-12-21 op if ((t = strrchr(path, '/')) == NULL)
298 65b37bc8 2022-12-21 op strlcpy(path, ".", len);
303 d3e1ab0c 2022-11-24 op static char **
304 d3e1ab0c 2022-11-24 op ftp_remote_files(const char *text, int start, int end)
306 847e80dc 2022-12-21 op char *t, dir[PATH_MAX];
308 847e80dc 2022-12-21 op strlcpy(dir, text, sizeof(dir));
309 847e80dc 2022-12-21 op cleanword(dir, 0);
311 847e80dc 2022-12-21 op if (!strcmp(dir, "..")) {
313 d3e1ab0c 2022-11-24 op if ((cs = calloc(2, sizeof(*cs))) == NULL)
315 d3e1ab0c 2022-11-24 op cs[0] = strdup("../");
320 847e80dc 2022-12-21 op if (!strncmp(t, "./", 2)) {
322 847e80dc 2022-12-21 op while (*t == '/')
324 847e80dc 2022-12-21 op memmove(dir, t, strlen(t) + 1);
327 65b37bc8 2022-12-21 op if (*dir && dir[strlen(dir) - 1] != '/')
328 65b37bc8 2022-12-21 op strlcat(dir, "/", sizeof(dir));
330 65b37bc8 2022-12-21 op set_compl_prfx(dir);
331 d3e1ab0c 2022-11-24 op compl_state_reset();
332 65b37bc8 2022-12-21 op if (dir_listing(dir, compl_add_entry, 0) == -1) {
334 65b37bc8 2022-12-21 op dir[strlen(dir) - 1] = '\0';
335 65b37bc8 2022-12-21 op inplace_dirname(dir, sizeof(dir));
336 65b37bc8 2022-12-21 op set_compl_prfx(dir);
338 65b37bc8 2022-12-21 op compl_state_reset();
339 65b37bc8 2022-12-21 op if (dir_listing(dir, compl_add_entry, 0) == -1)
343 d3e1ab0c 2022-11-24 op return rl_completion_matches(text, ftp_dirent_generator);
346 d3e1ab0c 2022-11-24 op static char **
347 d3e1ab0c 2022-11-24 op ftp_completion(const char *text, int start, int end)
349 d3e1ab0c 2022-11-24 op int argno, cmdtype;
352 d3e1ab0c 2022-11-24 op /* don't fall back on the default completion system by default */
353 d3e1ab0c 2022-11-24 op rl_attempted_completion_over = 1;
355 4c9b5bd4 2022-12-21 op /* append a space after selecting a command */
356 4c9b5bd4 2022-12-21 op rl_completion_append_character = ' ';
358 d3e1ab0c 2022-11-24 op if ((line = rl_copy_text(0, start)) == NULL)
361 d3e1ab0c 2022-11-24 op argno = tell_argno(line, &cmdtype);
363 d3e1ab0c 2022-11-24 op if (argno == 0)
364 d3e1ab0c 2022-11-24 op return rl_completion_matches(text, ftp_cmdname_generator);
366 4c9b5bd4 2022-12-21 op /* but not for file completions */
367 4c9b5bd4 2022-12-21 op rl_completion_append_character = '\0';
369 d3e1ab0c 2022-11-24 op switch (cmdtype) {
370 d3e1ab0c 2022-11-24 op case CMD_BELL:
371 d3e1ab0c 2022-11-24 op case CMD_HEXDUMP:
372 d3e1ab0c 2022-11-24 op case CMD_VERBOSE:
373 d3e1ab0c 2022-11-24 op if (argno != 1)
375 d3e1ab0c 2022-11-24 op return rl_completion_matches(text, ftp_bool_generator);
377 d3e1ab0c 2022-11-24 op case CMD_BYE:
378 d3e1ab0c 2022-11-24 op case CMD_LPWD:
379 d3e1ab0c 2022-11-24 op /* no args */
383 d3e1ab0c 2022-11-24 op case CMD_EDIT:
385 d3e1ab0c 2022-11-24 op case CMD_PAGE:
386 d3e1ab0c 2022-11-24 op if (argno != 1)
388 d3e1ab0c 2022-11-24 op /* fallthrough */
390 d3e1ab0c 2022-11-24 op return ftp_remote_files(text, start, end);
392 d3e1ab0c 2022-11-24 op case CMD_GET:
393 d3e1ab0c 2022-11-24 op if (argno > 2)
395 d3e1ab0c 2022-11-24 op if (argno == 2)
396 d3e1ab0c 2022-11-24 op return ftp_remote_files(text, start, end);
397 d3e1ab0c 2022-11-24 op /* try local */
398 d3e1ab0c 2022-11-24 op rl_attempted_completion_over = 0;
401 d3e1ab0c 2022-11-24 op case CMD_LCD:
402 d3e1ab0c 2022-11-24 op if (argno != 1)
404 d3e1ab0c 2022-11-24 op /* try local */
405 d3e1ab0c 2022-11-24 op rl_attempted_completion_over = 0;
408 d3e1ab0c 2022-11-24 op case CMD_PIPE:
409 d3e1ab0c 2022-11-24 op if (argno > 2)
411 d3e1ab0c 2022-11-24 op if (argno == 1)
412 d3e1ab0c 2022-11-24 op return ftp_remote_files(text, start, end);
413 d3e1ab0c 2022-11-24 op /* try local */
414 d3e1ab0c 2022-11-24 op rl_attempted_completion_over = 0;
417 d3e1ab0c 2022-11-24 op case CMD_PUT:
418 d3e1ab0c 2022-11-24 op if (argno > 2)
420 d3e1ab0c 2022-11-24 op if (argno == 1) {
421 d3e1ab0c 2022-11-24 op /* try local */
422 d3e1ab0c 2022-11-24 op rl_attempted_completion_over = 0;
425 d3e1ab0c 2022-11-24 op return ftp_remote_files(text, start, end);
427 d3e1ab0c 2022-11-24 op case CMD_RENAME:
428 d3e1ab0c 2022-11-24 op if (argno > 2)
430 d3e1ab0c 2022-11-24 op return ftp_remote_files(text, start, end);
437 d3e1ab0c 2022-11-24 op ftp_quoted(char *line, int index)
439 d3e1ab0c 2022-11-24 op if (index > 0 && line[index - 1] == '\\')
440 d3e1ab0c 2022-11-24 op return !ftp_quoted(line, index - 1);
445 d3e1ab0c 2022-11-24 op compl_setup(void)
447 d3e1ab0c 2022-11-24 op rl_attempted_completion_function = ftp_completion;
448 d3e1ab0c 2022-11-24 op rl_completer_word_break_characters = "\t ";
449 d3e1ab0c 2022-11-24 op rl_completer_quote_characters = "\"'";
450 d3e1ab0c 2022-11-24 op rl_char_is_quoted_p = ftp_quoted;