[openssh.git] / sftp.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.
 */

#include "includes.h"

RCSID("$OpenBSD: sftp.c,v 1.14 2001/04/12 23:17:54 mouring Exp $");

/* XXX: commandline mode */
/* XXX: copy between two remote hosts (commandline) */
/* XXX: short-form remote directory listings (like 'ls -C') */

#include "buffer.h"
#include "xmalloc.h"
#include "log.h"
#include "pathnames.h"

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

#ifdef HAVE___PROGNAME
extern char *__progname;
#else
char *__progname;
#endif

int use_ssh1 = 0;
char *ssh_program = _PATH_SSH_PROGRAM;
char *sftp_server = NULL;
FILE* infile;

void
connect_to_server(char **args, int *in, int *out, pid_t *sshpid)
{
	int c_in, c_out;
#ifdef USE_PIPES
	int pin[2], pout[2];
	if ((pipe(pin) == -1) || (pipe(pout) == -1))
		fatal("pipe: %s", strerror(errno));
	*in = pin[0];
	*out = pout[1];
	c_in = pout[0];
	c_out = pin[1];
#else /* USE_PIPES */
	int inout[2];
	if (socketpair(AF_UNIX, SOCK_STREAM, 0, inout) == -1)
		fatal("socketpair: %s", strerror(errno));
	*in = *out = inout[0];
	c_in = c_out = inout[1];
#endif /* USE_PIPES */

	if ((*sshpid = fork()) == -1)
		fatal("fork: %s", strerror(errno));
	else if (*sshpid == 0) {
		if ((dup2(c_in, STDIN_FILENO) == -1) ||
		    (dup2(c_out, STDOUT_FILENO) == -1)) {
			fprintf(stderr, "dup2: %s\n", strerror(errno));
			exit(1);
		}
		close(*in);
		close(*out);
		close(c_in);
		close(c_out);
		execv(ssh_program, args);
		fprintf(stderr, "exec: %s: %s\n", ssh_program, strerror(errno));
		exit(1);
	}

	close(c_in);
	close(c_out);
}

char **
make_ssh_args(char *add_arg)
{
	static char **args = NULL;
	static int nargs = 0;
	char debug_buf[4096];
	int i;

	/* Init args array */
	if (args == NULL) {
		nargs = 2;
		i = 0;
		args = xmalloc(sizeof(*args) * nargs);
		args[i++] = "ssh";
		args[i++] = NULL;
	}

	/* If asked to add args, then do so and return */
	if (add_arg) {
		i = nargs++ - 1;
		args = xrealloc(args, sizeof(*args) * nargs);
		args[i++] = add_arg;
		args[i++] = NULL;
		return(NULL);
	}

	/* no subsystem if the server-spec contains a '/' */
	if (sftp_server == NULL || strchr(sftp_server, '/') == NULL)
		make_ssh_args("-s");
	make_ssh_args("-oForwardX11=no");
	make_ssh_args("-oForwardAgent=no");
	make_ssh_args(use_ssh1 ? "-oProtocol=1" : "-oProtocol=2");

	/* Otherwise finish up and return the arg array */
	if (sftp_server != NULL)
		make_ssh_args(sftp_server);
	else
		make_ssh_args("sftp");

	/* XXX: overflow - doesn't grow debug_buf */
	debug_buf[0] = '\0';
	for(i = 0; args[i]; i++) {
		if (i)
			strlcat(debug_buf, " ", sizeof(debug_buf));

		strlcat(debug_buf, args[i], sizeof(debug_buf));
	}
	debug("SSH args \"%s\"", debug_buf);

	return(args);
}

void
usage(void)
{
	fprintf(stderr, "usage: sftp [-1vC] [-b batchfile] [-osshopt=value] [user@]host[:file [file]]\n");
	exit(1);
}

int
main(int argc, char **argv)
{
	int in, out, ch, debug_level, compress_flag;
	pid_t sshpid;
	char *file1 = NULL;
	char *host, *userhost, *cp, *file2;
	LogLevel ll;
	extern int optind;
	extern char *optarg;

	__progname = get_progname(argv[0]);
	infile = stdin;         /* Read from STDIN unless changed by -b */
	debug_level = compress_flag = 0;

	while ((ch = getopt(argc, argv, "1hvCo:s:S:b:")) != -1) {
		switch (ch) {
		case 'C':
			compress_flag = 1;
			break;
		case 'v':
			debug_level = MIN(3, debug_level + 1);
			break;
		case 'o':
			make_ssh_args("-o");
			make_ssh_args(optarg);
			break;
		case '1':
			use_ssh1 = 1;
			if (sftp_server == NULL)
				sftp_server = _PATH_SFTP_SERVER;
			break;
		case 's':
			sftp_server = optarg;
			break;
		case 'S':
			ssh_program = optarg;
			break;
		case 'b':
			if (infile == stdin) {
				infile = fopen(optarg, "r");
				if (infile == NULL)
					fatal("%s (%s).", strerror(errno), optarg);
			} else
				fatal("Filename already specified.");
			break;
		case 'h':
		default:
			usage();
		}
	}

	if (optind == argc || argc > (optind + 2))
		usage();

	userhost = xstrdup(argv[optind]);
	file2 = argv[optind+1];

	if ((cp = strchr(userhost, ':')) != NULL) {
		*cp++ = '\0';
		file1 = cp;
	}

	if ((host = strchr(userhost, '@')) == NULL)
		host = userhost;
	else {
		*host++ = '\0';
		if (!userhost[0]) {
			fprintf(stderr, "Missing username\n");
			usage();
		}
		make_ssh_args("-l");
		make_ssh_args(userhost);
	}

	if (!*host) {
		fprintf(stderr, "Missing hostname\n");
		usage();
	}

	/* Set up logging and debug '-d' arguments to ssh */
	ll = SYSLOG_LEVEL_INFO;
	switch (debug_level) {
	case 1:
		ll = SYSLOG_LEVEL_DEBUG1;
		make_ssh_args("-v");
		break;
	case 2:
		ll = SYSLOG_LEVEL_DEBUG2;
		make_ssh_args("-v");
		make_ssh_args("-v");
		break;
	case 3:
		ll = SYSLOG_LEVEL_DEBUG3;
		make_ssh_args("-v");
		make_ssh_args("-v");
		make_ssh_args("-v");
		break;
	}

	if (compress_flag)
		make_ssh_args("-C");

	log_init(argv[0], ll, SYSLOG_FACILITY_USER, 1);

	make_ssh_args(host);

	fprintf(stderr, "Connecting to %s...\n", host);

	connect_to_server(make_ssh_args(NULL), &in, &out, &sshpid);

	interactive_loop(in, out, file1, file2);

#if !defined(USE_PIPES)
        shutdown(in, SHUT_RDWR);
        shutdown(out, SHUT_RDWR);
#endif

	close(in);
	close(out);
	if (infile != stdin)
		fclose(infile);

	if (waitpid(sshpid, NULL, 0) == -1)
		fatal("Couldn't wait for ssh process: %s", strerror(errno));

	exit(0);
}
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	#include "includes.h"
	26
edeeab1e	27	RCSID("$OpenBSD: sftp.c,v 1.14 2001/04/12 23:17:54 mouring Exp $");
61e96248	28
	29	/* XXX: commandline mode */
	30	/* XXX: copy between two remote hosts (commandline) */
	31	/* XXX: short-form remote directory listings (like 'ls -C') */
	32
	33	#include "buffer.h"
	34	#include "xmalloc.h"
	35	#include "log.h"
	36	#include "pathnames.h"
	37
	38	#include "sftp.h"
	39	#include "sftp-common.h"
	40	#include "sftp-client.h"
	41	#include "sftp-int.h"
	42
5152d46f	43	#ifdef HAVE___PROGNAME
	44	extern char *__progname;
	45	#else
	46	char *__progname;
	47	#endif
	48
0426a3b4	49	int use_ssh1 = 0;
	50	char *ssh_program = _PATH_SSH_PROGRAM;
	51	char *sftp_server = NULL;
a5ec8a3d	52	FILE* infile;
0426a3b4	53
61e96248	54	void
	55	connect_to_server(char *args, int in, int out, pid_t sshpid)
	56	{
	57	int c_in, c_out;
	58	#ifdef USE_PIPES
	59	int pin[2], pout[2];
	60	if ((pipe(pin) == -1) \|\| (pipe(pout) == -1))
	61	fatal("pipe: %s", strerror(errno));
	62	*in = pin[0];
	63	*out = pout[1];
	64	c_in = pout[0];
	65	c_out = pin[1];
	66	#else /* USE_PIPES */
	67	int inout[2];
	68	if (socketpair(AF_UNIX, SOCK_STREAM, 0, inout) == -1)
	69	fatal("socketpair: %s", strerror(errno));
	70	in = out = inout[0];
	71	c_in = c_out = inout[1];
	72	#endif /* USE_PIPES */
	73
	74	if ((*sshpid = fork()) == -1)
	75	fatal("fork: %s", strerror(errno));
	76	else if (*sshpid == 0) {
	77	if ((dup2(c_in, STDIN_FILENO) == -1) \|\|
	78	(dup2(c_out, STDOUT_FILENO) == -1)) {
	79	fprintf(stderr, "dup2: %s\n", strerror(errno));
	80	exit(1);
	81	}
	82	close(*in);
	83	close(*out);
	84	close(c_in);
	85	close(c_out);
0426a3b4	86	execv(ssh_program, args);
0426a3b4	87	fprintf(stderr, "exec: %s: %s\n", ssh_program, strerror(errno));
61e96248	88	exit(1);
	89	}
	90
	91	close(c_in);
	92	close(c_out);
	93	}
	94
	95	char **
	96	make_ssh_args(char *add_arg)
	97	{
	98	static char **args = NULL;
	99	static int nargs = 0;
	100	char debug_buf[4096];
c630ce76	101	int i;
61e96248	102
	103	/* Init args array */
	104	if (args == NULL) {
c630ce76	105	nargs = 2;
61e96248	106	i = 0;
	107	args = xmalloc(sizeof(args) nargs);
	108	args[i++] = "ssh";
61e96248	109	args[i++] = NULL;
	110	}
	111
	112	/* If asked to add args, then do so and return */
	113	if (add_arg) {
	114	i = nargs++ - 1;
	115	args = xrealloc(args, sizeof(args) nargs);
	116	args[i++] = add_arg;
	117	args[i++] = NULL;
	118	return(NULL);
	119	}
	120
c630ce76	121	/* no subsystem if the server-spec contains a '/' */
	122	if (sftp_server == NULL \|\| strchr(sftp_server, '/') == NULL)
	123	make_ssh_args("-s");
	124	make_ssh_args("-oForwardX11=no");
	125	make_ssh_args("-oForwardAgent=no");
	126	make_ssh_args(use_ssh1 ? "-oProtocol=1" : "-oProtocol=2");
	127
61e96248	128	/* Otherwise finish up and return the arg array */
0426a3b4	129	if (sftp_server != NULL)
	130	make_ssh_args(sftp_server);
	131	else
	132	make_ssh_args("sftp");
61e96248	133
	134	/* XXX: overflow - doesn't grow debug_buf */
	135	debug_buf[0] = '\0';
	136	for(i = 0; args[i]; i++) {
	137	if (i)
	138	strlcat(debug_buf, " ", sizeof(debug_buf));
	139
	140	strlcat(debug_buf, args[i], sizeof(debug_buf));
	141	}
	142	debug("SSH args \"%s\"", debug_buf);
	143
	144	return(args);
	145	}
	146
2b87da3b	147	void
61e96248	148	usage(void)
61e96248	149	{
edeeab1e	150	fprintf(stderr, "usage: sftp [-1vC] [-b batchfile] [-osshopt=value] [user@]host[:file [file]]\n");
61e96248	151	exit(1);
	152	}
	153
