279 lines
6.7 KiB
C
279 lines
6.7 KiB
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.41 2004/01/27 10:08:10 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;
|
|
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;
|
|
infile = stdin;
|
|
|
|
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 ((host = strrchr(userhost, '@')) == NULL)
|
|
host = userhost;
|
|
else {
|
|
*host++ = '\0';
|
|
if (!userhost[0]) {
|
|
fprintf(stderr, "Missing username\n");
|
|
usage();
|
|
}
|
|
addargs(&args, "-l%s",userhost);
|
|
}
|
|
|
|
if ((cp = colon(host)) != NULL) {
|
|
*cp++ = '\0';
|
|
file1 = cp;
|
|
}
|
|
|
|
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);
|
|
}
|