blob: c4055b91ee47f7c2d5f19756354f4240f212d3d0 [file] [log] [blame]
Damien Miller33804262001-02-04 23:20:18 +11001/*
Damien Millerd14ee1e2002-02-05 12:27:31 +11002 * Copyright (c) 2001,2002 Damien Miller. All rights reserved.
Damien Miller33804262001-02-04 23:20:18 +11003 *
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
Ben Lindstrom8d631e82002-08-01 01:25:00 +000027RCSID("$OpenBSD: sftp.c,v 1.31 2002/07/25 01:16:59 mouring Exp $");
Damien Miller33804262001-02-04 23:20:18 +110028
Damien Miller33804262001-02-04 23:20:18 +110029/* XXX: short-form remote directory listings (like 'ls -C') */
30
31#include "buffer.h"
32#include "xmalloc.h"
33#include "log.h"
34#include "pathnames.h"
Ben Lindstrom4529b702001-05-03 23:39:53 +000035#include "misc.h"
Damien Miller33804262001-02-04 23:20:18 +110036
37#include "sftp.h"
38#include "sftp-common.h"
39#include "sftp-client.h"
40#include "sftp-int.h"
41
Kevin Steves12888d12001-03-05 19:50:57 +000042#ifdef HAVE___PROGNAME
43extern char *__progname;
44#else
45char *__progname;
46#endif
47
Ben Lindstrom562c26b2001-03-07 01:26:48 +000048FILE* infile;
Damien Miller8829d362002-02-08 22:04:05 +110049size_t copy_buffer_len = 32768;
Damien Miller16a13332002-02-13 14:03:56 +110050size_t num_requests = 16;
Damien Millerd7686fd2001-02-10 00:40:03 +110051
Ben Lindstrombba81212001-06-25 05:01:22 +000052static void
Damien Millerd14ee1e2002-02-05 12:27:31 +110053connect_to_server(char *path, char **args, int *in, int *out, pid_t *sshpid)
Damien Miller33804262001-02-04 23:20:18 +110054{
55 int c_in, c_out;
Ben Lindstromb1f483f2002-06-23 21:27:18 +000056
Damien Miller33804262001-02-04 23:20:18 +110057#ifdef USE_PIPES
58 int pin[2], pout[2];
Ben Lindstromb1f483f2002-06-23 21:27:18 +000059
Damien Miller33804262001-02-04 23:20:18 +110060 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];
Ben Lindstromb1f483f2002-06-23 21:27:18 +000068
Damien Miller33804262001-02-04 23:20:18 +110069 if (socketpair(AF_UNIX, SOCK_STREAM, 0, inout) == -1)
70 fatal("socketpair: %s", strerror(errno));
71 *in = *out = inout[0];
72 c_in = c_out = inout[1];
73#endif /* USE_PIPES */
74
75 if ((*sshpid = fork()) == -1)
76 fatal("fork: %s", strerror(errno));
77 else if (*sshpid == 0) {
78 if ((dup2(c_in, STDIN_FILENO) == -1) ||
79 (dup2(c_out, STDOUT_FILENO) == -1)) {
80 fprintf(stderr, "dup2: %s\n", strerror(errno));
81 exit(1);
82 }
83 close(*in);
84 close(*out);
85 close(c_in);
86 close(c_out);
Damien Millerd14ee1e2002-02-05 12:27:31 +110087 execv(path, args);
88 fprintf(stderr, "exec: %s: %s\n", path, strerror(errno));
Damien Miller33804262001-02-04 23:20:18 +110089 exit(1);
90 }
91
92 close(c_in);
93 close(c_out);
94}
95
Ben Lindstrombba81212001-06-25 05:01:22 +000096static void
Damien Miller33804262001-02-04 23:20:18 +110097usage(void)
98{
Damien Miller025e01c2002-02-08 22:06:29 +110099 extern char *__progname;
Ben Lindstrom6328ab32002-03-22 02:54:23 +0000100
Ben Lindstrom1e243242001-09-18 05:38:44 +0000101 fprintf(stderr,
Damien Miller025e01c2002-02-08 22:06:29 +1100102 "usage: %s [-vC1] [-b batchfile] [-o option] [-s subsystem|path] [-B buffer_size]\n"
103 " [-F config] [-P direct server path] [-S program]\n"
104 " [user@]host[:file [file]]\n", __progname);
Damien Miller33804262001-02-04 23:20:18 +1100105 exit(1);
106}
107
Kevin Stevesef4eea92001-02-05 12:42:17 +0000108int
Damien Miller33804262001-02-04 23:20:18 +1100109main(int argc, char **argv)
110{
Ben Lindstrom387c4722001-05-08 20:27:25 +0000111 int in, out, ch;
Damien Miller33804262001-02-04 23:20:18 +1100112 pid_t sshpid;
Ben Lindstrom63667f62001-04-13 00:00:14 +0000113 char *host, *userhost, *cp, *file2;
Ben Lindstrom387c4722001-05-08 20:27:25 +0000114 int debug_level = 0, sshver = 2;
115 char *file1 = NULL, *sftp_server = NULL;
Damien Millerd14ee1e2002-02-05 12:27:31 +1100116 char *ssh_program = _PATH_SSH_PROGRAM, *sftp_direct = NULL;
Ben Lindstrom387c4722001-05-08 20:27:25 +0000117 LogLevel ll = SYSLOG_LEVEL_INFO;
118 arglist args;
Damien Millerd7686fd2001-02-10 00:40:03 +1100119 extern int optind;
120 extern char *optarg;
Damien Miller33804262001-02-04 23:20:18 +1100121
Kevin Steves12888d12001-03-05 19:50:57 +0000122 __progname = get_progname(argv[0]);
Ben Lindstrom387c4722001-05-08 20:27:25 +0000123 args.list = NULL;
Damien Miller9f0f5c62001-12-21 14:45:46 +1100124 addargs(&args, "ssh"); /* overwritten with ssh_program */
Ben Lindstrom387c4722001-05-08 20:27:25 +0000125 addargs(&args, "-oForwardX11 no");
126 addargs(&args, "-oForwardAgent no");
Ben Lindstrom2b7a0e92001-09-20 00:57:55 +0000127 addargs(&args, "-oClearAllForwardings yes");
Ben Lindstrom387c4722001-05-08 20:27:25 +0000128 ll = SYSLOG_LEVEL_INFO;
129 infile = stdin; /* Read from STDIN unless changed by -b */
Damien Millerd7686fd2001-02-10 00:40:03 +1100130
Damien Miller16a13332002-02-13 14:03:56 +1100131 while ((ch = getopt(argc, argv, "1hvCo:s:S:b:B:F:P:R:")) != -1) {
Damien Millerd7686fd2001-02-10 00:40:03 +1100132 switch (ch) {
133 case 'C':
Ben Lindstrom387c4722001-05-08 20:27:25 +0000134 addargs(&args, "-C");
Damien Millerd7686fd2001-02-10 00:40:03 +1100135 break;
136 case 'v':
Ben Lindstrom387c4722001-05-08 20:27:25 +0000137 if (debug_level < 3) {
138 addargs(&args, "-v");
139 ll = SYSLOG_LEVEL_DEBUG1 + debug_level;
140 }
141 debug_level++;
Damien Millerd7686fd2001-02-10 00:40:03 +1100142 break;
Ben Lindstrom1e243242001-09-18 05:38:44 +0000143 case 'F':
Damien Millerd7686fd2001-02-10 00:40:03 +1100144 case 'o':
Ben Lindstrom1e243242001-09-18 05:38:44 +0000145 addargs(&args, "-%c%s", ch, optarg);
Damien Millerd7686fd2001-02-10 00:40:03 +1100146 break;
147 case '1':
Ben Lindstrom387c4722001-05-08 20:27:25 +0000148 sshver = 1;
Damien Millerd7686fd2001-02-10 00:40:03 +1100149 if (sftp_server == NULL)
150 sftp_server = _PATH_SFTP_SERVER;
151 break;
152 case 's':
153 sftp_server = optarg;
154 break;
155 case 'S':
156 ssh_program = optarg;
157 break;
Ben Lindstrom562c26b2001-03-07 01:26:48 +0000158 case 'b':
159 if (infile == stdin) {
160 infile = fopen(optarg, "r");
Ben Lindstroma3700052001-04-05 23:26:32 +0000161 if (infile == NULL)
Ben Lindstrom562c26b2001-03-07 01:26:48 +0000162 fatal("%s (%s).", strerror(errno), optarg);
Ben Lindstroma3700052001-04-05 23:26:32 +0000163 } else
Ben Lindstrom562c26b2001-03-07 01:26:48 +0000164 fatal("Filename already specified.");
165 break;
Damien Millerd14ee1e2002-02-05 12:27:31 +1100166 case 'P':
167 sftp_direct = optarg;
168 break;
Damien Miller8829d362002-02-08 22:04:05 +1100169 case 'B':
170 copy_buffer_len = strtol(optarg, &cp, 10);
171 if (copy_buffer_len == 0 || *cp != '\0')
172 fatal("Invalid buffer size \"%s\"", optarg);
173 break;
Damien Miller16a13332002-02-13 14:03:56 +1100174 case 'R':
175 num_requests = strtol(optarg, &cp, 10);
176 if (num_requests == 0 || *cp != '\0')
Ben Lindstrom6328ab32002-03-22 02:54:23 +0000177 fatal("Invalid number of requests \"%s\"",
Damien Miller16a13332002-02-13 14:03:56 +1100178 optarg);
179 break;
Damien Millerd7686fd2001-02-10 00:40:03 +1100180 case 'h':
181 default:
Damien Miller33804262001-02-04 23:20:18 +1100182 usage();
183 }
184 }
185
Ben Lindstrom2f3d52a2002-04-02 21:06:18 +0000186 log_init(argv[0], ll, SYSLOG_FACILITY_USER, 1);
187
Damien Millerd14ee1e2002-02-05 12:27:31 +1100188 if (sftp_direct == NULL) {
189 if (optind == argc || argc > (optind + 2))
190 usage();
Damien Miller33804262001-02-04 23:20:18 +1100191
Damien Millerd14ee1e2002-02-05 12:27:31 +1100192 userhost = xstrdup(argv[optind]);
193 file2 = argv[optind+1];
Ben Lindstrom63667f62001-04-13 00:00:14 +0000194
Damien Millerd14ee1e2002-02-05 12:27:31 +1100195 if ((cp = colon(userhost)) != NULL) {
196 *cp++ = '\0';
197 file1 = cp;
198 }
Damien Millerd7686fd2001-02-10 00:40:03 +1100199
Damien Millerd14ee1e2002-02-05 12:27:31 +1100200 if ((host = strchr(userhost, '@')) == NULL)
201 host = userhost;
202 else {
203 *host++ = '\0';
204 if (!userhost[0]) {
205 fprintf(stderr, "Missing username\n");
206 usage();
207 }
208 addargs(&args, "-l%s",userhost);
209 }
210
211 host = cleanhostname(host);
212 if (!*host) {
213 fprintf(stderr, "Missing hostname\n");
Damien Miller33804262001-02-04 23:20:18 +1100214 usage();
215 }
Damien Millerd14ee1e2002-02-05 12:27:31 +1100216
Damien Millerd14ee1e2002-02-05 12:27:31 +1100217 addargs(&args, "-oProtocol %d", sshver);
218
219 /* no subsystem if the server-spec contains a '/' */
220 if (sftp_server == NULL || strchr(sftp_server, '/') == NULL)
221 addargs(&args, "-s");
222
223 addargs(&args, "%s", host);
Ben Lindstrom6328ab32002-03-22 02:54:23 +0000224 addargs(&args, "%s", (sftp_server != NULL ?
Damien Millerd14ee1e2002-02-05 12:27:31 +1100225 sftp_server : "sftp"));
226 args.list[0] = ssh_program;
227
228 fprintf(stderr, "Connecting to %s...\n", host);
Ben Lindstrom6328ab32002-03-22 02:54:23 +0000229 connect_to_server(ssh_program, args.list, &in, &out,
Damien Millerd14ee1e2002-02-05 12:27:31 +1100230 &sshpid);
231 } else {
232 args.list = NULL;
233 addargs(&args, "sftp-server");
234
235 fprintf(stderr, "Attaching to %s...\n", sftp_direct);
Ben Lindstrom6328ab32002-03-22 02:54:23 +0000236 connect_to_server(sftp_direct, args.list, &in, &out,
Damien Millerd14ee1e2002-02-05 12:27:31 +1100237 &sshpid);
Damien Miller33804262001-02-04 23:20:18 +1100238 }
239
Ben Lindstrom63667f62001-04-13 00:00:14 +0000240 interactive_loop(in, out, file1, file2);
Damien Miller33804262001-02-04 23:20:18 +1100241
Ben Lindstrom10b9bf92001-02-26 20:04:45 +0000242#if !defined(USE_PIPES)
Ben Lindstrom5a6abda2002-06-09 19:41:48 +0000243 shutdown(in, SHUT_RDWR);
244 shutdown(out, SHUT_RDWR);
Ben Lindstrom10b9bf92001-02-26 20:04:45 +0000245#endif
246
Damien Miller33804262001-02-04 23:20:18 +1100247 close(in);
248 close(out);
Ben Lindstrom562c26b2001-03-07 01:26:48 +0000249 if (infile != stdin)
250 fclose(infile);
Damien Miller33804262001-02-04 23:20:18 +1100251
Ben Lindstrom47fd8112002-04-02 20:48:19 +0000252 while (waitpid(sshpid, NULL, 0) == -1)
253 if (errno != EINTR)
254 fatal("Couldn't wait for ssh process: %s",
255 strerror(errno));
Damien Miller33804262001-02-04 23:20:18 +1100256
257 exit(0);
258}