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 5
  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->qhead) {
 191                 exec_op(ftp, op, arg);
 192                 return 0;
 193         }
 194
 195         if(!(fop = malloc(sizeof *fop))) {
 196                 return -1;
 197         }
 198         if(!(fop->arg = strdup(arg))) {
 199                 free(fop);
 200                 return -1;
 201         }
 202         fop->op = op;
 203         fop->next = 0;
 204
 205         ftp->qtail->next = fop;
 206         ftp->qtail = fop;
 207         return 0;
 208 }
 209
 210 int ftp_waitresp(struct ftp *ftp, time_t timeout)
 211 {
 212         fd_set rdset;
 213         struct timeval tv;
 214         time_t start;
 215
 216         ftp->last_resp = -1;
 217         start = time(0);
 218
 219         for(;;) {
 220                 FD_ZERO(&rdset);
 221                 FD_SET(ftp->ctl, &rdset);
 222
 223                 if(timeout >= 0) {
 224                         tv.tv_sec = timeout;
 225                         tv.tv_usec = 0;
 226                 }
 227
 228                 if(select(ftp->ctl + 1, &rdset, 0, 0, timeout >= 0 ? &tv : 0) == -1 && errno == EINTR) {
 229                         continue;
 230                 }
 231
 232                 if(FD_ISSET(ftp->ctl, &rdset)) {
 233                         ftp->last_resp = -1;
 234                         ftp_handle(ftp, ftp->ctl);
 235                         if(ftp->last_resp) {
 236                                 break;
 237                         }
 238                 }
 239
 240                 if(timeout > 0) {
 241                         timeout -= time(0) - start;
 242                         if(timeout <= 0) {
 243                                 return -1;
 244                         }
 245                 }
 246
 247         }
 248         return ftp->last_resp;
 249 }
 250
 251 static int ftp_active(struct ftp *ftp)
 252 {
 253         struct sockaddr_in sa = {0};
 254         socklen_t len;
 255         unsigned long addr;
 256         unsigned short port;
 257
 258         if(ftp->lis >= 0) {
 259                 closesocket(ftp->lis);
 260         }
 261         if(ftp->data >= 0) {
 262                 closesocket(ftp->data);
 263                 ftp->data = -1;
 264         }
 265
 266         if((ftp->lis = socket(PF_INET, SOCK_STREAM, 0)) == -1) {
 267                 errmsg("ftp_active: failed to create listening socket\n");
 268                 return -1;
 269         }
 270
 271         sa.sin_family = AF_INET;
 272         sa.sin_port = 0;
 273         sa.sin_addr.s_addr = htonl(INADDR_ANY);
 274
 275         if(bind(ftp->lis, (struct sockaddr*)&sa, sizeof sa) == -1) {
 276                 errmsg("ftp_active: failed to bind listening socket\n");
 277                 closesocket(ftp->lis);
 278                 ftp->lis = -1;
 279                 return -1;
 280         }
 281         listen(ftp->lis, 1);
 282
 283         len = sizeof sa;
 284         if(getsockname(ftp->ctl, (struct sockaddr*)&sa, &len) == -1) {
 285                 errmsg("ftp_active: failed to retrieve listening socket address\n");
 286                 closesocket(ftp->lis);
 287                 ftp->lis = -1;
 288                 return -1;
 289         }
 290
 291         addr = ntohl(sa.sin_addr.s_addr);
 292         port = ntohs(sa.sin_port);
 293
 294         sendcmd(ftp, "PORT %d,%d,%d,%d,%d,%d", addr >> 24, (addr >> 16) & 0xff,
 295                         (addr >> 8) & 0xff, addr & 0xff, port >> 8, port & 0xff);
 296
 297         ftp->cproc = cproc_active;
 298         return 0;
 299 }
 300
 301 int ftp_handle(struct ftp *ftp, int s)
 302 {
 303         if(s == ftp->ctl) {
 304                 return handle_control(ftp);
 305         }
 306         if(s == ftp->data) {
 307                 return handle_data(ftp, s);
 308         }
 309         if(s == ftp->lis) {
 310                 int ns = accept(s, 0, 0);
 311                 if(ftp->data >= 0) {
 312                         closesocket(ns);
 313                 } else {
 314                         ftp->data = ns;
 315                 }
 316                 return 0;
 317         }
 318         return -1;
 319 }
 320
 321 static int sendcmd(struct ftp *ftp, const char *fmt, ...)
 322 {
 323         char buf[256];
 324         va_list ap;
 325
 326         if(ftp->ctl < 0) {
 327                 return -1;
 328         }
 329
 330         va_start(ap, fmt);
 331         vsprintf(buf, fmt, ap);
 332         va_end(ap);
 333         strcat(buf, "\r\n");
 334
 335         return send(ftp->ctl, buf, strlen(buf), 0);
 336 }
 337
 338 static int handle_control(struct ftp *ftp)
 339 {
 340         int i, sz, rd;
 341         char *buf, *start, *end;
 342
 343         while((sz = sizeof ftp->crecv - ftp->num_crecv) > 0) {
 344                 start = ftp->crecv + ftp->num_crecv;
 345                 if((rd = recv(ftp->ctl, start, sz, 0)) == -1) {
 346                         if(errno == EINTR) continue;
 347                         /* assume EWOULDBLOCK, try again next time */
 348                         return 0;
 349                 }
 350                 if(rd == 0) {
 351                         ftp_close(ftp);
 352                         return -1;
 353                 }
 354
 355                 end = start + rd;
 356                 buf = ftp->crecv;
 357                 for(i=0; i<rd; i++) {
 358                         if(start[i] == '\n') {
 359                                 start[i] = 0;
 360                                 proc_control(ftp, buf);
 361                                 buf = start + i + 1;
 362                         }
 363                 }
 364                 if(buf != ftp->crecv && buf < end) {
 365                         ftp->num_crecv = end - buf;
 366                         memmove(ftp->crecv, buf, ftp->num_crecv);
 367                 }
 368         }
 369         return 0;
 370 }
 371
 372 static int handle_data(struct ftp *ftp, int s)
 373 {
 374         int rd;
 375
 376         if(ftp->data == -1) {
 377                 return -1;
 378         }
 379
 380         for(;;) {
 381                 if((rd = recv(ftp->data, ftp->drecv, sizeof ftp->drecv, 0)) == -1) {
 382                         if(errno == EINTR) continue;
 383                         /* assume EWOULDBLOCK, try again next time */
 384                         break;
 385                 }
 386
 387                 /* call the callback first, so that we'll get a 0-count call to indicate
 388                  * EOF when the server closes the data connection
 389                  */
 390                 if(ftp->dproc) {
 391                         ftp->dproc(ftp, ftp->drecv, rd, ftp->dproc_cls);
 392                 }
 393
 394                 if(rd == 0) {
 395                         closesocket(ftp->data);
 396                         ftp->data = -1;
 397                         return -1;
 398                 }
 399         }
 400         return 0;
 401 }
 402
 403 static int respcode(const char *resp)
 404 {
 405         if(!isdigit(resp[0]) || !isdigit(resp[1]) || !isdigit(resp[2])) {
 406                 return 0;
 407         }
 408
 409         if(isspace(resp[3])) {
 410                 return atoi(resp);
 411         }
 412         if(resp[3] == '-') {
 413                 return -atoi(resp);
 414         }
 415         return 0;
 416 }
 417
 418 static void proc_control(struct ftp *ftp, const char *buf)
 419 {
 420         int code;
 421
 422         while(*buf && isspace(*buf)) buf++;
 423
 424         if((code = respcode(buf)) == 0) {
 425                 warnmsg("ignoring invalid response: %s\n", buf);
 426                 return;
 427         }
 428         if(code < 0) {
 429                 return; /* ignore continuations for now */
 430         }
 431
 432         ftp->last_resp = code;
 433
 434         if(ftp->cproc) {
 435                 if(ftp->cproc(ftp, code, buf, ftp->cproc_cls) <= 0) {
 436                         ftp->cproc = 0;
 437
 438                         /* execute next operation if there's one queued */
 439                         exec_queued(ftp);
 440                 }
 441                 return;
 442         }
 443
 444         switch(code) {
 445         case 220:
 446                 sendcmd(ftp, "user %s", ftp->user ? ftp->user : "anonymous");
 447                 break;
 448         case 331:
 449                 sendcmd(ftp, "pass %s", ftp->pass ? ftp->pass : "foobar");
 450                 break;
 451         case 230:
 452                 infomsg("login successful\n");
 453                 if(newconn(ftp) == -1) {
 454                         ftp_close(ftp);
 455                 }
 456                 break;
 457         case 530:
 458                 ftp->status = 0;
 459                 errmsg("login failed\n");
 460                 break;
 461         }
 462 }
 463
 464 static int newconn(struct ftp *ftp)
 465 {
 466         if(ftp_active(ftp) == -1) {
 467                 return -1;
 468         }
 469         ftp_queue(ftp, FTP_PWD, 0);
 470         ftp_queue(ftp, FTP_LIST, 0);
 471         return 0;
 472 }
 473
 474 int ftp_update(struct ftp *ftp)
 475 {
 476         return -1;
 477 }
 478
 479 int ftp_pwd(struct ftp *ftp)
 480 {
 481         sendcmd(ftp, "pwd");
 482         ftp->cproc = cproc_pwd;
 483         return 0;
 484 }
 485
 486 int ftp_chdir(struct ftp *ftp, const char *dirname)
 487 {
 488         sendcmd(ftp, "cwd %s", dirname);
 489         ftp->cproc = cproc_cwd;
 490         return 0;
 491 }
 492
 493 int ftp_mkdir(struct ftp *ftp, const char *dirname)
 494 {
 495         return -1;
 496 }
 497
 498 int ftp_rmdir(struct ftp *ftp, const char *dirname)
 499 {
 500         return -1;
 501 }
 502
 503 int ftp_delete(struct ftp *ftp, const char *fname)
 504 {
 505         return -1;
 506 }
 507
 508 struct recvbuf {
 509         char *buf;
 510         long size, bufsz;
 511 };
 512
 513 int ftp_list(struct ftp *ftp)
 514 {
 515         struct recvbuf *rbuf;
 516
 517         if(!(rbuf = malloc(sizeof *rbuf))) {
 518                 errmsg("failed to allocate receive buffer\n");
 519                 return -1;
 520         }
 521         rbuf->size = 0;
 522         rbuf->bufsz = 1024;
 523         if(!(rbuf->buf = malloc(rbuf->bufsz))) {
 524                 free(rbuf);
 525                 errmsg("failed to allocate receive buffer\n");
 526                 return -1;
 527         }
 528
 529         sendcmd(ftp, "list");
 530         ftp->cproc = cproc_list;
 531         ftp->dproc = dproc_list;
 532         ftp->cproc_cls = ftp->dproc_cls = rbuf;
 533         return 0;
 534 }
 535
 536 int ftp_retrieve(struct ftp *ftp, const char *fname)
 537 {
 538         return -1;
 539 }
 540
 541 int ftp_store(struct ftp *ftp, const char *fname)
 542 {
 543         return -1;
 544 }
 545
 546 static int get_quoted_text(const char *str, char *buf)
 547 {
 548         int len;
 549         const char *src, *end;
 550
 551         if(!(src = strchr(str, '"'))) {
 552                 return -1;
 553         }
 554         src++;
 555         end = src;
 556         while(*end && *end != '"') end++;
 557         if(!*end) return -1;
 558
 559         len = end - src;
 560         memcpy(buf, src, len);
 561         buf[len] = 0;
 562         return 0;
 563 }
 564
 565 static int cproc_active(struct ftp *ftp, int code, const char *buf, void *cls)
 566 {
 567         if(code != 200) {
 568                 errmsg("ftp_active failed\n");
 569                 ftp_close(ftp);
 570         } else {
 571                 ftp->status = FTP_CONN_ACT;
 572         }
 573         return 0;
 574 }
 575
 576 static int cproc_pwd(struct ftp *ftp, int code, const char *buf, void *cls)
 577 {
 578         char *dirname;
 579
 580         if(code != 257) {
 581                 warnmsg("pwd failed\n");
 582                 return -1;
 583         }
 584
 585         dirname = alloca(strlen(buf) + 1);
 586         if(get_quoted_text(buf, dirname) == -1) {
 587                 warnmsg("pwd: invalid response: %s\n", buf);
 588                 return -1;
 589         }
 590
 591         free(ftp->curdir_rem);
 592         ftp->curdir_rem = strdup_nf(dirname);
 593         ftp->modified = 1;
 594         return 0;
 595 }
 596
 597 static int cproc_cwd(struct ftp *ftp, int code, const char *buf, void *cls)
 598 {
 599         return -1;
 600 }
 601
 602 static int cproc_list(struct ftp *ftp, int code, const char *buf, void *cls)
 603 {
 604         if(code < 200) {
 605                 /* expect more */
 606                 return 1;
 607         }
 608
 609         if(code >= 400) {
 610                 errmsg("failed to retrieve directory listing\n");
 611         }
 612         return 0;
 613 }
 614
 615 static void dproc_list(struct ftp *ftp, const char *buf, int sz, void *cls)
 616 {
 617         struct recvbuf *rbuf = cls;
 618
 619         if(sz == 0) {
 620                 /* EOF condition, we got the whole list, update directory entries */
 621                 /* TODO */
 622                 rbuf->buf[rbuf->size] = 0;
 623                 fprintf(stderr, "%s\n", rbuf->buf);
 624
 625                 free(rbuf->buf);
 626                 free(rbuf);
 627                 ftp->dproc = 0;
 628                 return;
 629         }
 630
 631         if(rbuf->size + sz > rbuf->bufsz) {
 632                 char *tmp;
 633                 int newsz = rbuf->bufsz << 1;
 634
 635                 if(!(tmp = realloc(rbuf->buf, newsz))) {
 636                         errmsg("failed to resize receive buffer\n");
 637                         return;
 638                 }
 639                 rbuf->buf = tmp;
 640                 rbuf->bufsz = newsz;
 641         }
 642
 643         memcpy(rbuf->buf + rbuf->size, buf, sz);
 644         rbuf->size += sz;
 645 }