blob: 6dcd5de74695ccbfa067d7d32c2ccfafb8a10062 [file] [log] [blame]
Damien Miller33804262001-02-04 23:20:18 +11001/*
Damien Miller4e60ed72004-02-17 17:07:59 +11002 * Copyright (c) 2001-2004 Damien Miller <djm@openbsd.org>
Damien Miller33804262001-02-04 23:20:18 +11003 *
Damien Miller4e60ed72004-02-17 17:07:59 +11004 * Permission to use, copy, modify, and distribute this software for any
5 * purpose with or without fee is hereby granted, provided that the above
6 * copyright notice and this permission notice appear in all copies.
Damien Miller33804262001-02-04 23:20:18 +11007 *
Damien Miller4e60ed72004-02-17 17:07:59 +11008 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
9 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
10 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
11 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
12 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
13 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
14 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
Damien Miller33804262001-02-04 23:20:18 +110015 */
16
17/* XXX: memleaks */
18/* XXX: signed vs unsigned */
Damien Miller3db5f532002-02-13 14:10:32 +110019/* XXX: remove all logging, only return status codes */
Damien Miller33804262001-02-04 23:20:18 +110020/* XXX: copy between two remote sites */
21
22#include "includes.h"
Darren Tuckercdf547a2004-05-24 10:12:19 +100023RCSID("$OpenBSD: sftp-client.c,v 1.49 2004/05/19 12:17:33 djm Exp $");
Damien Miller16a13332002-02-13 14:03:56 +110024
Damien Miller9b481512002-09-12 10:43:29 +100025#include "openbsd-compat/sys-queue.h"
Damien Miller33804262001-02-04 23:20:18 +110026
Damien Miller33804262001-02-04 23:20:18 +110027#include "buffer.h"
28#include "bufaux.h"
29#include "getput.h"
30#include "xmalloc.h"
31#include "log.h"
32#include "atomicio.h"
Damien Miller62d57f62003-01-10 21:43:24 +110033#include "progressmeter.h"
Damien Miller33804262001-02-04 23:20:18 +110034
35#include "sftp.h"
36#include "sftp-common.h"
37#include "sftp-client.h"
38
Darren Tuckercdf547a2004-05-24 10:12:19 +100039extern volatile sig_atomic_t interrupted;
Damien Miller62d57f62003-01-10 21:43:24 +110040extern int showprogress;
41
Damien Miller16a13332002-02-13 14:03:56 +110042/* Minimum amount of data to read at at time */
43#define MIN_READ_SIZE 512
44
Damien Millera7f3aaa2003-01-10 21:43:58 +110045/* Maximum packet size */
46#define MAX_MSG_LENGTH (256 * 1024)
47
Damien Miller3db5f532002-02-13 14:10:32 +110048struct sftp_conn {
49 int fd_in;
50 int fd_out;
51 u_int transfer_buflen;
52 u_int num_requests;
53 u_int version;
54 u_int msg_id;
55};
Ben Lindstrom288cc392001-02-09 02:58:04 +000056
Ben Lindstrombba81212001-06-25 05:01:22 +000057static void
Damien Miller33804262001-02-04 23:20:18 +110058send_msg(int fd, Buffer *m)
59{
Damien Millera7f3aaa2003-01-10 21:43:58 +110060 u_char mlen[4];
Damien Miller33804262001-02-04 23:20:18 +110061
Damien Millera7f3aaa2003-01-10 21:43:58 +110062 if (buffer_len(m) > MAX_MSG_LENGTH)
63 fatal("Outbound message too long %u", buffer_len(m));
Damien Miller33804262001-02-04 23:20:18 +110064
Damien Millera7f3aaa2003-01-10 21:43:58 +110065 /* Send length first */
66 PUT_32BIT(mlen, buffer_len(m));
Darren Tucker9f63f222003-07-03 13:46:56 +100067 if (atomicio(vwrite, fd, mlen, sizeof(mlen)) <= 0)
Damien Miller33804262001-02-04 23:20:18 +110068 fatal("Couldn't send packet: %s", strerror(errno));
69
Darren Tucker9f63f222003-07-03 13:46:56 +100070 if (atomicio(vwrite, fd, buffer_ptr(m), buffer_len(m)) <= 0)
Damien Millera7f3aaa2003-01-10 21:43:58 +110071 fatal("Couldn't send packet: %s", strerror(errno));
72
73 buffer_clear(m);
Damien Miller33804262001-02-04 23:20:18 +110074}
75
Ben Lindstrombba81212001-06-25 05:01:22 +000076static void
Damien Miller33804262001-02-04 23:20:18 +110077get_msg(int fd, Buffer *m)
78{
Damien Millera7f3aaa2003-01-10 21:43:58 +110079 ssize_t len;
80 u_int msg_len;
Damien Miller33804262001-02-04 23:20:18 +110081
Damien Millera7f3aaa2003-01-10 21:43:58 +110082 buffer_append_space(m, 4);
83 len = atomicio(read, fd, buffer_ptr(m), 4);
Damien Millercafff192001-03-19 22:29:46 +110084 if (len == 0)
85 fatal("Connection closed");
86 else if (len == -1)
Damien Miller33804262001-02-04 23:20:18 +110087 fatal("Couldn't read packet: %s", strerror(errno));
88
Damien Millera7f3aaa2003-01-10 21:43:58 +110089 msg_len = buffer_get_int(m);
90 if (msg_len > MAX_MSG_LENGTH)
Ben Lindstromb1f483f2002-06-23 21:27:18 +000091 fatal("Received message too long %u", msg_len);
Damien Miller33804262001-02-04 23:20:18 +110092
Damien Millera7f3aaa2003-01-10 21:43:58 +110093 buffer_append_space(m, msg_len);
94 len = atomicio(read, fd, buffer_ptr(m), msg_len);
95 if (len == 0)
96 fatal("Connection closed");
97 else if (len == -1)
98 fatal("Read packet: %s", strerror(errno));
Damien Miller33804262001-02-04 23:20:18 +110099}
100
Ben Lindstrombba81212001-06-25 05:01:22 +0000101static void
Damien Miller33804262001-02-04 23:20:18 +1100102send_string_request(int fd, u_int id, u_int code, char *s,
103 u_int len)
104{
105 Buffer msg;
106
107 buffer_init(&msg);
108 buffer_put_char(&msg, code);
109 buffer_put_int(&msg, id);
110 buffer_put_string(&msg, s, len);
111 send_msg(fd, &msg);
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000112 debug3("Sent message fd %d T:%u I:%u", fd, code, id);
Damien Miller33804262001-02-04 23:20:18 +1100113 buffer_free(&msg);
114}
115
Ben Lindstrombba81212001-06-25 05:01:22 +0000116static void
Damien Miller33804262001-02-04 23:20:18 +1100117send_string_attrs_request(int fd, u_int id, u_int code, char *s,
118 u_int len, Attrib *a)
119{
120 Buffer msg;
121
122 buffer_init(&msg);
123 buffer_put_char(&msg, code);
124 buffer_put_int(&msg, id);
125 buffer_put_string(&msg, s, len);
126 encode_attrib(&msg, a);
127 send_msg(fd, &msg);
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000128 debug3("Sent message fd %d T:%u I:%u", fd, code, id);
Damien Miller33804262001-02-04 23:20:18 +1100129 buffer_free(&msg);
130}
131
Ben Lindstrombba81212001-06-25 05:01:22 +0000132static u_int
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000133get_status(int fd, u_int expected_id)
Damien Miller33804262001-02-04 23:20:18 +1100134{
135 Buffer msg;
136 u_int type, id, status;
137
138 buffer_init(&msg);
139 get_msg(fd, &msg);
140 type = buffer_get_char(&msg);
141 id = buffer_get_int(&msg);
142
143 if (id != expected_id)
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000144 fatal("ID mismatch (%u != %u)", id, expected_id);
Damien Miller33804262001-02-04 23:20:18 +1100145 if (type != SSH2_FXP_STATUS)
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000146 fatal("Expected SSH2_FXP_STATUS(%u) packet, got %u",
Damien Miller33804262001-02-04 23:20:18 +1100147 SSH2_FXP_STATUS, type);
148
149 status = buffer_get_int(&msg);
150 buffer_free(&msg);
151
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000152 debug3("SSH2_FXP_STATUS %u", status);
Damien Miller33804262001-02-04 23:20:18 +1100153
154 return(status);
155}
156
Ben Lindstrombba81212001-06-25 05:01:22 +0000157static char *
Damien Miller33804262001-02-04 23:20:18 +1100158get_handle(int fd, u_int expected_id, u_int *len)
159{
160 Buffer msg;
161 u_int type, id;
162 char *handle;
163
164 buffer_init(&msg);
165 get_msg(fd, &msg);
166 type = buffer_get_char(&msg);
167 id = buffer_get_int(&msg);
168
169 if (id != expected_id)
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000170 fatal("ID mismatch (%u != %u)", id, expected_id);
Damien Miller33804262001-02-04 23:20:18 +1100171 if (type == SSH2_FXP_STATUS) {
172 int status = buffer_get_int(&msg);
173
174 error("Couldn't get handle: %s", fx2txt(status));
175 return(NULL);
176 } else if (type != SSH2_FXP_HANDLE)
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000177 fatal("Expected SSH2_FXP_HANDLE(%u) packet, got %u",
Damien Miller33804262001-02-04 23:20:18 +1100178 SSH2_FXP_HANDLE, type);
179
180 handle = buffer_get_string(&msg, len);
181 buffer_free(&msg);
182
183 return(handle);
184}
185
Ben Lindstrombba81212001-06-25 05:01:22 +0000186static Attrib *
Ben Lindstromc8d1c302001-03-17 00:34:46 +0000187get_decode_stat(int fd, u_int expected_id, int quiet)
Damien Miller33804262001-02-04 23:20:18 +1100188{
189 Buffer msg;
190 u_int type, id;
191 Attrib *a;
192
193 buffer_init(&msg);
194 get_msg(fd, &msg);
195
196 type = buffer_get_char(&msg);
197 id = buffer_get_int(&msg);
198
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000199 debug3("Received stat reply T:%u I:%u", type, id);
Damien Miller33804262001-02-04 23:20:18 +1100200 if (id != expected_id)
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000201 fatal("ID mismatch (%u != %u)", id, expected_id);
Damien Miller33804262001-02-04 23:20:18 +1100202 if (type == SSH2_FXP_STATUS) {
203 int status = buffer_get_int(&msg);
204
Ben Lindstromc8d1c302001-03-17 00:34:46 +0000205 if (quiet)
206 debug("Couldn't stat remote file: %s", fx2txt(status));
207 else
208 error("Couldn't stat remote file: %s", fx2txt(status));
Damien Miller33804262001-02-04 23:20:18 +1100209 return(NULL);
210 } else if (type != SSH2_FXP_ATTRS) {
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000211 fatal("Expected SSH2_FXP_ATTRS(%u) packet, got %u",
Damien Miller33804262001-02-04 23:20:18 +1100212 SSH2_FXP_ATTRS, type);
213 }
214 a = decode_attrib(&msg);
215 buffer_free(&msg);
216
217 return(a);
218}
219
Damien Miller3db5f532002-02-13 14:10:32 +1100220struct sftp_conn *
221do_init(int fd_in, int fd_out, u_int transfer_buflen, u_int num_requests)
Damien Miller33804262001-02-04 23:20:18 +1100222{
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000223 u_int type;
224 int version;
Damien Miller33804262001-02-04 23:20:18 +1100225 Buffer msg;
Damien Miller3db5f532002-02-13 14:10:32 +1100226 struct sftp_conn *ret;
Damien Miller33804262001-02-04 23:20:18 +1100227
228 buffer_init(&msg);
229 buffer_put_char(&msg, SSH2_FXP_INIT);
230 buffer_put_int(&msg, SSH2_FILEXFER_VERSION);
231 send_msg(fd_out, &msg);
232
233 buffer_clear(&msg);
234
235 get_msg(fd_in, &msg);
236
Kevin Stevesef4eea92001-02-05 12:42:17 +0000237 /* Expecting a VERSION reply */
Damien Miller33804262001-02-04 23:20:18 +1100238 if ((type = buffer_get_char(&msg)) != SSH2_FXP_VERSION) {
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000239 error("Invalid packet back from SSH2_FXP_INIT (type %u)",
Damien Miller33804262001-02-04 23:20:18 +1100240 type);
241 buffer_free(&msg);
Damien Miller3db5f532002-02-13 14:10:32 +1100242 return(NULL);
Damien Miller33804262001-02-04 23:20:18 +1100243 }
244 version = buffer_get_int(&msg);
245
246 debug2("Remote version: %d", version);
247
248 /* Check for extensions */
249 while (buffer_len(&msg) > 0) {
250 char *name = buffer_get_string(&msg, NULL);
251 char *value = buffer_get_string(&msg, NULL);
252
253 debug2("Init extension: \"%s\"", name);
254 xfree(name);
255 xfree(value);
256 }
257
258 buffer_free(&msg);
Damien Miller058316f2001-03-08 10:08:49 +1100259
Damien Miller3db5f532002-02-13 14:10:32 +1100260 ret = xmalloc(sizeof(*ret));
261 ret->fd_in = fd_in;
262 ret->fd_out = fd_out;
263 ret->transfer_buflen = transfer_buflen;
264 ret->num_requests = num_requests;
265 ret->version = version;
266 ret->msg_id = 1;
267
268 /* Some filexfer v.0 servers don't support large packets */
269 if (version == 0)
Ben Lindstroma1d81142002-04-02 20:58:11 +0000270 ret->transfer_buflen = MIN(ret->transfer_buflen, 20480);
Damien Miller3db5f532002-02-13 14:10:32 +1100271
272 return(ret);
273}
274
275u_int
276sftp_proto_version(struct sftp_conn *conn)
277{
278 return(conn->version);
Damien Miller33804262001-02-04 23:20:18 +1100279}
280
281int
Damien Miller3db5f532002-02-13 14:10:32 +1100282do_close(struct sftp_conn *conn, char *handle, u_int handle_len)
Damien Miller33804262001-02-04 23:20:18 +1100283{
284 u_int id, status;
285 Buffer msg;
286
287 buffer_init(&msg);
288
Damien Miller3db5f532002-02-13 14:10:32 +1100289 id = conn->msg_id++;
Damien Miller33804262001-02-04 23:20:18 +1100290 buffer_put_char(&msg, SSH2_FXP_CLOSE);
291 buffer_put_int(&msg, id);
292 buffer_put_string(&msg, handle, handle_len);
Damien Miller3db5f532002-02-13 14:10:32 +1100293 send_msg(conn->fd_out, &msg);
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000294 debug3("Sent message SSH2_FXP_CLOSE I:%u", id);
Damien Miller33804262001-02-04 23:20:18 +1100295
Damien Miller3db5f532002-02-13 14:10:32 +1100296 status = get_status(conn->fd_in, id);
Damien Miller33804262001-02-04 23:20:18 +1100297 if (status != SSH2_FX_OK)
298 error("Couldn't close file: %s", fx2txt(status));
299
300 buffer_free(&msg);
301
302 return(status);
303}
304
Damien Miller4870afd2001-03-14 10:27:09 +1100305
Ben Lindstrombba81212001-06-25 05:01:22 +0000306static int
Damien Miller3db5f532002-02-13 14:10:32 +1100307do_lsreaddir(struct sftp_conn *conn, char *path, int printflag,
Damien Miller4870afd2001-03-14 10:27:09 +1100308 SFTP_DIRENT ***dir)
Damien Miller33804262001-02-04 23:20:18 +1100309{
310 Buffer msg;
Ben Lindstrom025df4a2001-03-14 15:16:34 +0000311 u_int type, id, handle_len, i, expected_id, ents = 0;
Damien Miller33804262001-02-04 23:20:18 +1100312 char *handle;
313
Damien Miller3db5f532002-02-13 14:10:32 +1100314 id = conn->msg_id++;
Damien Miller33804262001-02-04 23:20:18 +1100315
316 buffer_init(&msg);
317 buffer_put_char(&msg, SSH2_FXP_OPENDIR);
318 buffer_put_int(&msg, id);
319 buffer_put_cstring(&msg, path);
Damien Miller3db5f532002-02-13 14:10:32 +1100320 send_msg(conn->fd_out, &msg);
Damien Miller33804262001-02-04 23:20:18 +1100321
322 buffer_clear(&msg);
323
Damien Miller3db5f532002-02-13 14:10:32 +1100324 handle = get_handle(conn->fd_in, id, &handle_len);
Damien Miller33804262001-02-04 23:20:18 +1100325 if (handle == NULL)
326 return(-1);
327
Damien Miller4870afd2001-03-14 10:27:09 +1100328 if (dir) {
329 ents = 0;
330 *dir = xmalloc(sizeof(**dir));
331 (*dir)[0] = NULL;
332 }
Damien Miller4870afd2001-03-14 10:27:09 +1100333
Darren Tuckercdf547a2004-05-24 10:12:19 +1000334 for (; !interrupted;) {
Damien Miller33804262001-02-04 23:20:18 +1100335 int count;
336
Damien Miller3db5f532002-02-13 14:10:32 +1100337 id = expected_id = conn->msg_id++;
Damien Miller33804262001-02-04 23:20:18 +1100338
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000339 debug3("Sending SSH2_FXP_READDIR I:%u", id);
Damien Miller33804262001-02-04 23:20:18 +1100340
341 buffer_clear(&msg);
342 buffer_put_char(&msg, SSH2_FXP_READDIR);
343 buffer_put_int(&msg, id);
344 buffer_put_string(&msg, handle, handle_len);
Damien Miller3db5f532002-02-13 14:10:32 +1100345 send_msg(conn->fd_out, &msg);
Damien Miller33804262001-02-04 23:20:18 +1100346
347 buffer_clear(&msg);
348
Damien Miller3db5f532002-02-13 14:10:32 +1100349 get_msg(conn->fd_in, &msg);
Damien Miller33804262001-02-04 23:20:18 +1100350
351 type = buffer_get_char(&msg);
352 id = buffer_get_int(&msg);
353
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000354 debug3("Received reply T:%u I:%u", type, id);
Damien Miller33804262001-02-04 23:20:18 +1100355
356 if (id != expected_id)
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000357 fatal("ID mismatch (%u != %u)", id, expected_id);
Damien Miller33804262001-02-04 23:20:18 +1100358
359 if (type == SSH2_FXP_STATUS) {
360 int status = buffer_get_int(&msg);
361
362 debug3("Received SSH2_FXP_STATUS %d", status);
363
364 if (status == SSH2_FX_EOF) {
365 break;
366 } else {
367 error("Couldn't read directory: %s",
368 fx2txt(status));
Damien Miller3db5f532002-02-13 14:10:32 +1100369 do_close(conn, handle, handle_len);
Damien Miller00111382003-03-10 11:21:17 +1100370 xfree(handle);
Ben Lindstrom10ac33f2001-02-10 21:53:40 +0000371 return(status);
Damien Miller33804262001-02-04 23:20:18 +1100372 }
373 } else if (type != SSH2_FXP_NAME)
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000374 fatal("Expected SSH2_FXP_NAME(%u) packet, got %u",
Damien Miller33804262001-02-04 23:20:18 +1100375 SSH2_FXP_NAME, type);
376
377 count = buffer_get_int(&msg);
Damien Millerd7686fd2001-02-10 00:40:03 +1100378 if (count == 0)
379 break;
380 debug3("Received %d SSH2_FXP_NAME responses", count);
Damien Miller9f0f5c62001-12-21 14:45:46 +1100381 for (i = 0; i < count; i++) {
Damien Miller33804262001-02-04 23:20:18 +1100382 char *filename, *longname;
383 Attrib *a;
384
385 filename = buffer_get_string(&msg, NULL);
386 longname = buffer_get_string(&msg, NULL);
387 a = decode_attrib(&msg);
388
Damien Miller4870afd2001-03-14 10:27:09 +1100389 if (printflag)
390 printf("%s\n", longname);
391
392 if (dir) {
Ben Lindstroma3700052001-04-05 23:26:32 +0000393 *dir = xrealloc(*dir, sizeof(**dir) *
Damien Miller4870afd2001-03-14 10:27:09 +1100394 (ents + 2));
395 (*dir)[ents] = xmalloc(sizeof(***dir));
396 (*dir)[ents]->filename = xstrdup(filename);
397 (*dir)[ents]->longname = xstrdup(longname);
398 memcpy(&(*dir)[ents]->a, a, sizeof(*a));
399 (*dir)[++ents] = NULL;
400 }
Damien Miller33804262001-02-04 23:20:18 +1100401
402 xfree(filename);
403 xfree(longname);
404 }
405 }
406
407 buffer_free(&msg);
Damien Miller3db5f532002-02-13 14:10:32 +1100408 do_close(conn, handle, handle_len);
Damien Miller33804262001-02-04 23:20:18 +1100409 xfree(handle);
410
Darren Tuckercdf547a2004-05-24 10:12:19 +1000411 /* Don't return partial matches on interrupt */
412 if (interrupted && dir != NULL && *dir != NULL) {
413 free_sftp_dirents(*dir);
414 *dir = xmalloc(sizeof(**dir));
415 **dir = NULL;
416 }
417
Damien Miller33804262001-02-04 23:20:18 +1100418 return(0);
419}
420
421int
Damien Miller3db5f532002-02-13 14:10:32 +1100422do_readdir(struct sftp_conn *conn, char *path, SFTP_DIRENT ***dir)
Damien Miller4870afd2001-03-14 10:27:09 +1100423{
Damien Miller3db5f532002-02-13 14:10:32 +1100424 return(do_lsreaddir(conn, path, 0, dir));
Damien Miller4870afd2001-03-14 10:27:09 +1100425}
426
427void free_sftp_dirents(SFTP_DIRENT **s)
428{
429 int i;
Damien Miller9f0f5c62001-12-21 14:45:46 +1100430
431 for (i = 0; s[i]; i++) {
Damien Miller4870afd2001-03-14 10:27:09 +1100432 xfree(s[i]->filename);
433 xfree(s[i]->longname);
434 xfree(s[i]);
435 }
436 xfree(s);
437}
438
439int
Damien Miller3db5f532002-02-13 14:10:32 +1100440do_rm(struct sftp_conn *conn, char *path)
Damien Miller33804262001-02-04 23:20:18 +1100441{
442 u_int status, id;
443
444 debug2("Sending SSH2_FXP_REMOVE \"%s\"", path);
445
Damien Miller3db5f532002-02-13 14:10:32 +1100446 id = conn->msg_id++;
Ben Lindstrom6328ab32002-03-22 02:54:23 +0000447 send_string_request(conn->fd_out, id, SSH2_FXP_REMOVE, path,
Damien Miller3db5f532002-02-13 14:10:32 +1100448 strlen(path));
449 status = get_status(conn->fd_in, id);
Damien Miller33804262001-02-04 23:20:18 +1100450 if (status != SSH2_FX_OK)
451 error("Couldn't delete file: %s", fx2txt(status));
452 return(status);
453}
454
455int
Damien Miller3db5f532002-02-13 14:10:32 +1100456do_mkdir(struct sftp_conn *conn, char *path, Attrib *a)
Damien Miller33804262001-02-04 23:20:18 +1100457{
458 u_int status, id;
459
Damien Miller3db5f532002-02-13 14:10:32 +1100460 id = conn->msg_id++;
461 send_string_attrs_request(conn->fd_out, id, SSH2_FXP_MKDIR, path,
Damien Miller33804262001-02-04 23:20:18 +1100462 strlen(path), a);
463
Damien Miller3db5f532002-02-13 14:10:32 +1100464 status = get_status(conn->fd_in, id);
Damien Miller33804262001-02-04 23:20:18 +1100465 if (status != SSH2_FX_OK)
466 error("Couldn't create directory: %s", fx2txt(status));
467
468 return(status);
469}
470
471int
Damien Miller3db5f532002-02-13 14:10:32 +1100472do_rmdir(struct sftp_conn *conn, char *path)
Damien Miller33804262001-02-04 23:20:18 +1100473{
474 u_int status, id;
475
Damien Miller3db5f532002-02-13 14:10:32 +1100476 id = conn->msg_id++;
477 send_string_request(conn->fd_out, id, SSH2_FXP_RMDIR, path,
478 strlen(path));
Damien Miller33804262001-02-04 23:20:18 +1100479
Damien Miller3db5f532002-02-13 14:10:32 +1100480 status = get_status(conn->fd_in, id);
Damien Miller33804262001-02-04 23:20:18 +1100481 if (status != SSH2_FX_OK)
482 error("Couldn't remove directory: %s", fx2txt(status));
483
484 return(status);
485}
486
487Attrib *
Damien Miller3db5f532002-02-13 14:10:32 +1100488do_stat(struct sftp_conn *conn, char *path, int quiet)
Damien Miller33804262001-02-04 23:20:18 +1100489{
490 u_int id;
491
Damien Miller3db5f532002-02-13 14:10:32 +1100492 id = conn->msg_id++;
493
Ben Lindstrom6328ab32002-03-22 02:54:23 +0000494 send_string_request(conn->fd_out, id,
495 conn->version == 0 ? SSH2_FXP_STAT_VERSION_0 : SSH2_FXP_STAT,
Damien Miller3db5f532002-02-13 14:10:32 +1100496 path, strlen(path));
497
498 return(get_decode_stat(conn->fd_in, id, quiet));
Damien Miller33804262001-02-04 23:20:18 +1100499}
500
501Attrib *
Damien Miller3db5f532002-02-13 14:10:32 +1100502do_lstat(struct sftp_conn *conn, char *path, int quiet)
Damien Miller33804262001-02-04 23:20:18 +1100503{
504 u_int id;
505
Damien Miller3db5f532002-02-13 14:10:32 +1100506 if (conn->version == 0) {
507 if (quiet)
508 debug("Server version does not support lstat operation");
509 else
Damien Miller996acd22003-04-09 20:59:48 +1000510 logit("Server version does not support lstat operation");
Ben Lindstromf26ff5b2002-04-02 21:00:31 +0000511 return(do_stat(conn, path, quiet));
Damien Miller3db5f532002-02-13 14:10:32 +1100512 }
513
514 id = conn->msg_id++;
515 send_string_request(conn->fd_out, id, SSH2_FXP_LSTAT, path,
516 strlen(path));
517
518 return(get_decode_stat(conn->fd_in, id, quiet));
Damien Miller33804262001-02-04 23:20:18 +1100519}
520
521Attrib *
Damien Miller3db5f532002-02-13 14:10:32 +1100522do_fstat(struct sftp_conn *conn, char *handle, u_int handle_len, int quiet)
Damien Miller33804262001-02-04 23:20:18 +1100523{
524 u_int id;
525
Damien Miller3db5f532002-02-13 14:10:32 +1100526 id = conn->msg_id++;
527 send_string_request(conn->fd_out, id, SSH2_FXP_FSTAT, handle,
528 handle_len);
529
530 return(get_decode_stat(conn->fd_in, id, quiet));
Damien Miller33804262001-02-04 23:20:18 +1100531}
532
533int
Damien Miller3db5f532002-02-13 14:10:32 +1100534do_setstat(struct sftp_conn *conn, char *path, Attrib *a)
Damien Miller33804262001-02-04 23:20:18 +1100535{
536 u_int status, id;
537
Damien Miller3db5f532002-02-13 14:10:32 +1100538 id = conn->msg_id++;
539 send_string_attrs_request(conn->fd_out, id, SSH2_FXP_SETSTAT, path,
Damien Miller33804262001-02-04 23:20:18 +1100540 strlen(path), a);
541
Damien Miller3db5f532002-02-13 14:10:32 +1100542 status = get_status(conn->fd_in, id);
Damien Miller33804262001-02-04 23:20:18 +1100543 if (status != SSH2_FX_OK)
544 error("Couldn't setstat on \"%s\": %s", path,
545 fx2txt(status));
546
547 return(status);
548}
549
550int
Damien Miller3db5f532002-02-13 14:10:32 +1100551do_fsetstat(struct sftp_conn *conn, char *handle, u_int handle_len,
Damien Miller33804262001-02-04 23:20:18 +1100552 Attrib *a)
553{
554 u_int status, id;
555
Damien Miller3db5f532002-02-13 14:10:32 +1100556 id = conn->msg_id++;
557 send_string_attrs_request(conn->fd_out, id, SSH2_FXP_FSETSTAT, handle,
Damien Miller33804262001-02-04 23:20:18 +1100558 handle_len, a);
559
Damien Miller3db5f532002-02-13 14:10:32 +1100560 status = get_status(conn->fd_in, id);
Damien Miller33804262001-02-04 23:20:18 +1100561 if (status != SSH2_FX_OK)
562 error("Couldn't fsetstat: %s", fx2txt(status));
563
564 return(status);
565}
566
567char *
Damien Miller3db5f532002-02-13 14:10:32 +1100568do_realpath(struct sftp_conn *conn, char *path)
Damien Miller33804262001-02-04 23:20:18 +1100569{
570 Buffer msg;
571 u_int type, expected_id, count, id;
572 char *filename, *longname;
573 Attrib *a;
574
Damien Miller3db5f532002-02-13 14:10:32 +1100575 expected_id = id = conn->msg_id++;
576 send_string_request(conn->fd_out, id, SSH2_FXP_REALPATH, path,
577 strlen(path));
Damien Miller33804262001-02-04 23:20:18 +1100578
579 buffer_init(&msg);
580
Damien Miller3db5f532002-02-13 14:10:32 +1100581 get_msg(conn->fd_in, &msg);
Damien Miller33804262001-02-04 23:20:18 +1100582 type = buffer_get_char(&msg);
583 id = buffer_get_int(&msg);
584
585 if (id != expected_id)
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000586 fatal("ID mismatch (%u != %u)", id, expected_id);
Damien Miller33804262001-02-04 23:20:18 +1100587
588 if (type == SSH2_FXP_STATUS) {
589 u_int status = buffer_get_int(&msg);
590
591 error("Couldn't canonicalise: %s", fx2txt(status));
592 return(NULL);
593 } else if (type != SSH2_FXP_NAME)
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000594 fatal("Expected SSH2_FXP_NAME(%u) packet, got %u",
Damien Miller33804262001-02-04 23:20:18 +1100595 SSH2_FXP_NAME, type);
596
597 count = buffer_get_int(&msg);
598 if (count != 1)
599 fatal("Got multiple names (%d) from SSH_FXP_REALPATH", count);
600
601 filename = buffer_get_string(&msg, NULL);
602 longname = buffer_get_string(&msg, NULL);
603 a = decode_attrib(&msg);
604
605 debug3("SSH_FXP_REALPATH %s -> %s", path, filename);
606
607 xfree(longname);
608
609 buffer_free(&msg);
610
611 return(filename);
612}
613
614int
Damien Miller3db5f532002-02-13 14:10:32 +1100615do_rename(struct sftp_conn *conn, char *oldpath, char *newpath)
Damien Miller33804262001-02-04 23:20:18 +1100616{
617 Buffer msg;
618 u_int status, id;
619
620 buffer_init(&msg);
621
622 /* Send rename request */
Damien Miller3db5f532002-02-13 14:10:32 +1100623 id = conn->msg_id++;
Damien Miller33804262001-02-04 23:20:18 +1100624 buffer_put_char(&msg, SSH2_FXP_RENAME);
625 buffer_put_int(&msg, id);
626 buffer_put_cstring(&msg, oldpath);
627 buffer_put_cstring(&msg, newpath);
Damien Miller3db5f532002-02-13 14:10:32 +1100628 send_msg(conn->fd_out, &msg);
Damien Miller33804262001-02-04 23:20:18 +1100629 debug3("Sent message SSH2_FXP_RENAME \"%s\" -> \"%s\"", oldpath,
630 newpath);
631 buffer_free(&msg);
632
Damien Miller3db5f532002-02-13 14:10:32 +1100633 status = get_status(conn->fd_in, id);
Damien Miller33804262001-02-04 23:20:18 +1100634 if (status != SSH2_FX_OK)
Damien Miller3db5f532002-02-13 14:10:32 +1100635 error("Couldn't rename file \"%s\" to \"%s\": %s", oldpath,
636 newpath, fx2txt(status));
Damien Miller33804262001-02-04 23:20:18 +1100637
638 return(status);
639}
640
641int
Damien Miller3db5f532002-02-13 14:10:32 +1100642do_symlink(struct sftp_conn *conn, char *oldpath, char *newpath)
Damien Miller058316f2001-03-08 10:08:49 +1100643{
644 Buffer msg;
645 u_int status, id;
646
Damien Miller3db5f532002-02-13 14:10:32 +1100647 if (conn->version < 3) {
648 error("This server does not support the symlink operation");
649 return(SSH2_FX_OP_UNSUPPORTED);
650 }
651
Damien Miller058316f2001-03-08 10:08:49 +1100652 buffer_init(&msg);
653
Darren Tuckerdca6a4d2004-04-19 22:10:52 +1000654 /* Send symlink request */
Damien Miller3db5f532002-02-13 14:10:32 +1100655 id = conn->msg_id++;
Damien Miller058316f2001-03-08 10:08:49 +1100656 buffer_put_char(&msg, SSH2_FXP_SYMLINK);
657 buffer_put_int(&msg, id);
658 buffer_put_cstring(&msg, oldpath);
659 buffer_put_cstring(&msg, newpath);
Damien Miller3db5f532002-02-13 14:10:32 +1100660 send_msg(conn->fd_out, &msg);
Damien Miller058316f2001-03-08 10:08:49 +1100661 debug3("Sent message SSH2_FXP_SYMLINK \"%s\" -> \"%s\"", oldpath,
662 newpath);
663 buffer_free(&msg);
664
Damien Miller3db5f532002-02-13 14:10:32 +1100665 status = get_status(conn->fd_in, id);
Damien Miller058316f2001-03-08 10:08:49 +1100666 if (status != SSH2_FX_OK)
Ben Lindstrom8e879cf2002-11-09 15:48:49 +0000667 error("Couldn't symlink file \"%s\" to \"%s\": %s", oldpath,
Damien Miller3db5f532002-02-13 14:10:32 +1100668 newpath, fx2txt(status));
Damien Miller058316f2001-03-08 10:08:49 +1100669
670 return(status);
671}
672
673char *
Damien Miller3db5f532002-02-13 14:10:32 +1100674do_readlink(struct sftp_conn *conn, char *path)
Damien Miller058316f2001-03-08 10:08:49 +1100675{
676 Buffer msg;
677 u_int type, expected_id, count, id;
678 char *filename, *longname;
679 Attrib *a;
680
Damien Miller3db5f532002-02-13 14:10:32 +1100681 expected_id = id = conn->msg_id++;
682 send_string_request(conn->fd_out, id, SSH2_FXP_READLINK, path,
683 strlen(path));
Damien Miller058316f2001-03-08 10:08:49 +1100684
685 buffer_init(&msg);
686
Damien Miller3db5f532002-02-13 14:10:32 +1100687 get_msg(conn->fd_in, &msg);
Damien Miller058316f2001-03-08 10:08:49 +1100688 type = buffer_get_char(&msg);
689 id = buffer_get_int(&msg);
690
691 if (id != expected_id)
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000692 fatal("ID mismatch (%u != %u)", id, expected_id);
Damien Miller058316f2001-03-08 10:08:49 +1100693
694 if (type == SSH2_FXP_STATUS) {
695 u_int status = buffer_get_int(&msg);
696
697 error("Couldn't readlink: %s", fx2txt(status));
698 return(NULL);
699 } else if (type != SSH2_FXP_NAME)
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000700 fatal("Expected SSH2_FXP_NAME(%u) packet, got %u",
Damien Miller058316f2001-03-08 10:08:49 +1100701 SSH2_FXP_NAME, type);
702
703 count = buffer_get_int(&msg);
704 if (count != 1)
705 fatal("Got multiple names (%d) from SSH_FXP_READLINK", count);
706
707 filename = buffer_get_string(&msg, NULL);
708 longname = buffer_get_string(&msg, NULL);
709 a = decode_attrib(&msg);
710
711 debug3("SSH_FXP_READLINK %s -> %s", path, filename);
712
713 xfree(longname);
714
715 buffer_free(&msg);
716
717 return(filename);
718}
719
Damien Miller16a13332002-02-13 14:03:56 +1100720static void
721send_read_request(int fd_out, u_int id, u_int64_t offset, u_int len,
722 char *handle, u_int handle_len)
723{
724 Buffer msg;
Ben Lindstrom6328ab32002-03-22 02:54:23 +0000725
Damien Miller16a13332002-02-13 14:03:56 +1100726 buffer_init(&msg);
727 buffer_clear(&msg);
728 buffer_put_char(&msg, SSH2_FXP_READ);
729 buffer_put_int(&msg, id);
730 buffer_put_string(&msg, handle, handle_len);
731 buffer_put_int64(&msg, offset);
732 buffer_put_int(&msg, len);
733 send_msg(fd_out, &msg);
734 buffer_free(&msg);
Ben Lindstrom6328ab32002-03-22 02:54:23 +0000735}
Damien Miller16a13332002-02-13 14:03:56 +1100736
Damien Miller058316f2001-03-08 10:08:49 +1100737int
Damien Miller3db5f532002-02-13 14:10:32 +1100738do_download(struct sftp_conn *conn, char *remote_path, char *local_path,
739 int pflag)
Damien Miller33804262001-02-04 23:20:18 +1100740{
Damien Miller33804262001-02-04 23:20:18 +1100741 Attrib junk, *a;
Damien Miller16a13332002-02-13 14:03:56 +1100742 Buffer msg;
743 char *handle;
744 int local_fd, status, num_req, max_req, write_error;
745 int read_error, write_errno;
746 u_int64_t offset, size;
Damien Miller3db5f532002-02-13 14:10:32 +1100747 u_int handle_len, mode, type, id, buflen;
Damien Miller62d57f62003-01-10 21:43:24 +1100748 off_t progress_counter;
Damien Miller16a13332002-02-13 14:03:56 +1100749 struct request {
750 u_int id;
751 u_int len;
752 u_int64_t offset;
Ben Lindstrom6328ab32002-03-22 02:54:23 +0000753 TAILQ_ENTRY(request) tq;
Damien Miller16a13332002-02-13 14:03:56 +1100754 };
755 TAILQ_HEAD(reqhead, request) requests;
756 struct request *req;
757
758 TAILQ_INIT(&requests);
Damien Miller33804262001-02-04 23:20:18 +1100759
Damien Miller3db5f532002-02-13 14:10:32 +1100760 a = do_stat(conn, remote_path, 0);
Damien Miller33804262001-02-04 23:20:18 +1100761 if (a == NULL)
762 return(-1);
763
764 /* XXX: should we preserve set[ug]id? */
765 if (a->flags & SSH2_FILEXFER_ATTR_PERMISSIONS)
Damien Miller770b3742003-01-08 14:04:53 +1100766 mode = a->perm & 0777;
Damien Miller33804262001-02-04 23:20:18 +1100767 else
768 mode = 0666;
769
Ben Lindstromc8d1c302001-03-17 00:34:46 +0000770 if ((a->flags & SSH2_FILEXFER_ATTR_PERMISSIONS) &&
Damien Miller5fa01fd2003-01-14 22:24:47 +1100771 (!S_ISREG(a->perm))) {
772 error("Cannot download non-regular file: %s", remote_path);
Ben Lindstromc8d1c302001-03-17 00:34:46 +0000773 return(-1);
774 }
775
Damien Miller16a13332002-02-13 14:03:56 +1100776 if (a->flags & SSH2_FILEXFER_ATTR_SIZE)
777 size = a->size;
778 else
779 size = 0;
780
Damien Miller3db5f532002-02-13 14:10:32 +1100781 buflen = conn->transfer_buflen;
Damien Miller33804262001-02-04 23:20:18 +1100782 buffer_init(&msg);
783
784 /* Send open request */
Damien Miller3db5f532002-02-13 14:10:32 +1100785 id = conn->msg_id++;
Damien Miller33804262001-02-04 23:20:18 +1100786 buffer_put_char(&msg, SSH2_FXP_OPEN);
787 buffer_put_int(&msg, id);
788 buffer_put_cstring(&msg, remote_path);
789 buffer_put_int(&msg, SSH2_FXF_READ);
790 attrib_clear(&junk); /* Send empty attributes */
791 encode_attrib(&msg, &junk);
Damien Miller3db5f532002-02-13 14:10:32 +1100792 send_msg(conn->fd_out, &msg);
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000793 debug3("Sent message SSH2_FXP_OPEN I:%u P:%s", id, remote_path);
Damien Miller33804262001-02-04 23:20:18 +1100794
Damien Miller3db5f532002-02-13 14:10:32 +1100795 handle = get_handle(conn->fd_in, id, &handle_len);
Damien Miller33804262001-02-04 23:20:18 +1100796 if (handle == NULL) {
797 buffer_free(&msg);
Damien Miller33804262001-02-04 23:20:18 +1100798 return(-1);
799 }
800
Damien Millera8e06ce2003-11-21 23:48:55 +1100801 local_fd = open(local_path, O_WRONLY | O_CREAT | O_TRUNC,
Damien Miller770b3742003-01-08 14:04:53 +1100802 mode | S_IWRITE);
Damien Miller3db5f532002-02-13 14:10:32 +1100803 if (local_fd == -1) {
804 error("Couldn't open local file \"%s\" for writing: %s",
805 local_path, strerror(errno));
Ben Lindstrom021fcd32002-02-26 18:02:43 +0000806 buffer_free(&msg);
807 xfree(handle);
Damien Miller3db5f532002-02-13 14:10:32 +1100808 return(-1);
809 }
810
Damien Miller33804262001-02-04 23:20:18 +1100811 /* Read from remote and write to local */
Damien Miller16a13332002-02-13 14:03:56 +1100812 write_error = read_error = write_errno = num_req = offset = 0;
813 max_req = 1;
Damien Miller62d57f62003-01-10 21:43:24 +1100814 progress_counter = 0;
815
Damien Miller9ba30692004-03-08 23:12:02 +1100816 if (showprogress && size != 0)
817 start_progress_meter(remote_path, size, &progress_counter);
Damien Miller62d57f62003-01-10 21:43:24 +1100818
Damien Miller16a13332002-02-13 14:03:56 +1100819 while (num_req > 0 || max_req > 0) {
Damien Miller33804262001-02-04 23:20:18 +1100820 char *data;
Damien Miller16a13332002-02-13 14:03:56 +1100821 u_int len;
Damien Miller33804262001-02-04 23:20:18 +1100822
Darren Tuckercdf547a2004-05-24 10:12:19 +1000823 /*
824 * Simulate EOF on interrupt: stop sending new requests and
825 * allow outstanding requests to drain gracefully
826 */
827 if (interrupted) {
828 if (num_req == 0) /* If we haven't started yet... */
829 break;
830 max_req = 0;
831 }
832
Damien Miller16a13332002-02-13 14:03:56 +1100833 /* Send some more requests */
834 while (num_req < max_req) {
Ben Lindstrom6328ab32002-03-22 02:54:23 +0000835 debug3("Request range %llu -> %llu (%d/%d)",
Ben Lindstromd45f28c2002-03-22 01:00:57 +0000836 (unsigned long long)offset,
837 (unsigned long long)offset + buflen - 1,
838 num_req, max_req);
Damien Miller16a13332002-02-13 14:03:56 +1100839 req = xmalloc(sizeof(*req));
Damien Miller3db5f532002-02-13 14:10:32 +1100840 req->id = conn->msg_id++;
Damien Miller16a13332002-02-13 14:03:56 +1100841 req->len = buflen;
842 req->offset = offset;
843 offset += buflen;
844 num_req++;
845 TAILQ_INSERT_TAIL(&requests, req, tq);
Ben Lindstrom6328ab32002-03-22 02:54:23 +0000846 send_read_request(conn->fd_out, req->id, req->offset,
Damien Miller16a13332002-02-13 14:03:56 +1100847 req->len, handle, handle_len);
848 }
Damien Miller33804262001-02-04 23:20:18 +1100849
850 buffer_clear(&msg);
Damien Miller3db5f532002-02-13 14:10:32 +1100851 get_msg(conn->fd_in, &msg);
Damien Miller33804262001-02-04 23:20:18 +1100852 type = buffer_get_char(&msg);
853 id = buffer_get_int(&msg);
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000854 debug3("Received reply T:%u I:%u R:%d", type, id, max_req);
Damien Miller33804262001-02-04 23:20:18 +1100855
Damien Miller16a13332002-02-13 14:03:56 +1100856 /* Find the request in our queue */
857 for(req = TAILQ_FIRST(&requests);
858 req != NULL && req->id != id;
859 req = TAILQ_NEXT(req, tq))
860 ;
861 if (req == NULL)
862 fatal("Unexpected reply %u", id);
863
864 switch (type) {
865 case SSH2_FXP_STATUS:
866 status = buffer_get_int(&msg);
867 if (status != SSH2_FX_EOF)
868 read_error = 1;
869 max_req = 0;
870 TAILQ_REMOVE(&requests, req, tq);
871 xfree(req);
872 num_req--;
873 break;
874 case SSH2_FXP_DATA:
875 data = buffer_get_string(&msg, &len);
Ben Lindstromeb505452002-03-22 01:03:15 +0000876 debug3("Received data %llu -> %llu",
Ben Lindstrom6328ab32002-03-22 02:54:23 +0000877 (unsigned long long)req->offset,
Ben Lindstromeb505452002-03-22 01:03:15 +0000878 (unsigned long long)req->offset + len - 1);
Damien Miller16a13332002-02-13 14:03:56 +1100879 if (len > req->len)
880 fatal("Received more data than asked for "
Ben Lindstrom93576d92002-12-23 02:06:19 +0000881 "%u > %u", len, req->len);
Damien Miller16a13332002-02-13 14:03:56 +1100882 if ((lseek(local_fd, req->offset, SEEK_SET) == -1 ||
Darren Tucker9f63f222003-07-03 13:46:56 +1000883 atomicio(vwrite, local_fd, data, len) != len) &&
Damien Miller16a13332002-02-13 14:03:56 +1100884 !write_error) {
885 write_errno = errno;
886 write_error = 1;
887 max_req = 0;
Damien Miller33804262001-02-04 23:20:18 +1100888 }
Damien Miller62d57f62003-01-10 21:43:24 +1100889 progress_counter += len;
Damien Miller16a13332002-02-13 14:03:56 +1100890 xfree(data);
891
892 if (len == req->len) {
893 TAILQ_REMOVE(&requests, req, tq);
894 xfree(req);
895 num_req--;
896 } else {
897 /* Resend the request for the missing data */
898 debug3("Short data block, re-requesting "
Ben Lindstromeb505452002-03-22 01:03:15 +0000899 "%llu -> %llu (%2d)",
Ben Lindstrom6328ab32002-03-22 02:54:23 +0000900 (unsigned long long)req->offset + len,
Ben Lindstrom83b79e42002-03-22 01:05:27 +0000901 (unsigned long long)req->offset +
902 req->len - 1, num_req);
Damien Miller3db5f532002-02-13 14:10:32 +1100903 req->id = conn->msg_id++;
Damien Miller16a13332002-02-13 14:03:56 +1100904 req->len -= len;
905 req->offset += len;
Ben Lindstrom6328ab32002-03-22 02:54:23 +0000906 send_read_request(conn->fd_out, req->id,
Damien Miller3db5f532002-02-13 14:10:32 +1100907 req->offset, req->len, handle, handle_len);
Damien Miller16a13332002-02-13 14:03:56 +1100908 /* Reduce the request size */
909 if (len < buflen)
910 buflen = MAX(MIN_READ_SIZE, len);
911 }
912 if (max_req > 0) { /* max_req = 0 iff EOF received */
913 if (size > 0 && offset > size) {
914 /* Only one request at a time
915 * after the expected EOF */
916 debug3("Finish at %llu (%2d)",
Ben Lindstromeb505452002-03-22 01:03:15 +0000917 (unsigned long long)offset,
918 num_req);
Damien Miller16a13332002-02-13 14:03:56 +1100919 max_req = 1;
Darren Tuckercdf547a2004-05-24 10:12:19 +1000920 } else if (max_req <= conn->num_requests) {
Damien Miller16a13332002-02-13 14:03:56 +1100921 ++max_req;
922 }
923 }
924 break;
925 default:
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000926 fatal("Expected SSH2_FXP_DATA(%u) packet, got %u",
Damien Miller33804262001-02-04 23:20:18 +1100927 SSH2_FXP_DATA, type);
928 }
Damien Miller33804262001-02-04 23:20:18 +1100929 }
Damien Miller33804262001-02-04 23:20:18 +1100930
Damien Miller62d57f62003-01-10 21:43:24 +1100931 if (showprogress && size)
932 stop_progress_meter();
933
Damien Miller16a13332002-02-13 14:03:56 +1100934 /* Sanity check */
935 if (TAILQ_FIRST(&requests) != NULL)
936 fatal("Transfer complete, but requests still in queue");
937
938 if (read_error) {
Ben Lindstrom6328ab32002-03-22 02:54:23 +0000939 error("Couldn't read from remote file \"%s\" : %s",
Damien Miller3db5f532002-02-13 14:10:32 +1100940 remote_path, fx2txt(status));
941 do_close(conn, handle, handle_len);
Damien Miller16a13332002-02-13 14:03:56 +1100942 } else if (write_error) {
Damien Miller3db5f532002-02-13 14:10:32 +1100943 error("Couldn't write to \"%s\": %s", local_path,
944 strerror(write_errno));
945 status = -1;
946 do_close(conn, handle, handle_len);
Damien Miller16a13332002-02-13 14:03:56 +1100947 } else {
Damien Miller3db5f532002-02-13 14:10:32 +1100948 status = do_close(conn, handle, handle_len);
Damien Miller16a13332002-02-13 14:03:56 +1100949
950 /* Override umask and utimes if asked */
Ben Lindstrom6dc75f52001-02-17 16:47:47 +0000951#ifdef HAVE_FCHMOD
Damien Miller16a13332002-02-13 14:03:56 +1100952 if (pflag && fchmod(local_fd, mode) == -1)
Damien Millera8e06ce2003-11-21 23:48:55 +1100953#else
Damien Miller16a13332002-02-13 14:03:56 +1100954 if (pflag && chmod(local_path, mode) == -1)
Ben Lindstrom6dc75f52001-02-17 16:47:47 +0000955#endif /* HAVE_FCHMOD */
Damien Miller16a13332002-02-13 14:03:56 +1100956 error("Couldn't set mode on \"%s\": %s", local_path,
Ben Lindstrom93576d92002-12-23 02:06:19 +0000957 strerror(errno));
Damien Miller16a13332002-02-13 14:03:56 +1100958 if (pflag && (a->flags & SSH2_FILEXFER_ATTR_ACMODTIME)) {
959 struct timeval tv[2];
960 tv[0].tv_sec = a->atime;
961 tv[1].tv_sec = a->mtime;
962 tv[0].tv_usec = tv[1].tv_usec = 0;
963 if (utimes(local_path, tv) == -1)
964 error("Can't set times on \"%s\": %s",
Ben Lindstrom93576d92002-12-23 02:06:19 +0000965 local_path, strerror(errno));
Damien Miller16a13332002-02-13 14:03:56 +1100966 }
Ben Lindstrom9d4f2c82001-02-15 03:22:45 +0000967 }
Damien Millerd7686fd2001-02-10 00:40:03 +1100968 close(local_fd);
969 buffer_free(&msg);
970 xfree(handle);
Damien Miller3db5f532002-02-13 14:10:32 +1100971
972 return(status);
Damien Miller33804262001-02-04 23:20:18 +1100973}
974
975int
Damien Miller3db5f532002-02-13 14:10:32 +1100976do_upload(struct sftp_conn *conn, char *local_path, char *remote_path,
977 int pflag)
Damien Miller33804262001-02-04 23:20:18 +1100978{
Damien Miller8829d362002-02-08 22:04:05 +1100979 int local_fd, status;
Damien Miller5873dfd2002-02-13 14:04:37 +1100980 u_int handle_len, id, type;
Damien Miller33804262001-02-04 23:20:18 +1100981 u_int64_t offset;
Damien Miller8829d362002-02-08 22:04:05 +1100982 char *handle, *data;
Damien Miller33804262001-02-04 23:20:18 +1100983 Buffer msg;
984 struct stat sb;
985 Attrib a;
Damien Miller16a13332002-02-13 14:03:56 +1100986 u_int32_t startid;
987 u_int32_t ackid;
Damien Miller5873dfd2002-02-13 14:04:37 +1100988 struct outstanding_ack {
989 u_int id;
990 u_int len;
991 u_int64_t offset;
Ben Lindstrom6328ab32002-03-22 02:54:23 +0000992 TAILQ_ENTRY(outstanding_ack) tq;
Damien Miller5873dfd2002-02-13 14:04:37 +1100993 };
994 TAILQ_HEAD(ackhead, outstanding_ack) acks;
995 struct outstanding_ack *ack;
996
997 TAILQ_INIT(&acks);
Damien Miller33804262001-02-04 23:20:18 +1100998
999 if ((local_fd = open(local_path, O_RDONLY, 0)) == -1) {
1000 error("Couldn't open local file \"%s\" for reading: %s",
1001 local_path, strerror(errno));
1002 return(-1);
1003 }
1004 if (fstat(local_fd, &sb) == -1) {
1005 error("Couldn't fstat local file \"%s\": %s",
1006 local_path, strerror(errno));
1007 close(local_fd);
1008 return(-1);
1009 }
Damien Miller5fa01fd2003-01-14 22:24:47 +11001010 if (!S_ISREG(sb.st_mode)) {
1011 error("%s is not a regular file", local_path);
1012 close(local_fd);
1013 return(-1);
1014 }
Damien Miller33804262001-02-04 23:20:18 +11001015 stat_to_attrib(&sb, &a);
1016
1017 a.flags &= ~SSH2_FILEXFER_ATTR_SIZE;
1018 a.flags &= ~SSH2_FILEXFER_ATTR_UIDGID;
1019 a.perm &= 0777;
1020 if (!pflag)
1021 a.flags &= ~SSH2_FILEXFER_ATTR_ACMODTIME;
1022
1023 buffer_init(&msg);
1024
1025 /* Send open request */
Damien Miller3db5f532002-02-13 14:10:32 +11001026 id = conn->msg_id++;
Damien Miller33804262001-02-04 23:20:18 +11001027 buffer_put_char(&msg, SSH2_FXP_OPEN);
1028 buffer_put_int(&msg, id);
1029 buffer_put_cstring(&msg, remote_path);
1030 buffer_put_int(&msg, SSH2_FXF_WRITE|SSH2_FXF_CREAT|SSH2_FXF_TRUNC);
1031 encode_attrib(&msg, &a);
Damien Miller3db5f532002-02-13 14:10:32 +11001032 send_msg(conn->fd_out, &msg);
Ben Lindstromb1f483f2002-06-23 21:27:18 +00001033 debug3("Sent message SSH2_FXP_OPEN I:%u P:%s", id, remote_path);
Damien Miller33804262001-02-04 23:20:18 +11001034
1035 buffer_clear(&msg);
1036
Damien Miller3db5f532002-02-13 14:10:32 +11001037 handle = get_handle(conn->fd_in, id, &handle_len);
Damien Miller33804262001-02-04 23:20:18 +11001038 if (handle == NULL) {
1039 close(local_fd);
1040 buffer_free(&msg);
1041 return(-1);
1042 }
1043
Damien Miller16a13332002-02-13 14:03:56 +11001044 startid = ackid = id + 1;
Damien Miller3db5f532002-02-13 14:10:32 +11001045 data = xmalloc(conn->transfer_buflen);
Damien Miller8829d362002-02-08 22:04:05 +11001046
Damien Miller33804262001-02-04 23:20:18 +11001047 /* Read from local and write to remote */
1048 offset = 0;
Damien Miller62d57f62003-01-10 21:43:24 +11001049 if (showprogress)
1050 start_progress_meter(local_path, sb.st_size, &offset);
Damien Miller62d57f62003-01-10 21:43:24 +11001051
Damien Miller9f0f5c62001-12-21 14:45:46 +11001052 for (;;) {
Damien Miller33804262001-02-04 23:20:18 +11001053 int len;
Damien Miller33804262001-02-04 23:20:18 +11001054
1055 /*
Darren Tuckercdf547a2004-05-24 10:12:19 +10001056 * Can't use atomicio here because it returns 0 on EOF,
1057 * thus losing the last block of the file.
1058 * Simulate an EOF on interrupt, allowing ACKs from the
1059 * server to drain.
Damien Miller33804262001-02-04 23:20:18 +11001060 */
Darren Tuckercdf547a2004-05-24 10:12:19 +10001061 if (interrupted)
1062 len = 0;
1063 else do
Damien Miller3db5f532002-02-13 14:10:32 +11001064 len = read(local_fd, data, conn->transfer_buflen);
Damien Miller33804262001-02-04 23:20:18 +11001065 while ((len == -1) && (errno == EINTR || errno == EAGAIN));
1066
1067 if (len == -1)
1068 fatal("Couldn't read from \"%s\": %s", local_path,
1069 strerror(errno));
Damien Miller16a13332002-02-13 14:03:56 +11001070
1071 if (len != 0) {
Damien Miller5873dfd2002-02-13 14:04:37 +11001072 ack = xmalloc(sizeof(*ack));
1073 ack->id = ++id;
1074 ack->offset = offset;
1075 ack->len = len;
1076 TAILQ_INSERT_TAIL(&acks, ack, tq);
1077
Damien Miller16a13332002-02-13 14:03:56 +11001078 buffer_clear(&msg);
1079 buffer_put_char(&msg, SSH2_FXP_WRITE);
Damien Miller5873dfd2002-02-13 14:04:37 +11001080 buffer_put_int(&msg, ack->id);
Damien Miller16a13332002-02-13 14:03:56 +11001081 buffer_put_string(&msg, handle, handle_len);
1082 buffer_put_int64(&msg, offset);
1083 buffer_put_string(&msg, data, len);
Damien Miller3db5f532002-02-13 14:10:32 +11001084 send_msg(conn->fd_out, &msg);
Ben Lindstromb1f483f2002-06-23 21:27:18 +00001085 debug3("Sent message SSH2_FXP_WRITE I:%u O:%llu S:%u",
Ben Lindstrom93576d92002-12-23 02:06:19 +00001086 id, (unsigned long long)offset, len);
Damien Miller5873dfd2002-02-13 14:04:37 +11001087 } else if (TAILQ_FIRST(&acks) == NULL)
Damien Miller33804262001-02-04 23:20:18 +11001088 break;
1089
Damien Miller5873dfd2002-02-13 14:04:37 +11001090 if (ack == NULL)
1091 fatal("Unexpected ACK %u", id);
1092
Ben Lindstrom6328ab32002-03-22 02:54:23 +00001093 if (id == startid || len == 0 ||
Damien Miller3db5f532002-02-13 14:10:32 +11001094 id - ackid >= conn->num_requests) {
Ben Lindstrom5a6abda2002-06-09 19:41:48 +00001095 u_int r_id;
Ben Lindstrom06e95152002-04-06 04:16:45 +00001096
Damien Miller5873dfd2002-02-13 14:04:37 +11001097 buffer_clear(&msg);
Damien Miller3db5f532002-02-13 14:10:32 +11001098 get_msg(conn->fd_in, &msg);
Damien Miller5873dfd2002-02-13 14:04:37 +11001099 type = buffer_get_char(&msg);
Ben Lindstrom06e95152002-04-06 04:16:45 +00001100 r_id = buffer_get_int(&msg);
Damien Miller5873dfd2002-02-13 14:04:37 +11001101
1102 if (type != SSH2_FXP_STATUS)
1103 fatal("Expected SSH2_FXP_STATUS(%d) packet, "
1104 "got %d", SSH2_FXP_STATUS, type);
1105
1106 status = buffer_get_int(&msg);
1107 debug3("SSH2_FXP_STATUS %d", status);
1108
1109 /* Find the request in our queue */
1110 for(ack = TAILQ_FIRST(&acks);
Ben Lindstrom06e95152002-04-06 04:16:45 +00001111 ack != NULL && ack->id != r_id;
Damien Miller5873dfd2002-02-13 14:04:37 +11001112 ack = TAILQ_NEXT(ack, tq))
1113 ;
1114 if (ack == NULL)
Ben Lindstromb1f483f2002-06-23 21:27:18 +00001115 fatal("Can't find request for ID %u", r_id);
Damien Miller5873dfd2002-02-13 14:04:37 +11001116 TAILQ_REMOVE(&acks, ack, tq);
1117
Damien Miller16a13332002-02-13 14:03:56 +11001118 if (status != SSH2_FX_OK) {
1119 error("Couldn't write to remote file \"%s\": %s",
Ben Lindstrom93576d92002-12-23 02:06:19 +00001120 remote_path, fx2txt(status));
Damien Miller3db5f532002-02-13 14:10:32 +11001121 do_close(conn, handle, handle_len);
Damien Miller16a13332002-02-13 14:03:56 +11001122 close(local_fd);
Damien Miller00111382003-03-10 11:21:17 +11001123 xfree(data);
1124 xfree(ack);
Damien Miller16a13332002-02-13 14:03:56 +11001125 goto done;
1126 }
Ben Lindstromb1f483f2002-06-23 21:27:18 +00001127 debug3("In write loop, ack for %u %u bytes at %llu",
Ben Lindstromeb505452002-03-22 01:03:15 +00001128 ack->id, ack->len, (unsigned long long)ack->offset);
Damien Miller16a13332002-02-13 14:03:56 +11001129 ++ackid;
Ben Lindstromeec16fc2002-07-04 00:06:15 +00001130 xfree(ack);
Damien Miller33804262001-02-04 23:20:18 +11001131 }
Damien Miller33804262001-02-04 23:20:18 +11001132 offset += len;
1133 }
Damien Miller62d57f62003-01-10 21:43:24 +11001134 if (showprogress)
1135 stop_progress_meter();
Damien Miller8829d362002-02-08 22:04:05 +11001136 xfree(data);
Damien Miller33804262001-02-04 23:20:18 +11001137
1138 if (close(local_fd) == -1) {
1139 error("Couldn't close local file \"%s\": %s", local_path,
1140 strerror(errno));
Damien Miller3db5f532002-02-13 14:10:32 +11001141 do_close(conn, handle, handle_len);
Damien Millerd7686fd2001-02-10 00:40:03 +11001142 status = -1;
1143 goto done;
Damien Miller33804262001-02-04 23:20:18 +11001144 }
1145
Ben Lindstrom9d4f2c82001-02-15 03:22:45 +00001146 /* Override umask and utimes if asked */
1147 if (pflag)
Damien Miller3db5f532002-02-13 14:10:32 +11001148 do_fsetstat(conn, handle, handle_len, &a);
Ben Lindstrom9d4f2c82001-02-15 03:22:45 +00001149
Damien Miller3db5f532002-02-13 14:10:32 +11001150 status = do_close(conn, handle, handle_len);
Damien Millerd7686fd2001-02-10 00:40:03 +11001151
1152done:
1153 xfree(handle);
1154 buffer_free(&msg);
Damien Miller3db5f532002-02-13 14:10:32 +11001155 return(status);
Damien Miller33804262001-02-04 23:20:18 +11001156}