[openssh.git] / sftp-client.c

/*
 * Copyright (c) 2001 Damien Miller.  All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

/* XXX: memleaks */
/* XXX: signed vs unsigned */
/* XXX: redesign to allow concurrent overlapped operations */
/* XXX: we use fatal too much, error may be more appropriate in places */
/* XXX: copy between two remote sites */

#include "includes.h"
RCSID("$OpenBSD: sftp-client.c,v 1.9 2001/02/10 00:41:46 djm Exp $");

#include "ssh.h"
#include "buffer.h"
#include "bufaux.h"
#include "getput.h"
#include "xmalloc.h"
#include "log.h"
#include "atomicio.h"
#include "pathnames.h"

#include "sftp.h"
#include "sftp-common.h"
#include "sftp-client.h"

/* How much data to read/write at at time during copies */
/* XXX: what should this be? */
#define COPY_SIZE	8192

/* Message ID */
static u_int msg_id = 1;

void
send_msg(int fd, Buffer *m)
{
	int mlen = buffer_len(m);
	int len;
	Buffer oqueue;

	buffer_init(&oqueue);
	buffer_put_int(&oqueue, mlen);
	buffer_append(&oqueue, buffer_ptr(m), mlen);
	buffer_consume(m, mlen);

	len = atomicio(write, fd, buffer_ptr(&oqueue), buffer_len(&oqueue));
	if (len <= 0)
		fatal("Couldn't send packet: %s", strerror(errno));

	buffer_free(&oqueue);
}

void
get_msg(int fd, Buffer *m)
{
	u_int len, msg_len;
	unsigned char buf[4096];

	len = atomicio(read, fd, buf, 4);
	if (len != 4)
		fatal("Couldn't read packet: %s", strerror(errno));

	msg_len = GET_32BIT(buf);
	if (msg_len > 256 * 1024)
		fatal("Received message too long %d", msg_len);

	while (msg_len) {
		len = atomicio(read, fd, buf, MIN(msg_len, sizeof(buf)));
		if (len <= 0)
			fatal("Couldn't read packet: %s", strerror(errno));

		msg_len -= len;
		buffer_append(m, buf, len);
	}
}

void
send_string_request(int fd, u_int id, u_int code, char *s,
    u_int len)
{
	Buffer msg;

	buffer_init(&msg);
	buffer_put_char(&msg, code);
	buffer_put_int(&msg, id);
	buffer_put_string(&msg, s, len);
	send_msg(fd, &msg);
	debug3("Sent message fd %d T:%d I:%d", fd, code, id);
	buffer_free(&msg);
}

void
send_string_attrs_request(int fd, u_int id, u_int code, char *s,
    u_int len, Attrib *a)
{
	Buffer msg;

	buffer_init(&msg);
	buffer_put_char(&msg, code);
	buffer_put_int(&msg, id);
	buffer_put_string(&msg, s, len);
	encode_attrib(&msg, a);
	send_msg(fd, &msg);
	debug3("Sent message fd %d T:%d I:%d", fd, code, id);
	buffer_free(&msg);
}

u_int
get_status(int fd, int expected_id)
{
	Buffer msg;
	u_int type, id, status;

	buffer_init(&msg);
	get_msg(fd, &msg);
	type = buffer_get_char(&msg);
	id = buffer_get_int(&msg);

	if (id != expected_id)
		fatal("ID mismatch (%d != %d)", id, expected_id);
	if (type != SSH2_FXP_STATUS)
		fatal("Expected SSH2_FXP_STATUS(%d) packet, got %d",
		    SSH2_FXP_STATUS, type);

	status = buffer_get_int(&msg);
	buffer_free(&msg);

	debug3("SSH2_FXP_STATUS %d", status);

	return(status);
}

