xref: /minix3/usr.bin/ftp/complete.c (revision 04203a83a6848544e5157be39991291ba0c82525)
1 /*	$NetBSD: complete.c,v 1.46 2009/04/12 10:18:52 lukem Exp $	*/
2 
3 /*-
4  * Copyright (c) 1997-2009 The NetBSD Foundation, Inc.
5  * All rights reserved.
6  *
7  * This code is derived from software contributed to The NetBSD Foundation
8  * by Luke Mewburn.
9  *
10  * Redistribution and use in source and binary forms, with or without
11  * modification, are permitted provided that the following conditions
12  * are met:
13  * 1. Redistributions of source code must retain the above copyright
14  *    notice, this list of conditions and the following disclaimer.
15  * 2. Redistributions in binary form must reproduce the above copyright
16  *    notice, this list of conditions and the following disclaimer in the
17  *    documentation and/or other materials provided with the distribution.
18  *
19  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
20  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
21  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
22  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
23  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
24  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
25  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
26  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
27  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
28  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
29  * POSSIBILITY OF SUCH DAMAGE.
30  */
31 
32 #include <sys/cdefs.h>
33 #ifndef lint
34 __RCSID("$NetBSD: complete.c,v 1.46 2009/04/12 10:18:52 lukem Exp $");
35 #endif /* not lint */
36 
37 /*
38  * FTP user program - command and file completion routines
39  */
40 
41 #include <sys/stat.h>
42 
43 #include <ctype.h>
44 #include <err.h>
45 #include <dirent.h>
46 #include <stdio.h>
47 #include <stdlib.h>
48 #include <string.h>
49 
50 #include "ftp_var.h"
51 
52 #ifndef NO_EDITCOMPLETE
53 
54 static int	     comparstr		(const void *, const void *);
55 static unsigned char complete_ambiguous	(char *, int, StringList *);
56 static unsigned char complete_command	(char *, int);
57 static unsigned char complete_local	(char *, int);
58 static unsigned char complete_option	(char *, int);
59 static unsigned char complete_remote	(char *, int);
60 
61 static int
comparstr(const void * a,const void * b)62 comparstr(const void *a, const void *b)
63 {
64 	return (strcmp(*(const char * const *)a, *(const char * const *)b));
65 }
66 
67 /*
68  * Determine if complete is ambiguous. If unique, insert.
69  * If no choices, error. If unambiguous prefix, insert that.
70  * Otherwise, list choices. words is assumed to be filtered
71  * to only contain possible choices.
72  * Args:
73  *	word	word which started the match
74  *	list	list by default
75  *	words	stringlist containing possible matches
76  * Returns a result as per el_set(EL_ADDFN, ...)
77  */
78 static unsigned char
complete_ambiguous(char * word,int list,StringList * words)79 complete_ambiguous(char *word, int list, StringList *words)
80 {
81 	char insertstr[MAXPATHLEN];
82 	char *lastmatch, *p;
83 	size_t i, j;
84 	size_t matchlen, wordlen;
85 
86 	wordlen = strlen(word);
87 	if (words->sl_cur == 0)
88 		return (CC_ERROR);	/* no choices available */
89 
90 	if (words->sl_cur == 1) {	/* only once choice available */
91 		p = words->sl_str[0] + wordlen;
92 		if (*p == '\0')		/* at end of word? */
93 			return (CC_REFRESH);
94 		ftpvis(insertstr, sizeof(insertstr), p, strlen(p));
95 		if (el_insertstr(el, insertstr) == -1)
96 			return (CC_ERROR);
97 		else
98 			return (CC_REFRESH);
99 	}
100 
101 	if (!list) {
102 		matchlen = 0;
103 		lastmatch = words->sl_str[0];
104 		matchlen = strlen(lastmatch);
105 		for (i = 1 ; i < words->sl_cur ; i++) {
106 			for (j = wordlen ; j < strlen(words->sl_str[i]); j++)
107 				if (lastmatch[j] != words->sl_str[i][j])
108 					break;
109 			if (j < matchlen)
110 				matchlen = j;
111 		}
112 		if (matchlen > wordlen) {
113 			ftpvis(insertstr, sizeof(insertstr),
114 			    lastmatch + wordlen, matchlen - wordlen);
115 			if (el_insertstr(el, insertstr) == -1)
116 				return (CC_ERROR);
117 			else
118 				return (CC_REFRESH_BEEP);
119 		}
120 	}
121 
122 	putc('\n', ttyout);
123 	qsort(words->sl_str, words->sl_cur, sizeof(char *), comparstr);
124 	list_vertical(words);
125 	return (CC_REDISPLAY);
126 }
127 
128 /*
129  * Complete a command
130  */
131 static unsigned char
complete_command(char * word,int list)132 complete_command(char *word, int list)
133 {
134 	struct cmd *c;
135 	StringList *words;
136 	size_t wordlen;
137 	unsigned char rv;
138 
139 	words = ftp_sl_init();
140 	wordlen = strlen(word);
141 
142 	for (c = cmdtab; c->c_name != NULL; c++) {
143 		if (wordlen > strlen(c->c_name))
144 			continue;
145 		if (strncmp(word, c->c_name, wordlen) == 0)
146 			ftp_sl_add(words, ftp_strdup(c->c_name));
147 	}
148 
149 	rv = complete_ambiguous(word, list, words);
150 	if (rv == CC_REFRESH) {
151 		if (el_insertstr(el, " ") == -1)
152 			rv = CC_ERROR;
153 	}
154 	sl_free(words, 1);
155 	return (rv);
156 }
157 
158 /*
159  * Complete a local file
160  */
161 static unsigned char
complete_local(char * word,int list)162 complete_local(char *word, int list)
163 {
164 	StringList *words;
165 	char dir[MAXPATHLEN];
166 	char *file;
167 	DIR *dd;
168 	struct dirent *dp;
169 	unsigned char rv;
170 	size_t len;
171 
172 	if ((file = strrchr(word, '/')) == NULL) {
173 		dir[0] = '.';
174 		dir[1] = '\0';
175 		file = word;
176 	} else {
177 		if (file == word) {
178 			dir[0] = '/';
179 			dir[1] = '\0';
180 		} else
181 			(void)strlcpy(dir, word, file - word + 1);
182 		file++;
183 	}
184 	if (dir[0] == '~') {
185 		char *p;
186 
187 		if ((p = globulize(dir)) == NULL)
188 			return (CC_ERROR);
189 		(void)strlcpy(dir, p, sizeof(dir));
190 		free(p);
191 	}
192 
193 	if ((dd = opendir(dir)) == NULL)
194 		return (CC_ERROR);
195 
196 	words = ftp_sl_init();
197 	len = strlen(file);
198 
199 	for (dp = readdir(dd); dp != NULL; dp = readdir(dd)) {
200 		if (!strcmp(dp->d_name, ".") || !strcmp(dp->d_name, ".."))
201 			continue;
202 
203 #if defined(DIRENT_MISSING_D_NAMLEN)
204 		if (len > strlen(dp->d_name))
205 			continue;
206 #else
207 		if (len > dp->d_namlen)
208 			continue;
209 #endif
210 		if (strncmp(file, dp->d_name, len) == 0) {
211 			char *tcp;
212 
213 			tcp = ftp_strdup(dp->d_name);
214 			ftp_sl_add(words, tcp);
215 		}
216 	}
217 	closedir(dd);
218 
219 	rv = complete_ambiguous(file, list, words);
220 	if (rv == CC_REFRESH) {
221 		struct stat sb;
222 		char path[MAXPATHLEN];
223 
224 		(void)strlcpy(path, dir,		sizeof(path));
225 		(void)strlcat(path, "/",		sizeof(path));
226 		(void)strlcat(path, words->sl_str[0],	sizeof(path));
227 
228 		if (stat(path, &sb) >= 0) {
229 			char suffix[2] = " ";
230 
231 			if (S_ISDIR(sb.st_mode))
232 				suffix[0] = '/';
233 			if (el_insertstr(el, suffix) == -1)
234 				rv = CC_ERROR;
235 		}
236 	}
237 	sl_free(words, 1);
238 	return (rv);
239 }
240 /*
241  * Complete an option
242  */
243 static unsigned char
complete_option(char * word,int list)244 complete_option(char *word, int list)
245 {
246 	struct option *o;
247 	StringList *words;
248 	size_t wordlen;
249 	unsigned char rv;
250 
251 	words = ftp_sl_init();
252 	wordlen = strlen(word);
253 
254 	for (o = optiontab; o->name != NULL; o++) {
255 		if (wordlen > strlen(o->name))
256 			continue;
257 		if (strncmp(word, o->name, wordlen) == 0)
258 			ftp_sl_add(words, ftp_strdup(o->name));
259 	}
260 
261 	rv = complete_ambiguous(word, list, words);
262 	if (rv == CC_REFRESH) {
263 		if (el_insertstr(el, " ") == -1)
264 			rv = CC_ERROR;
265 	}
266 	sl_free(words, 1);
267 	return (rv);
268 }
269 
270 /*
271  * Complete a remote file
272  */
273 static unsigned char
complete_remote(char * word,int list)274 complete_remote(char *word, int list)
275 {
276 	static StringList *dirlist;
277 	static char	 lastdir[MAXPATHLEN];
278 	StringList	*words;
279 	char		 dir[MAXPATHLEN];
280 	char		*file, *cp;
281 	size_t		 i;
282 	unsigned char	 rv;
283 	char		 cmdbuf[MAX_C_NAME];
284 	char		*dummyargv[3] = { NULL, NULL, NULL };
285 
286 	(void)strlcpy(cmdbuf, "complete", sizeof(cmdbuf));
287 	dummyargv[0] = cmdbuf;
288 	dummyargv[1] = dir;
289 
290 	if ((file = strrchr(word, '/')) == NULL) {
291 		dir[0] = '\0';
292 		file = word;
293 	} else {
294 		cp = file;
295 		while (*cp == '/' && cp > word)
296 			cp--;
297 		(void)strlcpy(dir, word, cp - word + 2);
298 		file++;
299 	}
300 
301 	if (dirchange || dirlist == NULL ||
302 	    strcmp(dir, lastdir) != 0) {		/* dir not cached */
303 		const char *emesg;
304 
305 		if (dirlist != NULL)
306 			sl_free(dirlist, 1);
307 		dirlist = ftp_sl_init();
308 
309 		mflag = 1;
310 		emesg = NULL;
311 		while ((cp = remglob(dummyargv, 0, &emesg)) != NULL) {
312 			char *tcp;
313 
314 			if (!mflag)
315 				continue;
316 			if (*cp == '\0') {
317 				mflag = 0;
318 				continue;
319 			}
320 			tcp = strrchr(cp, '/');
321 			if (tcp)
322 				tcp++;
323 			else
324 				tcp = cp;
325 			tcp = ftp_strdup(tcp);
326 			ftp_sl_add(dirlist, tcp);
327 		}
328 		if (emesg != NULL) {
329 			fprintf(ttyout, "\n%s\n", emesg);
330 			return (CC_REDISPLAY);
331 		}
332 		(void)strlcpy(lastdir, dir, sizeof(lastdir));
333 		dirchange = 0;
334 	}
335 
336 	words = ftp_sl_init();
337 	for (i = 0; i < dirlist->sl_cur; i++) {
338 		cp = dirlist->sl_str[i];
339 		if (strlen(file) > strlen(cp))
340 			continue;
341 		if (strncmp(file, cp, strlen(file)) == 0)
342 			ftp_sl_add(words, cp);
343 	}
344 	rv = complete_ambiguous(file, list, words);
345 	sl_free(words, 0);
346 	return (rv);
347 }
348 
349 /*
350  * Generic complete routine
351  */
352 unsigned char
complete(EditLine * cel,int ch)353 complete(EditLine *cel, int ch)
354 {
355 	static char word[FTPBUFLEN];
356 	static size_t lastc_argc, lastc_argo;
357 
358 	struct cmd *c;
359 	const LineInfo *lf;
360 	int dolist, cmpltype;
361 	size_t celems, len;
362 
363 	lf = el_line(cel);
364 	len = lf->lastchar - lf->buffer;
365 	if (len >= sizeof(line))
366 		return (CC_ERROR);
367 	(void)strlcpy(line, lf->buffer, len + 1);
368 	cursor_pos = line + (lf->cursor - lf->buffer);
369 	lastc_argc = cursor_argc;	/* remember last cursor pos */
370 	lastc_argo = cursor_argo;
371 	makeargv();			/* build argc/argv of current line */
372 
373 	if (cursor_argo >= sizeof(word))
374 		return (CC_ERROR);
375 
376 	dolist = 0;
377 			/* if cursor and word is same, list alternatives */
378 	if (lastc_argc == cursor_argc && lastc_argo == cursor_argo
379 	    && strncmp(word, margv[cursor_argc] ? margv[cursor_argc] : "",
380 			cursor_argo) == 0)
381 		dolist = 1;
382 	else if (cursor_argc < (size_t)margc)
383 		(void)strlcpy(word, margv[cursor_argc], cursor_argo + 1);
384 	word[cursor_argo] = '\0';
385 
386 	if (cursor_argc == 0)
387 		return (complete_command(word, dolist));
388 
389 	c = getcmd(margv[0]);
390 	if (c == (struct cmd *)-1 || c == 0)
391 		return (CC_ERROR);
392 	celems = strlen(c->c_complete);
393 
394 		/* check for 'continuation' completes (which are uppercase) */
395 	if ((cursor_argc > celems) && (celems > 0)
396 	    && isupper((unsigned char) c->c_complete[celems-1]))
397 		cursor_argc = celems;
398 
399 	if (cursor_argc > celems)
400 		return (CC_ERROR);
401 
402 	cmpltype = c->c_complete[cursor_argc - 1];
403 	switch (cmpltype) {
404 		case 'c':			/* command complete */
405 		case 'C':
406 			return (complete_command(word, dolist));
407 		case 'l':			/* local complete */
408 		case 'L':
409 			return (complete_local(word, dolist));
410 		case 'n':			/* no complete */
411 		case 'N':			/* no complete */
412 			return (CC_ERROR);
413 		case 'o':			/* local complete */
414 		case 'O':
415 			return (complete_option(word, dolist));
416 		case 'r':			/* remote complete */
417 		case 'R':
418 			if (connected != -1) {
419 				fputs("\nMust be logged in to complete.\n",
420 				    ttyout);
421 				return (CC_REDISPLAY);
422 			}
423 			return (complete_remote(word, dolist));
424 		default:
425 			errx(1, "complete: unknown complete type `%c'",
426 			    cmpltype);
427 			return (CC_ERROR);
428 	}
429 	/* NOTREACHED */
430 }
431 
432 #endif /* !NO_EDITCOMPLETE */
433