blob: caa384b4e702cf593d7872f42193530e918ebc6a [file] [log] [blame]
Darren Tuckeraf1f9092010-12-05 09:02:47 +11001/* $OpenBSD: sftp-client.c,v 1.94 2010/12/04 00:18:01 djm 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
465 buffer_clear(&msg);
466
Damien Miller65e42f82010-09-24 22:15:11 +1000467 handle = get_handle(conn, id, &handle_len,
Darren Tuckerc22f0902009-10-07 08:24:19 +1100468 "remote readdir(\"%s\")", path);
Damien Miller33804262001-02-04 23:20:18 +1100469 if (handle == NULL)
Damien Miller65e42f82010-09-24 22:15:11 +1000470 return -1;
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);
Ben Lindstrom10ac33f2001-02-10 21:53:40 +0000513 return(status);
Damien Miller33804262001-02-04 23:20:18 +1100514 }
515 } else if (type != SSH2_FXP_NAME)
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000516 fatal("Expected SSH2_FXP_NAME(%u) packet, got %u",
Damien Miller33804262001-02-04 23:20:18 +1100517 SSH2_FXP_NAME, type);
518
519 count = buffer_get_int(&msg);
Damien Millerd7686fd2001-02-10 00:40:03 +1100520 if (count == 0)
521 break;
522 debug3("Received %d SSH2_FXP_NAME responses", count);
Damien Miller9f0f5c62001-12-21 14:45:46 +1100523 for (i = 0; i < count; i++) {
Damien Miller33804262001-02-04 23:20:18 +1100524 char *filename, *longname;
525 Attrib *a;
526
527 filename = buffer_get_string(&msg, NULL);
528 longname = buffer_get_string(&msg, NULL);
529 a = decode_attrib(&msg);
530
Damien Miller4870afd2001-03-14 10:27:09 +1100531 if (printflag)
532 printf("%s\n", longname);
533
Darren Tucker1b0dd172009-10-07 08:37:48 +1100534 /*
535 * Directory entries should never contain '/'
536 * These can be used to attack recursive ops
537 * (e.g. send '../../../../etc/passwd')
538 */
539 if (strchr(filename, '/') != NULL) {
540 error("Server sent suspect path \"%s\" "
541 "during readdir of \"%s\"", filename, path);
542 goto next;
543 }
544
Damien Miller4870afd2001-03-14 10:27:09 +1100545 if (dir) {
Damien Miller36812092006-03-26 14:22:47 +1100546 *dir = xrealloc(*dir, ents + 2, sizeof(**dir));
Damien Miller4870afd2001-03-14 10:27:09 +1100547 (*dir)[ents] = xmalloc(sizeof(***dir));
548 (*dir)[ents]->filename = xstrdup(filename);
549 (*dir)[ents]->longname = xstrdup(longname);
550 memcpy(&(*dir)[ents]->a, a, sizeof(*a));
551 (*dir)[++ents] = NULL;
552 }
Darren Tucker1b0dd172009-10-07 08:37:48 +1100553 next:
Damien Miller33804262001-02-04 23:20:18 +1100554 xfree(filename);
555 xfree(longname);
556 }
557 }
558
559 buffer_free(&msg);
Damien Miller3db5f532002-02-13 14:10:32 +1100560 do_close(conn, handle, handle_len);
Damien Miller33804262001-02-04 23:20:18 +1100561 xfree(handle);
562
Darren Tuckercdf547a2004-05-24 10:12:19 +1000563 /* Don't return partial matches on interrupt */
564 if (interrupted && dir != NULL && *dir != NULL) {
565 free_sftp_dirents(*dir);
566 *dir = xmalloc(sizeof(**dir));
567 **dir = NULL;
568 }
569
Damien Miller65e42f82010-09-24 22:15:11 +1000570 return 0;
Damien Miller33804262001-02-04 23:20:18 +1100571}
572
573int
Damien Miller3db5f532002-02-13 14:10:32 +1100574do_readdir(struct sftp_conn *conn, char *path, SFTP_DIRENT ***dir)
Damien Miller4870afd2001-03-14 10:27:09 +1100575{
Damien Miller3db5f532002-02-13 14:10:32 +1100576 return(do_lsreaddir(conn, path, 0, dir));
Damien Miller4870afd2001-03-14 10:27:09 +1100577}
578
579void free_sftp_dirents(SFTP_DIRENT **s)
580{
581 int i;
Damien Miller9f0f5c62001-12-21 14:45:46 +1100582
583 for (i = 0; s[i]; i++) {
Damien Miller4870afd2001-03-14 10:27:09 +1100584 xfree(s[i]->filename);
585 xfree(s[i]->longname);
586 xfree(s[i]);
587 }
588 xfree(s);
589}
590
591int
Damien Miller3db5f532002-02-13 14:10:32 +1100592do_rm(struct sftp_conn *conn, char *path)
Damien Miller33804262001-02-04 23:20:18 +1100593{
594 u_int status, id;
595
596 debug2("Sending SSH2_FXP_REMOVE \"%s\"", path);
597
Damien Miller3db5f532002-02-13 14:10:32 +1100598 id = conn->msg_id++;
Damien Miller65e42f82010-09-24 22:15:11 +1000599 send_string_request(conn, id, SSH2_FXP_REMOVE, path, strlen(path));
600 status = get_status(conn, id);
Damien Miller33804262001-02-04 23:20:18 +1100601 if (status != SSH2_FX_OK)
602 error("Couldn't delete file: %s", fx2txt(status));
603 return(status);
604}
605
606int
Darren Tucker1b0dd172009-10-07 08:37:48 +1100607do_mkdir(struct sftp_conn *conn, char *path, Attrib *a, int printflag)
Damien Miller33804262001-02-04 23:20:18 +1100608{
609 u_int status, id;
610
Damien Miller3db5f532002-02-13 14:10:32 +1100611 id = conn->msg_id++;
Damien Miller65e42f82010-09-24 22:15:11 +1000612 send_string_attrs_request(conn, id, SSH2_FXP_MKDIR, path,
Damien Miller33804262001-02-04 23:20:18 +1100613 strlen(path), a);
614
Damien Miller65e42f82010-09-24 22:15:11 +1000615 status = get_status(conn, id);
Darren Tucker1b0dd172009-10-07 08:37:48 +1100616 if (status != SSH2_FX_OK && printflag)
Damien Miller33804262001-02-04 23:20:18 +1100617 error("Couldn't create directory: %s", fx2txt(status));
618
619 return(status);
620}
621
622int
Damien Miller3db5f532002-02-13 14:10:32 +1100623do_rmdir(struct sftp_conn *conn, char *path)
Damien Miller33804262001-02-04 23:20:18 +1100624{
625 u_int status, id;
626
Damien Miller3db5f532002-02-13 14:10:32 +1100627 id = conn->msg_id++;
Damien Miller65e42f82010-09-24 22:15:11 +1000628 send_string_request(conn, id, SSH2_FXP_RMDIR, path,
Damien Miller3db5f532002-02-13 14:10:32 +1100629 strlen(path));
Damien Miller33804262001-02-04 23:20:18 +1100630
Damien Miller65e42f82010-09-24 22:15:11 +1000631 status = get_status(conn, id);
Damien Miller33804262001-02-04 23:20:18 +1100632 if (status != SSH2_FX_OK)
633 error("Couldn't remove directory: %s", fx2txt(status));
634
635 return(status);
636}
637
638Attrib *
Damien Miller3db5f532002-02-13 14:10:32 +1100639do_stat(struct sftp_conn *conn, char *path, int quiet)
Damien Miller33804262001-02-04 23:20:18 +1100640{
641 u_int id;
642
Damien Miller3db5f532002-02-13 14:10:32 +1100643 id = conn->msg_id++;
644
Damien Miller65e42f82010-09-24 22:15:11 +1000645 send_string_request(conn, id,
Ben Lindstrom6328ab32002-03-22 02:54:23 +0000646 conn->version == 0 ? SSH2_FXP_STAT_VERSION_0 : SSH2_FXP_STAT,
Damien Miller3db5f532002-02-13 14:10:32 +1100647 path, strlen(path));
648
Damien Miller65e42f82010-09-24 22:15:11 +1000649 return(get_decode_stat(conn, id, quiet));
Damien Miller33804262001-02-04 23:20:18 +1100650}
651
652Attrib *
Damien Miller3db5f532002-02-13 14:10:32 +1100653do_lstat(struct sftp_conn *conn, char *path, int quiet)
Damien Miller33804262001-02-04 23:20:18 +1100654{
655 u_int id;
656
Damien Miller3db5f532002-02-13 14:10:32 +1100657 if (conn->version == 0) {
658 if (quiet)
659 debug("Server version does not support lstat operation");
660 else
Damien Miller996acd22003-04-09 20:59:48 +1000661 logit("Server version does not support lstat operation");
Ben Lindstromf26ff5b2002-04-02 21:00:31 +0000662 return(do_stat(conn, path, quiet));
Damien Miller3db5f532002-02-13 14:10:32 +1100663 }
664
665 id = conn->msg_id++;
Damien Miller65e42f82010-09-24 22:15:11 +1000666 send_string_request(conn, id, SSH2_FXP_LSTAT, path,
Damien Miller3db5f532002-02-13 14:10:32 +1100667 strlen(path));
668
Damien Miller65e42f82010-09-24 22:15:11 +1000669 return(get_decode_stat(conn, id, quiet));
Damien Miller33804262001-02-04 23:20:18 +1100670}
671
Damien Millercfe23d32008-02-10 22:20:44 +1100672#ifdef notyet
Damien Miller33804262001-02-04 23:20:18 +1100673Attrib *
Damien Miller3db5f532002-02-13 14:10:32 +1100674do_fstat(struct sftp_conn *conn, char *handle, u_int handle_len, int quiet)
Damien Miller33804262001-02-04 23:20:18 +1100675{
676 u_int id;
677
Damien Miller3db5f532002-02-13 14:10:32 +1100678 id = conn->msg_id++;
Damien Miller65e42f82010-09-24 22:15:11 +1000679 send_string_request(conn, id, SSH2_FXP_FSTAT, handle,
Damien Miller3db5f532002-02-13 14:10:32 +1100680 handle_len);
681
Damien Miller65e42f82010-09-24 22:15:11 +1000682 return(get_decode_stat(conn, id, quiet));
Damien Miller33804262001-02-04 23:20:18 +1100683}
Damien Millercfe23d32008-02-10 22:20:44 +1100684#endif
Damien Miller33804262001-02-04 23:20:18 +1100685
686int
Damien Miller3db5f532002-02-13 14:10:32 +1100687do_setstat(struct sftp_conn *conn, char *path, Attrib *a)
Damien Miller33804262001-02-04 23:20:18 +1100688{
689 u_int status, id;
690
Damien Miller3db5f532002-02-13 14:10:32 +1100691 id = conn->msg_id++;
Damien Miller65e42f82010-09-24 22:15:11 +1000692 send_string_attrs_request(conn, id, SSH2_FXP_SETSTAT, path,
Damien Miller33804262001-02-04 23:20:18 +1100693 strlen(path), a);
694
Damien Miller65e42f82010-09-24 22:15:11 +1000695 status = get_status(conn, id);
Damien Miller33804262001-02-04 23:20:18 +1100696 if (status != SSH2_FX_OK)
697 error("Couldn't setstat on \"%s\": %s", path,
698 fx2txt(status));
699
700 return(status);
701}
702
703int
Damien Miller3db5f532002-02-13 14:10:32 +1100704do_fsetstat(struct sftp_conn *conn, char *handle, u_int handle_len,
Damien Miller33804262001-02-04 23:20:18 +1100705 Attrib *a)
706{
707 u_int status, id;
708
Damien Miller3db5f532002-02-13 14:10:32 +1100709 id = conn->msg_id++;
Damien Miller65e42f82010-09-24 22:15:11 +1000710 send_string_attrs_request(conn, id, SSH2_FXP_FSETSTAT, handle,
Damien Miller33804262001-02-04 23:20:18 +1100711 handle_len, a);
712
Damien Miller65e42f82010-09-24 22:15:11 +1000713 status = get_status(conn, id);
Damien Miller33804262001-02-04 23:20:18 +1100714 if (status != SSH2_FX_OK)
715 error("Couldn't fsetstat: %s", fx2txt(status));
716
717 return(status);
718}
719
720char *
Damien Miller3db5f532002-02-13 14:10:32 +1100721do_realpath(struct sftp_conn *conn, char *path)
Damien Miller33804262001-02-04 23:20:18 +1100722{
723 Buffer msg;
724 u_int type, expected_id, count, id;
725 char *filename, *longname;
726 Attrib *a;
727
Damien Miller3db5f532002-02-13 14:10:32 +1100728 expected_id = id = conn->msg_id++;
Damien Miller65e42f82010-09-24 22:15:11 +1000729 send_string_request(conn, id, SSH2_FXP_REALPATH, path,
Damien Miller3db5f532002-02-13 14:10:32 +1100730 strlen(path));
Damien Miller33804262001-02-04 23:20:18 +1100731
732 buffer_init(&msg);
733
Damien Miller65e42f82010-09-24 22:15:11 +1000734 get_msg(conn, &msg);
Damien Miller33804262001-02-04 23:20:18 +1100735 type = buffer_get_char(&msg);
736 id = buffer_get_int(&msg);
737
738 if (id != expected_id)
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000739 fatal("ID mismatch (%u != %u)", id, expected_id);
Damien Miller33804262001-02-04 23:20:18 +1100740
741 if (type == SSH2_FXP_STATUS) {
742 u_int status = buffer_get_int(&msg);
743
744 error("Couldn't canonicalise: %s", fx2txt(status));
Damien Miller49566312010-06-26 09:38:23 +1000745 buffer_free(&msg);
746 return NULL;
Damien Miller33804262001-02-04 23:20:18 +1100747 } else if (type != SSH2_FXP_NAME)
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000748 fatal("Expected SSH2_FXP_NAME(%u) packet, got %u",
Damien Miller33804262001-02-04 23:20:18 +1100749 SSH2_FXP_NAME, type);
750
751 count = buffer_get_int(&msg);
752 if (count != 1)
753 fatal("Got multiple names (%d) from SSH_FXP_REALPATH", count);
754
755 filename = buffer_get_string(&msg, NULL);
756 longname = buffer_get_string(&msg, NULL);
757 a = decode_attrib(&msg);
758
759 debug3("SSH_FXP_REALPATH %s -> %s", path, filename);
760
761 xfree(longname);
762
763 buffer_free(&msg);
764
765 return(filename);
766}
767
768int
Damien Miller3db5f532002-02-13 14:10:32 +1100769do_rename(struct sftp_conn *conn, char *oldpath, char *newpath)
Damien Miller33804262001-02-04 23:20:18 +1100770{
771 Buffer msg;
772 u_int status, id;
773
774 buffer_init(&msg);
775
776 /* Send rename request */
Damien Miller3db5f532002-02-13 14:10:32 +1100777 id = conn->msg_id++;
Damien Miller7a3e1d02008-03-27 10:59:57 +1100778 if ((conn->exts & SFTP_EXT_POSIX_RENAME)) {
779 buffer_put_char(&msg, SSH2_FXP_EXTENDED);
780 buffer_put_int(&msg, id);
781 buffer_put_cstring(&msg, "posix-rename@openssh.com");
782 } else {
783 buffer_put_char(&msg, SSH2_FXP_RENAME);
784 buffer_put_int(&msg, id);
785 }
Damien Miller33804262001-02-04 23:20:18 +1100786 buffer_put_cstring(&msg, oldpath);
787 buffer_put_cstring(&msg, newpath);
Damien Miller65e42f82010-09-24 22:15:11 +1000788 send_msg(conn, &msg);
Damien Miller7a3e1d02008-03-27 10:59:57 +1100789 debug3("Sent message %s \"%s\" -> \"%s\"",
790 (conn->exts & SFTP_EXT_POSIX_RENAME) ? "posix-rename@openssh.com" :
791 "SSH2_FXP_RENAME", oldpath, newpath);
Damien Miller33804262001-02-04 23:20:18 +1100792 buffer_free(&msg);
793
Damien Miller65e42f82010-09-24 22:15:11 +1000794 status = get_status(conn, id);
Damien Miller33804262001-02-04 23:20:18 +1100795 if (status != SSH2_FX_OK)
Damien Miller3db5f532002-02-13 14:10:32 +1100796 error("Couldn't rename file \"%s\" to \"%s\": %s", oldpath,
797 newpath, fx2txt(status));
Damien Miller33804262001-02-04 23:20:18 +1100798
799 return(status);
800}
801
802int
Darren Tuckeraf1f9092010-12-05 09:02:47 +1100803do_hardlink(struct sftp_conn *conn, char *oldpath, char *newpath)
804{
805 Buffer msg;
806 u_int status, id;
807
808 buffer_init(&msg);
809
810 /* Send link request */
811 id = conn->msg_id++;
812 if ((conn->exts & SFTP_EXT_HARDLINK) == 0) {
813 error("Server does not support hardlink@openssh.com extension");
814 return -1;
815 }
816
817 buffer_put_char(&msg, SSH2_FXP_EXTENDED);
818 buffer_put_int(&msg, id);
819 buffer_put_cstring(&msg, "hardlink@openssh.com");
820 buffer_put_cstring(&msg, oldpath);
821 buffer_put_cstring(&msg, newpath);
822 send_msg(conn, &msg);
823 debug3("Sent message hardlink@openssh.com \"%s\" -> \"%s\"",
824 oldpath, newpath);
825 buffer_free(&msg);
826
827 status = get_status(conn, id);
828 if (status != SSH2_FX_OK)
829 error("Couldn't link file \"%s\" to \"%s\": %s", oldpath,
830 newpath, fx2txt(status));
831
832 return(status);
833}
834
835int
Damien Miller3db5f532002-02-13 14:10:32 +1100836do_symlink(struct sftp_conn *conn, char *oldpath, char *newpath)
Damien Miller058316f2001-03-08 10:08:49 +1100837{
838 Buffer msg;
839 u_int status, id;
840
Damien Miller3db5f532002-02-13 14:10:32 +1100841 if (conn->version < 3) {
842 error("This server does not support the symlink operation");
843 return(SSH2_FX_OP_UNSUPPORTED);
844 }
845
Damien Miller058316f2001-03-08 10:08:49 +1100846 buffer_init(&msg);
847
Darren Tuckerdca6a4d2004-04-19 22:10:52 +1000848 /* Send symlink request */
Damien Miller3db5f532002-02-13 14:10:32 +1100849 id = conn->msg_id++;
Damien Miller058316f2001-03-08 10:08:49 +1100850 buffer_put_char(&msg, SSH2_FXP_SYMLINK);
851 buffer_put_int(&msg, id);
852 buffer_put_cstring(&msg, oldpath);
853 buffer_put_cstring(&msg, newpath);
Damien Miller65e42f82010-09-24 22:15:11 +1000854 send_msg(conn, &msg);
Damien Miller058316f2001-03-08 10:08:49 +1100855 debug3("Sent message SSH2_FXP_SYMLINK \"%s\" -> \"%s\"", oldpath,
856 newpath);
857 buffer_free(&msg);
858
Damien Miller65e42f82010-09-24 22:15:11 +1000859 status = get_status(conn, id);
Damien Miller058316f2001-03-08 10:08:49 +1100860 if (status != SSH2_FX_OK)
Ben Lindstrom8e879cf2002-11-09 15:48:49 +0000861 error("Couldn't symlink file \"%s\" to \"%s\": %s", oldpath,
Damien Miller3db5f532002-02-13 14:10:32 +1100862 newpath, fx2txt(status));
Damien Miller058316f2001-03-08 10:08:49 +1100863
864 return(status);
865}
866
Damien Millercfe23d32008-02-10 22:20:44 +1100867#ifdef notyet
Damien Miller058316f2001-03-08 10:08:49 +1100868char *
Damien Miller3db5f532002-02-13 14:10:32 +1100869do_readlink(struct sftp_conn *conn, char *path)
Damien Miller058316f2001-03-08 10:08:49 +1100870{
871 Buffer msg;
872 u_int type, expected_id, count, id;
873 char *filename, *longname;
874 Attrib *a;
875
Damien Miller3db5f532002-02-13 14:10:32 +1100876 expected_id = id = conn->msg_id++;
Damien Miller65e42f82010-09-24 22:15:11 +1000877 send_string_request(conn, id, SSH2_FXP_READLINK, path, strlen(path));
Damien Miller058316f2001-03-08 10:08:49 +1100878
879 buffer_init(&msg);
880
Damien Miller65e42f82010-09-24 22:15:11 +1000881 get_msg(conn, &msg);
Damien Miller058316f2001-03-08 10:08:49 +1100882 type = buffer_get_char(&msg);
883 id = buffer_get_int(&msg);
884
885 if (id != expected_id)
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000886 fatal("ID mismatch (%u != %u)", id, expected_id);
Damien Miller058316f2001-03-08 10:08:49 +1100887
888 if (type == SSH2_FXP_STATUS) {
889 u_int status = buffer_get_int(&msg);
890
891 error("Couldn't readlink: %s", fx2txt(status));
892 return(NULL);
893 } else if (type != SSH2_FXP_NAME)
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000894 fatal("Expected SSH2_FXP_NAME(%u) packet, got %u",
Damien Miller058316f2001-03-08 10:08:49 +1100895 SSH2_FXP_NAME, type);
896
897 count = buffer_get_int(&msg);
898 if (count != 1)
899 fatal("Got multiple names (%d) from SSH_FXP_READLINK", count);
900
901 filename = buffer_get_string(&msg, NULL);
902 longname = buffer_get_string(&msg, NULL);
903 a = decode_attrib(&msg);
904
905 debug3("SSH_FXP_READLINK %s -> %s", path, filename);
906
907 xfree(longname);
908
909 buffer_free(&msg);
910
911 return(filename);
912}
Damien Millercfe23d32008-02-10 22:20:44 +1100913#endif
Damien Miller058316f2001-03-08 10:08:49 +1100914
Damien Millerd671e5a2008-05-19 14:53:33 +1000915int
Darren Tucker7b598892008-06-09 22:49:36 +1000916do_statvfs(struct sftp_conn *conn, const char *path, struct sftp_statvfs *st,
Damien Millerd671e5a2008-05-19 14:53:33 +1000917 int quiet)
918{
919 Buffer msg;
920 u_int id;
921
922 if ((conn->exts & SFTP_EXT_STATVFS) == 0) {
923 error("Server does not support statvfs@openssh.com extension");
924 return -1;
925 }
926
927 id = conn->msg_id++;
928
929 buffer_init(&msg);
930 buffer_clear(&msg);
931 buffer_put_char(&msg, SSH2_FXP_EXTENDED);
932 buffer_put_int(&msg, id);
933 buffer_put_cstring(&msg, "statvfs@openssh.com");
934 buffer_put_cstring(&msg, path);
Damien Miller65e42f82010-09-24 22:15:11 +1000935 send_msg(conn, &msg);
Damien Millerd671e5a2008-05-19 14:53:33 +1000936 buffer_free(&msg);
937
Damien Miller65e42f82010-09-24 22:15:11 +1000938 return get_decode_statvfs(conn, st, id, quiet);
Damien Millerd671e5a2008-05-19 14:53:33 +1000939}
940
941#ifdef notyet
942int
943do_fstatvfs(struct sftp_conn *conn, const char *handle, u_int handle_len,
Darren Tucker7b598892008-06-09 22:49:36 +1000944 struct sftp_statvfs *st, int quiet)
Damien Millerd671e5a2008-05-19 14:53:33 +1000945{
946 Buffer msg;
947 u_int id;
948
949 if ((conn->exts & SFTP_EXT_FSTATVFS) == 0) {
950 error("Server does not support fstatvfs@openssh.com extension");
951 return -1;
952 }
953
954 id = conn->msg_id++;
955
956 buffer_init(&msg);
957 buffer_clear(&msg);
958 buffer_put_char(&msg, SSH2_FXP_EXTENDED);
959 buffer_put_int(&msg, id);
960 buffer_put_cstring(&msg, "fstatvfs@openssh.com");
961 buffer_put_string(&msg, handle, handle_len);
Damien Miller65e42f82010-09-24 22:15:11 +1000962 send_msg(conn, &msg);
Damien Millerd671e5a2008-05-19 14:53:33 +1000963 buffer_free(&msg);
964
Damien Miller65e42f82010-09-24 22:15:11 +1000965 return get_decode_statvfs(conn, st, id, quiet);
Damien Millerd671e5a2008-05-19 14:53:33 +1000966}
967#endif
968
Damien Miller16a13332002-02-13 14:03:56 +1100969static void
Damien Miller65e42f82010-09-24 22:15:11 +1000970send_read_request(struct sftp_conn *conn, u_int id, u_int64_t offset,
971 u_int len, char *handle, u_int handle_len)
Damien Miller16a13332002-02-13 14:03:56 +1100972{
973 Buffer msg;
Ben Lindstrom6328ab32002-03-22 02:54:23 +0000974
Damien Miller16a13332002-02-13 14:03:56 +1100975 buffer_init(&msg);
976 buffer_clear(&msg);
977 buffer_put_char(&msg, SSH2_FXP_READ);
978 buffer_put_int(&msg, id);
979 buffer_put_string(&msg, handle, handle_len);
980 buffer_put_int64(&msg, offset);
981 buffer_put_int(&msg, len);
Damien Miller65e42f82010-09-24 22:15:11 +1000982 send_msg(conn, &msg);
Damien Miller16a13332002-02-13 14:03:56 +1100983 buffer_free(&msg);
Ben Lindstrom6328ab32002-03-22 02:54:23 +0000984}
Damien Miller16a13332002-02-13 14:03:56 +1100985
Damien Miller058316f2001-03-08 10:08:49 +1100986int
Damien Miller3db5f532002-02-13 14:10:32 +1100987do_download(struct sftp_conn *conn, char *remote_path, char *local_path,
Darren Tucker1b0dd172009-10-07 08:37:48 +1100988 Attrib *a, int pflag)
Damien Miller33804262001-02-04 23:20:18 +1100989{
Darren Tucker1b0dd172009-10-07 08:37:48 +1100990 Attrib junk;
Damien Miller16a13332002-02-13 14:03:56 +1100991 Buffer msg;
992 char *handle;
Darren Tucker40858532005-08-02 17:07:07 +1000993 int local_fd, status = 0, write_error;
Damien Miller16a13332002-02-13 14:03:56 +1100994 int read_error, write_errno;
995 u_int64_t offset, size;
Damien Millereccb9de2005-06-17 12:59:34 +1000996 u_int handle_len, mode, type, id, buflen, num_req, max_req;
Damien Miller62d57f62003-01-10 21:43:24 +1100997 off_t progress_counter;
Damien Miller16a13332002-02-13 14:03:56 +1100998 struct request {
999 u_int id;
1000 u_int len;
1001 u_int64_t offset;
Ben Lindstrom6328ab32002-03-22 02:54:23 +00001002 TAILQ_ENTRY(request) tq;
Damien Miller16a13332002-02-13 14:03:56 +11001003 };
1004 TAILQ_HEAD(reqhead, request) requests;
1005 struct request *req;
1006
1007 TAILQ_INIT(&requests);
Damien Miller33804262001-02-04 23:20:18 +11001008
Darren Tucker1b0dd172009-10-07 08:37:48 +11001009 if (a == NULL && (a = do_stat(conn, remote_path, 0)) == NULL)
1010 return -1;
Damien Miller33804262001-02-04 23:20:18 +11001011
Damien Miller9e720282008-06-29 22:46:35 +10001012 /* Do not preserve set[ug]id here, as we do not preserve ownership */
Damien Miller33804262001-02-04 23:20:18 +11001013 if (a->flags & SSH2_FILEXFER_ATTR_PERMISSIONS)
Damien Miller770b3742003-01-08 14:04:53 +11001014 mode = a->perm & 0777;
Damien Miller33804262001-02-04 23:20:18 +11001015 else
1016 mode = 0666;
1017
Ben Lindstromc8d1c302001-03-17 00:34:46 +00001018 if ((a->flags & SSH2_FILEXFER_ATTR_PERMISSIONS) &&
Damien Miller5fa01fd2003-01-14 22:24:47 +11001019 (!S_ISREG(a->perm))) {
1020 error("Cannot download non-regular file: %s", remote_path);
Ben Lindstromc8d1c302001-03-17 00:34:46 +00001021 return(-1);
1022 }
1023
Damien Miller16a13332002-02-13 14:03:56 +11001024 if (a->flags & SSH2_FILEXFER_ATTR_SIZE)
1025 size = a->size;
1026 else
1027 size = 0;
1028
Damien Miller3db5f532002-02-13 14:10:32 +11001029 buflen = conn->transfer_buflen;
Damien Miller33804262001-02-04 23:20:18 +11001030 buffer_init(&msg);
1031
1032 /* Send open request */
Damien Miller3db5f532002-02-13 14:10:32 +11001033 id = conn->msg_id++;
Damien Miller33804262001-02-04 23:20:18 +11001034 buffer_put_char(&msg, SSH2_FXP_OPEN);
1035 buffer_put_int(&msg, id);
1036 buffer_put_cstring(&msg, remote_path);
1037 buffer_put_int(&msg, SSH2_FXF_READ);
1038 attrib_clear(&junk); /* Send empty attributes */
1039 encode_attrib(&msg, &junk);
Damien Miller65e42f82010-09-24 22:15:11 +10001040 send_msg(conn, &msg);
Ben Lindstromb1f483f2002-06-23 21:27:18 +00001041 debug3("Sent message SSH2_FXP_OPEN I:%u P:%s", id, remote_path);
Damien Miller33804262001-02-04 23:20:18 +11001042
Damien Miller65e42f82010-09-24 22:15:11 +10001043 handle = get_handle(conn, id, &handle_len,
Darren Tuckerc22f0902009-10-07 08:24:19 +11001044 "remote open(\"%s\")", remote_path);
Damien Miller33804262001-02-04 23:20:18 +11001045 if (handle == NULL) {
1046 buffer_free(&msg);
Damien Miller33804262001-02-04 23:20:18 +11001047 return(-1);
1048 }
1049
Damien Millera8e06ce2003-11-21 23:48:55 +11001050 local_fd = open(local_path, O_WRONLY | O_CREAT | O_TRUNC,
Damien Miller770b3742003-01-08 14:04:53 +11001051 mode | S_IWRITE);
Damien Miller3db5f532002-02-13 14:10:32 +11001052 if (local_fd == -1) {
1053 error("Couldn't open local file \"%s\" for writing: %s",
1054 local_path, strerror(errno));
Damien Miller6b0c8182008-02-10 22:23:41 +11001055 do_close(conn, handle, handle_len);
Ben Lindstrom021fcd32002-02-26 18:02:43 +00001056 buffer_free(&msg);
1057 xfree(handle);
Damien Miller3db5f532002-02-13 14:10:32 +11001058 return(-1);
1059 }
1060
Damien Miller33804262001-02-04 23:20:18 +11001061 /* Read from remote and write to local */
Damien Miller16a13332002-02-13 14:03:56 +11001062 write_error = read_error = write_errno = num_req = offset = 0;
1063 max_req = 1;
Damien Miller62d57f62003-01-10 21:43:24 +11001064 progress_counter = 0;
1065
Damien Miller9ba30692004-03-08 23:12:02 +11001066 if (showprogress && size != 0)
1067 start_progress_meter(remote_path, size, &progress_counter);
Damien Miller62d57f62003-01-10 21:43:24 +11001068
Damien Miller16a13332002-02-13 14:03:56 +11001069 while (num_req > 0 || max_req > 0) {
Damien Miller33804262001-02-04 23:20:18 +11001070 char *data;
Damien Miller16a13332002-02-13 14:03:56 +11001071 u_int len;
Damien Miller33804262001-02-04 23:20:18 +11001072
Darren Tuckercdf547a2004-05-24 10:12:19 +10001073 /*
Darren Tuckerfc959702004-07-17 16:12:08 +10001074 * Simulate EOF on interrupt: stop sending new requests and
Darren Tuckercdf547a2004-05-24 10:12:19 +10001075 * allow outstanding requests to drain gracefully
1076 */
1077 if (interrupted) {
1078 if (num_req == 0) /* If we haven't started yet... */
1079 break;
1080 max_req = 0;
1081 }
1082
Damien Miller16a13332002-02-13 14:03:56 +11001083 /* Send some more requests */
1084 while (num_req < max_req) {
Ben Lindstrom6328ab32002-03-22 02:54:23 +00001085 debug3("Request range %llu -> %llu (%d/%d)",
Ben Lindstromd45f28c2002-03-22 01:00:57 +00001086 (unsigned long long)offset,
1087 (unsigned long long)offset + buflen - 1,
1088 num_req, max_req);
Damien Miller16a13332002-02-13 14:03:56 +11001089 req = xmalloc(sizeof(*req));
Damien Miller3db5f532002-02-13 14:10:32 +11001090 req->id = conn->msg_id++;
Damien Miller16a13332002-02-13 14:03:56 +11001091 req->len = buflen;
1092 req->offset = offset;
1093 offset += buflen;
1094 num_req++;
1095 TAILQ_INSERT_TAIL(&requests, req, tq);
Damien Miller65e42f82010-09-24 22:15:11 +10001096 send_read_request(conn, req->id, req->offset,
Damien Miller16a13332002-02-13 14:03:56 +11001097 req->len, handle, handle_len);
1098 }
Damien Miller33804262001-02-04 23:20:18 +11001099
1100 buffer_clear(&msg);
Damien Miller65e42f82010-09-24 22:15:11 +10001101 get_msg(conn, &msg);
Damien Miller33804262001-02-04 23:20:18 +11001102 type = buffer_get_char(&msg);
1103 id = buffer_get_int(&msg);
Ben Lindstromb1f483f2002-06-23 21:27:18 +00001104 debug3("Received reply T:%u I:%u R:%d", type, id, max_req);
Damien Miller33804262001-02-04 23:20:18 +11001105
Damien Miller16a13332002-02-13 14:03:56 +11001106 /* Find the request in our queue */
Darren Tucker47eede72005-03-14 23:08:12 +11001107 for (req = TAILQ_FIRST(&requests);
Damien Miller16a13332002-02-13 14:03:56 +11001108 req != NULL && req->id != id;
1109 req = TAILQ_NEXT(req, tq))
1110 ;
1111 if (req == NULL)
1112 fatal("Unexpected reply %u", id);
1113
1114 switch (type) {
1115 case SSH2_FXP_STATUS:
1116 status = buffer_get_int(&msg);
1117 if (status != SSH2_FX_EOF)
1118 read_error = 1;
1119 max_req = 0;
1120 TAILQ_REMOVE(&requests, req, tq);
1121 xfree(req);
1122 num_req--;
1123 break;
1124 case SSH2_FXP_DATA:
1125 data = buffer_get_string(&msg, &len);
Ben Lindstromeb505452002-03-22 01:03:15 +00001126 debug3("Received data %llu -> %llu",
Ben Lindstrom6328ab32002-03-22 02:54:23 +00001127 (unsigned long long)req->offset,
Ben Lindstromeb505452002-03-22 01:03:15 +00001128 (unsigned long long)req->offset + len - 1);
Damien Miller16a13332002-02-13 14:03:56 +11001129 if (len > req->len)
1130 fatal("Received more data than asked for "
Ben Lindstrom93576d92002-12-23 02:06:19 +00001131 "%u > %u", len, req->len);
Damien Miller16a13332002-02-13 14:03:56 +11001132 if ((lseek(local_fd, req->offset, SEEK_SET) == -1 ||
Darren Tucker9f63f222003-07-03 13:46:56 +10001133 atomicio(vwrite, local_fd, data, len) != len) &&
Damien Miller16a13332002-02-13 14:03:56 +11001134 !write_error) {
1135 write_errno = errno;
1136 write_error = 1;
1137 max_req = 0;
Damien Miller33804262001-02-04 23:20:18 +11001138 }
Damien Miller62d57f62003-01-10 21:43:24 +11001139 progress_counter += len;
Damien Miller16a13332002-02-13 14:03:56 +11001140 xfree(data);
1141
1142 if (len == req->len) {
1143 TAILQ_REMOVE(&requests, req, tq);
1144 xfree(req);
1145 num_req--;
1146 } else {
1147 /* Resend the request for the missing data */
1148 debug3("Short data block, re-requesting "
Ben Lindstromeb505452002-03-22 01:03:15 +00001149 "%llu -> %llu (%2d)",
Ben Lindstrom6328ab32002-03-22 02:54:23 +00001150 (unsigned long long)req->offset + len,
Ben Lindstrom83b79e42002-03-22 01:05:27 +00001151 (unsigned long long)req->offset +
1152 req->len - 1, num_req);
Damien Miller3db5f532002-02-13 14:10:32 +11001153 req->id = conn->msg_id++;
Damien Miller16a13332002-02-13 14:03:56 +11001154 req->len -= len;
1155 req->offset += len;
Damien Miller65e42f82010-09-24 22:15:11 +10001156 send_read_request(conn, req->id,
Damien Miller3db5f532002-02-13 14:10:32 +11001157 req->offset, req->len, handle, handle_len);
Damien Miller16a13332002-02-13 14:03:56 +11001158 /* Reduce the request size */
1159 if (len < buflen)
1160 buflen = MAX(MIN_READ_SIZE, len);
1161 }
1162 if (max_req > 0) { /* max_req = 0 iff EOF received */
1163 if (size > 0 && offset > size) {
1164 /* Only one request at a time
1165 * after the expected EOF */
1166 debug3("Finish at %llu (%2d)",
Ben Lindstromeb505452002-03-22 01:03:15 +00001167 (unsigned long long)offset,
1168 num_req);
Damien Miller16a13332002-02-13 14:03:56 +11001169 max_req = 1;
Darren Tuckercdf547a2004-05-24 10:12:19 +10001170 } else if (max_req <= conn->num_requests) {
Damien Miller16a13332002-02-13 14:03:56 +11001171 ++max_req;
1172 }
1173 }
1174 break;
1175 default:
Ben Lindstromb1f483f2002-06-23 21:27:18 +00001176 fatal("Expected SSH2_FXP_DATA(%u) packet, got %u",
Damien Miller33804262001-02-04 23:20:18 +11001177 SSH2_FXP_DATA, type);
1178 }
Damien Miller33804262001-02-04 23:20:18 +11001179 }
Damien Miller33804262001-02-04 23:20:18 +11001180
Damien Miller62d57f62003-01-10 21:43:24 +11001181 if (showprogress && size)
1182 stop_progress_meter();
1183
Damien Miller16a13332002-02-13 14:03:56 +11001184 /* Sanity check */
1185 if (TAILQ_FIRST(&requests) != NULL)
1186 fatal("Transfer complete, but requests still in queue");
1187
1188 if (read_error) {
Ben Lindstrom6328ab32002-03-22 02:54:23 +00001189 error("Couldn't read from remote file \"%s\" : %s",
Damien Miller3db5f532002-02-13 14:10:32 +11001190 remote_path, fx2txt(status));
1191 do_close(conn, handle, handle_len);
Damien Miller16a13332002-02-13 14:03:56 +11001192 } else if (write_error) {
Damien Miller3db5f532002-02-13 14:10:32 +11001193 error("Couldn't write to \"%s\": %s", local_path,
1194 strerror(write_errno));
1195 status = -1;
1196 do_close(conn, handle, handle_len);
Damien Miller16a13332002-02-13 14:03:56 +11001197 } else {
Damien Miller3db5f532002-02-13 14:10:32 +11001198 status = do_close(conn, handle, handle_len);
Damien Miller16a13332002-02-13 14:03:56 +11001199
1200 /* Override umask and utimes if asked */
Ben Lindstrom6dc75f52001-02-17 16:47:47 +00001201#ifdef HAVE_FCHMOD
Damien Miller16a13332002-02-13 14:03:56 +11001202 if (pflag && fchmod(local_fd, mode) == -1)
Damien Millera8e06ce2003-11-21 23:48:55 +11001203#else
Damien Miller16a13332002-02-13 14:03:56 +11001204 if (pflag && chmod(local_path, mode) == -1)
Ben Lindstrom6dc75f52001-02-17 16:47:47 +00001205#endif /* HAVE_FCHMOD */
Damien Miller16a13332002-02-13 14:03:56 +11001206 error("Couldn't set mode on \"%s\": %s", local_path,
Ben Lindstrom93576d92002-12-23 02:06:19 +00001207 strerror(errno));
Damien Miller16a13332002-02-13 14:03:56 +11001208 if (pflag && (a->flags & SSH2_FILEXFER_ATTR_ACMODTIME)) {
1209 struct timeval tv[2];
1210 tv[0].tv_sec = a->atime;
1211 tv[1].tv_sec = a->mtime;
1212 tv[0].tv_usec = tv[1].tv_usec = 0;
1213 if (utimes(local_path, tv) == -1)
1214 error("Can't set times on \"%s\": %s",
Ben Lindstrom93576d92002-12-23 02:06:19 +00001215 local_path, strerror(errno));
Damien Miller16a13332002-02-13 14:03:56 +11001216 }
Ben Lindstrom9d4f2c82001-02-15 03:22:45 +00001217 }
Damien Millerd7686fd2001-02-10 00:40:03 +11001218 close(local_fd);
1219 buffer_free(&msg);
1220 xfree(handle);
Damien Miller3db5f532002-02-13 14:10:32 +11001221
1222 return(status);
Damien Miller33804262001-02-04 23:20:18 +11001223}
1224
Darren Tucker1b0dd172009-10-07 08:37:48 +11001225static int
1226download_dir_internal(struct sftp_conn *conn, char *src, char *dst,
1227 Attrib *dirattrib, int pflag, int printflag, int depth)
1228{
1229 int i, ret = 0;
1230 SFTP_DIRENT **dir_entries;
1231 char *filename, *new_src, *new_dst;
1232 mode_t mode = 0777;
1233
1234 if (depth >= MAX_DIR_DEPTH) {
1235 error("Maximum directory depth exceeded: %d levels", depth);
1236 return -1;
1237 }
1238
1239 if (dirattrib == NULL &&
1240 (dirattrib = do_stat(conn, src, 1)) == NULL) {
1241 error("Unable to stat remote directory \"%s\"", src);
1242 return -1;
1243 }
1244 if (!S_ISDIR(dirattrib->perm)) {
1245 error("\"%s\" is not a directory", src);
1246 return -1;
1247 }
1248 if (printflag)
1249 printf("Retrieving %s\n", src);
1250
1251 if (dirattrib->flags & SSH2_FILEXFER_ATTR_PERMISSIONS)
1252 mode = dirattrib->perm & 01777;
1253 else {
1254 debug("Server did not send permissions for "
1255 "directory \"%s\"", dst);
1256 }
1257
1258 if (mkdir(dst, mode) == -1 && errno != EEXIST) {
1259 error("mkdir %s: %s", dst, strerror(errno));
1260 return -1;
1261 }
1262
1263 if (do_readdir(conn, src, &dir_entries) == -1) {
1264 error("%s: Failed to get directory contents", src);
1265 return -1;
1266 }
1267
1268 for (i = 0; dir_entries[i] != NULL && !interrupted; i++) {
1269 filename = dir_entries[i]->filename;
1270
1271 new_dst = path_append(dst, filename);
1272 new_src = path_append(src, filename);
1273
1274 if (S_ISDIR(dir_entries[i]->a.perm)) {
1275 if (strcmp(filename, ".") == 0 ||
1276 strcmp(filename, "..") == 0)
1277 continue;
1278 if (download_dir_internal(conn, new_src, new_dst,
1279 &(dir_entries[i]->a), pflag, printflag,
1280 depth + 1) == -1)
1281 ret = -1;
1282 } else if (S_ISREG(dir_entries[i]->a.perm) ) {
1283 if (do_download(conn, new_src, new_dst,
1284 &(dir_entries[i]->a), pflag) == -1) {
1285 error("Download of file %s to %s failed",
1286 new_src, new_dst);
1287 ret = -1;
1288 }
1289 } else
1290 logit("%s: not a regular file\n", new_src);
1291
1292 xfree(new_dst);
1293 xfree(new_src);
1294 }
1295
1296 if (pflag) {
1297 if (dirattrib->flags & SSH2_FILEXFER_ATTR_ACMODTIME) {
1298 struct timeval tv[2];
1299 tv[0].tv_sec = dirattrib->atime;
1300 tv[1].tv_sec = dirattrib->mtime;
1301 tv[0].tv_usec = tv[1].tv_usec = 0;
1302 if (utimes(dst, tv) == -1)
1303 error("Can't set times on \"%s\": %s",
1304 dst, strerror(errno));
1305 } else
1306 debug("Server did not send times for directory "
1307 "\"%s\"", dst);
1308 }
1309
1310 free_sftp_dirents(dir_entries);
1311
1312 return ret;
1313}
1314
1315int
1316download_dir(struct sftp_conn *conn, char *src, char *dst,
1317 Attrib *dirattrib, int pflag, int printflag)
1318{
1319 char *src_canon;
1320 int ret;
1321
1322 if ((src_canon = do_realpath(conn, src)) == NULL) {
1323 error("Unable to canonicalise path \"%s\"", src);
1324 return -1;
1325 }
1326
1327 ret = download_dir_internal(conn, src_canon, dst,
1328 dirattrib, pflag, printflag, 0);
1329 xfree(src_canon);
1330 return ret;
1331}
1332
Damien Miller33804262001-02-04 23:20:18 +11001333int
Damien Miller3db5f532002-02-13 14:10:32 +11001334do_upload(struct sftp_conn *conn, char *local_path, char *remote_path,
1335 int pflag)
Damien Miller33804262001-02-04 23:20:18 +11001336{
Damien Milleracdf25b2008-02-10 22:27:24 +11001337 int local_fd;
1338 int status = SSH2_FX_OK;
Damien Miller5873dfd2002-02-13 14:04:37 +11001339 u_int handle_len, id, type;
Damien Miller8b3fdfb2007-09-17 16:12:03 +10001340 off_t offset;
Damien Miller8829d362002-02-08 22:04:05 +11001341 char *handle, *data;
Damien Miller33804262001-02-04 23:20:18 +11001342 Buffer msg;
1343 struct stat sb;
1344 Attrib a;
Damien Miller16a13332002-02-13 14:03:56 +11001345 u_int32_t startid;
1346 u_int32_t ackid;
Damien Miller5873dfd2002-02-13 14:04:37 +11001347 struct outstanding_ack {
1348 u_int id;
1349 u_int len;
Damien Miller8b3fdfb2007-09-17 16:12:03 +10001350 off_t offset;
Ben Lindstrom6328ab32002-03-22 02:54:23 +00001351 TAILQ_ENTRY(outstanding_ack) tq;
Damien Miller5873dfd2002-02-13 14:04:37 +11001352 };
1353 TAILQ_HEAD(ackhead, outstanding_ack) acks;
Damien Miller7cf17eb2004-06-15 10:28:56 +10001354 struct outstanding_ack *ack = NULL;
Damien Miller5873dfd2002-02-13 14:04:37 +11001355
1356 TAILQ_INIT(&acks);
Damien Miller33804262001-02-04 23:20:18 +11001357
1358 if ((local_fd = open(local_path, O_RDONLY, 0)) == -1) {
1359 error("Couldn't open local file \"%s\" for reading: %s",
1360 local_path, strerror(errno));
1361 return(-1);
1362 }
1363 if (fstat(local_fd, &sb) == -1) {
1364 error("Couldn't fstat local file \"%s\": %s",
1365 local_path, strerror(errno));
1366 close(local_fd);
1367 return(-1);
1368 }
Damien Miller5fa01fd2003-01-14 22:24:47 +11001369 if (!S_ISREG(sb.st_mode)) {
1370 error("%s is not a regular file", local_path);
1371 close(local_fd);
1372 return(-1);
1373 }
Damien Miller33804262001-02-04 23:20:18 +11001374 stat_to_attrib(&sb, &a);
1375
1376 a.flags &= ~SSH2_FILEXFER_ATTR_SIZE;
1377 a.flags &= ~SSH2_FILEXFER_ATTR_UIDGID;
1378 a.perm &= 0777;
1379 if (!pflag)
1380 a.flags &= ~SSH2_FILEXFER_ATTR_ACMODTIME;
1381
1382 buffer_init(&msg);
1383
1384 /* Send open request */
Damien Miller3db5f532002-02-13 14:10:32 +11001385 id = conn->msg_id++;
Damien Miller33804262001-02-04 23:20:18 +11001386 buffer_put_char(&msg, SSH2_FXP_OPEN);
1387 buffer_put_int(&msg, id);
1388 buffer_put_cstring(&msg, remote_path);
1389 buffer_put_int(&msg, SSH2_FXF_WRITE|SSH2_FXF_CREAT|SSH2_FXF_TRUNC);
1390 encode_attrib(&msg, &a);
Damien Miller65e42f82010-09-24 22:15:11 +10001391 send_msg(conn, &msg);
Ben Lindstromb1f483f2002-06-23 21:27:18 +00001392 debug3("Sent message SSH2_FXP_OPEN I:%u P:%s", id, remote_path);
Damien Miller33804262001-02-04 23:20:18 +11001393
1394 buffer_clear(&msg);
1395
Damien Miller65e42f82010-09-24 22:15:11 +10001396 handle = get_handle(conn, id, &handle_len,
Darren Tuckerc22f0902009-10-07 08:24:19 +11001397 "remote open(\"%s\")", remote_path);
Damien Miller33804262001-02-04 23:20:18 +11001398 if (handle == NULL) {
1399 close(local_fd);
1400 buffer_free(&msg);
Damien Milleracdf25b2008-02-10 22:27:24 +11001401 return -1;
Damien Miller33804262001-02-04 23:20:18 +11001402 }
1403
Damien Miller16a13332002-02-13 14:03:56 +11001404 startid = ackid = id + 1;
Damien Miller3db5f532002-02-13 14:10:32 +11001405 data = xmalloc(conn->transfer_buflen);
Damien Miller8829d362002-02-08 22:04:05 +11001406
Damien Miller33804262001-02-04 23:20:18 +11001407 /* Read from local and write to remote */
1408 offset = 0;
Damien Miller62d57f62003-01-10 21:43:24 +11001409 if (showprogress)
1410 start_progress_meter(local_path, sb.st_size, &offset);
Damien Miller62d57f62003-01-10 21:43:24 +11001411
Damien Miller9f0f5c62001-12-21 14:45:46 +11001412 for (;;) {
Damien Miller33804262001-02-04 23:20:18 +11001413 int len;
Damien Miller33804262001-02-04 23:20:18 +11001414
1415 /*
Darren Tuckerfc959702004-07-17 16:12:08 +10001416 * Can't use atomicio here because it returns 0 on EOF,
Darren Tuckercdf547a2004-05-24 10:12:19 +10001417 * thus losing the last block of the file.
Darren Tuckerfc959702004-07-17 16:12:08 +10001418 * Simulate an EOF on interrupt, allowing ACKs from the
Darren Tuckercdf547a2004-05-24 10:12:19 +10001419 * server to drain.
Damien Miller33804262001-02-04 23:20:18 +11001420 */
Damien Milleracdf25b2008-02-10 22:27:24 +11001421 if (interrupted || status != SSH2_FX_OK)
Darren Tuckercdf547a2004-05-24 10:12:19 +10001422 len = 0;
1423 else do
Damien Miller3db5f532002-02-13 14:10:32 +11001424 len = read(local_fd, data, conn->transfer_buflen);
Damien Millerd8968ad2008-07-04 23:10:49 +10001425 while ((len == -1) &&
1426 (errno == EINTR || errno == EAGAIN || errno == EWOULDBLOCK));
Damien Miller33804262001-02-04 23:20:18 +11001427
1428 if (len == -1)
1429 fatal("Couldn't read from \"%s\": %s", local_path,
1430 strerror(errno));
Damien Miller16a13332002-02-13 14:03:56 +11001431
1432 if (len != 0) {
Damien Miller5873dfd2002-02-13 14:04:37 +11001433 ack = xmalloc(sizeof(*ack));
1434 ack->id = ++id;
1435 ack->offset = offset;
1436 ack->len = len;
1437 TAILQ_INSERT_TAIL(&acks, ack, tq);
1438
Damien Miller16a13332002-02-13 14:03:56 +11001439 buffer_clear(&msg);
1440 buffer_put_char(&msg, SSH2_FXP_WRITE);
Damien Miller5873dfd2002-02-13 14:04:37 +11001441 buffer_put_int(&msg, ack->id);
Damien Miller16a13332002-02-13 14:03:56 +11001442 buffer_put_string(&msg, handle, handle_len);
1443 buffer_put_int64(&msg, offset);
1444 buffer_put_string(&msg, data, len);
Damien Miller65e42f82010-09-24 22:15:11 +10001445 send_msg(conn, &msg);
Ben Lindstromb1f483f2002-06-23 21:27:18 +00001446 debug3("Sent message SSH2_FXP_WRITE I:%u O:%llu S:%u",
Ben Lindstrom93576d92002-12-23 02:06:19 +00001447 id, (unsigned long long)offset, len);
Damien Miller5873dfd2002-02-13 14:04:37 +11001448 } else if (TAILQ_FIRST(&acks) == NULL)
Damien Miller33804262001-02-04 23:20:18 +11001449 break;
1450
Damien Miller5873dfd2002-02-13 14:04:37 +11001451 if (ack == NULL)
1452 fatal("Unexpected ACK %u", id);
1453
Ben Lindstrom6328ab32002-03-22 02:54:23 +00001454 if (id == startid || len == 0 ||
Damien Miller3db5f532002-02-13 14:10:32 +11001455 id - ackid >= conn->num_requests) {
Ben Lindstrom5a6abda2002-06-09 19:41:48 +00001456 u_int r_id;
Ben Lindstrom06e95152002-04-06 04:16:45 +00001457
Damien Miller5873dfd2002-02-13 14:04:37 +11001458 buffer_clear(&msg);
Damien Miller65e42f82010-09-24 22:15:11 +10001459 get_msg(conn, &msg);
Damien Miller5873dfd2002-02-13 14:04:37 +11001460 type = buffer_get_char(&msg);
Ben Lindstrom06e95152002-04-06 04:16:45 +00001461 r_id = buffer_get_int(&msg);
Damien Miller5873dfd2002-02-13 14:04:37 +11001462
1463 if (type != SSH2_FXP_STATUS)
1464 fatal("Expected SSH2_FXP_STATUS(%d) packet, "
1465 "got %d", SSH2_FXP_STATUS, type);
1466
1467 status = buffer_get_int(&msg);
1468 debug3("SSH2_FXP_STATUS %d", status);
1469
1470 /* Find the request in our queue */
Darren Tucker47eede72005-03-14 23:08:12 +11001471 for (ack = TAILQ_FIRST(&acks);
Ben Lindstrom06e95152002-04-06 04:16:45 +00001472 ack != NULL && ack->id != r_id;
Damien Miller5873dfd2002-02-13 14:04:37 +11001473 ack = TAILQ_NEXT(ack, tq))
1474 ;
1475 if (ack == NULL)
Ben Lindstromb1f483f2002-06-23 21:27:18 +00001476 fatal("Can't find request for ID %u", r_id);
Damien Miller5873dfd2002-02-13 14:04:37 +11001477 TAILQ_REMOVE(&acks, ack, tq);
Damien Miller8b3fdfb2007-09-17 16:12:03 +10001478 debug3("In write loop, ack for %u %u bytes at %lld",
1479 ack->id, ack->len, (long long)ack->offset);
Damien Miller16a13332002-02-13 14:03:56 +11001480 ++ackid;
Ben Lindstromeec16fc2002-07-04 00:06:15 +00001481 xfree(ack);
Damien Miller33804262001-02-04 23:20:18 +11001482 }
Damien Miller33804262001-02-04 23:20:18 +11001483 offset += len;
Damien Miller8b3fdfb2007-09-17 16:12:03 +10001484 if (offset < 0)
1485 fatal("%s: offset < 0", __func__);
Damien Miller33804262001-02-04 23:20:18 +11001486 }
Damien Milleracdf25b2008-02-10 22:27:24 +11001487 buffer_free(&msg);
1488
Damien Miller62d57f62003-01-10 21:43:24 +11001489 if (showprogress)
1490 stop_progress_meter();
Damien Miller8829d362002-02-08 22:04:05 +11001491 xfree(data);
Damien Miller33804262001-02-04 23:20:18 +11001492
Damien Milleracdf25b2008-02-10 22:27:24 +11001493 if (status != SSH2_FX_OK) {
1494 error("Couldn't write to remote file \"%s\": %s",
1495 remote_path, fx2txt(status));
1496 status = -1;
1497 }
1498
Damien Miller33804262001-02-04 23:20:18 +11001499 if (close(local_fd) == -1) {
1500 error("Couldn't close local file \"%s\": %s", local_path,
1501 strerror(errno));
Damien Millerd7686fd2001-02-10 00:40:03 +11001502 status = -1;
Damien Miller33804262001-02-04 23:20:18 +11001503 }
1504
Ben Lindstrom9d4f2c82001-02-15 03:22:45 +00001505 /* Override umask and utimes if asked */
1506 if (pflag)
Damien Miller3db5f532002-02-13 14:10:32 +11001507 do_fsetstat(conn, handle, handle_len, &a);
Ben Lindstrom9d4f2c82001-02-15 03:22:45 +00001508
Damien Milleracdf25b2008-02-10 22:27:24 +11001509 if (do_close(conn, handle, handle_len) != SSH2_FX_OK)
1510 status = -1;
Damien Millerd7686fd2001-02-10 00:40:03 +11001511 xfree(handle);
Damien Milleracdf25b2008-02-10 22:27:24 +11001512
1513 return status;
Damien Miller33804262001-02-04 23:20:18 +11001514}
Darren Tucker1b0dd172009-10-07 08:37:48 +11001515
1516static int
1517upload_dir_internal(struct sftp_conn *conn, char *src, char *dst,
1518 int pflag, int printflag, int depth)
1519{
1520 int ret = 0, status;
1521 DIR *dirp;
1522 struct dirent *dp;
1523 char *filename, *new_src, *new_dst;
1524 struct stat sb;
1525 Attrib a;
1526
1527 if (depth >= MAX_DIR_DEPTH) {
1528 error("Maximum directory depth exceeded: %d levels", depth);
1529 return -1;
1530 }
1531
1532 if (stat(src, &sb) == -1) {
1533 error("Couldn't stat directory \"%s\": %s",
1534 src, strerror(errno));
1535 return -1;
1536 }
1537 if (!S_ISDIR(sb.st_mode)) {
1538 error("\"%s\" is not a directory", src);
1539 return -1;
1540 }
1541 if (printflag)
1542 printf("Entering %s\n", src);
1543
1544 attrib_clear(&a);
1545 stat_to_attrib(&sb, &a);
1546 a.flags &= ~SSH2_FILEXFER_ATTR_SIZE;
1547 a.flags &= ~SSH2_FILEXFER_ATTR_UIDGID;
1548 a.perm &= 01777;
1549 if (!pflag)
1550 a.flags &= ~SSH2_FILEXFER_ATTR_ACMODTIME;
1551
1552 status = do_mkdir(conn, dst, &a, 0);
1553 /*
1554 * we lack a portable status for errno EEXIST,
1555 * so if we get a SSH2_FX_FAILURE back we must check
1556 * if it was created successfully.
1557 */
1558 if (status != SSH2_FX_OK) {
1559 if (status != SSH2_FX_FAILURE)
1560 return -1;
1561 if (do_stat(conn, dst, 0) == NULL)
1562 return -1;
1563 }
1564
1565 if ((dirp = opendir(src)) == NULL) {
1566 error("Failed to open dir \"%s\": %s", src, strerror(errno));
1567 return -1;
1568 }
1569
1570 while (((dp = readdir(dirp)) != NULL) && !interrupted) {
1571 if (dp->d_ino == 0)
1572 continue;
1573 filename = dp->d_name;
1574 new_dst = path_append(dst, filename);
1575 new_src = path_append(src, filename);
1576
Darren Tucker438b4732009-10-11 21:52:10 +11001577 if (lstat(new_src, &sb) == -1) {
1578 logit("%s: lstat failed: %s", filename,
1579 strerror(errno));
1580 ret = -1;
1581 } else if (S_ISDIR(sb.st_mode)) {
Darren Tucker1b0dd172009-10-07 08:37:48 +11001582 if (strcmp(filename, ".") == 0 ||
1583 strcmp(filename, "..") == 0)
1584 continue;
1585
1586 if (upload_dir_internal(conn, new_src, new_dst,
Damien Millerc4bb91c2010-08-03 16:04:22 +10001587 pflag, printflag, depth + 1) == -1)
Darren Tucker1b0dd172009-10-07 08:37:48 +11001588 ret = -1;
Darren Tucker438b4732009-10-11 21:52:10 +11001589 } else if (S_ISREG(sb.st_mode)) {
Darren Tucker1b0dd172009-10-07 08:37:48 +11001590 if (do_upload(conn, new_src, new_dst, pflag) == -1) {
1591 error("Uploading of file %s to %s failed!",
1592 new_src, new_dst);
1593 ret = -1;
1594 }
1595 } else
1596 logit("%s: not a regular file\n", filename);
1597 xfree(new_dst);
1598 xfree(new_src);
1599 }
1600
1601 do_setstat(conn, dst, &a);
1602
1603 (void) closedir(dirp);
1604 return ret;
1605}
1606
1607int
1608upload_dir(struct sftp_conn *conn, char *src, char *dst, int printflag,
1609 int pflag)
1610{
1611 char *dst_canon;
1612 int ret;
1613
1614 if ((dst_canon = do_realpath(conn, dst)) == NULL) {
1615 error("Unable to canonicalise path \"%s\"", dst);
1616 return -1;
1617 }
1618
1619 ret = upload_dir_internal(conn, src, dst_canon, pflag, printflag, 0);
1620 xfree(dst_canon);
1621 return ret;
1622}
1623
1624char *
1625path_append(char *p1, char *p2)
1626{
1627 char *ret;
1628 size_t len = strlen(p1) + strlen(p2) + 2;
1629
1630 ret = xmalloc(len);
1631 strlcpy(ret, p1, len);
1632 if (p1[0] != '\0' && p1[strlen(p1) - 1] != '/')
1633 strlcat(ret, "/", len);
1634 strlcat(ret, p2, len);
1635
1636 return(ret);
1637}
1638