char *
get_handle(int fd, u_int expected_id, u_int *len)
{
	Buffer msg;
	u_int type, id;
	char *handle;

	buffer_init(&msg);
	get_msg(fd, &msg);
	type = buffer_get_char(&msg);
	id = buffer_get_int(&msg);

	if (id != expected_id)
		fatal("ID mismatch (%d != %d)", id, expected_id);
	if (type == SSH2_FXP_STATUS) {
		int status = buffer_get_int(&msg);

		error("Couldn't get handle: %s", fx2txt(status));
		return(NULL);
	} else if (type != SSH2_FXP_HANDLE)
		fatal("Expected SSH2_FXP_HANDLE(%d) packet, got %d",
		    SSH2_FXP_HANDLE, type);

	handle = buffer_get_string(&msg, len);
	buffer_free(&msg);

	return(handle);
}

Attrib *
get_decode_stat(int fd, u_int expected_id)
{
	Buffer msg;
	u_int type, id;
	Attrib *a;

	buffer_init(&msg);
	get_msg(fd, &msg);

	type = buffer_get_char(&msg);
	id = buffer_get_int(&msg);

	debug3("Received stat reply T:%d I:%d", type, id);
	if (id != expected_id)
		fatal("ID mismatch (%d != %d)", id, expected_id);
	if (type == SSH2_FXP_STATUS) {
		int status = buffer_get_int(&msg);

		error("Couldn't stat remote file: %s", fx2txt(status));
		return(NULL);
	} else if (type != SSH2_FXP_ATTRS) {
		fatal("Expected SSH2_FXP_ATTRS(%d) packet, got %d",
		    SSH2_FXP_ATTRS, type);
	}
	a = decode_attrib(&msg);
	buffer_free(&msg);

	return(a);
}

int
do_init(int fd_in, int fd_out)
{
	int type, version;
	Buffer msg;

	buffer_init(&msg);
	buffer_put_char(&msg, SSH2_FXP_INIT);
	buffer_put_int(&msg, SSH2_FILEXFER_VERSION);
	send_msg(fd_out, &msg);

	buffer_clear(&msg);

	get_msg(fd_in, &msg);

	/* Expecting a VERSION reply */
	if ((type = buffer_get_char(&msg)) != SSH2_FXP_VERSION) {
		error("Invalid packet back from SSH2_FXP_INIT (type %d)",
		    type);
		buffer_free(&msg);
		return(-1);
	}
	version = buffer_get_int(&msg);

	debug2("Remote version: %d", version);

	/* Check for extensions */
	while (buffer_len(&msg) > 0) {
		char *name = buffer_get_string(&msg, NULL);
		char *value = buffer_get_string(&msg, NULL);

		debug2("Init extension: \"%s\"", name);
		xfree(name);
		xfree(value);
	}

	buffer_free(&msg);
	return(0);
}

int
do_close(int fd_in, int fd_out, char *handle, u_int handle_len)
{
	u_int id, status;
	Buffer msg;

	buffer_init(&msg);

	id = msg_id++;
	buffer_put_char(&msg, SSH2_FXP_CLOSE);
	buffer_put_int(&msg, id);
	buffer_put_string(&msg, handle, handle_len);
	send_msg(fd_out, &msg);
	debug3("Sent message SSH2_FXP_CLOSE I:%d", id);

	status = get_status(fd_in, id);
	if (status != SSH2_FX_OK)
		error("Couldn't close file: %s", fx2txt(status));

	buffer_free(&msg);

	return(status);
}

int
do_ls(int fd_in, int fd_out, char *path)
{
	Buffer msg;
	u_int type, id, handle_len, i, expected_id;
	char *handle;

	id = msg_id++;

	buffer_init(&msg);
	buffer_put_char(&msg, SSH2_FXP_OPENDIR);
	buffer_put_int(&msg, id);
	buffer_put_cstring(&msg, path);
	send_msg(fd_out, &msg);

	buffer_clear(&msg);

	handle = get_handle(fd_in, id, &handle_len);
	if (handle == NULL)
		return(-1);

	for(;;) {
		int count;

		id = expected_id = msg_id++;

		debug3("Sending SSH2_FXP_READDIR I:%d", id);

		buffer_clear(&msg);
		buffer_put_char(&msg, SSH2_FXP_READDIR);
		buffer_put_int(&msg, id);
		buffer_put_string(&msg, handle, handle_len);
		send_msg(fd_out, &msg);

		buffer_clear(&msg);

		get_msg(fd_in, &msg);

		type = buffer_get_char(&msg);
		id = buffer_get_int(&msg);

		debug3("Received reply T:%d I:%d", type, id);

		if (id != expected_id)
			fatal("ID mismatch (%d != %d)", id, expected_id);

		if (type == SSH2_FXP_STATUS) {
			int status = buffer_get_int(&msg);

			debug3("Received SSH2_FXP_STATUS %d", status);

			if (status == SSH2_FX_EOF) {
				break;
			} else {
				error("Couldn't read directory: %s",
				    fx2txt(status));
				do_close(fd_in, fd_out, handle, handle_len);
				return(status);
			}
		} else if (type != SSH2_FXP_NAME)
			fatal("Expected SSH2_FXP_NAME(%d) packet, got %d",
			    SSH2_FXP_NAME, type);

		count = buffer_get_int(&msg);
		if (count == 0)
			break;
		debug3("Received %d SSH2_FXP_NAME responses", count);
		for(i = 0; i < count; i++) {
			char *filename, *longname;
			Attrib *a;

			filename = buffer_get_string(&msg, NULL);
			longname = buffer_get_string(&msg, NULL);
			a = decode_attrib(&msg);

			printf("%s\n", longname);

			xfree(filename);
			xfree(longname);
		}
	}

	buffer_free(&msg);
	do_close(fd_in, fd_out, handle, handle_len);
	xfree(handle);

	return(0);
}

int
do_rm(int fd_in, int fd_out, char *path)
{
	u_int status, id;

	debug2("Sending SSH2_FXP_REMOVE \"%s\"", path);

	id = msg_id++;
	send_string_request(fd_out, id, SSH2_FXP_REMOVE, path, strlen(path));
	status = get_status(fd_in, id);
	if (status != SSH2_FX_OK)
		error("Couldn't delete file: %s", fx2txt(status));
	return(status);
}

int
do_mkdir(int fd_in, int fd_out, char *path, Attrib *a)
{
	u_int status, id;

	id = msg_id++;
	send_string_attrs_request(fd_out, id, SSH2_FXP_MKDIR, path,
	    strlen(path), a);

	status = get_status(fd_in, id);
	if (status != SSH2_FX_OK)
		error("Couldn't create directory: %s", fx2txt(status));

	return(status);
}

int
do_rmdir(int fd_in, int fd_out, char *path)
{
	u_int status, id;

	id = msg_id++;
	send_string_request(fd_out, id, SSH2_FXP_RMDIR, path, strlen(path));

	status = get_status(fd_in, id);
	if (status != SSH2_FX_OK)
		error("Couldn't remove directory: %s", fx2txt(status));

	return(status);
}

Attrib *
do_stat(int fd_in, int fd_out, char *path)
{
	u_int id;

	id = msg_id++;
	send_string_request(fd_out, id, SSH2_FXP_STAT, path, strlen(path));
	return(get_decode_stat(fd_in, id));
}

Attrib *
do_lstat(int fd_in, int fd_out, char *path)
{
	u_int id;

	id = msg_id++;
	send_string_request(fd_out, id, SSH2_FXP_LSTAT, path, strlen(path));
	return(get_decode_stat(fd_in, id));
}

Attrib *
do_fstat(int fd_in, int fd_out, char *handle,
    u_int handle_len)
{
	u_int id;

	id = msg_id++;
	send_string_request(fd_out, id, SSH2_FXP_FSTAT, handle, handle_len);
	return(get_decode_stat(fd_in, id));
}

int
do_setstat(int fd_in, int fd_out, char *path, Attrib *a)
{
	u_int status, id;

	id = msg_id++;
	send_string_attrs_request(fd_out, id, SSH2_FXP_SETSTAT, path,
	    strlen(path), a);

	status = get_status(fd_in, id);
	if (status != SSH2_FX_OK)
		error("Couldn't setstat on \"%s\": %s", path,
		    fx2txt(status));

	return(status);
}

int
do_fsetstat(int fd_in, int fd_out, char *handle, u_int handle_len,
    Attrib *a)
{
	u_int status, id;

	id = msg_id++;
	send_string_attrs_request(fd_out, id, SSH2_FXP_FSETSTAT, handle,
	    handle_len, a);

	status = get_status(fd_in, id);
	if (status != SSH2_FX_OK)
		error("Couldn't fsetstat: %s", fx2txt(status));

	return(status);
}

char *
do_realpath(int fd_in, int fd_out, char *path)
{
	Buffer msg;
	u_int type, expected_id, count, id;
	char *filename, *longname;
	Attrib *a;

	expected_id = id = msg_id++;
	send_string_request(fd_out, id, SSH2_FXP_REALPATH, path,
	    strlen(path));

	buffer_init(&msg);

	get_msg(fd_in, &msg);
	type = buffer_get_char(&msg);
	id = buffer_get_int(&msg);

	if (id != expected_id)
		fatal("ID mismatch (%d != %d)", id, expected_id);

	if (type == SSH2_FXP_STATUS) {
		u_int status = buffer_get_int(&msg);

		error("Couldn't canonicalise: %s", fx2txt(status));
		return(NULL);
	} else if (type != SSH2_FXP_NAME)
		fatal("Expected SSH2_FXP_NAME(%d) packet, got %d",
		    SSH2_FXP_NAME, type);

	count = buffer_get_int(&msg);
	if (count != 1)
		fatal("Got multiple names (%d) from SSH_FXP_REALPATH", count);

	filename = buffer_get_string(&msg, NULL);
	longname = buffer_get_string(&msg, NULL);
	a = decode_attrib(&msg);

	debug3("SSH_FXP_REALPATH %s -> %s", path, filename);

	xfree(longname);

	buffer_free(&msg);

	return(filename);
}

int
do_rename(int fd_in, int fd_out, char *oldpath, char *newpath)
{
	Buffer msg;
	u_int status, id;

	buffer_init(&msg);

	/* Send rename request */
	id = msg_id++;
	buffer_put_char(&msg, SSH2_FXP_RENAME);
	buffer_put_int(&msg, id);
	buffer_put_cstring(&msg, oldpath);
	buffer_put_cstring(&msg, newpath);
	send_msg(fd_out, &msg);
	debug3("Sent message SSH2_FXP_RENAME \"%s\" -> \"%s\"", oldpath,
	    newpath);
	buffer_free(&msg);

	status = get_status(fd_in, id);
	if (status != SSH2_FX_OK)
		error("Couldn't rename file \"%s\" to \"%s\": %s", oldpath, newpath,
		    fx2txt(status));

	return(status);
}

int
do_download(int fd_in, int fd_out, char *remote_path, char *local_path,
    int pflag)
{
	int local_fd;
	u_int expected_id, handle_len, mode, type, id;
	u_int64_t offset;
	char *handle;
	Buffer msg;
	Attrib junk, *a;
	int status;

	a = do_stat(fd_in, fd_out, remote_path);
	if (a == NULL)
		return(-1);

	/* XXX: should we preserve set[ug]id? */
	if (a->flags & SSH2_FILEXFER_ATTR_PERMISSIONS)
		mode = S_IWRITE | (a->perm & 0777);
	else
		mode = 0666;

	local_fd = open(local_path, O_WRONLY | O_CREAT | O_TRUNC, mode);
	if (local_fd == -1) {
		error("Couldn't open local file \"%s\" for writing: %s",
		    local_path, strerror(errno));
		return(errno);
	}

	/* Override umask and utimes if asked */
	if (pflag && fchmod(local_fd, mode) == -1)
		error("Couldn't set mode on \"%s\": %s", local_path,
		    strerror(errno));
	if (pflag && (a->flags & SSH2_FILEXFER_ATTR_ACMODTIME)) {
		struct timeval tv;

		tv.tv_sec = a->atime;
		tv.tv_usec = a->mtime;
		if (utimes(local_path, &tv) == -1)
			error("Can't set times on \"%s\": %s", local_path,
			    strerror(errno));
	}

	buffer_init(&msg);

	/* Send open request */
	id = msg_id++;
	buffer_put_char(&msg, SSH2_FXP_OPEN);
	buffer_put_int(&msg, id);
	buffer_put_cstring(&msg, remote_path);
	buffer_put_int(&msg, SSH2_FXF_READ);
	attrib_clear(&junk); /* Send empty attributes */
	encode_attrib(&msg, &junk);
	send_msg(fd_out, &msg);
	debug3("Sent message SSH2_FXP_OPEN I:%d P:%s", id, remote_path);

	handle = get_handle(fd_in, id, &handle_len);
	if (handle == NULL) {
		buffer_free(&msg);
		close(local_fd);
		return(-1);
	}

	/* Read from remote and write to local */
	offset = 0;
	for(;;) {
		u_int len;
		char *data;

		id = expected_id = msg_id++;

		buffer_clear(&msg);
		buffer_put_char(&msg, SSH2_FXP_READ);
		buffer_put_int(&msg, id);
		buffer_put_string(&msg, handle, handle_len);
		buffer_put_int64(&msg, offset);
		buffer_put_int(&msg, COPY_SIZE);
		send_msg(fd_out, &msg);
		debug3("Sent message SSH2_FXP_READ I:%d O:%llu S:%u",
		    id, (unsigned long long)offset, COPY_SIZE);

		buffer_clear(&msg);

		get_msg(fd_in, &msg);
		type = buffer_get_char(&msg);
		id = buffer_get_int(&msg);
		debug3("Received reply T:%d I:%d", type, id);
		if (id != expected_id)
			fatal("ID mismatch (%d != %d)", id, expected_id);
		if (type == SSH2_FXP_STATUS) {
			status = buffer_get_int(&msg);

			if (status == SSH2_FX_EOF)
				break;
			else {
				error("Couldn't read from remote "
				    "file \"%s\" : %s", remote_path,
				     fx2txt(status));
				do_close(fd_in, fd_out, handle, handle_len);
				goto done;
			}
		} else if (type != SSH2_FXP_DATA) {
			fatal("Expected SSH2_FXP_DATA(%d) packet, got %d",
			    SSH2_FXP_DATA, type);
		}

		data = buffer_get_string(&msg, &len);
		if (len > COPY_SIZE)
			fatal("Received more data than asked for %d > %d",
			    len, COPY_SIZE);

		debug3("In read loop, got %d offset %llu", len,
		    (unsigned long long)offset);
		if (atomicio(write, local_fd, data, len) != len) {
			error("Couldn't write to \"%s\": %s", local_path,
			    strerror(errno));
			do_close(fd_in, fd_out, handle, handle_len);
			status = -1;
			xfree(data);
			goto done;
		}

		offset += len;
		xfree(data);
	}
	status = do_close(fd_in, fd_out, handle, handle_len);

done:
	close(local_fd);
	buffer_free(&msg);
	xfree(handle);
	return status;
}

int
do_upload(int fd_in, int fd_out, char *local_path, char *remote_path,
    int pflag)
{
	int local_fd;
	u_int handle_len, id;
	u_int64_t offset;
	char *handle;
	Buffer msg;
	struct stat sb;
	Attrib a;
	int status;

	if ((local_fd = open(local_path, O_RDONLY, 0)) == -1) {
		error("Couldn't open local file \"%s\" for reading: %s",
		    local_path, strerror(errno));
		return(-1);
	}
	if (fstat(local_fd, &sb) == -1) {
		error("Couldn't fstat local file \"%s\": %s",
		    local_path, strerror(errno));
		close(local_fd);
		return(-1);
	}
	stat_to_attrib(&sb, &a);

	a.flags &= ~SSH2_FILEXFER_ATTR_SIZE;
	a.flags &= ~SSH2_FILEXFER_ATTR_UIDGID;
	a.perm &= 0777;
	if (!pflag)
		a.flags &= ~SSH2_FILEXFER_ATTR_ACMODTIME;

	buffer_init(&msg);

	/* Send open request */
	id = msg_id++;
	buffer_put_char(&msg, SSH2_FXP_OPEN);
	buffer_put_int(&msg, id);
	buffer_put_cstring(&msg, remote_path);
	buffer_put_int(&msg, SSH2_FXF_WRITE|SSH2_FXF_CREAT|SSH2_FXF_TRUNC);
	encode_attrib(&msg, &a);
	send_msg(fd_out, &msg);
	debug3("Sent message SSH2_FXP_OPEN I:%d P:%s", id, remote_path);

	buffer_clear(&msg);

	handle = get_handle(fd_in, id, &handle_len);
	if (handle == NULL) {
		close(local_fd);
		buffer_free(&msg);
		return(-1);
	}

	/* Override umask and utimes if asked */
	if (pflag)
		do_fsetstat(fd_in, fd_out, handle, handle_len, &a);

	/* Read from local and write to remote */
	offset = 0;
	for(;;) {
		int len;
		char data[COPY_SIZE];

		/*
		 * Can't use atomicio here because it returns 0 on EOF, thus losing
		 * the last block of the file
		 */
		do
			len = read(local_fd, data, COPY_SIZE);
		while ((len == -1) && (errno == EINTR || errno == EAGAIN));

		if (len == -1)
			fatal("Couldn't read from \"%s\": %s", local_path,
			    strerror(errno));
		if (len == 0)
			break;

		buffer_clear(&msg);
		buffer_put_char(&msg, SSH2_FXP_WRITE);
		buffer_put_int(&msg, ++id);
		buffer_put_string(&msg, handle, handle_len);
		buffer_put_int64(&msg, offset);
		buffer_put_string(&msg, data, len);
		send_msg(fd_out, &msg);
		debug3("Sent message SSH2_FXP_WRITE I:%d O:%llu S:%u",
		    id, (unsigned long long)offset, len);

		status = get_status(fd_in, id);
		if (status != SSH2_FX_OK) {
			error("Couldn't write to remote file \"%s\": %s",
			    remote_path, fx2txt(status));
			do_close(fd_in, fd_out, handle, handle_len);
			close(local_fd);
			goto done;
		}
		debug3("In write loop, got %d offset %llu", len,
		    (unsigned long long)offset);

		offset += len;
	}

	if (close(local_fd) == -1) {
		error("Couldn't close local file \"%s\": %s", local_path,
		    strerror(errno));
		do_close(fd_in, fd_out, handle, handle_len);
		status = -1;
		goto done;
	}

	status = do_close(fd_in, fd_out, handle, handle_len);

done:
	xfree(handle);
	buffer_free(&msg);
	return status;
}
Commit	Line	Data
61e96248	1	/*
	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"
b655a207	32	RCSID("$OpenBSD: sftp-client.c,v 1.9 2001/02/10 00:41:46 djm Exp $");
61e96248	33
	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
9c5a8165	51	/* Message ID */
	52	static u_int msg_id = 1;
	53
61e96248	54	void
	55	send_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
	73	void
	74	get_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
	97	void
	98	send_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
	112	void
	113	send_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
128	u_int
129	get_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
153	char *
154	get_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
182	Attrib *
183	get_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
213	int
214	do_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
2b87da3b	228	/* Expecting a VERSION reply */
61e96248	229	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
	253	int
	254	do_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
9c5a8165	261	id = msg_id++;
61e96248	262	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
	277	int
	278	do_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
9c5a8165	284	id = msg_id++;
61e96248	285
	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
9c5a8165	301	id = expected_id = msg_id++;
61e96248	302
	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);
b655a207	334	return(status);
61e96248	335	}
	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);
0426a3b4	341	if (count == 0)
	342	break;
	343	debug3("Received %d SSH2_FXP_NAME responses", count);
61e96248	344	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
	366	int
	367	do_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
