blob: a48d56e697ea7cba7c10fe393c112b0dafd824da [file] [log] [blame]
Damien Miller33804262001-02-04 23:20:18 +11001/*
Damien Millera7f3aaa2003-01-10 21:43:58 +11002 * Copyright (c) 2001-2003 Damien Miller. All rights reserved.
Damien Miller33804262001-02-04 23:20:18 +11003 *
4 * Redistribution and use in source and binary forms, with or without
5 * modification, are permitted provided that the following conditions
6 * are met:
7 * 1. Redistributions of source code must retain the above copyright
8 * notice, this list of conditions and the following disclaimer.
9 * 2. Redistributions in binary form must reproduce the above copyright
10 * notice, this list of conditions and the following disclaimer in the
11 * documentation and/or other materials provided with the distribution.
12 *
13 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
14 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
15 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
16 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
17 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
18 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
19 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
20 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
21 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
22 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
23 */
24
25/* XXX: memleaks */
26/* XXX: signed vs unsigned */
Damien Miller3db5f532002-02-13 14:10:32 +110027/* XXX: remove all logging, only return status codes */
Damien Miller33804262001-02-04 23:20:18 +110028/* XXX: copy between two remote sites */
29
30#include "includes.h"
Damien Millerd5580922003-05-14 13:40:06 +100031RCSID("$OpenBSD: sftp-client.c,v 1.43 2003/04/08 20:21:29 itojun Exp $");
Damien Miller16a13332002-02-13 14:03:56 +110032
Damien Miller9b481512002-09-12 10:43:29 +100033#include "openbsd-compat/sys-queue.h"
Damien Miller33804262001-02-04 23:20:18 +110034
Damien Miller33804262001-02-04 23:20:18 +110035#include "buffer.h"
36#include "bufaux.h"
37#include "getput.h"
38#include "xmalloc.h"
39#include "log.h"
40#include "atomicio.h"
Damien Miller62d57f62003-01-10 21:43:24 +110041#include "progressmeter.h"
Damien Miller33804262001-02-04 23:20:18 +110042
43#include "sftp.h"
44#include "sftp-common.h"
45#include "sftp-client.h"
46
Damien Miller62d57f62003-01-10 21:43:24 +110047extern int showprogress;
48
Damien Miller16a13332002-02-13 14:03:56 +110049/* Minimum amount of data to read at at time */
50#define MIN_READ_SIZE 512
51
Damien Millera7f3aaa2003-01-10 21:43:58 +110052/* Maximum packet size */
53#define MAX_MSG_LENGTH (256 * 1024)
54
Damien Miller3db5f532002-02-13 14:10:32 +110055struct sftp_conn {
56 int fd_in;
57 int fd_out;
58 u_int transfer_buflen;
59 u_int num_requests;
60 u_int version;
61 u_int msg_id;
62};
Ben Lindstrom288cc392001-02-09 02:58:04 +000063
Ben Lindstrombba81212001-06-25 05:01:22 +000064static void
Damien Miller33804262001-02-04 23:20:18 +110065send_msg(int fd, Buffer *m)
66{
Damien Millera7f3aaa2003-01-10 21:43:58 +110067 u_char mlen[4];
Damien Miller33804262001-02-04 23:20:18 +110068
Damien Millera7f3aaa2003-01-10 21:43:58 +110069 if (buffer_len(m) > MAX_MSG_LENGTH)
70 fatal("Outbound message too long %u", buffer_len(m));
Damien Miller33804262001-02-04 23:20:18 +110071
Damien Millera7f3aaa2003-01-10 21:43:58 +110072 /* Send length first */
73 PUT_32BIT(mlen, buffer_len(m));
74 if (atomicio(write, fd, mlen, sizeof(mlen)) <= 0)
Damien Miller33804262001-02-04 23:20:18 +110075 fatal("Couldn't send packet: %s", strerror(errno));
76
Damien Millera7f3aaa2003-01-10 21:43:58 +110077 if (atomicio(write, fd, buffer_ptr(m), buffer_len(m)) <= 0)
78 fatal("Couldn't send packet: %s", strerror(errno));
79
80 buffer_clear(m);
Damien Miller33804262001-02-04 23:20:18 +110081}
82
Ben Lindstrombba81212001-06-25 05:01:22 +000083static void
Damien Miller33804262001-02-04 23:20:18 +110084get_msg(int fd, Buffer *m)
85{
Damien Millera7f3aaa2003-01-10 21:43:58 +110086 ssize_t len;
87 u_int msg_len;
Damien Miller33804262001-02-04 23:20:18 +110088
Damien Millera7f3aaa2003-01-10 21:43:58 +110089 buffer_append_space(m, 4);
90 len = atomicio(read, fd, buffer_ptr(m), 4);
Damien Millercafff192001-03-19 22:29:46 +110091 if (len == 0)
92 fatal("Connection closed");
93 else if (len == -1)
Damien Miller33804262001-02-04 23:20:18 +110094 fatal("Couldn't read packet: %s", strerror(errno));
95
Damien Millera7f3aaa2003-01-10 21:43:58 +110096 msg_len = buffer_get_int(m);
97 if (msg_len > MAX_MSG_LENGTH)
Ben Lindstromb1f483f2002-06-23 21:27:18 +000098 fatal("Received message too long %u", msg_len);
Damien Miller33804262001-02-04 23:20:18 +110099
Damien Millera7f3aaa2003-01-10 21:43:58 +1100100 buffer_append_space(m, msg_len);
101 len = atomicio(read, fd, buffer_ptr(m), msg_len);
102 if (len == 0)
103 fatal("Connection closed");
104 else if (len == -1)
105 fatal("Read packet: %s", strerror(errno));
Damien Miller33804262001-02-04 23:20:18 +1100106}
107
Ben Lindstrombba81212001-06-25 05:01:22 +0000108static void
Damien Miller33804262001-02-04 23:20:18 +1100109send_string_request(int fd, u_int id, u_int code, char *s,
110 u_int len)
111{
112 Buffer msg;
113
114 buffer_init(&msg);
115 buffer_put_char(&msg, code);
116 buffer_put_int(&msg, id);
117 buffer_put_string(&msg, s, len);
118 send_msg(fd, &msg);
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000119 debug3("Sent message fd %d T:%u I:%u", fd, code, id);
Damien Miller33804262001-02-04 23:20:18 +1100120 buffer_free(&msg);
121}
122
Ben Lindstrombba81212001-06-25 05:01:22 +0000123static void
Damien Miller33804262001-02-04 23:20:18 +1100124send_string_attrs_request(int fd, u_int id, u_int code, char *s,
125 u_int len, Attrib *a)
126{
127 Buffer msg;
128
129 buffer_init(&msg);
130 buffer_put_char(&msg, code);
131 buffer_put_int(&msg, id);
132 buffer_put_string(&msg, s, len);
133 encode_attrib(&msg, a);
134 send_msg(fd, &msg);
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000135 debug3("Sent message fd %d T:%u I:%u", fd, code, id);
Damien Miller33804262001-02-04 23:20:18 +1100136 buffer_free(&msg);
137}
138
Ben Lindstrombba81212001-06-25 05:01:22 +0000139static u_int
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000140get_status(int fd, u_int expected_id)
Damien Miller33804262001-02-04 23:20:18 +1100141{
142 Buffer msg;
143 u_int type, id, status;
144
145 buffer_init(&msg);
146 get_msg(fd, &msg);
147 type = buffer_get_char(&msg);
148 id = buffer_get_int(&msg);
149
150 if (id != expected_id)
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000151 fatal("ID mismatch (%u != %u)", id, expected_id);
Damien Miller33804262001-02-04 23:20:18 +1100152 if (type != SSH2_FXP_STATUS)
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000153 fatal("Expected SSH2_FXP_STATUS(%u) packet, got %u",
Damien Miller33804262001-02-04 23:20:18 +1100154 SSH2_FXP_STATUS, type);
155
156 status = buffer_get_int(&msg);
157 buffer_free(&msg);
158
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000159 debug3("SSH2_FXP_STATUS %u", status);
Damien Miller33804262001-02-04 23:20:18 +1100160
161 return(status);
162}
163
Ben Lindstrombba81212001-06-25 05:01:22 +0000164static char *
Damien Miller33804262001-02-04 23:20:18 +1100165get_handle(int fd, u_int expected_id, u_int *len)
166{
167 Buffer msg;
168 u_int type, id;
169 char *handle;
170
171 buffer_init(&msg);
172 get_msg(fd, &msg);
173 type = buffer_get_char(&msg);
174 id = buffer_get_int(&msg);
175
176 if (id != expected_id)
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000177 fatal("ID mismatch (%u != %u)", id, expected_id);
Damien Miller33804262001-02-04 23:20:18 +1100178 if (type == SSH2_FXP_STATUS) {
179 int status = buffer_get_int(&msg);
180
181 error("Couldn't get handle: %s", fx2txt(status));
182 return(NULL);
183 } else if (type != SSH2_FXP_HANDLE)
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000184 fatal("Expected SSH2_FXP_HANDLE(%u) packet, got %u",
Damien Miller33804262001-02-04 23:20:18 +1100185 SSH2_FXP_HANDLE, type);
186
187 handle = buffer_get_string(&msg, len);
188 buffer_free(&msg);
189
190 return(handle);
191}
192
Ben Lindstrombba81212001-06-25 05:01:22 +0000193static Attrib *
Ben Lindstromc8d1c302001-03-17 00:34:46 +0000194get_decode_stat(int fd, u_int expected_id, int quiet)
Damien Miller33804262001-02-04 23:20:18 +1100195{
196 Buffer msg;
197 u_int type, id;
198 Attrib *a;
199
200 buffer_init(&msg);
201 get_msg(fd, &msg);
202
203 type = buffer_get_char(&msg);
204 id = buffer_get_int(&msg);
205
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000206 debug3("Received stat reply T:%u I:%u", type, id);
Damien Miller33804262001-02-04 23:20:18 +1100207 if (id != expected_id)
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000208 fatal("ID mismatch (%u != %u)", id, expected_id);
Damien Miller33804262001-02-04 23:20:18 +1100209 if (type == SSH2_FXP_STATUS) {
210 int status = buffer_get_int(&msg);
211
Ben Lindstromc8d1c302001-03-17 00:34:46 +0000212 if (quiet)
213 debug("Couldn't stat remote file: %s", fx2txt(status));
214 else
215 error("Couldn't stat remote file: %s", fx2txt(status));
Damien Miller33804262001-02-04 23:20:18 +1100216 return(NULL);
217 } else if (type != SSH2_FXP_ATTRS) {
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000218 fatal("Expected SSH2_FXP_ATTRS(%u) packet, got %u",
Damien Miller33804262001-02-04 23:20:18 +1100219 SSH2_FXP_ATTRS, type);
220 }
221 a = decode_attrib(&msg);
222 buffer_free(&msg);
223
224 return(a);
225}
226
Damien Miller3db5f532002-02-13 14:10:32 +1100227struct sftp_conn *
228do_init(int fd_in, int fd_out, u_int transfer_buflen, u_int num_requests)
Damien Miller33804262001-02-04 23:20:18 +1100229{
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000230 u_int type;
231 int version;
Damien Miller33804262001-02-04 23:20:18 +1100232 Buffer msg;
Damien Miller3db5f532002-02-13 14:10:32 +1100233 struct sftp_conn *ret;
Damien Miller33804262001-02-04 23:20:18 +1100234
235 buffer_init(&msg);
236 buffer_put_char(&msg, SSH2_FXP_INIT);
237 buffer_put_int(&msg, SSH2_FILEXFER_VERSION);
238 send_msg(fd_out, &msg);
239
240 buffer_clear(&msg);
241
242 get_msg(fd_in, &msg);
243
Kevin Stevesef4eea92001-02-05 12:42:17 +0000244 /* Expecting a VERSION reply */
Damien Miller33804262001-02-04 23:20:18 +1100245 if ((type = buffer_get_char(&msg)) != SSH2_FXP_VERSION) {
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000246 error("Invalid packet back from SSH2_FXP_INIT (type %u)",
Damien Miller33804262001-02-04 23:20:18 +1100247 type);
248 buffer_free(&msg);
Damien Miller3db5f532002-02-13 14:10:32 +1100249 return(NULL);
Damien Miller33804262001-02-04 23:20:18 +1100250 }
251 version = buffer_get_int(&msg);
252
253 debug2("Remote version: %d", version);
254
255 /* Check for extensions */
256 while (buffer_len(&msg) > 0) {
257 char *name = buffer_get_string(&msg, NULL);
258 char *value = buffer_get_string(&msg, NULL);
259
260 debug2("Init extension: \"%s\"", name);
261 xfree(name);
262 xfree(value);
263 }
264
265 buffer_free(&msg);
Damien Miller058316f2001-03-08 10:08:49 +1100266
Damien Miller3db5f532002-02-13 14:10:32 +1100267 ret = xmalloc(sizeof(*ret));
268 ret->fd_in = fd_in;
269 ret->fd_out = fd_out;
270 ret->transfer_buflen = transfer_buflen;
271 ret->num_requests = num_requests;
272 ret->version = version;
273 ret->msg_id = 1;
274
275 /* Some filexfer v.0 servers don't support large packets */
276 if (version == 0)
Ben Lindstroma1d81142002-04-02 20:58:11 +0000277 ret->transfer_buflen = MIN(ret->transfer_buflen, 20480);
Damien Miller3db5f532002-02-13 14:10:32 +1100278
279 return(ret);
280}
281
282u_int
283sftp_proto_version(struct sftp_conn *conn)
284{
285 return(conn->version);
Damien Miller33804262001-02-04 23:20:18 +1100286}
287
288int
Damien Miller3db5f532002-02-13 14:10:32 +1100289do_close(struct sftp_conn *conn, char *handle, u_int handle_len)
Damien Miller33804262001-02-04 23:20:18 +1100290{
291 u_int id, status;
292 Buffer msg;
293
294 buffer_init(&msg);
295
Damien Miller3db5f532002-02-13 14:10:32 +1100296 id = conn->msg_id++;
Damien Miller33804262001-02-04 23:20:18 +1100297 buffer_put_char(&msg, SSH2_FXP_CLOSE);
298 buffer_put_int(&msg, id);
299 buffer_put_string(&msg, handle, handle_len);
Damien Miller3db5f532002-02-13 14:10:32 +1100300 send_msg(conn->fd_out, &msg);
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000301 debug3("Sent message SSH2_FXP_CLOSE I:%u", id);
Damien Miller33804262001-02-04 23:20:18 +1100302
Damien Miller3db5f532002-02-13 14:10:32 +1100303 status = get_status(conn->fd_in, id);
Damien Miller33804262001-02-04 23:20:18 +1100304 if (status != SSH2_FX_OK)
305 error("Couldn't close file: %s", fx2txt(status));
306
307 buffer_free(&msg);
308
309 return(status);
310}
311
Damien Miller4870afd2001-03-14 10:27:09 +1100312
Ben Lindstrombba81212001-06-25 05:01:22 +0000313static int
Damien Miller3db5f532002-02-13 14:10:32 +1100314do_lsreaddir(struct sftp_conn *conn, char *path, int printflag,
Damien Miller4870afd2001-03-14 10:27:09 +1100315 SFTP_DIRENT ***dir)
Damien Miller33804262001-02-04 23:20:18 +1100316{
317 Buffer msg;
Ben Lindstrom025df4a2001-03-14 15:16:34 +0000318 u_int type, id, handle_len, i, expected_id, ents = 0;
Damien Miller33804262001-02-04 23:20:18 +1100319 char *handle;
320
Damien Miller3db5f532002-02-13 14:10:32 +1100321 id = conn->msg_id++;
Damien Miller33804262001-02-04 23:20:18 +1100322
323 buffer_init(&msg);
324 buffer_put_char(&msg, SSH2_FXP_OPENDIR);
325 buffer_put_int(&msg, id);
326 buffer_put_cstring(&msg, path);
Damien Miller3db5f532002-02-13 14:10:32 +1100327 send_msg(conn->fd_out, &msg);
Damien Miller33804262001-02-04 23:20:18 +1100328
329 buffer_clear(&msg);
330
Damien Miller3db5f532002-02-13 14:10:32 +1100331 handle = get_handle(conn->fd_in, id, &handle_len);
Damien Miller33804262001-02-04 23:20:18 +1100332 if (handle == NULL)
333 return(-1);
334
Damien Miller4870afd2001-03-14 10:27:09 +1100335 if (dir) {
336 ents = 0;
337 *dir = xmalloc(sizeof(**dir));
338 (*dir)[0] = NULL;
339 }
Damien Miller4870afd2001-03-14 10:27:09 +1100340
Damien Miller9f0f5c62001-12-21 14:45:46 +1100341 for (;;) {
Damien Miller33804262001-02-04 23:20:18 +1100342 int count;
343
Damien Miller3db5f532002-02-13 14:10:32 +1100344 id = expected_id = conn->msg_id++;
Damien Miller33804262001-02-04 23:20:18 +1100345
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000346 debug3("Sending SSH2_FXP_READDIR I:%u", id);
Damien Miller33804262001-02-04 23:20:18 +1100347
348 buffer_clear(&msg);
349 buffer_put_char(&msg, SSH2_FXP_READDIR);
350 buffer_put_int(&msg, id);
351 buffer_put_string(&msg, handle, handle_len);
Damien Miller3db5f532002-02-13 14:10:32 +1100352 send_msg(conn->fd_out, &msg);
Damien Miller33804262001-02-04 23:20:18 +1100353
354 buffer_clear(&msg);
355
Damien Miller3db5f532002-02-13 14:10:32 +1100356 get_msg(conn->fd_in, &msg);
Damien Miller33804262001-02-04 23:20:18 +1100357
358 type = buffer_get_char(&msg);
359 id = buffer_get_int(&msg);
360
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000361 debug3("Received reply T:%u I:%u", type, id);
Damien Miller33804262001-02-04 23:20:18 +1100362
363 if (id != expected_id)
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000364 fatal("ID mismatch (%u != %u)", id, expected_id);
Damien Miller33804262001-02-04 23:20:18 +1100365
366 if (type == SSH2_FXP_STATUS) {
367 int status = buffer_get_int(&msg);
368
369 debug3("Received SSH2_FXP_STATUS %d", status);
370
371 if (status == SSH2_FX_EOF) {
372 break;
373 } else {
374 error("Couldn't read directory: %s",
375 fx2txt(status));
Damien Miller3db5f532002-02-13 14:10:32 +1100376 do_close(conn, handle, handle_len);
Damien Miller00111382003-03-10 11:21:17 +1100377 xfree(handle);
Ben Lindstrom10ac33f2001-02-10 21:53:40 +0000378 return(status);
Damien Miller33804262001-02-04 23:20:18 +1100379 }
380 } else if (type != SSH2_FXP_NAME)
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000381 fatal("Expected SSH2_FXP_NAME(%u) packet, got %u",
Damien Miller33804262001-02-04 23:20:18 +1100382 SSH2_FXP_NAME, type);
383
384 count = buffer_get_int(&msg);
Damien Millerd7686fd2001-02-10 00:40:03 +1100385 if (count == 0)
386 break;
387 debug3("Received %d SSH2_FXP_NAME responses", count);
Damien Miller9f0f5c62001-12-21 14:45:46 +1100388 for (i = 0; i < count; i++) {
Damien Miller33804262001-02-04 23:20:18 +1100389 char *filename, *longname;
390 Attrib *a;
391
392 filename = buffer_get_string(&msg, NULL);
393 longname = buffer_get_string(&msg, NULL);
394 a = decode_attrib(&msg);
395
Damien Miller4870afd2001-03-14 10:27:09 +1100396 if (printflag)
397 printf("%s\n", longname);
398
399 if (dir) {
Ben Lindstroma3700052001-04-05 23:26:32 +0000400 *dir = xrealloc(*dir, sizeof(**dir) *
Damien Miller4870afd2001-03-14 10:27:09 +1100401 (ents + 2));
402 (*dir)[ents] = xmalloc(sizeof(***dir));
403 (*dir)[ents]->filename = xstrdup(filename);
404 (*dir)[ents]->longname = xstrdup(longname);
405 memcpy(&(*dir)[ents]->a, a, sizeof(*a));
406 (*dir)[++ents] = NULL;
407 }
Damien Miller33804262001-02-04 23:20:18 +1100408
409 xfree(filename);
410 xfree(longname);
411 }
412 }
413
414 buffer_free(&msg);
Damien Miller3db5f532002-02-13 14:10:32 +1100415 do_close(conn, handle, handle_len);
Damien Miller33804262001-02-04 23:20:18 +1100416 xfree(handle);
417
418 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
654 /* Send rename 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 Miller770b3742003-01-08 14:04:53 +1100801 local_fd = open(local_path, O_WRONLY | O_CREAT | O_TRUNC,
802 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
816 if (showprogress) {
817 if (size)
818 start_progress_meter(remote_path, size,
819 &progress_counter);
820 else
821 printf("Fetching %s to %s\n", remote_path, local_path);
822 }
823
Damien Miller16a13332002-02-13 14:03:56 +1100824 while (num_req > 0 || max_req > 0) {
Damien Miller33804262001-02-04 23:20:18 +1100825 char *data;
Damien Miller16a13332002-02-13 14:03:56 +1100826 u_int len;
Damien Miller33804262001-02-04 23:20:18 +1100827
Damien Miller16a13332002-02-13 14:03:56 +1100828 /* Send some more requests */
829 while (num_req < max_req) {
Ben Lindstrom6328ab32002-03-22 02:54:23 +0000830 debug3("Request range %llu -> %llu (%d/%d)",
Ben Lindstromd45f28c2002-03-22 01:00:57 +0000831 (unsigned long long)offset,
832 (unsigned long long)offset + buflen - 1,
833 num_req, max_req);
Damien Miller16a13332002-02-13 14:03:56 +1100834 req = xmalloc(sizeof(*req));
Damien Miller3db5f532002-02-13 14:10:32 +1100835 req->id = conn->msg_id++;
Damien Miller16a13332002-02-13 14:03:56 +1100836 req->len = buflen;
837 req->offset = offset;
838 offset += buflen;
839 num_req++;
840 TAILQ_INSERT_TAIL(&requests, req, tq);
Ben Lindstrom6328ab32002-03-22 02:54:23 +0000841 send_read_request(conn->fd_out, req->id, req->offset,
Damien Miller16a13332002-02-13 14:03:56 +1100842 req->len, handle, handle_len);
843 }
Damien Miller33804262001-02-04 23:20:18 +1100844
845 buffer_clear(&msg);
Damien Miller3db5f532002-02-13 14:10:32 +1100846 get_msg(conn->fd_in, &msg);
Damien Miller33804262001-02-04 23:20:18 +1100847 type = buffer_get_char(&msg);
848 id = buffer_get_int(&msg);
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000849 debug3("Received reply T:%u I:%u R:%d", type, id, max_req);
Damien Miller33804262001-02-04 23:20:18 +1100850
Damien Miller16a13332002-02-13 14:03:56 +1100851 /* Find the request in our queue */
852 for(req = TAILQ_FIRST(&requests);
853 req != NULL && req->id != id;
854 req = TAILQ_NEXT(req, tq))
855 ;
856 if (req == NULL)
857 fatal("Unexpected reply %u", id);
858
859 switch (type) {
860 case SSH2_FXP_STATUS:
861 status = buffer_get_int(&msg);
862 if (status != SSH2_FX_EOF)
863 read_error = 1;
864 max_req = 0;
865 TAILQ_REMOVE(&requests, req, tq);
866 xfree(req);
867 num_req--;
868 break;
869 case SSH2_FXP_DATA:
870 data = buffer_get_string(&msg, &len);
Ben Lindstromeb505452002-03-22 01:03:15 +0000871 debug3("Received data %llu -> %llu",
Ben Lindstrom6328ab32002-03-22 02:54:23 +0000872 (unsigned long long)req->offset,
Ben Lindstromeb505452002-03-22 01:03:15 +0000873 (unsigned long long)req->offset + len - 1);
Damien Miller16a13332002-02-13 14:03:56 +1100874 if (len > req->len)
875 fatal("Received more data than asked for "
Ben Lindstrom93576d92002-12-23 02:06:19 +0000876 "%u > %u", len, req->len);
Damien Miller16a13332002-02-13 14:03:56 +1100877 if ((lseek(local_fd, req->offset, SEEK_SET) == -1 ||
Ben Lindstrom93576d92002-12-23 02:06:19 +0000878 atomicio(write, local_fd, data, len) != len) &&
Damien Miller16a13332002-02-13 14:03:56 +1100879 !write_error) {
880 write_errno = errno;
881 write_error = 1;
882 max_req = 0;
Damien Miller33804262001-02-04 23:20:18 +1100883 }
Damien Miller62d57f62003-01-10 21:43:24 +1100884 progress_counter += len;
Damien Miller16a13332002-02-13 14:03:56 +1100885 xfree(data);
886
887 if (len == req->len) {
888 TAILQ_REMOVE(&requests, req, tq);
889 xfree(req);
890 num_req--;
891 } else {
892 /* Resend the request for the missing data */
893 debug3("Short data block, re-requesting "
Ben Lindstromeb505452002-03-22 01:03:15 +0000894 "%llu -> %llu (%2d)",
Ben Lindstrom6328ab32002-03-22 02:54:23 +0000895 (unsigned long long)req->offset + len,
Ben Lindstrom83b79e42002-03-22 01:05:27 +0000896 (unsigned long long)req->offset +
897 req->len - 1, num_req);
Damien Miller3db5f532002-02-13 14:10:32 +1100898 req->id = conn->msg_id++;
Damien Miller16a13332002-02-13 14:03:56 +1100899 req->len -= len;
900 req->offset += len;
Ben Lindstrom6328ab32002-03-22 02:54:23 +0000901 send_read_request(conn->fd_out, req->id,
Damien Miller3db5f532002-02-13 14:10:32 +1100902 req->offset, req->len, handle, handle_len);
Damien Miller16a13332002-02-13 14:03:56 +1100903 /* Reduce the request size */
904 if (len < buflen)
905 buflen = MAX(MIN_READ_SIZE, len);
906 }
907 if (max_req > 0) { /* max_req = 0 iff EOF received */
908 if (size > 0 && offset > size) {
909 /* Only one request at a time
910 * after the expected EOF */
911 debug3("Finish at %llu (%2d)",
Ben Lindstromeb505452002-03-22 01:03:15 +0000912 (unsigned long long)offset,
913 num_req);
Damien Miller16a13332002-02-13 14:03:56 +1100914 max_req = 1;
915 }
Damien Miller3db5f532002-02-13 14:10:32 +1100916 else if (max_req < conn->num_requests + 1) {
Damien Miller16a13332002-02-13 14:03:56 +1100917 ++max_req;
918 }
919 }
920 break;
921 default:
Ben Lindstromb1f483f2002-06-23 21:27:18 +0000922 fatal("Expected SSH2_FXP_DATA(%u) packet, got %u",
Damien Miller33804262001-02-04 23:20:18 +1100923 SSH2_FXP_DATA, type);
924 }
Damien Miller33804262001-02-04 23:20:18 +1100925 }
Damien Miller33804262001-02-04 23:20:18 +1100926
Damien Miller62d57f62003-01-10 21:43:24 +1100927 if (showprogress && size)
928 stop_progress_meter();
929
Damien Miller16a13332002-02-13 14:03:56 +1100930 /* Sanity check */
931 if (TAILQ_FIRST(&requests) != NULL)
932 fatal("Transfer complete, but requests still in queue");
933
934 if (read_error) {
Ben Lindstrom6328ab32002-03-22 02:54:23 +0000935 error("Couldn't read from remote file \"%s\" : %s",
Damien Miller3db5f532002-02-13 14:10:32 +1100936 remote_path, fx2txt(status));
937 do_close(conn, handle, handle_len);
Damien Miller16a13332002-02-13 14:03:56 +1100938 } else if (write_error) {
Damien Miller3db5f532002-02-13 14:10:32 +1100939 error("Couldn't write to \"%s\": %s", local_path,
940 strerror(write_errno));
941 status = -1;
942 do_close(conn, handle, handle_len);
Damien Miller16a13332002-02-13 14:03:56 +1100943 } else {
Damien Miller3db5f532002-02-13 14:10:32 +1100944 status = do_close(conn, handle, handle_len);
Damien Miller16a13332002-02-13 14:03:56 +1100945
946 /* Override umask and utimes if asked */
Ben Lindstrom6dc75f52001-02-17 16:47:47 +0000947#ifdef HAVE_FCHMOD
Damien Miller16a13332002-02-13 14:03:56 +1100948 if (pflag && fchmod(local_fd, mode) == -1)
Ben Lindstrom6dc75f52001-02-17 16:47:47 +0000949#else
Damien Miller16a13332002-02-13 14:03:56 +1100950 if (pflag && chmod(local_path, mode) == -1)
Ben Lindstrom6dc75f52001-02-17 16:47:47 +0000951#endif /* HAVE_FCHMOD */
Damien Miller16a13332002-02-13 14:03:56 +1100952 error("Couldn't set mode on \"%s\": %s", local_path,
Ben Lindstrom93576d92002-12-23 02:06:19 +0000953 strerror(errno));
Damien Miller16a13332002-02-13 14:03:56 +1100954 if (pflag && (a->flags & SSH2_FILEXFER_ATTR_ACMODTIME)) {
955 struct timeval tv[2];
956 tv[0].tv_sec = a->atime;
957 tv[1].tv_sec = a->mtime;
958 tv[0].tv_usec = tv[1].tv_usec = 0;
959 if (utimes(local_path, tv) == -1)
960 error("Can't set times on \"%s\": %s",
Ben Lindstrom93576d92002-12-23 02:06:19 +0000961 local_path, strerror(errno));
Damien Miller16a13332002-02-13 14:03:56 +1100962 }
Ben Lindstrom9d4f2c82001-02-15 03:22:45 +0000963 }
Damien Millerd7686fd2001-02-10 00:40:03 +1100964 close(local_fd);
965 buffer_free(&msg);
966 xfree(handle);
Damien Miller3db5f532002-02-13 14:10:32 +1100967
968 return(status);
Damien Miller33804262001-02-04 23:20:18 +1100969}
970
971int
Damien Miller3db5f532002-02-13 14:10:32 +1100972do_upload(struct sftp_conn *conn, char *local_path, char *remote_path,
973 int pflag)
Damien Miller33804262001-02-04 23:20:18 +1100974{
Damien Miller8829d362002-02-08 22:04:05 +1100975 int local_fd, status;
Damien Miller5873dfd2002-02-13 14:04:37 +1100976 u_int handle_len, id, type;
Damien Miller33804262001-02-04 23:20:18 +1100977 u_int64_t offset;
Damien Miller8829d362002-02-08 22:04:05 +1100978 char *handle, *data;
Damien Miller33804262001-02-04 23:20:18 +1100979 Buffer msg;
980 struct stat sb;
981 Attrib a;
Damien Miller16a13332002-02-13 14:03:56 +1100982 u_int32_t startid;
983 u_int32_t ackid;
Damien Miller5873dfd2002-02-13 14:04:37 +1100984 struct outstanding_ack {
985 u_int id;
986 u_int len;
987 u_int64_t offset;
Ben Lindstrom6328ab32002-03-22 02:54:23 +0000988 TAILQ_ENTRY(outstanding_ack) tq;
Damien Miller5873dfd2002-02-13 14:04:37 +1100989 };
990 TAILQ_HEAD(ackhead, outstanding_ack) acks;
991 struct outstanding_ack *ack;
992
993 TAILQ_INIT(&acks);
Damien Miller33804262001-02-04 23:20:18 +1100994
995 if ((local_fd = open(local_path, O_RDONLY, 0)) == -1) {
996 error("Couldn't open local file \"%s\" for reading: %s",
997 local_path, strerror(errno));
998 return(-1);
999 }
1000 if (fstat(local_fd, &sb) == -1) {
1001 error("Couldn't fstat local file \"%s\": %s",
1002 local_path, strerror(errno));
1003 close(local_fd);
1004 return(-1);
1005 }
Damien Miller5fa01fd2003-01-14 22:24:47 +11001006 if (!S_ISREG(sb.st_mode)) {
1007 error("%s is not a regular file", local_path);
1008 close(local_fd);
1009 return(-1);
1010 }
Damien Miller33804262001-02-04 23:20:18 +11001011 stat_to_attrib(&sb, &a);
1012
1013 a.flags &= ~SSH2_FILEXFER_ATTR_SIZE;
1014 a.flags &= ~SSH2_FILEXFER_ATTR_UIDGID;
1015 a.perm &= 0777;
1016 if (!pflag)
1017 a.flags &= ~SSH2_FILEXFER_ATTR_ACMODTIME;
1018
1019 buffer_init(&msg);
1020
1021 /* Send open request */
Damien Miller3db5f532002-02-13 14:10:32 +11001022 id = conn->msg_id++;
Damien Miller33804262001-02-04 23:20:18 +11001023 buffer_put_char(&msg, SSH2_FXP_OPEN);
1024 buffer_put_int(&msg, id);
1025 buffer_put_cstring(&msg, remote_path);
1026 buffer_put_int(&msg, SSH2_FXF_WRITE|SSH2_FXF_CREAT|SSH2_FXF_TRUNC);
1027 encode_attrib(&msg, &a);
Damien Miller3db5f532002-02-13 14:10:32 +11001028 send_msg(conn->fd_out, &msg);
Ben Lindstromb1f483f2002-06-23 21:27:18 +00001029 debug3("Sent message SSH2_FXP_OPEN I:%u P:%s", id, remote_path);
Damien Miller33804262001-02-04 23:20:18 +11001030
1031 buffer_clear(&msg);
1032
Damien Miller3db5f532002-02-13 14:10:32 +11001033 handle = get_handle(conn->fd_in, id, &handle_len);
Damien Miller33804262001-02-04 23:20:18 +11001034 if (handle == NULL) {
1035 close(local_fd);
1036 buffer_free(&msg);
1037 return(-1);
1038 }
1039
Damien Miller16a13332002-02-13 14:03:56 +11001040 startid = ackid = id + 1;
Damien Miller3db5f532002-02-13 14:10:32 +11001041 data = xmalloc(conn->transfer_buflen);
Damien Miller8829d362002-02-08 22:04:05 +11001042
Damien Miller33804262001-02-04 23:20:18 +11001043 /* Read from local and write to remote */
1044 offset = 0;
Damien Miller62d57f62003-01-10 21:43:24 +11001045 if (showprogress)
1046 start_progress_meter(local_path, sb.st_size, &offset);
1047 else
1048 printf("Uploading %s to %s\n", local_path, remote_path);
1049
Damien Miller9f0f5c62001-12-21 14:45:46 +11001050 for (;;) {
Damien Miller33804262001-02-04 23:20:18 +11001051 int len;
Damien Miller33804262001-02-04 23:20:18 +11001052
1053 /*
1054 * Can't use atomicio here because it returns 0 on EOF, thus losing
1055 * the last block of the file
1056 */
1057 do
Damien Miller3db5f532002-02-13 14:10:32 +11001058 len = read(local_fd, data, conn->transfer_buflen);
Damien Miller33804262001-02-04 23:20:18 +11001059 while ((len == -1) && (errno == EINTR || errno == EAGAIN));
1060
1061 if (len == -1)
1062 fatal("Couldn't read from \"%s\": %s", local_path,
1063 strerror(errno));
Damien Miller16a13332002-02-13 14:03:56 +11001064
1065 if (len != 0) {
Damien Miller5873dfd2002-02-13 14:04:37 +11001066 ack = xmalloc(sizeof(*ack));
1067 ack->id = ++id;
1068 ack->offset = offset;
1069 ack->len = len;
1070 TAILQ_INSERT_TAIL(&acks, ack, tq);
1071
Damien Miller16a13332002-02-13 14:03:56 +11001072 buffer_clear(&msg);
1073 buffer_put_char(&msg, SSH2_FXP_WRITE);
Damien Miller5873dfd2002-02-13 14:04:37 +11001074 buffer_put_int(&msg, ack->id);
Damien Miller16a13332002-02-13 14:03:56 +11001075 buffer_put_string(&msg, handle, handle_len);
1076 buffer_put_int64(&msg, offset);
1077 buffer_put_string(&msg, data, len);
Damien Miller3db5f532002-02-13 14:10:32 +11001078 send_msg(conn->fd_out, &msg);
Ben Lindstromb1f483f2002-06-23 21:27:18 +00001079 debug3("Sent message SSH2_FXP_WRITE I:%u O:%llu S:%u",
Ben Lindstrom93576d92002-12-23 02:06:19 +00001080 id, (unsigned long long)offset, len);
Damien Miller5873dfd2002-02-13 14:04:37 +11001081 } else if (TAILQ_FIRST(&acks) == NULL)
Damien Miller33804262001-02-04 23:20:18 +11001082 break;
1083
Damien Miller5873dfd2002-02-13 14:04:37 +11001084 if (ack == NULL)
1085 fatal("Unexpected ACK %u", id);
1086
Ben Lindstrom6328ab32002-03-22 02:54:23 +00001087 if (id == startid || len == 0 ||
Damien Miller3db5f532002-02-13 14:10:32 +11001088 id - ackid >= conn->num_requests) {
Ben Lindstrom5a6abda2002-06-09 19:41:48 +00001089 u_int r_id;
Ben Lindstrom06e95152002-04-06 04:16:45 +00001090
Damien Miller5873dfd2002-02-13 14:04:37 +11001091 buffer_clear(&msg);
Damien Miller3db5f532002-02-13 14:10:32 +11001092 get_msg(conn->fd_in, &msg);
Damien Miller5873dfd2002-02-13 14:04:37 +11001093 type = buffer_get_char(&msg);
Ben Lindstrom06e95152002-04-06 04:16:45 +00001094 r_id = buffer_get_int(&msg);
Damien Miller5873dfd2002-02-13 14:04:37 +11001095
1096 if (type != SSH2_FXP_STATUS)
1097 fatal("Expected SSH2_FXP_STATUS(%d) packet, "
1098 "got %d", SSH2_FXP_STATUS, type);
1099
1100 status = buffer_get_int(&msg);
1101 debug3("SSH2_FXP_STATUS %d", status);
1102
1103 /* Find the request in our queue */
1104 for(ack = TAILQ_FIRST(&acks);
Ben Lindstrom06e95152002-04-06 04:16:45 +00001105 ack != NULL && ack->id != r_id;
Damien Miller5873dfd2002-02-13 14:04:37 +11001106 ack = TAILQ_NEXT(ack, tq))
1107 ;
1108 if (ack == NULL)
Ben Lindstromb1f483f2002-06-23 21:27:18 +00001109 fatal("Can't find request for ID %u", r_id);
Damien Miller5873dfd2002-02-13 14:04:37 +11001110 TAILQ_REMOVE(&acks, ack, tq);
1111
Damien Miller16a13332002-02-13 14:03:56 +11001112 if (status != SSH2_FX_OK) {
1113 error("Couldn't write to remote file \"%s\": %s",
Ben Lindstrom93576d92002-12-23 02:06:19 +00001114 remote_path, fx2txt(status));
Damien Miller3db5f532002-02-13 14:10:32 +11001115 do_close(conn, handle, handle_len);
Damien Miller16a13332002-02-13 14:03:56 +11001116 close(local_fd);
Damien Miller00111382003-03-10 11:21:17 +11001117 xfree(data);
1118 xfree(ack);
Damien Miller16a13332002-02-13 14:03:56 +11001119 goto done;
1120 }
Ben Lindstromb1f483f2002-06-23 21:27:18 +00001121 debug3("In write loop, ack for %u %u bytes at %llu",
Ben Lindstromeb505452002-03-22 01:03:15 +00001122 ack->id, ack->len, (unsigned long long)ack->offset);
Damien Miller16a13332002-02-13 14:03:56 +11001123 ++ackid;
Ben Lindstromeec16fc2002-07-04 00:06:15 +00001124 xfree(ack);
Damien Miller33804262001-02-04 23:20:18 +11001125 }
Damien Miller33804262001-02-04 23:20:18 +11001126 offset += len;
1127 }
Damien Miller62d57f62003-01-10 21:43:24 +11001128 if (showprogress)
1129 stop_progress_meter();
Damien Miller8829d362002-02-08 22:04:05 +11001130 xfree(data);
Damien Miller33804262001-02-04 23:20:18 +11001131
1132 if (close(local_fd) == -1) {
1133 error("Couldn't close local file \"%s\": %s", local_path,
1134 strerror(errno));
Damien Miller3db5f532002-02-13 14:10:32 +11001135 do_close(conn, handle, handle_len);
Damien Millerd7686fd2001-02-10 00:40:03 +11001136 status = -1;
1137 goto done;
Damien Miller33804262001-02-04 23:20:18 +11001138 }
1139
Ben Lindstrom9d4f2c82001-02-15 03:22:45 +00001140 /* Override umask and utimes if asked */
1141 if (pflag)
Damien Miller3db5f532002-02-13 14:10:32 +11001142 do_fsetstat(conn, handle, handle_len, &a);
Ben Lindstrom9d4f2c82001-02-15 03:22:45 +00001143
Damien Miller3db5f532002-02-13 14:10:32 +11001144 status = do_close(conn, handle, handle_len);
Damien Millerd7686fd2001-02-10 00:40:03 +11001145
1146done:
1147 xfree(handle);
1148 buffer_free(&msg);
Damien Miller3db5f532002-02-13 14:10:32 +11001149 return(status);
Damien Miller33804262001-02-04 23:20:18 +11001150}