git.mutantstargoat.com Git - oftp/blob - src/ftp.c

   1 #include <stdio.h>
   2 #include <stdlib.h>
   3 #include <string.h>
   4 #include <stdarg.h>
   5 #include <errno.h>
   6 #include <ctype.h>
   7 #include <time.h>
   8 #include <sys/socket.h>
   9 #include <arpa/inet.h>
  10 #include <netinet/in.h>
  11 #include <netdb.h>
  12 #include "darray.h"
  13 #include "ftp.h"
  14 #include "util.h"
  15
  16 #ifdef __unix__
  17 #include <unistd.h>
  18 #include <fcntl.h>
  19 #define closesocket             close
  20 #define fcntlsocket             fcntl
  21 #endif
  22
  23 #define TIMEOUT 15
  24
  25 static int newconn(struct ftp *ftp);
  26 static int sendcmd(struct ftp *ftp, const char *fmt, ...);
  27 static int handle_control(struct ftp *ftp);
  28 static int handle_data(struct ftp *ftp, int s);
  29 static void proc_control(struct ftp *ftp, const char *buf);
  30
  31 static int cproc_active(struct ftp *ftp, int code, const char *buf, void *cls);
  32 static int cproc_pwd(struct ftp *ftp, int code, const char *buf, void *cls);
  33 static int cproc_cwd(struct ftp *ftp, int code, const char *buf, void *cls);
  34
  35 static int cproc_list(struct ftp *ftp, int code, const char *buf, void *cls);
  36 static void dproc_list(struct ftp *ftp, const char *buf, int sz, void *cls);
  37
  38
  39 struct ftp *ftp_alloc(void)
  40 {
  41         struct ftp *ftp;
  42
  43         if(!(ftp = calloc(1, sizeof *ftp))) {
  44                 return 0;
  45         }
  46         ftp->ctl = ftp->data = ftp->lis = -1;
  47         return ftp;
  48 }
  49
  50 void ftp_free(struct ftp *ftp)
  51 {
  52         if(!ftp) return;
  53
  54         if(ftp->ctl >= 0) {
  55                 ftp_close(ftp);
  56         }
  57 }
  58
  59 int ftp_connect(struct ftp *ftp, const char *hostname, int port)
  60 {
  61         struct hostent *host;
  62         struct sockaddr_in addr;
  63
  64         if((ftp->ctl = socket(PF_INET, SOCK_STREAM, 0)) == -1) {
  65                 errmsg("failed to create control socket\n");
  66                 return -1;
  67         }
  68
  69         if(!(host = gethostbyname(hostname))) {
  70                 errmsg("failed to resolve host: %s\n", hostname);
  71                 closesocket(ftp->ctl);
  72                 ftp->ctl = -1;
  73                 return -1;
  74         }
  75
  76         memset(&addr, 0, sizeof addr);
  77         addr.sin_family = AF_INET;
  78         addr.sin_addr = *((struct in_addr*)host->h_addr);
  79         addr.sin_port = htons(port);
  80
  81         if(connect(ftp->ctl, (struct sockaddr*)&addr, sizeof addr) == -1) {
  82                 errmsg("failed to connect to: %s (port: %d)\n", hostname, port);
  83                 closesocket(ftp->ctl);
  84                 ftp->ctl = -1;
  85                 return -1;
  86         }
  87
  88         fcntlsocket(ftp->ctl, F_SETFL, fcntlsocket(ftp->ctl, F_GETFL) | O_NONBLOCK);
  89         ftp->num_crecv = 0;
  90         return 0;
  91 }
  92
  93 void ftp_close(struct ftp *ftp)
  94 {
  95         if(ftp->ctl >= 0) {
  96                 closesocket(ftp->ctl);
  97                 ftp->ctl = -1;
  98                 ftp->num_crecv = 0;
  99         }
 100         if(ftp->lis >= 0) {
 101                 closesocket(ftp->lis);
 102                 ftp->lis = -1;
 103         }
 104         if(ftp->data >= 0) {
 105                 closesocket(ftp->data);
 106                 ftp->data = -1;
 107         }
 108 }
 109
 110 int ftp_sockets(struct ftp *ftp, int *sockv, int maxsize)
 111 {
 112         int *sptr = sockv;
 113         if(ftp->ctl >= 0 && maxsize-- > 0) {
 114                 *sptr++ = ftp->ctl;
 115         }
 116         if(ftp->lis >= 0 && maxsize-- > 0) {
 117                 *sptr++ = ftp->lis;
 118         }
 119         if(ftp->data >= 0 && maxsize-- > 0) {
 120                 *sptr++ = ftp->data;
 121         }
 122         return sptr - sockv;
 123 }
 124
 125 static void exec_op(struct ftp *ftp, int op, const char *arg)
 126 {
 127         switch(op) {
 128         case FTP_PWD:
 129                 ftp_pwd(ftp);
 130                 break;
 131
 132         case FTP_CHDIR:
 133                 ftp_chdir(ftp, arg);
 134                 break;
 135
 136         case FTP_MKDIR:
 137                 ftp_mkdir(ftp, arg);
 138                 break;
 139
 140         case FTP_RMDIR:
 141                 ftp_rmdir(ftp, arg);
 142                 break;
 143
 144         case FTP_DEL:
 145                 ftp_delete(ftp, arg);
 146                 break;
 147
 148         case FTP_LIST:
 149                 ftp_list(ftp);
 150                 break;
 151
 152         case FTP_RETR:
 153                 ftp_retrieve(ftp, arg);
 154                 break;
 155
 156         case FTP_STORE:
 157                 ftp_store(ftp, arg);
 158                 break;
 159
 160         default:
 161                 break;
 162         }
 163 }
 164
 165 static void exec_queued(struct ftp *ftp)
 166 {
 167         struct ftp_op *fop;
 168
 169         if(!(fop = ftp->qhead)) {
 170                 return;
 171         }
 172
 173         if(ftp->qtail == fop) {
 174                 ftp->qhead = ftp->qtail = 0;
 175         } else {
 176                 ftp->qhead = ftp->qhead->next;
 177         }
 178
 179         exec_op(ftp, fop->op, fop->arg);
 180
 181         free(fop->arg);
 182         free(fop);
 183 }
 184
 185
 186 int ftp_queue(struct ftp *ftp, int op, const char *arg)
 187 {
 188         struct ftp_op *fop;
 189
 190         if(!ftp->busy && !ftp->qhead) {
 191                 exec_op(ftp, op, arg);
 192                 return 0;
 193         }
 194
 195         if(!(fop = malloc(sizeof *fop))) {
 196                 return -1;
 197         }
 198         if(arg) {
 199                 if(!(fop->arg = strdup(arg))) {
 200                         free(fop);
 201                         return -1;
 202                 }
 203         } else {
 204                 fop->arg = 0;
 205         }
 206         fop->op = op;
 207         fop->next = 0;
 208
 209         if(ftp->qhead) {
 210                 ftp->qtail->next = fop;
 211                 ftp->qtail = fop;
 212         } else {
 213                 ftp->qhead = ftp->qtail = fop;
 214         }
 215         return 0;
 216 }
 217
 218 int ftp_waitresp(struct ftp *ftp, time_t timeout)
 219 {
 220         fd_set rdset;
 221         struct timeval tv;
 222         time_t start;
 223
 224         ftp->last_resp = -1;
 225         start = time(0);
 226
 227         for(;;) {
 228                 FD_ZERO(&rdset);
 229                 FD_SET(ftp->ctl, &rdset);
 230
 231                 if(timeout >= 0) {
 232                         tv.tv_sec = timeout;
 233                         tv.tv_usec = 0;
 234                 }
 235
 236                 if(select(ftp->ctl + 1, &rdset, 0, 0, timeout >= 0 ? &tv : 0) == -1 && errno == EINTR) {
 237                         continue;
 238                 }
 239
 240                 if(FD_ISSET(ftp->ctl, &rdset)) {
 241                         ftp->last_resp = -1;
 242                         ftp_handle(ftp, ftp->ctl);
 243                         if(ftp->last_resp) {
 244                                 break;
 245                         }
 246                 }
 247
 248                 if(timeout > 0) {
 249                         timeout -= time(0) - start;
 250                         if(timeout <= 0) {
 251                                 return -1;
 252                         }
 253                 }
 254
 255         }
 256         return ftp->last_resp;
 257 }
 258
 259 static int ftp_active(struct ftp *ftp)
 260 {
 261         struct sockaddr_in sa = {0};
 262         socklen_t len;
 263         unsigned long addr;
 264         unsigned short port;
 265
 266         if(ftp->lis >= 0) {
 267                 closesocket(ftp->lis);
 268         }
 269         if(ftp->data >= 0) {
 270                 closesocket(ftp->data);
 271                 ftp->data = -1;
 272         }
 273
 274         if((ftp->lis = socket(PF_INET, SOCK_STREAM, 0)) == -1) {
 275                 errmsg("ftp_active: failed to create listening socket\n");
 276                 return -1;
 277         }
 278
 279         len = sizeof sa;
 280         getsockname(ftp->ctl, (struct sockaddr*)&sa, &len);
 281
 282         sa.sin_family = AF_INET;
 283         sa.sin_port = 0;
 284
 285         if(bind(ftp->lis, (struct sockaddr*)&sa, sizeof sa) == -1) {
 286                 errmsg("ftp_active: failed to bind listening socket\n");
 287                 closesocket(ftp->lis);
 288                 ftp->lis = -1;
 289                 return -1;
 290         }
 291         listen(ftp->lis, 1);
 292
 293         len = sizeof sa;
 294         if(getsockname(ftp->lis, (struct sockaddr*)&sa, &len) == -1) {
 295                 errmsg("ftp_active: failed to retrieve listening socket address\n");
 296                 closesocket(ftp->lis);
 297                 ftp->lis = -1;
 298                 return -1;
 299         }
 300
 301         addr = ntohl(sa.sin_addr.s_addr);
 302         port = ntohs(sa.sin_port);
 303         infomsg("listening on %s port %d\n", inet_ntoa(sa.sin_addr), port);
 304
 305         sendcmd(ftp, "PORT %d,%d,%d,%d,%d,%d", addr >> 24, (addr >> 16) & 0xff,
 306                         (addr >> 8) & 0xff, addr & 0xff, port >> 8, port & 0xff);
 307
 308         ftp->cproc = cproc_active;
 309         return 0;
 310 }
 311
 312 int ftp_handle(struct ftp *ftp, int s)
 313 {
 314         if(s == ftp->ctl) {
 315                 return handle_control(ftp);
 316         }
 317         if(s == ftp->data) {
 318                 return handle_data(ftp, s);
 319         }
 320         if(s == ftp->lis) {
 321                 int ns = accept(s, 0, 0);
 322                 if(ftp->data >= 0) {
 323                         closesocket(ns);
 324                 } else {
 325                         ftp->data = ns;
 326                 }
 327                 return 0;
 328         }
 329         return -1;
 330 }
 331
 332 static int sendcmd(struct ftp *ftp, const char *fmt, ...)
 333 {
 334         char buf[256];
 335         va_list ap;
 336
 337         if(ftp->ctl < 0) {
 338                 return -1;
 339         }
 340
 341         ftp->busy = 1;
 342
 343         va_start(ap, fmt);
 344         vsprintf(buf, fmt, ap);
 345         va_end(ap);
 346         infomsg("send: %s\n", buf);
 347         strcat(buf, "\r\n");
 348         return send(ftp->ctl, buf, strlen(buf), 0);
 349 }
 350
 351 static int handle_control(struct ftp *ftp)
 352 {
 353         int i, sz, rd;
 354         char *buf, *start, *end;
 355
 356         for(;;) {
 357                 if((sz = sizeof ftp->crecv - ftp->num_crecv) <= 0) {
 358                         /* discard buffer */
 359                         warnmsg("discard buffer\n");
 360                         sz = sizeof ftp->crecv;
 361                         ftp->num_crecv = 0;
 362                 }
 363                 start = ftp->crecv + ftp->num_crecv;
 364                 if((rd = recv(ftp->ctl, start, sz, 0)) == -1) {
 365                         if(errno == EINTR) continue;
 366                         /* assume EWOULDBLOCK, try again next time */
 367                         return 0;
 368                 }
 369                 if(rd == 0) {
 370                         ftp_close(ftp);
 371                         return -1;
 372                 }
 373
 374                 end = start + rd;
 375                 buf = ftp->crecv;
 376                 for(i=0; i<rd; i++) {
 377                         if(start[i] == '\r') {
 378                                 start[i] = 0;
 379                         } else if(start[i] == '\n') {
 380                                 start[i] = 0;
 381                                 proc_control(ftp, buf);
 382                                 buf = start + i + 1;
 383                         }
 384                 }
 385                 if(buf != ftp->crecv && buf < end) {
 386                         ftp->num_crecv = end - buf;
 387                         memmove(ftp->crecv, buf, ftp->num_crecv);
 388                 } else {
 389                         ftp->num_crecv = 0;
 390                 }
 391         }
 392         return 0;
 393 }
 394
 395 static int handle_data(struct ftp *ftp, int s)
 396 {
 397         int rd;
 398
 399         if(ftp->data == -1) {
 400                 return -1;
 401         }
 402
 403         for(;;) {
 404                 if((rd = recv(ftp->data, ftp->drecv, sizeof ftp->drecv, 0)) == -1) {
 405                         if(errno == EINTR) continue;
 406                         /* assume EWOULDBLOCK, try again next time */
 407                         break;
 408                 }
 409
 410                 /* call the callback first, so that we'll get a 0-count call to indicate
 411                  * EOF when the server closes the data connection
 412                  */
 413                 if(ftp->dproc) {
 414                         ftp->dproc(ftp, ftp->drecv, rd, ftp->dproc_cls);
 415                 }
 416
 417                 if(rd == 0) {
 418                         closesocket(ftp->data);
 419                         ftp->data = -1;
 420                         return -1;
 421                 }
 422         }
 423         return 0;
 424 }
 425
 426 static int respcode(const char *resp)
 427 {
 428         if(!isdigit(resp[0]) || !isdigit(resp[1]) || !isdigit(resp[2])) {
 429                 return 0;
 430         }
 431
 432         if(isspace(resp[3])) {
 433                 return atoi(resp);
 434         }
 435         if(resp[3] == '-') {
 436                 return -atoi(resp);
 437         }
 438         return 0;
 439 }
 440
 441 static void proc_control(struct ftp *ftp, const char *buf)
 442 {
 443         int code;
 444         char *end;
 445
 446         while(*buf && isspace(*buf)) buf++;
 447         if((end = strchr(buf, '\r'))) {
 448                 *end = 0;
 449         }
 450
 451         infomsg("recv: %s\n", buf);
 452
 453         if((code = respcode(buf)) == 0) {
 454                 warnmsg("ignoring invalid response: %s\n", buf);
 455                 return;
 456         }
 457         if(code < 0) {
 458                 return; /* ignore continuations for now */
 459         }
 460
 461         ftp->last_resp = code;
 462
 463         if(ftp->cproc) {
 464                 if(ftp->cproc(ftp, code, buf, ftp->cproc_cls) <= 0) {
 465                         ftp->cproc = 0;
 466                         ftp->busy = 0;
 467
 468                         /* execute next operation if there's one queued */
 469                         exec_queued(ftp);
 470                 }
 471                 return;
 472         }
 473         ftp->busy = 0;
 474
 475         switch(code) {
 476         case 220:
 477                 sendcmd(ftp, "user %s", ftp->user ? ftp->user : "anonymous");
 478                 break;
 479         case 331:
 480                 sendcmd(ftp, "pass %s", ftp->pass ? ftp->pass : "foobar");
 481                 break;
 482         case 230:
 483                 infomsg("login successful\n");
 484                 if(newconn(ftp) == -1) {
 485                         ftp_close(ftp);
 486                 }
 487                 break;
 488         case 530:
 489                 ftp->status = 0;
 490                 errmsg("login failed\n");
 491                 break;
 492         }
 493 }
 494
 495 static int newconn(struct ftp *ftp)
 496 {
 497         if(ftp_active(ftp) == -1) {
 498                 return -1;
 499         }
 500         ftp_queue(ftp, FTP_PWD, 0);
 501         ftp_queue(ftp, FTP_LIST, 0);
 502         return 0;
 503 }
 504
 505 int ftp_update(struct ftp *ftp)
 506 {
 507         return -1;
 508 }
 509
 510 int ftp_pwd(struct ftp *ftp)
 511 {
 512         sendcmd(ftp, "pwd");
 513         ftp->cproc = cproc_pwd;
 514         return 0;
 515 }
 516
 517 int ftp_chdir(struct ftp *ftp, const char *dirname)
 518 {
 519         sendcmd(ftp, "cwd %s", dirname);
 520         ftp->cproc = cproc_cwd;
 521         return 0;
 522 }
 523
 524 int ftp_mkdir(struct ftp *ftp, const char *dirname)
 525 {
 526         return -1;
 527 }
 528
 529 int ftp_rmdir(struct ftp *ftp, const char *dirname)
 530 {
 531         return -1;
 532 }
 533
 534 int ftp_delete(struct ftp *ftp, const char *fname)
 535 {
 536         return -1;
 537 }
 538
 539 struct recvbuf {
 540         char *buf;
 541         long size, bufsz;
 542 };
 543
 544 int ftp_list(struct ftp *ftp)
 545 {
 546         struct recvbuf *rbuf;
 547
 548         if(!(rbuf = malloc(sizeof *rbuf))) {
 549                 errmsg("failed to allocate receive buffer\n");
 550                 return -1;
 551         }
 552         rbuf->size = 0;
 553         rbuf->bufsz = 1024;
 554         if(!(rbuf->buf = malloc(rbuf->bufsz))) {
 555                 free(rbuf);
 556                 errmsg("failed to allocate receive buffer\n");
 557                 return -1;
 558         }
 559
 560         sendcmd(ftp, "list");
 561         ftp->cproc = cproc_list;
 562         ftp->dproc = dproc_list;
 563         ftp->cproc_cls = ftp->dproc_cls = rbuf;
 564         return 0;
 565 }
 566
 567 int ftp_retrieve(struct ftp *ftp, const char *fname)
 568 {
 569         return -1;
 570 }
 571
 572 int ftp_store(struct ftp *ftp, const char *fname)
 573 {
 574         return -1;
 575 }
 576
 577 static int get_quoted_text(const char *str, char *buf)
 578 {
 579         int len;
 580         const char *src, *end;
 581
 582         if(!(src = strchr(str, '"'))) {
 583                 return -1;
 584         }
 585         src++;
 586         end = src;
 587         while(*end && *end != '"') end++;
 588         if(!*end) return -1;
 589
 590         len = end - src;
 591         memcpy(buf, src, len);
 592         buf[len] = 0;
 593         return 0;
 594 }
 595
 596 static int cproc_active(struct ftp *ftp, int code, const char *buf, void *cls)
 597 {
 598         if(code != 200) {
 599                 errmsg("ftp_active failed\n");
 600                 ftp_close(ftp);
 601         } else {
 602                 ftp->status = FTP_CONN_ACT;
 603         }
 604         return 0;
 605 }
 606
 607 static int cproc_pwd(struct ftp *ftp, int code, const char *buf, void *cls)
 608 {
 609         char *dirname;
 610
 611         if(code != 257) {
 612                 warnmsg("pwd failed\n");
 613                 return -1;
 614         }
 615
 616         dirname = alloca(strlen(buf) + 1);
 617         if(get_quoted_text(buf, dirname) == -1) {
 618                 warnmsg("pwd: invalid response: %s\n", buf);
 619                 return -1;
 620         }
 621
 622         free(ftp->curdir_rem);
 623         ftp->curdir_rem = strdup_nf(dirname);
 624         ftp->modified = 1;
 625         return 0;
 626 }
 627
 628 static int cproc_cwd(struct ftp *ftp, int code, const char *buf, void *cls)
 629 {
 630         return -1;
 631 }
 632
 633 static int cproc_list(struct ftp *ftp, int code, const char *buf, void *cls)
 634 {
 635         if(code < 200) {
 636                 /* expect more */
 637                 return 1;
 638         }
 639
 640         if(code >= 400) {
 641                 errmsg("failed to retrieve directory listing\n");
 642         }
 643         return 0;
 644 }
 645
 646 static void free_dirlist(struct ftp_dirent *list)
 647 {
 648         struct ftp_dirent *tmp;
 649
 650         while(list) {
 651                 tmp = list;
 652                 list = list->next;
 653
 654                 free(tmp->name);
 655                 free(tmp);
 656         }
 657 }
 658
 659 #define SKIP_FIELD(p) \
 660         do { \
 661                 while(*(p) && *(p) != '\n' && !isspace(*(p))) (p)++; \
 662                 while(*(p) && *(p) != '\n' && isspace(*(p))) (p)++; \
 663         } while(0)
 664
 665 static int parse_dirent(struct ftp_dirent *ent, const char *line)
 666 {
 667         int len;
 668         const char *ptr = line;
 669         const char *end;
 670
 671         if(!(end = strchr(line, '\r')) && !(end = strchr(line, '\n'))) {
 672                 return -1;
 673         }
 674
 675         if(line[0] == 'd') {
 676                 ent->type = FTP_DIR;
 677         } else {
 678                 ent->type = FTP_FILE;
 679         }
 680
 681         SKIP_FIELD(ptr);                /* skip mode */
 682         SKIP_FIELD(ptr);                /* skip links */
 683         SKIP_FIELD(ptr);                /* skip owner */
 684         SKIP_FIELD(ptr);                /* skip group */
 685
 686         if(ent->type == FTP_FILE) {
 687                 ent->size = atoi(ptr);
 688         }
 689         SKIP_FIELD(ptr);                /* skip size */
 690         SKIP_FIELD(ptr);                /* skip month */
 691         SKIP_FIELD(ptr);                /* skip day */
 692         SKIP_FIELD(ptr);                /* skip year */
 693
 694         if(ptr >= end) return -1;
 695
 696         len = end - ptr;
 697         ent->name = malloc(len + 1);
 698         memcpy(ent->name, ptr, len);
 699         ent->name[len] = 0;
 700
 701         return 0;
 702 }
 703
 704 static void dproc_list(struct ftp *ftp, const char *buf, int sz, void *cls)
 705 {
 706         struct recvbuf *rbuf = cls;
 707
 708         if(sz == 0) {
 709                 /* EOF condition, we got the whole list, update directory entries */
 710                 char *ptr = rbuf->buf;
 711                 char *end = rbuf->buf + rbuf->size;
 712                 struct ftp_dirent *tail = 0;
 713                 struct ftp_dirent *ent;
 714
 715                 free_dirlist(ftp->dent_rem);
 716                 ftp->dent_rem = 0;
 717
 718                 while(ptr < end) {
 719                         ent = malloc_nf(sizeof *ent);
 720                         if(parse_dirent(ent, ptr) != -1) {
 721                                 ent->next = 0;
 722                                 if(!tail) {
 723                                         ftp->dent_rem = tail = ent;
 724                                 } else {
 725                                         tail->next = ent;
 726                                         tail = ent;
 727                                 }
 728                         }
 729                         while(ptr < end && *ptr != '\n' && *ptr != '\r') ptr++;
 730                         while(ptr < end && (*ptr == '\r' || *ptr == '\n')) ptr++;
 731                 }
 732                 ftp->modified |= FTP_MOD_REMDIR;
 733
 734                 free(rbuf->buf);
 735                 free(rbuf);
 736                 ftp->dproc = 0;
 737                 return;
 738         }
 739
 740         if(rbuf->size + sz > rbuf->bufsz) {
 741                 char *tmp;
 742                 int newsz = rbuf->bufsz << 1;
 743
 744                 if(!(tmp = realloc(rbuf->buf, newsz))) {
 745                         errmsg("failed to resize receive buffer\n");
 746                         return;
 747                 }
 748                 rbuf->buf = tmp;
 749                 rbuf->bufsz = newsz;
 750         }
 751
 752         memcpy(rbuf->buf + rbuf->size, buf, sz);
 753         rbuf->size += sz;
 754 }