/* vi: set sw=4 ts=4: */ /* * ftpget * * Mini implementation of FTP to retrieve a remote file. * * Copyright (C) 2002 Jeff Angielski, The PTR Group <jeff@theptrgroup.com> * Copyright (C) 2002 Glenn McGrath * * Based on wget.c by Chip Rosenthal Covad Communications * <chip@laserlink.net> * * Licensed under GPLv2 or later, see file LICENSE in this source tree. */ //usage:#define ftpget_trivial_usage //usage: "[OPTIONS] HOST [LOCAL_FILE] REMOTE_FILE" //usage:#define ftpget_full_usage "\n\n" //usage: "Download a file via FTP\n" //usage: IF_FEATURE_FTPGETPUT_LONG_OPTIONS( //usage: "\n -c,--continue Continue previous transfer" //usage: "\n -v,--verbose Verbose" //usage: "\n -u,--username USER Username" //usage: "\n -p,--password PASS Password" //usage: "\n -P,--port NUM Port" //usage: ) //usage: IF_NOT_FEATURE_FTPGETPUT_LONG_OPTIONS( //usage: "\n -c Continue previous transfer" //usage: "\n -v Verbose" //usage: "\n -u USER Username" //usage: "\n -p PASS Password" //usage: "\n -P NUM Port" //usage: ) //usage: //usage:#define ftpput_trivial_usage //usage: "[OPTIONS] HOST [REMOTE_FILE] LOCAL_FILE" //usage:#define ftpput_full_usage "\n\n" //usage: "Upload a file to a FTP server\n" //usage: IF_FEATURE_FTPGETPUT_LONG_OPTIONS( //usage: "\n -v,--verbose Verbose" //usage: "\n -u,--username USER Username" //usage: "\n -p,--password PASS Password" //usage: "\n -P,--port NUM Port" //usage: ) //usage: IF_NOT_FEATURE_FTPGETPUT_LONG_OPTIONS( //usage: "\n -v Verbose" //usage: "\n -u USER Username" //usage: "\n -p PASS Password" //usage: "\n -P NUM Port number" //usage: ) #include "libbb.h" #include "common_bufsiz.h" struct globals { const char *user; const char *password; struct len_and_sockaddr *lsa; FILE *control_stream; int verbose_flag; int do_continue; char buf[4]; /* actually [BUFSZ] */ } FIX_ALIASING; #define G (*(struct globals*)bb_common_bufsiz1) enum { BUFSZ = COMMON_BUFSIZE - offsetof(struct globals, buf) }; #define user (G.user ) #define password (G.password ) #define lsa (G.lsa ) #define control_stream (G.control_stream) #define verbose_flag (G.verbose_flag ) #define do_continue (G.do_continue ) #define buf (G.buf ) #define INIT_G() do { \ setup_common_bufsiz(); \ BUILD_BUG_ON(sizeof(G) > COMMON_BUFSIZE); \ } while (0) static void ftp_die(const char *msg) NORETURN; static void ftp_die(const char *msg) { char *cp = buf; /* buf holds peer's response */ /* Guard against garbage from remote server */ while (*cp >= ' ' && *cp < '\x7f') cp++; *cp = '\0'; bb_error_msg_and_die("unexpected server response%s%s: %s", (msg ? " to " : ""), (msg ? msg : ""), buf); } static int ftpcmd(const char *s1, const char *s2) { unsigned n; if (verbose_flag) { bb_error_msg("cmd %s %s", s1, s2); } if (s1) { fprintf(control_stream, (s2 ? "%s %s\r\n" : "%s %s\r\n"+3), s1, s2); fflush(control_stream); } do { strcpy(buf, "EOF"); /* for ftp_die */ if (fgets(buf, BUFSZ - 2, control_stream) == NULL) { ftp_die(NULL); } } while (!isdigit(buf[0]) || buf[3] != ' '); buf[3] = '\0'; n = xatou(buf); buf[3] = ' '; return n; } static void ftp_login(void) { /* Connect to the command socket */ control_stream = fdopen(xconnect_stream(lsa), "r+"); if (control_stream == NULL) { /* fdopen failed - extremely unlikely */ bb_perror_nomsg_and_die(); } if (ftpcmd(NULL, NULL) != 220) { ftp_die(NULL); } /* Login to the server */ switch (ftpcmd("USER", user)) { case 230: break; case 331: if (ftpcmd("PASS", password) != 230) { ftp_die("PASS"); } break; default: ftp_die("USER"); } ftpcmd("TYPE I", NULL); } static int xconnect_ftpdata(void) { char *buf_ptr; unsigned port_num; /* TODO: PASV command will not work for IPv6. RFC2428 describes IPv6-capable "extended PASV" - EPSV. "EPSV [protocol]" asks server to bind to and listen on a data port in specified protocol. Protocol is 1 for IPv4, 2 for IPv6. If not specified, defaults to "same as used for control connection". If server understood you, it should answer "229 <some text>(|||port|)" where "|" are literal pipe chars and "port" is ASCII decimal port#. There is also an IPv6-capable replacement for PORT (EPRT), but we don't need that. NB: PASV may still work for some servers even over IPv6. For example, vsftp happily answers "227 Entering Passive Mode (0,0,0,0,n,n)" and proceeds as usual. TODO2: need to stop ignoring IP address in PASV response. */ if (ftpcmd("PASV", NULL) != 227) { ftp_die("PASV"); } /* Response is "NNN garbageN1,N2,N3,N4,P1,P2[)garbage] * Server's IP is N1.N2.N3.N4 (we ignore it) * Server's port for data connection is P1*256+P2 */ buf_ptr = strrchr(buf, ')'); if (buf_ptr) *buf_ptr = '\0'; buf_ptr = strrchr(buf, ','); *buf_ptr = '\0'; port_num = xatoul_range(buf_ptr + 1, 0, 255); buf_ptr = strrchr(buf, ','); *buf_ptr = '\0'; port_num += xatoul_range(buf_ptr + 1, 0, 255) * 256; set_nport(&lsa->u.sa, htons(port_num)); return xconnect_stream(lsa); } static int pump_data_and_QUIT(int from, int to) { /* copy the file */ if (bb_copyfd_eof(from, to) == -1) { /* error msg is already printed by bb_copyfd_eof */ return EXIT_FAILURE; } /* close data connection */ close(from); /* don't know which one is that, so we close both */ close(to); /* does server confirm that transfer is finished? */ if (ftpcmd(NULL, NULL) != 226) { ftp_die(NULL); } ftpcmd("QUIT", NULL); return EXIT_SUCCESS; } #if !ENABLE_FTPGET int ftp_receive(const char *local_path, char *server_path); #else static int ftp_receive(const char *local_path, char *server_path) { int fd_data; int fd_local = -1; off_t beg_range = 0; /* connect to the data socket */ fd_data = xconnect_ftpdata(); if (ftpcmd("SIZE", server_path) != 213) { do_continue = 0; } if (LONE_DASH(local_path)) { fd_local = STDOUT_FILENO; do_continue = 0; } if (do_continue) { struct stat sbuf; /* lstat would be wrong here! */ if (stat(local_path, &sbuf) < 0) { bb_perror_msg_and_die("stat"); } if (sbuf.st_size > 0) { beg_range = sbuf.st_size; } else { do_continue = 0; } } if (do_continue) { sprintf(buf, "REST %"OFF_FMT"u", beg_range); if (ftpcmd(buf, NULL) != 350) { do_continue = 0; } } if (ftpcmd("RETR", server_path) > 150) { ftp_die("RETR"); } /* create local file _after_ we know that remote file exists */ if (fd_local == -1) { fd_local = xopen(local_path, do_continue ? (O_APPEND | O_WRONLY) : (O_CREAT | O_TRUNC | O_WRONLY) ); } return pump_data_and_QUIT(fd_data, fd_local); } #endif #if !ENABLE_FTPPUT int ftp_send(const char *server_path, char *local_path); #else static int ftp_send(const char *server_path, char *local_path) { int fd_data; int fd_local; int response; /* connect to the data socket */ fd_data = xconnect_ftpdata(); /* get the local file */ fd_local = STDIN_FILENO; if (NOT_LONE_DASH(local_path)) fd_local = xopen(local_path, O_RDONLY); response = ftpcmd("STOR", server_path); switch (response) { case 125: case 150: break; default: ftp_die("STOR"); } return pump_data_and_QUIT(fd_local, fd_data); } #endif #if ENABLE_FEATURE_FTPGETPUT_LONG_OPTIONS static const char ftpgetput_longopts[] ALIGN1 = "continue\0" Required_argument "c" "verbose\0" No_argument "v" "username\0" Required_argument "u" "password\0" Required_argument "p" "port\0" Required_argument "P" ; #endif int ftpgetput_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; int ftpgetput_main(int argc UNUSED_PARAM, char **argv) { const char *port = "ftp"; /* socket to ftp server */ #if ENABLE_FTPPUT && !ENABLE_FTPGET # define ftp_action ftp_send #elif ENABLE_FTPGET && !ENABLE_FTPPUT # define ftp_action ftp_receive #else int (*ftp_action)(const char *, char *) = ftp_send; /* Check to see if the command is ftpget or ftput */ if (applet_name[3] == 'g') { ftp_action = ftp_receive; } #endif INIT_G(); /* Set default values */ user = "anonymous"; password = "busybox@"; /* * Decipher the command line */ #if ENABLE_FEATURE_FTPGETPUT_LONG_OPTIONS applet_long_options = ftpgetput_longopts; #endif opt_complementary = "-2:vv:cc"; /* must have 2 to 3 params; -v and -c count */ getopt32(argv, "cvu:p:P:", &user, &password, &port, &verbose_flag, &do_continue); argv += optind; /* We want to do exactly _one_ DNS lookup, since some * sites (i.e. ftp.us.debian.org) use round-robin DNS * and we want to connect to only one IP... */ lsa = xhost2sockaddr(argv[0], bb_lookup_port(port, "tcp", 21)); if (verbose_flag) { printf("Connecting to %s (%s)\n", argv[0], xmalloc_sockaddr2dotted(&lsa->u.sa)); } ftp_login(); return ftp_action(argv[1], argv[2] ? argv[2] : argv[1]); }