blob: c34f919a4097494198eb1db4f036617fa00e429e [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"
Damien Millerf17883e2006-03-15 11:45:54 +110023
24#include <sys/types.h>
25#ifdef HAVE_SYS_STAT_H
26# include <sys/stat.h>
27#endif
Damien Miller6645e7a2006-03-15 14:42:54 +110028# include <signal.h>
Damien Miller16a13332002-02-13 14:03:56 +110029
Damien Miller9b481512002-09-12 10:43:29 +100030#include "openbsd-compat/sys-queue.h"
Damien Miller33804262001-02-04 23:20:18 +110031
Damien Miller33804262001-02-04 23:20:18 +110032#include "buffer.h"
33#include "bufaux.h"
34#include "getput.h"
35#include "xmalloc.h"
36#include "log.h"
37#include "atomicio.h"
Damien Miller62d57f62003-01-10 21:43:24 +110038#include "progressmeter.h"
Damien Miller33804262001-02-04 23:20:18 +110039
40#include "sftp.h"
41#include "sftp-common.h"
42#include "sftp-client.h"
43
Darren Tuckercdf547a2004-05-24 10:12:19 +100044extern volatile sig_atomic_t interrupted;
Damien Miller62d57f62003-01-10 21:43:24 +110045extern int showprogress;
46
Damien Miller0c8d8f62006-03-15 11:34:25 +110047/* Minimum amount of data to read at a time */
Damien Miller16a13332002-02-13 14:03:56 +110048#define MIN_READ_SIZE 512
49
Damien Miller3db5f532002-02-13 14:10:32 +110050struct sftp_conn {
51 int fd_in;
52 int fd_out;
53 u_int transfer_buflen;
54 u_int num_requests;
55 u_int version;
56 u_int msg_id;
57};
Ben Lindstrom288cc392001-02-09 02:58:04 +000058
Ben Lindstrombba81212001-06-25 05:01:22 +000059static void
Damien Miller33804262001-02-04 23:20:18 +110060send_msg(int fd, Buffer *m)
61{
Damien Millera7f3aaa2003-01-10 21:43:58 +110062 u_char mlen[4];
Damien Miller33804262001-02-04 23:20:18 +110063
Damien Miller54446182006-01-02 23:40:50 +110064 if (buffer_len(m) > SFTP_MAX_MSG_LENGTH)
Damien Millera7f3aaa2003-01-10 21:43:58 +110065 fatal("Outbound message too long %u", buffer_len(m));
Damien Miller33804262001-02-04 23:20:18 +110066
Damien Millera7f3aaa2003-01-10 21:43:58 +110067 /* Send length first */
68 PUT_32BIT(mlen, buffer_len(m));
Damien Millerb253cc42005-05-26 12:23:44 +100069 if (atomicio(vwrite, fd, mlen, sizeof(mlen)) != sizeof(mlen))
Damien Miller33804262001-02-04 23:20:18 +110070 fatal("Couldn't send packet: %s", strerror(errno));
71
Damien Millerb253cc42005-05-26 12:23:44 +100072 if (atomicio(vwrite, fd, buffer_ptr(m), buffer_len(m)) != buffer_len(m))
Damien Millera7f3aaa2003-01-10 21:43:58 +110073 fatal("Couldn't send packet: %s", strerror(errno));
74
75 buffer_clear(m);
Damien Miller33804262001-02-04 23:20:18 +110076}
77
Ben Lindstrombba81212001-06-25 05:01:22 +000078static void
Damien Miller33804262001-02-04 23:20:18 +110079get_msg(int fd, Buffer *m)
80{
Damien Millera7f3aaa2003-01-10 21:43:58 +110081 u_int msg_len;
Damien Miller33804262001-02-04 23:20:18 +110082
Damien Millera7f3aaa2003-01-10 21:43:58 +110083 buffer_append_space(m, 4);
Damien Millerb253cc42005-05-26 12:23:44 +100084 if (atomicio(read, fd, buffer_ptr(m), 4) != 4) {
85 if (errno == EPIPE)
86 fatal("Connection closed");
87 else
88 fatal("Couldn't read packet: %s", strerror(errno));
89 }
Damien Miller33804262001-02-04 23:20:18 +110090
Damien Millera7f3aaa2003-01-10 21:43:58 +110091 msg_len = buffer_get_int(m);
Damien Miller54446182006-01-02 23:40:50 +110092 if (msg_len > SFTP_MAX_MSG_LENGTH)
Ben Lindstromb1f483f2002-06-23 21:27:18 +000093 fatal("Received message too long %u", msg_len);
Damien Miller33804262001-02-04 23:20:18 +110094
Damien Millera7f3aaa2003-01-10 21:43:58 +110095 buffer_append_space(m, msg_len);
Damien Millerb253cc42005-05-26 12:23:44 +100096 if (atomicio(read, fd, buffer_ptr(m), msg_len) != msg_len) {
97 if (errno == EPIPE)
98 fatal("Connection closed");
99 else
100 fatal("Read packet: %s", strerror(errno));
101 }
Damien Miller33804262001-02-04 23:20:18 +1100102}
103
Ben Lindstrombba81212001-06-25 05:01:22 +0000104static void
Damien Miller33804262001-02-04 23:20:18 +1100105send_string_request(int fd, u_int id, u_int code, char *s,
106 u_int len)
107{
108 Buffer msg;
109
110 buffer_init(&msg);
111 buffer_put_char(&msg, code);
112 buffer_put_int(&msg, id);
113 buffer_put_string(&msg, s, len);
114 send_msg(fd, &msg);
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000115 debug3("Sent message fd %d T:%u I:%u", fd, code, id);
Damien Miller33804262001-02-04 23:20:18 +1100116 buffer_free(&msg);
117}
118
Ben Lindstrombba81212001-06-25 05:01:22 +0000119static void
Damien Miller33804262001-02-04 23:20:18 +1100120send_string_attrs_request(int fd, u_int id, u_int code, char *s,
121 u_int len, Attrib *a)
122{
123 Buffer msg;
124
125 buffer_init(&msg);
126 buffer_put_char(&msg, code);
127 buffer_put_int(&msg, id);
128 buffer_put_string(&msg, s, len);
129 encode_attrib(&msg, a);
130 send_msg(fd, &msg);
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000131 debug3("Sent message fd %d T:%u I:%u", fd, code, id);
Damien Miller33804262001-02-04 23:20:18 +1100132 buffer_free(&msg);
133}
134
Ben Lindstrombba81212001-06-25 05:01:22 +0000135static u_int
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000136get_status(int fd, u_int expected_id)
Damien Miller33804262001-02-04 23:20:18 +1100137{
138 Buffer msg;
139 u_int type, id, status;
140
141 buffer_init(&msg);
142 get_msg(fd, &msg);
143 type = buffer_get_char(&msg);
144 id = buffer_get_int(&msg);
145
146 if (id != expected_id)
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000147 fatal("ID mismatch (%u != %u)", id, expected_id);
Damien Miller33804262001-02-04 23:20:18 +1100148 if (type != SSH2_FXP_STATUS)
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000149 fatal("Expected SSH2_FXP_STATUS(%u) packet, got %u",
Damien Miller33804262001-02-04 23:20:18 +1100150 SSH2_FXP_STATUS, type);
151
152 status = buffer_get_int(&msg);
153 buffer_free(&msg);
154
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000155 debug3("SSH2_FXP_STATUS %u", status);
Damien Miller33804262001-02-04 23:20:18 +1100156
157 return(status);
158}
159
Ben Lindstrombba81212001-06-25 05:01:22 +0000160static char *
Damien Miller33804262001-02-04 23:20:18 +1100161get_handle(int fd, u_int expected_id, u_int *len)
162{
163 Buffer msg;
164 u_int type, id;
165 char *handle;
166
167 buffer_init(&msg);
168 get_msg(fd, &msg);
169 type = buffer_get_char(&msg);
170 id = buffer_get_int(&msg);
171
172 if (id != expected_id)
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000173 fatal("ID mismatch (%u != %u)", id, expected_id);
Damien Miller33804262001-02-04 23:20:18 +1100174 if (type == SSH2_FXP_STATUS) {
175 int status = buffer_get_int(&msg);
176
177 error("Couldn't get handle: %s", fx2txt(status));
Darren Tuckercd516ef2004-12-06 22:43:43 +1100178 buffer_free(&msg);
Damien Miller33804262001-02-04 23:20:18 +1100179 return(NULL);
180 } else if (type != SSH2_FXP_HANDLE)
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000181 fatal("Expected SSH2_FXP_HANDLE(%u) packet, got %u",
Damien Miller33804262001-02-04 23:20:18 +1100182 SSH2_FXP_HANDLE, type);
183
184 handle = buffer_get_string(&msg, len);
185 buffer_free(&msg);
186
187 return(handle);
188}
189
Ben Lindstrombba81212001-06-25 05:01:22 +0000190static Attrib *
Ben Lindstromc8d1c302001-03-17 00:34:46 +0000191get_decode_stat(int fd, u_int expected_id, int quiet)
Damien Miller33804262001-02-04 23:20:18 +1100192{
193 Buffer msg;
194 u_int type, id;
195 Attrib *a;
196
197 buffer_init(&msg);
198 get_msg(fd, &msg);
199
200 type = buffer_get_char(&msg);
201 id = buffer_get_int(&msg);
202
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000203 debug3("Received stat reply T:%u I:%u", type, id);
Damien Miller33804262001-02-04 23:20:18 +1100204 if (id != expected_id)
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000205 fatal("ID mismatch (%u != %u)", id, expected_id);
Damien Miller33804262001-02-04 23:20:18 +1100206 if (type == SSH2_FXP_STATUS) {
207 int status = buffer_get_int(&msg);
208
Ben Lindstromc8d1c302001-03-17 00:34:46 +0000209 if (quiet)
210 debug("Couldn't stat remote file: %s", fx2txt(status));
211 else
212 error("Couldn't stat remote file: %s", fx2txt(status));
Darren Tuckercd516ef2004-12-06 22:43:43 +1100213 buffer_free(&msg);
Damien Miller33804262001-02-04 23:20:18 +1100214 return(NULL);
215 } else if (type != SSH2_FXP_ATTRS) {
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000216 fatal("Expected SSH2_FXP_ATTRS(%u) packet, got %u",
Damien Miller33804262001-02-04 23:20:18 +1100217 SSH2_FXP_ATTRS, type);
218 }
219 a = decode_attrib(&msg);
220 buffer_free(&msg);
221
222 return(a);
223}
224
Damien Miller3db5f532002-02-13 14:10:32 +1100225struct sftp_conn *
226do_init(int fd_in, int fd_out, u_int transfer_buflen, u_int num_requests)
Damien Miller33804262001-02-04 23:20:18 +1100227{
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000228 u_int type;
229 int version;
Damien Miller33804262001-02-04 23:20:18 +1100230 Buffer msg;
Damien Miller3db5f532002-02-13 14:10:32 +1100231 struct sftp_conn *ret;
Damien Miller33804262001-02-04 23:20:18 +1100232
233 buffer_init(&msg);
234 buffer_put_char(&msg, SSH2_FXP_INIT);
235 buffer_put_int(&msg, SSH2_FILEXFER_VERSION);
236 send_msg(fd_out, &msg);
237
238 buffer_clear(&msg);
239
240 get_msg(fd_in, &msg);
241
Kevin Stevesef4eea92001-02-05 12:42:17 +0000242 /* Expecting a VERSION reply */
Damien Miller33804262001-02-04 23:20:18 +1100243 if ((type = buffer_get_char(&msg)) != SSH2_FXP_VERSION) {
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000244 error("Invalid packet back from SSH2_FXP_INIT (type %u)",
Damien Miller33804262001-02-04 23:20:18 +1100245 type);
246 buffer_free(&msg);
Damien Miller3db5f532002-02-13 14:10:32 +1100247 return(NULL);
Damien Miller33804262001-02-04 23:20:18 +1100248 }
249 version = buffer_get_int(&msg);
250
251 debug2("Remote version: %d", version);
252
253 /* Check for extensions */
254 while (buffer_len(&msg) > 0) {
255 char *name = buffer_get_string(&msg, NULL);
256 char *value = buffer_get_string(&msg, NULL);
257
258 debug2("Init extension: \"%s\"", name);
259 xfree(name);
260 xfree(value);
261 }
262
263 buffer_free(&msg);
Damien Miller058316f2001-03-08 10:08:49 +1100264
Damien Miller3db5f532002-02-13 14:10:32 +1100265 ret = xmalloc(sizeof(*ret));
266 ret->fd_in = fd_in;
267 ret->fd_out = fd_out;
268 ret->transfer_buflen = transfer_buflen;
269 ret->num_requests = num_requests;
270 ret->version = version;
271 ret->msg_id = 1;
272
273 /* Some filexfer v.0 servers don't support large packets */
274 if (version == 0)
Ben Lindstroma1d81142002-04-02 20:58:11 +0000275 ret->transfer_buflen = MIN(ret->transfer_buflen, 20480);
Damien Miller3db5f532002-02-13 14:10:32 +1100276
277 return(ret);
278}
279
280u_int
281sftp_proto_version(struct sftp_conn *conn)
282{
283 return(conn->version);
Damien Miller33804262001-02-04 23:20:18 +1100284}
285
286int
Damien Miller3db5f532002-02-13 14:10:32 +1100287do_close(struct sftp_conn *conn, char *handle, u_int handle_len)
Damien Miller33804262001-02-04 23:20:18 +1100288{
289 u_int id, status;
290 Buffer msg;
291
292 buffer_init(&msg);
293
Damien Miller3db5f532002-02-13 14:10:32 +1100294 id = conn->msg_id++;
Damien Miller33804262001-02-04 23:20:18 +1100295 buffer_put_char(&msg, SSH2_FXP_CLOSE);
296 buffer_put_int(&msg, id);
297 buffer_put_string(&msg, handle, handle_len);
Damien Miller3db5f532002-02-13 14:10:32 +1100298 send_msg(conn->fd_out, &msg);
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000299 debug3("Sent message SSH2_FXP_CLOSE I:%u", id);
Damien Miller33804262001-02-04 23:20:18 +1100300
Damien Miller3db5f532002-02-13 14:10:32 +1100301 status = get_status(conn->fd_in, id);
Damien Miller33804262001-02-04 23:20:18 +1100302 if (status != SSH2_FX_OK)
303 error("Couldn't close file: %s", fx2txt(status));
304
305 buffer_free(&msg);
306
307 return(status);
308}
309
Damien Miller4870afd2001-03-14 10:27:09 +1100310
Ben Lindstrombba81212001-06-25 05:01:22 +0000311static int
Damien Miller3db5f532002-02-13 14:10:32 +1100312do_lsreaddir(struct sftp_conn *conn, char *path, int printflag,
Damien Miller4870afd2001-03-14 10:27:09 +1100313 SFTP_DIRENT ***dir)
Damien Miller33804262001-02-04 23:20:18 +1100314{
315 Buffer msg;
Damien Millereccb9de2005-06-17 12:59:34 +1000316 u_int count, type, id, handle_len, i, expected_id, ents = 0;
Damien Miller33804262001-02-04 23:20:18 +1100317 char *handle;
318
Damien Miller3db5f532002-02-13 14:10:32 +1100319 id = conn->msg_id++;
Damien Miller33804262001-02-04 23:20:18 +1100320
321 buffer_init(&msg);
322 buffer_put_char(&msg, SSH2_FXP_OPENDIR);
323 buffer_put_int(&msg, id);
324 buffer_put_cstring(&msg, path);
Damien Miller3db5f532002-02-13 14:10:32 +1100325 send_msg(conn->fd_out, &msg);
Damien Miller33804262001-02-04 23:20:18 +1100326
327 buffer_clear(&msg);
328
Damien Miller3db5f532002-02-13 14:10:32 +1100329 handle = get_handle(conn->fd_in, id, &handle_len);
Damien Miller33804262001-02-04 23:20:18 +1100330 if (handle == NULL)
331 return(-1);
332
Damien Miller4870afd2001-03-14 10:27:09 +1100333 if (dir) {
334 ents = 0;
335 *dir = xmalloc(sizeof(**dir));
336 (*dir)[0] = NULL;
337 }
Damien Miller4870afd2001-03-14 10:27:09 +1100338
Darren Tuckercdf547a2004-05-24 10:12:19 +1000339 for (; !interrupted;) {
Damien Miller3db5f532002-02-13 14:10:32 +1100340 id = expected_id = conn->msg_id++;
Damien Miller33804262001-02-04 23:20:18 +1100341
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000342 debug3("Sending SSH2_FXP_READDIR I:%u", id);
Damien Miller33804262001-02-04 23:20:18 +1100343
344 buffer_clear(&msg);
345 buffer_put_char(&msg, SSH2_FXP_READDIR);
346 buffer_put_int(&msg, id);
347 buffer_put_string(&msg, handle, handle_len);
Damien Miller3db5f532002-02-13 14:10:32 +1100348 send_msg(conn->fd_out, &msg);
Damien Miller33804262001-02-04 23:20:18 +1100349
350 buffer_clear(&msg);
351
Damien Miller3db5f532002-02-13 14:10:32 +1100352 get_msg(conn->fd_in, &msg);
Damien Miller33804262001-02-04 23:20:18 +1100353
354 type = buffer_get_char(&msg);
355 id = buffer_get_int(&msg);
356
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000357 debug3("Received reply T:%u I:%u", type, id);
Damien Miller33804262001-02-04 23:20:18 +1100358
359 if (id != expected_id)
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000360 fatal("ID mismatch (%u != %u)", id, expected_id);
Damien Miller33804262001-02-04 23:20:18 +1100361
362 if (type == SSH2_FXP_STATUS) {
363 int status = buffer_get_int(&msg);
364
365 debug3("Received SSH2_FXP_STATUS %d", status);
366
367 if (status == SSH2_FX_EOF) {
368 break;
369 } else {
370 error("Couldn't read directory: %s",
371 fx2txt(status));
Damien Miller3db5f532002-02-13 14:10:32 +1100372 do_close(conn, handle, handle_len);
Damien Miller00111382003-03-10 11:21:17 +1100373 xfree(handle);
Ben Lindstrom10ac33f2001-02-10 21:53:40 +0000374 return(status);
Damien Miller33804262001-02-04 23:20:18 +1100375 }
376 } else if (type != SSH2_FXP_NAME)
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000377 fatal("Expected SSH2_FXP_NAME(%u) packet, got %u",
Damien Miller33804262001-02-04 23:20:18 +1100378 SSH2_FXP_NAME, type);
379
380 count = buffer_get_int(&msg);
Damien Millerd7686fd2001-02-10 00:40:03 +1100381 if (count == 0)
382 break;
383 debug3("Received %d SSH2_FXP_NAME responses", count);
Damien Miller9f0f5c62001-12-21 14:45:46 +1100384 for (i = 0; i < count; i++) {
Damien Miller33804262001-02-04 23:20:18 +1100385 char *filename, *longname;
386 Attrib *a;
387
388 filename = buffer_get_string(&msg, NULL);
389 longname = buffer_get_string(&msg, NULL);
390 a = decode_attrib(&msg);
391
Damien Miller4870afd2001-03-14 10:27:09 +1100392 if (printflag)
393 printf("%s\n", longname);
394
395 if (dir) {
Ben Lindstroma3700052001-04-05 23:26:32 +0000396 *dir = xrealloc(*dir, sizeof(**dir) *
Damien Miller4870afd2001-03-14 10:27:09 +1100397 (ents + 2));
398 (*dir)[ents] = xmalloc(sizeof(***dir));
399 (*dir)[ents]->filename = xstrdup(filename);
400 (*dir)[ents]->longname = xstrdup(longname);
401 memcpy(&(*dir)[ents]->a, a, sizeof(*a));
402 (*dir)[++ents] = NULL;
403 }
Damien Miller33804262001-02-04 23:20:18 +1100404
405 xfree(filename);
406 xfree(longname);
407 }
408 }
409
410 buffer_free(&msg);
Damien Miller3db5f532002-02-13 14:10:32 +1100411 do_close(conn, handle, handle_len);
Damien Miller33804262001-02-04 23:20:18 +1100412 xfree(handle);
413
Darren Tuckercdf547a2004-05-24 10:12:19 +1000414 /* Don't return partial matches on interrupt */
415 if (interrupted && dir != NULL && *dir != NULL) {
416 free_sftp_dirents(*dir);
417 *dir = xmalloc(sizeof(**dir));
418 **dir = NULL;
419 }
420
Damien Miller33804262001-02-04 23:20:18 +1100421 return(0);
422}
423
424int
Damien Miller3db5f532002-02-13 14:10:32 +1100425do_readdir(struct sftp_conn *conn, char *path, SFTP_DIRENT ***dir)
Damien Miller4870afd2001-03-14 10:27:09 +1100426{
Damien Miller3db5f532002-02-13 14:10:32 +1100427 return(do_lsreaddir(conn, path, 0, dir));
Damien Miller4870afd2001-03-14 10:27:09 +1100428}
429
430void free_sftp_dirents(SFTP_DIRENT **s)
431{
432 int i;
Damien Miller9f0f5c62001-12-21 14:45:46 +1100433
434 for (i = 0; s[i]; i++) {
Damien Miller4870afd2001-03-14 10:27:09 +1100435 xfree(s[i]->filename);
436 xfree(s[i]->longname);
437 xfree(s[i]);
438 }
439 xfree(s);
440}
441
442int
Damien Miller3db5f532002-02-13 14:10:32 +1100443do_rm(struct sftp_conn *conn, char *path)
Damien Miller33804262001-02-04 23:20:18 +1100444{
445 u_int status, id;
446
447 debug2("Sending SSH2_FXP_REMOVE \"%s\"", path);
448
Damien Miller3db5f532002-02-13 14:10:32 +1100449 id = conn->msg_id++;
Ben Lindstrom6328ab32002-03-22 02:54:23 +0000450 send_string_request(conn->fd_out, id, SSH2_FXP_REMOVE, path,
Damien Miller3db5f532002-02-13 14:10:32 +1100451 strlen(path));
452 status = get_status(conn->fd_in, id);
Damien Miller33804262001-02-04 23:20:18 +1100453 if (status != SSH2_FX_OK)
454 error("Couldn't delete file: %s", fx2txt(status));
455 return(status);
456}
457
458int
Damien Miller3db5f532002-02-13 14:10:32 +1100459do_mkdir(struct sftp_conn *conn, char *path, Attrib *a)
Damien Miller33804262001-02-04 23:20:18 +1100460{
461 u_int status, id;
462
Damien Miller3db5f532002-02-13 14:10:32 +1100463 id = conn->msg_id++;
464 send_string_attrs_request(conn->fd_out, id, SSH2_FXP_MKDIR, path,
Damien Miller33804262001-02-04 23:20:18 +1100465 strlen(path), a);
466
Damien Miller3db5f532002-02-13 14:10:32 +1100467 status = get_status(conn->fd_in, id);
Damien Miller33804262001-02-04 23:20:18 +1100468 if (status != SSH2_FX_OK)
469 error("Couldn't create directory: %s", fx2txt(status));
470
471 return(status);
472}
473
474int
Damien Miller3db5f532002-02-13 14:10:32 +1100475do_rmdir(struct sftp_conn *conn, char *path)
Damien Miller33804262001-02-04 23:20:18 +1100476{
477 u_int status, id;
478
Damien Miller3db5f532002-02-13 14:10:32 +1100479 id = conn->msg_id++;
480 send_string_request(conn->fd_out, id, SSH2_FXP_RMDIR, path,
481 strlen(path));
Damien Miller33804262001-02-04 23:20:18 +1100482
Damien Miller3db5f532002-02-13 14:10:32 +1100483 status = get_status(conn->fd_in, id);
Damien Miller33804262001-02-04 23:20:18 +1100484 if (status != SSH2_FX_OK)
485 error("Couldn't remove directory: %s", fx2txt(status));
486
487 return(status);
488}
489
490Attrib *
Damien Miller3db5f532002-02-13 14:10:32 +1100491do_stat(struct sftp_conn *conn, char *path, int quiet)
Damien Miller33804262001-02-04 23:20:18 +1100492{
493 u_int id;
494
Damien Miller3db5f532002-02-13 14:10:32 +1100495 id = conn->msg_id++;
496
Ben Lindstrom6328ab32002-03-22 02:54:23 +0000497 send_string_request(conn->fd_out, id,
498 conn->version == 0 ? SSH2_FXP_STAT_VERSION_0 : SSH2_FXP_STAT,
Damien Miller3db5f532002-02-13 14:10:32 +1100499 path, strlen(path));
500
501 return(get_decode_stat(conn->fd_in, id, quiet));
Damien Miller33804262001-02-04 23:20:18 +1100502}
503
504Attrib *
Damien Miller3db5f532002-02-13 14:10:32 +1100505do_lstat(struct sftp_conn *conn, char *path, int quiet)
Damien Miller33804262001-02-04 23:20:18 +1100506{
507 u_int id;
508
Damien Miller3db5f532002-02-13 14:10:32 +1100509 if (conn->version == 0) {
510 if (quiet)
511 debug("Server version does not support lstat operation");
512 else
Damien Miller996acd22003-04-09 20:59:48 +1000513 logit("Server version does not support lstat operation");
Ben Lindstromf26ff5b2002-04-02 21:00:31 +0000514 return(do_stat(conn, path, quiet));
Damien Miller3db5f532002-02-13 14:10:32 +1100515 }
516
517 id = conn->msg_id++;
518 send_string_request(conn->fd_out, id, SSH2_FXP_LSTAT, path,
519 strlen(path));
520
521 return(get_decode_stat(conn->fd_in, id, quiet));
Damien Miller33804262001-02-04 23:20:18 +1100522}
523
524Attrib *
Damien Miller3db5f532002-02-13 14:10:32 +1100525do_fstat(struct sftp_conn *conn, char *handle, u_int handle_len, int quiet)
Damien Miller33804262001-02-04 23:20:18 +1100526{
527 u_int id;
528
Damien Miller3db5f532002-02-13 14:10:32 +1100529 id = conn->msg_id++;
530 send_string_request(conn->fd_out, id, SSH2_FXP_FSTAT, handle,
531 handle_len);
532
533 return(get_decode_stat(conn->fd_in, id, quiet));
Damien Miller33804262001-02-04 23:20:18 +1100534}
535
536int
Damien Miller3db5f532002-02-13 14:10:32 +1100537do_setstat(struct sftp_conn *conn, char *path, Attrib *a)
Damien Miller33804262001-02-04 23:20:18 +1100538{
539 u_int status, id;
540
Damien Miller3db5f532002-02-13 14:10:32 +1100541 id = conn->msg_id++;
542 send_string_attrs_request(conn->fd_out, id, SSH2_FXP_SETSTAT, path,
Damien Miller33804262001-02-04 23:20:18 +1100543 strlen(path), a);
544
Damien Miller3db5f532002-02-13 14:10:32 +1100545 status = get_status(conn->fd_in, id);
Damien Miller33804262001-02-04 23:20:18 +1100546 if (status != SSH2_FX_OK)
547 error("Couldn't setstat on \"%s\": %s", path,
548 fx2txt(status));
549
550 return(status);
551}
552
553int
Damien Miller3db5f532002-02-13 14:10:32 +1100554do_fsetstat(struct sftp_conn *conn, char *handle, u_int handle_len,
Damien Miller33804262001-02-04 23:20:18 +1100555 Attrib *a)
556{
557 u_int status, id;
558
Damien Miller3db5f532002-02-13 14:10:32 +1100559 id = conn->msg_id++;
560 send_string_attrs_request(conn->fd_out, id, SSH2_FXP_FSETSTAT, handle,
Damien Miller33804262001-02-04 23:20:18 +1100561 handle_len, a);
562
Damien Miller3db5f532002-02-13 14:10:32 +1100563 status = get_status(conn->fd_in, id);
Damien Miller33804262001-02-04 23:20:18 +1100564 if (status != SSH2_FX_OK)
565 error("Couldn't fsetstat: %s", fx2txt(status));
566
567 return(status);
568}
569
570char *
Damien Miller3db5f532002-02-13 14:10:32 +1100571do_realpath(struct sftp_conn *conn, char *path)
Damien Miller33804262001-02-04 23:20:18 +1100572{
573 Buffer msg;
574 u_int type, expected_id, count, id;
575 char *filename, *longname;
576 Attrib *a;
577
Damien Miller3db5f532002-02-13 14:10:32 +1100578 expected_id = id = conn->msg_id++;
579 send_string_request(conn->fd_out, id, SSH2_FXP_REALPATH, path,
580 strlen(path));
Damien Miller33804262001-02-04 23:20:18 +1100581
582 buffer_init(&msg);
583
Damien Miller3db5f532002-02-13 14:10:32 +1100584 get_msg(conn->fd_in, &msg);
Damien Miller33804262001-02-04 23:20:18 +1100585 type = buffer_get_char(&msg);
586 id = buffer_get_int(&msg);
587
588 if (id != expected_id)
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000589 fatal("ID mismatch (%u != %u)", id, expected_id);
Damien Miller33804262001-02-04 23:20:18 +1100590
591 if (type == SSH2_FXP_STATUS) {
592 u_int status = buffer_get_int(&msg);
593
594 error("Couldn't canonicalise: %s", fx2txt(status));
595 return(NULL);
596 } else if (type != SSH2_FXP_NAME)
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000597 fatal("Expected SSH2_FXP_NAME(%u) packet, got %u",
Damien Miller33804262001-02-04 23:20:18 +1100598 SSH2_FXP_NAME, type);
599
600 count = buffer_get_int(&msg);
601 if (count != 1)
602 fatal("Got multiple names (%d) from SSH_FXP_REALPATH", count);
603
604 filename = buffer_get_string(&msg, NULL);
605 longname = buffer_get_string(&msg, NULL);
606 a = decode_attrib(&msg);
607
608 debug3("SSH_FXP_REALPATH %s -> %s", path, filename);
609
610 xfree(longname);
611
612 buffer_free(&msg);
613
614 return(filename);
615}
616
617int
Damien Miller3db5f532002-02-13 14:10:32 +1100618do_rename(struct sftp_conn *conn, char *oldpath, char *newpath)
Damien Miller33804262001-02-04 23:20:18 +1100619{
620 Buffer msg;
621 u_int status, id;
622
623 buffer_init(&msg);
624
625 /* Send rename request */
Damien Miller3db5f532002-02-13 14:10:32 +1100626 id = conn->msg_id++;
Damien Miller33804262001-02-04 23:20:18 +1100627 buffer_put_char(&msg, SSH2_FXP_RENAME);
628 buffer_put_int(&msg, id);
629 buffer_put_cstring(&msg, oldpath);
630 buffer_put_cstring(&msg, newpath);
Damien Miller3db5f532002-02-13 14:10:32 +1100631 send_msg(conn->fd_out, &msg);
Damien Miller33804262001-02-04 23:20:18 +1100632 debug3("Sent message SSH2_FXP_RENAME \"%s\" -> \"%s\"", oldpath,
633 newpath);
634 buffer_free(&msg);
635
Damien Miller3db5f532002-02-13 14:10:32 +1100636 status = get_status(conn->fd_in, id);
Damien Miller33804262001-02-04 23:20:18 +1100637 if (status != SSH2_FX_OK)
Damien Miller3db5f532002-02-13 14:10:32 +1100638 error("Couldn't rename file \"%s\" to \"%s\": %s", oldpath,
639 newpath, fx2txt(status));
Damien Miller33804262001-02-04 23:20:18 +1100640
641 return(status);
642}
643
644int
Damien Miller3db5f532002-02-13 14:10:32 +1100645do_symlink(struct sftp_conn *conn, char *oldpath, char *newpath)
Damien Miller058316f2001-03-08 10:08:49 +1100646{
647 Buffer msg;
648 u_int status, id;
649
Damien Miller3db5f532002-02-13 14:10:32 +1100650 if (conn->version < 3) {
651 error("This server does not support the symlink operation");
652 return(SSH2_FX_OP_UNSUPPORTED);
653 }
654
Damien Miller058316f2001-03-08 10:08:49 +1100655 buffer_init(&msg);
656
Darren Tuckerdca6a4d2004-04-19 22:10:52 +1000657 /* Send symlink request */
Damien Miller3db5f532002-02-13 14:10:32 +1100658 id = conn->msg_id++;
Damien Miller058316f2001-03-08 10:08:49 +1100659 buffer_put_char(&msg, SSH2_FXP_SYMLINK);
660 buffer_put_int(&msg, id);
661 buffer_put_cstring(&msg, oldpath);
662 buffer_put_cstring(&msg, newpath);
Damien Miller3db5f532002-02-13 14:10:32 +1100663 send_msg(conn->fd_out, &msg);
Damien Miller058316f2001-03-08 10:08:49 +1100664 debug3("Sent message SSH2_FXP_SYMLINK \"%s\" -> \"%s\"", oldpath,
665 newpath);
666 buffer_free(&msg);
667
Damien Miller3db5f532002-02-13 14:10:32 +1100668 status = get_status(conn->fd_in, id);
Damien Miller058316f2001-03-08 10:08:49 +1100669 if (status != SSH2_FX_OK)
Ben Lindstrom8e879cf2002-11-09 15:48:49 +0000670 error("Couldn't symlink file \"%s\" to \"%s\": %s", oldpath,
Damien Miller3db5f532002-02-13 14:10:32 +1100671 newpath, fx2txt(status));
Damien Miller058316f2001-03-08 10:08:49 +1100672
673 return(status);
674}
675
676char *
Damien Miller3db5f532002-02-13 14:10:32 +1100677do_readlink(struct sftp_conn *conn, char *path)
Damien Miller058316f2001-03-08 10:08:49 +1100678{
679 Buffer msg;
680 u_int type, expected_id, count, id;
681 char *filename, *longname;
682 Attrib *a;
683
Damien Miller3db5f532002-02-13 14:10:32 +1100684 expected_id = id = conn->msg_id++;
685 send_string_request(conn->fd_out, id, SSH2_FXP_READLINK, path,
686 strlen(path));
Damien Miller058316f2001-03-08 10:08:49 +1100687
688 buffer_init(&msg);
689
Damien Miller3db5f532002-02-13 14:10:32 +1100690 get_msg(conn->fd_in, &msg);
Damien Miller058316f2001-03-08 10:08:49 +1100691 type = buffer_get_char(&msg);
692 id = buffer_get_int(&msg);
693
694 if (id != expected_id)
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000695 fatal("ID mismatch (%u != %u)", id, expected_id);
Damien Miller058316f2001-03-08 10:08:49 +1100696
697 if (type == SSH2_FXP_STATUS) {
698 u_int status = buffer_get_int(&msg);
699
700 error("Couldn't readlink: %s", fx2txt(status));
701 return(NULL);
702 } else if (type != SSH2_FXP_NAME)
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000703 fatal("Expected SSH2_FXP_NAME(%u) packet, got %u",
Damien Miller058316f2001-03-08 10:08:49 +1100704 SSH2_FXP_NAME, type);
705
706 count = buffer_get_int(&msg);
707 if (count != 1)
708 fatal("Got multiple names (%d) from SSH_FXP_READLINK", count);
709
710 filename = buffer_get_string(&msg, NULL);
711 longname = buffer_get_string(&msg, NULL);
712 a = decode_attrib(&msg);
713
714 debug3("SSH_FXP_READLINK %s -> %s", path, filename);
715
716 xfree(longname);
717
718 buffer_free(&msg);
719
720 return(filename);
721}
722
Damien Miller16a13332002-02-13 14:03:56 +1100723static void
724send_read_request(int fd_out, u_int id, u_int64_t offset, u_int len,
725 char *handle, u_int handle_len)
726{
727 Buffer msg;
Ben Lindstrom6328ab32002-03-22 02:54:23 +0000728
Damien Miller16a13332002-02-13 14:03:56 +1100729 buffer_init(&msg);
730 buffer_clear(&msg);
731 buffer_put_char(&msg, SSH2_FXP_READ);
732 buffer_put_int(&msg, id);
733 buffer_put_string(&msg, handle, handle_len);
734 buffer_put_int64(&msg, offset);
735 buffer_put_int(&msg, len);
736 send_msg(fd_out, &msg);
737 buffer_free(&msg);
Ben Lindstrom6328ab32002-03-22 02:54:23 +0000738}
Damien Miller16a13332002-02-13 14:03:56 +1100739
Damien Miller058316f2001-03-08 10:08:49 +1100740int
Damien Miller3db5f532002-02-13 14:10:32 +1100741do_download(struct sftp_conn *conn, char *remote_path, char *local_path,
742 int pflag)
Damien Miller33804262001-02-04 23:20:18 +1100743{
Damien Miller33804262001-02-04 23:20:18 +1100744 Attrib junk, *a;
Damien Miller16a13332002-02-13 14:03:56 +1100745 Buffer msg;
746 char *handle;
Darren Tucker40858532005-08-02 17:07:07 +1000747 int local_fd, status = 0, write_error;
Damien Miller16a13332002-02-13 14:03:56 +1100748 int read_error, write_errno;
749 u_int64_t offset, size;
Damien Millereccb9de2005-06-17 12:59:34 +1000750 u_int handle_len, mode, type, id, buflen, num_req, max_req;
Damien Miller62d57f62003-01-10 21:43:24 +1100751 off_t progress_counter;
Damien Miller16a13332002-02-13 14:03:56 +1100752 struct request {
753 u_int id;
754 u_int len;
755 u_int64_t offset;
Ben Lindstrom6328ab32002-03-22 02:54:23 +0000756 TAILQ_ENTRY(request) tq;
Damien Miller16a13332002-02-13 14:03:56 +1100757 };
758 TAILQ_HEAD(reqhead, request) requests;
759 struct request *req;
760
761 TAILQ_INIT(&requests);
Damien Miller33804262001-02-04 23:20:18 +1100762
Damien Miller3db5f532002-02-13 14:10:32 +1100763 a = do_stat(conn, remote_path, 0);
Damien Miller33804262001-02-04 23:20:18 +1100764 if (a == NULL)
765 return(-1);
766
767 /* XXX: should we preserve set[ug]id? */
768 if (a->flags & SSH2_FILEXFER_ATTR_PERMISSIONS)
Damien Miller770b3742003-01-08 14:04:53 +1100769 mode = a->perm & 0777;
Damien Miller33804262001-02-04 23:20:18 +1100770 else
771 mode = 0666;
772
Ben Lindstromc8d1c302001-03-17 00:34:46 +0000773 if ((a->flags & SSH2_FILEXFER_ATTR_PERMISSIONS) &&
Damien Miller5fa01fd2003-01-14 22:24:47 +1100774 (!S_ISREG(a->perm))) {
775 error("Cannot download non-regular file: %s", remote_path);
Ben Lindstromc8d1c302001-03-17 00:34:46 +0000776 return(-1);
777 }
778
Damien Miller16a13332002-02-13 14:03:56 +1100779 if (a->flags & SSH2_FILEXFER_ATTR_SIZE)
780 size = a->size;
781 else
782 size = 0;
783
Damien Miller3db5f532002-02-13 14:10:32 +1100784 buflen = conn->transfer_buflen;
Damien Miller33804262001-02-04 23:20:18 +1100785 buffer_init(&msg);
786
787 /* Send open request */
Damien Miller3db5f532002-02-13 14:10:32 +1100788 id = conn->msg_id++;
Damien Miller33804262001-02-04 23:20:18 +1100789 buffer_put_char(&msg, SSH2_FXP_OPEN);
790 buffer_put_int(&msg, id);
791 buffer_put_cstring(&msg, remote_path);
792 buffer_put_int(&msg, SSH2_FXF_READ);
793 attrib_clear(&junk); /* Send empty attributes */
794 encode_attrib(&msg, &junk);
Damien Miller3db5f532002-02-13 14:10:32 +1100795 send_msg(conn->fd_out, &msg);
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000796 debug3("Sent message SSH2_FXP_OPEN I:%u P:%s", id, remote_path);
Damien Miller33804262001-02-04 23:20:18 +1100797
Damien Miller3db5f532002-02-13 14:10:32 +1100798 handle = get_handle(conn->fd_in, id, &handle_len);
Damien Miller33804262001-02-04 23:20:18 +1100799 if (handle == NULL) {
800 buffer_free(&msg);
Damien Miller33804262001-02-04 23:20:18 +1100801 return(-1);
802 }
803
Damien Millera8e06ce2003-11-21 23:48:55 +1100804 local_fd = open(local_path, O_WRONLY | O_CREAT | O_TRUNC,
Damien Miller770b3742003-01-08 14:04:53 +1100805 mode | S_IWRITE);
Damien Miller3db5f532002-02-13 14:10:32 +1100806 if (local_fd == -1) {
807 error("Couldn't open local file \"%s\" for writing: %s",
808 local_path, strerror(errno));
Ben Lindstrom021fcd32002-02-26 18:02:43 +0000809 buffer_free(&msg);
810 xfree(handle);
Damien Miller3db5f532002-02-13 14:10:32 +1100811 return(-1);
812 }
813
Damien Miller33804262001-02-04 23:20:18 +1100814 /* Read from remote and write to local */
Damien Miller16a13332002-02-13 14:03:56 +1100815 write_error = read_error = write_errno = num_req = offset = 0;
816 max_req = 1;
Damien Miller62d57f62003-01-10 21:43:24 +1100817 progress_counter = 0;
818
Damien Miller9ba30692004-03-08 23:12:02 +1100819 if (showprogress && size != 0)
820 start_progress_meter(remote_path, size, &progress_counter);
Damien Miller62d57f62003-01-10 21:43:24 +1100821
Damien Miller16a13332002-02-13 14:03:56 +1100822 while (num_req > 0 || max_req > 0) {
Damien Miller33804262001-02-04 23:20:18 +1100823 char *data;
Damien Miller16a13332002-02-13 14:03:56 +1100824 u_int len;
Damien Miller33804262001-02-04 23:20:18 +1100825
Darren Tuckercdf547a2004-05-24 10:12:19 +1000826 /*
Darren Tuckerfc959702004-07-17 16:12:08 +1000827 * Simulate EOF on interrupt: stop sending new requests and
Darren Tuckercdf547a2004-05-24 10:12:19 +1000828 * allow outstanding requests to drain gracefully
829 */
830 if (interrupted) {
831 if (num_req == 0) /* If we haven't started yet... */
832 break;
833 max_req = 0;
834 }
835
Damien Miller16a13332002-02-13 14:03:56 +1100836 /* Send some more requests */
837 while (num_req < max_req) {
Ben Lindstrom6328ab32002-03-22 02:54:23 +0000838 debug3("Request range %llu -> %llu (%d/%d)",
Ben Lindstromd45f28c2002-03-22 01:00:57 +0000839 (unsigned long long)offset,
840 (unsigned long long)offset + buflen - 1,
841 num_req, max_req);
Damien Miller16a13332002-02-13 14:03:56 +1100842 req = xmalloc(sizeof(*req));
Damien Miller3db5f532002-02-13 14:10:32 +1100843 req->id = conn->msg_id++;
Damien Miller16a13332002-02-13 14:03:56 +1100844 req->len = buflen;
845 req->offset = offset;
846 offset += buflen;
847 num_req++;
848 TAILQ_INSERT_TAIL(&requests, req, tq);
Ben Lindstrom6328ab32002-03-22 02:54:23 +0000849 send_read_request(conn->fd_out, req->id, req->offset,
Damien Miller16a13332002-02-13 14:03:56 +1100850 req->len, handle, handle_len);
851 }
Damien Miller33804262001-02-04 23:20:18 +1100852
853 buffer_clear(&msg);
Damien Miller3db5f532002-02-13 14:10:32 +1100854 get_msg(conn->fd_in, &msg);
Damien Miller33804262001-02-04 23:20:18 +1100855 type = buffer_get_char(&msg);
856 id = buffer_get_int(&msg);
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000857 debug3("Received reply T:%u I:%u R:%d", type, id, max_req);
Damien Miller33804262001-02-04 23:20:18 +1100858
Damien Miller16a13332002-02-13 14:03:56 +1100859 /* Find the request in our queue */
Darren Tucker47eede72005-03-14 23:08:12 +1100860 for (req = TAILQ_FIRST(&requests);
Damien Miller16a13332002-02-13 14:03:56 +1100861 req != NULL && req->id != id;
862 req = TAILQ_NEXT(req, tq))
863 ;
864 if (req == NULL)
865 fatal("Unexpected reply %u", id);
866
867 switch (type) {
868 case SSH2_FXP_STATUS:
869 status = buffer_get_int(&msg);
870 if (status != SSH2_FX_EOF)
871 read_error = 1;
872 max_req = 0;
873 TAILQ_REMOVE(&requests, req, tq);
874 xfree(req);
875 num_req--;
876 break;
877 case SSH2_FXP_DATA:
878 data = buffer_get_string(&msg, &len);
Ben Lindstromeb505452002-03-22 01:03:15 +0000879 debug3("Received data %llu -> %llu",
Ben Lindstrom6328ab32002-03-22 02:54:23 +0000880 (unsigned long long)req->offset,
Ben Lindstromeb505452002-03-22 01:03:15 +0000881 (unsigned long long)req->offset + len - 1);
Damien Miller16a13332002-02-13 14:03:56 +1100882 if (len > req->len)
883 fatal("Received more data than asked for "
Ben Lindstrom93576d92002-12-23 02:06:19 +0000884 "%u > %u", len, req->len);
Damien Miller16a13332002-02-13 14:03:56 +1100885 if ((lseek(local_fd, req->offset, SEEK_SET) == -1 ||
Darren Tucker9f63f222003-07-03 13:46:56 +1000886 atomicio(vwrite, local_fd, data, len) != len) &&
Damien Miller16a13332002-02-13 14:03:56 +1100887 !write_error) {
888 write_errno = errno;
889 write_error = 1;
890 max_req = 0;
Damien Miller33804262001-02-04 23:20:18 +1100891 }
Damien Miller62d57f62003-01-10 21:43:24 +1100892 progress_counter += len;
Damien Miller16a13332002-02-13 14:03:56 +1100893 xfree(data);
894
895 if (len == req->len) {
896 TAILQ_REMOVE(&requests, req, tq);
897 xfree(req);
898 num_req--;
899 } else {
900 /* Resend the request for the missing data */
901 debug3("Short data block, re-requesting "
Ben Lindstromeb505452002-03-22 01:03:15 +0000902 "%llu -> %llu (%2d)",
Ben Lindstrom6328ab32002-03-22 02:54:23 +0000903 (unsigned long long)req->offset + len,
Ben Lindstrom83b79e42002-03-22 01:05:27 +0000904 (unsigned long long)req->offset +
905 req->len - 1, num_req);
Damien Miller3db5f532002-02-13 14:10:32 +1100906 req->id = conn->msg_id++;
Damien Miller16a13332002-02-13 14:03:56 +1100907 req->len -= len;
908 req->offset += len;
Ben Lindstrom6328ab32002-03-22 02:54:23 +0000909 send_read_request(conn->fd_out, req->id,
Damien Miller3db5f532002-02-13 14:10:32 +1100910 req->offset, req->len, handle, handle_len);
Damien Miller16a13332002-02-13 14:03:56 +1100911 /* Reduce the request size */
912 if (len < buflen)
913 buflen = MAX(MIN_READ_SIZE, len);
914 }
915 if (max_req > 0) { /* max_req = 0 iff EOF received */
916 if (size > 0 && offset > size) {
917 /* Only one request at a time
918 * after the expected EOF */
919 debug3("Finish at %llu (%2d)",
Ben Lindstromeb505452002-03-22 01:03:15 +0000920 (unsigned long long)offset,
921 num_req);
Damien Miller16a13332002-02-13 14:03:56 +1100922 max_req = 1;
Darren Tuckercdf547a2004-05-24 10:12:19 +1000923 } else if (max_req <= conn->num_requests) {
Damien Miller16a13332002-02-13 14:03:56 +1100924 ++max_req;
925 }
926 }
927 break;
928 default:
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000929 fatal("Expected SSH2_FXP_DATA(%u) packet, got %u",
Damien Miller33804262001-02-04 23:20:18 +1100930 SSH2_FXP_DATA, type);
931 }
Damien Miller33804262001-02-04 23:20:18 +1100932 }
Damien Miller33804262001-02-04 23:20:18 +1100933
Damien Miller62d57f62003-01-10 21:43:24 +1100934 if (showprogress && size)
935 stop_progress_meter();
936
Damien Miller16a13332002-02-13 14:03:56 +1100937 /* Sanity check */
938 if (TAILQ_FIRST(&requests) != NULL)
939 fatal("Transfer complete, but requests still in queue");
940
941 if (read_error) {
Ben Lindstrom6328ab32002-03-22 02:54:23 +0000942 error("Couldn't read from remote file \"%s\" : %s",
Damien Miller3db5f532002-02-13 14:10:32 +1100943 remote_path, fx2txt(status));
944 do_close(conn, handle, handle_len);
Damien Miller16a13332002-02-13 14:03:56 +1100945 } else if (write_error) {
Damien Miller3db5f532002-02-13 14:10:32 +1100946 error("Couldn't write to \"%s\": %s", local_path,
947 strerror(write_errno));
948 status = -1;
949 do_close(conn, handle, handle_len);
Damien Miller16a13332002-02-13 14:03:56 +1100950 } else {
Damien Miller3db5f532002-02-13 14:10:32 +1100951 status = do_close(conn, handle, handle_len);
Damien Miller16a13332002-02-13 14:03:56 +1100952
953 /* Override umask and utimes if asked */
Ben Lindstrom6dc75f52001-02-17 16:47:47 +0000954#ifdef HAVE_FCHMOD
Damien Miller16a13332002-02-13 14:03:56 +1100955 if (pflag && fchmod(local_fd, mode) == -1)
Damien Millera8e06ce2003-11-21 23:48:55 +1100956#else
Damien Miller16a13332002-02-13 14:03:56 +1100957 if (pflag && chmod(local_path, mode) == -1)
Ben Lindstrom6dc75f52001-02-17 16:47:47 +0000958#endif /* HAVE_FCHMOD */
Damien Miller16a13332002-02-13 14:03:56 +1100959 error("Couldn't set mode on \"%s\": %s", local_path,
Ben Lindstrom93576d92002-12-23 02:06:19 +0000960 strerror(errno));
Damien Miller16a13332002-02-13 14:03:56 +1100961 if (pflag && (a->flags & SSH2_FILEXFER_ATTR_ACMODTIME)) {
962 struct timeval tv[2];
963 tv[0].tv_sec = a->atime;
964 tv[1].tv_sec = a->mtime;
965 tv[0].tv_usec = tv[1].tv_usec = 0;
966 if (utimes(local_path, tv) == -1)
967 error("Can't set times on \"%s\": %s",
Ben Lindstrom93576d92002-12-23 02:06:19 +0000968 local_path, strerror(errno));
Damien Miller16a13332002-02-13 14:03:56 +1100969 }
Ben Lindstrom9d4f2c82001-02-15 03:22:45 +0000970 }
Damien Millerd7686fd2001-02-10 00:40:03 +1100971 close(local_fd);
972 buffer_free(&msg);
973 xfree(handle);
Damien Miller3db5f532002-02-13 14:10:32 +1100974
975 return(status);
Damien Miller33804262001-02-04 23:20:18 +1100976}
977
978int
Damien Miller3db5f532002-02-13 14:10:32 +1100979do_upload(struct sftp_conn *conn, char *local_path, char *remote_path,
980 int pflag)
Damien Miller33804262001-02-04 23:20:18 +1100981{
Damien Miller8829d362002-02-08 22:04:05 +1100982 int local_fd, status;
Damien Miller5873dfd2002-02-13 14:04:37 +1100983 u_int handle_len, id, type;
Damien Miller33804262001-02-04 23:20:18 +1100984 u_int64_t offset;
Damien Miller8829d362002-02-08 22:04:05 +1100985 char *handle, *data;
Damien Miller33804262001-02-04 23:20:18 +1100986 Buffer msg;
987 struct stat sb;
988 Attrib a;
Damien Miller16a13332002-02-13 14:03:56 +1100989 u_int32_t startid;
990 u_int32_t ackid;
Damien Miller5873dfd2002-02-13 14:04:37 +1100991 struct outstanding_ack {
992 u_int id;
993 u_int len;
994 u_int64_t offset;
Ben Lindstrom6328ab32002-03-22 02:54:23 +0000995 TAILQ_ENTRY(outstanding_ack) tq;
Damien Miller5873dfd2002-02-13 14:04:37 +1100996 };
997 TAILQ_HEAD(ackhead, outstanding_ack) acks;
Damien Miller7cf17eb2004-06-15 10:28:56 +1000998 struct outstanding_ack *ack = NULL;
Damien Miller5873dfd2002-02-13 14:04:37 +1100999
1000 TAILQ_INIT(&acks);
Damien Miller33804262001-02-04 23:20:18 +11001001
1002 if ((local_fd = open(local_path, O_RDONLY, 0)) == -1) {
1003 error("Couldn't open local file \"%s\" for reading: %s",
1004 local_path, strerror(errno));
1005 return(-1);
1006 }
1007 if (fstat(local_fd, &sb) == -1) {
1008 error("Couldn't fstat local file \"%s\": %s",
1009 local_path, strerror(errno));
1010 close(local_fd);
1011 return(-1);
1012 }
Damien Miller5fa01fd2003-01-14 22:24:47 +11001013 if (!S_ISREG(sb.st_mode)) {
1014 error("%s is not a regular file", local_path);
1015 close(local_fd);
1016 return(-1);
1017 }
Damien Miller33804262001-02-04 23:20:18 +11001018 stat_to_attrib(&sb, &a);
1019
1020 a.flags &= ~SSH2_FILEXFER_ATTR_SIZE;
1021 a.flags &= ~SSH2_FILEXFER_ATTR_UIDGID;
1022 a.perm &= 0777;
1023 if (!pflag)
1024 a.flags &= ~SSH2_FILEXFER_ATTR_ACMODTIME;
1025
1026 buffer_init(&msg);
1027
1028 /* Send open request */
Damien Miller3db5f532002-02-13 14:10:32 +11001029 id = conn->msg_id++;
Damien Miller33804262001-02-04 23:20:18 +11001030 buffer_put_char(&msg, SSH2_FXP_OPEN);
1031 buffer_put_int(&msg, id);
1032 buffer_put_cstring(&msg, remote_path);
1033 buffer_put_int(&msg, SSH2_FXF_WRITE|SSH2_FXF_CREAT|SSH2_FXF_TRUNC);
1034 encode_attrib(&msg, &a);
Damien Miller3db5f532002-02-13 14:10:32 +11001035 send_msg(conn->fd_out, &msg);
Ben Lindstromb1f483f2002-06-23 21:27:18 +00001036 debug3("Sent message SSH2_FXP_OPEN I:%u P:%s", id, remote_path);
Damien Miller33804262001-02-04 23:20:18 +11001037
1038 buffer_clear(&msg);
1039
Damien Miller3db5f532002-02-13 14:10:32 +11001040 handle = get_handle(conn->fd_in, id, &handle_len);
Damien Miller33804262001-02-04 23:20:18 +11001041 if (handle == NULL) {
1042 close(local_fd);
1043 buffer_free(&msg);
1044 return(-1);
1045 }
1046
Damien Miller16a13332002-02-13 14:03:56 +11001047 startid = ackid = id + 1;
Damien Miller3db5f532002-02-13 14:10:32 +11001048 data = xmalloc(conn->transfer_buflen);
Damien Miller8829d362002-02-08 22:04:05 +11001049
Damien Miller33804262001-02-04 23:20:18 +11001050 /* Read from local and write to remote */
1051 offset = 0;
Damien Miller62d57f62003-01-10 21:43:24 +11001052 if (showprogress)
1053 start_progress_meter(local_path, sb.st_size, &offset);
Damien Miller62d57f62003-01-10 21:43:24 +11001054
Damien Miller9f0f5c62001-12-21 14:45:46 +11001055 for (;;) {
Damien Miller33804262001-02-04 23:20:18 +11001056 int len;
Damien Miller33804262001-02-04 23:20:18 +11001057
1058 /*
Darren Tuckerfc959702004-07-17 16:12:08 +10001059 * Can't use atomicio here because it returns 0 on EOF,
Darren Tuckercdf547a2004-05-24 10:12:19 +10001060 * thus losing the last block of the file.
Darren Tuckerfc959702004-07-17 16:12:08 +10001061 * Simulate an EOF on interrupt, allowing ACKs from the
Darren Tuckercdf547a2004-05-24 10:12:19 +10001062 * server to drain.
Damien Miller33804262001-02-04 23:20:18 +11001063 */
Darren Tuckercdf547a2004-05-24 10:12:19 +10001064 if (interrupted)
1065 len = 0;
1066 else do
Damien Miller3db5f532002-02-13 14:10:32 +11001067 len = read(local_fd, data, conn->transfer_buflen);
Damien Miller33804262001-02-04 23:20:18 +11001068 while ((len == -1) && (errno == EINTR || errno == EAGAIN));
1069
1070 if (len == -1)
1071 fatal("Couldn't read from \"%s\": %s", local_path,
1072 strerror(errno));
Damien Miller16a13332002-02-13 14:03:56 +11001073
1074 if (len != 0) {
Damien Miller5873dfd2002-02-13 14:04:37 +11001075 ack = xmalloc(sizeof(*ack));
1076 ack->id = ++id;
1077 ack->offset = offset;
1078 ack->len = len;
1079 TAILQ_INSERT_TAIL(&acks, ack, tq);
1080
Damien Miller16a13332002-02-13 14:03:56 +11001081 buffer_clear(&msg);
1082 buffer_put_char(&msg, SSH2_FXP_WRITE);
Damien Miller5873dfd2002-02-13 14:04:37 +11001083 buffer_put_int(&msg, ack->id);
Damien Miller16a13332002-02-13 14:03:56 +11001084 buffer_put_string(&msg, handle, handle_len);
1085 buffer_put_int64(&msg, offset);
1086 buffer_put_string(&msg, data, len);
Damien Miller3db5f532002-02-13 14:10:32 +11001087 send_msg(conn->fd_out, &msg);
Ben Lindstromb1f483f2002-06-23 21:27:18 +00001088 debug3("Sent message SSH2_FXP_WRITE I:%u O:%llu S:%u",
Ben Lindstrom93576d92002-12-23 02:06:19 +00001089 id, (unsigned long long)offset, len);
Damien Miller5873dfd2002-02-13 14:04:37 +11001090 } else if (TAILQ_FIRST(&acks) == NULL)
Damien Miller33804262001-02-04 23:20:18 +11001091 break;
1092
Damien Miller5873dfd2002-02-13 14:04:37 +11001093 if (ack == NULL)
1094 fatal("Unexpected ACK %u", id);
1095
Ben Lindstrom6328ab32002-03-22 02:54:23 +00001096 if (id == startid || len == 0 ||
Damien Miller3db5f532002-02-13 14:10:32 +11001097 id - ackid >= conn->num_requests) {
Ben Lindstrom5a6abda2002-06-09 19:41:48 +00001098 u_int r_id;
Ben Lindstrom06e95152002-04-06 04:16:45 +00001099
Damien Miller5873dfd2002-02-13 14:04:37 +11001100 buffer_clear(&msg);
Damien Miller3db5f532002-02-13 14:10:32 +11001101 get_msg(conn->fd_in, &msg);
Damien Miller5873dfd2002-02-13 14:04:37 +11001102 type = buffer_get_char(&msg);
Ben Lindstrom06e95152002-04-06 04:16:45 +00001103 r_id = buffer_get_int(&msg);
Damien Miller5873dfd2002-02-13 14:04:37 +11001104
1105 if (type != SSH2_FXP_STATUS)
1106 fatal("Expected SSH2_FXP_STATUS(%d) packet, "
1107 "got %d", SSH2_FXP_STATUS, type);
1108
1109 status = buffer_get_int(&msg);
1110 debug3("SSH2_FXP_STATUS %d", status);
1111
1112 /* Find the request in our queue */
Darren Tucker47eede72005-03-14 23:08:12 +11001113 for (ack = TAILQ_FIRST(&acks);
Ben Lindstrom06e95152002-04-06 04:16:45 +00001114 ack != NULL && ack->id != r_id;
Damien Miller5873dfd2002-02-13 14:04:37 +11001115 ack = TAILQ_NEXT(ack, tq))
1116 ;
1117 if (ack == NULL)
Ben Lindstromb1f483f2002-06-23 21:27:18 +00001118 fatal("Can't find request for ID %u", r_id);
Damien Miller5873dfd2002-02-13 14:04:37 +11001119 TAILQ_REMOVE(&acks, ack, tq);
1120
Damien Miller16a13332002-02-13 14:03:56 +11001121 if (status != SSH2_FX_OK) {
1122 error("Couldn't write to remote file \"%s\": %s",
Ben Lindstrom93576d92002-12-23 02:06:19 +00001123 remote_path, fx2txt(status));
Damien Miller3db5f532002-02-13 14:10:32 +11001124 do_close(conn, handle, handle_len);
Damien Miller16a13332002-02-13 14:03:56 +11001125 close(local_fd);
Damien Miller00111382003-03-10 11:21:17 +11001126 xfree(data);
1127 xfree(ack);
Damien Miller16a13332002-02-13 14:03:56 +11001128 goto done;
1129 }
Ben Lindstromb1f483f2002-06-23 21:27:18 +00001130 debug3("In write loop, ack for %u %u bytes at %llu",
Damien Miller0dc1bef2005-07-17 17:22:45 +10001131 ack->id, ack->len, (unsigned long long)ack->offset);
Damien Miller16a13332002-02-13 14:03:56 +11001132 ++ackid;
Ben Lindstromeec16fc2002-07-04 00:06:15 +00001133 xfree(ack);
Damien Miller33804262001-02-04 23:20:18 +11001134 }
Damien Miller33804262001-02-04 23:20:18 +11001135 offset += len;
1136 }
Damien Miller62d57f62003-01-10 21:43:24 +11001137 if (showprogress)
1138 stop_progress_meter();
Damien Miller8829d362002-02-08 22:04:05 +11001139 xfree(data);
Damien Miller33804262001-02-04 23:20:18 +11001140
1141 if (close(local_fd) == -1) {
1142 error("Couldn't close local file \"%s\": %s", local_path,
1143 strerror(errno));
Damien Miller3db5f532002-02-13 14:10:32 +11001144 do_close(conn, handle, handle_len);
Damien Millerd7686fd2001-02-10 00:40:03 +11001145 status = -1;
1146 goto done;
Damien Miller33804262001-02-04 23:20:18 +11001147 }
1148
Ben Lindstrom9d4f2c82001-02-15 03:22:45 +00001149 /* Override umask and utimes if asked */
1150 if (pflag)
Damien Miller3db5f532002-02-13 14:10:32 +11001151 do_fsetstat(conn, handle, handle_len, &a);
Ben Lindstrom9d4f2c82001-02-15 03:22:45 +00001152
Damien Miller3db5f532002-02-13 14:10:32 +11001153 status = do_close(conn, handle, handle_len);
Damien Millerd7686fd2001-02-10 00:40:03 +11001154
1155done:
1156 xfree(handle);
1157 buffer_free(&msg);
Damien Miller3db5f532002-02-13 14:10:32 +11001158 return(status);
Damien Miller33804262001-02-04 23:20:18 +11001159}