Annotation of src/usr.bin/ftp/fetch.c, Revision 1.214
1.214 ! guenther 1: /* $OpenBSD: fetch.c,v 1.213 2022/12/26 19:16:01 jmc Exp $ */
1.15 millert 2: /* $NetBSD: fetch.c,v 1.14 1997/08/18 10:20:20 lukem Exp $ */
1.1 millert 3:
4: /*-
5: * Copyright (c) 1997 The NetBSD Foundation, Inc.
6: * All rights reserved.
7: *
8: * This code is derived from software contributed to The NetBSD Foundation
9: * by Jason Thorpe and Luke Mewburn.
10: *
11: * Redistribution and use in source and binary forms, with or without
12: * modification, are permitted provided that the following conditions
13: * are met:
14: * 1. Redistributions of source code must retain the above copyright
15: * notice, this list of conditions and the following disclaimer.
16: * 2. Redistributions in binary form must reproduce the above copyright
17: * notice, this list of conditions and the following disclaimer in the
18: * documentation and/or other materials provided with the distribution.
19: *
20: * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
21: * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
22: * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
1.15 millert 23: * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
24: * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
1.1 millert 25: * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
26: * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
27: * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
28: * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
29: * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
30: * POSSIBILITY OF SUCH DAMAGE.
31: */
32:
33: /*
34: * FTP User Program -- Command line file retrieval
35: */
36:
37: #include <sys/types.h>
38: #include <sys/socket.h>
1.22 deraadt 39: #include <sys/stat.h>
1.1 millert 40:
41: #include <netinet/in.h>
42:
43: #include <arpa/ftp.h>
44: #include <arpa/inet.h>
45:
46: #include <ctype.h>
47: #include <err.h>
1.17 millert 48: #include <libgen.h>
1.1 millert 49: #include <netdb.h>
50: #include <fcntl.h>
1.3 millert 51: #include <signal.h>
1.175 deraadt 52: #include <vis.h>
1.1 millert 53: #include <stdio.h>
1.61 deraadt 54: #include <stdarg.h>
1.19 deraadt 55: #include <errno.h>
1.1 millert 56: #include <stdlib.h>
57: #include <string.h>
58: #include <unistd.h>
1.73 drahn 59: #include <resolv.h>
1.200 robert 60: #include <utime.h>
1.1 millert 61:
1.154 deraadt 62: #ifndef NOSSL
1.134 jsing 63: #include <tls.h>
1.154 deraadt 64: #else /* !NOSSL */
1.134 jsing 65: struct tls;
1.154 deraadt 66: #endif /* !NOSSL */
1.61 deraadt 67:
1.1 millert 68: #include "ftp_var.h"
1.86 martynas 69: #include "cmds.h"
1.1 millert 70:
1.186 jca 71: static int file_get(const char *, const char *);
1.157 deraadt 72: static int url_get(const char *, const char *, const char *, int);
1.178 jca 73: static int save_chunked(FILE *, struct tls *, int , char *, size_t);
1.177 jca 74: static void aborthttp(int);
75: static char hextochar(const char *);
76: static char *urldecode(const char *);
77: static char *recode_credentials(const char *_userinfo);
1.189 jca 78: static void ftp_close(FILE **, struct tls **, int *);
1.177 jca 79: static const char *sockerror(struct tls *);
1.184 jca 80: #ifdef SMALL
81: #define ftp_printf(fp, ...) fprintf(fp, __VA_ARGS__)
82: #else
83: static int ftp_printf(FILE *, const char *, ...);
84: #endif /* SMALL */
1.158 jca 85: #ifndef NOSSL
1.177 jca 86: static int proxy_connect(int, char *, char *);
87: static int stdio_tls_write_wrapper(void *, const char *, int);
88: static int stdio_tls_read_wrapper(void *, char *, int);
1.158 jca 89: #endif /* !NOSSL */
1.14 millert 90:
1.1 millert 91: #define FTP_URL "ftp://" /* ftp URL prefix */
92: #define HTTP_URL "http://" /* http URL prefix */
1.61 deraadt 93: #define HTTPS_URL "https://" /* https URL prefix */
1.22 deraadt 94: #define FILE_URL "file:" /* file URL prefix */
1.6 millert 95: #define FTP_PROXY "ftp_proxy" /* env var with ftp proxy location */
1.1 millert 96: #define HTTP_PROXY "http_proxy" /* env var with http proxy location */
97:
98: #define EMPTYSTRING(x) ((x) == NULL || (*(x) == '\0'))
99:
1.110 guenther 100: static const char at_encoding_warning[] =
1.53 deraadt 101: "Extra `@' characters in usernames and passwords should be encoded as %%40";
1.37 heko 102:
1.177 jca 103: static jmp_buf httpabort;
1.1 millert 104:
1.54 fgsch 105: static int redirect_loop;
1.171 jca 106: static int retried;
1.54 fgsch 107:
1.1 millert 108: /*
1.206 kn 109: * Determine whether the character needs encoding, per RFC2396.
1.95 martynas 110: */
111: static int
1.206 kn 112: to_encode(const char *c0)
1.95 martynas 113: {
1.206 kn 114: /* 2.4.3. Excluded US-ASCII Characters */
115: const char *excluded_chars =
116: " " /* space */
117: "<>#\"" /* delims (modulo "%", see below) */
118: "{}|\\^[]`" /* unwise */
119: ;
1.138 jca 120: const unsigned char *c = (const unsigned char *)c0;
1.95 martynas 121:
122: /*
123: * No corresponding graphic US-ASCII.
124: * Control characters and octets not used in US-ASCII.
125: */
126: return (iscntrl(*c) || !isascii(*c) ||
127:
128: /*
1.206 kn 129: * '%' is also reserved, if is not followed by two
1.95 martynas 130: * hexadecimal digits.
131: */
1.206 kn 132: strchr(excluded_chars, *c) != NULL ||
1.204 deraadt 133: (*c == '%' && (!isxdigit(c[1]) || !isxdigit(c[2]))));
1.95 martynas 134: }
135:
136: /*
1.206 kn 137: * Encode given URL, per RFC2396.
1.95 martynas 138: * Allocate and return string to the caller.
139: */
140: static char *
141: url_encode(const char *path)
142: {
143: size_t i, length, new_length;
144: char *epath, *epathp;
145:
146: length = new_length = strlen(path);
147:
148: /*
149: * First pass:
1.206 kn 150: * Count characters to encode and determine length of the final URL.
1.95 martynas 151: */
152: for (i = 0; i < length; i++)
1.206 kn 153: if (to_encode(path + i))
1.95 martynas 154: new_length += 2;
155:
156: epath = epathp = malloc(new_length + 1); /* One more for '\0'. */
157: if (epath == NULL)
1.102 halex 158: err(1, "Can't allocate memory for URL encoding");
1.95 martynas 159:
160: /*
161: * Second pass:
162: * Encode, and copy final URL.
163: */
164: for (i = 0; i < length; i++)
1.206 kn 165: if (to_encode(path + i)) {
1.138 jca 166: snprintf(epathp, 4, "%%" "%02x",
167: (unsigned char)path[i]);
1.95 martynas 168: epathp += 3;
169: } else
170: *(epathp++) = path[i];
171:
172: *epathp = '\0';
173: return (epath);
174: }
175:
176: /*
1.186 jca 177: * Copy a local file (used by the OpenBSD installer).
178: * Returns -1 on failure, 0 on success
179: */
180: static int
181: file_get(const char *path, const char *outfile)
182: {
183: struct stat st;
1.195 jca 184: int fd, out = -1, rval = -1, save_errno;
1.186 jca 185: volatile sig_t oldintr, oldinti;
186: const char *savefile;
1.198 naddy 187: char *buf = NULL, *cp, *pathbuf = NULL;
1.186 jca 188: const size_t buflen = 128 * 1024;
189: off_t hashbytes;
190: ssize_t len, wlen;
191:
192: direction = "received";
193:
194: fd = open(path, O_RDONLY);
195: if (fd == -1) {
196: warn("Can't open file %s", path);
197: return -1;
198: }
199:
200: if (fstat(fd, &st) == -1)
201: filesize = -1;
202: else
203: filesize = st.st_size;
204:
205: if (outfile != NULL)
206: savefile = outfile;
207: else {
208: if (path[strlen(path) - 1] == '/') /* Consider no file */
209: savefile = NULL; /* after dir invalid. */
1.198 naddy 210: else {
211: pathbuf = strdup(path);
212: if (pathbuf == NULL)
213: errx(1, "Can't allocate memory for filename");
214: savefile = basename(pathbuf);
215: }
1.186 jca 216: }
217:
218: if (EMPTYSTRING(savefile)) {
219: warnx("No filename after directory (use -o): %s", path);
220: goto cleanup_copy;
221: }
222:
223: /* Open the output file. */
224: if (!pipeout) {
225: out = open(savefile, O_CREAT | O_WRONLY | O_TRUNC, 0666);
226: if (out == -1) {
227: warn("Can't open %s", savefile);
228: goto cleanup_copy;
229: }
230: } else
231: out = fileno(stdout);
232:
233: if ((buf = malloc(buflen)) == NULL)
234: errx(1, "Can't allocate memory for transfer buffer");
235:
236: /* Trap signals */
237: oldintr = NULL;
238: oldinti = NULL;
239: if (setjmp(httpabort)) {
240: if (oldintr)
241: (void)signal(SIGINT, oldintr);
242: if (oldinti)
243: (void)signal(SIGINFO, oldinti);
244: goto cleanup_copy;
245: }
1.197 kn 246: oldintr = signal(SIGINT, aborthttp);
1.186 jca 247:
248: bytes = 0;
249: hashbytes = mark;
250: progressmeter(-1, path);
251:
252: /* Finally, suck down the file. */
253: oldinti = signal(SIGINFO, psummary);
254: while ((len = read(fd, buf, buflen)) > 0) {
255: bytes += len;
256: for (cp = buf; len > 0; len -= wlen, cp += wlen) {
257: if ((wlen = write(out, cp, len)) == -1) {
258: warn("Writing %s", savefile);
1.196 jca 259: signal(SIGINT, oldintr);
1.186 jca 260: signal(SIGINFO, oldinti);
261: goto cleanup_copy;
262: }
263: }
264: if (hash && !progress) {
265: while (bytes >= hashbytes) {
266: (void)putc('#', ttyout);
267: hashbytes += mark;
268: }
269: (void)fflush(ttyout);
270: }
271: }
272: save_errno = errno;
1.196 jca 273: signal(SIGINT, oldintr);
1.186 jca 274: signal(SIGINFO, oldinti);
275: if (hash && !progress && bytes > 0) {
276: if (bytes < mark)
277: (void)putc('#', ttyout);
278: (void)putc('\n', ttyout);
279: (void)fflush(ttyout);
280: }
281: if (len == -1) {
282: warnc(save_errno, "Reading from file");
283: goto cleanup_copy;
284: }
285: progressmeter(1, NULL);
286: if (verbose)
287: ptransfer(0);
288:
289: rval = 0;
290:
291: cleanup_copy:
292: free(buf);
1.198 naddy 293: free(pathbuf);
1.186 jca 294: if (out >= 0 && out != fileno(stdout))
295: close(out);
296: close(fd);
297:
298: return rval;
299: }
300:
301: /*
1.6 millert 302: * Retrieve URL, via the proxy in $proxyvar if necessary.
1.1 millert 303: * Returns -1 on failure, 0 on success
304: */
1.14 millert 305: static int
1.157 deraadt 306: url_get(const char *origline, const char *proxyenv, const char *outfile, int lastfile)
1.1 millert 307: {
1.69 jsg 308: char pbuf[NI_MAXSERV], hbuf[NI_MAXHOST], *cp, *portnum, *path, ststr[4];
1.62 ray 309: char *hosttail, *cause = "unknown", *newline, *host, *port, *buf = NULL;
1.175 deraadt 310: char *epath, *redirurl, *loctail, *h, *p, gerror[200];
1.189 jca 311: int error, isftpurl = 0, isredirect = 0, rval = -1;
1.172 jca 312: int isunavail = 0, retryafter = -1;
1.152 krw 313: struct addrinfo hints, *res0, *res;
1.189 jca 314: const char *savefile;
1.198 naddy 315: char *pathbuf = NULL;
1.189 jca 316: char *proxyurl = NULL;
1.190 yasuoka 317: char *credentials = NULL, *proxy_credentials = NULL;
1.189 jca 318: int fd = -1, out = -1;
1.97 martynas 319: volatile sig_t oldintr, oldinti;
1.53 deraadt 320: FILE *fin = NULL;
1.1 millert 321: off_t hashbytes;
1.58 grunk 322: const char *errstr;
1.114 tedu 323: ssize_t len, wlen;
1.201 naddy 324: size_t bufsize;
1.136 bluhm 325: char *proxyhost = NULL;
1.154 deraadt 326: #ifndef NOSSL
1.61 deraadt 327: char *sslpath = NULL, *sslhost = NULL;
1.193 jca 328: int ishttpsurl = 0;
1.159 krw 329: #endif /* !NOSSL */
330: #ifndef SMALL
1.181 jca 331: char *full_host = NULL;
332: const char *scheme;
1.159 krw 333: char *locbase;
1.152 krw 334: struct addrinfo *ares = NULL;
1.200 robert 335: char tmbuf[32];
336: time_t mtime = 0;
337: struct stat stbuf;
338: struct tm lmt = { 0 };
339: struct timespec ts[2];
1.181 jca 340: #endif /* !SMALL */
1.134 jsing 341: struct tls *tls = NULL;
1.70 deraadt 342: int status;
1.105 haesbaer 343: int save_errno;
1.113 tedu 344: const size_t buflen = 128 * 1024;
1.178 jca 345: int chunked = 0;
1.14 millert 346:
1.97 martynas 347: direction = "received";
348:
1.62 ray 349: newline = strdup(origline);
350: if (newline == NULL)
1.14 millert 351: errx(1, "Can't allocate memory to parse URL");
1.102 halex 352: if (strncasecmp(newline, HTTP_URL, sizeof(HTTP_URL) - 1) == 0) {
1.62 ray 353: host = newline + sizeof(HTTP_URL) - 1;
1.102 halex 354: #ifndef SMALL
355: scheme = HTTP_URL;
356: #endif /* !SMALL */
357: } else if (strncasecmp(newline, FTP_URL, sizeof(FTP_URL) - 1) == 0) {
1.62 ray 358: host = newline + sizeof(FTP_URL) - 1;
1.14 millert 359: isftpurl = 1;
1.102 halex 360: #ifndef SMALL
361: scheme = FTP_URL;
362: #endif /* !SMALL */
1.183 jca 363: } else if (strncasecmp(newline, HTTPS_URL, sizeof(HTTPS_URL) - 1) == 0) {
1.154 deraadt 364: #ifndef NOSSL
1.62 ray 365: host = newline + sizeof(HTTPS_URL) - 1;
1.61 deraadt 366: ishttpsurl = 1;
1.183 jca 367: #else
368: errx(1, "%s: No HTTPS support", newline);
369: #endif /* !NOSSL */
1.181 jca 370: #ifndef SMALL
1.102 halex 371: scheme = HTTPS_URL;
1.181 jca 372: #endif /* !SMALL */
1.14 millert 373: } else
1.186 jca 374: errx(1, "%s: URL not permitted", newline);
1.6 millert 375:
1.186 jca 376: path = strchr(host, '/'); /* Find path */
1.191 jca 377:
378: /*
379: * Look for auth header in host.
380: * Basic auth from RFC 2617, valid characters for path are in
381: * RFC 3986 section 3.3.
382: */
1.193 jca 383: if (!isftpurl) {
1.191 jca 384: p = strchr(host, '@');
385: if (p != NULL && (path == NULL || p < path)) {
386: *p++ = '\0';
387: credentials = recode_credentials(host);
388:
389: /* Overwrite userinfo */
390: memmove(host, p, strlen(p) + 1);
391: path = strchr(host, '/');
392: }
393: }
394:
1.186 jca 395: if (EMPTYSTRING(path)) {
396: if (outfile) { /* No slash, but */
397: path = strchr(host,'\0'); /* we have outfile. */
398: goto noslash;
1.22 deraadt 399: }
1.186 jca 400: if (isftpurl)
401: goto noftpautologin;
402: warnx("No `/' after host (use -o): %s", origline);
403: goto cleanup_url_get;
404: }
405: *path++ = '\0';
406: if (EMPTYSTRING(path) && !outfile) {
407: if (isftpurl)
408: goto noftpautologin;
409: warnx("No filename after host (use -o): %s", origline);
410: goto cleanup_url_get;
1.14 millert 411: }
1.1 millert 412:
1.93 martynas 413: noslash:
1.17 millert 414: if (outfile)
415: savefile = outfile;
1.93 martynas 416: else {
417: if (path[strlen(path) - 1] == '/') /* Consider no file */
418: savefile = NULL; /* after dir invalid. */
1.198 naddy 419: else {
420: pathbuf = strdup(path);
421: if (pathbuf == NULL)
422: errx(1, "Can't allocate memory for filename");
423: savefile = basename(pathbuf);
424: }
1.75 martynas 425: }
426:
1.14 millert 427: if (EMPTYSTRING(savefile)) {
428: if (isftpurl)
429: goto noftpautologin;
1.94 martynas 430: warnx("No filename after directory (use -o): %s", origline);
1.6 millert 431: goto cleanup_url_get;
1.14 millert 432: }
1.1 millert 433:
1.93 martynas 434: #ifndef SMALL
1.100 halex 435: if (resume && pipeout) {
1.93 martynas 436: warnx("can't append to stdout");
437: goto cleanup_url_get;
438: }
439: #endif /* !SMALL */
440:
1.186 jca 441: if (proxyenv != NULL) { /* use proxy */
1.154 deraadt 442: #ifndef NOSSL
1.61 deraadt 443: if (ishttpsurl) {
444: sslpath = strdup(path);
445: sslhost = strdup(host);
446: if (! sslpath || ! sslhost)
447: errx(1, "Can't allocate memory for https path/host.");
448: }
1.154 deraadt 449: #endif /* !NOSSL */
1.136 bluhm 450: proxyhost = strdup(host);
451: if (proxyhost == NULL)
452: errx(1, "Can't allocate memory for proxy host.");
1.62 ray 453: proxyurl = strdup(proxyenv);
454: if (proxyurl == NULL)
1.14 millert 455: errx(1, "Can't allocate memory for proxy URL.");
1.62 ray 456: if (strncasecmp(proxyurl, HTTP_URL, sizeof(HTTP_URL) - 1) == 0)
457: host = proxyurl + sizeof(HTTP_URL) - 1;
458: else if (strncasecmp(proxyurl, FTP_URL, sizeof(FTP_URL) - 1) == 0)
459: host = proxyurl + sizeof(FTP_URL) - 1;
1.6 millert 460: else {
1.14 millert 461: warnx("Malformed proxy URL: %s", proxyenv);
1.6 millert 462: goto cleanup_url_get;
463: }
1.14 millert 464: if (EMPTYSTRING(host)) {
465: warnx("Malformed proxy URL: %s", proxyenv);
1.6 millert 466: goto cleanup_url_get;
1.14 millert 467: }
1.93 martynas 468: if (*--path == '\0')
469: *path = '/'; /* add / back to real path */
1.1 millert 470: path = strchr(host, '/'); /* remove trailing / on host */
1.42 deraadt 471: if (!EMPTYSTRING(path))
1.73 drahn 472: *path++ = '\0'; /* i guess this ++ is useless */
473:
474: path = strchr(host, '@'); /* look for credentials in proxy */
475: if (!EMPTYSTRING(path)) {
1.81 espie 476: *path = '\0';
1.123 guenther 477: if (strchr(host, ':') == NULL) {
1.73 drahn 478: warnx("Malformed proxy URL: %s", proxyenv);
479: goto cleanup_url_get;
480: }
1.190 yasuoka 481: proxy_credentials = recode_credentials(host);
1.81 espie 482: *path = '@'; /* restore @ in proxyurl */
1.123 guenther 483:
1.73 drahn 484: /*
1.81 espie 485: * This removes the password from proxyurl,
1.73 drahn 486: * filling with stars
487: */
1.81 espie 488: for (host = 1 + strchr(proxyurl + 5, ':'); *host != '@';
1.179 deraadt 489: host++)
1.73 drahn 490: *host = '*';
491:
1.81 espie 492: host = path + 1;
1.73 drahn 493: }
1.123 guenther 494:
1.62 ray 495: path = newline;
1.1 millert 496: }
497:
1.28 itojun 498: if (*host == '[' && (hosttail = strrchr(host, ']')) != NULL &&
499: (hosttail[1] == '\0' || hosttail[1] == ':')) {
500: host++;
501: *hosttail++ = '\0';
1.102 halex 502: #ifndef SMALL
503: if (asprintf(&full_host, "[%s]", host) == -1)
504: errx(1, "Cannot allocate memory for hostname");
505: #endif /* !SMALL */
1.28 itojun 506: } else
507: hosttail = host;
508:
509: portnum = strrchr(hosttail, ':'); /* find portnum */
1.1 millert 510: if (portnum != NULL)
511: *portnum++ = '\0';
1.154 deraadt 512: #ifndef NOSSL
1.150 millert 513: port = portnum ? portnum : (ishttpsurl ? httpsport : httpport);
1.154 deraadt 514: #else /* !NOSSL */
1.150 millert 515: port = portnum ? portnum : httpport;
1.154 deraadt 516: #endif /* !NOSSL */
1.1 millert 517:
1.80 martynas 518: #ifndef SMALL
1.102 halex 519: if (full_host == NULL)
520: if ((full_host = strdup(host)) == NULL)
521: errx(1, "Cannot allocate memory for hostname");
1.1 millert 522: if (debug)
1.106 haesbaer 523: fprintf(ttyout, "host %s, port %s, path %s, "
1.150 millert 524: "save as %s, auth %s.\n", host, port, path,
525: savefile, credentials ? credentials : "none");
1.80 martynas 526: #endif /* !SMALL */
1.1 millert 527:
1.25 itojun 528: memset(&hints, 0, sizeof(hints));
1.39 deraadt 529: hints.ai_family = family;
1.25 itojun 530: hints.ai_socktype = SOCK_STREAM;
531: error = getaddrinfo(host, port, &hints, &res0);
1.61 deraadt 532: /*
533: * If the services file is corrupt/missing, fall back
534: * on our hard-coded defines.
535: */
1.30 deraadt 536: if (error == EAI_SERVICE && port == httpport) {
537: snprintf(pbuf, sizeof(pbuf), "%d", HTTP_PORT);
538: error = getaddrinfo(host, pbuf, &hints, &res0);
1.154 deraadt 539: #ifndef NOSSL
1.61 deraadt 540: } else if (error == EAI_SERVICE && port == httpsport) {
541: snprintf(pbuf, sizeof(pbuf), "%d", HTTPS_PORT);
542: error = getaddrinfo(host, pbuf, &hints, &res0);
1.154 deraadt 543: #endif /* !NOSSL */
1.30 deraadt 544: }
1.25 itojun 545: if (error) {
1.122 guenther 546: warnx("%s: %s", host, gai_strerror(error));
1.25 itojun 547: goto cleanup_url_get;
1.1 millert 548: }
549:
1.105 haesbaer 550: #ifndef SMALL
551: if (srcaddr) {
552: hints.ai_flags |= AI_NUMERICHOST;
553: error = getaddrinfo(srcaddr, NULL, &hints, &ares);
554: if (error) {
1.122 guenther 555: warnx("%s: %s", srcaddr, gai_strerror(error));
1.105 haesbaer 556: goto cleanup_url_get;
557: }
558: }
559: #endif /* !SMALL */
1.135 deraadt 560:
561: /* ensure consistent order of the output */
562: if (verbose)
563: setvbuf(ttyout, NULL, _IOLBF, 0);
1.105 haesbaer 564:
1.166 procter 565: fd = -1;
1.25 itojun 566: for (res = res0; res; res = res->ai_next) {
1.44 itojun 567: if (getnameinfo(res->ai_addr, res->ai_addrlen, hbuf,
568: sizeof(hbuf), NULL, 0, NI_NUMERICHOST) != 0)
569: strlcpy(hbuf, "(unknown)", sizeof(hbuf));
1.41 deraadt 570: if (verbose)
1.44 itojun 571: fprintf(ttyout, "Trying %s...\n", hbuf);
1.14 millert 572:
1.166 procter 573: fd = socket(res->ai_family, res->ai_socktype, res->ai_protocol);
574: if (fd == -1) {
1.25 itojun 575: cause = "socket";
576: continue;
1.1 millert 577: }
578:
1.105 haesbaer 579: #ifndef SMALL
580: if (srcaddr) {
581: if (ares->ai_family != res->ai_family) {
1.166 procter 582: close(fd);
583: fd = -1;
1.105 haesbaer 584: errno = EINVAL;
585: cause = "bind";
586: continue;
587: }
1.170 deraadt 588: if (bind(fd, ares->ai_addr, ares->ai_addrlen) == -1) {
1.105 haesbaer 589: save_errno = errno;
1.166 procter 590: close(fd);
1.105 haesbaer 591: errno = save_errno;
1.166 procter 592: fd = -1;
1.105 haesbaer 593: cause = "bind";
594: continue;
595: }
596: }
597: #endif /* !SMALL */
598:
1.210 millert 599: error = timed_connect(fd, res->ai_addr, res->ai_addrlen,
600: connect_timeout);
1.149 millert 601: if (error != 0) {
1.57 otto 602: save_errno = errno;
1.166 procter 603: close(fd);
1.57 otto 604: errno = save_errno;
1.166 procter 605: fd = -1;
1.25 itojun 606: cause = "connect";
1.19 deraadt 607: continue;
608: }
1.25 itojun 609:
1.29 itojun 610: /* get port in numeric */
611: if (getnameinfo(res->ai_addr, res->ai_addrlen, NULL, 0,
612: pbuf, sizeof(pbuf), NI_NUMERICSERV) == 0)
613: port = pbuf;
614: else
615: port = NULL;
616:
1.158 jca 617: #ifndef NOSSL
1.61 deraadt 618: if (proxyenv && sslhost)
1.190 yasuoka 619: proxy_connect(fd, sslhost, proxy_credentials);
1.158 jca 620: #endif /* !NOSSL */
1.25 itojun 621: break;
622: }
623: freeaddrinfo(res0);
1.105 haesbaer 624: #ifndef SMALL
625: if (srcaddr)
626: freeaddrinfo(ares);
627: #endif /* !SMALL */
1.166 procter 628: if (fd < 0) {
1.33 millert 629: warn("%s", cause);
1.6 millert 630: goto cleanup_url_get;
1.1 millert 631: }
632:
1.154 deraadt 633: #ifndef NOSSL
1.61 deraadt 634: if (ishttpsurl) {
1.187 beck 635: ssize_t ret;
1.61 deraadt 636: if (proxyenv && sslpath) {
637: ishttpsurl = 0;
1.62 ray 638: proxyurl = NULL;
1.61 deraadt 639: path = sslpath;
640: }
1.131 jca 641: if (sslhost == NULL) {
642: sslhost = strdup(host);
643: if (sslhost == NULL)
644: errx(1, "Can't allocate memory for https host.");
1.116 jca 645: }
1.134 jsing 646: if ((tls = tls_client()) == NULL) {
1.126 jsing 647: fprintf(ttyout, "failed to create SSL client\n");
1.61 deraadt 648: goto cleanup_url_get;
649: }
1.134 jsing 650: if (tls_configure(tls, tls_config) != 0) {
1.187 beck 651: fprintf(ttyout, "TLS configuration failure: %s\n",
1.134 jsing 652: tls_error(tls));
1.61 deraadt 653: goto cleanup_url_get;
1.117 jca 654: }
1.166 procter 655: if (tls_connect_socket(tls, fd, sslhost) != 0) {
1.187 beck 656: fprintf(ttyout, "TLS connect failure: %s\n", tls_error(tls));
1.72 ray 657: goto cleanup_url_get;
1.118 jca 658: }
1.187 beck 659: do {
660: ret = tls_handshake(tls);
661: } while (ret == TLS_WANT_POLLIN || ret == TLS_WANT_POLLOUT);
662: if (ret != 0) {
663: fprintf(ttyout, "TLS handshake failure: %s\n", tls_error(tls));
1.176 jca 664: goto cleanup_url_get;
665: }
666: fin = funopen(tls, stdio_tls_read_wrapper,
667: stdio_tls_write_wrapper, NULL, NULL);
1.61 deraadt 668: } else {
1.166 procter 669: fin = fdopen(fd, "r+");
670: fd = -1;
1.61 deraadt 671: }
1.154 deraadt 672: #else /* !NOSSL */
1.166 procter 673: fin = fdopen(fd, "r+");
674: fd = -1;
1.154 deraadt 675: #endif /* !NOSSL */
1.155 deraadt 676:
1.157 deraadt 677: #ifdef SMALL
678: if (lastfile) {
679: if (pipeout) {
680: if (pledge("stdio rpath inet dns tty", NULL) == -1)
681: err(1, "pledge");
682: } else {
683: if (pledge("stdio rpath wpath cpath inet dns tty", NULL) == -1)
684: err(1, "pledge");
685: }
686: }
687: #endif
688:
1.1 millert 689: /*
1.40 fgsch 690: * Construct and send the request. Proxy requests don't want leading /.
1.1 millert 691: */
1.154 deraadt 692: #ifndef NOSSL
1.74 pyr 693: cookie_get(host, path, ishttpsurl, &buf);
1.154 deraadt 694: #endif /* !NOSSL */
1.95 martynas 695:
696: epath = url_encode(path);
1.62 ray 697: if (proxyurl) {
1.151 millert 698: if (verbose) {
699: fprintf(ttyout, "Requesting %s (via %s)\n",
1.208 kn 700: origline, proxyurl);
1.151 millert 701: }
1.32 itojun 702: /*
703: * Host: directive must use the destination host address for
1.136 bluhm 704: * the original URI (path).
1.32 itojun 705: */
1.190 yasuoka 706: ftp_printf(fin, "GET %s HTTP/1.1\r\n"
707: "Connection: close\r\n"
708: "Host: %s\r\n%s%s\r\n",
709: epath, proxyhost, buf ? buf : "", httpuseragent);
1.123 guenther 710: if (credentials)
1.190 yasuoka 711: ftp_printf(fin, "Authorization: Basic %s\r\n",
712: credentials);
713: if (proxy_credentials)
714: ftp_printf(fin, "Proxy-Authorization: Basic %s\r\n",
715: proxy_credentials);
716: ftp_printf(fin, "\r\n");
1.28 itojun 717: } else {
1.208 kn 718: if (verbose)
719: fprintf(ttyout, "Requesting %s\n", origline);
1.207 kn 720: #ifndef SMALL
1.200 robert 721: if (resume || timestamp) {
722: if (stat(savefile, &stbuf) == 0) {
723: if (resume)
724: restart_point = stbuf.st_size;
725: if (timestamp)
726: mtime = stbuf.st_mtime;
727: } else {
1.90 martynas 728: restart_point = 0;
1.200 robert 729: mtime = 0;
730: }
1.90 martynas 731: }
1.154 deraadt 732: #endif /* SMALL */
1.194 jca 733: ftp_printf(fin,
734: "GET /%s HTTP/1.1\r\n"
735: "Connection: close\r\n"
736: "Host: ", epath);
1.136 bluhm 737: if (proxyhost) {
1.184 jca 738: ftp_printf(fin, "%s", proxyhost);
1.136 bluhm 739: port = NULL;
740: } else if (strchr(host, ':')) {
1.55 fgsch 741: /*
742: * strip off scoped address portion, since it's
743: * local to node
744: */
1.32 itojun 745: h = strdup(host);
746: if (h == NULL)
747: errx(1, "Can't allocate memory.");
748: if ((p = strchr(h, '%')) != NULL)
749: *p = '\0';
1.184 jca 750: ftp_printf(fin, "[%s]", h);
1.32 itojun 751: free(h);
1.55 fgsch 752: } else
1.184 jca 753: ftp_printf(fin, "%s", host);
1.55 fgsch 754:
755: /*
756: * Send port number only if it's specified and does not equal
757: * 80. Some broken HTTP servers get confused if you explicitly
758: * send them the port number.
759: */
1.154 deraadt 760: #ifndef NOSSL
1.61 deraadt 761: if (port && strcmp(port, (ishttpsurl ? "443" : "80")) != 0)
1.184 jca 762: ftp_printf(fin, ":%s", port);
1.90 martynas 763: if (restart_point)
1.184 jca 764: ftp_printf(fin, "\r\nRange: bytes=%lld-",
1.75 martynas 765: (long long)restart_point);
1.154 deraadt 766: #else /* !NOSSL */
1.55 fgsch 767: if (port && strcmp(port, "80") != 0)
1.184 jca 768: ftp_printf(fin, ":%s", port);
1.154 deraadt 769: #endif /* !NOSSL */
1.200 robert 770:
771: #ifndef SMALL
772: if (mtime && (http_time(mtime, tmbuf, sizeof(tmbuf)) != 0))
773: ftp_printf(fin, "\r\nIf-Modified-Since: %s", tmbuf);
774: #endif /* SMALL */
775:
1.194 jca 776: ftp_printf(fin, "\r\n%s%s\r\n",
1.124 lteo 777: buf ? buf : "", httpuseragent);
1.194 jca 778: if (credentials)
779: ftp_printf(fin, "Authorization: Basic %s\r\n",
780: credentials);
781: ftp_printf(fin, "\r\n");
1.28 itojun 782: }
1.95 martynas 783: free(epath);
1.74 pyr 784:
1.154 deraadt 785: #ifndef NOSSL
1.74 pyr 786: free(buf);
1.156 tb 787: #endif /* !NOSSL */
1.74 pyr 788: buf = NULL;
1.201 naddy 789: bufsize = 0;
1.74 pyr 790:
1.176 jca 791: if (fflush(fin) == EOF) {
792: warnx("Writing HTTP request: %s", sockerror(tls));
1.6 millert 793: goto cleanup_url_get;
1.1 millert 794: }
1.201 naddy 795: if ((len = getline(&buf, &bufsize, fin)) == -1) {
1.176 jca 796: warnx("Receiving HTTP reply: %s", sockerror(tls));
1.40 fgsch 797: goto cleanup_url_get;
1.1 millert 798: }
1.40 fgsch 799:
800: while (len > 0 && (buf[len-1] == '\r' || buf[len-1] == '\n'))
801: buf[--len] = '\0';
1.80 martynas 802: #ifndef SMALL
1.40 fgsch 803: if (debug)
804: fprintf(ttyout, "received '%s'\n", buf);
1.80 martynas 805: #endif /* !SMALL */
1.40 fgsch 806:
1.1 millert 807: cp = strchr(buf, ' ');
808: if (cp == NULL)
809: goto improper;
810: else
811: cp++;
1.69 jsg 812:
813: strlcpy(ststr, cp, sizeof(ststr));
1.171 jca 814: status = strtonum(ststr, 200, 503, &errstr);
1.69 jsg 815: if (errstr) {
1.175 deraadt 816: strnvis(gerror, cp, sizeof gerror, VIS_SAFE);
817: warnx("Error retrieving %s: %s", origline, gerror);
1.69 jsg 818: goto cleanup_url_get;
819: }
820:
821: switch (status) {
822: case 200: /* OK */
1.75 martynas 823: #ifndef SMALL
1.109 sthen 824: /*
1.98 phessler 825: * When we request a partial file, and we receive an HTTP 200
826: * it is a good indication that the server doesn't support
827: * range requests, and is about to send us the entire file.
828: * If the restart_point == 0, then we are not actually
829: * requesting a partial file, and an HTTP 200 is appropriate.
830: */
831: if (resume && restart_point != 0) {
832: warnx("Server does not support resume.");
833: restart_point = resume = 0;
834: }
1.101 halex 835: /* FALLTHROUGH */
1.75 martynas 836: case 206: /* Partial Content */
1.91 halex 837: #endif /* !SMALL */
1.69 jsg 838: break;
839: case 301: /* Moved Permanently */
840: case 302: /* Found */
841: case 303: /* See Other */
842: case 307: /* Temporary Redirect */
1.199 chrisz 843: case 308: /* Permanent Redirect (RFC 7538) */
1.40 fgsch 844: isredirect++;
1.54 fgsch 845: if (redirect_loop++ > 10) {
846: warnx("Too many redirections requested");
847: goto cleanup_url_get;
848: }
1.69 jsg 849: break;
1.75 martynas 850: #ifndef SMALL
1.200 robert 851: case 304: /* Not Modified */
852: warnx("File is not modified on the server");
853: goto cleanup_url_get;
1.75 martynas 854: case 416: /* Requested Range Not Satisfiable */
855: warnx("File is already fully retrieved.");
856: goto cleanup_url_get;
1.78 martynas 857: #endif /* !SMALL */
1.171 jca 858: case 503:
1.172 jca 859: isunavail = 1;
860: break;
1.69 jsg 861: default:
1.175 deraadt 862: strnvis(gerror, cp, sizeof gerror, VIS_SAFE);
863: warnx("Error retrieving %s: %s", origline, gerror);
1.6 millert 864: goto cleanup_url_get;
1.1 millert 865: }
866:
867: /*
868: * Read the rest of the header.
869: */
1.40 fgsch 870: filesize = -1;
871:
1.62 ray 872: for (;;) {
1.201 naddy 873: if ((len = getline(&buf, &bufsize, fin)) == -1) {
1.176 jca 874: warnx("Receiving HTTP reply: %s", sockerror(tls));
1.40 fgsch 875: goto cleanup_url_get;
876: }
1.61 deraadt 877:
1.211 claudio 878: while (len > 0 && (buf[len-1] == '\r' || buf[len-1] == '\n' ||
879: buf[len-1] == ' ' || buf[len-1] == '\t'))
1.40 fgsch 880: buf[--len] = '\0';
881: if (len == 0)
1.1 millert 882: break;
1.80 martynas 883: #ifndef SMALL
1.40 fgsch 884: if (debug)
885: fprintf(ttyout, "received '%s'\n", buf);
1.80 martynas 886: #endif /* !SMALL */
1.1 millert 887:
1.40 fgsch 888: /* Look for some headers */
889: cp = buf;
1.209 claudio 890: #define CONTENTLEN "Content-Length:"
1.40 fgsch 891: if (strncasecmp(cp, CONTENTLEN, sizeof(CONTENTLEN) - 1) == 0) {
892: cp += sizeof(CONTENTLEN) - 1;
1.209 claudio 893: cp += strspn(cp, " \t");
894: cp[strcspn(cp, " \t")] = '\0';
1.58 grunk 895: filesize = strtonum(cp, 0, LLONG_MAX, &errstr);
896: if (errstr != NULL)
1.40 fgsch 897: goto improper;
1.75 martynas 898: #ifndef SMALL
1.90 martynas 899: if (restart_point)
1.75 martynas 900: filesize += restart_point;
1.78 martynas 901: #endif /* !SMALL */
1.209 claudio 902: #define LOCATION "Location:"
1.40 fgsch 903: } else if (isredirect &&
904: strncasecmp(cp, LOCATION, sizeof(LOCATION) - 1) == 0) {
905: cp += sizeof(LOCATION) - 1;
1.209 claudio 906: cp += strspn(cp, " \t");
1.144 sthen 907: /*
908: * If there is a colon before the first slash, this URI
909: * is not relative. RFC 3986 4.2
910: */
911: if (cp[strcspn(cp, ":/")] != ':') {
1.102 halex 912: #ifdef SMALL
913: errx(1, "Relative redirect not supported");
914: #else /* SMALL */
1.144 sthen 915: /* XXX doesn't handle protocol-relative URIs */
1.102 halex 916: if (*cp == '/') {
917: locbase = NULL;
918: cp++;
919: } else {
920: locbase = strdup(path);
921: if (locbase == NULL)
922: errx(1, "Can't allocate memory"
923: " for location base");
924: loctail = strchr(locbase, '#');
925: if (loctail != NULL)
926: *loctail = '\0';
927: loctail = strchr(locbase, '?');
928: if (loctail != NULL)
929: *loctail = '\0';
930: loctail = strrchr(locbase, '/');
931: if (loctail == NULL) {
932: free(locbase);
933: locbase = NULL;
934: } else
935: loctail[1] = '\0';
936: }
1.213 jmc 937: /* Construct URL from relative redirect */
1.102 halex 938: if (asprintf(&redirurl, "%s%s%s%s/%s%s",
939: scheme, full_host,
940: portnum ? ":" : "",
941: portnum ? portnum : "",
942: locbase ? locbase : "",
943: cp) == -1)
944: errx(1, "Cannot build "
945: "redirect URL");
946: free(locbase);
947: #endif /* SMALL */
948: } else if ((redirurl = strdup(cp)) == NULL)
949: errx(1, "Cannot allocate memory for URL");
950: loctail = strchr(redirurl, '#');
951: if (loctail != NULL)
952: *loctail = '\0';
1.212 claudio 953: if (verbose) {
954: char *visbuf;
955: if (stravis(&visbuf, redirurl, VIS_SAFE) == -1)
956: err(1, "Cannot vis redirect URL");
957: fprintf(ttyout, "Redirected to %s\n", visbuf);
958: free(visbuf);
959: }
1.173 jca 960: ftp_close(&fin, &tls, &fd);
1.157 deraadt 961: rval = url_get(redirurl, proxyenv, savefile, lastfile);
1.102 halex 962: free(redirurl);
963: goto cleanup_url_get;
1.209 claudio 964: #define RETRYAFTER "Retry-After:"
1.172 jca 965: } else if (isunavail &&
966: strncasecmp(cp, RETRYAFTER, sizeof(RETRYAFTER) - 1) == 0) {
967: size_t s;
968: cp += sizeof(RETRYAFTER) - 1;
1.209 claudio 969: cp += strspn(cp, " \t");
970: cp[strcspn(cp, " \t")] = '\0';
1.172 jca 971: retryafter = strtonum(cp, 0, 0, &errstr);
972: if (errstr != NULL)
973: retryafter = -1;
1.209 claudio 974: #define TRANSFER_ENCODING "Transfer-Encoding:"
1.178 jca 975: } else if (strncasecmp(cp, TRANSFER_ENCODING,
976: sizeof(TRANSFER_ENCODING) - 1) == 0) {
977: cp += sizeof(TRANSFER_ENCODING) - 1;
1.209 claudio 978: cp += strspn(cp, " \t");
1.178 jca 979: cp[strcspn(cp, " \t")] = '\0';
980: if (strcasecmp(cp, "chunked") == 0)
981: chunked = 1;
1.200 robert 982: #ifndef SMALL
1.209 claudio 983: #define LAST_MODIFIED "Last-Modified:"
1.200 robert 984: } else if (strncasecmp(cp, LAST_MODIFIED,
985: sizeof(LAST_MODIFIED) - 1) == 0) {
986: cp += sizeof(LAST_MODIFIED) - 1;
987: cp[strcspn(cp, "\t")] = '\0';
988: if (strptime(cp, "%a, %d %h %Y %T %Z", &lmt) == NULL)
989: server_timestamps = 0;
990: #endif /* !SMALL */
1.40 fgsch 991: }
1.172 jca 992: }
1.201 naddy 993: free(buf);
1.202 naddy 994: buf = NULL;
1.172 jca 995:
1.178 jca 996: /* Content-Length should be ignored for Transfer-Encoding: chunked */
997: if (chunked)
998: filesize = -1;
999:
1.172 jca 1000: if (isunavail) {
1001: if (retried || retryafter != 0)
1.175 deraadt 1002: warnx("Error retrieving %s: 503 Service Unavailable",
1003: origline);
1.172 jca 1004: else {
1005: if (verbose)
1006: fprintf(ttyout, "Retrying %s\n", origline);
1007: retried = 1;
1.174 jca 1008: ftp_close(&fin, &tls, &fd);
1.172 jca 1009: rval = url_get(origline, proxyenv, savefile, lastfile);
1010: }
1011: goto cleanup_url_get;
1.1 millert 1012: }
1013:
1.17 millert 1014: /* Open the output file. */
1.100 halex 1015: if (!pipeout) {
1.75 martynas 1016: #ifndef SMALL
1017: if (resume)
1.83 martynas 1018: out = open(savefile, O_CREAT | O_WRONLY | O_APPEND,
1019: 0666);
1.75 martynas 1020: else
1.78 martynas 1021: #endif /* !SMALL */
1.75 martynas 1022: out = open(savefile, O_CREAT | O_WRONLY | O_TRUNC,
1023: 0666);
1.170 deraadt 1024: if (out == -1) {
1.10 deraadt 1025: warn("Can't open %s", savefile);
1026: goto cleanup_url_get;
1027: }
1.157 deraadt 1028: } else {
1.17 millert 1029: out = fileno(stdout);
1.157 deraadt 1030: #ifdef SMALL
1031: if (lastfile) {
1032: if (pledge("stdio tty", NULL) == -1)
1033: err(1, "pledge");
1034: }
1035: #endif
1036: }
1.1 millert 1037:
1.189 jca 1038: if ((buf = malloc(buflen)) == NULL)
1039: errx(1, "Can't allocate memory for transfer buffer");
1040:
1.1 millert 1041: /* Trap signals */
1042: oldintr = NULL;
1.97 martynas 1043: oldinti = NULL;
1.1 millert 1044: if (setjmp(httpabort)) {
1045: if (oldintr)
1.2 millert 1046: (void)signal(SIGINT, oldintr);
1.97 martynas 1047: if (oldinti)
1048: (void)signal(SIGINFO, oldinti);
1.6 millert 1049: goto cleanup_url_get;
1.1 millert 1050: }
1051: oldintr = signal(SIGINT, aborthttp);
1052:
1053: bytes = 0;
1054: hashbytes = mark;
1.84 martynas 1055: progressmeter(-1, path);
1.43 millert 1056:
1.1 millert 1057: /* Finally, suck down the file. */
1.97 martynas 1058: oldinti = signal(SIGINFO, psummary);
1.178 jca 1059: if (chunked) {
1.182 jca 1060: error = save_chunked(fin, tls, out, buf, buflen);
1.196 jca 1061: signal(SIGINT, oldintr);
1.182 jca 1062: signal(SIGINFO, oldinti);
1063: if (error == -1)
1.178 jca 1064: goto cleanup_url_get;
1065: } else {
1.188 jca 1066: while ((len = fread(buf, 1, buflen, fin)) > 0) {
1.178 jca 1067: bytes += len;
1.188 jca 1068: for (cp = buf; len > 0; len -= wlen, cp += wlen) {
1069: if ((wlen = write(out, cp, len)) == -1) {
1.178 jca 1070: warn("Writing %s", savefile);
1.196 jca 1071: signal(SIGINT, oldintr);
1.178 jca 1072: signal(SIGINFO, oldinti);
1073: goto cleanup_url_get;
1.188 jca 1074: }
1.178 jca 1075: }
1076: if (hash && !progress) {
1077: while (bytes >= hashbytes) {
1078: (void)putc('#', ttyout);
1079: hashbytes += mark;
1080: }
1081: (void)fflush(ttyout);
1.1 millert 1082: }
1083: }
1.188 jca 1084: save_errno = errno;
1.196 jca 1085: signal(SIGINT, oldintr);
1.178 jca 1086: signal(SIGINFO, oldinti);
1087: if (hash && !progress && bytes > 0) {
1088: if (bytes < mark)
1.10 deraadt 1089: (void)putc('#', ttyout);
1.178 jca 1090: (void)putc('\n', ttyout);
1.10 deraadt 1091: (void)fflush(ttyout);
1.1 millert 1092: }
1.188 jca 1093: if (len == 0 && ferror(fin)) {
1094: errno = save_errno;
1.178 jca 1095: warnx("Reading from socket: %s", sockerror(tls));
1096: goto cleanup_url_get;
1097: }
1.1 millert 1098: }
1.84 martynas 1099: progressmeter(1, NULL);
1.75 martynas 1100: if (
1101: #ifndef SMALL
1102: !resume &&
1.78 martynas 1103: #endif /* !SMALL */
1.75 martynas 1104: filesize != -1 && len == 0 && bytes != filesize) {
1.24 deraadt 1105: if (verbose)
1106: fputs("Read short file.\n", ttyout);
1107: goto cleanup_url_get;
1108: }
1109:
1.1 millert 1110: if (verbose)
1.97 martynas 1111: ptransfer(0);
1.1 millert 1112:
1.40 fgsch 1113: rval = 0;
1114: goto cleanup_url_get;
1.1 millert 1115:
1.14 millert 1116: noftpautologin:
1117: warnx(
1118: "Auto-login using ftp URLs isn't supported when using $ftp_proxy");
1119: goto cleanup_url_get;
1120:
1.1 millert 1121: improper:
1.8 millert 1122: warnx("Improper response from %s", host);
1.14 millert 1123:
1.6 millert 1124: cleanup_url_get:
1.181 jca 1125: #ifndef SMALL
1126: free(full_host);
1127: #endif /* !SMALL */
1.154 deraadt 1128: #ifndef NOSSL
1.128 jca 1129: free(sslhost);
1.154 deraadt 1130: #endif /* !NOSSL */
1.173 jca 1131: ftp_close(&fin, &tls, &fd);
1.200 robert 1132: if (out >= 0 && out != fileno(stdout)) {
1133: #ifndef SMALL
1.203 sthen 1134: if (server_timestamps && lmt.tm_zone != 0 &&
1135: fstat(out, &stbuf) == 0 && S_ISREG(stbuf.st_mode) != 0) {
1.200 robert 1136: ts[0].tv_nsec = UTIME_NOW;
1137: ts[1].tv_nsec = 0;
1138: setenv("TZ", lmt.tm_zone, 1);
1139: if (((ts[1].tv_sec = mktime(&lmt)) != -1) &&
1140: (futimens(out, ts) == -1))
1141: warnx("Unable to set file modification time");
1142: }
1143: #endif /* !SMALL */
1.162 sthen 1144: close(out);
1.200 robert 1145: }
1.67 steven 1146: free(buf);
1.198 naddy 1147: free(pathbuf);
1.136 bluhm 1148: free(proxyhost);
1.67 steven 1149: free(proxyurl);
1.62 ray 1150: free(newline);
1.123 guenther 1151: free(credentials);
1.190 yasuoka 1152: free(proxy_credentials);
1.40 fgsch 1153: return (rval);
1.178 jca 1154: }
1155:
1156: static int
1157: save_chunked(FILE *fin, struct tls *tls, int out, char *buf, size_t buflen)
1158: {
1159:
1.201 naddy 1160: char *header = NULL, *end, *cp;
1.178 jca 1161: unsigned long chunksize;
1.201 naddy 1162: size_t hsize = 0, rlen, wlen;
1.178 jca 1163: ssize_t written;
1164: char cr, lf;
1165:
1166: for (;;) {
1.201 naddy 1167: if (getline(&header, &hsize, fin) == -1)
1.178 jca 1168: break;
1169: /* strip CRLF and any optional chunk extension */
1.211 claudio 1170: header[strcspn(header, "; \t\r\n")] = '\0';
1.178 jca 1171: errno = 0;
1172: chunksize = strtoul(header, &end, 16);
1173: if (errno || header[0] == '\0' || *end != '\0' ||
1174: chunksize > INT_MAX) {
1175: warnx("Invalid chunk size '%s'", header);
1176: free(header);
1177: return -1;
1178: }
1179:
1180: if (chunksize == 0) {
1181: /* We're done. Ignore optional trailer. */
1.201 naddy 1182: free(header);
1.178 jca 1183: return 0;
1184: }
1185:
1186: for (written = 0; chunksize != 0; chunksize -= rlen) {
1187: rlen = (chunksize < buflen) ? chunksize : buflen;
1188: rlen = fread(buf, 1, rlen, fin);
1189: if (rlen == 0)
1190: break;
1191: bytes += rlen;
1192: for (cp = buf, wlen = rlen; wlen > 0;
1.179 deraadt 1193: wlen -= written, cp += written) {
1.178 jca 1194: if ((written = write(out, cp, wlen)) == -1) {
1195: warn("Writing output file");
1.201 naddy 1196: free(header);
1.178 jca 1197: return -1;
1198: }
1199: }
1200: }
1201:
1202: if (rlen == 0 ||
1203: fread(&cr, 1, 1, fin) != 1 ||
1204: fread(&lf, 1, 1, fin) != 1)
1205: break;
1206:
1207: if (cr != '\r' || lf != '\n') {
1208: warnx("Invalid chunked encoding");
1.201 naddy 1209: free(header);
1.178 jca 1210: return -1;
1211: }
1212: }
1.201 naddy 1213: free(header);
1.178 jca 1214:
1215: if (ferror(fin))
1216: warnx("Error while reading from socket: %s", sockerror(tls));
1217: else
1218: warnx("Invalid chunked encoding: short read");
1219:
1220: return -1;
1.1 millert 1221: }
1222:
1223: /*
1224: * Abort a http retrieval
1225: */
1.177 jca 1226: static void
1.51 deraadt 1227: aborthttp(int signo)
1.1 millert 1228: {
1.197 kn 1229: const char errmsg[] = "\nfetch aborted.\n";
1.1 millert 1230:
1.197 kn 1231: write(fileno(ttyout), errmsg, sizeof(errmsg) - 1);
1.22 deraadt 1232: longjmp(httpabort, 1);
1233: }
1234:
1235: /*
1.1 millert 1236: * Retrieve multiple files from the command line, transferring
1237: * files of the form "host:path", "ftp://host/path" using the
1238: * ftp protocol, and files of the form "http://host/path" using
1239: * the http protocol.
1.2 millert 1240: * If path has a trailing "/", then return (-1);
1.1 millert 1241: * the path will be cd-ed into and the connection remains open,
1242: * and the function will return -1 (to indicate the connection
1243: * is alive).
1244: * If an error occurs the return value will be the offset+1 in
1245: * argv[] of the file that caused a problem (i.e, argv[x]
1246: * returns x+1)
1247: * Otherwise, 0 is returned if all files retrieved successfully.
1248: */
1249: int
1.50 deraadt 1250: auto_fetch(int argc, char *argv[], char *outfile)
1.1 millert 1251: {
1252: char *xargv[5];
1.62 ray 1253: char *cp, *url, *host, *dir, *file, *portnum;
1254: char *username, *pass, *pathstart;
1.6 millert 1255: char *ftpproxy, *httpproxy;
1.157 deraadt 1256: int rval, xargc, lastfile;
1.14 millert 1257: volatile int argpos;
1.49 krw 1258: int dirhasglob, filehasglob, oautologin;
1.137 deraadt 1259: char rempath[PATH_MAX];
1.1 millert 1260:
1261: argpos = 0;
1262:
1263: if (setjmp(toplevel)) {
1264: if (connected)
1265: disconnect(0, NULL);
1.2 millert 1266: return (argpos + 1);
1.1 millert 1267: }
1.3 millert 1268: (void)signal(SIGINT, (sig_t)intr);
1269: (void)signal(SIGPIPE, (sig_t)lostpeer);
1.1 millert 1270:
1.45 millert 1271: if ((ftpproxy = getenv(FTP_PROXY)) != NULL && *ftpproxy == '\0')
1272: ftpproxy = NULL;
1273: if ((httpproxy = getenv(HTTP_PROXY)) != NULL && *httpproxy == '\0')
1274: httpproxy = NULL;
1.6 millert 1275:
1.1 millert 1276: /*
1277: * Loop through as long as there's files to fetch.
1278: */
1.123 guenther 1279: username = pass = NULL;
1.62 ray 1280: for (rval = 0; (rval == 0) && (argpos < argc); free(url), argpos++) {
1.1 millert 1281: if (strchr(argv[argpos], ':') == NULL)
1282: break;
1.123 guenther 1283:
1284: free(username);
1285: free(pass);
1.62 ray 1286: host = dir = file = portnum = username = pass = NULL;
1.1 millert 1287:
1.157 deraadt 1288: lastfile = (argv[argpos+1] == NULL);
1289:
1.1 millert 1290: /*
1291: * We muck with the string, so we make a copy.
1292: */
1.62 ray 1293: url = strdup(argv[argpos]);
1294: if (url == NULL)
1.1 millert 1295: errx(1, "Can't allocate memory for auto-fetch.");
1296:
1.186 jca 1297: if (strncasecmp(url, FILE_URL, sizeof(FILE_URL) - 1) == 0) {
1298: if (file_get(url + sizeof(FILE_URL) - 1, outfile) == -1)
1299: rval = argpos + 1;
1300: continue;
1301: }
1302:
1.1 millert 1303: /*
1.186 jca 1304: * Try HTTP URL-style arguments next.
1.1 millert 1305: */
1.62 ray 1306: if (strncasecmp(url, HTTP_URL, sizeof(HTTP_URL) - 1) == 0 ||
1.186 jca 1307: strncasecmp(url, HTTPS_URL, sizeof(HTTPS_URL) -1) == 0) {
1.54 fgsch 1308: redirect_loop = 0;
1.171 jca 1309: retried = 0;
1.157 deraadt 1310: if (url_get(url, httpproxy, outfile, lastfile) == -1)
1.1 millert 1311: rval = argpos + 1;
1312: continue;
1313: }
1314:
1315: /*
1.6 millert 1316: * Try FTP URL-style arguments next. If ftpproxy is
1317: * set, use url_get() instead of standard ftp.
1318: * Finally, try host:file.
1.1 millert 1319: */
1.62 ray 1320: host = url;
1321: if (strncasecmp(url, FTP_URL, sizeof(FTP_URL) - 1) == 0) {
1.37 heko 1322: char *passend, *passagain, *userend;
1.31 itojun 1323:
1.6 millert 1324: if (ftpproxy) {
1.157 deraadt 1325: if (url_get(url, ftpproxy, outfile, lastfile) == -1)
1.6 millert 1326: rval = argpos + 1;
1327: continue;
1328: }
1.1 millert 1329: host += sizeof(FTP_URL) - 1;
1.8 millert 1330: dir = strchr(host, '/');
1.1 millert 1331:
1.8 millert 1332: /* Look for [user:pass@]host[:port] */
1.31 itojun 1333:
1334: /* check if we have "user:pass@" */
1.37 heko 1335: userend = strchr(host, ':');
1.31 itojun 1336: passend = strchr(host, '@');
1337: if (passend && userend && userend < passend &&
1338: (!dir || passend < dir)) {
1.62 ray 1339: username = host;
1.31 itojun 1340: pass = userend + 1;
1341: host = passend + 1;
1342: *userend = *passend = '\0';
1.37 heko 1343: passagain = strchr(host, '@');
1.42 deraadt 1344: if (strchr(pass, '@') != NULL ||
1.37 heko 1345: (passagain != NULL && passagain < dir)) {
1346: warnx(at_encoding_warning);
1.123 guenther 1347: username = pass = NULL;
1.37 heko 1348: goto bad_ftp_url;
1.42 deraadt 1349: }
1.31 itojun 1350:
1.77 martynas 1351: if (EMPTYSTRING(username)) {
1.11 millert 1352: bad_ftp_url:
1.31 itojun 1353: warnx("Invalid URL: %s", argv[argpos]);
1354: rval = argpos + 1;
1.123 guenther 1355: username = pass = NULL;
1.31 itojun 1356: continue;
1357: }
1.62 ray 1358: username = urldecode(username);
1.37 heko 1359: pass = urldecode(pass);
1.8 millert 1360: }
1.31 itojun 1361:
1362: /* check [host]:port, or [host] */
1363: if (host[0] == '[') {
1364: cp = strchr(host, ']');
1365: if (cp && (!dir || cp < dir)) {
1366: if (cp + 1 == dir || cp[1] == ':') {
1367: host++;
1368: *cp++ = '\0';
1369: } else
1370: cp = NULL;
1371: } else
1372: cp = host;
1.25 itojun 1373: } else
1374: cp = host;
1.31 itojun 1375:
1376: /* split off host[:port] if there is */
1377: if (cp) {
1378: portnum = strchr(cp, ':');
1.52 henning 1379: pathstart = strchr(cp, '/');
1380: /* : in path is not a port # indicator */
1381: if (portnum && pathstart &&
1382: pathstart < portnum)
1383: portnum = NULL;
1384:
1.31 itojun 1385: if (!portnum)
1386: ;
1387: else {
1388: if (!dir)
1389: ;
1390: else if (portnum + 1 < dir) {
1391: *portnum++ = '\0';
1392: /*
1393: * XXX should check if portnum
1394: * is decimal number
1395: */
1396: } else {
1397: /* empty portnum */
1398: goto bad_ftp_url;
1399: }
1400: }
1401: } else
1402: portnum = NULL;
1.8 millert 1403: } else { /* classic style `host:file' */
1404: dir = strchr(host, ':');
1405: }
1.1 millert 1406: if (EMPTYSTRING(host)) {
1407: rval = argpos + 1;
1408: continue;
1409: }
1410:
1411: /*
1.9 millert 1412: * If dir is NULL, the file wasn't specified
1.1 millert 1413: * (URL looked something like ftp://host)
1414: */
1.8 millert 1415: if (dir != NULL)
1416: *dir++ = '\0';
1.1 millert 1417:
1418: /*
1419: * Extract the file and (if present) directory name.
1420: */
1.42 deraadt 1421: if (!EMPTYSTRING(dir)) {
1.8 millert 1422: cp = strrchr(dir, '/');
1.1 millert 1423: if (cp != NULL) {
1424: *cp++ = '\0';
1425: file = cp;
1426: } else {
1427: file = dir;
1428: dir = NULL;
1429: }
1430: }
1.80 martynas 1431: #ifndef SMALL
1.1 millert 1432: if (debug)
1.42 deraadt 1433: fprintf(ttyout,
1434: "user %s:%s host %s port %s dir %s file %s\n",
1.76 martynas 1435: username, pass ? "XXXX" : NULL, host, portnum,
1436: dir, file);
1.80 martynas 1437: #endif /* !SMALL */
1.1 millert 1438:
1439: /*
1.49 krw 1440: * Set up the connection.
1.1 millert 1441: */
1.49 krw 1442: if (connected)
1443: disconnect(0, NULL);
1444: xargv[0] = __progname;
1445: xargv[1] = host;
1.8 millert 1446: xargv[2] = NULL;
1.49 krw 1447: xargc = 2;
1448: if (!EMPTYSTRING(portnum)) {
1449: xargv[2] = portnum;
1450: xargv[3] = NULL;
1451: xargc = 3;
1452: }
1453: oautologin = autologin;
1.88 martynas 1454: if (username == NULL)
1.89 halex 1455: anonftp = 1;
1456: else {
1457: anonftp = 0;
1.49 krw 1458: autologin = 0;
1.89 halex 1459: }
1.49 krw 1460: setpeer(xargc, xargv);
1461: autologin = oautologin;
1.87 martynas 1462: if (connected == 0 ||
1463: (connected == 1 && autologin && (username == NULL ||
1464: !ftp_login(host, username, pass)))) {
1.49 krw 1465: warnx("Can't connect or login to host `%s'", host);
1.8 millert 1466: rval = argpos + 1;
1467: continue;
1.1 millert 1468: }
1.49 krw 1469:
1470: /* Always use binary transfers. */
1471: setbinary(0, NULL);
1.1 millert 1472:
1.4 millert 1473: dirhasglob = filehasglob = 0;
1474: if (doglob) {
1.42 deraadt 1475: if (!EMPTYSTRING(dir) &&
1.4 millert 1476: strpbrk(dir, "*?[]{}") != NULL)
1477: dirhasglob = 1;
1.42 deraadt 1478: if (!EMPTYSTRING(file) &&
1.4 millert 1479: strpbrk(file, "*?[]{}") != NULL)
1480: filehasglob = 1;
1481: }
1482:
1.1 millert 1483: /* Change directories, if necessary. */
1.42 deraadt 1484: if (!EMPTYSTRING(dir) && !dirhasglob) {
1.1 millert 1485: xargv[0] = "cd";
1486: xargv[1] = dir;
1487: xargv[2] = NULL;
1488: cd(2, xargv);
1.42 deraadt 1489: if (!dirchange) {
1.1 millert 1490: rval = argpos + 1;
1491: continue;
1492: }
1493: }
1494:
1495: if (EMPTYSTRING(file)) {
1.86 martynas 1496: #ifndef SMALL
1.1 millert 1497: rval = -1;
1.86 martynas 1498: #else /* !SMALL */
1499: recvrequest("NLST", "-", NULL, "w", 0, 0);
1500: rval = 0;
1501: #endif /* !SMALL */
1.1 millert 1502: continue;
1503: }
1504:
1.21 marc 1505: if (verbose)
1.10 deraadt 1506: fprintf(ttyout, "Retrieving %s/%s\n", dir ? dir : "", file);
1.1 millert 1507:
1.4 millert 1508: if (dirhasglob) {
1509: snprintf(rempath, sizeof(rempath), "%s/%s", dir, file);
1510: file = rempath;
1511: }
1512:
1513: /* Fetch the file(s). */
1.10 deraadt 1514: xargc = 2;
1.1 millert 1515: xargv[0] = "get";
1516: xargv[1] = file;
1517: xargv[2] = NULL;
1.4 millert 1518: if (dirhasglob || filehasglob) {
1519: int ointeractive;
1520:
1521: ointeractive = interactive;
1522: interactive = 0;
1523: xargv[0] = "mget";
1.78 martynas 1524: #ifndef SMALL
1525: if (resume) {
1526: xargc = 3;
1527: xargv[1] = "-c";
1528: xargv[2] = file;
1529: xargv[3] = NULL;
1530: }
1531: #endif /* !SMALL */
1.10 deraadt 1532: mget(xargc, xargv);
1.5 millert 1533: interactive = ointeractive;
1.10 deraadt 1534: } else {
1.17 millert 1535: if (outfile != NULL) {
1536: xargv[2] = outfile;
1537: xargv[3] = NULL;
1.10 deraadt 1538: xargc++;
1539: }
1.75 martynas 1540: #ifndef SMALL
1541: if (resume)
1542: reget(xargc, xargv);
1543: else
1.78 martynas 1544: #endif /* !SMALL */
1.75 martynas 1545: get(xargc, xargv);
1.10 deraadt 1546: }
1.1 millert 1547:
1.4 millert 1548: if ((code / 100) != COMPLETE)
1.1 millert 1549: rval = argpos + 1;
1550: }
1551: if (connected && rval != -1)
1552: disconnect(0, NULL);
1553: return (rval);
1.37 heko 1554: }
1555:
1556: char *
1.50 deraadt 1557: urldecode(const char *str)
1.37 heko 1558: {
1.53 deraadt 1559: char *ret, c;
1560: int i, reallen;
1.37 heko 1561:
1.53 deraadt 1562: if (str == NULL)
1563: return NULL;
1.205 claudio 1564: if ((ret = malloc(strlen(str) + 1)) == NULL)
1.53 deraadt 1565: err(1, "Can't allocate memory for URL decoding");
1566: for (i = 0, reallen = 0; str[i] != '\0'; i++, reallen++, ret++) {
1567: c = str[i];
1568: if (c == '+') {
1569: *ret = ' ';
1570: continue;
1571: }
1.61 deraadt 1572:
1573: /* Cannot use strtol here because next char
1574: * after %xx may be a digit.
1575: */
1.205 claudio 1576: if (c == '%' && isxdigit((unsigned char)str[i + 1]) &&
1577: isxdigit((unsigned char)str[i + 2])) {
1578: *ret = hextochar(&str[i + 1]);
1579: i += 2;
1.53 deraadt 1580: continue;
1581: }
1582: *ret = c;
1583: }
1584: *ret = '\0';
1585:
1.205 claudio 1586: return ret - reallen;
1.123 guenther 1587: }
1588:
1.177 jca 1589: static char *
1.123 guenther 1590: recode_credentials(const char *userinfo)
1591: {
1592: char *ui, *creds;
1593: size_t ulen, credsize;
1594:
1595: /* url-decode the user and pass */
1596: ui = urldecode(userinfo);
1597:
1598: ulen = strlen(ui);
1599: credsize = (ulen + 2) / 3 * 4 + 1;
1600: creds = malloc(credsize);
1601: if (creds == NULL)
1602: errx(1, "out of memory");
1603: if (b64_ntop(ui, ulen, creds, credsize) == -1)
1604: errx(1, "error in base64 encoding");
1605: free(ui);
1606: return (creds);
1.37 heko 1607: }
1608:
1.177 jca 1609: static char
1.50 deraadt 1610: hextochar(const char *str)
1.37 heko 1611: {
1.143 guenther 1612: unsigned char c, ret;
1.37 heko 1613:
1.53 deraadt 1614: c = str[0];
1615: ret = c;
1616: if (isalpha(c))
1617: ret -= isupper(c) ? 'A' - 10 : 'a' - 10;
1618: else
1619: ret -= '0';
1620: ret *= 16;
1621:
1622: c = str[1];
1623: ret += c;
1624: if (isalpha(c))
1625: ret -= isupper(c) ? 'A' - 10 : 'a' - 10;
1626: else
1627: ret -= '0';
1628: return ret;
1.25 itojun 1629: }
1630:
1631: int
1.50 deraadt 1632: isurl(const char *p)
1.25 itojun 1633: {
1.27 millert 1634:
1.26 deraadt 1635: if (strncasecmp(p, FTP_URL, sizeof(FTP_URL) - 1) == 0 ||
1636: strncasecmp(p, HTTP_URL, sizeof(HTTP_URL) - 1) == 0 ||
1.154 deraadt 1637: #ifndef NOSSL
1.61 deraadt 1638: strncasecmp(p, HTTPS_URL, sizeof(HTTPS_URL) - 1) == 0 ||
1.154 deraadt 1639: #endif /* !NOSSL */
1.27 millert 1640: strncasecmp(p, FILE_URL, sizeof(FILE_URL) - 1) == 0 ||
1641: strstr(p, ":/"))
1642: return (1);
1643: return (0);
1.173 jca 1644: }
1.184 jca 1645:
1646: #ifndef SMALL
1647: static int
1648: ftp_printf(FILE *fp, const char *fmt, ...)
1649: {
1650: va_list ap;
1651: int ret;
1652:
1653: va_start(ap, fmt);
1654: ret = vfprintf(fp, fmt, ap);
1655: va_end(ap);
1656:
1657: if (debug) {
1658: va_start(ap, fmt);
1659: vfprintf(ttyout, fmt, ap);
1660: va_end(ap);
1661: }
1662:
1663: return ret;
1664: }
1665: #endif /* !SMALL */
1.173 jca 1666:
1.177 jca 1667: static void
1.189 jca 1668: ftp_close(FILE **fin, struct tls **tls, int *fd)
1.173 jca 1669: {
1670: #ifndef NOSSL
1671: int ret;
1672:
1673: if (*tls != NULL) {
1674: if (tls_session_fd != -1)
1675: dprintf(STDERR_FILENO, "tls session resumed: %s\n",
1676: tls_conn_session_resumed(*tls) ? "yes" : "no");
1677: do {
1678: ret = tls_close(*tls);
1679: } while (ret == TLS_WANT_POLLIN || ret == TLS_WANT_POLLOUT);
1680: tls_free(*tls);
1681: *tls = NULL;
1682: }
1.176 jca 1683: if (*fd != -1) {
1684: close(*fd);
1685: *fd = -1;
1686: }
1.173 jca 1687: #endif
1688: if (*fin != NULL) {
1689: fclose(*fin);
1690: *fin = NULL;
1691: }
1.61 deraadt 1692: }
1693:
1.177 jca 1694: static const char *
1.176 jca 1695: sockerror(struct tls *tls)
1696: {
1697: int save_errno = errno;
1.154 deraadt 1698: #ifndef NOSSL
1.176 jca 1699: if (tls != NULL) {
1700: const char *tlserr = tls_error(tls);
1701: if (tlserr != NULL)
1702: return tlserr;
1.139 bluhm 1703: }
1.176 jca 1704: #endif
1705: return strerror(save_errno);
1.61 deraadt 1706: }
1707:
1.176 jca 1708: #ifndef NOSSL
1.177 jca 1709: static int
1.81 espie 1710: proxy_connect(int socket, char *host, char *cookie)
1.61 deraadt 1711: {
1.66 ray 1712: int l;
1.61 deraadt 1713: char buf[1024];
1.192 yasuoka 1714: char *connstr, *hosttail, *port;
1.61 deraadt 1715:
1716: if (*host == '[' && (hosttail = strrchr(host, ']')) != NULL &&
1717: (hosttail[1] == '\0' || hosttail[1] == ':')) {
1718: host++;
1719: *hosttail++ = '\0';
1720: } else
1721: hosttail = host;
1722:
1.179 deraadt 1723: port = strrchr(hosttail, ':'); /* find portnum */
1.61 deraadt 1724: if (port != NULL)
1725: *port++ = '\0';
1726: if (!port)
1727: port = "443";
1728:
1.81 espie 1729: if (cookie) {
1730: l = asprintf(&connstr, "CONNECT %s:%s HTTP/1.1\r\n"
1731: "Proxy-Authorization: Basic %s\r\n%s\r\n\r\n",
1732: host, port, cookie, HTTP_USER_AGENT);
1733: } else {
1734: l = asprintf(&connstr, "CONNECT %s:%s HTTP/1.1\r\n%s\r\n\r\n",
1735: host, port, HTTP_USER_AGENT);
1736: }
1737:
1.66 ray 1738: if (l == -1)
1.61 deraadt 1739: errx(1, "Could not allocate memory to assemble connect string!");
1.80 martynas 1740: #ifndef SMALL
1.68 ray 1741: if (debug)
1742: printf("%s", connstr);
1.80 martynas 1743: #endif /* !SMALL */
1.66 ray 1744: if (write(socket, connstr, l) != l)
1.68 ray 1745: err(1, "Could not send connect string");
1.192 yasuoka 1746: read(socket, &buf, sizeof(buf)); /* only proxy header XXX: error handling? */
1.71 ray 1747: free(connstr);
1.61 deraadt 1748: return(200);
1.176 jca 1749: }
1750:
1.177 jca 1751: static int
1.176 jca 1752: stdio_tls_write_wrapper(void *arg, const char *buf, int len)
1753: {
1754: struct tls *tls = arg;
1755: ssize_t ret;
1756:
1757: do {
1758: ret = tls_write(tls, buf, len);
1759: } while (ret == TLS_WANT_POLLIN || ret == TLS_WANT_POLLOUT);
1760:
1761: return ret;
1762: }
1763:
1.177 jca 1764: static int
1.176 jca 1765: stdio_tls_read_wrapper(void *arg, char *buf, int len)
1766: {
1767: struct tls *tls = arg;
1768: ssize_t ret;
1769:
1770: do {
1771: ret = tls_read(tls, buf, len);
1772: } while (ret == TLS_WANT_POLLIN || ret == TLS_WANT_POLLOUT);
1773:
1774: return ret;
1.61 deraadt 1775: }
1.158 jca 1776: #endif /* !NOSSL */