bf9d17949
Denis Vlasenko
2009-03-08
Subject
adding forgotten new file
Binary
No binary (build failed)
Architecture & Compiler Flags
—
Build Log
(log yok)
Commit Diff GitHub →
networking/ftpd.c+1130 -0
| @@ -0,0 +1,1130 @@ | |
| + | /* vi: set sw=4 ts=4: */ |
| + | /* |
| + | * Simple FTP daemon, based on vsftpd 2.0.7 (written by Chris Evans) |
| + | * |
| + | * Author: Adam Tkac <[email protected]> |
| + | * |
| + | * Only subset of FTP protocol is implemented but vast majority of clients |
| + | * should not have any problem. You have to run this daemon via inetd. |
| + | * |
| + | * Options: |
| + | * -w - enable FTP write commands |
| + | */ |
| + | |
| + | #include "libbb.h" |
| + | #include <netinet/tcp.h> |
| + | |
| + | enum { |
| + | FTP_DATACONN = 150, |
| + | FTP_NOOPOK = 200, |
| + | FTP_TYPEOK = 200, |
| + | FTP_PORTOK = 200, |
| + | FTP_STRUOK = 200, |
| + | FTP_MODEOK = 200, |
| + | FTP_ALLOOK = 202, |
| + | FTP_STATOK = 211, |
| + | FTP_STATFILE_OK = 213, |
| + | FTP_HELP = 214, |
| + | FTP_SYSTOK = 215, |
| + | FTP_GREET = 220, |
| + | FTP_GOODBYE = 221, |
| + | FTP_TRANSFEROK = 226, |
| + | FTP_PASVOK = 227, |
| + | FTP_LOGINOK = 230, |
| + | FTP_CWDOK = 250, |
| + | #if ENABLE_FEATURE_FTP_WRITE |
| + | FTP_RMDIROK = 250, |
| + | FTP_DELEOK = 250, |
| + | FTP_RENAMEOK = 250, |
| + | #endif |
| + | FTP_PWDOK = 257, |
| + | #if ENABLE_FEATURE_FTP_WRITE |
| + | FTP_MKDIROK = 257, |
| + | #endif |
| + | FTP_GIVEPWORD = 331, |
| + | FTP_RESTOK = 350, |
| + | #if ENABLE_FEATURE_FTP_WRITE |
| + | FTP_RNFROK = 350, |
| + | #endif |
| + | FTP_BADSENDCONN = 425, |
| + | FTP_BADSENDNET = 426, |
| + | FTP_BADSENDFILE = 451, |
| + | FTP_BADCMD = 500, |
| + | FTP_COMMANDNOTIMPL = 502, |
| + | FTP_NEEDUSER = 503, |
| + | FTP_NEEDRNFR = 503, |
| + | FTP_BADSTRU = 504, |
| + | FTP_BADMODE = 504, |
| + | FTP_LOGINERR = 530, |
| + | FTP_FILEFAIL = 550, |
| + | FTP_NOPERM = 550, |
| + | FTP_UPLOADFAIL = 553, |
| + | |
| + | #define mk_const4(a,b,c,d) (((a * 0x100 + b) * 0x100 + c) * 0x100 + d) |
| + | #define mk_const3(a,b,c) ((a * 0x100 + b) * 0x100 + c) |
| + | const_ALLO = mk_const4('A', 'L', 'L', 'O'), |
| + | const_APPE = mk_const4('A', 'P', 'P', 'E'), |
| + | const_CDUP = mk_const4('C', 'D', 'U', 'P'), |
| + | const_CWD = mk_const3('C', 'W', 'D'), |
| + | const_DELE = mk_const4('D', 'E', 'L', 'E'), |
| + | const_HELP = mk_const4('H', 'E', 'L', 'P'), |
| + | const_LIST = mk_const4('L', 'I', 'S', 'T'), |
| + | const_MKD = mk_const3('M', 'K', 'D'), |
| + | const_MODE = mk_const4('M', 'O', 'D', 'E'), |
| + | const_NLST = mk_const4('N', 'L', 'S', 'T'), |
| + | const_NOOP = mk_const4('N', 'O', 'O', 'P'), |
| + | const_PASS = mk_const4('P', 'A', 'S', 'S'), |
| + | const_PASV = mk_const4('P', 'A', 'S', 'V'), |
| + | const_PORT = mk_const4('P', 'O', 'R', 'T'), |
| + | const_PWD = mk_const3('P', 'W', 'D'), |
| + | const_QUIT = mk_const4('Q', 'U', 'I', 'T'), |
| + | const_REST = mk_const4('R', 'E', 'S', 'T'), |
| + | const_RETR = mk_const4('R', 'E', 'T', 'R'), |
| + | const_RMD = mk_const3('R', 'M', 'D'), |
| + | const_RNFR = mk_const4('R', 'N', 'F', 'R'), |
| + | const_RNTO = mk_const4('R', 'N', 'T', 'O'), |
| + | const_STAT = mk_const4('S', 'T', 'A', 'T'), |
| + | const_STOR = mk_const4('S', 'T', 'O', 'R'), |
| + | const_STOU = mk_const4('S', 'T', 'O', 'U'), |
| + | const_STRU = mk_const4('S', 'T', 'R', 'U'), |
| + | const_SYST = mk_const4('S', 'Y', 'S', 'T'), |
| + | const_TYPE = mk_const4('T', 'Y', 'P', 'E'), |
| + | const_USER = mk_const4('U', 'S', 'E', 'R'), |
| + | }; |
| + | |
| + | struct globals { |
| + | char *p_control_line_buf; |
| + | len_and_sockaddr *local_addr; |
| + | len_and_sockaddr *port_addr; |
| + | int pasv_listen_fd; |
| + | int data_fd; |
| + | off_t restart_pos; |
| + | char *ftp_cmp; |
| + | char *ftp_arg; |
| + | #if ENABLE_FEATURE_FTP_WRITE |
| + | char *rnfr_filename; |
| + | smallint write_enable; |
| + | #endif |
| + | }; |
| + | #define G (*(struct globals*)&bb_common_bufsiz1) |
| + | #define INIT_G() do { } while (0) |
| + | |
| + | |
| + | static char * |
| + | replace_text(const char *str, const char from, const char *to) |
| + | { |
| + | size_t retlen, remainlen, chunklen, tolen; |
| + | const char *remain; |
| + | char *ret, *found; |
| + | |
| + | remain = str; |
| + | remainlen = strlen(str); |
| + | |
| + | tolen = strlen(to); |
| + | |
| + | /* simply alloc strlen(str)*strlen(to). To is max 2 so it's allowed */ |
| + | ret = xmalloc(remainlen * strlen(to) + 1); |
| + | retlen = 0; |
| + | |
| + | for (;;) { |
| + | found = strchr(remain, from); |
| + | if (found != NULL) { |
| + | chunklen = found - remain; |
| + | |
| + | /* Copy chunk which doesn't contain 'from' to ret */ |
| + | memcpy(&ret[retlen], remain, chunklen); |
| + | retlen += chunklen; |
| + | |
| + | /* Now copy 'to' instead of 'from' */ |
| + | memcpy(&ret[retlen], to, tolen); |
| + | retlen += tolen; |
| + | |
| + | remain = found + 1; |
| + | } else { |
| + | /* |
| + | * The last chunk. We are already sure that we have enough space |
| + | * so we can use strcpy. |
| + | */ |
| + | strcpy(&ret[retlen], remain); |
| + | break; |
| + | } |
| + | } |
| + | return ret; |
| + | } |
| + | |
| + | static void |
| + | replace_char(char *str, char from, char to) |
| + | { |
| + | char *ptr; |
| + | |
| + | /* Don't use strchr here...*/ |
| + | while ((ptr = strchr(str, from)) != NULL) { |
| + | *ptr = to; |
| + | str = ptr + 1; |
| + | } |
| + | } |
| + | |
| + | static void |
| + | str_netfd_write(const char *str, int fd) |
| + | { |
| + | xwrite(fd, str, strlen(str)); |
| + | } |
| + | |
| + | static void |
| + | ftp_write_str_common(unsigned int status, const char *str, char sep) |
| + | { |
| + | char *escaped_str, *response; |
| + | size_t len; |
| + | |
| + | escaped_str = replace_text(str, '\377', "\377\377"); |
| + | |
| + | response = xasprintf("%u%c%s\r\n", status, sep, escaped_str); |
| + | free(escaped_str); |
| + | |
| + | len = strlen(response); |
| + | replace_char(escaped_str, '\n', '\0'); |
| + | |
| + | /* Change trailing '\0' back to '\n' */ |
| + | response[len - 1] = '\n'; |
| + | xwrite(STDIN_FILENO, response, len); |
| + | } |
| + | |
| + | static void |
| + | cmdio_write(int status, const char *p_text) |
| + | { |
| + | ftp_write_str_common(status, p_text, ' '); |
| + | } |
| + | |
| + | static void |
| + | cmdio_write_hyphen(int status, const char *p_text) |
| + | { |
| + | ftp_write_str_common(status, p_text, '-'); |
| + | } |
| + | |
| + | static void |
| + | cmdio_write_raw(const char *p_text) |
| + | { |
| + | str_netfd_write(p_text, STDIN_FILENO); |
| + | } |
| + | |
| + | static uint32_t |
| + | cmdio_get_cmd_and_arg(void) |
| + | { |
| + | int len; |
| + | uint32_t cmdval; |
| + | char *cmd; |
| + | |
| + | free(G.ftp_cmp); |
| + | G.ftp_cmp = cmd = xmalloc_reads(STDIN_FILENO, NULL, NULL); |
| + | /* |
| + | * TODO: |
| + | * |
| + | * now we should change all '\0' to '\n' - xmalloc_reads will be improved, |
| + | * probably |
| + | */ |
| + | len = strlen(cmd) - 1; |
| + | while (len >= 0 && cmd[len] == '\r') { |
| + | cmd[len] = '\0'; |
| + | len--; |
| + | } |
| + | |
| + | G.ftp_arg = strchr(cmd, ' '); |
| + | if (G.ftp_arg != NULL) { |
| + | *G.ftp_arg = '\0'; |
| + | G.ftp_arg++; |
| + | } |
| + | cmdval = 0; |
| + | while (*cmd) |
| + | cmdval = (cmdval << 8) + ((unsigned char)*cmd++ & (unsigned char)~0x20); |
| + | |
| + | return cmdval; |
| + | } |
| + | |
| + | static void |
| + | init_data_sock_params(int sock_fd) |
| + | { |
| + | struct linger linger; |
| + | |
| + | G.data_fd = sock_fd; |
| + | |
| + | memset(&linger, 0, sizeof(linger)); |
| + | linger.l_onoff = 1; |
| + | linger.l_linger = 32767; |
| + | |
| + | setsockopt(sock_fd, SOL_SOCKET, SO_KEEPALIVE, &const_int_1, sizeof(const_int_1)); |
| + | setsockopt(sock_fd, SOL_SOCKET, SO_LINGER, &linger, sizeof(linger)); |
| + | } |
| + | |
| + | static int |
| + | ftpdataio_get_pasv_fd(void) |
| + | { |
| + | int remote_fd; |
| + | |
| + | remote_fd = accept(G.pasv_listen_fd, NULL, 0); |
| + | |
| + | if (remote_fd < 0) { |
| + | cmdio_write(FTP_BADSENDCONN, "Can't establish connection"); |
| + | return remote_fd; |
| + | } |
| + | |
| + | init_data_sock_params(remote_fd); |
| + | return remote_fd; |
| + | } |
| + | |
| + | static int |
| + | ftpdataio_get_port_fd(void) |
| + | { |
| + | int remote_fd; |
| + | |
| + | /* Do we want die or print error to client? */ |
| + | remote_fd = xconnect_stream(G.port_addr); |
| + | |
| + | init_data_sock_params(remote_fd); |
| + | return remote_fd; |
| + | } |
| + | |
| + | static void |
| + | ftpdataio_dispose_transfer_fd(void) |
| + | { |
| + | /* This close() blocks because we set SO_LINGER */ |
| + | if (G.data_fd > STDOUT_FILENO) { |
| + | if (close(G.data_fd) < 0) { |
| + | /* Do it again without blocking. */ |
| + | struct linger linger; |
| + | |
| + | memset(&linger, 0, sizeof(linger)); |
| + | setsockopt(G.data_fd, SOL_SOCKET, SO_LINGER, &linger, sizeof(linger)); |
| + | close(G.data_fd); |
| + | } |
| + | } |
| + | G.data_fd = -1; |
| + | } |
| + | |
| + | static int |
| + | port_active(void) |
| + | { |
| + | return (G.port_addr != NULL) ? 1: 0; |
| + | } |
| + | |
| + | static int |
| + | pasv_active(void) |
| + | { |
| + | return (G.pasv_listen_fd != -1) ? 1 : 0; |
| + | } |
| + | |
| + | static int |
| + | get_remote_transfer_fd(const char *p_status_msg) |
| + | { |
| + | int remote_fd; |
| + | |
| + | if (pasv_active()) |
| + | remote_fd = ftpdataio_get_pasv_fd(); |
| + | else |
| + | remote_fd = ftpdataio_get_port_fd(); |
| + | |
| + | if (remote_fd < 0) |
| + | return remote_fd; |
| + | |
| + | cmdio_write(FTP_DATACONN, p_status_msg); |
| + | return remote_fd; |
| + | } |
| + | |
| + | static void |
| + | handle_pwd(void) |
| + | { |
| + | char *cwd, *promoted_cwd, *response; |
| + | |
| + | cwd = xrealloc_getcwd_or_warn(NULL); |
| + | if (cwd == NULL) |
| + | cwd = xstrdup(""); |
| + | |
| + | /* We _have to_ promote each " to "" */ |
| + | promoted_cwd = replace_text(cwd, '\"', "\"\""); |
| + | free(cwd); |
| + | response = xasprintf("\"%s\"", promoted_cwd); |
| + | free(promoted_cwd); |
| + | cmdio_write(FTP_PWDOK, response); |
| + | free(response); |
| + | } |
| + | |
| + | static void |
| + | handle_cwd(void) |
| + | { |
| + | int retval; |
| + | |
| + | /* XXX Do we need check ftp_arg != NULL? */ |
| + | retval = chdir(G.ftp_arg); |
| + | if (retval == 0) |
| + | cmdio_write(FTP_CWDOK, "Directory changed"); |
| + | else |
| + | cmdio_write(FTP_FILEFAIL, "Can't change directory"); |
| + | } |
| + | |
| + | static void |
| + | handle_cdup(void) |
| + | { |
| + | G.ftp_arg = xstrdup(".."); |
| + | handle_cwd(); |
| + | free(G.ftp_arg); |
| + | } |
| + | |
| + | static int |
| + | data_transfer_checks_ok(void) |
| + | { |
| + | if (!pasv_active() && !port_active()) { |
| + | cmdio_write(FTP_BADSENDCONN, "Use PORT or PASV first"); |
| + | return 0; |
| + | } |
| + | |
| + | return 1; |
| + | } |
| + | |
| + | static void |
| + | port_cleanup(void) |
| + | { |
| + | if (G.port_addr != NULL) |
| + | free(G.port_addr); |
| + | |
| + | G.port_addr = NULL; |
| + | } |
| + | |
| + | static void |
| + | pasv_cleanup(void) |
| + | { |
| + | if (G.pasv_listen_fd > STDOUT_FILENO) |
| + | close(G.pasv_listen_fd); |
| + | G.pasv_listen_fd = -1; |
| + | } |
| + | |
| + | static char * |
| + | statbuf_getperms(const struct stat *statbuf) |
| + | { |
| + | char *perms; |
| + | enum { r = 'r', w = 'w', x = 'x', s = 's', S = 'S' }; |
| + | |
| + | perms = xmalloc(11); |
| + | memset(perms, '-', 10); |
| + | |
| + | perms[0] = '?'; |
| + | switch (statbuf->st_mode & S_IFMT) { |
| + | case S_IFREG: perms[0] = '-'; break; |
| + | case S_IFDIR: perms[0] = 'd'; break; |
| + | case S_IFLNK: perms[0] = 'l'; break; |
| + | case S_IFIFO: perms[0] = 'p'; break; |
| + | case S_IFSOCK: perms[0] = s; break; |
| + | case S_IFCHR: perms[0] = 'c'; break; |
| + | case S_IFBLK: perms[0] = 'b'; break; |
| + | } |
| + | |
| + | if (statbuf->st_mode & S_IRUSR) perms[1] = r; |
| + | if (statbuf->st_mode & S_IWUSR) perms[2] = w; |
| + | if (statbuf->st_mode & S_IXUSR) perms[3] = x; |
| + | if (statbuf->st_mode & S_IRGRP) perms[4] = r; |
| + | if (statbuf->st_mode & S_IWGRP) perms[5] = w; |
| + | if (statbuf->st_mode & S_IXGRP) perms[6] = x; |
| + | if (statbuf->st_mode & S_IROTH) perms[7] = r; |
| + | if (statbuf->st_mode & S_IWOTH) perms[8] = w; |
| + | if (statbuf->st_mode & S_IXOTH) perms[9] = x; |
| + | if (statbuf->st_mode & S_ISUID) perms[3] = (perms[3] == x) ? s : S; |
| + | if (statbuf->st_mode & S_ISGID) perms[6] = (perms[6] == x) ? s : S; |
| + | if (statbuf->st_mode & S_ISVTX) perms[9] = (perms[9] == x) ? 't' : 'T'; |
| + | |
| + | perms[10] = '\0'; |
| + | |
| + | return perms; |
| + | } |
| + | |
| + | static void |
| + | write_filestats(int fd, const char *filename, |
| + | const struct stat *statbuf) |
| + | { |
| + | off_t size; |
| + | char *stats, *lnkname = NULL, *perms; |
| + | const char *name; |
| + | int retval; |
| + | char timestr[32]; |
| + | struct tm *tm; |
| + | const char *format = "%b %d %H:%M"; |
| + | |
| + | name = bb_get_last_path_component_nostrip(filename); |
| + | |
| + | if (statbuf != NULL) { |
| + | size = statbuf->st_size; |
| + | |
| + | if (S_ISLNK(statbuf->st_mode)) |
| + | /* Damn symlink... */ |
| + | lnkname = xmalloc_readlink(filename); |
| + | |
| + | tm = gmtime(&statbuf->st_mtime); |
| + | retval = strftime(timestr, sizeof(timestr), format, tm); |
| + | if (retval == 0) |
| + | bb_error_msg_and_die("strftime"); |
| + | |
| + | timestr[sizeof(timestr) - 1] = '\0'; |
| + | |
| + | perms = statbuf_getperms(statbuf); |
| + | |
| + | stats = xasprintf("%s %u\tftp ftp %"OFF_FMT"u\t%s %s", |
| + | perms, (int) statbuf->st_nlink, |
| + | size, timestr, name); |
| + | |
| + | free(perms); |
| + | } else |
| + | stats = xstrdup(name); |
| + | |
| + | str_netfd_write(stats, fd); |
| + | free(stats); |
| + | if (lnkname != NULL) { |
| + | str_netfd_write(" -> ", fd); |
| + | str_netfd_write(lnkname, fd); |
| + | free(lnkname); |
| + | } |
| + | str_netfd_write("\r\n", fd); |
| + | } |
| + | |
| + | static void |
| + | write_dirstats(int fd, const char *dname, int details) |
| + | { |
| + | DIR *dir; |
| + | struct dirent *dirent; |
| + | struct stat statbuf; |
| + | char *filename; |
| + | |
| + | dir = xopendir(dname); |
| + | |
| + | for (;;) { |
| + | dirent = readdir(dir); |
| + | if (dirent == NULL) |
| + | break; |
| + | |
| + | /* Ignore . and .. */ |
| + | if (dirent->d_name[0] == '.') { |
| + | if (dirent->d_name[1] == '\0' |
| + | || (dirent->d_name[1] == '.' && dirent->d_name[2] == '\0') |
| + | ) { |
| + | continue; |
| + | } |
| + | } |
| + | |
| + | if (details) { |
| + | filename = xasprintf("%s/%s", dname, dirent->d_name); |
| + | if (lstat(filename, &statbuf) != 0) { |
| + | free(filename); |
| + | goto bail; |
| + | } |
| + | } else |
| + | filename = xstrdup(dirent->d_name); |
| + | |
| + | write_filestats(fd, filename, details ? &statbuf : NULL); |
| + | free(filename); |
| + | } |
| + | |
| + | bail: |
| + | closedir(dir); |
| + | } |
| + | |
| + | static void |
| + | handle_pasv(void) |
| + | { |
| + | int bind_retries = 10; |
| + | unsigned short port; |
| + | enum { min_port = 1024, max_port = 65535 }; |
| + | char *addr, *wire_addr, *response; |
| + | |
| + | pasv_cleanup(); |
| + | port_cleanup(); |
| + | G.pasv_listen_fd = xsocket(G.local_addr->u.sa.sa_family, SOCK_STREAM, 0); |
| + | setsockopt_reuseaddr(G.pasv_listen_fd); |
| + | |
| + | /* TODO bind() with port == 0 and then call getsockname */ |
| + | while (--bind_retries) { |
| + | port = rand() % max_port; |
| + | if (port < min_port) { |
| + | port += min_port; |
| + | } |
| + | |
| + | set_nport(G.local_addr, htons(port)); |
| + | /* We don't want to use xbind, it'll die if port is in use */ |
| + | if (bind(G.pasv_listen_fd, &G.local_addr->u.sa, G.local_addr->len) != 0) { |
| + | /* do we want check if errno == EADDRINUSE ? */ |
| + | continue; |
| + | } |
| + | xlisten(G.pasv_listen_fd, 1); |
| + | break; |
| + | } |
| + | |
| + | if (!bind_retries) |
| + | bb_error_msg_and_die("can't create pasv socket"); |
| + | |
| + | addr = xmalloc_sockaddr2dotted_noport(&G.local_addr->u.sa); |
| + | wire_addr = replace_text(addr, '.', ","); |
| + | free(addr); |
| + | |
| + | response = xasprintf("Entering Passive Mode (%s,%u,%u)", |
| + | wire_addr, (int)(port >> 8), (int)(port & 255)); |
| + | |
| + | cmdio_write(FTP_PASVOK, response); |
| + | free(wire_addr); |
| + | free(response); |
| + | } |
| + | |
| + | static void |
| + | handle_retr(void) |
| + | { |
| + | struct stat statbuf; |
| + | int trans_ret, retval; |
| + | int remote_fd; |
| + | int opened_file; |
| + | off_t offset = G.restart_pos; |
| + | char *response; |
| + | |
| + | G.restart_pos = 0; |
| + | |
| + | if (!data_transfer_checks_ok()) |
| + | return; |
| + | |
| + | /* XXX Do we need check if ftp_arg != NULL? */ |
| + | opened_file = open(G.ftp_arg, O_RDONLY | O_NONBLOCK); |
| + | if (opened_file < 0) { |
| + | cmdio_write(FTP_FILEFAIL, "Can't open file"); |
| + | return; |
| + | } |
| + | |
| + | retval = fstat(opened_file, &statbuf); |
| + | if (retval < 0 || !S_ISREG(statbuf.st_mode)) { |
| + | /* Note - pretend open failed */ |
| + | cmdio_write(FTP_FILEFAIL, "Can't open file"); |
| + | goto file_close_out; |
| + | } |
| + | |
| + | /* Now deactive O_NONBLOCK, otherwise we have a problem on DMAPI filesystems |
| + | * such as XFS DMAPI. |
| + | */ |
| + | ndelay_off(opened_file); |
| + | |
| + | /* Set the download offset (from REST) if any */ |
| + | if (offset != 0) |
| + | xlseek(opened_file, offset, SEEK_SET); |
| + | |
| + | response = xasprintf( |
| + | "Opening BINARY mode data connection for (%s %"OFF_FMT"u bytes).", |
| + | G.ftp_arg, statbuf.st_size); |
| + | |
| + | remote_fd = get_remote_transfer_fd(response); |
| + | free(response); |
| + | if (remote_fd < 0) |
| + | goto port_pasv_cleanup_out; |
| + | |
| + | trans_ret = bb_copyfd_eof(opened_file, remote_fd); |
| + | ftpdataio_dispose_transfer_fd(); |
| + | if (trans_ret < 0) |
| + | cmdio_write(FTP_BADSENDFILE, "Error sending local file"); |
| + | else |
| + | cmdio_write(FTP_TRANSFEROK, "File sent OK"); |
| + | |
| + | port_pasv_cleanup_out: |
| + | port_cleanup(); |
| + | pasv_cleanup(); |
| + | file_close_out: |
| + | close(opened_file); |
| + | } |
| + | |
| + | static void |
| + | handle_dir_common(int full_details, int stat_cmd) |
| + | { |
| + | int fd; |
| + | struct stat statbuf; |
| + | |
| + | if (!stat_cmd && !data_transfer_checks_ok()) |
| + | return; |
| + | |
| + | if (stat_cmd) { |
| + | fd = STDIN_FILENO; |
| + | cmdio_write_hyphen(FTP_STATFILE_OK, "Status follows:"); |
| + | } else { |
| + | fd = get_remote_transfer_fd("Here comes the directory listing"); |
| + | if (fd < 0) |
| + | goto bail; |
| + | } |
| + | |
| + | if (G.ftp_arg != NULL) { |
| + | if (lstat(G.ftp_arg, &statbuf) != 0) { |
| + | /* Dir doesn't exist => return ok to client */ |
| + | goto bail; |
| + | } |
| + | if (S_ISREG(statbuf.st_mode) || S_ISLNK(statbuf.st_mode)) |
| + | write_filestats(fd, G.ftp_arg, &statbuf); |
| + | else if (S_ISDIR(statbuf.st_mode)) |
| + | write_dirstats(fd, G.ftp_arg, full_details); |
| + | } else |
| + | write_dirstats(fd, ".", full_details); |
| + | |
| + | bail: |
| + | /* Well, if we can't open directory/file it doesn't matter */ |
| + | if (!stat_cmd) { |
| + | ftpdataio_dispose_transfer_fd(); |
| + | pasv_cleanup(); |
| + | port_cleanup(); |
| + | cmdio_write(FTP_TRANSFEROK, "OK"); |
| + | } else |
| + | cmdio_write(FTP_STATFILE_OK, "End of status"); |
| + | } |
| + | |
| + | static void |
| + | handle_list(void) |
| + | { |
| + | handle_dir_common(1, 0); |
| + | } |
| + | |
| + | static void |
| + | handle_type(void) |
| + | { |
| + | if (G.ftp_arg |
| + | && ( ((G.ftp_arg[0] | 0x20) == 'i' && G.ftp_arg[1] == '\0') |
| + | || !strcasecmp(G.ftp_arg, "L8") |
| + | || !strcasecmp(G.ftp_arg, "L 8") |
| + | ) |
| + | ) { |
| + | cmdio_write(FTP_TYPEOK, "Switching to Binary mode"); |
| + | } else { |
| + | cmdio_write(FTP_BADCMD, "Unrecognised TYPE command"); |
| + | } |
| + | } |
| + | |
| + | static void |
| + | handle_port(void) |
| + | { |
| + | unsigned short port; |
| + | char *raw = NULL, *port_part; |
| + | len_and_sockaddr *lsa = NULL; |
| + | |
| + | pasv_cleanup(); |
| + | port_cleanup(); |
| + | |
| + | if (G.ftp_arg == NULL) |
| + | goto bail; |
| + | |
| + | raw = replace_text(G.ftp_arg, ',', "."); |
| + | |
| + | port_part = strrchr(raw, '.'); |
| + | if (port_part == NULL) |
| + | goto bail; |
| + | |
| + | port = xatou16(&port_part[1]); |
| + | *port_part = '\0'; |
| + | |
| + | port_part = strrchr(raw, '.'); |
| + | if (port_part == NULL) |
| + | goto bail; |
| + | |
| + | port |= xatou16(&port_part[1]) << 8; |
| + | *port_part = '\0'; |
| + | |
| + | lsa = xdotted2sockaddr(raw, port); |
| + | |
| + | bail: |
| + | free(raw); |
| + | |
| + | if (lsa == NULL) { |
| + | cmdio_write(FTP_BADCMD, "Illegal PORT command"); |
| + | return; |
| + | } |
| + | |
| + | G.port_addr = lsa; |
| + | cmdio_write(FTP_PORTOK, "PORT command successful. Consider using PASV"); |
| + | } |
| + | |
| + | #if ENABLE_FEATURE_FTP_WRITE |
| + | static void |
| + | handle_upload_common(int is_append, int is_unique) |
| + | { |
| + | char *template = NULL; |
| + | int trans_ret; |
| + | int new_file_fd; |
| + | int remote_fd; |
| + | |
| + | enum { |
| + | fileflags = O_CREAT | O_WRONLY | O_APPEND, |
| + | }; |
| + | |
| + | off_t offset = G.restart_pos; |
| + | |
| + | G.restart_pos = 0; |
| + | if (!data_transfer_checks_ok()) |
| + | return; |
| + | |
| + | if (is_unique) { |
| + | template = xstrdup("uniq.XXXXXX"); |
| + | /* |
| + | * XXX Use mkostemp here? vsftpd opens file with O_CREAT, O_WRONLY, |
| + | * O_APPEND and O_EXCL flags... |
| + | */ |
| + | new_file_fd = mkstemp(template); |
| + | } else { |
| + | /* XXX Do we need check if ftp_arg != NULL? */ |
| + | if (!is_append && offset == 0) |
| + | new_file_fd = open(G.ftp_arg, O_CREAT | O_WRONLY | O_APPEND | O_NONBLOCK | O_TRUNC, 0666); |
| + | else |
| + | new_file_fd = open(G.ftp_arg, O_CREAT | O_WRONLY | O_APPEND | O_NONBLOCK, 0666); |
| + | } |
| + | |
| + | if (new_file_fd < 0) { |
| + | cmdio_write(FTP_UPLOADFAIL, "Can't create file"); |
| + | return; |
| + | } |
| + | |
| + | if (!is_append && offset != 0) { |
| + | /* warning, allows seek past end of file! Check for seek > size? */ |
| + | xlseek(new_file_fd, offset, SEEK_SET); |
| + | } |
| + | |
| + | if (is_unique) { |
| + | char *resp = xasprintf("FILE: %s", template); |
| + | remote_fd = get_remote_transfer_fd(resp); |
| + | free(resp); |
| + | free(template); |
| + | } else |
| + | remote_fd = get_remote_transfer_fd("Ok to send data"); |
| + | |
| + | if (remote_fd < 0) |
| + | goto bail; |
| + | |
| + | trans_ret = bb_copyfd_eof(remote_fd, new_file_fd); |
| + | ftpdataio_dispose_transfer_fd(); |
| + | |
| + | if (trans_ret < 0) |
| + | cmdio_write(FTP_BADSENDFILE, "Failure writing to local file"); |
| + | else |
| + | cmdio_write(FTP_TRANSFEROK, "File receive OK"); |
| + | |
| + | bail: |
| + | port_cleanup(); |
| + | pasv_cleanup(); |
| + | close(new_file_fd); |
| + | } |
| + | |
| + | static void |
| + | handle_stor(void) |
| + | { |
| + | handle_upload_common(0, 0); |
| + | } |
| + | |
| + | static void |
| + | handle_mkd(void) |
| + | { |
| + | int retval; |
| + | |
| + | /* Do we need check if ftp_arg != NULL? */ |
| + | retval = mkdir(G.ftp_arg, 0770); |
| + | if (retval != 0) { |
| + | cmdio_write(FTP_FILEFAIL, "Create directory operation failed"); |
| + | return; |
| + | } |
| + | |
| + | cmdio_write(FTP_MKDIROK, "created"); |
| + | } |
| + | |
| + | static void |
| + | handle_rmd(void) |
| + | { |
| + | int retval; |
| + | |
| + | /* Do we need check if ftp_arg != NULL? */ |
| + | retval = rmdir(G.ftp_arg); |
| + | if (retval != 0) |
| + | cmdio_write(FTP_FILEFAIL, "rmdir failed"); |
| + | else |
| + | cmdio_write(FTP_RMDIROK, "rmdir successful"); |
| + | } |
| + | |
| + | static void |
| + | handle_dele(void) |
| + | { |
| + | int retval; |
| + | |
| + | /* Do we need check if ftp_arg != NULL? */ |
| + | retval = unlink(G.ftp_arg); |
| + | if (retval != 0) |
| + | cmdio_write(FTP_FILEFAIL, "Delete failed"); |
| + | else |
| + | cmdio_write(FTP_DELEOK, "Delete successful"); |
| + | } |
| + | #endif /* ENABLE_FEATURE_FTP_WRITE */ |
| + | |
| + | static void |
| + | handle_rest(void) |
| + | { |
| + | /* When ftp_arg == NULL simply restart from beginning */ |
| + | G.restart_pos = xatoi_u(G.ftp_arg); |
| + | cmdio_write(FTP_RESTOK, "Restart OK"); |
| + | } |
| + | |
| + | #if ENABLE_FEATURE_FTP_WRITE |
| + | static void |
| + | handle_rnfr(void) |
| + | { |
| + | struct stat statbuf; |
| + | int retval; |
| + | |
| + | /* Clear old value */ |
| + | free(G.rnfr_filename); |
| + | |
| + | /* Does it exist? Do we need check if ftp_arg != NULL? */ |
| + | retval = stat(G.ftp_arg, &statbuf); |
| + | if (retval == 0) { |
| + | /* Yes */ |
| + | G.rnfr_filename = xstrdup(G.ftp_arg); |
| + | cmdio_write(FTP_RNFROK, "Ready for RNTO"); |
| + | } else |
| + | cmdio_write(FTP_FILEFAIL, "RNFR command failed"); |
| + | } |
| + | |
| + | static void |
| + | handle_rnto(void) |
| + | { |
| + | int retval; |
| + | |
| + | /* If we didn't get a RNFR, throw a wobbly */ |
| + | if (G.rnfr_filename == NULL) { |
| + | cmdio_write(FTP_NEEDRNFR, "RNFR required first"); |
| + | return; |
| + | } |
| + | |
| + | /* XXX Do we need check if ftp_arg != NULL? */ |
| + | retval = rename(G.rnfr_filename, G.ftp_arg); |
| + | |
| + | free(G.rnfr_filename); |
| + | |
| + | if (retval == 0) |
| + | cmdio_write(FTP_RENAMEOK, "Rename successful"); |
| + | else |
| + | cmdio_write(FTP_FILEFAIL, "Rename failed"); |
| + | } |
| + | #endif /* ENABLE_FEATURE_FTP_WRITE */ |
| + | |
| + | static void |
| + | handle_nlst(void) |
| + | { |
| + | handle_dir_common(0, 0); |
| + | } |
| + | |
| + | #if ENABLE_FEATURE_FTP_WRITE |
| + | static void |
| + | handle_appe(void) |
| + | { |
| + | handle_upload_common(1, 0); |
| + | } |
| + | #endif |
| + | |
| + | static void |
| + | handle_help(void) |
| + | { |
| + | cmdio_write_hyphen(FTP_HELP, "Recognized commands:"); |
| + | cmdio_write_raw(" ALLO CDUP CWD HELP LIST\r\n" |
| + | " MODE NLST NOOP PASS PASV PORT PWD QUIT\r\n" |
| + | " REST RETR STAT STRU SYST TYPE USER\r\n" |
| + | #if ENABLE_FEATURE_FTP_WRITE |
| + | " APPE DELE MKD RMD RNFR RNTO STOR STOU\r\n" |
| + | #endif |
| + | ); |
| + | cmdio_write(FTP_HELP, "Help OK"); |
| + | } |
| + | |
| + | #if ENABLE_FEATURE_FTP_WRITE |
| + | static void |
| + | handle_stou(void) |
| + | { |
| + | handle_upload_common(0, 1); |
| + | } |
| + | #endif |
| + | |
| + | static void |
| + | handle_stat(void) |
| + | { |
| + | cmdio_write_hyphen(FTP_STATOK, "FTP server status:"); |
| + | cmdio_write_raw(" TYPE: BINARY\r\n"); |
| + | cmdio_write(FTP_STATOK, "End of status"); |
| + | } |
| + | |
| + | static void |
| + | handle_stat_file(void) |
| + | { |
| + | handle_dir_common(1, 1); |
| + | } |
| + | |
| + | /* TODO: libbb candidate (tftp has another copy) */ |
| + | static len_and_sockaddr *get_sock_lsa(int s) |
| + | { |
| + | len_and_sockaddr *lsa; |
| + | socklen_t len = 0; |
| + | |
| + | if (getsockname(s, NULL, &len) != 0) |
| + | return NULL; |
| + | lsa = xzalloc(LSA_LEN_SIZE + len); |
| + | lsa->len = len; |
| + | getsockname(s, &lsa->u.sa, &lsa->len); |
| + | return lsa; |
| + | } |
| + | |
| + | int ftpd_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; |
| + | int ftpd_main(int argc UNUSED_PARAM, char **argv) |
| + | { |
| + | smallint user_was_specified = 0; |
| + | |
| + | INIT_G(); |
| + | |
| + | G.local_addr = get_sock_lsa(STDIN_FILENO); |
| + | if (!G.local_addr) { |
| + | /* This is confusing: |
| + | * bb_error_msg_and_die("stdin is not a socket"); |
| + | * Better: */ |
| + | bb_show_usage(); |
| + | /* Help text says that ftpd must be used as inetd service, |
| + | * which is by far the most usual cause of get_sock_lsa |
| + | * failure */ |
| + | } |
| + | |
| + | logmode = LOGMODE_SYSLOG; |
| + | |
| + | USE_FEATURE_FTP_WRITE(G.write_enable =) getopt32(argv, "" USE_FEATURE_FTP_WRITE("w")); |
| + | if (argv[optind]) { |
| + | xchdir(argv[optind]); |
| + | chroot("."); |
| + | } |
| + | |
| + | // if (G.local_addr->u.sa.sa_family != AF_INET) |
| + | // bb_error_msg_and_die("Only IPv4 is supported"); |
| + | |
| + | /* Signals. We'll always take -EPIPE rather than a rude signal, thanks */ |
| + | signal(SIGPIPE, SIG_IGN); |
| + | |
| + | /* Set up options on the command socket */ |
| + | setsockopt(STDIN_FILENO, IPPROTO_TCP, TCP_NODELAY, &const_int_1, sizeof(const_int_1)); |
| + | setsockopt(STDIN_FILENO, SOL_SOCKET, SO_KEEPALIVE, &const_int_1, sizeof(const_int_1)); |
| + | setsockopt(STDIN_FILENO, SOL_SOCKET, SO_OOBINLINE, &const_int_1, sizeof(const_int_1)); |
| + | |
| + | cmdio_write(FTP_GREET, "Welcome"); |
| + | |
| + | while (1) { |
| + | uint32_t cmdval = cmdio_get_cmd_and_arg(); |
| + | |
| + | if (cmdval == const_USER) { |
| + | if (G.ftp_arg == NULL || strcasecmp(G.ftp_arg, "anonymous") != 0) |
| + | cmdio_write(FTP_LOGINERR, "Server is anonymous only"); |
| + | else { |
| + | user_was_specified = 1; |
| + | cmdio_write(FTP_GIVEPWORD, "Please specify the password"); |
| + | } |
| + | } else if (cmdval == const_PASS) { |
| + | if (user_was_specified) |
| + | break; |
| + | cmdio_write(FTP_NEEDUSER, "Login with USER"); |
| + | } else if (cmdval == const_QUIT) { |
| + | cmdio_write(FTP_GOODBYE, "Goodbye"); |
| + | return 0; |
| + | } else { |
| + | cmdio_write(FTP_LOGINERR, |
| + | "Login with USER and PASS"); |
| + | } |
| + | } |
| + | |
| + | umask(077); |
| + | cmdio_write(FTP_LOGINOK, "Login successful"); |
| + | |
| + | while (1) { |
| + | uint32_t cmdval = cmdio_get_cmd_and_arg(); |
| + | |
| + | if (cmdval == const_QUIT) { |
| + | cmdio_write(FTP_GOODBYE, "Goodbye"); |
| + | return 0; |
| + | } |
| + | if (cmdval == const_PWD) |
| + | handle_pwd(); |
| + | else if (cmdval == const_CWD) |
| + | handle_cwd(); |
| + | else if (cmdval == const_CDUP) |
| + | handle_cdup(); |
| + | else if (cmdval == const_PASV) |
| + | handle_pasv(); |
| + | else if (cmdval == const_RETR) |
| + | handle_retr(); |
| + | else if (cmdval == const_NOOP) |
| + | cmdio_write(FTP_NOOPOK, "NOOP ok"); |
| + | else if (cmdval == const_SYST) |
| + | cmdio_write(FTP_SYSTOK, "UNIX Type: L8"); |
| + | else if (cmdval == const_HELP) |
| + | handle_help(); |
| + | else if (cmdval == const_LIST) |
| + | handle_list(); |
| + | else if (cmdval == const_TYPE) |
| + | handle_type(); |
| + | else if (cmdval == const_PORT) |
| + | handle_port(); |
| + | else if (cmdval == const_REST) |
| + | handle_rest(); |
| + | else if (cmdval == const_NLST) |
| + | handle_nlst(); |
| + | #if ENABLE_FEATURE_FTP_WRITE |
| + | else if (G.write_enable) { |
| + | if (cmdval == const_STOR) |
| + | handle_stor(); |
| + | else if (cmdval == const_MKD) |
| + | handle_mkd(); |
| + | else if (cmdval == const_RMD) |
| + | handle_rmd(); |
| + | else if (cmdval == const_DELE) |
| + | handle_dele(); |
| + | else if (cmdval == const_RNFR) |
| + | handle_rnfr(); |
| + | else if (cmdval == const_RNTO) |
| + | handle_rnto(); |
| + | else if (cmdval == const_APPE) |
| + | handle_appe(); |
| + | else if (cmdval == const_STOU) |
| + | handle_stou(); |
| + | } |
| + | #endif |
| + | else if (cmdval == const_STRU) { |
| + | if (G.ftp_arg |
| + | && (G.ftp_arg[0] | 0x20) == 'f' |
| + | && G.ftp_arg[1] == '\0' |
| + | ) { |
| + | cmdio_write(FTP_STRUOK, "Structure set to F"); |
| + | } else |
| + | cmdio_write(FTP_BADSTRU, "Bad STRU command"); |
| + | |
| + | } else if (cmdval == const_MODE) { |
| + | if (G.ftp_arg |
| + | && (G.ftp_arg[0] | 0x20) == 's' |
| + | && G.ftp_arg[1] == '\0' |
| + | ) { |
| + | cmdio_write(FTP_MODEOK, "Mode set to S"); |
| + | } else |
| + | cmdio_write(FTP_BADMODE, "Bad MODE command"); |
| + | } |
| + | else if (cmdval == const_ALLO) |
| + | cmdio_write(FTP_ALLOOK, "ALLO command ignored"); |
| + | else if (cmdval == const_STAT) { |
| + | if (G.ftp_arg == NULL) |
| + | handle_stat(); |
| + | else |
| + | handle_stat_file(); |
| + | } else if (cmdval == const_USER) |
| + | cmdio_write(FTP_LOGINERR, "Can't change to another user"); |
| + | else if (cmdval == const_PASS) |
| + | cmdio_write(FTP_LOGINOK, "Already logged in"); |
| + | else if (cmdval == const_STOR |
| + | || cmdval == const_MKD |
| + | || cmdval == const_RMD |
| + | || cmdval == const_DELE |
| + | || cmdval == const_RNFR |
| + | || cmdval == const_RNTO |
| + | || cmdval == const_APPE |
| + | || cmdval == const_STOU |
| + | ) { |
| + | cmdio_write(FTP_NOPERM, "Permission denied"); |
| + | } else { |
| + | cmdio_write(FTP_BADCMD, "Unknown command"); |
| + | } |
| + | } |
| + | } |