Annotation of src/usr.bin/ssh/sftp-int.c, Revision 1.58
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.58 ! mouring 28: RCSID("$OpenBSD: sftp-int.c,v 1.57 2003/03/05 22:33:43 markus 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:
1.54 djm 88: static 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.53 fgsch 142: printf("progress 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.55 djm 386: is_reg(char *path)
387: {
388: struct stat sb;
389:
390: if (stat(path, &sb) == -1)
391: fatal("stat %s: %s", path, strerror(errno));
392:
393: return(S_ISREG(sb.st_mode));
394: }
395:
396: static int
1.44 djm 397: remote_is_dir(struct sftp_conn *conn, char *path)
1.1 djm 398: {
1.29 djm 399: Attrib *a;
1.1 djm 400:
1.29 djm 401: /* XXX: report errors? */
1.44 djm 402: if ((a = do_stat(conn, path, 1)) == NULL)
1.29 djm 403: return(0);
404: if (!(a->flags & SSH2_FILEXFER_ATTR_PERMISSIONS))
1.1 djm 405: return(0);
1.29 djm 406: return(a->perm & S_IFDIR);
407: }
408:
1.37 itojun 409: static int
1.44 djm 410: process_get(struct sftp_conn *conn, char *src, char *dst, char *pwd, int pflag)
1.29 djm 411: {
412: char *abs_src = NULL;
413: char *abs_dst = NULL;
414: char *tmp;
415: glob_t g;
416: int err = 0;
417: int i;
1.30 markus 418:
1.29 djm 419: abs_src = xstrdup(src);
420: abs_src = make_absolute(abs_src, pwd);
421:
1.30 markus 422: memset(&g, 0, sizeof(g));
1.29 djm 423: debug3("Looking up %s", abs_src);
1.44 djm 424: if (remote_glob(conn, abs_src, 0, NULL, &g)) {
1.29 djm 425: error("File \"%s\" not found.", abs_src);
426: err = -1;
427: goto out;
428: }
429:
430: /* Only one match, dst may be file, directory or unspecified */
431: if (g.gl_pathv[0] && g.gl_matchc == 1) {
432: if (dst) {
433: /* If directory specified, append filename */
434: if (is_dir(dst)) {
435: if (infer_path(g.gl_pathv[0], &tmp)) {
436: err = 1;
437: goto out;
438: }
439: abs_dst = path_append(dst, tmp);
440: xfree(tmp);
441: } else
442: abs_dst = xstrdup(dst);
443: } else if (infer_path(g.gl_pathv[0], &abs_dst)) {
444: err = -1;
445: goto out;
446: }
1.44 djm 447: err = do_download(conn, g.gl_pathv[0], abs_dst, pflag);
1.29 djm 448: goto out;
449: }
450:
451: /* Multiple matches, dst may be directory or unspecified */
452: if (dst && !is_dir(dst)) {
1.30 markus 453: error("Multiple files match, but \"%s\" is not a directory",
1.29 djm 454: dst);
455: err = -1;
456: goto out;
457: }
1.30 markus 458:
1.41 deraadt 459: for (i = 0; g.gl_pathv[i]; i++) {
1.29 djm 460: if (infer_path(g.gl_pathv[i], &tmp)) {
461: err = -1;
462: goto out;
463: }
464: if (dst) {
465: abs_dst = path_append(dst, tmp);
466: xfree(tmp);
467: } else
468: abs_dst = tmp;
469:
470: printf("Fetching %s to %s\n", g.gl_pathv[i], abs_dst);
1.44 djm 471: if (do_download(conn, g.gl_pathv[i], abs_dst, pflag) == -1)
1.29 djm 472: err = -1;
473: xfree(abs_dst);
474: abs_dst = NULL;
1.1 djm 475: }
476:
1.29 djm 477: out:
478: xfree(abs_src);
479: if (abs_dst)
480: xfree(abs_dst);
481: globfree(&g);
482: return(err);
483: }
484:
1.37 itojun 485: static int
1.44 djm 486: process_put(struct sftp_conn *conn, char *src, char *dst, char *pwd, int pflag)
1.29 djm 487: {
488: char *tmp_dst = NULL;
489: char *abs_dst = NULL;
490: char *tmp;
491: glob_t g;
492: int err = 0;
493: int i;
494:
495: if (dst) {
496: tmp_dst = xstrdup(dst);
497: tmp_dst = make_absolute(tmp_dst, pwd);
498: }
499:
1.30 markus 500: memset(&g, 0, sizeof(g));
1.29 djm 501: debug3("Looking up %s", src);
502: if (glob(src, 0, NULL, &g)) {
503: error("File \"%s\" not found.", src);
504: err = -1;
505: goto out;
506: }
507:
508: /* Only one match, dst may be file, directory or unspecified */
509: if (g.gl_pathv[0] && g.gl_matchc == 1) {
1.56 djm 510: if (!is_reg(g.gl_pathv[0])) {
1.55 djm 511: error("Can't upload %s: not a regular file",
512: g.gl_pathv[0]);
513: err = 1;
514: goto out;
515: }
1.29 djm 516: if (tmp_dst) {
517: /* If directory specified, append filename */
1.44 djm 518: if (remote_is_dir(conn, tmp_dst)) {
1.29 djm 519: if (infer_path(g.gl_pathv[0], &tmp)) {
520: err = 1;
521: goto out;
522: }
523: abs_dst = path_append(tmp_dst, tmp);
524: xfree(tmp);
525: } else
526: abs_dst = xstrdup(tmp_dst);
1.32 markus 527: } else {
528: if (infer_path(g.gl_pathv[0], &abs_dst)) {
529: err = -1;
530: goto out;
531: }
532: abs_dst = make_absolute(abs_dst, pwd);
1.29 djm 533: }
1.44 djm 534: err = do_upload(conn, g.gl_pathv[0], abs_dst, pflag);
1.29 djm 535: goto out;
536: }
537:
538: /* Multiple matches, dst may be directory or unspecified */
1.44 djm 539: if (tmp_dst && !remote_is_dir(conn, tmp_dst)) {
1.30 markus 540: error("Multiple files match, but \"%s\" is not a directory",
1.29 djm 541: tmp_dst);
542: err = -1;
543: goto out;
544: }
545:
1.41 deraadt 546: for (i = 0; g.gl_pathv[i]; i++) {
1.55 djm 547: if (!is_reg(g.gl_pathv[i])) {
548: error("skipping non-regular file %s",
549: g.gl_pathv[i]);
550: continue;
551: }
1.29 djm 552: if (infer_path(g.gl_pathv[i], &tmp)) {
553: err = -1;
554: goto out;
555: }
556: if (tmp_dst) {
557: abs_dst = path_append(tmp_dst, tmp);
558: xfree(tmp);
559: } else
560: abs_dst = make_absolute(tmp, pwd);
561:
562: printf("Uploading %s to %s\n", g.gl_pathv[i], abs_dst);
1.44 djm 563: if (do_upload(conn, g.gl_pathv[i], abs_dst, pflag) == -1)
1.29 djm 564: err = -1;
1.1 djm 565: }
566:
1.29 djm 567: out:
568: if (abs_dst)
569: xfree(abs_dst);
570: if (tmp_dst)
571: xfree(tmp_dst);
1.58 ! mouring 572: globfree(&g);
1.29 djm 573: return(err);
1.1 djm 574: }
575:
1.37 itojun 576: static int
1.48 djm 577: sdirent_comp(const void *aa, const void *bb)
578: {
579: SFTP_DIRENT *a = *(SFTP_DIRENT **)aa;
580: SFTP_DIRENT *b = *(SFTP_DIRENT **)bb;
581:
1.50 deraadt 582: return (strcmp(a->filename, b->filename));
1.48 djm 583: }
584:
585: /* sftp ls.1 replacement for directories */
586: static int
587: do_ls_dir(struct sftp_conn *conn, char *path, char *strip_path, int lflag)
588: {
589: int n;
590: SFTP_DIRENT **d;
591:
592: if ((n = do_readdir(conn, path, &d)) != 0)
593: return (n);
594:
1.50 deraadt 595: /* Count entries for sort */
1.48 djm 596: for (n = 0; d[n] != NULL; n++)
597: ;
598:
599: qsort(d, n, sizeof(*d), sdirent_comp);
600:
601: for (n = 0; d[n] != NULL; n++) {
602: char *tmp, *fname;
1.50 deraadt 603:
1.48 djm 604: tmp = path_append(path, d[n]->filename);
605: fname = path_strip(tmp, strip_path);
606: xfree(tmp);
607:
608: if (lflag) {
609: char *lname;
610: struct stat sb;
611:
612: memset(&sb, 0, sizeof(sb));
613: attrib_to_stat(&d[n]->a, &sb);
614: lname = ls_file(fname, &sb, 1);
615: printf("%s\n", lname);
616: xfree(lname);
617: } else {
618: /* XXX - multicolumn display would be nice here */
619: printf("%s\n", fname);
620: }
1.50 deraadt 621:
1.48 djm 622: xfree(fname);
623: }
624:
625: free_sftp_dirents(d);
626: return (0);
627: }
628:
629: /* sftp ls.1 replacement which handles path globs */
630: static int
1.50 deraadt 631: do_globbed_ls(struct sftp_conn *conn, char *path, char *strip_path,
1.48 djm 632: int lflag)
633: {
634: glob_t g;
635: int i;
636: Attrib *a;
637: struct stat sb;
638:
639: memset(&g, 0, sizeof(g));
640:
1.50 deraadt 641: if (remote_glob(conn, path, GLOB_MARK|GLOB_NOCHECK|GLOB_BRACE,
1.48 djm 642: NULL, &g)) {
643: error("Can't ls: \"%s\" not found", path);
644: return (-1);
645: }
646:
647: /*
1.50 deraadt 648: * If the glob returns a single match, which is the same as the
1.48 djm 649: * input glob, and it is a directory, then just list its contents
650: */
1.50 deraadt 651: if (g.gl_pathc == 1 &&
1.48 djm 652: strncmp(path, g.gl_pathv[0], strlen(g.gl_pathv[0]) - 1) == 0) {
653: if ((a = do_lstat(conn, path, 1)) == NULL) {
654: globfree(&g);
655: return (-1);
656: }
1.50 deraadt 657: if ((a->flags & SSH2_FILEXFER_ATTR_PERMISSIONS) &&
1.48 djm 658: S_ISDIR(a->perm)) {
659: globfree(&g);
660: return (do_ls_dir(conn, path, strip_path, lflag));
661: }
662: }
663:
664: for (i = 0; g.gl_pathv[i]; i++) {
665: char *fname, *lname;
666:
667: fname = path_strip(g.gl_pathv[i], strip_path);
668:
669: if (lflag) {
670: /*
671: * XXX: this is slow - 1 roundtrip per path
1.50 deraadt 672: * A solution to this is to fork glob() and
673: * build a sftp specific version which keeps the
1.48 djm 674: * attribs (which currently get thrown away)
675: * that the server returns as well as the filenames.
676: */
677: memset(&sb, 0, sizeof(sb));
678: a = do_lstat(conn, g.gl_pathv[i], 1);
679: if (a != NULL)
680: attrib_to_stat(a, &sb);
681: lname = ls_file(fname, &sb, 1);
682: printf("%s\n", lname);
683: xfree(lname);
684: } else {
685: /* XXX - multicolumn display would be nice here */
686: printf("%s\n", fname);
687: }
688: xfree(fname);
689: }
690:
691: if (g.gl_pathc)
692: globfree(&g);
693:
694: return (0);
695: }
696:
697: static int
1.51 djm 698: parse_args(const char **cpp, int *pflag, int *lflag, int *iflag,
1.48 djm 699: unsigned long *n_arg, char **path1, char **path2)
1.1 djm 700: {
701: const char *cmd, *cp = *cpp;
1.21 stevesk 702: char *cp2;
1.4 markus 703: int base = 0;
1.21 stevesk 704: long l;
1.1 djm 705: int i, cmdnum;
706:
707: /* Skip leading whitespace */
708: cp = cp + strspn(cp, WHITESPACE);
709:
1.51 djm 710: /* Ignore blank lines and lines which begin with comment '#' char */
711: if (*cp == '\0' || *cp == '#')
712: return (0);
1.1 djm 713:
1.51 djm 714: /* Check for leading '-' (disable error processing) */
715: *iflag = 0;
716: if (*cp == '-') {
717: *iflag = 1;
718: cp++;
719: }
720:
1.1 djm 721: /* Figure out which command we have */
1.41 deraadt 722: for (i = 0; cmds[i].c; i++) {
1.1 djm 723: int cmdlen = strlen(cmds[i].c);
724:
725: /* Check for command followed by whitespace */
726: if (!strncasecmp(cp, cmds[i].c, cmdlen) &&
727: strchr(WHITESPACE, cp[cmdlen])) {
728: cp += cmdlen;
729: cp = cp + strspn(cp, WHITESPACE);
730: break;
731: }
732: }
733: cmdnum = cmds[i].n;
734: cmd = cmds[i].c;
735:
736: /* Special case */
737: if (*cp == '!') {
738: cp++;
739: cmdnum = I_SHELL;
740: } else if (cmdnum == -1) {
741: error("Invalid command.");
1.51 djm 742: return (-1);
1.1 djm 743: }
744:
745: /* Get arguments and parse flags */
1.48 djm 746: *lflag = *pflag = *n_arg = 0;
1.1 djm 747: *path1 = *path2 = NULL;
748: switch (cmdnum) {
749: case I_GET:
750: case I_PUT:
751: if (parse_getput_flags(&cp, pflag))
752: return(-1);
753: /* Get first pathname (mandatory) */
754: if (get_pathname(&cp, path1))
755: return(-1);
756: if (*path1 == NULL) {
757: error("You must specify at least one path after a "
758: "%s command.", cmd);
759: return(-1);
760: }
761: /* Try to get second pathname (optional) */
762: if (get_pathname(&cp, path2))
763: return(-1);
764: break;
765: case I_RENAME:
1.26 djm 766: case I_SYMLINK:
1.1 djm 767: if (get_pathname(&cp, path1))
768: return(-1);
769: if (get_pathname(&cp, path2))
770: return(-1);
771: if (!*path1 || !*path2) {
772: error("You must specify two paths after a %s "
773: "command.", cmd);
774: return(-1);
775: }
776: break;
777: case I_RM:
778: case I_MKDIR:
779: case I_RMDIR:
780: case I_CHDIR:
781: case I_LCHDIR:
782: case I_LMKDIR:
783: /* Get pathname (mandatory) */
784: if (get_pathname(&cp, path1))
785: return(-1);
786: if (*path1 == NULL) {
1.3 stevesk 787: error("You must specify a path after a %s command.",
1.1 djm 788: cmd);
789: return(-1);
790: }
791: break;
792: case I_LS:
1.48 djm 793: if (parse_ls_flags(&cp, lflag))
794: return(-1);
1.1 djm 795: /* Path is optional */
796: if (get_pathname(&cp, path1))
797: return(-1);
798: break;
799: case I_LLS:
800: case I_SHELL:
801: /* Uses the rest of the line */
802: break;
803: case I_LUMASK:
1.21 stevesk 804: base = 8;
1.1 djm 805: case I_CHMOD:
1.4 markus 806: base = 8;
1.1 djm 807: case I_CHOWN:
808: case I_CHGRP:
809: /* Get numeric arg (mandatory) */
1.21 stevesk 810: l = strtol(cp, &cp2, base);
811: if (cp2 == cp || ((l == LONG_MIN || l == LONG_MAX) &&
812: errno == ERANGE) || l < 0) {
1.1 djm 813: error("You must supply a numeric argument "
814: "to the %s command.", cmd);
815: return(-1);
816: }
1.21 stevesk 817: cp = cp2;
818: *n_arg = l;
819: if (cmdnum == I_LUMASK && strchr(WHITESPACE, *cp))
820: break;
821: if (cmdnum == I_LUMASK || !strchr(WHITESPACE, *cp)) {
1.1 djm 822: error("You must supply a numeric argument "
823: "to the %s command.", cmd);
824: return(-1);
825: }
826: cp += strspn(cp, WHITESPACE);
827:
828: /* Get pathname (mandatory) */
829: if (get_pathname(&cp, path1))
830: return(-1);
831: if (*path1 == NULL) {
1.3 stevesk 832: error("You must specify a path after a %s command.",
1.1 djm 833: cmd);
834: return(-1);
835: }
836: break;
837: case I_QUIT:
838: case I_PWD:
839: case I_LPWD:
840: case I_HELP:
1.28 markus 841: case I_VERSION:
1.52 fgsch 842: case I_PROGRESS:
1.1 djm 843: break;
844: default:
845: fatal("Command not implemented");
846: }
847:
848: *cpp = cp;
849: return(cmdnum);
850: }
851:
1.37 itojun 852: static int
1.51 djm 853: parse_dispatch_command(struct sftp_conn *conn, const char *cmd, char **pwd,
854: int err_abort)
1.1 djm 855: {
1.8 provos 856: char *path1, *path2, *tmp;
1.51 djm 857: int pflag, lflag, iflag, cmdnum, i;
1.1 djm 858: unsigned long n_arg;
859: Attrib a, *aa;
1.23 deraadt 860: char path_buf[MAXPATHLEN];
1.25 deraadt 861: int err = 0;
1.27 djm 862: glob_t g;
1.1 djm 863:
864: path1 = path2 = NULL;
1.51 djm 865: cmdnum = parse_args(&cmd, &pflag, &lflag, &iflag, &n_arg,
1.48 djm 866: &path1, &path2);
1.1 djm 867:
1.51 djm 868: if (iflag != 0)
869: err_abort = 0;
870:
1.29 djm 871: memset(&g, 0, sizeof(g));
872:
1.1 djm 873: /* Perform command */
874: switch (cmdnum) {
1.51 djm 875: case 0:
876: /* Blank line */
877: break;
1.1 djm 878: case -1:
1.51 djm 879: /* Unrecognized command */
880: err = -1;
1.1 djm 881: break;
882: case I_GET:
1.44 djm 883: err = process_get(conn, path1, path2, *pwd, pflag);
1.1 djm 884: break;
885: case I_PUT:
1.44 djm 886: err = process_put(conn, path1, path2, *pwd, pflag);
1.33 markus 887: break;
888: case I_RENAME:
1.1 djm 889: path1 = make_absolute(path1, *pwd);
890: path2 = make_absolute(path2, *pwd);
1.44 djm 891: err = do_rename(conn, path1, path2);
1.1 djm 892: break;
1.26 djm 893: case I_SYMLINK:
1.44 djm 894: path2 = make_absolute(path2, *pwd);
895: err = do_symlink(conn, path1, path2);
1.26 djm 896: break;
1.1 djm 897: case I_RM:
898: path1 = make_absolute(path1, *pwd);
1.44 djm 899: remote_glob(conn, path1, GLOB_NOCHECK, NULL, &g);
1.41 deraadt 900: for (i = 0; g.gl_pathv[i]; i++) {
1.27 djm 901: printf("Removing %s\n", g.gl_pathv[i]);
1.51 djm 902: err = do_rm(conn, g.gl_pathv[i]);
903: if (err != 0 && err_abort)
904: break;
1.27 djm 905: }
1.1 djm 906: break;
907: case I_MKDIR:
908: path1 = make_absolute(path1, *pwd);
909: attrib_clear(&a);
910: a.flags |= SSH2_FILEXFER_ATTR_PERMISSIONS;
911: a.perm = 0777;
1.44 djm 912: err = do_mkdir(conn, path1, &a);
1.1 djm 913: break;
914: case I_RMDIR:
915: path1 = make_absolute(path1, *pwd);
1.44 djm 916: err = do_rmdir(conn, path1);
1.1 djm 917: break;
918: case I_CHDIR:
919: path1 = make_absolute(path1, *pwd);
1.44 djm 920: if ((tmp = do_realpath(conn, path1)) == NULL) {
1.25 deraadt 921: err = 1;
1.11 markus 922: break;
1.25 deraadt 923: }
1.44 djm 924: if ((aa = do_stat(conn, tmp, 0)) == NULL) {
1.11 markus 925: xfree(tmp);
1.25 deraadt 926: err = 1;
1.11 markus 927: break;
928: }
1.9 djm 929: if (!(aa->flags & SSH2_FILEXFER_ATTR_PERMISSIONS)) {
930: error("Can't change directory: Can't check target");
931: xfree(tmp);
1.25 deraadt 932: err = 1;
1.9 djm 933: break;
934: }
935: if (!S_ISDIR(aa->perm)) {
936: error("Can't change directory: \"%s\" is not "
937: "a directory", tmp);
938: xfree(tmp);
1.25 deraadt 939: err = 1;
1.9 djm 940: break;
941: }
1.11 markus 942: xfree(*pwd);
943: *pwd = tmp;
1.1 djm 944: break;
945: case I_LS:
1.12 djm 946: if (!path1) {
1.48 djm 947: do_globbed_ls(conn, *pwd, *pwd, lflag);
1.12 djm 948: break;
949: }
1.50 deraadt 950:
1.48 djm 951: /* Strip pwd off beginning of non-absolute paths */
952: tmp = NULL;
953: if (*path1 != '/')
954: tmp = *pwd;
955:
1.1 djm 956: path1 = make_absolute(path1, *pwd);
1.51 djm 957: err = do_globbed_ls(conn, path1, tmp, lflag);
1.1 djm 958: break;
959: case I_LCHDIR:
1.25 deraadt 960: if (chdir(path1) == -1) {
1.1 djm 961: error("Couldn't change local directory to "
962: "\"%s\": %s", path1, strerror(errno));
1.25 deraadt 963: err = 1;
964: }
1.1 djm 965: break;
966: case I_LMKDIR:
1.25 deraadt 967: if (mkdir(path1, 0777) == -1) {
1.17 stevesk 968: error("Couldn't create local directory "
1.1 djm 969: "\"%s\": %s", path1, strerror(errno));
1.25 deraadt 970: err = 1;
971: }
1.1 djm 972: break;
973: case I_LLS:
974: local_do_ls(cmd);
975: break;
976: case I_SHELL:
977: local_do_shell(cmd);
978: break;
979: case I_LUMASK:
980: umask(n_arg);
1.21 stevesk 981: printf("Local umask: %03lo\n", n_arg);
1.1 djm 982: break;
983: case I_CHMOD:
984: path1 = make_absolute(path1, *pwd);
985: attrib_clear(&a);
986: a.flags |= SSH2_FILEXFER_ATTR_PERMISSIONS;
987: a.perm = n_arg;
1.44 djm 988: remote_glob(conn, path1, GLOB_NOCHECK, NULL, &g);
1.41 deraadt 989: for (i = 0; g.gl_pathv[i]; i++) {
1.27 djm 990: printf("Changing mode on %s\n", g.gl_pathv[i]);
1.51 djm 991: err = do_setstat(conn, g.gl_pathv[i], &a);
992: if (err != 0 && err_abort)
993: break;
1.27 djm 994: }
1.5 stevesk 995: break;
1.1 djm 996: case I_CHOWN:
1.51 djm 997: case I_CHGRP:
1.1 djm 998: path1 = make_absolute(path1, *pwd);
1.44 djm 999: remote_glob(conn, path1, GLOB_NOCHECK, NULL, &g);
1.41 deraadt 1000: for (i = 0; g.gl_pathv[i]; i++) {
1.51 djm 1001: if (!(aa = do_stat(conn, g.gl_pathv[i], 0))) {
1002: if (err != 0 && err_abort)
1003: break;
1004: else
1005: continue;
1006: }
1.27 djm 1007: if (!(aa->flags & SSH2_FILEXFER_ATTR_UIDGID)) {
1008: error("Can't get current ownership of "
1009: "remote file \"%s\"", g.gl_pathv[i]);
1.51 djm 1010: if (err != 0 && err_abort)
1011: break;
1012: else
1013: continue;
1.27 djm 1014: }
1015: aa->flags &= SSH2_FILEXFER_ATTR_UIDGID;
1.51 djm 1016: if (cmdnum == I_CHOWN) {
1017: printf("Changing owner on %s\n", g.gl_pathv[i]);
1018: aa->uid = n_arg;
1019: } else {
1020: printf("Changing group on %s\n", g.gl_pathv[i]);
1021: aa->gid = n_arg;
1.27 djm 1022: }
1.51 djm 1023: err = do_setstat(conn, g.gl_pathv[i], aa);
1024: if (err != 0 && err_abort)
1025: break;
1.1 djm 1026: }
1027: break;
1028: case I_PWD:
1029: printf("Remote working directory: %s\n", *pwd);
1030: break;
1031: case I_LPWD:
1.51 djm 1032: if (!getcwd(path_buf, sizeof(path_buf))) {
1033: error("Couldn't get local cwd: %s", strerror(errno));
1034: err = -1;
1035: break;
1036: }
1037: printf("Local working directory: %s\n", path_buf);
1.1 djm 1038: break;
1039: case I_QUIT:
1.51 djm 1040: /* Processed below */
1041: break;
1.1 djm 1042: case I_HELP:
1043: help();
1.28 markus 1044: break;
1045: case I_VERSION:
1.47 deraadt 1046: printf("SFTP protocol version %u\n", sftp_proto_version(conn));
1.52 fgsch 1047: break;
1048: case I_PROGRESS:
1049: showprogress = !showprogress;
1050: if (showprogress)
1051: printf("Progress meter enabled\n");
1052: else
1053: printf("Progress meter disabled\n");
1.1 djm 1054: break;
1055: default:
1056: fatal("%d is not implemented", cmdnum);
1057: }
1058:
1.29 djm 1059: if (g.gl_pathc)
1060: globfree(&g);
1.1 djm 1061: if (path1)
1062: xfree(path1);
1063: if (path2)
1064: xfree(path2);
1.25 deraadt 1065:
1.51 djm 1066: /* If an unignored error occurs in batch mode we should abort. */
1067: if (err_abort && err != 0)
1068: return (-1);
1069: else if (cmdnum == I_QUIT)
1070: return (1);
1.25 deraadt 1071:
1.51 djm 1072: return (0);
1.1 djm 1073: }
1074:
1.51 djm 1075: int
1.35 mouring 1076: interactive_loop(int fd_in, int fd_out, char *file1, char *file2)
1.1 djm 1077: {
1078: char *pwd;
1.35 mouring 1079: char *dir = NULL;
1.1 djm 1080: char cmd[2048];
1.44 djm 1081: struct sftp_conn *conn;
1.51 djm 1082: int err;
1.26 djm 1083:
1.44 djm 1084: conn = do_init(fd_in, fd_out, copy_buffer_len, num_requests);
1085: if (conn == NULL)
1.26 djm 1086: fatal("Couldn't initialise connection to server");
1.1 djm 1087:
1.44 djm 1088: pwd = do_realpath(conn, ".");
1.1 djm 1089: if (pwd == NULL)
1090: fatal("Need cwd");
1091:
1.35 mouring 1092: if (file1 != NULL) {
1093: dir = xstrdup(file1);
1094: dir = make_absolute(dir, pwd);
1095:
1.44 djm 1096: if (remote_is_dir(conn, dir) && file2 == NULL) {
1.35 mouring 1097: printf("Changing to: %s\n", dir);
1098: snprintf(cmd, sizeof cmd, "cd \"%s\"", dir);
1.51 djm 1099: if (parse_dispatch_command(conn, cmd, &pwd, 1) != 0)
1100: return (-1);
1.35 mouring 1101: } else {
1102: if (file2 == NULL)
1103: snprintf(cmd, sizeof cmd, "get %s", dir);
1104: else
1105: snprintf(cmd, sizeof cmd, "get %s %s", dir,
1106: file2);
1107:
1.51 djm 1108: err = parse_dispatch_command(conn, cmd, &pwd, 1);
1.45 mpech 1109: xfree(dir);
1.57 markus 1110: xfree(pwd);
1.51 djm 1111: return (err);
1.35 mouring 1112: }
1.45 mpech 1113: xfree(dir);
1.35 mouring 1114: }
1.51 djm 1115:
1.14 stevesk 1116: setvbuf(stdout, NULL, _IOLBF, 0);
1.25 deraadt 1117: setvbuf(infile, NULL, _IOLBF, 0);
1.1 djm 1118:
1.51 djm 1119: err = 0;
1.41 deraadt 1120: for (;;) {
1.1 djm 1121: char *cp;
1122:
1123: printf("sftp> ");
1124:
1125: /* XXX: use libedit */
1.25 deraadt 1126: if (fgets(cmd, sizeof(cmd), infile) == NULL) {
1.1 djm 1127: printf("\n");
1128: break;
1.25 deraadt 1129: } else if (infile != stdin) /* Bluff typing */
1130: printf("%s", cmd);
1131:
1.1 djm 1132: cp = strrchr(cmd, '\n');
1133: if (cp)
1134: *cp = '\0';
1.25 deraadt 1135:
1.51 djm 1136: err = parse_dispatch_command(conn, cmd, &pwd, infile != stdin);
1137: if (err != 0)
1.1 djm 1138: break;
1139: }
1140: xfree(pwd);
1.51 djm 1141:
1142: /* err == 1 signifies normal "quit" exit */
1143: return (err >= 0 ? 0 : -1);
1.1 djm 1144: }
1.51 djm 1145: