blob: 8338dbc9f7109ff999789059c88732a52a6aee23 [file] [log] [blame]
Damien Miller33804262001-02-04 23:20:18 +11001/*
2 * Copyright (c) 2001 Damien Miller. All rights reserved.
3 *
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 */
27/* XXX: redesign to allow concurrent overlapped operations */
28/* XXX: we use fatal too much, error may be more appropriate in places */
29/* XXX: copy between two remote sites */
30
31#include "includes.h"
Ben Lindstrom10ac33f2001-02-10 21:53:40 +000032RCSID("$OpenBSD: sftp-client.c,v 1.9 2001/02/10 00:41:46 djm Exp $");
Damien Miller33804262001-02-04 23:20:18 +110033
34#include "ssh.h"
35#include "buffer.h"
36#include "bufaux.h"
37#include "getput.h"
38#include "xmalloc.h"
39#include "log.h"
40#include "atomicio.h"
41#include "pathnames.h"
42
43#include "sftp.h"
44#include "sftp-common.h"
45#include "sftp-client.h"
46
47/* How much data to read/write at at time during copies */
48/* XXX: what should this be? */
49#define COPY_SIZE 8192
50
Ben Lindstrom288cc392001-02-09 02:58:04 +000051/* Message ID */
52static u_int msg_id = 1;
53
Damien Miller33804262001-02-04 23:20:18 +110054void
55send_msg(int fd, Buffer *m)
56{
57 int mlen = buffer_len(m);
58 int len;
59 Buffer oqueue;
60
61 buffer_init(&oqueue);
62 buffer_put_int(&oqueue, mlen);
63 buffer_append(&oqueue, buffer_ptr(m), mlen);
64 buffer_consume(m, mlen);
65
66 len = atomicio(write, fd, buffer_ptr(&oqueue), buffer_len(&oqueue));
67 if (len <= 0)
68 fatal("Couldn't send packet: %s", strerror(errno));
69
70 buffer_free(&oqueue);
71}
72
73void
74get_msg(int fd, Buffer *m)
75{
76 u_int len, msg_len;
77 unsigned char buf[4096];
78
79 len = atomicio(read, fd, buf, 4);
80 if (len != 4)
81 fatal("Couldn't read packet: %s", strerror(errno));
82
83 msg_len = GET_32BIT(buf);
84 if (msg_len > 256 * 1024)
85 fatal("Received message too long %d", msg_len);
86
87 while (msg_len) {
88 len = atomicio(read, fd, buf, MIN(msg_len, sizeof(buf)));
89 if (len <= 0)
90 fatal("Couldn't read packet: %s", strerror(errno));
91
92 msg_len -= len;
93 buffer_append(m, buf, len);
94 }
95}
96
97void
98send_string_request(int fd, u_int id, u_int code, char *s,
99 u_int len)
100{
101 Buffer msg;
102
103 buffer_init(&msg);
104 buffer_put_char(&msg, code);
105 buffer_put_int(&msg, id);
106 buffer_put_string(&msg, s, len);
107 send_msg(fd, &msg);
108 debug3("Sent message fd %d T:%d I:%d", fd, code, id);
109 buffer_free(&msg);
110}
111
112void
113send_string_attrs_request(int fd, u_int id, u_int code, char *s,
114 u_int len, Attrib *a)
115{
116 Buffer msg;
117
118 buffer_init(&msg);
119 buffer_put_char(&msg, code);
120 buffer_put_int(&msg, id);
121 buffer_put_string(&msg, s, len);
122 encode_attrib(&msg, a);
123 send_msg(fd, &msg);
124 debug3("Sent message fd %d T:%d I:%d", fd, code, id);
125 buffer_free(&msg);
126}
127
128u_int
129get_status(int fd, int expected_id)
130{
131 Buffer msg;
132 u_int type, id, status;
133
134 buffer_init(&msg);
135 get_msg(fd, &msg);
136 type = buffer_get_char(&msg);
137 id = buffer_get_int(&msg);
138
139 if (id != expected_id)
140 fatal("ID mismatch (%d != %d)", id, expected_id);
141 if (type != SSH2_FXP_STATUS)
142 fatal("Expected SSH2_FXP_STATUS(%d) packet, got %d",
143 SSH2_FXP_STATUS, type);
144
145 status = buffer_get_int(&msg);
146 buffer_free(&msg);
147
148 debug3("SSH2_FXP_STATUS %d", status);
149
150 return(status);
151}
152
153char *
154get_handle(int fd, u_int expected_id, u_int *len)
155{
156 Buffer msg;
157 u_int type, id;
158 char *handle;
159
160 buffer_init(&msg);
161 get_msg(fd, &msg);
162 type = buffer_get_char(&msg);
163 id = buffer_get_int(&msg);
164
165 if (id != expected_id)
166 fatal("ID mismatch (%d != %d)", id, expected_id);
167 if (type == SSH2_FXP_STATUS) {
168 int status = buffer_get_int(&msg);
169
170 error("Couldn't get handle: %s", fx2txt(status));
171 return(NULL);
172 } else if (type != SSH2_FXP_HANDLE)
173 fatal("Expected SSH2_FXP_HANDLE(%d) packet, got %d",
174 SSH2_FXP_HANDLE, type);
175
176 handle = buffer_get_string(&msg, len);
177 buffer_free(&msg);
178
179 return(handle);
180}
181
182Attrib *
183get_decode_stat(int fd, u_int expected_id)
184{
185 Buffer msg;
186 u_int type, id;
187 Attrib *a;
188
189 buffer_init(&msg);
190 get_msg(fd, &msg);
191
192 type = buffer_get_char(&msg);
193 id = buffer_get_int(&msg);
194
195 debug3("Received stat reply T:%d I:%d", type, id);
196 if (id != expected_id)
197 fatal("ID mismatch (%d != %d)", id, expected_id);
198 if (type == SSH2_FXP_STATUS) {
199 int status = buffer_get_int(&msg);
200
201 error("Couldn't stat remote file: %s", fx2txt(status));
202 return(NULL);
203 } else if (type != SSH2_FXP_ATTRS) {
204 fatal("Expected SSH2_FXP_ATTRS(%d) packet, got %d",
205 SSH2_FXP_ATTRS, type);
206 }
207 a = decode_attrib(&msg);
208 buffer_free(&msg);
209
210 return(a);
211}
212
213int
214do_init(int fd_in, int fd_out)
215{
216 int type, version;
217 Buffer msg;
218
219 buffer_init(&msg);
220 buffer_put_char(&msg, SSH2_FXP_INIT);
221 buffer_put_int(&msg, SSH2_FILEXFER_VERSION);
222 send_msg(fd_out, &msg);
223
224 buffer_clear(&msg);
225
226 get_msg(fd_in, &msg);
227
Kevin Stevesef4eea92001-02-05 12:42:17 +0000228 /* Expecting a VERSION reply */
Damien Miller33804262001-02-04 23:20:18 +1100229 if ((type = buffer_get_char(&msg)) != SSH2_FXP_VERSION) {
230 error("Invalid packet back from SSH2_FXP_INIT (type %d)",
231 type);
232 buffer_free(&msg);
233 return(-1);
234 }
235 version = buffer_get_int(&msg);
236
237 debug2("Remote version: %d", version);
238
239 /* Check for extensions */
240 while (buffer_len(&msg) > 0) {
241 char *name = buffer_get_string(&msg, NULL);
242 char *value = buffer_get_string(&msg, NULL);
243
244 debug2("Init extension: \"%s\"", name);
245 xfree(name);
246 xfree(value);
247 }
248
249 buffer_free(&msg);
250 return(0);
251}
252
253int
254do_close(int fd_in, int fd_out, char *handle, u_int handle_len)
255{
256 u_int id, status;
257 Buffer msg;
258
259 buffer_init(&msg);
260
Ben Lindstrom288cc392001-02-09 02:58:04 +0000261 id = msg_id++;
Damien Miller33804262001-02-04 23:20:18 +1100262 buffer_put_char(&msg, SSH2_FXP_CLOSE);
263 buffer_put_int(&msg, id);
264 buffer_put_string(&msg, handle, handle_len);
265 send_msg(fd_out, &msg);
266 debug3("Sent message SSH2_FXP_CLOSE I:%d", id);
267
268 status = get_status(fd_in, id);
269 if (status != SSH2_FX_OK)
270 error("Couldn't close file: %s", fx2txt(status));
271
272 buffer_free(&msg);
273
274 return(status);
275}
276
277int
278do_ls(int fd_in, int fd_out, char *path)
279{
280 Buffer msg;
281 u_int type, id, handle_len, i, expected_id;
282 char *handle;
283
Ben Lindstrom288cc392001-02-09 02:58:04 +0000284 id = msg_id++;
Damien Miller33804262001-02-04 23:20:18 +1100285
286 buffer_init(&msg);
287 buffer_put_char(&msg, SSH2_FXP_OPENDIR);
288 buffer_put_int(&msg, id);
289 buffer_put_cstring(&msg, path);
290 send_msg(fd_out, &msg);
291
292 buffer_clear(&msg);
293
294 handle = get_handle(fd_in, id, &handle_len);
295 if (handle == NULL)
296 return(-1);
297
298 for(;;) {
299 int count;
300
Ben Lindstrom288cc392001-02-09 02:58:04 +0000301 id = expected_id = msg_id++;
Damien Miller33804262001-02-04 23:20:18 +1100302
303 debug3("Sending SSH2_FXP_READDIR I:%d", id);
304
305 buffer_clear(&msg);
306 buffer_put_char(&msg, SSH2_FXP_READDIR);
307 buffer_put_int(&msg, id);
308 buffer_put_string(&msg, handle, handle_len);
309 send_msg(fd_out, &msg);
310
311 buffer_clear(&msg);
312
313 get_msg(fd_in, &msg);
314
315 type = buffer_get_char(&msg);
316 id = buffer_get_int(&msg);
317
318 debug3("Received reply T:%d I:%d", type, id);
319
320 if (id != expected_id)
321 fatal("ID mismatch (%d != %d)", id, expected_id);
322
323 if (type == SSH2_FXP_STATUS) {
324 int status = buffer_get_int(&msg);
325
326 debug3("Received SSH2_FXP_STATUS %d", status);
327
328 if (status == SSH2_FX_EOF) {
329 break;
330 } else {
331 error("Couldn't read directory: %s",
332 fx2txt(status));
333 do_close(fd_in, fd_out, handle, handle_len);
Ben Lindstrom10ac33f2001-02-10 21:53:40 +0000334 return(status);
Damien Miller33804262001-02-04 23:20:18 +1100335 }
336 } else if (type != SSH2_FXP_NAME)
337 fatal("Expected SSH2_FXP_NAME(%d) packet, got %d",
338 SSH2_FXP_NAME, type);
339
340 count = buffer_get_int(&msg);
Damien Millerd7686fd2001-02-10 00:40:03 +1100341 if (count == 0)
342 break;
343 debug3("Received %d SSH2_FXP_NAME responses", count);
Damien Miller33804262001-02-04 23:20:18 +1100344 for(i = 0; i < count; i++) {
345 char *filename, *longname;
346 Attrib *a;
347
348 filename = buffer_get_string(&msg, NULL);
349 longname = buffer_get_string(&msg, NULL);
350 a = decode_attrib(&msg);
351
352 printf("%s\n", longname);
353
354 xfree(filename);
355 xfree(longname);
356 }
357 }
358
359 buffer_free(&msg);
360 do_close(fd_in, fd_out, handle, handle_len);
361 xfree(handle);
362
363 return(0);
364}
365
366int
367do_rm(int fd_in, int fd_out, char *path)
368{
369 u_int status, id;
370
371 debug2("Sending SSH2_FXP_REMOVE \"%s\"", path);
372
Ben Lindstrom288cc392001-02-09 02:58:04 +0000373 id = msg_id++;
Damien Miller33804262001-02-04 23:20:18 +1100374 send_string_request(fd_out, id, SSH2_FXP_REMOVE, path, strlen(path));
375 status = get_status(fd_in, id);
376 if (status != SSH2_FX_OK)
377 error("Couldn't delete file: %s", fx2txt(status));
378 return(status);
379}
380
381int
382do_mkdir(int fd_in, int fd_out, char *path, Attrib *a)
383{
384 u_int status, id;
385
Ben Lindstrom288cc392001-02-09 02:58:04 +0000386 id = msg_id++;
Damien Miller33804262001-02-04 23:20:18 +1100387 send_string_attrs_request(fd_out, id, SSH2_FXP_MKDIR, path,
388 strlen(path), a);
389
390 status = get_status(fd_in, id);
391 if (status != SSH2_FX_OK)
392 error("Couldn't create directory: %s", fx2txt(status));
393
394 return(status);
395}
396
397int
398do_rmdir(int fd_in, int fd_out, char *path)
399{
400 u_int status, id;
401
Ben Lindstrom288cc392001-02-09 02:58:04 +0000402 id = msg_id++;
Damien Miller33804262001-02-04 23:20:18 +1100403 send_string_request(fd_out, id, SSH2_FXP_RMDIR, path, strlen(path));
404
405 status = get_status(fd_in, id);
406 if (status != SSH2_FX_OK)
407 error("Couldn't remove directory: %s", fx2txt(status));
408
409 return(status);
410}
411
412Attrib *
413do_stat(int fd_in, int fd_out, char *path)
414{
415 u_int id;
416
Ben Lindstrom288cc392001-02-09 02:58:04 +0000417 id = msg_id++;
Damien Miller33804262001-02-04 23:20:18 +1100418 send_string_request(fd_out, id, SSH2_FXP_STAT, path, strlen(path));
419 return(get_decode_stat(fd_in, id));
420}
421
422Attrib *
423do_lstat(int fd_in, int fd_out, char *path)
424{
425 u_int id;
426
Ben Lindstrom288cc392001-02-09 02:58:04 +0000427 id = msg_id++;
Damien Miller33804262001-02-04 23:20:18 +1100428 send_string_request(fd_out, id, SSH2_FXP_LSTAT, path, strlen(path));
429 return(get_decode_stat(fd_in, id));
430}
431
432Attrib *
433do_fstat(int fd_in, int fd_out, char *handle,
434 u_int handle_len)
435{
436 u_int id;
437
Ben Lindstrom288cc392001-02-09 02:58:04 +0000438 id = msg_id++;
Damien Miller33804262001-02-04 23:20:18 +1100439 send_string_request(fd_out, id, SSH2_FXP_FSTAT, handle, handle_len);
440 return(get_decode_stat(fd_in, id));
441}
442
443int
444do_setstat(int fd_in, int fd_out, char *path, Attrib *a)
445{
446 u_int status, id;
447
Ben Lindstrom288cc392001-02-09 02:58:04 +0000448 id = msg_id++;
Damien Miller33804262001-02-04 23:20:18 +1100449 send_string_attrs_request(fd_out, id, SSH2_FXP_SETSTAT, path,
450 strlen(path), a);
451
452 status = get_status(fd_in, id);
453 if (status != SSH2_FX_OK)
454 error("Couldn't setstat on \"%s\": %s", path,
455 fx2txt(status));
456
457 return(status);
458}
459
460int
461do_fsetstat(int fd_in, int fd_out, char *handle, u_int handle_len,
462 Attrib *a)
463{
464 u_int status, id;
465
Ben Lindstrom288cc392001-02-09 02:58:04 +0000466 id = msg_id++;
Damien Miller33804262001-02-04 23:20:18 +1100467 send_string_attrs_request(fd_out, id, SSH2_FXP_FSETSTAT, handle,
468 handle_len, a);
469
470 status = get_status(fd_in, id);
471 if (status != SSH2_FX_OK)
472 error("Couldn't fsetstat: %s", fx2txt(status));
473
474 return(status);
475}
476
477char *
478do_realpath(int fd_in, int fd_out, char *path)
479{
480 Buffer msg;
481 u_int type, expected_id, count, id;
482 char *filename, *longname;
483 Attrib *a;
484
Ben Lindstrom288cc392001-02-09 02:58:04 +0000485 expected_id = id = msg_id++;
Damien Miller33804262001-02-04 23:20:18 +1100486 send_string_request(fd_out, id, SSH2_FXP_REALPATH, path,
487 strlen(path));
488
489 buffer_init(&msg);
490
491 get_msg(fd_in, &msg);
492 type = buffer_get_char(&msg);
493 id = buffer_get_int(&msg);
494
495 if (id != expected_id)
496 fatal("ID mismatch (%d != %d)", id, expected_id);
497
498 if (type == SSH2_FXP_STATUS) {
499 u_int status = buffer_get_int(&msg);
500
501 error("Couldn't canonicalise: %s", fx2txt(status));
502 return(NULL);
503 } else if (type != SSH2_FXP_NAME)
504 fatal("Expected SSH2_FXP_NAME(%d) packet, got %d",
505 SSH2_FXP_NAME, type);
506
507 count = buffer_get_int(&msg);
508 if (count != 1)
509 fatal("Got multiple names (%d) from SSH_FXP_REALPATH", count);
510
511 filename = buffer_get_string(&msg, NULL);
512 longname = buffer_get_string(&msg, NULL);
513 a = decode_attrib(&msg);
514
515 debug3("SSH_FXP_REALPATH %s -> %s", path, filename);
516
517 xfree(longname);
518
519 buffer_free(&msg);
520
521 return(filename);
522}
523
524int
525do_rename(int fd_in, int fd_out, char *oldpath, char *newpath)
526{
527 Buffer msg;
528 u_int status, id;
529
530 buffer_init(&msg);
531
532 /* Send rename request */
Ben Lindstrom288cc392001-02-09 02:58:04 +0000533 id = msg_id++;
Damien Miller33804262001-02-04 23:20:18 +1100534 buffer_put_char(&msg, SSH2_FXP_RENAME);
535 buffer_put_int(&msg, id);
536 buffer_put_cstring(&msg, oldpath);
537 buffer_put_cstring(&msg, newpath);
538 send_msg(fd_out, &msg);
539 debug3("Sent message SSH2_FXP_RENAME \"%s\" -> \"%s\"", oldpath,
540 newpath);
541 buffer_free(&msg);
542
543 status = get_status(fd_in, id);
544 if (status != SSH2_FX_OK)
545 error("Couldn't rename file \"%s\" to \"%s\": %s", oldpath, newpath,
546 fx2txt(status));
547
548 return(status);
549}
550
551int
552do_download(int fd_in, int fd_out, char *remote_path, char *local_path,
553 int pflag)
554{
555 int local_fd;
556 u_int expected_id, handle_len, mode, type, id;
557 u_int64_t offset;
558 char *handle;
559 Buffer msg;
560 Attrib junk, *a;
Damien Millerd7686fd2001-02-10 00:40:03 +1100561 int status;
Damien Miller33804262001-02-04 23:20:18 +1100562
563 a = do_stat(fd_in, fd_out, remote_path);
564 if (a == NULL)
565 return(-1);
566
567 /* XXX: should we preserve set[ug]id? */
568 if (a->flags & SSH2_FILEXFER_ATTR_PERMISSIONS)
569 mode = S_IWRITE | (a->perm & 0777);
570 else
571 mode = 0666;
572
573 local_fd = open(local_path, O_WRONLY | O_CREAT | O_TRUNC, mode);
574 if (local_fd == -1) {
575 error("Couldn't open local file \"%s\" for writing: %s",
576 local_path, strerror(errno));
577 return(errno);
578 }
579
580 /* Override umask and utimes if asked */
581 if (pflag && fchmod(local_fd, mode) == -1)
582 error("Couldn't set mode on \"%s\": %s", local_path,
583 strerror(errno));
584 if (pflag && (a->flags & SSH2_FILEXFER_ATTR_ACMODTIME)) {
585 struct timeval tv;
586
587 tv.tv_sec = a->atime;
588 tv.tv_usec = a->mtime;
589 if (utimes(local_path, &tv) == -1)
590 error("Can't set times on \"%s\": %s", local_path,
591 strerror(errno));
592 }
593
594 buffer_init(&msg);
595
596 /* Send open request */
Ben Lindstrom288cc392001-02-09 02:58:04 +0000597 id = msg_id++;
Damien Miller33804262001-02-04 23:20:18 +1100598 buffer_put_char(&msg, SSH2_FXP_OPEN);
599 buffer_put_int(&msg, id);
600 buffer_put_cstring(&msg, remote_path);
601 buffer_put_int(&msg, SSH2_FXF_READ);
602 attrib_clear(&junk); /* Send empty attributes */
603 encode_attrib(&msg, &junk);
604 send_msg(fd_out, &msg);
605 debug3("Sent message SSH2_FXP_OPEN I:%d P:%s", id, remote_path);
606
607 handle = get_handle(fd_in, id, &handle_len);
608 if (handle == NULL) {
609 buffer_free(&msg);
610 close(local_fd);
611 return(-1);
612 }
613
614 /* Read from remote and write to local */
615 offset = 0;
616 for(;;) {
617 u_int len;
618 char *data;
619
Ben Lindstrom288cc392001-02-09 02:58:04 +0000620 id = expected_id = msg_id++;
Damien Miller33804262001-02-04 23:20:18 +1100621
622 buffer_clear(&msg);
623 buffer_put_char(&msg, SSH2_FXP_READ);
624 buffer_put_int(&msg, id);
625 buffer_put_string(&msg, handle, handle_len);
626 buffer_put_int64(&msg, offset);
627 buffer_put_int(&msg, COPY_SIZE);
628 send_msg(fd_out, &msg);
629 debug3("Sent message SSH2_FXP_READ I:%d O:%llu S:%u",
Kevin Stevesec1c1402001-02-05 15:39:22 +0000630 id, (unsigned long long)offset, COPY_SIZE);
Damien Miller33804262001-02-04 23:20:18 +1100631
632 buffer_clear(&msg);
633
634 get_msg(fd_in, &msg);
635 type = buffer_get_char(&msg);
636 id = buffer_get_int(&msg);
637 debug3("Received reply T:%d I:%d", type, id);
638 if (id != expected_id)
639 fatal("ID mismatch (%d != %d)", id, expected_id);
640 if (type == SSH2_FXP_STATUS) {
Damien Millerd7686fd2001-02-10 00:40:03 +1100641 status = buffer_get_int(&msg);
Damien Miller33804262001-02-04 23:20:18 +1100642
643 if (status == SSH2_FX_EOF)
644 break;
645 else {
646 error("Couldn't read from remote "
647 "file \"%s\" : %s", remote_path,
648 fx2txt(status));
649 do_close(fd_in, fd_out, handle, handle_len);
Damien Millerd7686fd2001-02-10 00:40:03 +1100650 goto done;
Damien Miller33804262001-02-04 23:20:18 +1100651 }
652 } else if (type != SSH2_FXP_DATA) {
653 fatal("Expected SSH2_FXP_DATA(%d) packet, got %d",
654 SSH2_FXP_DATA, type);
655 }
656
657 data = buffer_get_string(&msg, &len);
658 if (len > COPY_SIZE)
659 fatal("Received more data than asked for %d > %d",
660 len, COPY_SIZE);
661
Damien Millerd7686fd2001-02-10 00:40:03 +1100662 debug3("In read loop, got %d offset %llu", len,
Kevin Stevesec1c1402001-02-05 15:39:22 +0000663 (unsigned long long)offset);
Damien Miller33804262001-02-04 23:20:18 +1100664 if (atomicio(write, local_fd, data, len) != len) {
665 error("Couldn't write to \"%s\": %s", local_path,
666 strerror(errno));
667 do_close(fd_in, fd_out, handle, handle_len);
Damien Millerd7686fd2001-02-10 00:40:03 +1100668 status = -1;
Damien Miller33804262001-02-04 23:20:18 +1100669 xfree(data);
Damien Millerd7686fd2001-02-10 00:40:03 +1100670 goto done;
Damien Miller33804262001-02-04 23:20:18 +1100671 }
672
673 offset += len;
674 xfree(data);
675 }
Damien Millerd7686fd2001-02-10 00:40:03 +1100676 status = do_close(fd_in, fd_out, handle, handle_len);
Damien Miller33804262001-02-04 23:20:18 +1100677
Damien Millerd7686fd2001-02-10 00:40:03 +1100678done:
679 close(local_fd);
680 buffer_free(&msg);
681 xfree(handle);
682 return status;
Damien Miller33804262001-02-04 23:20:18 +1100683}
684
685int
686do_upload(int fd_in, int fd_out, char *local_path, char *remote_path,
687 int pflag)
688{
689 int local_fd;
690 u_int handle_len, id;
691 u_int64_t offset;
692 char *handle;
693 Buffer msg;
694 struct stat sb;
695 Attrib a;
Damien Millerd7686fd2001-02-10 00:40:03 +1100696 int status;
Damien Miller33804262001-02-04 23:20:18 +1100697
698 if ((local_fd = open(local_path, O_RDONLY, 0)) == -1) {
699 error("Couldn't open local file \"%s\" for reading: %s",
700 local_path, strerror(errno));
701 return(-1);
702 }
703 if (fstat(local_fd, &sb) == -1) {
704 error("Couldn't fstat local file \"%s\": %s",
705 local_path, strerror(errno));
706 close(local_fd);
707 return(-1);
708 }
709 stat_to_attrib(&sb, &a);
710
711 a.flags &= ~SSH2_FILEXFER_ATTR_SIZE;
712 a.flags &= ~SSH2_FILEXFER_ATTR_UIDGID;
713 a.perm &= 0777;
714 if (!pflag)
715 a.flags &= ~SSH2_FILEXFER_ATTR_ACMODTIME;
716
717 buffer_init(&msg);
718
719 /* Send open request */
Ben Lindstrom288cc392001-02-09 02:58:04 +0000720 id = msg_id++;
Damien Miller33804262001-02-04 23:20:18 +1100721 buffer_put_char(&msg, SSH2_FXP_OPEN);
722 buffer_put_int(&msg, id);
723 buffer_put_cstring(&msg, remote_path);
724 buffer_put_int(&msg, SSH2_FXF_WRITE|SSH2_FXF_CREAT|SSH2_FXF_TRUNC);
725 encode_attrib(&msg, &a);
726 send_msg(fd_out, &msg);
727 debug3("Sent message SSH2_FXP_OPEN I:%d P:%s", id, remote_path);
728
729 buffer_clear(&msg);
730
731 handle = get_handle(fd_in, id, &handle_len);
732 if (handle == NULL) {
733 close(local_fd);
734 buffer_free(&msg);
735 return(-1);
736 }
737
738 /* Override umask and utimes if asked */
739 if (pflag)
740 do_fsetstat(fd_in, fd_out, handle, handle_len, &a);
741
742 /* Read from local and write to remote */
743 offset = 0;
744 for(;;) {
745 int len;
746 char data[COPY_SIZE];
Damien Miller33804262001-02-04 23:20:18 +1100747
748 /*
749 * Can't use atomicio here because it returns 0 on EOF, thus losing
750 * the last block of the file
751 */
752 do
753 len = read(local_fd, data, COPY_SIZE);
754 while ((len == -1) && (errno == EINTR || errno == EAGAIN));
755
756 if (len == -1)
757 fatal("Couldn't read from \"%s\": %s", local_path,
758 strerror(errno));
759 if (len == 0)
760 break;
761
762 buffer_clear(&msg);
763 buffer_put_char(&msg, SSH2_FXP_WRITE);
764 buffer_put_int(&msg, ++id);
765 buffer_put_string(&msg, handle, handle_len);
766 buffer_put_int64(&msg, offset);
767 buffer_put_string(&msg, data, len);
768 send_msg(fd_out, &msg);
769 debug3("Sent message SSH2_FXP_WRITE I:%d O:%llu S:%u",
Kevin Stevesec1c1402001-02-05 15:39:22 +0000770 id, (unsigned long long)offset, len);
Damien Miller33804262001-02-04 23:20:18 +1100771
772 status = get_status(fd_in, id);
773 if (status != SSH2_FX_OK) {
774 error("Couldn't write to remote file \"%s\": %s",
775 remote_path, fx2txt(status));
776 do_close(fd_in, fd_out, handle, handle_len);
Damien Miller33804262001-02-04 23:20:18 +1100777 close(local_fd);
Damien Millerd7686fd2001-02-10 00:40:03 +1100778 goto done;
Damien Miller33804262001-02-04 23:20:18 +1100779 }
Kevin Stevesec1c1402001-02-05 15:39:22 +0000780 debug3("In write loop, got %d offset %llu", len,
781 (unsigned long long)offset);
Damien Miller33804262001-02-04 23:20:18 +1100782
783 offset += len;
784 }
Damien Miller33804262001-02-04 23:20:18 +1100785
786 if (close(local_fd) == -1) {
787 error("Couldn't close local file \"%s\": %s", local_path,
788 strerror(errno));
789 do_close(fd_in, fd_out, handle, handle_len);
Damien Millerd7686fd2001-02-10 00:40:03 +1100790 status = -1;
791 goto done;
Damien Miller33804262001-02-04 23:20:18 +1100792 }
793
Damien Millerd7686fd2001-02-10 00:40:03 +1100794 status = do_close(fd_in, fd_out, handle, handle_len);
795
796done:
797 xfree(handle);
798 buffer_free(&msg);
799 return status;
Damien Miller33804262001-02-04 23:20:18 +1100800}
Damien Millerd7686fd2001-02-10 00:40:03 +1100801
802