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