[openssh.git] / sftp.c

/*
 * Copyright (c) 2001,2002 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.39 2004/01/13 09:25:05 djm Exp $");

#include "buffer.h"
#include "xmalloc.h"
#include "log.h"
#include "pathnames.h"
#include "misc.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

FILE* infile = stdin;
int batchmode = 0;
size_t copy_buffer_len = 32768;
size_t num_requests = 16;
static pid_t sshpid = -1;

extern int showprogress;

static void
killchild(int signo)
{
	if (sshpid > 1)
		kill(sshpid, signo);

	_exit(1);
}

static void
connect_to_server(char *path, char **args, int *in, int *out)
{
	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(path, args);
		fprintf(stderr, "exec: %s: %s\n", path, strerror(errno));
		exit(1);
	}

	signal(SIGTERM, killchild);
	signal(SIGINT, killchild);
	signal(SIGHUP, killchild);
	close(c_in);
	close(c_out);
}

static void
usage(void)
{
	extern char *__progname;

	fprintf(stderr,
	    "usage: %s [-1Cv] [-B buffer_size] [-b batchfile] [-F ssh_config]\n"
	    "            [-o ssh_option] [-P sftp_server_path] [-R num_requests]\n"
	    "            [-S program] [-s subsystem | sftp_server] host\n"
	    "       %s [[user@]host[:file [file]]]\n"
	    "       %s [[user@]host[:dir[/]]]\n"
	    "       %s -b batchfile [user@]host\n", __progname, __progname, __progname, __progname);
	exit(1);
}

int
main(int argc, char **argv)
{
	int in, out, ch, err;
	char *host, *userhost, *cp, *file2;
	int debug_level = 0, sshver = 2;
	char *file1 = NULL, *sftp_server = NULL;
	char *ssh_program = _PATH_SSH_PROGRAM, *sftp_direct = NULL;
	LogLevel ll = SYSLOG_LEVEL_INFO;
	arglist args;
	extern int optind;
	extern char *optarg;

	__progname = ssh_get_progname(argv[0]);
	args.list = NULL;
	addargs(&args, "ssh");		/* overwritten with ssh_program */
	addargs(&args, "-oForwardX11 no");
	addargs(&args, "-oForwardAgent no");
	addargs(&args, "-oClearAllForwardings yes");
	ll = SYSLOG_LEVEL_INFO;

	while ((ch = getopt(argc, argv, "1hvCo:s:S:b:B:F:P:R:")) != -1) {
		switch (ch) {
		case 'C':
			addargs(&args, "-C");
			break;
		case 'v':
			if (debug_level < 3) {
				addargs(&args, "-v");
				ll = SYSLOG_LEVEL_DEBUG1 + debug_level;
			}
			debug_level++;
			break;
		case 'F':
		case 'o':
			addargs(&args, "-%c%s", ch, optarg);
			break;
		case '1':
			sshver = 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 (batchmode)
				fatal("Batch file already specified.");

			/* Allow "-" as stdin */
			if (strcmp(optarg, "-") != 0 && 
			   (infile = fopen(optarg, "r")) == NULL)
				fatal("%s (%s).", strerror(errno), optarg);
			showprogress = 0;
			batchmode = 1;
			break;
		case 'P':
			sftp_direct = optarg;
			break;
		case 'B':
			copy_buffer_len = strtol(optarg, &cp, 10);
			if (copy_buffer_len == 0 || *cp != '\0')
				fatal("Invalid buffer size \"%s\"", optarg);
			break;
		case 'R':
			num_requests = strtol(optarg, &cp, 10);
			if (num_requests == 0 || *cp != '\0')
				fatal("Invalid number of requests \"%s\"",
				    optarg);
			break;
		case 'h':
		default:
			usage();
		}
	}

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

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

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

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

		if ((host = strrchr(userhost, '@')) == NULL)
			host = userhost;
		else {
			*host++ = '\0';
			if (!userhost[0]) {
				fprintf(stderr, "Missing username\n");
				usage();
			}
			addargs(&args, "-l%s",userhost);
		}

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

		addargs(&args, "-oProtocol %d", sshver);

		/* no subsystem if the server-spec contains a '/' */
		if (sftp_server == NULL || strchr(sftp_server, '/') == NULL)
			addargs(&args, "-s");

		addargs(&args, "%s", host);
		addargs(&args, "%s", (sftp_server != NULL ?
		    sftp_server : "sftp"));
		args.list[0] = ssh_program;

		if (!batchmode)
			fprintf(stderr, "Connecting to %s...\n", host);
		connect_to_server(ssh_program, args.list, &in, &out);
	} else {
		args.list = NULL;
		addargs(&args, "sftp-server");

		if (!batchmode)
			fprintf(stderr, "Attaching to %s...\n", sftp_direct);
		connect_to_server(sftp_direct, args.list, &in, &out);
	}

	err = interactive_loop(in, out, file1, file2);

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

	close(in);
	close(out);
	if (batchmode)
		fclose(infile);

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

	exit(err == 0 ? 0 : 1);
}
Commit	Line	Data
61e96248	1	/*
a96fd7c2	2	* Copyright (c) 2001,2002 Damien Miller. All rights reserved.
61e96248	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
a8b64bb8	27	RCSID("$OpenBSD: sftp.c,v 1.39 2004/01/13 09:25:05 djm Exp $");
61e96248	28
	29	#include "buffer.h"
	30	#include "xmalloc.h"
	31	#include "log.h"
	32	#include "pathnames.h"
1fcde3fe	33	#include "misc.h"
61e96248	34
	35	#include "sftp.h"
	36	#include "sftp-common.h"
	37	#include "sftp-client.h"
	38	#include "sftp-int.h"
	39
5152d46f	40	#ifdef HAVE___PROGNAME
	41	extern char *__progname;
	42	#else
	43	char *__progname;
	44	#endif
	45
a8b64bb8	46	FILE* infile = stdin;
a8b64bb8	47	int batchmode = 0;
375f867e	48	size_t copy_buffer_len = 32768;
c25d3df7	49	size_t num_requests = 16;
1b558925	50	static pid_t sshpid = -1;
0426a3b4	51
b65c3807	52	extern int showprogress;
b65c3807	53
396c147e	54	static void
1b558925	55	killchild(int signo)
	56	{
	57	if (sshpid > 1)
	58	kill(sshpid, signo);
	59
	60	_exit(1);
	61	}
	62
	63	static void
	64	connect_to_server(char path, char args, int in, int *out)
61e96248	65	{
61e96248	66	int c_in, c_out;
9906a836	67
61e96248	68	#ifdef USE_PIPES
61e96248	69	int pin[2], pout[2];
9906a836	70
61e96248	71	if ((pipe(pin) == -1) \|\| (pipe(pout) == -1))
	72	fatal("pipe: %s", strerror(errno));
	73	*in = pin[0];
	74	*out = pout[1];
	75	c_in = pout[0];
	76	c_out = pin[1];
	77	#else /* USE_PIPES */
	78	int inout[2];
