2 * Copyright (c) 2001-2003 Damien Miller. All rights reserved.
4 * Redistribution and use in source and binary forms, with or without
5 * modification, are permitted provided that the following conditions
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.
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.
26 /* XXX: signed vs unsigned */
27 /* XXX: remove all logging, only return status codes */
28 /* XXX: copy between two remote sites */
31 RCSID("$OpenBSD: sftp-client.c,v 1.44 2003/06/28 16:23:06 deraadt Exp $");
33 #include "openbsd-compat/sys-queue.h"
41 #include "progressmeter.h"
44 #include "sftp-common.h"
45 #include "sftp-client.h"
47 extern int showprogress;
49 /* Minimum amount of data to read at at time */
50 #define MIN_READ_SIZE 512
52 /* Maximum packet size */
53 #define MAX_MSG_LENGTH (256 * 1024)
58 u_int transfer_buflen;
65 send_msg(int fd, Buffer *m)
69 if (buffer_len(m) > MAX_MSG_LENGTH)
70 fatal("Outbound message too long %u", buffer_len(m));
72 /* Send length first */
73 PUT_32BIT(mlen, buffer_len(m));
74 if (atomicio(vwrite, fd, mlen, sizeof(mlen)) <= 0)
75 fatal("Couldn't send packet: %s", strerror(errno));
77 if (atomicio(vwrite, fd, buffer_ptr(m), buffer_len(m)) <= 0)
78 fatal("Couldn't send packet: %s", strerror(errno));
84 get_msg(int fd, Buffer *m)
89 buffer_append_space(m, 4);
90 len = atomicio(read, fd, buffer_ptr(m), 4);
92 fatal("Connection closed");
94 fatal("Couldn't read packet: %s", strerror(errno));
96 msg_len = buffer_get_int(m);
97 if (msg_len > MAX_MSG_LENGTH)
98 fatal("Received message too long %u", msg_len);
100 buffer_append_space(m, msg_len);
101 len = atomicio(read, fd, buffer_ptr(m), msg_len);
103 fatal("Connection closed");
105 fatal("Read packet: %s", strerror(errno));
109 send_string_request(int fd, u_int id, u_int code, char *s,
115 buffer_put_char(&msg, code);
116 buffer_put_int(&msg, id);
117 buffer_put_string(&msg, s, len);
119 debug3("Sent message fd %d T:%u I:%u", fd, code, id);
124 send_string_attrs_request(int fd, u_int id, u_int code, char *s,
125 u_int len, Attrib *a)
130 buffer_put_char(&msg, code);
131 buffer_put_int(&msg, id);
132 buffer_put_string(&msg, s, len);
133 encode_attrib(&msg, a);
135 debug3("Sent message fd %d T:%u I:%u", fd, code, id);
140 get_status(int fd, u_int expected_id)
143 u_int type, id, status;
147 type = buffer_get_char(&msg);
148 id = buffer_get_int(&msg);
150 if (id != expected_id)
151 fatal("ID mismatch (%u != %u)", id, expected_id);
152 if (type != SSH2_FXP_STATUS)
153 fatal("Expected SSH2_FXP_STATUS(%u) packet, got %u",
154 SSH2_FXP_STATUS, type);
156 status = buffer_get_int(&msg);
159 debug3("SSH2_FXP_STATUS %u", status);
165 get_handle(int fd, u_int expected_id, u_int *len)
173 type = buffer_get_char(&msg);
174 id = buffer_get_int(&msg);
176 if (id != expected_id)
177 fatal("ID mismatch (%u != %u)", id, expected_id);
178 if (type == SSH2_FXP_STATUS) {
179 int status = buffer_get_int(&msg);
181 error("Couldn't get handle: %s", fx2txt(status));
183 } else if (type != SSH2_FXP_HANDLE)
184 fatal("Expected SSH2_FXP_HANDLE(%u) packet, got %u",
185 SSH2_FXP_HANDLE, type);
187 handle = buffer_get_string(&msg, len);
194 get_decode_stat(int fd, u_int expected_id, int quiet)
203 type = buffer_get_char(&msg);
204 id = buffer_get_int(&msg);
206 debug3("Received stat reply T:%u I:%u", type, id);
207 if (id != expected_id)
208 fatal("ID mismatch (%u != %u)", id, expected_id);
209 if (type == SSH2_FXP_STATUS) {
210 int status = buffer_get_int(&msg);
213 debug("Couldn't stat remote file: %s", fx2txt(status));
215 error("Couldn't stat remote file: %s", fx2txt(status));
217 } else if (type != SSH2_FXP_ATTRS) {
218 fatal("Expected SSH2_FXP_ATTRS(%u) packet, got %u",
219 SSH2_FXP_ATTRS, type);
221 a = decode_attrib(&msg);
228 do_init(int fd_in, int fd_out, u_int transfer_buflen, u_int num_requests)
233 struct sftp_conn *ret;
236 buffer_put_char(&msg, SSH2_FXP_INIT);
237 buffer_put_int(&msg, SSH2_FILEXFER_VERSION);
238 send_msg(fd_out, &msg);
242 get_msg(fd_in, &msg);
244 /* Expecting a VERSION reply */
245 if ((type = buffer_get_char(&msg)) != SSH2_FXP_VERSION) {
246 error("Invalid packet back from SSH2_FXP_INIT (type %u)",
251 version = buffer_get_int(&msg);
253 debug2("Remote version: %d", version);
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);
260 debug2("Init extension: \"%s\"", name);
267 ret = xmalloc(sizeof(*ret));
269 ret->fd_out = fd_out;
270 ret->transfer_buflen = transfer_buflen;
271 ret->num_requests = num_requests;
272 ret->version = version;
275 /* Some filexfer v.0 servers don't support large packets */
277 ret->transfer_buflen = MIN(ret->transfer_buflen, 20480);
283 sftp_proto_version(struct sftp_conn *conn)
285 return(conn->version);
289 do_close(struct sftp_conn *conn, char *handle, u_int handle_len)
297 buffer_put_char(&msg, SSH2_FXP_CLOSE);
298 buffer_put_int(&msg, id);
299 buffer_put_string(&msg, handle, handle_len);
300 send_msg(conn->fd_out, &msg);
301 debug3("Sent message SSH2_FXP_CLOSE I:%u", id);
303 status = get_status(conn->fd_in, id);
304 if (status != SSH2_FX_OK)
305 error("Couldn't close file: %s", fx2txt(status));
314 do_lsreaddir(struct sftp_conn *conn, char *path, int printflag,
318 u_int type, id, handle_len, i, expected_id, ents = 0;
324 buffer_put_char(&msg, SSH2_FXP_OPENDIR);
325 buffer_put_int(&msg, id);
326 buffer_put_cstring(&msg, path);
327 send_msg(conn->fd_out, &msg);
331 handle = get_handle(conn->fd_in, id, &handle_len);
337 *dir = xmalloc(sizeof(**dir));
344 id = expected_id = conn->msg_id++;
346 debug3("Sending SSH2_FXP_READDIR I:%u", id);
349 buffer_put_char(&msg, SSH2_FXP_READDIR);
350 buffer_put_int(&msg, id);
351 buffer_put_string(&msg, handle, handle_len);
352 send_msg(conn->fd_out, &msg);
356 get_msg(conn->fd_in, &msg);
358 type = buffer_get_char(&msg);
359 id = buffer_get_int(&msg);
361 debug3("Received reply T:%u I:%u", type, id);
363 if (id != expected_id)
364 fatal("ID mismatch (%u != %u)", id, expected_id);
366 if (type == SSH2_FXP_STATUS) {
367 int status = buffer_get_int(&msg);
369 debug3("Received SSH2_FXP_STATUS %d", status);
371 if (status == SSH2_FX_EOF) {
374 error("Couldn't read directory: %s",
376 do_close(conn, handle, handle_len);
380 } else if (type != SSH2_FXP_NAME)
381 fatal("Expected SSH2_FXP_NAME(%u) packet, got %u",
382 SSH2_FXP_NAME, type);
384 count = buffer_get_int(&msg);
387 debug3("Received %d SSH2_FXP_NAME responses", count);
388 for (i = 0; i < count; i++) {
389 char *filename, *longname;
392 filename = buffer_get_string(&msg, NULL);
393 longname = buffer_get_string(&msg, NULL);
394 a = decode_attrib(&msg);
397 printf("%s\n", longname);
400 *dir = xrealloc(*dir, sizeof(**dir) *
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;
415 do_close(conn, handle, handle_len);
422 do_readdir(struct sftp_conn *conn, char *path, SFTP_DIRENT ***dir)
424 return(do_lsreaddir(conn, path, 0, dir));
427 void free_sftp_dirents(SFTP_DIRENT **s)
431 for (i = 0; s[i]; i++) {
432 xfree(s[i]->filename);
433 xfree(s[i]->longname);
440 do_rm(struct sftp_conn *conn, char *path)
444 debug2("Sending SSH2_FXP_REMOVE \"%s\"", path);
447 send_string_request(conn->fd_out, id, SSH2_FXP_REMOVE, path,
449 status = get_status(conn->fd_in, id);
450 if (status != SSH2_FX_OK)
451 error("Couldn't delete file: %s", fx2txt(status));
456 do_mkdir(struct sftp_conn *conn, char *path, Attrib *a)
461 send_string_attrs_request(conn->fd_out, id, SSH2_FXP_MKDIR, path,
464 status = get_status(conn->fd_in, id);
465 if (status != SSH2_FX_OK)
466 error("Couldn't create directory: %s", fx2txt(status));
472 do_rmdir(struct sftp_conn *conn, char *path)
477 send_string_request(conn->fd_out, id, SSH2_FXP_RMDIR, path,
480 status = get_status(conn->fd_in, id);
481 if (status != SSH2_FX_OK)
482 error("Couldn't remove directory: %s", fx2txt(status));
488 do_stat(struct sftp_conn *conn, char *path, int quiet)
494 send_string_request(conn->fd_out, id,
495 conn->version == 0 ? SSH2_FXP_STAT_VERSION_0 : SSH2_FXP_STAT,
498 return(get_decode_stat(conn->fd_in, id, quiet));
502 do_lstat(struct sftp_conn *conn, char *path, int quiet)
506 if (conn->version == 0) {
508 debug("Server version does not support lstat operation");
510 logit("Server version does not support lstat operation");
511 return(do_stat(conn, path, quiet));
515 send_string_request(conn->fd_out, id, SSH2_FXP_LSTAT, path,
518 return(get_decode_stat(conn->fd_in, id, quiet));
522 do_fstat(struct sftp_conn *conn, char *handle, u_int handle_len, int quiet)
527 send_string_request(conn->fd_out, id, SSH2_FXP_FSTAT, handle,
530 return(get_decode_stat(conn->fd_in, id, quiet));
534 do_setstat(struct sftp_conn *conn, char *path, Attrib *a)
539 send_string_attrs_request(conn->fd_out, id, SSH2_FXP_SETSTAT, path,
542 status = get_status(conn->fd_in, id);
543 if (status != SSH2_FX_OK)
544 error("Couldn't setstat on \"%s\": %s", path,
551 do_fsetstat(struct sftp_conn *conn, char *handle, u_int handle_len,
557 send_string_attrs_request(conn->fd_out, id, SSH2_FXP_FSETSTAT, handle,
560 status = get_status(conn->fd_in, id);
561 if (status != SSH2_FX_OK)
562 error("Couldn't fsetstat: %s", fx2txt(status));
568 do_realpath(struct sftp_conn *conn, char *path)
571 u_int type, expected_id, count, id;
572 char *filename, *longname;
575 expected_id = id = conn->msg_id++;
576 send_string_request(conn->fd_out, id, SSH2_FXP_REALPATH, path,
581 get_msg(conn->fd_in, &msg);
582 type = buffer_get_char(&msg);
583 id = buffer_get_int(&msg);
585 if (id != expected_id)
586 fatal("ID mismatch (%u != %u)", id, expected_id);
588 if (type == SSH2_FXP_STATUS) {
589 u_int status = buffer_get_int(&msg);
591 error("Couldn't canonicalise: %s", fx2txt(status));
593 } else if (type != SSH2_FXP_NAME)
594 fatal("Expected SSH2_FXP_NAME(%u) packet, got %u",
595 SSH2_FXP_NAME, type);
597 count = buffer_get_int(&msg);
599 fatal("Got multiple names (%d) from SSH_FXP_REALPATH", count);
601 filename = buffer_get_string(&msg, NULL);
602 longname = buffer_get_string(&msg, NULL);
603 a = decode_attrib(&msg);
605 debug3("SSH_FXP_REALPATH %s -> %s", path, filename);
615 do_rename(struct sftp_conn *conn, char *oldpath, char *newpath)
622 /* Send rename request */
624 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);
628 send_msg(conn->fd_out, &msg);
629 debug3("Sent message SSH2_FXP_RENAME \"%s\" -> \"%s\"", oldpath,
633 status = get_status(conn->fd_in, id);
634 if (status != SSH2_FX_OK)
635 error("Couldn't rename file \"%s\" to \"%s\": %s", oldpath,
636 newpath, fx2txt(status));
642 do_symlink(struct sftp_conn *conn, char *oldpath, char *newpath)
647 if (conn->version < 3) {
648 error("This server does not support the symlink operation");
649 return(SSH2_FX_OP_UNSUPPORTED);
654 /* Send rename request */
656 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);
660 send_msg(conn->fd_out, &msg);
661 debug3("Sent message SSH2_FXP_SYMLINK \"%s\" -> \"%s\"", oldpath,
665 status = get_status(conn->fd_in, id);
666 if (status != SSH2_FX_OK)
667 error("Couldn't symlink file \"%s\" to \"%s\": %s", oldpath,
668 newpath, fx2txt(status));
674 do_readlink(struct sftp_conn *conn, char *path)
677 u_int type, expected_id, count, id;
678 char *filename, *longname;
681 expected_id = id = conn->msg_id++;
682 send_string_request(conn->fd_out, id, SSH2_FXP_READLINK, path,
687 get_msg(conn->fd_in, &msg);
688 type = buffer_get_char(&msg);
689 id = buffer_get_int(&msg);
691 if (id != expected_id)
692 fatal("ID mismatch (%u != %u)", id, expected_id);
694 if (type == SSH2_FXP_STATUS) {
695 u_int status = buffer_get_int(&msg);
697 error("Couldn't readlink: %s", fx2txt(status));
699 } else if (type != SSH2_FXP_NAME)
700 fatal("Expected SSH2_FXP_NAME(%u) packet, got %u",
701 SSH2_FXP_NAME, type);
703 count = buffer_get_int(&msg);
705 fatal("Got multiple names (%d) from SSH_FXP_READLINK", count);
707 filename = buffer_get_string(&msg, NULL);
708 longname = buffer_get_string(&msg, NULL);
709 a = decode_attrib(&msg);
711 debug3("SSH_FXP_READLINK %s -> %s", path, filename);
721 send_read_request(int fd_out, u_int id, u_int64_t offset, u_int len,
722 char *handle, u_int handle_len)
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);
738 do_download(struct sftp_conn *conn, char *remote_path, char *local_path,
744 int local_fd, status, num_req, max_req, write_error;
745 int read_error, write_errno;
746 u_int64_t offset, size;
747 u_int handle_len, mode, type, id, buflen;
748 off_t progress_counter;
753 TAILQ_ENTRY(request) tq;
755 TAILQ_HEAD(reqhead, request) requests;
758 TAILQ_INIT(&requests);
760 a = do_stat(conn, remote_path, 0);
764 /* XXX: should we preserve set[ug]id? */
765 if (a->flags & SSH2_FILEXFER_ATTR_PERMISSIONS)
766 mode = a->perm & 0777;
770 if ((a->flags & SSH2_FILEXFER_ATTR_PERMISSIONS) &&
771 (!S_ISREG(a->perm))) {
772 error("Cannot download non-regular file: %s", remote_path);
776 if (a->flags & SSH2_FILEXFER_ATTR_SIZE)
781 buflen = conn->transfer_buflen;
784 /* Send open request */
786 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);
792 send_msg(conn->fd_out, &msg);
793 debug3("Sent message SSH2_FXP_OPEN I:%u P:%s", id, remote_path);
795 handle = get_handle(conn->fd_in, id, &handle_len);
796 if (handle == NULL) {
801 local_fd = open(local_path, O_WRONLY | O_CREAT | O_TRUNC,
803 if (local_fd == -1) {
804 error("Couldn't open local file \"%s\" for writing: %s",
805 local_path, strerror(errno));
811 /* Read from remote and write to local */
812 write_error = read_error = write_errno = num_req = offset = 0;
814 progress_counter = 0;
818 start_progress_meter(remote_path, size,
821 printf("Fetching %s to %s\n", remote_path, local_path);
824 while (num_req > 0 || max_req > 0) {
828 /* Send some more requests */
829 while (num_req < max_req) {
830 debug3("Request range %llu -> %llu (%d/%d)",
831 (unsigned long long)offset,
832 (unsigned long long)offset + buflen - 1,
834 req = xmalloc(sizeof(*req));
835 req->id = conn->msg_id++;
837 req->offset = offset;
840 TAILQ_INSERT_TAIL(&requests, req, tq);
841 send_read_request(conn->fd_out, req->id, req->offset,
842 req->len, handle, handle_len);
846 get_msg(conn->fd_in, &msg);
847 type = buffer_get_char(&msg);
848 id = buffer_get_int(&msg);
849 debug3("Received reply T:%u I:%u R:%d", type, id, max_req);
851 /* Find the request in our queue */
852 for(req = TAILQ_FIRST(&requests);
853 req != NULL && req->id != id;
854 req = TAILQ_NEXT(req, tq))
857 fatal("Unexpected reply %u", id);
860 case SSH2_FXP_STATUS:
861 status = buffer_get_int(&msg);
862 if (status != SSH2_FX_EOF)
865 TAILQ_REMOVE(&requests, req, tq);
870 data = buffer_get_string(&msg, &len);
871 debug3("Received data %llu -> %llu",
872 (unsigned long long)req->offset,
873 (unsigned long long)req->offset + len - 1);
875 fatal("Received more data than asked for "
876 "%u > %u", len, req->len);
877 if ((lseek(local_fd, req->offset, SEEK_SET) == -1 ||
878 atomicio(vwrite, local_fd, data, len) != len) &&
884 progress_counter += len;
887 if (len == req->len) {
888 TAILQ_REMOVE(&requests, req, tq);
892 /* Resend the request for the missing data */
893 debug3("Short data block, re-requesting "
894 "%llu -> %llu (%2d)",
895 (unsigned long long)req->offset + len,
896 (unsigned long long)req->offset +
897 req->len - 1, num_req);
898 req->id = conn->msg_id++;
901 send_read_request(conn->fd_out, req->id,
902 req->offset, req->len, handle, handle_len);
903 /* Reduce the request size */
905 buflen = MAX(MIN_READ_SIZE, len);
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)",
912 (unsigned long long)offset,
916 else if (max_req < conn->num_requests + 1) {
922 fatal("Expected SSH2_FXP_DATA(%u) packet, got %u",
923 SSH2_FXP_DATA, type);
927 if (showprogress && size)
928 stop_progress_meter();
931 if (TAILQ_FIRST(&requests) != NULL)
932 fatal("Transfer complete, but requests still in queue");
935 error("Couldn't read from remote file \"%s\" : %s",
936 remote_path, fx2txt(status));
937 do_close(conn, handle, handle_len);
938 } else if (write_error) {
939 error("Couldn't write to \"%s\": %s", local_path,
940 strerror(write_errno));
942 do_close(conn, handle, handle_len);
944 status = do_close(conn, handle, handle_len);
946 /* Override umask and utimes if asked */
948 if (pflag && fchmod(local_fd, mode) == -1)
950 if (pflag && chmod(local_path, mode) == -1)
951 #endif /* HAVE_FCHMOD */
952 error("Couldn't set mode on \"%s\": %s", local_path,
954 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",
961 local_path, strerror(errno));
972 do_upload(struct sftp_conn *conn, char *local_path, char *remote_path,
975 int local_fd, status;
976 u_int handle_len, id, type;
984 struct outstanding_ack {
988 TAILQ_ENTRY(outstanding_ack) tq;
990 TAILQ_HEAD(ackhead, outstanding_ack) acks;
991 struct outstanding_ack *ack;
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));
1000 if (fstat(local_fd, &sb) == -1) {
1001 error("Couldn't fstat local file \"%s\": %s",
1002 local_path, strerror(errno));
1006 if (!S_ISREG(sb.st_mode)) {
1007 error("%s is not a regular file", local_path);
1011 stat_to_attrib(&sb, &a);
1013 a.flags &= ~SSH2_FILEXFER_ATTR_SIZE;
1014 a.flags &= ~SSH2_FILEXFER_ATTR_UIDGID;
1017 a.flags &= ~SSH2_FILEXFER_ATTR_ACMODTIME;
1021 /* Send open request */
1022 id = conn->msg_id++;
1023 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);
1028 send_msg(conn->fd_out, &msg);
1029 debug3("Sent message SSH2_FXP_OPEN I:%u P:%s", id, remote_path);
1033 handle = get_handle(conn->fd_in, id, &handle_len);
1034 if (handle == NULL) {
1040 startid = ackid = id + 1;
1041 data = xmalloc(conn->transfer_buflen);
1043 /* Read from local and write to remote */
1046 start_progress_meter(local_path, sb.st_size, &offset);
1048 printf("Uploading %s to %s\n", local_path, remote_path);
1054 * Can't use atomicio here because it returns 0 on EOF, thus losing
1055 * the last block of the file
1058 len = read(local_fd, data, conn->transfer_buflen);
1059 while ((len == -1) && (errno == EINTR || errno == EAGAIN));
1062 fatal("Couldn't read from \"%s\": %s", local_path,
1066 ack = xmalloc(sizeof(*ack));
1068 ack->offset = offset;
1070 TAILQ_INSERT_TAIL(&acks, ack, tq);
1073 buffer_put_char(&msg, SSH2_FXP_WRITE);
1074 buffer_put_int(&msg, ack->id);
1075 buffer_put_string(&msg, handle, handle_len);
1076 buffer_put_int64(&msg, offset);
1077 buffer_put_string(&msg, data, len);
1078 send_msg(conn->fd_out, &msg);
1079 debug3("Sent message SSH2_FXP_WRITE I:%u O:%llu S:%u",
1080 id, (unsigned long long)offset, len);
1081 } else if (TAILQ_FIRST(&acks) == NULL)
1085 fatal("Unexpected ACK %u", id);
1087 if (id == startid || len == 0 ||
1088 id - ackid >= conn->num_requests) {
1092 get_msg(conn->fd_in, &msg);
1093 type = buffer_get_char(&msg);
1094 r_id = buffer_get_int(&msg);
1096 if (type != SSH2_FXP_STATUS)
1097 fatal("Expected SSH2_FXP_STATUS(%d) packet, "
1098 "got %d", SSH2_FXP_STATUS, type);
1100 status = buffer_get_int(&msg);
1101 debug3("SSH2_FXP_STATUS %d", status);
1103 /* Find the request in our queue */
1104 for(ack = TAILQ_FIRST(&acks);
1105 ack != NULL && ack->id != r_id;
1106 ack = TAILQ_NEXT(ack, tq))
1109 fatal("Can't find request for ID %u", r_id);
1110 TAILQ_REMOVE(&acks, ack, tq);
1112 if (status != SSH2_FX_OK) {
1113 error("Couldn't write to remote file \"%s\": %s",
1114 remote_path, fx2txt(status));
1115 do_close(conn, handle, handle_len);
1121 debug3("In write loop, ack for %u %u bytes at %llu",
1122 ack->id, ack->len, (unsigned long long)ack->offset);
1129 stop_progress_meter();
1132 if (close(local_fd) == -1) {
1133 error("Couldn't close local file \"%s\": %s", local_path,
1135 do_close(conn, handle, handle_len);
1140 /* Override umask and utimes if asked */
1142 do_fsetstat(conn, handle, handle_len, &a);
1144 status = do_close(conn, handle, handle_len);