=================================================================== RCS file: /cvsrepo/anoncvs/cvs/src/usr.bin/ssh/sftp-client.c,v retrieving revision 1.60.2.2 retrieving revision 1.61 diff -u -r1.60.2.2 -r1.61 --- src/usr.bin/ssh/sftp-client.c 2006/11/08 00:17:14 1.60.2.2 +++ src/usr.bin/ssh/sftp-client.c 2006/03/19 18:51:18 1.61 @@ -1,4 +1,3 @@ -/* $OpenBSD: sftp-client.c,v 1.60.2.2 2006/11/08 00:17:14 brad Exp $ */ /* * Copyright (c) 2001-2004 Damien Miller * @@ -20,27 +19,19 @@ /* XXX: remove all logging, only return status codes */ /* XXX: copy between two remote sites */ -#include +#include "includes.h" + #include +#include #include -#include -#include -#include -#include -#include -#include -#include -#include -#include -#include - -#include "xmalloc.h" #include "buffer.h" +#include "bufaux.h" +#include "getput.h" +#include "xmalloc.h" #include "log.h" #include "atomicio.h" #include "progressmeter.h" -#include "misc.h" #include "sftp.h" #include "sftp-common.h" @@ -65,19 +56,16 @@ send_msg(int fd, Buffer *m) { u_char mlen[4]; - struct iovec iov[2]; if (buffer_len(m) > SFTP_MAX_MSG_LENGTH) fatal("Outbound message too long %u", buffer_len(m)); /* Send length first */ - put_u32(mlen, buffer_len(m)); - iov[0].iov_base = mlen; - iov[0].iov_len = sizeof(mlen); - iov[1].iov_base = buffer_ptr(m); - iov[1].iov_len = buffer_len(m); + PUT_32BIT(mlen, buffer_len(m)); + if (atomicio(vwrite, fd, mlen, sizeof(mlen)) != sizeof(mlen)) + fatal("Couldn't send packet: %s", strerror(errno)); - if (atomiciov(writev, fd, iov, 2) != buffer_len(m) + sizeof(mlen)) + if (atomicio(vwrite, fd, buffer_ptr(m), buffer_len(m)) != buffer_len(m)) fatal("Couldn't send packet: %s", strerror(errno)); buffer_clear(m); @@ -401,7 +389,8 @@ printf("%s\n", longname); if (dir) { - *dir = xrealloc(*dir, ents + 2, sizeof(**dir)); + *dir = xrealloc(*dir, sizeof(**dir) * + (ents + 2)); (*dir)[ents] = xmalloc(sizeof(***dir)); (*dir)[ents]->filename = xstrdup(filename); (*dir)[ents]->longname = xstrdup(longname); @@ -1124,8 +1113,6 @@ if (status != SSH2_FX_OK) { error("Couldn't write to remote file \"%s\": %s", remote_path, fx2txt(status)); - if (showprogress) - stop_progress_meter(); do_close(conn, handle, handle_len); close(local_fd); xfree(data);