ftp.c revision 63340
1/*- 2 * Copyright (c) 1998 Dag-Erling Co�dan Sm�rgrav 3 * All rights reserved. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions 7 * are met: 8 * 1. Redistributions of source code must retain the above copyright 9 * notice, this list of conditions and the following disclaimer 10 * in this position and unchanged. 11 * 2. Redistributions in binary form must reproduce the above copyright 12 * notice, this list of conditions and the following disclaimer in the 13 * documentation and/or other materials provided with the distribution. 14 * 3. The name of the author may not be used to endorse or promote products 15 * derived from this software without specific prior written permission 16 * 17 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR 18 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 19 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 20 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, 21 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 22 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 23 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 24 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 25 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 26 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 27 * 28 * $FreeBSD: head/lib/libfetch/ftp.c 63340 2000-07-17 21:25:00Z des $ 29 */ 30 31/* 32 * Portions of this code were taken from or based on ftpio.c: 33 * 34 * ---------------------------------------------------------------------------- 35 * "THE BEER-WARE LICENSE" (Revision 42): 36 * <phk@login.dknet.dk> wrote this file. As long as you retain this notice you 37 * can do whatever you want with this stuff. If we meet some day, and you think 38 * this stuff is worth it, you can buy me a beer in return. Poul-Henning Kamp 39 * ---------------------------------------------------------------------------- 40 * 41 * Major Changelog: 42 * 43 * Dag-Erling Co�dan Sm�rgrav 44 * 9 Jun 1998 45 * 46 * Incorporated into libfetch 47 * 48 * Jordan K. Hubbard 49 * 17 Jan 1996 50 * 51 * Turned inside out. Now returns xfers as new file ids, not as a special 52 * `state' of FTP_t 53 * 54 * $ftpioId: ftpio.c,v 1.30 1998/04/11 07:28:53 phk Exp $ 55 * 56 */ 57 58#include <sys/param.h> 59#include <sys/socket.h> 60#include <netinet/in.h> 61 62#include <ctype.h> 63#include <errno.h> 64#include <netdb.h> 65#include <stdarg.h> 66#include <stdio.h> 67#include <stdlib.h> 68#include <string.h> 69#include <time.h> 70#include <unistd.h> 71 72#include "fetch.h" 73#include "common.h" 74#include "ftperr.h" 75 76#define FTP_ANONYMOUS_USER "ftp" 77#define FTP_ANONYMOUS_PASSWORD "ftp" 78#define FTP_DEFAULT_PORT 21 79 80#define FTP_OPEN_DATA_CONNECTION 150 81#define FTP_OK 200 82#define FTP_FILE_STATUS 213 83#define FTP_SERVICE_READY 220 84#define FTP_PASSIVE_MODE 227 85#define FTP_LPASSIVE_MODE 228 86#define FTP_EPASSIVE_MODE 229 87#define FTP_LOGGED_IN 230 88#define FTP_FILE_ACTION_OK 250 89#define FTP_NEED_PASSWORD 331 90#define FTP_NEED_ACCOUNT 332 91#define FTP_FILE_OK 350 92#define FTP_SYNTAX_ERROR 500 93#define FTP_PROTOCOL_ERROR 999 94 95static struct url cached_host; 96static int cached_socket; 97 98static char *last_reply; 99static size_t lr_size, lr_length; 100static int last_code; 101 102#define isftpreply(foo) (isdigit(foo[0]) && isdigit(foo[1]) \ 103 && isdigit(foo[2]) \ 104 && (foo[3] == ' ' || foo[3] == '\0')) 105#define isftpinfo(foo) (isdigit(foo[0]) && isdigit(foo[1]) \ 106 && isdigit(foo[2]) && foo[3] == '-') 107 108/* translate IPv4 mapped IPv6 address to IPv4 address */ 109static void 110unmappedaddr(struct sockaddr_in6 *sin6) 111{ 112 struct sockaddr_in *sin4; 113 u_int32_t addr; 114 int port; 115 116 if (sin6->sin6_family != AF_INET6 || 117 !IN6_IS_ADDR_V4MAPPED(&sin6->sin6_addr)) 118 return; 119 sin4 = (struct sockaddr_in *)sin6; 120 addr = *(u_int32_t *)&sin6->sin6_addr.s6_addr[12]; 121 port = sin6->sin6_port; 122 memset(sin4, 0, sizeof(struct sockaddr_in)); 123 sin4->sin_addr.s_addr = addr; 124 sin4->sin_port = port; 125 sin4->sin_family = AF_INET; 126 sin4->sin_len = sizeof(struct sockaddr_in); 127} 128 129/* 130 * Get server response 131 */ 132static int 133_ftp_chkerr(int cd) 134{ 135 if (_fetch_getln(cd, &last_reply, &lr_size, &lr_length) == -1) { 136 _fetch_syserr(); 137 return -1; 138 } 139 if (isftpinfo(last_reply)) { 140 while (!isftpreply(last_reply)) { 141 if (_fetch_getln(cd, &last_reply, &lr_size, &lr_length) == -1) { 142 _fetch_syserr(); 143 return -1; 144 } 145 } 146 } 147 148 while (lr_length && isspace(last_reply[lr_length-1])) 149 lr_length--; 150 last_reply[lr_length] = 0; 151 152 if (!isftpreply(last_reply)) { 153 _ftp_seterr(FTP_PROTOCOL_ERROR); 154 return -1; 155 } 156 157 last_code = (last_reply[0] - '0') * 100 158 + (last_reply[1] - '0') * 10 159 + (last_reply[2] - '0'); 160 161 return last_code; 162} 163 164/* 165 * Send a command and check reply 166 */ 167static int 168_ftp_cmd(int cd, char *fmt, ...) 169{ 170 va_list ap; 171 size_t len; 172 char *msg; 173 int r; 174 175 va_start(ap, fmt); 176 len = vasprintf(&msg, fmt, ap); 177 va_end(ap); 178 179 if (msg == NULL) { 180 errno = ENOMEM; 181 _fetch_syserr(); 182 return -1; 183 } 184 185 r = _fetch_putln(cd, msg, len); 186 free(msg); 187 188 if (r == -1) { 189 _fetch_syserr(); 190 return -1; 191 } 192 193 return _ftp_chkerr(cd); 194} 195 196/* 197 * Return a pointer to the filename part of a path 198 */ 199static char * 200_ftp_filename(char *file) 201{ 202 char *s; 203 204 if ((s = strrchr(file, '/')) == NULL) 205 return file; 206 else 207 return s + 1; 208} 209 210/* 211 * Change working directory to the directory that contains the 212 * specified file. 213 */ 214static int 215_ftp_cwd(int cd, char *file) 216{ 217 char *s; 218 int e; 219 220 if ((s = strrchr(file, '/')) == NULL) { 221 e = _ftp_cmd(cd, "CWD /"); 222 } else { 223 e = _ftp_cmd(cd, "CWD %.*s", s - file, file); 224 } 225 if (e != FTP_FILE_ACTION_OK) { 226 _ftp_seterr(e); 227 return -1; 228 } 229 return 0; 230} 231 232/* 233 * Request and parse file stats 234 */ 235static int 236_ftp_stat(int cd, char *file, struct url_stat *us) 237{ 238 char *ln, *s; 239 struct tm tm; 240 time_t t; 241 int e; 242 243 if ((s = strrchr(file, '/')) == NULL) 244 s = file; 245 else 246 ++s; 247 248 if ((e = _ftp_cmd(cd, "SIZE %s", s)) != FTP_FILE_STATUS) { 249 _ftp_seterr(e); 250 return -1; 251 } 252 for (ln = last_reply + 4; *ln && isspace(*ln); ln++) 253 /* nothing */ ; 254 for (us->size = 0; *ln && isdigit(*ln); ln++) 255 us->size = us->size * 10 + *ln - '0'; 256 if (*ln && !isspace(*ln)) { 257 _ftp_seterr(FTP_PROTOCOL_ERROR); 258 return -1; 259 } 260 DEBUG(fprintf(stderr, "size: [\033[1m%lld\033[m]\n", us->size)); 261 262 if ((e = _ftp_cmd(cd, "MDTM %s", s)) != FTP_FILE_STATUS) { 263 _ftp_seterr(e); 264 return -1; 265 } 266 for (ln = last_reply + 4; *ln && isspace(*ln); ln++) 267 /* nothing */ ; 268 switch (strspn(ln, "0123456789")) { 269 case 14: 270 break; 271 case 15: 272 ln++; 273 ln[0] = '2'; 274 ln[1] = '0'; 275 break; 276 default: 277 _ftp_seterr(FTP_PROTOCOL_ERROR); 278 return -1; 279 } 280 if (sscanf(ln, "%04d%02d%02d%02d%02d%02d", 281 &tm.tm_year, &tm.tm_mon, &tm.tm_mday, 282 &tm.tm_hour, &tm.tm_min, &tm.tm_sec) != 6) { 283 _ftp_seterr(FTP_PROTOCOL_ERROR); 284 return -1; 285 } 286 tm.tm_mon--; 287 tm.tm_year -= 1900; 288 tm.tm_isdst = -1; 289 t = timegm(&tm); 290 if (t == (time_t)-1) 291 t = time(NULL); 292 us->mtime = t; 293 us->atime = t; 294 DEBUG(fprintf(stderr, "last modified: [\033[1m%04d-%02d-%02d " 295 "%02d:%02d:%02d\033[m]\n", 296 tm.tm_year + 1900, tm.tm_mon + 1, tm.tm_mday, 297 tm.tm_hour, tm.tm_min, tm.tm_sec)); 298 return 0; 299} 300 301/* 302 * Transfer file 303 */ 304static FILE * 305_ftp_transfer(int cd, char *oper, char *file, 306 char *mode, off_t offset, char *flags) 307{ 308 struct sockaddr_storage sin; 309 struct sockaddr_in6 *sin6; 310 struct sockaddr_in *sin4; 311 int pasv, high, verbose; 312 int e, sd = -1; 313 socklen_t l; 314 char *s; 315 FILE *df; 316 317 /* check flags */ 318 pasv = (flags && strchr(flags, 'p')); 319 high = (flags && strchr(flags, 'h')); 320 verbose = (flags && strchr(flags, 'v')); 321 322 /* passive mode */ 323 if (!pasv && (s = getenv("FTP_PASSIVE_MODE")) != NULL) 324 pasv = (strncasecmp(s, "no", 2) != 0); 325 326 /* find our own address, bind, and listen */ 327 l = sizeof sin; 328 if (getsockname(cd, (struct sockaddr *)&sin, &l) == -1) 329 goto sysouch; 330 if (sin.ss_family == AF_INET6) 331 unmappedaddr((struct sockaddr_in6 *)&sin); 332 333 /* open data socket */ 334 if ((sd = socket(sin.ss_family, SOCK_STREAM, IPPROTO_TCP)) == -1) { 335 _fetch_syserr(); 336 return NULL; 337 } 338 339 if (pasv) { 340 u_char addr[64]; 341 char *ln, *p; 342 int i; 343 int port; 344 345 /* send PASV command */ 346 if (verbose) 347 _fetch_info("setting passive mode"); 348 switch (sin.ss_family) { 349 case AF_INET: 350 if ((e = _ftp_cmd(cd, "PASV")) != FTP_PASSIVE_MODE) 351 goto ouch; 352 break; 353 case AF_INET6: 354 if ((e = _ftp_cmd(cd, "EPSV")) != FTP_EPASSIVE_MODE) { 355 if (e == -1) 356 goto ouch; 357 if ((e = _ftp_cmd(cd, "LPSV")) != FTP_LPASSIVE_MODE) 358 goto ouch; 359 } 360 break; 361 default: 362 e = FTP_PROTOCOL_ERROR; /* XXX: error code should be prepared */ 363 goto ouch; 364 } 365 366 /* 367 * Find address and port number. The reply to the PASV command 368 * is IMHO the one and only weak point in the FTP protocol. 369 */ 370 ln = last_reply; 371 switch (e) { 372 case FTP_PASSIVE_MODE: 373 case FTP_LPASSIVE_MODE: 374 for (p = ln + 3; *p && !isdigit(*p); p++) 375 /* nothing */ ; 376 if (!*p) { 377 e = FTP_PROTOCOL_ERROR; 378 goto ouch; 379 } 380 l = (e == FTP_PASSIVE_MODE ? 6 : 21); 381 for (i = 0; *p && i < l; i++, p++) 382 addr[i] = strtol(p, &p, 10); 383 if (i < l) { 384 e = FTP_PROTOCOL_ERROR; 385 goto ouch; 386 } 387 break; 388 case FTP_EPASSIVE_MODE: 389 for (p = ln + 3; *p && *p != '('; p++) 390 /* nothing */ ; 391 if (!*p) { 392 e = FTP_PROTOCOL_ERROR; 393 goto ouch; 394 } 395 ++p; 396 if (sscanf(p, "%c%c%c%d%c", &addr[0], &addr[1], &addr[2], 397 &port, &addr[3]) != 5 || 398 addr[0] != addr[1] || 399 addr[0] != addr[2] || addr[0] != addr[3]) { 400 e = FTP_PROTOCOL_ERROR; 401 goto ouch; 402 } 403 break; 404 } 405 406 /* seek to required offset */ 407 if (offset) 408 if (_ftp_cmd(cd, "REST %lu", (u_long)offset) != FTP_FILE_OK) 409 goto sysouch; 410 411 /* construct sockaddr for data socket */ 412 l = sizeof sin; 413 if (getpeername(cd, (struct sockaddr *)&sin, &l) == -1) 414 goto sysouch; 415 if (sin.ss_family == AF_INET6) 416 unmappedaddr((struct sockaddr_in6 *)&sin); 417 switch (sin.ss_family) { 418 case AF_INET6: 419 sin6 = (struct sockaddr_in6 *)&sin; 420 if (e == FTP_EPASSIVE_MODE) 421 sin6->sin6_port = htons(port); 422 else { 423 bcopy(addr + 2, (char *)&sin6->sin6_addr, 16); 424 bcopy(addr + 19, (char *)&sin6->sin6_port, 2); 425 } 426 break; 427 case AF_INET: 428 sin4 = (struct sockaddr_in *)&sin; 429 if (e == FTP_EPASSIVE_MODE) 430 sin4->sin_port = htons(port); 431 else { 432 bcopy(addr, (char *)&sin4->sin_addr, 4); 433 bcopy(addr + 4, (char *)&sin4->sin_port, 2); 434 } 435 break; 436 default: 437 e = FTP_PROTOCOL_ERROR; /* XXX: error code should be prepared */ 438 break; 439 } 440 441 /* connect to data port */ 442 if (verbose) 443 _fetch_info("opening data connection"); 444 if (connect(sd, (struct sockaddr *)&sin, sin.ss_len) == -1) 445 goto sysouch; 446 447 /* make the server initiate the transfer */ 448 if (verbose) 449 _fetch_info("initiating transfer"); 450 e = _ftp_cmd(cd, "%s %s", oper, _ftp_filename(file)); 451 if (e != FTP_OPEN_DATA_CONNECTION) 452 goto ouch; 453 454 } else { 455 u_int32_t a; 456 u_short p; 457 int arg, d; 458 char *ap; 459 char hname[INET6_ADDRSTRLEN]; 460 461 switch (sin.ss_family) { 462 case AF_INET6: 463 ((struct sockaddr_in6 *)&sin)->sin6_port = 0; 464#ifdef IPV6_PORTRANGE 465 arg = high ? IPV6_PORTRANGE_HIGH : IPV6_PORTRANGE_DEFAULT; 466 if (setsockopt(sd, IPPROTO_IPV6, IPV6_PORTRANGE, 467 (char *)&arg, sizeof(arg)) == -1) 468 goto sysouch; 469#endif 470 break; 471 case AF_INET: 472 ((struct sockaddr_in *)&sin)->sin_port = 0; 473 arg = high ? IP_PORTRANGE_HIGH : IP_PORTRANGE_DEFAULT; 474 if (setsockopt(sd, IPPROTO_IP, IP_PORTRANGE, 475 (char *)&arg, sizeof arg) == -1) 476 goto sysouch; 477 break; 478 } 479 if (verbose) 480 _fetch_info("binding data socket"); 481 if (bind(sd, (struct sockaddr *)&sin, sin.ss_len) == -1) 482 goto sysouch; 483 if (listen(sd, 1) == -1) 484 goto sysouch; 485 486 /* find what port we're on and tell the server */ 487 if (getsockname(sd, (struct sockaddr *)&sin, &l) == -1) 488 goto sysouch; 489 switch (sin.ss_family) { 490 case AF_INET: 491 sin4 = (struct sockaddr_in *)&sin; 492 a = ntohl(sin4->sin_addr.s_addr); 493 p = ntohs(sin4->sin_port); 494 e = _ftp_cmd(cd, "PORT %d,%d,%d,%d,%d,%d", 495 (a >> 24) & 0xff, (a >> 16) & 0xff, 496 (a >> 8) & 0xff, a & 0xff, 497 (p >> 8) & 0xff, p & 0xff); 498 break; 499 case AF_INET6: 500#define UC(b) (((int)b)&0xff) 501 e = -1; 502 sin6 = (struct sockaddr_in6 *)&sin; 503 if (getnameinfo((struct sockaddr *)&sin, sin.ss_len, 504 hname, sizeof(hname), 505 NULL, 0, NI_NUMERICHOST) == 0) { 506 e = _ftp_cmd(cd, "EPRT |%d|%s|%d|", 2, hname, 507 htons(sin6->sin6_port)); 508 if (e == -1) 509 goto ouch; 510 } 511 if (e != FTP_OK) { 512 ap = (char *)&sin6->sin6_addr; 513 e = _ftp_cmd(cd, 514 "LPRT %d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d", 515 6, 16, 516 UC(ap[0]), UC(ap[1]), UC(ap[2]), UC(ap[3]), 517 UC(ap[4]), UC(ap[5]), UC(ap[6]), UC(ap[7]), 518 UC(ap[8]), UC(ap[9]), UC(ap[10]), UC(ap[11]), 519 UC(ap[12]), UC(ap[13]), UC(ap[14]), UC(ap[15]), 520 2, 521 (ntohs(sin6->sin6_port) >> 8) & 0xff, 522 ntohs(sin6->sin6_port) & 0xff); 523 } 524 break; 525 default: 526 e = FTP_PROTOCOL_ERROR; /* XXX: error code should be prepared */ 527 goto ouch; 528 } 529 if (e != FTP_OK) 530 goto ouch; 531 532 /* seek to required offset */ 533 if (offset) 534 if (_ftp_cmd(cd, "REST %lu", (u_long)offset) != FTP_FILE_OK) 535 goto sysouch; 536 537 /* make the server initiate the transfer */ 538 if (verbose) 539 _fetch_info("initiating transfer"); 540 e = _ftp_cmd(cd, "%s %s", oper, _ftp_filename(file)); 541 if (e != FTP_OPEN_DATA_CONNECTION) 542 goto ouch; 543 544 /* accept the incoming connection and go to town */ 545 if ((d = accept(sd, NULL, NULL)) == -1) 546 goto sysouch; 547 close(sd); 548 sd = d; 549 } 550 551 if ((df = fdopen(sd, mode)) == NULL) 552 goto sysouch; 553 return df; 554 555sysouch: 556 _fetch_syserr(); 557 if (sd >= 0) 558 close(sd); 559 return NULL; 560 561ouch: 562 if (e != -1) 563 _ftp_seterr(e); 564 if (sd >= 0) 565 close(sd); 566 return NULL; 567} 568 569/* 570 * Log on to FTP server 571 */ 572static int 573_ftp_connect(char *host, int port, char *user, char *pwd, char *flags) 574{ 575 int cd, e, pp = 0, direct, verbose; 576#ifdef INET6 577 int af = AF_UNSPEC; 578#else 579 int af = AF_INET; 580#endif 581 char *p, *q; 582 const char *logname; 583 char localhost[MAXHOSTNAMELEN]; 584 char pbuf[MAXHOSTNAMELEN + MAXLOGNAME + 1]; 585 586 direct = (flags && strchr(flags, 'd')); 587 verbose = (flags && strchr(flags, 'v')); 588 if ((flags && strchr(flags, '4'))) 589 af = AF_INET; 590 else if ((flags && strchr(flags, '6'))) 591 af = AF_INET6; 592 593 /* check for proxy */ 594 if (!direct && (p = getenv("FTP_PROXY")) != NULL) { 595 char c = 0; 596 597#ifdef INET6 598 if (*p != '[' || (q = strchr(p + 1, ']')) == NULL || 599 (*++q != '\0' && *q != ':')) 600#endif 601 q = strchr(p, ':'); 602 if (q != NULL && *q == ':') { 603 if (strspn(q+1, "0123456789") != strlen(q+1) || strlen(q+1) > 5) { 604 /* XXX we should emit some kind of warning */ 605 } 606 pp = atoi(q+1); 607 if (pp < 1 || pp > 65535) { 608 /* XXX we should emit some kind of warning */ 609 } 610 } 611 if (!pp) { 612 struct servent *se; 613 614 if ((se = getservbyname("ftp", "tcp")) != NULL) 615 pp = ntohs(se->s_port); 616 else 617 pp = FTP_DEFAULT_PORT; 618 } 619 if (q) { 620#ifdef INET6 621 if (q > p && *p == '[' && *(q - 1) == ']') { 622 p++; 623 q--; 624 } 625#endif 626 c = *q; 627 *q = 0; 628 } 629 cd = _fetch_connect(p, pp, af, verbose); 630 if (q) 631 *q = c; 632 } else { 633 /* no proxy, go straight to target */ 634 cd = _fetch_connect(host, port, af, verbose); 635 p = NULL; 636 } 637 638 /* check connection */ 639 if (cd == -1) { 640 _fetch_syserr(); 641 return NULL; 642 } 643 644 /* expect welcome message */ 645 if ((e = _ftp_chkerr(cd)) != FTP_SERVICE_READY) 646 goto fouch; 647 648 /* send user name and password */ 649 if (!user || !*user) 650 user = FTP_ANONYMOUS_USER; 651 e = p ? _ftp_cmd(cd, "USER %s@%s@%d", user, host, port) 652 : _ftp_cmd(cd, "USER %s", user); 653 654 /* did the server request a password? */ 655 if (e == FTP_NEED_PASSWORD) { 656 if (!pwd || !*pwd) 657 pwd = getenv("FTP_PASSWORD"); 658 if (!pwd || !*pwd) { 659 if ((logname = getlogin()) == 0) 660 logname = FTP_ANONYMOUS_PASSWORD; 661 gethostname(localhost, sizeof localhost); 662 snprintf(pbuf, sizeof pbuf, "%s@%s", logname, localhost); 663 pwd = pbuf; 664 } 665 e = _ftp_cmd(cd, "PASS %s", pwd); 666 } 667 668 /* did the server request an account? */ 669 if (e == FTP_NEED_ACCOUNT) 670 goto fouch; 671 672 /* we should be done by now */ 673 if (e != FTP_LOGGED_IN) 674 goto fouch; 675 676 /* might as well select mode and type at once */ 677#ifdef FTP_FORCE_STREAM_MODE 678 if ((e = _ftp_cmd(cd, "MODE S")) != FTP_OK) /* default is S */ 679 goto fouch; 680#endif 681 if ((e = _ftp_cmd(cd, "TYPE I")) != FTP_OK) /* default is A */ 682 goto fouch; 683 684 /* done */ 685 return cd; 686 687fouch: 688 if (e != -1) 689 _ftp_seterr(e); 690 close(cd); 691 return NULL; 692} 693 694/* 695 * Disconnect from server 696 */ 697static void 698_ftp_disconnect(int cd) 699{ 700 (void)_ftp_cmd(cd, "QUIT"); 701 close(cd); 702} 703 704/* 705 * Check if we're already connected 706 */ 707static int 708_ftp_isconnected(struct url *url) 709{ 710 return (cached_socket 711 && (strcmp(url->host, cached_host.host) == 0) 712 && (strcmp(url->user, cached_host.user) == 0) 713 && (strcmp(url->pwd, cached_host.pwd) == 0) 714 && (url->port == cached_host.port)); 715} 716 717/* 718 * Check the cache, reconnect if no luck 719 */ 720static int 721_ftp_cached_connect(struct url *url, char *flags) 722{ 723 int e, cd; 724 725 cd = -1; 726 727 /* set default port */ 728 if (!url->port) { 729 struct servent *se; 730 731 if ((se = getservbyname("ftp", "tcp")) != NULL) 732 url->port = ntohs(se->s_port); 733 else 734 url->port = FTP_DEFAULT_PORT; 735 } 736 737 /* try to use previously cached connection */ 738 if (_ftp_isconnected(url)) { 739 e = _ftp_cmd(cached_socket, "NOOP"); 740 if (e == FTP_OK || e == FTP_SYNTAX_ERROR) 741 cd = cached_socket; 742 } 743 744 /* connect to server */ 745 if (cd == -1) { 746 cd = _ftp_connect(url->host, url->port, url->user, url->pwd, flags); 747 if (cd == -1) 748 return -1; 749 if (cached_socket) 750 _ftp_disconnect(cached_socket); 751 cached_socket = cd; 752 memcpy(&cached_host, url, sizeof *url); 753 } 754 755 return cd; 756} 757 758/* 759 * Get and stat file 760 */ 761FILE * 762fetchXGetFTP(struct url *url, struct url_stat *us, char *flags) 763{ 764 int cd; 765 766 /* connect to server */ 767 if ((cd = _ftp_cached_connect(url, flags)) == NULL) 768 return NULL; 769 770 /* change directory */ 771 if (_ftp_cwd(cd, url->doc) == -1) 772 return NULL; 773 774 /* stat file */ 775 if (us && _ftp_stat(cd, url->doc, us) == -1) 776 return NULL; 777 778 /* initiate the transfer */ 779 return _ftp_transfer(cd, "RETR", url->doc, "r", url->offset, flags); 780} 781 782/* 783 * Get file 784 */ 785FILE * 786fetchGetFTP(struct url *url, char *flags) 787{ 788 return fetchXGetFTP(url, NULL, flags); 789} 790 791/* 792 * Put file 793 */ 794FILE * 795fetchPutFTP(struct url *url, char *flags) 796{ 797 int cd; 798 799 /* connect to server */ 800 if ((cd = _ftp_cached_connect(url, flags)) == NULL) 801 return NULL; 802 803 /* change directory */ 804 if (_ftp_cwd(cd, url->doc) == -1) 805 return NULL; 806 807 /* initiate the transfer */ 808 return _ftp_transfer(cd, (flags && strchr(flags, 'a')) ? "APPE" : "STOR", 809 url->doc, "w", url->offset, flags); 810} 811 812/* 813 * Get file stats 814 */ 815int 816fetchStatFTP(struct url *url, struct url_stat *us, char *flags) 817{ 818 int cd; 819 820 us->size = -1; 821 us->atime = us->mtime = 0; 822 823 /* connect to server */ 824 if ((cd = _ftp_cached_connect(url, flags)) == NULL) 825 return -1; 826 827 /* change directory */ 828 if (_ftp_cwd(cd, url->doc) == -1) 829 return -1; 830 831 /* stat file */ 832 return _ftp_stat(cd, url->doc, us); 833} 834 835/* 836 * List a directory 837 */ 838extern void warnx(char *, ...); 839struct url_ent * 840fetchListFTP(struct url *url, char *flags) 841{ 842 warnx("fetchListFTP(): not implemented"); 843 return NULL; 844} 845