1*0fdf8faeSEd Maste /* $OpenBSD: sftp.c,v 1.239 2024/06/26 23:14:14 deraadt 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
113edf85781SEd 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
killchild(int signo)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
suspchild(int signo)235ca86bcf2SDag-Erling Smørgrav suspchild(int signo)
236ca86bcf2SDag-Erling Smørgrav {
237*0fdf8faeSEd Maste int save_errno = errno;
238ca86bcf2SDag-Erling Smørgrav if (sshpid > 1) {
239ca86bcf2SDag-Erling Smørgrav kill(sshpid, signo);
240ca86bcf2SDag-Erling Smørgrav while (waitpid(sshpid, NULL, WUNTRACED) == -1 && errno == EINTR)
241ca86bcf2SDag-Erling Smørgrav continue;
242ca86bcf2SDag-Erling Smørgrav }
243ca86bcf2SDag-Erling Smørgrav kill(getpid(), SIGSTOP);
244*0fdf8faeSEd Maste errno = save_errno;
245ca86bcf2SDag-Erling Smørgrav }
246ca86bcf2SDag-Erling Smørgrav
247ca86bcf2SDag-Erling Smørgrav static void
cmd_interrupt(int signo)248d74d50a8SDag-Erling Smørgrav cmd_interrupt(int signo)
249d74d50a8SDag-Erling Smørgrav {
250d74d50a8SDag-Erling Smørgrav const char msg[] = "\rInterrupt \n";
2515e8dbd04SDag-Erling Smørgrav int olderrno = errno;
252d74d50a8SDag-Erling Smørgrav
253e4a9863fSDag-Erling Smørgrav (void)write(STDERR_FILENO, msg, sizeof(msg) - 1);
254d74d50a8SDag-Erling Smørgrav interrupted = 1;
2555e8dbd04SDag-Erling Smørgrav errno = olderrno;
256d74d50a8SDag-Erling Smørgrav }
257d74d50a8SDag-Erling Smørgrav
258190cef3dSDag-Erling Smørgrav static void
read_interrupt(int signo)25919261079SEd Maste read_interrupt(int signo)
26019261079SEd Maste {
26119261079SEd Maste interrupted = 1;
26219261079SEd Maste }
26319261079SEd Maste
26419261079SEd Maste static void
sigchld_handler(int sig)265190cef3dSDag-Erling Smørgrav sigchld_handler(int sig)
266190cef3dSDag-Erling Smørgrav {
267190cef3dSDag-Erling Smørgrav int save_errno = errno;
268190cef3dSDag-Erling Smørgrav pid_t pid;
269190cef3dSDag-Erling Smørgrav const char msg[] = "\rConnection closed. \n";
270190cef3dSDag-Erling Smørgrav
271190cef3dSDag-Erling Smørgrav /* Report if ssh transport process dies. */
272190cef3dSDag-Erling Smørgrav while ((pid = waitpid(sshpid, NULL, WNOHANG)) == -1 && errno == EINTR)
273190cef3dSDag-Erling Smørgrav continue;
274190cef3dSDag-Erling Smørgrav if (pid == sshpid) {
275f374ba41SEd Maste if (!quiet)
276190cef3dSDag-Erling Smørgrav (void)write(STDERR_FILENO, msg, sizeof(msg) - 1);
277190cef3dSDag-Erling Smørgrav sshpid = -1;
278190cef3dSDag-Erling Smørgrav }
279190cef3dSDag-Erling Smørgrav
280190cef3dSDag-Erling Smørgrav errno = save_errno;
281190cef3dSDag-Erling Smørgrav }
282190cef3dSDag-Erling Smørgrav
283d74d50a8SDag-Erling Smørgrav static void
help(void)284efcad6b7SDag-Erling Smørgrav help(void)
285efcad6b7SDag-Erling Smørgrav {
286cce7d346SDag-Erling Smørgrav printf("Available commands:\n"
287cce7d346SDag-Erling Smørgrav "bye Quit sftp\n"
288cce7d346SDag-Erling Smørgrav "cd path Change remote directory to 'path'\n"
28919261079SEd Maste "chgrp [-h] grp path Change group of file 'path' to 'grp'\n"
29019261079SEd Maste "chmod [-h] mode path Change permissions of file 'path' to 'mode'\n"
29119261079SEd Maste "chown [-h] own path Change owner of file 'path' to 'own'\n"
29287c1498dSEd Maste "copy oldpath newpath Copy remote file\n"
29387c1498dSEd Maste "cp oldpath newpath Copy remote file\n"
294cce7d346SDag-Erling Smørgrav "df [-hi] [path] Display statistics for current directory or\n"
295cce7d346SDag-Erling Smørgrav " filesystem containing 'path'\n"
296cce7d346SDag-Erling Smørgrav "exit Quit sftp\n"
29719261079SEd Maste "get [-afpR] remote [local] Download file\n"
298cce7d346SDag-Erling Smørgrav "help Display this help text\n"
299cce7d346SDag-Erling Smørgrav "lcd path Change local directory to 'path'\n"
300cce7d346SDag-Erling Smørgrav "lls [ls-options [path]] Display local directory listing\n"
301cce7d346SDag-Erling Smørgrav "lmkdir path Create local directory\n"
3024a421b63SDag-Erling Smørgrav "ln [-s] oldpath newpath Link remote file (-s for symlink)\n"
303cce7d346SDag-Erling Smørgrav "lpwd Print local working directory\n"
304b15c8340SDag-Erling Smørgrav "ls [-1afhlnrSt] [path] Display remote directory listing\n"
305cce7d346SDag-Erling Smørgrav "lumask umask Set local umask to 'umask'\n"
306cce7d346SDag-Erling Smørgrav "mkdir path Create remote directory\n"
307cce7d346SDag-Erling Smørgrav "progress Toggle display of progress meter\n"
30819261079SEd Maste "put [-afpR] local [remote] Upload file\n"
309cce7d346SDag-Erling Smørgrav "pwd Display remote working directory\n"
310cce7d346SDag-Erling Smørgrav "quit Quit sftp\n"
31119261079SEd Maste "reget [-fpR] remote [local] Resume download file\n"
312cce7d346SDag-Erling Smørgrav "rename oldpath newpath Rename remote file\n"
31319261079SEd Maste "reput [-fpR] local [remote] Resume upload file\n"
314cce7d346SDag-Erling Smørgrav "rm path Delete remote file\n"
315cce7d346SDag-Erling Smørgrav "rmdir path Remove remote directory\n"
316cce7d346SDag-Erling Smørgrav "symlink oldpath newpath Symlink remote file\n"
317cce7d346SDag-Erling Smørgrav "version Show SFTP version\n"
318cce7d346SDag-Erling Smørgrav "!command Execute 'command' in local shell\n"
319cce7d346SDag-Erling Smørgrav "! Escape to local shell\n"
320cce7d346SDag-Erling Smørgrav "? Synonym for help\n");
321efcad6b7SDag-Erling Smørgrav }
322efcad6b7SDag-Erling Smørgrav
323efcad6b7SDag-Erling Smørgrav static void
local_do_shell(const char * args)324efcad6b7SDag-Erling Smørgrav local_do_shell(const char *args)
325efcad6b7SDag-Erling Smørgrav {
326efcad6b7SDag-Erling Smørgrav int status;
327efcad6b7SDag-Erling Smørgrav char *shell;
328efcad6b7SDag-Erling Smørgrav pid_t pid;
329efcad6b7SDag-Erling Smørgrav
330efcad6b7SDag-Erling Smørgrav if (!*args)
331efcad6b7SDag-Erling Smørgrav args = NULL;
332efcad6b7SDag-Erling Smørgrav
3334a421b63SDag-Erling Smørgrav if ((shell = getenv("SHELL")) == NULL || *shell == '\0')
334efcad6b7SDag-Erling Smørgrav shell = _PATH_BSHELL;
335efcad6b7SDag-Erling Smørgrav
336efcad6b7SDag-Erling Smørgrav if ((pid = fork()) == -1)
337efcad6b7SDag-Erling Smørgrav fatal("Couldn't fork: %s", strerror(errno));
338efcad6b7SDag-Erling Smørgrav
339efcad6b7SDag-Erling Smørgrav if (pid == 0) {
340efcad6b7SDag-Erling Smørgrav /* XXX: child has pipe fds to ssh subproc open - issue? */
341efcad6b7SDag-Erling Smørgrav if (args) {
342efcad6b7SDag-Erling Smørgrav debug3("Executing %s -c \"%s\"", shell, args);
343efcad6b7SDag-Erling Smørgrav execl(shell, shell, "-c", args, (char *)NULL);
344efcad6b7SDag-Erling Smørgrav } else {
345efcad6b7SDag-Erling Smørgrav debug3("Executing %s", shell);
346efcad6b7SDag-Erling Smørgrav execl(shell, shell, (char *)NULL);
347efcad6b7SDag-Erling Smørgrav }
348efcad6b7SDag-Erling Smørgrav fprintf(stderr, "Couldn't execute \"%s\": %s\n", shell,
349efcad6b7SDag-Erling Smørgrav strerror(errno));
350efcad6b7SDag-Erling Smørgrav _exit(1);
351efcad6b7SDag-Erling Smørgrav }
352efcad6b7SDag-Erling Smørgrav while (waitpid(pid, &status, 0) == -1)
353efcad6b7SDag-Erling Smørgrav if (errno != EINTR)
354efcad6b7SDag-Erling Smørgrav fatal("Couldn't wait for child: %s", strerror(errno));
355efcad6b7SDag-Erling Smørgrav if (!WIFEXITED(status))
356761efaa7SDag-Erling Smørgrav error("Shell exited abnormally");
357efcad6b7SDag-Erling Smørgrav else if (WEXITSTATUS(status))
358efcad6b7SDag-Erling Smørgrav error("Shell exited with status %d", WEXITSTATUS(status));
359efcad6b7SDag-Erling Smørgrav }
360efcad6b7SDag-Erling Smørgrav
361efcad6b7SDag-Erling Smørgrav static void
local_do_ls(const char * args)362efcad6b7SDag-Erling Smørgrav local_do_ls(const char *args)
363efcad6b7SDag-Erling Smørgrav {
364efcad6b7SDag-Erling Smørgrav if (!args || !*args)
365efcad6b7SDag-Erling Smørgrav local_do_shell(_PATH_LS);
366efcad6b7SDag-Erling Smørgrav else {
367efcad6b7SDag-Erling Smørgrav int len = strlen(_PATH_LS " ") + strlen(args) + 1;
368efcad6b7SDag-Erling Smørgrav char *buf = xmalloc(len);
369efcad6b7SDag-Erling Smørgrav
370efcad6b7SDag-Erling Smørgrav /* XXX: quoting - rip quoting code from ftp? */
371efcad6b7SDag-Erling Smørgrav snprintf(buf, len, _PATH_LS " %s", args);
372efcad6b7SDag-Erling Smørgrav local_do_shell(buf);
373e4a9863fSDag-Erling Smørgrav free(buf);
374efcad6b7SDag-Erling Smørgrav }
375efcad6b7SDag-Erling Smørgrav }
376efcad6b7SDag-Erling Smørgrav
377efcad6b7SDag-Erling Smørgrav /* Strip one path (usually the pwd) from the start of another */
378efcad6b7SDag-Erling Smørgrav static char *
path_strip(const char * path,const char * strip)379076ad2f8SDag-Erling Smørgrav path_strip(const char *path, const char *strip)
380efcad6b7SDag-Erling Smørgrav {
381efcad6b7SDag-Erling Smørgrav size_t len;
382efcad6b7SDag-Erling Smørgrav
383efcad6b7SDag-Erling Smørgrav if (strip == NULL)
384efcad6b7SDag-Erling Smørgrav return (xstrdup(path));
385efcad6b7SDag-Erling Smørgrav
386efcad6b7SDag-Erling Smørgrav len = strlen(strip);
3875e8dbd04SDag-Erling Smørgrav if (strncmp(path, strip, len) == 0) {
388efcad6b7SDag-Erling Smørgrav if (strip[len - 1] != '/' && path[len] == '/')
389efcad6b7SDag-Erling Smørgrav len++;
390efcad6b7SDag-Erling Smørgrav return (xstrdup(path + len));
391efcad6b7SDag-Erling Smørgrav }
392efcad6b7SDag-Erling Smørgrav
393efcad6b7SDag-Erling Smørgrav return (xstrdup(path));
394efcad6b7SDag-Erling Smørgrav }
395efcad6b7SDag-Erling Smørgrav
396efcad6b7SDag-Erling Smørgrav static int
parse_getput_flags(const char * cmd,char ** argv,int argc,int * aflag,int * fflag,int * pflag,int * rflag)397e4a9863fSDag-Erling Smørgrav parse_getput_flags(const char *cmd, char **argv, int argc,
398f7167e0eSDag-Erling Smørgrav int *aflag, int *fflag, int *pflag, int *rflag)
399efcad6b7SDag-Erling Smørgrav {
400d4af9e69SDag-Erling Smørgrav extern int opterr, optind, optopt, optreset;
401d4af9e69SDag-Erling Smørgrav int ch;
402efcad6b7SDag-Erling Smørgrav
403d4af9e69SDag-Erling Smørgrav optind = optreset = 1;
404d4af9e69SDag-Erling Smørgrav opterr = 0;
405d4af9e69SDag-Erling Smørgrav
406f7167e0eSDag-Erling Smørgrav *aflag = *fflag = *rflag = *pflag = 0;
407f7167e0eSDag-Erling Smørgrav while ((ch = getopt(argc, argv, "afPpRr")) != -1) {
408d4af9e69SDag-Erling Smørgrav switch (ch) {
409e4a9863fSDag-Erling Smørgrav case 'a':
410e4a9863fSDag-Erling Smørgrav *aflag = 1;
411e4a9863fSDag-Erling Smørgrav break;
412f7167e0eSDag-Erling Smørgrav case 'f':
413f7167e0eSDag-Erling Smørgrav *fflag = 1;
414f7167e0eSDag-Erling Smørgrav break;
415efcad6b7SDag-Erling Smørgrav case 'p':
416efcad6b7SDag-Erling Smørgrav case 'P':
417efcad6b7SDag-Erling Smørgrav *pflag = 1;
418efcad6b7SDag-Erling Smørgrav break;
419b15c8340SDag-Erling Smørgrav case 'r':
420b15c8340SDag-Erling Smørgrav case 'R':
421b15c8340SDag-Erling Smørgrav *rflag = 1;
422b15c8340SDag-Erling Smørgrav break;
423efcad6b7SDag-Erling Smørgrav default:
424d4af9e69SDag-Erling Smørgrav error("%s: Invalid flag -%c", cmd, optopt);
425d4af9e69SDag-Erling Smørgrav return -1;
426efcad6b7SDag-Erling Smørgrav }
427efcad6b7SDag-Erling Smørgrav }
428efcad6b7SDag-Erling Smørgrav
429d4af9e69SDag-Erling Smørgrav return optind;
430efcad6b7SDag-Erling Smørgrav }
431efcad6b7SDag-Erling Smørgrav
432efcad6b7SDag-Erling Smørgrav static int
parse_link_flags(const char * cmd,char ** argv,int argc,int * sflag)4334a421b63SDag-Erling Smørgrav parse_link_flags(const char *cmd, char **argv, int argc, int *sflag)
4344a421b63SDag-Erling Smørgrav {
4354a421b63SDag-Erling Smørgrav extern int opterr, optind, optopt, optreset;
4364a421b63SDag-Erling Smørgrav int ch;
4374a421b63SDag-Erling Smørgrav
4384a421b63SDag-Erling Smørgrav optind = optreset = 1;
4394a421b63SDag-Erling Smørgrav opterr = 0;
4404a421b63SDag-Erling Smørgrav
4414a421b63SDag-Erling Smørgrav *sflag = 0;
4424a421b63SDag-Erling Smørgrav while ((ch = getopt(argc, argv, "s")) != -1) {
4434a421b63SDag-Erling Smørgrav switch (ch) {
4444a421b63SDag-Erling Smørgrav case 's':
4454a421b63SDag-Erling Smørgrav *sflag = 1;
4464a421b63SDag-Erling Smørgrav break;
4474a421b63SDag-Erling Smørgrav default:
4484a421b63SDag-Erling Smørgrav error("%s: Invalid flag -%c", cmd, optopt);
4494a421b63SDag-Erling Smørgrav return -1;
4504a421b63SDag-Erling Smørgrav }
4514a421b63SDag-Erling Smørgrav }
4524a421b63SDag-Erling Smørgrav
4534a421b63SDag-Erling Smørgrav return optind;
4544a421b63SDag-Erling Smørgrav }
4554a421b63SDag-Erling Smørgrav
4564a421b63SDag-Erling Smørgrav static int
parse_rename_flags(const char * cmd,char ** argv,int argc,int * lflag)457f7167e0eSDag-Erling Smørgrav parse_rename_flags(const char *cmd, char **argv, int argc, int *lflag)
458f7167e0eSDag-Erling Smørgrav {
459f7167e0eSDag-Erling Smørgrav extern int opterr, optind, optopt, optreset;
460f7167e0eSDag-Erling Smørgrav int ch;
461f7167e0eSDag-Erling Smørgrav
462f7167e0eSDag-Erling Smørgrav optind = optreset = 1;
463f7167e0eSDag-Erling Smørgrav opterr = 0;
464f7167e0eSDag-Erling Smørgrav
465f7167e0eSDag-Erling Smørgrav *lflag = 0;
466f7167e0eSDag-Erling Smørgrav while ((ch = getopt(argc, argv, "l")) != -1) {
467f7167e0eSDag-Erling Smørgrav switch (ch) {
468f7167e0eSDag-Erling Smørgrav case 'l':
469f7167e0eSDag-Erling Smørgrav *lflag = 1;
470f7167e0eSDag-Erling Smørgrav break;
471f7167e0eSDag-Erling Smørgrav default:
472f7167e0eSDag-Erling Smørgrav error("%s: Invalid flag -%c", cmd, optopt);
473f7167e0eSDag-Erling Smørgrav return -1;
474f7167e0eSDag-Erling Smørgrav }
475f7167e0eSDag-Erling Smørgrav }
476f7167e0eSDag-Erling Smørgrav
477f7167e0eSDag-Erling Smørgrav return optind;
478f7167e0eSDag-Erling Smørgrav }
479f7167e0eSDag-Erling Smørgrav
480f7167e0eSDag-Erling Smørgrav static int
parse_ls_flags(char ** argv,int argc,int * lflag)481d4af9e69SDag-Erling Smørgrav parse_ls_flags(char **argv, int argc, int *lflag)
482efcad6b7SDag-Erling Smørgrav {
483d4af9e69SDag-Erling Smørgrav extern int opterr, optind, optopt, optreset;
484d4af9e69SDag-Erling Smørgrav int ch;
485efcad6b7SDag-Erling Smørgrav
486d4af9e69SDag-Erling Smørgrav optind = optreset = 1;
487d4af9e69SDag-Erling Smørgrav opterr = 0;
488d4af9e69SDag-Erling Smørgrav
489d74d50a8SDag-Erling Smørgrav *lflag = LS_NAME_SORT;
490b15c8340SDag-Erling Smørgrav while ((ch = getopt(argc, argv, "1Safhlnrt")) != -1) {
491d4af9e69SDag-Erling Smørgrav switch (ch) {
492efcad6b7SDag-Erling Smørgrav case '1':
493d74d50a8SDag-Erling Smørgrav *lflag &= ~VIEW_FLAGS;
494d74d50a8SDag-Erling Smørgrav *lflag |= LS_SHORT_VIEW;
495d74d50a8SDag-Erling Smørgrav break;
496d74d50a8SDag-Erling Smørgrav case 'S':
497d74d50a8SDag-Erling Smørgrav *lflag &= ~SORT_FLAGS;
498d74d50a8SDag-Erling Smørgrav *lflag |= LS_SIZE_SORT;
499d74d50a8SDag-Erling Smørgrav break;
500d4af9e69SDag-Erling Smørgrav case 'a':
501d4af9e69SDag-Erling Smørgrav *lflag |= LS_SHOW_ALL;
502d74d50a8SDag-Erling Smørgrav break;
503d74d50a8SDag-Erling Smørgrav case 'f':
504d74d50a8SDag-Erling Smørgrav *lflag &= ~SORT_FLAGS;
505d74d50a8SDag-Erling Smørgrav break;
506b15c8340SDag-Erling Smørgrav case 'h':
507b15c8340SDag-Erling Smørgrav *lflag |= LS_SI_UNITS;
508b15c8340SDag-Erling Smørgrav break;
509d4af9e69SDag-Erling Smørgrav case 'l':
510b15c8340SDag-Erling Smørgrav *lflag &= ~LS_SHORT_VIEW;
511d4af9e69SDag-Erling Smørgrav *lflag |= LS_LONG_VIEW;
512d4af9e69SDag-Erling Smørgrav break;
513d4af9e69SDag-Erling Smørgrav case 'n':
514b15c8340SDag-Erling Smørgrav *lflag &= ~LS_SHORT_VIEW;
515d4af9e69SDag-Erling Smørgrav *lflag |= LS_NUMERIC_VIEW|LS_LONG_VIEW;
516d4af9e69SDag-Erling Smørgrav break;
517d4af9e69SDag-Erling Smørgrav case 'r':
518d4af9e69SDag-Erling Smørgrav *lflag |= LS_REVERSE_SORT;
519d4af9e69SDag-Erling Smørgrav break;
520d4af9e69SDag-Erling Smørgrav case 't':
521d4af9e69SDag-Erling Smørgrav *lflag &= ~SORT_FLAGS;
522d4af9e69SDag-Erling Smørgrav *lflag |= LS_TIME_SORT;
523efcad6b7SDag-Erling Smørgrav break;
524efcad6b7SDag-Erling Smørgrav default:
525d4af9e69SDag-Erling Smørgrav error("ls: Invalid flag -%c", optopt);
526d4af9e69SDag-Erling Smørgrav return -1;
527efcad6b7SDag-Erling Smørgrav }
528efcad6b7SDag-Erling Smørgrav }
529efcad6b7SDag-Erling Smørgrav
530d4af9e69SDag-Erling Smørgrav return optind;
531efcad6b7SDag-Erling Smørgrav }
532efcad6b7SDag-Erling Smørgrav
533efcad6b7SDag-Erling Smørgrav static int
parse_df_flags(const char * cmd,char ** argv,int argc,int * hflag,int * iflag)534d4af9e69SDag-Erling Smørgrav parse_df_flags(const char *cmd, char **argv, int argc, int *hflag, int *iflag)
535efcad6b7SDag-Erling Smørgrav {
536d4af9e69SDag-Erling Smørgrav extern int opterr, optind, optopt, optreset;
537d4af9e69SDag-Erling Smørgrav int ch;
538efcad6b7SDag-Erling Smørgrav
539d4af9e69SDag-Erling Smørgrav optind = optreset = 1;
540d4af9e69SDag-Erling Smørgrav opterr = 0;
541efcad6b7SDag-Erling Smørgrav
542d4af9e69SDag-Erling Smørgrav *hflag = *iflag = 0;
543d4af9e69SDag-Erling Smørgrav while ((ch = getopt(argc, argv, "hi")) != -1) {
544d4af9e69SDag-Erling Smørgrav switch (ch) {
545d4af9e69SDag-Erling Smørgrav case 'h':
546d4af9e69SDag-Erling Smørgrav *hflag = 1;
547efcad6b7SDag-Erling Smørgrav break;
548d4af9e69SDag-Erling Smørgrav case 'i':
549d4af9e69SDag-Erling Smørgrav *iflag = 1;
550d4af9e69SDag-Erling Smørgrav break;
551d4af9e69SDag-Erling Smørgrav default:
552d4af9e69SDag-Erling Smørgrav error("%s: Invalid flag -%c", cmd, optopt);
553d4af9e69SDag-Erling Smørgrav return -1;
554efcad6b7SDag-Erling Smørgrav }
555efcad6b7SDag-Erling Smørgrav }
556efcad6b7SDag-Erling Smørgrav
557d4af9e69SDag-Erling Smørgrav return optind;
558efcad6b7SDag-Erling Smørgrav }
559efcad6b7SDag-Erling Smørgrav
560efcad6b7SDag-Erling Smørgrav static int
parse_ch_flags(const char * cmd,char ** argv,int argc,int * hflag)56119261079SEd Maste parse_ch_flags(const char *cmd, char **argv, int argc, int *hflag)
56219261079SEd Maste {
56319261079SEd Maste extern int opterr, optind, optopt, optreset;
56419261079SEd Maste int ch;
56519261079SEd Maste
56619261079SEd Maste optind = optreset = 1;
56719261079SEd Maste opterr = 0;
56819261079SEd Maste
56919261079SEd Maste *hflag = 0;
57019261079SEd Maste while ((ch = getopt(argc, argv, "h")) != -1) {
57119261079SEd Maste switch (ch) {
57219261079SEd Maste case 'h':
57319261079SEd Maste *hflag = 1;
57419261079SEd Maste break;
57519261079SEd Maste default:
57619261079SEd Maste error("%s: Invalid flag -%c", cmd, optopt);
57719261079SEd Maste return -1;
57819261079SEd Maste }
57919261079SEd Maste }
58019261079SEd Maste
58119261079SEd Maste return optind;
58219261079SEd Maste }
58319261079SEd Maste
58419261079SEd Maste static int
parse_no_flags(const char * cmd,char ** argv,int argc)585f7167e0eSDag-Erling Smørgrav parse_no_flags(const char *cmd, char **argv, int argc)
586f7167e0eSDag-Erling Smørgrav {
587f7167e0eSDag-Erling Smørgrav extern int opterr, optind, optopt, optreset;
588f7167e0eSDag-Erling Smørgrav int ch;
589f7167e0eSDag-Erling Smørgrav
590f7167e0eSDag-Erling Smørgrav optind = optreset = 1;
591f7167e0eSDag-Erling Smørgrav opterr = 0;
592f7167e0eSDag-Erling Smørgrav
593f7167e0eSDag-Erling Smørgrav while ((ch = getopt(argc, argv, "")) != -1) {
594f7167e0eSDag-Erling Smørgrav switch (ch) {
595f7167e0eSDag-Erling Smørgrav default:
596f7167e0eSDag-Erling Smørgrav error("%s: Invalid flag -%c", cmd, optopt);
597f7167e0eSDag-Erling Smørgrav return -1;
598f7167e0eSDag-Erling Smørgrav }
599f7167e0eSDag-Erling Smørgrav }
600f7167e0eSDag-Erling Smørgrav
601f7167e0eSDag-Erling Smørgrav return optind;
602f7167e0eSDag-Erling Smørgrav }
603f7167e0eSDag-Erling Smørgrav
60438a52bd3SEd Maste static char *
escape_glob(const char * s)60538a52bd3SEd Maste escape_glob(const char *s)
60638a52bd3SEd Maste {
60738a52bd3SEd Maste size_t i, o, len;
60838a52bd3SEd Maste char *ret;
60938a52bd3SEd Maste
61038a52bd3SEd Maste len = strlen(s);
61138a52bd3SEd Maste ret = xcalloc(2, len + 1);
61238a52bd3SEd Maste for (i = o = 0; i < len; i++) {
61338a52bd3SEd Maste if (strchr("[]?*\\", s[i]) != NULL)
61438a52bd3SEd Maste ret[o++] = '\\';
61538a52bd3SEd Maste ret[o++] = s[i];
61638a52bd3SEd Maste }
61738a52bd3SEd Maste ret[o++] = '\0';
61838a52bd3SEd Maste return ret;
61938a52bd3SEd Maste }
62038a52bd3SEd Maste
621535af610SEd Maste /*
622535af610SEd Maste * Arg p must be dynamically allocated. make_absolute will either return it
623535af610SEd Maste * or free it and allocate a new one. Caller must free returned string.
624535af610SEd Maste */
62538a52bd3SEd Maste static char *
make_absolute_pwd_glob(char * p,const char * pwd)62669c72a57SEd Maste make_absolute_pwd_glob(char *p, const char *pwd)
62738a52bd3SEd Maste {
62838a52bd3SEd Maste char *ret, *escpwd;
62938a52bd3SEd Maste
63038a52bd3SEd Maste escpwd = escape_glob(pwd);
63138a52bd3SEd Maste if (p == NULL)
63238a52bd3SEd Maste return escpwd;
633edf85781SEd Maste ret = sftp_make_absolute(p, escpwd);
63438a52bd3SEd Maste free(escpwd);
63538a52bd3SEd Maste return ret;
63638a52bd3SEd Maste }
63738a52bd3SEd Maste
638f7167e0eSDag-Erling Smørgrav static int
local_is_dir(const char * path)639edf85781SEd Maste local_is_dir(const char *path)
640edf85781SEd Maste {
641edf85781SEd Maste struct stat sb;
642edf85781SEd Maste
643edf85781SEd Maste if (stat(path, &sb) == -1)
644edf85781SEd Maste return 0;
645edf85781SEd Maste return S_ISDIR(sb.st_mode);
646edf85781SEd Maste }
647edf85781SEd Maste
648edf85781SEd Maste static int
process_get(struct sftp_conn * conn,const char * src,const char * dst,const char * pwd,int pflag,int rflag,int resume,int fflag)649076ad2f8SDag-Erling Smørgrav process_get(struct sftp_conn *conn, const char *src, const char *dst,
650076ad2f8SDag-Erling Smørgrav const char *pwd, int pflag, int rflag, int resume, int fflag)
651efcad6b7SDag-Erling Smørgrav {
65238a52bd3SEd Maste char *filename, *abs_src = NULL, *abs_dst = NULL, *tmp = NULL;
653efcad6b7SDag-Erling Smørgrav glob_t g;
654a0ee8cc6SDag-Erling Smørgrav int i, r, err = 0;
655efcad6b7SDag-Erling Smørgrav
65669c72a57SEd Maste abs_src = make_absolute_pwd_glob(xstrdup(src), pwd);
657efcad6b7SDag-Erling Smørgrav memset(&g, 0, sizeof(g));
658b15c8340SDag-Erling Smørgrav
659efcad6b7SDag-Erling Smørgrav debug3("Looking up %s", abs_src);
660edf85781SEd Maste if ((r = sftp_glob(conn, abs_src, GLOB_MARK, NULL, &g)) != 0) {
661a0ee8cc6SDag-Erling Smørgrav if (r == GLOB_NOSPACE) {
662a0ee8cc6SDag-Erling Smørgrav error("Too many matches for \"%s\".", abs_src);
663a0ee8cc6SDag-Erling Smørgrav } else {
664efcad6b7SDag-Erling Smørgrav error("File \"%s\" not found.", abs_src);
665a0ee8cc6SDag-Erling Smørgrav }
666efcad6b7SDag-Erling Smørgrav err = -1;
667efcad6b7SDag-Erling Smørgrav goto out;
668efcad6b7SDag-Erling Smørgrav }
669efcad6b7SDag-Erling Smørgrav
670b15c8340SDag-Erling Smørgrav /*
671b15c8340SDag-Erling Smørgrav * If multiple matches then dst must be a directory or
672b15c8340SDag-Erling Smørgrav * unspecified.
673b15c8340SDag-Erling Smørgrav */
67419261079SEd Maste if (g.gl_matchc > 1 && dst != NULL && !local_is_dir(dst)) {
675b15c8340SDag-Erling Smørgrav error("Multiple source paths, but destination "
676b15c8340SDag-Erling Smørgrav "\"%s\" is not a directory", dst);
677efcad6b7SDag-Erling Smørgrav err = -1;
678efcad6b7SDag-Erling Smørgrav goto out;
679efcad6b7SDag-Erling Smørgrav }
680efcad6b7SDag-Erling Smørgrav
681d74d50a8SDag-Erling Smørgrav for (i = 0; g.gl_pathv[i] && !interrupted; i++) {
682b15c8340SDag-Erling Smørgrav tmp = xstrdup(g.gl_pathv[i]);
683b15c8340SDag-Erling Smørgrav if ((filename = basename(tmp)) == NULL) {
684b15c8340SDag-Erling Smørgrav error("basename %s: %s", tmp, strerror(errno));
685e4a9863fSDag-Erling Smørgrav free(tmp);
686efcad6b7SDag-Erling Smørgrav err = -1;
687efcad6b7SDag-Erling Smørgrav goto out;
688efcad6b7SDag-Erling Smørgrav }
689efcad6b7SDag-Erling Smørgrav
690efcad6b7SDag-Erling Smørgrav if (g.gl_matchc == 1 && dst) {
69119261079SEd Maste if (local_is_dir(dst)) {
692edf85781SEd Maste abs_dst = sftp_path_append(dst, filename);
693b15c8340SDag-Erling Smørgrav } else {
694efcad6b7SDag-Erling Smørgrav abs_dst = xstrdup(dst);
695b15c8340SDag-Erling Smørgrav }
696efcad6b7SDag-Erling Smørgrav } else if (dst) {
697edf85781SEd Maste abs_dst = sftp_path_append(dst, filename);
698b15c8340SDag-Erling Smørgrav } else {
699b15c8340SDag-Erling Smørgrav abs_dst = xstrdup(filename);
700b15c8340SDag-Erling Smørgrav }
701e4a9863fSDag-Erling Smørgrav free(tmp);
702efcad6b7SDag-Erling Smørgrav
703e4a9863fSDag-Erling Smørgrav resume |= global_aflag;
704e4a9863fSDag-Erling Smørgrav if (!quiet && resume)
705076ad2f8SDag-Erling Smørgrav mprintf("Resuming %s to %s\n",
706076ad2f8SDag-Erling Smørgrav g.gl_pathv[i], abs_dst);
707e4a9863fSDag-Erling Smørgrav else if (!quiet && !resume)
708076ad2f8SDag-Erling Smørgrav mprintf("Fetching %s to %s\n",
709076ad2f8SDag-Erling Smørgrav g.gl_pathv[i], abs_dst);
71019261079SEd Maste /* XXX follow link flag */
711edf85781SEd Maste if (sftp_globpath_is_dir(g.gl_pathv[i]) &&
712edf85781SEd Maste (rflag || global_rflag)) {
713edf85781SEd Maste if (sftp_download_dir(conn, g.gl_pathv[i], abs_dst,
714edf85781SEd Maste NULL, pflag || global_pflag, 1, resume,
71538a52bd3SEd Maste fflag || global_fflag, 0, 0) == -1)
716efcad6b7SDag-Erling Smørgrav err = -1;
717b15c8340SDag-Erling Smørgrav } else {
718edf85781SEd Maste if (sftp_download(conn, g.gl_pathv[i], abs_dst, NULL,
719f7167e0eSDag-Erling Smørgrav pflag || global_pflag, resume,
72038a52bd3SEd Maste fflag || global_fflag, 0) == -1)
721b15c8340SDag-Erling Smørgrav err = -1;
722b15c8340SDag-Erling Smørgrav }
723e4a9863fSDag-Erling Smørgrav free(abs_dst);
724efcad6b7SDag-Erling Smørgrav abs_dst = NULL;
725efcad6b7SDag-Erling Smørgrav }
726efcad6b7SDag-Erling Smørgrav
727efcad6b7SDag-Erling Smørgrav out:
728e4a9863fSDag-Erling Smørgrav free(abs_src);
729efcad6b7SDag-Erling Smørgrav globfree(&g);
730efcad6b7SDag-Erling Smørgrav return(err);
731efcad6b7SDag-Erling Smørgrav }
732efcad6b7SDag-Erling Smørgrav
733efcad6b7SDag-Erling Smørgrav static int
process_put(struct sftp_conn * conn,const char * src,const char * dst,const char * pwd,int pflag,int rflag,int resume,int fflag)734076ad2f8SDag-Erling Smørgrav process_put(struct sftp_conn *conn, const char *src, const char *dst,
735076ad2f8SDag-Erling Smørgrav const char *pwd, int pflag, int rflag, int resume, int fflag)
736efcad6b7SDag-Erling Smørgrav {
737efcad6b7SDag-Erling Smørgrav char *tmp_dst = NULL;
738efcad6b7SDag-Erling Smørgrav char *abs_dst = NULL;
739b15c8340SDag-Erling Smørgrav char *tmp = NULL, *filename = NULL;
740efcad6b7SDag-Erling Smørgrav glob_t g;
741efcad6b7SDag-Erling Smørgrav int err = 0;
742b15c8340SDag-Erling Smørgrav int i, dst_is_dir = 1;
743d4af9e69SDag-Erling Smørgrav struct stat sb;
744efcad6b7SDag-Erling Smørgrav
745efcad6b7SDag-Erling Smørgrav if (dst) {
746efcad6b7SDag-Erling Smørgrav tmp_dst = xstrdup(dst);
747edf85781SEd Maste tmp_dst = sftp_make_absolute(tmp_dst, pwd);
748efcad6b7SDag-Erling Smørgrav }
749efcad6b7SDag-Erling Smørgrav
750efcad6b7SDag-Erling Smørgrav memset(&g, 0, sizeof(g));
751efcad6b7SDag-Erling Smørgrav debug3("Looking up %s", src);
752b15c8340SDag-Erling Smørgrav if (glob(src, GLOB_NOCHECK | GLOB_MARK, NULL, &g)) {
753efcad6b7SDag-Erling Smørgrav error("File \"%s\" not found.", src);
754efcad6b7SDag-Erling Smørgrav err = -1;
755efcad6b7SDag-Erling Smørgrav goto out;
756efcad6b7SDag-Erling Smørgrav }
757efcad6b7SDag-Erling Smørgrav
758b15c8340SDag-Erling Smørgrav /* If we aren't fetching to pwd then stash this status for later */
759b15c8340SDag-Erling Smørgrav if (tmp_dst != NULL)
760edf85781SEd Maste dst_is_dir = sftp_remote_is_dir(conn, tmp_dst);
761b15c8340SDag-Erling Smørgrav
762efcad6b7SDag-Erling Smørgrav /* If multiple matches, dst may be directory or unspecified */
763b15c8340SDag-Erling Smørgrav if (g.gl_matchc > 1 && tmp_dst && !dst_is_dir) {
764b15c8340SDag-Erling Smørgrav error("Multiple paths match, but destination "
765b15c8340SDag-Erling Smørgrav "\"%s\" is not a directory", tmp_dst);
766efcad6b7SDag-Erling Smørgrav err = -1;
767efcad6b7SDag-Erling Smørgrav goto out;
768efcad6b7SDag-Erling Smørgrav }
769efcad6b7SDag-Erling Smørgrav
770d74d50a8SDag-Erling Smørgrav for (i = 0; g.gl_pathv[i] && !interrupted; i++) {
771d4af9e69SDag-Erling Smørgrav if (stat(g.gl_pathv[i], &sb) == -1) {
772d4af9e69SDag-Erling Smørgrav err = -1;
773d4af9e69SDag-Erling Smørgrav error("stat %s: %s", g.gl_pathv[i], strerror(errno));
774d4af9e69SDag-Erling Smørgrav continue;
775d4af9e69SDag-Erling Smørgrav }
776d4af9e69SDag-Erling Smørgrav
777b15c8340SDag-Erling Smørgrav tmp = xstrdup(g.gl_pathv[i]);
778b15c8340SDag-Erling Smørgrav if ((filename = basename(tmp)) == NULL) {
779b15c8340SDag-Erling Smørgrav error("basename %s: %s", tmp, strerror(errno));
780e4a9863fSDag-Erling Smørgrav free(tmp);
781efcad6b7SDag-Erling Smørgrav err = -1;
782efcad6b7SDag-Erling Smørgrav goto out;
783efcad6b7SDag-Erling Smørgrav }
784efcad6b7SDag-Erling Smørgrav
785535af610SEd Maste free(abs_dst);
786535af610SEd Maste abs_dst = NULL;
787efcad6b7SDag-Erling Smørgrav if (g.gl_matchc == 1 && tmp_dst) {
788efcad6b7SDag-Erling Smørgrav /* If directory specified, append filename */
789b15c8340SDag-Erling Smørgrav if (dst_is_dir)
790edf85781SEd Maste abs_dst = sftp_path_append(tmp_dst, filename);
791b15c8340SDag-Erling Smørgrav else
792efcad6b7SDag-Erling Smørgrav abs_dst = xstrdup(tmp_dst);
793efcad6b7SDag-Erling Smørgrav } else if (tmp_dst) {
794edf85781SEd Maste abs_dst = sftp_path_append(tmp_dst, filename);
795b15c8340SDag-Erling Smørgrav } else {
796edf85781SEd Maste abs_dst = sftp_make_absolute(xstrdup(filename), pwd);
797b15c8340SDag-Erling Smørgrav }
798e4a9863fSDag-Erling Smørgrav free(tmp);
799efcad6b7SDag-Erling Smørgrav
800a0ee8cc6SDag-Erling Smørgrav resume |= global_aflag;
801a0ee8cc6SDag-Erling Smørgrav if (!quiet && resume)
802076ad2f8SDag-Erling Smørgrav mprintf("Resuming upload of %s to %s\n",
803076ad2f8SDag-Erling Smørgrav g.gl_pathv[i], abs_dst);
804a0ee8cc6SDag-Erling Smørgrav else if (!quiet && !resume)
805076ad2f8SDag-Erling Smørgrav mprintf("Uploading %s to %s\n",
806076ad2f8SDag-Erling Smørgrav g.gl_pathv[i], abs_dst);
80719261079SEd Maste /* XXX follow_link_flag */
808edf85781SEd Maste if (sftp_globpath_is_dir(g.gl_pathv[i]) &&
809edf85781SEd Maste (rflag || global_rflag)) {
810edf85781SEd Maste if (sftp_upload_dir(conn, g.gl_pathv[i], abs_dst,
811a0ee8cc6SDag-Erling Smørgrav pflag || global_pflag, 1, resume,
81238a52bd3SEd Maste fflag || global_fflag, 0, 0) == -1)
813efcad6b7SDag-Erling Smørgrav err = -1;
814b15c8340SDag-Erling Smørgrav } else {
815edf85781SEd Maste if (sftp_upload(conn, g.gl_pathv[i], abs_dst,
816a0ee8cc6SDag-Erling Smørgrav pflag || global_pflag, resume,
81738a52bd3SEd Maste fflag || global_fflag, 0) == -1)
818b15c8340SDag-Erling Smørgrav err = -1;
819b15c8340SDag-Erling Smørgrav }
820efcad6b7SDag-Erling Smørgrav }
821efcad6b7SDag-Erling Smørgrav
822efcad6b7SDag-Erling Smørgrav out:
823e4a9863fSDag-Erling Smørgrav free(abs_dst);
824e4a9863fSDag-Erling Smørgrav free(tmp_dst);
825efcad6b7SDag-Erling Smørgrav globfree(&g);
826efcad6b7SDag-Erling Smørgrav return(err);
827efcad6b7SDag-Erling Smørgrav }
828efcad6b7SDag-Erling Smørgrav
829efcad6b7SDag-Erling Smørgrav static int
sdirent_comp(const void * aa,const void * bb)830efcad6b7SDag-Erling Smørgrav sdirent_comp(const void *aa, const void *bb)
831efcad6b7SDag-Erling Smørgrav {
832efcad6b7SDag-Erling Smørgrav SFTP_DIRENT *a = *(SFTP_DIRENT **)aa;
833efcad6b7SDag-Erling Smørgrav SFTP_DIRENT *b = *(SFTP_DIRENT **)bb;
834d74d50a8SDag-Erling Smørgrav int rmul = sort_flag & LS_REVERSE_SORT ? -1 : 1;
835efcad6b7SDag-Erling Smørgrav
836d74d50a8SDag-Erling Smørgrav #define NCMP(a,b) (a == b ? 0 : (a < b ? 1 : -1))
837d74d50a8SDag-Erling Smørgrav if (sort_flag & LS_NAME_SORT)
838d74d50a8SDag-Erling Smørgrav return (rmul * strcmp(a->filename, b->filename));
839d74d50a8SDag-Erling Smørgrav else if (sort_flag & LS_TIME_SORT)
840d74d50a8SDag-Erling Smørgrav return (rmul * NCMP(a->a.mtime, b->a.mtime));
841d74d50a8SDag-Erling Smørgrav else if (sort_flag & LS_SIZE_SORT)
842d74d50a8SDag-Erling Smørgrav return (rmul * NCMP(a->a.size, b->a.size));
843d74d50a8SDag-Erling Smørgrav
844d74d50a8SDag-Erling Smørgrav fatal("Unknown ls sort type");
845efcad6b7SDag-Erling Smørgrav }
846efcad6b7SDag-Erling Smørgrav
847efcad6b7SDag-Erling Smørgrav /* sftp ls.1 replacement for directories */
848efcad6b7SDag-Erling Smørgrav static int
do_ls_dir(struct sftp_conn * conn,const char * path,const char * strip_path,int lflag)849076ad2f8SDag-Erling Smørgrav do_ls_dir(struct sftp_conn *conn, const char *path,
850076ad2f8SDag-Erling Smørgrav const char *strip_path, int lflag)
851efcad6b7SDag-Erling Smørgrav {
852043840dfSDag-Erling Smørgrav int n;
853043840dfSDag-Erling Smørgrav u_int c = 1, colspace = 0, columns = 1;
854efcad6b7SDag-Erling Smørgrav SFTP_DIRENT **d;
855efcad6b7SDag-Erling Smørgrav
856edf85781SEd Maste if ((n = sftp_readdir(conn, path, &d)) != 0)
857efcad6b7SDag-Erling Smørgrav return (n);
858efcad6b7SDag-Erling Smørgrav
859d74d50a8SDag-Erling Smørgrav if (!(lflag & LS_SHORT_VIEW)) {
860043840dfSDag-Erling Smørgrav u_int m = 0, width = 80;
861efcad6b7SDag-Erling Smørgrav struct winsize ws;
862efcad6b7SDag-Erling Smørgrav char *tmp;
863efcad6b7SDag-Erling Smørgrav
864efcad6b7SDag-Erling Smørgrav /* Count entries for sort and find longest filename */
865d74d50a8SDag-Erling Smørgrav for (n = 0; d[n] != NULL; n++) {
866d74d50a8SDag-Erling Smørgrav if (d[n]->filename[0] != '.' || (lflag & LS_SHOW_ALL))
867ca86bcf2SDag-Erling Smørgrav m = MAXIMUM(m, strlen(d[n]->filename));
868d74d50a8SDag-Erling Smørgrav }
869efcad6b7SDag-Erling Smørgrav
870efcad6b7SDag-Erling Smørgrav /* Add any subpath that also needs to be counted */
871efcad6b7SDag-Erling Smørgrav tmp = path_strip(path, strip_path);
872efcad6b7SDag-Erling Smørgrav m += strlen(tmp);
873e4a9863fSDag-Erling Smørgrav free(tmp);
874efcad6b7SDag-Erling Smørgrav
875efcad6b7SDag-Erling Smørgrav if (ioctl(fileno(stdin), TIOCGWINSZ, &ws) != -1)
876efcad6b7SDag-Erling Smørgrav width = ws.ws_col;
877efcad6b7SDag-Erling Smørgrav
878efcad6b7SDag-Erling Smørgrav columns = width / (m + 2);
879ca86bcf2SDag-Erling Smørgrav columns = MAXIMUM(columns, 1);
880efcad6b7SDag-Erling Smørgrav colspace = width / columns;
881ca86bcf2SDag-Erling Smørgrav colspace = MINIMUM(colspace, width);
882efcad6b7SDag-Erling Smørgrav }
883efcad6b7SDag-Erling Smørgrav
884d74d50a8SDag-Erling Smørgrav if (lflag & SORT_FLAGS) {
885021d409fSDag-Erling Smørgrav for (n = 0; d[n] != NULL; n++)
886021d409fSDag-Erling Smørgrav ; /* count entries */
887d74d50a8SDag-Erling Smørgrav sort_flag = lflag & (SORT_FLAGS|LS_REVERSE_SORT);
888efcad6b7SDag-Erling Smørgrav qsort(d, n, sizeof(*d), sdirent_comp);
889d74d50a8SDag-Erling Smørgrav }
890efcad6b7SDag-Erling Smørgrav
89138a52bd3SEd Maste get_remote_user_groups_from_dirents(conn, d);
892d74d50a8SDag-Erling Smørgrav for (n = 0; d[n] != NULL && !interrupted; n++) {
893efcad6b7SDag-Erling Smørgrav char *tmp, *fname;
894efcad6b7SDag-Erling Smørgrav
895d74d50a8SDag-Erling Smørgrav if (d[n]->filename[0] == '.' && !(lflag & LS_SHOW_ALL))
896d74d50a8SDag-Erling Smørgrav continue;
897d74d50a8SDag-Erling Smørgrav
898edf85781SEd Maste tmp = sftp_path_append(path, d[n]->filename);
899efcad6b7SDag-Erling Smørgrav fname = path_strip(tmp, strip_path);
900e4a9863fSDag-Erling Smørgrav free(tmp);
901efcad6b7SDag-Erling Smørgrav
902d74d50a8SDag-Erling Smørgrav if (lflag & LS_LONG_VIEW) {
90338a52bd3SEd Maste if ((lflag & (LS_NUMERIC_VIEW|LS_SI_UNITS)) != 0 ||
904edf85781SEd Maste sftp_can_get_users_groups_by_id(conn)) {
905efcad6b7SDag-Erling Smørgrav char *lname;
906efcad6b7SDag-Erling Smørgrav struct stat sb;
907efcad6b7SDag-Erling Smørgrav
908efcad6b7SDag-Erling Smørgrav memset(&sb, 0, sizeof(sb));
909efcad6b7SDag-Erling Smørgrav attrib_to_stat(&d[n]->a, &sb);
910b15c8340SDag-Erling Smørgrav lname = ls_file(fname, &sb, 1,
91138a52bd3SEd Maste (lflag & LS_SI_UNITS),
91238a52bd3SEd Maste ruser_name(sb.st_uid),
91338a52bd3SEd Maste rgroup_name(sb.st_gid));
914076ad2f8SDag-Erling Smørgrav mprintf("%s\n", lname);
915e4a9863fSDag-Erling Smørgrav free(lname);
916d74d50a8SDag-Erling Smørgrav } else
917076ad2f8SDag-Erling Smørgrav mprintf("%s\n", d[n]->longname);
918efcad6b7SDag-Erling Smørgrav } else {
919076ad2f8SDag-Erling Smørgrav mprintf("%-*s", colspace, fname);
920efcad6b7SDag-Erling Smørgrav if (c >= columns) {
921efcad6b7SDag-Erling Smørgrav printf("\n");
922efcad6b7SDag-Erling Smørgrav c = 1;
923efcad6b7SDag-Erling Smørgrav } else
924efcad6b7SDag-Erling Smørgrav c++;
925efcad6b7SDag-Erling Smørgrav }
926efcad6b7SDag-Erling Smørgrav
927e4a9863fSDag-Erling Smørgrav free(fname);
928efcad6b7SDag-Erling Smørgrav }
929efcad6b7SDag-Erling Smørgrav
930d74d50a8SDag-Erling Smørgrav if (!(lflag & LS_LONG_VIEW) && (c != 1))
931efcad6b7SDag-Erling Smørgrav printf("\n");
932efcad6b7SDag-Erling Smørgrav
933edf85781SEd Maste sftp_free_dirents(d);
934efcad6b7SDag-Erling Smørgrav return (0);
935efcad6b7SDag-Erling Smørgrav }
936efcad6b7SDag-Erling Smørgrav
9374f52dfbbSDag-Erling Smørgrav static int
sglob_comp(const void * aa,const void * bb)9384f52dfbbSDag-Erling Smørgrav sglob_comp(const void *aa, const void *bb)
9394f52dfbbSDag-Erling Smørgrav {
9404f52dfbbSDag-Erling Smørgrav u_int a = *(const u_int *)aa;
9414f52dfbbSDag-Erling Smørgrav u_int b = *(const u_int *)bb;
9424f52dfbbSDag-Erling Smørgrav const char *ap = sort_glob->gl_pathv[a];
9434f52dfbbSDag-Erling Smørgrav const char *bp = sort_glob->gl_pathv[b];
9444f52dfbbSDag-Erling Smørgrav const struct stat *as = sort_glob->gl_statv[a];
9454f52dfbbSDag-Erling Smørgrav const struct stat *bs = sort_glob->gl_statv[b];
9464f52dfbbSDag-Erling Smørgrav int rmul = sort_flag & LS_REVERSE_SORT ? -1 : 1;
9474f52dfbbSDag-Erling Smørgrav
9484f52dfbbSDag-Erling Smørgrav #define NCMP(a,b) (a == b ? 0 : (a < b ? 1 : -1))
9494f52dfbbSDag-Erling Smørgrav if (sort_flag & LS_NAME_SORT)
9504f52dfbbSDag-Erling Smørgrav return (rmul * strcmp(ap, bp));
9514f52dfbbSDag-Erling Smørgrav else if (sort_flag & LS_TIME_SORT) {
9524f52dfbbSDag-Erling Smørgrav #if defined(HAVE_STRUCT_STAT_ST_MTIM)
95319261079SEd Maste if (timespeccmp(&as->st_mtim, &bs->st_mtim, ==))
95419261079SEd Maste return 0;
95519261079SEd Maste return timespeccmp(&as->st_mtim, &bs->st_mtim, <) ?
95619261079SEd Maste rmul : -rmul;
9574f52dfbbSDag-Erling Smørgrav #elif defined(HAVE_STRUCT_STAT_ST_MTIME)
9584f52dfbbSDag-Erling Smørgrav return (rmul * NCMP(as->st_mtime, bs->st_mtime));
9594f52dfbbSDag-Erling Smørgrav #else
9604f52dfbbSDag-Erling Smørgrav return rmul * 1;
9614f52dfbbSDag-Erling Smørgrav #endif
9624f52dfbbSDag-Erling Smørgrav } else if (sort_flag & LS_SIZE_SORT)
9634f52dfbbSDag-Erling Smørgrav return (rmul * NCMP(as->st_size, bs->st_size));
9644f52dfbbSDag-Erling Smørgrav
9654f52dfbbSDag-Erling Smørgrav fatal("Unknown ls sort type");
9664f52dfbbSDag-Erling Smørgrav }
9674f52dfbbSDag-Erling Smørgrav
968efcad6b7SDag-Erling Smørgrav /* sftp ls.1 replacement which handles path globs */
969efcad6b7SDag-Erling Smørgrav static int
do_globbed_ls(struct sftp_conn * conn,const char * path,const char * strip_path,int lflag)970076ad2f8SDag-Erling Smørgrav do_globbed_ls(struct sftp_conn *conn, const char *path,
971076ad2f8SDag-Erling Smørgrav const char *strip_path, int lflag)
972efcad6b7SDag-Erling Smørgrav {
9734a421b63SDag-Erling Smørgrav char *fname, *lname;
9744a421b63SDag-Erling Smørgrav glob_t g;
975a0ee8cc6SDag-Erling Smørgrav int err, r;
9764a421b63SDag-Erling Smørgrav struct winsize ws;
9774f52dfbbSDag-Erling Smørgrav u_int i, j, nentries, *indices = NULL, c = 1;
9784f52dfbbSDag-Erling Smørgrav u_int colspace = 0, columns = 1, m = 0, width = 80;
979efcad6b7SDag-Erling Smørgrav
980efcad6b7SDag-Erling Smørgrav memset(&g, 0, sizeof(g));
981efcad6b7SDag-Erling Smørgrav
982edf85781SEd Maste if ((r = sftp_glob(conn, path,
983462c32cbSDag-Erling Smørgrav GLOB_MARK|GLOB_NOCHECK|GLOB_BRACE|GLOB_KEEPSTAT|GLOB_NOSORT,
984a0ee8cc6SDag-Erling Smørgrav NULL, &g)) != 0 ||
9854a421b63SDag-Erling Smørgrav (g.gl_pathc && !g.gl_matchc)) {
9865e8dbd04SDag-Erling Smørgrav if (g.gl_pathc)
9875e8dbd04SDag-Erling Smørgrav globfree(&g);
988a0ee8cc6SDag-Erling Smørgrav if (r == GLOB_NOSPACE) {
989a0ee8cc6SDag-Erling Smørgrav error("Can't ls: Too many matches for \"%s\"", path);
990a0ee8cc6SDag-Erling Smørgrav } else {
991efcad6b7SDag-Erling Smørgrav error("Can't ls: \"%s\" not found", path);
992a0ee8cc6SDag-Erling Smørgrav }
9934a421b63SDag-Erling Smørgrav return -1;
994efcad6b7SDag-Erling Smørgrav }
995efcad6b7SDag-Erling Smørgrav
996d74d50a8SDag-Erling Smørgrav if (interrupted)
997d74d50a8SDag-Erling Smørgrav goto out;
998d74d50a8SDag-Erling Smørgrav
999efcad6b7SDag-Erling Smørgrav /*
10005e8dbd04SDag-Erling Smørgrav * If the glob returns a single match and it is a directory,
10015e8dbd04SDag-Erling Smørgrav * then just list its contents.
1002efcad6b7SDag-Erling Smørgrav */
10034a421b63SDag-Erling Smørgrav if (g.gl_matchc == 1 && g.gl_statv[0] != NULL &&
10044a421b63SDag-Erling Smørgrav S_ISDIR(g.gl_statv[0]->st_mode)) {
10055e8dbd04SDag-Erling Smørgrav err = do_ls_dir(conn, g.gl_pathv[0], strip_path, lflag);
1006efcad6b7SDag-Erling Smørgrav globfree(&g);
10074a421b63SDag-Erling Smørgrav return err;
1008efcad6b7SDag-Erling Smørgrav }
1009efcad6b7SDag-Erling Smørgrav
1010efcad6b7SDag-Erling Smørgrav if (ioctl(fileno(stdin), TIOCGWINSZ, &ws) != -1)
1011efcad6b7SDag-Erling Smørgrav width = ws.ws_col;
1012efcad6b7SDag-Erling Smørgrav
10134a421b63SDag-Erling Smørgrav if (!(lflag & LS_SHORT_VIEW)) {
10144a421b63SDag-Erling Smørgrav /* Count entries for sort and find longest filename */
10154a421b63SDag-Erling Smørgrav for (i = 0; g.gl_pathv[i]; i++)
1016ca86bcf2SDag-Erling Smørgrav m = MAXIMUM(m, strlen(g.gl_pathv[i]));
10174a421b63SDag-Erling Smørgrav
1018efcad6b7SDag-Erling Smørgrav columns = width / (m + 2);
1019ca86bcf2SDag-Erling Smørgrav columns = MAXIMUM(columns, 1);
1020efcad6b7SDag-Erling Smørgrav colspace = width / columns;
1021efcad6b7SDag-Erling Smørgrav }
1022efcad6b7SDag-Erling Smørgrav
10234f52dfbbSDag-Erling Smørgrav /*
10244f52dfbbSDag-Erling Smørgrav * Sorting: rather than mess with the contents of glob_t, prepare
10254f52dfbbSDag-Erling Smørgrav * an array of indices into it and sort that. For the usual
10264f52dfbbSDag-Erling Smørgrav * unsorted case, the indices are just the identity 1=1, 2=2, etc.
10274f52dfbbSDag-Erling Smørgrav */
10284f52dfbbSDag-Erling Smørgrav for (nentries = 0; g.gl_pathv[nentries] != NULL; nentries++)
10294f52dfbbSDag-Erling Smørgrav ; /* count entries */
10304d3fc8b0SEd Maste indices = xcalloc(nentries, sizeof(*indices));
10314f52dfbbSDag-Erling Smørgrav for (i = 0; i < nentries; i++)
10324f52dfbbSDag-Erling Smørgrav indices[i] = i;
10334f52dfbbSDag-Erling Smørgrav
10344f52dfbbSDag-Erling Smørgrav if (lflag & SORT_FLAGS) {
10354f52dfbbSDag-Erling Smørgrav sort_glob = &g;
10364f52dfbbSDag-Erling Smørgrav sort_flag = lflag & (SORT_FLAGS|LS_REVERSE_SORT);
10374f52dfbbSDag-Erling Smørgrav qsort(indices, nentries, sizeof(*indices), sglob_comp);
10384f52dfbbSDag-Erling Smørgrav sort_glob = NULL;
10394f52dfbbSDag-Erling Smørgrav }
10404f52dfbbSDag-Erling Smørgrav
104138a52bd3SEd Maste get_remote_user_groups_from_glob(conn, &g);
10424f52dfbbSDag-Erling Smørgrav for (j = 0; j < nentries && !interrupted; j++) {
10434f52dfbbSDag-Erling Smørgrav i = indices[j];
1044efcad6b7SDag-Erling Smørgrav fname = path_strip(g.gl_pathv[i], strip_path);
1045d74d50a8SDag-Erling Smørgrav if (lflag & LS_LONG_VIEW) {
10464a421b63SDag-Erling Smørgrav if (g.gl_statv[i] == NULL) {
10474a421b63SDag-Erling Smørgrav error("no stat information for %s", fname);
10484d3fc8b0SEd Maste free(fname);
10494a421b63SDag-Erling Smørgrav continue;
10504a421b63SDag-Erling Smørgrav }
10514a421b63SDag-Erling Smørgrav lname = ls_file(fname, g.gl_statv[i], 1,
105238a52bd3SEd Maste (lflag & LS_SI_UNITS),
105338a52bd3SEd Maste ruser_name(g.gl_statv[i]->st_uid),
105438a52bd3SEd Maste rgroup_name(g.gl_statv[i]->st_gid));
1055076ad2f8SDag-Erling Smørgrav mprintf("%s\n", lname);
1056e4a9863fSDag-Erling Smørgrav free(lname);
1057efcad6b7SDag-Erling Smørgrav } else {
1058076ad2f8SDag-Erling Smørgrav mprintf("%-*s", colspace, fname);
1059efcad6b7SDag-Erling Smørgrav if (c >= columns) {
1060efcad6b7SDag-Erling Smørgrav printf("\n");
1061efcad6b7SDag-Erling Smørgrav c = 1;
1062efcad6b7SDag-Erling Smørgrav } else
1063efcad6b7SDag-Erling Smørgrav c++;
1064efcad6b7SDag-Erling Smørgrav }
1065e4a9863fSDag-Erling Smørgrav free(fname);
1066efcad6b7SDag-Erling Smørgrav }
1067efcad6b7SDag-Erling Smørgrav
1068d74d50a8SDag-Erling Smørgrav if (!(lflag & LS_LONG_VIEW) && (c != 1))
1069efcad6b7SDag-Erling Smørgrav printf("\n");
1070efcad6b7SDag-Erling Smørgrav
1071d74d50a8SDag-Erling Smørgrav out:
1072efcad6b7SDag-Erling Smørgrav if (g.gl_pathc)
1073efcad6b7SDag-Erling Smørgrav globfree(&g);
10744f52dfbbSDag-Erling Smørgrav free(indices);
1075efcad6b7SDag-Erling Smørgrav
10764a421b63SDag-Erling Smørgrav return 0;
1077efcad6b7SDag-Erling Smørgrav }
1078efcad6b7SDag-Erling Smørgrav
1079efcad6b7SDag-Erling Smørgrav static int
do_df(struct sftp_conn * conn,const char * path,int hflag,int iflag)1080076ad2f8SDag-Erling Smørgrav do_df(struct sftp_conn *conn, const char *path, int hflag, int iflag)
1081d4af9e69SDag-Erling Smørgrav {
1082d4af9e69SDag-Erling Smørgrav struct sftp_statvfs st;
1083d93a896eSDag-Erling Smørgrav char s_used[FMT_SCALED_STRSIZE], s_avail[FMT_SCALED_STRSIZE];
1084d93a896eSDag-Erling Smørgrav char s_root[FMT_SCALED_STRSIZE], s_total[FMT_SCALED_STRSIZE];
1085d93a896eSDag-Erling Smørgrav char s_icapacity[16], s_dcapacity[16];
1086d4af9e69SDag-Erling Smørgrav
1087edf85781SEd Maste if (sftp_statvfs(conn, path, &st, 1) == -1)
1088d4af9e69SDag-Erling Smørgrav return -1;
1089d93a896eSDag-Erling Smørgrav if (st.f_files == 0)
1090d93a896eSDag-Erling Smørgrav strlcpy(s_icapacity, "ERR", sizeof(s_icapacity));
1091d93a896eSDag-Erling Smørgrav else {
1092d93a896eSDag-Erling Smørgrav snprintf(s_icapacity, sizeof(s_icapacity), "%3llu%%",
1093d93a896eSDag-Erling Smørgrav (unsigned long long)(100 * (st.f_files - st.f_ffree) /
1094d93a896eSDag-Erling Smørgrav st.f_files));
1095d93a896eSDag-Erling Smørgrav }
1096d93a896eSDag-Erling Smørgrav if (st.f_blocks == 0)
1097d93a896eSDag-Erling Smørgrav strlcpy(s_dcapacity, "ERR", sizeof(s_dcapacity));
1098d93a896eSDag-Erling Smørgrav else {
1099d93a896eSDag-Erling Smørgrav snprintf(s_dcapacity, sizeof(s_dcapacity), "%3llu%%",
1100d93a896eSDag-Erling Smørgrav (unsigned long long)(100 * (st.f_blocks - st.f_bfree) /
1101d93a896eSDag-Erling Smørgrav st.f_blocks));
1102d93a896eSDag-Erling Smørgrav }
1103d4af9e69SDag-Erling Smørgrav if (iflag) {
1104d4af9e69SDag-Erling Smørgrav printf(" Inodes Used Avail "
1105d4af9e69SDag-Erling Smørgrav "(root) %%Capacity\n");
1106d93a896eSDag-Erling Smørgrav printf("%11llu %11llu %11llu %11llu %s\n",
1107d4af9e69SDag-Erling Smørgrav (unsigned long long)st.f_files,
1108d4af9e69SDag-Erling Smørgrav (unsigned long long)(st.f_files - st.f_ffree),
1109d4af9e69SDag-Erling Smørgrav (unsigned long long)st.f_favail,
1110d93a896eSDag-Erling Smørgrav (unsigned long long)st.f_ffree, s_icapacity);
1111d4af9e69SDag-Erling Smørgrav } else if (hflag) {
1112d4af9e69SDag-Erling Smørgrav strlcpy(s_used, "error", sizeof(s_used));
1113d4af9e69SDag-Erling Smørgrav strlcpy(s_avail, "error", sizeof(s_avail));
1114d4af9e69SDag-Erling Smørgrav strlcpy(s_root, "error", sizeof(s_root));
1115d4af9e69SDag-Erling Smørgrav strlcpy(s_total, "error", sizeof(s_total));
1116d4af9e69SDag-Erling Smørgrav fmt_scaled((st.f_blocks - st.f_bfree) * st.f_frsize, s_used);
1117d4af9e69SDag-Erling Smørgrav fmt_scaled(st.f_bavail * st.f_frsize, s_avail);
1118d4af9e69SDag-Erling Smørgrav fmt_scaled(st.f_bfree * st.f_frsize, s_root);
1119d4af9e69SDag-Erling Smørgrav fmt_scaled(st.f_blocks * st.f_frsize, s_total);
1120d4af9e69SDag-Erling Smørgrav printf(" Size Used Avail (root) %%Capacity\n");
1121d93a896eSDag-Erling Smørgrav printf("%7sB %7sB %7sB %7sB %s\n",
1122d93a896eSDag-Erling Smørgrav s_total, s_used, s_avail, s_root, s_dcapacity);
1123d4af9e69SDag-Erling Smørgrav } else {
1124d4af9e69SDag-Erling Smørgrav printf(" Size Used Avail "
1125d4af9e69SDag-Erling Smørgrav "(root) %%Capacity\n");
1126d93a896eSDag-Erling Smørgrav printf("%12llu %12llu %12llu %12llu %s\n",
1127d4af9e69SDag-Erling Smørgrav (unsigned long long)(st.f_frsize * st.f_blocks / 1024),
1128d4af9e69SDag-Erling Smørgrav (unsigned long long)(st.f_frsize *
1129d4af9e69SDag-Erling Smørgrav (st.f_blocks - st.f_bfree) / 1024),
1130d4af9e69SDag-Erling Smørgrav (unsigned long long)(st.f_frsize * st.f_bavail / 1024),
1131d4af9e69SDag-Erling Smørgrav (unsigned long long)(st.f_frsize * st.f_bfree / 1024),
1132d93a896eSDag-Erling Smørgrav s_dcapacity);
1133d4af9e69SDag-Erling Smørgrav }
1134d4af9e69SDag-Erling Smørgrav return 0;
1135d4af9e69SDag-Erling Smørgrav }
1136d4af9e69SDag-Erling Smørgrav
1137d4af9e69SDag-Erling Smørgrav /*
1138d4af9e69SDag-Erling Smørgrav * Undo escaping of glob sequences in place. Used to undo extra escaping
1139d4af9e69SDag-Erling Smørgrav * applied in makeargv() when the string is destined for a function that
1140d4af9e69SDag-Erling Smørgrav * does not glob it.
1141d4af9e69SDag-Erling Smørgrav */
1142d4af9e69SDag-Erling Smørgrav static void
undo_glob_escape(char * s)1143d4af9e69SDag-Erling Smørgrav undo_glob_escape(char *s)
1144d4af9e69SDag-Erling Smørgrav {
1145d4af9e69SDag-Erling Smørgrav size_t i, j;
1146d4af9e69SDag-Erling Smørgrav
1147d4af9e69SDag-Erling Smørgrav for (i = j = 0;;) {
1148d4af9e69SDag-Erling Smørgrav if (s[i] == '\0') {
1149d4af9e69SDag-Erling Smørgrav s[j] = '\0';
1150d4af9e69SDag-Erling Smørgrav return;
1151d4af9e69SDag-Erling Smørgrav }
1152d4af9e69SDag-Erling Smørgrav if (s[i] != '\\') {
1153d4af9e69SDag-Erling Smørgrav s[j++] = s[i++];
1154d4af9e69SDag-Erling Smørgrav continue;
1155d4af9e69SDag-Erling Smørgrav }
1156d4af9e69SDag-Erling Smørgrav /* s[i] == '\\' */
1157d4af9e69SDag-Erling Smørgrav ++i;
1158d4af9e69SDag-Erling Smørgrav switch (s[i]) {
1159d4af9e69SDag-Erling Smørgrav case '?':
1160d4af9e69SDag-Erling Smørgrav case '[':
1161d4af9e69SDag-Erling Smørgrav case '*':
1162d4af9e69SDag-Erling Smørgrav case '\\':
1163d4af9e69SDag-Erling Smørgrav s[j++] = s[i++];
1164d4af9e69SDag-Erling Smørgrav break;
1165d4af9e69SDag-Erling Smørgrav case '\0':
1166d4af9e69SDag-Erling Smørgrav s[j++] = '\\';
1167d4af9e69SDag-Erling Smørgrav s[j] = '\0';
1168d4af9e69SDag-Erling Smørgrav return;
1169d4af9e69SDag-Erling Smørgrav default:
1170d4af9e69SDag-Erling Smørgrav s[j++] = '\\';
1171d4af9e69SDag-Erling Smørgrav s[j++] = s[i++];
1172d4af9e69SDag-Erling Smørgrav break;
1173d4af9e69SDag-Erling Smørgrav }
1174d4af9e69SDag-Erling Smørgrav }
1175d4af9e69SDag-Erling Smørgrav }
1176d4af9e69SDag-Erling Smørgrav
1177d4af9e69SDag-Erling Smørgrav /*
1178d4af9e69SDag-Erling Smørgrav * Split a string into an argument vector using sh(1)-style quoting,
1179d4af9e69SDag-Erling Smørgrav * comment and escaping rules, but with some tweaks to handle glob(3)
1180d4af9e69SDag-Erling Smørgrav * wildcards.
1181b15c8340SDag-Erling Smørgrav * The "sloppy" flag allows for recovery from missing terminating quote, for
1182b15c8340SDag-Erling Smørgrav * use in parsing incomplete commandlines during tab autocompletion.
1183b15c8340SDag-Erling Smørgrav *
1184d4af9e69SDag-Erling Smørgrav * Returns NULL on error or a NULL-terminated array of arguments.
1185b15c8340SDag-Erling Smørgrav *
1186b15c8340SDag-Erling Smørgrav * If "lastquote" is not NULL, the quoting character used for the last
1187b15c8340SDag-Erling Smørgrav * argument is placed in *lastquote ("\0", "'" or "\"").
1188b15c8340SDag-Erling Smørgrav *
1189b15c8340SDag-Erling Smørgrav * If "terminated" is not NULL, *terminated will be set to 1 when the
1190b15c8340SDag-Erling Smørgrav * last argument's quote has been properly terminated or 0 otherwise.
1191b15c8340SDag-Erling Smørgrav * This parameter is only of use if "sloppy" is set.
1192d4af9e69SDag-Erling Smørgrav */
1193d4af9e69SDag-Erling Smørgrav #define MAXARGS 128
1194d4af9e69SDag-Erling Smørgrav #define MAXARGLEN 8192
1195d4af9e69SDag-Erling Smørgrav static char **
makeargv(const char * arg,int * argcp,int sloppy,char * lastquote,u_int * terminated)1196b15c8340SDag-Erling Smørgrav makeargv(const char *arg, int *argcp, int sloppy, char *lastquote,
1197b15c8340SDag-Erling Smørgrav u_int *terminated)
1198d4af9e69SDag-Erling Smørgrav {
1199d4af9e69SDag-Erling Smørgrav int argc, quot;
1200d4af9e69SDag-Erling Smørgrav size_t i, j;
1201d4af9e69SDag-Erling Smørgrav static char argvs[MAXARGLEN];
1202d4af9e69SDag-Erling Smørgrav static char *argv[MAXARGS + 1];
1203d4af9e69SDag-Erling Smørgrav enum { MA_START, MA_SQUOTE, MA_DQUOTE, MA_UNQUOTED } state, q;
1204d4af9e69SDag-Erling Smørgrav
1205d4af9e69SDag-Erling Smørgrav *argcp = argc = 0;
1206d4af9e69SDag-Erling Smørgrav if (strlen(arg) > sizeof(argvs) - 1) {
1207d4af9e69SDag-Erling Smørgrav args_too_longs:
1208d4af9e69SDag-Erling Smørgrav error("string too long");
1209d4af9e69SDag-Erling Smørgrav return NULL;
1210d4af9e69SDag-Erling Smørgrav }
1211b15c8340SDag-Erling Smørgrav if (terminated != NULL)
1212b15c8340SDag-Erling Smørgrav *terminated = 1;
1213b15c8340SDag-Erling Smørgrav if (lastquote != NULL)
1214b15c8340SDag-Erling Smørgrav *lastquote = '\0';
1215d4af9e69SDag-Erling Smørgrav state = MA_START;
1216d4af9e69SDag-Erling Smørgrav i = j = 0;
1217d4af9e69SDag-Erling Smørgrav for (;;) {
12186888a9beSDag-Erling Smørgrav if ((size_t)argc >= sizeof(argv) / sizeof(*argv)){
12196888a9beSDag-Erling Smørgrav error("Too many arguments.");
12206888a9beSDag-Erling Smørgrav return NULL;
12216888a9beSDag-Erling Smørgrav }
1222f7167e0eSDag-Erling Smørgrav if (isspace((unsigned char)arg[i])) {
1223d4af9e69SDag-Erling Smørgrav if (state == MA_UNQUOTED) {
1224d4af9e69SDag-Erling Smørgrav /* Terminate current argument */
1225d4af9e69SDag-Erling Smørgrav argvs[j++] = '\0';
1226d4af9e69SDag-Erling Smørgrav argc++;
1227d4af9e69SDag-Erling Smørgrav state = MA_START;
1228d4af9e69SDag-Erling Smørgrav } else if (state != MA_START)
1229d4af9e69SDag-Erling Smørgrav argvs[j++] = arg[i];
1230d4af9e69SDag-Erling Smørgrav } else if (arg[i] == '"' || arg[i] == '\'') {
1231d4af9e69SDag-Erling Smørgrav q = arg[i] == '"' ? MA_DQUOTE : MA_SQUOTE;
1232d4af9e69SDag-Erling Smørgrav if (state == MA_START) {
1233d4af9e69SDag-Erling Smørgrav argv[argc] = argvs + j;
1234d4af9e69SDag-Erling Smørgrav state = q;
1235b15c8340SDag-Erling Smørgrav if (lastquote != NULL)
1236b15c8340SDag-Erling Smørgrav *lastquote = arg[i];
1237d4af9e69SDag-Erling Smørgrav } else if (state == MA_UNQUOTED)
1238d4af9e69SDag-Erling Smørgrav state = q;
1239d4af9e69SDag-Erling Smørgrav else if (state == q)
1240d4af9e69SDag-Erling Smørgrav state = MA_UNQUOTED;
1241d4af9e69SDag-Erling Smørgrav else
1242d4af9e69SDag-Erling Smørgrav argvs[j++] = arg[i];
1243d4af9e69SDag-Erling Smørgrav } else if (arg[i] == '\\') {
1244d4af9e69SDag-Erling Smørgrav if (state == MA_SQUOTE || state == MA_DQUOTE) {
1245d4af9e69SDag-Erling Smørgrav quot = state == MA_SQUOTE ? '\'' : '"';
1246d4af9e69SDag-Erling Smørgrav /* Unescape quote we are in */
1247d4af9e69SDag-Erling Smørgrav /* XXX support \n and friends? */
1248d4af9e69SDag-Erling Smørgrav if (arg[i + 1] == quot) {
1249d4af9e69SDag-Erling Smørgrav i++;
1250d4af9e69SDag-Erling Smørgrav argvs[j++] = arg[i];
1251d4af9e69SDag-Erling Smørgrav } else if (arg[i + 1] == '?' ||
1252d4af9e69SDag-Erling Smørgrav arg[i + 1] == '[' || arg[i + 1] == '*') {
1253d4af9e69SDag-Erling Smørgrav /*
1254d4af9e69SDag-Erling Smørgrav * Special case for sftp: append
1255d4af9e69SDag-Erling Smørgrav * double-escaped glob sequence -
1256d4af9e69SDag-Erling Smørgrav * glob will undo one level of
1257d4af9e69SDag-Erling Smørgrav * escaping. NB. string can grow here.
1258d4af9e69SDag-Erling Smørgrav */
1259d4af9e69SDag-Erling Smørgrav if (j >= sizeof(argvs) - 5)
1260d4af9e69SDag-Erling Smørgrav goto args_too_longs;
1261d4af9e69SDag-Erling Smørgrav argvs[j++] = '\\';
1262d4af9e69SDag-Erling Smørgrav argvs[j++] = arg[i++];
1263d4af9e69SDag-Erling Smørgrav argvs[j++] = '\\';
1264d4af9e69SDag-Erling Smørgrav argvs[j++] = arg[i];
1265d4af9e69SDag-Erling Smørgrav } else {
1266d4af9e69SDag-Erling Smørgrav argvs[j++] = arg[i++];
1267d4af9e69SDag-Erling Smørgrav argvs[j++] = arg[i];
1268d4af9e69SDag-Erling Smørgrav }
1269d4af9e69SDag-Erling Smørgrav } else {
1270d4af9e69SDag-Erling Smørgrav if (state == MA_START) {
1271d4af9e69SDag-Erling Smørgrav argv[argc] = argvs + j;
1272d4af9e69SDag-Erling Smørgrav state = MA_UNQUOTED;
1273b15c8340SDag-Erling Smørgrav if (lastquote != NULL)
1274b15c8340SDag-Erling Smørgrav *lastquote = '\0';
1275d4af9e69SDag-Erling Smørgrav }
1276d4af9e69SDag-Erling Smørgrav if (arg[i + 1] == '?' || arg[i + 1] == '[' ||
1277d4af9e69SDag-Erling Smørgrav arg[i + 1] == '*' || arg[i + 1] == '\\') {
1278d4af9e69SDag-Erling Smørgrav /*
1279d4af9e69SDag-Erling Smørgrav * Special case for sftp: append
1280d4af9e69SDag-Erling Smørgrav * escaped glob sequence -
1281d4af9e69SDag-Erling Smørgrav * glob will undo one level of
1282d4af9e69SDag-Erling Smørgrav * escaping.
1283d4af9e69SDag-Erling Smørgrav */
1284d4af9e69SDag-Erling Smørgrav argvs[j++] = arg[i++];
1285d4af9e69SDag-Erling Smørgrav argvs[j++] = arg[i];
1286d4af9e69SDag-Erling Smørgrav } else {
1287d4af9e69SDag-Erling Smørgrav /* Unescape everything */
1288d4af9e69SDag-Erling Smørgrav /* XXX support \n and friends? */
1289d4af9e69SDag-Erling Smørgrav i++;
1290d4af9e69SDag-Erling Smørgrav argvs[j++] = arg[i];
1291d4af9e69SDag-Erling Smørgrav }
1292d4af9e69SDag-Erling Smørgrav }
1293d4af9e69SDag-Erling Smørgrav } else if (arg[i] == '#') {
1294d4af9e69SDag-Erling Smørgrav if (state == MA_SQUOTE || state == MA_DQUOTE)
1295d4af9e69SDag-Erling Smørgrav argvs[j++] = arg[i];
1296d4af9e69SDag-Erling Smørgrav else
1297d4af9e69SDag-Erling Smørgrav goto string_done;
1298d4af9e69SDag-Erling Smørgrav } else if (arg[i] == '\0') {
1299d4af9e69SDag-Erling Smørgrav if (state == MA_SQUOTE || state == MA_DQUOTE) {
1300b15c8340SDag-Erling Smørgrav if (sloppy) {
1301b15c8340SDag-Erling Smørgrav state = MA_UNQUOTED;
1302b15c8340SDag-Erling Smørgrav if (terminated != NULL)
1303b15c8340SDag-Erling Smørgrav *terminated = 0;
1304b15c8340SDag-Erling Smørgrav goto string_done;
1305b15c8340SDag-Erling Smørgrav }
1306d4af9e69SDag-Erling Smørgrav error("Unterminated quoted argument");
1307d4af9e69SDag-Erling Smørgrav return NULL;
1308d4af9e69SDag-Erling Smørgrav }
1309d4af9e69SDag-Erling Smørgrav string_done:
1310d4af9e69SDag-Erling Smørgrav if (state == MA_UNQUOTED) {
1311d4af9e69SDag-Erling Smørgrav argvs[j++] = '\0';
1312d4af9e69SDag-Erling Smørgrav argc++;
1313d4af9e69SDag-Erling Smørgrav }
1314d4af9e69SDag-Erling Smørgrav break;
1315d4af9e69SDag-Erling Smørgrav } else {
1316d4af9e69SDag-Erling Smørgrav if (state == MA_START) {
1317d4af9e69SDag-Erling Smørgrav argv[argc] = argvs + j;
1318d4af9e69SDag-Erling Smørgrav state = MA_UNQUOTED;
1319b15c8340SDag-Erling Smørgrav if (lastquote != NULL)
1320b15c8340SDag-Erling Smørgrav *lastquote = '\0';
1321d4af9e69SDag-Erling Smørgrav }
1322d4af9e69SDag-Erling Smørgrav if ((state == MA_SQUOTE || state == MA_DQUOTE) &&
1323d4af9e69SDag-Erling Smørgrav (arg[i] == '?' || arg[i] == '[' || arg[i] == '*')) {
1324d4af9e69SDag-Erling Smørgrav /*
1325d4af9e69SDag-Erling Smørgrav * Special case for sftp: escape quoted
1326d4af9e69SDag-Erling Smørgrav * glob(3) wildcards. NB. string can grow
1327d4af9e69SDag-Erling Smørgrav * here.
1328d4af9e69SDag-Erling Smørgrav */
1329d4af9e69SDag-Erling Smørgrav if (j >= sizeof(argvs) - 3)
1330d4af9e69SDag-Erling Smørgrav goto args_too_longs;
1331d4af9e69SDag-Erling Smørgrav argvs[j++] = '\\';
1332d4af9e69SDag-Erling Smørgrav argvs[j++] = arg[i];
1333d4af9e69SDag-Erling Smørgrav } else
1334d4af9e69SDag-Erling Smørgrav argvs[j++] = arg[i];
1335d4af9e69SDag-Erling Smørgrav }
1336d4af9e69SDag-Erling Smørgrav i++;
1337d4af9e69SDag-Erling Smørgrav }
1338d4af9e69SDag-Erling Smørgrav *argcp = argc;
1339d4af9e69SDag-Erling Smørgrav return argv;
1340d4af9e69SDag-Erling Smørgrav }
1341d4af9e69SDag-Erling Smørgrav
1342d4af9e69SDag-Erling Smørgrav static int
parse_args(const char ** cpp,int * ignore_errors,int * disable_echo,int * aflag,int * fflag,int * hflag,int * iflag,int * lflag,int * pflag,int * rflag,int * sflag,unsigned long * n_arg,char ** path1,char ** path2)134319261079SEd Maste parse_args(const char **cpp, int *ignore_errors, int *disable_echo, int *aflag,
1344a0ee8cc6SDag-Erling Smørgrav int *fflag, int *hflag, int *iflag, int *lflag, int *pflag,
1345a0ee8cc6SDag-Erling Smørgrav int *rflag, int *sflag,
1346f7167e0eSDag-Erling Smørgrav unsigned long *n_arg, char **path1, char **path2)
1347efcad6b7SDag-Erling Smørgrav {
1348efcad6b7SDag-Erling Smørgrav const char *cmd, *cp = *cpp;
1349d4af9e69SDag-Erling Smørgrav char *cp2, **argv;
1350efcad6b7SDag-Erling Smørgrav int base = 0;
135119261079SEd Maste long long ll;
135247dd1d1bSDag-Erling Smørgrav int path1_mandatory = 0, i, cmdnum, optidx, argc;
1353efcad6b7SDag-Erling Smørgrav
1354efcad6b7SDag-Erling Smørgrav /* Skip leading whitespace */
1355efcad6b7SDag-Erling Smørgrav cp = cp + strspn(cp, WHITESPACE);
1356efcad6b7SDag-Erling Smørgrav
135719261079SEd Maste /*
135819261079SEd Maste * Check for leading '-' (disable error processing) and '@' (suppress
135919261079SEd Maste * command echo)
136019261079SEd Maste */
1361f7167e0eSDag-Erling Smørgrav *ignore_errors = 0;
136219261079SEd Maste *disable_echo = 0;
136319261079SEd Maste for (;*cp != '\0'; cp++) {
1364efcad6b7SDag-Erling Smørgrav if (*cp == '-') {
1365f7167e0eSDag-Erling Smørgrav *ignore_errors = 1;
136619261079SEd Maste } else if (*cp == '@') {
136719261079SEd Maste *disable_echo = 1;
136819261079SEd Maste } else {
136919261079SEd Maste /* all other characters terminate prefix processing */
137019261079SEd Maste break;
1371efcad6b7SDag-Erling Smørgrav }
137219261079SEd Maste }
137319261079SEd Maste cp = cp + strspn(cp, WHITESPACE);
1374efcad6b7SDag-Erling Smørgrav
1375b15c8340SDag-Erling Smørgrav /* Ignore blank lines and lines which begin with comment '#' char */
1376b15c8340SDag-Erling Smørgrav if (*cp == '\0' || *cp == '#')
1377b15c8340SDag-Erling Smørgrav return (0);
1378b15c8340SDag-Erling Smørgrav
1379b15c8340SDag-Erling Smørgrav if ((argv = makeargv(cp, &argc, 0, NULL, NULL)) == NULL)
1380d4af9e69SDag-Erling Smørgrav return -1;
1381efcad6b7SDag-Erling Smørgrav
1382d4af9e69SDag-Erling Smørgrav /* Figure out which command we have */
1383d4af9e69SDag-Erling Smørgrav for (i = 0; cmds[i].c != NULL; i++) {
13846888a9beSDag-Erling Smørgrav if (argv[0] != NULL && strcasecmp(cmds[i].c, argv[0]) == 0)
1385efcad6b7SDag-Erling Smørgrav break;
1386efcad6b7SDag-Erling Smørgrav }
1387efcad6b7SDag-Erling Smørgrav cmdnum = cmds[i].n;
1388efcad6b7SDag-Erling Smørgrav cmd = cmds[i].c;
1389efcad6b7SDag-Erling Smørgrav
1390efcad6b7SDag-Erling Smørgrav /* Special case */
1391efcad6b7SDag-Erling Smørgrav if (*cp == '!') {
1392efcad6b7SDag-Erling Smørgrav cp++;
1393efcad6b7SDag-Erling Smørgrav cmdnum = I_SHELL;
1394efcad6b7SDag-Erling Smørgrav } else if (cmdnum == -1) {
1395efcad6b7SDag-Erling Smørgrav error("Invalid command.");
1396d4af9e69SDag-Erling Smørgrav return -1;
1397efcad6b7SDag-Erling Smørgrav }
1398efcad6b7SDag-Erling Smørgrav
1399efcad6b7SDag-Erling Smørgrav /* Get arguments and parse flags */
1400f7167e0eSDag-Erling Smørgrav *aflag = *fflag = *hflag = *iflag = *lflag = *pflag = 0;
1401f7167e0eSDag-Erling Smørgrav *rflag = *sflag = 0;
1402efcad6b7SDag-Erling Smørgrav *path1 = *path2 = NULL;
1403d4af9e69SDag-Erling Smørgrav optidx = 1;
1404efcad6b7SDag-Erling Smørgrav switch (cmdnum) {
1405efcad6b7SDag-Erling Smørgrav case I_GET:
1406e4a9863fSDag-Erling Smørgrav case I_REGET:
1407a0ee8cc6SDag-Erling Smørgrav case I_REPUT:
1408efcad6b7SDag-Erling Smørgrav case I_PUT:
14094a421b63SDag-Erling Smørgrav if ((optidx = parse_getput_flags(cmd, argv, argc,
1410f7167e0eSDag-Erling Smørgrav aflag, fflag, pflag, rflag)) == -1)
1411d4af9e69SDag-Erling Smørgrav return -1;
1412efcad6b7SDag-Erling Smørgrav /* Get first pathname (mandatory) */
1413d4af9e69SDag-Erling Smørgrav if (argc - optidx < 1) {
1414efcad6b7SDag-Erling Smørgrav error("You must specify at least one path after a "
1415efcad6b7SDag-Erling Smørgrav "%s command.", cmd);
1416d4af9e69SDag-Erling Smørgrav return -1;
1417efcad6b7SDag-Erling Smørgrav }
1418d4af9e69SDag-Erling Smørgrav *path1 = xstrdup(argv[optidx]);
1419d4af9e69SDag-Erling Smørgrav /* Get second pathname (optional) */
1420d4af9e69SDag-Erling Smørgrav if (argc - optidx > 1) {
1421d4af9e69SDag-Erling Smørgrav *path2 = xstrdup(argv[optidx + 1]);
1422d4af9e69SDag-Erling Smørgrav /* Destination is not globbed */
1423d4af9e69SDag-Erling Smørgrav undo_glob_escape(*path2);
1424d4af9e69SDag-Erling Smørgrav }
1425efcad6b7SDag-Erling Smørgrav break;
14264a421b63SDag-Erling Smørgrav case I_LINK:
14274a421b63SDag-Erling Smørgrav if ((optidx = parse_link_flags(cmd, argv, argc, sflag)) == -1)
14284a421b63SDag-Erling Smørgrav return -1;
1429f7167e0eSDag-Erling Smørgrav goto parse_two_paths;
143087c1498dSEd Maste case I_COPY:
143187c1498dSEd Maste if ((optidx = parse_no_flags(cmd, argv, argc)) == -1)
143287c1498dSEd Maste return -1;
143387c1498dSEd Maste goto parse_two_paths;
14344a421b63SDag-Erling Smørgrav case I_RENAME:
1435f7167e0eSDag-Erling Smørgrav if ((optidx = parse_rename_flags(cmd, argv, argc, lflag)) == -1)
1436f7167e0eSDag-Erling Smørgrav return -1;
1437f7167e0eSDag-Erling Smørgrav goto parse_two_paths;
1438f7167e0eSDag-Erling Smørgrav case I_SYMLINK:
1439f7167e0eSDag-Erling Smørgrav if ((optidx = parse_no_flags(cmd, argv, argc)) == -1)
1440f7167e0eSDag-Erling Smørgrav return -1;
1441f7167e0eSDag-Erling Smørgrav parse_two_paths:
1442d4af9e69SDag-Erling Smørgrav if (argc - optidx < 2) {
1443efcad6b7SDag-Erling Smørgrav error("You must specify two paths after a %s "
1444efcad6b7SDag-Erling Smørgrav "command.", cmd);
1445d4af9e69SDag-Erling Smørgrav return -1;
1446efcad6b7SDag-Erling Smørgrav }
1447d4af9e69SDag-Erling Smørgrav *path1 = xstrdup(argv[optidx]);
1448d4af9e69SDag-Erling Smørgrav *path2 = xstrdup(argv[optidx + 1]);
1449d4af9e69SDag-Erling Smørgrav /* Paths are not globbed */
1450d4af9e69SDag-Erling Smørgrav undo_glob_escape(*path1);
1451d4af9e69SDag-Erling Smørgrav undo_glob_escape(*path2);
1452efcad6b7SDag-Erling Smørgrav break;
1453efcad6b7SDag-Erling Smørgrav case I_RM:
1454efcad6b7SDag-Erling Smørgrav case I_MKDIR:
1455efcad6b7SDag-Erling Smørgrav case I_RMDIR:
145647dd1d1bSDag-Erling Smørgrav case I_LMKDIR:
145747dd1d1bSDag-Erling Smørgrav path1_mandatory = 1;
145847dd1d1bSDag-Erling Smørgrav /* FALLTHROUGH */
1459efcad6b7SDag-Erling Smørgrav case I_CHDIR:
1460efcad6b7SDag-Erling Smørgrav case I_LCHDIR:
1461f7167e0eSDag-Erling Smørgrav if ((optidx = parse_no_flags(cmd, argv, argc)) == -1)
1462f7167e0eSDag-Erling Smørgrav return -1;
1463efcad6b7SDag-Erling Smørgrav /* Get pathname (mandatory) */
1464d4af9e69SDag-Erling Smørgrav if (argc - optidx < 1) {
146547dd1d1bSDag-Erling Smørgrav if (!path1_mandatory)
146647dd1d1bSDag-Erling Smørgrav break; /* return a NULL path1 */
1467efcad6b7SDag-Erling Smørgrav error("You must specify a path after a %s command.",
1468efcad6b7SDag-Erling Smørgrav cmd);
1469d4af9e69SDag-Erling Smørgrav return -1;
1470d4af9e69SDag-Erling Smørgrav }
1471d4af9e69SDag-Erling Smørgrav *path1 = xstrdup(argv[optidx]);
1472d4af9e69SDag-Erling Smørgrav /* Only "rm" globs */
1473d4af9e69SDag-Erling Smørgrav if (cmdnum != I_RM)
1474d4af9e69SDag-Erling Smørgrav undo_glob_escape(*path1);
1475d4af9e69SDag-Erling Smørgrav break;
1476d4af9e69SDag-Erling Smørgrav case I_DF:
1477d4af9e69SDag-Erling Smørgrav if ((optidx = parse_df_flags(cmd, argv, argc, hflag,
1478d4af9e69SDag-Erling Smørgrav iflag)) == -1)
1479d4af9e69SDag-Erling Smørgrav return -1;
1480d4af9e69SDag-Erling Smørgrav /* Default to current directory if no path specified */
1481d4af9e69SDag-Erling Smørgrav if (argc - optidx < 1)
1482d4af9e69SDag-Erling Smørgrav *path1 = NULL;
1483d4af9e69SDag-Erling Smørgrav else {
1484d4af9e69SDag-Erling Smørgrav *path1 = xstrdup(argv[optidx]);
1485d4af9e69SDag-Erling Smørgrav undo_glob_escape(*path1);
1486efcad6b7SDag-Erling Smørgrav }
1487efcad6b7SDag-Erling Smørgrav break;
1488efcad6b7SDag-Erling Smørgrav case I_LS:
1489d4af9e69SDag-Erling Smørgrav if ((optidx = parse_ls_flags(argv, argc, lflag)) == -1)
1490efcad6b7SDag-Erling Smørgrav return(-1);
1491efcad6b7SDag-Erling Smørgrav /* Path is optional */
1492d4af9e69SDag-Erling Smørgrav if (argc - optidx > 0)
1493d4af9e69SDag-Erling Smørgrav *path1 = xstrdup(argv[optidx]);
1494efcad6b7SDag-Erling Smørgrav break;
1495efcad6b7SDag-Erling Smørgrav case I_LLS:
1496d4af9e69SDag-Erling Smørgrav /* Skip ls command and following whitespace */
1497d4af9e69SDag-Erling Smørgrav cp = cp + strlen(cmd) + strspn(cp, WHITESPACE);
1498efcad6b7SDag-Erling Smørgrav case I_SHELL:
1499efcad6b7SDag-Erling Smørgrav /* Uses the rest of the line */
1500efcad6b7SDag-Erling Smørgrav break;
1501efcad6b7SDag-Erling Smørgrav case I_LUMASK:
1502efcad6b7SDag-Erling Smørgrav case I_CHMOD:
1503efcad6b7SDag-Erling Smørgrav base = 8;
15042f513db7SEd Maste /* FALLTHROUGH */
1505efcad6b7SDag-Erling Smørgrav case I_CHOWN:
1506efcad6b7SDag-Erling Smørgrav case I_CHGRP:
150719261079SEd Maste if ((optidx = parse_ch_flags(cmd, argv, argc, hflag)) == -1)
1508f7167e0eSDag-Erling Smørgrav return -1;
1509efcad6b7SDag-Erling Smørgrav /* Get numeric arg (mandatory) */
1510d4af9e69SDag-Erling Smørgrav if (argc - optidx < 1)
1511d4af9e69SDag-Erling Smørgrav goto need_num_arg;
151292eb0aa1SDag-Erling Smørgrav errno = 0;
151319261079SEd Maste ll = strtoll(argv[optidx], &cp2, base);
1514d4af9e69SDag-Erling Smørgrav if (cp2 == argv[optidx] || *cp2 != '\0' ||
151519261079SEd Maste ((ll == LLONG_MIN || ll == LLONG_MAX) && errno == ERANGE) ||
151619261079SEd Maste ll < 0 || ll > UINT32_MAX) {
1517d4af9e69SDag-Erling Smørgrav need_num_arg:
1518efcad6b7SDag-Erling Smørgrav error("You must supply a numeric argument "
1519efcad6b7SDag-Erling Smørgrav "to the %s command.", cmd);
1520d4af9e69SDag-Erling Smørgrav return -1;
1521efcad6b7SDag-Erling Smørgrav }
152219261079SEd Maste *n_arg = ll;
1523d4af9e69SDag-Erling Smørgrav if (cmdnum == I_LUMASK)
1524efcad6b7SDag-Erling Smørgrav break;
1525efcad6b7SDag-Erling Smørgrav /* Get pathname (mandatory) */
1526d4af9e69SDag-Erling Smørgrav if (argc - optidx < 2) {
1527efcad6b7SDag-Erling Smørgrav error("You must specify a path after a %s command.",
1528efcad6b7SDag-Erling Smørgrav cmd);
1529d4af9e69SDag-Erling Smørgrav return -1;
1530efcad6b7SDag-Erling Smørgrav }
1531d4af9e69SDag-Erling Smørgrav *path1 = xstrdup(argv[optidx + 1]);
1532efcad6b7SDag-Erling Smørgrav break;
1533efcad6b7SDag-Erling Smørgrav case I_QUIT:
1534efcad6b7SDag-Erling Smørgrav case I_PWD:
1535efcad6b7SDag-Erling Smørgrav case I_LPWD:
1536efcad6b7SDag-Erling Smørgrav case I_HELP:
1537efcad6b7SDag-Erling Smørgrav case I_VERSION:
1538efcad6b7SDag-Erling Smørgrav case I_PROGRESS:
1539f7167e0eSDag-Erling Smørgrav if ((optidx = parse_no_flags(cmd, argv, argc)) == -1)
1540f7167e0eSDag-Erling Smørgrav return -1;
1541efcad6b7SDag-Erling Smørgrav break;
1542efcad6b7SDag-Erling Smørgrav default:
1543efcad6b7SDag-Erling Smørgrav fatal("Command not implemented");
1544efcad6b7SDag-Erling Smørgrav }
1545efcad6b7SDag-Erling Smørgrav
1546efcad6b7SDag-Erling Smørgrav *cpp = cp;
1547efcad6b7SDag-Erling Smørgrav return(cmdnum);
1548efcad6b7SDag-Erling Smørgrav }
1549efcad6b7SDag-Erling Smørgrav
1550efcad6b7SDag-Erling Smørgrav static int
parse_dispatch_command(struct sftp_conn * conn,const char * cmd,char ** pwd,const char * startdir,int err_abort,int echo_command)1551efcad6b7SDag-Erling Smørgrav parse_dispatch_command(struct sftp_conn *conn, const char *cmd, char **pwd,
155219261079SEd Maste const char *startdir, int err_abort, int echo_command)
1553efcad6b7SDag-Erling Smørgrav {
155419261079SEd Maste const char *ocmd = cmd;
1555efcad6b7SDag-Erling Smørgrav char *path1, *path2, *tmp;
155619261079SEd Maste int ignore_errors = 0, disable_echo = 1;
155719261079SEd Maste int aflag = 0, fflag = 0, hflag = 0, iflag = 0;
1558f7167e0eSDag-Erling Smørgrav int lflag = 0, pflag = 0, rflag = 0, sflag = 0;
15594a421b63SDag-Erling Smørgrav int cmdnum, i;
1560cce7d346SDag-Erling Smørgrav unsigned long n_arg = 0;
1561edf85781SEd Maste Attrib a, aa;
1562bc5531deSDag-Erling Smørgrav char path_buf[PATH_MAX];
1563efcad6b7SDag-Erling Smørgrav int err = 0;
1564efcad6b7SDag-Erling Smørgrav glob_t g;
1565efcad6b7SDag-Erling Smørgrav
1566efcad6b7SDag-Erling Smørgrav path1 = path2 = NULL;
156719261079SEd Maste cmdnum = parse_args(&cmd, &ignore_errors, &disable_echo, &aflag, &fflag,
156819261079SEd Maste &hflag, &iflag, &lflag, &pflag, &rflag, &sflag, &n_arg,
156919261079SEd Maste &path1, &path2);
1570f7167e0eSDag-Erling Smørgrav if (ignore_errors != 0)
1571efcad6b7SDag-Erling Smørgrav err_abort = 0;
1572efcad6b7SDag-Erling Smørgrav
157319261079SEd Maste if (echo_command && !disable_echo)
157419261079SEd Maste mprintf("sftp> %s\n", ocmd);
157519261079SEd Maste
1576efcad6b7SDag-Erling Smørgrav memset(&g, 0, sizeof(g));
1577efcad6b7SDag-Erling Smørgrav
1578efcad6b7SDag-Erling Smørgrav /* Perform command */
1579efcad6b7SDag-Erling Smørgrav switch (cmdnum) {
1580efcad6b7SDag-Erling Smørgrav case 0:
1581efcad6b7SDag-Erling Smørgrav /* Blank line */
1582efcad6b7SDag-Erling Smørgrav break;
1583efcad6b7SDag-Erling Smørgrav case -1:
1584efcad6b7SDag-Erling Smørgrav /* Unrecognized command */
1585efcad6b7SDag-Erling Smørgrav err = -1;
1586efcad6b7SDag-Erling Smørgrav break;
1587e4a9863fSDag-Erling Smørgrav case I_REGET:
1588e4a9863fSDag-Erling Smørgrav aflag = 1;
1589e4a9863fSDag-Erling Smørgrav /* FALLTHROUGH */
1590efcad6b7SDag-Erling Smørgrav case I_GET:
1591e4a9863fSDag-Erling Smørgrav err = process_get(conn, path1, path2, *pwd, pflag,
1592f7167e0eSDag-Erling Smørgrav rflag, aflag, fflag);
1593efcad6b7SDag-Erling Smørgrav break;
1594a0ee8cc6SDag-Erling Smørgrav case I_REPUT:
1595a0ee8cc6SDag-Erling Smørgrav aflag = 1;
1596a0ee8cc6SDag-Erling Smørgrav /* FALLTHROUGH */
1597efcad6b7SDag-Erling Smørgrav case I_PUT:
1598f7167e0eSDag-Erling Smørgrav err = process_put(conn, path1, path2, *pwd, pflag,
1599a0ee8cc6SDag-Erling Smørgrav rflag, aflag, fflag);
1600efcad6b7SDag-Erling Smørgrav break;
160187c1498dSEd Maste case I_COPY:
1602edf85781SEd Maste path1 = sftp_make_absolute(path1, *pwd);
1603edf85781SEd Maste path2 = sftp_make_absolute(path2, *pwd);
1604edf85781SEd Maste err = sftp_copy(conn, path1, path2);
160587c1498dSEd Maste break;
1606efcad6b7SDag-Erling Smørgrav case I_RENAME:
1607edf85781SEd Maste path1 = sftp_make_absolute(path1, *pwd);
1608edf85781SEd Maste path2 = sftp_make_absolute(path2, *pwd);
1609edf85781SEd Maste err = sftp_rename(conn, path1, path2, lflag);
1610efcad6b7SDag-Erling Smørgrav break;
1611efcad6b7SDag-Erling Smørgrav case I_SYMLINK:
16124a421b63SDag-Erling Smørgrav sflag = 1;
16132f513db7SEd Maste /* FALLTHROUGH */
16144a421b63SDag-Erling Smørgrav case I_LINK:
1615f7167e0eSDag-Erling Smørgrav if (!sflag)
1616edf85781SEd Maste path1 = sftp_make_absolute(path1, *pwd);
1617edf85781SEd Maste path2 = sftp_make_absolute(path2, *pwd);
1618edf85781SEd Maste err = (sflag ? sftp_symlink : sftp_hardlink)(conn,
1619edf85781SEd Maste path1, path2);
1620efcad6b7SDag-Erling Smørgrav break;
1621efcad6b7SDag-Erling Smørgrav case I_RM:
162238a52bd3SEd Maste path1 = make_absolute_pwd_glob(path1, *pwd);
1623edf85781SEd Maste sftp_glob(conn, path1, GLOB_NOCHECK, NULL, &g);
1624d74d50a8SDag-Erling Smørgrav for (i = 0; g.gl_pathv[i] && !interrupted; i++) {
1625e4a9863fSDag-Erling Smørgrav if (!quiet)
1626076ad2f8SDag-Erling Smørgrav mprintf("Removing %s\n", g.gl_pathv[i]);
1627edf85781SEd Maste err = sftp_rm(conn, g.gl_pathv[i]);
1628efcad6b7SDag-Erling Smørgrav if (err != 0 && err_abort)
1629efcad6b7SDag-Erling Smørgrav break;
1630efcad6b7SDag-Erling Smørgrav }
1631efcad6b7SDag-Erling Smørgrav break;
1632efcad6b7SDag-Erling Smørgrav case I_MKDIR:
1633edf85781SEd Maste path1 = sftp_make_absolute(path1, *pwd);
1634efcad6b7SDag-Erling Smørgrav attrib_clear(&a);
1635efcad6b7SDag-Erling Smørgrav a.flags |= SSH2_FILEXFER_ATTR_PERMISSIONS;
1636efcad6b7SDag-Erling Smørgrav a.perm = 0777;
1637edf85781SEd Maste err = sftp_mkdir(conn, path1, &a, 1);
1638efcad6b7SDag-Erling Smørgrav break;
1639efcad6b7SDag-Erling Smørgrav case I_RMDIR:
1640edf85781SEd Maste path1 = sftp_make_absolute(path1, *pwd);
1641edf85781SEd Maste err = sftp_rmdir(conn, path1);
1642efcad6b7SDag-Erling Smørgrav break;
1643efcad6b7SDag-Erling Smørgrav case I_CHDIR:
164447dd1d1bSDag-Erling Smørgrav if (path1 == NULL || *path1 == '\0')
164547dd1d1bSDag-Erling Smørgrav path1 = xstrdup(startdir);
1646edf85781SEd Maste path1 = sftp_make_absolute(path1, *pwd);
1647edf85781SEd Maste if ((tmp = sftp_realpath(conn, path1)) == NULL) {
1648efcad6b7SDag-Erling Smørgrav err = 1;
1649efcad6b7SDag-Erling Smørgrav break;
1650efcad6b7SDag-Erling Smørgrav }
1651edf85781SEd Maste if (sftp_stat(conn, tmp, 0, &aa) != 0) {
1652e4a9863fSDag-Erling Smørgrav free(tmp);
1653efcad6b7SDag-Erling Smørgrav err = 1;
1654efcad6b7SDag-Erling Smørgrav break;
1655efcad6b7SDag-Erling Smørgrav }
1656edf85781SEd Maste if (!(aa.flags & SSH2_FILEXFER_ATTR_PERMISSIONS)) {
1657efcad6b7SDag-Erling Smørgrav error("Can't change directory: Can't check target");
1658e4a9863fSDag-Erling Smørgrav free(tmp);
1659efcad6b7SDag-Erling Smørgrav err = 1;
1660efcad6b7SDag-Erling Smørgrav break;
1661efcad6b7SDag-Erling Smørgrav }
1662edf85781SEd Maste if (!S_ISDIR(aa.perm)) {
1663efcad6b7SDag-Erling Smørgrav error("Can't change directory: \"%s\" is not "
1664efcad6b7SDag-Erling Smørgrav "a directory", tmp);
1665e4a9863fSDag-Erling Smørgrav free(tmp);
1666efcad6b7SDag-Erling Smørgrav err = 1;
1667efcad6b7SDag-Erling Smørgrav break;
1668efcad6b7SDag-Erling Smørgrav }
1669e4a9863fSDag-Erling Smørgrav free(*pwd);
1670efcad6b7SDag-Erling Smørgrav *pwd = tmp;
1671efcad6b7SDag-Erling Smørgrav break;
1672efcad6b7SDag-Erling Smørgrav case I_LS:
1673efcad6b7SDag-Erling Smørgrav if (!path1) {
1674e2f6069cSDag-Erling Smørgrav do_ls_dir(conn, *pwd, *pwd, lflag);
1675efcad6b7SDag-Erling Smørgrav break;
1676efcad6b7SDag-Erling Smørgrav }
1677efcad6b7SDag-Erling Smørgrav
1678efcad6b7SDag-Erling Smørgrav /* Strip pwd off beginning of non-absolute paths */
1679efcad6b7SDag-Erling Smørgrav tmp = NULL;
168019261079SEd Maste if (!path_absolute(path1))
1681efcad6b7SDag-Erling Smørgrav tmp = *pwd;
1682efcad6b7SDag-Erling Smørgrav
168338a52bd3SEd Maste path1 = make_absolute_pwd_glob(path1, *pwd);
1684efcad6b7SDag-Erling Smørgrav err = do_globbed_ls(conn, path1, tmp, lflag);
1685efcad6b7SDag-Erling Smørgrav break;
1686d4af9e69SDag-Erling Smørgrav case I_DF:
1687d4af9e69SDag-Erling Smørgrav /* Default to current directory if no path specified */
1688d4af9e69SDag-Erling Smørgrav if (path1 == NULL)
1689d4af9e69SDag-Erling Smørgrav path1 = xstrdup(*pwd);
1690edf85781SEd Maste path1 = sftp_make_absolute(path1, *pwd);
1691d4af9e69SDag-Erling Smørgrav err = do_df(conn, path1, hflag, iflag);
1692d4af9e69SDag-Erling Smørgrav break;
1693efcad6b7SDag-Erling Smørgrav case I_LCHDIR:
169447dd1d1bSDag-Erling Smørgrav if (path1 == NULL || *path1 == '\0')
169547dd1d1bSDag-Erling Smørgrav path1 = xstrdup("~");
1696bc5531deSDag-Erling Smørgrav tmp = tilde_expand_filename(path1, getuid());
1697bc5531deSDag-Erling Smørgrav free(path1);
1698bc5531deSDag-Erling Smørgrav path1 = tmp;
1699efcad6b7SDag-Erling Smørgrav if (chdir(path1) == -1) {
1700efcad6b7SDag-Erling Smørgrav error("Couldn't change local directory to "
1701efcad6b7SDag-Erling Smørgrav "\"%s\": %s", path1, strerror(errno));
1702efcad6b7SDag-Erling Smørgrav err = 1;
1703efcad6b7SDag-Erling Smørgrav }
1704efcad6b7SDag-Erling Smørgrav break;
1705efcad6b7SDag-Erling Smørgrav case I_LMKDIR:
1706efcad6b7SDag-Erling Smørgrav if (mkdir(path1, 0777) == -1) {
1707efcad6b7SDag-Erling Smørgrav error("Couldn't create local directory "
1708efcad6b7SDag-Erling Smørgrav "\"%s\": %s", path1, strerror(errno));
1709efcad6b7SDag-Erling Smørgrav err = 1;
1710efcad6b7SDag-Erling Smørgrav }
1711efcad6b7SDag-Erling Smørgrav break;
1712efcad6b7SDag-Erling Smørgrav case I_LLS:
1713efcad6b7SDag-Erling Smørgrav local_do_ls(cmd);
1714efcad6b7SDag-Erling Smørgrav break;
1715efcad6b7SDag-Erling Smørgrav case I_SHELL:
1716efcad6b7SDag-Erling Smørgrav local_do_shell(cmd);
1717efcad6b7SDag-Erling Smørgrav break;
1718efcad6b7SDag-Erling Smørgrav case I_LUMASK:
1719efcad6b7SDag-Erling Smørgrav umask(n_arg);
1720efcad6b7SDag-Erling Smørgrav printf("Local umask: %03lo\n", n_arg);
1721efcad6b7SDag-Erling Smørgrav break;
1722efcad6b7SDag-Erling Smørgrav case I_CHMOD:
172338a52bd3SEd Maste path1 = make_absolute_pwd_glob(path1, *pwd);
1724efcad6b7SDag-Erling Smørgrav attrib_clear(&a);
1725efcad6b7SDag-Erling Smørgrav a.flags |= SSH2_FILEXFER_ATTR_PERMISSIONS;
1726efcad6b7SDag-Erling Smørgrav a.perm = n_arg;
1727edf85781SEd Maste sftp_glob(conn, path1, GLOB_NOCHECK, NULL, &g);
1728d74d50a8SDag-Erling Smørgrav for (i = 0; g.gl_pathv[i] && !interrupted; i++) {
1729e4a9863fSDag-Erling Smørgrav if (!quiet)
1730076ad2f8SDag-Erling Smørgrav mprintf("Changing mode on %s\n",
1731076ad2f8SDag-Erling Smørgrav g.gl_pathv[i]);
1732edf85781SEd Maste err = (hflag ? sftp_lsetstat : sftp_setstat)(conn,
173319261079SEd Maste g.gl_pathv[i], &a);
1734efcad6b7SDag-Erling Smørgrav if (err != 0 && err_abort)
1735efcad6b7SDag-Erling Smørgrav break;
1736efcad6b7SDag-Erling Smørgrav }
1737efcad6b7SDag-Erling Smørgrav break;
1738efcad6b7SDag-Erling Smørgrav case I_CHOWN:
1739efcad6b7SDag-Erling Smørgrav case I_CHGRP:
174038a52bd3SEd Maste path1 = make_absolute_pwd_glob(path1, *pwd);
1741edf85781SEd Maste sftp_glob(conn, path1, GLOB_NOCHECK, NULL, &g);
1742d74d50a8SDag-Erling Smørgrav for (i = 0; g.gl_pathv[i] && !interrupted; i++) {
1743edf85781SEd Maste if ((hflag ? sftp_lstat : sftp_stat)(conn,
1744edf85781SEd Maste g.gl_pathv[i], 0, &aa) != 0) {
1745cce7d346SDag-Erling Smørgrav if (err_abort) {
1746cce7d346SDag-Erling Smørgrav err = -1;
1747efcad6b7SDag-Erling Smørgrav break;
1748cce7d346SDag-Erling Smørgrav } else
1749efcad6b7SDag-Erling Smørgrav continue;
1750efcad6b7SDag-Erling Smørgrav }
1751edf85781SEd Maste if (!(aa.flags & SSH2_FILEXFER_ATTR_UIDGID)) {
1752efcad6b7SDag-Erling Smørgrav error("Can't get current ownership of "
1753efcad6b7SDag-Erling Smørgrav "remote file \"%s\"", g.gl_pathv[i]);
1754cce7d346SDag-Erling Smørgrav if (err_abort) {
1755cce7d346SDag-Erling Smørgrav err = -1;
1756efcad6b7SDag-Erling Smørgrav break;
1757cce7d346SDag-Erling Smørgrav } else
1758efcad6b7SDag-Erling Smørgrav continue;
1759efcad6b7SDag-Erling Smørgrav }
1760edf85781SEd Maste aa.flags &= SSH2_FILEXFER_ATTR_UIDGID;
1761efcad6b7SDag-Erling Smørgrav if (cmdnum == I_CHOWN) {
1762e4a9863fSDag-Erling Smørgrav if (!quiet)
1763076ad2f8SDag-Erling Smørgrav mprintf("Changing owner on %s\n",
1764e4a9863fSDag-Erling Smørgrav g.gl_pathv[i]);
1765edf85781SEd Maste aa.uid = n_arg;
1766efcad6b7SDag-Erling Smørgrav } else {
1767e4a9863fSDag-Erling Smørgrav if (!quiet)
1768076ad2f8SDag-Erling Smørgrav mprintf("Changing group on %s\n",
1769e4a9863fSDag-Erling Smørgrav g.gl_pathv[i]);
1770edf85781SEd Maste aa.gid = n_arg;
1771efcad6b7SDag-Erling Smørgrav }
1772edf85781SEd Maste err = (hflag ? sftp_lsetstat : sftp_setstat)(conn,
1773edf85781SEd Maste g.gl_pathv[i], &aa);
1774efcad6b7SDag-Erling Smørgrav if (err != 0 && err_abort)
1775efcad6b7SDag-Erling Smørgrav break;
1776efcad6b7SDag-Erling Smørgrav }
1777efcad6b7SDag-Erling Smørgrav break;
1778efcad6b7SDag-Erling Smørgrav case I_PWD:
1779076ad2f8SDag-Erling Smørgrav mprintf("Remote working directory: %s\n", *pwd);
1780efcad6b7SDag-Erling Smørgrav break;
1781efcad6b7SDag-Erling Smørgrav case I_LPWD:
1782efcad6b7SDag-Erling Smørgrav if (!getcwd(path_buf, sizeof(path_buf))) {
1783efcad6b7SDag-Erling Smørgrav error("Couldn't get local cwd: %s", strerror(errno));
1784efcad6b7SDag-Erling Smørgrav err = -1;
1785efcad6b7SDag-Erling Smørgrav break;
1786efcad6b7SDag-Erling Smørgrav }
1787076ad2f8SDag-Erling Smørgrav mprintf("Local working directory: %s\n", path_buf);
1788efcad6b7SDag-Erling Smørgrav break;
1789efcad6b7SDag-Erling Smørgrav case I_QUIT:
1790efcad6b7SDag-Erling Smørgrav /* Processed below */
1791efcad6b7SDag-Erling Smørgrav break;
1792efcad6b7SDag-Erling Smørgrav case I_HELP:
1793efcad6b7SDag-Erling Smørgrav help();
1794efcad6b7SDag-Erling Smørgrav break;
1795efcad6b7SDag-Erling Smørgrav case I_VERSION:
1796efcad6b7SDag-Erling Smørgrav printf("SFTP protocol version %u\n", sftp_proto_version(conn));
1797efcad6b7SDag-Erling Smørgrav break;
1798efcad6b7SDag-Erling Smørgrav case I_PROGRESS:
1799efcad6b7SDag-Erling Smørgrav showprogress = !showprogress;
1800efcad6b7SDag-Erling Smørgrav if (showprogress)
1801efcad6b7SDag-Erling Smørgrav printf("Progress meter enabled\n");
1802efcad6b7SDag-Erling Smørgrav else
1803efcad6b7SDag-Erling Smørgrav printf("Progress meter disabled\n");
1804efcad6b7SDag-Erling Smørgrav break;
1805efcad6b7SDag-Erling Smørgrav default:
1806efcad6b7SDag-Erling Smørgrav fatal("%d is not implemented", cmdnum);
1807efcad6b7SDag-Erling Smørgrav }
1808efcad6b7SDag-Erling Smørgrav
1809efcad6b7SDag-Erling Smørgrav if (g.gl_pathc)
1810efcad6b7SDag-Erling Smørgrav globfree(&g);
1811e4a9863fSDag-Erling Smørgrav free(path1);
1812e4a9863fSDag-Erling Smørgrav free(path2);
1813efcad6b7SDag-Erling Smørgrav
1814efcad6b7SDag-Erling Smørgrav /* If an unignored error occurs in batch mode we should abort. */
1815efcad6b7SDag-Erling Smørgrav if (err_abort && err != 0)
1816efcad6b7SDag-Erling Smørgrav return (-1);
1817efcad6b7SDag-Erling Smørgrav else if (cmdnum == I_QUIT)
1818efcad6b7SDag-Erling Smørgrav return (1);
1819efcad6b7SDag-Erling Smørgrav
1820efcad6b7SDag-Erling Smørgrav return (0);
1821efcad6b7SDag-Erling Smørgrav }
1822efcad6b7SDag-Erling Smørgrav
18235e8dbd04SDag-Erling Smørgrav #ifdef USE_LIBEDIT
18245e8dbd04SDag-Erling Smørgrav static char *
prompt(EditLine * el)18255e8dbd04SDag-Erling Smørgrav prompt(EditLine *el)
18265e8dbd04SDag-Erling Smørgrav {
18275e8dbd04SDag-Erling Smørgrav return ("sftp> ");
18285e8dbd04SDag-Erling Smørgrav }
1829b15c8340SDag-Erling Smørgrav
1830b15c8340SDag-Erling Smørgrav /* Display entries in 'list' after skipping the first 'len' chars */
1831b15c8340SDag-Erling Smørgrav static void
complete_display(char ** list,u_int len)1832b15c8340SDag-Erling Smørgrav complete_display(char **list, u_int len)
1833b15c8340SDag-Erling Smørgrav {
1834b15c8340SDag-Erling Smørgrav u_int y, m = 0, width = 80, columns = 1, colspace = 0, llen;
1835b15c8340SDag-Erling Smørgrav struct winsize ws;
1836b15c8340SDag-Erling Smørgrav char *tmp;
1837b15c8340SDag-Erling Smørgrav
1838b15c8340SDag-Erling Smørgrav /* Count entries for sort and find longest */
1839b15c8340SDag-Erling Smørgrav for (y = 0; list[y]; y++)
1840ca86bcf2SDag-Erling Smørgrav m = MAXIMUM(m, strlen(list[y]));
1841b15c8340SDag-Erling Smørgrav
1842b15c8340SDag-Erling Smørgrav if (ioctl(fileno(stdin), TIOCGWINSZ, &ws) != -1)
1843b15c8340SDag-Erling Smørgrav width = ws.ws_col;
1844b15c8340SDag-Erling Smørgrav
1845b15c8340SDag-Erling Smørgrav m = m > len ? m - len : 0;
1846b15c8340SDag-Erling Smørgrav columns = width / (m + 2);
1847ca86bcf2SDag-Erling Smørgrav columns = MAXIMUM(columns, 1);
1848b15c8340SDag-Erling Smørgrav colspace = width / columns;
1849ca86bcf2SDag-Erling Smørgrav colspace = MINIMUM(colspace, width);
1850b15c8340SDag-Erling Smørgrav
1851b15c8340SDag-Erling Smørgrav printf("\n");
1852b15c8340SDag-Erling Smørgrav m = 1;
1853b15c8340SDag-Erling Smørgrav for (y = 0; list[y]; y++) {
1854b15c8340SDag-Erling Smørgrav llen = strlen(list[y]);
1855b15c8340SDag-Erling Smørgrav tmp = llen > len ? list[y] + len : "";
1856076ad2f8SDag-Erling Smørgrav mprintf("%-*s", colspace, tmp);
1857b15c8340SDag-Erling Smørgrav if (m >= columns) {
1858b15c8340SDag-Erling Smørgrav printf("\n");
1859b15c8340SDag-Erling Smørgrav m = 1;
1860b15c8340SDag-Erling Smørgrav } else
1861b15c8340SDag-Erling Smørgrav m++;
1862b15c8340SDag-Erling Smørgrav }
1863b15c8340SDag-Erling Smørgrav printf("\n");
1864b15c8340SDag-Erling Smørgrav }
1865b15c8340SDag-Erling Smørgrav
1866b15c8340SDag-Erling Smørgrav /*
1867b15c8340SDag-Erling Smørgrav * Given a "list" of words that begin with a common prefix of "word",
1868b15c8340SDag-Erling Smørgrav * attempt to find an autocompletion to extends "word" by the next
1869b15c8340SDag-Erling Smørgrav * characters common to all entries in "list".
1870b15c8340SDag-Erling Smørgrav */
1871b15c8340SDag-Erling Smørgrav static char *
complete_ambiguous(const char * word,char ** list,size_t count)1872b15c8340SDag-Erling Smørgrav complete_ambiguous(const char *word, char **list, size_t count)
1873b15c8340SDag-Erling Smørgrav {
1874b15c8340SDag-Erling Smørgrav if (word == NULL)
1875b15c8340SDag-Erling Smørgrav return NULL;
1876b15c8340SDag-Erling Smørgrav
1877b15c8340SDag-Erling Smørgrav if (count > 0) {
1878b15c8340SDag-Erling Smørgrav u_int y, matchlen = strlen(list[0]);
1879b15c8340SDag-Erling Smørgrav
1880b15c8340SDag-Erling Smørgrav /* Find length of common stem */
1881b15c8340SDag-Erling Smørgrav for (y = 1; list[y]; y++) {
1882b15c8340SDag-Erling Smørgrav u_int x;
1883b15c8340SDag-Erling Smørgrav
1884b15c8340SDag-Erling Smørgrav for (x = 0; x < matchlen; x++)
1885b15c8340SDag-Erling Smørgrav if (list[0][x] != list[y][x])
1886b15c8340SDag-Erling Smørgrav break;
1887b15c8340SDag-Erling Smørgrav
1888b15c8340SDag-Erling Smørgrav matchlen = x;
1889b15c8340SDag-Erling Smørgrav }
1890b15c8340SDag-Erling Smørgrav
1891b15c8340SDag-Erling Smørgrav if (matchlen > strlen(word)) {
1892b15c8340SDag-Erling Smørgrav char *tmp = xstrdup(list[0]);
1893b15c8340SDag-Erling Smørgrav
1894b15c8340SDag-Erling Smørgrav tmp[matchlen] = '\0';
1895b15c8340SDag-Erling Smørgrav return tmp;
1896b15c8340SDag-Erling Smørgrav }
1897b15c8340SDag-Erling Smørgrav }
1898b15c8340SDag-Erling Smørgrav
1899b15c8340SDag-Erling Smørgrav return xstrdup(word);
1900b15c8340SDag-Erling Smørgrav }
1901b15c8340SDag-Erling Smørgrav
1902b15c8340SDag-Erling Smørgrav /* Autocomplete a sftp command */
1903b15c8340SDag-Erling Smørgrav static int
complete_cmd_parse(EditLine * el,char * cmd,int lastarg,char quote,int terminated)1904b15c8340SDag-Erling Smørgrav complete_cmd_parse(EditLine *el, char *cmd, int lastarg, char quote,
1905b15c8340SDag-Erling Smørgrav int terminated)
1906b15c8340SDag-Erling Smørgrav {
1907b15c8340SDag-Erling Smørgrav u_int y, count = 0, cmdlen, tmplen;
1908b15c8340SDag-Erling Smørgrav char *tmp, **list, argterm[3];
1909b15c8340SDag-Erling Smørgrav const LineInfo *lf;
1910b15c8340SDag-Erling Smørgrav
1911b15c8340SDag-Erling Smørgrav list = xcalloc((sizeof(cmds) / sizeof(*cmds)) + 1, sizeof(char *));
1912b15c8340SDag-Erling Smørgrav
1913b15c8340SDag-Erling Smørgrav /* No command specified: display all available commands */
1914b15c8340SDag-Erling Smørgrav if (cmd == NULL) {
1915b15c8340SDag-Erling Smørgrav for (y = 0; cmds[y].c; y++)
1916b15c8340SDag-Erling Smørgrav list[count++] = xstrdup(cmds[y].c);
1917b15c8340SDag-Erling Smørgrav
1918b15c8340SDag-Erling Smørgrav list[count] = NULL;
1919b15c8340SDag-Erling Smørgrav complete_display(list, 0);
1920b15c8340SDag-Erling Smørgrav
1921b15c8340SDag-Erling Smørgrav for (y = 0; list[y] != NULL; y++)
1922e4a9863fSDag-Erling Smørgrav free(list[y]);
1923e4a9863fSDag-Erling Smørgrav free(list);
1924b15c8340SDag-Erling Smørgrav return count;
1925b15c8340SDag-Erling Smørgrav }
1926b15c8340SDag-Erling Smørgrav
1927b15c8340SDag-Erling Smørgrav /* Prepare subset of commands that start with "cmd" */
1928b15c8340SDag-Erling Smørgrav cmdlen = strlen(cmd);
1929b15c8340SDag-Erling Smørgrav for (y = 0; cmds[y].c; y++) {
1930b15c8340SDag-Erling Smørgrav if (!strncasecmp(cmd, cmds[y].c, cmdlen))
1931b15c8340SDag-Erling Smørgrav list[count++] = xstrdup(cmds[y].c);
1932b15c8340SDag-Erling Smørgrav }
1933b15c8340SDag-Erling Smørgrav list[count] = NULL;
1934b15c8340SDag-Erling Smørgrav
1935462c32cbSDag-Erling Smørgrav if (count == 0) {
1936e4a9863fSDag-Erling Smørgrav free(list);
1937b15c8340SDag-Erling Smørgrav return 0;
1938462c32cbSDag-Erling Smørgrav }
1939b15c8340SDag-Erling Smørgrav
1940190cef3dSDag-Erling Smørgrav /* Complete ambiguous command */
1941b15c8340SDag-Erling Smørgrav tmp = complete_ambiguous(cmd, list, count);
1942b15c8340SDag-Erling Smørgrav if (count > 1)
1943b15c8340SDag-Erling Smørgrav complete_display(list, 0);
1944b15c8340SDag-Erling Smørgrav
1945b15c8340SDag-Erling Smørgrav for (y = 0; list[y]; y++)
1946e4a9863fSDag-Erling Smørgrav free(list[y]);
1947e4a9863fSDag-Erling Smørgrav free(list);
1948b15c8340SDag-Erling Smørgrav
1949b15c8340SDag-Erling Smørgrav if (tmp != NULL) {
1950b15c8340SDag-Erling Smørgrav tmplen = strlen(tmp);
1951b15c8340SDag-Erling Smørgrav cmdlen = strlen(cmd);
1952b15c8340SDag-Erling Smørgrav /* If cmd may be extended then do so */
1953b15c8340SDag-Erling Smørgrav if (tmplen > cmdlen)
1954b15c8340SDag-Erling Smørgrav if (el_insertstr(el, tmp + cmdlen) == -1)
1955b15c8340SDag-Erling Smørgrav fatal("el_insertstr failed.");
1956b15c8340SDag-Erling Smørgrav lf = el_line(el);
1957b15c8340SDag-Erling Smørgrav /* Terminate argument cleanly */
1958b15c8340SDag-Erling Smørgrav if (count == 1) {
1959b15c8340SDag-Erling Smørgrav y = 0;
1960b15c8340SDag-Erling Smørgrav if (!terminated)
1961b15c8340SDag-Erling Smørgrav argterm[y++] = quote;
1962b15c8340SDag-Erling Smørgrav if (lastarg || *(lf->cursor) != ' ')
1963b15c8340SDag-Erling Smørgrav argterm[y++] = ' ';
1964b15c8340SDag-Erling Smørgrav argterm[y] = '\0';
1965b15c8340SDag-Erling Smørgrav if (y > 0 && el_insertstr(el, argterm) == -1)
1966b15c8340SDag-Erling Smørgrav fatal("el_insertstr failed.");
1967b15c8340SDag-Erling Smørgrav }
1968e4a9863fSDag-Erling Smørgrav free(tmp);
1969b15c8340SDag-Erling Smørgrav }
1970b15c8340SDag-Erling Smørgrav
1971b15c8340SDag-Erling Smørgrav return count;
1972b15c8340SDag-Erling Smørgrav }
1973b15c8340SDag-Erling Smørgrav
1974b15c8340SDag-Erling Smørgrav /*
197538a52bd3SEd Maste * Determine whether a particular sftp command's arguments (if any) represent
197638a52bd3SEd Maste * local or remote files. The "cmdarg" argument specifies the actual argument
197738a52bd3SEd Maste * and accepts values 1 or 2.
1978b15c8340SDag-Erling Smørgrav */
1979b15c8340SDag-Erling Smørgrav static int
complete_is_remote(char * cmd,int cmdarg)198038a52bd3SEd Maste complete_is_remote(char *cmd, int cmdarg) {
1981b15c8340SDag-Erling Smørgrav int i;
1982b15c8340SDag-Erling Smørgrav
1983b15c8340SDag-Erling Smørgrav if (cmd == NULL)
1984b15c8340SDag-Erling Smørgrav return -1;
1985b15c8340SDag-Erling Smørgrav
1986b15c8340SDag-Erling Smørgrav for (i = 0; cmds[i].c; i++) {
198738a52bd3SEd Maste if (!strncasecmp(cmd, cmds[i].c, strlen(cmds[i].c))) {
198838a52bd3SEd Maste if (cmdarg == 1)
1989b15c8340SDag-Erling Smørgrav return cmds[i].t;
199038a52bd3SEd Maste else if (cmdarg == 2)
199138a52bd3SEd Maste return cmds[i].t2;
199238a52bd3SEd Maste break;
199338a52bd3SEd Maste }
1994b15c8340SDag-Erling Smørgrav }
1995b15c8340SDag-Erling Smørgrav
1996b15c8340SDag-Erling Smørgrav return -1;
1997b15c8340SDag-Erling Smørgrav }
1998b15c8340SDag-Erling Smørgrav
1999b15c8340SDag-Erling Smørgrav /* Autocomplete a filename "file" */
2000b15c8340SDag-Erling Smørgrav static int
complete_match(EditLine * el,struct sftp_conn * conn,char * remote_path,char * file,int remote,int lastarg,char quote,int terminated)2001b15c8340SDag-Erling Smørgrav complete_match(EditLine *el, struct sftp_conn *conn, char *remote_path,
2002b15c8340SDag-Erling Smørgrav char *file, int remote, int lastarg, char quote, int terminated)
2003b15c8340SDag-Erling Smørgrav {
2004b15c8340SDag-Erling Smørgrav glob_t g;
2005e4a9863fSDag-Erling Smørgrav char *tmp, *tmp2, ins[8];
20066888a9beSDag-Erling Smørgrav u_int i, hadglob, pwdlen, len, tmplen, filelen, cesc, isesc, isabs;
2007e4a9863fSDag-Erling Smørgrav int clen;
2008b15c8340SDag-Erling Smørgrav const LineInfo *lf;
2009b15c8340SDag-Erling Smørgrav
2010b15c8340SDag-Erling Smørgrav /* Glob from "file" location */
2011b15c8340SDag-Erling Smørgrav if (file == NULL)
2012b15c8340SDag-Erling Smørgrav tmp = xstrdup("*");
2013b15c8340SDag-Erling Smørgrav else
2014b15c8340SDag-Erling Smørgrav xasprintf(&tmp, "%s*", file);
2015b15c8340SDag-Erling Smørgrav
20166888a9beSDag-Erling Smørgrav /* Check if the path is absolute. */
201719261079SEd Maste isabs = path_absolute(tmp);
20186888a9beSDag-Erling Smørgrav
2019b15c8340SDag-Erling Smørgrav memset(&g, 0, sizeof(g));
2020b15c8340SDag-Erling Smørgrav if (remote != LOCAL) {
202169c72a57SEd Maste tmp = make_absolute_pwd_glob(tmp, remote_path);
2022edf85781SEd Maste sftp_glob(conn, tmp, GLOB_DOOFFS|GLOB_MARK, NULL, &g);
2023b15c8340SDag-Erling Smørgrav } else
2024535af610SEd Maste (void)glob(tmp, GLOB_DOOFFS|GLOB_MARK, NULL, &g);
2025b15c8340SDag-Erling Smørgrav
2026b15c8340SDag-Erling Smørgrav /* Determine length of pwd so we can trim completion display */
2027b15c8340SDag-Erling Smørgrav for (hadglob = tmplen = pwdlen = 0; tmp[tmplen] != 0; tmplen++) {
2028b15c8340SDag-Erling Smørgrav /* Terminate counting on first unescaped glob metacharacter */
2029b15c8340SDag-Erling Smørgrav if (tmp[tmplen] == '*' || tmp[tmplen] == '?') {
2030b15c8340SDag-Erling Smørgrav if (tmp[tmplen] != '*' || tmp[tmplen + 1] != '\0')
2031b15c8340SDag-Erling Smørgrav hadglob = 1;
2032b15c8340SDag-Erling Smørgrav break;
2033b15c8340SDag-Erling Smørgrav }
2034b15c8340SDag-Erling Smørgrav if (tmp[tmplen] == '\\' && tmp[tmplen + 1] != '\0')
2035b15c8340SDag-Erling Smørgrav tmplen++;
2036b15c8340SDag-Erling Smørgrav if (tmp[tmplen] == '/')
2037b15c8340SDag-Erling Smørgrav pwdlen = tmplen + 1; /* track last seen '/' */
2038b15c8340SDag-Erling Smørgrav }
2039e4a9863fSDag-Erling Smørgrav free(tmp);
2040a0ee8cc6SDag-Erling Smørgrav tmp = NULL;
2041b15c8340SDag-Erling Smørgrav
2042b15c8340SDag-Erling Smørgrav if (g.gl_matchc == 0)
2043b15c8340SDag-Erling Smørgrav goto out;
2044b15c8340SDag-Erling Smørgrav
2045b15c8340SDag-Erling Smørgrav if (g.gl_matchc > 1)
2046b15c8340SDag-Erling Smørgrav complete_display(g.gl_pathv, pwdlen);
2047b15c8340SDag-Erling Smørgrav
2048b15c8340SDag-Erling Smørgrav /* Don't try to extend globs */
2049b15c8340SDag-Erling Smørgrav if (file == NULL || hadglob)
2050b15c8340SDag-Erling Smørgrav goto out;
2051b15c8340SDag-Erling Smørgrav
2052b15c8340SDag-Erling Smørgrav tmp2 = complete_ambiguous(file, g.gl_pathv, g.gl_matchc);
20536888a9beSDag-Erling Smørgrav tmp = path_strip(tmp2, isabs ? NULL : remote_path);
2054e4a9863fSDag-Erling Smørgrav free(tmp2);
2055b15c8340SDag-Erling Smørgrav
2056b15c8340SDag-Erling Smørgrav if (tmp == NULL)
2057b15c8340SDag-Erling Smørgrav goto out;
2058b15c8340SDag-Erling Smørgrav
2059b15c8340SDag-Erling Smørgrav tmplen = strlen(tmp);
2060b15c8340SDag-Erling Smørgrav filelen = strlen(file);
2061b15c8340SDag-Erling Smørgrav
20626888a9beSDag-Erling Smørgrav /* Count the number of escaped characters in the input string. */
20636888a9beSDag-Erling Smørgrav cesc = isesc = 0;
20646888a9beSDag-Erling Smørgrav for (i = 0; i < filelen; i++) {
20656888a9beSDag-Erling Smørgrav if (!isesc && file[i] == '\\' && i + 1 < filelen){
20666888a9beSDag-Erling Smørgrav isesc = 1;
20676888a9beSDag-Erling Smørgrav cesc++;
20686888a9beSDag-Erling Smørgrav } else
20696888a9beSDag-Erling Smørgrav isesc = 0;
20706888a9beSDag-Erling Smørgrav }
20716888a9beSDag-Erling Smørgrav
20726888a9beSDag-Erling Smørgrav if (tmplen > (filelen - cesc)) {
20736888a9beSDag-Erling Smørgrav tmp2 = tmp + filelen - cesc;
2074b15c8340SDag-Erling Smørgrav len = strlen(tmp2);
2075b15c8340SDag-Erling Smørgrav /* quote argument on way out */
2076e4a9863fSDag-Erling Smørgrav for (i = 0; i < len; i += clen) {
2077e4a9863fSDag-Erling Smørgrav if ((clen = mblen(tmp2 + i, len - i)) < 0 ||
2078e4a9863fSDag-Erling Smørgrav (size_t)clen > sizeof(ins) - 2)
2079e4a9863fSDag-Erling Smørgrav fatal("invalid multibyte character");
2080b15c8340SDag-Erling Smørgrav ins[0] = '\\';
2081e4a9863fSDag-Erling Smørgrav memcpy(ins + 1, tmp2 + i, clen);
2082e4a9863fSDag-Erling Smørgrav ins[clen + 1] = '\0';
2083b15c8340SDag-Erling Smørgrav switch (tmp2[i]) {
2084b15c8340SDag-Erling Smørgrav case '\'':
2085b15c8340SDag-Erling Smørgrav case '"':
2086b15c8340SDag-Erling Smørgrav case '\\':
2087b15c8340SDag-Erling Smørgrav case '\t':
20884a421b63SDag-Erling Smørgrav case '[':
2089b15c8340SDag-Erling Smørgrav case ' ':
20906888a9beSDag-Erling Smørgrav case '#':
20916888a9beSDag-Erling Smørgrav case '*':
2092b15c8340SDag-Erling Smørgrav if (quote == '\0' || tmp2[i] == quote) {
2093b15c8340SDag-Erling Smørgrav if (el_insertstr(el, ins) == -1)
2094b15c8340SDag-Erling Smørgrav fatal("el_insertstr "
2095b15c8340SDag-Erling Smørgrav "failed.");
2096b15c8340SDag-Erling Smørgrav break;
2097b15c8340SDag-Erling Smørgrav }
2098b15c8340SDag-Erling Smørgrav /* FALLTHROUGH */
2099b15c8340SDag-Erling Smørgrav default:
2100b15c8340SDag-Erling Smørgrav if (el_insertstr(el, ins + 1) == -1)
2101b15c8340SDag-Erling Smørgrav fatal("el_insertstr failed.");
2102b15c8340SDag-Erling Smørgrav break;
2103b15c8340SDag-Erling Smørgrav }
2104b15c8340SDag-Erling Smørgrav }
2105b15c8340SDag-Erling Smørgrav }
2106b15c8340SDag-Erling Smørgrav
2107b15c8340SDag-Erling Smørgrav lf = el_line(el);
2108b15c8340SDag-Erling Smørgrav if (g.gl_matchc == 1) {
2109b15c8340SDag-Erling Smørgrav i = 0;
2110a0ee8cc6SDag-Erling Smørgrav if (!terminated && quote != '\0')
2111b15c8340SDag-Erling Smørgrav ins[i++] = quote;
2112b15c8340SDag-Erling Smørgrav if (*(lf->cursor - 1) != '/' &&
2113b15c8340SDag-Erling Smørgrav (lastarg || *(lf->cursor) != ' '))
2114b15c8340SDag-Erling Smørgrav ins[i++] = ' ';
2115b15c8340SDag-Erling Smørgrav ins[i] = '\0';
2116b15c8340SDag-Erling Smørgrav if (i > 0 && el_insertstr(el, ins) == -1)
2117b15c8340SDag-Erling Smørgrav fatal("el_insertstr failed.");
2118b15c8340SDag-Erling Smørgrav }
2119e4a9863fSDag-Erling Smørgrav free(tmp);
2120b15c8340SDag-Erling Smørgrav
2121b15c8340SDag-Erling Smørgrav out:
2122b15c8340SDag-Erling Smørgrav globfree(&g);
2123b15c8340SDag-Erling Smørgrav return g.gl_matchc;
2124b15c8340SDag-Erling Smørgrav }
2125b15c8340SDag-Erling Smørgrav
2126b15c8340SDag-Erling Smørgrav /* tab-completion hook function, called via libedit */
2127b15c8340SDag-Erling Smørgrav static unsigned char
complete(EditLine * el,int ch)2128b15c8340SDag-Erling Smørgrav complete(EditLine *el, int ch)
2129b15c8340SDag-Erling Smørgrav {
2130b15c8340SDag-Erling Smørgrav char **argv, *line, quote;
2131e4a9863fSDag-Erling Smørgrav int argc, carg;
2132e4a9863fSDag-Erling Smørgrav u_int cursor, len, terminated, ret = CC_ERROR;
2133b15c8340SDag-Erling Smørgrav const LineInfo *lf;
2134b15c8340SDag-Erling Smørgrav struct complete_ctx *complete_ctx;
2135b15c8340SDag-Erling Smørgrav
2136b15c8340SDag-Erling Smørgrav lf = el_line(el);
2137b15c8340SDag-Erling Smørgrav if (el_get(el, EL_CLIENTDATA, (void**)&complete_ctx) != 0)
213819261079SEd Maste fatal_f("el_get failed");
2139b15c8340SDag-Erling Smørgrav
2140b15c8340SDag-Erling Smørgrav /* Figure out which argument the cursor points to */
2141b15c8340SDag-Erling Smørgrav cursor = lf->cursor - lf->buffer;
2142fc1ba28aSDag-Erling Smørgrav line = xmalloc(cursor + 1);
2143b15c8340SDag-Erling Smørgrav memcpy(line, lf->buffer, cursor);
2144b15c8340SDag-Erling Smørgrav line[cursor] = '\0';
2145b15c8340SDag-Erling Smørgrav argv = makeargv(line, &carg, 1, "e, &terminated);
2146e4a9863fSDag-Erling Smørgrav free(line);
2147b15c8340SDag-Erling Smørgrav
2148b15c8340SDag-Erling Smørgrav /* Get all the arguments on the line */
2149b15c8340SDag-Erling Smørgrav len = lf->lastchar - lf->buffer;
2150fc1ba28aSDag-Erling Smørgrav line = xmalloc(len + 1);
2151b15c8340SDag-Erling Smørgrav memcpy(line, lf->buffer, len);
2152b15c8340SDag-Erling Smørgrav line[len] = '\0';
2153b15c8340SDag-Erling Smørgrav argv = makeargv(line, &argc, 1, NULL, NULL);
2154b15c8340SDag-Erling Smørgrav
2155b15c8340SDag-Erling Smørgrav /* Ensure cursor is at EOL or a argument boundary */
2156b15c8340SDag-Erling Smørgrav if (line[cursor] != ' ' && line[cursor] != '\0' &&
2157b15c8340SDag-Erling Smørgrav line[cursor] != '\n') {
2158e4a9863fSDag-Erling Smørgrav free(line);
2159b15c8340SDag-Erling Smørgrav return ret;
2160b15c8340SDag-Erling Smørgrav }
2161b15c8340SDag-Erling Smørgrav
2162b15c8340SDag-Erling Smørgrav if (carg == 0) {
2163b15c8340SDag-Erling Smørgrav /* Show all available commands */
2164b15c8340SDag-Erling Smørgrav complete_cmd_parse(el, NULL, argc == carg, '\0', 1);
2165b15c8340SDag-Erling Smørgrav ret = CC_REDISPLAY;
2166b15c8340SDag-Erling Smørgrav } else if (carg == 1 && cursor > 0 && line[cursor - 1] != ' ') {
2167b15c8340SDag-Erling Smørgrav /* Handle the command parsing */
2168b15c8340SDag-Erling Smørgrav if (complete_cmd_parse(el, argv[0], argc == carg,
2169b15c8340SDag-Erling Smørgrav quote, terminated) != 0)
2170b15c8340SDag-Erling Smørgrav ret = CC_REDISPLAY;
2171b15c8340SDag-Erling Smørgrav } else if (carg >= 1) {
2172b15c8340SDag-Erling Smørgrav /* Handle file parsing */
217338a52bd3SEd Maste int remote = 0;
217438a52bd3SEd Maste int i = 0, cmdarg = 0;
2175b15c8340SDag-Erling Smørgrav char *filematch = NULL;
2176b15c8340SDag-Erling Smørgrav
2177b15c8340SDag-Erling Smørgrav if (carg > 1 && line[cursor-1] != ' ')
2178b15c8340SDag-Erling Smørgrav filematch = argv[carg - 1];
2179b15c8340SDag-Erling Smørgrav
218038a52bd3SEd Maste for (i = 1; i < carg; i++) {
218138a52bd3SEd Maste /* Skip flags */
218238a52bd3SEd Maste if (argv[i][0] != '-')
218338a52bd3SEd Maste cmdarg++;
218438a52bd3SEd Maste }
218538a52bd3SEd Maste
218638a52bd3SEd Maste /*
218738a52bd3SEd Maste * If previous argument is complete, then offer completion
218838a52bd3SEd Maste * on the next one.
218938a52bd3SEd Maste */
219038a52bd3SEd Maste if (line[cursor - 1] == ' ')
219138a52bd3SEd Maste cmdarg++;
219238a52bd3SEd Maste
219338a52bd3SEd Maste remote = complete_is_remote(argv[0], cmdarg);
219438a52bd3SEd Maste
219538a52bd3SEd Maste if ((remote == REMOTE || remote == LOCAL) &&
2196b15c8340SDag-Erling Smørgrav complete_match(el, complete_ctx->conn,
2197b15c8340SDag-Erling Smørgrav *complete_ctx->remote_pathp, filematch,
2198b15c8340SDag-Erling Smørgrav remote, carg == argc, quote, terminated) != 0)
2199b15c8340SDag-Erling Smørgrav ret = CC_REDISPLAY;
2200b15c8340SDag-Erling Smørgrav }
2201b15c8340SDag-Erling Smørgrav
2202e4a9863fSDag-Erling Smørgrav free(line);
2203b15c8340SDag-Erling Smørgrav return ret;
2204b15c8340SDag-Erling Smørgrav }
2205b15c8340SDag-Erling Smørgrav #endif /* USE_LIBEDIT */
22065e8dbd04SDag-Erling Smørgrav
220747dd1d1bSDag-Erling Smørgrav static int
interactive_loop(struct sftp_conn * conn,char * file1,char * file2)2208b15c8340SDag-Erling Smørgrav interactive_loop(struct sftp_conn *conn, char *file1, char *file2)
2209efcad6b7SDag-Erling Smørgrav {
2210b15c8340SDag-Erling Smørgrav char *remote_path;
221147dd1d1bSDag-Erling Smørgrav char *dir = NULL, *startdir = NULL;
2212efcad6b7SDag-Erling Smørgrav char cmd[2048];
2213043840dfSDag-Erling Smørgrav int err, interactive;
22145e8dbd04SDag-Erling Smørgrav EditLine *el = NULL;
22155e8dbd04SDag-Erling Smørgrav #ifdef USE_LIBEDIT
22165e8dbd04SDag-Erling Smørgrav History *hl = NULL;
22175e8dbd04SDag-Erling Smørgrav HistEvent hev;
22185e8dbd04SDag-Erling Smørgrav extern char *__progname;
2219b15c8340SDag-Erling Smørgrav struct complete_ctx complete_ctx;
22205e8dbd04SDag-Erling Smørgrav
22215e8dbd04SDag-Erling Smørgrav if (!batchmode && isatty(STDIN_FILENO)) {
22225e8dbd04SDag-Erling Smørgrav if ((el = el_init(__progname, stdin, stdout, stderr)) == NULL)
22235e8dbd04SDag-Erling Smørgrav fatal("Couldn't initialise editline");
22245e8dbd04SDag-Erling Smørgrav if ((hl = history_init()) == NULL)
22255e8dbd04SDag-Erling Smørgrav fatal("Couldn't initialise editline history");
22265e8dbd04SDag-Erling Smørgrav history(hl, &hev, H_SETSIZE, 100);
22275e8dbd04SDag-Erling Smørgrav el_set(el, EL_HIST, history, hl);
22285e8dbd04SDag-Erling Smørgrav
22295e8dbd04SDag-Erling Smørgrav el_set(el, EL_PROMPT, prompt);
22305e8dbd04SDag-Erling Smørgrav el_set(el, EL_EDITOR, "emacs");
22315e8dbd04SDag-Erling Smørgrav el_set(el, EL_TERMINAL, NULL);
22325e8dbd04SDag-Erling Smørgrav el_set(el, EL_SIGNAL, 1);
22335e8dbd04SDag-Erling Smørgrav el_source(el, NULL);
2234b15c8340SDag-Erling Smørgrav
2235b15c8340SDag-Erling Smørgrav /* Tab Completion */
2236b15c8340SDag-Erling Smørgrav el_set(el, EL_ADDFN, "ftp-complete",
22374a421b63SDag-Erling Smørgrav "Context sensitive argument completion", complete);
2238b15c8340SDag-Erling Smørgrav complete_ctx.conn = conn;
2239b15c8340SDag-Erling Smørgrav complete_ctx.remote_pathp = &remote_path;
2240b15c8340SDag-Erling Smørgrav el_set(el, EL_CLIENTDATA, (void*)&complete_ctx);
2241b15c8340SDag-Erling Smørgrav el_set(el, EL_BIND, "^I", "ftp-complete", NULL);
2242f7167e0eSDag-Erling Smørgrav /* enable ctrl-left-arrow and ctrl-right-arrow */
2243f7167e0eSDag-Erling Smørgrav el_set(el, EL_BIND, "\\e[1;5C", "em-next-word", NULL);
224419261079SEd Maste el_set(el, EL_BIND, "\\e\\e[C", "em-next-word", NULL);
2245f7167e0eSDag-Erling Smørgrav el_set(el, EL_BIND, "\\e[1;5D", "ed-prev-word", NULL);
2246f7167e0eSDag-Erling Smørgrav el_set(el, EL_BIND, "\\e\\e[D", "ed-prev-word", NULL);
2247f7167e0eSDag-Erling Smørgrav /* make ^w match ksh behaviour */
2248f7167e0eSDag-Erling Smørgrav el_set(el, EL_BIND, "^w", "ed-delete-prev-word", NULL);
22495e8dbd04SDag-Erling Smørgrav }
22505e8dbd04SDag-Erling Smørgrav #endif /* USE_LIBEDIT */
2251efcad6b7SDag-Erling Smørgrav
2252edf85781SEd Maste if ((remote_path = sftp_realpath(conn, ".")) == NULL)
2253efcad6b7SDag-Erling Smørgrav fatal("Need cwd");
225447dd1d1bSDag-Erling Smørgrav startdir = xstrdup(remote_path);
2255efcad6b7SDag-Erling Smørgrav
2256efcad6b7SDag-Erling Smørgrav if (file1 != NULL) {
2257efcad6b7SDag-Erling Smørgrav dir = xstrdup(file1);
2258edf85781SEd Maste dir = sftp_make_absolute(dir, remote_path);
2259efcad6b7SDag-Erling Smørgrav
2260edf85781SEd Maste if (sftp_remote_is_dir(conn, dir) && file2 == NULL) {
2261e4a9863fSDag-Erling Smørgrav if (!quiet)
2262076ad2f8SDag-Erling Smørgrav mprintf("Changing to: %s\n", dir);
2263efcad6b7SDag-Erling Smørgrav snprintf(cmd, sizeof cmd, "cd \"%s\"", dir);
2264b15c8340SDag-Erling Smørgrav if (parse_dispatch_command(conn, cmd,
226519261079SEd Maste &remote_path, startdir, 1, 0) != 0) {
2266e4a9863fSDag-Erling Smørgrav free(dir);
226747dd1d1bSDag-Erling Smørgrav free(startdir);
2268e4a9863fSDag-Erling Smørgrav free(remote_path);
2269e4a9863fSDag-Erling Smørgrav free(conn);
2270efcad6b7SDag-Erling Smørgrav return (-1);
22715e8dbd04SDag-Erling Smørgrav }
2272efcad6b7SDag-Erling Smørgrav } else {
22736888a9beSDag-Erling Smørgrav /* XXX this is wrong wrt quoting */
2274e4a9863fSDag-Erling Smørgrav snprintf(cmd, sizeof cmd, "get%s %s%s%s",
2275e4a9863fSDag-Erling Smørgrav global_aflag ? " -a" : "", dir,
2276e4a9863fSDag-Erling Smørgrav file2 == NULL ? "" : " ",
2277e4a9863fSDag-Erling Smørgrav file2 == NULL ? "" : file2);
2278b15c8340SDag-Erling Smørgrav err = parse_dispatch_command(conn, cmd,
227919261079SEd Maste &remote_path, startdir, 1, 0);
2280e4a9863fSDag-Erling Smørgrav free(dir);
228147dd1d1bSDag-Erling Smørgrav free(startdir);
2282e4a9863fSDag-Erling Smørgrav free(remote_path);
2283e4a9863fSDag-Erling Smørgrav free(conn);
2284efcad6b7SDag-Erling Smørgrav return (err);
2285efcad6b7SDag-Erling Smørgrav }
2286e4a9863fSDag-Erling Smørgrav free(dir);
2287efcad6b7SDag-Erling Smørgrav }
2288efcad6b7SDag-Erling Smørgrav
2289bc5531deSDag-Erling Smørgrav setvbuf(stdout, NULL, _IOLBF, 0);
2290bc5531deSDag-Erling Smørgrav setvbuf(infile, NULL, _IOLBF, 0);
2291efcad6b7SDag-Erling Smørgrav
2292043840dfSDag-Erling Smørgrav interactive = !batchmode && isatty(STDIN_FILENO);
2293efcad6b7SDag-Erling Smørgrav err = 0;
2294efcad6b7SDag-Erling Smørgrav for (;;) {
2295e9e8876aSEd Maste struct sigaction sa;
2296e9e8876aSEd Maste
2297e9e8876aSEd Maste interrupted = 0;
2298e9e8876aSEd Maste memset(&sa, 0, sizeof(sa));
2299e9e8876aSEd Maste sa.sa_handler = interactive ? read_interrupt : killchild;
2300e9e8876aSEd Maste if (sigaction(SIGINT, &sa, NULL) == -1) {
2301e9e8876aSEd Maste debug3("sigaction(%s): %s", strsignal(SIGINT),
2302e9e8876aSEd Maste strerror(errno));
2303e9e8876aSEd Maste break;
2304e9e8876aSEd Maste }
23055e8dbd04SDag-Erling Smørgrav if (el == NULL) {
2306*0fdf8faeSEd Maste if (interactive) {
2307efcad6b7SDag-Erling Smørgrav printf("sftp> ");
2308*0fdf8faeSEd Maste fflush(stdout);
2309*0fdf8faeSEd Maste }
2310efcad6b7SDag-Erling Smørgrav if (fgets(cmd, sizeof(cmd), infile) == NULL) {
2311043840dfSDag-Erling Smørgrav if (interactive)
2312efcad6b7SDag-Erling Smørgrav printf("\n");
2313e9e8876aSEd Maste if (interrupted)
2314e9e8876aSEd Maste continue;
2315efcad6b7SDag-Erling Smørgrav break;
2316efcad6b7SDag-Erling Smørgrav }
23175e8dbd04SDag-Erling Smørgrav } else {
23185e8dbd04SDag-Erling Smørgrav #ifdef USE_LIBEDIT
23195e8dbd04SDag-Erling Smørgrav const char *line;
23205e8dbd04SDag-Erling Smørgrav int count = 0;
23215e8dbd04SDag-Erling Smørgrav
2322b15c8340SDag-Erling Smørgrav if ((line = el_gets(el, &count)) == NULL ||
2323b15c8340SDag-Erling Smørgrav count <= 0) {
2324043840dfSDag-Erling Smørgrav printf("\n");
232519261079SEd Maste if (interrupted)
232619261079SEd Maste continue;
23275e8dbd04SDag-Erling Smørgrav break;
2328043840dfSDag-Erling Smørgrav }
23295e8dbd04SDag-Erling Smørgrav history(hl, &hev, H_ENTER, line);
23305e8dbd04SDag-Erling Smørgrav if (strlcpy(cmd, line, sizeof(cmd)) >= sizeof(cmd)) {
23315e8dbd04SDag-Erling Smørgrav fprintf(stderr, "Error: input line too long\n");
23325e8dbd04SDag-Erling Smørgrav continue;
23335e8dbd04SDag-Erling Smørgrav }
23345e8dbd04SDag-Erling Smørgrav #endif /* USE_LIBEDIT */
23355e8dbd04SDag-Erling Smørgrav }
2336efcad6b7SDag-Erling Smørgrav
233719261079SEd Maste cmd[strcspn(cmd, "\n")] = '\0';
2338efcad6b7SDag-Erling Smørgrav
2339d74d50a8SDag-Erling Smørgrav /* Handle user interrupts gracefully during commands */
2340d74d50a8SDag-Erling Smørgrav interrupted = 0;
234119261079SEd Maste ssh_signal(SIGINT, cmd_interrupt);
2342d74d50a8SDag-Erling Smørgrav
2343b15c8340SDag-Erling Smørgrav err = parse_dispatch_command(conn, cmd, &remote_path,
234419261079SEd Maste startdir, batchmode, !interactive && el == NULL);
2345efcad6b7SDag-Erling Smørgrav if (err != 0)
2346efcad6b7SDag-Erling Smørgrav break;
2347efcad6b7SDag-Erling Smørgrav }
234819261079SEd Maste ssh_signal(SIGCHLD, SIG_DFL);
2349e4a9863fSDag-Erling Smørgrav free(remote_path);
235047dd1d1bSDag-Erling Smørgrav free(startdir);
2351e4a9863fSDag-Erling Smørgrav free(conn);
2352efcad6b7SDag-Erling Smørgrav
2353043840dfSDag-Erling Smørgrav #ifdef USE_LIBEDIT
2354043840dfSDag-Erling Smørgrav if (el != NULL)
2355043840dfSDag-Erling Smørgrav el_end(el);
2356043840dfSDag-Erling Smørgrav #endif /* USE_LIBEDIT */
2357043840dfSDag-Erling Smørgrav
2358efcad6b7SDag-Erling Smørgrav /* err == 1 signifies normal "quit" exit */
2359efcad6b7SDag-Erling Smørgrav return (err >= 0 ? 0 : -1);
2360efcad6b7SDag-Erling Smørgrav }
2361d0c8c0bcSDag-Erling Smørgrav
2362ae1f160dSDag-Erling Smørgrav static void
connect_to_server(char * path,char ** args,int * in,int * out)2363d95e11bfSDag-Erling Smørgrav connect_to_server(char *path, char **args, int *in, int *out)
23641e8db6e2SBrian Feldman {
23651e8db6e2SBrian Feldman int c_in, c_out;
23661e8db6e2SBrian Feldman #ifdef USE_PIPES
23671e8db6e2SBrian Feldman int pin[2], pout[2];
2368ee21a45fSDag-Erling Smørgrav
23691e8db6e2SBrian Feldman if ((pipe(pin) == -1) || (pipe(pout) == -1))
23701e8db6e2SBrian Feldman fatal("pipe: %s", strerror(errno));
23711e8db6e2SBrian Feldman *in = pin[0];
23721e8db6e2SBrian Feldman *out = pout[1];
23731e8db6e2SBrian Feldman c_in = pout[0];
23741e8db6e2SBrian Feldman c_out = pin[1];
23751e8db6e2SBrian Feldman #else /* USE_PIPES */
23761e8db6e2SBrian Feldman int inout[2];
2377ee21a45fSDag-Erling Smørgrav
23781e8db6e2SBrian Feldman if (socketpair(AF_UNIX, SOCK_STREAM, 0, inout) == -1)
23791e8db6e2SBrian Feldman fatal("socketpair: %s", strerror(errno));
23801e8db6e2SBrian Feldman *in = *out = inout[0];
23811e8db6e2SBrian Feldman c_in = c_out = inout[1];
23821e8db6e2SBrian Feldman #endif /* USE_PIPES */
23831e8db6e2SBrian Feldman
2384d95e11bfSDag-Erling Smørgrav if ((sshpid = fork()) == -1)
23851e8db6e2SBrian Feldman fatal("fork: %s", strerror(errno));
2386d95e11bfSDag-Erling Smørgrav else if (sshpid == 0) {
23871e8db6e2SBrian Feldman if ((dup2(c_in, STDIN_FILENO) == -1) ||
23881e8db6e2SBrian Feldman (dup2(c_out, STDOUT_FILENO) == -1)) {
23891e8db6e2SBrian Feldman fprintf(stderr, "dup2: %s\n", strerror(errno));
2390d74d50a8SDag-Erling Smørgrav _exit(1);
23911e8db6e2SBrian Feldman }
23921e8db6e2SBrian Feldman close(*in);
23931e8db6e2SBrian Feldman close(*out);
23941e8db6e2SBrian Feldman close(c_in);
23951e8db6e2SBrian Feldman close(c_out);
2396d74d50a8SDag-Erling Smørgrav
2397d74d50a8SDag-Erling Smørgrav /*
2398d74d50a8SDag-Erling Smørgrav * The underlying ssh is in the same process group, so we must
2399d74d50a8SDag-Erling Smørgrav * ignore SIGINT if we want to gracefully abort commands,
2400d74d50a8SDag-Erling Smørgrav * otherwise the signal will make it to the ssh process and
2401b15c8340SDag-Erling Smørgrav * kill it too. Contrawise, since sftp sends SIGTERMs to the
2402b15c8340SDag-Erling Smørgrav * underlying ssh, it must *not* ignore that signal.
2403d74d50a8SDag-Erling Smørgrav */
240419261079SEd Maste ssh_signal(SIGINT, SIG_IGN);
240519261079SEd Maste ssh_signal(SIGTERM, SIG_DFL);
2406d74d50a8SDag-Erling Smørgrav execvp(path, args);
2407ae1f160dSDag-Erling Smørgrav fprintf(stderr, "exec: %s: %s\n", path, strerror(errno));
2408d74d50a8SDag-Erling Smørgrav _exit(1);
24091e8db6e2SBrian Feldman }
24101e8db6e2SBrian Feldman
241119261079SEd Maste ssh_signal(SIGTERM, killchild);
241219261079SEd Maste ssh_signal(SIGINT, killchild);
241319261079SEd Maste ssh_signal(SIGHUP, killchild);
241419261079SEd Maste ssh_signal(SIGTSTP, suspchild);
241519261079SEd Maste ssh_signal(SIGTTIN, suspchild);
241619261079SEd Maste ssh_signal(SIGTTOU, suspchild);
241719261079SEd Maste ssh_signal(SIGCHLD, sigchld_handler);
24181e8db6e2SBrian Feldman close(c_in);
24191e8db6e2SBrian Feldman close(c_out);
24201e8db6e2SBrian Feldman }
24211e8db6e2SBrian Feldman
2422ae1f160dSDag-Erling Smørgrav static void
usage(void)24231e8db6e2SBrian Feldman usage(void)
24241e8db6e2SBrian Feldman {
2425ae1f160dSDag-Erling Smørgrav extern char *__progname;
2426ae1f160dSDag-Erling Smørgrav
2427ae1f160dSDag-Erling Smørgrav fprintf(stderr,
242819261079SEd Maste "usage: %s [-46AaCfNpqrv] [-B buffer_size] [-b batchfile] [-c cipher]\n"
242938a52bd3SEd Maste " [-D sftp_server_command] [-F ssh_config] [-i identity_file]\n"
243019261079SEd Maste " [-J destination] [-l limit] [-o ssh_option] [-P port]\n"
243119261079SEd Maste " [-R num_requests] [-S program] [-s subsystem | sftp_server]\n"
2432f374ba41SEd Maste " [-X sftp_option] destination\n",
243347dd1d1bSDag-Erling Smørgrav __progname);
24341e8db6e2SBrian Feldman exit(1);
24351e8db6e2SBrian Feldman }
24361e8db6e2SBrian Feldman
24371e8db6e2SBrian Feldman int
main(int argc,char ** argv)24381e8db6e2SBrian Feldman main(int argc, char **argv)
24391e8db6e2SBrian Feldman {
244038a52bd3SEd Maste int r, in, out, ch, err, tmp, port = -1, noisy = 0;
244138a52bd3SEd Maste char *host = NULL, *user, *cp, **cpp, *file2 = NULL;
244219261079SEd Maste int debug_level = 0;
2443ae1f160dSDag-Erling Smørgrav char *file1 = NULL, *sftp_server = NULL;
2444ae1f160dSDag-Erling Smørgrav char *ssh_program = _PATH_SSH_PROGRAM, *sftp_direct = NULL;
24454a421b63SDag-Erling Smørgrav const char *errstr;
2446ae1f160dSDag-Erling Smørgrav LogLevel ll = SYSLOG_LEVEL_INFO;
2447ae1f160dSDag-Erling Smørgrav arglist args;
24481e8db6e2SBrian Feldman extern int optind;
24491e8db6e2SBrian Feldman extern char *optarg;
2450b15c8340SDag-Erling Smørgrav struct sftp_conn *conn;
245119261079SEd Maste size_t copy_buffer_len = 0;
245219261079SEd Maste size_t num_requests = 0;
2453f374ba41SEd Maste long long llv, limit_kbps = 0;
24541e8db6e2SBrian Feldman
2455021d409fSDag-Erling Smørgrav /* Ensure that fds 0, 1 and 2 are open or directed to /dev/null */
2456021d409fSDag-Erling Smørgrav sanitise_stdfd();
2457ca86bcf2SDag-Erling Smørgrav msetlocale();
2458021d409fSDag-Erling Smørgrav
2459d95e11bfSDag-Erling Smørgrav __progname = ssh_get_progname(argv[0]);
2460021d409fSDag-Erling Smørgrav memset(&args, '\0', sizeof(args));
2461ae1f160dSDag-Erling Smørgrav args.list = NULL;
2462761efaa7SDag-Erling Smørgrav addargs(&args, "%s", ssh_program);
2463ae1f160dSDag-Erling Smørgrav addargs(&args, "-oForwardX11 no");
2464021d409fSDag-Erling Smørgrav addargs(&args, "-oPermitLocalCommand no");
2465ae1f160dSDag-Erling Smørgrav addargs(&args, "-oClearAllForwardings yes");
2466efcad6b7SDag-Erling Smørgrav
2467ae1f160dSDag-Erling Smørgrav ll = SYSLOG_LEVEL_INFO;
2468efcad6b7SDag-Erling Smørgrav infile = stdin;
24691e8db6e2SBrian Feldman
2470b15c8340SDag-Erling Smørgrav while ((ch = getopt(argc, argv,
2471f374ba41SEd Maste "1246AafhNpqrvCc:D:i:l:o:s:S:b:B:F:J:P:R:X:")) != -1) {
24721e8db6e2SBrian Feldman switch (ch) {
2473b15c8340SDag-Erling Smørgrav /* Passed through to ssh(1) */
247419261079SEd Maste case 'A':
2475b15c8340SDag-Erling Smørgrav case '4':
2476b15c8340SDag-Erling Smørgrav case '6':
24771e8db6e2SBrian Feldman case 'C':
2478b15c8340SDag-Erling Smørgrav addargs(&args, "-%c", ch);
2479b15c8340SDag-Erling Smørgrav break;
2480b15c8340SDag-Erling Smørgrav /* Passed through to ssh(1) with argument */
2481b15c8340SDag-Erling Smørgrav case 'F':
248219261079SEd Maste case 'J':
2483b15c8340SDag-Erling Smørgrav case 'c':
2484b15c8340SDag-Erling Smørgrav case 'i':
2485b15c8340SDag-Erling Smørgrav case 'o':
2486b15c8340SDag-Erling Smørgrav addargs(&args, "-%c", ch);
2487b15c8340SDag-Erling Smørgrav addargs(&args, "%s", optarg);
2488b15c8340SDag-Erling Smørgrav break;
2489b15c8340SDag-Erling Smørgrav case 'q':
2490e4a9863fSDag-Erling Smørgrav ll = SYSLOG_LEVEL_ERROR;
2491e4a9863fSDag-Erling Smørgrav quiet = 1;
2492b15c8340SDag-Erling Smørgrav showprogress = 0;
2493b15c8340SDag-Erling Smørgrav addargs(&args, "-%c", ch);
2494b15c8340SDag-Erling Smørgrav break;
2495b15c8340SDag-Erling Smørgrav case 'P':
249647dd1d1bSDag-Erling Smørgrav port = a2port(optarg);
249747dd1d1bSDag-Erling Smørgrav if (port <= 0)
249847dd1d1bSDag-Erling Smørgrav fatal("Bad port \"%s\"\n", optarg);
24991e8db6e2SBrian Feldman break;
25001e8db6e2SBrian Feldman case 'v':
2501ae1f160dSDag-Erling Smørgrav if (debug_level < 3) {
2502ae1f160dSDag-Erling Smørgrav addargs(&args, "-v");
2503ae1f160dSDag-Erling Smørgrav ll = SYSLOG_LEVEL_DEBUG1 + debug_level;
2504ae1f160dSDag-Erling Smørgrav }
2505ae1f160dSDag-Erling Smørgrav debug_level++;
25061e8db6e2SBrian Feldman break;
25071e8db6e2SBrian Feldman case '1':
250819261079SEd Maste fatal("SSH protocol v.1 is no longer supported");
25091e8db6e2SBrian Feldman break;
2510b15c8340SDag-Erling Smørgrav case '2':
251119261079SEd Maste /* accept silently */
25121e8db6e2SBrian Feldman break;
2513e4a9863fSDag-Erling Smørgrav case 'a':
2514e4a9863fSDag-Erling Smørgrav global_aflag = 1;
2515e4a9863fSDag-Erling Smørgrav break;
2516b15c8340SDag-Erling Smørgrav case 'B':
2517b15c8340SDag-Erling Smørgrav copy_buffer_len = strtol(optarg, &cp, 10);
2518b15c8340SDag-Erling Smørgrav if (copy_buffer_len == 0 || *cp != '\0')
2519b15c8340SDag-Erling Smørgrav fatal("Invalid buffer size \"%s\"", optarg);
25201e8db6e2SBrian Feldman break;
25211e8db6e2SBrian Feldman case 'b':
2522efcad6b7SDag-Erling Smørgrav if (batchmode)
2523efcad6b7SDag-Erling Smørgrav fatal("Batch file already specified.");
2524efcad6b7SDag-Erling Smørgrav
2525efcad6b7SDag-Erling Smørgrav /* Allow "-" as stdin */
2526efcad6b7SDag-Erling Smørgrav if (strcmp(optarg, "-") != 0 &&
2527efcad6b7SDag-Erling Smørgrav (infile = fopen(optarg, "r")) == NULL)
25281e8db6e2SBrian Feldman fatal("%s (%s).", strerror(errno), optarg);
2529d0c8c0bcSDag-Erling Smørgrav showprogress = 0;
2530e4a9863fSDag-Erling Smørgrav quiet = batchmode = 1;
25315e8dbd04SDag-Erling Smørgrav addargs(&args, "-obatchmode yes");
25321e8db6e2SBrian Feldman break;
2533f7167e0eSDag-Erling Smørgrav case 'f':
2534f7167e0eSDag-Erling Smørgrav global_fflag = 1;
2535f7167e0eSDag-Erling Smørgrav break;
253619261079SEd Maste case 'N':
253719261079SEd Maste noisy = 1; /* Used to clear quiet mode after getopt */
253819261079SEd Maste break;
2539b15c8340SDag-Erling Smørgrav case 'p':
2540b15c8340SDag-Erling Smørgrav global_pflag = 1;
2541b15c8340SDag-Erling Smørgrav break;
2542b15c8340SDag-Erling Smørgrav case 'D':
2543ae1f160dSDag-Erling Smørgrav sftp_direct = optarg;
2544ae1f160dSDag-Erling Smørgrav break;
25454a421b63SDag-Erling Smørgrav case 'l':
25464a421b63SDag-Erling Smørgrav limit_kbps = strtonum(optarg, 1, 100 * 1024 * 1024,
25474a421b63SDag-Erling Smørgrav &errstr);
25484a421b63SDag-Erling Smørgrav if (errstr != NULL)
25494a421b63SDag-Erling Smørgrav usage();
25504a421b63SDag-Erling Smørgrav limit_kbps *= 1024; /* kbps */
25514a421b63SDag-Erling Smørgrav break;
2552b15c8340SDag-Erling Smørgrav case 'r':
2553b15c8340SDag-Erling Smørgrav global_rflag = 1;
2554ae1f160dSDag-Erling Smørgrav break;
2555ae1f160dSDag-Erling Smørgrav case 'R':
2556ae1f160dSDag-Erling Smørgrav num_requests = strtol(optarg, &cp, 10);
2557ae1f160dSDag-Erling Smørgrav if (num_requests == 0 || *cp != '\0')
2558ae1f160dSDag-Erling Smørgrav fatal("Invalid number of requests \"%s\"",
2559ae1f160dSDag-Erling Smørgrav optarg);
2560ae1f160dSDag-Erling Smørgrav break;
2561b15c8340SDag-Erling Smørgrav case 's':
2562b15c8340SDag-Erling Smørgrav sftp_server = optarg;
2563b15c8340SDag-Erling Smørgrav break;
2564b15c8340SDag-Erling Smørgrav case 'S':
2565b15c8340SDag-Erling Smørgrav ssh_program = optarg;
2566b15c8340SDag-Erling Smørgrav replacearg(&args, 0, "%s", ssh_program);
2567b15c8340SDag-Erling Smørgrav break;
2568f374ba41SEd Maste case 'X':
2569f374ba41SEd Maste /* Please keep in sync with ssh.c -X */
2570f374ba41SEd Maste if (strncmp(optarg, "buffer=", 7) == 0) {
2571f374ba41SEd Maste r = scan_scaled(optarg + 7, &llv);
2572f374ba41SEd Maste if (r == 0 && (llv <= 0 || llv > 256 * 1024)) {
2573f374ba41SEd Maste r = -1;
2574f374ba41SEd Maste errno = EINVAL;
2575f374ba41SEd Maste }
2576f374ba41SEd Maste if (r == -1) {
2577f374ba41SEd Maste fatal("Invalid buffer size \"%s\": %s",
2578f374ba41SEd Maste optarg + 7, strerror(errno));
2579f374ba41SEd Maste }
2580f374ba41SEd Maste copy_buffer_len = (size_t)llv;
2581f374ba41SEd Maste } else if (strncmp(optarg, "nrequests=", 10) == 0) {
2582f374ba41SEd Maste llv = strtonum(optarg + 10, 1, 256 * 1024,
2583f374ba41SEd Maste &errstr);
2584f374ba41SEd Maste if (errstr != NULL) {
2585f374ba41SEd Maste fatal("Invalid number of requests "
2586f374ba41SEd Maste "\"%s\": %s", optarg + 10, errstr);
2587f374ba41SEd Maste }
2588f374ba41SEd Maste num_requests = (size_t)llv;
2589f374ba41SEd Maste } else {
2590f374ba41SEd Maste fatal("Invalid -X option");
2591f374ba41SEd Maste }
2592f374ba41SEd Maste break;
25931e8db6e2SBrian Feldman case 'h':
25941e8db6e2SBrian Feldman default:
25951e8db6e2SBrian Feldman usage();
25961e8db6e2SBrian Feldman }
25971e8db6e2SBrian Feldman }
25981e8db6e2SBrian Feldman
259919261079SEd Maste /* Do this last because we want the user to be able to override it */
260019261079SEd Maste addargs(&args, "-oForwardAgent no");
260119261079SEd Maste
260252028650SDag-Erling Smørgrav if (!isatty(STDERR_FILENO))
260352028650SDag-Erling Smørgrav showprogress = 0;
260452028650SDag-Erling Smørgrav
260519261079SEd Maste if (noisy)
260619261079SEd Maste quiet = 0;
260719261079SEd Maste
2608545d5ecaSDag-Erling Smørgrav log_init(argv[0], ll, SYSLOG_FACILITY_USER, 1);
2609545d5ecaSDag-Erling Smørgrav
2610ae1f160dSDag-Erling Smørgrav if (sftp_direct == NULL) {
26111e8db6e2SBrian Feldman if (optind == argc || argc > (optind + 2))
26121e8db6e2SBrian Feldman usage();
261347dd1d1bSDag-Erling Smørgrav argv += optind;
26141e8db6e2SBrian Feldman
261547dd1d1bSDag-Erling Smørgrav switch (parse_uri("sftp", *argv, &user, &host, &tmp, &file1)) {
261647dd1d1bSDag-Erling Smørgrav case -1:
26171e8db6e2SBrian Feldman usage();
261847dd1d1bSDag-Erling Smørgrav break;
261947dd1d1bSDag-Erling Smørgrav case 0:
262047dd1d1bSDag-Erling Smørgrav if (tmp != -1)
262147dd1d1bSDag-Erling Smørgrav port = tmp;
262247dd1d1bSDag-Erling Smørgrav break;
262347dd1d1bSDag-Erling Smørgrav default:
262419261079SEd Maste /* Try with user, host and path. */
262547dd1d1bSDag-Erling Smørgrav if (parse_user_host_path(*argv, &user, &host,
262619261079SEd Maste &file1) == 0)
262719261079SEd Maste break;
262819261079SEd Maste /* Try with user and host. */
262919261079SEd Maste if (parse_user_host_port(*argv, &user, &host, NULL)
263019261079SEd Maste == 0)
263119261079SEd Maste break;
263247dd1d1bSDag-Erling Smørgrav /* Treat as a plain hostname. */
263347dd1d1bSDag-Erling Smørgrav host = xstrdup(*argv);
26341e8db6e2SBrian Feldman host = cleanhostname(host);
263547dd1d1bSDag-Erling Smørgrav break;
263647dd1d1bSDag-Erling Smørgrav }
263747dd1d1bSDag-Erling Smørgrav file2 = *(argv + 1);
263847dd1d1bSDag-Erling Smørgrav
26391e8db6e2SBrian Feldman if (!*host) {
26401e8db6e2SBrian Feldman fprintf(stderr, "Missing hostname\n");
26411e8db6e2SBrian Feldman usage();
26421e8db6e2SBrian Feldman }
26431e8db6e2SBrian Feldman
264447dd1d1bSDag-Erling Smørgrav if (port != -1)
264547dd1d1bSDag-Erling Smørgrav addargs(&args, "-oPort %d", port);
264647dd1d1bSDag-Erling Smørgrav if (user != NULL) {
264747dd1d1bSDag-Erling Smørgrav addargs(&args, "-l");
264847dd1d1bSDag-Erling Smørgrav addargs(&args, "%s", user);
264947dd1d1bSDag-Erling Smørgrav }
26501e8db6e2SBrian Feldman
2651ae1f160dSDag-Erling Smørgrav /* no subsystem if the server-spec contains a '/' */
2652ae1f160dSDag-Erling Smørgrav if (sftp_server == NULL || strchr(sftp_server, '/') == NULL)
2653ae1f160dSDag-Erling Smørgrav addargs(&args, "-s");
2654ae1f160dSDag-Erling Smørgrav
2655b15c8340SDag-Erling Smørgrav addargs(&args, "--");
2656ae1f160dSDag-Erling Smørgrav addargs(&args, "%s", host);
2657ae1f160dSDag-Erling Smørgrav addargs(&args, "%s", (sftp_server != NULL ?
2658ae1f160dSDag-Erling Smørgrav sftp_server : "sftp"));
26591e8db6e2SBrian Feldman
2660d95e11bfSDag-Erling Smørgrav connect_to_server(ssh_program, args.list, &in, &out);
2661ae1f160dSDag-Erling Smørgrav } else {
266238a52bd3SEd Maste if ((r = argv_split(sftp_direct, &tmp, &cpp, 1)) != 0)
266338a52bd3SEd Maste fatal_r(r, "Parse -D arguments");
266438a52bd3SEd Maste if (cpp[0] == 0)
266538a52bd3SEd Maste fatal("No sftp server specified via -D");
266638a52bd3SEd Maste connect_to_server(cpp[0], cpp, &in, &out);
266738a52bd3SEd Maste argv_free(cpp, tmp);
2668ae1f160dSDag-Erling Smørgrav }
2669021d409fSDag-Erling Smørgrav freeargs(&args);
26701e8db6e2SBrian Feldman
2671edf85781SEd Maste conn = sftp_init(in, out, copy_buffer_len, num_requests, limit_kbps);
2672b15c8340SDag-Erling Smørgrav if (conn == NULL)
2673b15c8340SDag-Erling Smørgrav fatal("Couldn't initialise connection to server");
2674b15c8340SDag-Erling Smørgrav
2675e4a9863fSDag-Erling Smørgrav if (!quiet) {
2676b15c8340SDag-Erling Smørgrav if (sftp_direct == NULL)
2677b15c8340SDag-Erling Smørgrav fprintf(stderr, "Connected to %s.\n", host);
2678b15c8340SDag-Erling Smørgrav else
2679b15c8340SDag-Erling Smørgrav fprintf(stderr, "Attached to %s.\n", sftp_direct);
2680b15c8340SDag-Erling Smørgrav }
2681b15c8340SDag-Erling Smørgrav
2682b15c8340SDag-Erling Smørgrav err = interactive_loop(conn, file1, file2);
26831e8db6e2SBrian Feldman
268483d2307dSDag-Erling Smørgrav #if !defined(USE_PIPES)
268583d2307dSDag-Erling Smørgrav shutdown(in, SHUT_RDWR);
268683d2307dSDag-Erling Smørgrav shutdown(out, SHUT_RDWR);
268783d2307dSDag-Erling Smørgrav #endif
268883d2307dSDag-Erling Smørgrav
26891e8db6e2SBrian Feldman close(in);
26901e8db6e2SBrian Feldman close(out);
2691efcad6b7SDag-Erling Smørgrav if (batchmode)
26921e8db6e2SBrian Feldman fclose(infile);
26931e8db6e2SBrian Feldman
2694190cef3dSDag-Erling Smørgrav while (waitpid(sshpid, NULL, 0) == -1 && sshpid > 1)
2695545d5ecaSDag-Erling Smørgrav if (errno != EINTR)
2696545d5ecaSDag-Erling Smørgrav fatal("Couldn't wait for ssh process: %s",
2697545d5ecaSDag-Erling Smørgrav strerror(errno));
26981e8db6e2SBrian Feldman
2699d0c8c0bcSDag-Erling Smørgrav exit(err == 0 ? 0 : 1);
27001e8db6e2SBrian Feldman }
2701