blob: 8621a4409b7aab658c3b31fbd6a544ae3c7291a8 [file] [log] [blame]
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001/*
Damien Millere4340be2000-09-16 13:29:08 +11002 * scp - secure remote copy. This is basically patched BSD rcp which
3 * uses ssh to do the data transfer (instead of using rcmd).
Damien Miller4af51302000-04-16 11:18:38 +10004 *
Damien Millere4340be2000-09-16 13:29:08 +11005 * NOTE: This version should NOT be suid root. (This uses ssh to
6 * do the transfer and ssh has the necessary privileges.)
Damien Miller4af51302000-04-16 11:18:38 +10007 *
Damien Miller95def091999-11-25 00:26:21 +11008 * 1995 Timo Rinne <tri@iki.fi>, Tatu Ylonen <ylo@cs.hut.fi>
Damien Miller4af51302000-04-16 11:18:38 +10009 *
Damien Millere4340be2000-09-16 13:29:08 +110010 * As far as I am concerned, the code I have written for this software
11 * can be used freely for any purpose. Any derived versions of this
12 * software must be clearly marked as such, and if the derived work is
13 * incompatible with the protocol description in the RFC file, it must be
14 * called by a name other than "ssh" or "Secure Shell".
15 */
16/*
Ben Lindstrom92a2e382001-03-05 06:59:27 +000017 * Copyright (c) 1999 Theo de Raadt. All rights reserved.
18 * Copyright (c) 1999 Aaron Campbell. All rights reserved.
Damien Millere4340be2000-09-16 13:29:08 +110019 *
20 * Redistribution and use in source and binary forms, with or without
21 * modification, are permitted provided that the following conditions
22 * are met:
23 * 1. Redistributions of source code must retain the above copyright
24 * notice, this list of conditions and the following disclaimer.
25 * 2. Redistributions in binary form must reproduce the above copyright
26 * notice, this list of conditions and the following disclaimer in the
27 * documentation and/or other materials provided with the distribution.
28 *
29 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
30 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
31 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
32 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
33 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
34 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
35 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
36 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
37 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
38 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
39 */
Damien Millerd4a8b7e1999-10-27 13:42:43 +100040
41/*
Damien Millerad833b32000-08-23 10:46:23 +100042 * Parts from:
43 *
Damien Millerd4a8b7e1999-10-27 13:42:43 +100044 * Copyright (c) 1983, 1990, 1992, 1993, 1995
45 * The Regents of the University of California. All rights reserved.
46 *
47 * Redistribution and use in source and binary forms, with or without
48 * modification, are permitted provided that the following conditions
49 * are met:
50 * 1. Redistributions of source code must retain the above copyright
51 * notice, this list of conditions and the following disclaimer.
52 * 2. Redistributions in binary form must reproduce the above copyright
53 * notice, this list of conditions and the following disclaimer in the
54 * documentation and/or other materials provided with the distribution.
Damien Millerdafb12e2003-06-03 13:06:18 +100055 * 3. Neither the name of the University nor the names of its contributors
Damien Millerd4a8b7e1999-10-27 13:42:43 +100056 * may be used to endorse or promote products derived from this software
57 * without specific prior written permission.
58 *
59 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
60 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
61 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
62 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
63 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
64 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
65 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
66 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
67 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
68 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
69 * SUCH DAMAGE.
70 *
Damien Millerd4a8b7e1999-10-27 13:42:43 +100071 */
72
73#include "includes.h"
Darren Tuckerbddc2b02004-04-19 23:50:16 +100074RCSID("$OpenBSD: scp.c,v 1.114 2004/04/01 12:19:57 markus Exp $");
Damien Millerd4a8b7e1999-10-27 13:42:43 +100075
Damien Millerd4a8b7e1999-10-27 13:42:43 +100076#include "xmalloc.h"
Ben Lindstrom226cfa02001-01-22 05:34:40 +000077#include "atomicio.h"
78#include "pathnames.h"
79#include "log.h"
Kevin Steves93c17d92001-02-18 03:55:16 +000080#include "misc.h"
Damien Miller62d57f62003-01-10 21:43:24 +110081#include "progressmeter.h"
Damien Millerd4a8b7e1999-10-27 13:42:43 +100082
Ben Lindstrom49a79c02000-11-17 03:47:20 +000083#ifdef HAVE___PROGNAME
84extern char *__progname;
85#else
86char *__progname;
87#endif
88
Damien Millerff74d742003-01-24 11:36:58 +110089void bwlimit(int);
90
Ben Lindstrom387c4722001-05-08 20:27:25 +000091/* Struct for addargs */
92arglist args;
Damien Miller874d77b2000-10-14 16:23:11 +110093
Damien Millerff74d742003-01-24 11:36:58 +110094/* Bandwidth limit */
Damien Miller4da295c2003-11-22 14:39:04 +110095off_t limit_rate = 0;
Damien Millerff74d742003-01-24 11:36:58 +110096
Damien Millerd4a8b7e1999-10-27 13:42:43 +100097/* Name of current file being transferred. */
98char *curfile;
99
100/* This is set to non-zero to enable verbose mode. */
Damien Miller95def091999-11-25 00:26:21 +1100101int verbose_mode = 0;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000102
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000103/* This is set to zero if the progressmeter is not desired. */
104int showprogress = 1;
105
Damien Millerad833b32000-08-23 10:46:23 +1000106/* This is the program to execute for the secured connection. ("ssh" or -S) */
Ben Lindstrom226cfa02001-01-22 05:34:40 +0000107char *ssh_program = _PATH_SSH_PROGRAM;
Damien Millerad833b32000-08-23 10:46:23 +1000108
Ben Lindstrom007eb912002-11-09 15:54:08 +0000109/* This is used to store the pid of ssh_program */
Damien Millerb69aaa82003-06-04 22:51:24 +1000110pid_t do_cmd_pid = -1;
111
112static void
113killchild(int signo)
114{
115 if (do_cmd_pid > 1)
116 kill(do_cmd_pid, signo);
117
118 _exit(1);
119}
Ben Lindstrom007eb912002-11-09 15:54:08 +0000120
Damien Miller5428f641999-11-25 11:54:57 +1100121/*
122 * This function executes the given command as the specified user on the
123 * given host. This returns < 0 if execution fails, and >= 0 otherwise. This
124 * assigns the input and output file descriptors on success.
125 */
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000126
Damien Miller4af51302000-04-16 11:18:38 +1000127int
Damien Millerad833b32000-08-23 10:46:23 +1000128do_cmd(char *host, char *remuser, char *cmd, int *fdin, int *fdout, int argc)
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000129{
Damien Miller95def091999-11-25 00:26:21 +1100130 int pin[2], pout[2], reserved[2];
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000131
Damien Miller95def091999-11-25 00:26:21 +1100132 if (verbose_mode)
Ben Lindstrom2772a3f2001-08-06 21:17:12 +0000133 fprintf(stderr,
134 "Executing: program %s host %s, user %s, command %s\n",
135 ssh_program, host,
136 remuser ? remuser : "(unspecified)", cmd);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000137
Damien Miller5428f641999-11-25 11:54:57 +1100138 /*
139 * Reserve two descriptors so that the real pipes won't get
140 * descriptors 0 and 1 because that will screw up dup2 below.
141 */
Damien Miller95def091999-11-25 00:26:21 +1100142 pipe(reserved);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000143
Damien Miller95def091999-11-25 00:26:21 +1100144 /* Create a socket pair for communicating with ssh. */
145 if (pipe(pin) < 0)
146 fatal("pipe: %s", strerror(errno));
147 if (pipe(pout) < 0)
148 fatal("pipe: %s", strerror(errno));
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000149
Damien Miller95def091999-11-25 00:26:21 +1100150 /* Free the reserved descriptors. */
151 close(reserved[0]);
152 close(reserved[1]);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000153
Damien Millerb2cdcb52003-06-18 20:26:34 +1000154 /* Fork a child to execute the command on the remote host using ssh. */
Ben Lindstrom007eb912002-11-09 15:54:08 +0000155 do_cmd_pid = fork();
156 if (do_cmd_pid == 0) {
Damien Miller95def091999-11-25 00:26:21 +1100157 /* Child. */
158 close(pin[1]);
159 close(pout[0]);
160 dup2(pin[0], 0);
161 dup2(pout[1], 1);
162 close(pin[0]);
163 close(pout[1]);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000164
Damien Miller874d77b2000-10-14 16:23:11 +1100165 args.list[0] = ssh_program;
166 if (remuser != NULL)
Ben Lindstrom387c4722001-05-08 20:27:25 +0000167 addargs(&args, "-l%s", remuser);
168 addargs(&args, "%s", host);
169 addargs(&args, "%s", cmd);
Damien Miller95def091999-11-25 00:26:21 +1100170
Damien Miller874d77b2000-10-14 16:23:11 +1100171 execvp(ssh_program, args.list);
Damien Millerad833b32000-08-23 10:46:23 +1000172 perror(ssh_program);
Damien Miller95def091999-11-25 00:26:21 +1100173 exit(1);
Ben Lindstrom007eb912002-11-09 15:54:08 +0000174 } else if (do_cmd_pid == -1) {
175 fatal("fork: %s", strerror(errno));
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000176 }
Damien Miller95def091999-11-25 00:26:21 +1100177 /* Parent. Close the other side, and return the local side. */
178 close(pin[0]);
179 *fdout = pin[1];
180 close(pout[1]);
181 *fdin = pout[0];
Damien Millerb69aaa82003-06-04 22:51:24 +1000182 signal(SIGTERM, killchild);
183 signal(SIGINT, killchild);
184 signal(SIGHUP, killchild);
Damien Miller95def091999-11-25 00:26:21 +1100185 return 0;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000186}
187
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000188typedef struct {
189 int cnt;
190 char *buf;
191} BUF;
192
Damien Miller95def091999-11-25 00:26:21 +1100193BUF *allocbuf(BUF *, int, int);
Damien Miller95def091999-11-25 00:26:21 +1100194void lostconn(int);
195void nospace(void);
196int okname(char *);
197void run_err(const char *,...);
198void verifydir(char *);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000199
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000200struct passwd *pwd;
Damien Miller95def091999-11-25 00:26:21 +1100201uid_t userid;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000202int errs, remin, remout;
203int pflag, iamremote, iamrecursive, targetshouldbedirectory;
204
205#define CMDNEEDS 64
206char cmd[CMDNEEDS]; /* must hold "rcp -r -p -d\0" */
207
Damien Miller95def091999-11-25 00:26:21 +1100208int response(void);
209void rsource(char *, struct stat *);
210void sink(int, char *[]);
211void source(int, char *[]);
212void tolocal(int, char *[]);
213void toremote(char *, int, char *[]);
214void usage(void);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000215
216int
Damien Miller65d1f572003-06-04 22:51:08 +1000217main(int argc, char **argv)
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000218{
Ben Lindstrom007eb912002-11-09 15:54:08 +0000219 int ch, fflag, tflag, status;
Damien Millerff74d742003-01-24 11:36:58 +1100220 double speed;
221 char *targ, *endp;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000222 extern char *optarg;
223 extern int optind;
224
Damien Miller59d3d5b2003-08-22 09:34:41 +1000225 __progname = ssh_get_progname(argv[0]);
Ben Lindstrom49a79c02000-11-17 03:47:20 +0000226
Damien Miller874d77b2000-10-14 16:23:11 +1100227 args.list = NULL;
Ben Lindstromcb72e4f2002-06-21 00:41:51 +0000228 addargs(&args, "ssh"); /* overwritten with ssh_program */
Ben Lindstrom387c4722001-05-08 20:27:25 +0000229 addargs(&args, "-x");
Ben Lindstrom4213c552001-09-12 18:45:09 +0000230 addargs(&args, "-oForwardAgent no");
Ben Lindstrom2b7a0e92001-09-20 00:57:55 +0000231 addargs(&args, "-oClearAllForwardings yes");
Damien Miller874d77b2000-10-14 16:23:11 +1100232
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000233 fflag = tflag = 0;
Damien Miller8e121472003-01-24 11:37:38 +1100234 while ((ch = getopt(argc, argv, "dfl:prtvBCc:i:P:q1246S:o:F:")) != -1)
Damien Miller95def091999-11-25 00:26:21 +1100235 switch (ch) {
236 /* User-visible flags. */
Damien Miller8e121472003-01-24 11:37:38 +1100237 case '1':
238 case '2':
Damien Miller34132e52000-01-14 15:45:46 +1100239 case '4':
Damien Miller34132e52000-01-14 15:45:46 +1100240 case '6':
Damien Miller874d77b2000-10-14 16:23:11 +1100241 case 'C':
Ben Lindstrom387c4722001-05-08 20:27:25 +0000242 addargs(&args, "-%c", ch);
Damien Miller874d77b2000-10-14 16:23:11 +1100243 break;
244 case 'o':
245 case 'c':
246 case 'i':
Ben Lindstrom1e243242001-09-18 05:38:44 +0000247 case 'F':
Ben Lindstrom387c4722001-05-08 20:27:25 +0000248 addargs(&args, "-%c%s", ch, optarg);
Damien Miller874d77b2000-10-14 16:23:11 +1100249 break;
250 case 'P':
Ben Lindstrom387c4722001-05-08 20:27:25 +0000251 addargs(&args, "-p%s", optarg);
Damien Miller874d77b2000-10-14 16:23:11 +1100252 break;
253 case 'B':
Ben Lindstrom387c4722001-05-08 20:27:25 +0000254 addargs(&args, "-oBatchmode yes");
Damien Miller34132e52000-01-14 15:45:46 +1100255 break;
Damien Millerff74d742003-01-24 11:36:58 +1100256 case 'l':
257 speed = strtod(optarg, &endp);
258 if (speed <= 0 || *endp != '\0')
259 usage();
Damien Miller4da295c2003-11-22 14:39:04 +1100260 limit_rate = speed * 1024;
Damien Millerff74d742003-01-24 11:36:58 +1100261 break;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000262 case 'p':
263 pflag = 1;
264 break;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000265 case 'r':
266 iamrecursive = 1;
267 break;
Damien Millerad833b32000-08-23 10:46:23 +1000268 case 'S':
Damien Miller874d77b2000-10-14 16:23:11 +1100269 ssh_program = xstrdup(optarg);
270 break;
271 case 'v':
Ben Lindstrom9cc94642001-06-09 01:15:11 +0000272 addargs(&args, "-v");
Damien Miller874d77b2000-10-14 16:23:11 +1100273 verbose_mode = 1;
274 break;
275 case 'q':
Damien Millerc1f27922003-11-17 21:19:05 +1100276 addargs(&args, "-q");
Damien Miller874d77b2000-10-14 16:23:11 +1100277 showprogress = 0;
Damien Millerad833b32000-08-23 10:46:23 +1000278 break;
279
Damien Miller95def091999-11-25 00:26:21 +1100280 /* Server options. */
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000281 case 'd':
282 targetshouldbedirectory = 1;
283 break;
Damien Miller95def091999-11-25 00:26:21 +1100284 case 'f': /* "from" */
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000285 iamremote = 1;
286 fflag = 1;
287 break;
Damien Miller95def091999-11-25 00:26:21 +1100288 case 't': /* "to" */
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000289 iamremote = 1;
290 tflag = 1;
Damien Miller402b3312001-04-14 00:28:42 +1000291#ifdef HAVE_CYGWIN
292 setmode(0, O_BINARY);
293#endif
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000294 break;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000295 default:
296 usage();
297 }
298 argc -= optind;
299 argv += optind;
300
301 if ((pwd = getpwuid(userid = getuid())) == NULL)
Darren Tucker03a29ba2003-07-19 20:07:45 +1000302 fatal("unknown user %u", (u_int) userid);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000303
Damien Miller95def091999-11-25 00:26:21 +1100304 if (!isatty(STDERR_FILENO))
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000305 showprogress = 0;
306
307 remin = STDIN_FILENO;
308 remout = STDOUT_FILENO;
309
Kevin Stevesef4eea92001-02-05 12:42:17 +0000310 if (fflag) {
Damien Miller95def091999-11-25 00:26:21 +1100311 /* Follow "protocol", send data. */
312 (void) response();
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000313 source(argc, argv);
314 exit(errs != 0);
315 }
Damien Miller95def091999-11-25 00:26:21 +1100316 if (tflag) {
317 /* Receive data. */
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000318 sink(argc, argv);
319 exit(errs != 0);
320 }
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000321 if (argc < 2)
322 usage();
323 if (argc > 2)
324 targetshouldbedirectory = 1;
325
326 remin = remout = -1;
Ben Lindstrom007eb912002-11-09 15:54:08 +0000327 do_cmd_pid = -1;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000328 /* Command to be executed on remote system using "ssh". */
Ben Lindstromf6b7b092001-02-09 01:23:39 +0000329 (void) snprintf(cmd, sizeof cmd, "scp%s%s%s%s",
330 verbose_mode ? " -v" : "",
Damien Millerad833b32000-08-23 10:46:23 +1000331 iamrecursive ? " -r" : "", pflag ? " -p" : "",
332 targetshouldbedirectory ? " -d" : "");
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000333
Damien Miller95def091999-11-25 00:26:21 +1100334 (void) signal(SIGPIPE, lostconn);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000335
336 if ((targ = colon(argv[argc - 1]))) /* Dest is remote host. */
337 toremote(targ, argc, argv);
338 else {
Damien Miller95def091999-11-25 00:26:21 +1100339 tolocal(argc, argv); /* Dest is local host. */
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000340 if (targetshouldbedirectory)
341 verifydir(argv[argc - 1]);
342 }
Ben Lindstrom007eb912002-11-09 15:54:08 +0000343 /*
344 * Finally check the exit status of the ssh process, if one was forked
345 * and no error has occured yet
346 */
347 if (do_cmd_pid != -1 && errs == 0) {
348 if (remin != -1)
349 (void) close(remin);
350 if (remout != -1)
351 (void) close(remout);
352 if (waitpid(do_cmd_pid, &status, 0) == -1)
353 errs = 1;
354 else {
355 if (!WIFEXITED(status) || WEXITSTATUS(status) != 0)
356 errs = 1;
357 }
358 }
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000359 exit(errs != 0);
360}
361
362void
Damien Miller65d1f572003-06-04 22:51:08 +1000363toremote(char *targ, int argc, char **argv)
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000364{
365 int i, len;
366 char *bp, *host, *src, *suser, *thost, *tuser;
367
368 *targ++ = 0;
369 if (*targ == 0)
370 targ = ".";
371
Ben Lindstromc276c122002-12-23 02:14:51 +0000372 if ((thost = strrchr(argv[argc - 1], '@'))) {
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000373 /* user@host */
374 *thost++ = 0;
375 tuser = argv[argc - 1];
376 if (*tuser == '\0')
377 tuser = NULL;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000378 } else {
379 thost = argv[argc - 1];
380 tuser = NULL;
381 }
382
383 for (i = 0; i < argc - 1; i++) {
384 src = colon(argv[i]);
Damien Miller95def091999-11-25 00:26:21 +1100385 if (src) { /* remote to remote */
Ben Lindstrom2b7a0e92001-09-20 00:57:55 +0000386 static char *ssh_options =
Ben Lindstromaf0c6d62002-06-09 20:06:29 +0000387 "-x -o'ClearAllForwardings yes'";
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000388 *src++ = 0;
389 if (*src == 0)
390 src = ".";
Ben Lindstromc276c122002-12-23 02:14:51 +0000391 host = strrchr(argv[i], '@');
Damien Millerad833b32000-08-23 10:46:23 +1000392 len = strlen(ssh_program) + strlen(argv[i]) +
393 strlen(src) + (tuser ? strlen(tuser) : 0) +
Ben Lindstrom2b7a0e92001-09-20 00:57:55 +0000394 strlen(thost) + strlen(targ) +
395 strlen(ssh_options) + CMDNEEDS + 20;
Damien Miller95def091999-11-25 00:26:21 +1100396 bp = xmalloc(len);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000397 if (host) {
398 *host++ = 0;
Damien Miller34132e52000-01-14 15:45:46 +1100399 host = cleanhostname(host);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000400 suser = argv[i];
401 if (*suser == '\0')
402 suser = pwd->pw_name;
Damien Miller00111382003-03-10 11:21:17 +1100403 else if (!okname(suser)) {
404 xfree(bp);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000405 continue;
Damien Miller00111382003-03-10 11:21:17 +1100406 }
407 if (tuser && !okname(tuser)) {
408 xfree(bp);
Damien Millereeeeb352003-02-24 11:52:58 +1100409 continue;
Damien Miller00111382003-03-10 11:21:17 +1100410 }
Ben Lindstromf6b7b092001-02-09 01:23:39 +0000411 snprintf(bp, len,
Ben Lindstrom2b7a0e92001-09-20 00:57:55 +0000412 "%s%s %s -n "
Ben Lindstromf6b7b092001-02-09 01:23:39 +0000413 "-l %s %s %s %s '%s%s%s:%s'",
Kevin Steves7d00ba42000-12-15 23:03:10 +0000414 ssh_program, verbose_mode ? " -v" : "",
Ben Lindstrom2b7a0e92001-09-20 00:57:55 +0000415 ssh_options, suser, host, cmd, src,
Kevin Steves7d00ba42000-12-15 23:03:10 +0000416 tuser ? tuser : "", tuser ? "@" : "",
417 thost, targ);
Damien Miller34132e52000-01-14 15:45:46 +1100418 } else {
419 host = cleanhostname(argv[i]);
Ben Lindstromf6b7b092001-02-09 01:23:39 +0000420 snprintf(bp, len,
Ben Lindstrom2b7a0e92001-09-20 00:57:55 +0000421 "exec %s%s %s -n %s "
Ben Lindstromf6b7b092001-02-09 01:23:39 +0000422 "%s %s '%s%s%s:%s'",
Kevin Steves7d00ba42000-12-15 23:03:10 +0000423 ssh_program, verbose_mode ? " -v" : "",
Ben Lindstrom2b7a0e92001-09-20 00:57:55 +0000424 ssh_options, host, cmd, src,
Kevin Steves7d00ba42000-12-15 23:03:10 +0000425 tuser ? tuser : "", tuser ? "@" : "",
426 thost, targ);
Damien Miller34132e52000-01-14 15:45:46 +1100427 }
Damien Miller95def091999-11-25 00:26:21 +1100428 if (verbose_mode)
429 fprintf(stderr, "Executing: %s\n", bp);
Darren Tucker8654d162003-09-22 21:14:55 +1000430 if (system(bp) != 0)
431 errs = 1;
Damien Miller95def091999-11-25 00:26:21 +1100432 (void) xfree(bp);
433 } else { /* local to remote */
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000434 if (remin == -1) {
435 len = strlen(targ) + CMDNEEDS + 20;
Damien Miller95def091999-11-25 00:26:21 +1100436 bp = xmalloc(len);
Ben Lindstromf6b7b092001-02-09 01:23:39 +0000437 (void) snprintf(bp, len, "%s -t %s", cmd, targ);
Damien Miller34132e52000-01-14 15:45:46 +1100438 host = cleanhostname(thost);
Damien Millerad833b32000-08-23 10:46:23 +1000439 if (do_cmd(host, tuser, bp, &remin,
440 &remout, argc) < 0)
Damien Miller95def091999-11-25 00:26:21 +1100441 exit(1);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000442 if (response() < 0)
443 exit(1);
Damien Miller95def091999-11-25 00:26:21 +1100444 (void) xfree(bp);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000445 }
Damien Miller95def091999-11-25 00:26:21 +1100446 source(1, argv + i);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000447 }
448 }
449}
450
451void
Damien Miller65d1f572003-06-04 22:51:08 +1000452tolocal(int argc, char **argv)
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000453{
454 int i, len;
455 char *bp, *host, *src, *suser;
456
457 for (i = 0; i < argc - 1; i++) {
Damien Miller95def091999-11-25 00:26:21 +1100458 if (!(src = colon(argv[i]))) { /* Local to local. */
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000459 len = strlen(_PATH_CP) + strlen(argv[i]) +
Damien Millerad833b32000-08-23 10:46:23 +1000460 strlen(argv[argc - 1]) + 20;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000461 bp = xmalloc(len);
Ben Lindstromf6b7b092001-02-09 01:23:39 +0000462 (void) snprintf(bp, len, "exec %s%s%s %s %s", _PATH_CP,
Damien Millerad833b32000-08-23 10:46:23 +1000463 iamrecursive ? " -r" : "", pflag ? " -p" : "",
464 argv[i], argv[argc - 1]);
Damien Miller95def091999-11-25 00:26:21 +1100465 if (verbose_mode)
466 fprintf(stderr, "Executing: %s\n", bp);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000467 if (system(bp))
468 ++errs;
Damien Miller95def091999-11-25 00:26:21 +1100469 (void) xfree(bp);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000470 continue;
471 }
472 *src++ = 0;
473 if (*src == 0)
474 src = ".";
Ben Lindstromc276c122002-12-23 02:14:51 +0000475 if ((host = strrchr(argv[i], '@')) == NULL) {
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000476 host = argv[i];
477 suser = NULL;
478 } else {
479 *host++ = 0;
480 suser = argv[i];
481 if (*suser == '\0')
482 suser = pwd->pw_name;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000483 }
Damien Miller34132e52000-01-14 15:45:46 +1100484 host = cleanhostname(host);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000485 len = strlen(src) + CMDNEEDS + 20;
Damien Miller95def091999-11-25 00:26:21 +1100486 bp = xmalloc(len);
Ben Lindstromf6b7b092001-02-09 01:23:39 +0000487 (void) snprintf(bp, len, "%s -f %s", cmd, src);
Damien Millerad833b32000-08-23 10:46:23 +1000488 if (do_cmd(host, suser, bp, &remin, &remout, argc) < 0) {
Damien Miller95def091999-11-25 00:26:21 +1100489 (void) xfree(bp);
490 ++errs;
491 continue;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000492 }
Damien Miller95def091999-11-25 00:26:21 +1100493 xfree(bp);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000494 sink(1, argv + argc - 1);
Damien Miller95def091999-11-25 00:26:21 +1100495 (void) close(remin);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000496 remin = remout = -1;
497 }
498}
499
500void
Damien Miller65d1f572003-06-04 22:51:08 +1000501source(int argc, char **argv)
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000502{
503 struct stat stb;
504 static BUF buffer;
505 BUF *bp;
Damien Miller62d57f62003-01-10 21:43:24 +1100506 off_t i, amt, result, statbytes;
Ben Lindstrom4eda71d2001-04-22 17:13:20 +0000507 int fd, haderr, indx;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000508 char *last, *name, buf[2048];
Ben Lindstromd47cf4d2001-04-07 01:14:38 +0000509 int len;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000510
511 for (indx = 0; indx < argc; ++indx) {
Damien Miller95def091999-11-25 00:26:21 +1100512 name = argv[indx];
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000513 statbytes = 0;
Ben Lindstromd47cf4d2001-04-07 01:14:38 +0000514 len = strlen(name);
515 while (len > 1 && name[len-1] == '/')
516 name[--len] = '\0';
Ben Lindstrom3e45e4c2001-10-03 17:30:58 +0000517 if (strchr(name, '\n') != NULL) {
518 run_err("%s: skipping, filename contains a newline",
519 name);
520 goto next;
521 }
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000522 if ((fd = open(name, O_RDONLY, 0)) < 0)
523 goto syserr;
524 if (fstat(fd, &stb) < 0) {
525syserr: run_err("%s: %s", name, strerror(errno));
526 goto next;
527 }
528 switch (stb.st_mode & S_IFMT) {
529 case S_IFREG:
530 break;
531 case S_IFDIR:
532 if (iamrecursive) {
533 rsource(name, &stb);
534 goto next;
535 }
536 /* FALLTHROUGH */
537 default:
538 run_err("%s: not a regular file", name);
539 goto next;
540 }
541 if ((last = strrchr(name, '/')) == NULL)
542 last = name;
543 else
544 ++last;
545 curfile = last;
546 if (pflag) {
547 /*
548 * Make it compatible with possible future
549 * versions expecting microseconds.
550 */
Ben Lindstromf6b7b092001-02-09 01:23:39 +0000551 (void) snprintf(buf, sizeof buf, "T%lu 0 %lu 0\n",
Ben Lindstrom46c16222000-12-22 01:43:59 +0000552 (u_long) stb.st_mtime,
553 (u_long) stb.st_atime);
Darren Tucker9f63f222003-07-03 13:46:56 +1000554 (void) atomicio(vwrite, remout, buf, strlen(buf));
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000555 if (response() < 0)
556 goto next;
557 }
558#define FILEMODEMASK (S_ISUID|S_ISGID|S_IRWXU|S_IRWXG|S_IRWXO)
Ben Lindstroma4c57662001-03-17 00:10:20 +0000559 snprintf(buf, sizeof buf, "C%04o %lld %s\n",
Ben Lindstrom46c16222000-12-22 01:43:59 +0000560 (u_int) (stb.st_mode & FILEMODEMASK),
Ben Lindstrom0e7f4362003-04-28 23:30:43 +0000561 (int64_t)stb.st_size, last);
Damien Miller95def091999-11-25 00:26:21 +1100562 if (verbose_mode) {
563 fprintf(stderr, "Sending file modes: %s", buf);
Damien Miller95def091999-11-25 00:26:21 +1100564 }
Darren Tucker9f63f222003-07-03 13:46:56 +1000565 (void) atomicio(vwrite, remout, buf, strlen(buf));
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000566 if (response() < 0)
567 goto next;
568 if ((bp = allocbuf(&buffer, fd, 2048)) == NULL) {
Damien Miller95def091999-11-25 00:26:21 +1100569next: (void) close(fd);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000570 continue;
571 }
Damien Miller62d57f62003-01-10 21:43:24 +1100572 if (showprogress)
573 start_progress_meter(curfile, stb.st_size, &statbytes);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000574 /* Keep writing after an error so that we stay sync'd up. */
575 for (haderr = i = 0; i < stb.st_size; i += bp->cnt) {
576 amt = bp->cnt;
577 if (i + amt > stb.st_size)
578 amt = stb.st_size - i;
579 if (!haderr) {
Damien Millere247cc42000-05-07 12:03:14 +1000580 result = atomicio(read, fd, bp->buf, amt);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000581 if (result != amt)
582 haderr = result >= 0 ? EIO : errno;
583 }
584 if (haderr)
Darren Tucker9f63f222003-07-03 13:46:56 +1000585 (void) atomicio(vwrite, remout, bp->buf, amt);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000586 else {
Darren Tucker9f63f222003-07-03 13:46:56 +1000587 result = atomicio(vwrite, remout, bp->buf, amt);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000588 if (result != amt)
589 haderr = result >= 0 ? EIO : errno;
590 statbytes += result;
591 }
Damien Miller4da295c2003-11-22 14:39:04 +1100592 if (limit_rate)
Damien Millerff74d742003-01-24 11:36:58 +1100593 bwlimit(amt);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000594 }
Damien Miller95def091999-11-25 00:26:21 +1100595 if (showprogress)
Damien Miller62d57f62003-01-10 21:43:24 +1100596 stop_progress_meter();
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000597
598 if (close(fd) < 0 && !haderr)
599 haderr = errno;
600 if (!haderr)
Darren Tucker9f63f222003-07-03 13:46:56 +1000601 (void) atomicio(vwrite, remout, "", 1);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000602 else
603 run_err("%s: %s", name, strerror(haderr));
Damien Miller95def091999-11-25 00:26:21 +1100604 (void) response();
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000605 }
606}
607
608void
Damien Miller65d1f572003-06-04 22:51:08 +1000609rsource(char *name, struct stat *statp)
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000610{
611 DIR *dirp;
612 struct dirent *dp;
613 char *last, *vect[1], path[1100];
614
615 if (!(dirp = opendir(name))) {
616 run_err("%s: %s", name, strerror(errno));
617 return;
618 }
619 last = strrchr(name, '/');
620 if (last == 0)
621 last = name;
622 else
623 last++;
624 if (pflag) {
Ben Lindstromf6b7b092001-02-09 01:23:39 +0000625 (void) snprintf(path, sizeof(path), "T%lu 0 %lu 0\n",
Ben Lindstrom46c16222000-12-22 01:43:59 +0000626 (u_long) statp->st_mtime,
627 (u_long) statp->st_atime);
Darren Tucker9f63f222003-07-03 13:46:56 +1000628 (void) atomicio(vwrite, remout, path, strlen(path));
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000629 if (response() < 0) {
630 closedir(dirp);
631 return;
632 }
633 }
Ben Lindstromf6b7b092001-02-09 01:23:39 +0000634 (void) snprintf(path, sizeof path, "D%04o %d %.1024s\n",
Ben Lindstrom46c16222000-12-22 01:43:59 +0000635 (u_int) (statp->st_mode & FILEMODEMASK), 0, last);
Damien Miller95def091999-11-25 00:26:21 +1100636 if (verbose_mode)
637 fprintf(stderr, "Entering directory: %s", path);
Darren Tucker9f63f222003-07-03 13:46:56 +1000638 (void) atomicio(vwrite, remout, path, strlen(path));
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000639 if (response() < 0) {
640 closedir(dirp);
641 return;
642 }
Ben Lindstrombd472262001-03-29 00:39:55 +0000643 while ((dp = readdir(dirp)) != NULL) {
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000644 if (dp->d_ino == 0)
645 continue;
646 if (!strcmp(dp->d_name, ".") || !strcmp(dp->d_name, ".."))
647 continue;
648 if (strlen(name) + 1 + strlen(dp->d_name) >= sizeof(path) - 1) {
649 run_err("%s/%s: name too long", name, dp->d_name);
650 continue;
651 }
Ben Lindstromf6b7b092001-02-09 01:23:39 +0000652 (void) snprintf(path, sizeof path, "%s/%s", name, dp->d_name);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000653 vect[0] = path;
654 source(1, vect);
655 }
Damien Miller95def091999-11-25 00:26:21 +1100656 (void) closedir(dirp);
Darren Tucker9f63f222003-07-03 13:46:56 +1000657 (void) atomicio(vwrite, remout, "E\n", 2);
Damien Miller95def091999-11-25 00:26:21 +1100658 (void) response();
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000659}
660
661void
Damien Millerff74d742003-01-24 11:36:58 +1100662bwlimit(int amount)
663{
664 static struct timeval bwstart, bwend;
665 static int lamt, thresh = 16384;
666 u_int64_t wait;
667 struct timespec ts, rm;
668
669 if (!timerisset(&bwstart)) {
670 gettimeofday(&bwstart, NULL);
671 return;
672 }
673
674 lamt += amount;
675 if (lamt < thresh)
676 return;
677
678 gettimeofday(&bwend, NULL);
679 timersub(&bwend, &bwstart, &bwend);
680 if (!timerisset(&bwend))
681 return;
682
683 lamt *= 8;
Damien Miller4da295c2003-11-22 14:39:04 +1100684 wait = (double)1000000L * lamt / limit_rate;
Damien Millerff74d742003-01-24 11:36:58 +1100685
686 bwstart.tv_sec = wait / 1000000L;
687 bwstart.tv_usec = wait % 1000000L;
688
689 if (timercmp(&bwstart, &bwend, >)) {
690 timersub(&bwstart, &bwend, &bwend);
691
692 /* Adjust the wait time */
693 if (bwend.tv_sec) {
694 thresh /= 2;
695 if (thresh < 2048)
696 thresh = 2048;
697 } else if (bwend.tv_usec < 100) {
698 thresh *= 2;
699 if (thresh > 32768)
700 thresh = 32768;
701 }
702
703 TIMEVAL_TO_TIMESPEC(&bwend, &ts);
704 while (nanosleep(&ts, &rm) == -1) {
705 if (errno != EINTR)
706 break;
707 ts = rm;
708 }
709 }
710
711 lamt = 0;
712 gettimeofday(&bwstart, NULL);
713}
714
715void
Damien Miller65d1f572003-06-04 22:51:08 +1000716sink(int argc, char **argv)
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000717{
718 static BUF buffer;
719 struct stat stb;
Damien Miller95def091999-11-25 00:26:21 +1100720 enum {
721 YES, NO, DISPLAYED
722 } wrerr;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000723 BUF *bp;
724 off_t i, j;
725 int amt, count, exists, first, mask, mode, ofd, omode;
Damien Miller62d57f62003-01-10 21:43:24 +1100726 off_t size, statbytes;
Damien Millercaf6dd62000-08-29 11:33:50 +1100727 int setimes, targisdir, wrerrno = 0;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000728 char ch, *cp, *np, *targ, *why, *vect[1], buf[2048];
Damien Miller62cee002000-09-23 17:15:56 +1100729 struct timeval tv[2];
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000730
Ben Lindstromf719a202001-04-14 23:14:22 +0000731#define atime tv[0]
732#define mtime tv[1]
Ben Lindstrom7d5ed3a2001-06-25 04:28:30 +0000733#define SCREWUP(str) do { why = str; goto screwup; } while (0)
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000734
735 setimes = targisdir = 0;
736 mask = umask(0);
737 if (!pflag)
Damien Miller95def091999-11-25 00:26:21 +1100738 (void) umask(mask);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000739 if (argc != 1) {
740 run_err("ambiguous target");
741 exit(1);
742 }
743 targ = *argv;
744 if (targetshouldbedirectory)
745 verifydir(targ);
Damien Miller95def091999-11-25 00:26:21 +1100746
Darren Tucker9f63f222003-07-03 13:46:56 +1000747 (void) atomicio(vwrite, remout, "", 1);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000748 if (stat(targ, &stb) == 0 && S_ISDIR(stb.st_mode))
749 targisdir = 1;
750 for (first = 1;; first = 0) {
751 cp = buf;
Damien Millere247cc42000-05-07 12:03:14 +1000752 if (atomicio(read, remin, cp, 1) <= 0)
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000753 return;
754 if (*cp++ == '\n')
755 SCREWUP("unexpected <newline>");
756 do {
Damien Millere247cc42000-05-07 12:03:14 +1000757 if (atomicio(read, remin, &ch, sizeof(ch)) != sizeof(ch))
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000758 SCREWUP("lost connection");
759 *cp++ = ch;
760 } while (cp < &buf[sizeof(buf) - 1] && ch != '\n');
761 *cp = 0;
Darren Tuckerbddc2b02004-04-19 23:50:16 +1000762 if (verbose_mode)
763 fprintf(stderr, "Sink: %s", buf);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000764
765 if (buf[0] == '\01' || buf[0] == '\02') {
766 if (iamremote == 0)
Darren Tucker9f63f222003-07-03 13:46:56 +1000767 (void) atomicio(vwrite, STDERR_FILENO,
Kevin Steves7d00ba42000-12-15 23:03:10 +0000768 buf + 1, strlen(buf + 1));
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000769 if (buf[0] == '\02')
770 exit(1);
771 ++errs;
772 continue;
773 }
774 if (buf[0] == 'E') {
Darren Tucker9f63f222003-07-03 13:46:56 +1000775 (void) atomicio(vwrite, remout, "", 1);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000776 return;
777 }
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000778 if (ch == '\n')
779 *--cp = 0;
780
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000781 cp = buf;
782 if (*cp == 'T') {
783 setimes++;
784 cp++;
Ben Lindstromf719a202001-04-14 23:14:22 +0000785 mtime.tv_sec = strtol(cp, &cp, 10);
786 if (!cp || *cp++ != ' ')
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000787 SCREWUP("mtime.sec not delimited");
Ben Lindstromf719a202001-04-14 23:14:22 +0000788 mtime.tv_usec = strtol(cp, &cp, 10);
789 if (!cp || *cp++ != ' ')
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000790 SCREWUP("mtime.usec not delimited");
Ben Lindstromf719a202001-04-14 23:14:22 +0000791 atime.tv_sec = strtol(cp, &cp, 10);
792 if (!cp || *cp++ != ' ')
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000793 SCREWUP("atime.sec not delimited");
Ben Lindstromf719a202001-04-14 23:14:22 +0000794 atime.tv_usec = strtol(cp, &cp, 10);
795 if (!cp || *cp++ != '\0')
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000796 SCREWUP("atime.usec not delimited");
Darren Tucker9f63f222003-07-03 13:46:56 +1000797 (void) atomicio(vwrite, remout, "", 1);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000798 continue;
799 }
800 if (*cp != 'C' && *cp != 'D') {
801 /*
802 * Check for the case "rcp remote:foo\* local:bar".
803 * In this case, the line "No match." can be returned
804 * by the shell before the rcp command on the remote is
805 * executed so the ^Aerror_message convention isn't
806 * followed.
807 */
808 if (first) {
809 run_err("%s", cp);
810 exit(1);
811 }
812 SCREWUP("expected control record");
813 }
814 mode = 0;
815 for (++cp; cp < buf + 5; cp++) {
816 if (*cp < '0' || *cp > '7')
817 SCREWUP("bad mode");
818 mode = (mode << 3) | (*cp - '0');
819 }
820 if (*cp++ != ' ')
821 SCREWUP("mode not delimited");
822
Ben Lindstrombd472262001-03-29 00:39:55 +0000823 for (size = 0; isdigit(*cp);)
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000824 size = size * 10 + (*cp++ - '0');
825 if (*cp++ != ' ')
826 SCREWUP("size not delimited");
Darren Tuckerbddc2b02004-04-19 23:50:16 +1000827 if ((strchr(cp, '/') != NULL) || (strcmp(cp, "..") == 0)) {
828 run_err("error: unexpected filename: %s", cp);
829 exit(1);
830 }
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000831 if (targisdir) {
832 static char *namebuf;
833 static int cursize;
834 size_t need;
835
836 need = strlen(targ) + strlen(cp) + 250;
Ben Lindstromf6b7b092001-02-09 01:23:39 +0000837 if (need > cursize) {
838 if (namebuf)
839 xfree(namebuf);
Damien Miller95def091999-11-25 00:26:21 +1100840 namebuf = xmalloc(need);
Ben Lindstromf6b7b092001-02-09 01:23:39 +0000841 cursize = need;
842 }
843 (void) snprintf(namebuf, need, "%s%s%s", targ,
Ben Lindstromde3895d2002-04-06 18:29:59 +0000844 strcmp(targ, "/") ? "/" : "", cp);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000845 np = namebuf;
846 } else
847 np = targ;
848 curfile = cp;
849 exists = stat(np, &stb) == 0;
850 if (buf[0] == 'D') {
851 int mod_flag = pflag;
Darren Tuckerbddc2b02004-04-19 23:50:16 +1000852 if (!iamrecursive)
853 SCREWUP("received directory without -r");
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000854 if (exists) {
855 if (!S_ISDIR(stb.st_mode)) {
856 errno = ENOTDIR;
857 goto bad;
858 }
859 if (pflag)
Damien Miller95def091999-11-25 00:26:21 +1100860 (void) chmod(np, mode);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000861 } else {
Damien Miller95def091999-11-25 00:26:21 +1100862 /* Handle copying from a read-only
863 directory */
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000864 mod_flag = 1;
865 if (mkdir(np, mode | S_IRWXU) < 0)
866 goto bad;
867 }
Ben Lindstrom550bc542001-02-10 21:50:00 +0000868 vect[0] = xstrdup(np);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000869 sink(1, vect);
870 if (setimes) {
871 setimes = 0;
Damien Miller225736c2001-02-19 21:51:08 +1100872 if (utimes(vect[0], tv) < 0)
Damien Miller95def091999-11-25 00:26:21 +1100873 run_err("%s: set times: %s",
Damien Miller225736c2001-02-19 21:51:08 +1100874 vect[0], strerror(errno));
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000875 }
876 if (mod_flag)
Damien Miller225736c2001-02-19 21:51:08 +1100877 (void) chmod(vect[0], mode);
878 if (vect[0])
879 xfree(vect[0]);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000880 continue;
881 }
882 omode = mode;
883 mode |= S_IWRITE;
Ben Lindstrom7bad55b2001-06-05 19:31:41 +0000884 if ((ofd = open(np, O_WRONLY|O_CREAT, mode)) < 0) {
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000885bad: run_err("%s: %s", np, strerror(errno));
886 continue;
887 }
Darren Tucker9f63f222003-07-03 13:46:56 +1000888 (void) atomicio(vwrite, remout, "", 1);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000889 if ((bp = allocbuf(&buffer, ofd, 4096)) == NULL) {
Damien Miller95def091999-11-25 00:26:21 +1100890 (void) close(ofd);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000891 continue;
892 }
893 cp = bp->buf;
894 wrerr = NO;
895
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000896 statbytes = 0;
Damien Miller62d57f62003-01-10 21:43:24 +1100897 if (showprogress)
898 start_progress_meter(curfile, size, &statbytes);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000899 for (count = i = 0; i < size; i += 4096) {
900 amt = 4096;
901 if (i + amt > size)
902 amt = size - i;
903 count += amt;
904 do {
Damien Miller69b69aa2000-10-28 14:19:58 +1100905 j = read(remin, cp, amt);
Ben Lindstrom2772a3f2001-08-06 21:17:12 +0000906 if (j == -1 && (errno == EINTR ||
907 errno == EAGAIN)) {
Damien Miller69b69aa2000-10-28 14:19:58 +1100908 continue;
909 } else if (j <= 0) {
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000910 run_err("%s", j ? strerror(errno) :
Ben Lindstrombd472262001-03-29 00:39:55 +0000911 "dropped connection");
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000912 exit(1);
913 }
914 amt -= j;
915 cp += j;
Damien Miller95def091999-11-25 00:26:21 +1100916 statbytes += j;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000917 } while (amt > 0);
Damien Miller787b2ec2003-11-21 23:56:47 +1100918
Damien Miller4da295c2003-11-22 14:39:04 +1100919 if (limit_rate)
Damien Millerff74d742003-01-24 11:36:58 +1100920 bwlimit(4096);
921
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000922 if (count == bp->cnt) {
923 /* Keep reading so we stay sync'd up. */
924 if (wrerr == NO) {
Darren Tucker9f63f222003-07-03 13:46:56 +1000925 j = atomicio(vwrite, ofd, bp->buf, count);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000926 if (j != count) {
927 wrerr = YES;
Damien Miller95def091999-11-25 00:26:21 +1100928 wrerrno = j >= 0 ? EIO : errno;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000929 }
930 }
931 count = 0;
932 cp = bp->buf;
933 }
934 }
935 if (showprogress)
Damien Miller62d57f62003-01-10 21:43:24 +1100936 stop_progress_meter();
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000937 if (count != 0 && wrerr == NO &&
Darren Tucker9f63f222003-07-03 13:46:56 +1000938 (j = atomicio(vwrite, ofd, bp->buf, count)) != count) {
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000939 wrerr = YES;
Damien Miller95def091999-11-25 00:26:21 +1100940 wrerrno = j >= 0 ? EIO : errno;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000941 }
Damien Miller86b781c2003-01-10 21:44:48 +1100942 if (wrerr == NO && ftruncate(ofd, size) != 0) {
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000943 run_err("%s: truncate: %s", np, strerror(errno));
944 wrerr = DISPLAYED;
945 }
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000946 if (pflag) {
947 if (exists || omode != mode)
Damien Miller78315eb2000-09-29 23:01:36 +1100948#ifdef HAVE_FCHMOD
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000949 if (fchmod(ofd, omode))
Damien Miller78315eb2000-09-29 23:01:36 +1100950#else /* HAVE_FCHMOD */
951 if (chmod(np, omode))
952#endif /* HAVE_FCHMOD */
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000953 run_err("%s: set mode: %s",
Ben Lindstrombd472262001-03-29 00:39:55 +0000954 np, strerror(errno));
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000955 } else {
956 if (!exists && omode != mode)
Damien Miller78315eb2000-09-29 23:01:36 +1100957#ifdef HAVE_FCHMOD
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000958 if (fchmod(ofd, omode & ~mask))
Damien Miller78315eb2000-09-29 23:01:36 +1100959#else /* HAVE_FCHMOD */
960 if (chmod(np, omode & ~mask))
961#endif /* HAVE_FCHMOD */
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000962 run_err("%s: set mode: %s",
Ben Lindstrombd472262001-03-29 00:39:55 +0000963 np, strerror(errno));
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000964 }
Damien Millerbe484b52000-07-15 14:14:16 +1000965 if (close(ofd) == -1) {
966 wrerr = YES;
967 wrerrno = errno;
968 }
Damien Miller95def091999-11-25 00:26:21 +1100969 (void) response();
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000970 if (setimes && wrerr == NO) {
971 setimes = 0;
Damien Miller62cee002000-09-23 17:15:56 +1100972 if (utimes(np, tv) < 0) {
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000973 run_err("%s: set times: %s",
Ben Lindstrombd472262001-03-29 00:39:55 +0000974 np, strerror(errno));
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000975 wrerr = DISPLAYED;
976 }
977 }
Damien Miller95def091999-11-25 00:26:21 +1100978 switch (wrerr) {
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000979 case YES:
980 run_err("%s: %s", np, strerror(wrerrno));
981 break;
982 case NO:
Darren Tucker9f63f222003-07-03 13:46:56 +1000983 (void) atomicio(vwrite, remout, "", 1);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000984 break;
985 case DISPLAYED:
986 break;
987 }
988 }
989screwup:
990 run_err("protocol error: %s", why);
991 exit(1);
992}
993
994int
Ben Lindstrom3c36bb22001-12-06 17:55:26 +0000995response(void)
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000996{
997 char ch, *cp, resp, rbuf[2048];
998
Damien Millere247cc42000-05-07 12:03:14 +1000999 if (atomicio(read, remin, &resp, sizeof(resp)) != sizeof(resp))
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001000 lostconn(0);
1001
1002 cp = rbuf;
Damien Miller95def091999-11-25 00:26:21 +11001003 switch (resp) {
1004 case 0: /* ok */
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001005 return (0);
1006 default:
1007 *cp++ = resp;
1008 /* FALLTHROUGH */
Damien Miller95def091999-11-25 00:26:21 +11001009 case 1: /* error, followed by error msg */
1010 case 2: /* fatal error, "" */
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001011 do {
Damien Millere247cc42000-05-07 12:03:14 +10001012 if (atomicio(read, remin, &ch, sizeof(ch)) != sizeof(ch))
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001013 lostconn(0);
1014 *cp++ = ch;
1015 } while (cp < &rbuf[sizeof(rbuf) - 1] && ch != '\n');
1016
1017 if (!iamremote)
Darren Tucker9f63f222003-07-03 13:46:56 +10001018 (void) atomicio(vwrite, STDERR_FILENO, rbuf, cp - rbuf);
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001019 ++errs;
1020 if (resp == 1)
1021 return (-1);
1022 exit(1);
1023 }
1024 /* NOTREACHED */
1025}
1026
1027void
Ben Lindstrom3c36bb22001-12-06 17:55:26 +00001028usage(void)
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001029{
Ben Lindstrom1e243242001-09-18 05:38:44 +00001030 (void) fprintf(stderr,
Darren Tucker1f203942003-10-15 15:50:42 +10001031 "usage: scp [-1246BCpqrv] [-c cipher] [-F ssh_config] [-i identity_file]\n"
1032 " [-l limit] [-o ssh_option] [-P port] [-S program]\n"
Ben Lindstrom45933dd2002-06-21 00:10:58 +00001033 " [[user@]host1:]file1 [...] [[user@]host2:]file2\n");
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001034 exit(1);
1035}
1036
1037void
Damien Miller95def091999-11-25 00:26:21 +11001038run_err(const char *fmt,...)
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001039{
1040 static FILE *fp;
1041 va_list ap;
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001042
1043 ++errs;
1044 if (fp == NULL && !(fp = fdopen(remout, "w")))
1045 return;
Damien Miller95def091999-11-25 00:26:21 +11001046 (void) fprintf(fp, "%c", 0x01);
1047 (void) fprintf(fp, "scp: ");
Damien Miller03783f01999-12-31 09:16:40 +11001048 va_start(ap, fmt);
Damien Miller95def091999-11-25 00:26:21 +11001049 (void) vfprintf(fp, fmt, ap);
Damien Miller03783f01999-12-31 09:16:40 +11001050 va_end(ap);
Damien Miller95def091999-11-25 00:26:21 +11001051 (void) fprintf(fp, "\n");
1052 (void) fflush(fp);
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001053
Damien Miller95def091999-11-25 00:26:21 +11001054 if (!iamremote) {
Damien Miller03783f01999-12-31 09:16:40 +11001055 va_start(ap, fmt);
Damien Miller95def091999-11-25 00:26:21 +11001056 vfprintf(stderr, fmt, ap);
Damien Miller03783f01999-12-31 09:16:40 +11001057 va_end(ap);
Damien Miller95def091999-11-25 00:26:21 +11001058 fprintf(stderr, "\n");
1059 }
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001060}
1061
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001062void
Damien Miller65d1f572003-06-04 22:51:08 +10001063verifydir(char *cp)
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001064{
1065 struct stat stb;
1066
1067 if (!stat(cp, &stb)) {
1068 if (S_ISDIR(stb.st_mode))
1069 return;
1070 errno = ENOTDIR;
1071 }
1072 run_err("%s: %s", cp, strerror(errno));
1073 exit(1);
1074}
1075
1076int
Damien Miller65d1f572003-06-04 22:51:08 +10001077okname(char *cp0)
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001078{
1079 int c;
1080 char *cp;
1081
1082 cp = cp0;
1083 do {
Ben Lindstrom7d5ed3a2001-06-25 04:28:30 +00001084 c = (int)*cp;
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001085 if (c & 0200)
1086 goto bad;
Damien Millereeeeb352003-02-24 11:52:58 +11001087 if (!isalpha(c) && !isdigit(c)) {
1088 switch (c) {
1089 case '\'':
1090 case '"':
1091 case '`':
1092 case ' ':
1093 case '#':
1094 goto bad;
1095 default:
1096 break;
1097 }
1098 }
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001099 } while (*++cp);
1100 return (1);
1101
Damien Miller98c7ad62000-03-09 21:27:49 +11001102bad: fprintf(stderr, "%s: invalid user name\n", cp0);
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001103 return (0);
1104}
1105
1106BUF *
Damien Miller65d1f572003-06-04 22:51:08 +10001107allocbuf(BUF *bp, int fd, int blksize)
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001108{
1109 size_t size;
Tim Rice13aae5e2001-10-21 17:53:58 -07001110#ifdef HAVE_STRUCT_STAT_ST_BLKSIZE
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001111 struct stat stb;
1112
1113 if (fstat(fd, &stb) < 0) {
1114 run_err("fstat: %s", strerror(errno));
1115 return (0);
1116 }
Ben Lindstrom418e0782002-12-23 02:22:09 +00001117 size = roundup(stb.st_blksize, blksize);
1118 if (size == 0)
Damien Miller95def091999-11-25 00:26:21 +11001119 size = blksize;
Tim Rice13aae5e2001-10-21 17:53:58 -07001120#else /* HAVE_STRUCT_STAT_ST_BLKSIZE */
Kevin Stevesef4eea92001-02-05 12:42:17 +00001121 size = blksize;
Tim Rice13aae5e2001-10-21 17:53:58 -07001122#endif /* HAVE_STRUCT_STAT_ST_BLKSIZE */
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001123 if (bp->cnt >= size)
1124 return (bp);
Damien Miller95def091999-11-25 00:26:21 +11001125 if (bp->buf == NULL)
1126 bp->buf = xmalloc(size);
1127 else
1128 bp->buf = xrealloc(bp->buf, size);
Ben Lindstrom5fccbc22001-09-12 17:49:48 +00001129 memset(bp->buf, 0, size);
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001130 bp->cnt = size;
1131 return (bp);
1132}
1133
1134void
Damien Miller65d1f572003-06-04 22:51:08 +10001135lostconn(int signo)
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001136{
1137 if (!iamremote)
Ben Lindstrom738f51e2001-06-21 03:08:58 +00001138 write(STDERR_FILENO, "lost connection\n", 16);
1139 if (signo)
1140 _exit(1);
1141 else
1142 exit(1);
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001143}