sftp.c

   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.17 2001/05/08 19:45:25 mouring 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 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 void
  94 usage(void)
  95 {
  96         fprintf(stderr, "usage: sftp [-1vC] [-b batchfile] [-osshopt=value] [user@]host[:file [file]]\n");
  97         exit(1);
  98 }
  99
 100 int
 101 main(int argc, char **argv)
 102 {
 103         int in, out, ch;
 104         pid_t sshpid;
 105         char *host, *userhost, *cp, *file2;
 106         int debug_level = 0, sshver = 2;
 107         char *file1 = NULL, *sftp_server = NULL;
 108         LogLevel ll = SYSLOG_LEVEL_INFO;
 109         arglist args;
 110         extern int optind;
 111         extern char *optarg;
 112
 113         __progname = get_progname(argv[0]);
 114         args.list = NULL;
 115         addargs(&args, "ssh");         /* overwritten with ssh_program */
 116         addargs(&args, "-oFallBackToRsh no");
 117         addargs(&args, "-oForwardX11 no");
 118         addargs(&args, "-oForwardAgent no");
 119         ll = SYSLOG_LEVEL_INFO;
 120         infile = stdin;         /* Read from STDIN unless changed by -b */
 121
 122         while ((ch = getopt(argc, argv, "1hvCo:s:S:b:")) != -1) {
 123                 switch (ch) {
 124                 case 'C':
 125                         addargs(&args, "-C");
 126                         break;
 127                 case 'v':
 128                         if (debug_level < 3) {
 129                                 addargs(&args, "-v");
 130                                 ll = SYSLOG_LEVEL_DEBUG1 + debug_level;
 131                         }
 132                         debug_level++;
 133                         break;
 134                 case 'o':
 135                         addargs(&args, "-o%s", optarg);
 136                         break;
 137                 case '1':
 138                         sshver = 1;
 139                         if (sftp_server == NULL)
 140                                 sftp_server = _PATH_SFTP_SERVER;
 141                         break;
 142                 case 's':
 143                         sftp_server = optarg;
 144                         break;
 145                 case 'S':
 146                         ssh_program = optarg;
 147                         break;
 148                 case 'b':
 149                         if (infile == stdin) {
 150                                 infile = fopen(optarg, "r");
 151                                 if (infile == NULL)
 152                                         fatal("%s (%s).", strerror(errno), optarg);
 153                         } else
 154                                 fatal("Filename already specified.");
 155                         break;
 156                 case 'h':
 157                 default:
 158                         usage();
 159                 }
 160         }
 161
 162         if (optind == argc || argc > (optind + 2))
 163                 usage();
 164
 165         userhost = xstrdup(argv[optind]);
 166         file2 = argv[optind+1];
 167
 168         if ((cp = colon(userhost)) != NULL) {
 169                 *cp++ = '\0';
 170                 file1 = cp;
 171         }
 172
 173         if ((host = strchr(userhost, '@')) == NULL)
 174                 host = userhost;
 175         else {
 176                 *host++ = '\0';
 177                 if (!userhost[0]) {
 178                         fprintf(stderr, "Missing username\n");
 179                         usage();
 180                 }
 181                 addargs(&args, "-l%s",userhost);
 182         }
 183
 184         host = cleanhostname(host);
 185         if (!*host) {
 186                 fprintf(stderr, "Missing hostname\n");
 187                 usage();
 188         }
 189
 190         log_init(argv[0], ll, SYSLOG_FACILITY_USER, 1);
 191         addargs(&args, "-oProtocol %d", sshver);
 192
 193         /* no subsystem if the server-spec contains a '/' */
 194         if (sftp_server == NULL || strchr(sftp_server, '/') == NULL)
 195                 addargs(&args, "-s");
 196
 197         addargs(&args, "%s", host);
 198         addargs(&args, "%s", (sftp_server != NULL ? sftp_server : "sftp"));
 199         args.list[0] = ssh_program;
 200
 201         fprintf(stderr, "Connecting to %s...\n", host);
 202
 203         connect_to_server(args.list, &in, &out, &sshpid);
 204
 205         interactive_loop(in, out, file1, file2);
 206
 207 #if !defined(USE_PIPES)
 208         shutdown(in, SHUT_RDWR);
 209         shutdown(out, SHUT_RDWR);
 210 #endif
 211
 212         close(in);
 213         close(out);
 214         if (infile != stdin)
 215                 fclose(infile);
 216
 217         if (waitpid(sshpid, NULL, 0) == -1)
 218                 fatal("Couldn't wait for ssh process: %s", strerror(errno));
 219
 220         exit(0);
 221 }