Annotation of src/usr.bin/ssh/sftp-int.c, Revision 1.52
1.1 djm 1: /*
1.44 djm 2: * Copyright (c) 2001,2002 Damien Miller. All rights reserved.
1.1 djm 3: *
4: * Redistribution and use in source and binary forms, with or without
5: * modification, are permitted provided that the following conditions
6: * are met:
7: * 1. Redistributions of source code must retain the above copyright
8: * notice, this list of conditions and the following disclaimer.
9: * 2. Redistributions in binary form must reproduce the above copyright
10: * notice, this list of conditions and the following disclaimer in the
11: * documentation and/or other materials provided with the distribution.
12: *
13: * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
14: * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
15: * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
16: * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
17: * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
18: * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
19: * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
20: * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
21: * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
22: * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
23: */
24:
25: /* XXX: recursive operations */
26:
27: #include "includes.h"
1.51 djm 28: RCSID("$OpenBSD: sftp-int.c,v 1.50 2002/11/21 23:03:51 deraadt Exp $");
1.27 djm 29:
30: #include <glob.h>
1.1 djm 31:
32: #include "buffer.h"
33: #include "xmalloc.h"
34: #include "log.h"
35: #include "pathnames.h"
36:
37: #include "sftp.h"
38: #include "sftp-common.h"
1.27 djm 39: #include "sftp-glob.h"
1.1 djm 40: #include "sftp-client.h"
41: #include "sftp-int.h"
42:
1.26 djm 43: /* File to read commands from */
44: extern FILE *infile;
45:
1.42 djm 46: /* Size of buffer used when copying files */
47: extern size_t copy_buffer_len;
48:
1.43 djm 49: /* Number of concurrent outstanding requests */
50: extern int num_requests;
51:
1.52 ! fgsch 52: /* This is set to 0 if the progressmeter is not desired. */
! 53: int showprogress = 1;
! 54:
1.1 djm 55: /* Seperators for interactive commands */
56: #define WHITESPACE " \t\r\n"
57:
58: /* Commands for interactive mode */
59: #define I_CHDIR 1
60: #define I_CHGRP 2
61: #define I_CHMOD 3
62: #define I_CHOWN 4
63: #define I_GET 5
64: #define I_HELP 6
65: #define I_LCHDIR 7
66: #define I_LLS 8
67: #define I_LMKDIR 9
68: #define I_LPWD 10
69: #define I_LS 11
70: #define I_LUMASK 12
71: #define I_MKDIR 13
72: #define I_PUT 14
73: #define I_PWD 15
74: #define I_QUIT 16
75: #define I_RENAME 17
76: #define I_RM 18
77: #define I_RMDIR 19
78: #define I_SHELL 20
1.26 djm 79: #define I_SYMLINK 21
1.28 markus 80: #define I_VERSION 22
1.52 ! fgsch 81: #define I_PROGRESS 23
1.1 djm 82:
83: struct CMD {
1.6 deraadt 84: const char *c;
1.1 djm 85: const int n;
86: };
87:
88: const struct CMD cmds[] = {
1.40 markus 89: { "bye", I_QUIT },
1.15 stevesk 90: { "cd", I_CHDIR },
91: { "chdir", I_CHDIR },
92: { "chgrp", I_CHGRP },
93: { "chmod", I_CHMOD },
94: { "chown", I_CHOWN },
95: { "dir", I_LS },
96: { "exit", I_QUIT },
97: { "get", I_GET },
1.34 djm 98: { "mget", I_GET },
1.15 stevesk 99: { "help", I_HELP },
100: { "lcd", I_LCHDIR },
101: { "lchdir", I_LCHDIR },
102: { "lls", I_LLS },
103: { "lmkdir", I_LMKDIR },
1.26 djm 104: { "ln", I_SYMLINK },
1.15 stevesk 105: { "lpwd", I_LPWD },
106: { "ls", I_LS },
107: { "lumask", I_LUMASK },
108: { "mkdir", I_MKDIR },
1.52 ! fgsch 109: { "progress", I_PROGRESS },
1.15 stevesk 110: { "put", I_PUT },
1.34 djm 111: { "mput", I_PUT },
1.15 stevesk 112: { "pwd", I_PWD },
113: { "quit", I_QUIT },
114: { "rename", I_RENAME },
115: { "rm", I_RM },
116: { "rmdir", I_RMDIR },
1.26 djm 117: { "symlink", I_SYMLINK },
1.28 markus 118: { "version", I_VERSION },
1.6 deraadt 119: { "!", I_SHELL },
1.7 deraadt 120: { "?", I_HELP },
1.6 deraadt 121: { NULL, -1}
1.1 djm 122: };
123:
1.37 itojun 124: static void
1.1 djm 125: help(void)
126: {
127: printf("Available commands:\n");
1.13 stevesk 128: printf("cd path Change remote directory to 'path'\n");
129: printf("lcd path Change local directory to 'path'\n");
130: printf("chgrp grp path Change group of file 'path' to 'grp'\n");
131: printf("chmod mode path Change permissions of file 'path' to 'mode'\n");
132: printf("chown own path Change owner of file 'path' to 'own'\n");
133: printf("help Display this help text\n");
134: printf("get remote-path [local-path] Download file\n");
135: printf("lls [ls-options [path]] Display local directory listing\n");
1.26 djm 136: printf("ln oldpath newpath Symlink remote file\n");
1.13 stevesk 137: printf("lmkdir path Create local directory\n");
138: printf("lpwd Print local working directory\n");
139: printf("ls [path] Display remote directory listing\n");
140: printf("lumask umask Set local umask to 'umask'\n");
141: printf("mkdir path Create remote directory\n");
1.52 ! fgsch 142: printf("preogress Toggle display of progress meter\n");
1.13 stevesk 143: printf("put local-path [remote-path] Upload file\n");
144: printf("pwd Display remote working directory\n");
145: printf("exit Quit sftp\n");
146: printf("quit Quit sftp\n");
147: printf("rename oldpath newpath Rename remote file\n");
148: printf("rmdir path Remove remote directory\n");
149: printf("rm path Delete remote file\n");
1.26 djm 150: printf("symlink oldpath newpath Symlink remote file\n");
1.28 markus 151: printf("version Show SFTP version\n");
1.1 djm 152: printf("!command Execute 'command' in local shell\n");
153: printf("! Escape to local shell\n");
1.13 stevesk 154: printf("? Synonym for help\n");
1.1 djm 155: }
156:
1.37 itojun 157: static void
1.1 djm 158: local_do_shell(const char *args)
159: {
1.36 markus 160: int status;
1.1 djm 161: char *shell;
162: pid_t pid;
1.3 stevesk 163:
1.1 djm 164: if (!*args)
165: args = NULL;
1.3 stevesk 166:
1.1 djm 167: if ((shell = getenv("SHELL")) == NULL)
168: shell = _PATH_BSHELL;
169:
170: if ((pid = fork()) == -1)
171: fatal("Couldn't fork: %s", strerror(errno));
172:
173: if (pid == 0) {
174: /* XXX: child has pipe fds to ssh subproc open - issue? */
175: if (args) {
176: debug3("Executing %s -c \"%s\"", shell, args);
1.38 deraadt 177: execl(shell, shell, "-c", args, (char *)NULL);
1.1 djm 178: } else {
179: debug3("Executing %s", shell);
1.38 deraadt 180: execl(shell, shell, (char *)NULL);
1.1 djm 181: }
1.3 stevesk 182: fprintf(stderr, "Couldn't execute \"%s\": %s\n", shell,
1.1 djm 183: strerror(errno));
184: _exit(1);
185: }
1.46 markus 186: while (waitpid(pid, &status, 0) == -1)
187: if (errno != EINTR)
188: fatal("Couldn't wait for child: %s", strerror(errno));
1.1 djm 189: if (!WIFEXITED(status))
190: error("Shell exited abormally");
191: else if (WEXITSTATUS(status))
192: error("Shell exited with status %d", WEXITSTATUS(status));
193: }
194:
1.37 itojun 195: static void
1.1 djm 196: local_do_ls(const char *args)
197: {
198: if (!args || !*args)
1.18 stevesk 199: local_do_shell(_PATH_LS);
1.1 djm 200: else {
1.18 stevesk 201: int len = strlen(_PATH_LS " ") + strlen(args) + 1;
1.16 deraadt 202: char *buf = xmalloc(len);
1.1 djm 203:
204: /* XXX: quoting - rip quoting code from ftp? */
1.18 stevesk 205: snprintf(buf, len, _PATH_LS " %s", args);
1.1 djm 206: local_do_shell(buf);
1.16 deraadt 207: xfree(buf);
1.1 djm 208: }
209: }
210:
1.48 djm 211: /* Strip one path (usually the pwd) from the start of another */
212: static char *
213: path_strip(char *path, char *strip)
214: {
215: size_t len;
216:
217: if (strip == NULL)
218: return (xstrdup(path));
219:
220: len = strlen(strip);
221: if (strip != NULL && strncmp(path, strip, len) == 0) {
222: if (strip[len - 1] != '/' && path[len] == '/')
223: len++;
224: return (xstrdup(path + len));
225: }
226:
227: return (xstrdup(path));
228: }
229:
1.37 itojun 230: static char *
1.29 djm 231: path_append(char *p1, char *p2)
232: {
233: char *ret;
1.31 markus 234: int len = strlen(p1) + strlen(p2) + 2;
1.29 djm 235:
1.31 markus 236: ret = xmalloc(len);
237: strlcpy(ret, p1, len);
1.48 djm 238: if (p1[strlen(p1) - 1] != '/')
1.39 jakob 239: strlcat(ret, "/", len);
1.31 markus 240: strlcat(ret, p2, len);
1.29 djm 241:
242: return(ret);
243: }
244:
1.37 itojun 245: static char *
1.1 djm 246: make_absolute(char *p, char *pwd)
247: {
1.29 djm 248: char *abs;
1.1 djm 249:
250: /* Derelativise */
251: if (p && p[0] != '/') {
1.29 djm 252: abs = path_append(pwd, p);
1.1 djm 253: xfree(p);
1.29 djm 254: return(abs);
255: } else
256: return(p);
257: }
258:
1.37 itojun 259: static int
1.29 djm 260: infer_path(const char *p, char **ifp)
261: {
262: char *cp;
263:
264: cp = strrchr(p, '/');
265: if (cp == NULL) {
266: *ifp = xstrdup(p);
267: return(0);
1.1 djm 268: }
269:
1.29 djm 270: if (!cp[1]) {
271: error("Invalid path");
272: return(-1);
273: }
274:
275: *ifp = xstrdup(cp + 1);
276: return(0);
1.1 djm 277: }
278:
1.37 itojun 279: static int
1.1 djm 280: parse_getput_flags(const char **cpp, int *pflag)
281: {
282: const char *cp = *cpp;
283:
284: /* Check for flags */
285: if (cp[0] == '-' && cp[1] && strchr(WHITESPACE, cp[2])) {
1.20 djm 286: switch (cp[1]) {
1.22 djm 287: case 'p':
1.1 djm 288: case 'P':
289: *pflag = 1;
290: break;
291: default:
1.22 djm 292: error("Invalid flag -%c", cp[1]);
1.1 djm 293: return(-1);
294: }
295: cp += 2;
296: *cpp = cp + strspn(cp, WHITESPACE);
297: }
298:
299: return(0);
300: }
301:
1.37 itojun 302: static int
1.48 djm 303: parse_ls_flags(const char **cpp, int *lflag)
304: {
305: const char *cp = *cpp;
306:
307: /* Check for flags */
308: if (cp++[0] == '-') {
309: for(; strchr(WHITESPACE, *cp) == NULL; cp++) {
310: switch (*cp) {
311: case 'l':
312: *lflag = 1;
313: break;
314: default:
315: error("Invalid flag -%c", *cp);
316: return(-1);
317: }
318: }
319: *cpp = cp + strspn(cp, WHITESPACE);
320: }
321:
322: return(0);
323: }
324:
325: static int
1.1 djm 326: get_pathname(const char **cpp, char **path)
327: {
1.8 provos 328: const char *cp = *cpp, *end;
329: char quot;
1.1 djm 330: int i;
331:
332: cp += strspn(cp, WHITESPACE);
333: if (!*cp) {
334: *cpp = cp;
335: *path = NULL;
1.8 provos 336: return (0);
1.1 djm 337: }
338:
339: /* Check for quoted filenames */
340: if (*cp == '\"' || *cp == '\'') {
1.8 provos 341: quot = *cp++;
1.30 markus 342:
1.8 provos 343: end = strchr(cp, quot);
344: if (end == NULL) {
1.1 djm 345: error("Unterminated quote");
1.8 provos 346: goto fail;
1.1 djm 347: }
1.8 provos 348: if (cp == end) {
1.1 djm 349: error("Empty quotes");
1.8 provos 350: goto fail;
1.1 djm 351: }
1.8 provos 352: *cpp = end + 1 + strspn(end + 1, WHITESPACE);
353: } else {
354: /* Read to end of filename */
355: end = strpbrk(cp, WHITESPACE);
356: if (end == NULL)
357: end = strchr(cp, '\0');
358: *cpp = end + strspn(end, WHITESPACE);
1.1 djm 359: }
360:
1.8 provos 361: i = end - cp;
1.1 djm 362:
363: *path = xmalloc(i + 1);
364: memcpy(*path, cp, i);
365: (*path)[i] = '\0';
366: return(0);
1.8 provos 367:
368: fail:
369: *path = NULL;
370: return (-1);
1.1 djm 371: }
372:
1.37 itojun 373: static int
1.29 djm 374: is_dir(char *path)
375: {
376: struct stat sb;
377:
378: /* XXX: report errors? */
379: if (stat(path, &sb) == -1)
380: return(0);
381:
382: return(sb.st_mode & S_IFDIR);
383: }
384:
1.37 itojun 385: static int
1.44 djm 386: remote_is_dir(struct sftp_conn *conn, char *path)
1.1 djm 387: {
1.29 djm 388: Attrib *a;
1.1 djm 389:
1.29 djm 390: /* XXX: report errors? */
1.44 djm 391: if ((a = do_stat(conn, path, 1)) == NULL)
1.29 djm 392: return(0);
393: if (!(a->flags & SSH2_FILEXFER_ATTR_PERMISSIONS))
1.1 djm 394: return(0);
1.29 djm 395: return(a->perm & S_IFDIR);
396: }
397:
1.37 itojun 398: static int
1.44 djm 399: process_get(struct sftp_conn *conn, char *src, char *dst, char *pwd, int pflag)
1.29 djm 400: {
401: char *abs_src = NULL;
402: char *abs_dst = NULL;
403: char *tmp;
404: glob_t g;
405: int err = 0;
406: int i;
1.30 markus 407:
1.29 djm 408: abs_src = xstrdup(src);
409: abs_src = make_absolute(abs_src, pwd);
410:
1.30 markus 411: memset(&g, 0, sizeof(g));
1.29 djm 412: debug3("Looking up %s", abs_src);
1.44 djm 413: if (remote_glob(conn, abs_src, 0, NULL, &g)) {
1.29 djm 414: error("File \"%s\" not found.", abs_src);
415: err = -1;
416: goto out;
417: }
418:
419: /* Only one match, dst may be file, directory or unspecified */
420: if (g.gl_pathv[0] && g.gl_matchc == 1) {
421: if (dst) {
422: /* If directory specified, append filename */
423: if (is_dir(dst)) {
424: if (infer_path(g.gl_pathv[0], &tmp)) {
425: err = 1;
426: goto out;
427: }
428: abs_dst = path_append(dst, tmp);
429: xfree(tmp);
430: } else
431: abs_dst = xstrdup(dst);
432: } else if (infer_path(g.gl_pathv[0], &abs_dst)) {
433: err = -1;
434: goto out;
435: }
1.44 djm 436: err = do_download(conn, g.gl_pathv[0], abs_dst, pflag);
1.29 djm 437: goto out;
438: }
439:
440: /* Multiple matches, dst may be directory or unspecified */
441: if (dst && !is_dir(dst)) {
1.30 markus 442: error("Multiple files match, but \"%s\" is not a directory",
1.29 djm 443: dst);
444: err = -1;
445: goto out;
446: }
1.30 markus 447:
1.41 deraadt 448: for (i = 0; g.gl_pathv[i]; i++) {
1.29 djm 449: if (infer_path(g.gl_pathv[i], &tmp)) {
450: err = -1;
451: goto out;
452: }
453: if (dst) {
454: abs_dst = path_append(dst, tmp);
455: xfree(tmp);
456: } else
457: abs_dst = tmp;
458:
459: printf("Fetching %s to %s\n", g.gl_pathv[i], abs_dst);
1.44 djm 460: if (do_download(conn, g.gl_pathv[i], abs_dst, pflag) == -1)
1.29 djm 461: err = -1;
462: xfree(abs_dst);
463: abs_dst = NULL;
1.1 djm 464: }
465:
1.29 djm 466: out:
467: xfree(abs_src);
468: if (abs_dst)
469: xfree(abs_dst);
470: globfree(&g);
471: return(err);
472: }
473:
1.37 itojun 474: static int
1.44 djm 475: process_put(struct sftp_conn *conn, char *src, char *dst, char *pwd, int pflag)
1.29 djm 476: {
477: char *tmp_dst = NULL;
478: char *abs_dst = NULL;
479: char *tmp;
480: glob_t g;
481: int err = 0;
482: int i;
483:
484: if (dst) {
485: tmp_dst = xstrdup(dst);
486: tmp_dst = make_absolute(tmp_dst, pwd);
487: }
488:
1.30 markus 489: memset(&g, 0, sizeof(g));
1.29 djm 490: debug3("Looking up %s", src);
491: if (glob(src, 0, NULL, &g)) {
492: error("File \"%s\" not found.", src);
493: err = -1;
494: goto out;
495: }
496:
497: /* Only one match, dst may be file, directory or unspecified */
498: if (g.gl_pathv[0] && g.gl_matchc == 1) {
499: if (tmp_dst) {
500: /* If directory specified, append filename */
1.44 djm 501: if (remote_is_dir(conn, tmp_dst)) {
1.29 djm 502: if (infer_path(g.gl_pathv[0], &tmp)) {
503: err = 1;
504: goto out;
505: }
506: abs_dst = path_append(tmp_dst, tmp);
507: xfree(tmp);
508: } else
509: abs_dst = xstrdup(tmp_dst);
1.32 markus 510: } else {
511: if (infer_path(g.gl_pathv[0], &abs_dst)) {
512: err = -1;
513: goto out;
514: }
515: abs_dst = make_absolute(abs_dst, pwd);
1.29 djm 516: }
1.44 djm 517: err = do_upload(conn, g.gl_pathv[0], abs_dst, pflag);
1.29 djm 518: goto out;
519: }
520:
521: /* Multiple matches, dst may be directory or unspecified */
1.44 djm 522: if (tmp_dst && !remote_is_dir(conn, tmp_dst)) {
1.30 markus 523: error("Multiple files match, but \"%s\" is not a directory",
1.29 djm 524: tmp_dst);
525: err = -1;
526: goto out;
527: }
528:
1.41 deraadt 529: for (i = 0; g.gl_pathv[i]; i++) {
1.29 djm 530: if (infer_path(g.gl_pathv[i], &tmp)) {
531: err = -1;
532: goto out;
533: }
534: if (tmp_dst) {
535: abs_dst = path_append(tmp_dst, tmp);
536: xfree(tmp);
537: } else
538: abs_dst = make_absolute(tmp, pwd);
539:
540: printf("Uploading %s to %s\n", g.gl_pathv[i], abs_dst);
1.44 djm 541: if (do_upload(conn, g.gl_pathv[i], abs_dst, pflag) == -1)
1.29 djm 542: err = -1;
1.1 djm 543: }
544:
1.29 djm 545: out:
546: if (abs_dst)
547: xfree(abs_dst);
548: if (tmp_dst)
549: xfree(tmp_dst);
550: return(err);
1.1 djm 551: }
552:
1.37 itojun 553: static int
1.48 djm 554: sdirent_comp(const void *aa, const void *bb)
555: {
556: SFTP_DIRENT *a = *(SFTP_DIRENT **)aa;
557: SFTP_DIRENT *b = *(SFTP_DIRENT **)bb;
558:
1.50 deraadt 559: return (strcmp(a->filename, b->filename));
1.48 djm 560: }
561:
562: /* sftp ls.1 replacement for directories */
563: static int
564: do_ls_dir(struct sftp_conn *conn, char *path, char *strip_path, int lflag)
565: {
566: int n;
567: SFTP_DIRENT **d;
568:
569: if ((n = do_readdir(conn, path, &d)) != 0)
570: return (n);
571:
1.50 deraadt 572: /* Count entries for sort */
1.48 djm 573: for (n = 0; d[n] != NULL; n++)
574: ;
575:
576: qsort(d, n, sizeof(*d), sdirent_comp);
577:
578: for (n = 0; d[n] != NULL; n++) {
579: char *tmp, *fname;
1.50 deraadt 580:
1.48 djm 581: tmp = path_append(path, d[n]->filename);
582: fname = path_strip(tmp, strip_path);
583: xfree(tmp);
584:
585: if (lflag) {
586: char *lname;
587: struct stat sb;
588:
589: memset(&sb, 0, sizeof(sb));
590: attrib_to_stat(&d[n]->a, &sb);
591: lname = ls_file(fname, &sb, 1);
592: printf("%s\n", lname);
593: xfree(lname);
594: } else {
595: /* XXX - multicolumn display would be nice here */
596: printf("%s\n", fname);
597: }
1.50 deraadt 598:
1.48 djm 599: xfree(fname);
600: }
601:
602: free_sftp_dirents(d);
603: return (0);
604: }
605:
606: /* sftp ls.1 replacement which handles path globs */
607: static int
1.50 deraadt 608: do_globbed_ls(struct sftp_conn *conn, char *path, char *strip_path,
1.48 djm 609: int lflag)
610: {
611: glob_t g;
612: int i;
613: Attrib *a;
614: struct stat sb;
615:
616: memset(&g, 0, sizeof(g));
617:
1.50 deraadt 618: if (remote_glob(conn, path, GLOB_MARK|GLOB_NOCHECK|GLOB_BRACE,
1.48 djm 619: NULL, &g)) {
620: error("Can't ls: \"%s\" not found", path);
621: return (-1);
622: }
623:
624: /*
1.50 deraadt 625: * If the glob returns a single match, which is the same as the
1.48 djm 626: * input glob, and it is a directory, then just list its contents
627: */
1.50 deraadt 628: if (g.gl_pathc == 1 &&
1.48 djm 629: strncmp(path, g.gl_pathv[0], strlen(g.gl_pathv[0]) - 1) == 0) {
630: if ((a = do_lstat(conn, path, 1)) == NULL) {
631: globfree(&g);
632: return (-1);
633: }
1.50 deraadt 634: if ((a->flags & SSH2_FILEXFER_ATTR_PERMISSIONS) &&
1.48 djm 635: S_ISDIR(a->perm)) {
636: globfree(&g);
637: return (do_ls_dir(conn, path, strip_path, lflag));
638: }
639: }
640:
641: for (i = 0; g.gl_pathv[i]; i++) {
642: char *fname, *lname;
643:
644: fname = path_strip(g.gl_pathv[i], strip_path);
645:
646: if (lflag) {
647: /*
648: * XXX: this is slow - 1 roundtrip per path
1.50 deraadt 649: * A solution to this is to fork glob() and
650: * build a sftp specific version which keeps the
1.48 djm 651: * attribs (which currently get thrown away)
652: * that the server returns as well as the filenames.
653: */
654: memset(&sb, 0, sizeof(sb));
655: a = do_lstat(conn, g.gl_pathv[i], 1);
656: if (a != NULL)
657: attrib_to_stat(a, &sb);
658: lname = ls_file(fname, &sb, 1);
659: printf("%s\n", lname);
660: xfree(lname);
661: } else {
662: /* XXX - multicolumn display would be nice here */
663: printf("%s\n", fname);
664: }
665: xfree(fname);
666: }
667:
668: if (g.gl_pathc)
669: globfree(&g);
670:
671: return (0);
672: }
673:
674: static int
1.51 djm 675: parse_args(const char **cpp, int *pflag, int *lflag, int *iflag,
1.48 djm 676: unsigned long *n_arg, char **path1, char **path2)
1.1 djm 677: {
678: const char *cmd, *cp = *cpp;
1.21 stevesk 679: char *cp2;
1.4 markus 680: int base = 0;
1.21 stevesk 681: long l;
1.1 djm 682: int i, cmdnum;
683:
684: /* Skip leading whitespace */
685: cp = cp + strspn(cp, WHITESPACE);
686:
1.51 djm 687: /* Ignore blank lines and lines which begin with comment '#' char */
688: if (*cp == '\0' || *cp == '#')
689: return (0);
1.1 djm 690:
1.51 djm 691: /* Check for leading '-' (disable error processing) */
692: *iflag = 0;
693: if (*cp == '-') {
694: *iflag = 1;
695: cp++;
696: }
697:
1.1 djm 698: /* Figure out which command we have */
1.41 deraadt 699: for (i = 0; cmds[i].c; i++) {
1.1 djm 700: int cmdlen = strlen(cmds[i].c);
701:
702: /* Check for command followed by whitespace */
703: if (!strncasecmp(cp, cmds[i].c, cmdlen) &&
704: strchr(WHITESPACE, cp[cmdlen])) {
705: cp += cmdlen;
706: cp = cp + strspn(cp, WHITESPACE);
707: break;
708: }
709: }
710: cmdnum = cmds[i].n;
711: cmd = cmds[i].c;
712:
713: /* Special case */
714: if (*cp == '!') {
715: cp++;
716: cmdnum = I_SHELL;
717: } else if (cmdnum == -1) {
718: error("Invalid command.");
1.51 djm 719: return (-1);
1.1 djm 720: }
721:
722: /* Get arguments and parse flags */
1.48 djm 723: *lflag = *pflag = *n_arg = 0;
1.1 djm 724: *path1 = *path2 = NULL;
725: switch (cmdnum) {
726: case I_GET:
727: case I_PUT:
728: if (parse_getput_flags(&cp, pflag))
729: return(-1);
730: /* Get first pathname (mandatory) */
731: if (get_pathname(&cp, path1))
732: return(-1);
733: if (*path1 == NULL) {
734: error("You must specify at least one path after a "
735: "%s command.", cmd);
736: return(-1);
737: }
738: /* Try to get second pathname (optional) */
739: if (get_pathname(&cp, path2))
740: return(-1);
741: break;
742: case I_RENAME:
1.26 djm 743: case I_SYMLINK:
1.1 djm 744: if (get_pathname(&cp, path1))
745: return(-1);
746: if (get_pathname(&cp, path2))
747: return(-1);
748: if (!*path1 || !*path2) {
749: error("You must specify two paths after a %s "
750: "command.", cmd);
751: return(-1);
752: }
753: break;
754: case I_RM:
755: case I_MKDIR:
756: case I_RMDIR:
757: case I_CHDIR:
758: case I_LCHDIR:
759: case I_LMKDIR:
760: /* Get pathname (mandatory) */
761: if (get_pathname(&cp, path1))
762: return(-1);
763: if (*path1 == NULL) {
1.3 stevesk 764: error("You must specify a path after a %s command.",
1.1 djm 765: cmd);
766: return(-1);
767: }
768: break;
769: case I_LS:
1.48 djm 770: if (parse_ls_flags(&cp, lflag))
771: return(-1);
1.1 djm 772: /* Path is optional */
773: if (get_pathname(&cp, path1))
774: return(-1);
775: break;
776: case I_LLS:
777: case I_SHELL:
778: /* Uses the rest of the line */
779: break;
780: case I_LUMASK:
1.21 stevesk 781: base = 8;
1.1 djm 782: case I_CHMOD:
1.4 markus 783: base = 8;
1.1 djm 784: case I_CHOWN:
785: case I_CHGRP:
786: /* Get numeric arg (mandatory) */
1.21 stevesk 787: l = strtol(cp, &cp2, base);
788: if (cp2 == cp || ((l == LONG_MIN || l == LONG_MAX) &&
789: errno == ERANGE) || l < 0) {
1.1 djm 790: error("You must supply a numeric argument "
791: "to the %s command.", cmd);
792: return(-1);
793: }
1.21 stevesk 794: cp = cp2;
795: *n_arg = l;
796: if (cmdnum == I_LUMASK && strchr(WHITESPACE, *cp))
797: break;
798: if (cmdnum == I_LUMASK || !strchr(WHITESPACE, *cp)) {
1.1 djm 799: error("You must supply a numeric argument "
800: "to the %s command.", cmd);
801: return(-1);
802: }
803: cp += strspn(cp, WHITESPACE);
804:
805: /* Get pathname (mandatory) */
806: if (get_pathname(&cp, path1))
807: return(-1);
808: if (*path1 == NULL) {
1.3 stevesk 809: error("You must specify a path after a %s command.",
1.1 djm 810: cmd);
811: return(-1);
812: }
813: break;
814: case I_QUIT:
815: case I_PWD:
816: case I_LPWD:
817: case I_HELP:
1.28 markus 818: case I_VERSION:
1.52 ! fgsch 819: case I_PROGRESS:
1.1 djm 820: break;
821: default:
822: fatal("Command not implemented");
823: }
824:
825: *cpp = cp;
826: return(cmdnum);
827: }
828:
1.37 itojun 829: static int
1.51 djm 830: parse_dispatch_command(struct sftp_conn *conn, const char *cmd, char **pwd,
831: int err_abort)
1.1 djm 832: {
1.8 provos 833: char *path1, *path2, *tmp;
1.51 djm 834: int pflag, lflag, iflag, cmdnum, i;
1.1 djm 835: unsigned long n_arg;
836: Attrib a, *aa;
1.23 deraadt 837: char path_buf[MAXPATHLEN];
1.25 deraadt 838: int err = 0;
1.27 djm 839: glob_t g;
1.1 djm 840:
841: path1 = path2 = NULL;
1.51 djm 842: cmdnum = parse_args(&cmd, &pflag, &lflag, &iflag, &n_arg,
1.48 djm 843: &path1, &path2);
1.1 djm 844:
1.51 djm 845: if (iflag != 0)
846: err_abort = 0;
847:
1.29 djm 848: memset(&g, 0, sizeof(g));
849:
1.1 djm 850: /* Perform command */
851: switch (cmdnum) {
1.51 djm 852: case 0:
853: /* Blank line */
854: break;
1.1 djm 855: case -1:
1.51 djm 856: /* Unrecognized command */
857: err = -1;
1.1 djm 858: break;
859: case I_GET:
1.44 djm 860: err = process_get(conn, path1, path2, *pwd, pflag);
1.1 djm 861: break;
862: case I_PUT:
1.44 djm 863: err = process_put(conn, path1, path2, *pwd, pflag);
1.33 markus 864: break;
865: case I_RENAME:
1.1 djm 866: path1 = make_absolute(path1, *pwd);
867: path2 = make_absolute(path2, *pwd);
1.44 djm 868: err = do_rename(conn, path1, path2);
1.1 djm 869: break;
1.26 djm 870: case I_SYMLINK:
1.44 djm 871: path2 = make_absolute(path2, *pwd);
872: err = do_symlink(conn, path1, path2);
1.26 djm 873: break;
1.1 djm 874: case I_RM:
875: path1 = make_absolute(path1, *pwd);
1.44 djm 876: remote_glob(conn, path1, GLOB_NOCHECK, NULL, &g);
1.41 deraadt 877: for (i = 0; g.gl_pathv[i]; i++) {
1.27 djm 878: printf("Removing %s\n", g.gl_pathv[i]);
1.51 djm 879: err = do_rm(conn, g.gl_pathv[i]);
880: if (err != 0 && err_abort)
881: break;
1.27 djm 882: }
1.1 djm 883: break;
884: case I_MKDIR:
885: path1 = make_absolute(path1, *pwd);
886: attrib_clear(&a);
887: a.flags |= SSH2_FILEXFER_ATTR_PERMISSIONS;
888: a.perm = 0777;
1.44 djm 889: err = do_mkdir(conn, path1, &a);
1.1 djm 890: break;
891: case I_RMDIR:
892: path1 = make_absolute(path1, *pwd);
1.44 djm 893: err = do_rmdir(conn, path1);
1.1 djm 894: break;
895: case I_CHDIR:
896: path1 = make_absolute(path1, *pwd);
1.44 djm 897: if ((tmp = do_realpath(conn, path1)) == NULL) {
1.25 deraadt 898: err = 1;
1.11 markus 899: break;
1.25 deraadt 900: }
1.44 djm 901: if ((aa = do_stat(conn, tmp, 0)) == NULL) {
1.11 markus 902: xfree(tmp);
1.25 deraadt 903: err = 1;
1.11 markus 904: break;
905: }
1.9 djm 906: if (!(aa->flags & SSH2_FILEXFER_ATTR_PERMISSIONS)) {
907: error("Can't change directory: Can't check target");
908: xfree(tmp);
1.25 deraadt 909: err = 1;
1.9 djm 910: break;
911: }
912: if (!S_ISDIR(aa->perm)) {
913: error("Can't change directory: \"%s\" is not "
914: "a directory", tmp);
915: xfree(tmp);
1.25 deraadt 916: err = 1;
1.9 djm 917: break;
918: }
1.11 markus 919: xfree(*pwd);
920: *pwd = tmp;
1.1 djm 921: break;
922: case I_LS:
1.12 djm 923: if (!path1) {
1.48 djm 924: do_globbed_ls(conn, *pwd, *pwd, lflag);
1.12 djm 925: break;
926: }
1.50 deraadt 927:
1.48 djm 928: /* Strip pwd off beginning of non-absolute paths */
929: tmp = NULL;
930: if (*path1 != '/')
931: tmp = *pwd;
932:
1.1 djm 933: path1 = make_absolute(path1, *pwd);
1.51 djm 934: err = do_globbed_ls(conn, path1, tmp, lflag);
1.1 djm 935: break;
936: case I_LCHDIR:
1.25 deraadt 937: if (chdir(path1) == -1) {
1.1 djm 938: error("Couldn't change local directory to "
939: "\"%s\": %s", path1, strerror(errno));
1.25 deraadt 940: err = 1;
941: }
1.1 djm 942: break;
943: case I_LMKDIR:
1.25 deraadt 944: if (mkdir(path1, 0777) == -1) {
1.17 stevesk 945: error("Couldn't create local directory "
1.1 djm 946: "\"%s\": %s", path1, strerror(errno));
1.25 deraadt 947: err = 1;
948: }
1.1 djm 949: break;
950: case I_LLS:
951: local_do_ls(cmd);
952: break;
953: case I_SHELL:
954: local_do_shell(cmd);
955: break;
956: case I_LUMASK:
957: umask(n_arg);
1.21 stevesk 958: printf("Local umask: %03lo\n", n_arg);
1.1 djm 959: break;
960: case I_CHMOD:
961: path1 = make_absolute(path1, *pwd);
962: attrib_clear(&a);
963: a.flags |= SSH2_FILEXFER_ATTR_PERMISSIONS;
964: a.perm = n_arg;
1.44 djm 965: remote_glob(conn, path1, GLOB_NOCHECK, NULL, &g);
1.41 deraadt 966: for (i = 0; g.gl_pathv[i]; i++) {
1.27 djm 967: printf("Changing mode on %s\n", g.gl_pathv[i]);
1.51 djm 968: err = do_setstat(conn, g.gl_pathv[i], &a);
969: if (err != 0 && err_abort)
970: break;
1.27 djm 971: }
1.5 stevesk 972: break;
1.1 djm 973: case I_CHOWN:
1.51 djm 974: case I_CHGRP:
1.1 djm 975: path1 = make_absolute(path1, *pwd);
1.44 djm 976: remote_glob(conn, path1, GLOB_NOCHECK, NULL, &g);
1.41 deraadt 977: for (i = 0; g.gl_pathv[i]; i++) {
1.51 djm 978: if (!(aa = do_stat(conn, g.gl_pathv[i], 0))) {
979: if (err != 0 && err_abort)
980: break;
981: else
982: continue;
983: }
1.27 djm 984: if (!(aa->flags & SSH2_FILEXFER_ATTR_UIDGID)) {
985: error("Can't get current ownership of "
986: "remote file \"%s\"", g.gl_pathv[i]);
1.51 djm 987: if (err != 0 && err_abort)
988: break;
989: else
990: continue;
1.27 djm 991: }
992: aa->flags &= SSH2_FILEXFER_ATTR_UIDGID;
1.51 djm 993: if (cmdnum == I_CHOWN) {
994: printf("Changing owner on %s\n", g.gl_pathv[i]);
995: aa->uid = n_arg;
996: } else {
997: printf("Changing group on %s\n", g.gl_pathv[i]);
998: aa->gid = n_arg;
1.27 djm 999: }
1.51 djm 1000: err = do_setstat(conn, g.gl_pathv[i], aa);
1001: if (err != 0 && err_abort)
1002: break;
1.1 djm 1003: }
1004: break;
1005: case I_PWD:
1006: printf("Remote working directory: %s\n", *pwd);
1007: break;
1008: case I_LPWD:
1.51 djm 1009: if (!getcwd(path_buf, sizeof(path_buf))) {
1010: error("Couldn't get local cwd: %s", strerror(errno));
1011: err = -1;
1012: break;
1013: }
1014: printf("Local working directory: %s\n", path_buf);
1.1 djm 1015: break;
1016: case I_QUIT:
1.51 djm 1017: /* Processed below */
1018: break;
1.1 djm 1019: case I_HELP:
1020: help();
1.28 markus 1021: break;
1022: case I_VERSION:
1.47 deraadt 1023: printf("SFTP protocol version %u\n", sftp_proto_version(conn));
1.52 ! fgsch 1024: break;
! 1025: case I_PROGRESS:
! 1026: showprogress = !showprogress;
! 1027: if (showprogress)
! 1028: printf("Progress meter enabled\n");
! 1029: else
! 1030: printf("Progress meter disabled\n");
1.1 djm 1031: break;
1032: default:
1033: fatal("%d is not implemented", cmdnum);
1034: }
1035:
1.29 djm 1036: if (g.gl_pathc)
1037: globfree(&g);
1.1 djm 1038: if (path1)
1039: xfree(path1);
1040: if (path2)
1041: xfree(path2);
1.25 deraadt 1042:
1.51 djm 1043: /* If an unignored error occurs in batch mode we should abort. */
1044: if (err_abort && err != 0)
1045: return (-1);
1046: else if (cmdnum == I_QUIT)
1047: return (1);
1.25 deraadt 1048:
1.51 djm 1049: return (0);
1.1 djm 1050: }
1051:
1.51 djm 1052: int
1.35 mouring 1053: interactive_loop(int fd_in, int fd_out, char *file1, char *file2)
1.1 djm 1054: {
1055: char *pwd;
1.35 mouring 1056: char *dir = NULL;
1.1 djm 1057: char cmd[2048];
1.44 djm 1058: struct sftp_conn *conn;
1.51 djm 1059: int err;
1.26 djm 1060:
1.44 djm 1061: conn = do_init(fd_in, fd_out, copy_buffer_len, num_requests);
1062: if (conn == NULL)
1.26 djm 1063: fatal("Couldn't initialise connection to server");
1.1 djm 1064:
1.44 djm 1065: pwd = do_realpath(conn, ".");
1.1 djm 1066: if (pwd == NULL)
1067: fatal("Need cwd");
1068:
1.35 mouring 1069: if (file1 != NULL) {
1070: dir = xstrdup(file1);
1071: dir = make_absolute(dir, pwd);
1072:
1.44 djm 1073: if (remote_is_dir(conn, dir) && file2 == NULL) {
1.35 mouring 1074: printf("Changing to: %s\n", dir);
1075: snprintf(cmd, sizeof cmd, "cd \"%s\"", dir);
1.51 djm 1076: if (parse_dispatch_command(conn, cmd, &pwd, 1) != 0)
1077: return (-1);
1.35 mouring 1078: } else {
1079: if (file2 == NULL)
1080: snprintf(cmd, sizeof cmd, "get %s", dir);
1081: else
1082: snprintf(cmd, sizeof cmd, "get %s %s", dir,
1083: file2);
1084:
1.51 djm 1085: err = parse_dispatch_command(conn, cmd, &pwd, 1);
1.45 mpech 1086: xfree(dir);
1.51 djm 1087: return (err);
1.35 mouring 1088: }
1.45 mpech 1089: xfree(dir);
1.35 mouring 1090: }
1.51 djm 1091:
1.14 stevesk 1092: setvbuf(stdout, NULL, _IOLBF, 0);
1.25 deraadt 1093: setvbuf(infile, NULL, _IOLBF, 0);
1.1 djm 1094:
1.51 djm 1095: err = 0;
1.41 deraadt 1096: for (;;) {
1.1 djm 1097: char *cp;
1098:
1099: printf("sftp> ");
1100:
1101: /* XXX: use libedit */
1.25 deraadt 1102: if (fgets(cmd, sizeof(cmd), infile) == NULL) {
1.1 djm 1103: printf("\n");
1104: break;
1.25 deraadt 1105: } else if (infile != stdin) /* Bluff typing */
1106: printf("%s", cmd);
1107:
1.1 djm 1108: cp = strrchr(cmd, '\n');
1109: if (cp)
1110: *cp = '\0';
1.25 deraadt 1111:
1.51 djm 1112: err = parse_dispatch_command(conn, cmd, &pwd, infile != stdin);
1113: if (err != 0)
1.1 djm 1114: break;
1115: }
1116: xfree(pwd);
1.51 djm 1117:
1118: /* err == 1 signifies normal "quit" exit */
1119: return (err >= 0 ? 0 : -1);
1.1 djm 1120: }
1.51 djm 1121: