xref: /netbsd-src/external/bsd/fetch/dist/libfetch/ftp.c (revision 466a16a118933bd295a8a104f095714fadf9cf68)
1 /*	$NetBSD: ftp.c,v 1.1.1.3 2008/10/29 16:18:14 joerg Exp $	*/
2 /*-
3  * Copyright (c) 1998-2004 Dag-Erling Co�dan Sm�rgrav
4  * Copyright (c) 2008 Joerg Sonnenberger <joerg@NetBSD.org>
5  * All rights reserved.
6  *
7  * Redistribution and use in source and binary forms, with or without
8  * modification, are permitted provided that the following conditions
9  * are met:
10  * 1. Redistributions of source code must retain the above copyright
11  *    notice, this list of conditions and the following disclaimer
12  *    in this position and unchanged.
13  * 2. Redistributions in binary form must reproduce the above copyright
14  *    notice, this list of conditions and the following disclaimer in the
15  *    documentation and/or other materials provided with the distribution.
16  * 3. The name of the author may not be used to endorse or promote products
17  *    derived from this software without specific prior written permission
18  *
19  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
20  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
21  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
22  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
23  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
24  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
25  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
26  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
27  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
28  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
29  *
30  * $FreeBSD: ftp.c,v 1.101 2008/01/23 20:57:59 des Exp $
31  */
32 
33 /*
34  * Portions of this code were taken from or based on ftpio.c:
35  *
36  * ----------------------------------------------------------------------------
37  * "THE BEER-WARE LICENSE" (Revision 42):
38  * <phk@FreeBSD.org> wrote this file.  As long as you retain this notice you
39  * can do whatever you want with this stuff. If we meet some day, and you think
40  * this stuff is worth it, you can buy me a beer in return.   Poul-Henning Kamp
41  * ----------------------------------------------------------------------------
42  *
43  * Major Changelog:
44  *
45  * Dag-Erling Co�dan Sm�rgrav
46  * 9 Jun 1998
47  *
48  * Incorporated into libfetch
49  *
50  * Jordan K. Hubbard
51  * 17 Jan 1996
52  *
53  * Turned inside out. Now returns xfers as new file ids, not as a special
54  * `state' of FTP_t
55  *
56  * $ftpioId: ftpio.c,v 1.30 1998/04/11 07:28:53 phk Exp $
57  *
58  */
59 
60 #if HAVE_CONFIG_H
61 #include "config.h"
62 #endif
63 #ifndef NETBSD
64 #include <nbcompat.h>
65 #endif
66 
67 #include <sys/types.h>
68 #include <sys/socket.h>
69 
70 #include <arpa/inet.h>
71 #include <netinet/in.h>
72 
73 #include <ctype.h>
74 #include <errno.h>
75 #include <fcntl.h>
76 #if defined(HAVE_INTTYPES_H) || defined(NETBSD)
77 #include <inttypes.h>
78 #endif
79 #include <stdarg.h>
80 #ifndef NETBSD
81 #include <nbcompat/netdb.h>
82 #include <nbcompat/stdio.h>
83 #else
84 #include <netdb.h>
85 #include <stdio.h>
86 #endif
87 #include <stdlib.h>
88 #include <string.h>
89 #include <time.h>
90 #include <unistd.h>
91 
92 #include "fetch.h"
93 #include "common.h"
94 #include "ftperr.h"
95 
96 #define FTP_ANONYMOUS_USER	"anonymous"
97 
98 #define FTP_CONNECTION_ALREADY_OPEN	125
99 #define FTP_OPEN_DATA_CONNECTION	150
100 #define FTP_OK				200
101 #define FTP_FILE_STATUS			213
102 #define FTP_SERVICE_READY		220
103 #define FTP_TRANSFER_COMPLETE		226
104 #define FTP_PASSIVE_MODE		227
105 #define FTP_LPASSIVE_MODE		228
106 #define FTP_EPASSIVE_MODE		229
107 #define FTP_LOGGED_IN			230
108 #define FTP_FILE_ACTION_OK		250
109 #define FTP_DIRECTORY_CREATED		257 /* multiple meanings */
110 #define FTP_FILE_CREATED		257 /* multiple meanings */
111 #define FTP_WORKING_DIRECTORY		257 /* multiple meanings */
112 #define FTP_NEED_PASSWORD		331
113 #define FTP_NEED_ACCOUNT		332
114 #define FTP_FILE_OK			350
115 #define FTP_SYNTAX_ERROR		500
116 #define FTP_PROTOCOL_ERROR		999
117 
118 static struct url cached_host;
119 static conn_t	*cached_connection;
120 
121 #define isftpreply(foo)				\
122 	(isdigit((unsigned char)foo[0]) &&	\
123 	    isdigit((unsigned char)foo[1]) &&	\
124 	    isdigit((unsigned char)foo[2]) &&	\
125 	    (foo[3] == ' ' || foo[3] == '\0'))
126 #define isftpinfo(foo) \
127 	(isdigit((unsigned char)foo[0]) &&	\
128 	    isdigit((unsigned char)foo[1]) &&	\
129 	    isdigit((unsigned char)foo[2]) &&	\
130 	    foo[3] == '-')
131 
132 /*
133  * Translate IPv4 mapped IPv6 address to IPv4 address
134  */
135 static void
136 unmappedaddr(struct sockaddr_in6 *sin6, socklen_t *len)
137 {
138 	struct sockaddr_in *sin4;
139 	uint32_t addr;
140 	int port;
141 
142 	if (sin6->sin6_family != AF_INET6 ||
143 	    !IN6_IS_ADDR_V4MAPPED(&sin6->sin6_addr))
144 		return;
145 	sin4 = (struct sockaddr_in *)sin6;
146 	addr = *(uint32_t *)&sin6->sin6_addr.s6_addr[12];
147 	port = sin6->sin6_port;
148 	memset(sin4, 0, sizeof(struct sockaddr_in));
149 	sin4->sin_addr.s_addr = addr;
150 	sin4->sin_port = port;
151 	sin4->sin_family = AF_INET;
152 	*len = sizeof(struct sockaddr_in);
153 #ifdef HAVE_SA_LEN
154 	sin4->sin_len = sizeof(struct sockaddr_in);
155 #endif
156 }
157 
158 /*
159  * Get server response
160  */
161 static int
162 ftp_chkerr(conn_t *conn)
163 {
164 	if (fetch_getln(conn) == -1) {
165 		fetch_syserr();
166 		return (-1);
167 	}
168 	if (isftpinfo(conn->buf)) {
169 		while (conn->buflen && !isftpreply(conn->buf)) {
170 			if (fetch_getln(conn) == -1) {
171 				fetch_syserr();
172 				return (-1);
173 			}
174 		}
175 	}
176 
177 	while (conn->buflen &&
178 	    isspace((unsigned char)conn->buf[conn->buflen - 1]))
179 		conn->buflen--;
180 	conn->buf[conn->buflen] = '\0';
181 
182 	if (!isftpreply(conn->buf)) {
183 		ftp_seterr(FTP_PROTOCOL_ERROR);
184 		return (-1);
185 	}
186 
187 	conn->err = (conn->buf[0] - '0') * 100
188 	    + (conn->buf[1] - '0') * 10
189 	    + (conn->buf[2] - '0');
190 
191 	return (conn->err);
192 }
193 
194 /*
195  * Send a command and check reply
196  */
197 static int
198 ftp_cmd(conn_t *conn, const char *fmt, ...)
199 {
200 	va_list ap;
201 	size_t len;
202 	char *msg;
203 	int r;
204 
205 	va_start(ap, fmt);
206 	len = vasprintf(&msg, fmt, ap);
207 	va_end(ap);
208 
209 	if (msg == NULL) {
210 		errno = ENOMEM;
211 		fetch_syserr();
212 		return (-1);
213 	}
214 
215 	r = fetch_putln(conn, msg, len);
216 	free(msg);
217 
218 	if (r == -1) {
219 		fetch_syserr();
220 		return (-1);
221 	}
222 
223 	return (ftp_chkerr(conn));
224 }
225 
226 /*
227  * Return a pointer to the filename part of a path
228  */
229 static const char *
230 ftp_filename(const char *file, int *len, int *type, int subdir)
231 {
232 	const char *s;
233 
234 	if ((s = strrchr(file, '/')) == NULL || subdir)
235 		s = file;
236 	else
237 		s = s + 1;
238 	*len = strlen(s);
239 	if (*len > 7 && strncmp(s + *len - 7, ";type=", 6) == 0) {
240 		*type = s[*len - 1];
241 		*len -= 7;
242 	} else {
243 		*type = '\0';
244 	}
245 	return (s);
246 }
247 
248 /*
249  * Get current working directory from the reply to a CWD, PWD or CDUP
250  * command.
251  */
252 static int
253 ftp_pwd(conn_t *conn, char *pwd, size_t pwdlen)
254 {
255 	char *src, *dst, *end;
256 	int q;
257 
258 	if (conn->err != FTP_WORKING_DIRECTORY &&
259 	    conn->err != FTP_FILE_ACTION_OK)
260 		return (FTP_PROTOCOL_ERROR);
261 	end = conn->buf + conn->buflen;
262 	src = conn->buf + 4;
263 	if (src >= end || *src++ != '"')
264 		return (FTP_PROTOCOL_ERROR);
265 	for (q = 0, dst = pwd; src < end && pwdlen--; ++src) {
266 		if (!q && *src == '"')
267 			q = 1;
268 		else if (q && *src != '"')
269 			break;
270 		else if (q)
271 			*dst++ = '"', q = 0;
272 		else
273 			*dst++ = *src;
274 	}
275 	if (!pwdlen)
276 		return (FTP_PROTOCOL_ERROR);
277 	*dst = '\0';
278 	return (FTP_OK);
279 }
280 
281 /*
282  * Change working directory to the directory that contains the specified
283  * file.
284  */
285 static int
286 ftp_cwd(conn_t *conn, const char *file, int subdir)
287 {
288 	const char *beg, *end;
289 	char pwd[PATH_MAX];
290 	int e, i, len;
291 
292 	/* If no slashes in name, no need to change dirs. */
293 	if (subdir)
294 		end = file + strlen(file);
295 	else if ((end = strrchr(file, '/')) == NULL)
296 		return (0);
297 	if ((e = ftp_cmd(conn, "PWD")) != FTP_WORKING_DIRECTORY ||
298 	    (e = ftp_pwd(conn, pwd, sizeof(pwd))) != FTP_OK) {
299 		ftp_seterr(e);
300 		return (-1);
301 	}
302 	for (;;) {
303 		len = strlen(pwd);
304 
305 		/* Look for a common prefix between PWD and dir to fetch. */
306 		for (i = 0; i <= len && i <= end - file; ++i)
307 			if (pwd[i] != file[i])
308 				break;
309 		/* Keep going up a dir until we have a matching prefix. */
310 		if (strcmp(pwd, "/") == 0)
311 			break;
312 		if (pwd[i] == '\0' && (file[i - 1] == '/' || file[i] == '/'))
313 			break;
314 		if ((e = ftp_cmd(conn, "CDUP")) != FTP_FILE_ACTION_OK ||
315 		    (e = ftp_cmd(conn, "PWD")) != FTP_WORKING_DIRECTORY ||
316 		    (e = ftp_pwd(conn, pwd, sizeof(pwd))) != FTP_OK) {
317 			ftp_seterr(e);
318 			return (-1);
319 		}
320 	}
321 
322 #ifdef FTP_COMBINE_CWDS
323 	/* Skip leading slashes, even "////". */
324 	for (beg = file + i; beg < end && *beg == '/'; ++beg, ++i)
325 		/* nothing */ ;
326 
327 	/* If there is no trailing dir, we're already there. */
328 	if (beg >= end)
329 		return (0);
330 
331 	/* Change to the directory all in one chunk (e.g., foo/bar/baz). */
332 	e = ftp_cmd(conn, "CWD %.*s", (int)(end - beg), beg);
333 	if (e == FTP_FILE_ACTION_OK)
334 		return (0);
335 #endif /* FTP_COMBINE_CWDS */
336 
337 	/* That didn't work so go back to legacy behavior (multiple CWDs). */
338 	for (beg = file + i; beg < end; beg = file + i + 1) {
339 		while (*beg == '/')
340 			++beg, ++i;
341 		for (++i; file + i < end && file[i] != '/'; ++i)
342 			/* nothing */ ;
343 		e = ftp_cmd(conn, "CWD %.*s", file + i - beg, beg);
344 		if (e != FTP_FILE_ACTION_OK) {
345 			ftp_seterr(e);
346 			return (-1);
347 		}
348 	}
349 	return (0);
350 }
351 
352 /*
353  * Set transfer mode and data type
354  */
355 static int
356 ftp_mode_type(conn_t *conn, int mode, int type)
357 {
358 	int e;
359 
360 	switch (mode) {
361 	case 0:
362 	case 's':
363 		mode = 'S';
364 	case 'S':
365 		break;
366 	default:
367 		return (FTP_PROTOCOL_ERROR);
368 	}
369 	if ((e = ftp_cmd(conn, "MODE %c", mode)) != FTP_OK) {
370 		if (mode == 'S') {
371 			/*
372 			 * Stream mode is supposed to be the default - so
373 			 * much so that some servers not only do not
374 			 * support any other mode, but do not support the
375 			 * MODE command at all.
376 			 *
377 			 * If "MODE S" fails, it is unlikely that we
378 			 * previously succeeded in setting a different
379 			 * mode.  Therefore, we simply hope that the
380 			 * server is already in the correct mode, and
381 			 * silently ignore the failure.
382 			 */
383 		} else {
384 			return (e);
385 		}
386 	}
387 
388 	switch (type) {
389 	case 0:
390 	case 'i':
391 		type = 'I';
392 	case 'I':
393 		break;
394 	case 'a':
395 		type = 'A';
396 	case 'A':
397 		break;
398 	case 'd':
399 		type = 'D';
400 	case 'D':
401 		/* can't handle yet */
402 	default:
403 		return (FTP_PROTOCOL_ERROR);
404 	}
405 	if ((e = ftp_cmd(conn, "TYPE %c", type)) != FTP_OK)
406 		return (e);
407 
408 	return (FTP_OK);
409 }
410 
411 /*
412  * Request and parse file stats
413  */
414 static int
415 ftp_stat(conn_t *conn, const char *file, struct url_stat *us)
416 {
417 	char *ln;
418 	const char *filename;
419 	int filenamelen, type;
420 	struct tm tm;
421 	time_t t;
422 	int e;
423 
424 	us->size = -1;
425 	us->atime = us->mtime = 0;
426 
427 	filename = ftp_filename(file, &filenamelen, &type, 0);
428 
429 	if ((e = ftp_mode_type(conn, 0, type)) != FTP_OK) {
430 		ftp_seterr(e);
431 		return (-1);
432 	}
433 
434 	e = ftp_cmd(conn, "SIZE %.*s", filenamelen, filename);
435 	if (e != FTP_FILE_STATUS) {
436 		ftp_seterr(e);
437 		return (-1);
438 	}
439 	for (ln = conn->buf + 4; *ln && isspace((unsigned char)*ln); ln++)
440 		/* nothing */ ;
441 	for (us->size = 0; *ln && isdigit((unsigned char)*ln); ln++)
442 		us->size = us->size * 10 + *ln - '0';
443 	if (*ln && !isspace((unsigned char)*ln)) {
444 		ftp_seterr(FTP_PROTOCOL_ERROR);
445 		us->size = -1;
446 		return (-1);
447 	}
448 	if (us->size == 0)
449 		us->size = -1;
450 
451 	e = ftp_cmd(conn, "MDTM %.*s", filenamelen, filename);
452 	if (e != FTP_FILE_STATUS) {
453 		ftp_seterr(e);
454 		return (-1);
455 	}
456 	for (ln = conn->buf + 4; *ln && isspace((unsigned char)*ln); ln++)
457 		/* nothing */ ;
458 	switch (strspn(ln, "0123456789")) {
459 	case 14:
460 		break;
461 	case 15:
462 		ln++;
463 		ln[0] = '2';
464 		ln[1] = '0';
465 		break;
466 	default:
467 		ftp_seterr(FTP_PROTOCOL_ERROR);
468 		return (-1);
469 	}
470 	if (sscanf(ln, "%04d%02d%02d%02d%02d%02d",
471 	    &tm.tm_year, &tm.tm_mon, &tm.tm_mday,
472 	    &tm.tm_hour, &tm.tm_min, &tm.tm_sec) != 6) {
473 		ftp_seterr(FTP_PROTOCOL_ERROR);
474 		return (-1);
475 	}
476 	tm.tm_mon--;
477 	tm.tm_year -= 1900;
478 	tm.tm_isdst = -1;
479 	t = timegm(&tm);
480 	if (t == (time_t)-1)
481 		t = time(NULL);
482 	us->mtime = t;
483 	us->atime = t;
484 
485 	return (0);
486 }
487 
488 /*
489  * I/O functions for FTP
490  */
491 struct ftpio {
492 	conn_t	*cconn;		/* Control connection */
493 	conn_t	*dconn;		/* Data connection */
494 	int	 dir;		/* Direction */
495 	int	 eof;		/* EOF reached */
496 	int	 err;		/* Error code */
497 };
498 
499 static ssize_t	 ftp_readfn(void *, void *, size_t);
500 static ssize_t	 ftp_writefn(void *, const void *, size_t);
501 static void	 ftp_closefn(void *);
502 
503 static ssize_t
504 ftp_readfn(void *v, void *buf, size_t len)
505 {
506 	struct ftpio *io;
507 	int r;
508 
509 	io = (struct ftpio *)v;
510 	if (io == NULL) {
511 		errno = EBADF;
512 		return (-1);
513 	}
514 	if (io->cconn == NULL || io->dconn == NULL || io->dir == O_WRONLY) {
515 		errno = EBADF;
516 		return (-1);
517 	}
518 	if (io->err) {
519 		errno = io->err;
520 		return (-1);
521 	}
522 	if (io->eof)
523 		return (0);
524 	r = fetch_read(io->dconn, buf, len);
525 	if (r > 0)
526 		return (r);
527 	if (r == 0) {
528 		io->eof = 1;
529 		return (0);
530 	}
531 	if (errno != EINTR)
532 		io->err = errno;
533 	return (-1);
534 }
535 
536 static ssize_t
537 ftp_writefn(void *v, const void *buf, size_t len)
538 {
539 	struct ftpio *io;
540 	int w;
541 
542 	io = (struct ftpio *)v;
543 	if (io == NULL) {
544 		errno = EBADF;
545 		return (-1);
546 	}
547 	if (io->cconn == NULL || io->dconn == NULL || io->dir == O_RDONLY) {
548 		errno = EBADF;
549 		return (-1);
550 	}
551 	if (io->err) {
552 		errno = io->err;
553 		return (-1);
554 	}
555 	w = fetch_write(io->dconn, buf, len);
556 	if (w >= 0)
557 		return (w);
558 	if (errno != EINTR)
559 		io->err = errno;
560 	return (-1);
561 }
562 
563 static void
564 ftp_closefn(void *v)
565 {
566 	struct ftpio *io;
567 	int r;
568 
569 	io = (struct ftpio *)v;
570 	if (io == NULL) {
571 		errno = EBADF;
572 		return;
573 	}
574 	if (io->dir == -1)
575 		return;
576 	if (io->cconn == NULL || io->dconn == NULL) {
577 		errno = EBADF;
578 		return;
579 	}
580 	fetch_close(io->dconn);
581 	io->dir = -1;
582 	io->dconn = NULL;
583 	r = ftp_chkerr(io->cconn);
584 	if (io->cconn == cached_connection && io->cconn->ref == 1)
585 		cached_connection = NULL;
586 	fetch_close(io->cconn);
587 	free(io);
588 	return;
589 }
590 
591 static fetchIO *
592 ftp_setup(conn_t *cconn, conn_t *dconn, int mode)
593 {
594 	struct ftpio *io;
595 	fetchIO *f;
596 
597 	if (cconn == NULL || dconn == NULL)
598 		return (NULL);
599 	if ((io = malloc(sizeof(*io))) == NULL)
600 		return (NULL);
601 	io->cconn = cconn;
602 	io->dconn = dconn;
603 	io->dir = mode;
604 	io->eof = io->err = 0;
605 	f = fetchIO_unopen(io, ftp_readfn, ftp_writefn, ftp_closefn);
606 	if (f == NULL)
607 		free(io);
608 	return (f);
609 }
610 
611 /*
612  * Transfer file
613  */
614 static fetchIO *
615 ftp_transfer(conn_t *conn, const char *oper, const char *file, const char *op_arg,
616     int mode, off_t offset, const char *flags)
617 {
618 	struct sockaddr_storage sa;
619 	struct sockaddr_in6 *sin6;
620 	struct sockaddr_in *sin4;
621 	const char *bindaddr;
622 	const char *filename;
623 	int filenamelen, type;
624 	int low, pasv, verbose;
625 	int e, sd = -1;
626 	socklen_t l;
627 	char *s;
628 	fetchIO *df;
629 
630 	/* check flags */
631 	low = CHECK_FLAG('l');
632 	pasv = !CHECK_FLAG('a');
633 	verbose = CHECK_FLAG('v');
634 
635 	/* passive mode */
636 	if (!pasv)
637 		pasv = ((s = getenv("FTP_PASSIVE_MODE")) != NULL &&
638 		    strncasecmp(s, "no", 2) != 0);
639 
640 	/* isolate filename */
641 	filename = ftp_filename(file, &filenamelen, &type, op_arg != NULL);
642 
643 	/* set transfer mode and data type */
644 	if ((e = ftp_mode_type(conn, 0, type)) != FTP_OK)
645 		goto ouch;
646 
647 	/* find our own address, bind, and listen */
648 	l = sizeof(sa);
649 	if (getsockname(conn->sd, (struct sockaddr *)&sa, &l) == -1)
650 		goto sysouch;
651 	if (sa.ss_family == AF_INET6)
652 		unmappedaddr((struct sockaddr_in6 *)&sa, &l);
653 
654 retry_mode:
655 
656 	/* open data socket */
657 	if ((sd = socket(sa.ss_family, SOCK_STREAM, IPPROTO_TCP)) == -1) {
658 		fetch_syserr();
659 		return (NULL);
660 	}
661 
662 	if (pasv) {
663 		unsigned char addr[64];
664 		char *ln, *p;
665 		unsigned int i;
666 		int port;
667 
668 		/* send PASV command */
669 		if (verbose)
670 			fetch_info("setting passive mode");
671 		switch (sa.ss_family) {
672 		case AF_INET:
673 			if ((e = ftp_cmd(conn, "PASV")) != FTP_PASSIVE_MODE)
674 				goto ouch;
675 			break;
676 		case AF_INET6:
677 			if ((e = ftp_cmd(conn, "EPSV")) != FTP_EPASSIVE_MODE) {
678 				if (e == -1)
679 					goto ouch;
680 				if ((e = ftp_cmd(conn, "LPSV")) !=
681 				    FTP_LPASSIVE_MODE)
682 					goto ouch;
683 			}
684 			break;
685 		default:
686 			e = FTP_PROTOCOL_ERROR; /* XXX: error code should be prepared */
687 			goto ouch;
688 		}
689 
690 		/*
691 		 * Find address and port number. The reply to the PASV command
692 		 * is IMHO the one and only weak point in the FTP protocol.
693 		 */
694 		ln = conn->buf;
695 		switch (e) {
696 		case FTP_PASSIVE_MODE:
697 		case FTP_LPASSIVE_MODE:
698 			for (p = ln + 3; *p && !isdigit((unsigned char)*p); p++)
699 				/* nothing */ ;
700 			if (!*p) {
701 				e = FTP_PROTOCOL_ERROR;
702 				goto ouch;
703 			}
704 			l = (e == FTP_PASSIVE_MODE ? 6 : 21);
705 			for (i = 0; *p && i < l; i++, p++)
706 				addr[i] = strtol(p, &p, 10);
707 			if (i < l) {
708 				e = FTP_PROTOCOL_ERROR;
709 				goto ouch;
710 			}
711 			break;
712 		case FTP_EPASSIVE_MODE:
713 			for (p = ln + 3; *p && *p != '('; p++)
714 				/* nothing */ ;
715 			if (!*p) {
716 				e = FTP_PROTOCOL_ERROR;
717 				goto ouch;
718 			}
719 			++p;
720 			if (sscanf(p, "%c%c%c%d%c", &addr[0], &addr[1], &addr[2],
721 				&port, &addr[3]) != 5 ||
722 			    addr[0] != addr[1] ||
723 			    addr[0] != addr[2] || addr[0] != addr[3]) {
724 				e = FTP_PROTOCOL_ERROR;
725 				goto ouch;
726 			}
727 			break;
728 		case FTP_SYNTAX_ERROR:
729 			if (verbose)
730 				fetch_info("passive mode failed");
731 			/* Close socket and retry with passive mode. */
732 			pasv = 0;
733 			close(sd);
734 			sd = -1;
735 			goto retry_mode;
736 		}
737 
738 		/* seek to required offset */
739 		if (offset)
740 			if (ftp_cmd(conn, "REST %lu", (unsigned long)offset) != FTP_FILE_OK)
741 				goto sysouch;
742 
743 		/* construct sockaddr for data socket */
744 		l = sizeof(sa);
745 		if (getpeername(conn->sd, (struct sockaddr *)&sa, &l) == -1)
746 			goto sysouch;
747 		if (sa.ss_family == AF_INET6)
748 			unmappedaddr((struct sockaddr_in6 *)&sa, &l);
749 		switch (sa.ss_family) {
750 		case AF_INET6:
751 			sin6 = (struct sockaddr_in6 *)&sa;
752 			if (e == FTP_EPASSIVE_MODE)
753 				sin6->sin6_port = htons(port);
754 			else {
755 				memcpy(&sin6->sin6_addr, addr + 2, 16);
756 				memcpy(&sin6->sin6_port, addr + 19, 2);
757 			}
758 			break;
759 		case AF_INET:
760 			sin4 = (struct sockaddr_in *)&sa;
761 			if (e == FTP_EPASSIVE_MODE)
762 				sin4->sin_port = htons(port);
763 			else {
764 				memcpy(&sin4->sin_addr, addr, 4);
765 				memcpy(&sin4->sin_port, addr + 4, 2);
766 			}
767 			break;
768 		default:
769 			e = FTP_PROTOCOL_ERROR; /* XXX: error code should be prepared */
770 			break;
771 		}
772 
773 		/* connect to data port */
774 		if (verbose)
775 			fetch_info("opening data connection");
776 		bindaddr = getenv("FETCH_BIND_ADDRESS");
777 		if (bindaddr != NULL && *bindaddr != '\0' &&
778 		    fetch_bind(sd, sa.ss_family, bindaddr) != 0)
779 			goto sysouch;
780 		if (connect(sd, (struct sockaddr *)&sa, l) == -1)
781 			goto sysouch;
782 
783 		/* make the server initiate the transfer */
784 		if (verbose)
785 			fetch_info("initiating transfer");
786 		if (op_arg)
787 			e = ftp_cmd(conn, "%s%s%s", oper, *op_arg ? " " : "", op_arg);
788 		else
789 			e = ftp_cmd(conn, "%s %.*s", oper,
790 			    filenamelen, filename);
791 		if (e != FTP_CONNECTION_ALREADY_OPEN && e != FTP_OPEN_DATA_CONNECTION)
792 			goto ouch;
793 
794 	} else {
795 		uint32_t a;
796 		uint16_t p;
797 #if defined(IPV6_PORTRANGE) || defined(IP_PORTRANGE)
798 		int arg;
799 #endif
800 		int d;
801 		char *ap;
802 		char hname[INET6_ADDRSTRLEN];
803 
804 		switch (sa.ss_family) {
805 		case AF_INET6:
806 			((struct sockaddr_in6 *)&sa)->sin6_port = 0;
807 #ifdef IPV6_PORTRANGE
808 			arg = low ? IPV6_PORTRANGE_DEFAULT : IPV6_PORTRANGE_HIGH;
809 			if (setsockopt(sd, IPPROTO_IPV6, IPV6_PORTRANGE,
810 				(char *)&arg, sizeof(arg)) == -1)
811 				goto sysouch;
812 #endif
813 			break;
814 		case AF_INET:
815 			((struct sockaddr_in *)&sa)->sin_port = 0;
816 #ifdef IP_PORTRANGE
817 			arg = low ? IP_PORTRANGE_DEFAULT : IP_PORTRANGE_HIGH;
818 			if (setsockopt(sd, IPPROTO_IP, IP_PORTRANGE,
819 				(char *)&arg, sizeof(arg)) == -1)
820 				goto sysouch;
821 #endif
822 			break;
823 		}
824 		if (verbose)
825 			fetch_info("binding data socket");
826 		if (bind(sd, (struct sockaddr *)&sa, l) == -1)
827 			goto sysouch;
828 		if (listen(sd, 1) == -1)
829 			goto sysouch;
830 
831 		/* find what port we're on and tell the server */
832 		if (getsockname(sd, (struct sockaddr *)&sa, &l) == -1)
833 			goto sysouch;
834 		switch (sa.ss_family) {
835 		case AF_INET:
836 			sin4 = (struct sockaddr_in *)&sa;
837 			a = ntohl(sin4->sin_addr.s_addr);
838 			p = ntohs(sin4->sin_port);
839 			e = ftp_cmd(conn, "PORT %d,%d,%d,%d,%d,%d",
840 			    (a >> 24) & 0xff, (a >> 16) & 0xff,
841 			    (a >> 8) & 0xff, a & 0xff,
842 			    (p >> 8) & 0xff, p & 0xff);
843 			break;
844 		case AF_INET6:
845 #define UC(b)	(((int)b)&0xff)
846 			e = -1;
847 			sin6 = (struct sockaddr_in6 *)&sa;
848 			sin6->sin6_scope_id = 0;
849 			if (getnameinfo((struct sockaddr *)&sa, l,
850 				hname, sizeof(hname),
851 				NULL, 0, NI_NUMERICHOST) == 0) {
852 				e = ftp_cmd(conn, "EPRT |%d|%s|%d|", 2, hname,
853 				    htons(sin6->sin6_port));
854 				if (e == -1)
855 					goto ouch;
856 			}
857 			if (e != FTP_OK) {
858 				ap = (char *)&sin6->sin6_addr;
859 				e = ftp_cmd(conn,
860 				    "LPRT %d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d",
861 				    6, 16,
862 				    UC(ap[0]), UC(ap[1]), UC(ap[2]), UC(ap[3]),
863 				    UC(ap[4]), UC(ap[5]), UC(ap[6]), UC(ap[7]),
864 				    UC(ap[8]), UC(ap[9]), UC(ap[10]), UC(ap[11]),
865 				    UC(ap[12]), UC(ap[13]), UC(ap[14]), UC(ap[15]),
866 				    2,
867 				    (ntohs(sin6->sin6_port) >> 8) & 0xff,
868 				    ntohs(sin6->sin6_port)        & 0xff);
869 			}
870 			break;
871 		default:
872 			e = FTP_PROTOCOL_ERROR; /* XXX: error code should be prepared */
873 			goto ouch;
874 		}
875 		if (e != FTP_OK)
876 			goto ouch;
877 
878 		/* seek to required offset */
879 		if (offset)
880 			if (ftp_cmd(conn, "REST %llu", (unsigned long long)offset) != FTP_FILE_OK)
881 				goto sysouch;
882 
883 		/* make the server initiate the transfer */
884 		if (verbose)
885 			fetch_info("initiating transfer");
886 		if (op_arg)
887 			e = ftp_cmd(conn, "%s%s%s", oper, *op_arg ? " " : "", op_arg);
888 		else
889 			e = ftp_cmd(conn, "%s %.*s", oper,
890 			    filenamelen, filename);
891 		if (e != FTP_CONNECTION_ALREADY_OPEN && e != FTP_OPEN_DATA_CONNECTION)
892 			goto ouch;
893 
894 		/* accept the incoming connection and go to town */
895 		if ((d = accept(sd, NULL, NULL)) == -1)
896 			goto sysouch;
897 		close(sd);
898 		sd = d;
899 	}
900 
901 	if ((df = ftp_setup(conn, fetch_reopen(sd), mode)) == NULL)
902 		goto sysouch;
903 	return (df);
904 
905 sysouch:
906 	fetch_syserr();
907 	if (sd >= 0)
908 		close(sd);
909 	return (NULL);
910 
911 ouch:
912 	if (e != -1)
913 		ftp_seterr(e);
914 	if (sd >= 0)
915 		close(sd);
916 	return (NULL);
917 }
918 
919 /*
920  * Authenticate
921  */
922 static int
923 ftp_authenticate(conn_t *conn, struct url *url, struct url *purl)
924 {
925 	const char *user, *pwd, *login_name;
926 	char pbuf[URL_USERLEN + 1 + URL_HOSTLEN + 1];
927 	int e, len;
928 
929 	/* XXX FTP_AUTH, and maybe .netrc */
930 
931 	/* send user name and password */
932 	if (url->user[0] == '\0')
933 		fetch_netrc_auth(url);
934 	user = url->user;
935 	if (*user == '\0')
936 		user = getenv("FTP_LOGIN");
937 	if (user == NULL || *user == '\0')
938 		user = FTP_ANONYMOUS_USER;
939 	if (purl && url->port == fetch_default_port(url->scheme))
940 		e = ftp_cmd(conn, "USER %s@%s", user, url->host);
941 	else if (purl)
942 		e = ftp_cmd(conn, "USER %s@%s@%d", user, url->host, url->port);
943 	else
944 		e = ftp_cmd(conn, "USER %s", user);
945 
946 	/* did the server request a password? */
947 	if (e == FTP_NEED_PASSWORD) {
948 		pwd = url->pwd;
949 		if (*pwd == '\0')
950 			pwd = getenv("FTP_PASSWORD");
951 		if (pwd == NULL || *pwd == '\0') {
952 			if ((login_name = getlogin()) == 0)
953 				login_name = FTP_ANONYMOUS_USER;
954 			if ((len = snprintf(pbuf, URL_USERLEN + 2, "%s@", login_name)) < 0)
955 				len = 0;
956 			else if (len > URL_USERLEN + 1)
957 				len = URL_USERLEN + 1;
958 			gethostname(pbuf + len, sizeof(pbuf) - len);
959 			/* MAXHOSTNAMELEN can differ from URL_HOSTLEN + 1 */
960 			pbuf[sizeof(pbuf) - 1] = '\0';
961 			pwd = pbuf;
962 		}
963 		e = ftp_cmd(conn, "PASS %s", pwd);
964 	}
965 
966 	return (e);
967 }
968 
969 /*
970  * Log on to FTP server
971  */
972 static conn_t *
973 ftp_connect(struct url *url, struct url *purl, const char *flags)
974 {
975 	conn_t *conn;
976 	int e, direct, verbose;
977 #ifdef INET6
978 	int af = AF_UNSPEC;
979 #else
980 	int af = AF_INET;
981 #endif
982 
983 	direct = CHECK_FLAG('d');
984 	verbose = CHECK_FLAG('v');
985 	if (CHECK_FLAG('4'))
986 		af = AF_INET;
987 	else if (CHECK_FLAG('6'))
988 		af = AF_INET6;
989 
990 	if (direct)
991 		purl = NULL;
992 
993 	/* check for proxy */
994 	if (purl) {
995 		/* XXX proxy authentication! */
996 		conn = fetch_connect(purl->host, purl->port, af, verbose);
997 	} else {
998 		/* no proxy, go straight to target */
999 		conn = fetch_connect(url->host, url->port, af, verbose);
1000 		purl = NULL;
1001 	}
1002 
1003 	/* check connection */
1004 	if (conn == NULL)
1005 		/* fetch_connect() has already set an error code */
1006 		return (NULL);
1007 
1008 	/* expect welcome message */
1009 	if ((e = ftp_chkerr(conn)) != FTP_SERVICE_READY)
1010 		goto fouch;
1011 
1012 	/* authenticate */
1013 	if ((e = ftp_authenticate(conn, url, purl)) != FTP_LOGGED_IN)
1014 		goto fouch;
1015 
1016 	/* TODO: Request extended features supported, if any (RFC 3659). */
1017 
1018 	/* done */
1019 	return (conn);
1020 
1021 fouch:
1022 	if (e != -1)
1023 		ftp_seterr(e);
1024 	fetch_close(conn);
1025 	return (NULL);
1026 }
1027 
1028 /*
1029  * Disconnect from server
1030  */
1031 static void
1032 ftp_disconnect(conn_t *conn)
1033 {
1034 	(void)ftp_cmd(conn, "QUIT");
1035 	if (conn == cached_connection && conn->ref == 1)
1036 		cached_connection = NULL;
1037 	fetch_close(conn);
1038 }
1039 
1040 /*
1041  * Check if we're already connected
1042  */
1043 static int
1044 ftp_isconnected(struct url *url)
1045 {
1046 	return (cached_connection
1047 	    && (strcmp(url->host, cached_host.host) == 0)
1048 	    && (strcmp(url->user, cached_host.user) == 0)
1049 	    && (strcmp(url->pwd, cached_host.pwd) == 0)
1050 	    && (url->port == cached_host.port));
1051 }
1052 
1053 /*
1054  * Check the cache, reconnect if no luck
1055  */
1056 static conn_t *
1057 ftp_cached_connect(struct url *url, struct url *purl, const char *flags)
1058 {
1059 	conn_t *conn;
1060 	int e;
1061 
1062 	/* set default port */
1063 	if (!url->port)
1064 		url->port = fetch_default_port(url->scheme);
1065 
1066 	/* try to use previously cached connection */
1067 	if (ftp_isconnected(url)) {
1068 		e = ftp_cmd(cached_connection, "NOOP");
1069 		if (e == FTP_OK || e == FTP_SYNTAX_ERROR)
1070 			return (fetch_ref(cached_connection));
1071 	}
1072 
1073 	/* connect to server */
1074 	if ((conn = ftp_connect(url, purl, flags)) == NULL)
1075 		return (NULL);
1076 	if (cached_connection)
1077 		ftp_disconnect(cached_connection);
1078 	cached_connection = fetch_ref(conn);
1079 	memcpy(&cached_host, url, sizeof(*url));
1080 	return (conn);
1081 }
1082 
1083 /*
1084  * Check the proxy settings
1085  */
1086 static struct url *
1087 ftp_get_proxy(struct url * url, const char *flags)
1088 {
1089 	struct url *purl;
1090 	char *p;
1091 
1092 	if (flags != NULL && strchr(flags, 'd') != NULL)
1093 		return (NULL);
1094 	if (fetch_no_proxy_match(url->host))
1095 		return (NULL);
1096 	if (((p = getenv("FTP_PROXY")) || (p = getenv("ftp_proxy")) ||
1097 		(p = getenv("HTTP_PROXY")) || (p = getenv("http_proxy"))) &&
1098 	    *p && (purl = fetchParseURL(p)) != NULL) {
1099 		if (!*purl->scheme) {
1100 			if (getenv("FTP_PROXY") || getenv("ftp_proxy"))
1101 				strcpy(purl->scheme, SCHEME_FTP);
1102 			else
1103 				strcpy(purl->scheme, SCHEME_HTTP);
1104 		}
1105 		if (!purl->port)
1106 			purl->port = fetch_default_proxy_port(purl->scheme);
1107 		if (strcasecmp(purl->scheme, SCHEME_FTP) == 0 ||
1108 		    strcasecmp(purl->scheme, SCHEME_HTTP) == 0)
1109 			return (purl);
1110 		fetchFreeURL(purl);
1111 	}
1112 	return (NULL);
1113 }
1114 
1115 /*
1116  * Process an FTP request
1117  */
1118 fetchIO *
1119 ftp_request(struct url *url, const char *op, const char *op_arg,
1120     struct url_stat *us, struct url *purl, const char *flags)
1121 {
1122 	fetchIO *f;
1123 	char *path;
1124 	conn_t *conn;
1125 	int oflag;
1126 
1127 	/* check if we should use HTTP instead */
1128 	if (purl && strcasecmp(purl->scheme, SCHEME_HTTP) == 0) {
1129 		if (strcmp(op, "STAT") == 0)
1130 			return (http_request(url, "HEAD", us, purl, flags));
1131 		else if (strcmp(op, "RETR") == 0)
1132 			return (http_request(url, "GET", us, purl, flags));
1133 		/*
1134 		 * Our HTTP code doesn't support PUT requests yet, so try
1135 		 * a direct connection.
1136 		 */
1137 	}
1138 
1139 	/* connect to server */
1140 	conn = ftp_cached_connect(url, purl, flags);
1141 	if (purl)
1142 		fetchFreeURL(purl);
1143 	if (conn == NULL)
1144 		return (NULL);
1145 
1146 	if ((path = fetchUnquotePath(url)) == NULL) {
1147 		fetch_syserr();
1148 		return NULL;
1149 	}
1150 
1151 	/* change directory */
1152 	if (ftp_cwd(conn, path, op_arg != NULL) == -1) {
1153 		free(path);
1154 		return (NULL);
1155 	}
1156 
1157 	/* stat file */
1158 	if (us && ftp_stat(conn, path, us) == -1
1159 	    && fetchLastErrCode != FETCH_PROTO
1160 	    && fetchLastErrCode != FETCH_UNAVAIL) {
1161 		free(path);
1162 		return (NULL);
1163 	}
1164 
1165 	/* just a stat */
1166 	if (strcmp(op, "STAT") == 0)
1167 		return fetchIO_unopen(NULL, NULL, NULL, NULL);
1168 	if (strcmp(op, "STOR") == 0 || strcmp(op, "APPE") == 0)
1169 		oflag = O_WRONLY;
1170 	else
1171 		oflag = O_RDONLY;
1172 
1173 	/* initiate the transfer */
1174 	f = (ftp_transfer(conn, op, path, op_arg, oflag, url->offset, flags));
1175 	free(path);
1176 	return f;
1177 }
1178 
1179 /*
1180  * Get and stat file
1181  */
1182 fetchIO *
1183 fetchXGetFTP(struct url *url, struct url_stat *us, const char *flags)
1184 {
1185 	return (ftp_request(url, "RETR", NULL, us, ftp_get_proxy(url, flags), flags));
1186 }
1187 
1188 /*
1189  * Get file
1190  */
1191 fetchIO *
1192 fetchGetFTP(struct url *url, const char *flags)
1193 {
1194 	return (fetchXGetFTP(url, NULL, flags));
1195 }
1196 
1197 /*
1198  * Put file
1199  */
1200 fetchIO *
1201 fetchPutFTP(struct url *url, const char *flags)
1202 {
1203 	return (ftp_request(url, CHECK_FLAG('a') ? "APPE" : "STOR", NULL, NULL,
1204 	    ftp_get_proxy(url, flags), flags));
1205 }
1206 
1207 /*
1208  * Get file stats
1209  */
1210 int
1211 fetchStatFTP(struct url *url, struct url_stat *us, const char *flags)
1212 {
1213 	fetchIO *f;
1214 
1215 	f = ftp_request(url, "STAT", NULL, us, ftp_get_proxy(url, flags), flags);
1216 	if (f == NULL)
1217 		return (-1);
1218 	fetchIO_close(f);
1219 	return (0);
1220 }
1221 
1222 /*
1223  * List a directory
1224  */
1225 int
1226 fetchListFTP(struct url_list *ue, struct url *url, const char *pattern, const char *flags)
1227 {
1228 	fetchIO *f;
1229 	char buf[2 * PATH_MAX], *eol, *eos;
1230 	ssize_t len;
1231 	size_t cur_off;
1232 
1233 	/* XXX What about proxies? */
1234 	if (pattern == NULL || strcmp(pattern, "*") == 0)
1235 		pattern = "";
1236 	f = ftp_request(url, "NLST", pattern, NULL, ftp_get_proxy(url, flags), flags);
1237 	if (f == NULL)
1238 		return -1;
1239 
1240 	cur_off = 0;
1241 	while ((len = fetchIO_read(f, buf + cur_off, sizeof(buf) - cur_off)) > 0) {
1242 		cur_off += len;
1243 		while ((eol = memchr(buf, '\n', cur_off)) != NULL) {
1244 			if (len == eol - buf)
1245 				break;
1246 			if (eol != buf) {
1247 				if (eol[-1] == '\r')
1248 					eos = eol - 1;
1249 				else
1250 					eos = eol;
1251 				*eos = '\0';
1252 				fetch_add_entry(ue, url, buf, 0);
1253 				cur_off -= eol - buf + 1;
1254 				memmove(buf, eol + 1, cur_off);
1255 			}
1256 		}
1257 	}
1258 	if (cur_off != 0 || len < 0) {
1259 		/* Not RFC conform, bail out. */
1260 		fetchIO_close(f);
1261 		return -1;
1262 	}
1263 	fetchIO_close(f);
1264 	return 0;
1265 }
1266