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