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