[gssapi-openssh.git] / openssh / 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.21 2001/09/19 19:24:19 stevesk Exp $");

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

#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

char *ssh_program = _PATH_SSH_PROGRAM;
FILE* infile;

static 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);
}

static void
usage(void)
{
	fprintf(stderr,
	    "usage: sftp [-1Cv] [-b batchfile] [-F config] [-o option] [-s subsystem|path]\n"
	    "            [-S program] [user@]host[:file [file]]\n");
	exit(1);
}

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

	__progname = get_progname(argv[0]);
	args.list = NULL;
	addargs(&args, "ssh");         /* overwritten with ssh_program */
	addargs(&args, "-oFallBackToRsh no");
	addargs(&args, "-oForwardX11 no");
	addargs(&args, "-oForwardAgent no");
	addargs(&args, "-oClearAllForwardings yes");
	ll = SYSLOG_LEVEL_INFO;
	infile = stdin;		/* Read from STDIN unless changed by -b */

	while ((ch = getopt(argc, argv, "1hvCo:s:S:b:F:")) != -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 (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 = colon(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();
		}
		addargs(&args, "-l%s",userhost);
	}

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

	log_init(argv[0], ll, SYSLOG_FACILITY_USER, 1);
	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;

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

	connect_to_server(args.list, &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
3c0ef626	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
	27	RCSID("$OpenBSD: sftp.c,v 1.21 2001/09/19 19:24:19 stevesk Exp $");
	28
	29	/* XXX: commandline mode */
	30	/* XXX: short-form remote directory listings (like 'ls -C') */
	31
	32	#include "buffer.h"
	33	#include "xmalloc.h"
	34	#include "log.h"
	35	#include "pathnames.h"
	36	#include "misc.h"
	37
	38	#include "sftp.h"
	39	#include "sftp-common.h"
	40	#include "sftp-client.h"
	41	#include "sftp-int.h"
	42
	43	#ifdef HAVE___PROGNAME
	44	extern char *__progname;
	45	#else
	46	char *__progname;
	47	#endif
	48
	49	char *ssh_program = _PATH_SSH_PROGRAM;
	50	FILE* infile;
	51
	52	static void
	53	connect_to_server(char *args, int in, int out, pid_t sshpid)
	54	{
	55	int c_in, c_out;
	56	#ifdef USE_PIPES
	57	int pin[2], pout[2];
	58	if ((pipe(pin) == -1) \|\| (pipe(pout) == -1))
	59	fatal("pipe: %s", strerror(errno));
	60	*in = pin[0];
	61	*out = pout[1];
	62	c_in = pout[0];
	63	c_out = pin[1];
	64	#else /* USE_PIPES */
65	int inout[2];
66	if (socketpair(AF_UNIX, SOCK_STREAM, 0, inout) == -1)
67	fatal("socketpair: %s", strerror(errno));
68	in = out = inout[0];
69	c_in = c_out = inout[1];
70	#endif /* USE_PIPES */
71
72	if ((*sshpid = fork()) == -1)
73	fatal("fork: %s", strerror(errno));
74	else if (*sshpid == 0) {
75	if ((dup2(c_in, STDIN_FILENO) == -1) \|\|
76	(dup2(c_out, STDOUT_FILENO) == -1)) {
77	fprintf(stderr, "dup2: %s\n", strerror(errno));
78	exit(1);
79	}
80	close(*in);
81	close(*out);
82	close(c_in);
83	close(c_out);
84	execv(ssh_program, args);
85	fprintf(stderr, "exec: %s: %s\n", ssh_program, strerror(errno));
86	exit(1);
87	}
88
89	close(c_in);
90	close(c_out);
91	}
92
93	static void
94	usage(void)
95	{
96	fprintf(stderr,
97	"usage: sftp [-1Cv] [-b batchfile] [-F config] [-o option] [-s subsystem\|path]\n"
98	" [-S program] [user@]host[:file [file]]\n");
99	exit(1);
100	}
101
102	int
103	main(int argc, char **argv)
104	{
105	int in, out, ch;
106	pid_t sshpid;
107	char host, userhost, cp, file2;
108	int debug_level = 0, sshver = 2;
109	char file1 = NULL, sftp_server = NULL;
110	LogLevel ll = SYSLOG_LEVEL_INFO;
111	arglist args;
112	extern int optind;
113	extern char *optarg;
114
115	__progname = get_progname(argv[0]);
116	args.list = NULL;
117	addargs(&args, "ssh"); /* overwritten with ssh_program */
118	addargs(&args, "-oFallBackToRsh no");
119	addargs(&args, "-oForwardX11 no");
120	addargs(&args, "-oForwardAgent no");
121	addargs(&args, "-oClearAllForwardings yes");
122	ll = SYSLOG_LEVEL_INFO;
123	infile = stdin; /* Read from STDIN unless changed by -b */
124
125	while ((ch = getopt(argc, argv, "1hvCo:s:S:b:F:")) != -1) {
126	switch (ch) {
127	case 'C':
128	addargs(&args, "-C");
129	break;
130	case 'v':
131	if (debug_level < 3) {
132	addargs(&args, "-v");
133	ll = SYSLOG_LEVEL_DEBUG1 + debug_level;
134	}
135	debug_level++;
136	break;
137	case 'F':
138	case 'o':
139	addargs(&args, "-%c%s", ch, optarg);
140	break;
141	case '1':
142	sshver = 1;
143	if (sftp_server == NULL)
144	sftp_server = _PATH_SFTP_SERVER;
145	break;
146	case 's':
147	sftp_server = optarg;
148	break;
149	case 'S':
150	ssh_program = optarg;
151	break;
152	case 'b':
153	if (infile == stdin) {
154	infile = fopen(optarg, "r");
155	if (infile == NULL)
156	fatal("%s (%s).", strerror(errno), optarg);
157	} else
158	fatal("Filename already specified.");
159	break;
160	case 'h':
161	default:
162	usage();
163	}
164	}
165
166	if (optind == argc \|\| argc > (optind + 2))
167	usage();
168
169	userhost = xstrdup(argv[optind]);
170	file2 = argv[optind+1];
171
172	if ((cp = colon(userhost)) != NULL) {
173	*cp++ = '\0';
174	file1 = cp;
175	}
176
177	if ((host = strchr(userhost, '@')) == NULL)
178	host = userhost;
179	else {
180	*host++ = '\0';
181	if (!userhost[0]) {
182	fprintf(stderr, "Missing username\n");
183	usage();
184	}
185	addargs(&args, "-l%s",userhost);
186	}
187
188	host = cleanhostname(host);
189	if (!*host) {
190	fprintf(stderr, "Missing hostname\n");
191	usage();
192	}
193
194	log_init(argv[0], ll, SYSLOG_FACILITY_USER, 1);
195	addargs(&args, "-oProtocol %d", sshver);
196
197	/* no subsystem if the server-spec contains a '/' */
198	if (sftp_server == NULL \|\| strchr(sftp_server, '/') == NULL)
199	addargs(&args, "-s");
200
201	addargs(&args, "%s", host);
202	addargs(&args, "%s", (sftp_server != NULL ? sftp_server : "sftp"));
203	args.list[0] = ssh_program;
204
205	fprintf(stderr, "Connecting to %s...\n", host);
206
207	connect_to_server(args.list, &in, &out, &sshpid);
208
209	interactive_loop(in, out, file1, file2);
210
211	#if !defined(USE_PIPES)
212	shutdown(in, SHUT_RDWR);
213	shutdown(out, SHUT_RDWR);
214	#endif
215
216	close(in);
217	close(out);
218	if (infile != stdin)
219	fclose(infile);
220
221	if (waitpid(sshpid, NULL, 0) == -1)
222	fatal("Couldn't wait for ssh process: %s", strerror(errno));
223
224	exit(0);
225	}