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 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 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 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 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 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 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 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