2b87da3b	154	int
61e96248	155	main(int argc, char **argv)
61e96248	156	{
0426a3b4	157	int in, out, ch, debug_level, compress_flag;
61e96248	158	pid_t sshpid;
edeeab1e	159	char *file1 = NULL;
edeeab1e	160	char host, userhost, cp, file2;
61e96248	161	LogLevel ll;
0426a3b4	162	extern int optind;
0426a3b4	163	extern char *optarg;
61e96248	164
5152d46f	165	__progname = get_progname(argv[0]);
a5ec8a3d	166	infile = stdin; /* Read from STDIN unless changed by -b */
61e96248	167	debug_level = compress_flag = 0;
0426a3b4	168
a5ec8a3d	169	while ((ch = getopt(argc, argv, "1hvCo:s:S:b:")) != -1) {
0426a3b4	170	switch (ch) {
0426a3b4	171	case 'C':
61e96248	172	compress_flag = 1;
0426a3b4	173	break;
	174	case 'v':
	175	debug_level = MIN(3, debug_level + 1);
	176	break;
	177	case 'o':
	178	make_ssh_args("-o");
	179	make_ssh_args(optarg);
	180	break;
	181	case '1':
	182	use_ssh1 = 1;
	183	if (sftp_server == NULL)
	184	sftp_server = _PATH_SFTP_SERVER;
	185	break;
	186	case 's':
	187	sftp_server = optarg;
	188	break;
	189	case 'S':
	190	ssh_program = optarg;
	191	break;
a5ec8a3d	192	case 'b':
	193	if (infile == stdin) {
	194	infile = fopen(optarg, "r");
cd332296	195	if (infile == NULL)
a5ec8a3d	196	fatal("%s (%s).", strerror(errno), optarg);
cd332296	197	} else
a5ec8a3d	198	fatal("Filename already specified.");
a5ec8a3d	199	break;
0426a3b4	200	case 'h':
0426a3b4	201	default:
61e96248	202	usage();
	203	}
	204	}
	205
edeeab1e	206	if (optind == argc \|\| argc > (optind + 2))
61e96248	207	usage();
61e96248	208
0a85ab61	209	userhost = xstrdup(argv[optind]);
edeeab1e	210	file2 = argv[optind+1];
	211
	212	if ((cp = strchr(userhost, ':')) != NULL) {
	213	*cp++ = '\0';
	214	file1 = cp;
	215	}
0426a3b4	216
	217	if ((host = strchr(userhost, '@')) == NULL)
	218	host = userhost;
61e96248	219	else {
edeeab1e	220	*host++ = '\0';
0426a3b4	221	if (!userhost[0]) {
61e96248	222	fprintf(stderr, "Missing username\n");
	223	usage();
	224	}
	225	make_ssh_args("-l");
0426a3b4	226	make_ssh_args(userhost);
61e96248	227	}
61e96248	228
0426a3b4	229	if (!*host) {
61e96248	230	fprintf(stderr, "Missing hostname\n");
	231	usage();
	232	}
	233
	234	/* Set up logging and debug '-d' arguments to ssh */
	235	ll = SYSLOG_LEVEL_INFO;
	236	switch (debug_level) {
	237	case 1:
	238	ll = SYSLOG_LEVEL_DEBUG1;
	239	make_ssh_args("-v");
	240	break;
	241	case 2:
	242	ll = SYSLOG_LEVEL_DEBUG2;
	243	make_ssh_args("-v");
	244	make_ssh_args("-v");
	245	break;
	246	case 3:
	247	ll = SYSLOG_LEVEL_DEBUG3;
	248	make_ssh_args("-v");
	249	make_ssh_args("-v");
	250	make_ssh_args("-v");
	251	break;
	252	}
	253
	254	if (compress_flag)
	255	make_ssh_args("-C");
	256
	257	log_init(argv[0], ll, SYSLOG_FACILITY_USER, 1);
	258
0426a3b4	259	make_ssh_args(host);
61e96248	260
0426a3b4	261	fprintf(stderr, "Connecting to %s...\n", host);
61e96248	262
	263	connect_to_server(make_ssh_args(NULL), &in, &out, &sshpid);
	264
edeeab1e	265	interactive_loop(in, out, file1, file2);
61e96248	266
51fb577a	267	#if !defined(USE_PIPES)
	268	shutdown(in, SHUT_RDWR);
	269	shutdown(out, SHUT_RDWR);
	270	#endif
	271
61e96248	272	close(in);
61e96248	273	close(out);
a5ec8a3d	274	if (infile != stdin)
a5ec8a3d	275	fclose(infile);
61e96248	276
0426a3b4	277	if (waitpid(sshpid, NULL, 0) == -1)
0426a3b4	278	fatal("Couldn't wait for ssh process: %s", strerror(errno));
61e96248	279
	280	exit(0);
	281	}