fetch.c revision 1.35 1 1.35 lukem /* $NetBSD: fetch.c,v 1.35 1998/09/01 04:42:49 lukem Exp $ */
2 1.1 lukem
3 1.1 lukem /*-
4 1.25 lukem * Copyright (c) 1997, 1998 The NetBSD Foundation, Inc.
5 1.1 lukem * All rights reserved.
6 1.1 lukem *
7 1.1 lukem * This code is derived from software contributed to The NetBSD Foundation
8 1.1 lukem * by Jason Thorpe and Luke Mewburn.
9 1.1 lukem *
10 1.1 lukem * Redistribution and use in source and binary forms, with or without
11 1.1 lukem * modification, are permitted provided that the following conditions
12 1.1 lukem * are met:
13 1.1 lukem * 1. Redistributions of source code must retain the above copyright
14 1.1 lukem * notice, this list of conditions and the following disclaimer.
15 1.1 lukem * 2. Redistributions in binary form must reproduce the above copyright
16 1.1 lukem * notice, this list of conditions and the following disclaimer in the
17 1.1 lukem * documentation and/or other materials provided with the distribution.
18 1.1 lukem * 3. All advertising materials mentioning features or use of this software
19 1.1 lukem * must display the following acknowledgement:
20 1.1 lukem * This product includes software developed by the NetBSD
21 1.1 lukem * Foundation, Inc. and its contributors.
22 1.1 lukem * 4. Neither the name of The NetBSD Foundation nor the names of its
23 1.1 lukem * contributors may be used to endorse or promote products derived
24 1.1 lukem * from this software without specific prior written permission.
25 1.1 lukem *
26 1.1 lukem * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
27 1.1 lukem * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
28 1.1 lukem * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
29 1.14 lukem * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
30 1.14 lukem * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
31 1.1 lukem * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
32 1.1 lukem * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
33 1.1 lukem * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
34 1.1 lukem * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
35 1.1 lukem * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
36 1.1 lukem * POSSIBILITY OF SUCH DAMAGE.
37 1.1 lukem */
38 1.1 lukem
39 1.12 lukem #include <sys/cdefs.h>
40 1.1 lukem #ifndef lint
41 1.35 lukem __RCSID("$NetBSD: fetch.c,v 1.35 1998/09/01 04:42:49 lukem Exp $");
42 1.1 lukem #endif /* not lint */
43 1.1 lukem
44 1.1 lukem /*
45 1.1 lukem * FTP User Program -- Command line file retrieval
46 1.1 lukem */
47 1.1 lukem
48 1.1 lukem #include <sys/types.h>
49 1.1 lukem #include <sys/param.h>
50 1.1 lukem #include <sys/socket.h>
51 1.30 lukem #include <sys/stat.h>
52 1.32 lukem #include <sys/time.h>
53 1.1 lukem
54 1.1 lukem #include <netinet/in.h>
55 1.1 lukem
56 1.2 lukem #include <arpa/ftp.h>
57 1.1 lukem #include <arpa/inet.h>
58 1.1 lukem
59 1.1 lukem #include <ctype.h>
60 1.1 lukem #include <err.h>
61 1.22 lukem #include <errno.h>
62 1.1 lukem #include <netdb.h>
63 1.1 lukem #include <fcntl.h>
64 1.3 lukem #include <signal.h>
65 1.1 lukem #include <stdio.h>
66 1.1 lukem #include <stdlib.h>
67 1.1 lukem #include <string.h>
68 1.1 lukem #include <unistd.h>
69 1.24 lukem #include <util.h>
70 1.1 lukem
71 1.1 lukem #include "ftp_var.h"
72 1.1 lukem
73 1.27 lukem typedef enum {
74 1.27 lukem UNKNOWN_URL_T=-1,
75 1.27 lukem HTTP_URL_T,
76 1.27 lukem FTP_URL_T,
77 1.27 lukem FILE_URL_T
78 1.27 lukem } url_t;
79 1.27 lukem
80 1.27 lukem static int parse_url __P((const char *, const char *, url_t *, char **,
81 1.27 lukem char **, char **, in_port_t *, char **));
82 1.22 lukem static int url_get __P((const char *, const char *, const char *));
83 1.12 lukem void aborthttp __P((int));
84 1.12 lukem
85 1.12 lukem
86 1.25 lukem #define ABOUT_URL "about:" /* propaganda */
87 1.25 lukem #define FILE_URL "file://" /* file URL prefix */
88 1.1 lukem #define FTP_URL "ftp://" /* ftp URL prefix */
89 1.1 lukem #define HTTP_URL "http://" /* http URL prefix */
90 1.5 lukem #define FTP_PROXY "ftp_proxy" /* env var with ftp proxy location */
91 1.1 lukem #define HTTP_PROXY "http_proxy" /* env var with http proxy location */
92 1.27 lukem #define NO_PROXY "no_proxy" /* env var with list of non-proxied
93 1.27 lukem * hosts, comma or space separated */
94 1.1 lukem
95 1.1 lukem
96 1.1 lukem #define EMPTYSTRING(x) ((x) == NULL || (*(x) == '\0'))
97 1.27 lukem #define FREEPTR(x) if ((x) != NULL) { free(x); (x) = NULL; }
98 1.27 lukem
99 1.27 lukem /*
100 1.27 lukem * Parse URL of form:
101 1.27 lukem * <type>://[<user>[:<password>@]]<host>[:<port>]/<url-path>
102 1.27 lukem * Returns -1 if a parse error occurred, otherwise 0.
103 1.27 lukem * Sets type to url_t, each of the given char ** pointers to a
104 1.27 lukem * malloc(3)ed strings of the relevant section, and port to
105 1.27 lukem * 0 if not given, or the number given.
106 1.27 lukem */
107 1.27 lukem static int
108 1.27 lukem parse_url(url, desc, type, user, pass, host, port, path)
109 1.27 lukem const char *url;
110 1.27 lukem const char *desc;
111 1.27 lukem url_t *type;
112 1.27 lukem char **user;
113 1.27 lukem char **pass;
114 1.27 lukem char **host;
115 1.27 lukem in_port_t *port;
116 1.27 lukem char **path;
117 1.27 lukem {
118 1.27 lukem char *cp, *ep, *thost;
119 1.27 lukem
120 1.27 lukem if (url == NULL || desc == NULL || type == NULL || user == NULL
121 1.27 lukem || pass == NULL || host == NULL || port == NULL || path == NULL)
122 1.27 lukem errx(1, "parse_url: invoked with NULL argument!");
123 1.27 lukem
124 1.27 lukem *type = UNKNOWN_URL_T;
125 1.27 lukem *user = *pass = *host = *path = NULL;
126 1.27 lukem *port = 0;
127 1.27 lukem
128 1.27 lukem if (strncasecmp(url, HTTP_URL, sizeof(HTTP_URL) - 1) == 0) {
129 1.27 lukem url += sizeof(HTTP_URL) - 1;
130 1.27 lukem *type = HTTP_URL_T;
131 1.27 lukem } else if (strncasecmp(url, FTP_URL, sizeof(FTP_URL) - 1) == 0) {
132 1.27 lukem url += sizeof(FTP_URL) - 1;
133 1.27 lukem *type = FTP_URL_T;
134 1.27 lukem } else if (strncasecmp(url, FILE_URL, sizeof(FILE_URL) - 1) == 0) {
135 1.27 lukem url += sizeof(FILE_URL) - 1;
136 1.27 lukem *type = FILE_URL_T;
137 1.27 lukem } else {
138 1.27 lukem warnx("Invalid %s `%s'", desc, url);
139 1.27 lukem cleanup_parse_url:
140 1.27 lukem FREEPTR(*user);
141 1.27 lukem FREEPTR(*pass);
142 1.27 lukem FREEPTR(*host);
143 1.27 lukem FREEPTR(*path);
144 1.27 lukem return (-1);
145 1.27 lukem }
146 1.27 lukem
147 1.27 lukem if (*url == '\0')
148 1.27 lukem return (0);
149 1.27 lukem
150 1.27 lukem /* find [user[:pass]@]host[:port] */
151 1.27 lukem ep = strchr(url, '/');
152 1.27 lukem if (ep == NULL)
153 1.27 lukem thost = xstrdup(url);
154 1.27 lukem else {
155 1.27 lukem size_t len = ep - url;
156 1.27 lukem thost = (char *)xmalloc(len + 1);
157 1.27 lukem strncpy(thost, url, len);
158 1.27 lukem thost[len] = '\0';
159 1.27 lukem *path = xstrdup(ep);
160 1.27 lukem }
161 1.27 lukem
162 1.27 lukem cp = strchr(thost, '@');
163 1.27 lukem if (cp != NULL) {
164 1.27 lukem *user = thost;
165 1.27 lukem *cp = '\0';
166 1.27 lukem *host = xstrdup(cp + 1);
167 1.27 lukem cp = strchr(*user, ':');
168 1.27 lukem if (cp != NULL) {
169 1.27 lukem *cp = '\0';
170 1.27 lukem *pass = xstrdup(cp + 1);
171 1.27 lukem }
172 1.27 lukem } else
173 1.27 lukem *host = thost;
174 1.27 lukem
175 1.27 lukem /* look for [:port] */
176 1.27 lukem cp = strrchr(*host, ':');
177 1.27 lukem if (cp != NULL) {
178 1.27 lukem long nport;
179 1.27 lukem
180 1.27 lukem *cp = '\0';
181 1.27 lukem nport = strtol(cp + 1, &ep, 10);
182 1.27 lukem if (nport < 1 || nport > MAX_IN_PORT_T || *ep != '\0') {
183 1.27 lukem warnx("Invalid port `%s' in %s `%s'", cp, desc, line);
184 1.27 lukem goto cleanup_parse_url;
185 1.27 lukem }
186 1.27 lukem *port = htons((in_port_t)nport);
187 1.27 lukem }
188 1.27 lukem
189 1.27 lukem if (debug)
190 1.27 lukem fprintf(ttyout,
191 1.27 lukem "parse_url: user `%s', pass `%s', host %s:%d, path `%s'\n",
192 1.27 lukem *user ? *user : "", *pass ? *pass : "", *host ? *host : "",
193 1.27 lukem ntohs(*port), *path ? *path : "");
194 1.27 lukem
195 1.27 lukem return (0);
196 1.27 lukem }
197 1.27 lukem
198 1.1 lukem
199 1.2 lukem jmp_buf httpabort;
200 1.2 lukem
201 1.1 lukem /*
202 1.5 lukem * Retrieve URL, via the proxy in $proxyvar if necessary.
203 1.1 lukem * Modifies the string argument given.
204 1.1 lukem * Returns -1 on failure, 0 on success
205 1.1 lukem */
206 1.12 lukem static int
207 1.27 lukem url_get(url, proxyenv, outfile)
208 1.27 lukem const char *url;
209 1.9 lukem const char *proxyenv;
210 1.22 lukem const char *outfile;
211 1.1 lukem {
212 1.1 lukem struct sockaddr_in sin;
213 1.27 lukem int isredirected, isproxy;
214 1.12 lukem volatile int s;
215 1.13 lukem size_t len;
216 1.27 lukem char *cp, *ep;
217 1.29 lukem char *buf, *savefile;
218 1.22 lukem volatile sig_t oldintr, oldintp;
219 1.1 lukem off_t hashbytes;
220 1.22 lukem struct hostent *hp = NULL;
221 1.22 lukem int (*closefunc) __P((FILE *));
222 1.24 lukem FILE *fin, *fout;
223 1.24 lukem int retval;
224 1.25 lukem time_t mtime;
225 1.27 lukem url_t urltype;
226 1.27 lukem char *user, *pass, *host;
227 1.27 lukem in_port_t port;
228 1.27 lukem char *path;
229 1.1 lukem
230 1.22 lukem closefunc = NULL;
231 1.24 lukem fin = fout = NULL;
232 1.1 lukem s = -1;
233 1.29 lukem buf = savefile = NULL;
234 1.27 lukem isredirected = isproxy = 0;
235 1.24 lukem retval = -1;
236 1.1 lukem
237 1.27 lukem #ifdef __GNUC__ /* shut up gcc warnings */
238 1.22 lukem (void)&closefunc;
239 1.24 lukem (void)&fin;
240 1.22 lukem (void)&fout;
241 1.24 lukem (void)&buf;
242 1.12 lukem (void)&savefile;
243 1.24 lukem (void)&retval;
244 1.28 fvdl (void)&isproxy;
245 1.12 lukem #endif
246 1.12 lukem
247 1.27 lukem if (parse_url(url, "URL", &urltype, &user, &pass, &host, &port, &path)
248 1.27 lukem == -1)
249 1.27 lukem goto cleanup_url_get;
250 1.27 lukem if (port == 0)
251 1.27 lukem port = httpport;
252 1.5 lukem
253 1.27 lukem if (urltype == FILE_URL_T && ! EMPTYSTRING(host)
254 1.27 lukem && strcasecmp(host, "localhost") != 0) {
255 1.27 lukem warnx("No support for non local file URL `%s'", url);
256 1.5 lukem goto cleanup_url_get;
257 1.9 lukem }
258 1.27 lukem
259 1.9 lukem if (EMPTYSTRING(path)) {
260 1.25 lukem if (urltype == FTP_URL_T)
261 1.11 lukem goto noftpautologin;
262 1.27 lukem if (urltype != HTTP_URL_T || outfile == NULL) {
263 1.27 lukem warnx("Invalid URL (no file after host) `%s'", url);
264 1.27 lukem goto cleanup_url_get;
265 1.27 lukem }
266 1.9 lukem }
267 1.1 lukem
268 1.22 lukem if (outfile)
269 1.29 lukem savefile = xstrdup(outfile);
270 1.22 lukem else {
271 1.29 lukem cp = strrchr(path, '/'); /* find savefile */
272 1.29 lukem if (cp != NULL)
273 1.29 lukem savefile = xstrdup(cp + 1);
274 1.22 lukem else
275 1.29 lukem savefile = xstrdup(path);
276 1.22 lukem }
277 1.9 lukem if (EMPTYSTRING(savefile)) {
278 1.25 lukem if (urltype == FTP_URL_T)
279 1.11 lukem goto noftpautologin;
280 1.27 lukem warnx("Invalid URL (no file after directory) `%s'", url);
281 1.5 lukem goto cleanup_url_get;
282 1.9 lukem }
283 1.1 lukem
284 1.25 lukem filesize = -1;
285 1.25 lukem mtime = -1;
286 1.25 lukem if (urltype == FILE_URL_T) { /* file:// URLs */
287 1.25 lukem struct stat sb;
288 1.25 lukem
289 1.25 lukem direction = "copied";
290 1.25 lukem fin = fopen(path, "r");
291 1.25 lukem if (fin == NULL) {
292 1.27 lukem warn("Cannot open file `%s'", path);
293 1.5 lukem goto cleanup_url_get;
294 1.5 lukem }
295 1.25 lukem if (fstat(fileno(fin), &sb) == 0) {
296 1.25 lukem mtime = sb.st_mtime;
297 1.25 lukem filesize = sb.st_size;
298 1.25 lukem }
299 1.25 lukem fprintf(ttyout, "Copying %s\n", path);
300 1.25 lukem } else { /* ftp:// or http:// URLs */
301 1.25 lukem direction = "retrieved";
302 1.25 lukem if (proxyenv != NULL) { /* use proxy */
303 1.27 lukem url_t purltype;
304 1.27 lukem char *puser, *ppass, *phost;
305 1.27 lukem in_port_t pport;
306 1.27 lukem char *ppath;
307 1.27 lukem char *no_proxy;
308 1.27 lukem
309 1.27 lukem isproxy = 1;
310 1.27 lukem
311 1.27 lukem /* check URL against list of no_proxied sites */
312 1.27 lukem no_proxy = getenv(NO_PROXY);
313 1.27 lukem if (no_proxy != NULL) {
314 1.27 lukem char *np, *np_copy;
315 1.27 lukem long np_port;
316 1.27 lukem size_t hlen, plen;
317 1.27 lukem
318 1.27 lukem np_copy = xstrdup(no_proxy);
319 1.27 lukem hlen = strlen(host);
320 1.27 lukem while ((cp = strsep(&np_copy, " ,")) != NULL) {
321 1.27 lukem if (*cp == '\0')
322 1.27 lukem continue;
323 1.27 lukem if ((np = strchr(cp, ':')) != NULL) {
324 1.27 lukem *np = '\0';
325 1.27 lukem np_port =
326 1.27 lukem strtol(np + 1, &ep, 10);
327 1.27 lukem if (*ep != '\0')
328 1.27 lukem continue;
329 1.27 lukem if (port !=
330 1.27 lukem htons((in_port_t)np_port))
331 1.27 lukem continue;
332 1.27 lukem }
333 1.27 lukem plen = strlen(cp);
334 1.27 lukem if (strncasecmp(host + hlen - plen,
335 1.27 lukem cp, plen) == 0) {
336 1.27 lukem isproxy = 0;
337 1.27 lukem break;
338 1.27 lukem }
339 1.27 lukem }
340 1.27 lukem FREEPTR(np_copy);
341 1.25 lukem }
342 1.1 lukem
343 1.27 lukem if (isproxy) {
344 1.27 lukem if (parse_url(proxyenv, "proxy URL", &purltype,
345 1.27 lukem &puser, &ppass, &phost, &pport, &ppath)
346 1.27 lukem == -1)
347 1.27 lukem goto cleanup_url_get;
348 1.27 lukem
349 1.27 lukem if ((purltype != HTTP_URL_T
350 1.27 lukem && purltype != FTP_URL_T) ||
351 1.27 lukem EMPTYSTRING(phost) ||
352 1.27 lukem (! EMPTYSTRING(ppath)
353 1.27 lukem && strcmp(ppath, "/") != 0)) {
354 1.27 lukem warnx("Malformed proxy URL `%s'",
355 1.27 lukem proxyenv);
356 1.27 lukem FREEPTR(puser);
357 1.27 lukem FREEPTR(ppass);
358 1.27 lukem FREEPTR(phost);
359 1.27 lukem FREEPTR(ppath);
360 1.27 lukem goto cleanup_url_get;
361 1.27 lukem }
362 1.12 lukem
363 1.27 lukem FREEPTR(user);
364 1.27 lukem user = puser;
365 1.27 lukem FREEPTR(pass);
366 1.27 lukem pass = ppass;
367 1.27 lukem FREEPTR(host);
368 1.27 lukem host = phost;
369 1.27 lukem if (pport == 0)
370 1.27 lukem port = httpport;
371 1.27 lukem else
372 1.27 lukem port = pport;
373 1.27 lukem FREEPTR(path);
374 1.27 lukem FREEPTR(ppath);
375 1.27 lukem path = xstrdup(url);
376 1.27 lukem }
377 1.27 lukem } /* proxyenv != NULL */
378 1.25 lukem
379 1.25 lukem memset(&sin, 0, sizeof(sin));
380 1.25 lukem sin.sin_family = AF_INET;
381 1.25 lukem
382 1.25 lukem if (isdigit((unsigned char)host[0])) {
383 1.25 lukem if (inet_aton(host, &sin.sin_addr) == 0) {
384 1.27 lukem warnx("Invalid IP address `%s'", host);
385 1.25 lukem goto cleanup_url_get;
386 1.25 lukem }
387 1.25 lukem } else {
388 1.25 lukem hp = gethostbyname(host);
389 1.25 lukem if (hp == NULL) {
390 1.25 lukem warnx("%s: %s", host, hstrerror(h_errno));
391 1.25 lukem goto cleanup_url_get;
392 1.25 lukem }
393 1.25 lukem if (hp->h_addrtype != AF_INET) {
394 1.27 lukem warnx("`%s': not an Internet address?", host);
395 1.25 lukem goto cleanup_url_get;
396 1.25 lukem }
397 1.25 lukem memcpy(&sin.sin_addr, hp->h_addr, hp->h_length);
398 1.1 lukem }
399 1.1 lukem
400 1.27 lukem if (port == 0)
401 1.25 lukem port = httpport;
402 1.25 lukem sin.sin_port = port;
403 1.25 lukem
404 1.25 lukem s = socket(AF_INET, SOCK_STREAM, 0);
405 1.25 lukem if (s == -1) {
406 1.25 lukem warn("Can't create socket");
407 1.25 lukem goto cleanup_url_get;
408 1.22 lukem }
409 1.1 lukem
410 1.25 lukem while (xconnect(s, (struct sockaddr *)&sin,
411 1.25 lukem sizeof(sin)) == -1) {
412 1.25 lukem if (errno == EINTR)
413 1.25 lukem continue;
414 1.25 lukem if (hp && hp->h_addr_list[1]) {
415 1.25 lukem int oerrno = errno;
416 1.25 lukem char *ia;
417 1.25 lukem
418 1.25 lukem ia = inet_ntoa(sin.sin_addr);
419 1.25 lukem errno = oerrno;
420 1.27 lukem warn("Connect to address `%s'", ia);
421 1.25 lukem hp->h_addr_list++;
422 1.25 lukem memcpy(&sin.sin_addr, hp->h_addr_list[0],
423 1.25 lukem (size_t)hp->h_length);
424 1.25 lukem fprintf(ttyout, "Trying %s...\n",
425 1.25 lukem inet_ntoa(sin.sin_addr));
426 1.25 lukem (void)close(s);
427 1.25 lukem s = socket(AF_INET, SOCK_STREAM, 0);
428 1.25 lukem if (s < 0) {
429 1.25 lukem warn("Can't create socket");
430 1.25 lukem goto cleanup_url_get;
431 1.25 lukem }
432 1.25 lukem continue;
433 1.25 lukem }
434 1.27 lukem warn("Can't connect to `%s'", host);
435 1.25 lukem goto cleanup_url_get;
436 1.25 lukem }
437 1.24 lukem
438 1.25 lukem fin = fdopen(s, "r+");
439 1.25 lukem /*
440 1.27 lukem * Construct and send the request.
441 1.27 lukem * Proxy requests don't want leading /.
442 1.25 lukem */
443 1.27 lukem if (isproxy) {
444 1.27 lukem fprintf(ttyout, "Requesting %s\n (via %s)\n",
445 1.27 lukem url, proxyenv);
446 1.27 lukem fprintf(fin, "GET %s HTTP/1.0\r\n\r\n", path);
447 1.27 lukem } else {
448 1.27 lukem fprintf(ttyout, "Requesting %s\n", url);
449 1.27 lukem fprintf(fin, "GET %s HTTP/1.1\r\n", path);
450 1.25 lukem fprintf(fin, "Host: %s\r\n", host);
451 1.25 lukem fprintf(fin, "Connection: close\r\n\r\n");
452 1.25 lukem }
453 1.25 lukem if (fflush(fin) == EOF) {
454 1.25 lukem warn("Writing HTTP request");
455 1.25 lukem goto cleanup_url_get;
456 1.25 lukem }
457 1.1 lukem
458 1.25 lukem /* Read the response */
459 1.24 lukem if ((buf = fparseln(fin, &len, NULL, "\0\0\0", 0)) == NULL) {
460 1.24 lukem warn("Receiving HTTP reply");
461 1.24 lukem goto cleanup_url_get;
462 1.24 lukem }
463 1.24 lukem while (len > 0 && (buf[len-1] == '\r' || buf[len-1] == '\n'))
464 1.24 lukem buf[--len] = '\0';
465 1.24 lukem if (debug)
466 1.27 lukem fprintf(ttyout, "received `%s'\n", buf);
467 1.1 lukem
468 1.25 lukem cp = strchr(buf, ' ');
469 1.25 lukem if (cp == NULL)
470 1.25 lukem goto improper;
471 1.25 lukem else
472 1.25 lukem cp++;
473 1.25 lukem if (strncmp(cp, "301", 3) == 0 || strncmp(cp, "302", 3) == 0) {
474 1.25 lukem isredirected++;
475 1.25 lukem } else if (strncmp(cp, "200", 3)) {
476 1.27 lukem warnx("Error retrieving file `%s'", cp);
477 1.25 lukem goto cleanup_url_get;
478 1.25 lukem }
479 1.25 lukem
480 1.25 lukem /* Read the rest of the header. */
481 1.27 lukem FREEPTR(buf);
482 1.25 lukem while (1) {
483 1.25 lukem if ((buf = fparseln(fin, &len, NULL, "\0\0\0", 0))
484 1.25 lukem == NULL) {
485 1.25 lukem warn("Receiving HTTP reply");
486 1.25 lukem goto cleanup_url_get;
487 1.25 lukem }
488 1.25 lukem while (len > 0 &&
489 1.25 lukem (buf[len-1] == '\r' || buf[len-1] == '\n'))
490 1.25 lukem buf[--len] = '\0';
491 1.25 lukem if (len == 0)
492 1.25 lukem break;
493 1.25 lukem if (debug)
494 1.27 lukem fprintf(ttyout, "received `%s'\n", buf);
495 1.25 lukem
496 1.25 lukem /* Look for some headers */
497 1.25 lukem cp = buf;
498 1.1 lukem #define CONTENTLEN "Content-Length: "
499 1.25 lukem if (strncasecmp(cp, CONTENTLEN,
500 1.25 lukem sizeof(CONTENTLEN) - 1) == 0) {
501 1.25 lukem cp += sizeof(CONTENTLEN) - 1;
502 1.25 lukem filesize = strtol(cp, &ep, 10);
503 1.25 lukem if (filesize < 1 || *ep != '\0')
504 1.25 lukem goto improper;
505 1.25 lukem if (debug)
506 1.25 lukem fprintf(ttyout,
507 1.25 lukem #ifndef NO_QUAD
508 1.25 lukem "parsed length as: %qd\n",
509 1.25 lukem (long long)filesize);
510 1.25 lukem #else
511 1.25 lukem "parsed length as: %ld\n",
512 1.25 lukem (long)filesize);
513 1.25 lukem #endif
514 1.25 lukem #define LASTMOD "Last-Modified: "
515 1.25 lukem } else if (strncasecmp(cp, LASTMOD,
516 1.25 lukem sizeof(LASTMOD) - 1) == 0) {
517 1.25 lukem struct tm parsed;
518 1.25 lukem char *t;
519 1.25 lukem
520 1.25 lukem cp += sizeof(LASTMOD) - 1;
521 1.25 lukem /* RFC 1123 */
522 1.25 lukem if ((t = strptime(cp,
523 1.25 lukem "%a, %d %b %Y %H:%M:%S GMT",
524 1.25 lukem &parsed))
525 1.25 lukem /* RFC 850 */
526 1.25 lukem || (t = strptime(cp,
527 1.25 lukem "%a, %d-%b-%y %H:%M:%S GMT",
528 1.25 lukem &parsed))
529 1.25 lukem /* asctime */
530 1.25 lukem || (t = strptime(cp,
531 1.25 lukem "%a, %b %d %H:%M:%S %Y",
532 1.25 lukem &parsed))) {
533 1.25 lukem if (*t == '\0')
534 1.25 lukem mtime = mktime(&parsed);
535 1.25 lukem if (debug && mtime != -1)
536 1.25 lukem fprintf(ttyout,
537 1.25 lukem "parsed date as: %s",
538 1.25 lukem ctime(&mtime));
539 1.25 lukem }
540 1.24 lukem #define LOCATION "Location: "
541 1.25 lukem } else if (isredirected &&
542 1.25 lukem strncasecmp(cp, LOCATION,
543 1.25 lukem sizeof(LOCATION) - 1) == 0) {
544 1.25 lukem cp += sizeof(LOCATION) - 1;
545 1.25 lukem if (debug)
546 1.25 lukem fprintf(ttyout,
547 1.25 lukem "parsed location as: %s\n", cp);
548 1.25 lukem if (verbose)
549 1.25 lukem fprintf(ttyout,
550 1.25 lukem "Redirected to %s\n", cp);
551 1.25 lukem retval = url_get(cp, proxyenv, outfile);
552 1.27 lukem goto cleanup_url_get;
553 1.25 lukem }
554 1.24 lukem }
555 1.27 lukem FREEPTR(buf);
556 1.1 lukem }
557 1.1 lukem
558 1.22 lukem oldintr = oldintp = NULL;
559 1.22 lukem
560 1.22 lukem /* Open the output file. */
561 1.22 lukem if (strcmp(savefile, "-") == 0) {
562 1.22 lukem fout = stdout;
563 1.22 lukem } else if (*savefile == '|') {
564 1.22 lukem oldintp = signal(SIGPIPE, SIG_IGN);
565 1.22 lukem fout = popen(savefile + 1, "w");
566 1.22 lukem if (fout == NULL) {
567 1.27 lukem warn("Can't run `%s'", savefile + 1);
568 1.22 lukem goto cleanup_url_get;
569 1.22 lukem }
570 1.22 lukem closefunc = pclose;
571 1.22 lukem } else {
572 1.22 lukem fout = fopen(savefile, "w");
573 1.22 lukem if (fout == NULL) {
574 1.27 lukem warn("Can't open `%s'", savefile);
575 1.22 lukem goto cleanup_url_get;
576 1.22 lukem }
577 1.22 lukem closefunc = fclose;
578 1.1 lukem }
579 1.1 lukem
580 1.22 lukem /* Trap signals */
581 1.2 lukem if (setjmp(httpabort)) {
582 1.2 lukem if (oldintr)
583 1.3 lukem (void)signal(SIGINT, oldintr);
584 1.22 lukem if (oldintp)
585 1.22 lukem (void)signal(SIGPIPE, oldintp);
586 1.5 lukem goto cleanup_url_get;
587 1.2 lukem }
588 1.2 lukem oldintr = signal(SIGINT, aborthttp);
589 1.2 lukem
590 1.1 lukem bytes = 0;
591 1.1 lukem hashbytes = mark;
592 1.1 lukem progressmeter(-1);
593 1.2 lukem
594 1.24 lukem /* Finally, suck down the file. */
595 1.27 lukem buf = xmalloc(BUFSIZ);
596 1.24 lukem while ((len = fread(buf, sizeof(char), BUFSIZ, fin)) > 0) {
597 1.1 lukem bytes += len;
598 1.25 lukem if (fwrite(buf, sizeof(char), len, fout) != len) {
599 1.27 lukem warn("Writing `%s'", savefile);
600 1.25 lukem goto cleanup_url_get;
601 1.1 lukem }
602 1.1 lukem if (hash && !progress) {
603 1.1 lukem while (bytes >= hashbytes) {
604 1.22 lukem (void)putc('#', ttyout);
605 1.1 lukem hashbytes += mark;
606 1.1 lukem }
607 1.22 lukem (void)fflush(ttyout);
608 1.1 lukem }
609 1.1 lukem }
610 1.1 lukem if (hash && !progress && bytes > 0) {
611 1.1 lukem if (bytes < mark)
612 1.22 lukem (void)putc('#', ttyout);
613 1.22 lukem (void)putc('\n', ttyout);
614 1.22 lukem (void)fflush(ttyout);
615 1.1 lukem }
616 1.25 lukem if (ferror(fin)) {
617 1.25 lukem warn("Reading file");
618 1.5 lukem goto cleanup_url_get;
619 1.1 lukem }
620 1.2 lukem progressmeter(1);
621 1.25 lukem (void)fflush(fout);
622 1.3 lukem (void)signal(SIGINT, oldintr);
623 1.22 lukem if (oldintp)
624 1.22 lukem (void)signal(SIGPIPE, oldintp);
625 1.25 lukem if (closefunc == fclose && mtime != -1) {
626 1.25 lukem struct timeval tval[2];
627 1.25 lukem
628 1.25 lukem (void)gettimeofday(&tval[0], NULL);
629 1.25 lukem tval[1].tv_sec = mtime;
630 1.25 lukem tval[1].tv_usec = 0;
631 1.31 lukem (*closefunc)(fout);
632 1.31 lukem fout = NULL;
633 1.31 lukem
634 1.31 lukem if (utimes(savefile, tval) == -1) {
635 1.25 lukem fprintf(ttyout,
636 1.25 lukem "Can't change modification time to %s",
637 1.25 lukem asctime(localtime(&mtime)));
638 1.25 lukem }
639 1.25 lukem }
640 1.25 lukem if (bytes > 0)
641 1.25 lukem ptransfer(0);
642 1.1 lukem
643 1.24 lukem retval = 0;
644 1.24 lukem goto cleanup_url_get;
645 1.1 lukem
646 1.11 lukem noftpautologin:
647 1.11 lukem warnx(
648 1.11 lukem "Auto-login using ftp URLs isn't supported when using $ftp_proxy");
649 1.11 lukem goto cleanup_url_get;
650 1.11 lukem
651 1.1 lukem improper:
652 1.27 lukem warnx("Improper response from `%s'", host);
653 1.11 lukem
654 1.5 lukem cleanup_url_get:
655 1.23 thorpej resetsockbufsize();
656 1.24 lukem if (fin != NULL)
657 1.24 lukem fclose(fin);
658 1.24 lukem else if (s != -1)
659 1.1 lukem close(s);
660 1.22 lukem if (closefunc != NULL && fout != NULL)
661 1.22 lukem (*closefunc)(fout);
662 1.29 lukem FREEPTR(savefile);
663 1.27 lukem FREEPTR(user);
664 1.27 lukem FREEPTR(pass);
665 1.27 lukem FREEPTR(host);
666 1.27 lukem FREEPTR(path);
667 1.27 lukem FREEPTR(buf);
668 1.24 lukem return (retval);
669 1.1 lukem }
670 1.1 lukem
671 1.1 lukem /*
672 1.2 lukem * Abort a http retrieval
673 1.2 lukem */
674 1.2 lukem void
675 1.3 lukem aborthttp(notused)
676 1.3 lukem int notused;
677 1.2 lukem {
678 1.2 lukem
679 1.2 lukem alarmtimer(0);
680 1.24 lukem fputs("\nHTTP fetch aborted.\n", ttyout);
681 1.22 lukem (void)fflush(ttyout);
682 1.2 lukem longjmp(httpabort, 1);
683 1.2 lukem }
684 1.2 lukem
685 1.2 lukem /*
686 1.1 lukem * Retrieve multiple files from the command line, transferring
687 1.25 lukem * URLs of the form "host:path", "ftp://host/path" using the
688 1.25 lukem * ftp protocol, URLs of the form "http://host/path" using the
689 1.25 lukem * http protocol, and URLs of the form "file:///" by simple
690 1.25 lukem * copying.
691 1.3 lukem * If path has a trailing "/", then return (-1);
692 1.1 lukem * the path will be cd-ed into and the connection remains open,
693 1.1 lukem * and the function will return -1 (to indicate the connection
694 1.1 lukem * is alive).
695 1.1 lukem * If an error occurs the return value will be the offset+1 in
696 1.1 lukem * argv[] of the file that caused a problem (i.e, argv[x]
697 1.1 lukem * returns x+1)
698 1.1 lukem * Otherwise, 0 is returned if all files retrieved successfully.
699 1.1 lukem */
700 1.1 lukem int
701 1.22 lukem auto_fetch(argc, argv, outfile)
702 1.1 lukem int argc;
703 1.1 lukem char *argv[];
704 1.22 lukem char *outfile;
705 1.1 lukem {
706 1.1 lukem static char lasthost[MAXHOSTNAMELEN];
707 1.27 lukem char portnum[6]; /* large enough for "65535\0" */
708 1.1 lukem char *xargv[5];
709 1.27 lukem const char *line;
710 1.27 lukem char *cp, *host, *path, *dir, *file;
711 1.6 lukem char *user, *pass;
712 1.27 lukem in_port_t port;
713 1.5 lukem char *ftpproxy, *httpproxy;
714 1.12 lukem int rval, xargc;
715 1.12 lukem volatile int argpos;
716 1.3 lukem int dirhasglob, filehasglob;
717 1.3 lukem char rempath[MAXPATHLEN];
718 1.1 lukem
719 1.22 lukem #ifdef __GNUC__ /* to shut up gcc warnings */
720 1.22 lukem (void)&outfile;
721 1.22 lukem #endif
722 1.22 lukem
723 1.2 lukem argpos = 0;
724 1.1 lukem
725 1.2 lukem if (setjmp(toplevel)) {
726 1.2 lukem if (connected)
727 1.2 lukem disconnect(0, NULL);
728 1.3 lukem return (argpos + 1);
729 1.2 lukem }
730 1.3 lukem (void)signal(SIGINT, (sig_t)intr);
731 1.3 lukem (void)signal(SIGPIPE, (sig_t)lostpeer);
732 1.1 lukem
733 1.5 lukem ftpproxy = getenv(FTP_PROXY);
734 1.5 lukem httpproxy = getenv(HTTP_PROXY);
735 1.5 lukem
736 1.1 lukem /*
737 1.1 lukem * Loop through as long as there's files to fetch.
738 1.1 lukem */
739 1.27 lukem for (rval = 0; (rval == 0) && (argpos < argc); argpos++) {
740 1.1 lukem if (strchr(argv[argpos], ':') == NULL)
741 1.1 lukem break;
742 1.27 lukem host = path = dir = file = user = pass = NULL;
743 1.32 lukem port = 0;
744 1.27 lukem line = argv[argpos];
745 1.1 lukem
746 1.25 lukem #ifndef SMALL
747 1.25 lukem /*
748 1.25 lukem * Check for about:*
749 1.25 lukem */
750 1.25 lukem if (strncasecmp(line, ABOUT_URL, sizeof(ABOUT_URL) - 1) == 0) {
751 1.27 lukem line += sizeof(ABOUT_URL) -1;
752 1.27 lukem if (strcasecmp(line, "ftp") == 0) {
753 1.25 lukem fprintf(ttyout, "%s\n%s\n",
754 1.26 lukem "This version of ftp has been enhanced by Luke Mewburn <lukem (at) netbsd.org>.",
755 1.25 lukem "Execute 'man ftp' for more details");
756 1.27 lukem } else if (strcasecmp(line, "netbsd") == 0) {
757 1.25 lukem fprintf(ttyout, "%s\n%s\n",
758 1.25 lukem "NetBSD is a freely available and redistributable UNIX-like operating system.",
759 1.25 lukem "For more information, see http://www.netbsd.org/index.html");
760 1.25 lukem } else {
761 1.25 lukem fprintf(ttyout,
762 1.27 lukem "`%s' is an interesting topic.\n", line);
763 1.25 lukem }
764 1.25 lukem continue;
765 1.25 lukem }
766 1.25 lukem #endif /* SMALL */
767 1.25 lukem
768 1.1 lukem /*
769 1.25 lukem * Check for file:// and http:// URLs.
770 1.1 lukem */
771 1.25 lukem if (strncasecmp(line, HTTP_URL, sizeof(HTTP_URL) - 1) == 0 ||
772 1.25 lukem strncasecmp(line, FILE_URL, sizeof(FILE_URL) - 1) == 0) {
773 1.22 lukem if (url_get(line, httpproxy, outfile) == -1)
774 1.1 lukem rval = argpos + 1;
775 1.1 lukem continue;
776 1.1 lukem }
777 1.1 lukem
778 1.1 lukem /*
779 1.5 lukem * Try FTP URL-style arguments next. If ftpproxy is
780 1.5 lukem * set, use url_get() instead of standard ftp.
781 1.5 lukem * Finally, try host:file.
782 1.1 lukem */
783 1.6 lukem if (strncasecmp(line, FTP_URL, sizeof(FTP_URL) - 1) == 0) {
784 1.34 lukem url_t urltype;
785 1.27 lukem
786 1.5 lukem if (ftpproxy) {
787 1.22 lukem if (url_get(line, ftpproxy, outfile) == -1)
788 1.5 lukem rval = argpos + 1;
789 1.5 lukem continue;
790 1.5 lukem }
791 1.27 lukem if ((parse_url(line, "URL", &urltype, &user, &pass,
792 1.27 lukem &host, &port, &path) == -1) ||
793 1.27 lukem (user != NULL && *user == '\0') ||
794 1.27 lukem (pass != NULL && *pass == '\0') ||
795 1.27 lukem EMPTYSTRING(host)) {
796 1.27 lukem warnx("Invalid URL `%s'", argv[argpos]);
797 1.6 lukem rval = argpos + 1;
798 1.27 lukem break;
799 1.6 lukem }
800 1.6 lukem } else { /* classic style `host:file' */
801 1.27 lukem host = xstrdup(line);
802 1.27 lukem cp = strchr(host, ':');
803 1.27 lukem if (cp != NULL) {
804 1.27 lukem *cp = '\0';
805 1.27 lukem path = xstrdup(cp + 1);
806 1.27 lukem }
807 1.6 lukem }
808 1.1 lukem if (EMPTYSTRING(host)) {
809 1.1 lukem rval = argpos + 1;
810 1.27 lukem break;
811 1.1 lukem }
812 1.1 lukem
813 1.1 lukem /*
814 1.33 lukem * Extract the file and (if present) directory name.
815 1.1 lukem */
816 1.27 lukem dir = path;
817 1.1 lukem if (! EMPTYSTRING(dir)) {
818 1.33 lukem if (*dir == '/')
819 1.33 lukem dir++; /* skip leading / */
820 1.6 lukem cp = strrchr(dir, '/');
821 1.1 lukem if (cp != NULL) {
822 1.1 lukem *cp++ = '\0';
823 1.1 lukem file = cp;
824 1.1 lukem } else {
825 1.1 lukem file = dir;
826 1.1 lukem dir = NULL;
827 1.1 lukem }
828 1.1 lukem }
829 1.1 lukem if (debug)
830 1.22 lukem fprintf(ttyout,
831 1.27 lukem "auto_fetch: user `%s', pass `%s', host %s:%d, path, `%s', dir `%s', file `%s'\n",
832 1.27 lukem user ? user : "", pass ? pass : "",
833 1.27 lukem host ? host : "", ntohs(port), path ? path : "",
834 1.27 lukem dir ? dir : "", file ? file : "");
835 1.1 lukem
836 1.1 lukem /*
837 1.1 lukem * Set up the connection if we don't have one.
838 1.1 lukem */
839 1.27 lukem if (strcasecmp(host, lasthost) != 0) {
840 1.6 lukem int oautologin;
841 1.6 lukem
842 1.4 lukem (void)strcpy(lasthost, host);
843 1.1 lukem if (connected)
844 1.1 lukem disconnect(0, NULL);
845 1.1 lukem xargv[0] = __progname;
846 1.1 lukem xargv[1] = host;
847 1.1 lukem xargv[2] = NULL;
848 1.1 lukem xargc = 2;
849 1.27 lukem if (port) {
850 1.27 lukem snprintf(portnum, sizeof(portnum),
851 1.35 lukem "%d", ntohs(port));
852 1.1 lukem xargv[2] = portnum;
853 1.1 lukem xargv[3] = NULL;
854 1.1 lukem xargc = 3;
855 1.1 lukem }
856 1.6 lukem oautologin = autologin;
857 1.6 lukem if (user != NULL)
858 1.6 lukem autologin = 0;
859 1.1 lukem setpeer(xargc, xargv);
860 1.6 lukem autologin = oautologin;
861 1.6 lukem if ((connected == 0)
862 1.24 lukem || ((connected == 1) &&
863 1.24 lukem !ftp_login(host, user, pass)) ) {
864 1.6 lukem warnx("Can't connect or login to host `%s'",
865 1.6 lukem host);
866 1.1 lukem rval = argpos + 1;
867 1.27 lukem break;
868 1.1 lukem }
869 1.1 lukem
870 1.1 lukem /* Always use binary transfers. */
871 1.1 lukem setbinary(0, NULL);
872 1.1 lukem }
873 1.27 lukem /* cd back to `/' */
874 1.6 lukem xargv[0] = "cd";
875 1.6 lukem xargv[1] = "/";
876 1.6 lukem xargv[2] = NULL;
877 1.6 lukem cd(2, xargv);
878 1.6 lukem if (! dirchange) {
879 1.6 lukem rval = argpos + 1;
880 1.27 lukem break;
881 1.1 lukem }
882 1.1 lukem
883 1.3 lukem dirhasglob = filehasglob = 0;
884 1.3 lukem if (doglob) {
885 1.3 lukem if (! EMPTYSTRING(dir) &&
886 1.3 lukem strpbrk(dir, "*?[]{}") != NULL)
887 1.3 lukem dirhasglob = 1;
888 1.3 lukem if (! EMPTYSTRING(file) &&
889 1.3 lukem strpbrk(file, "*?[]{}") != NULL)
890 1.3 lukem filehasglob = 1;
891 1.3 lukem }
892 1.3 lukem
893 1.27 lukem /* Change directories, if necessary. */
894 1.3 lukem if (! EMPTYSTRING(dir) && !dirhasglob) {
895 1.1 lukem xargv[0] = "cd";
896 1.1 lukem xargv[1] = dir;
897 1.1 lukem xargv[2] = NULL;
898 1.1 lukem cd(2, xargv);
899 1.1 lukem if (! dirchange) {
900 1.1 lukem rval = argpos + 1;
901 1.27 lukem break;
902 1.1 lukem }
903 1.1 lukem }
904 1.1 lukem
905 1.1 lukem if (EMPTYSTRING(file)) {
906 1.1 lukem rval = -1;
907 1.27 lukem break;
908 1.1 lukem }
909 1.1 lukem
910 1.2 lukem if (!verbose)
911 1.22 lukem fprintf(ttyout, "Retrieving %s/%s\n", dir ? dir : "",
912 1.22 lukem file);
913 1.2 lukem
914 1.3 lukem if (dirhasglob) {
915 1.3 lukem snprintf(rempath, sizeof(rempath), "%s/%s", dir, file);
916 1.3 lukem file = rempath;
917 1.3 lukem }
918 1.3 lukem
919 1.27 lukem /* Fetch the file(s). */
920 1.22 lukem xargc = 2;
921 1.1 lukem xargv[0] = "get";
922 1.1 lukem xargv[1] = file;
923 1.1 lukem xargv[2] = NULL;
924 1.3 lukem if (dirhasglob || filehasglob) {
925 1.3 lukem int ointeractive;
926 1.3 lukem
927 1.3 lukem ointeractive = interactive;
928 1.3 lukem interactive = 0;
929 1.3 lukem xargv[0] = "mget";
930 1.22 lukem mget(xargc, xargv);
931 1.4 lukem interactive = ointeractive;
932 1.22 lukem } else {
933 1.22 lukem if (outfile != NULL) {
934 1.22 lukem xargv[2] = outfile;
935 1.22 lukem xargv[3] = NULL;
936 1.22 lukem xargc++;
937 1.22 lukem }
938 1.22 lukem get(xargc, xargv);
939 1.22 lukem if (outfile != NULL && strcmp(outfile, "-") != 0
940 1.22 lukem && outfile[0] != '|')
941 1.22 lukem outfile = NULL;
942 1.22 lukem }
943 1.2 lukem
944 1.3 lukem if ((code / 100) != COMPLETE)
945 1.2 lukem rval = argpos + 1;
946 1.1 lukem }
947 1.1 lukem if (connected && rval != -1)
948 1.1 lukem disconnect(0, NULL);
949 1.27 lukem FREEPTR(host);
950 1.27 lukem FREEPTR(path);
951 1.27 lukem FREEPTR(user);
952 1.27 lukem FREEPTR(pass);
953 1.1 lukem return (rval);
954 1.1 lukem }
955