?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
dma: constify bounce reason and avoid strdup
[dragonfly.git] / crypto / openssh / sftp.c
blob66bd111b122c47ad5ca7380cda5261a424f675e0
1 /* $OpenBSD: sftp.c,v 1.107 2009/02/02 11:15:14 dtucker Exp $ */
2 /*
3 * Copyright (c) 2001-2004 Damien Miller <djm@openbsd.org>
4 *
5 * Permission to use, copy, modify, and distribute this software for any
6 * purpose with or without fee is hereby granted, provided that the above
7 * copyright notice and this permission notice appear in all copies.
8 *
9 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
10 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
11 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
12 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
13 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
14 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
15 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
16 */
17
18 #include "includes.h"
19
20 #include <sys/types.h>
21 #include <sys/ioctl.h>
22 #ifdef HAVE_SYS_STAT_H
23 # include <sys/stat.h>
24 #endif
25 #include <sys/param.h>
26 #include <sys/socket.h>
27 #include <sys/wait.h>
28 #ifdef HAVE_SYS_STATVFS_H
29 #include <sys/statvfs.h>
30 #endif
31
32 #include <ctype.h>
33 #include <errno.h>
34
35 #ifdef HAVE_PATHS_H
36 # include <paths.h>
37 #endif
38 #ifdef USE_LIBEDIT
39 #include <histedit.h>
40 #else
41 typedef void EditLine;
42 #endif
43 #include <signal.h>
44 #include <stdlib.h>
45 #include <stdio.h>
46 #include <string.h>
47 #include <unistd.h>
48 #include <stdarg.h>
49
50 #ifdef HAVE_UTIL_H
51 # include <util.h>
52 #endif
53
54 #ifdef HAVE_LIBUTIL_H
55 # include <libutil.h>
56 #endif
57
58 #include "xmalloc.h"
59 #include "log.h"
60 #include "pathnames.h"
61 #include "misc.h"
62
63 #include "sftp.h"
64 #include "buffer.h"
65 #include "sftp-common.h"
66 #include "sftp-client.h"
67
68 /* File to read commands from */
69 FILE* infile;
70
71 /* Are we in batchfile mode? */
72 int batchmode = 0;
73
74 /* Size of buffer used when copying files */
75 size_t copy_buffer_len = 32768;
76
77 /* Number of concurrent outstanding requests */
78 size_t num_requests = 64;
79
80 /* PID of ssh transport process */
81 static pid_t sshpid = -1;
82
83 /* This is set to 0 if the progressmeter is not desired. */
84 int showprogress = 1;
85
86 /* SIGINT received during command processing */
87 volatile sig_atomic_t interrupted = 0;
88
89 /* I wish qsort() took a separate ctx for the comparison function...*/
90 int sort_flag;
91
92 int remote_glob(struct sftp_conn *, const char *, int,
93 int (*)(const char *, int), glob_t *); /* proto for sftp-glob.c */
94
95 extern char *__progname;
96
97 /* Separators for interactive commands */
98 #define WHITESPACE " \t\r\n"
99
100 /* ls flags */
101 #define LS_LONG_VIEW 0x01 /* Full view ala ls -l */
102 #define LS_SHORT_VIEW 0x02 /* Single row view ala ls -1 */
103 #define LS_NUMERIC_VIEW 0x04 /* Long view with numeric uid/gid */
104 #define LS_NAME_SORT 0x08 /* Sort by name (default) */
105 #define LS_TIME_SORT 0x10 /* Sort by mtime */
106 #define LS_SIZE_SORT 0x20 /* Sort by file size */
107 #define LS_REVERSE_SORT 0x40 /* Reverse sort order */
108 #define LS_SHOW_ALL 0x80 /* Don't skip filenames starting with '.' */
109
110 #define VIEW_FLAGS (LS_LONG_VIEW|LS_SHORT_VIEW|LS_NUMERIC_VIEW)
111 #define SORT_FLAGS (LS_NAME_SORT|LS_TIME_SORT|LS_SIZE_SORT)
112
113 /* Commands for interactive mode */
114 #define I_CHDIR 1
115 #define I_CHGRP 2
116 #define I_CHMOD 3
117 #define I_CHOWN 4
118 #define I_DF 24
119 #define I_GET 5
120 #define I_HELP 6
121 #define I_LCHDIR 7
122 #define I_LLS 8
123 #define I_LMKDIR 9
124 #define I_LPWD 10
125 #define I_LS 11
126 #define I_LUMASK 12
127 #define I_MKDIR 13
128 #define I_PUT 14
129 #define I_PWD 15
130 #define I_QUIT 16
131 #define I_RENAME 17
132 #define I_RM 18
133 #define I_RMDIR 19
134 #define I_SHELL 20
135 #define I_SYMLINK 21
136 #define I_VERSION 22
137 #define I_PROGRESS 23
138
139 struct CMD {
140 const char *c;
141 const int n;
142 };
143
144 static const struct CMD cmds[] = {
145 { "bye", I_QUIT },
146 { "cd", I_CHDIR },
147 { "chdir", I_CHDIR },
148 { "chgrp", I_CHGRP },
149 { "chmod", I_CHMOD },
150 { "chown", I_CHOWN },
151 { "df", I_DF },
152 { "dir", I_LS },
153 { "exit", I_QUIT },
154 { "get", I_GET },
155 { "mget", I_GET },
156 { "help", I_HELP },
157 { "lcd", I_LCHDIR },
158 { "lchdir", I_LCHDIR },
159 { "lls", I_LLS },
160 { "lmkdir", I_LMKDIR },
161 { "ln", I_SYMLINK },
162 { "lpwd", I_LPWD },
163 { "ls", I_LS },
164 { "lumask", I_LUMASK },
165 { "mkdir", I_MKDIR },
166 { "progress", I_PROGRESS },
167 { "put", I_PUT },
168 { "mput", I_PUT },
169 { "pwd", I_PWD },
170 { "quit", I_QUIT },
171 { "rename", I_RENAME },
172 { "rm", I_RM },
173 { "rmdir", I_RMDIR },
174 { "symlink", I_SYMLINK },
175 { "version", I_VERSION },
176 { "!", I_SHELL },
177 { "?", I_HELP },
178 { NULL, -1}
179 };
180
181 int interactive_loop(int fd_in, int fd_out, char *file1, char *file2);
182
183 /* ARGSUSED */
184 static void
185 killchild(int signo)
186 {
187 if (sshpid > 1) {
188 kill(sshpid, SIGTERM);
189 waitpid(sshpid, NULL, 0);
190 }
191
192 _exit(1);
193 }
194
195 /* ARGSUSED */
196 static void
197 cmd_interrupt(int signo)
198 {
199 const char msg[] = "\rInterrupt \n";
200 int olderrno = errno;
201
202 write(STDERR_FILENO, msg, sizeof(msg) - 1);
203 interrupted = 1;
204 errno = olderrno;
205 }
206
207 static void
208 help(void)
209 {
210 printf("Available commands:\n"
211 "bye Quit sftp\n"
212 "cd path Change remote directory to 'path'\n"
213 "chgrp grp path Change group of file 'path' to 'grp'\n"
214 "chmod mode path Change permissions of file 'path' to 'mode'\n"
215 "chown own path Change owner of file 'path' to 'own'\n"
216 "df [-hi] [path] Display statistics for current directory or\n"
217 " filesystem containing 'path'\n"
218 "exit Quit sftp\n"
219 "get [-P] remote-path [local-path] Download file\n"
220 "help Display this help text\n"
221 "lcd path Change local directory to 'path'\n"
222 "lls [ls-options [path]] Display local directory listing\n"
223 "lmkdir path Create local directory\n"
224 "ln oldpath newpath Symlink remote file\n"
225 "lpwd Print local working directory\n"
226 "ls [-1aflnrSt] [path] Display remote directory listing\n"
227 "lumask umask Set local umask to 'umask'\n"
228 "mkdir path Create remote directory\n"
229 "progress Toggle display of progress meter\n"
230 "put [-P] local-path [remote-path] Upload file\n"
231 "pwd Display remote working directory\n"
232 "quit Quit sftp\n"
233 "rename oldpath newpath Rename remote file\n"
234 "rm path Delete remote file\n"
235 "rmdir path Remove remote directory\n"
236 "symlink oldpath newpath Symlink remote file\n"
237 "version Show SFTP version\n"
238 "!command Execute 'command' in local shell\n"
239 "! Escape to local shell\n"
240 "? Synonym for help\n");
241 }
242
243 static void
244 local_do_shell(const char *args)
245 {
246 int status;
247 char *shell;
248 pid_t pid;
249
250 if (!*args)
251 args = NULL;
252
253 if ((shell = getenv("SHELL")) == NULL)
254 shell = _PATH_BSHELL;
255
256 if ((pid = fork()) == -1)
257 fatal("Couldn't fork: %s", strerror(errno));
258
259 if (pid == 0) {
260 /* XXX: child has pipe fds to ssh subproc open - issue? */
261 if (args) {
262 debug3("Executing %s -c \"%s\"", shell, args);
263 execl(shell, shell, "-c", args, (char *)NULL);
264 } else {
265 debug3("Executing %s", shell);
266 execl(shell, shell, (char *)NULL);
267 }
268 fprintf(stderr, "Couldn't execute \"%s\": %s\n", shell,
269 strerror(errno));
270 _exit(1);
271 }
272 while (waitpid(pid, &status, 0) == -1)
273 if (errno != EINTR)
274 fatal("Couldn't wait for child: %s", strerror(errno));
275 if (!WIFEXITED(status))
276 error("Shell exited abnormally");
277 else if (WEXITSTATUS(status))
278 error("Shell exited with status %d", WEXITSTATUS(status));
279 }
280
281 static void
282 local_do_ls(const char *args)
283 {
284 if (!args || !*args)
285 local_do_shell(_PATH_LS);
286 else {
287 int len = strlen(_PATH_LS " ") + strlen(args) + 1;
288 char *buf = xmalloc(len);
289
290 /* XXX: quoting - rip quoting code from ftp? */
291 snprintf(buf, len, _PATH_LS " %s", args);
292 local_do_shell(buf);
293 xfree(buf);
294 }
295 }
296
297 /* Strip one path (usually the pwd) from the start of another */
298 static char *
299 path_strip(char *path, char *strip)
300 {
301 size_t len;
302
303 if (strip == NULL)
304 return (xstrdup(path));
305
306 len = strlen(strip);
307 if (strncmp(path, strip, len) == 0) {
308 if (strip[len - 1] != '/' && path[len] == '/')
309 len++;
310 return (xstrdup(path + len));
311 }
312
313 return (xstrdup(path));
314 }
315
316 static char *
317 path_append(char *p1, char *p2)
318 {
319 char *ret;
320 size_t len = strlen(p1) + strlen(p2) + 2;
321
322 ret = xmalloc(len);
323 strlcpy(ret, p1, len);
324 if (p1[0] != '\0' && p1[strlen(p1) - 1] != '/')
325 strlcat(ret, "/", len);
326 strlcat(ret, p2, len);
327
328 return(ret);
329 }
330
331 static char *
332 make_absolute(char *p, char *pwd)
333 {
334 char *abs_str;
335
336 /* Derelativise */
337 if (p && p[0] != '/') {
338 abs_str = path_append(pwd, p);
339 xfree(p);
340 return(abs_str);
341 } else
342 return(p);
343 }
344
345 static int
346 infer_path(const char *p, char **ifp)
347 {
348 char *cp;
349
350 cp = strrchr(p, '/');
351 if (cp == NULL) {
352 *ifp = xstrdup(p);
353 return(0);
354 }
355
356 if (!cp[1]) {
357 error("Invalid path");
358 return(-1);
359 }
360
361 *ifp = xstrdup(cp + 1);
362 return(0);
363 }
364
365 static int
366 parse_getput_flags(const char *cmd, char **argv, int argc, int *pflag)
367 {
368 extern int opterr, optind, optopt, optreset;
369 int ch;
370
371 optind = optreset = 1;
372 opterr = 0;
373
374 *pflag = 0;
375 while ((ch = getopt(argc, argv, "Pp")) != -1) {
376 switch (ch) {
377 case 'p':
378 case 'P':
379 *pflag = 1;
380 break;
381 default:
382 error("%s: Invalid flag -%c", cmd, optopt);
383 return -1;
384 }
385 }
386
387 return optind;
388 }
389
390 static int
391 parse_ls_flags(char **argv, int argc, int *lflag)
392 {
393 extern int opterr, optind, optopt, optreset;
394 int ch;
395
396 optind = optreset = 1;
397 opterr = 0;
398
399 *lflag = LS_NAME_SORT;
400 while ((ch = getopt(argc, argv, "1Saflnrt")) != -1) {
401 switch (ch) {
402 case '1':
403 *lflag &= ~VIEW_FLAGS;
404 *lflag |= LS_SHORT_VIEW;
405 break;
406 case 'S':
407 *lflag &= ~SORT_FLAGS;
408 *lflag |= LS_SIZE_SORT;
409 break;
410 case 'a':
411 *lflag |= LS_SHOW_ALL;
412 break;
413 case 'f':
414 *lflag &= ~SORT_FLAGS;
415 break;
416 case 'l':
417 *lflag &= ~VIEW_FLAGS;
418 *lflag |= LS_LONG_VIEW;
419 break;
420 case 'n':
421 *lflag &= ~VIEW_FLAGS;
422 *lflag |= LS_NUMERIC_VIEW|LS_LONG_VIEW;
423 break;
424 case 'r':
425 *lflag |= LS_REVERSE_SORT;
426 break;
427 case 't':
428 *lflag &= ~SORT_FLAGS;
429 *lflag |= LS_TIME_SORT;
430 break;
431 default:
432 error("ls: Invalid flag -%c", optopt);
433 return -1;
434 }
435 }
436
437 return optind;
438 }
439
440 static int
441 parse_df_flags(const char *cmd, char **argv, int argc, int *hflag, int *iflag)
442 {
443 extern int opterr, optind, optopt, optreset;
444 int ch;
445
446 optind = optreset = 1;
447 opterr = 0;
448
449 *hflag = *iflag = 0;
450 while ((ch = getopt(argc, argv, "hi")) != -1) {
451 switch (ch) {
452 case 'h':
453 *hflag = 1;
454 break;
455 case 'i':
456 *iflag = 1;
457 break;
458 default:
459 error("%s: Invalid flag -%c", cmd, optopt);
460 return -1;
461 }
462 }
463
464 return optind;
465 }
466
467 static int
468 is_dir(char *path)
469 {
470 struct stat sb;
471
472 /* XXX: report errors? */
473 if (stat(path, &sb) == -1)
474 return(0);
475
476 return(S_ISDIR(sb.st_mode));
477 }
478
479 static int
480 remote_is_dir(struct sftp_conn *conn, char *path)
481 {
482 Attrib *a;
483
484 /* XXX: report errors? */
485 if ((a = do_stat(conn, path, 1)) == NULL)
486 return(0);
487 if (!(a->flags & SSH2_FILEXFER_ATTR_PERMISSIONS))
488 return(0);
489 return(S_ISDIR(a->perm));
490 }
491
492 static int
493 process_get(struct sftp_conn *conn, char *src, char *dst, char *pwd, int pflag)
494 {
495 char *abs_src = NULL;
496 char *abs_dst = NULL;
497 char *tmp;
498 glob_t g;
499 int err = 0;
500 int i;
501
502 abs_src = xstrdup(src);
503 abs_src = make_absolute(abs_src, pwd);
504
505 memset(&g, 0, sizeof(g));
506 debug3("Looking up %s", abs_src);
507 if (remote_glob(conn, abs_src, 0, NULL, &g)) {
508 error("File \"%s\" not found.", abs_src);
509 err = -1;
510 goto out;
511 }
512
513 /* If multiple matches, dst must be a directory or unspecified */
514 if (g.gl_matchc > 1 && dst && !is_dir(dst)) {
515 error("Multiple files match, but \"%s\" is not a directory",
516 dst);
517 err = -1;
518 goto out;
519 }
520
521 for (i = 0; g.gl_pathv[i] && !interrupted; i++) {
522 if (infer_path(g.gl_pathv[i], &tmp)) {
523 err = -1;
524 goto out;
525 }
526
527 if (g.gl_matchc == 1 && dst) {
528 /* If directory specified, append filename */
529 xfree(tmp);
530 if (is_dir(dst)) {
531 if (infer_path(g.gl_pathv[0], &tmp)) {
532 err = 1;
533 goto out;
534 }
535 abs_dst = path_append(dst, tmp);
536 xfree(tmp);
537 } else
538 abs_dst = xstrdup(dst);
539 } else if (dst) {
540 abs_dst = path_append(dst, tmp);
541 xfree(tmp);
542 } else
543 abs_dst = tmp;
544
545 printf("Fetching %s to %s\n", g.gl_pathv[i], abs_dst);
546 if (do_download(conn, g.gl_pathv[i], abs_dst, pflag) == -1)
547 err = -1;
548 xfree(abs_dst);
549 abs_dst = NULL;
550 }
551
552 out:
553 xfree(abs_src);
554 globfree(&g);
555 return(err);
556 }
557
558 static int
559 process_put(struct sftp_conn *conn, char *src, char *dst, char *pwd, int pflag)
560 {
561 char *tmp_dst = NULL;
562 char *abs_dst = NULL;
563 char *tmp;
564 glob_t g;
565 int err = 0;
566 int i;
567 struct stat sb;
568
569 if (dst) {
570 tmp_dst = xstrdup(dst);
571 tmp_dst = make_absolute(tmp_dst, pwd);
572 }
573
574 memset(&g, 0, sizeof(g));
575 debug3("Looking up %s", src);
576 if (glob(src, GLOB_NOCHECK, NULL, &g)) {
577 error("File \"%s\" not found.", src);
578 err = -1;
579 goto out;
580 }
581
582 /* If multiple matches, dst may be directory or unspecified */
583 if (g.gl_matchc > 1 && tmp_dst && !remote_is_dir(conn, tmp_dst)) {
584 error("Multiple files match, but \"%s\" is not a directory",
585 tmp_dst);
586 err = -1;
587 goto out;
588 }
589
590 for (i = 0; g.gl_pathv[i] && !interrupted; i++) {
591 if (stat(g.gl_pathv[i], &sb) == -1) {
592 err = -1;
593 error("stat %s: %s", g.gl_pathv[i], strerror(errno));
594 continue;
595 }
596
597 if (!S_ISREG(sb.st_mode)) {
598 error("skipping non-regular file %s",
599 g.gl_pathv[i]);
600 continue;
601 }
602 if (infer_path(g.gl_pathv[i], &tmp)) {
603 err = -1;
604 goto out;
605 }
606
607 if (g.gl_matchc == 1 && tmp_dst) {
608 /* If directory specified, append filename */
609 if (remote_is_dir(conn, tmp_dst)) {
610 if (infer_path(g.gl_pathv[0], &tmp)) {
611 err = 1;
612 goto out;
613 }
614 abs_dst = path_append(tmp_dst, tmp);
615 xfree(tmp);
616 } else
617 abs_dst = xstrdup(tmp_dst);
618
619 } else if (tmp_dst) {
620 abs_dst = path_append(tmp_dst, tmp);
621 xfree(tmp);
622 } else
623 abs_dst = make_absolute(tmp, pwd);
624
625 printf("Uploading %s to %s\n", g.gl_pathv[i], abs_dst);
626 if (do_upload(conn, g.gl_pathv[i], abs_dst, pflag) == -1)
627 err = -1;
628 }
629
630 out:
631 if (abs_dst)
632 xfree(abs_dst);
633 if (tmp_dst)
634 xfree(tmp_dst);
635 globfree(&g);
636 return(err);
637 }
638
639 static int
640 sdirent_comp(const void *aa, const void *bb)
641 {
642 SFTP_DIRENT *a = *(SFTP_DIRENT **)aa;
643 SFTP_DIRENT *b = *(SFTP_DIRENT **)bb;
644 int rmul = sort_flag & LS_REVERSE_SORT ? -1 : 1;
645
646 #define NCMP(a,b) (a == b ? 0 : (a < b ? 1 : -1))
647 if (sort_flag & LS_NAME_SORT)
648 return (rmul * strcmp(a->filename, b->filename));
649 else if (sort_flag & LS_TIME_SORT)
650 return (rmul * NCMP(a->a.mtime, b->a.mtime));
651 else if (sort_flag & LS_SIZE_SORT)
652 return (rmul * NCMP(a->a.size, b->a.size));
653
654 fatal("Unknown ls sort type");
655 }
656
657 /* sftp ls.1 replacement for directories */
658 static int
659 do_ls_dir(struct sftp_conn *conn, char *path, char *strip_path, int lflag)
660 {
661 int n;
662 u_int c = 1, colspace = 0, columns = 1;
663 SFTP_DIRENT **d;
664
665 if ((n = do_readdir(conn, path, &d)) != 0)
666 return (n);
667
668 if (!(lflag & LS_SHORT_VIEW)) {
669 u_int m = 0, width = 80;
670 struct winsize ws;
671 char *tmp;
672
673 /* Count entries for sort and find longest filename */
674 for (n = 0; d[n] != NULL; n++) {
675 if (d[n]->filename[0] != '.' || (lflag & LS_SHOW_ALL))
676 m = MAX(m, strlen(d[n]->filename));
677 }
678
679 /* Add any subpath that also needs to be counted */
680 tmp = path_strip(path, strip_path);
681 m += strlen(tmp);
682 xfree(tmp);
683
684 if (ioctl(fileno(stdin), TIOCGWINSZ, &ws) != -1)
685 width = ws.ws_col;
686
687 columns = width / (m + 2);
688 columns = MAX(columns, 1);
689 colspace = width / columns;
690 colspace = MIN(colspace, width);
691 }
692
693 if (lflag & SORT_FLAGS) {
694 for (n = 0; d[n] != NULL; n++)
695 ; /* count entries */
696 sort_flag = lflag & (SORT_FLAGS|LS_REVERSE_SORT);
697 qsort(d, n, sizeof(*d), sdirent_comp);
698 }
699
700 for (n = 0; d[n] != NULL && !interrupted; n++) {
701 char *tmp, *fname;
702
703 if (d[n]->filename[0] == '.' && !(lflag & LS_SHOW_ALL))
704 continue;
705
706 tmp = path_append(path, d[n]->filename);
707 fname = path_strip(tmp, strip_path);
708 xfree(tmp);
709
710 if (lflag & LS_LONG_VIEW) {
711 if (lflag & LS_NUMERIC_VIEW) {
712 char *lname;
713 struct stat sb;
714
715 memset(&sb, 0, sizeof(sb));
716 attrib_to_stat(&d[n]->a, &sb);
717 lname = ls_file(fname, &sb, 1);
718 printf("%s\n", lname);
719 xfree(lname);
720 } else
721 printf("%s\n", d[n]->longname);
722 } else {
723 printf("%-*s", colspace, fname);
724 if (c >= columns) {
725 printf("\n");
726 c = 1;
727 } else
728 c++;
729 }
730
731 xfree(fname);
732 }
733
734 if (!(lflag & LS_LONG_VIEW) && (c != 1))
735 printf("\n");
736
737 free_sftp_dirents(d);
738 return (0);
739 }
740
741 /* sftp ls.1 replacement which handles path globs */
742 static int
743 do_globbed_ls(struct sftp_conn *conn, char *path, char *strip_path,
744 int lflag)
745 {
746 glob_t g;
747 u_int i, c = 1, colspace = 0, columns = 1;
748 Attrib *a = NULL;
749
750 memset(&g, 0, sizeof(g));
751
752 if (remote_glob(conn, path, GLOB_MARK|GLOB_NOCHECK|GLOB_BRACE,
753 NULL, &g) || (g.gl_pathc && !g.gl_matchc)) {
754 if (g.gl_pathc)
755 globfree(&g);
756 error("Can't ls: \"%s\" not found", path);
757 return (-1);
758 }
759
760 if (interrupted)
761 goto out;
762
763 /*
764 * If the glob returns a single match and it is a directory,
765 * then just list its contents.
766 */
767 if (g.gl_matchc == 1) {
768 if ((a = do_lstat(conn, g.gl_pathv[0], 1)) == NULL) {
769 globfree(&g);
770 return (-1);
771 }
772 if ((a->flags & SSH2_FILEXFER_ATTR_PERMISSIONS) &&
773 S_ISDIR(a->perm)) {
774 int err;
775
776 err = do_ls_dir(conn, g.gl_pathv[0], strip_path, lflag);
777 globfree(&g);
778 return (err);
779 }
780 }
781
782 if (!(lflag & LS_SHORT_VIEW)) {
783 u_int m = 0, width = 80;
784 struct winsize ws;
785
786 /* Count entries for sort and find longest filename */
787 for (i = 0; g.gl_pathv[i]; i++)
788 m = MAX(m, strlen(g.gl_pathv[i]));
789
790 if (ioctl(fileno(stdin), TIOCGWINSZ, &ws) != -1)
791 width = ws.ws_col;
792
793 columns = width / (m + 2);
794 columns = MAX(columns, 1);
795 colspace = width / columns;
796 }
797
798 for (i = 0; g.gl_pathv[i] && !interrupted; i++, a = NULL) {
799 char *fname;
800
801 fname = path_strip(g.gl_pathv[i], strip_path);
802
803 if (lflag & LS_LONG_VIEW) {
804 char *lname;
805 struct stat sb;
806
807 /*
808 * XXX: this is slow - 1 roundtrip per path
809 * A solution to this is to fork glob() and
810 * build a sftp specific version which keeps the
811 * attribs (which currently get thrown away)
812 * that the server returns as well as the filenames.
813 */
814 memset(&sb, 0, sizeof(sb));
815 if (a == NULL)
816 a = do_lstat(conn, g.gl_pathv[i], 1);
817 if (a != NULL)
818 attrib_to_stat(a, &sb);
819 lname = ls_file(fname, &sb, 1);
820 printf("%s\n", lname);
821 xfree(lname);
822 } else {
823 printf("%-*s", colspace, fname);
824 if (c >= columns) {
825 printf("\n");
826 c = 1;
827 } else
828 c++;
829 }
830 xfree(fname);
831 }
832
833 if (!(lflag & LS_LONG_VIEW) && (c != 1))
834 printf("\n");
835
836 out:
837 if (g.gl_pathc)
838 globfree(&g);
839
840 return (0);
841 }
842
843 static int
844 do_df(struct sftp_conn *conn, char *path, int hflag, int iflag)
845 {
846 struct sftp_statvfs st;
847 char s_used[FMT_SCALED_STRSIZE];
848 char s_avail[FMT_SCALED_STRSIZE];
849 char s_root[FMT_SCALED_STRSIZE];
850 char s_total[FMT_SCALED_STRSIZE];
851
852 if (do_statvfs(conn, path, &st, 1) == -1)
853 return -1;
854 if (iflag) {
855 printf(" Inodes Used Avail "
856 "(root) %%Capacity\n");
857 printf("%11llu %11llu %11llu %11llu %3llu%%\n",
858 (unsigned long long)st.f_files,
859 (unsigned long long)(st.f_files - st.f_ffree),
860 (unsigned long long)st.f_favail,
861 (unsigned long long)st.f_ffree,
862 (unsigned long long)(100 * (st.f_files - st.f_ffree) /
863 st.f_files));
864 } else if (hflag) {
865 strlcpy(s_used, "error", sizeof(s_used));
866 strlcpy(s_avail, "error", sizeof(s_avail));
867 strlcpy(s_root, "error", sizeof(s_root));
868 strlcpy(s_total, "error", sizeof(s_total));
869 fmt_scaled((st.f_blocks - st.f_bfree) * st.f_frsize, s_used);
870 fmt_scaled(st.f_bavail * st.f_frsize, s_avail);
871 fmt_scaled(st.f_bfree * st.f_frsize, s_root);
872 fmt_scaled(st.f_blocks * st.f_frsize, s_total);
873 printf(" Size Used Avail (root) %%Capacity\n");
874 printf("%7sB %7sB %7sB %7sB %3llu%%\n",
875 s_total, s_used, s_avail, s_root,
876 (unsigned long long)(100 * (st.f_blocks - st.f_bfree) /
877 st.f_blocks));
878 } else {
879 printf(" Size Used Avail "
880 "(root) %%Capacity\n");
881 printf("%12llu %12llu %12llu %12llu %3llu%%\n",
882 (unsigned long long)(st.f_frsize * st.f_blocks / 1024),
883 (unsigned long long)(st.f_frsize *
884 (st.f_blocks - st.f_bfree) / 1024),
885 (unsigned long long)(st.f_frsize * st.f_bavail / 1024),
886 (unsigned long long)(st.f_frsize * st.f_bfree / 1024),
887 (unsigned long long)(100 * (st.f_blocks - st.f_bfree) /
888 st.f_blocks));
889 }
890 return 0;
891 }
892
893 /*
894 * Undo escaping of glob sequences in place. Used to undo extra escaping
895 * applied in makeargv() when the string is destined for a function that
896 * does not glob it.
897 */
898 static void
899 undo_glob_escape(char *s)
900 {
901 size_t i, j;
902
903 for (i = j = 0;;) {
904 if (s[i] == '\0') {
905 s[j] = '\0';
906 return;
907 }
908 if (s[i] != '\\') {
909 s[j++] = s[i++];
910 continue;
911 }
912 /* s[i] == '\\' */
913 ++i;
914 switch (s[i]) {
915 case '?':
916 case '[':
917 case '*':
918 case '\\':
919 s[j++] = s[i++];
920 break;
921 case '\0':
922 s[j++] = '\\';
923 s[j] = '\0';
924 return;
925 default:
926 s[j++] = '\\';
927 s[j++] = s[i++];
928 break;
929 }
930 }
931 }
932
933 /*
934 * Split a string into an argument vector using sh(1)-style quoting,
935 * comment and escaping rules, but with some tweaks to handle glob(3)
936 * wildcards.
937 * Returns NULL on error or a NULL-terminated array of arguments.
938 */
939 #define MAXARGS 128
940 #define MAXARGLEN 8192
941 static char **
942 makeargv(const char *arg, int *argcp)
943 {
944 int argc, quot;
945 size_t i, j;
946 static char argvs[MAXARGLEN];
947 static char *argv[MAXARGS + 1];
948 enum { MA_START, MA_SQUOTE, MA_DQUOTE, MA_UNQUOTED } state, q;
949
950 *argcp = argc = 0;
951 if (strlen(arg) > sizeof(argvs) - 1) {
952 args_too_longs:
953 error("string too long");
954 return NULL;
955 }
956 state = MA_START;
957 i = j = 0;
958 for (;;) {
959 if (isspace(arg[i])) {
960 if (state == MA_UNQUOTED) {
961 /* Terminate current argument */
962 argvs[j++] = '\0';
963 argc++;
964 state = MA_START;
965 } else if (state != MA_START)
966 argvs[j++] = arg[i];
967 } else if (arg[i] == '"' || arg[i] == '\'') {
968 q = arg[i] == '"' ? MA_DQUOTE : MA_SQUOTE;
969 if (state == MA_START) {
970 argv[argc] = argvs + j;
971 state = q;
972 } else if (state == MA_UNQUOTED)
973 state = q;
974 else if (state == q)
975 state = MA_UNQUOTED;
976 else
977 argvs[j++] = arg[i];
978 } else if (arg[i] == '\\') {
979 if (state == MA_SQUOTE || state == MA_DQUOTE) {
980 quot = state == MA_SQUOTE ? '\'' : '"';
981 /* Unescape quote we are in */
982 /* XXX support \n and friends? */
983 if (arg[i + 1] == quot) {
984 i++;
985 argvs[j++] = arg[i];
986 } else if (arg[i + 1] == '?' ||
987 arg[i + 1] == '[' || arg[i + 1] == '*') {
988 /*
989 * Special case for sftp: append
990 * double-escaped glob sequence -
991 * glob will undo one level of
992 * escaping. NB. string can grow here.
993 */
994 if (j >= sizeof(argvs) - 5)
995 goto args_too_longs;
996 argvs[j++] = '\\';
997 argvs[j++] = arg[i++];
998 argvs[j++] = '\\';
999 argvs[j++] = arg[i];
1000 } else {
1001 argvs[j++] = arg[i++];
1002 argvs[j++] = arg[i];
1003 }
1004 } else {
1005 if (state == MA_START) {
1006 argv[argc] = argvs + j;
1007 state = MA_UNQUOTED;
1008 }
1009 if (arg[i + 1] == '?' || arg[i + 1] == '[' ||
1010 arg[i + 1] == '*' || arg[i + 1] == '\\') {
1011 /*
1012 * Special case for sftp: append
1013 * escaped glob sequence -
1014 * glob will undo one level of
1015 * escaping.
1016 */
1017 argvs[j++] = arg[i++];
1018 argvs[j++] = arg[i];
1019 } else {
1020 /* Unescape everything */
1021 /* XXX support \n and friends? */
1022 i++;
1023 argvs[j++] = arg[i];
1024 }
1025 }
1026 } else if (arg[i] == '#') {
1027 if (state == MA_SQUOTE || state == MA_DQUOTE)
1028 argvs[j++] = arg[i];
1029 else
1030 goto string_done;
1031 } else if (arg[i] == '\0') {
1032 if (state == MA_SQUOTE || state == MA_DQUOTE) {
1033 error("Unterminated quoted argument");
1034 return NULL;
1035 }
1036 string_done:
1037 if (state == MA_UNQUOTED) {
1038 argvs[j++] = '\0';
1039 argc++;
1040 }
1041 break;
1042 } else {
1043 if (state == MA_START) {
1044 argv[argc] = argvs + j;
1045 state = MA_UNQUOTED;
1046 }
1047 if ((state == MA_SQUOTE || state == MA_DQUOTE) &&
1048 (arg[i] == '?' || arg[i] == '[' || arg[i] == '*')) {
1049 /*
1050 * Special case for sftp: escape quoted
1051 * glob(3) wildcards. NB. string can grow
1052 * here.
1053 */
1054 if (j >= sizeof(argvs) - 3)
1055 goto args_too_longs;
1056 argvs[j++] = '\\';
1057 argvs[j++] = arg[i];
1058 } else
1059 argvs[j++] = arg[i];
1060 }
1061 i++;
1062 }
1063 *argcp = argc;
1064 return argv;
1065 }
1066
1067 static int
1068 parse_args(const char **cpp, int *pflag, int *lflag, int *iflag, int *hflag,
1069 unsigned long *n_arg, char **path1, char **path2)
1070 {
1071 const char *cmd, *cp = *cpp;
1072 char *cp2, **argv;
1073 int base = 0;
1074 long l;
1075 int i, cmdnum, optidx, argc;
1076
1077 /* Skip leading whitespace */
1078 cp = cp + strspn(cp, WHITESPACE);
1079
1080 /* Ignore blank lines and lines which begin with comment '#' char */
1081 if (*cp == '\0' || *cp == '#')
1082 return (0);
1083
1084 /* Check for leading '-' (disable error processing) */
1085 *iflag = 0;
1086 if (*cp == '-') {
1087 *iflag = 1;
1088 cp++;
1089 }
1090
1091 if ((argv = makeargv(cp, &argc)) == NULL)
1092 return -1;
1093
1094 /* Figure out which command we have */
1095 for (i = 0; cmds[i].c != NULL; i++) {
1096 if (strcasecmp(cmds[i].c, argv[0]) == 0)
1097 break;
1098 }
1099 cmdnum = cmds[i].n;
1100 cmd = cmds[i].c;
1101
1102 /* Special case */
1103 if (*cp == '!') {
1104 cp++;
1105 cmdnum = I_SHELL;
1106 } else if (cmdnum == -1) {
1107 error("Invalid command.");
1108 return -1;
1109 }
1110
1111 /* Get arguments and parse flags */
1112 *lflag = *pflag = *hflag = *n_arg = 0;
1113 *path1 = *path2 = NULL;
1114 optidx = 1;
1115 switch (cmdnum) {
1116 case I_GET:
1117 case I_PUT:
1118 if ((optidx = parse_getput_flags(cmd, argv, argc, pflag)) == -1)
1119 return -1;
1120 /* Get first pathname (mandatory) */
1121 if (argc - optidx < 1) {
1122 error("You must specify at least one path after a "
1123 "%s command.", cmd);
1124 return -1;
1125 }
1126 *path1 = xstrdup(argv[optidx]);
1127 /* Get second pathname (optional) */
1128 if (argc - optidx > 1) {
1129 *path2 = xstrdup(argv[optidx + 1]);
1130 /* Destination is not globbed */
1131 undo_glob_escape(*path2);
1132 }
1133 break;
1134 case I_RENAME:
1135 case I_SYMLINK:
1136 if (argc - optidx < 2) {
1137 error("You must specify two paths after a %s "
1138 "command.", cmd);
1139 return -1;
1140 }
1141 *path1 = xstrdup(argv[optidx]);
1142 *path2 = xstrdup(argv[optidx + 1]);
1143 /* Paths are not globbed */
1144 undo_glob_escape(*path1);
1145 undo_glob_escape(*path2);
1146 break;
1147 case I_RM:
1148 case I_MKDIR:
1149 case I_RMDIR:
1150 case I_CHDIR:
1151 case I_LCHDIR:
1152 case I_LMKDIR:
1153 /* Get pathname (mandatory) */
1154 if (argc - optidx < 1) {
1155 error("You must specify a path after a %s command.",
1156 cmd);
1157 return -1;
1158 }
1159 *path1 = xstrdup(argv[optidx]);
1160 /* Only "rm" globs */
1161 if (cmdnum != I_RM)
1162 undo_glob_escape(*path1);
1163 break;
1164 case I_DF:
1165 if ((optidx = parse_df_flags(cmd, argv, argc, hflag,
1166 iflag)) == -1)
1167 return -1;
1168 /* Default to current directory if no path specified */
1169 if (argc - optidx < 1)
1170 *path1 = NULL;
1171 else {
1172 *path1 = xstrdup(argv[optidx]);
1173 undo_glob_escape(*path1);
1174 }
1175 break;
1176 case I_LS:
1177 if ((optidx = parse_ls_flags(argv, argc, lflag)) == -1)
1178 return(-1);
1179 /* Path is optional */
1180 if (argc - optidx > 0)
1181 *path1 = xstrdup(argv[optidx]);
1182 break;
1183 case I_LLS:
1184 /* Skip ls command and following whitespace */
1185 cp = cp + strlen(cmd) + strspn(cp, WHITESPACE);
1186 case I_SHELL:
1187 /* Uses the rest of the line */
1188 break;
1189 case I_LUMASK:
1190 case I_CHMOD:
1191 base = 8;
1192 case I_CHOWN:
1193 case I_CHGRP:
1194 /* Get numeric arg (mandatory) */
1195 if (argc - optidx < 1)
1196 goto need_num_arg;
1197 errno = 0;
1198 l = strtol(argv[optidx], &cp2, base);
1199 if (cp2 == argv[optidx] || *cp2 != '\0' ||
1200 ((l == LONG_MIN || l == LONG_MAX) && errno == ERANGE) ||
1201 l < 0) {
1202 need_num_arg:
1203 error("You must supply a numeric argument "
1204 "to the %s command.", cmd);
1205 return -1;
1206 }
1207 *n_arg = l;
1208 if (cmdnum == I_LUMASK)
1209 break;
1210 /* Get pathname (mandatory) */
1211 if (argc - optidx < 2) {
1212 error("You must specify a path after a %s command.",
1213 cmd);
1214 return -1;
1215 }
1216 *path1 = xstrdup(argv[optidx + 1]);
1217 break;
1218 case I_QUIT:
1219 case I_PWD:
1220 case I_LPWD:
1221 case I_HELP:
1222 case I_VERSION:
1223 case I_PROGRESS:
1224 break;
1225 default:
1226 fatal("Command not implemented");
1227 }
1228
1229 *cpp = cp;
1230 return(cmdnum);
1231 }
1232
1233 static int
1234 parse_dispatch_command(struct sftp_conn *conn, const char *cmd, char **pwd,
1235 int err_abort)
1236 {
1237 char *path1, *path2, *tmp;
1238 int pflag = 0, lflag = 0, iflag = 0, hflag = 0, cmdnum, i;
1239 unsigned long n_arg = 0;
1240 Attrib a, *aa;
1241 char path_buf[MAXPATHLEN];
1242 int err = 0;
1243 glob_t g;
1244
1245 path1 = path2 = NULL;
1246 cmdnum = parse_args(&cmd, &pflag, &lflag, &iflag, &hflag, &n_arg,
1247 &path1, &path2);
1248
1249 if (iflag != 0)
1250 err_abort = 0;
1251
1252 memset(&g, 0, sizeof(g));
1253
1254 /* Perform command */
1255 switch (cmdnum) {
1256 case 0:
1257 /* Blank line */
1258 break;
1259 case -1:
1260 /* Unrecognized command */
1261 err = -1;
1262 break;
1263 case I_GET:
1264 err = process_get(conn, path1, path2, *pwd, pflag);
1265 break;
1266 case I_PUT:
1267 err = process_put(conn, path1, path2, *pwd, pflag);
1268 break;
1269 case I_RENAME:
1270 path1 = make_absolute(path1, *pwd);
1271 path2 = make_absolute(path2, *pwd);
1272 err = do_rename(conn, path1, path2);
1273 break;
1274 case I_SYMLINK:
1275 path2 = make_absolute(path2, *pwd);
1276 err = do_symlink(conn, path1, path2);
1277 break;
1278 case I_RM:
1279 path1 = make_absolute(path1, *pwd);
1280 remote_glob(conn, path1, GLOB_NOCHECK, NULL, &g);
1281 for (i = 0; g.gl_pathv[i] && !interrupted; i++) {
1282 printf("Removing %s\n", g.gl_pathv[i]);
1283 err = do_rm(conn, g.gl_pathv[i]);
1284 if (err != 0 && err_abort)
1285 break;
1286 }
1287 break;
1288 case I_MKDIR:
1289 path1 = make_absolute(path1, *pwd);
1290 attrib_clear(&a);
1291 a.flags |= SSH2_FILEXFER_ATTR_PERMISSIONS;
1292 a.perm = 0777;
1293 err = do_mkdir(conn, path1, &a);
1294 break;
1295 case I_RMDIR:
1296 path1 = make_absolute(path1, *pwd);
1297 err = do_rmdir(conn, path1);
1298 break;
1299 case I_CHDIR:
1300 path1 = make_absolute(path1, *pwd);
1301 if ((tmp = do_realpath(conn, path1)) == NULL) {
1302 err = 1;
1303 break;
1304 }
1305 if ((aa = do_stat(conn, tmp, 0)) == NULL) {
1306 xfree(tmp);
1307 err = 1;
1308 break;
1309 }
1310 if (!(aa->flags & SSH2_FILEXFER_ATTR_PERMISSIONS)) {
1311 error("Can't change directory: Can't check target");
1312 xfree(tmp);
1313 err = 1;
1314 break;
1315 }
1316 if (!S_ISDIR(aa->perm)) {
1317 error("Can't change directory: \"%s\" is not "
1318 "a directory", tmp);
1319 xfree(tmp);
1320 err = 1;
1321 break;
1322 }
1323 xfree(*pwd);
1324 *pwd = tmp;
1325 break;
1326 case I_LS:
1327 if (!path1) {
1328 do_globbed_ls(conn, *pwd, *pwd, lflag);
1329 break;
1330 }
1331
1332 /* Strip pwd off beginning of non-absolute paths */
1333 tmp = NULL;
1334 if (*path1 != '/')
1335 tmp = *pwd;
1336
1337 path1 = make_absolute(path1, *pwd);
1338 err = do_globbed_ls(conn, path1, tmp, lflag);
1339 break;
1340 case I_DF:
1341 /* Default to current directory if no path specified */
1342 if (path1 == NULL)
1343 path1 = xstrdup(*pwd);
1344 path1 = make_absolute(path1, *pwd);
1345 err = do_df(conn, path1, hflag, iflag);
1346 break;
1347 case I_LCHDIR:
1348 if (chdir(path1) == -1) {
1349 error("Couldn't change local directory to "
1350 "\"%s\": %s", path1, strerror(errno));
1351 err = 1;
1352 }
1353 break;
1354 case I_LMKDIR:
1355 if (mkdir(path1, 0777) == -1) {
1356 error("Couldn't create local directory "
1357 "\"%s\": %s", path1, strerror(errno));
1358 err = 1;
1359 }
1360 break;
1361 case I_LLS:
1362 local_do_ls(cmd);
1363 break;
1364 case I_SHELL:
1365 local_do_shell(cmd);
1366 break;
1367 case I_LUMASK:
1368 umask(n_arg);
1369 printf("Local umask: %03lo\n", n_arg);
1370 break;
1371 case I_CHMOD:
1372 path1 = make_absolute(path1, *pwd);
1373 attrib_clear(&a);
1374 a.flags |= SSH2_FILEXFER_ATTR_PERMISSIONS;
1375 a.perm = n_arg;
1376 remote_glob(conn, path1, GLOB_NOCHECK, NULL, &g);
1377 for (i = 0; g.gl_pathv[i] && !interrupted; i++) {
1378 printf("Changing mode on %s\n", g.gl_pathv[i]);
1379 err = do_setstat(conn, g.gl_pathv[i], &a);
1380 if (err != 0 && err_abort)
1381 break;
1382 }
1383 break;
1384 case I_CHOWN:
1385 case I_CHGRP:
1386 path1 = make_absolute(path1, *pwd);
1387 remote_glob(conn, path1, GLOB_NOCHECK, NULL, &g);
1388 for (i = 0; g.gl_pathv[i] && !interrupted; i++) {
1389 if (!(aa = do_stat(conn, g.gl_pathv[i], 0))) {
1390 if (err_abort) {
1391 err = -1;
1392 break;
1393 } else
1394 continue;
1395 }
1396 if (!(aa->flags & SSH2_FILEXFER_ATTR_UIDGID)) {
1397 error("Can't get current ownership of "
1398 "remote file \"%s\"", g.gl_pathv[i]);
1399 if (err_abort) {
1400 err = -1;
1401 break;
1402 } else
1403 continue;
1404 }
1405 aa->flags &= SSH2_FILEXFER_ATTR_UIDGID;
1406 if (cmdnum == I_CHOWN) {
1407 printf("Changing owner on %s\n", g.gl_pathv[i]);
1408 aa->uid = n_arg;
1409 } else {
1410 printf("Changing group on %s\n", g.gl_pathv[i]);
1411 aa->gid = n_arg;
1412 }
1413 err = do_setstat(conn, g.gl_pathv[i], aa);
1414 if (err != 0 && err_abort)
1415 break;
1416 }
1417 break;
1418 case I_PWD:
1419 printf("Remote working directory: %s\n", *pwd);
1420 break;
1421 case I_LPWD:
1422 if (!getcwd(path_buf, sizeof(path_buf))) {
1423 error("Couldn't get local cwd: %s", strerror(errno));
1424 err = -1;
1425 break;
1426 }
1427 printf("Local working directory: %s\n", path_buf);
1428 break;
1429 case I_QUIT:
1430 /* Processed below */
1431 break;
1432 case I_HELP:
1433 help();
1434 break;
1435 case I_VERSION:
1436 printf("SFTP protocol version %u\n", sftp_proto_version(conn));
1437 break;
1438 case I_PROGRESS:
1439 showprogress = !showprogress;
1440 if (showprogress)
1441 printf("Progress meter enabled\n");
1442 else
1443 printf("Progress meter disabled\n");
1444 break;
1445 default:
1446 fatal("%d is not implemented", cmdnum);
1447 }
1448
1449 if (g.gl_pathc)
1450 globfree(&g);
1451 if (path1)
1452 xfree(path1);
1453 if (path2)
1454 xfree(path2);
1455
1456 /* If an unignored error occurs in batch mode we should abort. */
1457 if (err_abort && err != 0)
1458 return (-1);
1459 else if (cmdnum == I_QUIT)
1460 return (1);
1461
1462 return (0);
1463 }
1464
1465 #ifdef USE_LIBEDIT
1466 static char *
1467 prompt(EditLine *el)
1468 {
1469 return ("sftp> ");
1470 }
1471 #endif
1472
1473 int
1474 interactive_loop(int fd_in, int fd_out, char *file1, char *file2)
1475 {
1476 char *pwd;
1477 char *dir = NULL;
1478 char cmd[2048];
1479 struct sftp_conn *conn;
1480 int err, interactive;
1481 EditLine *el = NULL;
1482 #ifdef USE_LIBEDIT
1483 History *hl = NULL;
1484 HistEvent hev;
1485 extern char *__progname;
1486
1487 if (!batchmode && isatty(STDIN_FILENO)) {
1488 if ((el = el_init(__progname, stdin, stdout, stderr)) == NULL)
1489 fatal("Couldn't initialise editline");
1490 if ((hl = history_init()) == NULL)
1491 fatal("Couldn't initialise editline history");
1492 history(hl, &hev, H_SETSIZE, 100);
1493 el_set(el, EL_HIST, history, hl);
1494
1495 el_set(el, EL_PROMPT, prompt);
1496 el_set(el, EL_EDITOR, "emacs");
1497 el_set(el, EL_TERMINAL, NULL);
1498 el_set(el, EL_SIGNAL, 1);
1499 el_source(el, NULL);
1500 }
1501 #endif /* USE_LIBEDIT */
1502
1503 conn = do_init(fd_in, fd_out, copy_buffer_len, num_requests);
1504 if (conn == NULL)
1505 fatal("Couldn't initialise connection to server");
1506
1507 pwd = do_realpath(conn, ".");
1508 if (pwd == NULL)
1509 fatal("Need cwd");
1510
1511 if (file1 != NULL) {
1512 dir = xstrdup(file1);
1513 dir = make_absolute(dir, pwd);
1514
1515 if (remote_is_dir(conn, dir) && file2 == NULL) {
1516 printf("Changing to: %s\n", dir);
1517 snprintf(cmd, sizeof cmd, "cd \"%s\"", dir);
1518 if (parse_dispatch_command(conn, cmd, &pwd, 1) != 0) {
1519 xfree(dir);
1520 xfree(pwd);
1521 xfree(conn);
1522 return (-1);
1523 }
1524 } else {
1525 if (file2 == NULL)
1526 snprintf(cmd, sizeof cmd, "get %s", dir);
1527 else
1528 snprintf(cmd, sizeof cmd, "get %s %s", dir,
1529 file2);
1530
1531 err = parse_dispatch_command(conn, cmd, &pwd, 1);
1532 xfree(dir);
1533 xfree(pwd);
1534 xfree(conn);
1535 return (err);
1536 }
1537 xfree(dir);
1538 }
1539
1540 #if defined(HAVE_SETVBUF) && !defined(BROKEN_SETVBUF)
1541 setvbuf(stdout, NULL, _IOLBF, 0);
1542 setvbuf(infile, NULL, _IOLBF, 0);
1543 #else
1544 setlinebuf(stdout);
1545 setlinebuf(infile);
1546 #endif
1547
1548 interactive = !batchmode && isatty(STDIN_FILENO);
1549 err = 0;
1550 for (;;) {
1551 char *cp;
1552
1553 signal(SIGINT, SIG_IGN);
1554
1555 if (el == NULL) {
1556 if (interactive)
1557 printf("sftp> ");
1558 if (fgets(cmd, sizeof(cmd), infile) == NULL) {
1559 if (interactive)
1560 printf("\n");
1561 break;
1562 }
1563 if (!interactive) { /* Echo command */
1564 printf("sftp> %s", cmd);
1565 if (strlen(cmd) > 0 &&
1566 cmd[strlen(cmd) - 1] != '\n')
1567 printf("\n");
1568 }
1569 } else {
1570 #ifdef USE_LIBEDIT
1571 const char *line;
1572 int count = 0;
1573
1574 if ((line = el_gets(el, &count)) == NULL || count <= 0) {
1575 printf("\n");
1576 break;
1577 }
1578 history(hl, &hev, H_ENTER, line);
1579 if (strlcpy(cmd, line, sizeof(cmd)) >= sizeof(cmd)) {
1580 fprintf(stderr, "Error: input line too long\n");
1581 continue;
1582 }
1583 #endif /* USE_LIBEDIT */
1584 }
1585
1586 cp = strrchr(cmd, '\n');
1587 if (cp)
1588 *cp = '\0';
1589
1590 /* Handle user interrupts gracefully during commands */
1591 interrupted = 0;
1592 signal(SIGINT, cmd_interrupt);
1593
1594 err = parse_dispatch_command(conn, cmd, &pwd, batchmode);
1595 if (err != 0)
1596 break;
1597 }
1598 xfree(pwd);
1599 xfree(conn);
1600
1601 #ifdef USE_LIBEDIT
1602 if (el != NULL)
1603 el_end(el);
1604 #endif /* USE_LIBEDIT */
1605
1606 /* err == 1 signifies normal "quit" exit */
1607 return (err >= 0 ? 0 : -1);
1608 }
1609
1610 static void
1611 connect_to_server(char *path, char **args, int *in, int *out)
1612 {
1613 int c_in, c_out;
1614
1615 #ifdef USE_PIPES
1616 int pin[2], pout[2];
1617
1618 if ((pipe(pin) == -1) || (pipe(pout) == -1))
1619 fatal("pipe: %s", strerror(errno));
1620 *in = pin[0];
1621 *out = pout[1];
1622 c_in = pout[0];
1623 c_out = pin[1];
1624 #else /* USE_PIPES */
1625 int inout[2];
1626
1627 if (socketpair(AF_UNIX, SOCK_STREAM, 0, inout) == -1)
1628 fatal("socketpair: %s", strerror(errno));
1629 *in = *out = inout[0];
1630 c_in = c_out = inout[1];
1631 #endif /* USE_PIPES */
1632
1633 if ((sshpid = fork()) == -1)
1634 fatal("fork: %s", strerror(errno));
1635 else if (sshpid == 0) {
1636 if ((dup2(c_in, STDIN_FILENO) == -1) ||
1637 (dup2(c_out, STDOUT_FILENO) == -1)) {
1638 fprintf(stderr, "dup2: %s\n", strerror(errno));
1639 _exit(1);
1640 }
1641 close(*in);
1642 close(*out);
1643 close(c_in);
1644 close(c_out);
1645
1646 /*
1647 * The underlying ssh is in the same process group, so we must
1648 * ignore SIGINT if we want to gracefully abort commands,
1649 * otherwise the signal will make it to the ssh process and
1650 * kill it too
1651 */
1652 signal(SIGINT, SIG_IGN);
1653 execvp(path, args);
1654 fprintf(stderr, "exec: %s: %s\n", path, strerror(errno));
1655 _exit(1);
1656 }
1657
1658 signal(SIGTERM, killchild);
1659 signal(SIGINT, killchild);
1660 signal(SIGHUP, killchild);
1661 close(c_in);
1662 close(c_out);
1663 }
1664
1665 static void
1666 usage(void)
1667 {
1668 extern char *__progname;
1669
1670 fprintf(stderr,
1671 "usage: %s [-1Cv] [-B buffer_size] [-b batchfile] [-F ssh_config]\n"
1672 " [-o ssh_option] [-P sftp_server_path] [-R num_requests]\n"
1673 " [-S program] [-s subsystem | sftp_server] host\n"
1674 " %s [user@]host[:file ...]\n"
1675 " %s [user@]host[:dir[/]]\n"
1676 " %s -b batchfile [user@]host\n", __progname, __progname, __progname, __progname);
1677 exit(1);
1678 }
1679
1680 int
1681 main(int argc, char **argv)
1682 {
1683 int in, out, ch, err;
1684 char *host, *userhost, *cp, *file2 = NULL;
1685 int debug_level = 0, sshver = 2;
1686 char *file1 = NULL, *sftp_server = NULL;
1687 char *ssh_program = _PATH_SSH_PROGRAM, *sftp_direct = NULL;
1688 LogLevel ll = SYSLOG_LEVEL_INFO;
1689 arglist args;
1690 extern int optind;
1691 extern char *optarg;
1692
1693 /* Ensure that fds 0, 1 and 2 are open or directed to /dev/null */
1694 sanitise_stdfd();
1695
1696 __progname = ssh_get_progname(argv[0]);
1697 memset(&args, '\0', sizeof(args));
1698 args.list = NULL;
1699 addargs(&args, "%s", ssh_program);
1700 addargs(&args, "-oForwardX11 no");
1701 addargs(&args, "-oForwardAgent no");
1702 addargs(&args, "-oPermitLocalCommand no");
1703 addargs(&args, "-oClearAllForwardings yes");
1704
1705 ll = SYSLOG_LEVEL_INFO;
1706 infile = stdin;
1707
1708 while ((ch = getopt(argc, argv, "1hvCo:s:S:b:B:F:P:R:")) != -1) {
1709 switch (ch) {
1710 case 'C':
1711 addargs(&args, "-C");
1712 break;
1713 case 'v':
1714 if (debug_level < 3) {
1715 addargs(&args, "-v");
1716 ll = SYSLOG_LEVEL_DEBUG1 + debug_level;
1717 }
1718 debug_level++;
1719 break;
1720 case 'F':
1721 case 'o':
1722 addargs(&args, "-%c%s", ch, optarg);
1723 break;
1724 case '1':
1725 sshver = 1;
1726 if (sftp_server == NULL)
1727 sftp_server = _PATH_SFTP_SERVER;
1728 break;
1729 case 's':
1730 sftp_server = optarg;
1731 break;
1732 case 'S':
1733 ssh_program = optarg;
1734 replacearg(&args, 0, "%s", ssh_program);
1735 break;
1736 case 'b':
1737 if (batchmode)
1738 fatal("Batch file already specified.");
1739
1740 /* Allow "-" as stdin */
1741 if (strcmp(optarg, "-") != 0 &&
1742 (infile = fopen(optarg, "r")) == NULL)
1743 fatal("%s (%s).", strerror(errno), optarg);
1744 showprogress = 0;
1745 batchmode = 1;
1746 addargs(&args, "-obatchmode yes");
1747 break;
1748 case 'P':
1749 sftp_direct = optarg;
1750 break;
1751 case 'B':
1752 copy_buffer_len = strtol(optarg, &cp, 10);
1753 if (copy_buffer_len == 0 || *cp != '\0')
1754 fatal("Invalid buffer size \"%s\"", optarg);
1755 break;
1756 case 'R':
1757 num_requests = strtol(optarg, &cp, 10);
1758 if (num_requests == 0 || *cp != '\0')
1759 fatal("Invalid number of requests \"%s\"",
1760 optarg);
1761 break;
1762 case 'h':
1763 default:
1764 usage();
1765 }
1766 }
1767
1768 if (!isatty(STDERR_FILENO))
1769 showprogress = 0;
1770
1771 log_init(argv[0], ll, SYSLOG_FACILITY_USER, 1);
1772
1773 if (sftp_direct == NULL) {
1774 if (optind == argc || argc > (optind + 2))
1775 usage();
1776
1777 userhost = xstrdup(argv[optind]);
1778 file2 = argv[optind+1];
1779
1780 if ((host = strrchr(userhost, '@')) == NULL)
1781 host = userhost;
1782 else {
1783 *host++ = '\0';
1784 if (!userhost[0]) {
1785 fprintf(stderr, "Missing username\n");
1786 usage();
1787 }
1788 addargs(&args, "-l%s", userhost);
1789 }
1790
1791 if ((cp = colon(host)) != NULL) {
1792 *cp++ = '\0';
1793 file1 = cp;
1794 }
1795
1796 host = cleanhostname(host);
1797 if (!*host) {
1798 fprintf(stderr, "Missing hostname\n");
1799 usage();
1800 }
1801
1802 addargs(&args, "-oProtocol %d", sshver);
1803
1804 /* no subsystem if the server-spec contains a '/' */
1805 if (sftp_server == NULL || strchr(sftp_server, '/') == NULL)
1806 addargs(&args, "-s");
1807
1808 addargs(&args, "%s", host);
1809 addargs(&args, "%s", (sftp_server != NULL ?
1810 sftp_server : "sftp"));
1811
1812 if (!batchmode)
1813 fprintf(stderr, "Connecting to %s...\n", host);
1814 connect_to_server(ssh_program, args.list, &in, &out);
1815 } else {
1816 args.list = NULL;
1817 addargs(&args, "sftp-server");
1818
1819 if (!batchmode)
1820 fprintf(stderr, "Attaching to %s...\n", sftp_direct);
1821 connect_to_server(sftp_direct, args.list, &in, &out);
1822 }
1823 freeargs(&args);
1824
1825 err = interactive_loop(in, out, file1, file2);
1826
1827 #if !defined(USE_PIPES)
1828 shutdown(in, SHUT_RDWR);
1829 shutdown(out, SHUT_RDWR);
1830 #endif
1831
1832 close(in);
1833 close(out);
1834 if (batchmode)
1835 fclose(infile);
1836
1837 while (waitpid(sshpid, NULL, 0) == -1)
1838 if (errno != EINTR)
1839 fatal("Couldn't wait for ssh process: %s",
1840 strerror(errno));
1841
1842 exit(err == 0 ? 0 : 1);
1843 }
DragonFly BSD