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