Annotation of src/usr.bin/ssh/sftp-int.c, Revision 1.49.2.1
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.49.2.1! margarid 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.49.2.1! margarid 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.49.2.1! margarid 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.49.2.1! margarid 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.49.2.1! margarid 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.49.2.1! margarid 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.49.2.1! margarid 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.49.2.1! margarid 510: if (!is_reg(g.gl_pathv[0])) {
! 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.49.2.1! margarid 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);
572: return(err);
1.1 djm 573: }
574:
1.37 itojun 575: static int
1.48 djm 576: sdirent_comp(const void *aa, const void *bb)
577: {
578: SFTP_DIRENT *a = *(SFTP_DIRENT **)aa;
579: SFTP_DIRENT *b = *(SFTP_DIRENT **)bb;
580:
1.49.2.1! margarid 581: return (strcmp(a->filename, b->filename));
1.48 djm 582: }
583:
584: /* sftp ls.1 replacement for directories */
585: static int
586: do_ls_dir(struct sftp_conn *conn, char *path, char *strip_path, int lflag)
587: {
588: int n;
589: SFTP_DIRENT **d;
590:
591: if ((n = do_readdir(conn, path, &d)) != 0)
592: return (n);
593:
1.49.2.1! margarid 594: /* Count entries for sort */
1.48 djm 595: for (n = 0; d[n] != NULL; n++)
596: ;
597:
598: qsort(d, n, sizeof(*d), sdirent_comp);
599:
600: for (n = 0; d[n] != NULL; n++) {
601: char *tmp, *fname;
1.49.2.1! margarid 602:
1.48 djm 603: tmp = path_append(path, d[n]->filename);
604: fname = path_strip(tmp, strip_path);
605: xfree(tmp);
606:
607: if (lflag) {
608: char *lname;
609: struct stat sb;
610:
611: memset(&sb, 0, sizeof(sb));
612: attrib_to_stat(&d[n]->a, &sb);
613: lname = ls_file(fname, &sb, 1);
614: printf("%s\n", lname);
615: xfree(lname);
616: } else {
617: /* XXX - multicolumn display would be nice here */
618: printf("%s\n", fname);
619: }
1.49.2.1! margarid 620:
1.48 djm 621: xfree(fname);
622: }
623:
624: free_sftp_dirents(d);
625: return (0);
626: }
627:
628: /* sftp ls.1 replacement which handles path globs */
629: static int
1.49.2.1! margarid 630: do_globbed_ls(struct sftp_conn *conn, char *path, char *strip_path,
1.48 djm 631: int lflag)
632: {
633: glob_t g;
634: int i;
635: Attrib *a;
636: struct stat sb;
637:
638: memset(&g, 0, sizeof(g));
639:
1.49.2.1! margarid 640: if (remote_glob(conn, path, GLOB_MARK|GLOB_NOCHECK|GLOB_BRACE,
1.48 djm 641: NULL, &g)) {
642: error("Can't ls: \"%s\" not found", path);
643: return (-1);
644: }
645:
646: /*
1.49.2.1! margarid 647: * If the glob returns a single match, which is the same as the
1.48 djm 648: * input glob, and it is a directory, then just list its contents
649: */
1.49.2.1! margarid 650: if (g.gl_pathc == 1 &&
1.48 djm 651: strncmp(path, g.gl_pathv[0], strlen(g.gl_pathv[0]) - 1) == 0) {
652: if ((a = do_lstat(conn, path, 1)) == NULL) {
653: globfree(&g);
654: return (-1);
655: }
1.49.2.1! margarid 656: if ((a->flags & SSH2_FILEXFER_ATTR_PERMISSIONS) &&
1.48 djm 657: S_ISDIR(a->perm)) {
658: globfree(&g);
659: return (do_ls_dir(conn, path, strip_path, lflag));
660: }
661: }
662:
663: for (i = 0; g.gl_pathv[i]; i++) {
664: char *fname, *lname;
665:
666: fname = path_strip(g.gl_pathv[i], strip_path);
667:
668: if (lflag) {
669: /*
670: * XXX: this is slow - 1 roundtrip per path
1.49.2.1! margarid 671: * A solution to this is to fork glob() and
! 672: * build a sftp specific version which keeps the
1.48 djm 673: * attribs (which currently get thrown away)
674: * that the server returns as well as the filenames.
675: */
676: memset(&sb, 0, sizeof(sb));
677: a = do_lstat(conn, g.gl_pathv[i], 1);
678: if (a != NULL)
679: attrib_to_stat(a, &sb);
680: lname = ls_file(fname, &sb, 1);
681: printf("%s\n", lname);
682: xfree(lname);
683: } else {
684: /* XXX - multicolumn display would be nice here */
685: printf("%s\n", fname);
686: }
687: xfree(fname);
688: }
689:
690: if (g.gl_pathc)
691: globfree(&g);
692:
693: return (0);
694: }
695:
696: static int
1.49.2.1! margarid 697: parse_args(const char **cpp, int *pflag, int *lflag, int *iflag,
1.48 djm 698: unsigned long *n_arg, char **path1, char **path2)
1.1 djm 699: {
700: const char *cmd, *cp = *cpp;
1.21 stevesk 701: char *cp2;
1.4 markus 702: int base = 0;
1.21 stevesk 703: long l;
1.1 djm 704: int i, cmdnum;
705:
706: /* Skip leading whitespace */
707: cp = cp + strspn(cp, WHITESPACE);
708:
1.49.2.1! margarid 709: /* Ignore blank lines and lines which begin with comment '#' char */
! 710: if (*cp == '\0' || *cp == '#')
! 711: return (0);
1.1 djm 712:
1.49.2.1! margarid 713: /* Check for leading '-' (disable error processing) */
! 714: *iflag = 0;
! 715: if (*cp == '-') {
! 716: *iflag = 1;
! 717: cp++;
! 718: }
! 719:
1.1 djm 720: /* Figure out which command we have */
1.41 deraadt 721: for (i = 0; cmds[i].c; i++) {
1.1 djm 722: int cmdlen = strlen(cmds[i].c);
723:
724: /* Check for command followed by whitespace */
725: if (!strncasecmp(cp, cmds[i].c, cmdlen) &&
726: strchr(WHITESPACE, cp[cmdlen])) {
727: cp += cmdlen;
728: cp = cp + strspn(cp, WHITESPACE);
729: break;
730: }
731: }
732: cmdnum = cmds[i].n;
733: cmd = cmds[i].c;
734:
735: /* Special case */
736: if (*cp == '!') {
737: cp++;
738: cmdnum = I_SHELL;
739: } else if (cmdnum == -1) {
740: error("Invalid command.");
1.49.2.1! margarid 741: return (-1);
1.1 djm 742: }
743:
744: /* Get arguments and parse flags */
1.48 djm 745: *lflag = *pflag = *n_arg = 0;
1.1 djm 746: *path1 = *path2 = NULL;
747: switch (cmdnum) {
748: case I_GET:
749: case I_PUT:
750: if (parse_getput_flags(&cp, pflag))
751: return(-1);
752: /* Get first pathname (mandatory) */
753: if (get_pathname(&cp, path1))
754: return(-1);
755: if (*path1 == NULL) {
756: error("You must specify at least one path after a "
757: "%s command.", cmd);
758: return(-1);
759: }
760: /* Try to get second pathname (optional) */
761: if (get_pathname(&cp, path2))
762: return(-1);
763: break;
764: case I_RENAME:
1.26 djm 765: case I_SYMLINK:
1.1 djm 766: if (get_pathname(&cp, path1))
767: return(-1);
768: if (get_pathname(&cp, path2))
769: return(-1);
770: if (!*path1 || !*path2) {
771: error("You must specify two paths after a %s "
772: "command.", cmd);
773: return(-1);
774: }
775: break;
776: case I_RM:
777: case I_MKDIR:
778: case I_RMDIR:
779: case I_CHDIR:
780: case I_LCHDIR:
781: case I_LMKDIR:
782: /* Get pathname (mandatory) */
783: if (get_pathname(&cp, path1))
784: return(-1);
785: if (*path1 == NULL) {
1.3 stevesk 786: error("You must specify a path after a %s command.",
1.1 djm 787: cmd);
788: return(-1);
789: }
790: break;
791: case I_LS:
1.48 djm 792: if (parse_ls_flags(&cp, lflag))
793: return(-1);
1.1 djm 794: /* Path is optional */
795: if (get_pathname(&cp, path1))
796: return(-1);
797: break;
798: case I_LLS:
799: case I_SHELL:
800: /* Uses the rest of the line */
801: break;
802: case I_LUMASK:
1.21 stevesk 803: base = 8;
1.1 djm 804: case I_CHMOD:
1.4 markus 805: base = 8;
1.1 djm 806: case I_CHOWN:
807: case I_CHGRP:
808: /* Get numeric arg (mandatory) */
1.21 stevesk 809: l = strtol(cp, &cp2, base);
810: if (cp2 == cp || ((l == LONG_MIN || l == LONG_MAX) &&
811: errno == ERANGE) || l < 0) {
1.1 djm 812: error("You must supply a numeric argument "
813: "to the %s command.", cmd);
814: return(-1);
815: }
1.21 stevesk 816: cp = cp2;
817: *n_arg = l;
818: if (cmdnum == I_LUMASK && strchr(WHITESPACE, *cp))
819: break;
820: if (cmdnum == I_LUMASK || !strchr(WHITESPACE, *cp)) {
1.1 djm 821: error("You must supply a numeric argument "
822: "to the %s command.", cmd);
823: return(-1);
824: }
825: cp += strspn(cp, WHITESPACE);
826:
827: /* Get pathname (mandatory) */
828: if (get_pathname(&cp, path1))
829: return(-1);
830: if (*path1 == NULL) {
1.3 stevesk 831: error("You must specify a path after a %s command.",
1.1 djm 832: cmd);
833: return(-1);
834: }
835: break;
836: case I_QUIT:
837: case I_PWD:
838: case I_LPWD:
839: case I_HELP:
1.28 markus 840: case I_VERSION:
1.49.2.1! margarid 841: case I_PROGRESS:
1.1 djm 842: break;
843: default:
844: fatal("Command not implemented");
845: }
846:
847: *cpp = cp;
848: return(cmdnum);
849: }
850:
1.37 itojun 851: static int
1.49.2.1! margarid 852: parse_dispatch_command(struct sftp_conn *conn, const char *cmd, char **pwd,
! 853: int err_abort)
1.1 djm 854: {
1.8 provos 855: char *path1, *path2, *tmp;
1.49.2.1! margarid 856: int pflag, lflag, iflag, cmdnum, i;
1.1 djm 857: unsigned long n_arg;
858: Attrib a, *aa;
1.23 deraadt 859: char path_buf[MAXPATHLEN];
1.25 deraadt 860: int err = 0;
1.27 djm 861: glob_t g;
1.1 djm 862:
863: path1 = path2 = NULL;
1.49.2.1! margarid 864: cmdnum = parse_args(&cmd, &pflag, &lflag, &iflag, &n_arg,
1.48 djm 865: &path1, &path2);
1.1 djm 866:
1.49.2.1! margarid 867: if (iflag != 0)
! 868: err_abort = 0;
! 869:
1.29 djm 870: memset(&g, 0, sizeof(g));
871:
1.1 djm 872: /* Perform command */
873: switch (cmdnum) {
1.49.2.1! margarid 874: case 0:
! 875: /* Blank line */
! 876: break;
1.1 djm 877: case -1:
1.49.2.1! margarid 878: /* Unrecognized command */
! 879: err = -1;
1.1 djm 880: break;
881: case I_GET:
1.44 djm 882: err = process_get(conn, path1, path2, *pwd, pflag);
1.1 djm 883: break;
884: case I_PUT:
1.44 djm 885: err = process_put(conn, path1, path2, *pwd, pflag);
1.33 markus 886: break;
887: case I_RENAME:
1.1 djm 888: path1 = make_absolute(path1, *pwd);
889: path2 = make_absolute(path2, *pwd);
1.44 djm 890: err = do_rename(conn, path1, path2);
1.1 djm 891: break;
1.26 djm 892: case I_SYMLINK:
1.44 djm 893: path2 = make_absolute(path2, *pwd);
894: err = do_symlink(conn, path1, path2);
1.26 djm 895: break;
1.1 djm 896: case I_RM:
897: path1 = make_absolute(path1, *pwd);
1.44 djm 898: remote_glob(conn, path1, GLOB_NOCHECK, NULL, &g);
1.41 deraadt 899: for (i = 0; g.gl_pathv[i]; i++) {
1.27 djm 900: printf("Removing %s\n", g.gl_pathv[i]);
1.49.2.1! margarid 901: err = do_rm(conn, g.gl_pathv[i]);
! 902: if (err != 0 && err_abort)
! 903: break;
1.27 djm 904: }
1.1 djm 905: break;
906: case I_MKDIR:
907: path1 = make_absolute(path1, *pwd);
908: attrib_clear(&a);
909: a.flags |= SSH2_FILEXFER_ATTR_PERMISSIONS;
910: a.perm = 0777;
1.44 djm 911: err = do_mkdir(conn, path1, &a);
1.1 djm 912: break;
913: case I_RMDIR:
914: path1 = make_absolute(path1, *pwd);
1.44 djm 915: err = do_rmdir(conn, path1);
1.1 djm 916: break;
917: case I_CHDIR:
918: path1 = make_absolute(path1, *pwd);
1.44 djm 919: if ((tmp = do_realpath(conn, path1)) == NULL) {
1.25 deraadt 920: err = 1;
1.11 markus 921: break;
1.25 deraadt 922: }
1.44 djm 923: if ((aa = do_stat(conn, tmp, 0)) == NULL) {
1.11 markus 924: xfree(tmp);
1.25 deraadt 925: err = 1;
1.11 markus 926: break;
927: }
1.9 djm 928: if (!(aa->flags & SSH2_FILEXFER_ATTR_PERMISSIONS)) {
929: error("Can't change directory: Can't check target");
930: xfree(tmp);
1.25 deraadt 931: err = 1;
1.9 djm 932: break;
933: }
934: if (!S_ISDIR(aa->perm)) {
935: error("Can't change directory: \"%s\" is not "
936: "a directory", tmp);
937: xfree(tmp);
1.25 deraadt 938: err = 1;
1.9 djm 939: break;
940: }
1.11 markus 941: xfree(*pwd);
942: *pwd = tmp;
1.1 djm 943: break;
944: case I_LS:
1.12 djm 945: if (!path1) {
1.48 djm 946: do_globbed_ls(conn, *pwd, *pwd, lflag);
1.12 djm 947: break;
948: }
1.49.2.1! margarid 949:
1.48 djm 950: /* Strip pwd off beginning of non-absolute paths */
951: tmp = NULL;
952: if (*path1 != '/')
953: tmp = *pwd;
954:
1.1 djm 955: path1 = make_absolute(path1, *pwd);
1.49.2.1! margarid 956: err = do_globbed_ls(conn, path1, tmp, lflag);
1.1 djm 957: break;
958: case I_LCHDIR:
1.25 deraadt 959: if (chdir(path1) == -1) {
1.1 djm 960: error("Couldn't change local directory to "
961: "\"%s\": %s", path1, strerror(errno));
1.25 deraadt 962: err = 1;
963: }
1.1 djm 964: break;
965: case I_LMKDIR:
1.25 deraadt 966: if (mkdir(path1, 0777) == -1) {
1.17 stevesk 967: error("Couldn't create local directory "
1.1 djm 968: "\"%s\": %s", path1, strerror(errno));
1.25 deraadt 969: err = 1;
970: }
1.1 djm 971: break;
972: case I_LLS:
973: local_do_ls(cmd);
974: break;
975: case I_SHELL:
976: local_do_shell(cmd);
977: break;
978: case I_LUMASK:
979: umask(n_arg);
1.21 stevesk 980: printf("Local umask: %03lo\n", n_arg);
1.1 djm 981: break;
982: case I_CHMOD:
983: path1 = make_absolute(path1, *pwd);
984: attrib_clear(&a);
985: a.flags |= SSH2_FILEXFER_ATTR_PERMISSIONS;
986: a.perm = n_arg;
1.44 djm 987: remote_glob(conn, path1, GLOB_NOCHECK, NULL, &g);
1.41 deraadt 988: for (i = 0; g.gl_pathv[i]; i++) {
1.27 djm 989: printf("Changing mode on %s\n", g.gl_pathv[i]);
1.49.2.1! margarid 990: err = do_setstat(conn, g.gl_pathv[i], &a);
! 991: if (err != 0 && err_abort)
! 992: break;
1.27 djm 993: }
1.5 stevesk 994: break;
1.1 djm 995: case I_CHOWN:
996: case I_CHGRP:
997: path1 = make_absolute(path1, *pwd);
1.44 djm 998: remote_glob(conn, path1, GLOB_NOCHECK, NULL, &g);
1.41 deraadt 999: for (i = 0; g.gl_pathv[i]; i++) {
1.49.2.1! margarid 1000: if (!(aa = do_stat(conn, g.gl_pathv[i], 0))) {
! 1001: if (err != 0 && err_abort)
! 1002: break;
! 1003: else
! 1004: continue;
! 1005: }
1.27 djm 1006: if (!(aa->flags & SSH2_FILEXFER_ATTR_UIDGID)) {
1007: error("Can't get current ownership of "
1008: "remote file \"%s\"", g.gl_pathv[i]);
1.49.2.1! margarid 1009: if (err != 0 && err_abort)
! 1010: break;
! 1011: else
! 1012: continue;
1.27 djm 1013: }
1014: aa->flags &= SSH2_FILEXFER_ATTR_UIDGID;
1.49.2.1! margarid 1015: if (cmdnum == I_CHOWN) {
! 1016: printf("Changing owner on %s\n", g.gl_pathv[i]);
! 1017: aa->uid = n_arg;
! 1018: } else {
! 1019: printf("Changing group on %s\n", g.gl_pathv[i]);
! 1020: aa->gid = n_arg;
! 1021: }
! 1022: err = do_setstat(conn, g.gl_pathv[i], aa);
! 1023: if (err != 0 && err_abort)
! 1024: break;
1.1 djm 1025: }
1026: break;
1027: case I_PWD:
1028: printf("Remote working directory: %s\n", *pwd);
1029: break;
1030: case I_LPWD:
1.49.2.1! margarid 1031: if (!getcwd(path_buf, sizeof(path_buf))) {
! 1032: error("Couldn't get local cwd: %s", strerror(errno));
! 1033: err = -1;
! 1034: break;
! 1035: }
! 1036: printf("Local working directory: %s\n", path_buf);
1.1 djm 1037: break;
1038: case I_QUIT:
1.49.2.1! margarid 1039: /* Processed below */
! 1040: break;
1.1 djm 1041: case I_HELP:
1042: help();
1.28 markus 1043: break;
1044: case I_VERSION:
1.47 deraadt 1045: printf("SFTP protocol version %u\n", sftp_proto_version(conn));
1.1 djm 1046: break;
1.49.2.1! margarid 1047: case I_PROGRESS:
! 1048: showprogress = !showprogress;
! 1049: if (showprogress)
! 1050: printf("Progress meter enabled\n");
! 1051: else
! 1052: printf("Progress meter disabled\n");
! 1053: break;
1.1 djm 1054: default:
1055: fatal("%d is not implemented", cmdnum);
1056: }
1057:
1.29 djm 1058: if (g.gl_pathc)
1059: globfree(&g);
1.1 djm 1060: if (path1)
1061: xfree(path1);
1062: if (path2)
1063: xfree(path2);
1.25 deraadt 1064:
1.49.2.1! margarid 1065: /* If an unignored error occurs in batch mode we should abort. */
! 1066: if (err_abort && err != 0)
! 1067: return (-1);
! 1068: else if (cmdnum == I_QUIT)
! 1069: return (1);
1.25 deraadt 1070:
1.49.2.1! margarid 1071: return (0);
1.1 djm 1072: }
1073:
1.49.2.1! margarid 1074: int
1.35 mouring 1075: interactive_loop(int fd_in, int fd_out, char *file1, char *file2)
1.1 djm 1076: {
1077: char *pwd;
1.35 mouring 1078: char *dir = NULL;
1.1 djm 1079: char cmd[2048];
1.44 djm 1080: struct sftp_conn *conn;
1.49.2.1! margarid 1081: int err;
1.26 djm 1082:
1.44 djm 1083: conn = do_init(fd_in, fd_out, copy_buffer_len, num_requests);
1084: if (conn == NULL)
1.26 djm 1085: fatal("Couldn't initialise connection to server");
1.1 djm 1086:
1.44 djm 1087: pwd = do_realpath(conn, ".");
1.1 djm 1088: if (pwd == NULL)
1089: fatal("Need cwd");
1090:
1.35 mouring 1091: if (file1 != NULL) {
1092: dir = xstrdup(file1);
1093: dir = make_absolute(dir, pwd);
1094:
1.44 djm 1095: if (remote_is_dir(conn, dir) && file2 == NULL) {
1.35 mouring 1096: printf("Changing to: %s\n", dir);
1097: snprintf(cmd, sizeof cmd, "cd \"%s\"", dir);
1.49.2.1! margarid 1098: if (parse_dispatch_command(conn, cmd, &pwd, 1) != 0)
! 1099: return (-1);
1.35 mouring 1100: } else {
1101: if (file2 == NULL)
1102: snprintf(cmd, sizeof cmd, "get %s", dir);
1103: else
1104: snprintf(cmd, sizeof cmd, "get %s %s", dir,
1105: file2);
1106:
1.49.2.1! margarid 1107: err = parse_dispatch_command(conn, cmd, &pwd, 1);
1.45 mpech 1108: xfree(dir);
1.49.2.1! margarid 1109: xfree(pwd);
! 1110: return (err);
1.35 mouring 1111: }
1.45 mpech 1112: xfree(dir);
1.35 mouring 1113: }
1.49.2.1! margarid 1114:
1.14 stevesk 1115: setvbuf(stdout, NULL, _IOLBF, 0);
1.25 deraadt 1116: setvbuf(infile, NULL, _IOLBF, 0);
1.1 djm 1117:
1.49.2.1! margarid 1118: err = 0;
1.41 deraadt 1119: for (;;) {
1.1 djm 1120: char *cp;
1121:
1122: printf("sftp> ");
1123:
1124: /* XXX: use libedit */
1.25 deraadt 1125: if (fgets(cmd, sizeof(cmd), infile) == NULL) {
1.1 djm 1126: printf("\n");
1127: break;
1.25 deraadt 1128: } else if (infile != stdin) /* Bluff typing */
1129: printf("%s", cmd);
1130:
1.1 djm 1131: cp = strrchr(cmd, '\n');
1132: if (cp)
1133: *cp = '\0';
1.25 deraadt 1134:
1.49.2.1! margarid 1135: err = parse_dispatch_command(conn, cmd, &pwd, infile != stdin);
! 1136: if (err != 0)
1.1 djm 1137: break;
1138: }
1139: xfree(pwd);
1.49.2.1! margarid 1140:
! 1141: /* err == 1 signifies normal "quit" exit */
! 1142: return (err >= 0 ? 0 : -1);
1.1 djm 1143: }
1.49.2.1! margarid 1144: