blob: d7eff70b438c5237914b796006b18e514fd09acb [file] [log] [blame]
Damien Miller57c38ac2011-09-22 21:42:45 +10001/* $OpenBSD: sftp-client.c,v 1.96 2011/09/12 08:46:15 markus Exp $ */
Damien Miller33804262001-02-04 23:20:18 +11002/*
Damien Miller4e60ed72004-02-17 17:07:59 +11003 * Copyright (c) 2001-2004 Damien Miller <djm@openbsd.org>
Damien Miller33804262001-02-04 23:20:18 +11004 *
Damien Miller4e60ed72004-02-17 17:07:59 +11005 * Permission to use, copy, modify, and distribute this software for any
6 * purpose with or without fee is hereby granted, provided that the above
7 * copyright notice and this permission notice appear in all copies.
Damien Miller33804262001-02-04 23:20:18 +11008 *
Damien Miller4e60ed72004-02-17 17:07:59 +11009 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
10 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
11 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
12 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
13 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
14 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
15 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
Damien Miller33804262001-02-04 23:20:18 +110016 */
17
18/* XXX: memleaks */
19/* XXX: signed vs unsigned */
Damien Miller3db5f532002-02-13 14:10:32 +110020/* XXX: remove all logging, only return status codes */
Damien Miller33804262001-02-04 23:20:18 +110021/* XXX: copy between two remote sites */
22
23#include "includes.h"
Damien Millerf17883e2006-03-15 11:45:54 +110024
25#include <sys/types.h>
Damien Miller8dbffe72006-08-05 11:02:17 +100026#include <sys/param.h>
Darren Tucker5b2e2ba2008-06-08 09:25:28 +100027#ifdef HAVE_SYS_STATVFS_H
Damien Millerd671e5a2008-05-19 14:53:33 +100028#include <sys/statvfs.h>
Darren Tucker5b2e2ba2008-06-08 09:25:28 +100029#endif
Damien Millerd7834352006-08-05 12:39:39 +100030#include "openbsd-compat/sys-queue.h"
Damien Millerf17883e2006-03-15 11:45:54 +110031#ifdef HAVE_SYS_STAT_H
32# include <sys/stat.h>
33#endif
Damien Miller9aec9192006-08-05 10:57:45 +100034#ifdef HAVE_SYS_TIME_H
35# include <sys/time.h>
36#endif
Damien Millerd7834352006-08-05 12:39:39 +100037#include <sys/uio.h>
Damien Miller57cf6382006-07-10 21:13:46 +100038
Darren Tucker1b0dd172009-10-07 08:37:48 +110039#include <dirent.h>
Darren Tucker39972492006-07-12 22:22:46 +100040#include <errno.h>
Damien Miller57cf6382006-07-10 21:13:46 +100041#include <fcntl.h>
42#include <signal.h>
Damien Millerd7834352006-08-05 12:39:39 +100043#include <stdarg.h>
Damien Millera7a73ee2006-08-05 11:37:59 +100044#include <stdio.h>
Damien Millere3476ed2006-07-24 14:13:33 +100045#include <string.h>
Damien Millere6b3b612006-07-24 14:01:23 +100046#include <unistd.h>
Damien Miller16a13332002-02-13 14:03:56 +110047
Damien Miller33804262001-02-04 23:20:18 +110048#include "xmalloc.h"
Damien Millerd7834352006-08-05 12:39:39 +100049#include "buffer.h"
Damien Miller33804262001-02-04 23:20:18 +110050#include "log.h"
51#include "atomicio.h"
Damien Miller62d57f62003-01-10 21:43:24 +110052#include "progressmeter.h"
Damien Miller3f941882006-03-31 23:13:02 +110053#include "misc.h"
Damien Miller33804262001-02-04 23:20:18 +110054
55#include "sftp.h"
56#include "sftp-common.h"
57#include "sftp-client.h"
58
Darren Tuckercdf547a2004-05-24 10:12:19 +100059extern volatile sig_atomic_t interrupted;
Damien Miller62d57f62003-01-10 21:43:24 +110060extern int showprogress;
61
Damien Miller0c8d8f62006-03-15 11:34:25 +110062/* Minimum amount of data to read at a time */
Damien Miller16a13332002-02-13 14:03:56 +110063#define MIN_READ_SIZE 512
64
Darren Tucker1b0dd172009-10-07 08:37:48 +110065/* Maximum depth to descend in directory trees */
66#define MAX_DIR_DEPTH 64
67
Damien Miller3db5f532002-02-13 14:10:32 +110068struct sftp_conn {
69 int fd_in;
70 int fd_out;
71 u_int transfer_buflen;
72 u_int num_requests;
73 u_int version;
74 u_int msg_id;
Damien Millerd671e5a2008-05-19 14:53:33 +100075#define SFTP_EXT_POSIX_RENAME 0x00000001
76#define SFTP_EXT_STATVFS 0x00000002
77#define SFTP_EXT_FSTATVFS 0x00000004
Darren Tuckeraf1f9092010-12-05 09:02:47 +110078#define SFTP_EXT_HARDLINK 0x00000008
Damien Miller7a3e1d02008-03-27 10:59:57 +110079 u_int exts;
Damien Miller65e42f82010-09-24 22:15:11 +100080 u_int64_t limit_kbps;
81 struct bwlimit bwlimit_in, bwlimit_out;
Damien Miller3db5f532002-02-13 14:10:32 +110082};
Ben Lindstrom288cc392001-02-09 02:58:04 +000083
Darren Tuckerc22f0902009-10-07 08:24:19 +110084static char *
Damien Miller65e42f82010-09-24 22:15:11 +100085get_handle(struct sftp_conn *conn, u_int expected_id, u_int *len,
86 const char *errfmt, ...) __attribute__((format(printf, 4, 5)));
87
88/* ARGSUSED */
89static int
90sftpio(void *_bwlimit, size_t amount)
91{
92 struct bwlimit *bwlimit = (struct bwlimit *)_bwlimit;
93
94 bandwidth_limit(bwlimit, amount);
95 return 0;
96}
Darren Tuckerc22f0902009-10-07 08:24:19 +110097
Ben Lindstrombba81212001-06-25 05:01:22 +000098static void
Damien Miller65e42f82010-09-24 22:15:11 +100099send_msg(struct sftp_conn *conn, Buffer *m)
Damien Miller33804262001-02-04 23:20:18 +1100100{
Damien Millera7f3aaa2003-01-10 21:43:58 +1100101 u_char mlen[4];
Damien Miller58ca98b2006-04-23 12:06:35 +1000102 struct iovec iov[2];
Damien Miller33804262001-02-04 23:20:18 +1100103
Damien Miller54446182006-01-02 23:40:50 +1100104 if (buffer_len(m) > SFTP_MAX_MSG_LENGTH)
Damien Millera7f3aaa2003-01-10 21:43:58 +1100105 fatal("Outbound message too long %u", buffer_len(m));
Damien Miller33804262001-02-04 23:20:18 +1100106
Damien Millera7f3aaa2003-01-10 21:43:58 +1100107 /* Send length first */
Damien Miller3f941882006-03-31 23:13:02 +1100108 put_u32(mlen, buffer_len(m));
Damien Miller58ca98b2006-04-23 12:06:35 +1000109 iov[0].iov_base = mlen;
110 iov[0].iov_len = sizeof(mlen);
111 iov[1].iov_base = buffer_ptr(m);
112 iov[1].iov_len = buffer_len(m);
Damien Millerd7834352006-08-05 12:39:39 +1000113
Damien Miller65e42f82010-09-24 22:15:11 +1000114 if (atomiciov6(writev, conn->fd_out, iov, 2,
115 conn->limit_kbps > 0 ? sftpio : NULL, &conn->bwlimit_out) !=
116 buffer_len(m) + sizeof(mlen))
Damien Millera7f3aaa2003-01-10 21:43:58 +1100117 fatal("Couldn't send packet: %s", strerror(errno));
118
119 buffer_clear(m);
Damien Miller33804262001-02-04 23:20:18 +1100120}
121
Ben Lindstrombba81212001-06-25 05:01:22 +0000122static void
Damien Miller65e42f82010-09-24 22:15:11 +1000123get_msg(struct sftp_conn *conn, Buffer *m)
Damien Miller33804262001-02-04 23:20:18 +1100124{
Damien Millera7f3aaa2003-01-10 21:43:58 +1100125 u_int msg_len;
Damien Miller33804262001-02-04 23:20:18 +1100126
Damien Millera7f3aaa2003-01-10 21:43:58 +1100127 buffer_append_space(m, 4);
Damien Miller65e42f82010-09-24 22:15:11 +1000128 if (atomicio6(read, conn->fd_in, buffer_ptr(m), 4,
129 conn->limit_kbps > 0 ? sftpio : NULL, &conn->bwlimit_in) != 4) {
Damien Millerb253cc42005-05-26 12:23:44 +1000130 if (errno == EPIPE)
131 fatal("Connection closed");
132 else
133 fatal("Couldn't read packet: %s", strerror(errno));
134 }
Damien Miller33804262001-02-04 23:20:18 +1100135
Damien Millera7f3aaa2003-01-10 21:43:58 +1100136 msg_len = buffer_get_int(m);
Damien Miller54446182006-01-02 23:40:50 +1100137 if (msg_len > SFTP_MAX_MSG_LENGTH)
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000138 fatal("Received message too long %u", msg_len);
Damien Miller33804262001-02-04 23:20:18 +1100139
Damien Millera7f3aaa2003-01-10 21:43:58 +1100140 buffer_append_space(m, msg_len);
Damien Miller65e42f82010-09-24 22:15:11 +1000141 if (atomicio6(read, conn->fd_in, buffer_ptr(m), msg_len,
142 conn->limit_kbps > 0 ? sftpio : NULL, &conn->bwlimit_in)
143 != msg_len) {
Damien Millerb253cc42005-05-26 12:23:44 +1000144 if (errno == EPIPE)
145 fatal("Connection closed");
146 else
147 fatal("Read packet: %s", strerror(errno));
148 }
Damien Miller33804262001-02-04 23:20:18 +1100149}
150
Ben Lindstrombba81212001-06-25 05:01:22 +0000151static void
Damien Miller65e42f82010-09-24 22:15:11 +1000152send_string_request(struct sftp_conn *conn, u_int id, u_int code, char *s,
Damien Miller33804262001-02-04 23:20:18 +1100153 u_int len)
154{
155 Buffer msg;
156
157 buffer_init(&msg);
158 buffer_put_char(&msg, code);
159 buffer_put_int(&msg, id);
160 buffer_put_string(&msg, s, len);
Damien Miller65e42f82010-09-24 22:15:11 +1000161 send_msg(conn, &msg);
162 debug3("Sent message fd %d T:%u I:%u", conn->fd_out, code, id);
Damien Miller33804262001-02-04 23:20:18 +1100163 buffer_free(&msg);
164}
165
Ben Lindstrombba81212001-06-25 05:01:22 +0000166static void
Damien Miller65e42f82010-09-24 22:15:11 +1000167send_string_attrs_request(struct sftp_conn *conn, u_int id, u_int code,
168 char *s, u_int len, Attrib *a)
Damien Miller33804262001-02-04 23:20:18 +1100169{
170 Buffer msg;
171
172 buffer_init(&msg);
173 buffer_put_char(&msg, code);
174 buffer_put_int(&msg, id);
175 buffer_put_string(&msg, s, len);
176 encode_attrib(&msg, a);
Damien Miller65e42f82010-09-24 22:15:11 +1000177 send_msg(conn, &msg);
178 debug3("Sent message fd %d T:%u I:%u", conn->fd_out, code, id);
Damien Miller33804262001-02-04 23:20:18 +1100179 buffer_free(&msg);
180}
181
Ben Lindstrombba81212001-06-25 05:01:22 +0000182static u_int
Damien Miller65e42f82010-09-24 22:15:11 +1000183get_status(struct sftp_conn *conn, u_int expected_id)
Damien Miller33804262001-02-04 23:20:18 +1100184{
185 Buffer msg;
186 u_int type, id, status;
187
188 buffer_init(&msg);
Damien Miller65e42f82010-09-24 22:15:11 +1000189 get_msg(conn, &msg);
Damien Miller33804262001-02-04 23:20:18 +1100190 type = buffer_get_char(&msg);
191 id = buffer_get_int(&msg);
192
193 if (id != expected_id)
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000194 fatal("ID mismatch (%u != %u)", id, expected_id);
Damien Miller33804262001-02-04 23:20:18 +1100195 if (type != SSH2_FXP_STATUS)
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000196 fatal("Expected SSH2_FXP_STATUS(%u) packet, got %u",
Damien Miller33804262001-02-04 23:20:18 +1100197 SSH2_FXP_STATUS, type);
198
199 status = buffer_get_int(&msg);
200 buffer_free(&msg);
201
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000202 debug3("SSH2_FXP_STATUS %u", status);
Damien Miller33804262001-02-04 23:20:18 +1100203
Damien Miller65e42f82010-09-24 22:15:11 +1000204 return status;
Damien Miller33804262001-02-04 23:20:18 +1100205}
206
Ben Lindstrombba81212001-06-25 05:01:22 +0000207static char *
Damien Miller65e42f82010-09-24 22:15:11 +1000208get_handle(struct sftp_conn *conn, u_int expected_id, u_int *len,
209 const char *errfmt, ...)
Damien Miller33804262001-02-04 23:20:18 +1100210{
211 Buffer msg;
212 u_int type, id;
Darren Tuckerc22f0902009-10-07 08:24:19 +1100213 char *handle, errmsg[256];
214 va_list args;
215 int status;
216
217 va_start(args, errfmt);
218 if (errfmt != NULL)
219 vsnprintf(errmsg, sizeof(errmsg), errfmt, args);
220 va_end(args);
Damien Miller33804262001-02-04 23:20:18 +1100221
222 buffer_init(&msg);
Damien Miller65e42f82010-09-24 22:15:11 +1000223 get_msg(conn, &msg);
Damien Miller33804262001-02-04 23:20:18 +1100224 type = buffer_get_char(&msg);
225 id = buffer_get_int(&msg);
226
227 if (id != expected_id)
Darren Tuckerc22f0902009-10-07 08:24:19 +1100228 fatal("%s: ID mismatch (%u != %u)",
229 errfmt == NULL ? __func__ : errmsg, id, expected_id);
Damien Miller33804262001-02-04 23:20:18 +1100230 if (type == SSH2_FXP_STATUS) {
Darren Tuckerc22f0902009-10-07 08:24:19 +1100231 status = buffer_get_int(&msg);
232 if (errfmt != NULL)
233 error("%s: %s", errmsg, fx2txt(status));
Darren Tuckercd516ef2004-12-06 22:43:43 +1100234 buffer_free(&msg);
Damien Miller33804262001-02-04 23:20:18 +1100235 return(NULL);
236 } else if (type != SSH2_FXP_HANDLE)
Darren Tuckerc22f0902009-10-07 08:24:19 +1100237 fatal("%s: Expected SSH2_FXP_HANDLE(%u) packet, got %u",
238 errfmt == NULL ? __func__ : errmsg, SSH2_FXP_HANDLE, type);
Damien Miller33804262001-02-04 23:20:18 +1100239
240 handle = buffer_get_string(&msg, len);
241 buffer_free(&msg);
242
243 return(handle);
244}
245
Ben Lindstrombba81212001-06-25 05:01:22 +0000246static Attrib *
Damien Miller65e42f82010-09-24 22:15:11 +1000247get_decode_stat(struct sftp_conn *conn, u_int expected_id, int quiet)
Damien Miller33804262001-02-04 23:20:18 +1100248{
249 Buffer msg;
250 u_int type, id;
251 Attrib *a;
252
253 buffer_init(&msg);
Damien Miller65e42f82010-09-24 22:15:11 +1000254 get_msg(conn, &msg);
Damien Miller33804262001-02-04 23:20:18 +1100255
256 type = buffer_get_char(&msg);
257 id = buffer_get_int(&msg);
258
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000259 debug3("Received stat reply T:%u I:%u", type, id);
Damien Miller33804262001-02-04 23:20:18 +1100260 if (id != expected_id)
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000261 fatal("ID mismatch (%u != %u)", id, expected_id);
Damien Miller33804262001-02-04 23:20:18 +1100262 if (type == SSH2_FXP_STATUS) {
263 int status = buffer_get_int(&msg);
264
Ben Lindstromc8d1c302001-03-17 00:34:46 +0000265 if (quiet)
266 debug("Couldn't stat remote file: %s", fx2txt(status));
267 else
268 error("Couldn't stat remote file: %s", fx2txt(status));
Darren Tuckercd516ef2004-12-06 22:43:43 +1100269 buffer_free(&msg);
Damien Miller33804262001-02-04 23:20:18 +1100270 return(NULL);
271 } else if (type != SSH2_FXP_ATTRS) {
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000272 fatal("Expected SSH2_FXP_ATTRS(%u) packet, got %u",
Damien Miller33804262001-02-04 23:20:18 +1100273 SSH2_FXP_ATTRS, type);
274 }
275 a = decode_attrib(&msg);
276 buffer_free(&msg);
277
278 return(a);
279}
280
Damien Millerd671e5a2008-05-19 14:53:33 +1000281static int
Damien Miller65e42f82010-09-24 22:15:11 +1000282get_decode_statvfs(struct sftp_conn *conn, struct sftp_statvfs *st,
283 u_int expected_id, int quiet)
Damien Millerd671e5a2008-05-19 14:53:33 +1000284{
285 Buffer msg;
286 u_int type, id, flag;
287
288 buffer_init(&msg);
Damien Miller65e42f82010-09-24 22:15:11 +1000289 get_msg(conn, &msg);
Damien Millerd671e5a2008-05-19 14:53:33 +1000290
291 type = buffer_get_char(&msg);
292 id = buffer_get_int(&msg);
293
294 debug3("Received statvfs reply T:%u I:%u", type, id);
295 if (id != expected_id)
296 fatal("ID mismatch (%u != %u)", id, expected_id);
297 if (type == SSH2_FXP_STATUS) {
298 int status = buffer_get_int(&msg);
299
300 if (quiet)
301 debug("Couldn't statvfs: %s", fx2txt(status));
302 else
303 error("Couldn't statvfs: %s", fx2txt(status));
304 buffer_free(&msg);
305 return -1;
306 } else if (type != SSH2_FXP_EXTENDED_REPLY) {
307 fatal("Expected SSH2_FXP_EXTENDED_REPLY(%u) packet, got %u",
308 SSH2_FXP_EXTENDED_REPLY, type);
309 }
310
311 bzero(st, sizeof(*st));
Darren Tucker7b598892008-06-09 22:49:36 +1000312 st->f_bsize = buffer_get_int64(&msg);
313 st->f_frsize = buffer_get_int64(&msg);
Damien Millerd671e5a2008-05-19 14:53:33 +1000314 st->f_blocks = buffer_get_int64(&msg);
315 st->f_bfree = buffer_get_int64(&msg);
316 st->f_bavail = buffer_get_int64(&msg);
317 st->f_files = buffer_get_int64(&msg);
318 st->f_ffree = buffer_get_int64(&msg);
319 st->f_favail = buffer_get_int64(&msg);
Darren Tucker294b8412008-06-08 12:57:08 +1000320 st->f_fsid = buffer_get_int64(&msg);
Darren Tucker7b598892008-06-09 22:49:36 +1000321 flag = buffer_get_int64(&msg);
322 st->f_namemax = buffer_get_int64(&msg);
Damien Millerd671e5a2008-05-19 14:53:33 +1000323
324 st->f_flag = (flag & SSH2_FXE_STATVFS_ST_RDONLY) ? ST_RDONLY : 0;
325 st->f_flag |= (flag & SSH2_FXE_STATVFS_ST_NOSUID) ? ST_NOSUID : 0;
326
327 buffer_free(&msg);
328
329 return 0;
330}
331
Damien Miller3db5f532002-02-13 14:10:32 +1100332struct sftp_conn *
Damien Miller65e42f82010-09-24 22:15:11 +1000333do_init(int fd_in, int fd_out, u_int transfer_buflen, u_int num_requests,
334 u_int64_t limit_kbps)
Damien Miller33804262001-02-04 23:20:18 +1100335{
Damien Miller65e42f82010-09-24 22:15:11 +1000336 u_int type;
Damien Miller33804262001-02-04 23:20:18 +1100337 Buffer msg;
Damien Miller3db5f532002-02-13 14:10:32 +1100338 struct sftp_conn *ret;
Damien Miller33804262001-02-04 23:20:18 +1100339
Damien Miller65e42f82010-09-24 22:15:11 +1000340 ret = xmalloc(sizeof(*ret));
341 ret->fd_in = fd_in;
342 ret->fd_out = fd_out;
343 ret->transfer_buflen = transfer_buflen;
344 ret->num_requests = num_requests;
345 ret->exts = 0;
346 ret->limit_kbps = 0;
347
Damien Miller33804262001-02-04 23:20:18 +1100348 buffer_init(&msg);
349 buffer_put_char(&msg, SSH2_FXP_INIT);
350 buffer_put_int(&msg, SSH2_FILEXFER_VERSION);
Damien Miller65e42f82010-09-24 22:15:11 +1000351 send_msg(ret, &msg);
Damien Miller33804262001-02-04 23:20:18 +1100352
353 buffer_clear(&msg);
354
Damien Miller65e42f82010-09-24 22:15:11 +1000355 get_msg(ret, &msg);
Damien Miller33804262001-02-04 23:20:18 +1100356
Kevin Stevesef4eea92001-02-05 12:42:17 +0000357 /* Expecting a VERSION reply */
Damien Miller33804262001-02-04 23:20:18 +1100358 if ((type = buffer_get_char(&msg)) != SSH2_FXP_VERSION) {
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000359 error("Invalid packet back from SSH2_FXP_INIT (type %u)",
Damien Miller33804262001-02-04 23:20:18 +1100360 type);
361 buffer_free(&msg);
Damien Miller3db5f532002-02-13 14:10:32 +1100362 return(NULL);
Damien Miller33804262001-02-04 23:20:18 +1100363 }
Damien Miller65e42f82010-09-24 22:15:11 +1000364 ret->version = buffer_get_int(&msg);
Damien Miller33804262001-02-04 23:20:18 +1100365
Damien Miller65e42f82010-09-24 22:15:11 +1000366 debug2("Remote version: %u", ret->version);
Damien Miller33804262001-02-04 23:20:18 +1100367
368 /* Check for extensions */
369 while (buffer_len(&msg) > 0) {
370 char *name = buffer_get_string(&msg, NULL);
371 char *value = buffer_get_string(&msg, NULL);
Darren Tuckera64ab332008-06-13 07:01:29 +1000372 int known = 0;
Damien Miller33804262001-02-04 23:20:18 +1100373
Damien Millerd671e5a2008-05-19 14:53:33 +1000374 if (strcmp(name, "posix-rename@openssh.com") == 0 &&
Darren Tuckera64ab332008-06-13 07:01:29 +1000375 strcmp(value, "1") == 0) {
Damien Miller65e42f82010-09-24 22:15:11 +1000376 ret->exts |= SFTP_EXT_POSIX_RENAME;
Darren Tuckera64ab332008-06-13 07:01:29 +1000377 known = 1;
378 } else if (strcmp(name, "statvfs@openssh.com") == 0 &&
379 strcmp(value, "2") == 0) {
Damien Miller65e42f82010-09-24 22:15:11 +1000380 ret->exts |= SFTP_EXT_STATVFS;
Darren Tuckera64ab332008-06-13 07:01:29 +1000381 known = 1;
Darren Tuckeraf1f9092010-12-05 09:02:47 +1100382 } else if (strcmp(name, "fstatvfs@openssh.com") == 0 &&
Darren Tuckera64ab332008-06-13 07:01:29 +1000383 strcmp(value, "2") == 0) {
Damien Miller65e42f82010-09-24 22:15:11 +1000384 ret->exts |= SFTP_EXT_FSTATVFS;
Darren Tuckera64ab332008-06-13 07:01:29 +1000385 known = 1;
Darren Tuckeraf1f9092010-12-05 09:02:47 +1100386 } else if (strcmp(name, "hardlink@openssh.com") == 0 &&
387 strcmp(value, "1") == 0) {
388 ret->exts |= SFTP_EXT_HARDLINK;
389 known = 1;
Darren Tuckera64ab332008-06-13 07:01:29 +1000390 }
391 if (known) {
392 debug2("Server supports extension \"%s\" revision %s",
393 name, value);
394 } else {
395 debug2("Unrecognised server extension \"%s\"", name);
396 }
Damien Miller33804262001-02-04 23:20:18 +1100397 xfree(name);
398 xfree(value);
399 }
400
401 buffer_free(&msg);
Damien Miller058316f2001-03-08 10:08:49 +1100402
Damien Miller3db5f532002-02-13 14:10:32 +1100403 /* Some filexfer v.0 servers don't support large packets */
Damien Miller65e42f82010-09-24 22:15:11 +1000404 if (ret->version == 0)
Ben Lindstroma1d81142002-04-02 20:58:11 +0000405 ret->transfer_buflen = MIN(ret->transfer_buflen, 20480);
Damien Miller3db5f532002-02-13 14:10:32 +1100406
Damien Miller65e42f82010-09-24 22:15:11 +1000407 ret->limit_kbps = limit_kbps;
408 if (ret->limit_kbps > 0) {
409 bandwidth_limit_init(&ret->bwlimit_in, ret->limit_kbps,
410 ret->transfer_buflen);
411 bandwidth_limit_init(&ret->bwlimit_out, ret->limit_kbps,
412 ret->transfer_buflen);
413 }
414
415 return ret;
Damien Miller3db5f532002-02-13 14:10:32 +1100416}
417
418u_int
419sftp_proto_version(struct sftp_conn *conn)
420{
Damien Miller65e42f82010-09-24 22:15:11 +1000421 return conn->version;
Damien Miller33804262001-02-04 23:20:18 +1100422}
423
424int
Damien Miller3db5f532002-02-13 14:10:32 +1100425do_close(struct sftp_conn *conn, char *handle, u_int handle_len)
Damien Miller33804262001-02-04 23:20:18 +1100426{
427 u_int id, status;
428 Buffer msg;
429
430 buffer_init(&msg);
431
Damien Miller3db5f532002-02-13 14:10:32 +1100432 id = conn->msg_id++;
Damien Miller33804262001-02-04 23:20:18 +1100433 buffer_put_char(&msg, SSH2_FXP_CLOSE);
434 buffer_put_int(&msg, id);
435 buffer_put_string(&msg, handle, handle_len);
Damien Miller65e42f82010-09-24 22:15:11 +1000436 send_msg(conn, &msg);
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000437 debug3("Sent message SSH2_FXP_CLOSE I:%u", id);
Damien Miller33804262001-02-04 23:20:18 +1100438
Damien Miller65e42f82010-09-24 22:15:11 +1000439 status = get_status(conn, id);
Damien Miller33804262001-02-04 23:20:18 +1100440 if (status != SSH2_FX_OK)
441 error("Couldn't close file: %s", fx2txt(status));
442
443 buffer_free(&msg);
444
Damien Miller65e42f82010-09-24 22:15:11 +1000445 return status;
Damien Miller33804262001-02-04 23:20:18 +1100446}
447
Damien Miller4870afd2001-03-14 10:27:09 +1100448
Ben Lindstrombba81212001-06-25 05:01:22 +0000449static int
Damien Miller3db5f532002-02-13 14:10:32 +1100450do_lsreaddir(struct sftp_conn *conn, char *path, int printflag,
Damien Miller4870afd2001-03-14 10:27:09 +1100451 SFTP_DIRENT ***dir)
Damien Miller33804262001-02-04 23:20:18 +1100452{
453 Buffer msg;
Damien Millereccb9de2005-06-17 12:59:34 +1000454 u_int count, type, id, handle_len, i, expected_id, ents = 0;
Damien Miller33804262001-02-04 23:20:18 +1100455 char *handle;
456
Damien Miller3db5f532002-02-13 14:10:32 +1100457 id = conn->msg_id++;
Damien Miller33804262001-02-04 23:20:18 +1100458
459 buffer_init(&msg);
460 buffer_put_char(&msg, SSH2_FXP_OPENDIR);
461 buffer_put_int(&msg, id);
462 buffer_put_cstring(&msg, path);
Damien Miller65e42f82010-09-24 22:15:11 +1000463 send_msg(conn, &msg);
Damien Miller33804262001-02-04 23:20:18 +1100464
Damien Miller65e42f82010-09-24 22:15:11 +1000465 handle = get_handle(conn, id, &handle_len,
Darren Tuckerc22f0902009-10-07 08:24:19 +1100466 "remote readdir(\"%s\")", path);
Damien Miller57c38ac2011-09-22 21:42:45 +1000467 if (handle == NULL) {
468 buffer_free(&msg);
Damien Miller65e42f82010-09-24 22:15:11 +1000469 return -1;
Damien Miller57c38ac2011-09-22 21:42:45 +1000470 }
Damien Miller33804262001-02-04 23:20:18 +1100471
Damien Miller4870afd2001-03-14 10:27:09 +1100472 if (dir) {
473 ents = 0;
474 *dir = xmalloc(sizeof(**dir));
475 (*dir)[0] = NULL;
476 }
Damien Miller4870afd2001-03-14 10:27:09 +1100477
Darren Tuckercdf547a2004-05-24 10:12:19 +1000478 for (; !interrupted;) {
Damien Miller3db5f532002-02-13 14:10:32 +1100479 id = expected_id = conn->msg_id++;
Damien Miller33804262001-02-04 23:20:18 +1100480
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000481 debug3("Sending SSH2_FXP_READDIR I:%u", id);
Damien Miller33804262001-02-04 23:20:18 +1100482
483 buffer_clear(&msg);
484 buffer_put_char(&msg, SSH2_FXP_READDIR);
485 buffer_put_int(&msg, id);
486 buffer_put_string(&msg, handle, handle_len);
Damien Miller65e42f82010-09-24 22:15:11 +1000487 send_msg(conn, &msg);
Damien Miller33804262001-02-04 23:20:18 +1100488
489 buffer_clear(&msg);
490
Damien Miller65e42f82010-09-24 22:15:11 +1000491 get_msg(conn, &msg);
Damien Miller33804262001-02-04 23:20:18 +1100492
493 type = buffer_get_char(&msg);
494 id = buffer_get_int(&msg);
495
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000496 debug3("Received reply T:%u I:%u", type, id);
Damien Miller33804262001-02-04 23:20:18 +1100497
498 if (id != expected_id)
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000499 fatal("ID mismatch (%u != %u)", id, expected_id);
Damien Miller33804262001-02-04 23:20:18 +1100500
501 if (type == SSH2_FXP_STATUS) {
502 int status = buffer_get_int(&msg);
503
504 debug3("Received SSH2_FXP_STATUS %d", status);
505
506 if (status == SSH2_FX_EOF) {
507 break;
508 } else {
509 error("Couldn't read directory: %s",
510 fx2txt(status));
Damien Miller3db5f532002-02-13 14:10:32 +1100511 do_close(conn, handle, handle_len);
Damien Miller00111382003-03-10 11:21:17 +1100512 xfree(handle);
Damien Miller57c38ac2011-09-22 21:42:45 +1000513 buffer_free(&msg);
Ben Lindstrom10ac33f2001-02-10 21:53:40 +0000514 return(status);
Damien Miller33804262001-02-04 23:20:18 +1100515 }
516 } else if (type != SSH2_FXP_NAME)
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000517 fatal("Expected SSH2_FXP_NAME(%u) packet, got %u",
Damien Miller33804262001-02-04 23:20:18 +1100518 SSH2_FXP_NAME, type);
519
520 count = buffer_get_int(&msg);
Damien Millerd7686fd2001-02-10 00:40:03 +1100521 if (count == 0)
522 break;
523 debug3("Received %d SSH2_FXP_NAME responses", count);
Damien Miller9f0f5c62001-12-21 14:45:46 +1100524 for (i = 0; i < count; i++) {
Damien Miller33804262001-02-04 23:20:18 +1100525 char *filename, *longname;
526 Attrib *a;
527
528 filename = buffer_get_string(&msg, NULL);
529 longname = buffer_get_string(&msg, NULL);
530 a = decode_attrib(&msg);
531
Damien Miller4870afd2001-03-14 10:27:09 +1100532 if (printflag)
533 printf("%s\n", longname);
534
Darren Tucker1b0dd172009-10-07 08:37:48 +1100535 /*
536 * Directory entries should never contain '/'
537 * These can be used to attack recursive ops
538 * (e.g. send '../../../../etc/passwd')
539 */
540 if (strchr(filename, '/') != NULL) {
541 error("Server sent suspect path \"%s\" "
542 "during readdir of \"%s\"", filename, path);
543 goto next;
544 }
545
Damien Miller4870afd2001-03-14 10:27:09 +1100546 if (dir) {
Damien Miller36812092006-03-26 14:22:47 +1100547 *dir = xrealloc(*dir, ents + 2, sizeof(**dir));
Damien Miller4870afd2001-03-14 10:27:09 +1100548 (*dir)[ents] = xmalloc(sizeof(***dir));
549 (*dir)[ents]->filename = xstrdup(filename);
550 (*dir)[ents]->longname = xstrdup(longname);
551 memcpy(&(*dir)[ents]->a, a, sizeof(*a));
552 (*dir)[++ents] = NULL;
553 }
Darren Tucker1b0dd172009-10-07 08:37:48 +1100554 next:
Damien Miller33804262001-02-04 23:20:18 +1100555 xfree(filename);
556 xfree(longname);
557 }
558 }
559
560 buffer_free(&msg);
Damien Miller3db5f532002-02-13 14:10:32 +1100561 do_close(conn, handle, handle_len);
Damien Miller33804262001-02-04 23:20:18 +1100562 xfree(handle);
563
Darren Tuckercdf547a2004-05-24 10:12:19 +1000564 /* Don't return partial matches on interrupt */
565 if (interrupted && dir != NULL && *dir != NULL) {
566 free_sftp_dirents(*dir);
567 *dir = xmalloc(sizeof(**dir));
568 **dir = NULL;
569 }
570
Damien Miller65e42f82010-09-24 22:15:11 +1000571 return 0;
Damien Miller33804262001-02-04 23:20:18 +1100572}
573
574int
Damien Miller3db5f532002-02-13 14:10:32 +1100575do_readdir(struct sftp_conn *conn, char *path, SFTP_DIRENT ***dir)
Damien Miller4870afd2001-03-14 10:27:09 +1100576{
Damien Miller3db5f532002-02-13 14:10:32 +1100577 return(do_lsreaddir(conn, path, 0, dir));
Damien Miller4870afd2001-03-14 10:27:09 +1100578}
579
580void free_sftp_dirents(SFTP_DIRENT **s)
581{
582 int i;
Damien Miller9f0f5c62001-12-21 14:45:46 +1100583
584 for (i = 0; s[i]; i++) {
Damien Miller4870afd2001-03-14 10:27:09 +1100585 xfree(s[i]->filename);
586 xfree(s[i]->longname);
587 xfree(s[i]);
588 }
589 xfree(s);
590}
591
592int
Damien Miller3db5f532002-02-13 14:10:32 +1100593do_rm(struct sftp_conn *conn, char *path)
Damien Miller33804262001-02-04 23:20:18 +1100594{
595 u_int status, id;
596
597 debug2("Sending SSH2_FXP_REMOVE \"%s\"", path);
598
Damien Miller3db5f532002-02-13 14:10:32 +1100599 id = conn->msg_id++;
Damien Miller65e42f82010-09-24 22:15:11 +1000600 send_string_request(conn, id, SSH2_FXP_REMOVE, path, strlen(path));
601 status = get_status(conn, id);
Damien Miller33804262001-02-04 23:20:18 +1100602 if (status != SSH2_FX_OK)
603 error("Couldn't delete file: %s", fx2txt(status));
604 return(status);
605}
606
607int
Darren Tucker1b0dd172009-10-07 08:37:48 +1100608do_mkdir(struct sftp_conn *conn, char *path, Attrib *a, int printflag)
Damien Miller33804262001-02-04 23:20:18 +1100609{
610 u_int status, id;
611
Damien Miller3db5f532002-02-13 14:10:32 +1100612 id = conn->msg_id++;
Damien Miller65e42f82010-09-24 22:15:11 +1000613 send_string_attrs_request(conn, id, SSH2_FXP_MKDIR, path,
Damien Miller33804262001-02-04 23:20:18 +1100614 strlen(path), a);
615
Damien Miller65e42f82010-09-24 22:15:11 +1000616 status = get_status(conn, id);
Darren Tucker1b0dd172009-10-07 08:37:48 +1100617 if (status != SSH2_FX_OK && printflag)
Damien Miller33804262001-02-04 23:20:18 +1100618 error("Couldn't create directory: %s", fx2txt(status));
619
620 return(status);
621}
622
623int
Damien Miller3db5f532002-02-13 14:10:32 +1100624do_rmdir(struct sftp_conn *conn, char *path)
Damien Miller33804262001-02-04 23:20:18 +1100625{
626 u_int status, id;
627
Damien Miller3db5f532002-02-13 14:10:32 +1100628 id = conn->msg_id++;
Damien Miller65e42f82010-09-24 22:15:11 +1000629 send_string_request(conn, id, SSH2_FXP_RMDIR, path,
Damien Miller3db5f532002-02-13 14:10:32 +1100630 strlen(path));
Damien Miller33804262001-02-04 23:20:18 +1100631
Damien Miller65e42f82010-09-24 22:15:11 +1000632 status = get_status(conn, id);
Damien Miller33804262001-02-04 23:20:18 +1100633 if (status != SSH2_FX_OK)
634 error("Couldn't remove directory: %s", fx2txt(status));
635
636 return(status);
637}
638
639Attrib *
Damien Miller3db5f532002-02-13 14:10:32 +1100640do_stat(struct sftp_conn *conn, char *path, int quiet)
Damien Miller33804262001-02-04 23:20:18 +1100641{
642 u_int id;
643
Damien Miller3db5f532002-02-13 14:10:32 +1100644 id = conn->msg_id++;
645
Damien Miller65e42f82010-09-24 22:15:11 +1000646 send_string_request(conn, id,
Ben Lindstrom6328ab32002-03-22 02:54:23 +0000647 conn->version == 0 ? SSH2_FXP_STAT_VERSION_0 : SSH2_FXP_STAT,
Damien Miller3db5f532002-02-13 14:10:32 +1100648 path, strlen(path));
649
Damien Miller65e42f82010-09-24 22:15:11 +1000650 return(get_decode_stat(conn, id, quiet));
Damien Miller33804262001-02-04 23:20:18 +1100651}
652
653Attrib *
Damien Miller3db5f532002-02-13 14:10:32 +1100654do_lstat(struct sftp_conn *conn, char *path, int quiet)
Damien Miller33804262001-02-04 23:20:18 +1100655{
656 u_int id;
657
Damien Miller3db5f532002-02-13 14:10:32 +1100658 if (conn->version == 0) {
659 if (quiet)
660 debug("Server version does not support lstat operation");
661 else
Damien Miller996acd22003-04-09 20:59:48 +1000662 logit("Server version does not support lstat operation");
Ben Lindstromf26ff5b2002-04-02 21:00:31 +0000663 return(do_stat(conn, path, quiet));
Damien Miller3db5f532002-02-13 14:10:32 +1100664 }
665
666 id = conn->msg_id++;
Damien Miller65e42f82010-09-24 22:15:11 +1000667 send_string_request(conn, id, SSH2_FXP_LSTAT, path,
Damien Miller3db5f532002-02-13 14:10:32 +1100668 strlen(path));
669
Damien Miller65e42f82010-09-24 22:15:11 +1000670 return(get_decode_stat(conn, id, quiet));
Damien Miller33804262001-02-04 23:20:18 +1100671}
672
Damien Millercfe23d32008-02-10 22:20:44 +1100673#ifdef notyet
Damien Miller33804262001-02-04 23:20:18 +1100674Attrib *
Damien Miller3db5f532002-02-13 14:10:32 +1100675do_fstat(struct sftp_conn *conn, char *handle, u_int handle_len, int quiet)
Damien Miller33804262001-02-04 23:20:18 +1100676{
677 u_int id;
678
Damien Miller3db5f532002-02-13 14:10:32 +1100679 id = conn->msg_id++;
Damien Miller65e42f82010-09-24 22:15:11 +1000680 send_string_request(conn, id, SSH2_FXP_FSTAT, handle,
Damien Miller3db5f532002-02-13 14:10:32 +1100681 handle_len);
682
Damien Miller65e42f82010-09-24 22:15:11 +1000683 return(get_decode_stat(conn, id, quiet));
Damien Miller33804262001-02-04 23:20:18 +1100684}
Damien Millercfe23d32008-02-10 22:20:44 +1100685#endif
Damien Miller33804262001-02-04 23:20:18 +1100686
687int
Damien Miller3db5f532002-02-13 14:10:32 +1100688do_setstat(struct sftp_conn *conn, char *path, Attrib *a)
Damien Miller33804262001-02-04 23:20:18 +1100689{
690 u_int status, id;
691
Damien Miller3db5f532002-02-13 14:10:32 +1100692 id = conn->msg_id++;
Damien Miller65e42f82010-09-24 22:15:11 +1000693 send_string_attrs_request(conn, id, SSH2_FXP_SETSTAT, path,
Damien Miller33804262001-02-04 23:20:18 +1100694 strlen(path), a);
695
Damien Miller65e42f82010-09-24 22:15:11 +1000696 status = get_status(conn, id);
Damien Miller33804262001-02-04 23:20:18 +1100697 if (status != SSH2_FX_OK)
698 error("Couldn't setstat on \"%s\": %s", path,
699 fx2txt(status));
700
701 return(status);
702}
703
704int
Damien Miller3db5f532002-02-13 14:10:32 +1100705do_fsetstat(struct sftp_conn *conn, char *handle, u_int handle_len,
Damien Miller33804262001-02-04 23:20:18 +1100706 Attrib *a)
707{
708 u_int status, id;
709
Damien Miller3db5f532002-02-13 14:10:32 +1100710 id = conn->msg_id++;
Damien Miller65e42f82010-09-24 22:15:11 +1000711 send_string_attrs_request(conn, id, SSH2_FXP_FSETSTAT, handle,
Damien Miller33804262001-02-04 23:20:18 +1100712 handle_len, a);
713
Damien Miller65e42f82010-09-24 22:15:11 +1000714 status = get_status(conn, id);
Damien Miller33804262001-02-04 23:20:18 +1100715 if (status != SSH2_FX_OK)
716 error("Couldn't fsetstat: %s", fx2txt(status));
717
718 return(status);
719}
720
721char *
Damien Miller3db5f532002-02-13 14:10:32 +1100722do_realpath(struct sftp_conn *conn, char *path)
Damien Miller33804262001-02-04 23:20:18 +1100723{
724 Buffer msg;
725 u_int type, expected_id, count, id;
726 char *filename, *longname;
727 Attrib *a;
728
Damien Miller3db5f532002-02-13 14:10:32 +1100729 expected_id = id = conn->msg_id++;
Damien Miller65e42f82010-09-24 22:15:11 +1000730 send_string_request(conn, id, SSH2_FXP_REALPATH, path,
Damien Miller3db5f532002-02-13 14:10:32 +1100731 strlen(path));
Damien Miller33804262001-02-04 23:20:18 +1100732
733 buffer_init(&msg);
734
Damien Miller65e42f82010-09-24 22:15:11 +1000735 get_msg(conn, &msg);
Damien Miller33804262001-02-04 23:20:18 +1100736 type = buffer_get_char(&msg);
737 id = buffer_get_int(&msg);
738
739 if (id != expected_id)
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000740 fatal("ID mismatch (%u != %u)", id, expected_id);
Damien Miller33804262001-02-04 23:20:18 +1100741
742 if (type == SSH2_FXP_STATUS) {
743 u_int status = buffer_get_int(&msg);
744
745 error("Couldn't canonicalise: %s", fx2txt(status));
Damien Miller49566312010-06-26 09:38:23 +1000746 buffer_free(&msg);
747 return NULL;
Damien Miller33804262001-02-04 23:20:18 +1100748 } else if (type != SSH2_FXP_NAME)
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000749 fatal("Expected SSH2_FXP_NAME(%u) packet, got %u",
Damien Miller33804262001-02-04 23:20:18 +1100750 SSH2_FXP_NAME, type);
751
752 count = buffer_get_int(&msg);
753 if (count != 1)
754 fatal("Got multiple names (%d) from SSH_FXP_REALPATH", count);
755
756 filename = buffer_get_string(&msg, NULL);
757 longname = buffer_get_string(&msg, NULL);
758 a = decode_attrib(&msg);
759
760 debug3("SSH_FXP_REALPATH %s -> %s", path, filename);
761
762 xfree(longname);
763
764 buffer_free(&msg);
765
766 return(filename);
767}
768
769int
Damien Miller3db5f532002-02-13 14:10:32 +1100770do_rename(struct sftp_conn *conn, char *oldpath, char *newpath)
Damien Miller33804262001-02-04 23:20:18 +1100771{
772 Buffer msg;
773 u_int status, id;
774
775 buffer_init(&msg);
776
777 /* Send rename request */
Damien Miller3db5f532002-02-13 14:10:32 +1100778 id = conn->msg_id++;
Damien Miller7a3e1d02008-03-27 10:59:57 +1100779 if ((conn->exts & SFTP_EXT_POSIX_RENAME)) {
780 buffer_put_char(&msg, SSH2_FXP_EXTENDED);
781 buffer_put_int(&msg, id);
782 buffer_put_cstring(&msg, "posix-rename@openssh.com");
783 } else {
784 buffer_put_char(&msg, SSH2_FXP_RENAME);
785 buffer_put_int(&msg, id);
786 }
Damien Miller33804262001-02-04 23:20:18 +1100787 buffer_put_cstring(&msg, oldpath);
788 buffer_put_cstring(&msg, newpath);
Damien Miller65e42f82010-09-24 22:15:11 +1000789 send_msg(conn, &msg);
Damien Miller7a3e1d02008-03-27 10:59:57 +1100790 debug3("Sent message %s \"%s\" -> \"%s\"",
791 (conn->exts & SFTP_EXT_POSIX_RENAME) ? "posix-rename@openssh.com" :
792 "SSH2_FXP_RENAME", oldpath, newpath);
Damien Miller33804262001-02-04 23:20:18 +1100793 buffer_free(&msg);
794
Damien Miller65e42f82010-09-24 22:15:11 +1000795 status = get_status(conn, id);
Damien Miller33804262001-02-04 23:20:18 +1100796 if (status != SSH2_FX_OK)
Damien Miller3db5f532002-02-13 14:10:32 +1100797 error("Couldn't rename file \"%s\" to \"%s\": %s", oldpath,
798 newpath, fx2txt(status));
Damien Miller33804262001-02-04 23:20:18 +1100799
800 return(status);
801}
802
803int
Darren Tuckeraf1f9092010-12-05 09:02:47 +1100804do_hardlink(struct sftp_conn *conn, char *oldpath, char *newpath)
805{
806 Buffer msg;
807 u_int status, id;
808
Darren Tuckeraf1f9092010-12-05 09:02:47 +1100809 if ((conn->exts & SFTP_EXT_HARDLINK) == 0) {
810 error("Server does not support hardlink@openssh.com extension");
811 return -1;
812 }
813
Damien Miller3decdba2011-09-22 21:41:05 +1000814 buffer_init(&msg);
815
816 /* Send link request */
817 id = conn->msg_id++;
Darren Tuckeraf1f9092010-12-05 09:02:47 +1100818 buffer_put_char(&msg, SSH2_FXP_EXTENDED);
819 buffer_put_int(&msg, id);
820 buffer_put_cstring(&msg, "hardlink@openssh.com");
821 buffer_put_cstring(&msg, oldpath);
822 buffer_put_cstring(&msg, newpath);
823 send_msg(conn, &msg);
824 debug3("Sent message hardlink@openssh.com \"%s\" -> \"%s\"",
825 oldpath, newpath);
826 buffer_free(&msg);
827
828 status = get_status(conn, id);
829 if (status != SSH2_FX_OK)
830 error("Couldn't link file \"%s\" to \"%s\": %s", oldpath,
831 newpath, fx2txt(status));
832
833 return(status);
834}
835
836int
Damien Miller3db5f532002-02-13 14:10:32 +1100837do_symlink(struct sftp_conn *conn, char *oldpath, char *newpath)
Damien Miller058316f2001-03-08 10:08:49 +1100838{
839 Buffer msg;
840 u_int status, id;
841
Damien Miller3db5f532002-02-13 14:10:32 +1100842 if (conn->version < 3) {
843 error("This server does not support the symlink operation");
844 return(SSH2_FX_OP_UNSUPPORTED);
845 }
846
Damien Miller058316f2001-03-08 10:08:49 +1100847 buffer_init(&msg);
848
Darren Tuckerdca6a4d2004-04-19 22:10:52 +1000849 /* Send symlink request */
Damien Miller3db5f532002-02-13 14:10:32 +1100850 id = conn->msg_id++;
Damien Miller058316f2001-03-08 10:08:49 +1100851 buffer_put_char(&msg, SSH2_FXP_SYMLINK);
852 buffer_put_int(&msg, id);
853 buffer_put_cstring(&msg, oldpath);
854 buffer_put_cstring(&msg, newpath);
Damien Miller65e42f82010-09-24 22:15:11 +1000855 send_msg(conn, &msg);
Damien Miller058316f2001-03-08 10:08:49 +1100856 debug3("Sent message SSH2_FXP_SYMLINK \"%s\" -> \"%s\"", oldpath,
857 newpath);
858 buffer_free(&msg);
859
Damien Miller65e42f82010-09-24 22:15:11 +1000860 status = get_status(conn, id);
Damien Miller058316f2001-03-08 10:08:49 +1100861 if (status != SSH2_FX_OK)
Ben Lindstrom8e879cf2002-11-09 15:48:49 +0000862 error("Couldn't symlink file \"%s\" to \"%s\": %s", oldpath,
Damien Miller3db5f532002-02-13 14:10:32 +1100863 newpath, fx2txt(status));
Damien Miller058316f2001-03-08 10:08:49 +1100864
865 return(status);
866}
867
Damien Millercfe23d32008-02-10 22:20:44 +1100868#ifdef notyet
Damien Miller058316f2001-03-08 10:08:49 +1100869char *
Damien Miller3db5f532002-02-13 14:10:32 +1100870do_readlink(struct sftp_conn *conn, char *path)
Damien Miller058316f2001-03-08 10:08:49 +1100871{
872 Buffer msg;
873 u_int type, expected_id, count, id;
874 char *filename, *longname;
875 Attrib *a;
876
Damien Miller3db5f532002-02-13 14:10:32 +1100877 expected_id = id = conn->msg_id++;
Damien Miller65e42f82010-09-24 22:15:11 +1000878 send_string_request(conn, id, SSH2_FXP_READLINK, path, strlen(path));
Damien Miller058316f2001-03-08 10:08:49 +1100879
880 buffer_init(&msg);
881
Damien Miller65e42f82010-09-24 22:15:11 +1000882 get_msg(conn, &msg);
Damien Miller058316f2001-03-08 10:08:49 +1100883 type = buffer_get_char(&msg);
884 id = buffer_get_int(&msg);
885
886 if (id != expected_id)
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000887 fatal("ID mismatch (%u != %u)", id, expected_id);
Damien Miller058316f2001-03-08 10:08:49 +1100888
889 if (type == SSH2_FXP_STATUS) {
890 u_int status = buffer_get_int(&msg);
891
892 error("Couldn't readlink: %s", fx2txt(status));
Damien Miller3decdba2011-09-22 21:41:05 +1000893 buffer_free(&msg);
Damien Miller058316f2001-03-08 10:08:49 +1100894 return(NULL);
895 } else if (type != SSH2_FXP_NAME)
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000896 fatal("Expected SSH2_FXP_NAME(%u) packet, got %u",
Damien Miller058316f2001-03-08 10:08:49 +1100897 SSH2_FXP_NAME, type);
898
899 count = buffer_get_int(&msg);
900 if (count != 1)
901 fatal("Got multiple names (%d) from SSH_FXP_READLINK", count);
902
903 filename = buffer_get_string(&msg, NULL);
904 longname = buffer_get_string(&msg, NULL);
905 a = decode_attrib(&msg);
906
907 debug3("SSH_FXP_READLINK %s -> %s", path, filename);
908
909 xfree(longname);
910
911 buffer_free(&msg);
912
913 return(filename);
914}
Damien Millercfe23d32008-02-10 22:20:44 +1100915#endif
Damien Miller058316f2001-03-08 10:08:49 +1100916
Damien Millerd671e5a2008-05-19 14:53:33 +1000917int
Darren Tucker7b598892008-06-09 22:49:36 +1000918do_statvfs(struct sftp_conn *conn, const char *path, struct sftp_statvfs *st,
Damien Millerd671e5a2008-05-19 14:53:33 +1000919 int quiet)
920{
921 Buffer msg;
922 u_int id;
923
924 if ((conn->exts & SFTP_EXT_STATVFS) == 0) {
925 error("Server does not support statvfs@openssh.com extension");
926 return -1;
927 }
928
929 id = conn->msg_id++;
930
931 buffer_init(&msg);
932 buffer_clear(&msg);
933 buffer_put_char(&msg, SSH2_FXP_EXTENDED);
934 buffer_put_int(&msg, id);
935 buffer_put_cstring(&msg, "statvfs@openssh.com");
936 buffer_put_cstring(&msg, path);
Damien Miller65e42f82010-09-24 22:15:11 +1000937 send_msg(conn, &msg);
Damien Millerd671e5a2008-05-19 14:53:33 +1000938 buffer_free(&msg);
939
Damien Miller65e42f82010-09-24 22:15:11 +1000940 return get_decode_statvfs(conn, st, id, quiet);
Damien Millerd671e5a2008-05-19 14:53:33 +1000941}
942
943#ifdef notyet
944int
945do_fstatvfs(struct sftp_conn *conn, const char *handle, u_int handle_len,
Darren Tucker7b598892008-06-09 22:49:36 +1000946 struct sftp_statvfs *st, int quiet)
Damien Millerd671e5a2008-05-19 14:53:33 +1000947{
948 Buffer msg;
949 u_int id;
950
951 if ((conn->exts & SFTP_EXT_FSTATVFS) == 0) {
952 error("Server does not support fstatvfs@openssh.com extension");
953 return -1;
954 }
955
956 id = conn->msg_id++;
957
958 buffer_init(&msg);
959 buffer_clear(&msg);
960 buffer_put_char(&msg, SSH2_FXP_EXTENDED);
961 buffer_put_int(&msg, id);
962 buffer_put_cstring(&msg, "fstatvfs@openssh.com");
963 buffer_put_string(&msg, handle, handle_len);
Damien Miller65e42f82010-09-24 22:15:11 +1000964 send_msg(conn, &msg);
Damien Millerd671e5a2008-05-19 14:53:33 +1000965 buffer_free(&msg);
966
Damien Miller65e42f82010-09-24 22:15:11 +1000967 return get_decode_statvfs(conn, st, id, quiet);
Damien Millerd671e5a2008-05-19 14:53:33 +1000968}
969#endif
970
Damien Miller16a13332002-02-13 14:03:56 +1100971static void
Damien Miller65e42f82010-09-24 22:15:11 +1000972send_read_request(struct sftp_conn *conn, u_int id, u_int64_t offset,
973 u_int len, char *handle, u_int handle_len)
Damien Miller16a13332002-02-13 14:03:56 +1100974{
975 Buffer msg;
Ben Lindstrom6328ab32002-03-22 02:54:23 +0000976
Damien Miller16a13332002-02-13 14:03:56 +1100977 buffer_init(&msg);
978 buffer_clear(&msg);
979 buffer_put_char(&msg, SSH2_FXP_READ);
980 buffer_put_int(&msg, id);
981 buffer_put_string(&msg, handle, handle_len);
982 buffer_put_int64(&msg, offset);
983 buffer_put_int(&msg, len);
Damien Miller65e42f82010-09-24 22:15:11 +1000984 send_msg(conn, &msg);
Damien Miller16a13332002-02-13 14:03:56 +1100985 buffer_free(&msg);
Ben Lindstrom6328ab32002-03-22 02:54:23 +0000986}
Damien Miller16a13332002-02-13 14:03:56 +1100987
Damien Miller058316f2001-03-08 10:08:49 +1100988int
Damien Miller3db5f532002-02-13 14:10:32 +1100989do_download(struct sftp_conn *conn, char *remote_path, char *local_path,
Darren Tucker1b0dd172009-10-07 08:37:48 +1100990 Attrib *a, int pflag)
Damien Miller33804262001-02-04 23:20:18 +1100991{
Darren Tucker1b0dd172009-10-07 08:37:48 +1100992 Attrib junk;
Damien Miller16a13332002-02-13 14:03:56 +1100993 Buffer msg;
994 char *handle;
Darren Tucker40858532005-08-02 17:07:07 +1000995 int local_fd, status = 0, write_error;
Damien Miller16a13332002-02-13 14:03:56 +1100996 int read_error, write_errno;
997 u_int64_t offset, size;
Damien Millereccb9de2005-06-17 12:59:34 +1000998 u_int handle_len, mode, type, id, buflen, num_req, max_req;
Damien Miller62d57f62003-01-10 21:43:24 +1100999 off_t progress_counter;
Damien Miller16a13332002-02-13 14:03:56 +11001000 struct request {
1001 u_int id;
1002 u_int len;
1003 u_int64_t offset;
Ben Lindstrom6328ab32002-03-22 02:54:23 +00001004 TAILQ_ENTRY(request) tq;
Damien Miller16a13332002-02-13 14:03:56 +11001005 };
1006 TAILQ_HEAD(reqhead, request) requests;
1007 struct request *req;
1008
1009 TAILQ_INIT(&requests);
Damien Miller33804262001-02-04 23:20:18 +11001010
Darren Tucker1b0dd172009-10-07 08:37:48 +11001011 if (a == NULL && (a = do_stat(conn, remote_path, 0)) == NULL)
1012 return -1;
Damien Miller33804262001-02-04 23:20:18 +11001013
Damien Miller9e720282008-06-29 22:46:35 +10001014 /* Do not preserve set[ug]id here, as we do not preserve ownership */
Damien Miller33804262001-02-04 23:20:18 +11001015 if (a->flags & SSH2_FILEXFER_ATTR_PERMISSIONS)
Damien Miller770b3742003-01-08 14:04:53 +11001016 mode = a->perm & 0777;
Damien Miller33804262001-02-04 23:20:18 +11001017 else
1018 mode = 0666;
1019
Ben Lindstromc8d1c302001-03-17 00:34:46 +00001020 if ((a->flags & SSH2_FILEXFER_ATTR_PERMISSIONS) &&
Damien Miller5fa01fd2003-01-14 22:24:47 +11001021 (!S_ISREG(a->perm))) {
1022 error("Cannot download non-regular file: %s", remote_path);
Ben Lindstromc8d1c302001-03-17 00:34:46 +00001023 return(-1);
1024 }
1025
Damien Miller16a13332002-02-13 14:03:56 +11001026 if (a->flags & SSH2_FILEXFER_ATTR_SIZE)
1027 size = a->size;
1028 else
1029 size = 0;
1030
Damien Miller3db5f532002-02-13 14:10:32 +11001031 buflen = conn->transfer_buflen;
Damien Miller33804262001-02-04 23:20:18 +11001032 buffer_init(&msg);
1033
1034 /* Send open request */
Damien Miller3db5f532002-02-13 14:10:32 +11001035 id = conn->msg_id++;
Damien Miller33804262001-02-04 23:20:18 +11001036 buffer_put_char(&msg, SSH2_FXP_OPEN);
1037 buffer_put_int(&msg, id);
1038 buffer_put_cstring(&msg, remote_path);
1039 buffer_put_int(&msg, SSH2_FXF_READ);
1040 attrib_clear(&junk); /* Send empty attributes */
1041 encode_attrib(&msg, &junk);
Damien Miller65e42f82010-09-24 22:15:11 +10001042 send_msg(conn, &msg);
Ben Lindstromb1f483f2002-06-23 21:27:18 +00001043 debug3("Sent message SSH2_FXP_OPEN I:%u P:%s", id, remote_path);
Damien Miller33804262001-02-04 23:20:18 +11001044
Damien Miller65e42f82010-09-24 22:15:11 +10001045 handle = get_handle(conn, id, &handle_len,
Darren Tuckerc22f0902009-10-07 08:24:19 +11001046 "remote open(\"%s\")", remote_path);
Damien Miller33804262001-02-04 23:20:18 +11001047 if (handle == NULL) {
1048 buffer_free(&msg);
Damien Miller33804262001-02-04 23:20:18 +11001049 return(-1);
1050 }
1051
Damien Millera8e06ce2003-11-21 23:48:55 +11001052 local_fd = open(local_path, O_WRONLY | O_CREAT | O_TRUNC,
Damien Miller770b3742003-01-08 14:04:53 +11001053 mode | S_IWRITE);
Damien Miller3db5f532002-02-13 14:10:32 +11001054 if (local_fd == -1) {
1055 error("Couldn't open local file \"%s\" for writing: %s",
1056 local_path, strerror(errno));
Damien Miller6b0c8182008-02-10 22:23:41 +11001057 do_close(conn, handle, handle_len);
Ben Lindstrom021fcd32002-02-26 18:02:43 +00001058 buffer_free(&msg);
1059 xfree(handle);
Damien Miller3db5f532002-02-13 14:10:32 +11001060 return(-1);
1061 }
1062
Damien Miller33804262001-02-04 23:20:18 +11001063 /* Read from remote and write to local */
Damien Miller16a13332002-02-13 14:03:56 +11001064 write_error = read_error = write_errno = num_req = offset = 0;
1065 max_req = 1;
Damien Miller62d57f62003-01-10 21:43:24 +11001066 progress_counter = 0;
1067
Damien Miller9ba30692004-03-08 23:12:02 +11001068 if (showprogress && size != 0)
1069 start_progress_meter(remote_path, size, &progress_counter);
Damien Miller62d57f62003-01-10 21:43:24 +11001070
Damien Miller16a13332002-02-13 14:03:56 +11001071 while (num_req > 0 || max_req > 0) {
Damien Miller33804262001-02-04 23:20:18 +11001072 char *data;
Damien Miller16a13332002-02-13 14:03:56 +11001073 u_int len;
Damien Miller33804262001-02-04 23:20:18 +11001074
Darren Tuckercdf547a2004-05-24 10:12:19 +10001075 /*
Darren Tuckerfc959702004-07-17 16:12:08 +10001076 * Simulate EOF on interrupt: stop sending new requests and
Darren Tuckercdf547a2004-05-24 10:12:19 +10001077 * allow outstanding requests to drain gracefully
1078 */
1079 if (interrupted) {
1080 if (num_req == 0) /* If we haven't started yet... */
1081 break;
1082 max_req = 0;
1083 }
1084
Damien Miller16a13332002-02-13 14:03:56 +11001085 /* Send some more requests */
1086 while (num_req < max_req) {
Ben Lindstrom6328ab32002-03-22 02:54:23 +00001087 debug3("Request range %llu -> %llu (%d/%d)",
Ben Lindstromd45f28c2002-03-22 01:00:57 +00001088 (unsigned long long)offset,
1089 (unsigned long long)offset + buflen - 1,
1090 num_req, max_req);
Damien Miller16a13332002-02-13 14:03:56 +11001091 req = xmalloc(sizeof(*req));
Damien Miller3db5f532002-02-13 14:10:32 +11001092 req->id = conn->msg_id++;
Damien Miller16a13332002-02-13 14:03:56 +11001093 req->len = buflen;
1094 req->offset = offset;
1095 offset += buflen;
1096 num_req++;
1097 TAILQ_INSERT_TAIL(&requests, req, tq);
Damien Miller65e42f82010-09-24 22:15:11 +10001098 send_read_request(conn, req->id, req->offset,
Damien Miller16a13332002-02-13 14:03:56 +11001099 req->len, handle, handle_len);
1100 }
Damien Miller33804262001-02-04 23:20:18 +11001101
1102 buffer_clear(&msg);
Damien Miller65e42f82010-09-24 22:15:11 +10001103 get_msg(conn, &msg);
Damien Miller33804262001-02-04 23:20:18 +11001104 type = buffer_get_char(&msg);
1105 id = buffer_get_int(&msg);
Ben Lindstromb1f483f2002-06-23 21:27:18 +00001106 debug3("Received reply T:%u I:%u R:%d", type, id, max_req);
Damien Miller33804262001-02-04 23:20:18 +11001107
Damien Miller16a13332002-02-13 14:03:56 +11001108 /* Find the request in our queue */
Darren Tucker47eede72005-03-14 23:08:12 +11001109 for (req = TAILQ_FIRST(&requests);
Damien Miller16a13332002-02-13 14:03:56 +11001110 req != NULL && req->id != id;
1111 req = TAILQ_NEXT(req, tq))
1112 ;
1113 if (req == NULL)
1114 fatal("Unexpected reply %u", id);
1115
1116 switch (type) {
1117 case SSH2_FXP_STATUS:
1118 status = buffer_get_int(&msg);
1119 if (status != SSH2_FX_EOF)
1120 read_error = 1;
1121 max_req = 0;
1122 TAILQ_REMOVE(&requests, req, tq);
1123 xfree(req);
1124 num_req--;
1125 break;
1126 case SSH2_FXP_DATA:
1127 data = buffer_get_string(&msg, &len);
Ben Lindstromeb505452002-03-22 01:03:15 +00001128 debug3("Received data %llu -> %llu",
Ben Lindstrom6328ab32002-03-22 02:54:23 +00001129 (unsigned long long)req->offset,
Ben Lindstromeb505452002-03-22 01:03:15 +00001130 (unsigned long long)req->offset + len - 1);
Damien Miller16a13332002-02-13 14:03:56 +11001131 if (len > req->len)
1132 fatal("Received more data than asked for "
Ben Lindstrom93576d92002-12-23 02:06:19 +00001133 "%u > %u", len, req->len);
Damien Miller16a13332002-02-13 14:03:56 +11001134 if ((lseek(local_fd, req->offset, SEEK_SET) == -1 ||
Darren Tucker9f63f222003-07-03 13:46:56 +10001135 atomicio(vwrite, local_fd, data, len) != len) &&
Damien Miller16a13332002-02-13 14:03:56 +11001136 !write_error) {
1137 write_errno = errno;
1138 write_error = 1;
1139 max_req = 0;
Damien Miller33804262001-02-04 23:20:18 +11001140 }
Damien Miller62d57f62003-01-10 21:43:24 +11001141 progress_counter += len;
Damien Miller16a13332002-02-13 14:03:56 +11001142 xfree(data);
1143
1144 if (len == req->len) {
1145 TAILQ_REMOVE(&requests, req, tq);
1146 xfree(req);
1147 num_req--;
1148 } else {
1149 /* Resend the request for the missing data */
1150 debug3("Short data block, re-requesting "
Ben Lindstromeb505452002-03-22 01:03:15 +00001151 "%llu -> %llu (%2d)",
Ben Lindstrom6328ab32002-03-22 02:54:23 +00001152 (unsigned long long)req->offset + len,
Ben Lindstrom83b79e42002-03-22 01:05:27 +00001153 (unsigned long long)req->offset +
1154 req->len - 1, num_req);
Damien Miller3db5f532002-02-13 14:10:32 +11001155 req->id = conn->msg_id++;
Damien Miller16a13332002-02-13 14:03:56 +11001156 req->len -= len;
1157 req->offset += len;
Damien Miller65e42f82010-09-24 22:15:11 +10001158 send_read_request(conn, req->id,
Damien Miller3db5f532002-02-13 14:10:32 +11001159 req->offset, req->len, handle, handle_len);
Damien Miller16a13332002-02-13 14:03:56 +11001160 /* Reduce the request size */
1161 if (len < buflen)
1162 buflen = MAX(MIN_READ_SIZE, len);
1163 }
1164 if (max_req > 0) { /* max_req = 0 iff EOF received */
1165 if (size > 0 && offset > size) {
1166 /* Only one request at a time
1167 * after the expected EOF */
1168 debug3("Finish at %llu (%2d)",
Ben Lindstromeb505452002-03-22 01:03:15 +00001169 (unsigned long long)offset,
1170 num_req);
Damien Miller16a13332002-02-13 14:03:56 +11001171 max_req = 1;
Darren Tuckercdf547a2004-05-24 10:12:19 +10001172 } else if (max_req <= conn->num_requests) {
Damien Miller16a13332002-02-13 14:03:56 +11001173 ++max_req;
1174 }
1175 }
1176 break;
1177 default:
Ben Lindstromb1f483f2002-06-23 21:27:18 +00001178 fatal("Expected SSH2_FXP_DATA(%u) packet, got %u",
Damien Miller33804262001-02-04 23:20:18 +11001179 SSH2_FXP_DATA, type);
1180 }
Damien Miller33804262001-02-04 23:20:18 +11001181 }
Damien Miller33804262001-02-04 23:20:18 +11001182
Damien Miller62d57f62003-01-10 21:43:24 +11001183 if (showprogress && size)
1184 stop_progress_meter();
1185
Damien Miller16a13332002-02-13 14:03:56 +11001186 /* Sanity check */
1187 if (TAILQ_FIRST(&requests) != NULL)
1188 fatal("Transfer complete, but requests still in queue");
1189
1190 if (read_error) {
Ben Lindstrom6328ab32002-03-22 02:54:23 +00001191 error("Couldn't read from remote file \"%s\" : %s",
Damien Miller3db5f532002-02-13 14:10:32 +11001192 remote_path, fx2txt(status));
1193 do_close(conn, handle, handle_len);
Damien Miller16a13332002-02-13 14:03:56 +11001194 } else if (write_error) {
Damien Miller3db5f532002-02-13 14:10:32 +11001195 error("Couldn't write to \"%s\": %s", local_path,
1196 strerror(write_errno));
1197 status = -1;
1198 do_close(conn, handle, handle_len);
Damien Miller16a13332002-02-13 14:03:56 +11001199 } else {
Damien Miller3db5f532002-02-13 14:10:32 +11001200 status = do_close(conn, handle, handle_len);
Damien Miller16a13332002-02-13 14:03:56 +11001201
1202 /* Override umask and utimes if asked */
Ben Lindstrom6dc75f52001-02-17 16:47:47 +00001203#ifdef HAVE_FCHMOD
Damien Miller16a13332002-02-13 14:03:56 +11001204 if (pflag && fchmod(local_fd, mode) == -1)
Damien Millera8e06ce2003-11-21 23:48:55 +11001205#else
Damien Miller16a13332002-02-13 14:03:56 +11001206 if (pflag && chmod(local_path, mode) == -1)
Ben Lindstrom6dc75f52001-02-17 16:47:47 +00001207#endif /* HAVE_FCHMOD */
Damien Miller16a13332002-02-13 14:03:56 +11001208 error("Couldn't set mode on \"%s\": %s", local_path,
Ben Lindstrom93576d92002-12-23 02:06:19 +00001209 strerror(errno));
Damien Miller16a13332002-02-13 14:03:56 +11001210 if (pflag && (a->flags & SSH2_FILEXFER_ATTR_ACMODTIME)) {
1211 struct timeval tv[2];
1212 tv[0].tv_sec = a->atime;
1213 tv[1].tv_sec = a->mtime;
1214 tv[0].tv_usec = tv[1].tv_usec = 0;
1215 if (utimes(local_path, tv) == -1)
1216 error("Can't set times on \"%s\": %s",
Ben Lindstrom93576d92002-12-23 02:06:19 +00001217 local_path, strerror(errno));
Damien Miller16a13332002-02-13 14:03:56 +11001218 }
Ben Lindstrom9d4f2c82001-02-15 03:22:45 +00001219 }
Damien Millerd7686fd2001-02-10 00:40:03 +11001220 close(local_fd);
1221 buffer_free(&msg);
1222 xfree(handle);
Damien Miller3db5f532002-02-13 14:10:32 +11001223
1224 return(status);
Damien Miller33804262001-02-04 23:20:18 +11001225}
1226
Darren Tucker1b0dd172009-10-07 08:37:48 +11001227static int
1228download_dir_internal(struct sftp_conn *conn, char *src, char *dst,
1229 Attrib *dirattrib, int pflag, int printflag, int depth)
1230{
1231 int i, ret = 0;
1232 SFTP_DIRENT **dir_entries;
1233 char *filename, *new_src, *new_dst;
1234 mode_t mode = 0777;
1235
1236 if (depth >= MAX_DIR_DEPTH) {
1237 error("Maximum directory depth exceeded: %d levels", depth);
1238 return -1;
1239 }
1240
1241 if (dirattrib == NULL &&
1242 (dirattrib = do_stat(conn, src, 1)) == NULL) {
1243 error("Unable to stat remote directory \"%s\"", src);
1244 return -1;
1245 }
1246 if (!S_ISDIR(dirattrib->perm)) {
1247 error("\"%s\" is not a directory", src);
1248 return -1;
1249 }
1250 if (printflag)
1251 printf("Retrieving %s\n", src);
1252
1253 if (dirattrib->flags & SSH2_FILEXFER_ATTR_PERMISSIONS)
1254 mode = dirattrib->perm & 01777;
1255 else {
1256 debug("Server did not send permissions for "
1257 "directory \"%s\"", dst);
1258 }
1259
1260 if (mkdir(dst, mode) == -1 && errno != EEXIST) {
1261 error("mkdir %s: %s", dst, strerror(errno));
1262 return -1;
1263 }
1264
1265 if (do_readdir(conn, src, &dir_entries) == -1) {
1266 error("%s: Failed to get directory contents", src);
1267 return -1;
1268 }
1269
1270 for (i = 0; dir_entries[i] != NULL && !interrupted; i++) {
1271 filename = dir_entries[i]->filename;
1272
1273 new_dst = path_append(dst, filename);
1274 new_src = path_append(src, filename);
1275
1276 if (S_ISDIR(dir_entries[i]->a.perm)) {
1277 if (strcmp(filename, ".") == 0 ||
1278 strcmp(filename, "..") == 0)
1279 continue;
1280 if (download_dir_internal(conn, new_src, new_dst,
1281 &(dir_entries[i]->a), pflag, printflag,
1282 depth + 1) == -1)
1283 ret = -1;
1284 } else if (S_ISREG(dir_entries[i]->a.perm) ) {
1285 if (do_download(conn, new_src, new_dst,
1286 &(dir_entries[i]->a), pflag) == -1) {
1287 error("Download of file %s to %s failed",
1288 new_src, new_dst);
1289 ret = -1;
1290 }
1291 } else
1292 logit("%s: not a regular file\n", new_src);
1293
1294 xfree(new_dst);
1295 xfree(new_src);
1296 }
1297
1298 if (pflag) {
1299 if (dirattrib->flags & SSH2_FILEXFER_ATTR_ACMODTIME) {
1300 struct timeval tv[2];
1301 tv[0].tv_sec = dirattrib->atime;
1302 tv[1].tv_sec = dirattrib->mtime;
1303 tv[0].tv_usec = tv[1].tv_usec = 0;
1304 if (utimes(dst, tv) == -1)
1305 error("Can't set times on \"%s\": %s",
1306 dst, strerror(errno));
1307 } else
1308 debug("Server did not send times for directory "
1309 "\"%s\"", dst);
1310 }
1311
1312 free_sftp_dirents(dir_entries);
1313
1314 return ret;
1315}
1316
1317int
1318download_dir(struct sftp_conn *conn, char *src, char *dst,
1319 Attrib *dirattrib, int pflag, int printflag)
1320{
1321 char *src_canon;
1322 int ret;
1323
1324 if ((src_canon = do_realpath(conn, src)) == NULL) {
1325 error("Unable to canonicalise path \"%s\"", src);
1326 return -1;
1327 }
1328
1329 ret = download_dir_internal(conn, src_canon, dst,
1330 dirattrib, pflag, printflag, 0);
1331 xfree(src_canon);
1332 return ret;
1333}
1334
Damien Miller33804262001-02-04 23:20:18 +11001335int
Damien Miller3db5f532002-02-13 14:10:32 +11001336do_upload(struct sftp_conn *conn, char *local_path, char *remote_path,
1337 int pflag)
Damien Miller33804262001-02-04 23:20:18 +11001338{
Damien Milleracdf25b2008-02-10 22:27:24 +11001339 int local_fd;
1340 int status = SSH2_FX_OK;
Damien Miller5873dfd2002-02-13 14:04:37 +11001341 u_int handle_len, id, type;
Damien Miller8b3fdfb2007-09-17 16:12:03 +10001342 off_t offset;
Damien Miller8829d362002-02-08 22:04:05 +11001343 char *handle, *data;
Damien Miller33804262001-02-04 23:20:18 +11001344 Buffer msg;
1345 struct stat sb;
1346 Attrib a;
Damien Miller16a13332002-02-13 14:03:56 +11001347 u_int32_t startid;
1348 u_int32_t ackid;
Damien Miller5873dfd2002-02-13 14:04:37 +11001349 struct outstanding_ack {
1350 u_int id;
1351 u_int len;
Damien Miller8b3fdfb2007-09-17 16:12:03 +10001352 off_t offset;
Ben Lindstrom6328ab32002-03-22 02:54:23 +00001353 TAILQ_ENTRY(outstanding_ack) tq;
Damien Miller5873dfd2002-02-13 14:04:37 +11001354 };
1355 TAILQ_HEAD(ackhead, outstanding_ack) acks;
Damien Miller7cf17eb2004-06-15 10:28:56 +10001356 struct outstanding_ack *ack = NULL;
Damien Miller5873dfd2002-02-13 14:04:37 +11001357
1358 TAILQ_INIT(&acks);
Damien Miller33804262001-02-04 23:20:18 +11001359
1360 if ((local_fd = open(local_path, O_RDONLY, 0)) == -1) {
1361 error("Couldn't open local file \"%s\" for reading: %s",
1362 local_path, strerror(errno));
1363 return(-1);
1364 }
1365 if (fstat(local_fd, &sb) == -1) {
1366 error("Couldn't fstat local file \"%s\": %s",
1367 local_path, strerror(errno));
1368 close(local_fd);
1369 return(-1);
1370 }
Damien Miller5fa01fd2003-01-14 22:24:47 +11001371 if (!S_ISREG(sb.st_mode)) {
1372 error("%s is not a regular file", local_path);
1373 close(local_fd);
1374 return(-1);
1375 }
Damien Miller33804262001-02-04 23:20:18 +11001376 stat_to_attrib(&sb, &a);
1377
1378 a.flags &= ~SSH2_FILEXFER_ATTR_SIZE;
1379 a.flags &= ~SSH2_FILEXFER_ATTR_UIDGID;
1380 a.perm &= 0777;
1381 if (!pflag)
1382 a.flags &= ~SSH2_FILEXFER_ATTR_ACMODTIME;
1383
1384 buffer_init(&msg);
1385
1386 /* Send open request */
Damien Miller3db5f532002-02-13 14:10:32 +11001387 id = conn->msg_id++;
Damien Miller33804262001-02-04 23:20:18 +11001388 buffer_put_char(&msg, SSH2_FXP_OPEN);
1389 buffer_put_int(&msg, id);
1390 buffer_put_cstring(&msg, remote_path);
1391 buffer_put_int(&msg, SSH2_FXF_WRITE|SSH2_FXF_CREAT|SSH2_FXF_TRUNC);
1392 encode_attrib(&msg, &a);
Damien Miller65e42f82010-09-24 22:15:11 +10001393 send_msg(conn, &msg);
Ben Lindstromb1f483f2002-06-23 21:27:18 +00001394 debug3("Sent message SSH2_FXP_OPEN I:%u P:%s", id, remote_path);
Damien Miller33804262001-02-04 23:20:18 +11001395
1396 buffer_clear(&msg);
1397
Damien Miller65e42f82010-09-24 22:15:11 +10001398 handle = get_handle(conn, id, &handle_len,
Darren Tuckerc22f0902009-10-07 08:24:19 +11001399 "remote open(\"%s\")", remote_path);
Damien Miller33804262001-02-04 23:20:18 +11001400 if (handle == NULL) {
1401 close(local_fd);
1402 buffer_free(&msg);
Damien Milleracdf25b2008-02-10 22:27:24 +11001403 return -1;
Damien Miller33804262001-02-04 23:20:18 +11001404 }
1405
Damien Miller16a13332002-02-13 14:03:56 +11001406 startid = ackid = id + 1;
Damien Miller3db5f532002-02-13 14:10:32 +11001407 data = xmalloc(conn->transfer_buflen);
Damien Miller8829d362002-02-08 22:04:05 +11001408
Damien Miller33804262001-02-04 23:20:18 +11001409 /* Read from local and write to remote */
1410 offset = 0;
Damien Miller62d57f62003-01-10 21:43:24 +11001411 if (showprogress)
1412 start_progress_meter(local_path, sb.st_size, &offset);
Damien Miller62d57f62003-01-10 21:43:24 +11001413
Damien Miller9f0f5c62001-12-21 14:45:46 +11001414 for (;;) {
Damien Miller33804262001-02-04 23:20:18 +11001415 int len;
Damien Miller33804262001-02-04 23:20:18 +11001416
1417 /*
Darren Tuckerfc959702004-07-17 16:12:08 +10001418 * Can't use atomicio here because it returns 0 on EOF,
Darren Tuckercdf547a2004-05-24 10:12:19 +10001419 * thus losing the last block of the file.
Darren Tuckerfc959702004-07-17 16:12:08 +10001420 * Simulate an EOF on interrupt, allowing ACKs from the
Darren Tuckercdf547a2004-05-24 10:12:19 +10001421 * server to drain.
Damien Miller33804262001-02-04 23:20:18 +11001422 */
Damien Milleracdf25b2008-02-10 22:27:24 +11001423 if (interrupted || status != SSH2_FX_OK)
Darren Tuckercdf547a2004-05-24 10:12:19 +10001424 len = 0;
1425 else do
Damien Miller3db5f532002-02-13 14:10:32 +11001426 len = read(local_fd, data, conn->transfer_buflen);
Damien Millerd8968ad2008-07-04 23:10:49 +10001427 while ((len == -1) &&
1428 (errno == EINTR || errno == EAGAIN || errno == EWOULDBLOCK));
Damien Miller33804262001-02-04 23:20:18 +11001429
1430 if (len == -1)
1431 fatal("Couldn't read from \"%s\": %s", local_path,
1432 strerror(errno));
Damien Miller16a13332002-02-13 14:03:56 +11001433
1434 if (len != 0) {
Damien Miller5873dfd2002-02-13 14:04:37 +11001435 ack = xmalloc(sizeof(*ack));
1436 ack->id = ++id;
1437 ack->offset = offset;
1438 ack->len = len;
1439 TAILQ_INSERT_TAIL(&acks, ack, tq);
1440
Damien Miller16a13332002-02-13 14:03:56 +11001441 buffer_clear(&msg);
1442 buffer_put_char(&msg, SSH2_FXP_WRITE);
Damien Miller5873dfd2002-02-13 14:04:37 +11001443 buffer_put_int(&msg, ack->id);
Damien Miller16a13332002-02-13 14:03:56 +11001444 buffer_put_string(&msg, handle, handle_len);
1445 buffer_put_int64(&msg, offset);
1446 buffer_put_string(&msg, data, len);
Damien Miller65e42f82010-09-24 22:15:11 +10001447 send_msg(conn, &msg);
Ben Lindstromb1f483f2002-06-23 21:27:18 +00001448 debug3("Sent message SSH2_FXP_WRITE I:%u O:%llu S:%u",
Ben Lindstrom93576d92002-12-23 02:06:19 +00001449 id, (unsigned long long)offset, len);
Damien Miller5873dfd2002-02-13 14:04:37 +11001450 } else if (TAILQ_FIRST(&acks) == NULL)
Damien Miller33804262001-02-04 23:20:18 +11001451 break;
1452
Damien Miller5873dfd2002-02-13 14:04:37 +11001453 if (ack == NULL)
1454 fatal("Unexpected ACK %u", id);
1455
Ben Lindstrom6328ab32002-03-22 02:54:23 +00001456 if (id == startid || len == 0 ||
Damien Miller3db5f532002-02-13 14:10:32 +11001457 id - ackid >= conn->num_requests) {
Ben Lindstrom5a6abda2002-06-09 19:41:48 +00001458 u_int r_id;
Ben Lindstrom06e95152002-04-06 04:16:45 +00001459
Damien Miller5873dfd2002-02-13 14:04:37 +11001460 buffer_clear(&msg);
Damien Miller65e42f82010-09-24 22:15:11 +10001461 get_msg(conn, &msg);
Damien Miller5873dfd2002-02-13 14:04:37 +11001462 type = buffer_get_char(&msg);
Ben Lindstrom06e95152002-04-06 04:16:45 +00001463 r_id = buffer_get_int(&msg);
Damien Miller5873dfd2002-02-13 14:04:37 +11001464
1465 if (type != SSH2_FXP_STATUS)
1466 fatal("Expected SSH2_FXP_STATUS(%d) packet, "
1467 "got %d", SSH2_FXP_STATUS, type);
1468
1469 status = buffer_get_int(&msg);
1470 debug3("SSH2_FXP_STATUS %d", status);
1471
1472 /* Find the request in our queue */
Darren Tucker47eede72005-03-14 23:08:12 +11001473 for (ack = TAILQ_FIRST(&acks);
Ben Lindstrom06e95152002-04-06 04:16:45 +00001474 ack != NULL && ack->id != r_id;
Damien Miller5873dfd2002-02-13 14:04:37 +11001475 ack = TAILQ_NEXT(ack, tq))
1476 ;
1477 if (ack == NULL)
Ben Lindstromb1f483f2002-06-23 21:27:18 +00001478 fatal("Can't find request for ID %u", r_id);
Damien Miller5873dfd2002-02-13 14:04:37 +11001479 TAILQ_REMOVE(&acks, ack, tq);
Damien Miller8b3fdfb2007-09-17 16:12:03 +10001480 debug3("In write loop, ack for %u %u bytes at %lld",
1481 ack->id, ack->len, (long long)ack->offset);
Damien Miller16a13332002-02-13 14:03:56 +11001482 ++ackid;
Ben Lindstromeec16fc2002-07-04 00:06:15 +00001483 xfree(ack);
Damien Miller33804262001-02-04 23:20:18 +11001484 }
Damien Miller33804262001-02-04 23:20:18 +11001485 offset += len;
Damien Miller8b3fdfb2007-09-17 16:12:03 +10001486 if (offset < 0)
1487 fatal("%s: offset < 0", __func__);
Damien Miller33804262001-02-04 23:20:18 +11001488 }
Damien Milleracdf25b2008-02-10 22:27:24 +11001489 buffer_free(&msg);
1490
Damien Miller62d57f62003-01-10 21:43:24 +11001491 if (showprogress)
1492 stop_progress_meter();
Damien Miller8829d362002-02-08 22:04:05 +11001493 xfree(data);
Damien Miller33804262001-02-04 23:20:18 +11001494
Damien Milleracdf25b2008-02-10 22:27:24 +11001495 if (status != SSH2_FX_OK) {
1496 error("Couldn't write to remote file \"%s\": %s",
1497 remote_path, fx2txt(status));
1498 status = -1;
1499 }
1500
Damien Miller33804262001-02-04 23:20:18 +11001501 if (close(local_fd) == -1) {
1502 error("Couldn't close local file \"%s\": %s", local_path,
1503 strerror(errno));
Damien Millerd7686fd2001-02-10 00:40:03 +11001504 status = -1;
Damien Miller33804262001-02-04 23:20:18 +11001505 }
1506
Ben Lindstrom9d4f2c82001-02-15 03:22:45 +00001507 /* Override umask and utimes if asked */
1508 if (pflag)
Damien Miller3db5f532002-02-13 14:10:32 +11001509 do_fsetstat(conn, handle, handle_len, &a);
Ben Lindstrom9d4f2c82001-02-15 03:22:45 +00001510
Damien Milleracdf25b2008-02-10 22:27:24 +11001511 if (do_close(conn, handle, handle_len) != SSH2_FX_OK)
1512 status = -1;
Damien Millerd7686fd2001-02-10 00:40:03 +11001513 xfree(handle);
Damien Milleracdf25b2008-02-10 22:27:24 +11001514
1515 return status;
Damien Miller33804262001-02-04 23:20:18 +11001516}
Darren Tucker1b0dd172009-10-07 08:37:48 +11001517
1518static int
1519upload_dir_internal(struct sftp_conn *conn, char *src, char *dst,
1520 int pflag, int printflag, int depth)
1521{
1522 int ret = 0, status;
1523 DIR *dirp;
1524 struct dirent *dp;
1525 char *filename, *new_src, *new_dst;
1526 struct stat sb;
1527 Attrib a;
1528
1529 if (depth >= MAX_DIR_DEPTH) {
1530 error("Maximum directory depth exceeded: %d levels", depth);
1531 return -1;
1532 }
1533
1534 if (stat(src, &sb) == -1) {
1535 error("Couldn't stat directory \"%s\": %s",
1536 src, strerror(errno));
1537 return -1;
1538 }
1539 if (!S_ISDIR(sb.st_mode)) {
1540 error("\"%s\" is not a directory", src);
1541 return -1;
1542 }
1543 if (printflag)
1544 printf("Entering %s\n", src);
1545
1546 attrib_clear(&a);
1547 stat_to_attrib(&sb, &a);
1548 a.flags &= ~SSH2_FILEXFER_ATTR_SIZE;
1549 a.flags &= ~SSH2_FILEXFER_ATTR_UIDGID;
1550 a.perm &= 01777;
1551 if (!pflag)
1552 a.flags &= ~SSH2_FILEXFER_ATTR_ACMODTIME;
1553
1554 status = do_mkdir(conn, dst, &a, 0);
1555 /*
1556 * we lack a portable status for errno EEXIST,
1557 * so if we get a SSH2_FX_FAILURE back we must check
1558 * if it was created successfully.
1559 */
1560 if (status != SSH2_FX_OK) {
1561 if (status != SSH2_FX_FAILURE)
1562 return -1;
1563 if (do_stat(conn, dst, 0) == NULL)
1564 return -1;
1565 }
1566
1567 if ((dirp = opendir(src)) == NULL) {
1568 error("Failed to open dir \"%s\": %s", src, strerror(errno));
1569 return -1;
1570 }
1571
1572 while (((dp = readdir(dirp)) != NULL) && !interrupted) {
1573 if (dp->d_ino == 0)
1574 continue;
1575 filename = dp->d_name;
1576 new_dst = path_append(dst, filename);
1577 new_src = path_append(src, filename);
1578
Darren Tucker438b4732009-10-11 21:52:10 +11001579 if (lstat(new_src, &sb) == -1) {
1580 logit("%s: lstat failed: %s", filename,
1581 strerror(errno));
1582 ret = -1;
1583 } else if (S_ISDIR(sb.st_mode)) {
Darren Tucker1b0dd172009-10-07 08:37:48 +11001584 if (strcmp(filename, ".") == 0 ||
1585 strcmp(filename, "..") == 0)
1586 continue;
1587
1588 if (upload_dir_internal(conn, new_src, new_dst,
Damien Millerc4bb91c2010-08-03 16:04:22 +10001589 pflag, printflag, depth + 1) == -1)
Darren Tucker1b0dd172009-10-07 08:37:48 +11001590 ret = -1;
Darren Tucker438b4732009-10-11 21:52:10 +11001591 } else if (S_ISREG(sb.st_mode)) {
Darren Tucker1b0dd172009-10-07 08:37:48 +11001592 if (do_upload(conn, new_src, new_dst, pflag) == -1) {
1593 error("Uploading of file %s to %s failed!",
1594 new_src, new_dst);
1595 ret = -1;
1596 }
1597 } else
1598 logit("%s: not a regular file\n", filename);
1599 xfree(new_dst);
1600 xfree(new_src);
1601 }
1602
1603 do_setstat(conn, dst, &a);
1604
1605 (void) closedir(dirp);
1606 return ret;
1607}
1608
1609int
1610upload_dir(struct sftp_conn *conn, char *src, char *dst, int printflag,
1611 int pflag)
1612{
1613 char *dst_canon;
1614 int ret;
1615
1616 if ((dst_canon = do_realpath(conn, dst)) == NULL) {
1617 error("Unable to canonicalise path \"%s\"", dst);
1618 return -1;
1619 }
1620
1621 ret = upload_dir_internal(conn, src, dst_canon, pflag, printflag, 0);
1622 xfree(dst_canon);
1623 return ret;
1624}
1625
1626char *
1627path_append(char *p1, char *p2)
1628{
1629 char *ret;
1630 size_t len = strlen(p1) + strlen(p2) + 2;
1631
1632 ret = xmalloc(len);
1633 strlcpy(ret, p1, len);
1634 if (p1[0] != '\0' && p1[strlen(p1) - 1] != '/')
1635 strlcat(ret, "/", len);
1636 strlcat(ret, p2, len);
1637
1638 return(ret);
1639}
1640