sftp.c

   1 /*
   2  * Copyright (c) 2001,2002 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.41 2004/01/27 10:08:10 djm Exp $");
  28
  29 #include "buffer.h"
  30 #include "xmalloc.h"
  31 #include "log.h"
  32 #include "pathnames.h"
  33 #include "misc.h"
  34
  35 #include "sftp.h"
  36 #include "sftp-common.h"
  37 #include "sftp-client.h"
  38 #include "sftp-int.h"
  39
  40 #ifdef HAVE___PROGNAME
  41 extern char *__progname;
  42 #else
  43 char *__progname;
  44 #endif
  45
  46 FILE* infile;
  47 int batchmode = 0;
  48 size_t copy_buffer_len = 32768;
  49 size_t num_requests = 16;
  50 static pid_t sshpid = -1;
  51
  52 extern int showprogress;
  53
  54 static void
  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)
  65 {
  66         int c_in, c_out;
  67
  68 #ifdef USE_PIPES
  69         int pin[2], pout[2];
  70
  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];
  79
  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
  86         if ((sshpid = fork()) == -1)
  87                 fatal("fork: %s", strerror(errno));
  88         else if (sshpid == 0) {
  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);
  98                 execv(path, args);
  99                 fprintf(stderr, "exec: %s: %s\n", path, strerror(errno));
 100                 exit(1);
 101         }
 102
 103         signal(SIGTERM, killchild);
 104         signal(SIGINT, killchild);
 105         signal(SIGHUP, killchild);
 106         close(c_in);
 107         close(c_out);
 108 }
 109
 110 static void
 111 usage(void)
 112 {
 113         extern char *__progname;
 114
 115         fprintf(stderr,
 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);
 122         exit(1);
 123 }
 124
 125 int
 126 main(int argc, char **argv)
 127 {
 128         int in, out, ch, err;
 129         char *host, *userhost, *cp, *file2;
 130         int debug_level = 0, sshver = 2;
 131         char *file1 = NULL, *sftp_server = NULL;
 132         char *ssh_program = _PATH_SSH_PROGRAM, *sftp_direct = NULL;
 133         LogLevel ll = SYSLOG_LEVEL_INFO;
 134         arglist args;
 135         extern int optind;
 136         extern char *optarg;
 137
 138         __progname = ssh_get_progname(argv[0]);
 139         args.list = NULL;
 140         addargs(&args, "ssh");          /* overwritten with ssh_program */
 141         addargs(&args, "-oForwardX11 no");
 142         addargs(&args, "-oForwardAgent no");
 143         addargs(&args, "-oClearAllForwardings yes");
 144
 145         ll = SYSLOG_LEVEL_INFO;
 146         infile = stdin;
 147
 148         while ((ch = getopt(argc, argv, "1hvCo:s:S:b:B:F:P:R:")) != -1) {
 149                 switch (ch) {
 150                 case 'C':
 151                         addargs(&args, "-C");
 152                         break;
 153                 case 'v':
 154                         if (debug_level < 3) {
 155                                 addargs(&args, "-v");
 156                                 ll = SYSLOG_LEVEL_DEBUG1 + debug_level;
 157                         }
 158                         debug_level++;
 159                         break;
 160                 case 'F':
 161                 case 'o':
 162                         addargs(&args, "-%c%s", ch, optarg);
 163                         break;
 164                 case '1':
 165                         sshver = 1;
 166                         if (sftp_server == NULL)
 167                                 sftp_server = _PATH_SFTP_SERVER;
 168                         break;
 169                 case 's':
 170                         sftp_server = optarg;
 171                         break;
 172                 case 'S':
 173                         ssh_program = optarg;
 174                         break;
 175                 case 'b':
 176                         if (batchmode)
 177                                 fatal("Batch file already specified.");
 178
 179                         /* Allow "-" as stdin */
 180                         if (strcmp(optarg, "-") != 0 &&
 181                            (infile = fopen(optarg, "r")) == NULL)
 182                                 fatal("%s (%s).", strerror(errno), optarg);
 183                         showprogress = 0;
 184                         batchmode = 1;
 185                         break;
 186                 case 'P':
 187                         sftp_direct = optarg;
 188                         break;
 189                 case 'B':
 190                         copy_buffer_len = strtol(optarg, &cp, 10);
 191                         if (copy_buffer_len == 0 || *cp != '\0')
 192                                 fatal("Invalid buffer size \"%s\"", optarg);
 193                         break;
 194                 case 'R':
 195                         num_requests = strtol(optarg, &cp, 10);
 196                         if (num_requests == 0 || *cp != '\0')
 197                                 fatal("Invalid number of requests \"%s\"",
 198                                     optarg);
 199                         break;
 200                 case 'h':
 201                 default:
 202                         usage();
 203                 }
 204         }
 205
 206         log_init(argv[0], ll, SYSLOG_FACILITY_USER, 1);
 207
 208         if (sftp_direct == NULL) {
 209                 if (optind == argc || argc > (optind + 2))
 210                         usage();
 211
 212                 userhost = xstrdup(argv[optind]);
 213                 file2 = argv[optind+1];
 214
 215                 if ((host = strrchr(userhost, '@')) == NULL)
 216                         host = userhost;
 217                 else {
 218                         *host++ = '\0';
 219                         if (!userhost[0]) {
 220                                 fprintf(stderr, "Missing username\n");
 221                                 usage();
 222                         }
 223                         addargs(&args, "-l%s",userhost);
 224                 }
 225
 226                 if ((cp = colon(host)) != NULL) {
 227                         *cp++ = '\0';
 228                         file1 = cp;
 229                 }
 230
 231                 host = cleanhostname(host);
 232                 if (!*host) {
 233                         fprintf(stderr, "Missing hostname\n");
 234                         usage();
 235                 }
 236
 237                 addargs(&args, "-oProtocol %d", sshver);
 238
 239                 /* no subsystem if the server-spec contains a '/' */
 240                 if (sftp_server == NULL || strchr(sftp_server, '/') == NULL)
 241                         addargs(&args, "-s");
 242
 243                 addargs(&args, "%s", host);
 244                 addargs(&args, "%s", (sftp_server != NULL ?
 245                     sftp_server : "sftp"));
 246                 args.list[0] = ssh_program;
 247
 248                 if (!batchmode)
 249                         fprintf(stderr, "Connecting to %s...\n", host);
 250                 connect_to_server(ssh_program, args.list, &in, &out);
 251         } else {
 252                 args.list = NULL;
 253                 addargs(&args, "sftp-server");
 254
 255                 if (!batchmode)
 256                         fprintf(stderr, "Attaching to %s...\n", sftp_direct);
 257                 connect_to_server(sftp_direct, args.list, &in, &out);
 258         }
 259
 260         err = interactive_loop(in, out, file1, file2);
 261
 262 #if !defined(USE_PIPES)
 263         shutdown(in, SHUT_RDWR);
 264         shutdown(out, SHUT_RDWR);
 265 #endif
 266
 267         close(in);
 268         close(out);
 269         if (batchmode)
 270                 fclose(infile);
 271
 272         while (waitpid(sshpid, NULL, 0) == -1)
 273                 if (errno != EINTR)
 274                         fatal("Couldn't wait for ssh process: %s",
 275                             strerror(errno));
 276
 277         exit(err == 0 ? 0 : 1);
 278 }