8 #include <sys/socket.h>
9 #include <sys/select.h>
11 #include <arpa/inet.h>
12 #include <netinet/in.h>
21 #define closesocket close
22 #define fcntlsocket fcntl
24 #define select select_s
27 #define O_NONBLOCK 0x2000
37 static void free_dir(struct ftp_dirent *dir);
39 static int newconn(struct ftp *ftp);
40 static int sendcmd(struct ftp *ftp, const char *fmt, ...);
41 static int handle_control(struct ftp *ftp);
42 static int handle_data(struct ftp *ftp, int s);
43 static void proc_control(struct ftp *ftp, const char *buf);
45 static int cproc_active(struct ftp *ftp, int code, const char *buf, void *cls);
46 static int cproc_pasv(struct ftp *ftp, int code, const char *buf, void *cls);
47 static int cproc_pwd(struct ftp *ftp, int code, const char *buf, void *cls);
48 static int cproc_cwd(struct ftp *ftp, int code, const char *buf, void *cls);
50 static int cproc_list(struct ftp *ftp, int code, const char *buf, void *cls);
51 static void dproc_list(struct ftp *ftp, const char *buf, int sz, void *cls);
53 static int cproc_xfer(struct ftp *ftp, int code, const char *buf, void *cls);
54 static void dproc_xfer(struct ftp *ftp, const char *buf, int sz, void *cls);
57 struct ftp *ftp_alloc(void)
61 if(!(ftp = calloc(1, sizeof *ftp))) {
64 ftp->ctl = ftp->data = ftp->lis = -1;
67 ftp->dirent = darr_alloc(0, sizeof *ftp->dirent);
71 void ftp_free(struct ftp *ftp)
79 free_dir(ftp->dirent);
82 static void free_dir(struct ftp_dirent *dir)
85 for(i=0; i<darr_size(dir); i++) {
91 int ftp_connect(struct ftp *ftp, const char *hostname, int port)
94 struct sockaddr_in addr;
96 if((ftp->ctl = socket(PF_INET, SOCK_STREAM, 0)) == -1) {
97 errmsg("failed to create control socket\n");
101 if(!(host = gethostbyname(hostname))) {
102 errmsg("failed to resolve host: %s\n", hostname);
103 closesocket(ftp->ctl);
108 memset(&addr, 0, sizeof addr);
109 addr.sin_family = AF_INET;
110 addr.sin_addr = *((struct in_addr*)host->h_addr);
111 addr.sin_port = htons(port);
113 if(connect(ftp->ctl, (struct sockaddr*)&addr, sizeof addr) == -1) {
114 errmsg("failed to connect to: %s (port: %d)\n", hostname, port);
115 closesocket(ftp->ctl);
120 fcntlsocket(ftp->ctl, F_SETFL, fcntlsocket(ftp->ctl, F_GETFL) | O_NONBLOCK);
125 void ftp_close(struct ftp *ftp)
128 closesocket(ftp->ctl);
133 closesocket(ftp->lis);
137 closesocket(ftp->data);
142 int ftp_sockets(struct ftp *ftp, int *sockv, int maxsize)
145 if(ftp->ctl >= 0 && maxsize-- > 0) {
148 if(ftp->lis >= 0 && maxsize-- > 0) {
151 if(ftp->data >= 0 && maxsize-- > 0) {
157 static void exec_op(struct ftp *ftp, struct ftp_op *fop)
161 ftp_type(ftp, fop->arg);
169 ftp_chdir(ftp, fop->arg);
173 ftp_chdir(ftp, "..");
177 ftp_mkdir(ftp, fop->arg);
181 ftp_rmdir(ftp, fop->arg);
185 ftp_delete(ftp, fop->arg);
193 ftp_retrieve(ftp, fop->arg);
197 ftp_store(ftp, fop->arg);
201 ftp_transfer(ftp, fop->data);
209 static void exec_queued(struct ftp *ftp)
213 if(!(fop = ftp->qhead)) {
217 if(ftp->qtail == fop) {
218 ftp->qhead = ftp->qtail = 0;
220 ftp->qhead = ftp->qhead->next;
230 int ftp_queue(struct ftp *ftp, int op, const char *arg)
234 if(!ftp->busy && !ftp->qhead) {
235 struct ftp_op tmp = {0};
237 tmp.arg = (char*)arg;
242 if(!(fop = malloc(sizeof *fop))) {
246 if(!(fop->arg = strdup(arg))) {
257 ftp->qtail->next = fop;
260 ftp->qhead = ftp->qtail = fop;
265 int ftp_queue_transfer(struct ftp *ftp, struct ftp_transfer *xfer)
269 if(!ftp->busy && !ftp->qhead) {
270 struct ftp_op tmp = {0};
277 if(!(fop = malloc(sizeof *fop))) {
286 ftp->qtail->next = fop;
289 ftp->qhead = ftp->qtail = fop;
294 int ftp_waitresp(struct ftp *ftp, long timeout)
305 FD_SET(ftp->ctl, &rdset);
312 if(select(ftp->ctl + 1, &rdset, 0, 0, timeout >= 0 ? &tv : 0) == -1 && errno == EINTR) {
316 if(FD_ISSET(ftp->ctl, &rdset)) {
318 ftp_handle(ftp, ftp->ctl);
325 timeout -= time(0) - start;
332 return ftp->last_resp;
335 static int ftp_active(struct ftp *ftp)
337 struct sockaddr_in sa = {0};
343 closesocket(ftp->lis);
346 closesocket(ftp->data);
350 if((ftp->lis = socket(PF_INET, SOCK_STREAM, 0)) == -1) {
351 errmsg("ftp_active: failed to create listening socket\n");
356 getsockname(ftp->ctl, (struct sockaddr*)&sa, &len);
358 sa.sin_family = AF_INET;
361 if(bind(ftp->lis, (struct sockaddr*)&sa, sizeof sa) == -1) {
362 errmsg("ftp_active: failed to bind listening socket\n");
363 closesocket(ftp->lis);
370 if(getsockname(ftp->lis, (struct sockaddr*)&sa, &len) == -1) {
371 errmsg("ftp_active: failed to retrieve listening socket address\n");
372 closesocket(ftp->lis);
377 addr = ntohl(sa.sin_addr.s_addr);
378 port = ntohs(sa.sin_port);
379 infomsg("listening on %s port %d\n", inet_ntoa(sa.sin_addr), port);
381 sendcmd(ftp, "PORT %d,%d,%d,%d,%d,%d", addr >> 24, (addr >> 16) & 0xff,
382 (addr >> 8) & 0xff, addr & 0xff, port >> 8, port & 0xff);
384 ftp->cproc = cproc_active;
388 static int ftp_passive(struct ftp *ftp)
391 closesocket(ftp->lis);
395 closesocket(ftp->data);
399 sendcmd(ftp, "PASV");
400 ftp->cproc = cproc_pasv;
404 int ftp_handle(struct ftp *ftp, int s)
407 return handle_control(ftp);
410 return handle_data(ftp, s);
413 int ns = accept(s, 0, 0);
424 static int sendcmd(struct ftp *ftp, const char *fmt, ...)
436 vsprintf(buf, fmt, ap);
438 infomsg("send: %s\n", buf);
440 return send(ftp->ctl, buf, strlen(buf), 0);
443 static int handle_control(struct ftp *ftp)
446 char *buf, *start, *end;
449 if((sz = sizeof ftp->crecv - ftp->num_crecv) <= 0) {
451 warnmsg("discard buffer\n");
452 sz = sizeof ftp->crecv;
455 start = ftp->crecv + ftp->num_crecv;
456 if((rd = recv(ftp->ctl, start, sz, 0)) == -1) {
457 if(errno == EINTR) continue;
458 /* assume EWOULDBLOCK, try again next time */
468 for(i=0; i<rd; i++) {
469 if(start[i] == '\r') {
471 } else if(start[i] == '\n') {
473 proc_control(ftp, buf);
477 if(buf != ftp->crecv && buf < end) {
478 ftp->num_crecv = end - buf;
479 memmove(ftp->crecv, buf, ftp->num_crecv);
487 static int handle_data(struct ftp *ftp, int s)
491 if(ftp->data == -1) {
495 /* get at most 4 packets at a time, to allow returning back to the main loop
499 if((rd = recv(ftp->data, ftp->drecv, sizeof ftp->drecv, 0)) == -1) {
500 if(errno == EINTR) continue;
501 /* assume EWOULDBLOCK, try again next time */
505 /* call the callback first, so that we'll get a 0-count call to indicate
506 * EOF when the server closes the data connection
509 ftp->dproc(ftp, ftp->drecv, rd, ftp->dproc_cls);
513 closesocket(ftp->data);
521 static int respcode(const char *resp)
523 if(!isdigit(resp[0]) || !isdigit(resp[1]) || !isdigit(resp[2])) {
527 if(isspace(resp[3])) {
536 static void proc_control(struct ftp *ftp, const char *buf)
541 while(*buf && isspace(*buf)) buf++;
542 if((end = strchr(buf, '\r'))) {
546 infomsg("recv: %s\n", buf);
548 if((code = respcode(buf)) == 0) {
549 warnmsg("ignoring invalid response: %s\n", buf);
553 return; /* ignore continuations for now */
556 ftp->last_resp = code;
559 if(ftp->cproc(ftp, code, buf, ftp->cproc_cls) <= 0) {
563 /* execute next operation if there's one queued */
572 sendcmd(ftp, "user %s", ftp->user ? ftp->user : "anonymous");
575 sendcmd(ftp, "pass %s", ftp->pass ? ftp->pass : "foobar");
578 infomsg("login successful\n");
579 if(newconn(ftp) == -1) {
585 errmsg("login failed\n");
591 /* execute next operation if there's one queued */
596 static int newconn(struct ftp *ftp)
598 ftp_queue(ftp, FTP_TYPE, "I");
599 ftp_queue(ftp, FTP_PWD, 0);
600 ftp_queue(ftp, FTP_LIST, 0);
604 int ftp_update(struct ftp *ftp)
609 int ftp_type(struct ftp *ftp, const char *type)
611 int tc = toupper(type[0]);
612 if(tc != 'I' && tc != 'A') {
615 sendcmd(ftp, "type %c", tc);
619 int ftp_pwd(struct ftp *ftp)
622 ftp->cproc = cproc_pwd;
626 int ftp_chdir(struct ftp *ftp, const char *dirname)
628 if(strcmp(dirname, "..") == 0) {
629 sendcmd(ftp, "cdup");
631 sendcmd(ftp, "cwd %s", dirname);
633 ftp->cproc = cproc_cwd;
637 int ftp_mkdir(struct ftp *ftp, const char *dirname)
642 int ftp_rmdir(struct ftp *ftp, const char *dirname)
647 int ftp_delete(struct ftp *ftp, const char *fname)
652 static int prepare_data(struct ftp *ftp)
660 ftp_waitresp(ftp, TIMEOUT);
662 if(ftp_active(ftp) == -1) {
666 if(ftp->data == -1) {
677 int ftp_list(struct ftp *ftp)
679 struct recvbuf *rbuf;
681 if(prepare_data(ftp) == -1) {
685 if(!(rbuf = malloc(sizeof *rbuf))) {
686 errmsg("failed to allocate receive buffer\n");
691 if(!(rbuf->buf = malloc(rbuf->bufsz))) {
693 errmsg("failed to allocate receive buffer\n");
697 sendcmd(ftp, "list");
698 ftp->cproc = cproc_list;
699 ftp->dproc = dproc_list;
700 ftp->cproc_cls = ftp->dproc_cls = rbuf;
704 int ftp_retrieve(struct ftp *ftp, const char *fname)
709 int ftp_store(struct ftp *ftp, const char *fname)
714 #define XFER_BUF_SIZE 1024
716 int ftp_transfer(struct ftp *ftp, struct ftp_transfer *xfer)
718 if(prepare_data(ftp) == -1) {
722 if(xfer->op == FTP_RETR) {
723 sendcmd(ftp, "retr %s", xfer->rname);
724 ftp->cproc = cproc_xfer;
725 ftp->dproc = dproc_xfer;
726 ftp->cproc_cls = ftp->dproc_cls = xfer;
733 static int get_quoted_text(const char *str, char *buf)
736 const char *src, *end;
738 if(!(src = strchr(str, '"'))) {
743 while(*end && *end != '"') end++;
747 memcpy(buf, src, len);
752 static int cproc_active(struct ftp *ftp, int code, const char *buf, void *cls)
755 errmsg("ftp_active failed\n");
758 ftp->status = FTP_CONN_ACT;
763 static int cproc_pasv(struct ftp *ftp, int code, const char *buf, void *cls)
766 unsigned int addr[6];
767 struct sockaddr_in sa;
772 errmsg("ftp_passive failed\n");
778 if(sscanf(str, "%u,%u,%u,%u,%u,%u", addr, addr + 1, addr + 2, addr + 3,
779 addr + 4, addr + 5) == 6) {
784 if(!*str || (addr[0] | addr[1] | addr[2] | addr[3] | addr[4] | addr[5]) >= 256) {
785 errmsg("ftp_passive: failed to parse response: %s\n", buf);
788 port = (addr[4] << 8) | addr[5];
789 ipaddr = ((unsigned int)addr[0] << 24) | ((unsigned int)addr[1] << 16) |
790 ((unsigned int)addr[2] << 8) | addr[3];
792 if((ftp->data = socket(PF_INET, SOCK_STREAM, 0)) == -1) {
793 fprintf(stderr, "ftp_passive: failed to allocate socket\n");
797 infomsg("passive mode: %d.%d.%d.%d port %d\n", addr[0], addr[1], addr[2], addr[3],
800 sa.sin_family = AF_INET;
801 sa.sin_addr.s_addr = htonl(ipaddr);
802 sa.sin_port = htons(port);
804 if(connect(ftp->data, (struct sockaddr*)&sa, sizeof sa) == -1) {
805 errmsg("ftp_passive: failed to connect to %s:%p\n", inet_ntoa(sa.sin_addr), port);
806 closesocket(ftp->data);
811 ftp->status = FTP_CONN_PASV;
820 static int cproc_pwd(struct ftp *ftp, int code, const char *buf, void *cls)
825 warnmsg("pwd failed\n");
829 dirname = alloca(strlen(buf) + 1);
830 if(get_quoted_text(buf, dirname) == -1) {
831 warnmsg("pwd: invalid response: %s\n", buf);
836 ftp->curdir = strdup_nf(dirname);
837 ftp->modified = FTP_MOD_DIR;
841 static int cproc_cwd(struct ftp *ftp, int code, const char *buf, void *cls)
844 warnmsg("cwd failed\n");
848 ftp_queue(ftp, FTP_PWD, 0);
849 ftp_queue(ftp, FTP_LIST, 0);
853 static int cproc_list(struct ftp *ftp, int code, const char *buf, void *cls)
861 errmsg("failed to retrieve directory listing\n");
866 #define SKIP_FIELD(p) \
868 while(*(p) && *(p) != '\n' && !isspace(*(p))) (p)++; \
869 while(*(p) && *(p) != '\n' && isspace(*(p))) (p)++; \
872 static int parse_dirent(struct ftp_dirent *ent, const char *line)
875 const char *ptr = line;
878 if(!(end = strchr(line, '\r')) && !(end = strchr(line, '\n'))) {
885 ent->type = FTP_FILE;
888 SKIP_FIELD(ptr); /* skip mode */
889 SKIP_FIELD(ptr); /* skip links */
890 SKIP_FIELD(ptr); /* skip owner */
891 SKIP_FIELD(ptr); /* skip group */
893 if(ent->type == FTP_FILE) {
894 ent->size = atoi(ptr);
896 SKIP_FIELD(ptr); /* skip size */
897 SKIP_FIELD(ptr); /* skip month */
898 SKIP_FIELD(ptr); /* skip day */
899 SKIP_FIELD(ptr); /* skip year */
901 if(ptr >= end) return -1;
904 ent->name = malloc(len + 1);
905 memcpy(ent->name, ptr, len);
911 int ftp_direntcmp(const void *a, const void *b)
913 const struct ftp_dirent *da = a, *db = b;
915 if(da->type == db->type) {
916 return strcmp(da->name, db->name);
918 return da->type == FTP_DIR ? -1 : 1;
921 static void dproc_list(struct ftp *ftp, const char *buf, int sz, void *cls)
924 struct recvbuf *rbuf = cls;
927 /* EOF condition, we got the whole list, update directory entries */
928 char *ptr = rbuf->buf;
929 char *end = rbuf->buf + rbuf->size;
930 struct ftp_dirent ent;
932 darr_clear(ftp->dirent);
935 if(parse_dirent(&ent, ptr) != -1) {
936 darr_push(ftp->dirent, &ent);
938 while(ptr < end && *ptr != '\n' && *ptr != '\r') ptr++;
939 while(ptr < end && (*ptr == '\r' || *ptr == '\n')) ptr++;
941 ftp->modified |= FTP_MOD_DIR;
947 num = darr_size(ftp->dirent);
948 qsort(ftp->dirent, num, sizeof *ftp->dirent, ftp_direntcmp);
952 if(rbuf->size + sz > rbuf->bufsz) {
954 int newsz = rbuf->bufsz << 1;
956 if(!(tmp = realloc(rbuf->buf, newsz))) {
957 errmsg("failed to resize receive buffer\n");
964 memcpy(rbuf->buf + rbuf->size, buf, sz);
968 static int cproc_xfer(struct ftp *ftp, int code, const char *buf, void *cls)
971 struct ftp_transfer *xfer = cls;
975 if(code == 150 && (ptr = strchr(buf, '('))) {
976 /* update total size */
977 sscanf(ptr, "(%ld bytes)", &xfer->total);
983 errmsg("failed to retrieve file\n");
988 static void dproc_xfer(struct ftp *ftp, const char *buf, int sz, void *cls)
990 struct ftp_transfer *xfer = cls;
992 if(xfer->op == FTP_RETR) {
994 /* EOF condition, got the whole file */
1001 xfer->done(ftp, xfer);
1007 ((char*)buf)[sz] = 0;
1008 fwrite(buf, 1, sz, xfer->fp);
1010 } else if(xfer->mem) {
1011 int prevsz = darr_size(xfer->mem);
1012 darr_resize(xfer->mem, prevsz + sz);
1013 memcpy(xfer->mem + prevsz, buf, sz);
1017 if(xfer->count > xfer->total) {
1018 xfer->count = xfer->total;
1021 } else { /* FTP_STOR */
1026 const char *ftp_curdir(struct ftp *ftp)
1031 int ftp_num_dirent(struct ftp *ftp)
1033 return darr_size(ftp->dirent);
1036 struct ftp_dirent *ftp_dirent(struct ftp *ftp, int idx)
1038 return ftp->dirent + idx;