9906a836	79
61e96248	80	if (socketpair(AF_UNIX, SOCK_STREAM, 0, inout) == -1)
	81	fatal("socketpair: %s", strerror(errno));
	82	in = out = inout[0];
	83	c_in = c_out = inout[1];
	84	#endif /* USE_PIPES */
	85
1b558925	86	if ((sshpid = fork()) == -1)
61e96248	87	fatal("fork: %s", strerror(errno));
1b558925	88	else if (sshpid == 0) {
61e96248	89	if ((dup2(c_in, STDIN_FILENO) == -1) \|\|
	90	(dup2(c_out, STDOUT_FILENO) == -1)) {
	91	fprintf(stderr, "dup2: %s\n", strerror(errno));
	92	exit(1);
	93	}
	94	close(*in);
	95	close(*out);
	96	close(c_in);
	97	close(c_out);
a96fd7c2	98	execv(path, args);
a96fd7c2	99	fprintf(stderr, "exec: %s: %s\n", path, strerror(errno));
61e96248	100	exit(1);
	101	}
	102
1b558925	103	signal(SIGTERM, killchild);
	104	signal(SIGINT, killchild);
	105	signal(SIGHUP, killchild);
61e96248	106	close(c_in);
	107	close(c_out);
	108	}
	109
396c147e	110	static void
61e96248	111	usage(void)
61e96248	112	{
22be05a5	113	extern char *__progname;
762715ce	114
f1278af7	115	fprintf(stderr,
433e60ac	116	"usage: %s [-1Cv] [-B buffer_size] [-b batchfile] [-F ssh_config]\n"
	117	" [-o ssh_option] [-P sftp_server_path] [-R num_requests]\n"
	118	" [-S program] [-s subsystem \| sftp_server] host\n"
	119	" %s [[user@]host[:file [file]]]\n"
	120	" %s [[user@]host[:dir[/]]]\n"
	121	" %s -b batchfile [user@]host\n", __progname, __progname, __progname, __progname);
61e96248	122	exit(1);
	123	}
	124
