xref: /freebsd/crypto/openssh/sftp.c (revision ca86bcf2531c7b149c95244a67853d44323e7855)
1*ca86bcf2SDag-Erling Smørgrav /* $OpenBSD: sftp.c,v 1.177 2016/10/18 12:41:22 millert Exp $ */
21e8db6e2SBrian Feldman /*
3efcad6b7SDag-Erling Smørgrav  * Copyright (c) 2001-2004 Damien Miller <djm@openbsd.org>
41e8db6e2SBrian Feldman  *
5efcad6b7SDag-Erling Smørgrav  * Permission to use, copy, modify, and distribute this software for any
6efcad6b7SDag-Erling Smørgrav  * purpose with or without fee is hereby granted, provided that the above
7efcad6b7SDag-Erling Smørgrav  * copyright notice and this permission notice appear in all copies.
81e8db6e2SBrian Feldman  *
9efcad6b7SDag-Erling Smørgrav  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
10efcad6b7SDag-Erling Smørgrav  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
11efcad6b7SDag-Erling Smørgrav  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
12efcad6b7SDag-Erling Smørgrav  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
13efcad6b7SDag-Erling Smørgrav  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
14efcad6b7SDag-Erling Smørgrav  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
15efcad6b7SDag-Erling Smørgrav  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
161e8db6e2SBrian Feldman  */
171e8db6e2SBrian Feldman 
181e8db6e2SBrian Feldman #include "includes.h"
191e8db6e2SBrian Feldman 
20761efaa7SDag-Erling Smørgrav #include <sys/types.h>
21761efaa7SDag-Erling Smørgrav #include <sys/ioctl.h>
22761efaa7SDag-Erling Smørgrav #ifdef HAVE_SYS_STAT_H
23761efaa7SDag-Erling Smørgrav # include <sys/stat.h>
24761efaa7SDag-Erling Smørgrav #endif
25761efaa7SDag-Erling Smørgrav #include <sys/param.h>
26761efaa7SDag-Erling Smørgrav #include <sys/socket.h>
27761efaa7SDag-Erling Smørgrav #include <sys/wait.h>
28d4af9e69SDag-Erling Smørgrav #ifdef HAVE_SYS_STATVFS_H
29d4af9e69SDag-Erling Smørgrav #include <sys/statvfs.h>
30d4af9e69SDag-Erling Smørgrav #endif
315e8dbd04SDag-Erling Smørgrav 
32d4af9e69SDag-Erling Smørgrav #include <ctype.h>
33761efaa7SDag-Erling Smørgrav #include <errno.h>
34761efaa7SDag-Erling Smørgrav 
35761efaa7SDag-Erling Smørgrav #ifdef HAVE_PATHS_H
36761efaa7SDag-Erling Smørgrav # include <paths.h>
37761efaa7SDag-Erling Smørgrav #endif
38b15c8340SDag-Erling Smørgrav #ifdef HAVE_LIBGEN_H
39b15c8340SDag-Erling Smørgrav #include <libgen.h>
40b15c8340SDag-Erling Smørgrav #endif
41e4a9863fSDag-Erling Smørgrav #ifdef HAVE_LOCALE_H
42e4a9863fSDag-Erling Smørgrav # include <locale.h>
43e4a9863fSDag-Erling Smørgrav #endif
445e8dbd04SDag-Erling Smørgrav #ifdef USE_LIBEDIT
455e8dbd04SDag-Erling Smørgrav #include <histedit.h>
465e8dbd04SDag-Erling Smørgrav #else
475e8dbd04SDag-Erling Smørgrav typedef void EditLine;
485e8dbd04SDag-Erling Smørgrav #endif
49bc5531deSDag-Erling Smørgrav #include <limits.h>
50761efaa7SDag-Erling Smørgrav #include <signal.h>
51076ad2f8SDag-Erling Smørgrav #include <stdarg.h>
52761efaa7SDag-Erling Smørgrav #include <stdlib.h>
53761efaa7SDag-Erling Smørgrav #include <stdio.h>
54761efaa7SDag-Erling Smørgrav #include <string.h>
55761efaa7SDag-Erling Smørgrav #include <unistd.h>
56761efaa7SDag-Erling Smørgrav #include <stdarg.h>
571e8db6e2SBrian Feldman 
58d4af9e69SDag-Erling Smørgrav #ifdef HAVE_UTIL_H
59d4af9e69SDag-Erling Smørgrav # include <util.h>
60d4af9e69SDag-Erling Smørgrav #endif
61d4af9e69SDag-Erling Smørgrav 
621e8db6e2SBrian Feldman #include "xmalloc.h"
631e8db6e2SBrian Feldman #include "log.h"
641e8db6e2SBrian Feldman #include "pathnames.h"
65ae1f160dSDag-Erling Smørgrav #include "misc.h"
66076ad2f8SDag-Erling Smørgrav #include "utf8.h"
671e8db6e2SBrian Feldman 
681e8db6e2SBrian Feldman #include "sftp.h"
69bc5531deSDag-Erling Smørgrav #include "ssherr.h"
70bc5531deSDag-Erling Smørgrav #include "sshbuf.h"
711e8db6e2SBrian Feldman #include "sftp-common.h"
721e8db6e2SBrian Feldman #include "sftp-client.h"
73efcad6b7SDag-Erling Smørgrav 
74b15c8340SDag-Erling Smørgrav #define DEFAULT_COPY_BUFLEN	32768	/* Size of buffer for up/download */
7560c59fadSDag-Erling Smørgrav #define DEFAULT_NUM_REQUESTS	64	/* # concurrent outstanding requests */
76b15c8340SDag-Erling Smørgrav 
77efcad6b7SDag-Erling Smørgrav /* File to read commands from */
78efcad6b7SDag-Erling Smørgrav FILE* infile;
79efcad6b7SDag-Erling Smørgrav 
80efcad6b7SDag-Erling Smørgrav /* Are we in batchfile mode? */
81efcad6b7SDag-Erling Smørgrav int batchmode = 0;
82efcad6b7SDag-Erling Smørgrav 
83efcad6b7SDag-Erling Smørgrav /* PID of ssh transport process */
84efcad6b7SDag-Erling Smørgrav static pid_t sshpid = -1;
85efcad6b7SDag-Erling Smørgrav 
86e4a9863fSDag-Erling Smørgrav /* Suppress diagnositic messages */
87e4a9863fSDag-Erling Smørgrav int quiet = 0;
88e4a9863fSDag-Erling Smørgrav 
89efcad6b7SDag-Erling Smørgrav /* This is set to 0 if the progressmeter is not desired. */
9052028650SDag-Erling Smørgrav int showprogress = 1;
91efcad6b7SDag-Erling Smørgrav 
92b15c8340SDag-Erling Smørgrav /* When this option is set, we always recursively download/upload directories */
93b15c8340SDag-Erling Smørgrav int global_rflag = 0;
94b15c8340SDag-Erling Smørgrav 
95a0ee8cc6SDag-Erling Smørgrav /* When this option is set, we resume download or upload if possible */
96e4a9863fSDag-Erling Smørgrav int global_aflag = 0;
97e4a9863fSDag-Erling Smørgrav 
98b15c8340SDag-Erling Smørgrav /* When this option is set, the file transfers will always preserve times */
99b15c8340SDag-Erling Smørgrav int global_pflag = 0;
100b15c8340SDag-Erling Smørgrav 
101f7167e0eSDag-Erling Smørgrav /* When this option is set, transfers will have fsync() called on each file */
102f7167e0eSDag-Erling Smørgrav int global_fflag = 0;
103f7167e0eSDag-Erling Smørgrav 
104d74d50a8SDag-Erling Smørgrav /* SIGINT received during command processing */
105d74d50a8SDag-Erling Smørgrav volatile sig_atomic_t interrupted = 0;
106d74d50a8SDag-Erling Smørgrav 
107d74d50a8SDag-Erling Smørgrav /* I wish qsort() took a separate ctx for the comparison function...*/
108d74d50a8SDag-Erling Smørgrav int sort_flag;
109d74d50a8SDag-Erling Smørgrav 
110b15c8340SDag-Erling Smørgrav /* Context used for commandline completion */
111b15c8340SDag-Erling Smørgrav struct complete_ctx {
112b15c8340SDag-Erling Smørgrav 	struct sftp_conn *conn;
113b15c8340SDag-Erling Smørgrav 	char **remote_pathp;
114b15c8340SDag-Erling Smørgrav };
115b15c8340SDag-Erling Smørgrav 
116efcad6b7SDag-Erling Smørgrav int remote_glob(struct sftp_conn *, const char *, int,
117efcad6b7SDag-Erling Smørgrav     int (*)(const char *, int), glob_t *); /* proto for sftp-glob.c */
1181e8db6e2SBrian Feldman 
11983d2307dSDag-Erling Smørgrav extern char *__progname;
12083d2307dSDag-Erling Smørgrav 
121efcad6b7SDag-Erling Smørgrav /* Separators for interactive commands */
122efcad6b7SDag-Erling Smørgrav #define WHITESPACE " \t\r\n"
1231e8db6e2SBrian Feldman 
124d74d50a8SDag-Erling Smørgrav /* ls flags */
125b15c8340SDag-Erling Smørgrav #define LS_LONG_VIEW	0x0001	/* Full view ala ls -l */
126b15c8340SDag-Erling Smørgrav #define LS_SHORT_VIEW	0x0002	/* Single row view ala ls -1 */
127b15c8340SDag-Erling Smørgrav #define LS_NUMERIC_VIEW	0x0004	/* Long view with numeric uid/gid */
128b15c8340SDag-Erling Smørgrav #define LS_NAME_SORT	0x0008	/* Sort by name (default) */
129b15c8340SDag-Erling Smørgrav #define LS_TIME_SORT	0x0010	/* Sort by mtime */
130b15c8340SDag-Erling Smørgrav #define LS_SIZE_SORT	0x0020	/* Sort by file size */
131b15c8340SDag-Erling Smørgrav #define LS_REVERSE_SORT	0x0040	/* Reverse sort order */
132b15c8340SDag-Erling Smørgrav #define LS_SHOW_ALL	0x0080	/* Don't skip filenames starting with '.' */
133b15c8340SDag-Erling Smørgrav #define LS_SI_UNITS	0x0100	/* Display sizes as K, M, G, etc. */
134d74d50a8SDag-Erling Smørgrav 
135b15c8340SDag-Erling Smørgrav #define VIEW_FLAGS	(LS_LONG_VIEW|LS_SHORT_VIEW|LS_NUMERIC_VIEW|LS_SI_UNITS)
136d74d50a8SDag-Erling Smørgrav #define SORT_FLAGS	(LS_NAME_SORT|LS_TIME_SORT|LS_SIZE_SORT)
137efcad6b7SDag-Erling Smørgrav 
138efcad6b7SDag-Erling Smørgrav /* Commands for interactive mode */
139f7167e0eSDag-Erling Smørgrav enum sftp_command {
140f7167e0eSDag-Erling Smørgrav 	I_CHDIR = 1,
141f7167e0eSDag-Erling Smørgrav 	I_CHGRP,
142f7167e0eSDag-Erling Smørgrav 	I_CHMOD,
143f7167e0eSDag-Erling Smørgrav 	I_CHOWN,
144f7167e0eSDag-Erling Smørgrav 	I_DF,
145f7167e0eSDag-Erling Smørgrav 	I_GET,
146f7167e0eSDag-Erling Smørgrav 	I_HELP,
147f7167e0eSDag-Erling Smørgrav 	I_LCHDIR,
148f7167e0eSDag-Erling Smørgrav 	I_LINK,
149f7167e0eSDag-Erling Smørgrav 	I_LLS,
150f7167e0eSDag-Erling Smørgrav 	I_LMKDIR,
151f7167e0eSDag-Erling Smørgrav 	I_LPWD,
152f7167e0eSDag-Erling Smørgrav 	I_LS,
153f7167e0eSDag-Erling Smørgrav 	I_LUMASK,
154f7167e0eSDag-Erling Smørgrav 	I_MKDIR,
155f7167e0eSDag-Erling Smørgrav 	I_PUT,
156f7167e0eSDag-Erling Smørgrav 	I_PWD,
157f7167e0eSDag-Erling Smørgrav 	I_QUIT,
158a0ee8cc6SDag-Erling Smørgrav 	I_REGET,
159f7167e0eSDag-Erling Smørgrav 	I_RENAME,
160a0ee8cc6SDag-Erling Smørgrav 	I_REPUT,
161f7167e0eSDag-Erling Smørgrav 	I_RM,
162f7167e0eSDag-Erling Smørgrav 	I_RMDIR,
163f7167e0eSDag-Erling Smørgrav 	I_SHELL,
164f7167e0eSDag-Erling Smørgrav 	I_SYMLINK,
165f7167e0eSDag-Erling Smørgrav 	I_VERSION,
166f7167e0eSDag-Erling Smørgrav 	I_PROGRESS,
167f7167e0eSDag-Erling Smørgrav };
168efcad6b7SDag-Erling Smørgrav 
169efcad6b7SDag-Erling Smørgrav struct CMD {
170efcad6b7SDag-Erling Smørgrav 	const char *c;
171efcad6b7SDag-Erling Smørgrav 	const int n;
172b15c8340SDag-Erling Smørgrav 	const int t;
173efcad6b7SDag-Erling Smørgrav };
174efcad6b7SDag-Erling Smørgrav 
175b15c8340SDag-Erling Smørgrav /* Type of completion */
176b15c8340SDag-Erling Smørgrav #define NOARGS	0
177b15c8340SDag-Erling Smørgrav #define REMOTE	1
178b15c8340SDag-Erling Smørgrav #define LOCAL	2
179b15c8340SDag-Erling Smørgrav 
180efcad6b7SDag-Erling Smørgrav static const struct CMD cmds[] = {
181b15c8340SDag-Erling Smørgrav 	{ "bye",	I_QUIT,		NOARGS	},
182b15c8340SDag-Erling Smørgrav 	{ "cd",		I_CHDIR,	REMOTE	},
183b15c8340SDag-Erling Smørgrav 	{ "chdir",	I_CHDIR,	REMOTE	},
184b15c8340SDag-Erling Smørgrav 	{ "chgrp",	I_CHGRP,	REMOTE	},
185b15c8340SDag-Erling Smørgrav 	{ "chmod",	I_CHMOD,	REMOTE	},
186b15c8340SDag-Erling Smørgrav 	{ "chown",	I_CHOWN,	REMOTE	},
187b15c8340SDag-Erling Smørgrav 	{ "df",		I_DF,		REMOTE	},
188b15c8340SDag-Erling Smørgrav 	{ "dir",	I_LS,		REMOTE	},
189b15c8340SDag-Erling Smørgrav 	{ "exit",	I_QUIT,		NOARGS	},
190b15c8340SDag-Erling Smørgrav 	{ "get",	I_GET,		REMOTE	},
191b15c8340SDag-Erling Smørgrav 	{ "help",	I_HELP,		NOARGS	},
192b15c8340SDag-Erling Smørgrav 	{ "lcd",	I_LCHDIR,	LOCAL	},
193b15c8340SDag-Erling Smørgrav 	{ "lchdir",	I_LCHDIR,	LOCAL	},
194b15c8340SDag-Erling Smørgrav 	{ "lls",	I_LLS,		LOCAL	},
195b15c8340SDag-Erling Smørgrav 	{ "lmkdir",	I_LMKDIR,	LOCAL	},
1964a421b63SDag-Erling Smørgrav 	{ "ln",		I_LINK,		REMOTE	},
197b15c8340SDag-Erling Smørgrav 	{ "lpwd",	I_LPWD,		LOCAL	},
198b15c8340SDag-Erling Smørgrav 	{ "ls",		I_LS,		REMOTE	},
199b15c8340SDag-Erling Smørgrav 	{ "lumask",	I_LUMASK,	NOARGS	},
200b15c8340SDag-Erling Smørgrav 	{ "mkdir",	I_MKDIR,	REMOTE	},
201e2f6069cSDag-Erling Smørgrav 	{ "mget",	I_GET,		REMOTE	},
202e2f6069cSDag-Erling Smørgrav 	{ "mput",	I_PUT,		LOCAL	},
203b15c8340SDag-Erling Smørgrav 	{ "progress",	I_PROGRESS,	NOARGS	},
204b15c8340SDag-Erling Smørgrav 	{ "put",	I_PUT,		LOCAL	},
205b15c8340SDag-Erling Smørgrav 	{ "pwd",	I_PWD,		REMOTE	},
206b15c8340SDag-Erling Smørgrav 	{ "quit",	I_QUIT,		NOARGS	},
207e4a9863fSDag-Erling Smørgrav 	{ "reget",	I_REGET,	REMOTE	},
208b15c8340SDag-Erling Smørgrav 	{ "rename",	I_RENAME,	REMOTE	},
209a0ee8cc6SDag-Erling Smørgrav 	{ "reput",	I_REPUT,	LOCAL	},
210b15c8340SDag-Erling Smørgrav 	{ "rm",		I_RM,		REMOTE	},
211b15c8340SDag-Erling Smørgrav 	{ "rmdir",	I_RMDIR,	REMOTE	},
212b15c8340SDag-Erling Smørgrav 	{ "symlink",	I_SYMLINK,	REMOTE	},
213b15c8340SDag-Erling Smørgrav 	{ "version",	I_VERSION,	NOARGS	},
214b15c8340SDag-Erling Smørgrav 	{ "!",		I_SHELL,	NOARGS	},
215b15c8340SDag-Erling Smørgrav 	{ "?",		I_HELP,		NOARGS	},
216b15c8340SDag-Erling Smørgrav 	{ NULL,		-1,		-1	}
217efcad6b7SDag-Erling Smørgrav };
218efcad6b7SDag-Erling Smørgrav 
219b15c8340SDag-Erling Smørgrav int interactive_loop(struct sftp_conn *, char *file1, char *file2);
220efcad6b7SDag-Erling Smørgrav 
221d4af9e69SDag-Erling Smørgrav /* ARGSUSED */
222efcad6b7SDag-Erling Smørgrav static void
223d74d50a8SDag-Erling Smørgrav killchild(int signo)
224d74d50a8SDag-Erling Smørgrav {
2255e8dbd04SDag-Erling Smørgrav 	if (sshpid > 1) {
226d74d50a8SDag-Erling Smørgrav 		kill(sshpid, SIGTERM);
2275e8dbd04SDag-Erling Smørgrav 		waitpid(sshpid, NULL, 0);
2285e8dbd04SDag-Erling Smørgrav 	}
229d74d50a8SDag-Erling Smørgrav 
230d74d50a8SDag-Erling Smørgrav 	_exit(1);
231d74d50a8SDag-Erling Smørgrav }
232d74d50a8SDag-Erling Smørgrav 
233d4af9e69SDag-Erling Smørgrav /* ARGSUSED */
234d74d50a8SDag-Erling Smørgrav static void
235*ca86bcf2SDag-Erling Smørgrav suspchild(int signo)
236*ca86bcf2SDag-Erling Smørgrav {
237*ca86bcf2SDag-Erling Smørgrav 	if (sshpid > 1) {
238*ca86bcf2SDag-Erling Smørgrav 		kill(sshpid, signo);
239*ca86bcf2SDag-Erling Smørgrav 		while (waitpid(sshpid, NULL, WUNTRACED) == -1 && errno == EINTR)
240*ca86bcf2SDag-Erling Smørgrav 			continue;
241*ca86bcf2SDag-Erling Smørgrav 	}
242*ca86bcf2SDag-Erling Smørgrav 	kill(getpid(), SIGSTOP);
243*ca86bcf2SDag-Erling Smørgrav }
244*ca86bcf2SDag-Erling Smørgrav 
245*ca86bcf2SDag-Erling Smørgrav /* ARGSUSED */
246*ca86bcf2SDag-Erling Smørgrav static void
247d74d50a8SDag-Erling Smørgrav cmd_interrupt(int signo)
248d74d50a8SDag-Erling Smørgrav {
249d74d50a8SDag-Erling Smørgrav 	const char msg[] = "\rInterrupt  \n";
2505e8dbd04SDag-Erling Smørgrav 	int olderrno = errno;
251d74d50a8SDag-Erling Smørgrav 
252e4a9863fSDag-Erling Smørgrav 	(void)write(STDERR_FILENO, msg, sizeof(msg) - 1);
253d74d50a8SDag-Erling Smørgrav 	interrupted = 1;
2545e8dbd04SDag-Erling Smørgrav 	errno = olderrno;
255d74d50a8SDag-Erling Smørgrav }
256d74d50a8SDag-Erling Smørgrav 
257d74d50a8SDag-Erling Smørgrav static void
258efcad6b7SDag-Erling Smørgrav help(void)
259efcad6b7SDag-Erling Smørgrav {
260cce7d346SDag-Erling Smørgrav 	printf("Available commands:\n"
261cce7d346SDag-Erling Smørgrav 	    "bye                                Quit sftp\n"
262cce7d346SDag-Erling Smørgrav 	    "cd path                            Change remote directory to 'path'\n"
263cce7d346SDag-Erling Smørgrav 	    "chgrp grp path                     Change group of file 'path' to 'grp'\n"
264cce7d346SDag-Erling Smørgrav 	    "chmod mode path                    Change permissions of file 'path' to 'mode'\n"
265cce7d346SDag-Erling Smørgrav 	    "chown own path                     Change owner of file 'path' to 'own'\n"
266cce7d346SDag-Erling Smørgrav 	    "df [-hi] [path]                    Display statistics for current directory or\n"
267cce7d346SDag-Erling Smørgrav 	    "                                   filesystem containing 'path'\n"
268cce7d346SDag-Erling Smørgrav 	    "exit                               Quit sftp\n"
269bc5531deSDag-Erling Smørgrav 	    "get [-afPpRr] remote [local]       Download file\n"
270bc5531deSDag-Erling Smørgrav 	    "reget [-fPpRr] remote [local]      Resume download file\n"
271bc5531deSDag-Erling Smørgrav 	    "reput [-fPpRr] [local] remote      Resume upload file\n"
272cce7d346SDag-Erling Smørgrav 	    "help                               Display this help text\n"
273cce7d346SDag-Erling Smørgrav 	    "lcd path                           Change local directory to 'path'\n"
274cce7d346SDag-Erling Smørgrav 	    "lls [ls-options [path]]            Display local directory listing\n"
275cce7d346SDag-Erling Smørgrav 	    "lmkdir path                        Create local directory\n"
2764a421b63SDag-Erling Smørgrav 	    "ln [-s] oldpath newpath            Link remote file (-s for symlink)\n"
277cce7d346SDag-Erling Smørgrav 	    "lpwd                               Print local working directory\n"
278b15c8340SDag-Erling Smørgrav 	    "ls [-1afhlnrSt] [path]             Display remote directory listing\n"
279cce7d346SDag-Erling Smørgrav 	    "lumask umask                       Set local umask to 'umask'\n"
280cce7d346SDag-Erling Smørgrav 	    "mkdir path                         Create remote directory\n"
281cce7d346SDag-Erling Smørgrav 	    "progress                           Toggle display of progress meter\n"
282bc5531deSDag-Erling Smørgrav 	    "put [-afPpRr] local [remote]       Upload file\n"
283cce7d346SDag-Erling Smørgrav 	    "pwd                                Display remote working directory\n"
284cce7d346SDag-Erling Smørgrav 	    "quit                               Quit sftp\n"
285cce7d346SDag-Erling Smørgrav 	    "rename oldpath newpath             Rename remote file\n"
286cce7d346SDag-Erling Smørgrav 	    "rm path                            Delete remote file\n"
287cce7d346SDag-Erling Smørgrav 	    "rmdir path                         Remove remote directory\n"
288cce7d346SDag-Erling Smørgrav 	    "symlink oldpath newpath            Symlink remote file\n"
289cce7d346SDag-Erling Smørgrav 	    "version                            Show SFTP version\n"
290cce7d346SDag-Erling Smørgrav 	    "!command                           Execute 'command' in local shell\n"
291cce7d346SDag-Erling Smørgrav 	    "!                                  Escape to local shell\n"
292cce7d346SDag-Erling Smørgrav 	    "?                                  Synonym for help\n");
293efcad6b7SDag-Erling Smørgrav }
294efcad6b7SDag-Erling Smørgrav 
295efcad6b7SDag-Erling Smørgrav static void
296efcad6b7SDag-Erling Smørgrav local_do_shell(const char *args)
297efcad6b7SDag-Erling Smørgrav {
298efcad6b7SDag-Erling Smørgrav 	int status;
299efcad6b7SDag-Erling Smørgrav 	char *shell;
300efcad6b7SDag-Erling Smørgrav 	pid_t pid;
301efcad6b7SDag-Erling Smørgrav 
302efcad6b7SDag-Erling Smørgrav 	if (!*args)
303efcad6b7SDag-Erling Smørgrav 		args = NULL;
304efcad6b7SDag-Erling Smørgrav 
3054a421b63SDag-Erling Smørgrav 	if ((shell = getenv("SHELL")) == NULL || *shell == '\0')
306efcad6b7SDag-Erling Smørgrav 		shell = _PATH_BSHELL;
307efcad6b7SDag-Erling Smørgrav 
308efcad6b7SDag-Erling Smørgrav 	if ((pid = fork()) == -1)
309efcad6b7SDag-Erling Smørgrav 		fatal("Couldn't fork: %s", strerror(errno));
310efcad6b7SDag-Erling Smørgrav 
311efcad6b7SDag-Erling Smørgrav 	if (pid == 0) {
312efcad6b7SDag-Erling Smørgrav 		/* XXX: child has pipe fds to ssh subproc open - issue? */
313efcad6b7SDag-Erling Smørgrav 		if (args) {
314efcad6b7SDag-Erling Smørgrav 			debug3("Executing %s -c \"%s\"", shell, args);
315efcad6b7SDag-Erling Smørgrav 			execl(shell, shell, "-c", args, (char *)NULL);
316efcad6b7SDag-Erling Smørgrav 		} else {
317efcad6b7SDag-Erling Smørgrav 			debug3("Executing %s", shell);
318efcad6b7SDag-Erling Smørgrav 			execl(shell, shell, (char *)NULL);
319efcad6b7SDag-Erling Smørgrav 		}
320efcad6b7SDag-Erling Smørgrav 		fprintf(stderr, "Couldn't execute \"%s\": %s\n", shell,
321efcad6b7SDag-Erling Smørgrav 		    strerror(errno));
322efcad6b7SDag-Erling Smørgrav 		_exit(1);
323efcad6b7SDag-Erling Smørgrav 	}
324efcad6b7SDag-Erling Smørgrav 	while (waitpid(pid, &status, 0) == -1)
325efcad6b7SDag-Erling Smørgrav 		if (errno != EINTR)
326efcad6b7SDag-Erling Smørgrav 			fatal("Couldn't wait for child: %s", strerror(errno));
327efcad6b7SDag-Erling Smørgrav 	if (!WIFEXITED(status))
328761efaa7SDag-Erling Smørgrav 		error("Shell exited abnormally");
329efcad6b7SDag-Erling Smørgrav 	else if (WEXITSTATUS(status))
330efcad6b7SDag-Erling Smørgrav 		error("Shell exited with status %d", WEXITSTATUS(status));
331efcad6b7SDag-Erling Smørgrav }
332efcad6b7SDag-Erling Smørgrav 
333efcad6b7SDag-Erling Smørgrav static void
334efcad6b7SDag-Erling Smørgrav local_do_ls(const char *args)
335efcad6b7SDag-Erling Smørgrav {
336efcad6b7SDag-Erling Smørgrav 	if (!args || !*args)
337efcad6b7SDag-Erling Smørgrav 		local_do_shell(_PATH_LS);
338efcad6b7SDag-Erling Smørgrav 	else {
339efcad6b7SDag-Erling Smørgrav 		int len = strlen(_PATH_LS " ") + strlen(args) + 1;
340efcad6b7SDag-Erling Smørgrav 		char *buf = xmalloc(len);
341efcad6b7SDag-Erling Smørgrav 
342efcad6b7SDag-Erling Smørgrav 		/* XXX: quoting - rip quoting code from ftp? */
343efcad6b7SDag-Erling Smørgrav 		snprintf(buf, len, _PATH_LS " %s", args);
344efcad6b7SDag-Erling Smørgrav 		local_do_shell(buf);
345e4a9863fSDag-Erling Smørgrav 		free(buf);
346efcad6b7SDag-Erling Smørgrav 	}
347efcad6b7SDag-Erling Smørgrav }
348efcad6b7SDag-Erling Smørgrav 
349efcad6b7SDag-Erling Smørgrav /* Strip one path (usually the pwd) from the start of another */
350efcad6b7SDag-Erling Smørgrav static char *
351076ad2f8SDag-Erling Smørgrav path_strip(const char *path, const char *strip)
352efcad6b7SDag-Erling Smørgrav {
353efcad6b7SDag-Erling Smørgrav 	size_t len;
354efcad6b7SDag-Erling Smørgrav 
355efcad6b7SDag-Erling Smørgrav 	if (strip == NULL)
356efcad6b7SDag-Erling Smørgrav 		return (xstrdup(path));
357efcad6b7SDag-Erling Smørgrav 
358efcad6b7SDag-Erling Smørgrav 	len = strlen(strip);
3595e8dbd04SDag-Erling Smørgrav 	if (strncmp(path, strip, len) == 0) {
360efcad6b7SDag-Erling Smørgrav 		if (strip[len - 1] != '/' && path[len] == '/')
361efcad6b7SDag-Erling Smørgrav 			len++;
362efcad6b7SDag-Erling Smørgrav 		return (xstrdup(path + len));
363efcad6b7SDag-Erling Smørgrav 	}
364efcad6b7SDag-Erling Smørgrav 
365efcad6b7SDag-Erling Smørgrav 	return (xstrdup(path));
366efcad6b7SDag-Erling Smørgrav }
367efcad6b7SDag-Erling Smørgrav 
368efcad6b7SDag-Erling Smørgrav static char *
369076ad2f8SDag-Erling Smørgrav make_absolute(char *p, const char *pwd)
370efcad6b7SDag-Erling Smørgrav {
371d74d50a8SDag-Erling Smørgrav 	char *abs_str;
372efcad6b7SDag-Erling Smørgrav 
373efcad6b7SDag-Erling Smørgrav 	/* Derelativise */
374efcad6b7SDag-Erling Smørgrav 	if (p && p[0] != '/') {
375d74d50a8SDag-Erling Smørgrav 		abs_str = path_append(pwd, p);
376e4a9863fSDag-Erling Smørgrav 		free(p);
377d74d50a8SDag-Erling Smørgrav 		return(abs_str);
378efcad6b7SDag-Erling Smørgrav 	} else
379efcad6b7SDag-Erling Smørgrav 		return(p);
380efcad6b7SDag-Erling Smørgrav }
381efcad6b7SDag-Erling Smørgrav 
382efcad6b7SDag-Erling Smørgrav static int
383e4a9863fSDag-Erling Smørgrav parse_getput_flags(const char *cmd, char **argv, int argc,
384f7167e0eSDag-Erling Smørgrav     int *aflag, int *fflag, int *pflag, int *rflag)
385efcad6b7SDag-Erling Smørgrav {
386d4af9e69SDag-Erling Smørgrav 	extern int opterr, optind, optopt, optreset;
387d4af9e69SDag-Erling Smørgrav 	int ch;
388efcad6b7SDag-Erling Smørgrav 
389d4af9e69SDag-Erling Smørgrav 	optind = optreset = 1;
390d4af9e69SDag-Erling Smørgrav 	opterr = 0;
391d4af9e69SDag-Erling Smørgrav 
392f7167e0eSDag-Erling Smørgrav 	*aflag = *fflag = *rflag = *pflag = 0;
393f7167e0eSDag-Erling Smørgrav 	while ((ch = getopt(argc, argv, "afPpRr")) != -1) {
394d4af9e69SDag-Erling Smørgrav 		switch (ch) {
395e4a9863fSDag-Erling Smørgrav 		case 'a':
396e4a9863fSDag-Erling Smørgrav 			*aflag = 1;
397e4a9863fSDag-Erling Smørgrav 			break;
398f7167e0eSDag-Erling Smørgrav 		case 'f':
399f7167e0eSDag-Erling Smørgrav 			*fflag = 1;
400f7167e0eSDag-Erling Smørgrav 			break;
401efcad6b7SDag-Erling Smørgrav 		case 'p':
402efcad6b7SDag-Erling Smørgrav 		case 'P':
403efcad6b7SDag-Erling Smørgrav 			*pflag = 1;
404efcad6b7SDag-Erling Smørgrav 			break;
405b15c8340SDag-Erling Smørgrav 		case 'r':
406b15c8340SDag-Erling Smørgrav 		case 'R':
407b15c8340SDag-Erling Smørgrav 			*rflag = 1;
408b15c8340SDag-Erling Smørgrav 			break;
409efcad6b7SDag-Erling Smørgrav 		default:
410d4af9e69SDag-Erling Smørgrav 			error("%s: Invalid flag -%c", cmd, optopt);
411d4af9e69SDag-Erling Smørgrav 			return -1;
412efcad6b7SDag-Erling Smørgrav 		}
413efcad6b7SDag-Erling Smørgrav 	}
414efcad6b7SDag-Erling Smørgrav 
415d4af9e69SDag-Erling Smørgrav 	return optind;
416efcad6b7SDag-Erling Smørgrav }
417efcad6b7SDag-Erling Smørgrav 
418efcad6b7SDag-Erling Smørgrav static int
4194a421b63SDag-Erling Smørgrav parse_link_flags(const char *cmd, char **argv, int argc, int *sflag)
4204a421b63SDag-Erling Smørgrav {
4214a421b63SDag-Erling Smørgrav 	extern int opterr, optind, optopt, optreset;
4224a421b63SDag-Erling Smørgrav 	int ch;
4234a421b63SDag-Erling Smørgrav 
4244a421b63SDag-Erling Smørgrav 	optind = optreset = 1;
4254a421b63SDag-Erling Smørgrav 	opterr = 0;
4264a421b63SDag-Erling Smørgrav 
4274a421b63SDag-Erling Smørgrav 	*sflag = 0;
4284a421b63SDag-Erling Smørgrav 	while ((ch = getopt(argc, argv, "s")) != -1) {
4294a421b63SDag-Erling Smørgrav 		switch (ch) {
4304a421b63SDag-Erling Smørgrav 		case 's':
4314a421b63SDag-Erling Smørgrav 			*sflag = 1;
4324a421b63SDag-Erling Smørgrav 			break;
4334a421b63SDag-Erling Smørgrav 		default:
4344a421b63SDag-Erling Smørgrav 			error("%s: Invalid flag -%c", cmd, optopt);
4354a421b63SDag-Erling Smørgrav 			return -1;
4364a421b63SDag-Erling Smørgrav 		}
4374a421b63SDag-Erling Smørgrav 	}
4384a421b63SDag-Erling Smørgrav 
4394a421b63SDag-Erling Smørgrav 	return optind;
4404a421b63SDag-Erling Smørgrav }
4414a421b63SDag-Erling Smørgrav 
4424a421b63SDag-Erling Smørgrav static int
443f7167e0eSDag-Erling Smørgrav parse_rename_flags(const char *cmd, char **argv, int argc, int *lflag)
444f7167e0eSDag-Erling Smørgrav {
445f7167e0eSDag-Erling Smørgrav 	extern int opterr, optind, optopt, optreset;
446f7167e0eSDag-Erling Smørgrav 	int ch;
447f7167e0eSDag-Erling Smørgrav 
448f7167e0eSDag-Erling Smørgrav 	optind = optreset = 1;
449f7167e0eSDag-Erling Smørgrav 	opterr = 0;
450f7167e0eSDag-Erling Smørgrav 
451f7167e0eSDag-Erling Smørgrav 	*lflag = 0;
452f7167e0eSDag-Erling Smørgrav 	while ((ch = getopt(argc, argv, "l")) != -1) {
453f7167e0eSDag-Erling Smørgrav 		switch (ch) {
454f7167e0eSDag-Erling Smørgrav 		case 'l':
455f7167e0eSDag-Erling Smørgrav 			*lflag = 1;
456f7167e0eSDag-Erling Smørgrav 			break;
457f7167e0eSDag-Erling Smørgrav 		default:
458f7167e0eSDag-Erling Smørgrav 			error("%s: Invalid flag -%c", cmd, optopt);
459f7167e0eSDag-Erling Smørgrav 			return -1;
460f7167e0eSDag-Erling Smørgrav 		}
461f7167e0eSDag-Erling Smørgrav 	}
462f7167e0eSDag-Erling Smørgrav 
463f7167e0eSDag-Erling Smørgrav 	return optind;
464f7167e0eSDag-Erling Smørgrav }
465f7167e0eSDag-Erling Smørgrav 
466f7167e0eSDag-Erling Smørgrav static int
467d4af9e69SDag-Erling Smørgrav parse_ls_flags(char **argv, int argc, int *lflag)
468efcad6b7SDag-Erling Smørgrav {
469d4af9e69SDag-Erling Smørgrav 	extern int opterr, optind, optopt, optreset;
470d4af9e69SDag-Erling Smørgrav 	int ch;
471efcad6b7SDag-Erling Smørgrav 
472d4af9e69SDag-Erling Smørgrav 	optind = optreset = 1;
473d4af9e69SDag-Erling Smørgrav 	opterr = 0;
474d4af9e69SDag-Erling Smørgrav 
475d74d50a8SDag-Erling Smørgrav 	*lflag = LS_NAME_SORT;
476b15c8340SDag-Erling Smørgrav 	while ((ch = getopt(argc, argv, "1Safhlnrt")) != -1) {
477d4af9e69SDag-Erling Smørgrav 		switch (ch) {
478efcad6b7SDag-Erling Smørgrav 		case '1':
479d74d50a8SDag-Erling Smørgrav 			*lflag &= ~VIEW_FLAGS;
480d74d50a8SDag-Erling Smørgrav 			*lflag |= LS_SHORT_VIEW;
481d74d50a8SDag-Erling Smørgrav 			break;
482d74d50a8SDag-Erling Smørgrav 		case 'S':
483d74d50a8SDag-Erling Smørgrav 			*lflag &= ~SORT_FLAGS;
484d74d50a8SDag-Erling Smørgrav 			*lflag |= LS_SIZE_SORT;
485d74d50a8SDag-Erling Smørgrav 			break;
486d4af9e69SDag-Erling Smørgrav 		case 'a':
487d4af9e69SDag-Erling Smørgrav 			*lflag |= LS_SHOW_ALL;
488d74d50a8SDag-Erling Smørgrav 			break;
489d74d50a8SDag-Erling Smørgrav 		case 'f':
490d74d50a8SDag-Erling Smørgrav 			*lflag &= ~SORT_FLAGS;
491d74d50a8SDag-Erling Smørgrav 			break;
492b15c8340SDag-Erling Smørgrav 		case 'h':
493b15c8340SDag-Erling Smørgrav 			*lflag |= LS_SI_UNITS;
494b15c8340SDag-Erling Smørgrav 			break;
495d4af9e69SDag-Erling Smørgrav 		case 'l':
496b15c8340SDag-Erling Smørgrav 			*lflag &= ~LS_SHORT_VIEW;
497d4af9e69SDag-Erling Smørgrav 			*lflag |= LS_LONG_VIEW;
498d4af9e69SDag-Erling Smørgrav 			break;
499d4af9e69SDag-Erling Smørgrav 		case 'n':
500b15c8340SDag-Erling Smørgrav 			*lflag &= ~LS_SHORT_VIEW;
501d4af9e69SDag-Erling Smørgrav 			*lflag |= LS_NUMERIC_VIEW|LS_LONG_VIEW;
502d4af9e69SDag-Erling Smørgrav 			break;
503d4af9e69SDag-Erling Smørgrav 		case 'r':
504d4af9e69SDag-Erling Smørgrav 			*lflag |= LS_REVERSE_SORT;
505d4af9e69SDag-Erling Smørgrav 			break;
506d4af9e69SDag-Erling Smørgrav 		case 't':
507d4af9e69SDag-Erling Smørgrav 			*lflag &= ~SORT_FLAGS;
508d4af9e69SDag-Erling Smørgrav 			*lflag |= LS_TIME_SORT;
509efcad6b7SDag-Erling Smørgrav 			break;
510efcad6b7SDag-Erling Smørgrav 		default:
511d4af9e69SDag-Erling Smørgrav 			error("ls: Invalid flag -%c", optopt);
512d4af9e69SDag-Erling Smørgrav 			return -1;
513efcad6b7SDag-Erling Smørgrav 		}
514efcad6b7SDag-Erling Smørgrav 	}
515efcad6b7SDag-Erling Smørgrav 
516d4af9e69SDag-Erling Smørgrav 	return optind;
517efcad6b7SDag-Erling Smørgrav }
518efcad6b7SDag-Erling Smørgrav 
519efcad6b7SDag-Erling Smørgrav static int
520d4af9e69SDag-Erling Smørgrav parse_df_flags(const char *cmd, char **argv, int argc, int *hflag, int *iflag)
521efcad6b7SDag-Erling Smørgrav {
522d4af9e69SDag-Erling Smørgrav 	extern int opterr, optind, optopt, optreset;
523d4af9e69SDag-Erling Smørgrav 	int ch;
524efcad6b7SDag-Erling Smørgrav 
525d4af9e69SDag-Erling Smørgrav 	optind = optreset = 1;
526d4af9e69SDag-Erling Smørgrav 	opterr = 0;
527efcad6b7SDag-Erling Smørgrav 
528d4af9e69SDag-Erling Smørgrav 	*hflag = *iflag = 0;
529d4af9e69SDag-Erling Smørgrav 	while ((ch = getopt(argc, argv, "hi")) != -1) {
530d4af9e69SDag-Erling Smørgrav 		switch (ch) {
531d4af9e69SDag-Erling Smørgrav 		case 'h':
532d4af9e69SDag-Erling Smørgrav 			*hflag = 1;
533efcad6b7SDag-Erling Smørgrav 			break;
534d4af9e69SDag-Erling Smørgrav 		case 'i':
535d4af9e69SDag-Erling Smørgrav 			*iflag = 1;
536d4af9e69SDag-Erling Smørgrav 			break;
537d4af9e69SDag-Erling Smørgrav 		default:
538d4af9e69SDag-Erling Smørgrav 			error("%s: Invalid flag -%c", cmd, optopt);
539d4af9e69SDag-Erling Smørgrav 			return -1;
540efcad6b7SDag-Erling Smørgrav 		}
541efcad6b7SDag-Erling Smørgrav 	}
542efcad6b7SDag-Erling Smørgrav 
543d4af9e69SDag-Erling Smørgrav 	return optind;
544efcad6b7SDag-Erling Smørgrav }
545efcad6b7SDag-Erling Smørgrav 
546efcad6b7SDag-Erling Smørgrav static int
547f7167e0eSDag-Erling Smørgrav parse_no_flags(const char *cmd, char **argv, int argc)
548f7167e0eSDag-Erling Smørgrav {
549f7167e0eSDag-Erling Smørgrav 	extern int opterr, optind, optopt, optreset;
550f7167e0eSDag-Erling Smørgrav 	int ch;
551f7167e0eSDag-Erling Smørgrav 
552f7167e0eSDag-Erling Smørgrav 	optind = optreset = 1;
553f7167e0eSDag-Erling Smørgrav 	opterr = 0;
554f7167e0eSDag-Erling Smørgrav 
555f7167e0eSDag-Erling Smørgrav 	while ((ch = getopt(argc, argv, "")) != -1) {
556f7167e0eSDag-Erling Smørgrav 		switch (ch) {
557f7167e0eSDag-Erling Smørgrav 		default:
558f7167e0eSDag-Erling Smørgrav 			error("%s: Invalid flag -%c", cmd, optopt);
559f7167e0eSDag-Erling Smørgrav 			return -1;
560f7167e0eSDag-Erling Smørgrav 		}
561f7167e0eSDag-Erling Smørgrav 	}
562f7167e0eSDag-Erling Smørgrav 
563f7167e0eSDag-Erling Smørgrav 	return optind;
564f7167e0eSDag-Erling Smørgrav }
565f7167e0eSDag-Erling Smørgrav 
566f7167e0eSDag-Erling Smørgrav static int
567076ad2f8SDag-Erling Smørgrav is_dir(const char *path)
568efcad6b7SDag-Erling Smørgrav {
569efcad6b7SDag-Erling Smørgrav 	struct stat sb;
570efcad6b7SDag-Erling Smørgrav 
571efcad6b7SDag-Erling Smørgrav 	/* XXX: report errors? */
572efcad6b7SDag-Erling Smørgrav 	if (stat(path, &sb) == -1)
573efcad6b7SDag-Erling Smørgrav 		return(0);
574efcad6b7SDag-Erling Smørgrav 
575761efaa7SDag-Erling Smørgrav 	return(S_ISDIR(sb.st_mode));
576efcad6b7SDag-Erling Smørgrav }
577efcad6b7SDag-Erling Smørgrav 
578efcad6b7SDag-Erling Smørgrav static int
579076ad2f8SDag-Erling Smørgrav remote_is_dir(struct sftp_conn *conn, const char *path)
580efcad6b7SDag-Erling Smørgrav {
581efcad6b7SDag-Erling Smørgrav 	Attrib *a;
582efcad6b7SDag-Erling Smørgrav 
583efcad6b7SDag-Erling Smørgrav 	/* XXX: report errors? */
584efcad6b7SDag-Erling Smørgrav 	if ((a = do_stat(conn, path, 1)) == NULL)
585efcad6b7SDag-Erling Smørgrav 		return(0);
586efcad6b7SDag-Erling Smørgrav 	if (!(a->flags & SSH2_FILEXFER_ATTR_PERMISSIONS))
587efcad6b7SDag-Erling Smørgrav 		return(0);
588761efaa7SDag-Erling Smørgrav 	return(S_ISDIR(a->perm));
589efcad6b7SDag-Erling Smørgrav }
590efcad6b7SDag-Erling Smørgrav 
591b15c8340SDag-Erling Smørgrav /* Check whether path returned from glob(..., GLOB_MARK, ...) is a directory */
592efcad6b7SDag-Erling Smørgrav static int
593076ad2f8SDag-Erling Smørgrav pathname_is_dir(const char *pathname)
594b15c8340SDag-Erling Smørgrav {
595b15c8340SDag-Erling Smørgrav 	size_t l = strlen(pathname);
596b15c8340SDag-Erling Smørgrav 
597b15c8340SDag-Erling Smørgrav 	return l > 0 && pathname[l - 1] == '/';
598b15c8340SDag-Erling Smørgrav }
599b15c8340SDag-Erling Smørgrav 
600b15c8340SDag-Erling Smørgrav static int
601076ad2f8SDag-Erling Smørgrav process_get(struct sftp_conn *conn, const char *src, const char *dst,
602076ad2f8SDag-Erling Smørgrav     const char *pwd, int pflag, int rflag, int resume, int fflag)
603efcad6b7SDag-Erling Smørgrav {
604efcad6b7SDag-Erling Smørgrav 	char *abs_src = NULL;
605efcad6b7SDag-Erling Smørgrav 	char *abs_dst = NULL;
606efcad6b7SDag-Erling Smørgrav 	glob_t g;
607b15c8340SDag-Erling Smørgrav 	char *filename, *tmp=NULL;
608a0ee8cc6SDag-Erling Smørgrav 	int i, r, err = 0;
609efcad6b7SDag-Erling Smørgrav 
610efcad6b7SDag-Erling Smørgrav 	abs_src = xstrdup(src);
611efcad6b7SDag-Erling Smørgrav 	abs_src = make_absolute(abs_src, pwd);
612efcad6b7SDag-Erling Smørgrav 	memset(&g, 0, sizeof(g));
613b15c8340SDag-Erling Smørgrav 
614efcad6b7SDag-Erling Smørgrav 	debug3("Looking up %s", abs_src);
615a0ee8cc6SDag-Erling Smørgrav 	if ((r = remote_glob(conn, abs_src, GLOB_MARK, NULL, &g)) != 0) {
616a0ee8cc6SDag-Erling Smørgrav 		if (r == GLOB_NOSPACE) {
617a0ee8cc6SDag-Erling Smørgrav 			error("Too many matches for \"%s\".", abs_src);
618a0ee8cc6SDag-Erling Smørgrav 		} else {
619efcad6b7SDag-Erling Smørgrav 			error("File \"%s\" not found.", abs_src);
620a0ee8cc6SDag-Erling Smørgrav 		}
621efcad6b7SDag-Erling Smørgrav 		err = -1;
622efcad6b7SDag-Erling Smørgrav 		goto out;
623efcad6b7SDag-Erling Smørgrav 	}
624efcad6b7SDag-Erling Smørgrav 
625b15c8340SDag-Erling Smørgrav 	/*
626b15c8340SDag-Erling Smørgrav 	 * If multiple matches then dst must be a directory or
627b15c8340SDag-Erling Smørgrav 	 * unspecified.
628b15c8340SDag-Erling Smørgrav 	 */
629b15c8340SDag-Erling Smørgrav 	if (g.gl_matchc > 1 && dst != NULL && !is_dir(dst)) {
630b15c8340SDag-Erling Smørgrav 		error("Multiple source paths, but destination "
631b15c8340SDag-Erling Smørgrav 		    "\"%s\" is not a directory", dst);
632efcad6b7SDag-Erling Smørgrav 		err = -1;
633efcad6b7SDag-Erling Smørgrav 		goto out;
634efcad6b7SDag-Erling Smørgrav 	}
635efcad6b7SDag-Erling Smørgrav 
636d74d50a8SDag-Erling Smørgrav 	for (i = 0; g.gl_pathv[i] && !interrupted; i++) {
637b15c8340SDag-Erling Smørgrav 		tmp = xstrdup(g.gl_pathv[i]);
638b15c8340SDag-Erling Smørgrav 		if ((filename = basename(tmp)) == NULL) {
639b15c8340SDag-Erling Smørgrav 			error("basename %s: %s", tmp, strerror(errno));
640e4a9863fSDag-Erling Smørgrav 			free(tmp);
641efcad6b7SDag-Erling Smørgrav 			err = -1;
642efcad6b7SDag-Erling Smørgrav 			goto out;
643efcad6b7SDag-Erling Smørgrav 		}
644efcad6b7SDag-Erling Smørgrav 
645efcad6b7SDag-Erling Smørgrav 		if (g.gl_matchc == 1 && dst) {
646efcad6b7SDag-Erling Smørgrav 			if (is_dir(dst)) {
647b15c8340SDag-Erling Smørgrav 				abs_dst = path_append(dst, filename);
648b15c8340SDag-Erling Smørgrav 			} else {
649efcad6b7SDag-Erling Smørgrav 				abs_dst = xstrdup(dst);
650b15c8340SDag-Erling Smørgrav 			}
651efcad6b7SDag-Erling Smørgrav 		} else if (dst) {
652b15c8340SDag-Erling Smørgrav 			abs_dst = path_append(dst, filename);
653b15c8340SDag-Erling Smørgrav 		} else {
654b15c8340SDag-Erling Smørgrav 			abs_dst = xstrdup(filename);
655b15c8340SDag-Erling Smørgrav 		}
656e4a9863fSDag-Erling Smørgrav 		free(tmp);
657efcad6b7SDag-Erling Smørgrav 
658e4a9863fSDag-Erling Smørgrav 		resume |= global_aflag;
659e4a9863fSDag-Erling Smørgrav 		if (!quiet && resume)
660076ad2f8SDag-Erling Smørgrav 			mprintf("Resuming %s to %s\n",
661076ad2f8SDag-Erling Smørgrav 			    g.gl_pathv[i], abs_dst);
662e4a9863fSDag-Erling Smørgrav 		else if (!quiet && !resume)
663076ad2f8SDag-Erling Smørgrav 			mprintf("Fetching %s to %s\n",
664076ad2f8SDag-Erling Smørgrav 			    g.gl_pathv[i], abs_dst);
665b15c8340SDag-Erling Smørgrav 		if (pathname_is_dir(g.gl_pathv[i]) && (rflag || global_rflag)) {
666b15c8340SDag-Erling Smørgrav 			if (download_dir(conn, g.gl_pathv[i], abs_dst, NULL,
667f7167e0eSDag-Erling Smørgrav 			    pflag || global_pflag, 1, resume,
668f7167e0eSDag-Erling Smørgrav 			    fflag || global_fflag) == -1)
669efcad6b7SDag-Erling Smørgrav 				err = -1;
670b15c8340SDag-Erling Smørgrav 		} else {
671b15c8340SDag-Erling Smørgrav 			if (do_download(conn, g.gl_pathv[i], abs_dst, NULL,
672f7167e0eSDag-Erling Smørgrav 			    pflag || global_pflag, resume,
673f7167e0eSDag-Erling Smørgrav 			    fflag || global_fflag) == -1)
674b15c8340SDag-Erling Smørgrav 				err = -1;
675b15c8340SDag-Erling Smørgrav 		}
676e4a9863fSDag-Erling Smørgrav 		free(abs_dst);
677efcad6b7SDag-Erling Smørgrav 		abs_dst = NULL;
678efcad6b7SDag-Erling Smørgrav 	}
679efcad6b7SDag-Erling Smørgrav 
680efcad6b7SDag-Erling Smørgrav out:
681e4a9863fSDag-Erling Smørgrav 	free(abs_src);
682efcad6b7SDag-Erling Smørgrav 	globfree(&g);
683efcad6b7SDag-Erling Smørgrav 	return(err);
684efcad6b7SDag-Erling Smørgrav }
685efcad6b7SDag-Erling Smørgrav 
686efcad6b7SDag-Erling Smørgrav static int
687076ad2f8SDag-Erling Smørgrav process_put(struct sftp_conn *conn, const char *src, const char *dst,
688076ad2f8SDag-Erling Smørgrav     const char *pwd, int pflag, int rflag, int resume, int fflag)
689efcad6b7SDag-Erling Smørgrav {
690efcad6b7SDag-Erling Smørgrav 	char *tmp_dst = NULL;
691efcad6b7SDag-Erling Smørgrav 	char *abs_dst = NULL;
692b15c8340SDag-Erling Smørgrav 	char *tmp = NULL, *filename = NULL;
693efcad6b7SDag-Erling Smørgrav 	glob_t g;
694efcad6b7SDag-Erling Smørgrav 	int err = 0;
695b15c8340SDag-Erling Smørgrav 	int i, dst_is_dir = 1;
696d4af9e69SDag-Erling Smørgrav 	struct stat sb;
697efcad6b7SDag-Erling Smørgrav 
698efcad6b7SDag-Erling Smørgrav 	if (dst) {
699efcad6b7SDag-Erling Smørgrav 		tmp_dst = xstrdup(dst);
700efcad6b7SDag-Erling Smørgrav 		tmp_dst = make_absolute(tmp_dst, pwd);
701efcad6b7SDag-Erling Smørgrav 	}
702efcad6b7SDag-Erling Smørgrav 
703efcad6b7SDag-Erling Smørgrav 	memset(&g, 0, sizeof(g));
704efcad6b7SDag-Erling Smørgrav 	debug3("Looking up %s", src);
705b15c8340SDag-Erling Smørgrav 	if (glob(src, GLOB_NOCHECK | GLOB_MARK, NULL, &g)) {
706efcad6b7SDag-Erling Smørgrav 		error("File \"%s\" not found.", src);
707efcad6b7SDag-Erling Smørgrav 		err = -1;
708efcad6b7SDag-Erling Smørgrav 		goto out;
709efcad6b7SDag-Erling Smørgrav 	}
710efcad6b7SDag-Erling Smørgrav 
711b15c8340SDag-Erling Smørgrav 	/* If we aren't fetching to pwd then stash this status for later */
712b15c8340SDag-Erling Smørgrav 	if (tmp_dst != NULL)
713b15c8340SDag-Erling Smørgrav 		dst_is_dir = remote_is_dir(conn, tmp_dst);
714b15c8340SDag-Erling Smørgrav 
715efcad6b7SDag-Erling Smørgrav 	/* If multiple matches, dst may be directory or unspecified */
716b15c8340SDag-Erling Smørgrav 	if (g.gl_matchc > 1 && tmp_dst && !dst_is_dir) {
717b15c8340SDag-Erling Smørgrav 		error("Multiple paths match, but destination "
718b15c8340SDag-Erling Smørgrav 		    "\"%s\" is not a directory", tmp_dst);
719efcad6b7SDag-Erling Smørgrav 		err = -1;
720efcad6b7SDag-Erling Smørgrav 		goto out;
721efcad6b7SDag-Erling Smørgrav 	}
722efcad6b7SDag-Erling Smørgrav 
723d74d50a8SDag-Erling Smørgrav 	for (i = 0; g.gl_pathv[i] && !interrupted; i++) {
724d4af9e69SDag-Erling Smørgrav 		if (stat(g.gl_pathv[i], &sb) == -1) {
725d4af9e69SDag-Erling Smørgrav 			err = -1;
726d4af9e69SDag-Erling Smørgrav 			error("stat %s: %s", g.gl_pathv[i], strerror(errno));
727d4af9e69SDag-Erling Smørgrav 			continue;
728d4af9e69SDag-Erling Smørgrav 		}
729d4af9e69SDag-Erling Smørgrav 
730b15c8340SDag-Erling Smørgrav 		tmp = xstrdup(g.gl_pathv[i]);
731b15c8340SDag-Erling Smørgrav 		if ((filename = basename(tmp)) == NULL) {
732b15c8340SDag-Erling Smørgrav 			error("basename %s: %s", tmp, strerror(errno));
733e4a9863fSDag-Erling Smørgrav 			free(tmp);
734efcad6b7SDag-Erling Smørgrav 			err = -1;
735efcad6b7SDag-Erling Smørgrav 			goto out;
736efcad6b7SDag-Erling Smørgrav 		}
737efcad6b7SDag-Erling Smørgrav 
738efcad6b7SDag-Erling Smørgrav 		if (g.gl_matchc == 1 && tmp_dst) {
739efcad6b7SDag-Erling Smørgrav 			/* If directory specified, append filename */
740b15c8340SDag-Erling Smørgrav 			if (dst_is_dir)
741b15c8340SDag-Erling Smørgrav 				abs_dst = path_append(tmp_dst, filename);
742b15c8340SDag-Erling Smørgrav 			else
743efcad6b7SDag-Erling Smørgrav 				abs_dst = xstrdup(tmp_dst);
744efcad6b7SDag-Erling Smørgrav 		} else if (tmp_dst) {
745b15c8340SDag-Erling Smørgrav 			abs_dst = path_append(tmp_dst, filename);
746b15c8340SDag-Erling Smørgrav 		} else {
747b15c8340SDag-Erling Smørgrav 			abs_dst = make_absolute(xstrdup(filename), pwd);
748b15c8340SDag-Erling Smørgrav 		}
749e4a9863fSDag-Erling Smørgrav 		free(tmp);
750efcad6b7SDag-Erling Smørgrav 
751a0ee8cc6SDag-Erling Smørgrav                 resume |= global_aflag;
752a0ee8cc6SDag-Erling Smørgrav 		if (!quiet && resume)
753076ad2f8SDag-Erling Smørgrav 			mprintf("Resuming upload of %s to %s\n",
754076ad2f8SDag-Erling Smørgrav 			    g.gl_pathv[i], abs_dst);
755a0ee8cc6SDag-Erling Smørgrav 		else if (!quiet && !resume)
756076ad2f8SDag-Erling Smørgrav 			mprintf("Uploading %s to %s\n",
757076ad2f8SDag-Erling Smørgrav 			    g.gl_pathv[i], abs_dst);
758b15c8340SDag-Erling Smørgrav 		if (pathname_is_dir(g.gl_pathv[i]) && (rflag || global_rflag)) {
759b15c8340SDag-Erling Smørgrav 			if (upload_dir(conn, g.gl_pathv[i], abs_dst,
760a0ee8cc6SDag-Erling Smørgrav 			    pflag || global_pflag, 1, resume,
761f7167e0eSDag-Erling Smørgrav 			    fflag || global_fflag) == -1)
762efcad6b7SDag-Erling Smørgrav 				err = -1;
763b15c8340SDag-Erling Smørgrav 		} else {
764b15c8340SDag-Erling Smørgrav 			if (do_upload(conn, g.gl_pathv[i], abs_dst,
765a0ee8cc6SDag-Erling Smørgrav 			    pflag || global_pflag, resume,
766f7167e0eSDag-Erling Smørgrav 			    fflag || global_fflag) == -1)
767b15c8340SDag-Erling Smørgrav 				err = -1;
768b15c8340SDag-Erling Smørgrav 		}
769efcad6b7SDag-Erling Smørgrav 	}
770efcad6b7SDag-Erling Smørgrav 
771efcad6b7SDag-Erling Smørgrav out:
772e4a9863fSDag-Erling Smørgrav 	free(abs_dst);
773e4a9863fSDag-Erling Smørgrav 	free(tmp_dst);
774efcad6b7SDag-Erling Smørgrav 	globfree(&g);
775efcad6b7SDag-Erling Smørgrav 	return(err);
776efcad6b7SDag-Erling Smørgrav }
777efcad6b7SDag-Erling Smørgrav 
778efcad6b7SDag-Erling Smørgrav static int
779efcad6b7SDag-Erling Smørgrav sdirent_comp(const void *aa, const void *bb)
780efcad6b7SDag-Erling Smørgrav {
781efcad6b7SDag-Erling Smørgrav 	SFTP_DIRENT *a = *(SFTP_DIRENT **)aa;
782efcad6b7SDag-Erling Smørgrav 	SFTP_DIRENT *b = *(SFTP_DIRENT **)bb;
783d74d50a8SDag-Erling Smørgrav 	int rmul = sort_flag & LS_REVERSE_SORT ? -1 : 1;
784efcad6b7SDag-Erling Smørgrav 
785d74d50a8SDag-Erling Smørgrav #define NCMP(a,b) (a == b ? 0 : (a < b ? 1 : -1))
786d74d50a8SDag-Erling Smørgrav 	if (sort_flag & LS_NAME_SORT)
787d74d50a8SDag-Erling Smørgrav 		return (rmul * strcmp(a->filename, b->filename));
788d74d50a8SDag-Erling Smørgrav 	else if (sort_flag & LS_TIME_SORT)
789d74d50a8SDag-Erling Smørgrav 		return (rmul * NCMP(a->a.mtime, b->a.mtime));
790d74d50a8SDag-Erling Smørgrav 	else if (sort_flag & LS_SIZE_SORT)
791d74d50a8SDag-Erling Smørgrav 		return (rmul * NCMP(a->a.size, b->a.size));
792d74d50a8SDag-Erling Smørgrav 
793d74d50a8SDag-Erling Smørgrav 	fatal("Unknown ls sort type");
794efcad6b7SDag-Erling Smørgrav }
795efcad6b7SDag-Erling Smørgrav 
796efcad6b7SDag-Erling Smørgrav /* sftp ls.1 replacement for directories */
797efcad6b7SDag-Erling Smørgrav static int
798076ad2f8SDag-Erling Smørgrav do_ls_dir(struct sftp_conn *conn, const char *path,
799076ad2f8SDag-Erling Smørgrav     const char *strip_path, int lflag)
800efcad6b7SDag-Erling Smørgrav {
801043840dfSDag-Erling Smørgrav 	int n;
802043840dfSDag-Erling Smørgrav 	u_int c = 1, colspace = 0, columns = 1;
803efcad6b7SDag-Erling Smørgrav 	SFTP_DIRENT **d;
804efcad6b7SDag-Erling Smørgrav 
805efcad6b7SDag-Erling Smørgrav 	if ((n = do_readdir(conn, path, &d)) != 0)
806efcad6b7SDag-Erling Smørgrav 		return (n);
807efcad6b7SDag-Erling Smørgrav 
808d74d50a8SDag-Erling Smørgrav 	if (!(lflag & LS_SHORT_VIEW)) {
809043840dfSDag-Erling Smørgrav 		u_int m = 0, width = 80;
810efcad6b7SDag-Erling Smørgrav 		struct winsize ws;
811efcad6b7SDag-Erling Smørgrav 		char *tmp;
812efcad6b7SDag-Erling Smørgrav 
813efcad6b7SDag-Erling Smørgrav 		/* Count entries for sort and find longest filename */
814d74d50a8SDag-Erling Smørgrav 		for (n = 0; d[n] != NULL; n++) {
815d74d50a8SDag-Erling Smørgrav 			if (d[n]->filename[0] != '.' || (lflag & LS_SHOW_ALL))
816*ca86bcf2SDag-Erling Smørgrav 				m = MAXIMUM(m, strlen(d[n]->filename));
817d74d50a8SDag-Erling Smørgrav 		}
818efcad6b7SDag-Erling Smørgrav 
819efcad6b7SDag-Erling Smørgrav 		/* Add any subpath that also needs to be counted */
820efcad6b7SDag-Erling Smørgrav 		tmp = path_strip(path, strip_path);
821efcad6b7SDag-Erling Smørgrav 		m += strlen(tmp);
822e4a9863fSDag-Erling Smørgrav 		free(tmp);
823efcad6b7SDag-Erling Smørgrav 
824efcad6b7SDag-Erling Smørgrav 		if (ioctl(fileno(stdin), TIOCGWINSZ, &ws) != -1)
825efcad6b7SDag-Erling Smørgrav 			width = ws.ws_col;
826efcad6b7SDag-Erling Smørgrav 
827efcad6b7SDag-Erling Smørgrav 		columns = width / (m + 2);
828*ca86bcf2SDag-Erling Smørgrav 		columns = MAXIMUM(columns, 1);
829efcad6b7SDag-Erling Smørgrav 		colspace = width / columns;
830*ca86bcf2SDag-Erling Smørgrav 		colspace = MINIMUM(colspace, width);
831efcad6b7SDag-Erling Smørgrav 	}
832efcad6b7SDag-Erling Smørgrav 
833d74d50a8SDag-Erling Smørgrav 	if (lflag & SORT_FLAGS) {
834021d409fSDag-Erling Smørgrav 		for (n = 0; d[n] != NULL; n++)
835021d409fSDag-Erling Smørgrav 			;	/* count entries */
836d74d50a8SDag-Erling Smørgrav 		sort_flag = lflag & (SORT_FLAGS|LS_REVERSE_SORT);
837efcad6b7SDag-Erling Smørgrav 		qsort(d, n, sizeof(*d), sdirent_comp);
838d74d50a8SDag-Erling Smørgrav 	}
839efcad6b7SDag-Erling Smørgrav 
840d74d50a8SDag-Erling Smørgrav 	for (n = 0; d[n] != NULL && !interrupted; n++) {
841efcad6b7SDag-Erling Smørgrav 		char *tmp, *fname;
842efcad6b7SDag-Erling Smørgrav 
843d74d50a8SDag-Erling Smørgrav 		if (d[n]->filename[0] == '.' && !(lflag & LS_SHOW_ALL))
844d74d50a8SDag-Erling Smørgrav 			continue;
845d74d50a8SDag-Erling Smørgrav 
846efcad6b7SDag-Erling Smørgrav 		tmp = path_append(path, d[n]->filename);
847efcad6b7SDag-Erling Smørgrav 		fname = path_strip(tmp, strip_path);
848e4a9863fSDag-Erling Smørgrav 		free(tmp);
849efcad6b7SDag-Erling Smørgrav 
850d74d50a8SDag-Erling Smørgrav 		if (lflag & LS_LONG_VIEW) {
851b15c8340SDag-Erling Smørgrav 			if (lflag & (LS_NUMERIC_VIEW|LS_SI_UNITS)) {
852efcad6b7SDag-Erling Smørgrav 				char *lname;
853efcad6b7SDag-Erling Smørgrav 				struct stat sb;
854efcad6b7SDag-Erling Smørgrav 
855efcad6b7SDag-Erling Smørgrav 				memset(&sb, 0, sizeof(sb));
856efcad6b7SDag-Erling Smørgrav 				attrib_to_stat(&d[n]->a, &sb);
857b15c8340SDag-Erling Smørgrav 				lname = ls_file(fname, &sb, 1,
858b15c8340SDag-Erling Smørgrav 				    (lflag & LS_SI_UNITS));
859076ad2f8SDag-Erling Smørgrav 				mprintf("%s\n", lname);
860e4a9863fSDag-Erling Smørgrav 				free(lname);
861d74d50a8SDag-Erling Smørgrav 			} else
862076ad2f8SDag-Erling Smørgrav 				mprintf("%s\n", d[n]->longname);
863efcad6b7SDag-Erling Smørgrav 		} else {
864076ad2f8SDag-Erling Smørgrav 			mprintf("%-*s", colspace, fname);
865efcad6b7SDag-Erling Smørgrav 			if (c >= columns) {
866efcad6b7SDag-Erling Smørgrav 				printf("\n");
867efcad6b7SDag-Erling Smørgrav 				c = 1;
868efcad6b7SDag-Erling Smørgrav 			} else
869efcad6b7SDag-Erling Smørgrav 				c++;
870efcad6b7SDag-Erling Smørgrav 		}
871efcad6b7SDag-Erling Smørgrav 
872e4a9863fSDag-Erling Smørgrav 		free(fname);
873efcad6b7SDag-Erling Smørgrav 	}
874efcad6b7SDag-Erling Smørgrav 
875d74d50a8SDag-Erling Smørgrav 	if (!(lflag & LS_LONG_VIEW) && (c != 1))
876efcad6b7SDag-Erling Smørgrav 		printf("\n");
877efcad6b7SDag-Erling Smørgrav 
878efcad6b7SDag-Erling Smørgrav 	free_sftp_dirents(d);
879efcad6b7SDag-Erling Smørgrav 	return (0);
880efcad6b7SDag-Erling Smørgrav }
881efcad6b7SDag-Erling Smørgrav 
882efcad6b7SDag-Erling Smørgrav /* sftp ls.1 replacement which handles path globs */
883efcad6b7SDag-Erling Smørgrav static int
884076ad2f8SDag-Erling Smørgrav do_globbed_ls(struct sftp_conn *conn, const char *path,
885076ad2f8SDag-Erling Smørgrav     const char *strip_path, int lflag)
886efcad6b7SDag-Erling Smørgrav {
8874a421b63SDag-Erling Smørgrav 	char *fname, *lname;
8884a421b63SDag-Erling Smørgrav 	glob_t g;
889a0ee8cc6SDag-Erling Smørgrav 	int err, r;
8904a421b63SDag-Erling Smørgrav 	struct winsize ws;
8914a421b63SDag-Erling Smørgrav 	u_int i, c = 1, colspace = 0, columns = 1, m = 0, width = 80;
892efcad6b7SDag-Erling Smørgrav 
893efcad6b7SDag-Erling Smørgrav 	memset(&g, 0, sizeof(g));
894efcad6b7SDag-Erling Smørgrav 
895a0ee8cc6SDag-Erling Smørgrav 	if ((r = remote_glob(conn, path,
896462c32cbSDag-Erling Smørgrav 	    GLOB_MARK|GLOB_NOCHECK|GLOB_BRACE|GLOB_KEEPSTAT|GLOB_NOSORT,
897a0ee8cc6SDag-Erling Smørgrav 	    NULL, &g)) != 0 ||
8984a421b63SDag-Erling Smørgrav 	    (g.gl_pathc && !g.gl_matchc)) {
8995e8dbd04SDag-Erling Smørgrav 		if (g.gl_pathc)
9005e8dbd04SDag-Erling Smørgrav 			globfree(&g);
901a0ee8cc6SDag-Erling Smørgrav 		if (r == GLOB_NOSPACE) {
902a0ee8cc6SDag-Erling Smørgrav 			error("Can't ls: Too many matches for \"%s\"", path);
903a0ee8cc6SDag-Erling Smørgrav 		} else {
904efcad6b7SDag-Erling Smørgrav 			error("Can't ls: \"%s\" not found", path);
905a0ee8cc6SDag-Erling Smørgrav 		}
9064a421b63SDag-Erling Smørgrav 		return -1;
907efcad6b7SDag-Erling Smørgrav 	}
908efcad6b7SDag-Erling Smørgrav 
909d74d50a8SDag-Erling Smørgrav 	if (interrupted)
910d74d50a8SDag-Erling Smørgrav 		goto out;
911d74d50a8SDag-Erling Smørgrav 
912efcad6b7SDag-Erling Smørgrav 	/*
9135e8dbd04SDag-Erling Smørgrav 	 * If the glob returns a single match and it is a directory,
9145e8dbd04SDag-Erling Smørgrav 	 * then just list its contents.
915efcad6b7SDag-Erling Smørgrav 	 */
9164a421b63SDag-Erling Smørgrav 	if (g.gl_matchc == 1 && g.gl_statv[0] != NULL &&
9174a421b63SDag-Erling Smørgrav 	    S_ISDIR(g.gl_statv[0]->st_mode)) {
9185e8dbd04SDag-Erling Smørgrav 		err = do_ls_dir(conn, g.gl_pathv[0], strip_path, lflag);
919efcad6b7SDag-Erling Smørgrav 		globfree(&g);
9204a421b63SDag-Erling Smørgrav 		return err;
921efcad6b7SDag-Erling Smørgrav 	}
922efcad6b7SDag-Erling Smørgrav 
923efcad6b7SDag-Erling Smørgrav 	if (ioctl(fileno(stdin), TIOCGWINSZ, &ws) != -1)
924efcad6b7SDag-Erling Smørgrav 		width = ws.ws_col;
925efcad6b7SDag-Erling Smørgrav 
9264a421b63SDag-Erling Smørgrav 	if (!(lflag & LS_SHORT_VIEW)) {
9274a421b63SDag-Erling Smørgrav 		/* Count entries for sort and find longest filename */
9284a421b63SDag-Erling Smørgrav 		for (i = 0; g.gl_pathv[i]; i++)
929*ca86bcf2SDag-Erling Smørgrav 			m = MAXIMUM(m, strlen(g.gl_pathv[i]));
9304a421b63SDag-Erling Smørgrav 
931efcad6b7SDag-Erling Smørgrav 		columns = width / (m + 2);
932*ca86bcf2SDag-Erling Smørgrav 		columns = MAXIMUM(columns, 1);
933efcad6b7SDag-Erling Smørgrav 		colspace = width / columns;
934efcad6b7SDag-Erling Smørgrav 	}
935efcad6b7SDag-Erling Smørgrav 
936462c32cbSDag-Erling Smørgrav 	for (i = 0; g.gl_pathv[i] && !interrupted; i++) {
937efcad6b7SDag-Erling Smørgrav 		fname = path_strip(g.gl_pathv[i], strip_path);
938d74d50a8SDag-Erling Smørgrav 		if (lflag & LS_LONG_VIEW) {
9394a421b63SDag-Erling Smørgrav 			if (g.gl_statv[i] == NULL) {
9404a421b63SDag-Erling Smørgrav 				error("no stat information for %s", fname);
9414a421b63SDag-Erling Smørgrav 				continue;
9424a421b63SDag-Erling Smørgrav 			}
9434a421b63SDag-Erling Smørgrav 			lname = ls_file(fname, g.gl_statv[i], 1,
9444a421b63SDag-Erling Smørgrav 			    (lflag & LS_SI_UNITS));
945076ad2f8SDag-Erling Smørgrav 			mprintf("%s\n", lname);
946e4a9863fSDag-Erling Smørgrav 			free(lname);
947efcad6b7SDag-Erling Smørgrav 		} else {
948076ad2f8SDag-Erling Smørgrav 			mprintf("%-*s", colspace, fname);
949efcad6b7SDag-Erling Smørgrav 			if (c >= columns) {
950efcad6b7SDag-Erling Smørgrav 				printf("\n");
951efcad6b7SDag-Erling Smørgrav 				c = 1;
952efcad6b7SDag-Erling Smørgrav 			} else
953efcad6b7SDag-Erling Smørgrav 				c++;
954efcad6b7SDag-Erling Smørgrav 		}
955e4a9863fSDag-Erling Smørgrav 		free(fname);
956efcad6b7SDag-Erling Smørgrav 	}
957efcad6b7SDag-Erling Smørgrav 
958d74d50a8SDag-Erling Smørgrav 	if (!(lflag & LS_LONG_VIEW) && (c != 1))
959efcad6b7SDag-Erling Smørgrav 		printf("\n");
960efcad6b7SDag-Erling Smørgrav 
961d74d50a8SDag-Erling Smørgrav  out:
962efcad6b7SDag-Erling Smørgrav 	if (g.gl_pathc)
963efcad6b7SDag-Erling Smørgrav 		globfree(&g);
964efcad6b7SDag-Erling Smørgrav 
9654a421b63SDag-Erling Smørgrav 	return 0;
966efcad6b7SDag-Erling Smørgrav }
967efcad6b7SDag-Erling Smørgrav 
968efcad6b7SDag-Erling Smørgrav static int
969076ad2f8SDag-Erling Smørgrav do_df(struct sftp_conn *conn, const char *path, int hflag, int iflag)
970d4af9e69SDag-Erling Smørgrav {
971d4af9e69SDag-Erling Smørgrav 	struct sftp_statvfs st;
972d4af9e69SDag-Erling Smørgrav 	char s_used[FMT_SCALED_STRSIZE];
973d4af9e69SDag-Erling Smørgrav 	char s_avail[FMT_SCALED_STRSIZE];
974d4af9e69SDag-Erling Smørgrav 	char s_root[FMT_SCALED_STRSIZE];
975d4af9e69SDag-Erling Smørgrav 	char s_total[FMT_SCALED_STRSIZE];
976b15c8340SDag-Erling Smørgrav 	unsigned long long ffree;
977d4af9e69SDag-Erling Smørgrav 
978d4af9e69SDag-Erling Smørgrav 	if (do_statvfs(conn, path, &st, 1) == -1)
979d4af9e69SDag-Erling Smørgrav 		return -1;
980d4af9e69SDag-Erling Smørgrav 	if (iflag) {
981b15c8340SDag-Erling Smørgrav 		ffree = st.f_files ? (100 * (st.f_files - st.f_ffree) / st.f_files) : 0;
982d4af9e69SDag-Erling Smørgrav 		printf("     Inodes        Used       Avail      "
983d4af9e69SDag-Erling Smørgrav 		    "(root)    %%Capacity\n");
984d4af9e69SDag-Erling Smørgrav 		printf("%11llu %11llu %11llu %11llu         %3llu%%\n",
985d4af9e69SDag-Erling Smørgrav 		    (unsigned long long)st.f_files,
986d4af9e69SDag-Erling Smørgrav 		    (unsigned long long)(st.f_files - st.f_ffree),
987d4af9e69SDag-Erling Smørgrav 		    (unsigned long long)st.f_favail,
988b15c8340SDag-Erling Smørgrav 		    (unsigned long long)st.f_ffree, ffree);
989d4af9e69SDag-Erling Smørgrav 	} else if (hflag) {
990d4af9e69SDag-Erling Smørgrav 		strlcpy(s_used, "error", sizeof(s_used));
991d4af9e69SDag-Erling Smørgrav 		strlcpy(s_avail, "error", sizeof(s_avail));
992d4af9e69SDag-Erling Smørgrav 		strlcpy(s_root, "error", sizeof(s_root));
993d4af9e69SDag-Erling Smørgrav 		strlcpy(s_total, "error", sizeof(s_total));
994d4af9e69SDag-Erling Smørgrav 		fmt_scaled((st.f_blocks - st.f_bfree) * st.f_frsize, s_used);
995d4af9e69SDag-Erling Smørgrav 		fmt_scaled(st.f_bavail * st.f_frsize, s_avail);
996d4af9e69SDag-Erling Smørgrav 		fmt_scaled(st.f_bfree * st.f_frsize, s_root);
997d4af9e69SDag-Erling Smørgrav 		fmt_scaled(st.f_blocks * st.f_frsize, s_total);
998d4af9e69SDag-Erling Smørgrav 		printf("    Size     Used    Avail   (root)    %%Capacity\n");
999d4af9e69SDag-Erling Smørgrav 		printf("%7sB %7sB %7sB %7sB         %3llu%%\n",
1000d4af9e69SDag-Erling Smørgrav 		    s_total, s_used, s_avail, s_root,
1001d4af9e69SDag-Erling Smørgrav 		    (unsigned long long)(100 * (st.f_blocks - st.f_bfree) /
1002d4af9e69SDag-Erling Smørgrav 		    st.f_blocks));
1003d4af9e69SDag-Erling Smørgrav 	} else {
1004d4af9e69SDag-Erling Smørgrav 		printf("        Size         Used        Avail       "
1005d4af9e69SDag-Erling Smørgrav 		    "(root)    %%Capacity\n");
1006d4af9e69SDag-Erling Smørgrav 		printf("%12llu %12llu %12llu %12llu         %3llu%%\n",
1007d4af9e69SDag-Erling Smørgrav 		    (unsigned long long)(st.f_frsize * st.f_blocks / 1024),
1008d4af9e69SDag-Erling Smørgrav 		    (unsigned long long)(st.f_frsize *
1009d4af9e69SDag-Erling Smørgrav 		    (st.f_blocks - st.f_bfree) / 1024),
1010d4af9e69SDag-Erling Smørgrav 		    (unsigned long long)(st.f_frsize * st.f_bavail / 1024),
1011d4af9e69SDag-Erling Smørgrav 		    (unsigned long long)(st.f_frsize * st.f_bfree / 1024),
1012d4af9e69SDag-Erling Smørgrav 		    (unsigned long long)(100 * (st.f_blocks - st.f_bfree) /
1013d4af9e69SDag-Erling Smørgrav 		    st.f_blocks));
1014d4af9e69SDag-Erling Smørgrav 	}
1015d4af9e69SDag-Erling Smørgrav 	return 0;
1016d4af9e69SDag-Erling Smørgrav }
1017d4af9e69SDag-Erling Smørgrav 
1018d4af9e69SDag-Erling Smørgrav /*
1019d4af9e69SDag-Erling Smørgrav  * Undo escaping of glob sequences in place. Used to undo extra escaping
1020d4af9e69SDag-Erling Smørgrav  * applied in makeargv() when the string is destined for a function that
1021d4af9e69SDag-Erling Smørgrav  * does not glob it.
1022d4af9e69SDag-Erling Smørgrav  */
1023d4af9e69SDag-Erling Smørgrav static void
1024d4af9e69SDag-Erling Smørgrav undo_glob_escape(char *s)
1025d4af9e69SDag-Erling Smørgrav {
1026d4af9e69SDag-Erling Smørgrav 	size_t i, j;
1027d4af9e69SDag-Erling Smørgrav 
1028d4af9e69SDag-Erling Smørgrav 	for (i = j = 0;;) {
1029d4af9e69SDag-Erling Smørgrav 		if (s[i] == '\0') {
1030d4af9e69SDag-Erling Smørgrav 			s[j] = '\0';
1031d4af9e69SDag-Erling Smørgrav 			return;
1032d4af9e69SDag-Erling Smørgrav 		}
1033d4af9e69SDag-Erling Smørgrav 		if (s[i] != '\\') {
1034d4af9e69SDag-Erling Smørgrav 			s[j++] = s[i++];
1035d4af9e69SDag-Erling Smørgrav 			continue;
1036d4af9e69SDag-Erling Smørgrav 		}
1037d4af9e69SDag-Erling Smørgrav 		/* s[i] == '\\' */
1038d4af9e69SDag-Erling Smørgrav 		++i;
1039d4af9e69SDag-Erling Smørgrav 		switch (s[i]) {
1040d4af9e69SDag-Erling Smørgrav 		case '?':
1041d4af9e69SDag-Erling Smørgrav 		case '[':
1042d4af9e69SDag-Erling Smørgrav 		case '*':
1043d4af9e69SDag-Erling Smørgrav 		case '\\':
1044d4af9e69SDag-Erling Smørgrav 			s[j++] = s[i++];
1045d4af9e69SDag-Erling Smørgrav 			break;
1046d4af9e69SDag-Erling Smørgrav 		case '\0':
1047d4af9e69SDag-Erling Smørgrav 			s[j++] = '\\';
1048d4af9e69SDag-Erling Smørgrav 			s[j] = '\0';
1049d4af9e69SDag-Erling Smørgrav 			return;
1050d4af9e69SDag-Erling Smørgrav 		default:
1051d4af9e69SDag-Erling Smørgrav 			s[j++] = '\\';
1052d4af9e69SDag-Erling Smørgrav 			s[j++] = s[i++];
1053d4af9e69SDag-Erling Smørgrav 			break;
1054d4af9e69SDag-Erling Smørgrav 		}
1055d4af9e69SDag-Erling Smørgrav 	}
1056d4af9e69SDag-Erling Smørgrav }
1057d4af9e69SDag-Erling Smørgrav 
1058d4af9e69SDag-Erling Smørgrav /*
1059d4af9e69SDag-Erling Smørgrav  * Split a string into an argument vector using sh(1)-style quoting,
1060d4af9e69SDag-Erling Smørgrav  * comment and escaping rules, but with some tweaks to handle glob(3)
1061d4af9e69SDag-Erling Smørgrav  * wildcards.
1062b15c8340SDag-Erling Smørgrav  * The "sloppy" flag allows for recovery from missing terminating quote, for
1063b15c8340SDag-Erling Smørgrav  * use in parsing incomplete commandlines during tab autocompletion.
1064b15c8340SDag-Erling Smørgrav  *
1065d4af9e69SDag-Erling Smørgrav  * Returns NULL on error or a NULL-terminated array of arguments.
1066b15c8340SDag-Erling Smørgrav  *
1067b15c8340SDag-Erling Smørgrav  * If "lastquote" is not NULL, the quoting character used for the last
1068b15c8340SDag-Erling Smørgrav  * argument is placed in *lastquote ("\0", "'" or "\"").
1069b15c8340SDag-Erling Smørgrav  *
1070b15c8340SDag-Erling Smørgrav  * If "terminated" is not NULL, *terminated will be set to 1 when the
1071b15c8340SDag-Erling Smørgrav  * last argument's quote has been properly terminated or 0 otherwise.
1072b15c8340SDag-Erling Smørgrav  * This parameter is only of use if "sloppy" is set.
1073d4af9e69SDag-Erling Smørgrav  */
1074d4af9e69SDag-Erling Smørgrav #define MAXARGS 	128
1075d4af9e69SDag-Erling Smørgrav #define MAXARGLEN	8192
1076d4af9e69SDag-Erling Smørgrav static char **
1077b15c8340SDag-Erling Smørgrav makeargv(const char *arg, int *argcp, int sloppy, char *lastquote,
1078b15c8340SDag-Erling Smørgrav     u_int *terminated)
1079d4af9e69SDag-Erling Smørgrav {
1080d4af9e69SDag-Erling Smørgrav 	int argc, quot;
1081d4af9e69SDag-Erling Smørgrav 	size_t i, j;
1082d4af9e69SDag-Erling Smørgrav 	static char argvs[MAXARGLEN];
1083d4af9e69SDag-Erling Smørgrav 	static char *argv[MAXARGS + 1];
1084d4af9e69SDag-Erling Smørgrav 	enum { MA_START, MA_SQUOTE, MA_DQUOTE, MA_UNQUOTED } state, q;
1085d4af9e69SDag-Erling Smørgrav 
1086d4af9e69SDag-Erling Smørgrav 	*argcp = argc = 0;
1087d4af9e69SDag-Erling Smørgrav 	if (strlen(arg) > sizeof(argvs) - 1) {
1088d4af9e69SDag-Erling Smørgrav  args_too_longs:
1089d4af9e69SDag-Erling Smørgrav 		error("string too long");
1090d4af9e69SDag-Erling Smørgrav 		return NULL;
1091d4af9e69SDag-Erling Smørgrav 	}
1092b15c8340SDag-Erling Smørgrav 	if (terminated != NULL)
1093b15c8340SDag-Erling Smørgrav 		*terminated = 1;
1094b15c8340SDag-Erling Smørgrav 	if (lastquote != NULL)
1095b15c8340SDag-Erling Smørgrav 		*lastquote = '\0';
1096d4af9e69SDag-Erling Smørgrav 	state = MA_START;
1097d4af9e69SDag-Erling Smørgrav 	i = j = 0;
1098d4af9e69SDag-Erling Smørgrav 	for (;;) {
10996888a9beSDag-Erling Smørgrav 		if ((size_t)argc >= sizeof(argv) / sizeof(*argv)){
11006888a9beSDag-Erling Smørgrav 			error("Too many arguments.");
11016888a9beSDag-Erling Smørgrav 			return NULL;
11026888a9beSDag-Erling Smørgrav 		}
1103f7167e0eSDag-Erling Smørgrav 		if (isspace((unsigned char)arg[i])) {
1104d4af9e69SDag-Erling Smørgrav 			if (state == MA_UNQUOTED) {
1105d4af9e69SDag-Erling Smørgrav 				/* Terminate current argument */
1106d4af9e69SDag-Erling Smørgrav 				argvs[j++] = '\0';
1107d4af9e69SDag-Erling Smørgrav 				argc++;
1108d4af9e69SDag-Erling Smørgrav 				state = MA_START;
1109d4af9e69SDag-Erling Smørgrav 			} else if (state != MA_START)
1110d4af9e69SDag-Erling Smørgrav 				argvs[j++] = arg[i];
1111d4af9e69SDag-Erling Smørgrav 		} else if (arg[i] == '"' || arg[i] == '\'') {
1112d4af9e69SDag-Erling Smørgrav 			q = arg[i] == '"' ? MA_DQUOTE : MA_SQUOTE;
1113d4af9e69SDag-Erling Smørgrav 			if (state == MA_START) {
1114d4af9e69SDag-Erling Smørgrav 				argv[argc] = argvs + j;
1115d4af9e69SDag-Erling Smørgrav 				state = q;
1116b15c8340SDag-Erling Smørgrav 				if (lastquote != NULL)
1117b15c8340SDag-Erling Smørgrav 					*lastquote = arg[i];
1118d4af9e69SDag-Erling Smørgrav 			} else if (state == MA_UNQUOTED)
1119d4af9e69SDag-Erling Smørgrav 				state = q;
1120d4af9e69SDag-Erling Smørgrav 			else if (state == q)
1121d4af9e69SDag-Erling Smørgrav 				state = MA_UNQUOTED;
1122d4af9e69SDag-Erling Smørgrav 			else
1123d4af9e69SDag-Erling Smørgrav 				argvs[j++] = arg[i];
1124d4af9e69SDag-Erling Smørgrav 		} else if (arg[i] == '\\') {
1125d4af9e69SDag-Erling Smørgrav 			if (state == MA_SQUOTE || state == MA_DQUOTE) {
1126d4af9e69SDag-Erling Smørgrav 				quot = state == MA_SQUOTE ? '\'' : '"';
1127d4af9e69SDag-Erling Smørgrav 				/* Unescape quote we are in */
1128d4af9e69SDag-Erling Smørgrav 				/* XXX support \n and friends? */
1129d4af9e69SDag-Erling Smørgrav 				if (arg[i + 1] == quot) {
1130d4af9e69SDag-Erling Smørgrav 					i++;
1131d4af9e69SDag-Erling Smørgrav 					argvs[j++] = arg[i];
1132d4af9e69SDag-Erling Smørgrav 				} else if (arg[i + 1] == '?' ||
1133d4af9e69SDag-Erling Smørgrav 				    arg[i + 1] == '[' || arg[i + 1] == '*') {
1134d4af9e69SDag-Erling Smørgrav 					/*
1135d4af9e69SDag-Erling Smørgrav 					 * Special case for sftp: append
1136d4af9e69SDag-Erling Smørgrav 					 * double-escaped glob sequence -
1137d4af9e69SDag-Erling Smørgrav 					 * glob will undo one level of
1138d4af9e69SDag-Erling Smørgrav 					 * escaping. NB. string can grow here.
1139d4af9e69SDag-Erling Smørgrav 					 */
1140d4af9e69SDag-Erling Smørgrav 					if (j >= sizeof(argvs) - 5)
1141d4af9e69SDag-Erling Smørgrav 						goto args_too_longs;
1142d4af9e69SDag-Erling Smørgrav 					argvs[j++] = '\\';
1143d4af9e69SDag-Erling Smørgrav 					argvs[j++] = arg[i++];
1144d4af9e69SDag-Erling Smørgrav 					argvs[j++] = '\\';
1145d4af9e69SDag-Erling Smørgrav 					argvs[j++] = arg[i];
1146d4af9e69SDag-Erling Smørgrav 				} else {
1147d4af9e69SDag-Erling Smørgrav 					argvs[j++] = arg[i++];
1148d4af9e69SDag-Erling Smørgrav 					argvs[j++] = arg[i];
1149d4af9e69SDag-Erling Smørgrav 				}
1150d4af9e69SDag-Erling Smørgrav 			} else {
1151d4af9e69SDag-Erling Smørgrav 				if (state == MA_START) {
1152d4af9e69SDag-Erling Smørgrav 					argv[argc] = argvs + j;
1153d4af9e69SDag-Erling Smørgrav 					state = MA_UNQUOTED;
1154b15c8340SDag-Erling Smørgrav 					if (lastquote != NULL)
1155b15c8340SDag-Erling Smørgrav 						*lastquote = '\0';
1156d4af9e69SDag-Erling Smørgrav 				}
1157d4af9e69SDag-Erling Smørgrav 				if (arg[i + 1] == '?' || arg[i + 1] == '[' ||
1158d4af9e69SDag-Erling Smørgrav 				    arg[i + 1] == '*' || arg[i + 1] == '\\') {
1159d4af9e69SDag-Erling Smørgrav 					/*
1160d4af9e69SDag-Erling Smørgrav 					 * Special case for sftp: append
1161d4af9e69SDag-Erling Smørgrav 					 * escaped glob sequence -
1162d4af9e69SDag-Erling Smørgrav 					 * glob will undo one level of
1163d4af9e69SDag-Erling Smørgrav 					 * escaping.
1164d4af9e69SDag-Erling Smørgrav 					 */
1165d4af9e69SDag-Erling Smørgrav 					argvs[j++] = arg[i++];
1166d4af9e69SDag-Erling Smørgrav 					argvs[j++] = arg[i];
1167d4af9e69SDag-Erling Smørgrav 				} else {
1168d4af9e69SDag-Erling Smørgrav 					/* Unescape everything */
1169d4af9e69SDag-Erling Smørgrav 					/* XXX support \n and friends? */
1170d4af9e69SDag-Erling Smørgrav 					i++;
1171d4af9e69SDag-Erling Smørgrav 					argvs[j++] = arg[i];
1172d4af9e69SDag-Erling Smørgrav 				}
1173d4af9e69SDag-Erling Smørgrav 			}
1174d4af9e69SDag-Erling Smørgrav 		} else if (arg[i] == '#') {
1175d4af9e69SDag-Erling Smørgrav 			if (state == MA_SQUOTE || state == MA_DQUOTE)
1176d4af9e69SDag-Erling Smørgrav 				argvs[j++] = arg[i];
1177d4af9e69SDag-Erling Smørgrav 			else
1178d4af9e69SDag-Erling Smørgrav 				goto string_done;
1179d4af9e69SDag-Erling Smørgrav 		} else if (arg[i] == '\0') {
1180d4af9e69SDag-Erling Smørgrav 			if (state == MA_SQUOTE || state == MA_DQUOTE) {
1181b15c8340SDag-Erling Smørgrav 				if (sloppy) {
1182b15c8340SDag-Erling Smørgrav 					state = MA_UNQUOTED;
1183b15c8340SDag-Erling Smørgrav 					if (terminated != NULL)
1184b15c8340SDag-Erling Smørgrav 						*terminated = 0;
1185b15c8340SDag-Erling Smørgrav 					goto string_done;
1186b15c8340SDag-Erling Smørgrav 				}
1187d4af9e69SDag-Erling Smørgrav 				error("Unterminated quoted argument");
1188d4af9e69SDag-Erling Smørgrav 				return NULL;
1189d4af9e69SDag-Erling Smørgrav 			}
1190d4af9e69SDag-Erling Smørgrav  string_done:
1191d4af9e69SDag-Erling Smørgrav 			if (state == MA_UNQUOTED) {
1192d4af9e69SDag-Erling Smørgrav 				argvs[j++] = '\0';
1193d4af9e69SDag-Erling Smørgrav 				argc++;
1194d4af9e69SDag-Erling Smørgrav 			}
1195d4af9e69SDag-Erling Smørgrav 			break;
1196d4af9e69SDag-Erling Smørgrav 		} else {
1197d4af9e69SDag-Erling Smørgrav 			if (state == MA_START) {
1198d4af9e69SDag-Erling Smørgrav 				argv[argc] = argvs + j;
1199d4af9e69SDag-Erling Smørgrav 				state = MA_UNQUOTED;
1200b15c8340SDag-Erling Smørgrav 				if (lastquote != NULL)
1201b15c8340SDag-Erling Smørgrav 					*lastquote = '\0';
1202d4af9e69SDag-Erling Smørgrav 			}
1203d4af9e69SDag-Erling Smørgrav 			if ((state == MA_SQUOTE || state == MA_DQUOTE) &&
1204d4af9e69SDag-Erling Smørgrav 			    (arg[i] == '?' || arg[i] == '[' || arg[i] == '*')) {
1205d4af9e69SDag-Erling Smørgrav 				/*
1206d4af9e69SDag-Erling Smørgrav 				 * Special case for sftp: escape quoted
1207d4af9e69SDag-Erling Smørgrav 				 * glob(3) wildcards. NB. string can grow
1208d4af9e69SDag-Erling Smørgrav 				 * here.
1209d4af9e69SDag-Erling Smørgrav 				 */
1210d4af9e69SDag-Erling Smørgrav 				if (j >= sizeof(argvs) - 3)
1211d4af9e69SDag-Erling Smørgrav 					goto args_too_longs;
1212d4af9e69SDag-Erling Smørgrav 				argvs[j++] = '\\';
1213d4af9e69SDag-Erling Smørgrav 				argvs[j++] = arg[i];
1214d4af9e69SDag-Erling Smørgrav 			} else
1215d4af9e69SDag-Erling Smørgrav 				argvs[j++] = arg[i];
1216d4af9e69SDag-Erling Smørgrav 		}
1217d4af9e69SDag-Erling Smørgrav 		i++;
1218d4af9e69SDag-Erling Smørgrav 	}
1219d4af9e69SDag-Erling Smørgrav 	*argcp = argc;
1220d4af9e69SDag-Erling Smørgrav 	return argv;
1221d4af9e69SDag-Erling Smørgrav }
1222d4af9e69SDag-Erling Smørgrav 
1223d4af9e69SDag-Erling Smørgrav static int
1224a0ee8cc6SDag-Erling Smørgrav parse_args(const char **cpp, int *ignore_errors, int *aflag,
1225a0ee8cc6SDag-Erling Smørgrav 	  int *fflag, int *hflag, int *iflag, int *lflag, int *pflag,
1226a0ee8cc6SDag-Erling Smørgrav 	  int *rflag, int *sflag,
1227f7167e0eSDag-Erling Smørgrav     unsigned long *n_arg, char **path1, char **path2)
1228efcad6b7SDag-Erling Smørgrav {
1229efcad6b7SDag-Erling Smørgrav 	const char *cmd, *cp = *cpp;
1230d4af9e69SDag-Erling Smørgrav 	char *cp2, **argv;
1231efcad6b7SDag-Erling Smørgrav 	int base = 0;
1232efcad6b7SDag-Erling Smørgrav 	long l;
1233d4af9e69SDag-Erling Smørgrav 	int i, cmdnum, optidx, argc;
1234efcad6b7SDag-Erling Smørgrav 
1235efcad6b7SDag-Erling Smørgrav 	/* Skip leading whitespace */
1236efcad6b7SDag-Erling Smørgrav 	cp = cp + strspn(cp, WHITESPACE);
1237efcad6b7SDag-Erling Smørgrav 
1238efcad6b7SDag-Erling Smørgrav 	/* Check for leading '-' (disable error processing) */
1239f7167e0eSDag-Erling Smørgrav 	*ignore_errors = 0;
1240efcad6b7SDag-Erling Smørgrav 	if (*cp == '-') {
1241f7167e0eSDag-Erling Smørgrav 		*ignore_errors = 1;
1242efcad6b7SDag-Erling Smørgrav 		cp++;
1243b15c8340SDag-Erling Smørgrav 		cp = cp + strspn(cp, WHITESPACE);
1244efcad6b7SDag-Erling Smørgrav 	}
1245efcad6b7SDag-Erling Smørgrav 
1246b15c8340SDag-Erling Smørgrav 	/* Ignore blank lines and lines which begin with comment '#' char */
1247b15c8340SDag-Erling Smørgrav 	if (*cp == '\0' || *cp == '#')
1248b15c8340SDag-Erling Smørgrav 		return (0);
1249b15c8340SDag-Erling Smørgrav 
1250b15c8340SDag-Erling Smørgrav 	if ((argv = makeargv(cp, &argc, 0, NULL, NULL)) == NULL)
1251d4af9e69SDag-Erling Smørgrav 		return -1;
1252efcad6b7SDag-Erling Smørgrav 
1253d4af9e69SDag-Erling Smørgrav 	/* Figure out which command we have */
1254d4af9e69SDag-Erling Smørgrav 	for (i = 0; cmds[i].c != NULL; i++) {
12556888a9beSDag-Erling Smørgrav 		if (argv[0] != NULL && strcasecmp(cmds[i].c, argv[0]) == 0)
1256efcad6b7SDag-Erling Smørgrav 			break;
1257efcad6b7SDag-Erling Smørgrav 	}
1258efcad6b7SDag-Erling Smørgrav 	cmdnum = cmds[i].n;
1259efcad6b7SDag-Erling Smørgrav 	cmd = cmds[i].c;
1260efcad6b7SDag-Erling Smørgrav 
1261efcad6b7SDag-Erling Smørgrav 	/* Special case */
1262efcad6b7SDag-Erling Smørgrav 	if (*cp == '!') {
1263efcad6b7SDag-Erling Smørgrav 		cp++;
1264efcad6b7SDag-Erling Smørgrav 		cmdnum = I_SHELL;
1265efcad6b7SDag-Erling Smørgrav 	} else if (cmdnum == -1) {
1266efcad6b7SDag-Erling Smørgrav 		error("Invalid command.");
1267d4af9e69SDag-Erling Smørgrav 		return -1;
1268efcad6b7SDag-Erling Smørgrav 	}
1269efcad6b7SDag-Erling Smørgrav 
1270efcad6b7SDag-Erling Smørgrav 	/* Get arguments and parse flags */
1271f7167e0eSDag-Erling Smørgrav 	*aflag = *fflag = *hflag = *iflag = *lflag = *pflag = 0;
1272f7167e0eSDag-Erling Smørgrav 	*rflag = *sflag = 0;
1273efcad6b7SDag-Erling Smørgrav 	*path1 = *path2 = NULL;
1274d4af9e69SDag-Erling Smørgrav 	optidx = 1;
1275efcad6b7SDag-Erling Smørgrav 	switch (cmdnum) {
1276efcad6b7SDag-Erling Smørgrav 	case I_GET:
1277e4a9863fSDag-Erling Smørgrav 	case I_REGET:
1278a0ee8cc6SDag-Erling Smørgrav 	case I_REPUT:
1279efcad6b7SDag-Erling Smørgrav 	case I_PUT:
12804a421b63SDag-Erling Smørgrav 		if ((optidx = parse_getput_flags(cmd, argv, argc,
1281f7167e0eSDag-Erling Smørgrav 		    aflag, fflag, pflag, rflag)) == -1)
1282d4af9e69SDag-Erling Smørgrav 			return -1;
1283efcad6b7SDag-Erling Smørgrav 		/* Get first pathname (mandatory) */
1284d4af9e69SDag-Erling Smørgrav 		if (argc - optidx < 1) {
1285efcad6b7SDag-Erling Smørgrav 			error("You must specify at least one path after a "
1286efcad6b7SDag-Erling Smørgrav 			    "%s command.", cmd);
1287d4af9e69SDag-Erling Smørgrav 			return -1;
1288efcad6b7SDag-Erling Smørgrav 		}
1289d4af9e69SDag-Erling Smørgrav 		*path1 = xstrdup(argv[optidx]);
1290d4af9e69SDag-Erling Smørgrav 		/* Get second pathname (optional) */
1291d4af9e69SDag-Erling Smørgrav 		if (argc - optidx > 1) {
1292d4af9e69SDag-Erling Smørgrav 			*path2 = xstrdup(argv[optidx + 1]);
1293d4af9e69SDag-Erling Smørgrav 			/* Destination is not globbed */
1294d4af9e69SDag-Erling Smørgrav 			undo_glob_escape(*path2);
1295d4af9e69SDag-Erling Smørgrav 		}
1296efcad6b7SDag-Erling Smørgrav 		break;
12974a421b63SDag-Erling Smørgrav 	case I_LINK:
12984a421b63SDag-Erling Smørgrav 		if ((optidx = parse_link_flags(cmd, argv, argc, sflag)) == -1)
12994a421b63SDag-Erling Smørgrav 			return -1;
1300f7167e0eSDag-Erling Smørgrav 		goto parse_two_paths;
13014a421b63SDag-Erling Smørgrav 	case I_RENAME:
1302f7167e0eSDag-Erling Smørgrav 		if ((optidx = parse_rename_flags(cmd, argv, argc, lflag)) == -1)
1303f7167e0eSDag-Erling Smørgrav 			return -1;
1304f7167e0eSDag-Erling Smørgrav 		goto parse_two_paths;
1305f7167e0eSDag-Erling Smørgrav 	case I_SYMLINK:
1306f7167e0eSDag-Erling Smørgrav 		if ((optidx = parse_no_flags(cmd, argv, argc)) == -1)
1307f7167e0eSDag-Erling Smørgrav 			return -1;
1308f7167e0eSDag-Erling Smørgrav  parse_two_paths:
1309d4af9e69SDag-Erling Smørgrav 		if (argc - optidx < 2) {
1310efcad6b7SDag-Erling Smørgrav 			error("You must specify two paths after a %s "
1311efcad6b7SDag-Erling Smørgrav 			    "command.", cmd);
1312d4af9e69SDag-Erling Smørgrav 			return -1;
1313efcad6b7SDag-Erling Smørgrav 		}
1314d4af9e69SDag-Erling Smørgrav 		*path1 = xstrdup(argv[optidx]);
1315d4af9e69SDag-Erling Smørgrav 		*path2 = xstrdup(argv[optidx + 1]);
1316d4af9e69SDag-Erling Smørgrav 		/* Paths are not globbed */
1317d4af9e69SDag-Erling Smørgrav 		undo_glob_escape(*path1);
1318d4af9e69SDag-Erling Smørgrav 		undo_glob_escape(*path2);
1319efcad6b7SDag-Erling Smørgrav 		break;
1320efcad6b7SDag-Erling Smørgrav 	case I_RM:
1321efcad6b7SDag-Erling Smørgrav 	case I_MKDIR:
1322efcad6b7SDag-Erling Smørgrav 	case I_RMDIR:
1323efcad6b7SDag-Erling Smørgrav 	case I_CHDIR:
1324efcad6b7SDag-Erling Smørgrav 	case I_LCHDIR:
1325efcad6b7SDag-Erling Smørgrav 	case I_LMKDIR:
1326f7167e0eSDag-Erling Smørgrav 		if ((optidx = parse_no_flags(cmd, argv, argc)) == -1)
1327f7167e0eSDag-Erling Smørgrav 			return -1;
1328efcad6b7SDag-Erling Smørgrav 		/* Get pathname (mandatory) */
1329d4af9e69SDag-Erling Smørgrav 		if (argc - optidx < 1) {
1330efcad6b7SDag-Erling Smørgrav 			error("You must specify a path after a %s command.",
1331efcad6b7SDag-Erling Smørgrav 			    cmd);
1332d4af9e69SDag-Erling Smørgrav 			return -1;
1333d4af9e69SDag-Erling Smørgrav 		}
1334d4af9e69SDag-Erling Smørgrav 		*path1 = xstrdup(argv[optidx]);
1335d4af9e69SDag-Erling Smørgrav 		/* Only "rm" globs */
1336d4af9e69SDag-Erling Smørgrav 		if (cmdnum != I_RM)
1337d4af9e69SDag-Erling Smørgrav 			undo_glob_escape(*path1);
1338d4af9e69SDag-Erling Smørgrav 		break;
1339d4af9e69SDag-Erling Smørgrav 	case I_DF:
1340d4af9e69SDag-Erling Smørgrav 		if ((optidx = parse_df_flags(cmd, argv, argc, hflag,
1341d4af9e69SDag-Erling Smørgrav 		    iflag)) == -1)
1342d4af9e69SDag-Erling Smørgrav 			return -1;
1343d4af9e69SDag-Erling Smørgrav 		/* Default to current directory if no path specified */
1344d4af9e69SDag-Erling Smørgrav 		if (argc - optidx < 1)
1345d4af9e69SDag-Erling Smørgrav 			*path1 = NULL;
1346d4af9e69SDag-Erling Smørgrav 		else {
1347d4af9e69SDag-Erling Smørgrav 			*path1 = xstrdup(argv[optidx]);
1348d4af9e69SDag-Erling Smørgrav 			undo_glob_escape(*path1);
1349efcad6b7SDag-Erling Smørgrav 		}
1350efcad6b7SDag-Erling Smørgrav 		break;
1351efcad6b7SDag-Erling Smørgrav 	case I_LS:
1352d4af9e69SDag-Erling Smørgrav 		if ((optidx = parse_ls_flags(argv, argc, lflag)) == -1)
1353efcad6b7SDag-Erling Smørgrav 			return(-1);
1354efcad6b7SDag-Erling Smørgrav 		/* Path is optional */
1355d4af9e69SDag-Erling Smørgrav 		if (argc - optidx > 0)
1356d4af9e69SDag-Erling Smørgrav 			*path1 = xstrdup(argv[optidx]);
1357efcad6b7SDag-Erling Smørgrav 		break;
1358efcad6b7SDag-Erling Smørgrav 	case I_LLS:
1359d4af9e69SDag-Erling Smørgrav 		/* Skip ls command and following whitespace */
1360d4af9e69SDag-Erling Smørgrav 		cp = cp + strlen(cmd) + strspn(cp, WHITESPACE);
1361efcad6b7SDag-Erling Smørgrav 	case I_SHELL:
1362efcad6b7SDag-Erling Smørgrav 		/* Uses the rest of the line */
1363efcad6b7SDag-Erling Smørgrav 		break;
1364efcad6b7SDag-Erling Smørgrav 	case I_LUMASK:
1365efcad6b7SDag-Erling Smørgrav 	case I_CHMOD:
1366efcad6b7SDag-Erling Smørgrav 		base = 8;
1367efcad6b7SDag-Erling Smørgrav 	case I_CHOWN:
1368efcad6b7SDag-Erling Smørgrav 	case I_CHGRP:
1369f7167e0eSDag-Erling Smørgrav 		if ((optidx = parse_no_flags(cmd, argv, argc)) == -1)
1370f7167e0eSDag-Erling Smørgrav 			return -1;
1371efcad6b7SDag-Erling Smørgrav 		/* Get numeric arg (mandatory) */
1372d4af9e69SDag-Erling Smørgrav 		if (argc - optidx < 1)
1373d4af9e69SDag-Erling Smørgrav 			goto need_num_arg;
137492eb0aa1SDag-Erling Smørgrav 		errno = 0;
1375d4af9e69SDag-Erling Smørgrav 		l = strtol(argv[optidx], &cp2, base);
1376d4af9e69SDag-Erling Smørgrav 		if (cp2 == argv[optidx] || *cp2 != '\0' ||
1377d4af9e69SDag-Erling Smørgrav 		    ((l == LONG_MIN || l == LONG_MAX) && errno == ERANGE) ||
1378d4af9e69SDag-Erling Smørgrav 		    l < 0) {
1379d4af9e69SDag-Erling Smørgrav  need_num_arg:
1380efcad6b7SDag-Erling Smørgrav 			error("You must supply a numeric argument "
1381efcad6b7SDag-Erling Smørgrav 			    "to the %s command.", cmd);
1382d4af9e69SDag-Erling Smørgrav 			return -1;
1383efcad6b7SDag-Erling Smørgrav 		}
1384efcad6b7SDag-Erling Smørgrav 		*n_arg = l;
1385d4af9e69SDag-Erling Smørgrav 		if (cmdnum == I_LUMASK)
1386efcad6b7SDag-Erling Smørgrav 			break;
1387efcad6b7SDag-Erling Smørgrav 		/* Get pathname (mandatory) */
1388d4af9e69SDag-Erling Smørgrav 		if (argc - optidx < 2) {
1389efcad6b7SDag-Erling Smørgrav 			error("You must specify a path after a %s command.",
1390efcad6b7SDag-Erling Smørgrav 			    cmd);
1391d4af9e69SDag-Erling Smørgrav 			return -1;
1392efcad6b7SDag-Erling Smørgrav 		}
1393d4af9e69SDag-Erling Smørgrav 		*path1 = xstrdup(argv[optidx + 1]);
1394efcad6b7SDag-Erling Smørgrav 		break;
1395efcad6b7SDag-Erling Smørgrav 	case I_QUIT:
1396efcad6b7SDag-Erling Smørgrav 	case I_PWD:
1397efcad6b7SDag-Erling Smørgrav 	case I_LPWD:
1398efcad6b7SDag-Erling Smørgrav 	case I_HELP:
1399efcad6b7SDag-Erling Smørgrav 	case I_VERSION:
1400efcad6b7SDag-Erling Smørgrav 	case I_PROGRESS:
1401f7167e0eSDag-Erling Smørgrav 		if ((optidx = parse_no_flags(cmd, argv, argc)) == -1)
1402f7167e0eSDag-Erling Smørgrav 			return -1;
1403efcad6b7SDag-Erling Smørgrav 		break;
1404efcad6b7SDag-Erling Smørgrav 	default:
1405efcad6b7SDag-Erling Smørgrav 		fatal("Command not implemented");
1406efcad6b7SDag-Erling Smørgrav 	}
1407efcad6b7SDag-Erling Smørgrav 
1408efcad6b7SDag-Erling Smørgrav 	*cpp = cp;
1409efcad6b7SDag-Erling Smørgrav 	return(cmdnum);
1410efcad6b7SDag-Erling Smørgrav }
1411efcad6b7SDag-Erling Smørgrav 
1412efcad6b7SDag-Erling Smørgrav static int
1413efcad6b7SDag-Erling Smørgrav parse_dispatch_command(struct sftp_conn *conn, const char *cmd, char **pwd,
1414efcad6b7SDag-Erling Smørgrav     int err_abort)
1415efcad6b7SDag-Erling Smørgrav {
1416efcad6b7SDag-Erling Smørgrav 	char *path1, *path2, *tmp;
1417a0ee8cc6SDag-Erling Smørgrav 	int ignore_errors = 0, aflag = 0, fflag = 0, hflag = 0,
1418a0ee8cc6SDag-Erling Smørgrav 	iflag = 0;
1419f7167e0eSDag-Erling Smørgrav 	int lflag = 0, pflag = 0, rflag = 0, sflag = 0;
14204a421b63SDag-Erling Smørgrav 	int cmdnum, i;
1421cce7d346SDag-Erling Smørgrav 	unsigned long n_arg = 0;
1422efcad6b7SDag-Erling Smørgrav 	Attrib a, *aa;
1423bc5531deSDag-Erling Smørgrav 	char path_buf[PATH_MAX];
1424efcad6b7SDag-Erling Smørgrav 	int err = 0;
1425efcad6b7SDag-Erling Smørgrav 	glob_t g;
1426efcad6b7SDag-Erling Smørgrav 
1427efcad6b7SDag-Erling Smørgrav 	path1 = path2 = NULL;
1428f7167e0eSDag-Erling Smørgrav 	cmdnum = parse_args(&cmd, &ignore_errors, &aflag, &fflag, &hflag,
1429f7167e0eSDag-Erling Smørgrav 	    &iflag, &lflag, &pflag, &rflag, &sflag, &n_arg, &path1, &path2);
1430f7167e0eSDag-Erling Smørgrav 	if (ignore_errors != 0)
1431efcad6b7SDag-Erling Smørgrav 		err_abort = 0;
1432efcad6b7SDag-Erling Smørgrav 
1433efcad6b7SDag-Erling Smørgrav 	memset(&g, 0, sizeof(g));
1434efcad6b7SDag-Erling Smørgrav 
1435efcad6b7SDag-Erling Smørgrav 	/* Perform command */
1436efcad6b7SDag-Erling Smørgrav 	switch (cmdnum) {
1437efcad6b7SDag-Erling Smørgrav 	case 0:
1438efcad6b7SDag-Erling Smørgrav 		/* Blank line */
1439efcad6b7SDag-Erling Smørgrav 		break;
1440efcad6b7SDag-Erling Smørgrav 	case -1:
1441efcad6b7SDag-Erling Smørgrav 		/* Unrecognized command */
1442efcad6b7SDag-Erling Smørgrav 		err = -1;
1443efcad6b7SDag-Erling Smørgrav 		break;
1444e4a9863fSDag-Erling Smørgrav 	case I_REGET:
1445e4a9863fSDag-Erling Smørgrav 		aflag = 1;
1446e4a9863fSDag-Erling Smørgrav 		/* FALLTHROUGH */
1447efcad6b7SDag-Erling Smørgrav 	case I_GET:
1448e4a9863fSDag-Erling Smørgrav 		err = process_get(conn, path1, path2, *pwd, pflag,
1449f7167e0eSDag-Erling Smørgrav 		    rflag, aflag, fflag);
1450efcad6b7SDag-Erling Smørgrav 		break;
1451a0ee8cc6SDag-Erling Smørgrav 	case I_REPUT:
1452a0ee8cc6SDag-Erling Smørgrav 		aflag = 1;
1453a0ee8cc6SDag-Erling Smørgrav 		/* FALLTHROUGH */
1454efcad6b7SDag-Erling Smørgrav 	case I_PUT:
1455f7167e0eSDag-Erling Smørgrav 		err = process_put(conn, path1, path2, *pwd, pflag,
1456a0ee8cc6SDag-Erling Smørgrav 		    rflag, aflag, fflag);
1457efcad6b7SDag-Erling Smørgrav 		break;
1458efcad6b7SDag-Erling Smørgrav 	case I_RENAME:
1459efcad6b7SDag-Erling Smørgrav 		path1 = make_absolute(path1, *pwd);
1460efcad6b7SDag-Erling Smørgrav 		path2 = make_absolute(path2, *pwd);
1461f7167e0eSDag-Erling Smørgrav 		err = do_rename(conn, path1, path2, lflag);
1462efcad6b7SDag-Erling Smørgrav 		break;
1463efcad6b7SDag-Erling Smørgrav 	case I_SYMLINK:
14644a421b63SDag-Erling Smørgrav 		sflag = 1;
14654a421b63SDag-Erling Smørgrav 	case I_LINK:
1466f7167e0eSDag-Erling Smørgrav 		if (!sflag)
14674a421b63SDag-Erling Smørgrav 			path1 = make_absolute(path1, *pwd);
1468efcad6b7SDag-Erling Smørgrav 		path2 = make_absolute(path2, *pwd);
14694a421b63SDag-Erling Smørgrav 		err = (sflag ? do_symlink : do_hardlink)(conn, path1, path2);
1470efcad6b7SDag-Erling Smørgrav 		break;
1471efcad6b7SDag-Erling Smørgrav 	case I_RM:
1472efcad6b7SDag-Erling Smørgrav 		path1 = make_absolute(path1, *pwd);
1473efcad6b7SDag-Erling Smørgrav 		remote_glob(conn, path1, GLOB_NOCHECK, NULL, &g);
1474d74d50a8SDag-Erling Smørgrav 		for (i = 0; g.gl_pathv[i] && !interrupted; i++) {
1475e4a9863fSDag-Erling Smørgrav 			if (!quiet)
1476076ad2f8SDag-Erling Smørgrav 				mprintf("Removing %s\n", g.gl_pathv[i]);
1477efcad6b7SDag-Erling Smørgrav 			err = do_rm(conn, g.gl_pathv[i]);
1478efcad6b7SDag-Erling Smørgrav 			if (err != 0 && err_abort)
1479efcad6b7SDag-Erling Smørgrav 				break;
1480efcad6b7SDag-Erling Smørgrav 		}
1481efcad6b7SDag-Erling Smørgrav 		break;
1482efcad6b7SDag-Erling Smørgrav 	case I_MKDIR:
1483efcad6b7SDag-Erling Smørgrav 		path1 = make_absolute(path1, *pwd);
1484efcad6b7SDag-Erling Smørgrav 		attrib_clear(&a);
1485efcad6b7SDag-Erling Smørgrav 		a.flags |= SSH2_FILEXFER_ATTR_PERMISSIONS;
1486efcad6b7SDag-Erling Smørgrav 		a.perm = 0777;
1487b15c8340SDag-Erling Smørgrav 		err = do_mkdir(conn, path1, &a, 1);
1488efcad6b7SDag-Erling Smørgrav 		break;
1489efcad6b7SDag-Erling Smørgrav 	case I_RMDIR:
1490efcad6b7SDag-Erling Smørgrav 		path1 = make_absolute(path1, *pwd);
1491efcad6b7SDag-Erling Smørgrav 		err = do_rmdir(conn, path1);
1492efcad6b7SDag-Erling Smørgrav 		break;
1493efcad6b7SDag-Erling Smørgrav 	case I_CHDIR:
1494efcad6b7SDag-Erling Smørgrav 		path1 = make_absolute(path1, *pwd);
1495efcad6b7SDag-Erling Smørgrav 		if ((tmp = do_realpath(conn, path1)) == NULL) {
1496efcad6b7SDag-Erling Smørgrav 			err = 1;
1497efcad6b7SDag-Erling Smørgrav 			break;
1498efcad6b7SDag-Erling Smørgrav 		}
1499efcad6b7SDag-Erling Smørgrav 		if ((aa = do_stat(conn, tmp, 0)) == NULL) {
1500e4a9863fSDag-Erling Smørgrav 			free(tmp);
1501efcad6b7SDag-Erling Smørgrav 			err = 1;
1502efcad6b7SDag-Erling Smørgrav 			break;
1503efcad6b7SDag-Erling Smørgrav 		}
1504efcad6b7SDag-Erling Smørgrav 		if (!(aa->flags & SSH2_FILEXFER_ATTR_PERMISSIONS)) {
1505efcad6b7SDag-Erling Smørgrav 			error("Can't change directory: Can't check target");
1506e4a9863fSDag-Erling Smørgrav 			free(tmp);
1507efcad6b7SDag-Erling Smørgrav 			err = 1;
1508efcad6b7SDag-Erling Smørgrav 			break;
1509efcad6b7SDag-Erling Smørgrav 		}
1510efcad6b7SDag-Erling Smørgrav 		if (!S_ISDIR(aa->perm)) {
1511efcad6b7SDag-Erling Smørgrav 			error("Can't change directory: \"%s\" is not "
1512efcad6b7SDag-Erling Smørgrav 			    "a directory", tmp);
1513e4a9863fSDag-Erling Smørgrav 			free(tmp);
1514efcad6b7SDag-Erling Smørgrav 			err = 1;
1515efcad6b7SDag-Erling Smørgrav 			break;
1516efcad6b7SDag-Erling Smørgrav 		}
1517e4a9863fSDag-Erling Smørgrav 		free(*pwd);
1518efcad6b7SDag-Erling Smørgrav 		*pwd = tmp;
1519efcad6b7SDag-Erling Smørgrav 		break;
1520efcad6b7SDag-Erling Smørgrav 	case I_LS:
1521efcad6b7SDag-Erling Smørgrav 		if (!path1) {
1522e2f6069cSDag-Erling Smørgrav 			do_ls_dir(conn, *pwd, *pwd, lflag);
1523efcad6b7SDag-Erling Smørgrav 			break;
1524efcad6b7SDag-Erling Smørgrav 		}
1525efcad6b7SDag-Erling Smørgrav 
1526efcad6b7SDag-Erling Smørgrav 		/* Strip pwd off beginning of non-absolute paths */
1527efcad6b7SDag-Erling Smørgrav 		tmp = NULL;
1528efcad6b7SDag-Erling Smørgrav 		if (*path1 != '/')
1529efcad6b7SDag-Erling Smørgrav 			tmp = *pwd;
1530efcad6b7SDag-Erling Smørgrav 
1531efcad6b7SDag-Erling Smørgrav 		path1 = make_absolute(path1, *pwd);
1532efcad6b7SDag-Erling Smørgrav 		err = do_globbed_ls(conn, path1, tmp, lflag);
1533efcad6b7SDag-Erling Smørgrav 		break;
1534d4af9e69SDag-Erling Smørgrav 	case I_DF:
1535d4af9e69SDag-Erling Smørgrav 		/* Default to current directory if no path specified */
1536d4af9e69SDag-Erling Smørgrav 		if (path1 == NULL)
1537d4af9e69SDag-Erling Smørgrav 			path1 = xstrdup(*pwd);
1538d4af9e69SDag-Erling Smørgrav 		path1 = make_absolute(path1, *pwd);
1539d4af9e69SDag-Erling Smørgrav 		err = do_df(conn, path1, hflag, iflag);
1540d4af9e69SDag-Erling Smørgrav 		break;
1541efcad6b7SDag-Erling Smørgrav 	case I_LCHDIR:
1542bc5531deSDag-Erling Smørgrav 		tmp = tilde_expand_filename(path1, getuid());
1543bc5531deSDag-Erling Smørgrav 		free(path1);
1544bc5531deSDag-Erling Smørgrav 		path1 = tmp;
1545efcad6b7SDag-Erling Smørgrav 		if (chdir(path1) == -1) {
1546efcad6b7SDag-Erling Smørgrav 			error("Couldn't change local directory to "
1547efcad6b7SDag-Erling Smørgrav 			    "\"%s\": %s", path1, strerror(errno));
1548efcad6b7SDag-Erling Smørgrav 			err = 1;
1549efcad6b7SDag-Erling Smørgrav 		}
1550efcad6b7SDag-Erling Smørgrav 		break;
1551efcad6b7SDag-Erling Smørgrav 	case I_LMKDIR:
1552efcad6b7SDag-Erling Smørgrav 		if (mkdir(path1, 0777) == -1) {
1553efcad6b7SDag-Erling Smørgrav 			error("Couldn't create local directory "
1554efcad6b7SDag-Erling Smørgrav 			    "\"%s\": %s", path1, strerror(errno));
1555efcad6b7SDag-Erling Smørgrav 			err = 1;
1556efcad6b7SDag-Erling Smørgrav 		}
1557efcad6b7SDag-Erling Smørgrav 		break;
1558efcad6b7SDag-Erling Smørgrav 	case I_LLS:
1559efcad6b7SDag-Erling Smørgrav 		local_do_ls(cmd);
1560efcad6b7SDag-Erling Smørgrav 		break;
1561efcad6b7SDag-Erling Smørgrav 	case I_SHELL:
1562efcad6b7SDag-Erling Smørgrav 		local_do_shell(cmd);
1563efcad6b7SDag-Erling Smørgrav 		break;
1564efcad6b7SDag-Erling Smørgrav 	case I_LUMASK:
1565efcad6b7SDag-Erling Smørgrav 		umask(n_arg);
1566efcad6b7SDag-Erling Smørgrav 		printf("Local umask: %03lo\n", n_arg);
1567efcad6b7SDag-Erling Smørgrav 		break;
1568efcad6b7SDag-Erling Smørgrav 	case I_CHMOD:
1569efcad6b7SDag-Erling Smørgrav 		path1 = make_absolute(path1, *pwd);
1570efcad6b7SDag-Erling Smørgrav 		attrib_clear(&a);
1571efcad6b7SDag-Erling Smørgrav 		a.flags |= SSH2_FILEXFER_ATTR_PERMISSIONS;
1572efcad6b7SDag-Erling Smørgrav 		a.perm = n_arg;
1573efcad6b7SDag-Erling Smørgrav 		remote_glob(conn, path1, GLOB_NOCHECK, NULL, &g);
1574d74d50a8SDag-Erling Smørgrav 		for (i = 0; g.gl_pathv[i] && !interrupted; i++) {
1575e4a9863fSDag-Erling Smørgrav 			if (!quiet)
1576076ad2f8SDag-Erling Smørgrav 				mprintf("Changing mode on %s\n",
1577076ad2f8SDag-Erling Smørgrav 				    g.gl_pathv[i]);
1578efcad6b7SDag-Erling Smørgrav 			err = do_setstat(conn, g.gl_pathv[i], &a);
1579efcad6b7SDag-Erling Smørgrav 			if (err != 0 && err_abort)
1580efcad6b7SDag-Erling Smørgrav 				break;
1581efcad6b7SDag-Erling Smørgrav 		}
1582efcad6b7SDag-Erling Smørgrav 		break;
1583efcad6b7SDag-Erling Smørgrav 	case I_CHOWN:
1584efcad6b7SDag-Erling Smørgrav 	case I_CHGRP:
1585efcad6b7SDag-Erling Smørgrav 		path1 = make_absolute(path1, *pwd);
1586efcad6b7SDag-Erling Smørgrav 		remote_glob(conn, path1, GLOB_NOCHECK, NULL, &g);
1587d74d50a8SDag-Erling Smørgrav 		for (i = 0; g.gl_pathv[i] && !interrupted; i++) {
1588efcad6b7SDag-Erling Smørgrav 			if (!(aa = do_stat(conn, g.gl_pathv[i], 0))) {
1589cce7d346SDag-Erling Smørgrav 				if (err_abort) {
1590cce7d346SDag-Erling Smørgrav 					err = -1;
1591efcad6b7SDag-Erling Smørgrav 					break;
1592cce7d346SDag-Erling Smørgrav 				} else
1593efcad6b7SDag-Erling Smørgrav 					continue;
1594efcad6b7SDag-Erling Smørgrav 			}
1595efcad6b7SDag-Erling Smørgrav 			if (!(aa->flags & SSH2_FILEXFER_ATTR_UIDGID)) {
1596efcad6b7SDag-Erling Smørgrav 				error("Can't get current ownership of "
1597efcad6b7SDag-Erling Smørgrav 				    "remote file \"%s\"", g.gl_pathv[i]);
1598cce7d346SDag-Erling Smørgrav 				if (err_abort) {
1599cce7d346SDag-Erling Smørgrav 					err = -1;
1600efcad6b7SDag-Erling Smørgrav 					break;
1601cce7d346SDag-Erling Smørgrav 				} else
1602efcad6b7SDag-Erling Smørgrav 					continue;
1603efcad6b7SDag-Erling Smørgrav 			}
1604efcad6b7SDag-Erling Smørgrav 			aa->flags &= SSH2_FILEXFER_ATTR_UIDGID;
1605efcad6b7SDag-Erling Smørgrav 			if (cmdnum == I_CHOWN) {
1606e4a9863fSDag-Erling Smørgrav 				if (!quiet)
1607076ad2f8SDag-Erling Smørgrav 					mprintf("Changing owner on %s\n",
1608e4a9863fSDag-Erling Smørgrav 					    g.gl_pathv[i]);
1609efcad6b7SDag-Erling Smørgrav 				aa->uid = n_arg;
1610efcad6b7SDag-Erling Smørgrav 			} else {
1611e4a9863fSDag-Erling Smørgrav 				if (!quiet)
1612076ad2f8SDag-Erling Smørgrav 					mprintf("Changing group on %s\n",
1613e4a9863fSDag-Erling Smørgrav 					    g.gl_pathv[i]);
1614efcad6b7SDag-Erling Smørgrav 				aa->gid = n_arg;
1615efcad6b7SDag-Erling Smørgrav 			}
1616efcad6b7SDag-Erling Smørgrav 			err = do_setstat(conn, g.gl_pathv[i], aa);
1617efcad6b7SDag-Erling Smørgrav 			if (err != 0 && err_abort)
1618efcad6b7SDag-Erling Smørgrav 				break;
1619efcad6b7SDag-Erling Smørgrav 		}
1620efcad6b7SDag-Erling Smørgrav 		break;
1621efcad6b7SDag-Erling Smørgrav 	case I_PWD:
1622076ad2f8SDag-Erling Smørgrav 		mprintf("Remote working directory: %s\n", *pwd);
1623efcad6b7SDag-Erling Smørgrav 		break;
1624efcad6b7SDag-Erling Smørgrav 	case I_LPWD:
1625efcad6b7SDag-Erling Smørgrav 		if (!getcwd(path_buf, sizeof(path_buf))) {
1626efcad6b7SDag-Erling Smørgrav 			error("Couldn't get local cwd: %s", strerror(errno));
1627efcad6b7SDag-Erling Smørgrav 			err = -1;
1628efcad6b7SDag-Erling Smørgrav 			break;
1629efcad6b7SDag-Erling Smørgrav 		}
1630076ad2f8SDag-Erling Smørgrav 		mprintf("Local working directory: %s\n", path_buf);
1631efcad6b7SDag-Erling Smørgrav 		break;
1632efcad6b7SDag-Erling Smørgrav 	case I_QUIT:
1633efcad6b7SDag-Erling Smørgrav 		/* Processed below */
1634efcad6b7SDag-Erling Smørgrav 		break;
1635efcad6b7SDag-Erling Smørgrav 	case I_HELP:
1636efcad6b7SDag-Erling Smørgrav 		help();
1637efcad6b7SDag-Erling Smørgrav 		break;
1638efcad6b7SDag-Erling Smørgrav 	case I_VERSION:
1639efcad6b7SDag-Erling Smørgrav 		printf("SFTP protocol version %u\n", sftp_proto_version(conn));
1640efcad6b7SDag-Erling Smørgrav 		break;
1641efcad6b7SDag-Erling Smørgrav 	case I_PROGRESS:
1642efcad6b7SDag-Erling Smørgrav 		showprogress = !showprogress;
1643efcad6b7SDag-Erling Smørgrav 		if (showprogress)
1644efcad6b7SDag-Erling Smørgrav 			printf("Progress meter enabled\n");
1645efcad6b7SDag-Erling Smørgrav 		else
1646efcad6b7SDag-Erling Smørgrav 			printf("Progress meter disabled\n");
1647efcad6b7SDag-Erling Smørgrav 		break;
1648efcad6b7SDag-Erling Smørgrav 	default:
1649efcad6b7SDag-Erling Smørgrav 		fatal("%d is not implemented", cmdnum);
1650efcad6b7SDag-Erling Smørgrav 	}
1651efcad6b7SDag-Erling Smørgrav 
1652efcad6b7SDag-Erling Smørgrav 	if (g.gl_pathc)
1653efcad6b7SDag-Erling Smørgrav 		globfree(&g);
1654e4a9863fSDag-Erling Smørgrav 	free(path1);
1655e4a9863fSDag-Erling Smørgrav 	free(path2);
1656efcad6b7SDag-Erling Smørgrav 
1657efcad6b7SDag-Erling Smørgrav 	/* If an unignored error occurs in batch mode we should abort. */
1658efcad6b7SDag-Erling Smørgrav 	if (err_abort && err != 0)
1659efcad6b7SDag-Erling Smørgrav 		return (-1);
1660efcad6b7SDag-Erling Smørgrav 	else if (cmdnum == I_QUIT)
1661efcad6b7SDag-Erling Smørgrav 		return (1);
1662efcad6b7SDag-Erling Smørgrav 
1663efcad6b7SDag-Erling Smørgrav 	return (0);
1664efcad6b7SDag-Erling Smørgrav }
1665efcad6b7SDag-Erling Smørgrav 
16665e8dbd04SDag-Erling Smørgrav #ifdef USE_LIBEDIT
16675e8dbd04SDag-Erling Smørgrav static char *
16685e8dbd04SDag-Erling Smørgrav prompt(EditLine *el)
16695e8dbd04SDag-Erling Smørgrav {
16705e8dbd04SDag-Erling Smørgrav 	return ("sftp> ");
16715e8dbd04SDag-Erling Smørgrav }
1672b15c8340SDag-Erling Smørgrav 
1673b15c8340SDag-Erling Smørgrav /* Display entries in 'list' after skipping the first 'len' chars */
1674b15c8340SDag-Erling Smørgrav static void
1675b15c8340SDag-Erling Smørgrav complete_display(char **list, u_int len)
1676b15c8340SDag-Erling Smørgrav {
1677b15c8340SDag-Erling Smørgrav 	u_int y, m = 0, width = 80, columns = 1, colspace = 0, llen;
1678b15c8340SDag-Erling Smørgrav 	struct winsize ws;
1679b15c8340SDag-Erling Smørgrav 	char *tmp;
1680b15c8340SDag-Erling Smørgrav 
1681b15c8340SDag-Erling Smørgrav 	/* Count entries for sort and find longest */
1682b15c8340SDag-Erling Smørgrav 	for (y = 0; list[y]; y++)
1683*ca86bcf2SDag-Erling Smørgrav 		m = MAXIMUM(m, strlen(list[y]));
1684b15c8340SDag-Erling Smørgrav 
1685b15c8340SDag-Erling Smørgrav 	if (ioctl(fileno(stdin), TIOCGWINSZ, &ws) != -1)
1686b15c8340SDag-Erling Smørgrav 		width = ws.ws_col;
1687b15c8340SDag-Erling Smørgrav 
1688b15c8340SDag-Erling Smørgrav 	m = m > len ? m - len : 0;
1689b15c8340SDag-Erling Smørgrav 	columns = width / (m + 2);
1690*ca86bcf2SDag-Erling Smørgrav 	columns = MAXIMUM(columns, 1);
1691b15c8340SDag-Erling Smørgrav 	colspace = width / columns;
1692*ca86bcf2SDag-Erling Smørgrav 	colspace = MINIMUM(colspace, width);
1693b15c8340SDag-Erling Smørgrav 
1694b15c8340SDag-Erling Smørgrav 	printf("\n");
1695b15c8340SDag-Erling Smørgrav 	m = 1;
1696b15c8340SDag-Erling Smørgrav 	for (y = 0; list[y]; y++) {
1697b15c8340SDag-Erling Smørgrav 		llen = strlen(list[y]);
1698b15c8340SDag-Erling Smørgrav 		tmp = llen > len ? list[y] + len : "";
1699076ad2f8SDag-Erling Smørgrav 		mprintf("%-*s", colspace, tmp);
1700b15c8340SDag-Erling Smørgrav 		if (m >= columns) {
1701b15c8340SDag-Erling Smørgrav 			printf("\n");
1702b15c8340SDag-Erling Smørgrav 			m = 1;
1703b15c8340SDag-Erling Smørgrav 		} else
1704b15c8340SDag-Erling Smørgrav 			m++;
1705b15c8340SDag-Erling Smørgrav 	}
1706b15c8340SDag-Erling Smørgrav 	printf("\n");
1707b15c8340SDag-Erling Smørgrav }
1708b15c8340SDag-Erling Smørgrav 
1709b15c8340SDag-Erling Smørgrav /*
1710b15c8340SDag-Erling Smørgrav  * Given a "list" of words that begin with a common prefix of "word",
1711b15c8340SDag-Erling Smørgrav  * attempt to find an autocompletion to extends "word" by the next
1712b15c8340SDag-Erling Smørgrav  * characters common to all entries in "list".
1713b15c8340SDag-Erling Smørgrav  */
1714b15c8340SDag-Erling Smørgrav static char *
1715b15c8340SDag-Erling Smørgrav complete_ambiguous(const char *word, char **list, size_t count)
1716b15c8340SDag-Erling Smørgrav {
1717b15c8340SDag-Erling Smørgrav 	if (word == NULL)
1718b15c8340SDag-Erling Smørgrav 		return NULL;
1719b15c8340SDag-Erling Smørgrav 
1720b15c8340SDag-Erling Smørgrav 	if (count > 0) {
1721b15c8340SDag-Erling Smørgrav 		u_int y, matchlen = strlen(list[0]);
1722b15c8340SDag-Erling Smørgrav 
1723b15c8340SDag-Erling Smørgrav 		/* Find length of common stem */
1724b15c8340SDag-Erling Smørgrav 		for (y = 1; list[y]; y++) {
1725b15c8340SDag-Erling Smørgrav 			u_int x;
1726b15c8340SDag-Erling Smørgrav 
1727b15c8340SDag-Erling Smørgrav 			for (x = 0; x < matchlen; x++)
1728b15c8340SDag-Erling Smørgrav 				if (list[0][x] != list[y][x])
1729b15c8340SDag-Erling Smørgrav 					break;
1730b15c8340SDag-Erling Smørgrav 
1731b15c8340SDag-Erling Smørgrav 			matchlen = x;
1732b15c8340SDag-Erling Smørgrav 		}
1733b15c8340SDag-Erling Smørgrav 
1734b15c8340SDag-Erling Smørgrav 		if (matchlen > strlen(word)) {
1735b15c8340SDag-Erling Smørgrav 			char *tmp = xstrdup(list[0]);
1736b15c8340SDag-Erling Smørgrav 
1737b15c8340SDag-Erling Smørgrav 			tmp[matchlen] = '\0';
1738b15c8340SDag-Erling Smørgrav 			return tmp;
1739b15c8340SDag-Erling Smørgrav 		}
1740b15c8340SDag-Erling Smørgrav 	}
1741b15c8340SDag-Erling Smørgrav 
1742b15c8340SDag-Erling Smørgrav 	return xstrdup(word);
1743b15c8340SDag-Erling Smørgrav }
1744b15c8340SDag-Erling Smørgrav 
1745b15c8340SDag-Erling Smørgrav /* Autocomplete a sftp command */
1746b15c8340SDag-Erling Smørgrav static int
1747b15c8340SDag-Erling Smørgrav complete_cmd_parse(EditLine *el, char *cmd, int lastarg, char quote,
1748b15c8340SDag-Erling Smørgrav     int terminated)
1749b15c8340SDag-Erling Smørgrav {
1750b15c8340SDag-Erling Smørgrav 	u_int y, count = 0, cmdlen, tmplen;
1751b15c8340SDag-Erling Smørgrav 	char *tmp, **list, argterm[3];
1752b15c8340SDag-Erling Smørgrav 	const LineInfo *lf;
1753b15c8340SDag-Erling Smørgrav 
1754b15c8340SDag-Erling Smørgrav 	list = xcalloc((sizeof(cmds) / sizeof(*cmds)) + 1, sizeof(char *));
1755b15c8340SDag-Erling Smørgrav 
1756b15c8340SDag-Erling Smørgrav 	/* No command specified: display all available commands */
1757b15c8340SDag-Erling Smørgrav 	if (cmd == NULL) {
1758b15c8340SDag-Erling Smørgrav 		for (y = 0; cmds[y].c; y++)
1759b15c8340SDag-Erling Smørgrav 			list[count++] = xstrdup(cmds[y].c);
1760b15c8340SDag-Erling Smørgrav 
1761b15c8340SDag-Erling Smørgrav 		list[count] = NULL;
1762b15c8340SDag-Erling Smørgrav 		complete_display(list, 0);
1763b15c8340SDag-Erling Smørgrav 
1764b15c8340SDag-Erling Smørgrav 		for (y = 0; list[y] != NULL; y++)
1765e4a9863fSDag-Erling Smørgrav 			free(list[y]);
1766e4a9863fSDag-Erling Smørgrav 		free(list);
1767b15c8340SDag-Erling Smørgrav 		return count;
1768b15c8340SDag-Erling Smørgrav 	}
1769b15c8340SDag-Erling Smørgrav 
1770b15c8340SDag-Erling Smørgrav 	/* Prepare subset of commands that start with "cmd" */
1771b15c8340SDag-Erling Smørgrav 	cmdlen = strlen(cmd);
1772b15c8340SDag-Erling Smørgrav 	for (y = 0; cmds[y].c; y++)  {
1773b15c8340SDag-Erling Smørgrav 		if (!strncasecmp(cmd, cmds[y].c, cmdlen))
1774b15c8340SDag-Erling Smørgrav 			list[count++] = xstrdup(cmds[y].c);
1775b15c8340SDag-Erling Smørgrav 	}
1776b15c8340SDag-Erling Smørgrav 	list[count] = NULL;
1777b15c8340SDag-Erling Smørgrav 
1778462c32cbSDag-Erling Smørgrav 	if (count == 0) {
1779e4a9863fSDag-Erling Smørgrav 		free(list);
1780b15c8340SDag-Erling Smørgrav 		return 0;
1781462c32cbSDag-Erling Smørgrav 	}
1782b15c8340SDag-Erling Smørgrav 
1783b15c8340SDag-Erling Smørgrav 	/* Complete ambigious command */
1784b15c8340SDag-Erling Smørgrav 	tmp = complete_ambiguous(cmd, list, count);
1785b15c8340SDag-Erling Smørgrav 	if (count > 1)
1786b15c8340SDag-Erling Smørgrav 		complete_display(list, 0);
1787b15c8340SDag-Erling Smørgrav 
1788b15c8340SDag-Erling Smørgrav 	for (y = 0; list[y]; y++)
1789e4a9863fSDag-Erling Smørgrav 		free(list[y]);
1790e4a9863fSDag-Erling Smørgrav 	free(list);
1791b15c8340SDag-Erling Smørgrav 
1792b15c8340SDag-Erling Smørgrav 	if (tmp != NULL) {
1793b15c8340SDag-Erling Smørgrav 		tmplen = strlen(tmp);
1794b15c8340SDag-Erling Smørgrav 		cmdlen = strlen(cmd);
1795b15c8340SDag-Erling Smørgrav 		/* If cmd may be extended then do so */
1796b15c8340SDag-Erling Smørgrav 		if (tmplen > cmdlen)
1797b15c8340SDag-Erling Smørgrav 			if (el_insertstr(el, tmp + cmdlen) == -1)
1798b15c8340SDag-Erling Smørgrav 				fatal("el_insertstr failed.");
1799b15c8340SDag-Erling Smørgrav 		lf = el_line(el);
1800b15c8340SDag-Erling Smørgrav 		/* Terminate argument cleanly */
1801b15c8340SDag-Erling Smørgrav 		if (count == 1) {
1802b15c8340SDag-Erling Smørgrav 			y = 0;
1803b15c8340SDag-Erling Smørgrav 			if (!terminated)
1804b15c8340SDag-Erling Smørgrav 				argterm[y++] = quote;
1805b15c8340SDag-Erling Smørgrav 			if (lastarg || *(lf->cursor) != ' ')
1806b15c8340SDag-Erling Smørgrav 				argterm[y++] = ' ';
1807b15c8340SDag-Erling Smørgrav 			argterm[y] = '\0';
1808b15c8340SDag-Erling Smørgrav 			if (y > 0 && el_insertstr(el, argterm) == -1)
1809b15c8340SDag-Erling Smørgrav 				fatal("el_insertstr failed.");
1810b15c8340SDag-Erling Smørgrav 		}
1811e4a9863fSDag-Erling Smørgrav 		free(tmp);
1812b15c8340SDag-Erling Smørgrav 	}
1813b15c8340SDag-Erling Smørgrav 
1814b15c8340SDag-Erling Smørgrav 	return count;
1815b15c8340SDag-Erling Smørgrav }
1816b15c8340SDag-Erling Smørgrav 
1817b15c8340SDag-Erling Smørgrav /*
1818b15c8340SDag-Erling Smørgrav  * Determine whether a particular sftp command's arguments (if any)
1819b15c8340SDag-Erling Smørgrav  * represent local or remote files.
1820b15c8340SDag-Erling Smørgrav  */
1821b15c8340SDag-Erling Smørgrav static int
1822b15c8340SDag-Erling Smørgrav complete_is_remote(char *cmd) {
1823b15c8340SDag-Erling Smørgrav 	int i;
1824b15c8340SDag-Erling Smørgrav 
1825b15c8340SDag-Erling Smørgrav 	if (cmd == NULL)
1826b15c8340SDag-Erling Smørgrav 		return -1;
1827b15c8340SDag-Erling Smørgrav 
1828b15c8340SDag-Erling Smørgrav 	for (i = 0; cmds[i].c; i++) {
1829b15c8340SDag-Erling Smørgrav 		if (!strncasecmp(cmd, cmds[i].c, strlen(cmds[i].c)))
1830b15c8340SDag-Erling Smørgrav 			return cmds[i].t;
1831b15c8340SDag-Erling Smørgrav 	}
1832b15c8340SDag-Erling Smørgrav 
1833b15c8340SDag-Erling Smørgrav 	return -1;
1834b15c8340SDag-Erling Smørgrav }
1835b15c8340SDag-Erling Smørgrav 
1836b15c8340SDag-Erling Smørgrav /* Autocomplete a filename "file" */
1837b15c8340SDag-Erling Smørgrav static int
1838b15c8340SDag-Erling Smørgrav complete_match(EditLine *el, struct sftp_conn *conn, char *remote_path,
1839b15c8340SDag-Erling Smørgrav     char *file, int remote, int lastarg, char quote, int terminated)
1840b15c8340SDag-Erling Smørgrav {
1841b15c8340SDag-Erling Smørgrav 	glob_t g;
1842e4a9863fSDag-Erling Smørgrav 	char *tmp, *tmp2, ins[8];
18436888a9beSDag-Erling Smørgrav 	u_int i, hadglob, pwdlen, len, tmplen, filelen, cesc, isesc, isabs;
1844e4a9863fSDag-Erling Smørgrav 	int clen;
1845b15c8340SDag-Erling Smørgrav 	const LineInfo *lf;
1846b15c8340SDag-Erling Smørgrav 
1847b15c8340SDag-Erling Smørgrav 	/* Glob from "file" location */
1848b15c8340SDag-Erling Smørgrav 	if (file == NULL)
1849b15c8340SDag-Erling Smørgrav 		tmp = xstrdup("*");
1850b15c8340SDag-Erling Smørgrav 	else
1851b15c8340SDag-Erling Smørgrav 		xasprintf(&tmp, "%s*", file);
1852b15c8340SDag-Erling Smørgrav 
18536888a9beSDag-Erling Smørgrav 	/* Check if the path is absolute. */
18546888a9beSDag-Erling Smørgrav 	isabs = tmp[0] == '/';
18556888a9beSDag-Erling Smørgrav 
1856b15c8340SDag-Erling Smørgrav 	memset(&g, 0, sizeof(g));
1857b15c8340SDag-Erling Smørgrav 	if (remote != LOCAL) {
1858b15c8340SDag-Erling Smørgrav 		tmp = make_absolute(tmp, remote_path);
1859b15c8340SDag-Erling Smørgrav 		remote_glob(conn, tmp, GLOB_DOOFFS|GLOB_MARK, NULL, &g);
1860b15c8340SDag-Erling Smørgrav 	} else
1861b15c8340SDag-Erling Smørgrav 		glob(tmp, GLOB_DOOFFS|GLOB_MARK, NULL, &g);
1862b15c8340SDag-Erling Smørgrav 
1863b15c8340SDag-Erling Smørgrav 	/* Determine length of pwd so we can trim completion display */
1864b15c8340SDag-Erling Smørgrav 	for (hadglob = tmplen = pwdlen = 0; tmp[tmplen] != 0; tmplen++) {
1865b15c8340SDag-Erling Smørgrav 		/* Terminate counting on first unescaped glob metacharacter */
1866b15c8340SDag-Erling Smørgrav 		if (tmp[tmplen] == '*' || tmp[tmplen] == '?') {
1867b15c8340SDag-Erling Smørgrav 			if (tmp[tmplen] != '*' || tmp[tmplen + 1] != '\0')
1868b15c8340SDag-Erling Smørgrav 				hadglob = 1;
1869b15c8340SDag-Erling Smørgrav 			break;
1870b15c8340SDag-Erling Smørgrav 		}
1871b15c8340SDag-Erling Smørgrav 		if (tmp[tmplen] == '\\' && tmp[tmplen + 1] != '\0')
1872b15c8340SDag-Erling Smørgrav 			tmplen++;
1873b15c8340SDag-Erling Smørgrav 		if (tmp[tmplen] == '/')
1874b15c8340SDag-Erling Smørgrav 			pwdlen = tmplen + 1;	/* track last seen '/' */
1875b15c8340SDag-Erling Smørgrav 	}
1876e4a9863fSDag-Erling Smørgrav 	free(tmp);
1877a0ee8cc6SDag-Erling Smørgrav 	tmp = NULL;
1878b15c8340SDag-Erling Smørgrav 
1879b15c8340SDag-Erling Smørgrav 	if (g.gl_matchc == 0)
1880b15c8340SDag-Erling Smørgrav 		goto out;
1881b15c8340SDag-Erling Smørgrav 
1882b15c8340SDag-Erling Smørgrav 	if (g.gl_matchc > 1)
1883b15c8340SDag-Erling Smørgrav 		complete_display(g.gl_pathv, pwdlen);
1884b15c8340SDag-Erling Smørgrav 
1885b15c8340SDag-Erling Smørgrav 	/* Don't try to extend globs */
1886b15c8340SDag-Erling Smørgrav 	if (file == NULL || hadglob)
1887b15c8340SDag-Erling Smørgrav 		goto out;
1888b15c8340SDag-Erling Smørgrav 
1889b15c8340SDag-Erling Smørgrav 	tmp2 = complete_ambiguous(file, g.gl_pathv, g.gl_matchc);
18906888a9beSDag-Erling Smørgrav 	tmp = path_strip(tmp2, isabs ? NULL : remote_path);
1891e4a9863fSDag-Erling Smørgrav 	free(tmp2);
1892b15c8340SDag-Erling Smørgrav 
1893b15c8340SDag-Erling Smørgrav 	if (tmp == NULL)
1894b15c8340SDag-Erling Smørgrav 		goto out;
1895b15c8340SDag-Erling Smørgrav 
1896b15c8340SDag-Erling Smørgrav 	tmplen = strlen(tmp);
1897b15c8340SDag-Erling Smørgrav 	filelen = strlen(file);
1898b15c8340SDag-Erling Smørgrav 
18996888a9beSDag-Erling Smørgrav 	/* Count the number of escaped characters in the input string. */
19006888a9beSDag-Erling Smørgrav 	cesc = isesc = 0;
19016888a9beSDag-Erling Smørgrav 	for (i = 0; i < filelen; i++) {
19026888a9beSDag-Erling Smørgrav 		if (!isesc && file[i] == '\\' && i + 1 < filelen){
19036888a9beSDag-Erling Smørgrav 			isesc = 1;
19046888a9beSDag-Erling Smørgrav 			cesc++;
19056888a9beSDag-Erling Smørgrav 		} else
19066888a9beSDag-Erling Smørgrav 			isesc = 0;
19076888a9beSDag-Erling Smørgrav 	}
19086888a9beSDag-Erling Smørgrav 
19096888a9beSDag-Erling Smørgrav 	if (tmplen > (filelen - cesc)) {
19106888a9beSDag-Erling Smørgrav 		tmp2 = tmp + filelen - cesc;
1911b15c8340SDag-Erling Smørgrav 		len = strlen(tmp2);
1912b15c8340SDag-Erling Smørgrav 		/* quote argument on way out */
1913e4a9863fSDag-Erling Smørgrav 		for (i = 0; i < len; i += clen) {
1914e4a9863fSDag-Erling Smørgrav 			if ((clen = mblen(tmp2 + i, len - i)) < 0 ||
1915e4a9863fSDag-Erling Smørgrav 			    (size_t)clen > sizeof(ins) - 2)
1916e4a9863fSDag-Erling Smørgrav 				fatal("invalid multibyte character");
1917b15c8340SDag-Erling Smørgrav 			ins[0] = '\\';
1918e4a9863fSDag-Erling Smørgrav 			memcpy(ins + 1, tmp2 + i, clen);
1919e4a9863fSDag-Erling Smørgrav 			ins[clen + 1] = '\0';
1920b15c8340SDag-Erling Smørgrav 			switch (tmp2[i]) {
1921b15c8340SDag-Erling Smørgrav 			case '\'':
1922b15c8340SDag-Erling Smørgrav 			case '"':
1923b15c8340SDag-Erling Smørgrav 			case '\\':
1924b15c8340SDag-Erling Smørgrav 			case '\t':
19254a421b63SDag-Erling Smørgrav 			case '[':
1926b15c8340SDag-Erling Smørgrav 			case ' ':
19276888a9beSDag-Erling Smørgrav 			case '#':
19286888a9beSDag-Erling Smørgrav 			case '*':
1929b15c8340SDag-Erling Smørgrav 				if (quote == '\0' || tmp2[i] == quote) {
1930b15c8340SDag-Erling Smørgrav 					if (el_insertstr(el, ins) == -1)
1931b15c8340SDag-Erling Smørgrav 						fatal("el_insertstr "
1932b15c8340SDag-Erling Smørgrav 						    "failed.");
1933b15c8340SDag-Erling Smørgrav 					break;
1934b15c8340SDag-Erling Smørgrav 				}
1935b15c8340SDag-Erling Smørgrav 				/* FALLTHROUGH */
1936b15c8340SDag-Erling Smørgrav 			default:
1937b15c8340SDag-Erling Smørgrav 				if (el_insertstr(el, ins + 1) == -1)
1938b15c8340SDag-Erling Smørgrav 					fatal("el_insertstr failed.");
1939b15c8340SDag-Erling Smørgrav 				break;
1940b15c8340SDag-Erling Smørgrav 			}
1941b15c8340SDag-Erling Smørgrav 		}
1942b15c8340SDag-Erling Smørgrav 	}
1943b15c8340SDag-Erling Smørgrav 
1944b15c8340SDag-Erling Smørgrav 	lf = el_line(el);
1945b15c8340SDag-Erling Smørgrav 	if (g.gl_matchc == 1) {
1946b15c8340SDag-Erling Smørgrav 		i = 0;
1947a0ee8cc6SDag-Erling Smørgrav 		if (!terminated && quote != '\0')
1948b15c8340SDag-Erling Smørgrav 			ins[i++] = quote;
1949b15c8340SDag-Erling Smørgrav 		if (*(lf->cursor - 1) != '/' &&
1950b15c8340SDag-Erling Smørgrav 		    (lastarg || *(lf->cursor) != ' '))
1951b15c8340SDag-Erling Smørgrav 			ins[i++] = ' ';
1952b15c8340SDag-Erling Smørgrav 		ins[i] = '\0';
1953b15c8340SDag-Erling Smørgrav 		if (i > 0 && el_insertstr(el, ins) == -1)
1954b15c8340SDag-Erling Smørgrav 			fatal("el_insertstr failed.");
1955b15c8340SDag-Erling Smørgrav 	}
1956e4a9863fSDag-Erling Smørgrav 	free(tmp);
1957b15c8340SDag-Erling Smørgrav 
1958b15c8340SDag-Erling Smørgrav  out:
1959b15c8340SDag-Erling Smørgrav 	globfree(&g);
1960b15c8340SDag-Erling Smørgrav 	return g.gl_matchc;
1961b15c8340SDag-Erling Smørgrav }
1962b15c8340SDag-Erling Smørgrav 
1963b15c8340SDag-Erling Smørgrav /* tab-completion hook function, called via libedit */
1964b15c8340SDag-Erling Smørgrav static unsigned char
1965b15c8340SDag-Erling Smørgrav complete(EditLine *el, int ch)
1966b15c8340SDag-Erling Smørgrav {
1967b15c8340SDag-Erling Smørgrav 	char **argv, *line, quote;
1968e4a9863fSDag-Erling Smørgrav 	int argc, carg;
1969e4a9863fSDag-Erling Smørgrav 	u_int cursor, len, terminated, ret = CC_ERROR;
1970b15c8340SDag-Erling Smørgrav 	const LineInfo *lf;
1971b15c8340SDag-Erling Smørgrav 	struct complete_ctx *complete_ctx;
1972b15c8340SDag-Erling Smørgrav 
1973b15c8340SDag-Erling Smørgrav 	lf = el_line(el);
1974b15c8340SDag-Erling Smørgrav 	if (el_get(el, EL_CLIENTDATA, (void**)&complete_ctx) != 0)
1975b15c8340SDag-Erling Smørgrav 		fatal("%s: el_get failed", __func__);
1976b15c8340SDag-Erling Smørgrav 
1977b15c8340SDag-Erling Smørgrav 	/* Figure out which argument the cursor points to */
1978b15c8340SDag-Erling Smørgrav 	cursor = lf->cursor - lf->buffer;
1979fc1ba28aSDag-Erling Smørgrav 	line = xmalloc(cursor + 1);
1980b15c8340SDag-Erling Smørgrav 	memcpy(line, lf->buffer, cursor);
1981b15c8340SDag-Erling Smørgrav 	line[cursor] = '\0';
1982b15c8340SDag-Erling Smørgrav 	argv = makeargv(line, &carg, 1, &quote, &terminated);
1983e4a9863fSDag-Erling Smørgrav 	free(line);
1984b15c8340SDag-Erling Smørgrav 
1985b15c8340SDag-Erling Smørgrav 	/* Get all the arguments on the line */
1986b15c8340SDag-Erling Smørgrav 	len = lf->lastchar - lf->buffer;
1987fc1ba28aSDag-Erling Smørgrav 	line = xmalloc(len + 1);
1988b15c8340SDag-Erling Smørgrav 	memcpy(line, lf->buffer, len);
1989b15c8340SDag-Erling Smørgrav 	line[len] = '\0';
1990b15c8340SDag-Erling Smørgrav 	argv = makeargv(line, &argc, 1, NULL, NULL);
1991b15c8340SDag-Erling Smørgrav 
1992b15c8340SDag-Erling Smørgrav 	/* Ensure cursor is at EOL or a argument boundary */
1993b15c8340SDag-Erling Smørgrav 	if (line[cursor] != ' ' && line[cursor] != '\0' &&
1994b15c8340SDag-Erling Smørgrav 	    line[cursor] != '\n') {
1995e4a9863fSDag-Erling Smørgrav 		free(line);
1996b15c8340SDag-Erling Smørgrav 		return ret;
1997b15c8340SDag-Erling Smørgrav 	}
1998b15c8340SDag-Erling Smørgrav 
1999b15c8340SDag-Erling Smørgrav 	if (carg == 0) {
2000b15c8340SDag-Erling Smørgrav 		/* Show all available commands */
2001b15c8340SDag-Erling Smørgrav 		complete_cmd_parse(el, NULL, argc == carg, '\0', 1);
2002b15c8340SDag-Erling Smørgrav 		ret = CC_REDISPLAY;
2003b15c8340SDag-Erling Smørgrav 	} else if (carg == 1 && cursor > 0 && line[cursor - 1] != ' ')  {
2004b15c8340SDag-Erling Smørgrav 		/* Handle the command parsing */
2005b15c8340SDag-Erling Smørgrav 		if (complete_cmd_parse(el, argv[0], argc == carg,
2006b15c8340SDag-Erling Smørgrav 		    quote, terminated) != 0)
2007b15c8340SDag-Erling Smørgrav 			ret = CC_REDISPLAY;
2008b15c8340SDag-Erling Smørgrav 	} else if (carg >= 1) {
2009b15c8340SDag-Erling Smørgrav 		/* Handle file parsing */
2010b15c8340SDag-Erling Smørgrav 		int remote = complete_is_remote(argv[0]);
2011b15c8340SDag-Erling Smørgrav 		char *filematch = NULL;
2012b15c8340SDag-Erling Smørgrav 
2013b15c8340SDag-Erling Smørgrav 		if (carg > 1 && line[cursor-1] != ' ')
2014b15c8340SDag-Erling Smørgrav 			filematch = argv[carg - 1];
2015b15c8340SDag-Erling Smørgrav 
2016b15c8340SDag-Erling Smørgrav 		if (remote != 0 &&
2017b15c8340SDag-Erling Smørgrav 		    complete_match(el, complete_ctx->conn,
2018b15c8340SDag-Erling Smørgrav 		    *complete_ctx->remote_pathp, filematch,
2019b15c8340SDag-Erling Smørgrav 		    remote, carg == argc, quote, terminated) != 0)
2020b15c8340SDag-Erling Smørgrav 			ret = CC_REDISPLAY;
2021b15c8340SDag-Erling Smørgrav 	}
2022b15c8340SDag-Erling Smørgrav 
2023e4a9863fSDag-Erling Smørgrav 	free(line);
2024b15c8340SDag-Erling Smørgrav 	return ret;
2025b15c8340SDag-Erling Smørgrav }
2026b15c8340SDag-Erling Smørgrav #endif /* USE_LIBEDIT */
20275e8dbd04SDag-Erling Smørgrav 
2028efcad6b7SDag-Erling Smørgrav int
2029b15c8340SDag-Erling Smørgrav interactive_loop(struct sftp_conn *conn, char *file1, char *file2)
2030efcad6b7SDag-Erling Smørgrav {
2031b15c8340SDag-Erling Smørgrav 	char *remote_path;
2032efcad6b7SDag-Erling Smørgrav 	char *dir = NULL;
2033efcad6b7SDag-Erling Smørgrav 	char cmd[2048];
2034043840dfSDag-Erling Smørgrav 	int err, interactive;
20355e8dbd04SDag-Erling Smørgrav 	EditLine *el = NULL;
20365e8dbd04SDag-Erling Smørgrav #ifdef USE_LIBEDIT
20375e8dbd04SDag-Erling Smørgrav 	History *hl = NULL;
20385e8dbd04SDag-Erling Smørgrav 	HistEvent hev;
20395e8dbd04SDag-Erling Smørgrav 	extern char *__progname;
2040b15c8340SDag-Erling Smørgrav 	struct complete_ctx complete_ctx;
20415e8dbd04SDag-Erling Smørgrav 
20425e8dbd04SDag-Erling Smørgrav 	if (!batchmode && isatty(STDIN_FILENO)) {
20435e8dbd04SDag-Erling Smørgrav 		if ((el = el_init(__progname, stdin, stdout, stderr)) == NULL)
20445e8dbd04SDag-Erling Smørgrav 			fatal("Couldn't initialise editline");
20455e8dbd04SDag-Erling Smørgrav 		if ((hl = history_init()) == NULL)
20465e8dbd04SDag-Erling Smørgrav 			fatal("Couldn't initialise editline history");
20475e8dbd04SDag-Erling Smørgrav 		history(hl, &hev, H_SETSIZE, 100);
20485e8dbd04SDag-Erling Smørgrav 		el_set(el, EL_HIST, history, hl);
20495e8dbd04SDag-Erling Smørgrav 
20505e8dbd04SDag-Erling Smørgrav 		el_set(el, EL_PROMPT, prompt);
20515e8dbd04SDag-Erling Smørgrav 		el_set(el, EL_EDITOR, "emacs");
20525e8dbd04SDag-Erling Smørgrav 		el_set(el, EL_TERMINAL, NULL);
20535e8dbd04SDag-Erling Smørgrav 		el_set(el, EL_SIGNAL, 1);
20545e8dbd04SDag-Erling Smørgrav 		el_source(el, NULL);
2055b15c8340SDag-Erling Smørgrav 
2056b15c8340SDag-Erling Smørgrav 		/* Tab Completion */
2057b15c8340SDag-Erling Smørgrav 		el_set(el, EL_ADDFN, "ftp-complete",
20584a421b63SDag-Erling Smørgrav 		    "Context sensitive argument completion", complete);
2059b15c8340SDag-Erling Smørgrav 		complete_ctx.conn = conn;
2060b15c8340SDag-Erling Smørgrav 		complete_ctx.remote_pathp = &remote_path;
2061b15c8340SDag-Erling Smørgrav 		el_set(el, EL_CLIENTDATA, (void*)&complete_ctx);
2062b15c8340SDag-Erling Smørgrav 		el_set(el, EL_BIND, "^I", "ftp-complete", NULL);
2063f7167e0eSDag-Erling Smørgrav 		/* enable ctrl-left-arrow and ctrl-right-arrow */
2064f7167e0eSDag-Erling Smørgrav 		el_set(el, EL_BIND, "\\e[1;5C", "em-next-word", NULL);
2065f7167e0eSDag-Erling Smørgrav 		el_set(el, EL_BIND, "\\e[5C", "em-next-word", NULL);
2066f7167e0eSDag-Erling Smørgrav 		el_set(el, EL_BIND, "\\e[1;5D", "ed-prev-word", NULL);
2067f7167e0eSDag-Erling Smørgrav 		el_set(el, EL_BIND, "\\e\\e[D", "ed-prev-word", NULL);
2068f7167e0eSDag-Erling Smørgrav 		/* make ^w match ksh behaviour */
2069f7167e0eSDag-Erling Smørgrav 		el_set(el, EL_BIND, "^w", "ed-delete-prev-word", NULL);
20705e8dbd04SDag-Erling Smørgrav 	}
20715e8dbd04SDag-Erling Smørgrav #endif /* USE_LIBEDIT */
2072efcad6b7SDag-Erling Smørgrav 
2073b15c8340SDag-Erling Smørgrav 	remote_path = do_realpath(conn, ".");
2074b15c8340SDag-Erling Smørgrav 	if (remote_path == NULL)
2075efcad6b7SDag-Erling Smørgrav 		fatal("Need cwd");
2076efcad6b7SDag-Erling Smørgrav 
2077efcad6b7SDag-Erling Smørgrav 	if (file1 != NULL) {
2078efcad6b7SDag-Erling Smørgrav 		dir = xstrdup(file1);
2079b15c8340SDag-Erling Smørgrav 		dir = make_absolute(dir, remote_path);
2080efcad6b7SDag-Erling Smørgrav 
2081efcad6b7SDag-Erling Smørgrav 		if (remote_is_dir(conn, dir) && file2 == NULL) {
2082e4a9863fSDag-Erling Smørgrav 			if (!quiet)
2083076ad2f8SDag-Erling Smørgrav 				mprintf("Changing to: %s\n", dir);
2084efcad6b7SDag-Erling Smørgrav 			snprintf(cmd, sizeof cmd, "cd \"%s\"", dir);
2085b15c8340SDag-Erling Smørgrav 			if (parse_dispatch_command(conn, cmd,
2086b15c8340SDag-Erling Smørgrav 			    &remote_path, 1) != 0) {
2087e4a9863fSDag-Erling Smørgrav 				free(dir);
2088e4a9863fSDag-Erling Smørgrav 				free(remote_path);
2089e4a9863fSDag-Erling Smørgrav 				free(conn);
2090efcad6b7SDag-Erling Smørgrav 				return (-1);
20915e8dbd04SDag-Erling Smørgrav 			}
2092efcad6b7SDag-Erling Smørgrav 		} else {
20936888a9beSDag-Erling Smørgrav 			/* XXX this is wrong wrt quoting */
2094e4a9863fSDag-Erling Smørgrav 			snprintf(cmd, sizeof cmd, "get%s %s%s%s",
2095e4a9863fSDag-Erling Smørgrav 			    global_aflag ? " -a" : "", dir,
2096e4a9863fSDag-Erling Smørgrav 			    file2 == NULL ? "" : " ",
2097e4a9863fSDag-Erling Smørgrav 			    file2 == NULL ? "" : file2);
2098b15c8340SDag-Erling Smørgrav 			err = parse_dispatch_command(conn, cmd,
2099b15c8340SDag-Erling Smørgrav 			    &remote_path, 1);
2100e4a9863fSDag-Erling Smørgrav 			free(dir);
2101e4a9863fSDag-Erling Smørgrav 			free(remote_path);
2102e4a9863fSDag-Erling Smørgrav 			free(conn);
2103efcad6b7SDag-Erling Smørgrav 			return (err);
2104efcad6b7SDag-Erling Smørgrav 		}
2105e4a9863fSDag-Erling Smørgrav 		free(dir);
2106efcad6b7SDag-Erling Smørgrav 	}
2107efcad6b7SDag-Erling Smørgrav 
2108bc5531deSDag-Erling Smørgrav 	setvbuf(stdout, NULL, _IOLBF, 0);
2109bc5531deSDag-Erling Smørgrav 	setvbuf(infile, NULL, _IOLBF, 0);
2110efcad6b7SDag-Erling Smørgrav 
2111043840dfSDag-Erling Smørgrav 	interactive = !batchmode && isatty(STDIN_FILENO);
2112efcad6b7SDag-Erling Smørgrav 	err = 0;
2113efcad6b7SDag-Erling Smørgrav 	for (;;) {
2114efcad6b7SDag-Erling Smørgrav 		char *cp;
2115efcad6b7SDag-Erling Smørgrav 
2116d74d50a8SDag-Erling Smørgrav 		signal(SIGINT, SIG_IGN);
2117d74d50a8SDag-Erling Smørgrav 
21185e8dbd04SDag-Erling Smørgrav 		if (el == NULL) {
2119043840dfSDag-Erling Smørgrav 			if (interactive)
2120efcad6b7SDag-Erling Smørgrav 				printf("sftp> ");
2121efcad6b7SDag-Erling Smørgrav 			if (fgets(cmd, sizeof(cmd), infile) == NULL) {
2122043840dfSDag-Erling Smørgrav 				if (interactive)
2123efcad6b7SDag-Erling Smørgrav 					printf("\n");
2124efcad6b7SDag-Erling Smørgrav 				break;
2125efcad6b7SDag-Erling Smørgrav 			}
2126043840dfSDag-Erling Smørgrav 			if (!interactive) { /* Echo command */
2127076ad2f8SDag-Erling Smørgrav 				mprintf("sftp> %s", cmd);
2128043840dfSDag-Erling Smørgrav 				if (strlen(cmd) > 0 &&
2129043840dfSDag-Erling Smørgrav 				    cmd[strlen(cmd) - 1] != '\n')
2130043840dfSDag-Erling Smørgrav 					printf("\n");
2131043840dfSDag-Erling Smørgrav 			}
21325e8dbd04SDag-Erling Smørgrav 		} else {
21335e8dbd04SDag-Erling Smørgrav #ifdef USE_LIBEDIT
21345e8dbd04SDag-Erling Smørgrav 			const char *line;
21355e8dbd04SDag-Erling Smørgrav 			int count = 0;
21365e8dbd04SDag-Erling Smørgrav 
2137b15c8340SDag-Erling Smørgrav 			if ((line = el_gets(el, &count)) == NULL ||
2138b15c8340SDag-Erling Smørgrav 			    count <= 0) {
2139043840dfSDag-Erling Smørgrav 				printf("\n");
21405e8dbd04SDag-Erling Smørgrav  				break;
2141043840dfSDag-Erling Smørgrav 			}
21425e8dbd04SDag-Erling Smørgrav 			history(hl, &hev, H_ENTER, line);
21435e8dbd04SDag-Erling Smørgrav 			if (strlcpy(cmd, line, sizeof(cmd)) >= sizeof(cmd)) {
21445e8dbd04SDag-Erling Smørgrav 				fprintf(stderr, "Error: input line too long\n");
21455e8dbd04SDag-Erling Smørgrav 				continue;
21465e8dbd04SDag-Erling Smørgrav 			}
21475e8dbd04SDag-Erling Smørgrav #endif /* USE_LIBEDIT */
21485e8dbd04SDag-Erling Smørgrav 		}
2149efcad6b7SDag-Erling Smørgrav 
2150efcad6b7SDag-Erling Smørgrav 		cp = strrchr(cmd, '\n');
2151efcad6b7SDag-Erling Smørgrav 		if (cp)
2152efcad6b7SDag-Erling Smørgrav 			*cp = '\0';
2153efcad6b7SDag-Erling Smørgrav 
2154d74d50a8SDag-Erling Smørgrav 		/* Handle user interrupts gracefully during commands */
2155d74d50a8SDag-Erling Smørgrav 		interrupted = 0;
2156d74d50a8SDag-Erling Smørgrav 		signal(SIGINT, cmd_interrupt);
2157d74d50a8SDag-Erling Smørgrav 
2158b15c8340SDag-Erling Smørgrav 		err = parse_dispatch_command(conn, cmd, &remote_path,
2159b15c8340SDag-Erling Smørgrav 		    batchmode);
2160efcad6b7SDag-Erling Smørgrav 		if (err != 0)
2161efcad6b7SDag-Erling Smørgrav 			break;
2162efcad6b7SDag-Erling Smørgrav 	}
2163e4a9863fSDag-Erling Smørgrav 	free(remote_path);
2164e4a9863fSDag-Erling Smørgrav 	free(conn);
2165efcad6b7SDag-Erling Smørgrav 
2166043840dfSDag-Erling Smørgrav #ifdef USE_LIBEDIT
2167043840dfSDag-Erling Smørgrav 	if (el != NULL)
2168043840dfSDag-Erling Smørgrav 		el_end(el);
2169043840dfSDag-Erling Smørgrav #endif /* USE_LIBEDIT */
2170043840dfSDag-Erling Smørgrav 
2171efcad6b7SDag-Erling Smørgrav 	/* err == 1 signifies normal "quit" exit */
2172efcad6b7SDag-Erling Smørgrav 	return (err >= 0 ? 0 : -1);
2173efcad6b7SDag-Erling Smørgrav }
2174d0c8c0bcSDag-Erling Smørgrav 
2175ae1f160dSDag-Erling Smørgrav static void
2176d95e11bfSDag-Erling Smørgrav connect_to_server(char *path, char **args, int *in, int *out)
21771e8db6e2SBrian Feldman {
21781e8db6e2SBrian Feldman 	int c_in, c_out;
2179ee21a45fSDag-Erling Smørgrav 
21801e8db6e2SBrian Feldman #ifdef USE_PIPES
21811e8db6e2SBrian Feldman 	int pin[2], pout[2];
2182ee21a45fSDag-Erling Smørgrav 
21831e8db6e2SBrian Feldman 	if ((pipe(pin) == -1) || (pipe(pout) == -1))
21841e8db6e2SBrian Feldman 		fatal("pipe: %s", strerror(errno));
21851e8db6e2SBrian Feldman 	*in = pin[0];
21861e8db6e2SBrian Feldman 	*out = pout[1];
21871e8db6e2SBrian Feldman 	c_in = pout[0];
21881e8db6e2SBrian Feldman 	c_out = pin[1];
21891e8db6e2SBrian Feldman #else /* USE_PIPES */
21901e8db6e2SBrian Feldman 	int inout[2];
2191ee21a45fSDag-Erling Smørgrav 
21921e8db6e2SBrian Feldman 	if (socketpair(AF_UNIX, SOCK_STREAM, 0, inout) == -1)
21931e8db6e2SBrian Feldman 		fatal("socketpair: %s", strerror(errno));
21941e8db6e2SBrian Feldman 	*in = *out = inout[0];
21951e8db6e2SBrian Feldman 	c_in = c_out = inout[1];
21961e8db6e2SBrian Feldman #endif /* USE_PIPES */
21971e8db6e2SBrian Feldman 
2198d95e11bfSDag-Erling Smørgrav 	if ((sshpid = fork()) == -1)
21991e8db6e2SBrian Feldman 		fatal("fork: %s", strerror(errno));
2200d95e11bfSDag-Erling Smørgrav 	else if (sshpid == 0) {
22011e8db6e2SBrian Feldman 		if ((dup2(c_in, STDIN_FILENO) == -1) ||
22021e8db6e2SBrian Feldman 		    (dup2(c_out, STDOUT_FILENO) == -1)) {
22031e8db6e2SBrian Feldman 			fprintf(stderr, "dup2: %s\n", strerror(errno));
2204d74d50a8SDag-Erling Smørgrav 			_exit(1);
22051e8db6e2SBrian Feldman 		}
22061e8db6e2SBrian Feldman 		close(*in);
22071e8db6e2SBrian Feldman 		close(*out);
22081e8db6e2SBrian Feldman 		close(c_in);
22091e8db6e2SBrian Feldman 		close(c_out);
2210d74d50a8SDag-Erling Smørgrav 
2211d74d50a8SDag-Erling Smørgrav 		/*
2212d74d50a8SDag-Erling Smørgrav 		 * The underlying ssh is in the same process group, so we must
2213d74d50a8SDag-Erling Smørgrav 		 * ignore SIGINT if we want to gracefully abort commands,
2214d74d50a8SDag-Erling Smørgrav 		 * otherwise the signal will make it to the ssh process and
2215b15c8340SDag-Erling Smørgrav 		 * kill it too.  Contrawise, since sftp sends SIGTERMs to the
2216b15c8340SDag-Erling Smørgrav 		 * underlying ssh, it must *not* ignore that signal.
2217d74d50a8SDag-Erling Smørgrav 		 */
2218d74d50a8SDag-Erling Smørgrav 		signal(SIGINT, SIG_IGN);
2219b15c8340SDag-Erling Smørgrav 		signal(SIGTERM, SIG_DFL);
2220d74d50a8SDag-Erling Smørgrav 		execvp(path, args);
2221ae1f160dSDag-Erling Smørgrav 		fprintf(stderr, "exec: %s: %s\n", path, strerror(errno));
2222d74d50a8SDag-Erling Smørgrav 		_exit(1);
22231e8db6e2SBrian Feldman 	}
22241e8db6e2SBrian Feldman 
2225d95e11bfSDag-Erling Smørgrav 	signal(SIGTERM, killchild);
2226d95e11bfSDag-Erling Smørgrav 	signal(SIGINT, killchild);
2227d95e11bfSDag-Erling Smørgrav 	signal(SIGHUP, killchild);
2228*ca86bcf2SDag-Erling Smørgrav 	signal(SIGTSTP, suspchild);
2229*ca86bcf2SDag-Erling Smørgrav 	signal(SIGTTIN, suspchild);
2230*ca86bcf2SDag-Erling Smørgrav 	signal(SIGTTOU, suspchild);
22311e8db6e2SBrian Feldman 	close(c_in);
22321e8db6e2SBrian Feldman 	close(c_out);
22331e8db6e2SBrian Feldman }
22341e8db6e2SBrian Feldman 
2235ae1f160dSDag-Erling Smørgrav static void
22361e8db6e2SBrian Feldman usage(void)
22371e8db6e2SBrian Feldman {
2238ae1f160dSDag-Erling Smørgrav 	extern char *__progname;
2239ae1f160dSDag-Erling Smørgrav 
2240ae1f160dSDag-Erling Smørgrav 	fprintf(stderr,
2241f7167e0eSDag-Erling Smørgrav 	    "usage: %s [-1246aCfpqrv] [-B buffer_size] [-b batchfile] [-c cipher]\n"
2242b15c8340SDag-Erling Smørgrav 	    "          [-D sftp_server_path] [-F ssh_config] "
22434a421b63SDag-Erling Smørgrav 	    "[-i identity_file] [-l limit]\n"
2244b15c8340SDag-Erling Smørgrav 	    "          [-o ssh_option] [-P port] [-R num_requests] "
2245b15c8340SDag-Erling Smørgrav 	    "[-S program]\n"
2246b15c8340SDag-Erling Smørgrav 	    "          [-s subsystem | sftp_server] host\n"
2247cce7d346SDag-Erling Smørgrav 	    "       %s [user@]host[:file ...]\n"
2248cce7d346SDag-Erling Smørgrav 	    "       %s [user@]host[:dir[/]]\n"
2249b15c8340SDag-Erling Smørgrav 	    "       %s -b batchfile [user@]host\n",
2250b15c8340SDag-Erling Smørgrav 	    __progname, __progname, __progname, __progname);
22511e8db6e2SBrian Feldman 	exit(1);
22521e8db6e2SBrian Feldman }
22531e8db6e2SBrian Feldman 
22541e8db6e2SBrian Feldman int
22551e8db6e2SBrian Feldman main(int argc, char **argv)
22561e8db6e2SBrian Feldman {
2257d0c8c0bcSDag-Erling Smørgrav 	int in, out, ch, err;
2258b15c8340SDag-Erling Smørgrav 	char *host = NULL, *userhost, *cp, *file2 = NULL;
2259ae1f160dSDag-Erling Smørgrav 	int debug_level = 0, sshver = 2;
2260ae1f160dSDag-Erling Smørgrav 	char *file1 = NULL, *sftp_server = NULL;
2261ae1f160dSDag-Erling Smørgrav 	char *ssh_program = _PATH_SSH_PROGRAM, *sftp_direct = NULL;
22624a421b63SDag-Erling Smørgrav 	const char *errstr;
2263ae1f160dSDag-Erling Smørgrav 	LogLevel ll = SYSLOG_LEVEL_INFO;
2264ae1f160dSDag-Erling Smørgrav 	arglist args;
22651e8db6e2SBrian Feldman 	extern int optind;
22661e8db6e2SBrian Feldman 	extern char *optarg;
2267b15c8340SDag-Erling Smørgrav 	struct sftp_conn *conn;
2268b15c8340SDag-Erling Smørgrav 	size_t copy_buffer_len = DEFAULT_COPY_BUFLEN;
2269b15c8340SDag-Erling Smørgrav 	size_t num_requests = DEFAULT_NUM_REQUESTS;
22704a421b63SDag-Erling Smørgrav 	long long limit_kbps = 0;
22711e8db6e2SBrian Feldman 
2272acc1a9efSDag-Erling Smørgrav 	ssh_malloc_init();	/* must be called before any mallocs */
2273021d409fSDag-Erling Smørgrav 	/* Ensure that fds 0, 1 and 2 are open or directed to /dev/null */
2274021d409fSDag-Erling Smørgrav 	sanitise_stdfd();
2275*ca86bcf2SDag-Erling Smørgrav 	msetlocale();
2276021d409fSDag-Erling Smørgrav 
2277d95e11bfSDag-Erling Smørgrav 	__progname = ssh_get_progname(argv[0]);
2278021d409fSDag-Erling Smørgrav 	memset(&args, '\0', sizeof(args));
2279ae1f160dSDag-Erling Smørgrav 	args.list = NULL;
2280761efaa7SDag-Erling Smørgrav 	addargs(&args, "%s", ssh_program);
2281ae1f160dSDag-Erling Smørgrav 	addargs(&args, "-oForwardX11 no");
2282ae1f160dSDag-Erling Smørgrav 	addargs(&args, "-oForwardAgent no");
2283021d409fSDag-Erling Smørgrav 	addargs(&args, "-oPermitLocalCommand no");
2284ae1f160dSDag-Erling Smørgrav 	addargs(&args, "-oClearAllForwardings yes");
2285efcad6b7SDag-Erling Smørgrav 
2286ae1f160dSDag-Erling Smørgrav 	ll = SYSLOG_LEVEL_INFO;
2287efcad6b7SDag-Erling Smørgrav 	infile = stdin;
22881e8db6e2SBrian Feldman 
2289b15c8340SDag-Erling Smørgrav 	while ((ch = getopt(argc, argv,
2290f7167e0eSDag-Erling Smørgrav 	    "1246afhpqrvCc:D:i:l:o:s:S:b:B:F:P:R:")) != -1) {
22911e8db6e2SBrian Feldman 		switch (ch) {
2292b15c8340SDag-Erling Smørgrav 		/* Passed through to ssh(1) */
2293b15c8340SDag-Erling Smørgrav 		case '4':
2294b15c8340SDag-Erling Smørgrav 		case '6':
22951e8db6e2SBrian Feldman 		case 'C':
2296b15c8340SDag-Erling Smørgrav 			addargs(&args, "-%c", ch);
2297b15c8340SDag-Erling Smørgrav 			break;
2298b15c8340SDag-Erling Smørgrav 		/* Passed through to ssh(1) with argument */
2299b15c8340SDag-Erling Smørgrav 		case 'F':
2300b15c8340SDag-Erling Smørgrav 		case 'c':
2301b15c8340SDag-Erling Smørgrav 		case 'i':
2302b15c8340SDag-Erling Smørgrav 		case 'o':
2303b15c8340SDag-Erling Smørgrav 			addargs(&args, "-%c", ch);
2304b15c8340SDag-Erling Smørgrav 			addargs(&args, "%s", optarg);
2305b15c8340SDag-Erling Smørgrav 			break;
2306b15c8340SDag-Erling Smørgrav 		case 'q':
2307e4a9863fSDag-Erling Smørgrav 			ll = SYSLOG_LEVEL_ERROR;
2308e4a9863fSDag-Erling Smørgrav 			quiet = 1;
2309b15c8340SDag-Erling Smørgrav 			showprogress = 0;
2310b15c8340SDag-Erling Smørgrav 			addargs(&args, "-%c", ch);
2311b15c8340SDag-Erling Smørgrav 			break;
2312b15c8340SDag-Erling Smørgrav 		case 'P':
2313b15c8340SDag-Erling Smørgrav 			addargs(&args, "-oPort %s", optarg);
23141e8db6e2SBrian Feldman 			break;
23151e8db6e2SBrian Feldman 		case 'v':
2316ae1f160dSDag-Erling Smørgrav 			if (debug_level < 3) {
2317ae1f160dSDag-Erling Smørgrav 				addargs(&args, "-v");
2318ae1f160dSDag-Erling Smørgrav 				ll = SYSLOG_LEVEL_DEBUG1 + debug_level;
2319ae1f160dSDag-Erling Smørgrav 			}
2320ae1f160dSDag-Erling Smørgrav 			debug_level++;
23211e8db6e2SBrian Feldman 			break;
23221e8db6e2SBrian Feldman 		case '1':
2323ae1f160dSDag-Erling Smørgrav 			sshver = 1;
23241e8db6e2SBrian Feldman 			if (sftp_server == NULL)
23251e8db6e2SBrian Feldman 				sftp_server = _PATH_SFTP_SERVER;
23261e8db6e2SBrian Feldman 			break;
2327b15c8340SDag-Erling Smørgrav 		case '2':
2328b15c8340SDag-Erling Smørgrav 			sshver = 2;
23291e8db6e2SBrian Feldman 			break;
2330e4a9863fSDag-Erling Smørgrav 		case 'a':
2331e4a9863fSDag-Erling Smørgrav 			global_aflag = 1;
2332e4a9863fSDag-Erling Smørgrav 			break;
2333b15c8340SDag-Erling Smørgrav 		case 'B':
2334b15c8340SDag-Erling Smørgrav 			copy_buffer_len = strtol(optarg, &cp, 10);
2335b15c8340SDag-Erling Smørgrav 			if (copy_buffer_len == 0 || *cp != '\0')
2336b15c8340SDag-Erling Smørgrav 				fatal("Invalid buffer size \"%s\"", optarg);
23371e8db6e2SBrian Feldman 			break;
23381e8db6e2SBrian Feldman 		case 'b':
2339efcad6b7SDag-Erling Smørgrav 			if (batchmode)
2340efcad6b7SDag-Erling Smørgrav 				fatal("Batch file already specified.");
2341efcad6b7SDag-Erling Smørgrav 
2342efcad6b7SDag-Erling Smørgrav 			/* Allow "-" as stdin */
2343efcad6b7SDag-Erling Smørgrav 			if (strcmp(optarg, "-") != 0 &&
2344efcad6b7SDag-Erling Smørgrav 			    (infile = fopen(optarg, "r")) == NULL)
23451e8db6e2SBrian Feldman 				fatal("%s (%s).", strerror(errno), optarg);
2346d0c8c0bcSDag-Erling Smørgrav 			showprogress = 0;
2347e4a9863fSDag-Erling Smørgrav 			quiet = batchmode = 1;
23485e8dbd04SDag-Erling Smørgrav 			addargs(&args, "-obatchmode yes");
23491e8db6e2SBrian Feldman 			break;
2350f7167e0eSDag-Erling Smørgrav 		case 'f':
2351f7167e0eSDag-Erling Smørgrav 			global_fflag = 1;
2352f7167e0eSDag-Erling Smørgrav 			break;
2353b15c8340SDag-Erling Smørgrav 		case 'p':
2354b15c8340SDag-Erling Smørgrav 			global_pflag = 1;
2355b15c8340SDag-Erling Smørgrav 			break;
2356b15c8340SDag-Erling Smørgrav 		case 'D':
2357ae1f160dSDag-Erling Smørgrav 			sftp_direct = optarg;
2358ae1f160dSDag-Erling Smørgrav 			break;
23594a421b63SDag-Erling Smørgrav 		case 'l':
23604a421b63SDag-Erling Smørgrav 			limit_kbps = strtonum(optarg, 1, 100 * 1024 * 1024,
23614a421b63SDag-Erling Smørgrav 			    &errstr);
23624a421b63SDag-Erling Smørgrav 			if (errstr != NULL)
23634a421b63SDag-Erling Smørgrav 				usage();
23644a421b63SDag-Erling Smørgrav 			limit_kbps *= 1024; /* kbps */
23654a421b63SDag-Erling Smørgrav 			break;
2366b15c8340SDag-Erling Smørgrav 		case 'r':
2367b15c8340SDag-Erling Smørgrav 			global_rflag = 1;
2368ae1f160dSDag-Erling Smørgrav 			break;
2369ae1f160dSDag-Erling Smørgrav 		case 'R':
2370ae1f160dSDag-Erling Smørgrav 			num_requests = strtol(optarg, &cp, 10);
2371ae1f160dSDag-Erling Smørgrav 			if (num_requests == 0 || *cp != '\0')
2372ae1f160dSDag-Erling Smørgrav 				fatal("Invalid number of requests \"%s\"",
2373ae1f160dSDag-Erling Smørgrav 				    optarg);
2374ae1f160dSDag-Erling Smørgrav 			break;
2375b15c8340SDag-Erling Smørgrav 		case 's':
2376b15c8340SDag-Erling Smørgrav 			sftp_server = optarg;
2377b15c8340SDag-Erling Smørgrav 			break;
2378b15c8340SDag-Erling Smørgrav 		case 'S':
2379b15c8340SDag-Erling Smørgrav 			ssh_program = optarg;
2380b15c8340SDag-Erling Smørgrav 			replacearg(&args, 0, "%s", ssh_program);
2381b15c8340SDag-Erling Smørgrav 			break;
23821e8db6e2SBrian Feldman 		case 'h':
23831e8db6e2SBrian Feldman 		default:
23841e8db6e2SBrian Feldman 			usage();
23851e8db6e2SBrian Feldman 		}
23861e8db6e2SBrian Feldman 	}
23871e8db6e2SBrian Feldman 
238852028650SDag-Erling Smørgrav 	if (!isatty(STDERR_FILENO))
238952028650SDag-Erling Smørgrav 		showprogress = 0;
239052028650SDag-Erling Smørgrav 
2391545d5ecaSDag-Erling Smørgrav 	log_init(argv[0], ll, SYSLOG_FACILITY_USER, 1);
2392545d5ecaSDag-Erling Smørgrav 
2393ae1f160dSDag-Erling Smørgrav 	if (sftp_direct == NULL) {
23941e8db6e2SBrian Feldman 		if (optind == argc || argc > (optind + 2))
23951e8db6e2SBrian Feldman 			usage();
23961e8db6e2SBrian Feldman 
23971e8db6e2SBrian Feldman 		userhost = xstrdup(argv[optind]);
23981e8db6e2SBrian Feldman 		file2 = argv[optind+1];
23991e8db6e2SBrian Feldman 
2400d0c8c0bcSDag-Erling Smørgrav 		if ((host = strrchr(userhost, '@')) == NULL)
24011e8db6e2SBrian Feldman 			host = userhost;
24021e8db6e2SBrian Feldman 		else {
24031e8db6e2SBrian Feldman 			*host++ = '\0';
24041e8db6e2SBrian Feldman 			if (!userhost[0]) {
24051e8db6e2SBrian Feldman 				fprintf(stderr, "Missing username\n");
24061e8db6e2SBrian Feldman 				usage();
24071e8db6e2SBrian Feldman 			}
2408b15c8340SDag-Erling Smørgrav 			addargs(&args, "-l");
2409b15c8340SDag-Erling Smørgrav 			addargs(&args, "%s", userhost);
24101e8db6e2SBrian Feldman 		}
24111e8db6e2SBrian Feldman 
2412efcad6b7SDag-Erling Smørgrav 		if ((cp = colon(host)) != NULL) {
2413efcad6b7SDag-Erling Smørgrav 			*cp++ = '\0';
2414efcad6b7SDag-Erling Smørgrav 			file1 = cp;
2415efcad6b7SDag-Erling Smørgrav 		}
2416efcad6b7SDag-Erling Smørgrav 
24171e8db6e2SBrian Feldman 		host = cleanhostname(host);
24181e8db6e2SBrian Feldman 		if (!*host) {
24191e8db6e2SBrian Feldman 			fprintf(stderr, "Missing hostname\n");
24201e8db6e2SBrian Feldman 			usage();
24211e8db6e2SBrian Feldman 		}
24221e8db6e2SBrian Feldman 
2423ae1f160dSDag-Erling Smørgrav 		addargs(&args, "-oProtocol %d", sshver);
24241e8db6e2SBrian Feldman 
2425ae1f160dSDag-Erling Smørgrav 		/* no subsystem if the server-spec contains a '/' */
2426ae1f160dSDag-Erling Smørgrav 		if (sftp_server == NULL || strchr(sftp_server, '/') == NULL)
2427ae1f160dSDag-Erling Smørgrav 			addargs(&args, "-s");
2428ae1f160dSDag-Erling Smørgrav 
2429b15c8340SDag-Erling Smørgrav 		addargs(&args, "--");
2430ae1f160dSDag-Erling Smørgrav 		addargs(&args, "%s", host);
2431ae1f160dSDag-Erling Smørgrav 		addargs(&args, "%s", (sftp_server != NULL ?
2432ae1f160dSDag-Erling Smørgrav 		    sftp_server : "sftp"));
24331e8db6e2SBrian Feldman 
2434d95e11bfSDag-Erling Smørgrav 		connect_to_server(ssh_program, args.list, &in, &out);
2435ae1f160dSDag-Erling Smørgrav 	} else {
2436ae1f160dSDag-Erling Smørgrav 		args.list = NULL;
2437ae1f160dSDag-Erling Smørgrav 		addargs(&args, "sftp-server");
24381e8db6e2SBrian Feldman 
2439d95e11bfSDag-Erling Smørgrav 		connect_to_server(sftp_direct, args.list, &in, &out);
2440ae1f160dSDag-Erling Smørgrav 	}
2441021d409fSDag-Erling Smørgrav 	freeargs(&args);
24421e8db6e2SBrian Feldman 
24434a421b63SDag-Erling Smørgrav 	conn = do_init(in, out, copy_buffer_len, num_requests, limit_kbps);
2444b15c8340SDag-Erling Smørgrav 	if (conn == NULL)
2445b15c8340SDag-Erling Smørgrav 		fatal("Couldn't initialise connection to server");
2446b15c8340SDag-Erling Smørgrav 
2447e4a9863fSDag-Erling Smørgrav 	if (!quiet) {
2448b15c8340SDag-Erling Smørgrav 		if (sftp_direct == NULL)
2449b15c8340SDag-Erling Smørgrav 			fprintf(stderr, "Connected to %s.\n", host);
2450b15c8340SDag-Erling Smørgrav 		else
2451b15c8340SDag-Erling Smørgrav 			fprintf(stderr, "Attached to %s.\n", sftp_direct);
2452b15c8340SDag-Erling Smørgrav 	}
2453b15c8340SDag-Erling Smørgrav 
2454b15c8340SDag-Erling Smørgrav 	err = interactive_loop(conn, file1, file2);
24551e8db6e2SBrian Feldman 
245683d2307dSDag-Erling Smørgrav #if !defined(USE_PIPES)
245783d2307dSDag-Erling Smørgrav 	shutdown(in, SHUT_RDWR);
245883d2307dSDag-Erling Smørgrav 	shutdown(out, SHUT_RDWR);
245983d2307dSDag-Erling Smørgrav #endif
246083d2307dSDag-Erling Smørgrav 
24611e8db6e2SBrian Feldman 	close(in);
24621e8db6e2SBrian Feldman 	close(out);
2463efcad6b7SDag-Erling Smørgrav 	if (batchmode)
24641e8db6e2SBrian Feldman 		fclose(infile);
24651e8db6e2SBrian Feldman 
2466545d5ecaSDag-Erling Smørgrav 	while (waitpid(sshpid, NULL, 0) == -1)
2467545d5ecaSDag-Erling Smørgrav 		if (errno != EINTR)
2468545d5ecaSDag-Erling Smørgrav 			fatal("Couldn't wait for ssh process: %s",
2469545d5ecaSDag-Erling Smørgrav 			    strerror(errno));
24701e8db6e2SBrian Feldman 
2471d0c8c0bcSDag-Erling Smørgrav 	exit(err == 0 ? 0 : 1);
24721e8db6e2SBrian Feldman }
2473