xref: /freebsd/crypto/openssh/sftp.c (revision edf8578117e8844e02c0121147f45e4609b30680)
1*edf85781SEd Maste /* $OpenBSD: sftp.c,v 1.236 2023/09/10 23:12:32 djm 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/socket.h>
26761efaa7SDag-Erling Smørgrav #include <sys/wait.h>
27d4af9e69SDag-Erling Smørgrav #ifdef HAVE_SYS_STATVFS_H
28d4af9e69SDag-Erling Smørgrav #include <sys/statvfs.h>
29d4af9e69SDag-Erling Smørgrav #endif
305e8dbd04SDag-Erling Smørgrav 
31d4af9e69SDag-Erling Smørgrav #include <ctype.h>
32761efaa7SDag-Erling Smørgrav #include <errno.h>
33761efaa7SDag-Erling Smørgrav 
34761efaa7SDag-Erling Smørgrav #ifdef HAVE_PATHS_H
35761efaa7SDag-Erling Smørgrav # include <paths.h>
36761efaa7SDag-Erling Smørgrav #endif
37b15c8340SDag-Erling Smørgrav #ifdef HAVE_LIBGEN_H
38b15c8340SDag-Erling Smørgrav #include <libgen.h>
39b15c8340SDag-Erling Smørgrav #endif
40e4a9863fSDag-Erling Smørgrav #ifdef HAVE_LOCALE_H
41e4a9863fSDag-Erling Smørgrav # include <locale.h>
42e4a9863fSDag-Erling Smørgrav #endif
435e8dbd04SDag-Erling Smørgrav #ifdef USE_LIBEDIT
445e8dbd04SDag-Erling Smørgrav #include <histedit.h>
455e8dbd04SDag-Erling Smørgrav #else
465e8dbd04SDag-Erling Smørgrav typedef void EditLine;
475e8dbd04SDag-Erling Smørgrav #endif
48bc5531deSDag-Erling Smørgrav #include <limits.h>
49761efaa7SDag-Erling Smørgrav #include <signal.h>
50076ad2f8SDag-Erling Smørgrav #include <stdarg.h>
51761efaa7SDag-Erling Smørgrav #include <stdlib.h>
52761efaa7SDag-Erling Smørgrav #include <stdio.h>
53761efaa7SDag-Erling Smørgrav #include <string.h>
54761efaa7SDag-Erling Smørgrav #include <unistd.h>
551e8db6e2SBrian Feldman 
56d4af9e69SDag-Erling Smørgrav #ifdef HAVE_UTIL_H
57d4af9e69SDag-Erling Smørgrav # include <util.h>
58d4af9e69SDag-Erling Smørgrav #endif
59d4af9e69SDag-Erling Smørgrav 
601e8db6e2SBrian Feldman #include "xmalloc.h"
611e8db6e2SBrian Feldman #include "log.h"
621e8db6e2SBrian Feldman #include "pathnames.h"
63ae1f160dSDag-Erling Smørgrav #include "misc.h"
64076ad2f8SDag-Erling Smørgrav #include "utf8.h"
651e8db6e2SBrian Feldman 
661e8db6e2SBrian Feldman #include "sftp.h"
67bc5531deSDag-Erling Smørgrav #include "ssherr.h"
68bc5531deSDag-Erling Smørgrav #include "sshbuf.h"
691e8db6e2SBrian Feldman #include "sftp-common.h"
701e8db6e2SBrian Feldman #include "sftp-client.h"
7138a52bd3SEd Maste #include "sftp-usergroup.h"
72efcad6b7SDag-Erling Smørgrav 
73efcad6b7SDag-Erling Smørgrav /* File to read commands from */
74efcad6b7SDag-Erling Smørgrav FILE* infile;
75efcad6b7SDag-Erling Smørgrav 
76efcad6b7SDag-Erling Smørgrav /* Are we in batchfile mode? */
77efcad6b7SDag-Erling Smørgrav int batchmode = 0;
78efcad6b7SDag-Erling Smørgrav 
79efcad6b7SDag-Erling Smørgrav /* PID of ssh transport process */
80190cef3dSDag-Erling Smørgrav static volatile pid_t sshpid = -1;
81efcad6b7SDag-Erling Smørgrav 
8219261079SEd Maste /* Suppress diagnostic messages */
83e4a9863fSDag-Erling Smørgrav int quiet = 0;
84e4a9863fSDag-Erling Smørgrav 
85efcad6b7SDag-Erling Smørgrav /* This is set to 0 if the progressmeter is not desired. */
8652028650SDag-Erling Smørgrav int showprogress = 1;
87efcad6b7SDag-Erling Smørgrav 
88b15c8340SDag-Erling Smørgrav /* When this option is set, we always recursively download/upload directories */
89b15c8340SDag-Erling Smørgrav int global_rflag = 0;
90b15c8340SDag-Erling Smørgrav 
91a0ee8cc6SDag-Erling Smørgrav /* When this option is set, we resume download or upload if possible */
92e4a9863fSDag-Erling Smørgrav int global_aflag = 0;
93e4a9863fSDag-Erling Smørgrav 
94b15c8340SDag-Erling Smørgrav /* When this option is set, the file transfers will always preserve times */
95b15c8340SDag-Erling Smørgrav int global_pflag = 0;
96b15c8340SDag-Erling Smørgrav 
97f7167e0eSDag-Erling Smørgrav /* When this option is set, transfers will have fsync() called on each file */
98f7167e0eSDag-Erling Smørgrav int global_fflag = 0;
99f7167e0eSDag-Erling Smørgrav 
100d74d50a8SDag-Erling Smørgrav /* SIGINT received during command processing */
101d74d50a8SDag-Erling Smørgrav volatile sig_atomic_t interrupted = 0;
102d74d50a8SDag-Erling Smørgrav 
103d74d50a8SDag-Erling Smørgrav /* I wish qsort() took a separate ctx for the comparison function...*/
104d74d50a8SDag-Erling Smørgrav int sort_flag;
1054f52dfbbSDag-Erling Smørgrav glob_t *sort_glob;
106d74d50a8SDag-Erling Smørgrav 
107b15c8340SDag-Erling Smørgrav /* Context used for commandline completion */
108b15c8340SDag-Erling Smørgrav struct complete_ctx {
109b15c8340SDag-Erling Smørgrav 	struct sftp_conn *conn;
110b15c8340SDag-Erling Smørgrav 	char **remote_pathp;
111b15c8340SDag-Erling Smørgrav };
112b15c8340SDag-Erling Smørgrav 
113*edf85781SEd Maste int sftp_glob(struct sftp_conn *, const char *, int,
114efcad6b7SDag-Erling Smørgrav     int (*)(const char *, int), glob_t *); /* proto for sftp-glob.c */
1151e8db6e2SBrian Feldman 
11683d2307dSDag-Erling Smørgrav extern char *__progname;
11783d2307dSDag-Erling Smørgrav 
118efcad6b7SDag-Erling Smørgrav /* Separators for interactive commands */
119efcad6b7SDag-Erling Smørgrav #define WHITESPACE " \t\r\n"
1201e8db6e2SBrian Feldman 
121d74d50a8SDag-Erling Smørgrav /* ls flags */
122b15c8340SDag-Erling Smørgrav #define LS_LONG_VIEW	0x0001	/* Full view ala ls -l */
123b15c8340SDag-Erling Smørgrav #define LS_SHORT_VIEW	0x0002	/* Single row view ala ls -1 */
124b15c8340SDag-Erling Smørgrav #define LS_NUMERIC_VIEW	0x0004	/* Long view with numeric uid/gid */
125b15c8340SDag-Erling Smørgrav #define LS_NAME_SORT	0x0008	/* Sort by name (default) */
126b15c8340SDag-Erling Smørgrav #define LS_TIME_SORT	0x0010	/* Sort by mtime */
127b15c8340SDag-Erling Smørgrav #define LS_SIZE_SORT	0x0020	/* Sort by file size */
128b15c8340SDag-Erling Smørgrav #define LS_REVERSE_SORT	0x0040	/* Reverse sort order */
129b15c8340SDag-Erling Smørgrav #define LS_SHOW_ALL	0x0080	/* Don't skip filenames starting with '.' */
130b15c8340SDag-Erling Smørgrav #define LS_SI_UNITS	0x0100	/* Display sizes as K, M, G, etc. */
131d74d50a8SDag-Erling Smørgrav 
132b15c8340SDag-Erling Smørgrav #define VIEW_FLAGS	(LS_LONG_VIEW|LS_SHORT_VIEW|LS_NUMERIC_VIEW|LS_SI_UNITS)
133d74d50a8SDag-Erling Smørgrav #define SORT_FLAGS	(LS_NAME_SORT|LS_TIME_SORT|LS_SIZE_SORT)
134efcad6b7SDag-Erling Smørgrav 
135efcad6b7SDag-Erling Smørgrav /* Commands for interactive mode */
136f7167e0eSDag-Erling Smørgrav enum sftp_command {
137f7167e0eSDag-Erling Smørgrav 	I_CHDIR = 1,
138f7167e0eSDag-Erling Smørgrav 	I_CHGRP,
139f7167e0eSDag-Erling Smørgrav 	I_CHMOD,
140f7167e0eSDag-Erling Smørgrav 	I_CHOWN,
14187c1498dSEd Maste 	I_COPY,
142f7167e0eSDag-Erling Smørgrav 	I_DF,
143f7167e0eSDag-Erling Smørgrav 	I_GET,
144f7167e0eSDag-Erling Smørgrav 	I_HELP,
145f7167e0eSDag-Erling Smørgrav 	I_LCHDIR,
146f7167e0eSDag-Erling Smørgrav 	I_LINK,
147f7167e0eSDag-Erling Smørgrav 	I_LLS,
148f7167e0eSDag-Erling Smørgrav 	I_LMKDIR,
149f7167e0eSDag-Erling Smørgrav 	I_LPWD,
150f7167e0eSDag-Erling Smørgrav 	I_LS,
151f7167e0eSDag-Erling Smørgrav 	I_LUMASK,
152f7167e0eSDag-Erling Smørgrav 	I_MKDIR,
153f7167e0eSDag-Erling Smørgrav 	I_PUT,
154f7167e0eSDag-Erling Smørgrav 	I_PWD,
155f7167e0eSDag-Erling Smørgrav 	I_QUIT,
156a0ee8cc6SDag-Erling Smørgrav 	I_REGET,
157f7167e0eSDag-Erling Smørgrav 	I_RENAME,
158a0ee8cc6SDag-Erling Smørgrav 	I_REPUT,
159f7167e0eSDag-Erling Smørgrav 	I_RM,
160f7167e0eSDag-Erling Smørgrav 	I_RMDIR,
161f7167e0eSDag-Erling Smørgrav 	I_SHELL,
162f7167e0eSDag-Erling Smørgrav 	I_SYMLINK,
163f7167e0eSDag-Erling Smørgrav 	I_VERSION,
164f7167e0eSDag-Erling Smørgrav 	I_PROGRESS,
165f7167e0eSDag-Erling Smørgrav };
166efcad6b7SDag-Erling Smørgrav 
167efcad6b7SDag-Erling Smørgrav struct CMD {
168efcad6b7SDag-Erling Smørgrav 	const char *c;
169efcad6b7SDag-Erling Smørgrav 	const int n;
17038a52bd3SEd Maste 	const int t;	/* Completion type for the first argument */
17138a52bd3SEd Maste 	const int t2;	/* completion type for the optional second argument */
172efcad6b7SDag-Erling Smørgrav };
173efcad6b7SDag-Erling Smørgrav 
174b15c8340SDag-Erling Smørgrav /* Type of completion */
175b15c8340SDag-Erling Smørgrav #define NOARGS	0
176b15c8340SDag-Erling Smørgrav #define REMOTE	1
177b15c8340SDag-Erling Smørgrav #define LOCAL	2
178b15c8340SDag-Erling Smørgrav 
179efcad6b7SDag-Erling Smørgrav static const struct CMD cmds[] = {
18038a52bd3SEd Maste 	{ "bye",	I_QUIT,		NOARGS, 	NOARGS	},
18138a52bd3SEd Maste 	{ "cd",		I_CHDIR,	REMOTE, 	NOARGS	},
18238a52bd3SEd Maste 	{ "chdir",	I_CHDIR,	REMOTE, 	NOARGS	},
18338a52bd3SEd Maste 	{ "chgrp",	I_CHGRP,	REMOTE, 	NOARGS	},
18438a52bd3SEd Maste 	{ "chmod",	I_CHMOD,	REMOTE, 	NOARGS	},
18538a52bd3SEd Maste 	{ "chown",	I_CHOWN,	REMOTE, 	NOARGS	},
18638a52bd3SEd Maste 	{ "copy",	I_COPY,		REMOTE, 	LOCAL	},
18738a52bd3SEd Maste 	{ "cp",		I_COPY,		REMOTE, 	LOCAL	},
18838a52bd3SEd Maste 	{ "df",		I_DF,		REMOTE, 	NOARGS	},
18938a52bd3SEd Maste 	{ "dir",	I_LS,		REMOTE, 	NOARGS	},
19038a52bd3SEd Maste 	{ "exit",	I_QUIT,		NOARGS, 	NOARGS	},
19138a52bd3SEd Maste 	{ "get",	I_GET,		REMOTE, 	LOCAL	},
19238a52bd3SEd Maste 	{ "help",	I_HELP,		NOARGS, 	NOARGS	},
19338a52bd3SEd Maste 	{ "lcd",	I_LCHDIR,	LOCAL,		NOARGS	},
19438a52bd3SEd Maste 	{ "lchdir",	I_LCHDIR,	LOCAL,		NOARGS	},
19538a52bd3SEd Maste 	{ "lls",	I_LLS,		LOCAL,		NOARGS	},
19638a52bd3SEd Maste 	{ "lmkdir",	I_LMKDIR,	LOCAL,		NOARGS	},
19738a52bd3SEd Maste 	{ "ln",		I_LINK,		REMOTE, 	REMOTE	},
19838a52bd3SEd Maste 	{ "lpwd",	I_LPWD,		LOCAL,		NOARGS	},
19938a52bd3SEd Maste 	{ "ls",		I_LS,		REMOTE,		NOARGS	},
20038a52bd3SEd Maste 	{ "lumask",	I_LUMASK,	NOARGS,		NOARGS	},
20138a52bd3SEd Maste 	{ "mkdir",	I_MKDIR,	REMOTE,		NOARGS	},
20238a52bd3SEd Maste 	{ "mget",	I_GET,		REMOTE,		LOCAL	},
20338a52bd3SEd Maste 	{ "mput",	I_PUT,		LOCAL,		REMOTE	},
20438a52bd3SEd Maste 	{ "progress",	I_PROGRESS,	NOARGS,		NOARGS	},
20538a52bd3SEd Maste 	{ "put",	I_PUT,		LOCAL,		REMOTE	},
20638a52bd3SEd Maste 	{ "pwd",	I_PWD,		REMOTE, 	NOARGS	},
20738a52bd3SEd Maste 	{ "quit",	I_QUIT,		NOARGS, 	NOARGS	},
20838a52bd3SEd Maste 	{ "reget",	I_REGET,	REMOTE, 	LOCAL	},
20938a52bd3SEd Maste 	{ "rename",	I_RENAME,	REMOTE, 	REMOTE	},
21038a52bd3SEd Maste 	{ "reput",	I_REPUT,	LOCAL,		REMOTE	},
21138a52bd3SEd Maste 	{ "rm",		I_RM,		REMOTE,		NOARGS	},
21238a52bd3SEd Maste 	{ "rmdir",	I_RMDIR,	REMOTE,		NOARGS	},
21338a52bd3SEd Maste 	{ "symlink",	I_SYMLINK,	REMOTE,		REMOTE	},
21438a52bd3SEd Maste 	{ "version",	I_VERSION,	NOARGS, 	NOARGS	},
21538a52bd3SEd Maste 	{ "!",		I_SHELL,	NOARGS, 	NOARGS	},
21638a52bd3SEd Maste 	{ "?",		I_HELP,		NOARGS, 	NOARGS	},
21738a52bd3SEd Maste 	{ NULL,		-1,		-1,		-1	}
218efcad6b7SDag-Erling Smørgrav };
219efcad6b7SDag-Erling Smørgrav 
220efcad6b7SDag-Erling Smørgrav static void
221d74d50a8SDag-Erling Smørgrav killchild(int signo)
222d74d50a8SDag-Erling Smørgrav {
22319261079SEd Maste 	pid_t pid;
22419261079SEd Maste 
22519261079SEd Maste 	pid = sshpid;
22619261079SEd Maste 	if (pid > 1) {
22719261079SEd Maste 		kill(pid, SIGTERM);
228535af610SEd Maste 		(void)waitpid(pid, NULL, 0);
2295e8dbd04SDag-Erling Smørgrav 	}
230d74d50a8SDag-Erling Smørgrav 
231d74d50a8SDag-Erling Smørgrav 	_exit(1);
232d74d50a8SDag-Erling Smørgrav }
233d74d50a8SDag-Erling Smørgrav 
234d74d50a8SDag-Erling Smørgrav static void
235ca86bcf2SDag-Erling Smørgrav suspchild(int signo)
236ca86bcf2SDag-Erling Smørgrav {
237ca86bcf2SDag-Erling Smørgrav 	if (sshpid > 1) {
238ca86bcf2SDag-Erling Smørgrav 		kill(sshpid, signo);
239ca86bcf2SDag-Erling Smørgrav 		while (waitpid(sshpid, NULL, WUNTRACED) == -1 && errno == EINTR)
240ca86bcf2SDag-Erling Smørgrav 			continue;
241ca86bcf2SDag-Erling Smørgrav 	}
242ca86bcf2SDag-Erling Smørgrav 	kill(getpid(), SIGSTOP);
243ca86bcf2SDag-Erling Smørgrav }
244ca86bcf2SDag-Erling Smørgrav 
245ca86bcf2SDag-Erling Smørgrav static void
246d74d50a8SDag-Erling Smørgrav cmd_interrupt(int signo)
247d74d50a8SDag-Erling Smørgrav {
248d74d50a8SDag-Erling Smørgrav 	const char msg[] = "\rInterrupt  \n";
2495e8dbd04SDag-Erling Smørgrav 	int olderrno = errno;
250d74d50a8SDag-Erling Smørgrav 
251e4a9863fSDag-Erling Smørgrav 	(void)write(STDERR_FILENO, msg, sizeof(msg) - 1);
252d74d50a8SDag-Erling Smørgrav 	interrupted = 1;
2535e8dbd04SDag-Erling Smørgrav 	errno = olderrno;
254d74d50a8SDag-Erling Smørgrav }
255d74d50a8SDag-Erling Smørgrav 
256190cef3dSDag-Erling Smørgrav static void
25719261079SEd Maste read_interrupt(int signo)
25819261079SEd Maste {
25919261079SEd Maste 	interrupted = 1;
26019261079SEd Maste }
26119261079SEd Maste 
26219261079SEd Maste static void
263190cef3dSDag-Erling Smørgrav sigchld_handler(int sig)
264190cef3dSDag-Erling Smørgrav {
265190cef3dSDag-Erling Smørgrav 	int save_errno = errno;
266190cef3dSDag-Erling Smørgrav 	pid_t pid;
267190cef3dSDag-Erling Smørgrav 	const char msg[] = "\rConnection closed.  \n";
268190cef3dSDag-Erling Smørgrav 
269190cef3dSDag-Erling Smørgrav 	/* Report if ssh transport process dies. */
270190cef3dSDag-Erling Smørgrav 	while ((pid = waitpid(sshpid, NULL, WNOHANG)) == -1 && errno == EINTR)
271190cef3dSDag-Erling Smørgrav 		continue;
272190cef3dSDag-Erling Smørgrav 	if (pid == sshpid) {
273f374ba41SEd Maste 		if (!quiet)
274190cef3dSDag-Erling Smørgrav 		    (void)write(STDERR_FILENO, msg, sizeof(msg) - 1);
275190cef3dSDag-Erling Smørgrav 		sshpid = -1;
276190cef3dSDag-Erling Smørgrav 	}
277190cef3dSDag-Erling Smørgrav 
278190cef3dSDag-Erling Smørgrav 	errno = save_errno;
279190cef3dSDag-Erling Smørgrav }
280190cef3dSDag-Erling Smørgrav 
281d74d50a8SDag-Erling Smørgrav static void
282efcad6b7SDag-Erling Smørgrav help(void)
283efcad6b7SDag-Erling Smørgrav {
284cce7d346SDag-Erling Smørgrav 	printf("Available commands:\n"
285cce7d346SDag-Erling Smørgrav 	    "bye                                Quit sftp\n"
286cce7d346SDag-Erling Smørgrav 	    "cd path                            Change remote directory to 'path'\n"
28719261079SEd Maste 	    "chgrp [-h] grp path                Change group of file 'path' to 'grp'\n"
28819261079SEd Maste 	    "chmod [-h] mode path               Change permissions of file 'path' to 'mode'\n"
28919261079SEd Maste 	    "chown [-h] own path                Change owner of file 'path' to 'own'\n"
29087c1498dSEd Maste 	    "copy oldpath newpath               Copy remote file\n"
29187c1498dSEd Maste 	    "cp oldpath newpath                 Copy remote file\n"
292cce7d346SDag-Erling Smørgrav 	    "df [-hi] [path]                    Display statistics for current directory or\n"
293cce7d346SDag-Erling Smørgrav 	    "                                   filesystem containing 'path'\n"
294cce7d346SDag-Erling Smørgrav 	    "exit                               Quit sftp\n"
29519261079SEd Maste 	    "get [-afpR] remote [local]         Download file\n"
296cce7d346SDag-Erling Smørgrav 	    "help                               Display this help text\n"
297cce7d346SDag-Erling Smørgrav 	    "lcd path                           Change local directory to 'path'\n"
298cce7d346SDag-Erling Smørgrav 	    "lls [ls-options [path]]            Display local directory listing\n"
299cce7d346SDag-Erling Smørgrav 	    "lmkdir path                        Create local directory\n"
3004a421b63SDag-Erling Smørgrav 	    "ln [-s] oldpath newpath            Link remote file (-s for symlink)\n"
301cce7d346SDag-Erling Smørgrav 	    "lpwd                               Print local working directory\n"
302b15c8340SDag-Erling Smørgrav 	    "ls [-1afhlnrSt] [path]             Display remote directory listing\n"
303cce7d346SDag-Erling Smørgrav 	    "lumask umask                       Set local umask to 'umask'\n"
304cce7d346SDag-Erling Smørgrav 	    "mkdir path                         Create remote directory\n"
305cce7d346SDag-Erling Smørgrav 	    "progress                           Toggle display of progress meter\n"
30619261079SEd Maste 	    "put [-afpR] local [remote]         Upload file\n"
307cce7d346SDag-Erling Smørgrav 	    "pwd                                Display remote working directory\n"
308cce7d346SDag-Erling Smørgrav 	    "quit                               Quit sftp\n"
30919261079SEd Maste 	    "reget [-fpR] remote [local]        Resume download file\n"
310cce7d346SDag-Erling Smørgrav 	    "rename oldpath newpath             Rename remote file\n"
31119261079SEd Maste 	    "reput [-fpR] local [remote]        Resume upload file\n"
312cce7d346SDag-Erling Smørgrav 	    "rm path                            Delete remote file\n"
313cce7d346SDag-Erling Smørgrav 	    "rmdir path                         Remove remote directory\n"
314cce7d346SDag-Erling Smørgrav 	    "symlink oldpath newpath            Symlink remote file\n"
315cce7d346SDag-Erling Smørgrav 	    "version                            Show SFTP version\n"
316cce7d346SDag-Erling Smørgrav 	    "!command                           Execute 'command' in local shell\n"
317cce7d346SDag-Erling Smørgrav 	    "!                                  Escape to local shell\n"
318cce7d346SDag-Erling Smørgrav 	    "?                                  Synonym for help\n");
319efcad6b7SDag-Erling Smørgrav }
320efcad6b7SDag-Erling Smørgrav 
321efcad6b7SDag-Erling Smørgrav static void
322efcad6b7SDag-Erling Smørgrav local_do_shell(const char *args)
323efcad6b7SDag-Erling Smørgrav {
324efcad6b7SDag-Erling Smørgrav 	int status;
325efcad6b7SDag-Erling Smørgrav 	char *shell;
326efcad6b7SDag-Erling Smørgrav 	pid_t pid;
327efcad6b7SDag-Erling Smørgrav 
328efcad6b7SDag-Erling Smørgrav 	if (!*args)
329efcad6b7SDag-Erling Smørgrav 		args = NULL;
330efcad6b7SDag-Erling Smørgrav 
3314a421b63SDag-Erling Smørgrav 	if ((shell = getenv("SHELL")) == NULL || *shell == '\0')
332efcad6b7SDag-Erling Smørgrav 		shell = _PATH_BSHELL;
333efcad6b7SDag-Erling Smørgrav 
334efcad6b7SDag-Erling Smørgrav 	if ((pid = fork()) == -1)
335efcad6b7SDag-Erling Smørgrav 		fatal("Couldn't fork: %s", strerror(errno));
336efcad6b7SDag-Erling Smørgrav 
337efcad6b7SDag-Erling Smørgrav 	if (pid == 0) {
338efcad6b7SDag-Erling Smørgrav 		/* XXX: child has pipe fds to ssh subproc open - issue? */
339efcad6b7SDag-Erling Smørgrav 		if (args) {
340efcad6b7SDag-Erling Smørgrav 			debug3("Executing %s -c \"%s\"", shell, args);
341efcad6b7SDag-Erling Smørgrav 			execl(shell, shell, "-c", args, (char *)NULL);
342efcad6b7SDag-Erling Smørgrav 		} else {
343efcad6b7SDag-Erling Smørgrav 			debug3("Executing %s", shell);
344efcad6b7SDag-Erling Smørgrav 			execl(shell, shell, (char *)NULL);
345efcad6b7SDag-Erling Smørgrav 		}
346efcad6b7SDag-Erling Smørgrav 		fprintf(stderr, "Couldn't execute \"%s\": %s\n", shell,
347efcad6b7SDag-Erling Smørgrav 		    strerror(errno));
348efcad6b7SDag-Erling Smørgrav 		_exit(1);
349efcad6b7SDag-Erling Smørgrav 	}
350efcad6b7SDag-Erling Smørgrav 	while (waitpid(pid, &status, 0) == -1)
351efcad6b7SDag-Erling Smørgrav 		if (errno != EINTR)
352efcad6b7SDag-Erling Smørgrav 			fatal("Couldn't wait for child: %s", strerror(errno));
353efcad6b7SDag-Erling Smørgrav 	if (!WIFEXITED(status))
354761efaa7SDag-Erling Smørgrav 		error("Shell exited abnormally");
355efcad6b7SDag-Erling Smørgrav 	else if (WEXITSTATUS(status))
356efcad6b7SDag-Erling Smørgrav 		error("Shell exited with status %d", WEXITSTATUS(status));
357efcad6b7SDag-Erling Smørgrav }
358efcad6b7SDag-Erling Smørgrav 
359efcad6b7SDag-Erling Smørgrav static void
360efcad6b7SDag-Erling Smørgrav local_do_ls(const char *args)
361efcad6b7SDag-Erling Smørgrav {
362efcad6b7SDag-Erling Smørgrav 	if (!args || !*args)
363efcad6b7SDag-Erling Smørgrav 		local_do_shell(_PATH_LS);
364efcad6b7SDag-Erling Smørgrav 	else {
365efcad6b7SDag-Erling Smørgrav 		int len = strlen(_PATH_LS " ") + strlen(args) + 1;
366efcad6b7SDag-Erling Smørgrav 		char *buf = xmalloc(len);
367efcad6b7SDag-Erling Smørgrav 
368efcad6b7SDag-Erling Smørgrav 		/* XXX: quoting - rip quoting code from ftp? */
369efcad6b7SDag-Erling Smørgrav 		snprintf(buf, len, _PATH_LS " %s", args);
370efcad6b7SDag-Erling Smørgrav 		local_do_shell(buf);
371e4a9863fSDag-Erling Smørgrav 		free(buf);
372efcad6b7SDag-Erling Smørgrav 	}
373efcad6b7SDag-Erling Smørgrav }
374efcad6b7SDag-Erling Smørgrav 
375efcad6b7SDag-Erling Smørgrav /* Strip one path (usually the pwd) from the start of another */
376efcad6b7SDag-Erling Smørgrav static char *
377076ad2f8SDag-Erling Smørgrav path_strip(const char *path, const char *strip)
378efcad6b7SDag-Erling Smørgrav {
379efcad6b7SDag-Erling Smørgrav 	size_t len;
380efcad6b7SDag-Erling Smørgrav 
381efcad6b7SDag-Erling Smørgrav 	if (strip == NULL)
382efcad6b7SDag-Erling Smørgrav 		return (xstrdup(path));
383efcad6b7SDag-Erling Smørgrav 
384efcad6b7SDag-Erling Smørgrav 	len = strlen(strip);
3855e8dbd04SDag-Erling Smørgrav 	if (strncmp(path, strip, len) == 0) {
386efcad6b7SDag-Erling Smørgrav 		if (strip[len - 1] != '/' && path[len] == '/')
387efcad6b7SDag-Erling Smørgrav 			len++;
388efcad6b7SDag-Erling Smørgrav 		return (xstrdup(path + len));
389efcad6b7SDag-Erling Smørgrav 	}
390efcad6b7SDag-Erling Smørgrav 
391efcad6b7SDag-Erling Smørgrav 	return (xstrdup(path));
392efcad6b7SDag-Erling Smørgrav }
393efcad6b7SDag-Erling Smørgrav 
394efcad6b7SDag-Erling Smørgrav static int
395e4a9863fSDag-Erling Smørgrav parse_getput_flags(const char *cmd, char **argv, int argc,
396f7167e0eSDag-Erling Smørgrav     int *aflag, int *fflag, int *pflag, int *rflag)
397efcad6b7SDag-Erling Smørgrav {
398d4af9e69SDag-Erling Smørgrav 	extern int opterr, optind, optopt, optreset;
399d4af9e69SDag-Erling Smørgrav 	int ch;
400efcad6b7SDag-Erling Smørgrav 
401d4af9e69SDag-Erling Smørgrav 	optind = optreset = 1;
402d4af9e69SDag-Erling Smørgrav 	opterr = 0;
403d4af9e69SDag-Erling Smørgrav 
404f7167e0eSDag-Erling Smørgrav 	*aflag = *fflag = *rflag = *pflag = 0;
405f7167e0eSDag-Erling Smørgrav 	while ((ch = getopt(argc, argv, "afPpRr")) != -1) {
406d4af9e69SDag-Erling Smørgrav 		switch (ch) {
407e4a9863fSDag-Erling Smørgrav 		case 'a':
408e4a9863fSDag-Erling Smørgrav 			*aflag = 1;
409e4a9863fSDag-Erling Smørgrav 			break;
410f7167e0eSDag-Erling Smørgrav 		case 'f':
411f7167e0eSDag-Erling Smørgrav 			*fflag = 1;
412f7167e0eSDag-Erling Smørgrav 			break;
413efcad6b7SDag-Erling Smørgrav 		case 'p':
414efcad6b7SDag-Erling Smørgrav 		case 'P':
415efcad6b7SDag-Erling Smørgrav 			*pflag = 1;
416efcad6b7SDag-Erling Smørgrav 			break;
417b15c8340SDag-Erling Smørgrav 		case 'r':
418b15c8340SDag-Erling Smørgrav 		case 'R':
419b15c8340SDag-Erling Smørgrav 			*rflag = 1;
420b15c8340SDag-Erling Smørgrav 			break;
421efcad6b7SDag-Erling Smørgrav 		default:
422d4af9e69SDag-Erling Smørgrav 			error("%s: Invalid flag -%c", cmd, optopt);
423d4af9e69SDag-Erling Smørgrav 			return -1;
424efcad6b7SDag-Erling Smørgrav 		}
425efcad6b7SDag-Erling Smørgrav 	}
426efcad6b7SDag-Erling Smørgrav 
427d4af9e69SDag-Erling Smørgrav 	return optind;
428efcad6b7SDag-Erling Smørgrav }
429efcad6b7SDag-Erling Smørgrav 
430efcad6b7SDag-Erling Smørgrav static int
4314a421b63SDag-Erling Smørgrav parse_link_flags(const char *cmd, char **argv, int argc, int *sflag)
4324a421b63SDag-Erling Smørgrav {
4334a421b63SDag-Erling Smørgrav 	extern int opterr, optind, optopt, optreset;
4344a421b63SDag-Erling Smørgrav 	int ch;
4354a421b63SDag-Erling Smørgrav 
4364a421b63SDag-Erling Smørgrav 	optind = optreset = 1;
4374a421b63SDag-Erling Smørgrav 	opterr = 0;
4384a421b63SDag-Erling Smørgrav 
4394a421b63SDag-Erling Smørgrav 	*sflag = 0;
4404a421b63SDag-Erling Smørgrav 	while ((ch = getopt(argc, argv, "s")) != -1) {
4414a421b63SDag-Erling Smørgrav 		switch (ch) {
4424a421b63SDag-Erling Smørgrav 		case 's':
4434a421b63SDag-Erling Smørgrav 			*sflag = 1;
4444a421b63SDag-Erling Smørgrav 			break;
4454a421b63SDag-Erling Smørgrav 		default:
4464a421b63SDag-Erling Smørgrav 			error("%s: Invalid flag -%c", cmd, optopt);
4474a421b63SDag-Erling Smørgrav 			return -1;
4484a421b63SDag-Erling Smørgrav 		}
4494a421b63SDag-Erling Smørgrav 	}
4504a421b63SDag-Erling Smørgrav 
4514a421b63SDag-Erling Smørgrav 	return optind;
4524a421b63SDag-Erling Smørgrav }
4534a421b63SDag-Erling Smørgrav 
4544a421b63SDag-Erling Smørgrav static int
455f7167e0eSDag-Erling Smørgrav parse_rename_flags(const char *cmd, char **argv, int argc, int *lflag)
456f7167e0eSDag-Erling Smørgrav {
457f7167e0eSDag-Erling Smørgrav 	extern int opterr, optind, optopt, optreset;
458f7167e0eSDag-Erling Smørgrav 	int ch;
459f7167e0eSDag-Erling Smørgrav 
460f7167e0eSDag-Erling Smørgrav 	optind = optreset = 1;
461f7167e0eSDag-Erling Smørgrav 	opterr = 0;
462f7167e0eSDag-Erling Smørgrav 
463f7167e0eSDag-Erling Smørgrav 	*lflag = 0;
464f7167e0eSDag-Erling Smørgrav 	while ((ch = getopt(argc, argv, "l")) != -1) {
465f7167e0eSDag-Erling Smørgrav 		switch (ch) {
466f7167e0eSDag-Erling Smørgrav 		case 'l':
467f7167e0eSDag-Erling Smørgrav 			*lflag = 1;
468f7167e0eSDag-Erling Smørgrav 			break;
469f7167e0eSDag-Erling Smørgrav 		default:
470f7167e0eSDag-Erling Smørgrav 			error("%s: Invalid flag -%c", cmd, optopt);
471f7167e0eSDag-Erling Smørgrav 			return -1;
472f7167e0eSDag-Erling Smørgrav 		}
473f7167e0eSDag-Erling Smørgrav 	}
474f7167e0eSDag-Erling Smørgrav 
475f7167e0eSDag-Erling Smørgrav 	return optind;
476f7167e0eSDag-Erling Smørgrav }
477f7167e0eSDag-Erling Smørgrav 
478f7167e0eSDag-Erling Smørgrav static int
479d4af9e69SDag-Erling Smørgrav parse_ls_flags(char **argv, int argc, int *lflag)
480efcad6b7SDag-Erling Smørgrav {
481d4af9e69SDag-Erling Smørgrav 	extern int opterr, optind, optopt, optreset;
482d4af9e69SDag-Erling Smørgrav 	int ch;
483efcad6b7SDag-Erling Smørgrav 
484d4af9e69SDag-Erling Smørgrav 	optind = optreset = 1;
485d4af9e69SDag-Erling Smørgrav 	opterr = 0;
486d4af9e69SDag-Erling Smørgrav 
487d74d50a8SDag-Erling Smørgrav 	*lflag = LS_NAME_SORT;
488b15c8340SDag-Erling Smørgrav 	while ((ch = getopt(argc, argv, "1Safhlnrt")) != -1) {
489d4af9e69SDag-Erling Smørgrav 		switch (ch) {
490efcad6b7SDag-Erling Smørgrav 		case '1':
491d74d50a8SDag-Erling Smørgrav 			*lflag &= ~VIEW_FLAGS;
492d74d50a8SDag-Erling Smørgrav 			*lflag |= LS_SHORT_VIEW;
493d74d50a8SDag-Erling Smørgrav 			break;
494d74d50a8SDag-Erling Smørgrav 		case 'S':
495d74d50a8SDag-Erling Smørgrav 			*lflag &= ~SORT_FLAGS;
496d74d50a8SDag-Erling Smørgrav 			*lflag |= LS_SIZE_SORT;
497d74d50a8SDag-Erling Smørgrav 			break;
498d4af9e69SDag-Erling Smørgrav 		case 'a':
499d4af9e69SDag-Erling Smørgrav 			*lflag |= LS_SHOW_ALL;
500d74d50a8SDag-Erling Smørgrav 			break;
501d74d50a8SDag-Erling Smørgrav 		case 'f':
502d74d50a8SDag-Erling Smørgrav 			*lflag &= ~SORT_FLAGS;
503d74d50a8SDag-Erling Smørgrav 			break;
504b15c8340SDag-Erling Smørgrav 		case 'h':
505b15c8340SDag-Erling Smørgrav 			*lflag |= LS_SI_UNITS;
506b15c8340SDag-Erling Smørgrav 			break;
507d4af9e69SDag-Erling Smørgrav 		case 'l':
508b15c8340SDag-Erling Smørgrav 			*lflag &= ~LS_SHORT_VIEW;
509d4af9e69SDag-Erling Smørgrav 			*lflag |= LS_LONG_VIEW;
510d4af9e69SDag-Erling Smørgrav 			break;
511d4af9e69SDag-Erling Smørgrav 		case 'n':
512b15c8340SDag-Erling Smørgrav 			*lflag &= ~LS_SHORT_VIEW;
513d4af9e69SDag-Erling Smørgrav 			*lflag |= LS_NUMERIC_VIEW|LS_LONG_VIEW;
514d4af9e69SDag-Erling Smørgrav 			break;
515d4af9e69SDag-Erling Smørgrav 		case 'r':
516d4af9e69SDag-Erling Smørgrav 			*lflag |= LS_REVERSE_SORT;
517d4af9e69SDag-Erling Smørgrav 			break;
518d4af9e69SDag-Erling Smørgrav 		case 't':
519d4af9e69SDag-Erling Smørgrav 			*lflag &= ~SORT_FLAGS;
520d4af9e69SDag-Erling Smørgrav 			*lflag |= LS_TIME_SORT;
521efcad6b7SDag-Erling Smørgrav 			break;
522efcad6b7SDag-Erling Smørgrav 		default:
523d4af9e69SDag-Erling Smørgrav 			error("ls: Invalid flag -%c", optopt);
524d4af9e69SDag-Erling Smørgrav 			return -1;
525efcad6b7SDag-Erling Smørgrav 		}
526efcad6b7SDag-Erling Smørgrav 	}
527efcad6b7SDag-Erling Smørgrav 
528d4af9e69SDag-Erling Smørgrav 	return optind;
529efcad6b7SDag-Erling Smørgrav }
530efcad6b7SDag-Erling Smørgrav 
531efcad6b7SDag-Erling Smørgrav static int
532d4af9e69SDag-Erling Smørgrav parse_df_flags(const char *cmd, char **argv, int argc, int *hflag, int *iflag)
533efcad6b7SDag-Erling Smørgrav {
534d4af9e69SDag-Erling Smørgrav 	extern int opterr, optind, optopt, optreset;
535d4af9e69SDag-Erling Smørgrav 	int ch;
536efcad6b7SDag-Erling Smørgrav 
537d4af9e69SDag-Erling Smørgrav 	optind = optreset = 1;
538d4af9e69SDag-Erling Smørgrav 	opterr = 0;
539efcad6b7SDag-Erling Smørgrav 
540d4af9e69SDag-Erling Smørgrav 	*hflag = *iflag = 0;
541d4af9e69SDag-Erling Smørgrav 	while ((ch = getopt(argc, argv, "hi")) != -1) {
542d4af9e69SDag-Erling Smørgrav 		switch (ch) {
543d4af9e69SDag-Erling Smørgrav 		case 'h':
544d4af9e69SDag-Erling Smørgrav 			*hflag = 1;
545efcad6b7SDag-Erling Smørgrav 			break;
546d4af9e69SDag-Erling Smørgrav 		case 'i':
547d4af9e69SDag-Erling Smørgrav 			*iflag = 1;
548d4af9e69SDag-Erling Smørgrav 			break;
549d4af9e69SDag-Erling Smørgrav 		default:
550d4af9e69SDag-Erling Smørgrav 			error("%s: Invalid flag -%c", cmd, optopt);
551d4af9e69SDag-Erling Smørgrav 			return -1;
552efcad6b7SDag-Erling Smørgrav 		}
553efcad6b7SDag-Erling Smørgrav 	}
554efcad6b7SDag-Erling Smørgrav 
555d4af9e69SDag-Erling Smørgrav 	return optind;
556efcad6b7SDag-Erling Smørgrav }
557efcad6b7SDag-Erling Smørgrav 
558efcad6b7SDag-Erling Smørgrav static int
55919261079SEd Maste parse_ch_flags(const char *cmd, char **argv, int argc, int *hflag)
56019261079SEd Maste {
56119261079SEd Maste 	extern int opterr, optind, optopt, optreset;
56219261079SEd Maste 	int ch;
56319261079SEd Maste 
56419261079SEd Maste 	optind = optreset = 1;
56519261079SEd Maste 	opterr = 0;
56619261079SEd Maste 
56719261079SEd Maste 	*hflag = 0;
56819261079SEd Maste 	while ((ch = getopt(argc, argv, "h")) != -1) {
56919261079SEd Maste 		switch (ch) {
57019261079SEd Maste 		case 'h':
57119261079SEd Maste 			*hflag = 1;
57219261079SEd Maste 			break;
57319261079SEd Maste 		default:
57419261079SEd Maste 			error("%s: Invalid flag -%c", cmd, optopt);
57519261079SEd Maste 			return -1;
57619261079SEd Maste 		}
57719261079SEd Maste 	}
57819261079SEd Maste 
57919261079SEd Maste 	return optind;
58019261079SEd Maste }
58119261079SEd Maste 
58219261079SEd Maste static int
583f7167e0eSDag-Erling Smørgrav parse_no_flags(const char *cmd, char **argv, int argc)
584f7167e0eSDag-Erling Smørgrav {
585f7167e0eSDag-Erling Smørgrav 	extern int opterr, optind, optopt, optreset;
586f7167e0eSDag-Erling Smørgrav 	int ch;
587f7167e0eSDag-Erling Smørgrav 
588f7167e0eSDag-Erling Smørgrav 	optind = optreset = 1;
589f7167e0eSDag-Erling Smørgrav 	opterr = 0;
590f7167e0eSDag-Erling Smørgrav 
591f7167e0eSDag-Erling Smørgrav 	while ((ch = getopt(argc, argv, "")) != -1) {
592f7167e0eSDag-Erling Smørgrav 		switch (ch) {
593f7167e0eSDag-Erling Smørgrav 		default:
594f7167e0eSDag-Erling Smørgrav 			error("%s: Invalid flag -%c", cmd, optopt);
595f7167e0eSDag-Erling Smørgrav 			return -1;
596f7167e0eSDag-Erling Smørgrav 		}
597f7167e0eSDag-Erling Smørgrav 	}
598f7167e0eSDag-Erling Smørgrav 
599f7167e0eSDag-Erling Smørgrav 	return optind;
600f7167e0eSDag-Erling Smørgrav }
601f7167e0eSDag-Erling Smørgrav 
60238a52bd3SEd Maste static char *
60338a52bd3SEd Maste escape_glob(const char *s)
60438a52bd3SEd Maste {
60538a52bd3SEd Maste 	size_t i, o, len;
60638a52bd3SEd Maste 	char *ret;
60738a52bd3SEd Maste 
60838a52bd3SEd Maste 	len = strlen(s);
60938a52bd3SEd Maste 	ret = xcalloc(2, len + 1);
61038a52bd3SEd Maste 	for (i = o = 0; i < len; i++) {
61138a52bd3SEd Maste 		if (strchr("[]?*\\", s[i]) != NULL)
61238a52bd3SEd Maste 			ret[o++] = '\\';
61338a52bd3SEd Maste 		ret[o++] = s[i];
61438a52bd3SEd Maste 	}
61538a52bd3SEd Maste 	ret[o++] = '\0';
61638a52bd3SEd Maste 	return ret;
61738a52bd3SEd Maste }
61838a52bd3SEd Maste 
619535af610SEd Maste /*
620535af610SEd Maste  * Arg p must be dynamically allocated.  make_absolute will either return it
621535af610SEd Maste  * or free it and allocate a new one.  Caller must free returned string.
622535af610SEd Maste  */
62338a52bd3SEd Maste static char *
62469c72a57SEd Maste make_absolute_pwd_glob(char *p, const char *pwd)
62538a52bd3SEd Maste {
62638a52bd3SEd Maste 	char *ret, *escpwd;
62738a52bd3SEd Maste 
62838a52bd3SEd Maste 	escpwd = escape_glob(pwd);
62938a52bd3SEd Maste 	if (p == NULL)
63038a52bd3SEd Maste 		return escpwd;
631*edf85781SEd Maste 	ret = sftp_make_absolute(p, escpwd);
63238a52bd3SEd Maste 	free(escpwd);
63338a52bd3SEd Maste 	return ret;
63438a52bd3SEd Maste }
63538a52bd3SEd Maste 
636f7167e0eSDag-Erling Smørgrav static int
637*edf85781SEd Maste local_is_dir(const char *path)
638*edf85781SEd Maste {
639*edf85781SEd Maste 	struct stat sb;
640*edf85781SEd Maste 
641*edf85781SEd Maste 	if (stat(path, &sb) == -1)
642*edf85781SEd Maste 		return 0;
643*edf85781SEd Maste 	return S_ISDIR(sb.st_mode);
644*edf85781SEd Maste }
645*edf85781SEd Maste 
646*edf85781SEd Maste static int
647076ad2f8SDag-Erling Smørgrav process_get(struct sftp_conn *conn, const char *src, const char *dst,
648076ad2f8SDag-Erling Smørgrav     const char *pwd, int pflag, int rflag, int resume, int fflag)
649efcad6b7SDag-Erling Smørgrav {
65038a52bd3SEd Maste 	char *filename, *abs_src = NULL, *abs_dst = NULL, *tmp = NULL;
651efcad6b7SDag-Erling Smørgrav 	glob_t g;
652a0ee8cc6SDag-Erling Smørgrav 	int i, r, err = 0;
653efcad6b7SDag-Erling Smørgrav 
65469c72a57SEd Maste 	abs_src = make_absolute_pwd_glob(xstrdup(src), pwd);
655efcad6b7SDag-Erling Smørgrav 	memset(&g, 0, sizeof(g));
656b15c8340SDag-Erling Smørgrav 
657efcad6b7SDag-Erling Smørgrav 	debug3("Looking up %s", abs_src);
658*edf85781SEd Maste 	if ((r = sftp_glob(conn, abs_src, GLOB_MARK, NULL, &g)) != 0) {
659a0ee8cc6SDag-Erling Smørgrav 		if (r == GLOB_NOSPACE) {
660a0ee8cc6SDag-Erling Smørgrav 			error("Too many matches for \"%s\".", abs_src);
661a0ee8cc6SDag-Erling Smørgrav 		} else {
662efcad6b7SDag-Erling Smørgrav 			error("File \"%s\" not found.", abs_src);
663a0ee8cc6SDag-Erling Smørgrav 		}
664efcad6b7SDag-Erling Smørgrav 		err = -1;
665efcad6b7SDag-Erling Smørgrav 		goto out;
666efcad6b7SDag-Erling Smørgrav 	}
667efcad6b7SDag-Erling Smørgrav 
668b15c8340SDag-Erling Smørgrav 	/*
669b15c8340SDag-Erling Smørgrav 	 * If multiple matches then dst must be a directory or
670b15c8340SDag-Erling Smørgrav 	 * unspecified.
671b15c8340SDag-Erling Smørgrav 	 */
67219261079SEd Maste 	if (g.gl_matchc > 1 && dst != NULL && !local_is_dir(dst)) {
673b15c8340SDag-Erling Smørgrav 		error("Multiple source paths, but destination "
674b15c8340SDag-Erling Smørgrav 		    "\"%s\" is not a directory", dst);
675efcad6b7SDag-Erling Smørgrav 		err = -1;
676efcad6b7SDag-Erling Smørgrav 		goto out;
677efcad6b7SDag-Erling Smørgrav 	}
678efcad6b7SDag-Erling Smørgrav 
679d74d50a8SDag-Erling Smørgrav 	for (i = 0; g.gl_pathv[i] && !interrupted; i++) {
680b15c8340SDag-Erling Smørgrav 		tmp = xstrdup(g.gl_pathv[i]);
681b15c8340SDag-Erling Smørgrav 		if ((filename = basename(tmp)) == NULL) {
682b15c8340SDag-Erling Smørgrav 			error("basename %s: %s", tmp, strerror(errno));
683e4a9863fSDag-Erling Smørgrav 			free(tmp);
684efcad6b7SDag-Erling Smørgrav 			err = -1;
685efcad6b7SDag-Erling Smørgrav 			goto out;
686efcad6b7SDag-Erling Smørgrav 		}
687efcad6b7SDag-Erling Smørgrav 
688efcad6b7SDag-Erling Smørgrav 		if (g.gl_matchc == 1 && dst) {
68919261079SEd Maste 			if (local_is_dir(dst)) {
690*edf85781SEd Maste 				abs_dst = sftp_path_append(dst, filename);
691b15c8340SDag-Erling Smørgrav 			} else {
692efcad6b7SDag-Erling Smørgrav 				abs_dst = xstrdup(dst);
693b15c8340SDag-Erling Smørgrav 			}
694efcad6b7SDag-Erling Smørgrav 		} else if (dst) {
695*edf85781SEd Maste 			abs_dst = sftp_path_append(dst, filename);
696b15c8340SDag-Erling Smørgrav 		} else {
697b15c8340SDag-Erling Smørgrav 			abs_dst = xstrdup(filename);
698b15c8340SDag-Erling Smørgrav 		}
699e4a9863fSDag-Erling Smørgrav 		free(tmp);
700efcad6b7SDag-Erling Smørgrav 
701e4a9863fSDag-Erling Smørgrav 		resume |= global_aflag;
702e4a9863fSDag-Erling Smørgrav 		if (!quiet && resume)
703076ad2f8SDag-Erling Smørgrav 			mprintf("Resuming %s to %s\n",
704076ad2f8SDag-Erling Smørgrav 			    g.gl_pathv[i], abs_dst);
705e4a9863fSDag-Erling Smørgrav 		else if (!quiet && !resume)
706076ad2f8SDag-Erling Smørgrav 			mprintf("Fetching %s to %s\n",
707076ad2f8SDag-Erling Smørgrav 			    g.gl_pathv[i], abs_dst);
70819261079SEd Maste 		/* XXX follow link flag */
709*edf85781SEd Maste 		if (sftp_globpath_is_dir(g.gl_pathv[i]) &&
710*edf85781SEd Maste 		    (rflag || global_rflag)) {
711*edf85781SEd Maste 			if (sftp_download_dir(conn, g.gl_pathv[i], abs_dst,
712*edf85781SEd Maste 			    NULL, pflag || global_pflag, 1, resume,
71338a52bd3SEd Maste 			    fflag || global_fflag, 0, 0) == -1)
714efcad6b7SDag-Erling Smørgrav 				err = -1;
715b15c8340SDag-Erling Smørgrav 		} else {
716*edf85781SEd Maste 			if (sftp_download(conn, g.gl_pathv[i], abs_dst, NULL,
717f7167e0eSDag-Erling Smørgrav 			    pflag || global_pflag, resume,
71838a52bd3SEd Maste 			    fflag || global_fflag, 0) == -1)
719b15c8340SDag-Erling Smørgrav 				err = -1;
720b15c8340SDag-Erling Smørgrav 		}
721e4a9863fSDag-Erling Smørgrav 		free(abs_dst);
722efcad6b7SDag-Erling Smørgrav 		abs_dst = NULL;
723efcad6b7SDag-Erling Smørgrav 	}
724efcad6b7SDag-Erling Smørgrav 
725efcad6b7SDag-Erling Smørgrav out:
726e4a9863fSDag-Erling Smørgrav 	free(abs_src);
727efcad6b7SDag-Erling Smørgrav 	globfree(&g);
728efcad6b7SDag-Erling Smørgrav 	return(err);
729efcad6b7SDag-Erling Smørgrav }
730efcad6b7SDag-Erling Smørgrav 
731efcad6b7SDag-Erling Smørgrav static int
732076ad2f8SDag-Erling Smørgrav process_put(struct sftp_conn *conn, const char *src, const char *dst,
733076ad2f8SDag-Erling Smørgrav     const char *pwd, int pflag, int rflag, int resume, int fflag)
734efcad6b7SDag-Erling Smørgrav {
735efcad6b7SDag-Erling Smørgrav 	char *tmp_dst = NULL;
736efcad6b7SDag-Erling Smørgrav 	char *abs_dst = NULL;
737b15c8340SDag-Erling Smørgrav 	char *tmp = NULL, *filename = NULL;
738efcad6b7SDag-Erling Smørgrav 	glob_t g;
739efcad6b7SDag-Erling Smørgrav 	int err = 0;
740b15c8340SDag-Erling Smørgrav 	int i, dst_is_dir = 1;
741d4af9e69SDag-Erling Smørgrav 	struct stat sb;
742efcad6b7SDag-Erling Smørgrav 
743efcad6b7SDag-Erling Smørgrav 	if (dst) {
744efcad6b7SDag-Erling Smørgrav 		tmp_dst = xstrdup(dst);
745*edf85781SEd Maste 		tmp_dst = sftp_make_absolute(tmp_dst, pwd);
746efcad6b7SDag-Erling Smørgrav 	}
747efcad6b7SDag-Erling Smørgrav 
748efcad6b7SDag-Erling Smørgrav 	memset(&g, 0, sizeof(g));
749efcad6b7SDag-Erling Smørgrav 	debug3("Looking up %s", src);
750b15c8340SDag-Erling Smørgrav 	if (glob(src, GLOB_NOCHECK | GLOB_MARK, NULL, &g)) {
751efcad6b7SDag-Erling Smørgrav 		error("File \"%s\" not found.", src);
752efcad6b7SDag-Erling Smørgrav 		err = -1;
753efcad6b7SDag-Erling Smørgrav 		goto out;
754efcad6b7SDag-Erling Smørgrav 	}
755efcad6b7SDag-Erling Smørgrav 
756b15c8340SDag-Erling Smørgrav 	/* If we aren't fetching to pwd then stash this status for later */
757b15c8340SDag-Erling Smørgrav 	if (tmp_dst != NULL)
758*edf85781SEd Maste 		dst_is_dir = sftp_remote_is_dir(conn, tmp_dst);
759b15c8340SDag-Erling Smørgrav 
760efcad6b7SDag-Erling Smørgrav 	/* If multiple matches, dst may be directory or unspecified */
761b15c8340SDag-Erling Smørgrav 	if (g.gl_matchc > 1 && tmp_dst && !dst_is_dir) {
762b15c8340SDag-Erling Smørgrav 		error("Multiple paths match, but destination "
763b15c8340SDag-Erling Smørgrav 		    "\"%s\" is not a directory", tmp_dst);
764efcad6b7SDag-Erling Smørgrav 		err = -1;
765efcad6b7SDag-Erling Smørgrav 		goto out;
766efcad6b7SDag-Erling Smørgrav 	}
767efcad6b7SDag-Erling Smørgrav 
768d74d50a8SDag-Erling Smørgrav 	for (i = 0; g.gl_pathv[i] && !interrupted; i++) {
769d4af9e69SDag-Erling Smørgrav 		if (stat(g.gl_pathv[i], &sb) == -1) {
770d4af9e69SDag-Erling Smørgrav 			err = -1;
771d4af9e69SDag-Erling Smørgrav 			error("stat %s: %s", g.gl_pathv[i], strerror(errno));
772d4af9e69SDag-Erling Smørgrav 			continue;
773d4af9e69SDag-Erling Smørgrav 		}
774d4af9e69SDag-Erling Smørgrav 
775b15c8340SDag-Erling Smørgrav 		tmp = xstrdup(g.gl_pathv[i]);
776b15c8340SDag-Erling Smørgrav 		if ((filename = basename(tmp)) == NULL) {
777b15c8340SDag-Erling Smørgrav 			error("basename %s: %s", tmp, strerror(errno));
778e4a9863fSDag-Erling Smørgrav 			free(tmp);
779efcad6b7SDag-Erling Smørgrav 			err = -1;
780efcad6b7SDag-Erling Smørgrav 			goto out;
781efcad6b7SDag-Erling Smørgrav 		}
782efcad6b7SDag-Erling Smørgrav 
783535af610SEd Maste 		free(abs_dst);
784535af610SEd Maste 		abs_dst = NULL;
785efcad6b7SDag-Erling Smørgrav 		if (g.gl_matchc == 1 && tmp_dst) {
786efcad6b7SDag-Erling Smørgrav 			/* If directory specified, append filename */
787b15c8340SDag-Erling Smørgrav 			if (dst_is_dir)
788*edf85781SEd Maste 				abs_dst = sftp_path_append(tmp_dst, filename);
789b15c8340SDag-Erling Smørgrav 			else
790efcad6b7SDag-Erling Smørgrav 				abs_dst = xstrdup(tmp_dst);
791efcad6b7SDag-Erling Smørgrav 		} else if (tmp_dst) {
792*edf85781SEd Maste 			abs_dst = sftp_path_append(tmp_dst, filename);
793b15c8340SDag-Erling Smørgrav 		} else {
794*edf85781SEd Maste 			abs_dst = sftp_make_absolute(xstrdup(filename), pwd);
795b15c8340SDag-Erling Smørgrav 		}
796e4a9863fSDag-Erling Smørgrav 		free(tmp);
797efcad6b7SDag-Erling Smørgrav 
798a0ee8cc6SDag-Erling Smørgrav 		resume |= global_aflag;
799a0ee8cc6SDag-Erling Smørgrav 		if (!quiet && resume)
800076ad2f8SDag-Erling Smørgrav 			mprintf("Resuming upload of %s to %s\n",
801076ad2f8SDag-Erling Smørgrav 			    g.gl_pathv[i], abs_dst);
802a0ee8cc6SDag-Erling Smørgrav 		else if (!quiet && !resume)
803076ad2f8SDag-Erling Smørgrav 			mprintf("Uploading %s to %s\n",
804076ad2f8SDag-Erling Smørgrav 			    g.gl_pathv[i], abs_dst);
80519261079SEd Maste 		/* XXX follow_link_flag */
806*edf85781SEd Maste 		if (sftp_globpath_is_dir(g.gl_pathv[i]) &&
807*edf85781SEd Maste 		    (rflag || global_rflag)) {
808*edf85781SEd Maste 			if (sftp_upload_dir(conn, g.gl_pathv[i], abs_dst,
809a0ee8cc6SDag-Erling Smørgrav 			    pflag || global_pflag, 1, resume,
81038a52bd3SEd Maste 			    fflag || global_fflag, 0, 0) == -1)
811efcad6b7SDag-Erling Smørgrav 				err = -1;
812b15c8340SDag-Erling Smørgrav 		} else {
813*edf85781SEd Maste 			if (sftp_upload(conn, g.gl_pathv[i], abs_dst,
814a0ee8cc6SDag-Erling Smørgrav 			    pflag || global_pflag, resume,
81538a52bd3SEd Maste 			    fflag || global_fflag, 0) == -1)
816b15c8340SDag-Erling Smørgrav 				err = -1;
817b15c8340SDag-Erling Smørgrav 		}
818efcad6b7SDag-Erling Smørgrav 	}
819efcad6b7SDag-Erling Smørgrav 
820efcad6b7SDag-Erling Smørgrav out:
821e4a9863fSDag-Erling Smørgrav 	free(abs_dst);
822e4a9863fSDag-Erling Smørgrav 	free(tmp_dst);
823efcad6b7SDag-Erling Smørgrav 	globfree(&g);
824efcad6b7SDag-Erling Smørgrav 	return(err);
825efcad6b7SDag-Erling Smørgrav }
826efcad6b7SDag-Erling Smørgrav 
827efcad6b7SDag-Erling Smørgrav static int
828efcad6b7SDag-Erling Smørgrav sdirent_comp(const void *aa, const void *bb)
829efcad6b7SDag-Erling Smørgrav {
830efcad6b7SDag-Erling Smørgrav 	SFTP_DIRENT *a = *(SFTP_DIRENT **)aa;
831efcad6b7SDag-Erling Smørgrav 	SFTP_DIRENT *b = *(SFTP_DIRENT **)bb;
832d74d50a8SDag-Erling Smørgrav 	int rmul = sort_flag & LS_REVERSE_SORT ? -1 : 1;
833efcad6b7SDag-Erling Smørgrav 
834d74d50a8SDag-Erling Smørgrav #define NCMP(a,b) (a == b ? 0 : (a < b ? 1 : -1))
835d74d50a8SDag-Erling Smørgrav 	if (sort_flag & LS_NAME_SORT)
836d74d50a8SDag-Erling Smørgrav 		return (rmul * strcmp(a->filename, b->filename));
837d74d50a8SDag-Erling Smørgrav 	else if (sort_flag & LS_TIME_SORT)
838d74d50a8SDag-Erling Smørgrav 		return (rmul * NCMP(a->a.mtime, b->a.mtime));
839d74d50a8SDag-Erling Smørgrav 	else if (sort_flag & LS_SIZE_SORT)
840d74d50a8SDag-Erling Smørgrav 		return (rmul * NCMP(a->a.size, b->a.size));
841d74d50a8SDag-Erling Smørgrav 
842d74d50a8SDag-Erling Smørgrav 	fatal("Unknown ls sort type");
843efcad6b7SDag-Erling Smørgrav }
844efcad6b7SDag-Erling Smørgrav 
845efcad6b7SDag-Erling Smørgrav /* sftp ls.1 replacement for directories */
846efcad6b7SDag-Erling Smørgrav static int
847076ad2f8SDag-Erling Smørgrav do_ls_dir(struct sftp_conn *conn, const char *path,
848076ad2f8SDag-Erling Smørgrav     const char *strip_path, int lflag)
849efcad6b7SDag-Erling Smørgrav {
850043840dfSDag-Erling Smørgrav 	int n;
851043840dfSDag-Erling Smørgrav 	u_int c = 1, colspace = 0, columns = 1;
852efcad6b7SDag-Erling Smørgrav 	SFTP_DIRENT **d;
853efcad6b7SDag-Erling Smørgrav 
854*edf85781SEd Maste 	if ((n = sftp_readdir(conn, path, &d)) != 0)
855efcad6b7SDag-Erling Smørgrav 		return (n);
856efcad6b7SDag-Erling Smørgrav 
857d74d50a8SDag-Erling Smørgrav 	if (!(lflag & LS_SHORT_VIEW)) {
858043840dfSDag-Erling Smørgrav 		u_int m = 0, width = 80;
859efcad6b7SDag-Erling Smørgrav 		struct winsize ws;
860efcad6b7SDag-Erling Smørgrav 		char *tmp;
861efcad6b7SDag-Erling Smørgrav 
862efcad6b7SDag-Erling Smørgrav 		/* Count entries for sort and find longest filename */
863d74d50a8SDag-Erling Smørgrav 		for (n = 0; d[n] != NULL; n++) {
864d74d50a8SDag-Erling Smørgrav 			if (d[n]->filename[0] != '.' || (lflag & LS_SHOW_ALL))
865ca86bcf2SDag-Erling Smørgrav 				m = MAXIMUM(m, strlen(d[n]->filename));
866d74d50a8SDag-Erling Smørgrav 		}
867efcad6b7SDag-Erling Smørgrav 
868efcad6b7SDag-Erling Smørgrav 		/* Add any subpath that also needs to be counted */
869efcad6b7SDag-Erling Smørgrav 		tmp = path_strip(path, strip_path);
870efcad6b7SDag-Erling Smørgrav 		m += strlen(tmp);
871e4a9863fSDag-Erling Smørgrav 		free(tmp);
872efcad6b7SDag-Erling Smørgrav 
873efcad6b7SDag-Erling Smørgrav 		if (ioctl(fileno(stdin), TIOCGWINSZ, &ws) != -1)
874efcad6b7SDag-Erling Smørgrav 			width = ws.ws_col;
875efcad6b7SDag-Erling Smørgrav 
876efcad6b7SDag-Erling Smørgrav 		columns = width / (m + 2);
877ca86bcf2SDag-Erling Smørgrav 		columns = MAXIMUM(columns, 1);
878efcad6b7SDag-Erling Smørgrav 		colspace = width / columns;
879ca86bcf2SDag-Erling Smørgrav 		colspace = MINIMUM(colspace, width);
880efcad6b7SDag-Erling Smørgrav 	}
881efcad6b7SDag-Erling Smørgrav 
882d74d50a8SDag-Erling Smørgrav 	if (lflag & SORT_FLAGS) {
883021d409fSDag-Erling Smørgrav 		for (n = 0; d[n] != NULL; n++)
884021d409fSDag-Erling Smørgrav 			;	/* count entries */
885d74d50a8SDag-Erling Smørgrav 		sort_flag = lflag & (SORT_FLAGS|LS_REVERSE_SORT);
886efcad6b7SDag-Erling Smørgrav 		qsort(d, n, sizeof(*d), sdirent_comp);
887d74d50a8SDag-Erling Smørgrav 	}
888efcad6b7SDag-Erling Smørgrav 
88938a52bd3SEd Maste 	get_remote_user_groups_from_dirents(conn, d);
890d74d50a8SDag-Erling Smørgrav 	for (n = 0; d[n] != NULL && !interrupted; n++) {
891efcad6b7SDag-Erling Smørgrav 		char *tmp, *fname;
892efcad6b7SDag-Erling Smørgrav 
893d74d50a8SDag-Erling Smørgrav 		if (d[n]->filename[0] == '.' && !(lflag & LS_SHOW_ALL))
894d74d50a8SDag-Erling Smørgrav 			continue;
895d74d50a8SDag-Erling Smørgrav 
896*edf85781SEd Maste 		tmp = sftp_path_append(path, d[n]->filename);
897efcad6b7SDag-Erling Smørgrav 		fname = path_strip(tmp, strip_path);
898e4a9863fSDag-Erling Smørgrav 		free(tmp);
899efcad6b7SDag-Erling Smørgrav 
900d74d50a8SDag-Erling Smørgrav 		if (lflag & LS_LONG_VIEW) {
90138a52bd3SEd Maste 			if ((lflag & (LS_NUMERIC_VIEW|LS_SI_UNITS)) != 0 ||
902*edf85781SEd Maste 			    sftp_can_get_users_groups_by_id(conn)) {
903efcad6b7SDag-Erling Smørgrav 				char *lname;
904efcad6b7SDag-Erling Smørgrav 				struct stat sb;
905efcad6b7SDag-Erling Smørgrav 
906efcad6b7SDag-Erling Smørgrav 				memset(&sb, 0, sizeof(sb));
907efcad6b7SDag-Erling Smørgrav 				attrib_to_stat(&d[n]->a, &sb);
908b15c8340SDag-Erling Smørgrav 				lname = ls_file(fname, &sb, 1,
90938a52bd3SEd Maste 				    (lflag & LS_SI_UNITS),
91038a52bd3SEd Maste 				    ruser_name(sb.st_uid),
91138a52bd3SEd Maste 				    rgroup_name(sb.st_gid));
912076ad2f8SDag-Erling Smørgrav 				mprintf("%s\n", lname);
913e4a9863fSDag-Erling Smørgrav 				free(lname);
914d74d50a8SDag-Erling Smørgrav 			} else
915076ad2f8SDag-Erling Smørgrav 				mprintf("%s\n", d[n]->longname);
916efcad6b7SDag-Erling Smørgrav 		} else {
917076ad2f8SDag-Erling Smørgrav 			mprintf("%-*s", colspace, fname);
918efcad6b7SDag-Erling Smørgrav 			if (c >= columns) {
919efcad6b7SDag-Erling Smørgrav 				printf("\n");
920efcad6b7SDag-Erling Smørgrav 				c = 1;
921efcad6b7SDag-Erling Smørgrav 			} else
922efcad6b7SDag-Erling Smørgrav 				c++;
923efcad6b7SDag-Erling Smørgrav 		}
924efcad6b7SDag-Erling Smørgrav 
925e4a9863fSDag-Erling Smørgrav 		free(fname);
926efcad6b7SDag-Erling Smørgrav 	}
927efcad6b7SDag-Erling Smørgrav 
928d74d50a8SDag-Erling Smørgrav 	if (!(lflag & LS_LONG_VIEW) && (c != 1))
929efcad6b7SDag-Erling Smørgrav 		printf("\n");
930efcad6b7SDag-Erling Smørgrav 
931*edf85781SEd Maste 	sftp_free_dirents(d);
932efcad6b7SDag-Erling Smørgrav 	return (0);
933efcad6b7SDag-Erling Smørgrav }
934efcad6b7SDag-Erling Smørgrav 
9354f52dfbbSDag-Erling Smørgrav static int
9364f52dfbbSDag-Erling Smørgrav sglob_comp(const void *aa, const void *bb)
9374f52dfbbSDag-Erling Smørgrav {
9384f52dfbbSDag-Erling Smørgrav 	u_int a = *(const u_int *)aa;
9394f52dfbbSDag-Erling Smørgrav 	u_int b = *(const u_int *)bb;
9404f52dfbbSDag-Erling Smørgrav 	const char *ap = sort_glob->gl_pathv[a];
9414f52dfbbSDag-Erling Smørgrav 	const char *bp = sort_glob->gl_pathv[b];
9424f52dfbbSDag-Erling Smørgrav 	const struct stat *as = sort_glob->gl_statv[a];
9434f52dfbbSDag-Erling Smørgrav 	const struct stat *bs = sort_glob->gl_statv[b];
9444f52dfbbSDag-Erling Smørgrav 	int rmul = sort_flag & LS_REVERSE_SORT ? -1 : 1;
9454f52dfbbSDag-Erling Smørgrav 
9464f52dfbbSDag-Erling Smørgrav #define NCMP(a,b) (a == b ? 0 : (a < b ? 1 : -1))
9474f52dfbbSDag-Erling Smørgrav 	if (sort_flag & LS_NAME_SORT)
9484f52dfbbSDag-Erling Smørgrav 		return (rmul * strcmp(ap, bp));
9494f52dfbbSDag-Erling Smørgrav 	else if (sort_flag & LS_TIME_SORT) {
9504f52dfbbSDag-Erling Smørgrav #if defined(HAVE_STRUCT_STAT_ST_MTIM)
95119261079SEd Maste 		if (timespeccmp(&as->st_mtim, &bs->st_mtim, ==))
95219261079SEd Maste 			return 0;
95319261079SEd Maste 		return timespeccmp(&as->st_mtim, &bs->st_mtim, <) ?
95419261079SEd Maste 		    rmul : -rmul;
9554f52dfbbSDag-Erling Smørgrav #elif defined(HAVE_STRUCT_STAT_ST_MTIME)
9564f52dfbbSDag-Erling Smørgrav 		return (rmul * NCMP(as->st_mtime, bs->st_mtime));
9574f52dfbbSDag-Erling Smørgrav #else
9584f52dfbbSDag-Erling Smørgrav 	return rmul * 1;
9594f52dfbbSDag-Erling Smørgrav #endif
9604f52dfbbSDag-Erling Smørgrav 	} else if (sort_flag & LS_SIZE_SORT)
9614f52dfbbSDag-Erling Smørgrav 		return (rmul * NCMP(as->st_size, bs->st_size));
9624f52dfbbSDag-Erling Smørgrav 
9634f52dfbbSDag-Erling Smørgrav 	fatal("Unknown ls sort type");
9644f52dfbbSDag-Erling Smørgrav }
9654f52dfbbSDag-Erling Smørgrav 
966efcad6b7SDag-Erling Smørgrav /* sftp ls.1 replacement which handles path globs */
967efcad6b7SDag-Erling Smørgrav static int
968076ad2f8SDag-Erling Smørgrav do_globbed_ls(struct sftp_conn *conn, const char *path,
969076ad2f8SDag-Erling Smørgrav     const char *strip_path, int lflag)
970efcad6b7SDag-Erling Smørgrav {
9714a421b63SDag-Erling Smørgrav 	char *fname, *lname;
9724a421b63SDag-Erling Smørgrav 	glob_t g;
973a0ee8cc6SDag-Erling Smørgrav 	int err, r;
9744a421b63SDag-Erling Smørgrav 	struct winsize ws;
9754f52dfbbSDag-Erling Smørgrav 	u_int i, j, nentries, *indices = NULL, c = 1;
9764f52dfbbSDag-Erling Smørgrav 	u_int colspace = 0, columns = 1, m = 0, width = 80;
977efcad6b7SDag-Erling Smørgrav 
978efcad6b7SDag-Erling Smørgrav 	memset(&g, 0, sizeof(g));
979efcad6b7SDag-Erling Smørgrav 
980*edf85781SEd Maste 	if ((r = sftp_glob(conn, path,
981462c32cbSDag-Erling Smørgrav 	    GLOB_MARK|GLOB_NOCHECK|GLOB_BRACE|GLOB_KEEPSTAT|GLOB_NOSORT,
982a0ee8cc6SDag-Erling Smørgrav 	    NULL, &g)) != 0 ||
9834a421b63SDag-Erling Smørgrav 	    (g.gl_pathc && !g.gl_matchc)) {
9845e8dbd04SDag-Erling Smørgrav 		if (g.gl_pathc)
9855e8dbd04SDag-Erling Smørgrav 			globfree(&g);
986a0ee8cc6SDag-Erling Smørgrav 		if (r == GLOB_NOSPACE) {
987a0ee8cc6SDag-Erling Smørgrav 			error("Can't ls: Too many matches for \"%s\"", path);
988a0ee8cc6SDag-Erling Smørgrav 		} else {
989efcad6b7SDag-Erling Smørgrav 			error("Can't ls: \"%s\" not found", path);
990a0ee8cc6SDag-Erling Smørgrav 		}
9914a421b63SDag-Erling Smørgrav 		return -1;
992efcad6b7SDag-Erling Smørgrav 	}
993efcad6b7SDag-Erling Smørgrav 
994d74d50a8SDag-Erling Smørgrav 	if (interrupted)
995d74d50a8SDag-Erling Smørgrav 		goto out;
996d74d50a8SDag-Erling Smørgrav 
997efcad6b7SDag-Erling Smørgrav 	/*
9985e8dbd04SDag-Erling Smørgrav 	 * If the glob returns a single match and it is a directory,
9995e8dbd04SDag-Erling Smørgrav 	 * then just list its contents.
1000efcad6b7SDag-Erling Smørgrav 	 */
10014a421b63SDag-Erling Smørgrav 	if (g.gl_matchc == 1 && g.gl_statv[0] != NULL &&
10024a421b63SDag-Erling Smørgrav 	    S_ISDIR(g.gl_statv[0]->st_mode)) {
10035e8dbd04SDag-Erling Smørgrav 		err = do_ls_dir(conn, g.gl_pathv[0], strip_path, lflag);
1004efcad6b7SDag-Erling Smørgrav 		globfree(&g);
10054a421b63SDag-Erling Smørgrav 		return err;
1006efcad6b7SDag-Erling Smørgrav 	}
1007efcad6b7SDag-Erling Smørgrav 
1008efcad6b7SDag-Erling Smørgrav 	if (ioctl(fileno(stdin), TIOCGWINSZ, &ws) != -1)
1009efcad6b7SDag-Erling Smørgrav 		width = ws.ws_col;
1010efcad6b7SDag-Erling Smørgrav 
10114a421b63SDag-Erling Smørgrav 	if (!(lflag & LS_SHORT_VIEW)) {
10124a421b63SDag-Erling Smørgrav 		/* Count entries for sort and find longest filename */
10134a421b63SDag-Erling Smørgrav 		for (i = 0; g.gl_pathv[i]; i++)
1014ca86bcf2SDag-Erling Smørgrav 			m = MAXIMUM(m, strlen(g.gl_pathv[i]));
10154a421b63SDag-Erling Smørgrav 
1016efcad6b7SDag-Erling Smørgrav 		columns = width / (m + 2);
1017ca86bcf2SDag-Erling Smørgrav 		columns = MAXIMUM(columns, 1);
1018efcad6b7SDag-Erling Smørgrav 		colspace = width / columns;
1019efcad6b7SDag-Erling Smørgrav 	}
1020efcad6b7SDag-Erling Smørgrav 
10214f52dfbbSDag-Erling Smørgrav 	/*
10224f52dfbbSDag-Erling Smørgrav 	 * Sorting: rather than mess with the contents of glob_t, prepare
10234f52dfbbSDag-Erling Smørgrav 	 * an array of indices into it and sort that. For the usual
10244f52dfbbSDag-Erling Smørgrav 	 * unsorted case, the indices are just the identity 1=1, 2=2, etc.
10254f52dfbbSDag-Erling Smørgrav 	 */
10264f52dfbbSDag-Erling Smørgrav 	for (nentries = 0; g.gl_pathv[nentries] != NULL; nentries++)
10274f52dfbbSDag-Erling Smørgrav 		;	/* count entries */
10284d3fc8b0SEd Maste 	indices = xcalloc(nentries, sizeof(*indices));
10294f52dfbbSDag-Erling Smørgrav 	for (i = 0; i < nentries; i++)
10304f52dfbbSDag-Erling Smørgrav 		indices[i] = i;
10314f52dfbbSDag-Erling Smørgrav 
10324f52dfbbSDag-Erling Smørgrav 	if (lflag & SORT_FLAGS) {
10334f52dfbbSDag-Erling Smørgrav 		sort_glob = &g;
10344f52dfbbSDag-Erling Smørgrav 		sort_flag = lflag & (SORT_FLAGS|LS_REVERSE_SORT);
10354f52dfbbSDag-Erling Smørgrav 		qsort(indices, nentries, sizeof(*indices), sglob_comp);
10364f52dfbbSDag-Erling Smørgrav 		sort_glob = NULL;
10374f52dfbbSDag-Erling Smørgrav 	}
10384f52dfbbSDag-Erling Smørgrav 
103938a52bd3SEd Maste 	get_remote_user_groups_from_glob(conn, &g);
10404f52dfbbSDag-Erling Smørgrav 	for (j = 0; j < nentries && !interrupted; j++) {
10414f52dfbbSDag-Erling Smørgrav 		i = indices[j];
1042efcad6b7SDag-Erling Smørgrav 		fname = path_strip(g.gl_pathv[i], strip_path);
1043d74d50a8SDag-Erling Smørgrav 		if (lflag & LS_LONG_VIEW) {
10444a421b63SDag-Erling Smørgrav 			if (g.gl_statv[i] == NULL) {
10454a421b63SDag-Erling Smørgrav 				error("no stat information for %s", fname);
10464d3fc8b0SEd Maste 				free(fname);
10474a421b63SDag-Erling Smørgrav 				continue;
10484a421b63SDag-Erling Smørgrav 			}
10494a421b63SDag-Erling Smørgrav 			lname = ls_file(fname, g.gl_statv[i], 1,
105038a52bd3SEd Maste 			    (lflag & LS_SI_UNITS),
105138a52bd3SEd Maste 			    ruser_name(g.gl_statv[i]->st_uid),
105238a52bd3SEd Maste 			    rgroup_name(g.gl_statv[i]->st_gid));
1053076ad2f8SDag-Erling Smørgrav 			mprintf("%s\n", lname);
1054e4a9863fSDag-Erling Smørgrav 			free(lname);
1055efcad6b7SDag-Erling Smørgrav 		} else {
1056076ad2f8SDag-Erling Smørgrav 			mprintf("%-*s", colspace, fname);
1057efcad6b7SDag-Erling Smørgrav 			if (c >= columns) {
1058efcad6b7SDag-Erling Smørgrav 				printf("\n");
1059efcad6b7SDag-Erling Smørgrav 				c = 1;
1060efcad6b7SDag-Erling Smørgrav 			} else
1061efcad6b7SDag-Erling Smørgrav 				c++;
1062efcad6b7SDag-Erling Smørgrav 		}
1063e4a9863fSDag-Erling Smørgrav 		free(fname);
1064efcad6b7SDag-Erling Smørgrav 	}
1065efcad6b7SDag-Erling Smørgrav 
1066d74d50a8SDag-Erling Smørgrav 	if (!(lflag & LS_LONG_VIEW) && (c != 1))
1067efcad6b7SDag-Erling Smørgrav 		printf("\n");
1068efcad6b7SDag-Erling Smørgrav 
1069d74d50a8SDag-Erling Smørgrav  out:
1070efcad6b7SDag-Erling Smørgrav 	if (g.gl_pathc)
1071efcad6b7SDag-Erling Smørgrav 		globfree(&g);
10724f52dfbbSDag-Erling Smørgrav 	free(indices);
1073efcad6b7SDag-Erling Smørgrav 
10744a421b63SDag-Erling Smørgrav 	return 0;
1075efcad6b7SDag-Erling Smørgrav }
1076efcad6b7SDag-Erling Smørgrav 
1077efcad6b7SDag-Erling Smørgrav static int
1078076ad2f8SDag-Erling Smørgrav do_df(struct sftp_conn *conn, const char *path, int hflag, int iflag)
1079d4af9e69SDag-Erling Smørgrav {
1080d4af9e69SDag-Erling Smørgrav 	struct sftp_statvfs st;
1081d93a896eSDag-Erling Smørgrav 	char s_used[FMT_SCALED_STRSIZE], s_avail[FMT_SCALED_STRSIZE];
1082d93a896eSDag-Erling Smørgrav 	char s_root[FMT_SCALED_STRSIZE], s_total[FMT_SCALED_STRSIZE];
1083d93a896eSDag-Erling Smørgrav 	char s_icapacity[16], s_dcapacity[16];
1084d4af9e69SDag-Erling Smørgrav 
1085*edf85781SEd Maste 	if (sftp_statvfs(conn, path, &st, 1) == -1)
1086d4af9e69SDag-Erling Smørgrav 		return -1;
1087d93a896eSDag-Erling Smørgrav 	if (st.f_files == 0)
1088d93a896eSDag-Erling Smørgrav 		strlcpy(s_icapacity, "ERR", sizeof(s_icapacity));
1089d93a896eSDag-Erling Smørgrav 	else {
1090d93a896eSDag-Erling Smørgrav 		snprintf(s_icapacity, sizeof(s_icapacity), "%3llu%%",
1091d93a896eSDag-Erling Smørgrav 		    (unsigned long long)(100 * (st.f_files - st.f_ffree) /
1092d93a896eSDag-Erling Smørgrav 		    st.f_files));
1093d93a896eSDag-Erling Smørgrav 	}
1094d93a896eSDag-Erling Smørgrav 	if (st.f_blocks == 0)
1095d93a896eSDag-Erling Smørgrav 		strlcpy(s_dcapacity, "ERR", sizeof(s_dcapacity));
1096d93a896eSDag-Erling Smørgrav 	else {
1097d93a896eSDag-Erling Smørgrav 		snprintf(s_dcapacity, sizeof(s_dcapacity), "%3llu%%",
1098d93a896eSDag-Erling Smørgrav 		    (unsigned long long)(100 * (st.f_blocks - st.f_bfree) /
1099d93a896eSDag-Erling Smørgrav 		    st.f_blocks));
1100d93a896eSDag-Erling Smørgrav 	}
1101d4af9e69SDag-Erling Smørgrav 	if (iflag) {
1102d4af9e69SDag-Erling Smørgrav 		printf("     Inodes        Used       Avail      "
1103d4af9e69SDag-Erling Smørgrav 		    "(root)    %%Capacity\n");
1104d93a896eSDag-Erling Smørgrav 		printf("%11llu %11llu %11llu %11llu         %s\n",
1105d4af9e69SDag-Erling Smørgrav 		    (unsigned long long)st.f_files,
1106d4af9e69SDag-Erling Smørgrav 		    (unsigned long long)(st.f_files - st.f_ffree),
1107d4af9e69SDag-Erling Smørgrav 		    (unsigned long long)st.f_favail,
1108d93a896eSDag-Erling Smørgrav 		    (unsigned long long)st.f_ffree, s_icapacity);
1109d4af9e69SDag-Erling Smørgrav 	} else if (hflag) {
1110d4af9e69SDag-Erling Smørgrav 		strlcpy(s_used, "error", sizeof(s_used));
1111d4af9e69SDag-Erling Smørgrav 		strlcpy(s_avail, "error", sizeof(s_avail));
1112d4af9e69SDag-Erling Smørgrav 		strlcpy(s_root, "error", sizeof(s_root));
1113d4af9e69SDag-Erling Smørgrav 		strlcpy(s_total, "error", sizeof(s_total));
1114d4af9e69SDag-Erling Smørgrav 		fmt_scaled((st.f_blocks - st.f_bfree) * st.f_frsize, s_used);
1115d4af9e69SDag-Erling Smørgrav 		fmt_scaled(st.f_bavail * st.f_frsize, s_avail);
1116d4af9e69SDag-Erling Smørgrav 		fmt_scaled(st.f_bfree * st.f_frsize, s_root);
1117d4af9e69SDag-Erling Smørgrav 		fmt_scaled(st.f_blocks * st.f_frsize, s_total);
1118d4af9e69SDag-Erling Smørgrav 		printf("    Size     Used    Avail   (root)    %%Capacity\n");
1119d93a896eSDag-Erling Smørgrav 		printf("%7sB %7sB %7sB %7sB         %s\n",
1120d93a896eSDag-Erling Smørgrav 		    s_total, s_used, s_avail, s_root, s_dcapacity);
1121d4af9e69SDag-Erling Smørgrav 	} else {
1122d4af9e69SDag-Erling Smørgrav 		printf("        Size         Used        Avail       "
1123d4af9e69SDag-Erling Smørgrav 		    "(root)    %%Capacity\n");
1124d93a896eSDag-Erling Smørgrav 		printf("%12llu %12llu %12llu %12llu         %s\n",
1125d4af9e69SDag-Erling Smørgrav 		    (unsigned long long)(st.f_frsize * st.f_blocks / 1024),
1126d4af9e69SDag-Erling Smørgrav 		    (unsigned long long)(st.f_frsize *
1127d4af9e69SDag-Erling Smørgrav 		    (st.f_blocks - st.f_bfree) / 1024),
1128d4af9e69SDag-Erling Smørgrav 		    (unsigned long long)(st.f_frsize * st.f_bavail / 1024),
1129d4af9e69SDag-Erling Smørgrav 		    (unsigned long long)(st.f_frsize * st.f_bfree / 1024),
1130d93a896eSDag-Erling Smørgrav 		    s_dcapacity);
1131d4af9e69SDag-Erling Smørgrav 	}
1132d4af9e69SDag-Erling Smørgrav 	return 0;
1133d4af9e69SDag-Erling Smørgrav }
1134d4af9e69SDag-Erling Smørgrav 
1135d4af9e69SDag-Erling Smørgrav /*
1136d4af9e69SDag-Erling Smørgrav  * Undo escaping of glob sequences in place. Used to undo extra escaping
1137d4af9e69SDag-Erling Smørgrav  * applied in makeargv() when the string is destined for a function that
1138d4af9e69SDag-Erling Smørgrav  * does not glob it.
1139d4af9e69SDag-Erling Smørgrav  */
1140d4af9e69SDag-Erling Smørgrav static void
1141d4af9e69SDag-Erling Smørgrav undo_glob_escape(char *s)
1142d4af9e69SDag-Erling Smørgrav {
1143d4af9e69SDag-Erling Smørgrav 	size_t i, j;
1144d4af9e69SDag-Erling Smørgrav 
1145d4af9e69SDag-Erling Smørgrav 	for (i = j = 0;;) {
1146d4af9e69SDag-Erling Smørgrav 		if (s[i] == '\0') {
1147d4af9e69SDag-Erling Smørgrav 			s[j] = '\0';
1148d4af9e69SDag-Erling Smørgrav 			return;
1149d4af9e69SDag-Erling Smørgrav 		}
1150d4af9e69SDag-Erling Smørgrav 		if (s[i] != '\\') {
1151d4af9e69SDag-Erling Smørgrav 			s[j++] = s[i++];
1152d4af9e69SDag-Erling Smørgrav 			continue;
1153d4af9e69SDag-Erling Smørgrav 		}
1154d4af9e69SDag-Erling Smørgrav 		/* s[i] == '\\' */
1155d4af9e69SDag-Erling Smørgrav 		++i;
1156d4af9e69SDag-Erling Smørgrav 		switch (s[i]) {
1157d4af9e69SDag-Erling Smørgrav 		case '?':
1158d4af9e69SDag-Erling Smørgrav 		case '[':
1159d4af9e69SDag-Erling Smørgrav 		case '*':
1160d4af9e69SDag-Erling Smørgrav 		case '\\':
1161d4af9e69SDag-Erling Smørgrav 			s[j++] = s[i++];
1162d4af9e69SDag-Erling Smørgrav 			break;
1163d4af9e69SDag-Erling Smørgrav 		case '\0':
1164d4af9e69SDag-Erling Smørgrav 			s[j++] = '\\';
1165d4af9e69SDag-Erling Smørgrav 			s[j] = '\0';
1166d4af9e69SDag-Erling Smørgrav 			return;
1167d4af9e69SDag-Erling Smørgrav 		default:
1168d4af9e69SDag-Erling Smørgrav 			s[j++] = '\\';
1169d4af9e69SDag-Erling Smørgrav 			s[j++] = s[i++];
1170d4af9e69SDag-Erling Smørgrav 			break;
1171d4af9e69SDag-Erling Smørgrav 		}
1172d4af9e69SDag-Erling Smørgrav 	}
1173d4af9e69SDag-Erling Smørgrav }
1174d4af9e69SDag-Erling Smørgrav 
1175d4af9e69SDag-Erling Smørgrav /*
1176d4af9e69SDag-Erling Smørgrav  * Split a string into an argument vector using sh(1)-style quoting,
1177d4af9e69SDag-Erling Smørgrav  * comment and escaping rules, but with some tweaks to handle glob(3)
1178d4af9e69SDag-Erling Smørgrav  * wildcards.
1179b15c8340SDag-Erling Smørgrav  * The "sloppy" flag allows for recovery from missing terminating quote, for
1180b15c8340SDag-Erling Smørgrav  * use in parsing incomplete commandlines during tab autocompletion.
1181b15c8340SDag-Erling Smørgrav  *
1182d4af9e69SDag-Erling Smørgrav  * Returns NULL on error or a NULL-terminated array of arguments.
1183b15c8340SDag-Erling Smørgrav  *
1184b15c8340SDag-Erling Smørgrav  * If "lastquote" is not NULL, the quoting character used for the last
1185b15c8340SDag-Erling Smørgrav  * argument is placed in *lastquote ("\0", "'" or "\"").
1186b15c8340SDag-Erling Smørgrav  *
1187b15c8340SDag-Erling Smørgrav  * If "terminated" is not NULL, *terminated will be set to 1 when the
1188b15c8340SDag-Erling Smørgrav  * last argument's quote has been properly terminated or 0 otherwise.
1189b15c8340SDag-Erling Smørgrav  * This parameter is only of use if "sloppy" is set.
1190d4af9e69SDag-Erling Smørgrav  */
1191d4af9e69SDag-Erling Smørgrav #define MAXARGS		128
1192d4af9e69SDag-Erling Smørgrav #define MAXARGLEN	8192
1193d4af9e69SDag-Erling Smørgrav static char **
1194b15c8340SDag-Erling Smørgrav makeargv(const char *arg, int *argcp, int sloppy, char *lastquote,
1195b15c8340SDag-Erling Smørgrav     u_int *terminated)
1196d4af9e69SDag-Erling Smørgrav {
1197d4af9e69SDag-Erling Smørgrav 	int argc, quot;
1198d4af9e69SDag-Erling Smørgrav 	size_t i, j;
1199d4af9e69SDag-Erling Smørgrav 	static char argvs[MAXARGLEN];
1200d4af9e69SDag-Erling Smørgrav 	static char *argv[MAXARGS + 1];
1201d4af9e69SDag-Erling Smørgrav 	enum { MA_START, MA_SQUOTE, MA_DQUOTE, MA_UNQUOTED } state, q;
1202d4af9e69SDag-Erling Smørgrav 
1203d4af9e69SDag-Erling Smørgrav 	*argcp = argc = 0;
1204d4af9e69SDag-Erling Smørgrav 	if (strlen(arg) > sizeof(argvs) - 1) {
1205d4af9e69SDag-Erling Smørgrav  args_too_longs:
1206d4af9e69SDag-Erling Smørgrav 		error("string too long");
1207d4af9e69SDag-Erling Smørgrav 		return NULL;
1208d4af9e69SDag-Erling Smørgrav 	}
1209b15c8340SDag-Erling Smørgrav 	if (terminated != NULL)
1210b15c8340SDag-Erling Smørgrav 		*terminated = 1;
1211b15c8340SDag-Erling Smørgrav 	if (lastquote != NULL)
1212b15c8340SDag-Erling Smørgrav 		*lastquote = '\0';
1213d4af9e69SDag-Erling Smørgrav 	state = MA_START;
1214d4af9e69SDag-Erling Smørgrav 	i = j = 0;
1215d4af9e69SDag-Erling Smørgrav 	for (;;) {
12166888a9beSDag-Erling Smørgrav 		if ((size_t)argc >= sizeof(argv) / sizeof(*argv)){
12176888a9beSDag-Erling Smørgrav 			error("Too many arguments.");
12186888a9beSDag-Erling Smørgrav 			return NULL;
12196888a9beSDag-Erling Smørgrav 		}
1220f7167e0eSDag-Erling Smørgrav 		if (isspace((unsigned char)arg[i])) {
1221d4af9e69SDag-Erling Smørgrav 			if (state == MA_UNQUOTED) {
1222d4af9e69SDag-Erling Smørgrav 				/* Terminate current argument */
1223d4af9e69SDag-Erling Smørgrav 				argvs[j++] = '\0';
1224d4af9e69SDag-Erling Smørgrav 				argc++;
1225d4af9e69SDag-Erling Smørgrav 				state = MA_START;
1226d4af9e69SDag-Erling Smørgrav 			} else if (state != MA_START)
1227d4af9e69SDag-Erling Smørgrav 				argvs[j++] = arg[i];
1228d4af9e69SDag-Erling Smørgrav 		} else if (arg[i] == '"' || arg[i] == '\'') {
1229d4af9e69SDag-Erling Smørgrav 			q = arg[i] == '"' ? MA_DQUOTE : MA_SQUOTE;
1230d4af9e69SDag-Erling Smørgrav 			if (state == MA_START) {
1231d4af9e69SDag-Erling Smørgrav 				argv[argc] = argvs + j;
1232d4af9e69SDag-Erling Smørgrav 				state = q;
1233b15c8340SDag-Erling Smørgrav 				if (lastquote != NULL)
1234b15c8340SDag-Erling Smørgrav 					*lastquote = arg[i];
1235d4af9e69SDag-Erling Smørgrav 			} else if (state == MA_UNQUOTED)
1236d4af9e69SDag-Erling Smørgrav 				state = q;
1237d4af9e69SDag-Erling Smørgrav 			else if (state == q)
1238d4af9e69SDag-Erling Smørgrav 				state = MA_UNQUOTED;
1239d4af9e69SDag-Erling Smørgrav 			else
1240d4af9e69SDag-Erling Smørgrav 				argvs[j++] = arg[i];
1241d4af9e69SDag-Erling Smørgrav 		} else if (arg[i] == '\\') {
1242d4af9e69SDag-Erling Smørgrav 			if (state == MA_SQUOTE || state == MA_DQUOTE) {
1243d4af9e69SDag-Erling Smørgrav 				quot = state == MA_SQUOTE ? '\'' : '"';
1244d4af9e69SDag-Erling Smørgrav 				/* Unescape quote we are in */
1245d4af9e69SDag-Erling Smørgrav 				/* XXX support \n and friends? */
1246d4af9e69SDag-Erling Smørgrav 				if (arg[i + 1] == quot) {
1247d4af9e69SDag-Erling Smørgrav 					i++;
1248d4af9e69SDag-Erling Smørgrav 					argvs[j++] = arg[i];
1249d4af9e69SDag-Erling Smørgrav 				} else if (arg[i + 1] == '?' ||
1250d4af9e69SDag-Erling Smørgrav 				    arg[i + 1] == '[' || arg[i + 1] == '*') {
1251d4af9e69SDag-Erling Smørgrav 					/*
1252d4af9e69SDag-Erling Smørgrav 					 * Special case for sftp: append
1253d4af9e69SDag-Erling Smørgrav 					 * double-escaped glob sequence -
1254d4af9e69SDag-Erling Smørgrav 					 * glob will undo one level of
1255d4af9e69SDag-Erling Smørgrav 					 * escaping. NB. string can grow here.
1256d4af9e69SDag-Erling Smørgrav 					 */
1257d4af9e69SDag-Erling Smørgrav 					if (j >= sizeof(argvs) - 5)
1258d4af9e69SDag-Erling Smørgrav 						goto args_too_longs;
1259d4af9e69SDag-Erling Smørgrav 					argvs[j++] = '\\';
1260d4af9e69SDag-Erling Smørgrav 					argvs[j++] = arg[i++];
1261d4af9e69SDag-Erling Smørgrav 					argvs[j++] = '\\';
1262d4af9e69SDag-Erling Smørgrav 					argvs[j++] = arg[i];
1263d4af9e69SDag-Erling Smørgrav 				} else {
1264d4af9e69SDag-Erling Smørgrav 					argvs[j++] = arg[i++];
1265d4af9e69SDag-Erling Smørgrav 					argvs[j++] = arg[i];
1266d4af9e69SDag-Erling Smørgrav 				}
1267d4af9e69SDag-Erling Smørgrav 			} else {
1268d4af9e69SDag-Erling Smørgrav 				if (state == MA_START) {
1269d4af9e69SDag-Erling Smørgrav 					argv[argc] = argvs + j;
1270d4af9e69SDag-Erling Smørgrav 					state = MA_UNQUOTED;
1271b15c8340SDag-Erling Smørgrav 					if (lastquote != NULL)
1272b15c8340SDag-Erling Smørgrav 						*lastquote = '\0';
1273d4af9e69SDag-Erling Smørgrav 				}
1274d4af9e69SDag-Erling Smørgrav 				if (arg[i + 1] == '?' || arg[i + 1] == '[' ||
1275d4af9e69SDag-Erling Smørgrav 				    arg[i + 1] == '*' || arg[i + 1] == '\\') {
1276d4af9e69SDag-Erling Smørgrav 					/*
1277d4af9e69SDag-Erling Smørgrav 					 * Special case for sftp: append
1278d4af9e69SDag-Erling Smørgrav 					 * escaped glob sequence -
1279d4af9e69SDag-Erling Smørgrav 					 * glob will undo one level of
1280d4af9e69SDag-Erling Smørgrav 					 * escaping.
1281d4af9e69SDag-Erling Smørgrav 					 */
1282d4af9e69SDag-Erling Smørgrav 					argvs[j++] = arg[i++];
1283d4af9e69SDag-Erling Smørgrav 					argvs[j++] = arg[i];
1284d4af9e69SDag-Erling Smørgrav 				} else {
1285d4af9e69SDag-Erling Smørgrav 					/* Unescape everything */
1286d4af9e69SDag-Erling Smørgrav 					/* XXX support \n and friends? */
1287d4af9e69SDag-Erling Smørgrav 					i++;
1288d4af9e69SDag-Erling Smørgrav 					argvs[j++] = arg[i];
1289d4af9e69SDag-Erling Smørgrav 				}
1290d4af9e69SDag-Erling Smørgrav 			}
1291d4af9e69SDag-Erling Smørgrav 		} else if (arg[i] == '#') {
1292d4af9e69SDag-Erling Smørgrav 			if (state == MA_SQUOTE || state == MA_DQUOTE)
1293d4af9e69SDag-Erling Smørgrav 				argvs[j++] = arg[i];
1294d4af9e69SDag-Erling Smørgrav 			else
1295d4af9e69SDag-Erling Smørgrav 				goto string_done;
1296d4af9e69SDag-Erling Smørgrav 		} else if (arg[i] == '\0') {
1297d4af9e69SDag-Erling Smørgrav 			if (state == MA_SQUOTE || state == MA_DQUOTE) {
1298b15c8340SDag-Erling Smørgrav 				if (sloppy) {
1299b15c8340SDag-Erling Smørgrav 					state = MA_UNQUOTED;
1300b15c8340SDag-Erling Smørgrav 					if (terminated != NULL)
1301b15c8340SDag-Erling Smørgrav 						*terminated = 0;
1302b15c8340SDag-Erling Smørgrav 					goto string_done;
1303b15c8340SDag-Erling Smørgrav 				}
1304d4af9e69SDag-Erling Smørgrav 				error("Unterminated quoted argument");
1305d4af9e69SDag-Erling Smørgrav 				return NULL;
1306d4af9e69SDag-Erling Smørgrav 			}
1307d4af9e69SDag-Erling Smørgrav  string_done:
1308d4af9e69SDag-Erling Smørgrav 			if (state == MA_UNQUOTED) {
1309d4af9e69SDag-Erling Smørgrav 				argvs[j++] = '\0';
1310d4af9e69SDag-Erling Smørgrav 				argc++;
1311d4af9e69SDag-Erling Smørgrav 			}
1312d4af9e69SDag-Erling Smørgrav 			break;
1313d4af9e69SDag-Erling Smørgrav 		} else {
1314d4af9e69SDag-Erling Smørgrav 			if (state == MA_START) {
1315d4af9e69SDag-Erling Smørgrav 				argv[argc] = argvs + j;
1316d4af9e69SDag-Erling Smørgrav 				state = MA_UNQUOTED;
1317b15c8340SDag-Erling Smørgrav 				if (lastquote != NULL)
1318b15c8340SDag-Erling Smørgrav 					*lastquote = '\0';
1319d4af9e69SDag-Erling Smørgrav 			}
1320d4af9e69SDag-Erling Smørgrav 			if ((state == MA_SQUOTE || state == MA_DQUOTE) &&
1321d4af9e69SDag-Erling Smørgrav 			    (arg[i] == '?' || arg[i] == '[' || arg[i] == '*')) {
1322d4af9e69SDag-Erling Smørgrav 				/*
1323d4af9e69SDag-Erling Smørgrav 				 * Special case for sftp: escape quoted
1324d4af9e69SDag-Erling Smørgrav 				 * glob(3) wildcards. NB. string can grow
1325d4af9e69SDag-Erling Smørgrav 				 * here.
1326d4af9e69SDag-Erling Smørgrav 				 */
1327d4af9e69SDag-Erling Smørgrav 				if (j >= sizeof(argvs) - 3)
1328d4af9e69SDag-Erling Smørgrav 					goto args_too_longs;
1329d4af9e69SDag-Erling Smørgrav 				argvs[j++] = '\\';
1330d4af9e69SDag-Erling Smørgrav 				argvs[j++] = arg[i];
1331d4af9e69SDag-Erling Smørgrav 			} else
1332d4af9e69SDag-Erling Smørgrav 				argvs[j++] = arg[i];
1333d4af9e69SDag-Erling Smørgrav 		}
1334d4af9e69SDag-Erling Smørgrav 		i++;
1335d4af9e69SDag-Erling Smørgrav 	}
1336d4af9e69SDag-Erling Smørgrav 	*argcp = argc;
1337d4af9e69SDag-Erling Smørgrav 	return argv;
1338d4af9e69SDag-Erling Smørgrav }
1339d4af9e69SDag-Erling Smørgrav 
1340d4af9e69SDag-Erling Smørgrav static int
134119261079SEd Maste parse_args(const char **cpp, int *ignore_errors, int *disable_echo, int *aflag,
1342a0ee8cc6SDag-Erling Smørgrav 	  int *fflag, int *hflag, int *iflag, int *lflag, int *pflag,
1343a0ee8cc6SDag-Erling Smørgrav 	  int *rflag, int *sflag,
1344f7167e0eSDag-Erling Smørgrav     unsigned long *n_arg, char **path1, char **path2)
1345efcad6b7SDag-Erling Smørgrav {
1346efcad6b7SDag-Erling Smørgrav 	const char *cmd, *cp = *cpp;
1347d4af9e69SDag-Erling Smørgrav 	char *cp2, **argv;
1348efcad6b7SDag-Erling Smørgrav 	int base = 0;
134919261079SEd Maste 	long long ll;
135047dd1d1bSDag-Erling Smørgrav 	int path1_mandatory = 0, i, cmdnum, optidx, argc;
1351efcad6b7SDag-Erling Smørgrav 
1352efcad6b7SDag-Erling Smørgrav 	/* Skip leading whitespace */
1353efcad6b7SDag-Erling Smørgrav 	cp = cp + strspn(cp, WHITESPACE);
1354efcad6b7SDag-Erling Smørgrav 
135519261079SEd Maste 	/*
135619261079SEd Maste 	 * Check for leading '-' (disable error processing) and '@' (suppress
135719261079SEd Maste 	 * command echo)
135819261079SEd Maste 	 */
1359f7167e0eSDag-Erling Smørgrav 	*ignore_errors = 0;
136019261079SEd Maste 	*disable_echo = 0;
136119261079SEd Maste 	for (;*cp != '\0'; cp++) {
1362efcad6b7SDag-Erling Smørgrav 		if (*cp == '-') {
1363f7167e0eSDag-Erling Smørgrav 			*ignore_errors = 1;
136419261079SEd Maste 		} else if (*cp == '@') {
136519261079SEd Maste 			*disable_echo = 1;
136619261079SEd Maste 		} else {
136719261079SEd Maste 			/* all other characters terminate prefix processing */
136819261079SEd Maste 			break;
1369efcad6b7SDag-Erling Smørgrav 		}
137019261079SEd Maste 	}
137119261079SEd Maste 	cp = cp + strspn(cp, WHITESPACE);
1372efcad6b7SDag-Erling Smørgrav 
1373b15c8340SDag-Erling Smørgrav 	/* Ignore blank lines and lines which begin with comment '#' char */
1374b15c8340SDag-Erling Smørgrav 	if (*cp == '\0' || *cp == '#')
1375b15c8340SDag-Erling Smørgrav 		return (0);
1376b15c8340SDag-Erling Smørgrav 
1377b15c8340SDag-Erling Smørgrav 	if ((argv = makeargv(cp, &argc, 0, NULL, NULL)) == NULL)
1378d4af9e69SDag-Erling Smørgrav 		return -1;
1379efcad6b7SDag-Erling Smørgrav 
1380d4af9e69SDag-Erling Smørgrav 	/* Figure out which command we have */
1381d4af9e69SDag-Erling Smørgrav 	for (i = 0; cmds[i].c != NULL; i++) {
13826888a9beSDag-Erling Smørgrav 		if (argv[0] != NULL && strcasecmp(cmds[i].c, argv[0]) == 0)
1383efcad6b7SDag-Erling Smørgrav 			break;
1384efcad6b7SDag-Erling Smørgrav 	}
1385efcad6b7SDag-Erling Smørgrav 	cmdnum = cmds[i].n;
1386efcad6b7SDag-Erling Smørgrav 	cmd = cmds[i].c;
1387efcad6b7SDag-Erling Smørgrav 
1388efcad6b7SDag-Erling Smørgrav 	/* Special case */
1389efcad6b7SDag-Erling Smørgrav 	if (*cp == '!') {
1390efcad6b7SDag-Erling Smørgrav 		cp++;
1391efcad6b7SDag-Erling Smørgrav 		cmdnum = I_SHELL;
1392efcad6b7SDag-Erling Smørgrav 	} else if (cmdnum == -1) {
1393efcad6b7SDag-Erling Smørgrav 		error("Invalid command.");
1394d4af9e69SDag-Erling Smørgrav 		return -1;
1395efcad6b7SDag-Erling Smørgrav 	}
1396efcad6b7SDag-Erling Smørgrav 
1397efcad6b7SDag-Erling Smørgrav 	/* Get arguments and parse flags */
1398f7167e0eSDag-Erling Smørgrav 	*aflag = *fflag = *hflag = *iflag = *lflag = *pflag = 0;
1399f7167e0eSDag-Erling Smørgrav 	*rflag = *sflag = 0;
1400efcad6b7SDag-Erling Smørgrav 	*path1 = *path2 = NULL;
1401d4af9e69SDag-Erling Smørgrav 	optidx = 1;
1402efcad6b7SDag-Erling Smørgrav 	switch (cmdnum) {
1403efcad6b7SDag-Erling Smørgrav 	case I_GET:
1404e4a9863fSDag-Erling Smørgrav 	case I_REGET:
1405a0ee8cc6SDag-Erling Smørgrav 	case I_REPUT:
1406efcad6b7SDag-Erling Smørgrav 	case I_PUT:
14074a421b63SDag-Erling Smørgrav 		if ((optidx = parse_getput_flags(cmd, argv, argc,
1408f7167e0eSDag-Erling Smørgrav 		    aflag, fflag, pflag, rflag)) == -1)
1409d4af9e69SDag-Erling Smørgrav 			return -1;
1410efcad6b7SDag-Erling Smørgrav 		/* Get first pathname (mandatory) */
1411d4af9e69SDag-Erling Smørgrav 		if (argc - optidx < 1) {
1412efcad6b7SDag-Erling Smørgrav 			error("You must specify at least one path after a "
1413efcad6b7SDag-Erling Smørgrav 			    "%s command.", cmd);
1414d4af9e69SDag-Erling Smørgrav 			return -1;
1415efcad6b7SDag-Erling Smørgrav 		}
1416d4af9e69SDag-Erling Smørgrav 		*path1 = xstrdup(argv[optidx]);
1417d4af9e69SDag-Erling Smørgrav 		/* Get second pathname (optional) */
1418d4af9e69SDag-Erling Smørgrav 		if (argc - optidx > 1) {
1419d4af9e69SDag-Erling Smørgrav 			*path2 = xstrdup(argv[optidx + 1]);
1420d4af9e69SDag-Erling Smørgrav 			/* Destination is not globbed */
1421d4af9e69SDag-Erling Smørgrav 			undo_glob_escape(*path2);
1422d4af9e69SDag-Erling Smørgrav 		}
1423efcad6b7SDag-Erling Smørgrav 		break;
14244a421b63SDag-Erling Smørgrav 	case I_LINK:
14254a421b63SDag-Erling Smørgrav 		if ((optidx = parse_link_flags(cmd, argv, argc, sflag)) == -1)
14264a421b63SDag-Erling Smørgrav 			return -1;
1427f7167e0eSDag-Erling Smørgrav 		goto parse_two_paths;
142887c1498dSEd Maste 	case I_COPY:
142987c1498dSEd Maste 		if ((optidx = parse_no_flags(cmd, argv, argc)) == -1)
143087c1498dSEd Maste 			return -1;
143187c1498dSEd Maste 		goto parse_two_paths;
14324a421b63SDag-Erling Smørgrav 	case I_RENAME:
1433f7167e0eSDag-Erling Smørgrav 		if ((optidx = parse_rename_flags(cmd, argv, argc, lflag)) == -1)
1434f7167e0eSDag-Erling Smørgrav 			return -1;
1435f7167e0eSDag-Erling Smørgrav 		goto parse_two_paths;
1436f7167e0eSDag-Erling Smørgrav 	case I_SYMLINK:
1437f7167e0eSDag-Erling Smørgrav 		if ((optidx = parse_no_flags(cmd, argv, argc)) == -1)
1438f7167e0eSDag-Erling Smørgrav 			return -1;
1439f7167e0eSDag-Erling Smørgrav  parse_two_paths:
1440d4af9e69SDag-Erling Smørgrav 		if (argc - optidx < 2) {
1441efcad6b7SDag-Erling Smørgrav 			error("You must specify two paths after a %s "
1442efcad6b7SDag-Erling Smørgrav 			    "command.", cmd);
1443d4af9e69SDag-Erling Smørgrav 			return -1;
1444efcad6b7SDag-Erling Smørgrav 		}
1445d4af9e69SDag-Erling Smørgrav 		*path1 = xstrdup(argv[optidx]);
1446d4af9e69SDag-Erling Smørgrav 		*path2 = xstrdup(argv[optidx + 1]);
1447d4af9e69SDag-Erling Smørgrav 		/* Paths are not globbed */
1448d4af9e69SDag-Erling Smørgrav 		undo_glob_escape(*path1);
1449d4af9e69SDag-Erling Smørgrav 		undo_glob_escape(*path2);
1450efcad6b7SDag-Erling Smørgrav 		break;
1451efcad6b7SDag-Erling Smørgrav 	case I_RM:
1452efcad6b7SDag-Erling Smørgrav 	case I_MKDIR:
1453efcad6b7SDag-Erling Smørgrav 	case I_RMDIR:
145447dd1d1bSDag-Erling Smørgrav 	case I_LMKDIR:
145547dd1d1bSDag-Erling Smørgrav 		path1_mandatory = 1;
145647dd1d1bSDag-Erling Smørgrav 		/* FALLTHROUGH */
1457efcad6b7SDag-Erling Smørgrav 	case I_CHDIR:
1458efcad6b7SDag-Erling Smørgrav 	case I_LCHDIR:
1459f7167e0eSDag-Erling Smørgrav 		if ((optidx = parse_no_flags(cmd, argv, argc)) == -1)
1460f7167e0eSDag-Erling Smørgrav 			return -1;
1461efcad6b7SDag-Erling Smørgrav 		/* Get pathname (mandatory) */
1462d4af9e69SDag-Erling Smørgrav 		if (argc - optidx < 1) {
146347dd1d1bSDag-Erling Smørgrav 			if (!path1_mandatory)
146447dd1d1bSDag-Erling Smørgrav 				break; /* return a NULL path1 */
1465efcad6b7SDag-Erling Smørgrav 			error("You must specify a path after a %s command.",
1466efcad6b7SDag-Erling Smørgrav 			    cmd);
1467d4af9e69SDag-Erling Smørgrav 			return -1;
1468d4af9e69SDag-Erling Smørgrav 		}
1469d4af9e69SDag-Erling Smørgrav 		*path1 = xstrdup(argv[optidx]);
1470d4af9e69SDag-Erling Smørgrav 		/* Only "rm" globs */
1471d4af9e69SDag-Erling Smørgrav 		if (cmdnum != I_RM)
1472d4af9e69SDag-Erling Smørgrav 			undo_glob_escape(*path1);
1473d4af9e69SDag-Erling Smørgrav 		break;
1474d4af9e69SDag-Erling Smørgrav 	case I_DF:
1475d4af9e69SDag-Erling Smørgrav 		if ((optidx = parse_df_flags(cmd, argv, argc, hflag,
1476d4af9e69SDag-Erling Smørgrav 		    iflag)) == -1)
1477d4af9e69SDag-Erling Smørgrav 			return -1;
1478d4af9e69SDag-Erling Smørgrav 		/* Default to current directory if no path specified */
1479d4af9e69SDag-Erling Smørgrav 		if (argc - optidx < 1)
1480d4af9e69SDag-Erling Smørgrav 			*path1 = NULL;
1481d4af9e69SDag-Erling Smørgrav 		else {
1482d4af9e69SDag-Erling Smørgrav 			*path1 = xstrdup(argv[optidx]);
1483d4af9e69SDag-Erling Smørgrav 			undo_glob_escape(*path1);
1484efcad6b7SDag-Erling Smørgrav 		}
1485efcad6b7SDag-Erling Smørgrav 		break;
1486efcad6b7SDag-Erling Smørgrav 	case I_LS:
1487d4af9e69SDag-Erling Smørgrav 		if ((optidx = parse_ls_flags(argv, argc, lflag)) == -1)
1488efcad6b7SDag-Erling Smørgrav 			return(-1);
1489efcad6b7SDag-Erling Smørgrav 		/* Path is optional */
1490d4af9e69SDag-Erling Smørgrav 		if (argc - optidx > 0)
1491d4af9e69SDag-Erling Smørgrav 			*path1 = xstrdup(argv[optidx]);
1492efcad6b7SDag-Erling Smørgrav 		break;
1493efcad6b7SDag-Erling Smørgrav 	case I_LLS:
1494d4af9e69SDag-Erling Smørgrav 		/* Skip ls command and following whitespace */
1495d4af9e69SDag-Erling Smørgrav 		cp = cp + strlen(cmd) + strspn(cp, WHITESPACE);
1496efcad6b7SDag-Erling Smørgrav 	case I_SHELL:
1497efcad6b7SDag-Erling Smørgrav 		/* Uses the rest of the line */
1498efcad6b7SDag-Erling Smørgrav 		break;
1499efcad6b7SDag-Erling Smørgrav 	case I_LUMASK:
1500efcad6b7SDag-Erling Smørgrav 	case I_CHMOD:
1501efcad6b7SDag-Erling Smørgrav 		base = 8;
15022f513db7SEd Maste 		/* FALLTHROUGH */
1503efcad6b7SDag-Erling Smørgrav 	case I_CHOWN:
1504efcad6b7SDag-Erling Smørgrav 	case I_CHGRP:
150519261079SEd Maste 		if ((optidx = parse_ch_flags(cmd, argv, argc, hflag)) == -1)
1506f7167e0eSDag-Erling Smørgrav 			return -1;
1507efcad6b7SDag-Erling Smørgrav 		/* Get numeric arg (mandatory) */
1508d4af9e69SDag-Erling Smørgrav 		if (argc - optidx < 1)
1509d4af9e69SDag-Erling Smørgrav 			goto need_num_arg;
151092eb0aa1SDag-Erling Smørgrav 		errno = 0;
151119261079SEd Maste 		ll = strtoll(argv[optidx], &cp2, base);
1512d4af9e69SDag-Erling Smørgrav 		if (cp2 == argv[optidx] || *cp2 != '\0' ||
151319261079SEd Maste 		    ((ll == LLONG_MIN || ll == LLONG_MAX) && errno == ERANGE) ||
151419261079SEd Maste 		    ll < 0 || ll > UINT32_MAX) {
1515d4af9e69SDag-Erling Smørgrav  need_num_arg:
1516efcad6b7SDag-Erling Smørgrav 			error("You must supply a numeric argument "
1517efcad6b7SDag-Erling Smørgrav 			    "to the %s command.", cmd);
1518d4af9e69SDag-Erling Smørgrav 			return -1;
1519efcad6b7SDag-Erling Smørgrav 		}
152019261079SEd Maste 		*n_arg = ll;
1521d4af9e69SDag-Erling Smørgrav 		if (cmdnum == I_LUMASK)
1522efcad6b7SDag-Erling Smørgrav 			break;
1523efcad6b7SDag-Erling Smørgrav 		/* Get pathname (mandatory) */
1524d4af9e69SDag-Erling Smørgrav 		if (argc - optidx < 2) {
1525efcad6b7SDag-Erling Smørgrav 			error("You must specify a path after a %s command.",
1526efcad6b7SDag-Erling Smørgrav 			    cmd);
1527d4af9e69SDag-Erling Smørgrav 			return -1;
1528efcad6b7SDag-Erling Smørgrav 		}
1529d4af9e69SDag-Erling Smørgrav 		*path1 = xstrdup(argv[optidx + 1]);
1530efcad6b7SDag-Erling Smørgrav 		break;
1531efcad6b7SDag-Erling Smørgrav 	case I_QUIT:
1532efcad6b7SDag-Erling Smørgrav 	case I_PWD:
1533efcad6b7SDag-Erling Smørgrav 	case I_LPWD:
1534efcad6b7SDag-Erling Smørgrav 	case I_HELP:
1535efcad6b7SDag-Erling Smørgrav 	case I_VERSION:
1536efcad6b7SDag-Erling Smørgrav 	case I_PROGRESS:
1537f7167e0eSDag-Erling Smørgrav 		if ((optidx = parse_no_flags(cmd, argv, argc)) == -1)
1538f7167e0eSDag-Erling Smørgrav 			return -1;
1539efcad6b7SDag-Erling Smørgrav 		break;
1540efcad6b7SDag-Erling Smørgrav 	default:
1541efcad6b7SDag-Erling Smørgrav 		fatal("Command not implemented");
1542efcad6b7SDag-Erling Smørgrav 	}
1543efcad6b7SDag-Erling Smørgrav 
1544efcad6b7SDag-Erling Smørgrav 	*cpp = cp;
1545efcad6b7SDag-Erling Smørgrav 	return(cmdnum);
1546efcad6b7SDag-Erling Smørgrav }
1547efcad6b7SDag-Erling Smørgrav 
1548efcad6b7SDag-Erling Smørgrav static int
1549efcad6b7SDag-Erling Smørgrav parse_dispatch_command(struct sftp_conn *conn, const char *cmd, char **pwd,
155019261079SEd Maste     const char *startdir, int err_abort, int echo_command)
1551efcad6b7SDag-Erling Smørgrav {
155219261079SEd Maste 	const char *ocmd = cmd;
1553efcad6b7SDag-Erling Smørgrav 	char *path1, *path2, *tmp;
155419261079SEd Maste 	int ignore_errors = 0, disable_echo = 1;
155519261079SEd Maste 	int aflag = 0, fflag = 0, hflag = 0, iflag = 0;
1556f7167e0eSDag-Erling Smørgrav 	int lflag = 0, pflag = 0, rflag = 0, sflag = 0;
15574a421b63SDag-Erling Smørgrav 	int cmdnum, i;
1558cce7d346SDag-Erling Smørgrav 	unsigned long n_arg = 0;
1559*edf85781SEd Maste 	Attrib a, aa;
1560bc5531deSDag-Erling Smørgrav 	char path_buf[PATH_MAX];
1561efcad6b7SDag-Erling Smørgrav 	int err = 0;
1562efcad6b7SDag-Erling Smørgrav 	glob_t g;
1563efcad6b7SDag-Erling Smørgrav 
1564efcad6b7SDag-Erling Smørgrav 	path1 = path2 = NULL;
156519261079SEd Maste 	cmdnum = parse_args(&cmd, &ignore_errors, &disable_echo, &aflag, &fflag,
156619261079SEd Maste 	    &hflag, &iflag, &lflag, &pflag, &rflag, &sflag, &n_arg,
156719261079SEd Maste 	    &path1, &path2);
1568f7167e0eSDag-Erling Smørgrav 	if (ignore_errors != 0)
1569efcad6b7SDag-Erling Smørgrav 		err_abort = 0;
1570efcad6b7SDag-Erling Smørgrav 
157119261079SEd Maste 	if (echo_command && !disable_echo)
157219261079SEd Maste 		mprintf("sftp> %s\n", ocmd);
157319261079SEd Maste 
1574efcad6b7SDag-Erling Smørgrav 	memset(&g, 0, sizeof(g));
1575efcad6b7SDag-Erling Smørgrav 
1576efcad6b7SDag-Erling Smørgrav 	/* Perform command */
1577efcad6b7SDag-Erling Smørgrav 	switch (cmdnum) {
1578efcad6b7SDag-Erling Smørgrav 	case 0:
1579efcad6b7SDag-Erling Smørgrav 		/* Blank line */
1580efcad6b7SDag-Erling Smørgrav 		break;
1581efcad6b7SDag-Erling Smørgrav 	case -1:
1582efcad6b7SDag-Erling Smørgrav 		/* Unrecognized command */
1583efcad6b7SDag-Erling Smørgrav 		err = -1;
1584efcad6b7SDag-Erling Smørgrav 		break;
1585e4a9863fSDag-Erling Smørgrav 	case I_REGET:
1586e4a9863fSDag-Erling Smørgrav 		aflag = 1;
1587e4a9863fSDag-Erling Smørgrav 		/* FALLTHROUGH */
1588efcad6b7SDag-Erling Smørgrav 	case I_GET:
1589e4a9863fSDag-Erling Smørgrav 		err = process_get(conn, path1, path2, *pwd, pflag,
1590f7167e0eSDag-Erling Smørgrav 		    rflag, aflag, fflag);
1591efcad6b7SDag-Erling Smørgrav 		break;
1592a0ee8cc6SDag-Erling Smørgrav 	case I_REPUT:
1593a0ee8cc6SDag-Erling Smørgrav 		aflag = 1;
1594a0ee8cc6SDag-Erling Smørgrav 		/* FALLTHROUGH */
1595efcad6b7SDag-Erling Smørgrav 	case I_PUT:
1596f7167e0eSDag-Erling Smørgrav 		err = process_put(conn, path1, path2, *pwd, pflag,
1597a0ee8cc6SDag-Erling Smørgrav 		    rflag, aflag, fflag);
1598efcad6b7SDag-Erling Smørgrav 		break;
159987c1498dSEd Maste 	case I_COPY:
1600*edf85781SEd Maste 		path1 = sftp_make_absolute(path1, *pwd);
1601*edf85781SEd Maste 		path2 = sftp_make_absolute(path2, *pwd);
1602*edf85781SEd Maste 		err = sftp_copy(conn, path1, path2);
160387c1498dSEd Maste 		break;
1604efcad6b7SDag-Erling Smørgrav 	case I_RENAME:
1605*edf85781SEd Maste 		path1 = sftp_make_absolute(path1, *pwd);
1606*edf85781SEd Maste 		path2 = sftp_make_absolute(path2, *pwd);
1607*edf85781SEd Maste 		err = sftp_rename(conn, path1, path2, lflag);
1608efcad6b7SDag-Erling Smørgrav 		break;
1609efcad6b7SDag-Erling Smørgrav 	case I_SYMLINK:
16104a421b63SDag-Erling Smørgrav 		sflag = 1;
16112f513db7SEd Maste 		/* FALLTHROUGH */
16124a421b63SDag-Erling Smørgrav 	case I_LINK:
1613f7167e0eSDag-Erling Smørgrav 		if (!sflag)
1614*edf85781SEd Maste 			path1 = sftp_make_absolute(path1, *pwd);
1615*edf85781SEd Maste 		path2 = sftp_make_absolute(path2, *pwd);
1616*edf85781SEd Maste 		err = (sflag ? sftp_symlink : sftp_hardlink)(conn,
1617*edf85781SEd Maste 		    path1, path2);
1618efcad6b7SDag-Erling Smørgrav 		break;
1619efcad6b7SDag-Erling Smørgrav 	case I_RM:
162038a52bd3SEd Maste 		path1 = make_absolute_pwd_glob(path1, *pwd);
1621*edf85781SEd Maste 		sftp_glob(conn, path1, GLOB_NOCHECK, NULL, &g);
1622d74d50a8SDag-Erling Smørgrav 		for (i = 0; g.gl_pathv[i] && !interrupted; i++) {
1623e4a9863fSDag-Erling Smørgrav 			if (!quiet)
1624076ad2f8SDag-Erling Smørgrav 				mprintf("Removing %s\n", g.gl_pathv[i]);
1625*edf85781SEd Maste 			err = sftp_rm(conn, g.gl_pathv[i]);
1626efcad6b7SDag-Erling Smørgrav 			if (err != 0 && err_abort)
1627efcad6b7SDag-Erling Smørgrav 				break;
1628efcad6b7SDag-Erling Smørgrav 		}
1629efcad6b7SDag-Erling Smørgrav 		break;
1630efcad6b7SDag-Erling Smørgrav 	case I_MKDIR:
1631*edf85781SEd Maste 		path1 = sftp_make_absolute(path1, *pwd);
1632efcad6b7SDag-Erling Smørgrav 		attrib_clear(&a);
1633efcad6b7SDag-Erling Smørgrav 		a.flags |= SSH2_FILEXFER_ATTR_PERMISSIONS;
1634efcad6b7SDag-Erling Smørgrav 		a.perm = 0777;
1635*edf85781SEd Maste 		err = sftp_mkdir(conn, path1, &a, 1);
1636efcad6b7SDag-Erling Smørgrav 		break;
1637efcad6b7SDag-Erling Smørgrav 	case I_RMDIR:
1638*edf85781SEd Maste 		path1 = sftp_make_absolute(path1, *pwd);
1639*edf85781SEd Maste 		err = sftp_rmdir(conn, path1);
1640efcad6b7SDag-Erling Smørgrav 		break;
1641efcad6b7SDag-Erling Smørgrav 	case I_CHDIR:
164247dd1d1bSDag-Erling Smørgrav 		if (path1 == NULL || *path1 == '\0')
164347dd1d1bSDag-Erling Smørgrav 			path1 = xstrdup(startdir);
1644*edf85781SEd Maste 		path1 = sftp_make_absolute(path1, *pwd);
1645*edf85781SEd Maste 		if ((tmp = sftp_realpath(conn, path1)) == NULL) {
1646efcad6b7SDag-Erling Smørgrav 			err = 1;
1647efcad6b7SDag-Erling Smørgrav 			break;
1648efcad6b7SDag-Erling Smørgrav 		}
1649*edf85781SEd Maste 		if (sftp_stat(conn, tmp, 0, &aa) != 0) {
1650e4a9863fSDag-Erling Smørgrav 			free(tmp);
1651efcad6b7SDag-Erling Smørgrav 			err = 1;
1652efcad6b7SDag-Erling Smørgrav 			break;
1653efcad6b7SDag-Erling Smørgrav 		}
1654*edf85781SEd Maste 		if (!(aa.flags & SSH2_FILEXFER_ATTR_PERMISSIONS)) {
1655efcad6b7SDag-Erling Smørgrav 			error("Can't change directory: Can't check target");
1656e4a9863fSDag-Erling Smørgrav 			free(tmp);
1657efcad6b7SDag-Erling Smørgrav 			err = 1;
1658efcad6b7SDag-Erling Smørgrav 			break;
1659efcad6b7SDag-Erling Smørgrav 		}
1660*edf85781SEd Maste 		if (!S_ISDIR(aa.perm)) {
1661efcad6b7SDag-Erling Smørgrav 			error("Can't change directory: \"%s\" is not "
1662efcad6b7SDag-Erling Smørgrav 			    "a directory", tmp);
1663e4a9863fSDag-Erling Smørgrav 			free(tmp);
1664efcad6b7SDag-Erling Smørgrav 			err = 1;
1665efcad6b7SDag-Erling Smørgrav 			break;
1666efcad6b7SDag-Erling Smørgrav 		}
1667e4a9863fSDag-Erling Smørgrav 		free(*pwd);
1668efcad6b7SDag-Erling Smørgrav 		*pwd = tmp;
1669efcad6b7SDag-Erling Smørgrav 		break;
1670efcad6b7SDag-Erling Smørgrav 	case I_LS:
1671efcad6b7SDag-Erling Smørgrav 		if (!path1) {
1672e2f6069cSDag-Erling Smørgrav 			do_ls_dir(conn, *pwd, *pwd, lflag);
1673efcad6b7SDag-Erling Smørgrav 			break;
1674efcad6b7SDag-Erling Smørgrav 		}
1675efcad6b7SDag-Erling Smørgrav 
1676efcad6b7SDag-Erling Smørgrav 		/* Strip pwd off beginning of non-absolute paths */
1677efcad6b7SDag-Erling Smørgrav 		tmp = NULL;
167819261079SEd Maste 		if (!path_absolute(path1))
1679efcad6b7SDag-Erling Smørgrav 			tmp = *pwd;
1680efcad6b7SDag-Erling Smørgrav 
168138a52bd3SEd Maste 		path1 = make_absolute_pwd_glob(path1, *pwd);
1682efcad6b7SDag-Erling Smørgrav 		err = do_globbed_ls(conn, path1, tmp, lflag);
1683efcad6b7SDag-Erling Smørgrav 		break;
1684d4af9e69SDag-Erling Smørgrav 	case I_DF:
1685d4af9e69SDag-Erling Smørgrav 		/* Default to current directory if no path specified */
1686d4af9e69SDag-Erling Smørgrav 		if (path1 == NULL)
1687d4af9e69SDag-Erling Smørgrav 			path1 = xstrdup(*pwd);
1688*edf85781SEd Maste 		path1 = sftp_make_absolute(path1, *pwd);
1689d4af9e69SDag-Erling Smørgrav 		err = do_df(conn, path1, hflag, iflag);
1690d4af9e69SDag-Erling Smørgrav 		break;
1691efcad6b7SDag-Erling Smørgrav 	case I_LCHDIR:
169247dd1d1bSDag-Erling Smørgrav 		if (path1 == NULL || *path1 == '\0')
169347dd1d1bSDag-Erling Smørgrav 			path1 = xstrdup("~");
1694bc5531deSDag-Erling Smørgrav 		tmp = tilde_expand_filename(path1, getuid());
1695bc5531deSDag-Erling Smørgrav 		free(path1);
1696bc5531deSDag-Erling Smørgrav 		path1 = tmp;
1697efcad6b7SDag-Erling Smørgrav 		if (chdir(path1) == -1) {
1698efcad6b7SDag-Erling Smørgrav 			error("Couldn't change local directory to "
1699efcad6b7SDag-Erling Smørgrav 			    "\"%s\": %s", path1, strerror(errno));
1700efcad6b7SDag-Erling Smørgrav 			err = 1;
1701efcad6b7SDag-Erling Smørgrav 		}
1702efcad6b7SDag-Erling Smørgrav 		break;
1703efcad6b7SDag-Erling Smørgrav 	case I_LMKDIR:
1704efcad6b7SDag-Erling Smørgrav 		if (mkdir(path1, 0777) == -1) {
1705efcad6b7SDag-Erling Smørgrav 			error("Couldn't create local directory "
1706efcad6b7SDag-Erling Smørgrav 			    "\"%s\": %s", path1, strerror(errno));
1707efcad6b7SDag-Erling Smørgrav 			err = 1;
1708efcad6b7SDag-Erling Smørgrav 		}
1709efcad6b7SDag-Erling Smørgrav 		break;
1710efcad6b7SDag-Erling Smørgrav 	case I_LLS:
1711efcad6b7SDag-Erling Smørgrav 		local_do_ls(cmd);
1712efcad6b7SDag-Erling Smørgrav 		break;
1713efcad6b7SDag-Erling Smørgrav 	case I_SHELL:
1714efcad6b7SDag-Erling Smørgrav 		local_do_shell(cmd);
1715efcad6b7SDag-Erling Smørgrav 		break;
1716efcad6b7SDag-Erling Smørgrav 	case I_LUMASK:
1717efcad6b7SDag-Erling Smørgrav 		umask(n_arg);
1718efcad6b7SDag-Erling Smørgrav 		printf("Local umask: %03lo\n", n_arg);
1719efcad6b7SDag-Erling Smørgrav 		break;
1720efcad6b7SDag-Erling Smørgrav 	case I_CHMOD:
172138a52bd3SEd Maste 		path1 = make_absolute_pwd_glob(path1, *pwd);
1722efcad6b7SDag-Erling Smørgrav 		attrib_clear(&a);
1723efcad6b7SDag-Erling Smørgrav 		a.flags |= SSH2_FILEXFER_ATTR_PERMISSIONS;
1724efcad6b7SDag-Erling Smørgrav 		a.perm = n_arg;
1725*edf85781SEd Maste 		sftp_glob(conn, path1, GLOB_NOCHECK, NULL, &g);
1726d74d50a8SDag-Erling Smørgrav 		for (i = 0; g.gl_pathv[i] && !interrupted; i++) {
1727e4a9863fSDag-Erling Smørgrav 			if (!quiet)
1728076ad2f8SDag-Erling Smørgrav 				mprintf("Changing mode on %s\n",
1729076ad2f8SDag-Erling Smørgrav 				    g.gl_pathv[i]);
1730*edf85781SEd Maste 			err = (hflag ? sftp_lsetstat : sftp_setstat)(conn,
173119261079SEd Maste 			    g.gl_pathv[i], &a);
1732efcad6b7SDag-Erling Smørgrav 			if (err != 0 && err_abort)
1733efcad6b7SDag-Erling Smørgrav 				break;
1734efcad6b7SDag-Erling Smørgrav 		}
1735efcad6b7SDag-Erling Smørgrav 		break;
1736efcad6b7SDag-Erling Smørgrav 	case I_CHOWN:
1737efcad6b7SDag-Erling Smørgrav 	case I_CHGRP:
173838a52bd3SEd Maste 		path1 = make_absolute_pwd_glob(path1, *pwd);
1739*edf85781SEd Maste 		sftp_glob(conn, path1, GLOB_NOCHECK, NULL, &g);
1740d74d50a8SDag-Erling Smørgrav 		for (i = 0; g.gl_pathv[i] && !interrupted; i++) {
1741*edf85781SEd Maste 			if ((hflag ? sftp_lstat : sftp_stat)(conn,
1742*edf85781SEd Maste 			    g.gl_pathv[i], 0, &aa) != 0) {
1743cce7d346SDag-Erling Smørgrav 				if (err_abort) {
1744cce7d346SDag-Erling Smørgrav 					err = -1;
1745efcad6b7SDag-Erling Smørgrav 					break;
1746cce7d346SDag-Erling Smørgrav 				} else
1747efcad6b7SDag-Erling Smørgrav 					continue;
1748efcad6b7SDag-Erling Smørgrav 			}
1749*edf85781SEd Maste 			if (!(aa.flags & SSH2_FILEXFER_ATTR_UIDGID)) {
1750efcad6b7SDag-Erling Smørgrav 				error("Can't get current ownership of "
1751efcad6b7SDag-Erling Smørgrav 				    "remote file \"%s\"", g.gl_pathv[i]);
1752cce7d346SDag-Erling Smørgrav 				if (err_abort) {
1753cce7d346SDag-Erling Smørgrav 					err = -1;
1754efcad6b7SDag-Erling Smørgrav 					break;
1755cce7d346SDag-Erling Smørgrav 				} else
1756efcad6b7SDag-Erling Smørgrav 					continue;
1757efcad6b7SDag-Erling Smørgrav 			}
1758*edf85781SEd Maste 			aa.flags &= SSH2_FILEXFER_ATTR_UIDGID;
1759efcad6b7SDag-Erling Smørgrav 			if (cmdnum == I_CHOWN) {
1760e4a9863fSDag-Erling Smørgrav 				if (!quiet)
1761076ad2f8SDag-Erling Smørgrav 					mprintf("Changing owner on %s\n",
1762e4a9863fSDag-Erling Smørgrav 					    g.gl_pathv[i]);
1763*edf85781SEd Maste 				aa.uid = n_arg;
1764efcad6b7SDag-Erling Smørgrav 			} else {
1765e4a9863fSDag-Erling Smørgrav 				if (!quiet)
1766076ad2f8SDag-Erling Smørgrav 					mprintf("Changing group on %s\n",
1767e4a9863fSDag-Erling Smørgrav 					    g.gl_pathv[i]);
1768*edf85781SEd Maste 				aa.gid = n_arg;
1769efcad6b7SDag-Erling Smørgrav 			}
1770*edf85781SEd Maste 			err = (hflag ? sftp_lsetstat : sftp_setstat)(conn,
1771*edf85781SEd Maste 			    g.gl_pathv[i], &aa);
1772efcad6b7SDag-Erling Smørgrav 			if (err != 0 && err_abort)
1773efcad6b7SDag-Erling Smørgrav 				break;
1774efcad6b7SDag-Erling Smørgrav 		}
1775efcad6b7SDag-Erling Smørgrav 		break;
1776efcad6b7SDag-Erling Smørgrav 	case I_PWD:
1777076ad2f8SDag-Erling Smørgrav 		mprintf("Remote working directory: %s\n", *pwd);
1778efcad6b7SDag-Erling Smørgrav 		break;
1779efcad6b7SDag-Erling Smørgrav 	case I_LPWD:
1780efcad6b7SDag-Erling Smørgrav 		if (!getcwd(path_buf, sizeof(path_buf))) {
1781efcad6b7SDag-Erling Smørgrav 			error("Couldn't get local cwd: %s", strerror(errno));
1782efcad6b7SDag-Erling Smørgrav 			err = -1;
1783efcad6b7SDag-Erling Smørgrav 			break;
1784efcad6b7SDag-Erling Smørgrav 		}
1785076ad2f8SDag-Erling Smørgrav 		mprintf("Local working directory: %s\n", path_buf);
1786efcad6b7SDag-Erling Smørgrav 		break;
1787efcad6b7SDag-Erling Smørgrav 	case I_QUIT:
1788efcad6b7SDag-Erling Smørgrav 		/* Processed below */
1789efcad6b7SDag-Erling Smørgrav 		break;
1790efcad6b7SDag-Erling Smørgrav 	case I_HELP:
1791efcad6b7SDag-Erling Smørgrav 		help();
1792efcad6b7SDag-Erling Smørgrav 		break;
1793efcad6b7SDag-Erling Smørgrav 	case I_VERSION:
1794efcad6b7SDag-Erling Smørgrav 		printf("SFTP protocol version %u\n", sftp_proto_version(conn));
1795efcad6b7SDag-Erling Smørgrav 		break;
1796efcad6b7SDag-Erling Smørgrav 	case I_PROGRESS:
1797efcad6b7SDag-Erling Smørgrav 		showprogress = !showprogress;
1798efcad6b7SDag-Erling Smørgrav 		if (showprogress)
1799efcad6b7SDag-Erling Smørgrav 			printf("Progress meter enabled\n");
1800efcad6b7SDag-Erling Smørgrav 		else
1801efcad6b7SDag-Erling Smørgrav 			printf("Progress meter disabled\n");
1802efcad6b7SDag-Erling Smørgrav 		break;
1803efcad6b7SDag-Erling Smørgrav 	default:
1804efcad6b7SDag-Erling Smørgrav 		fatal("%d is not implemented", cmdnum);
1805efcad6b7SDag-Erling Smørgrav 	}
1806efcad6b7SDag-Erling Smørgrav 
1807efcad6b7SDag-Erling Smørgrav 	if (g.gl_pathc)
1808efcad6b7SDag-Erling Smørgrav 		globfree(&g);
1809e4a9863fSDag-Erling Smørgrav 	free(path1);
1810e4a9863fSDag-Erling Smørgrav 	free(path2);
1811efcad6b7SDag-Erling Smørgrav 
1812efcad6b7SDag-Erling Smørgrav 	/* If an unignored error occurs in batch mode we should abort. */
1813efcad6b7SDag-Erling Smørgrav 	if (err_abort && err != 0)
1814efcad6b7SDag-Erling Smørgrav 		return (-1);
1815efcad6b7SDag-Erling Smørgrav 	else if (cmdnum == I_QUIT)
1816efcad6b7SDag-Erling Smørgrav 		return (1);
1817efcad6b7SDag-Erling Smørgrav 
1818efcad6b7SDag-Erling Smørgrav 	return (0);
1819efcad6b7SDag-Erling Smørgrav }
1820efcad6b7SDag-Erling Smørgrav 
18215e8dbd04SDag-Erling Smørgrav #ifdef USE_LIBEDIT
18225e8dbd04SDag-Erling Smørgrav static char *
18235e8dbd04SDag-Erling Smørgrav prompt(EditLine *el)
18245e8dbd04SDag-Erling Smørgrav {
18255e8dbd04SDag-Erling Smørgrav 	return ("sftp> ");
18265e8dbd04SDag-Erling Smørgrav }
1827b15c8340SDag-Erling Smørgrav 
1828b15c8340SDag-Erling Smørgrav /* Display entries in 'list' after skipping the first 'len' chars */
1829b15c8340SDag-Erling Smørgrav static void
1830b15c8340SDag-Erling Smørgrav complete_display(char **list, u_int len)
1831b15c8340SDag-Erling Smørgrav {
1832b15c8340SDag-Erling Smørgrav 	u_int y, m = 0, width = 80, columns = 1, colspace = 0, llen;
1833b15c8340SDag-Erling Smørgrav 	struct winsize ws;
1834b15c8340SDag-Erling Smørgrav 	char *tmp;
1835b15c8340SDag-Erling Smørgrav 
1836b15c8340SDag-Erling Smørgrav 	/* Count entries for sort and find longest */
1837b15c8340SDag-Erling Smørgrav 	for (y = 0; list[y]; y++)
1838ca86bcf2SDag-Erling Smørgrav 		m = MAXIMUM(m, strlen(list[y]));
1839b15c8340SDag-Erling Smørgrav 
1840b15c8340SDag-Erling Smørgrav 	if (ioctl(fileno(stdin), TIOCGWINSZ, &ws) != -1)
1841b15c8340SDag-Erling Smørgrav 		width = ws.ws_col;
1842b15c8340SDag-Erling Smørgrav 
1843b15c8340SDag-Erling Smørgrav 	m = m > len ? m - len : 0;
1844b15c8340SDag-Erling Smørgrav 	columns = width / (m + 2);
1845ca86bcf2SDag-Erling Smørgrav 	columns = MAXIMUM(columns, 1);
1846b15c8340SDag-Erling Smørgrav 	colspace = width / columns;
1847ca86bcf2SDag-Erling Smørgrav 	colspace = MINIMUM(colspace, width);
1848b15c8340SDag-Erling Smørgrav 
1849b15c8340SDag-Erling Smørgrav 	printf("\n");
1850b15c8340SDag-Erling Smørgrav 	m = 1;
1851b15c8340SDag-Erling Smørgrav 	for (y = 0; list[y]; y++) {
1852b15c8340SDag-Erling Smørgrav 		llen = strlen(list[y]);
1853b15c8340SDag-Erling Smørgrav 		tmp = llen > len ? list[y] + len : "";
1854076ad2f8SDag-Erling Smørgrav 		mprintf("%-*s", colspace, tmp);
1855b15c8340SDag-Erling Smørgrav 		if (m >= columns) {
1856b15c8340SDag-Erling Smørgrav 			printf("\n");
1857b15c8340SDag-Erling Smørgrav 			m = 1;
1858b15c8340SDag-Erling Smørgrav 		} else
1859b15c8340SDag-Erling Smørgrav 			m++;
1860b15c8340SDag-Erling Smørgrav 	}
1861b15c8340SDag-Erling Smørgrav 	printf("\n");
1862b15c8340SDag-Erling Smørgrav }
1863b15c8340SDag-Erling Smørgrav 
1864b15c8340SDag-Erling Smørgrav /*
1865b15c8340SDag-Erling Smørgrav  * Given a "list" of words that begin with a common prefix of "word",
1866b15c8340SDag-Erling Smørgrav  * attempt to find an autocompletion to extends "word" by the next
1867b15c8340SDag-Erling Smørgrav  * characters common to all entries in "list".
1868b15c8340SDag-Erling Smørgrav  */
1869b15c8340SDag-Erling Smørgrav static char *
1870b15c8340SDag-Erling Smørgrav complete_ambiguous(const char *word, char **list, size_t count)
1871b15c8340SDag-Erling Smørgrav {
1872b15c8340SDag-Erling Smørgrav 	if (word == NULL)
1873b15c8340SDag-Erling Smørgrav 		return NULL;
1874b15c8340SDag-Erling Smørgrav 
1875b15c8340SDag-Erling Smørgrav 	if (count > 0) {
1876b15c8340SDag-Erling Smørgrav 		u_int y, matchlen = strlen(list[0]);
1877b15c8340SDag-Erling Smørgrav 
1878b15c8340SDag-Erling Smørgrav 		/* Find length of common stem */
1879b15c8340SDag-Erling Smørgrav 		for (y = 1; list[y]; y++) {
1880b15c8340SDag-Erling Smørgrav 			u_int x;
1881b15c8340SDag-Erling Smørgrav 
1882b15c8340SDag-Erling Smørgrav 			for (x = 0; x < matchlen; x++)
1883b15c8340SDag-Erling Smørgrav 				if (list[0][x] != list[y][x])
1884b15c8340SDag-Erling Smørgrav 					break;
1885b15c8340SDag-Erling Smørgrav 
1886b15c8340SDag-Erling Smørgrav 			matchlen = x;
1887b15c8340SDag-Erling Smørgrav 		}
1888b15c8340SDag-Erling Smørgrav 
1889b15c8340SDag-Erling Smørgrav 		if (matchlen > strlen(word)) {
1890b15c8340SDag-Erling Smørgrav 			char *tmp = xstrdup(list[0]);
1891b15c8340SDag-Erling Smørgrav 
1892b15c8340SDag-Erling Smørgrav 			tmp[matchlen] = '\0';
1893b15c8340SDag-Erling Smørgrav 			return tmp;
1894b15c8340SDag-Erling Smørgrav 		}
1895b15c8340SDag-Erling Smørgrav 	}
1896b15c8340SDag-Erling Smørgrav 
1897b15c8340SDag-Erling Smørgrav 	return xstrdup(word);
1898b15c8340SDag-Erling Smørgrav }
1899b15c8340SDag-Erling Smørgrav 
1900b15c8340SDag-Erling Smørgrav /* Autocomplete a sftp command */
1901b15c8340SDag-Erling Smørgrav static int
1902b15c8340SDag-Erling Smørgrav complete_cmd_parse(EditLine *el, char *cmd, int lastarg, char quote,
1903b15c8340SDag-Erling Smørgrav     int terminated)
1904b15c8340SDag-Erling Smørgrav {
1905b15c8340SDag-Erling Smørgrav 	u_int y, count = 0, cmdlen, tmplen;
1906b15c8340SDag-Erling Smørgrav 	char *tmp, **list, argterm[3];
1907b15c8340SDag-Erling Smørgrav 	const LineInfo *lf;
1908b15c8340SDag-Erling Smørgrav 
1909b15c8340SDag-Erling Smørgrav 	list = xcalloc((sizeof(cmds) / sizeof(*cmds)) + 1, sizeof(char *));
1910b15c8340SDag-Erling Smørgrav 
1911b15c8340SDag-Erling Smørgrav 	/* No command specified: display all available commands */
1912b15c8340SDag-Erling Smørgrav 	if (cmd == NULL) {
1913b15c8340SDag-Erling Smørgrav 		for (y = 0; cmds[y].c; y++)
1914b15c8340SDag-Erling Smørgrav 			list[count++] = xstrdup(cmds[y].c);
1915b15c8340SDag-Erling Smørgrav 
1916b15c8340SDag-Erling Smørgrav 		list[count] = NULL;
1917b15c8340SDag-Erling Smørgrav 		complete_display(list, 0);
1918b15c8340SDag-Erling Smørgrav 
1919b15c8340SDag-Erling Smørgrav 		for (y = 0; list[y] != NULL; y++)
1920e4a9863fSDag-Erling Smørgrav 			free(list[y]);
1921e4a9863fSDag-Erling Smørgrav 		free(list);
1922b15c8340SDag-Erling Smørgrav 		return count;
1923b15c8340SDag-Erling Smørgrav 	}
1924b15c8340SDag-Erling Smørgrav 
1925b15c8340SDag-Erling Smørgrav 	/* Prepare subset of commands that start with "cmd" */
1926b15c8340SDag-Erling Smørgrav 	cmdlen = strlen(cmd);
1927b15c8340SDag-Erling Smørgrav 	for (y = 0; cmds[y].c; y++)  {
1928b15c8340SDag-Erling Smørgrav 		if (!strncasecmp(cmd, cmds[y].c, cmdlen))
1929b15c8340SDag-Erling Smørgrav 			list[count++] = xstrdup(cmds[y].c);
1930b15c8340SDag-Erling Smørgrav 	}
1931b15c8340SDag-Erling Smørgrav 	list[count] = NULL;
1932b15c8340SDag-Erling Smørgrav 
1933462c32cbSDag-Erling Smørgrav 	if (count == 0) {
1934e4a9863fSDag-Erling Smørgrav 		free(list);
1935b15c8340SDag-Erling Smørgrav 		return 0;
1936462c32cbSDag-Erling Smørgrav 	}
1937b15c8340SDag-Erling Smørgrav 
1938190cef3dSDag-Erling Smørgrav 	/* Complete ambiguous command */
1939b15c8340SDag-Erling Smørgrav 	tmp = complete_ambiguous(cmd, list, count);
1940b15c8340SDag-Erling Smørgrav 	if (count > 1)
1941b15c8340SDag-Erling Smørgrav 		complete_display(list, 0);
1942b15c8340SDag-Erling Smørgrav 
1943b15c8340SDag-Erling Smørgrav 	for (y = 0; list[y]; y++)
1944e4a9863fSDag-Erling Smørgrav 		free(list[y]);
1945e4a9863fSDag-Erling Smørgrav 	free(list);
1946b15c8340SDag-Erling Smørgrav 
1947b15c8340SDag-Erling Smørgrav 	if (tmp != NULL) {
1948b15c8340SDag-Erling Smørgrav 		tmplen = strlen(tmp);
1949b15c8340SDag-Erling Smørgrav 		cmdlen = strlen(cmd);
1950b15c8340SDag-Erling Smørgrav 		/* If cmd may be extended then do so */
1951b15c8340SDag-Erling Smørgrav 		if (tmplen > cmdlen)
1952b15c8340SDag-Erling Smørgrav 			if (el_insertstr(el, tmp + cmdlen) == -1)
1953b15c8340SDag-Erling Smørgrav 				fatal("el_insertstr failed.");
1954b15c8340SDag-Erling Smørgrav 		lf = el_line(el);
1955b15c8340SDag-Erling Smørgrav 		/* Terminate argument cleanly */
1956b15c8340SDag-Erling Smørgrav 		if (count == 1) {
1957b15c8340SDag-Erling Smørgrav 			y = 0;
1958b15c8340SDag-Erling Smørgrav 			if (!terminated)
1959b15c8340SDag-Erling Smørgrav 				argterm[y++] = quote;
1960b15c8340SDag-Erling Smørgrav 			if (lastarg || *(lf->cursor) != ' ')
1961b15c8340SDag-Erling Smørgrav 				argterm[y++] = ' ';
1962b15c8340SDag-Erling Smørgrav 			argterm[y] = '\0';
1963b15c8340SDag-Erling Smørgrav 			if (y > 0 && el_insertstr(el, argterm) == -1)
1964b15c8340SDag-Erling Smørgrav 				fatal("el_insertstr failed.");
1965b15c8340SDag-Erling Smørgrav 		}
1966e4a9863fSDag-Erling Smørgrav 		free(tmp);
1967b15c8340SDag-Erling Smørgrav 	}
1968b15c8340SDag-Erling Smørgrav 
1969b15c8340SDag-Erling Smørgrav 	return count;
1970b15c8340SDag-Erling Smørgrav }
1971b15c8340SDag-Erling Smørgrav 
1972b15c8340SDag-Erling Smørgrav /*
197338a52bd3SEd Maste  * Determine whether a particular sftp command's arguments (if any) represent
197438a52bd3SEd Maste  * local or remote files. The "cmdarg" argument specifies the actual argument
197538a52bd3SEd Maste  * and accepts values 1 or 2.
1976b15c8340SDag-Erling Smørgrav  */
1977b15c8340SDag-Erling Smørgrav static int
197838a52bd3SEd Maste complete_is_remote(char *cmd, int cmdarg) {
1979b15c8340SDag-Erling Smørgrav 	int i;
1980b15c8340SDag-Erling Smørgrav 
1981b15c8340SDag-Erling Smørgrav 	if (cmd == NULL)
1982b15c8340SDag-Erling Smørgrav 		return -1;
1983b15c8340SDag-Erling Smørgrav 
1984b15c8340SDag-Erling Smørgrav 	for (i = 0; cmds[i].c; i++) {
198538a52bd3SEd Maste 		if (!strncasecmp(cmd, cmds[i].c, strlen(cmds[i].c))) {
198638a52bd3SEd Maste 			if (cmdarg == 1)
1987b15c8340SDag-Erling Smørgrav 				return cmds[i].t;
198838a52bd3SEd Maste 			else if (cmdarg == 2)
198938a52bd3SEd Maste 				return cmds[i].t2;
199038a52bd3SEd Maste 			break;
199138a52bd3SEd Maste 		}
1992b15c8340SDag-Erling Smørgrav 	}
1993b15c8340SDag-Erling Smørgrav 
1994b15c8340SDag-Erling Smørgrav 	return -1;
1995b15c8340SDag-Erling Smørgrav }
1996b15c8340SDag-Erling Smørgrav 
1997b15c8340SDag-Erling Smørgrav /* Autocomplete a filename "file" */
1998b15c8340SDag-Erling Smørgrav static int
1999b15c8340SDag-Erling Smørgrav complete_match(EditLine *el, struct sftp_conn *conn, char *remote_path,
2000b15c8340SDag-Erling Smørgrav     char *file, int remote, int lastarg, char quote, int terminated)
2001b15c8340SDag-Erling Smørgrav {
2002b15c8340SDag-Erling Smørgrav 	glob_t g;
2003e4a9863fSDag-Erling Smørgrav 	char *tmp, *tmp2, ins[8];
20046888a9beSDag-Erling Smørgrav 	u_int i, hadglob, pwdlen, len, tmplen, filelen, cesc, isesc, isabs;
2005e4a9863fSDag-Erling Smørgrav 	int clen;
2006b15c8340SDag-Erling Smørgrav 	const LineInfo *lf;
2007b15c8340SDag-Erling Smørgrav 
2008b15c8340SDag-Erling Smørgrav 	/* Glob from "file" location */
2009b15c8340SDag-Erling Smørgrav 	if (file == NULL)
2010b15c8340SDag-Erling Smørgrav 		tmp = xstrdup("*");
2011b15c8340SDag-Erling Smørgrav 	else
2012b15c8340SDag-Erling Smørgrav 		xasprintf(&tmp, "%s*", file);
2013b15c8340SDag-Erling Smørgrav 
20146888a9beSDag-Erling Smørgrav 	/* Check if the path is absolute. */
201519261079SEd Maste 	isabs = path_absolute(tmp);
20166888a9beSDag-Erling Smørgrav 
2017b15c8340SDag-Erling Smørgrav 	memset(&g, 0, sizeof(g));
2018b15c8340SDag-Erling Smørgrav 	if (remote != LOCAL) {
201969c72a57SEd Maste 		tmp = make_absolute_pwd_glob(tmp, remote_path);
2020*edf85781SEd Maste 		sftp_glob(conn, tmp, GLOB_DOOFFS|GLOB_MARK, NULL, &g);
2021b15c8340SDag-Erling Smørgrav 	} else
2022535af610SEd Maste 		(void)glob(tmp, GLOB_DOOFFS|GLOB_MARK, NULL, &g);
2023b15c8340SDag-Erling Smørgrav 
2024b15c8340SDag-Erling Smørgrav 	/* Determine length of pwd so we can trim completion display */
2025b15c8340SDag-Erling Smørgrav 	for (hadglob = tmplen = pwdlen = 0; tmp[tmplen] != 0; tmplen++) {
2026b15c8340SDag-Erling Smørgrav 		/* Terminate counting on first unescaped glob metacharacter */
2027b15c8340SDag-Erling Smørgrav 		if (tmp[tmplen] == '*' || tmp[tmplen] == '?') {
2028b15c8340SDag-Erling Smørgrav 			if (tmp[tmplen] != '*' || tmp[tmplen + 1] != '\0')
2029b15c8340SDag-Erling Smørgrav 				hadglob = 1;
2030b15c8340SDag-Erling Smørgrav 			break;
2031b15c8340SDag-Erling Smørgrav 		}
2032b15c8340SDag-Erling Smørgrav 		if (tmp[tmplen] == '\\' && tmp[tmplen + 1] != '\0')
2033b15c8340SDag-Erling Smørgrav 			tmplen++;
2034b15c8340SDag-Erling Smørgrav 		if (tmp[tmplen] == '/')
2035b15c8340SDag-Erling Smørgrav 			pwdlen = tmplen + 1;	/* track last seen '/' */
2036b15c8340SDag-Erling Smørgrav 	}
2037e4a9863fSDag-Erling Smørgrav 	free(tmp);
2038a0ee8cc6SDag-Erling Smørgrav 	tmp = NULL;
2039b15c8340SDag-Erling Smørgrav 
2040b15c8340SDag-Erling Smørgrav 	if (g.gl_matchc == 0)
2041b15c8340SDag-Erling Smørgrav 		goto out;
2042b15c8340SDag-Erling Smørgrav 
2043b15c8340SDag-Erling Smørgrav 	if (g.gl_matchc > 1)
2044b15c8340SDag-Erling Smørgrav 		complete_display(g.gl_pathv, pwdlen);
2045b15c8340SDag-Erling Smørgrav 
2046b15c8340SDag-Erling Smørgrav 	/* Don't try to extend globs */
2047b15c8340SDag-Erling Smørgrav 	if (file == NULL || hadglob)
2048b15c8340SDag-Erling Smørgrav 		goto out;
2049b15c8340SDag-Erling Smørgrav 
2050b15c8340SDag-Erling Smørgrav 	tmp2 = complete_ambiguous(file, g.gl_pathv, g.gl_matchc);
20516888a9beSDag-Erling Smørgrav 	tmp = path_strip(tmp2, isabs ? NULL : remote_path);
2052e4a9863fSDag-Erling Smørgrav 	free(tmp2);
2053b15c8340SDag-Erling Smørgrav 
2054b15c8340SDag-Erling Smørgrav 	if (tmp == NULL)
2055b15c8340SDag-Erling Smørgrav 		goto out;
2056b15c8340SDag-Erling Smørgrav 
2057b15c8340SDag-Erling Smørgrav 	tmplen = strlen(tmp);
2058b15c8340SDag-Erling Smørgrav 	filelen = strlen(file);
2059b15c8340SDag-Erling Smørgrav 
20606888a9beSDag-Erling Smørgrav 	/* Count the number of escaped characters in the input string. */
20616888a9beSDag-Erling Smørgrav 	cesc = isesc = 0;
20626888a9beSDag-Erling Smørgrav 	for (i = 0; i < filelen; i++) {
20636888a9beSDag-Erling Smørgrav 		if (!isesc && file[i] == '\\' && i + 1 < filelen){
20646888a9beSDag-Erling Smørgrav 			isesc = 1;
20656888a9beSDag-Erling Smørgrav 			cesc++;
20666888a9beSDag-Erling Smørgrav 		} else
20676888a9beSDag-Erling Smørgrav 			isesc = 0;
20686888a9beSDag-Erling Smørgrav 	}
20696888a9beSDag-Erling Smørgrav 
20706888a9beSDag-Erling Smørgrav 	if (tmplen > (filelen - cesc)) {
20716888a9beSDag-Erling Smørgrav 		tmp2 = tmp + filelen - cesc;
2072b15c8340SDag-Erling Smørgrav 		len = strlen(tmp2);
2073b15c8340SDag-Erling Smørgrav 		/* quote argument on way out */
2074e4a9863fSDag-Erling Smørgrav 		for (i = 0; i < len; i += clen) {
2075e4a9863fSDag-Erling Smørgrav 			if ((clen = mblen(tmp2 + i, len - i)) < 0 ||
2076e4a9863fSDag-Erling Smørgrav 			    (size_t)clen > sizeof(ins) - 2)
2077e4a9863fSDag-Erling Smørgrav 				fatal("invalid multibyte character");
2078b15c8340SDag-Erling Smørgrav 			ins[0] = '\\';
2079e4a9863fSDag-Erling Smørgrav 			memcpy(ins + 1, tmp2 + i, clen);
2080e4a9863fSDag-Erling Smørgrav 			ins[clen + 1] = '\0';
2081b15c8340SDag-Erling Smørgrav 			switch (tmp2[i]) {
2082b15c8340SDag-Erling Smørgrav 			case '\'':
2083b15c8340SDag-Erling Smørgrav 			case '"':
2084b15c8340SDag-Erling Smørgrav 			case '\\':
2085b15c8340SDag-Erling Smørgrav 			case '\t':
20864a421b63SDag-Erling Smørgrav 			case '[':
2087b15c8340SDag-Erling Smørgrav 			case ' ':
20886888a9beSDag-Erling Smørgrav 			case '#':
20896888a9beSDag-Erling Smørgrav 			case '*':
2090b15c8340SDag-Erling Smørgrav 				if (quote == '\0' || tmp2[i] == quote) {
2091b15c8340SDag-Erling Smørgrav 					if (el_insertstr(el, ins) == -1)
2092b15c8340SDag-Erling Smørgrav 						fatal("el_insertstr "
2093b15c8340SDag-Erling Smørgrav 						    "failed.");
2094b15c8340SDag-Erling Smørgrav 					break;
2095b15c8340SDag-Erling Smørgrav 				}
2096b15c8340SDag-Erling Smørgrav 				/* FALLTHROUGH */
2097b15c8340SDag-Erling Smørgrav 			default:
2098b15c8340SDag-Erling Smørgrav 				if (el_insertstr(el, ins + 1) == -1)
2099b15c8340SDag-Erling Smørgrav 					fatal("el_insertstr failed.");
2100b15c8340SDag-Erling Smørgrav 				break;
2101b15c8340SDag-Erling Smørgrav 			}
2102b15c8340SDag-Erling Smørgrav 		}
2103b15c8340SDag-Erling Smørgrav 	}
2104b15c8340SDag-Erling Smørgrav 
2105b15c8340SDag-Erling Smørgrav 	lf = el_line(el);
2106b15c8340SDag-Erling Smørgrav 	if (g.gl_matchc == 1) {
2107b15c8340SDag-Erling Smørgrav 		i = 0;
2108a0ee8cc6SDag-Erling Smørgrav 		if (!terminated && quote != '\0')
2109b15c8340SDag-Erling Smørgrav 			ins[i++] = quote;
2110b15c8340SDag-Erling Smørgrav 		if (*(lf->cursor - 1) != '/' &&
2111b15c8340SDag-Erling Smørgrav 		    (lastarg || *(lf->cursor) != ' '))
2112b15c8340SDag-Erling Smørgrav 			ins[i++] = ' ';
2113b15c8340SDag-Erling Smørgrav 		ins[i] = '\0';
2114b15c8340SDag-Erling Smørgrav 		if (i > 0 && el_insertstr(el, ins) == -1)
2115b15c8340SDag-Erling Smørgrav 			fatal("el_insertstr failed.");
2116b15c8340SDag-Erling Smørgrav 	}
2117e4a9863fSDag-Erling Smørgrav 	free(tmp);
2118b15c8340SDag-Erling Smørgrav 
2119b15c8340SDag-Erling Smørgrav  out:
2120b15c8340SDag-Erling Smørgrav 	globfree(&g);
2121b15c8340SDag-Erling Smørgrav 	return g.gl_matchc;
2122b15c8340SDag-Erling Smørgrav }
2123b15c8340SDag-Erling Smørgrav 
2124b15c8340SDag-Erling Smørgrav /* tab-completion hook function, called via libedit */
2125b15c8340SDag-Erling Smørgrav static unsigned char
2126b15c8340SDag-Erling Smørgrav complete(EditLine *el, int ch)
2127b15c8340SDag-Erling Smørgrav {
2128b15c8340SDag-Erling Smørgrav 	char **argv, *line, quote;
2129e4a9863fSDag-Erling Smørgrav 	int argc, carg;
2130e4a9863fSDag-Erling Smørgrav 	u_int cursor, len, terminated, ret = CC_ERROR;
2131b15c8340SDag-Erling Smørgrav 	const LineInfo *lf;
2132b15c8340SDag-Erling Smørgrav 	struct complete_ctx *complete_ctx;
2133b15c8340SDag-Erling Smørgrav 
2134b15c8340SDag-Erling Smørgrav 	lf = el_line(el);
2135b15c8340SDag-Erling Smørgrav 	if (el_get(el, EL_CLIENTDATA, (void**)&complete_ctx) != 0)
213619261079SEd Maste 		fatal_f("el_get failed");
2137b15c8340SDag-Erling Smørgrav 
2138b15c8340SDag-Erling Smørgrav 	/* Figure out which argument the cursor points to */
2139b15c8340SDag-Erling Smørgrav 	cursor = lf->cursor - lf->buffer;
2140fc1ba28aSDag-Erling Smørgrav 	line = xmalloc(cursor + 1);
2141b15c8340SDag-Erling Smørgrav 	memcpy(line, lf->buffer, cursor);
2142b15c8340SDag-Erling Smørgrav 	line[cursor] = '\0';
2143b15c8340SDag-Erling Smørgrav 	argv = makeargv(line, &carg, 1, &quote, &terminated);
2144e4a9863fSDag-Erling Smørgrav 	free(line);
2145b15c8340SDag-Erling Smørgrav 
2146b15c8340SDag-Erling Smørgrav 	/* Get all the arguments on the line */
2147b15c8340SDag-Erling Smørgrav 	len = lf->lastchar - lf->buffer;
2148fc1ba28aSDag-Erling Smørgrav 	line = xmalloc(len + 1);
2149b15c8340SDag-Erling Smørgrav 	memcpy(line, lf->buffer, len);
2150b15c8340SDag-Erling Smørgrav 	line[len] = '\0';
2151b15c8340SDag-Erling Smørgrav 	argv = makeargv(line, &argc, 1, NULL, NULL);
2152b15c8340SDag-Erling Smørgrav 
2153b15c8340SDag-Erling Smørgrav 	/* Ensure cursor is at EOL or a argument boundary */
2154b15c8340SDag-Erling Smørgrav 	if (line[cursor] != ' ' && line[cursor] != '\0' &&
2155b15c8340SDag-Erling Smørgrav 	    line[cursor] != '\n') {
2156e4a9863fSDag-Erling Smørgrav 		free(line);
2157b15c8340SDag-Erling Smørgrav 		return ret;
2158b15c8340SDag-Erling Smørgrav 	}
2159b15c8340SDag-Erling Smørgrav 
2160b15c8340SDag-Erling Smørgrav 	if (carg == 0) {
2161b15c8340SDag-Erling Smørgrav 		/* Show all available commands */
2162b15c8340SDag-Erling Smørgrav 		complete_cmd_parse(el, NULL, argc == carg, '\0', 1);
2163b15c8340SDag-Erling Smørgrav 		ret = CC_REDISPLAY;
2164b15c8340SDag-Erling Smørgrav 	} else if (carg == 1 && cursor > 0 && line[cursor - 1] != ' ')  {
2165b15c8340SDag-Erling Smørgrav 		/* Handle the command parsing */
2166b15c8340SDag-Erling Smørgrav 		if (complete_cmd_parse(el, argv[0], argc == carg,
2167b15c8340SDag-Erling Smørgrav 		    quote, terminated) != 0)
2168b15c8340SDag-Erling Smørgrav 			ret = CC_REDISPLAY;
2169b15c8340SDag-Erling Smørgrav 	} else if (carg >= 1) {
2170b15c8340SDag-Erling Smørgrav 		/* Handle file parsing */
217138a52bd3SEd Maste 		int remote = 0;
217238a52bd3SEd Maste 		int i = 0, cmdarg = 0;
2173b15c8340SDag-Erling Smørgrav 		char *filematch = NULL;
2174b15c8340SDag-Erling Smørgrav 
2175b15c8340SDag-Erling Smørgrav 		if (carg > 1 && line[cursor-1] != ' ')
2176b15c8340SDag-Erling Smørgrav 			filematch = argv[carg - 1];
2177b15c8340SDag-Erling Smørgrav 
217838a52bd3SEd Maste 		for (i = 1; i < carg; i++) {
217938a52bd3SEd Maste 			/* Skip flags */
218038a52bd3SEd Maste 			if (argv[i][0] != '-')
218138a52bd3SEd Maste 				cmdarg++;
218238a52bd3SEd Maste 		}
218338a52bd3SEd Maste 
218438a52bd3SEd Maste 		/*
218538a52bd3SEd Maste 		 * If previous argument is complete, then offer completion
218638a52bd3SEd Maste 		 * on the next one.
218738a52bd3SEd Maste 		 */
218838a52bd3SEd Maste 		if (line[cursor - 1] == ' ')
218938a52bd3SEd Maste 			cmdarg++;
219038a52bd3SEd Maste 
219138a52bd3SEd Maste 		remote = complete_is_remote(argv[0], cmdarg);
219238a52bd3SEd Maste 
219338a52bd3SEd Maste 		if ((remote == REMOTE || remote == LOCAL) &&
2194b15c8340SDag-Erling Smørgrav 		    complete_match(el, complete_ctx->conn,
2195b15c8340SDag-Erling Smørgrav 		    *complete_ctx->remote_pathp, filematch,
2196b15c8340SDag-Erling Smørgrav 		    remote, carg == argc, quote, terminated) != 0)
2197b15c8340SDag-Erling Smørgrav 			ret = CC_REDISPLAY;
2198b15c8340SDag-Erling Smørgrav 	}
2199b15c8340SDag-Erling Smørgrav 
2200e4a9863fSDag-Erling Smørgrav 	free(line);
2201b15c8340SDag-Erling Smørgrav 	return ret;
2202b15c8340SDag-Erling Smørgrav }
2203b15c8340SDag-Erling Smørgrav #endif /* USE_LIBEDIT */
22045e8dbd04SDag-Erling Smørgrav 
220547dd1d1bSDag-Erling Smørgrav static int
2206b15c8340SDag-Erling Smørgrav interactive_loop(struct sftp_conn *conn, char *file1, char *file2)
2207efcad6b7SDag-Erling Smørgrav {
2208b15c8340SDag-Erling Smørgrav 	char *remote_path;
220947dd1d1bSDag-Erling Smørgrav 	char *dir = NULL, *startdir = NULL;
2210efcad6b7SDag-Erling Smørgrav 	char cmd[2048];
2211043840dfSDag-Erling Smørgrav 	int err, interactive;
22125e8dbd04SDag-Erling Smørgrav 	EditLine *el = NULL;
22135e8dbd04SDag-Erling Smørgrav #ifdef USE_LIBEDIT
22145e8dbd04SDag-Erling Smørgrav 	History *hl = NULL;
22155e8dbd04SDag-Erling Smørgrav 	HistEvent hev;
22165e8dbd04SDag-Erling Smørgrav 	extern char *__progname;
2217b15c8340SDag-Erling Smørgrav 	struct complete_ctx complete_ctx;
22185e8dbd04SDag-Erling Smørgrav 
22195e8dbd04SDag-Erling Smørgrav 	if (!batchmode && isatty(STDIN_FILENO)) {
22205e8dbd04SDag-Erling Smørgrav 		if ((el = el_init(__progname, stdin, stdout, stderr)) == NULL)
22215e8dbd04SDag-Erling Smørgrav 			fatal("Couldn't initialise editline");
22225e8dbd04SDag-Erling Smørgrav 		if ((hl = history_init()) == NULL)
22235e8dbd04SDag-Erling Smørgrav 			fatal("Couldn't initialise editline history");
22245e8dbd04SDag-Erling Smørgrav 		history(hl, &hev, H_SETSIZE, 100);
22255e8dbd04SDag-Erling Smørgrav 		el_set(el, EL_HIST, history, hl);
22265e8dbd04SDag-Erling Smørgrav 
22275e8dbd04SDag-Erling Smørgrav 		el_set(el, EL_PROMPT, prompt);
22285e8dbd04SDag-Erling Smørgrav 		el_set(el, EL_EDITOR, "emacs");
22295e8dbd04SDag-Erling Smørgrav 		el_set(el, EL_TERMINAL, NULL);
22305e8dbd04SDag-Erling Smørgrav 		el_set(el, EL_SIGNAL, 1);
22315e8dbd04SDag-Erling Smørgrav 		el_source(el, NULL);
2232b15c8340SDag-Erling Smørgrav 
2233b15c8340SDag-Erling Smørgrav 		/* Tab Completion */
2234b15c8340SDag-Erling Smørgrav 		el_set(el, EL_ADDFN, "ftp-complete",
22354a421b63SDag-Erling Smørgrav 		    "Context sensitive argument completion", complete);
2236b15c8340SDag-Erling Smørgrav 		complete_ctx.conn = conn;
2237b15c8340SDag-Erling Smørgrav 		complete_ctx.remote_pathp = &remote_path;
2238b15c8340SDag-Erling Smørgrav 		el_set(el, EL_CLIENTDATA, (void*)&complete_ctx);
2239b15c8340SDag-Erling Smørgrav 		el_set(el, EL_BIND, "^I", "ftp-complete", NULL);
2240f7167e0eSDag-Erling Smørgrav 		/* enable ctrl-left-arrow and ctrl-right-arrow */
2241f7167e0eSDag-Erling Smørgrav 		el_set(el, EL_BIND, "\\e[1;5C", "em-next-word", NULL);
224219261079SEd Maste 		el_set(el, EL_BIND, "\\e\\e[C", "em-next-word", NULL);
2243f7167e0eSDag-Erling Smørgrav 		el_set(el, EL_BIND, "\\e[1;5D", "ed-prev-word", NULL);
2244f7167e0eSDag-Erling Smørgrav 		el_set(el, EL_BIND, "\\e\\e[D", "ed-prev-word", NULL);
2245f7167e0eSDag-Erling Smørgrav 		/* make ^w match ksh behaviour */
2246f7167e0eSDag-Erling Smørgrav 		el_set(el, EL_BIND, "^w", "ed-delete-prev-word", NULL);
22475e8dbd04SDag-Erling Smørgrav 	}
22485e8dbd04SDag-Erling Smørgrav #endif /* USE_LIBEDIT */
2249efcad6b7SDag-Erling Smørgrav 
2250*edf85781SEd Maste 	if ((remote_path = sftp_realpath(conn, ".")) == NULL)
2251efcad6b7SDag-Erling Smørgrav 		fatal("Need cwd");
225247dd1d1bSDag-Erling Smørgrav 	startdir = xstrdup(remote_path);
2253efcad6b7SDag-Erling Smørgrav 
2254efcad6b7SDag-Erling Smørgrav 	if (file1 != NULL) {
2255efcad6b7SDag-Erling Smørgrav 		dir = xstrdup(file1);
2256*edf85781SEd Maste 		dir = sftp_make_absolute(dir, remote_path);
2257efcad6b7SDag-Erling Smørgrav 
2258*edf85781SEd Maste 		if (sftp_remote_is_dir(conn, dir) && file2 == NULL) {
2259e4a9863fSDag-Erling Smørgrav 			if (!quiet)
2260076ad2f8SDag-Erling Smørgrav 				mprintf("Changing to: %s\n", dir);
2261efcad6b7SDag-Erling Smørgrav 			snprintf(cmd, sizeof cmd, "cd \"%s\"", dir);
2262b15c8340SDag-Erling Smørgrav 			if (parse_dispatch_command(conn, cmd,
226319261079SEd Maste 			    &remote_path, startdir, 1, 0) != 0) {
2264e4a9863fSDag-Erling Smørgrav 				free(dir);
226547dd1d1bSDag-Erling Smørgrav 				free(startdir);
2266e4a9863fSDag-Erling Smørgrav 				free(remote_path);
2267e4a9863fSDag-Erling Smørgrav 				free(conn);
2268efcad6b7SDag-Erling Smørgrav 				return (-1);
22695e8dbd04SDag-Erling Smørgrav 			}
2270efcad6b7SDag-Erling Smørgrav 		} else {
22716888a9beSDag-Erling Smørgrav 			/* XXX this is wrong wrt quoting */
2272e4a9863fSDag-Erling Smørgrav 			snprintf(cmd, sizeof cmd, "get%s %s%s%s",
2273e4a9863fSDag-Erling Smørgrav 			    global_aflag ? " -a" : "", dir,
2274e4a9863fSDag-Erling Smørgrav 			    file2 == NULL ? "" : " ",
2275e4a9863fSDag-Erling Smørgrav 			    file2 == NULL ? "" : file2);
2276b15c8340SDag-Erling Smørgrav 			err = parse_dispatch_command(conn, cmd,
227719261079SEd Maste 			    &remote_path, startdir, 1, 0);
2278e4a9863fSDag-Erling Smørgrav 			free(dir);
227947dd1d1bSDag-Erling Smørgrav 			free(startdir);
2280e4a9863fSDag-Erling Smørgrav 			free(remote_path);
2281e4a9863fSDag-Erling Smørgrav 			free(conn);
2282efcad6b7SDag-Erling Smørgrav 			return (err);
2283efcad6b7SDag-Erling Smørgrav 		}
2284e4a9863fSDag-Erling Smørgrav 		free(dir);
2285efcad6b7SDag-Erling Smørgrav 	}
2286efcad6b7SDag-Erling Smørgrav 
2287bc5531deSDag-Erling Smørgrav 	setvbuf(stdout, NULL, _IOLBF, 0);
2288bc5531deSDag-Erling Smørgrav 	setvbuf(infile, NULL, _IOLBF, 0);
2289efcad6b7SDag-Erling Smørgrav 
2290043840dfSDag-Erling Smørgrav 	interactive = !batchmode && isatty(STDIN_FILENO);
2291efcad6b7SDag-Erling Smørgrav 	err = 0;
2292efcad6b7SDag-Erling Smørgrav 	for (;;) {
2293e9e8876aSEd Maste 		struct sigaction sa;
2294e9e8876aSEd Maste 
2295e9e8876aSEd Maste 		interrupted = 0;
2296e9e8876aSEd Maste 		memset(&sa, 0, sizeof(sa));
2297e9e8876aSEd Maste 		sa.sa_handler = interactive ? read_interrupt : killchild;
2298e9e8876aSEd Maste 		if (sigaction(SIGINT, &sa, NULL) == -1) {
2299e9e8876aSEd Maste 			debug3("sigaction(%s): %s", strsignal(SIGINT),
2300e9e8876aSEd Maste 			    strerror(errno));
2301e9e8876aSEd Maste 			break;
2302e9e8876aSEd Maste 		}
23035e8dbd04SDag-Erling Smørgrav 		if (el == NULL) {
2304043840dfSDag-Erling Smørgrav 			if (interactive)
2305efcad6b7SDag-Erling Smørgrav 				printf("sftp> ");
2306efcad6b7SDag-Erling Smørgrav 			if (fgets(cmd, sizeof(cmd), infile) == NULL) {
2307043840dfSDag-Erling Smørgrav 				if (interactive)
2308efcad6b7SDag-Erling Smørgrav 					printf("\n");
2309e9e8876aSEd Maste 				if (interrupted)
2310e9e8876aSEd Maste 					continue;
2311efcad6b7SDag-Erling Smørgrav 				break;
2312efcad6b7SDag-Erling Smørgrav 			}
23135e8dbd04SDag-Erling Smørgrav 		} else {
23145e8dbd04SDag-Erling Smørgrav #ifdef USE_LIBEDIT
23155e8dbd04SDag-Erling Smørgrav 			const char *line;
23165e8dbd04SDag-Erling Smørgrav 			int count = 0;
23175e8dbd04SDag-Erling Smørgrav 
2318b15c8340SDag-Erling Smørgrav 			if ((line = el_gets(el, &count)) == NULL ||
2319b15c8340SDag-Erling Smørgrav 			    count <= 0) {
2320043840dfSDag-Erling Smørgrav 				printf("\n");
232119261079SEd Maste 				if (interrupted)
232219261079SEd Maste 					continue;
23235e8dbd04SDag-Erling Smørgrav 				break;
2324043840dfSDag-Erling Smørgrav 			}
23255e8dbd04SDag-Erling Smørgrav 			history(hl, &hev, H_ENTER, line);
23265e8dbd04SDag-Erling Smørgrav 			if (strlcpy(cmd, line, sizeof(cmd)) >= sizeof(cmd)) {
23275e8dbd04SDag-Erling Smørgrav 				fprintf(stderr, "Error: input line too long\n");
23285e8dbd04SDag-Erling Smørgrav 				continue;
23295e8dbd04SDag-Erling Smørgrav 			}
23305e8dbd04SDag-Erling Smørgrav #endif /* USE_LIBEDIT */
23315e8dbd04SDag-Erling Smørgrav 		}
2332efcad6b7SDag-Erling Smørgrav 
233319261079SEd Maste 		cmd[strcspn(cmd, "\n")] = '\0';
2334efcad6b7SDag-Erling Smørgrav 
2335d74d50a8SDag-Erling Smørgrav 		/* Handle user interrupts gracefully during commands */
2336d74d50a8SDag-Erling Smørgrav 		interrupted = 0;
233719261079SEd Maste 		ssh_signal(SIGINT, cmd_interrupt);
2338d74d50a8SDag-Erling Smørgrav 
2339b15c8340SDag-Erling Smørgrav 		err = parse_dispatch_command(conn, cmd, &remote_path,
234019261079SEd Maste 		    startdir, batchmode, !interactive && el == NULL);
2341efcad6b7SDag-Erling Smørgrav 		if (err != 0)
2342efcad6b7SDag-Erling Smørgrav 			break;
2343efcad6b7SDag-Erling Smørgrav 	}
234419261079SEd Maste 	ssh_signal(SIGCHLD, SIG_DFL);
2345e4a9863fSDag-Erling Smørgrav 	free(remote_path);
234647dd1d1bSDag-Erling Smørgrav 	free(startdir);
2347e4a9863fSDag-Erling Smørgrav 	free(conn);
2348efcad6b7SDag-Erling Smørgrav 
2349043840dfSDag-Erling Smørgrav #ifdef USE_LIBEDIT
2350043840dfSDag-Erling Smørgrav 	if (el != NULL)
2351043840dfSDag-Erling Smørgrav 		el_end(el);
2352043840dfSDag-Erling Smørgrav #endif /* USE_LIBEDIT */
2353043840dfSDag-Erling Smørgrav 
2354efcad6b7SDag-Erling Smørgrav 	/* err == 1 signifies normal "quit" exit */
2355efcad6b7SDag-Erling Smørgrav 	return (err >= 0 ? 0 : -1);
2356efcad6b7SDag-Erling Smørgrav }
2357d0c8c0bcSDag-Erling Smørgrav 
2358ae1f160dSDag-Erling Smørgrav static void
2359d95e11bfSDag-Erling Smørgrav connect_to_server(char *path, char **args, int *in, int *out)
23601e8db6e2SBrian Feldman {
23611e8db6e2SBrian Feldman 	int c_in, c_out;
23621e8db6e2SBrian Feldman #ifdef USE_PIPES
23631e8db6e2SBrian Feldman 	int pin[2], pout[2];
2364ee21a45fSDag-Erling Smørgrav 
23651e8db6e2SBrian Feldman 	if ((pipe(pin) == -1) || (pipe(pout) == -1))
23661e8db6e2SBrian Feldman 		fatal("pipe: %s", strerror(errno));
23671e8db6e2SBrian Feldman 	*in = pin[0];
23681e8db6e2SBrian Feldman 	*out = pout[1];
23691e8db6e2SBrian Feldman 	c_in = pout[0];
23701e8db6e2SBrian Feldman 	c_out = pin[1];
23711e8db6e2SBrian Feldman #else /* USE_PIPES */
23721e8db6e2SBrian Feldman 	int inout[2];
2373ee21a45fSDag-Erling Smørgrav 
23741e8db6e2SBrian Feldman 	if (socketpair(AF_UNIX, SOCK_STREAM, 0, inout) == -1)
23751e8db6e2SBrian Feldman 		fatal("socketpair: %s", strerror(errno));
23761e8db6e2SBrian Feldman 	*in = *out = inout[0];
23771e8db6e2SBrian Feldman 	c_in = c_out = inout[1];
23781e8db6e2SBrian Feldman #endif /* USE_PIPES */
23791e8db6e2SBrian Feldman 
2380d95e11bfSDag-Erling Smørgrav 	if ((sshpid = fork()) == -1)
23811e8db6e2SBrian Feldman 		fatal("fork: %s", strerror(errno));
2382d95e11bfSDag-Erling Smørgrav 	else if (sshpid == 0) {
23831e8db6e2SBrian Feldman 		if ((dup2(c_in, STDIN_FILENO) == -1) ||
23841e8db6e2SBrian Feldman 		    (dup2(c_out, STDOUT_FILENO) == -1)) {
23851e8db6e2SBrian Feldman 			fprintf(stderr, "dup2: %s\n", strerror(errno));
2386d74d50a8SDag-Erling Smørgrav 			_exit(1);
23871e8db6e2SBrian Feldman 		}
23881e8db6e2SBrian Feldman 		close(*in);
23891e8db6e2SBrian Feldman 		close(*out);
23901e8db6e2SBrian Feldman 		close(c_in);
23911e8db6e2SBrian Feldman 		close(c_out);
2392d74d50a8SDag-Erling Smørgrav 
2393d74d50a8SDag-Erling Smørgrav 		/*
2394d74d50a8SDag-Erling Smørgrav 		 * The underlying ssh is in the same process group, so we must
2395d74d50a8SDag-Erling Smørgrav 		 * ignore SIGINT if we want to gracefully abort commands,
2396d74d50a8SDag-Erling Smørgrav 		 * otherwise the signal will make it to the ssh process and
2397b15c8340SDag-Erling Smørgrav 		 * kill it too.  Contrawise, since sftp sends SIGTERMs to the
2398b15c8340SDag-Erling Smørgrav 		 * underlying ssh, it must *not* ignore that signal.
2399d74d50a8SDag-Erling Smørgrav 		 */
240019261079SEd Maste 		ssh_signal(SIGINT, SIG_IGN);
240119261079SEd Maste 		ssh_signal(SIGTERM, SIG_DFL);
2402d74d50a8SDag-Erling Smørgrav 		execvp(path, args);
2403ae1f160dSDag-Erling Smørgrav 		fprintf(stderr, "exec: %s: %s\n", path, strerror(errno));
2404d74d50a8SDag-Erling Smørgrav 		_exit(1);
24051e8db6e2SBrian Feldman 	}
24061e8db6e2SBrian Feldman 
240719261079SEd Maste 	ssh_signal(SIGTERM, killchild);
240819261079SEd Maste 	ssh_signal(SIGINT, killchild);
240919261079SEd Maste 	ssh_signal(SIGHUP, killchild);
241019261079SEd Maste 	ssh_signal(SIGTSTP, suspchild);
241119261079SEd Maste 	ssh_signal(SIGTTIN, suspchild);
241219261079SEd Maste 	ssh_signal(SIGTTOU, suspchild);
241319261079SEd Maste 	ssh_signal(SIGCHLD, sigchld_handler);
24141e8db6e2SBrian Feldman 	close(c_in);
24151e8db6e2SBrian Feldman 	close(c_out);
24161e8db6e2SBrian Feldman }
24171e8db6e2SBrian Feldman 
2418ae1f160dSDag-Erling Smørgrav static void
24191e8db6e2SBrian Feldman usage(void)
24201e8db6e2SBrian Feldman {
2421ae1f160dSDag-Erling Smørgrav 	extern char *__progname;
2422ae1f160dSDag-Erling Smørgrav 
2423ae1f160dSDag-Erling Smørgrav 	fprintf(stderr,
242419261079SEd Maste 	    "usage: %s [-46AaCfNpqrv] [-B buffer_size] [-b batchfile] [-c cipher]\n"
242538a52bd3SEd Maste 	    "          [-D sftp_server_command] [-F ssh_config] [-i identity_file]\n"
242619261079SEd Maste 	    "          [-J destination] [-l limit] [-o ssh_option] [-P port]\n"
242719261079SEd Maste 	    "          [-R num_requests] [-S program] [-s subsystem | sftp_server]\n"
2428f374ba41SEd Maste 	    "          [-X sftp_option] destination\n",
242947dd1d1bSDag-Erling Smørgrav 	    __progname);
24301e8db6e2SBrian Feldman 	exit(1);
24311e8db6e2SBrian Feldman }
24321e8db6e2SBrian Feldman 
24331e8db6e2SBrian Feldman int
24341e8db6e2SBrian Feldman main(int argc, char **argv)
24351e8db6e2SBrian Feldman {
243638a52bd3SEd Maste 	int r, in, out, ch, err, tmp, port = -1, noisy = 0;
243738a52bd3SEd Maste 	char *host = NULL, *user, *cp, **cpp, *file2 = NULL;
243819261079SEd Maste 	int debug_level = 0;
2439ae1f160dSDag-Erling Smørgrav 	char *file1 = NULL, *sftp_server = NULL;
2440ae1f160dSDag-Erling Smørgrav 	char *ssh_program = _PATH_SSH_PROGRAM, *sftp_direct = NULL;
24414a421b63SDag-Erling Smørgrav 	const char *errstr;
2442ae1f160dSDag-Erling Smørgrav 	LogLevel ll = SYSLOG_LEVEL_INFO;
2443ae1f160dSDag-Erling Smørgrav 	arglist args;
24441e8db6e2SBrian Feldman 	extern int optind;
24451e8db6e2SBrian Feldman 	extern char *optarg;
2446b15c8340SDag-Erling Smørgrav 	struct sftp_conn *conn;
244719261079SEd Maste 	size_t copy_buffer_len = 0;
244819261079SEd Maste 	size_t num_requests = 0;
2449f374ba41SEd Maste 	long long llv, limit_kbps = 0;
24501e8db6e2SBrian Feldman 
2451021d409fSDag-Erling Smørgrav 	/* Ensure that fds 0, 1 and 2 are open or directed to /dev/null */
2452021d409fSDag-Erling Smørgrav 	sanitise_stdfd();
2453ca86bcf2SDag-Erling Smørgrav 	msetlocale();
2454021d409fSDag-Erling Smørgrav 
2455d95e11bfSDag-Erling Smørgrav 	__progname = ssh_get_progname(argv[0]);
2456021d409fSDag-Erling Smørgrav 	memset(&args, '\0', sizeof(args));
2457ae1f160dSDag-Erling Smørgrav 	args.list = NULL;
2458761efaa7SDag-Erling Smørgrav 	addargs(&args, "%s", ssh_program);
2459ae1f160dSDag-Erling Smørgrav 	addargs(&args, "-oForwardX11 no");
2460021d409fSDag-Erling Smørgrav 	addargs(&args, "-oPermitLocalCommand no");
2461ae1f160dSDag-Erling Smørgrav 	addargs(&args, "-oClearAllForwardings yes");
2462efcad6b7SDag-Erling Smørgrav 
2463ae1f160dSDag-Erling Smørgrav 	ll = SYSLOG_LEVEL_INFO;
2464efcad6b7SDag-Erling Smørgrav 	infile = stdin;
24651e8db6e2SBrian Feldman 
2466b15c8340SDag-Erling Smørgrav 	while ((ch = getopt(argc, argv,
2467f374ba41SEd Maste 	    "1246AafhNpqrvCc:D:i:l:o:s:S:b:B:F:J:P:R:X:")) != -1) {
24681e8db6e2SBrian Feldman 		switch (ch) {
2469b15c8340SDag-Erling Smørgrav 		/* Passed through to ssh(1) */
247019261079SEd Maste 		case 'A':
2471b15c8340SDag-Erling Smørgrav 		case '4':
2472b15c8340SDag-Erling Smørgrav 		case '6':
24731e8db6e2SBrian Feldman 		case 'C':
2474b15c8340SDag-Erling Smørgrav 			addargs(&args, "-%c", ch);
2475b15c8340SDag-Erling Smørgrav 			break;
2476b15c8340SDag-Erling Smørgrav 		/* Passed through to ssh(1) with argument */
2477b15c8340SDag-Erling Smørgrav 		case 'F':
247819261079SEd Maste 		case 'J':
2479b15c8340SDag-Erling Smørgrav 		case 'c':
2480b15c8340SDag-Erling Smørgrav 		case 'i':
2481b15c8340SDag-Erling Smørgrav 		case 'o':
2482b15c8340SDag-Erling Smørgrav 			addargs(&args, "-%c", ch);
2483b15c8340SDag-Erling Smørgrav 			addargs(&args, "%s", optarg);
2484b15c8340SDag-Erling Smørgrav 			break;
2485b15c8340SDag-Erling Smørgrav 		case 'q':
2486e4a9863fSDag-Erling Smørgrav 			ll = SYSLOG_LEVEL_ERROR;
2487e4a9863fSDag-Erling Smørgrav 			quiet = 1;
2488b15c8340SDag-Erling Smørgrav 			showprogress = 0;
2489b15c8340SDag-Erling Smørgrav 			addargs(&args, "-%c", ch);
2490b15c8340SDag-Erling Smørgrav 			break;
2491b15c8340SDag-Erling Smørgrav 		case 'P':
249247dd1d1bSDag-Erling Smørgrav 			port = a2port(optarg);
249347dd1d1bSDag-Erling Smørgrav 			if (port <= 0)
249447dd1d1bSDag-Erling Smørgrav 				fatal("Bad port \"%s\"\n", optarg);
24951e8db6e2SBrian Feldman 			break;
24961e8db6e2SBrian Feldman 		case 'v':
2497ae1f160dSDag-Erling Smørgrav 			if (debug_level < 3) {
2498ae1f160dSDag-Erling Smørgrav 				addargs(&args, "-v");
2499ae1f160dSDag-Erling Smørgrav 				ll = SYSLOG_LEVEL_DEBUG1 + debug_level;
2500ae1f160dSDag-Erling Smørgrav 			}
2501ae1f160dSDag-Erling Smørgrav 			debug_level++;
25021e8db6e2SBrian Feldman 			break;
25031e8db6e2SBrian Feldman 		case '1':
250419261079SEd Maste 			fatal("SSH protocol v.1 is no longer supported");
25051e8db6e2SBrian Feldman 			break;
2506b15c8340SDag-Erling Smørgrav 		case '2':
250719261079SEd Maste 			/* accept silently */
25081e8db6e2SBrian Feldman 			break;
2509e4a9863fSDag-Erling Smørgrav 		case 'a':
2510e4a9863fSDag-Erling Smørgrav 			global_aflag = 1;
2511e4a9863fSDag-Erling Smørgrav 			break;
2512b15c8340SDag-Erling Smørgrav 		case 'B':
2513b15c8340SDag-Erling Smørgrav 			copy_buffer_len = strtol(optarg, &cp, 10);
2514b15c8340SDag-Erling Smørgrav 			if (copy_buffer_len == 0 || *cp != '\0')
2515b15c8340SDag-Erling Smørgrav 				fatal("Invalid buffer size \"%s\"", optarg);
25161e8db6e2SBrian Feldman 			break;
25171e8db6e2SBrian Feldman 		case 'b':
2518efcad6b7SDag-Erling Smørgrav 			if (batchmode)
2519efcad6b7SDag-Erling Smørgrav 				fatal("Batch file already specified.");
2520efcad6b7SDag-Erling Smørgrav 
2521efcad6b7SDag-Erling Smørgrav 			/* Allow "-" as stdin */
2522efcad6b7SDag-Erling Smørgrav 			if (strcmp(optarg, "-") != 0 &&
2523efcad6b7SDag-Erling Smørgrav 			    (infile = fopen(optarg, "r")) == NULL)
25241e8db6e2SBrian Feldman 				fatal("%s (%s).", strerror(errno), optarg);
2525d0c8c0bcSDag-Erling Smørgrav 			showprogress = 0;
2526e4a9863fSDag-Erling Smørgrav 			quiet = batchmode = 1;
25275e8dbd04SDag-Erling Smørgrav 			addargs(&args, "-obatchmode yes");
25281e8db6e2SBrian Feldman 			break;
2529f7167e0eSDag-Erling Smørgrav 		case 'f':
2530f7167e0eSDag-Erling Smørgrav 			global_fflag = 1;
2531f7167e0eSDag-Erling Smørgrav 			break;
253219261079SEd Maste 		case 'N':
253319261079SEd Maste 			noisy = 1; /* Used to clear quiet mode after getopt */
253419261079SEd Maste 			break;
2535b15c8340SDag-Erling Smørgrav 		case 'p':
2536b15c8340SDag-Erling Smørgrav 			global_pflag = 1;
2537b15c8340SDag-Erling Smørgrav 			break;
2538b15c8340SDag-Erling Smørgrav 		case 'D':
2539ae1f160dSDag-Erling Smørgrav 			sftp_direct = optarg;
2540ae1f160dSDag-Erling Smørgrav 			break;
25414a421b63SDag-Erling Smørgrav 		case 'l':
25424a421b63SDag-Erling Smørgrav 			limit_kbps = strtonum(optarg, 1, 100 * 1024 * 1024,
25434a421b63SDag-Erling Smørgrav 			    &errstr);
25444a421b63SDag-Erling Smørgrav 			if (errstr != NULL)
25454a421b63SDag-Erling Smørgrav 				usage();
25464a421b63SDag-Erling Smørgrav 			limit_kbps *= 1024; /* kbps */
25474a421b63SDag-Erling Smørgrav 			break;
2548b15c8340SDag-Erling Smørgrav 		case 'r':
2549b15c8340SDag-Erling Smørgrav 			global_rflag = 1;
2550ae1f160dSDag-Erling Smørgrav 			break;
2551ae1f160dSDag-Erling Smørgrav 		case 'R':
2552ae1f160dSDag-Erling Smørgrav 			num_requests = strtol(optarg, &cp, 10);
2553ae1f160dSDag-Erling Smørgrav 			if (num_requests == 0 || *cp != '\0')
2554ae1f160dSDag-Erling Smørgrav 				fatal("Invalid number of requests \"%s\"",
2555ae1f160dSDag-Erling Smørgrav 				    optarg);
2556ae1f160dSDag-Erling Smørgrav 			break;
2557b15c8340SDag-Erling Smørgrav 		case 's':
2558b15c8340SDag-Erling Smørgrav 			sftp_server = optarg;
2559b15c8340SDag-Erling Smørgrav 			break;
2560b15c8340SDag-Erling Smørgrav 		case 'S':
2561b15c8340SDag-Erling Smørgrav 			ssh_program = optarg;
2562b15c8340SDag-Erling Smørgrav 			replacearg(&args, 0, "%s", ssh_program);
2563b15c8340SDag-Erling Smørgrav 			break;
2564f374ba41SEd Maste 		case 'X':
2565f374ba41SEd Maste 			/* Please keep in sync with ssh.c -X */
2566f374ba41SEd Maste 			if (strncmp(optarg, "buffer=", 7) == 0) {
2567f374ba41SEd Maste 				r = scan_scaled(optarg + 7, &llv);
2568f374ba41SEd Maste 				if (r == 0 && (llv <= 0 || llv > 256 * 1024)) {
2569f374ba41SEd Maste 					r = -1;
2570f374ba41SEd Maste 					errno = EINVAL;
2571f374ba41SEd Maste 				}
2572f374ba41SEd Maste 				if (r == -1) {
2573f374ba41SEd Maste 					fatal("Invalid buffer size \"%s\": %s",
2574f374ba41SEd Maste 					     optarg + 7, strerror(errno));
2575f374ba41SEd Maste 				}
2576f374ba41SEd Maste 				copy_buffer_len = (size_t)llv;
2577f374ba41SEd Maste 			} else if (strncmp(optarg, "nrequests=", 10) == 0) {
2578f374ba41SEd Maste 				llv = strtonum(optarg + 10, 1, 256 * 1024,
2579f374ba41SEd Maste 				    &errstr);
2580f374ba41SEd Maste 				if (errstr != NULL) {
2581f374ba41SEd Maste 					fatal("Invalid number of requests "
2582f374ba41SEd Maste 					    "\"%s\": %s", optarg + 10, errstr);
2583f374ba41SEd Maste 				}
2584f374ba41SEd Maste 				num_requests = (size_t)llv;
2585f374ba41SEd Maste 			} else {
2586f374ba41SEd Maste 				fatal("Invalid -X option");
2587f374ba41SEd Maste 			}
2588f374ba41SEd Maste 			break;
25891e8db6e2SBrian Feldman 		case 'h':
25901e8db6e2SBrian Feldman 		default:
25911e8db6e2SBrian Feldman 			usage();
25921e8db6e2SBrian Feldman 		}
25931e8db6e2SBrian Feldman 	}
25941e8db6e2SBrian Feldman 
259519261079SEd Maste 	/* Do this last because we want the user to be able to override it */
259619261079SEd Maste 	addargs(&args, "-oForwardAgent no");
259719261079SEd Maste 
259852028650SDag-Erling Smørgrav 	if (!isatty(STDERR_FILENO))
259952028650SDag-Erling Smørgrav 		showprogress = 0;
260052028650SDag-Erling Smørgrav 
260119261079SEd Maste 	if (noisy)
260219261079SEd Maste 		quiet = 0;
260319261079SEd Maste 
2604545d5ecaSDag-Erling Smørgrav 	log_init(argv[0], ll, SYSLOG_FACILITY_USER, 1);
2605545d5ecaSDag-Erling Smørgrav 
2606ae1f160dSDag-Erling Smørgrav 	if (sftp_direct == NULL) {
26071e8db6e2SBrian Feldman 		if (optind == argc || argc > (optind + 2))
26081e8db6e2SBrian Feldman 			usage();
260947dd1d1bSDag-Erling Smørgrav 		argv += optind;
26101e8db6e2SBrian Feldman 
261147dd1d1bSDag-Erling Smørgrav 		switch (parse_uri("sftp", *argv, &user, &host, &tmp, &file1)) {
261247dd1d1bSDag-Erling Smørgrav 		case -1:
26131e8db6e2SBrian Feldman 			usage();
261447dd1d1bSDag-Erling Smørgrav 			break;
261547dd1d1bSDag-Erling Smørgrav 		case 0:
261647dd1d1bSDag-Erling Smørgrav 			if (tmp != -1)
261747dd1d1bSDag-Erling Smørgrav 				port = tmp;
261847dd1d1bSDag-Erling Smørgrav 			break;
261947dd1d1bSDag-Erling Smørgrav 		default:
262019261079SEd Maste 			/* Try with user, host and path. */
262147dd1d1bSDag-Erling Smørgrav 			if (parse_user_host_path(*argv, &user, &host,
262219261079SEd Maste 			    &file1) == 0)
262319261079SEd Maste 				break;
262419261079SEd Maste 			/* Try with user and host. */
262519261079SEd Maste 			if (parse_user_host_port(*argv, &user, &host, NULL)
262619261079SEd Maste 			    == 0)
262719261079SEd Maste 				break;
262847dd1d1bSDag-Erling Smørgrav 			/* Treat as a plain hostname. */
262947dd1d1bSDag-Erling Smørgrav 			host = xstrdup(*argv);
26301e8db6e2SBrian Feldman 			host = cleanhostname(host);
263147dd1d1bSDag-Erling Smørgrav 			break;
263247dd1d1bSDag-Erling Smørgrav 		}
263347dd1d1bSDag-Erling Smørgrav 		file2 = *(argv + 1);
263447dd1d1bSDag-Erling Smørgrav 
26351e8db6e2SBrian Feldman 		if (!*host) {
26361e8db6e2SBrian Feldman 			fprintf(stderr, "Missing hostname\n");
26371e8db6e2SBrian Feldman 			usage();
26381e8db6e2SBrian Feldman 		}
26391e8db6e2SBrian Feldman 
264047dd1d1bSDag-Erling Smørgrav 		if (port != -1)
264147dd1d1bSDag-Erling Smørgrav 			addargs(&args, "-oPort %d", port);
264247dd1d1bSDag-Erling Smørgrav 		if (user != NULL) {
264347dd1d1bSDag-Erling Smørgrav 			addargs(&args, "-l");
264447dd1d1bSDag-Erling Smørgrav 			addargs(&args, "%s", user);
264547dd1d1bSDag-Erling Smørgrav 		}
26461e8db6e2SBrian Feldman 
2647ae1f160dSDag-Erling Smørgrav 		/* no subsystem if the server-spec contains a '/' */
2648ae1f160dSDag-Erling Smørgrav 		if (sftp_server == NULL || strchr(sftp_server, '/') == NULL)
2649ae1f160dSDag-Erling Smørgrav 			addargs(&args, "-s");
2650ae1f160dSDag-Erling Smørgrav 
2651b15c8340SDag-Erling Smørgrav 		addargs(&args, "--");
2652ae1f160dSDag-Erling Smørgrav 		addargs(&args, "%s", host);
2653ae1f160dSDag-Erling Smørgrav 		addargs(&args, "%s", (sftp_server != NULL ?
2654ae1f160dSDag-Erling Smørgrav 		    sftp_server : "sftp"));
26551e8db6e2SBrian Feldman 
2656d95e11bfSDag-Erling Smørgrav 		connect_to_server(ssh_program, args.list, &in, &out);
2657ae1f160dSDag-Erling Smørgrav 	} else {
265838a52bd3SEd Maste 		if ((r = argv_split(sftp_direct, &tmp, &cpp, 1)) != 0)
265938a52bd3SEd Maste 			fatal_r(r, "Parse -D arguments");
266038a52bd3SEd Maste 		if (cpp[0] == 0)
266138a52bd3SEd Maste 			fatal("No sftp server specified via -D");
266238a52bd3SEd Maste 		connect_to_server(cpp[0], cpp, &in, &out);
266338a52bd3SEd Maste 		argv_free(cpp, tmp);
2664ae1f160dSDag-Erling Smørgrav 	}
2665021d409fSDag-Erling Smørgrav 	freeargs(&args);
26661e8db6e2SBrian Feldman 
2667*edf85781SEd Maste 	conn = sftp_init(in, out, copy_buffer_len, num_requests, limit_kbps);
2668b15c8340SDag-Erling Smørgrav 	if (conn == NULL)
2669b15c8340SDag-Erling Smørgrav 		fatal("Couldn't initialise connection to server");
2670b15c8340SDag-Erling Smørgrav 
2671e4a9863fSDag-Erling Smørgrav 	if (!quiet) {
2672b15c8340SDag-Erling Smørgrav 		if (sftp_direct == NULL)
2673b15c8340SDag-Erling Smørgrav 			fprintf(stderr, "Connected to %s.\n", host);
2674b15c8340SDag-Erling Smørgrav 		else
2675b15c8340SDag-Erling Smørgrav 			fprintf(stderr, "Attached to %s.\n", sftp_direct);
2676b15c8340SDag-Erling Smørgrav 	}
2677b15c8340SDag-Erling Smørgrav 
2678b15c8340SDag-Erling Smørgrav 	err = interactive_loop(conn, file1, file2);
26791e8db6e2SBrian Feldman 
268083d2307dSDag-Erling Smørgrav #if !defined(USE_PIPES)
268183d2307dSDag-Erling Smørgrav 	shutdown(in, SHUT_RDWR);
268283d2307dSDag-Erling Smørgrav 	shutdown(out, SHUT_RDWR);
268383d2307dSDag-Erling Smørgrav #endif
268483d2307dSDag-Erling Smørgrav 
26851e8db6e2SBrian Feldman 	close(in);
26861e8db6e2SBrian Feldman 	close(out);
2687efcad6b7SDag-Erling Smørgrav 	if (batchmode)
26881e8db6e2SBrian Feldman 		fclose(infile);
26891e8db6e2SBrian Feldman 
2690190cef3dSDag-Erling Smørgrav 	while (waitpid(sshpid, NULL, 0) == -1 && sshpid > 1)
2691545d5ecaSDag-Erling Smørgrav 		if (errno != EINTR)
2692545d5ecaSDag-Erling Smørgrav 			fatal("Couldn't wait for ssh process: %s",
2693545d5ecaSDag-Erling Smørgrav 			    strerror(errno));
26941e8db6e2SBrian Feldman 
2695d0c8c0bcSDag-Erling Smørgrav 	exit(err == 0 ? 0 : 1);
26961e8db6e2SBrian Feldman }
2697