2b87da3b	125	int
61e96248	126	main(int argc, char **argv)
61e96248	127	{
9a36208d	128	int in, out, ch, err;
edeeab1e	129	char host, userhost, cp, file2;
8a624ebf	130	int debug_level = 0, sshver = 2;
8a624ebf	131	char file1 = NULL, sftp_server = NULL;
a96fd7c2	132	char ssh_program = _PATH_SSH_PROGRAM, sftp_direct = NULL;
8a624ebf	133	LogLevel ll = SYSLOG_LEVEL_INFO;
8a624ebf	134	arglist args;
0426a3b4	135	extern int optind;
0426a3b4	136	extern char *optarg;
61e96248	137
fda04d7d	138	__progname = ssh_get_progname(argv[0]);
8a624ebf	139	args.list = NULL;
184eed6a	140	addargs(&args, "ssh"); /* overwritten with ssh_program */
8a624ebf	141	addargs(&args, "-oForwardX11 no");
8a624ebf	142	addargs(&args, "-oForwardAgent no");
e1c5bfaf	143	addargs(&args, "-oClearAllForwardings yes");
8a624ebf	144	ll = SYSLOG_LEVEL_INFO;
0426a3b4	145
c25d3df7	146	while ((ch = getopt(argc, argv, "1hvCo:s:S:b:B:F:P:R:")) != -1) {
0426a3b4	147	switch (ch) {
0426a3b4	148	case 'C':
8a624ebf	149	addargs(&args, "-C");
0426a3b4	150	break;
0426a3b4	151	case 'v':
8a624ebf	152	if (debug_level < 3) {
	153	addargs(&args, "-v");
	154	ll = SYSLOG_LEVEL_DEBUG1 + debug_level;
	155	}
	156	debug_level++;
0426a3b4	157	break;
f1278af7	158	case 'F':
0426a3b4	159	case 'o':
f1278af7	160	addargs(&args, "-%c%s", ch, optarg);
0426a3b4	161	break;
0426a3b4	162	case '1':
8a624ebf	163	sshver = 1;
0426a3b4	164	if (sftp_server == NULL)
	165	sftp_server = _PATH_SFTP_SERVER;
	166	break;
	167	case 's':
	168	sftp_server = optarg;
	169	break;
	170	case 'S':
	171	ssh_program = optarg;
	172	break;
a5ec8a3d	173	case 'b':
a8b64bb8	174	if (batchmode)
	175	fatal("Batch file already specified.");
	176
	177	/* Allow "-" as stdin */
	178	if (strcmp(optarg, "-") != 0 &&
	179	(infile = fopen(optarg, "r")) == NULL)
	180	fatal("%s (%s).", strerror(errno), optarg);
b65c3807	181	showprogress = 0;
a8b64bb8	182	batchmode = 1;
a5ec8a3d	183	break;
a96fd7c2	184	case 'P':
	185	sftp_direct = optarg;
	186	break;
375f867e	187	case 'B':
	188	copy_buffer_len = strtol(optarg, &cp, 10);
	189	if (copy_buffer_len == 0 \|\| *cp != '\0')
	190	fatal("Invalid buffer size \"%s\"", optarg);
	191	break;
c25d3df7	192	case 'R':
	193	num_requests = strtol(optarg, &cp, 10);
	194	if (num_requests == 0 \|\| *cp != '\0')
762715ce	195	fatal("Invalid number of requests \"%s\"",
c25d3df7	196	optarg);
c25d3df7	197	break;
0426a3b4	198	case 'h':
0426a3b4	199	default:
61e96248	200	usage();
	201	}
	202	}
	203
b69145c2	204	log_init(argv[0], ll, SYSLOG_FACILITY_USER, 1);
b69145c2	205
a96fd7c2	206	if (sftp_direct == NULL) {
	207	if (optind == argc \|\| argc > (optind + 2))
	208	usage();
61e96248	209
a96fd7c2	210	userhost = xstrdup(argv[optind]);
a96fd7c2	211	file2 = argv[optind+1];
edeeab1e	212
a96fd7c2	213	if ((cp = colon(userhost)) != NULL) {
	214	*cp++ = '\0';
	215	file1 = cp;
	216	}
0426a3b4	217
15748b4d	218	if ((host = strrchr(userhost, '@')) == NULL)
a96fd7c2	219	host = userhost;
	220	else {
	221	*host++ = '\0';
	222	if (!userhost[0]) {
	223	fprintf(stderr, "Missing username\n");
	224	usage();
	225	}
	226	addargs(&args, "-l%s",userhost);
61e96248	227	}
61e96248	228
a96fd7c2	229	host = cleanhostname(host);
	230	if (!*host) {
	231	fprintf(stderr, "Missing hostname\n");
	232	usage();
	233	}
61e96248	234
a96fd7c2	235	addargs(&args, "-oProtocol %d", sshver);
8a624ebf	236
a96fd7c2	237	/* no subsystem if the server-spec contains a '/' */
	238	if (sftp_server == NULL \|\| strchr(sftp_server, '/') == NULL)
	239	addargs(&args, "-s");
61e96248	240
a96fd7c2	241	addargs(&args, "%s", host);
762715ce	242	addargs(&args, "%s", (sftp_server != NULL ?
a96fd7c2	243	sftp_server : "sftp"));
a96fd7c2	244	args.list[0] = ssh_program;
61e96248	245
a8b64bb8	246	if (!batchmode)
a8b64bb8	247	fprintf(stderr, "Connecting to %s...\n", host);
1b558925	248	connect_to_server(ssh_program, args.list, &in, &out);
a96fd7c2	249	} else {
	250	args.list = NULL;
	251	addargs(&args, "sftp-server");
61e96248	252
a8b64bb8	253	if (!batchmode)
a8b64bb8	254	fprintf(stderr, "Attaching to %s...\n", sftp_direct);
1b558925	255	connect_to_server(sftp_direct, args.list, &in, &out);
a96fd7c2	256	}
61e96248	257
9a36208d	258	err = interactive_loop(in, out, file1, file2);
61e96248	259
51fb577a	260	#if !defined(USE_PIPES)
875ec275	261	shutdown(in, SHUT_RDWR);
875ec275	262	shutdown(out, SHUT_RDWR);
51fb577a	263	#endif
51fb577a	264
61e96248	265	close(in);
61e96248	266	close(out);
a8b64bb8	267	if (batchmode)
a5ec8a3d	268	fclose(infile);
61e96248	269
8c38e88b	270	while (waitpid(sshpid, NULL, 0) == -1)
	271	if (errno != EINTR)
	272	fatal("Couldn't wait for ssh process: %s",
	273	strerror(errno));
61e96248	274
9a36208d	275	exit(err == 0 ? 0 : 1);
61e96248	276	}