Open the root image O_EXLOCK|O_NONBLOCK and exit with an error message
[dragonfly/vkernel-mp.git] / lib / libfetch / ftp.c
blob826a8c3b0b3d8debd59714f754a1d6609c3f49f6
1 /*-
2 * Copyright (c) 1998 Dag-Erling Coïdan Smørgrav
3 * All rights reserved.
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
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.
28 * $FreeBSD: src/lib/libfetch/ftp.c,v 1.16.2.31 2003/06/06 06:45:25 des Exp $
29 * $DragonFly: src/lib/libfetch/ftp.c,v 1.3 2003/12/01 23:58:44 drhodus Exp $
33 * Portions of this code were taken from or based on ftpio.c:
35 * ----------------------------------------------------------------------------
36 * "THE BEER-WARE LICENSE" (Revision 42):
37 * <phk@FreeBSD.org> wrote this file. As long as you retain this notice you
38 * can do whatever you want with this stuff. If we meet some day, and you think
39 * this stuff is worth it, you can buy me a beer in return. Poul-Henning Kamp
40 * ----------------------------------------------------------------------------
42 * Major Changelog:
44 * Dag-Erling Coïdan Smørgrav
45 * 9 Jun 1998
47 * Incorporated into libfetch
49 * Jordan K. Hubbard
50 * 17 Jan 1996
52 * Turned inside out. Now returns xfers as new file ids, not as a special
53 * `state' of FTP_t
55 * $ftpioId: ftpio.c,v 1.30 1998/04/11 07:28:53 phk Exp $
59 #include <sys/param.h>
60 #include <sys/socket.h>
61 #include <netinet/in.h>
63 #include <ctype.h>
64 #include <err.h>
65 #include <errno.h>
66 #include <fcntl.h>
67 #include <netdb.h>
68 #include <stdarg.h>
69 #include <stdio.h>
70 #include <stdlib.h>
71 #include <string.h>
72 #include <time.h>
73 #include <unistd.h>
75 #include "fetch.h"
76 #include "common.h"
77 #include "ftperr.h"
79 #define FTP_ANONYMOUS_USER "anonymous"
81 #define FTP_CONNECTION_ALREADY_OPEN 125
82 #define FTP_OPEN_DATA_CONNECTION 150
83 #define FTP_OK 200
84 #define FTP_FILE_STATUS 213
85 #define FTP_SERVICE_READY 220
86 #define FTP_TRANSFER_COMPLETE 226
87 #define FTP_PASSIVE_MODE 227
88 #define FTP_LPASSIVE_MODE 228
89 #define FTP_EPASSIVE_MODE 229
90 #define FTP_LOGGED_IN 230
91 #define FTP_FILE_ACTION_OK 250
92 #define FTP_NEED_PASSWORD 331
93 #define FTP_NEED_ACCOUNT 332
94 #define FTP_FILE_OK 350
95 #define FTP_SYNTAX_ERROR 500
96 #define FTP_PROTOCOL_ERROR 999
98 static struct url cached_host;
99 static conn_t *cached_connection;
101 #define isftpreply(foo) (isdigit(foo[0]) && isdigit(foo[1]) \
102 && isdigit(foo[2]) \
103 && (foo[3] == ' ' || foo[3] == '\0'))
104 #define isftpinfo(foo) (isdigit(foo[0]) && isdigit(foo[1]) \
105 && isdigit(foo[2]) && foo[3] == '-')
108 * Translate IPv4 mapped IPv6 address to IPv4 address
110 static void
111 unmappedaddr(struct sockaddr_in6 *sin6)
113 struct sockaddr_in *sin4;
114 u_int32_t addr;
115 int port;
117 if (sin6->sin6_family != AF_INET6 ||
118 !IN6_IS_ADDR_V4MAPPED(&sin6->sin6_addr))
119 return;
120 sin4 = (struct sockaddr_in *)sin6;
121 addr = *(u_int32_t *)&sin6->sin6_addr.s6_addr[12];
122 port = sin6->sin6_port;
123 memset(sin4, 0, sizeof(struct sockaddr_in));
124 sin4->sin_addr.s_addr = addr;
125 sin4->sin_port = port;
126 sin4->sin_family = AF_INET;
127 sin4->sin_len = sizeof(struct sockaddr_in);
131 * Get server response
133 static int
134 _ftp_chkerr(conn_t *conn)
136 if (_fetch_getln(conn) == -1) {
137 _fetch_syserr();
138 return (-1);
140 if (isftpinfo(conn->buf)) {
141 while (conn->buflen && !isftpreply(conn->buf)) {
142 if (_fetch_getln(conn) == -1) {
143 _fetch_syserr();
144 return (-1);
149 while (conn->buflen && isspace(conn->buf[conn->buflen - 1]))
150 conn->buflen--;
151 conn->buf[conn->buflen] = '\0';
153 if (!isftpreply(conn->buf)) {
154 _ftp_seterr(FTP_PROTOCOL_ERROR);
155 return (-1);
158 conn->err = (conn->buf[0] - '0') * 100
159 + (conn->buf[1] - '0') * 10
160 + (conn->buf[2] - '0');
162 return (conn->err);
166 * Send a command and check reply
168 static int
169 _ftp_cmd(conn_t *conn, const char *fmt, ...)
171 va_list ap;
172 size_t len;
173 char *msg;
174 int r;
176 va_start(ap, fmt);
177 len = vasprintf(&msg, fmt, ap);
178 va_end(ap);
180 if (msg == NULL) {
181 errno = ENOMEM;
182 _fetch_syserr();
183 return (-1);
186 r = _fetch_putln(conn, msg, len);
187 free(msg);
189 if (r == -1) {
190 _fetch_syserr();
191 return (-1);
194 return (_ftp_chkerr(conn));
198 * Return a pointer to the filename part of a path
200 static const char *
201 _ftp_filename(const char *file)
203 char *s;
205 if ((s = strrchr(file, '/')) == NULL)
206 return (file);
207 else
208 return (s + 1);
212 * Change working directory to the directory that contains the specified
213 * file.
215 static int
216 _ftp_cwd(conn_t *conn, const char *file)
218 char *s;
219 int e;
221 if ((s = strrchr(file, '/')) == NULL || s == file) {
222 e = _ftp_cmd(conn, "CWD /");
223 } else {
224 e = _ftp_cmd(conn, "CWD %.*s", s - file, file);
226 if (e != FTP_FILE_ACTION_OK) {
227 _ftp_seterr(e);
228 return (-1);
230 return (0);
234 * Request and parse file stats
236 static int
237 _ftp_stat(conn_t *conn, const char *file, struct url_stat *us)
239 char *ln;
240 const char *s;
241 struct tm tm;
242 time_t t;
243 int e;
245 us->size = -1;
246 us->atime = us->mtime = 0;
248 if ((s = strrchr(file, '/')) == NULL)
249 s = file;
250 else
251 ++s;
253 if ((e = _ftp_cmd(conn, "SIZE %s", s)) != FTP_FILE_STATUS) {
254 _ftp_seterr(e);
255 return (-1);
257 for (ln = conn->buf + 4; *ln && isspace(*ln); ln++)
258 /* nothing */ ;
259 for (us->size = 0; *ln && isdigit(*ln); ln++)
260 us->size = us->size * 10 + *ln - '0';
261 if (*ln && !isspace(*ln)) {
262 _ftp_seterr(FTP_PROTOCOL_ERROR);
263 us->size = -1;
264 return (-1);
266 if (us->size == 0)
267 us->size = -1;
268 DEBUG(fprintf(stderr, "size: [%lld]\n", (long long)us->size));
270 if ((e = _ftp_cmd(conn, "MDTM %s", s)) != FTP_FILE_STATUS) {
271 _ftp_seterr(e);
272 return (-1);
274 for (ln = conn->buf + 4; *ln && isspace(*ln); ln++)
275 /* nothing */ ;
276 switch (strspn(ln, "0123456789")) {
277 case 14:
278 break;
279 case 15:
280 ln++;
281 ln[0] = '2';
282 ln[1] = '0';
283 break;
284 default:
285 _ftp_seterr(FTP_PROTOCOL_ERROR);
286 return (-1);
288 if (sscanf(ln, "%04d%02d%02d%02d%02d%02d",
289 &tm.tm_year, &tm.tm_mon, &tm.tm_mday,
290 &tm.tm_hour, &tm.tm_min, &tm.tm_sec) != 6) {
291 _ftp_seterr(FTP_PROTOCOL_ERROR);
292 return (-1);
294 tm.tm_mon--;
295 tm.tm_year -= 1900;
296 tm.tm_isdst = -1;
297 t = timegm(&tm);
298 if (t == (time_t)-1)
299 t = time(NULL);
300 us->mtime = t;
301 us->atime = t;
302 DEBUG(fprintf(stderr,
303 "last modified: [%04d-%02d-%02d %02d:%02d:%02d]\n",
304 tm.tm_year + 1900, tm.tm_mon + 1, tm.tm_mday,
305 tm.tm_hour, tm.tm_min, tm.tm_sec));
306 return (0);
310 * I/O functions for FTP
312 struct ftpio {
313 conn_t *cconn; /* Control connection */
314 conn_t *dconn; /* Data connection */
315 int dir; /* Direction */
316 int eof; /* EOF reached */
317 int err; /* Error code */
320 static int _ftp_readfn(void *, char *, int);
321 static int _ftp_writefn(void *, const char *, int);
322 static fpos_t _ftp_seekfn(void *, fpos_t, int);
323 static int _ftp_closefn(void *);
325 static int
326 _ftp_readfn(void *v, char *buf, int len)
328 struct ftpio *io;
329 int r;
331 io = (struct ftpio *)v;
332 if (io == NULL) {
333 errno = EBADF;
334 return (-1);
336 if (io->cconn == NULL || io->dconn == NULL || io->dir == O_WRONLY) {
337 errno = EBADF;
338 return (-1);
340 if (io->err) {
341 errno = io->err;
342 return (-1);
344 if (io->eof)
345 return (0);
346 r = _fetch_read(io->dconn, buf, len);
347 if (r > 0)
348 return (r);
349 if (r == 0) {
350 io->eof = 1;
351 return (0);
353 if (errno != EINTR)
354 io->err = errno;
355 return (-1);
358 static int
359 _ftp_writefn(void *v, const char *buf, int len)
361 struct ftpio *io;
362 int w;
364 io = (struct ftpio *)v;
365 if (io == NULL) {
366 errno = EBADF;
367 return (-1);
369 if (io->cconn == NULL || io->dconn == NULL || io->dir == O_RDONLY) {
370 errno = EBADF;
371 return (-1);
373 if (io->err) {
374 errno = io->err;
375 return (-1);
377 w = _fetch_write(io->dconn, buf, len);
378 if (w >= 0)
379 return (w);
380 if (errno != EINTR)
381 io->err = errno;
382 return (-1);
385 static fpos_t
386 _ftp_seekfn(void *v, fpos_t pos __unused, int whence __unused)
388 struct ftpio *io;
390 io = (struct ftpio *)v;
391 if (io == NULL) {
392 errno = EBADF;
393 return (-1);
395 errno = ESPIPE;
396 return (-1);
399 static int
400 _ftp_closefn(void *v)
402 struct ftpio *io;
403 int r;
405 io = (struct ftpio *)v;
406 if (io == NULL) {
407 errno = EBADF;
408 return (-1);
410 if (io->dir == -1)
411 return (0);
412 if (io->cconn == NULL || io->dconn == NULL) {
413 errno = EBADF;
414 return (-1);
416 _fetch_close(io->dconn);
417 io->dir = -1;
418 io->dconn = NULL;
419 DEBUG(fprintf(stderr, "Waiting for final status\n"));
420 r = _ftp_chkerr(io->cconn);
421 if (io->cconn == cached_connection && io->cconn->ref == 1)
422 cached_connection = NULL;
423 _fetch_close(io->cconn);
424 free(io);
425 return (r == FTP_TRANSFER_COMPLETE) ? 0 : -1;
428 static FILE *
429 _ftp_setup(conn_t *cconn, conn_t *dconn, int mode)
431 struct ftpio *io;
432 FILE *f;
434 if (cconn == NULL || dconn == NULL)
435 return (NULL);
436 if ((io = malloc(sizeof(*io))) == NULL)
437 return (NULL);
438 io->cconn = cconn;
439 io->dconn = dconn;
440 io->dir = mode;
441 io->eof = io->err = 0;
442 f = funopen(io, _ftp_readfn, _ftp_writefn, _ftp_seekfn, _ftp_closefn);
443 if (f == NULL)
444 free(io);
445 return (f);
449 * Transfer file
451 static FILE *
452 _ftp_transfer(conn_t *conn, const char *oper, const char *file,
453 int mode, off_t offset, const char *flags)
455 struct sockaddr_storage sa;
456 struct sockaddr_in6 *sin6;
457 struct sockaddr_in *sin4;
458 int low, pasv, verbose;
459 int e, sd = -1;
460 socklen_t l;
461 char *s;
462 FILE *df;
464 /* check flags */
465 low = CHECK_FLAG('l');
466 pasv = CHECK_FLAG('p');
467 verbose = CHECK_FLAG('v');
469 /* passive mode */
470 if (!pasv)
471 pasv = ((s = getenv("FTP_PASSIVE_MODE")) != NULL &&
472 strncasecmp(s, "no", 2) != 0);
474 /* find our own address, bind, and listen */
475 l = sizeof(sa);
476 if (getsockname(conn->sd, (struct sockaddr *)&sa, &l) == -1)
477 goto sysouch;
478 if (sa.ss_family == AF_INET6)
479 unmappedaddr((struct sockaddr_in6 *)&sa);
481 /* open data socket */
482 if ((sd = socket(sa.ss_family, SOCK_STREAM, IPPROTO_TCP)) == -1) {
483 _fetch_syserr();
484 return (NULL);
487 if (pasv) {
488 u_char addr[64];
489 char *ln, *p;
490 unsigned int i;
491 int port;
493 /* send PASV command */
494 if (verbose)
495 _fetch_info("setting passive mode");
496 switch (sa.ss_family) {
497 case AF_INET:
498 if ((e = _ftp_cmd(conn, "PASV")) != FTP_PASSIVE_MODE)
499 goto ouch;
500 break;
501 case AF_INET6:
502 if ((e = _ftp_cmd(conn, "EPSV")) != FTP_EPASSIVE_MODE) {
503 if (e == -1)
504 goto ouch;
505 if ((e = _ftp_cmd(conn, "LPSV")) !=
506 FTP_LPASSIVE_MODE)
507 goto ouch;
509 break;
510 default:
511 e = FTP_PROTOCOL_ERROR; /* XXX: error code should be prepared */
512 goto ouch;
516 * Find address and port number. The reply to the PASV command
517 * is IMHO the one and only weak point in the FTP protocol.
519 ln = conn->buf;
520 switch (e) {
521 case FTP_PASSIVE_MODE:
522 case FTP_LPASSIVE_MODE:
523 for (p = ln + 3; *p && !isdigit(*p); p++)
524 /* nothing */ ;
525 if (!*p) {
526 e = FTP_PROTOCOL_ERROR;
527 goto ouch;
529 l = (e == FTP_PASSIVE_MODE ? 6 : 21);
530 for (i = 0; *p && i < l; i++, p++)
531 addr[i] = strtol(p, &p, 10);
532 if (i < l) {
533 e = FTP_PROTOCOL_ERROR;
534 goto ouch;
536 break;
537 case FTP_EPASSIVE_MODE:
538 for (p = ln + 3; *p && *p != '('; p++)
539 /* nothing */ ;
540 if (!*p) {
541 e = FTP_PROTOCOL_ERROR;
542 goto ouch;
544 ++p;
545 if (sscanf(p, "%c%c%c%d%c", &addr[0], &addr[1], &addr[2],
546 &port, &addr[3]) != 5 ||
547 addr[0] != addr[1] ||
548 addr[0] != addr[2] || addr[0] != addr[3]) {
549 e = FTP_PROTOCOL_ERROR;
550 goto ouch;
552 break;
555 /* seek to required offset */
556 if (offset)
557 if (_ftp_cmd(conn, "REST %lu", (u_long)offset) != FTP_FILE_OK)
558 goto sysouch;
560 /* construct sockaddr for data socket */
561 l = sizeof(sa);
562 if (getpeername(conn->sd, (struct sockaddr *)&sa, &l) == -1)
563 goto sysouch;
564 if (sa.ss_family == AF_INET6)
565 unmappedaddr((struct sockaddr_in6 *)&sa);
566 switch (sa.ss_family) {
567 case AF_INET6:
568 sin6 = (struct sockaddr_in6 *)&sa;
569 if (e == FTP_EPASSIVE_MODE)
570 sin6->sin6_port = htons(port);
571 else {
572 bcopy(addr + 2, (char *)&sin6->sin6_addr, 16);
573 bcopy(addr + 19, (char *)&sin6->sin6_port, 2);
575 break;
576 case AF_INET:
577 sin4 = (struct sockaddr_in *)&sa;
578 if (e == FTP_EPASSIVE_MODE)
579 sin4->sin_port = htons(port);
580 else {
581 bcopy(addr, (char *)&sin4->sin_addr, 4);
582 bcopy(addr + 4, (char *)&sin4->sin_port, 2);
584 break;
585 default:
586 e = FTP_PROTOCOL_ERROR; /* XXX: error code should be prepared */
587 break;
590 /* connect to data port */
591 if (verbose)
592 _fetch_info("opening data connection");
593 if (connect(sd, (struct sockaddr *)&sa, sa.ss_len) == -1)
594 goto sysouch;
596 /* make the server initiate the transfer */
597 if (verbose)
598 _fetch_info("initiating transfer");
599 e = _ftp_cmd(conn, "%s %s", oper, _ftp_filename(file));
600 if (e != FTP_CONNECTION_ALREADY_OPEN && e != FTP_OPEN_DATA_CONNECTION)
601 goto ouch;
603 } else {
604 u_int32_t a;
605 u_short p;
606 int arg, d;
607 char *ap;
608 char hname[INET6_ADDRSTRLEN];
610 switch (sa.ss_family) {
611 case AF_INET6:
612 ((struct sockaddr_in6 *)&sa)->sin6_port = 0;
613 #ifdef IPV6_PORTRANGE
614 arg = low ? IPV6_PORTRANGE_DEFAULT : IPV6_PORTRANGE_HIGH;
615 if (setsockopt(sd, IPPROTO_IPV6, IPV6_PORTRANGE,
616 (char *)&arg, sizeof(arg)) == -1)
617 goto sysouch;
618 #endif
619 break;
620 case AF_INET:
621 ((struct sockaddr_in *)&sa)->sin_port = 0;
622 arg = low ? IP_PORTRANGE_DEFAULT : IP_PORTRANGE_HIGH;
623 if (setsockopt(sd, IPPROTO_IP, IP_PORTRANGE,
624 (char *)&arg, sizeof(arg)) == -1)
625 goto sysouch;
626 break;
628 if (verbose)
629 _fetch_info("binding data socket");
630 if (bind(sd, (struct sockaddr *)&sa, sa.ss_len) == -1)
631 goto sysouch;
632 if (listen(sd, 1) == -1)
633 goto sysouch;
635 /* find what port we're on and tell the server */
636 if (getsockname(sd, (struct sockaddr *)&sa, &l) == -1)
637 goto sysouch;
638 switch (sa.ss_family) {
639 case AF_INET:
640 sin4 = (struct sockaddr_in *)&sa;
641 a = ntohl(sin4->sin_addr.s_addr);
642 p = ntohs(sin4->sin_port);
643 e = _ftp_cmd(conn, "PORT %d,%d,%d,%d,%d,%d",
644 (a >> 24) & 0xff, (a >> 16) & 0xff,
645 (a >> 8) & 0xff, a & 0xff,
646 (p >> 8) & 0xff, p & 0xff);
647 break;
648 case AF_INET6:
649 #define UC(b) (((int)b)&0xff)
650 e = -1;
651 sin6 = (struct sockaddr_in6 *)&sa;
652 sin6->sin6_scope_id = 0;
653 if (getnameinfo((struct sockaddr *)&sa, sa.ss_len,
654 hname, sizeof(hname),
655 NULL, 0, NI_NUMERICHOST) == 0) {
656 e = _ftp_cmd(conn, "EPRT |%d|%s|%d|", 2, hname,
657 htons(sin6->sin6_port));
658 if (e == -1)
659 goto ouch;
661 if (e != FTP_OK) {
662 ap = (char *)&sin6->sin6_addr;
663 e = _ftp_cmd(conn,
664 "LPRT %d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d",
665 6, 16,
666 UC(ap[0]), UC(ap[1]), UC(ap[2]), UC(ap[3]),
667 UC(ap[4]), UC(ap[5]), UC(ap[6]), UC(ap[7]),
668 UC(ap[8]), UC(ap[9]), UC(ap[10]), UC(ap[11]),
669 UC(ap[12]), UC(ap[13]), UC(ap[14]), UC(ap[15]),
671 (ntohs(sin6->sin6_port) >> 8) & 0xff,
672 ntohs(sin6->sin6_port) & 0xff);
674 break;
675 default:
676 e = FTP_PROTOCOL_ERROR; /* XXX: error code should be prepared */
677 goto ouch;
679 if (e != FTP_OK)
680 goto ouch;
682 /* seek to required offset */
683 if (offset)
684 if (_ftp_cmd(conn, "REST %llu",
685 (unsigned long long)offset) != FTP_FILE_OK)
686 goto sysouch;
688 /* make the server initiate the transfer */
689 if (verbose)
690 _fetch_info("initiating transfer");
691 e = _ftp_cmd(conn, "%s %s", oper, _ftp_filename(file));
692 if (e != FTP_OPEN_DATA_CONNECTION)
693 goto ouch;
695 /* accept the incoming connection and go to town */
696 if ((d = accept(sd, NULL, NULL)) == -1)
697 goto sysouch;
698 close(sd);
699 sd = d;
702 if ((df = _ftp_setup(conn, _fetch_reopen(sd), mode)) == NULL)
703 goto sysouch;
704 return (df);
706 sysouch:
707 _fetch_syserr();
708 if (sd >= 0)
709 close(sd);
710 return (NULL);
712 ouch:
713 if (e != -1)
714 _ftp_seterr(e);
715 if (sd >= 0)
716 close(sd);
717 return (NULL);
721 * Authenticate
723 static int
724 _ftp_authenticate(conn_t *conn, struct url *url, struct url *purl)
726 const char *user, *pwd, *logname;
727 char pbuf[MAXHOSTNAMELEN + MAXLOGNAME + 1];
728 int e, len;
730 /* XXX FTP_AUTH, and maybe .netrc */
732 /* send user name and password */
733 if (url->user[0] == '\0')
734 _fetch_netrc_auth(url);
735 user = url->user;
736 if (*user == '\0')
737 user = getenv("FTP_LOGIN");
738 if (user == NULL || *user == '\0')
739 user = FTP_ANONYMOUS_USER;
740 if (purl && url->port == _fetch_default_port(url->scheme))
741 e = _ftp_cmd(conn, "USER %s@%s", user, url->host);
742 else if (purl)
743 e = _ftp_cmd(conn, "USER %s@%s@%d", user, url->host, url->port);
744 else
745 e = _ftp_cmd(conn, "USER %s", user);
747 /* did the server request a password? */
748 if (e == FTP_NEED_PASSWORD) {
749 pwd = url->pwd;
750 if (*pwd == '\0')
751 pwd = getenv("FTP_PASSWORD");
752 if (pwd == NULL || *pwd == '\0') {
753 if ((logname = getlogin()) == 0)
754 logname = FTP_ANONYMOUS_USER;
755 if ((len = snprintf(pbuf, MAXLOGNAME + 1, "%s@", logname)) < 0)
756 len = 0;
757 else if (len > MAXLOGNAME)
758 len = MAXLOGNAME;
759 gethostname(pbuf + len, sizeof(pbuf) - len);
760 pwd = pbuf;
762 e = _ftp_cmd(conn, "PASS %s", pwd);
765 return (e);
769 * Log on to FTP server
771 static conn_t *
772 _ftp_connect(struct url *url, struct url *purl, const char *flags)
774 conn_t *conn;
775 int e, direct, verbose;
776 #ifdef INET6
777 int af = AF_UNSPEC;
778 #else
779 int af = AF_INET;
780 #endif
782 direct = CHECK_FLAG('d');
783 verbose = CHECK_FLAG('v');
784 if (CHECK_FLAG('4'))
785 af = AF_INET;
786 else if (CHECK_FLAG('6'))
787 af = AF_INET6;
789 if (direct)
790 purl = NULL;
792 /* check for proxy */
793 if (purl) {
794 /* XXX proxy authentication! */
795 conn = _fetch_connect(purl->host, purl->port, af, verbose);
796 } else {
797 /* no proxy, go straight to target */
798 conn = _fetch_connect(url->host, url->port, af, verbose);
799 purl = NULL;
802 /* check connection */
803 if (conn == NULL)
804 /* _fetch_connect() has already set an error code */
805 return (NULL);
807 /* expect welcome message */
808 if ((e = _ftp_chkerr(conn)) != FTP_SERVICE_READY)
809 goto fouch;
811 /* authenticate */
812 if ((e = _ftp_authenticate(conn, url, purl)) != FTP_LOGGED_IN)
813 goto fouch;
815 /* might as well select mode and type at once */
816 #ifdef FTP_FORCE_STREAM_MODE
817 if ((e = _ftp_cmd(conn, "MODE S")) != FTP_OK) /* default is S */
818 goto fouch;
819 #endif
820 if ((e = _ftp_cmd(conn, "TYPE I")) != FTP_OK) /* default is A */
821 goto fouch;
823 /* done */
824 return (conn);
826 fouch:
827 if (e != -1)
828 _ftp_seterr(e);
829 _fetch_close(conn);
830 return (NULL);
834 * Disconnect from server
836 static void
837 _ftp_disconnect(conn_t *conn)
839 (void)_ftp_cmd(conn, "QUIT");
840 if (conn == cached_connection && conn->ref == 1)
841 cached_connection = NULL;
842 _fetch_close(conn);
846 * Check if we're already connected
848 static int
849 _ftp_isconnected(struct url *url)
851 return (cached_connection
852 && (strcmp(url->host, cached_host.host) == 0)
853 && (strcmp(url->user, cached_host.user) == 0)
854 && (strcmp(url->pwd, cached_host.pwd) == 0)
855 && (url->port == cached_host.port));
859 * Check the cache, reconnect if no luck
861 static conn_t *
862 _ftp_cached_connect(struct url *url, struct url *purl, const char *flags)
864 conn_t *conn;
865 int e;
867 /* set default port */
868 if (!url->port)
869 url->port = _fetch_default_port(url->scheme);
871 /* try to use previously cached connection */
872 if (_ftp_isconnected(url)) {
873 e = _ftp_cmd(cached_connection, "NOOP");
874 if (e == FTP_OK || e == FTP_SYNTAX_ERROR)
875 return (_fetch_ref(cached_connection));
878 /* connect to server */
879 if ((conn = _ftp_connect(url, purl, flags)) == NULL)
880 return (NULL);
881 if (cached_connection)
882 _ftp_disconnect(cached_connection);
883 cached_connection = _fetch_ref(conn);
884 memcpy(&cached_host, url, sizeof(*url));
885 return (conn);
889 * Check the proxy settings
891 static struct url *
892 _ftp_get_proxy(const char *flags)
894 struct url *purl;
895 char *p;
897 if (flags != NULL && strchr(flags, 'd') != NULL)
898 return (NULL);
899 if (((p = getenv("FTP_PROXY")) || (p = getenv("ftp_proxy")) ||
900 (p = getenv("HTTP_PROXY")) || (p = getenv("http_proxy"))) &&
901 *p && (purl = fetchParseURL(p)) != NULL) {
902 if (!*purl->scheme) {
903 if (getenv("FTP_PROXY") || getenv("ftp_proxy"))
904 strcpy(purl->scheme, SCHEME_FTP);
905 else
906 strcpy(purl->scheme, SCHEME_HTTP);
908 if (!purl->port)
909 purl->port = _fetch_default_proxy_port(purl->scheme);
910 if (strcasecmp(purl->scheme, SCHEME_FTP) == 0 ||
911 strcasecmp(purl->scheme, SCHEME_HTTP) == 0)
912 return (purl);
913 fetchFreeURL(purl);
915 return (NULL);
919 * Process an FTP request
921 FILE *
922 _ftp_request(struct url *url, const char *op, struct url_stat *us,
923 struct url *purl, const char *flags)
925 conn_t *conn;
926 int oflag;
928 /* check if we should use HTTP instead */
929 if (purl && strcasecmp(purl->scheme, SCHEME_HTTP) == 0) {
930 if (strcmp(op, "STAT") == 0)
931 return (_http_request(url, "HEAD", us, purl, flags));
932 else if (strcmp(op, "RETR") == 0)
933 return (_http_request(url, "GET", us, purl, flags));
935 * Our HTTP code doesn't support PUT requests yet, so try
936 * a direct connection.
940 /* connect to server */
941 conn = _ftp_cached_connect(url, purl, flags);
942 if (purl)
943 fetchFreeURL(purl);
944 if (conn == NULL)
945 return (NULL);
947 /* change directory */
948 if (_ftp_cwd(conn, url->doc) == -1)
949 return (NULL);
951 /* stat file */
952 if (us && _ftp_stat(conn, url->doc, us) == -1
953 && fetchLastErrCode != FETCH_PROTO
954 && fetchLastErrCode != FETCH_UNAVAIL)
955 return (NULL);
957 /* just a stat */
958 if (strcmp(op, "STAT") == 0)
959 return (FILE *)1; /* bogus return value */
960 if (strcmp(op, "STOR") == 0 || strcmp(op, "APPE") == 0)
961 oflag = O_WRONLY;
962 else
963 oflag = O_RDONLY;
965 /* initiate the transfer */
966 return (_ftp_transfer(conn, op, url->doc, oflag, url->offset, flags));
970 * Get and stat file
972 FILE *
973 fetchXGetFTP(struct url *url, struct url_stat *us, const char *flags)
975 return (_ftp_request(url, "RETR", us, _ftp_get_proxy(flags), flags));
979 * Get file
981 FILE *
982 fetchGetFTP(struct url *url, const char *flags)
984 return (fetchXGetFTP(url, NULL, flags));
988 * Put file
990 FILE *
991 fetchPutFTP(struct url *url, const char *flags)
994 return (_ftp_request(url, CHECK_FLAG('a') ? "APPE" : "STOR", NULL,
995 _ftp_get_proxy(flags), flags));
999 * Get file stats
1002 fetchStatFTP(struct url *url, struct url_stat *us, const char *flags)
1004 FILE *f;
1006 f = _ftp_request(url, "STAT", us, _ftp_get_proxy(flags), flags);
1007 if (f == NULL)
1008 return (-1);
1009 return (0);
1013 * List a directory
1015 struct url_ent *
1016 fetchListFTP(struct url *url __unused, const char *flags __unused)
1018 warnx("fetchListFTP(): not implemented");
1019 return (NULL);