9c5a8165	373	id = msg_id++;
61e96248	374	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
	381	int
	382	do_mkdir(int fd_in, int fd_out, char path, Attrib a)
	383	{
	384	u_int status, id;
	385
9c5a8165	386	id = msg_id++;
61e96248	387	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
	397	int
	398	do_rmdir(int fd_in, int fd_out, char *path)
	399	{
	400	u_int status, id;
	401
9c5a8165	402	id = msg_id++;
61e96248	403	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
	412	Attrib *
	413	do_stat(int fd_in, int fd_out, char *path)
	414	{
	415	u_int id;
	416
9c5a8165	417	id = msg_id++;
61e96248	418	send_string_request(fd_out, id, SSH2_FXP_STAT, path, strlen(path));
	419	return(get_decode_stat(fd_in, id));
	420	}
	421
	422	Attrib *
	423	do_lstat(int fd_in, int fd_out, char *path)
	424	{
	425	u_int id;
	426
9c5a8165	427	id = msg_id++;
61e96248	428	send_string_request(fd_out, id, SSH2_FXP_LSTAT, path, strlen(path));
	429	return(get_decode_stat(fd_in, id));
	430	}
	431
	432	Attrib *
	433	do_fstat(int fd_in, int fd_out, char *handle,
	434	u_int handle_len)
	435	{
	436	u_int id;
	437
9c5a8165	438	id = msg_id++;
61e96248	439	send_string_request(fd_out, id, SSH2_FXP_FSTAT, handle, handle_len);
	440	return(get_decode_stat(fd_in, id));
	441	}
	442
	443	int
	444	do_setstat(int fd_in, int fd_out, char path, Attrib a)
	445	{
	446	u_int status, id;
	447
9c5a8165	448	id = msg_id++;
61e96248	449	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
	460	int
	461	do_fsetstat(int fd_in, int fd_out, char *handle, u_int handle_len,
	462	Attrib *a)
	463	{
	464	u_int status, id;
	465
9c5a8165	466	id = msg_id++;
61e96248	467	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
	477	char *
	478	do_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
9c5a8165	485	expected_id = id = msg_id++;
61e96248	486	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
	524	int
	525	do_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 */
9c5a8165	533	id = msg_id++;
61e96248	534	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
	551	int
	552	do_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;
0426a3b4	561	int status;
61e96248	562
	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 */
9c5a8165	597	id = msg_id++;
61e96248	598	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
9c5a8165	620	id = expected_id = msg_id++;
61e96248	621
	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",
9378f292	630	id, (unsigned long long)offset, COPY_SIZE);
61e96248	631
	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) {
0426a3b4	641	status = buffer_get_int(&msg);
61e96248	642
	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);
0426a3b4	650	goto done;
61e96248	651	}
	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
0426a3b4	662	debug3("In read loop, got %d offset %llu", len,
9378f292	663	(unsigned long long)offset);
61e96248	664	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);
0426a3b4	668	status = -1;
61e96248	669	xfree(data);
0426a3b4	670	goto done;
61e96248	671	}
	672
	673	offset += len;
	674	xfree(data);
	675	}
0426a3b4	676	status = do_close(fd_in, fd_out, handle, handle_len);
61e96248	677
0426a3b4	678	done:
	679	close(local_fd);
	680	buffer_free(&msg);
	681	xfree(handle);
	682	return status;
61e96248	683	}
	684
	685	int
	686	do_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;
0426a3b4	696	int status;
61e96248	697
	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 */
9c5a8165	720	id = msg_id++;
61e96248	721	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];
61e96248	747
	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",
9378f292	770	id, (unsigned long long)offset, len);
61e96248	771
	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);
61e96248	777	close(local_fd);
0426a3b4	778	goto done;
61e96248	779	}
9378f292	780	debug3("In write loop, got %d offset %llu", len,
9378f292	781	(unsigned long long)offset);
61e96248	782
	783	offset += len;
	784	}
61e96248	785
	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);
0426a3b4	790	status = -1;
0426a3b4	791	goto done;
61e96248	792	}
61e96248	793
0426a3b4	794	status = do_close(fd_in, fd_out, handle, handle_len);
	795
	796	done:
	797	xfree(handle);
	798	buffer_free(&msg);
	799	return status;
61e96248	800	}
0426a3b4	801
0426a3b4	802