http: Add support for reading http POST reply headers
[ffmpeg.git] / libavformat / http.c
1 /*
2  * HTTP protocol for avconv client
3  * Copyright (c) 2000, 2001 Fabrice Bellard
4  *
5  * This file is part of Libav.
6  *
7  * Libav is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Lesser General Public
9  * License as published by the Free Software Foundation; either
10  * version 2.1 of the License, or (at your option) any later version.
11  *
12  * Libav is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15  * Lesser General Public License for more details.
16  *
17  * You should have received a copy of the GNU Lesser General Public
18  * License along with Libav; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20  */
21
22 #include "libavutil/avstring.h"
23 #include "avformat.h"
24 #include <unistd.h>
25 #include "internal.h"
26 #include "network.h"
27 #include "http.h"
28 #include "os_support.h"
29 #include "httpauth.h"
30 #include "url.h"
31 #include "libavutil/opt.h"
32
33 /* XXX: POST protocol is not completely implemented because avconv uses
34    only a subset of it. */
35
36 /* used for protocol handling */
37 #define BUFFER_SIZE 1024
38 #define MAX_REDIRECTS 8
39
40 typedef struct {
41     const AVClass *class;
42     URLContext *hd;
43     unsigned char buffer[BUFFER_SIZE], *buf_ptr, *buf_end;
44     int line_count;
45     int http_code;
46     int64_t chunksize;      /**< Used if "Transfer-Encoding: chunked" otherwise -1. */
47     int64_t off, filesize;
48     char location[MAX_URL_SIZE];
49     HTTPAuthState auth_state;
50     HTTPAuthState proxy_auth_state;
51     char *headers;
52     int willclose;          /**< Set if the server correctly handles Connection: close and will close the connection after feeding us the content. */
53     int chunked_post;
54     int end_chunked_post;   /**< A flag which indicates if the end of chunked encoding has been sent. */
55     int end_header;         /**< A flag which indicates we have finished to read POST reply. */
56 } HTTPContext;
57
58 #define OFFSET(x) offsetof(HTTPContext, x)
59 #define D AV_OPT_FLAG_DECODING_PARAM
60 #define E AV_OPT_FLAG_ENCODING_PARAM
61 static const AVOption options[] = {
62 {"chunked_post", "use chunked transfer-encoding for posts", OFFSET(chunked_post), AV_OPT_TYPE_INT, {.dbl = 1}, 0, 1, E },
63 {"headers", "custom HTTP headers, can override built in default headers", OFFSET(headers), AV_OPT_TYPE_STRING, { 0 }, 0, 0, D|E },
64 {NULL}
65 };
66 #define HTTP_CLASS(flavor)\
67 static const AVClass flavor ## _context_class = {\
68     .class_name     = #flavor,\
69     .item_name      = av_default_item_name,\
70     .option         = options,\
71     .version        = LIBAVUTIL_VERSION_INT,\
72 }
73
74 HTTP_CLASS(http);
75 HTTP_CLASS(https);
76
77 static int http_connect(URLContext *h, const char *path, const char *local_path,
78                         const char *hoststr, const char *auth,
79                         const char *proxyauth, int *new_location);
80
81 void ff_http_init_auth_state(URLContext *dest, const URLContext *src)
82 {
83     memcpy(&((HTTPContext*)dest->priv_data)->auth_state,
84            &((HTTPContext*)src->priv_data)->auth_state, sizeof(HTTPAuthState));
85     memcpy(&((HTTPContext*)dest->priv_data)->proxy_auth_state,
86            &((HTTPContext*)src->priv_data)->proxy_auth_state,
87            sizeof(HTTPAuthState));
88 }
89
90 /* return non zero if error */
91 static int http_open_cnx(URLContext *h)
92 {
93     const char *path, *proxy_path, *lower_proto = "tcp", *local_path;
94     char hostname[1024], hoststr[1024], proto[10];
95     char auth[1024], proxyauth[1024] = "";
96     char path1[1024];
97     char buf[1024], urlbuf[1024];
98     int port, use_proxy, err, location_changed = 0, redirects = 0, attempts = 0;
99     HTTPAuthType cur_auth_type, cur_proxy_auth_type;
100     HTTPContext *s = h->priv_data;
101     URLContext *hd = NULL;
102
103     proxy_path = getenv("http_proxy");
104     use_proxy = (proxy_path != NULL) && !getenv("no_proxy") &&
105         av_strstart(proxy_path, "http://", NULL);
106
107     /* fill the dest addr */
108  redo:
109     /* needed in any case to build the host string */
110     av_url_split(proto, sizeof(proto), auth, sizeof(auth),
111                  hostname, sizeof(hostname), &port,
112                  path1, sizeof(path1), s->location);
113     ff_url_join(hoststr, sizeof(hoststr), NULL, NULL, hostname, port, NULL);
114
115     if (!strcmp(proto, "https")) {
116         lower_proto = "tls";
117         use_proxy = 0;
118         if (port < 0)
119             port = 443;
120     }
121     if (port < 0)
122         port = 80;
123
124     if (path1[0] == '\0')
125         path = "/";
126     else
127         path = path1;
128     local_path = path;
129     if (use_proxy) {
130         /* Reassemble the request URL without auth string - we don't
131          * want to leak the auth to the proxy. */
132         ff_url_join(urlbuf, sizeof(urlbuf), proto, NULL, hostname, port, "%s",
133                     path1);
134         path = urlbuf;
135         av_url_split(NULL, 0, proxyauth, sizeof(proxyauth),
136                      hostname, sizeof(hostname), &port, NULL, 0, proxy_path);
137     }
138
139     ff_url_join(buf, sizeof(buf), lower_proto, NULL, hostname, port, NULL);
140     err = ffurl_open(&hd, buf, AVIO_FLAG_READ_WRITE,
141                      &h->interrupt_callback, NULL);
142     if (err < 0)
143         goto fail;
144
145     s->hd = hd;
146     cur_auth_type = s->auth_state.auth_type;
147     cur_proxy_auth_type = s->auth_state.auth_type;
148     if (http_connect(h, path, local_path, hoststr, auth, proxyauth, &location_changed) < 0)
149         goto fail;
150     attempts++;
151     if (s->http_code == 401) {
152         if ((cur_auth_type == HTTP_AUTH_NONE || s->auth_state.stale) &&
153             s->auth_state.auth_type != HTTP_AUTH_NONE && attempts < 4) {
154             ffurl_close(hd);
155             goto redo;
156         } else
157             goto fail;
158     }
159     if (s->http_code == 407) {
160         if ((cur_proxy_auth_type == HTTP_AUTH_NONE || s->proxy_auth_state.stale) &&
161             s->proxy_auth_state.auth_type != HTTP_AUTH_NONE && attempts < 4) {
162             ffurl_close(hd);
163             goto redo;
164         } else
165             goto fail;
166     }
167     if ((s->http_code == 301 || s->http_code == 302 || s->http_code == 303 || s->http_code == 307)
168         && location_changed == 1) {
169         /* url moved, get next */
170         ffurl_close(hd);
171         if (redirects++ >= MAX_REDIRECTS)
172             return AVERROR(EIO);
173         /* Restart the authentication process with the new target, which
174          * might use a different auth mechanism. */
175         memset(&s->auth_state, 0, sizeof(s->auth_state));
176         attempts = 0;
177         location_changed = 0;
178         goto redo;
179     }
180     return 0;
181  fail:
182     if (hd)
183         ffurl_close(hd);
184     s->hd = NULL;
185     return AVERROR(EIO);
186 }
187
188 static int http_open(URLContext *h, const char *uri, int flags)
189 {
190     HTTPContext *s = h->priv_data;
191
192     h->is_streamed = 1;
193
194     s->filesize = -1;
195     av_strlcpy(s->location, uri, sizeof(s->location));
196
197     if (s->headers) {
198         int len = strlen(s->headers);
199         if (len < 2 || strcmp("\r\n", s->headers + len - 2))
200             av_log(h, AV_LOG_WARNING, "No trailing CRLF found in HTTP header.\n");
201     }
202
203     return http_open_cnx(h);
204 }
205 static int http_getc(HTTPContext *s)
206 {
207     int len;
208     if (s->buf_ptr >= s->buf_end) {
209         len = ffurl_read(s->hd, s->buffer, BUFFER_SIZE);
210         if (len < 0) {
211             return AVERROR(EIO);
212         } else if (len == 0) {
213             return -1;
214         } else {
215             s->buf_ptr = s->buffer;
216             s->buf_end = s->buffer + len;
217         }
218     }
219     return *s->buf_ptr++;
220 }
221
222 static int http_get_line(HTTPContext *s, char *line, int line_size)
223 {
224     int ch;
225     char *q;
226
227     q = line;
228     for(;;) {
229         ch = http_getc(s);
230         if (ch < 0)
231             return AVERROR(EIO);
232         if (ch == '\n') {
233             /* process line */
234             if (q > line && q[-1] == '\r')
235                 q--;
236             *q = '\0';
237
238             return 0;
239         } else {
240             if ((q - line) < line_size - 1)
241                 *q++ = ch;
242         }
243     }
244 }
245
246 static int process_line(URLContext *h, char *line, int line_count,
247                         int *new_location)
248 {
249     HTTPContext *s = h->priv_data;
250     char *tag, *p, *end;
251
252     /* end of header */
253     if (line[0] == '\0') {
254         s->end_header = 1;
255         return 0;
256     }
257
258     p = line;
259     if (line_count == 0) {
260         while (!isspace(*p) && *p != '\0')
261             p++;
262         while (isspace(*p))
263             p++;
264         s->http_code = strtol(p, &end, 10);
265
266         av_dlog(NULL, "http_code=%d\n", s->http_code);
267
268         /* error codes are 4xx and 5xx, but regard 401 as a success, so we
269          * don't abort until all headers have been parsed. */
270         if (s->http_code >= 400 && s->http_code < 600 && (s->http_code != 401
271             || s->auth_state.auth_type != HTTP_AUTH_NONE) &&
272             (s->http_code != 407 || s->proxy_auth_state.auth_type != HTTP_AUTH_NONE)) {
273             end += strspn(end, SPACE_CHARS);
274             av_log(h, AV_LOG_WARNING, "HTTP error %d %s\n",
275                    s->http_code, end);
276             return -1;
277         }
278     } else {
279         while (*p != '\0' && *p != ':')
280             p++;
281         if (*p != ':')
282             return 1;
283
284         *p = '\0';
285         tag = line;
286         p++;
287         while (isspace(*p))
288             p++;
289         if (!av_strcasecmp(tag, "Location")) {
290             strcpy(s->location, p);
291             *new_location = 1;
292         } else if (!av_strcasecmp (tag, "Content-Length") && s->filesize == -1) {
293             s->filesize = atoll(p);
294         } else if (!av_strcasecmp (tag, "Content-Range")) {
295             /* "bytes $from-$to/$document_size" */
296             const char *slash;
297             if (!strncmp (p, "bytes ", 6)) {
298                 p += 6;
299                 s->off = atoll(p);
300                 if ((slash = strchr(p, '/')) && strlen(slash) > 0)
301                     s->filesize = atoll(slash+1);
302             }
303             h->is_streamed = 0; /* we _can_ in fact seek */
304         } else if (!av_strcasecmp(tag, "Accept-Ranges") && !strncmp(p, "bytes", 5)) {
305             h->is_streamed = 0;
306         } else if (!av_strcasecmp (tag, "Transfer-Encoding") && !av_strncasecmp(p, "chunked", 7)) {
307             s->filesize = -1;
308             s->chunksize = 0;
309         } else if (!av_strcasecmp (tag, "WWW-Authenticate")) {
310             ff_http_auth_handle_header(&s->auth_state, tag, p);
311         } else if (!av_strcasecmp (tag, "Authentication-Info")) {
312             ff_http_auth_handle_header(&s->auth_state, tag, p);
313         } else if (!av_strcasecmp (tag, "Proxy-Authenticate")) {
314             ff_http_auth_handle_header(&s->proxy_auth_state, tag, p);
315         } else if (!av_strcasecmp (tag, "Connection")) {
316             if (!strcmp(p, "close"))
317                 s->willclose = 1;
318         }
319     }
320     return 1;
321 }
322
323 static inline int has_header(const char *str, const char *header)
324 {
325     /* header + 2 to skip over CRLF prefix. (make sure you have one!) */
326     if (!str)
327         return 0;
328     return av_stristart(str, header + 2, NULL) || av_stristr(str, header);
329 }
330
331 static int http_read_header(URLContext *h, int *new_location)
332 {
333     HTTPContext *s = h->priv_data;
334     char line[1024];
335     int err = 0;
336
337     for (;;) {
338         if (http_get_line(s, line, sizeof(line)) < 0)
339             return AVERROR(EIO);
340
341         av_dlog(NULL, "header='%s'\n", line);
342
343         err = process_line(h, line, s->line_count, new_location);
344         if (err < 0)
345             return err;
346         if (err == 0)
347             break;
348         s->line_count++;
349     }
350
351     return err;
352 }
353
354 static int http_connect(URLContext *h, const char *path, const char *local_path,
355                         const char *hoststr, const char *auth,
356                         const char *proxyauth, int *new_location)
357 {
358     HTTPContext *s = h->priv_data;
359     int post, err;
360     char headers[1024] = "";
361     char *authstr = NULL, *proxyauthstr = NULL;
362     int64_t off = s->off;
363     int len = 0;
364     const char *method;
365
366
367     /* send http header */
368     post = h->flags & AVIO_FLAG_WRITE;
369     method = post ? "POST" : "GET";
370     authstr = ff_http_auth_create_response(&s->auth_state, auth, local_path,
371                                            method);
372     proxyauthstr = ff_http_auth_create_response(&s->proxy_auth_state, proxyauth,
373                                                 local_path, method);
374
375     /* set default headers if needed */
376     if (!has_header(s->headers, "\r\nUser-Agent: "))
377        len += av_strlcatf(headers + len, sizeof(headers) - len,
378                           "User-Agent: %s\r\n", LIBAVFORMAT_IDENT);
379     if (!has_header(s->headers, "\r\nAccept: "))
380         len += av_strlcpy(headers + len, "Accept: */*\r\n",
381                           sizeof(headers) - len);
382     if (!has_header(s->headers, "\r\nRange: ") && !post)
383         len += av_strlcatf(headers + len, sizeof(headers) - len,
384                            "Range: bytes=%"PRId64"-\r\n", s->off);
385     if (!has_header(s->headers, "\r\nConnection: "))
386         len += av_strlcpy(headers + len, "Connection: close\r\n",
387                           sizeof(headers)-len);
388     if (!has_header(s->headers, "\r\nHost: "))
389         len += av_strlcatf(headers + len, sizeof(headers) - len,
390                            "Host: %s\r\n", hoststr);
391
392     /* now add in custom headers */
393     if (s->headers)
394         av_strlcpy(headers + len, s->headers, sizeof(headers) - len);
395
396     snprintf(s->buffer, sizeof(s->buffer),
397              "%s %s HTTP/1.1\r\n"
398              "%s"
399              "%s"
400              "%s"
401              "%s%s"
402              "\r\n",
403              method,
404              path,
405              post && s->chunked_post ? "Transfer-Encoding: chunked\r\n" : "",
406              headers,
407              authstr ? authstr : "",
408              proxyauthstr ? "Proxy-" : "", proxyauthstr ? proxyauthstr : "");
409
410     av_freep(&authstr);
411     av_freep(&proxyauthstr);
412     if (ffurl_write(s->hd, s->buffer, strlen(s->buffer)) < 0)
413         return AVERROR(EIO);
414
415     /* init input buffer */
416     s->buf_ptr = s->buffer;
417     s->buf_end = s->buffer;
418     s->line_count = 0;
419     s->off = 0;
420     s->filesize = -1;
421     s->willclose = 0;
422     s->end_chunked_post = 0;
423     if (post) {
424         /* Pretend that it did work. We didn't read any header yet, since
425          * we've still to send the POST data, but the code calling this
426          * function will check http_code after we return. */
427         s->http_code = 200;
428         return 0;
429     }
430     s->chunksize = -1;
431
432     /* wait for header */
433     err = http_read_header(h, new_location);
434     if (err < 0)
435         return err;
436
437     return (off == s->off) ? 0 : -1;
438 }
439
440
441 static int http_buf_read(URLContext *h, uint8_t *buf, int size)
442 {
443     HTTPContext *s = h->priv_data;
444     int len;
445     /* read bytes from input buffer first */
446     len = s->buf_end - s->buf_ptr;
447     if (len > 0) {
448         if (len > size)
449             len = size;
450         memcpy(buf, s->buf_ptr, len);
451         s->buf_ptr += len;
452     } else {
453         if (!s->willclose && s->filesize >= 0 && s->off >= s->filesize)
454             return AVERROR_EOF;
455         len = ffurl_read(s->hd, buf, size);
456     }
457     if (len > 0) {
458         s->off += len;
459         if (s->chunksize > 0)
460             s->chunksize -= len;
461     }
462     return len;
463 }
464
465 static int http_read(URLContext *h, uint8_t *buf, int size)
466 {
467     HTTPContext *s = h->priv_data;
468     int err, new_location;
469
470     if (s->end_chunked_post) {
471         if (!s->end_header) {
472             err = http_read_header(h, &new_location);
473             if (err < 0)
474                 return err;
475         }
476
477         return http_buf_read(h, buf, size);
478     }
479
480     if (s->chunksize >= 0) {
481         if (!s->chunksize) {
482             char line[32];
483
484             for(;;) {
485                 do {
486                     if (http_get_line(s, line, sizeof(line)) < 0)
487                         return AVERROR(EIO);
488                 } while (!*line);    /* skip CR LF from last chunk */
489
490                 s->chunksize = strtoll(line, NULL, 16);
491
492                 av_dlog(NULL, "Chunked encoding data size: %"PRId64"'\n", s->chunksize);
493
494                 if (!s->chunksize)
495                     return 0;
496                 break;
497             }
498         }
499         size = FFMIN(size, s->chunksize);
500     }
501     return http_buf_read(h, buf, size);
502 }
503
504 /* used only when posting data */
505 static int http_write(URLContext *h, const uint8_t *buf, int size)
506 {
507     char temp[11] = "";  /* 32-bit hex + CRLF + nul */
508     int ret;
509     char crlf[] = "\r\n";
510     HTTPContext *s = h->priv_data;
511
512     if (!s->chunked_post) {
513         /* non-chunked data is sent without any special encoding */
514         return ffurl_write(s->hd, buf, size);
515     }
516
517     /* silently ignore zero-size data since chunk encoding that would
518      * signal EOF */
519     if (size > 0) {
520         /* upload data using chunked encoding */
521         snprintf(temp, sizeof(temp), "%x\r\n", size);
522
523         if ((ret = ffurl_write(s->hd, temp, strlen(temp))) < 0 ||
524             (ret = ffurl_write(s->hd, buf, size)) < 0 ||
525             (ret = ffurl_write(s->hd, crlf, sizeof(crlf) - 1)) < 0)
526             return ret;
527     }
528     return size;
529 }
530
531 static int http_shutdown(URLContext *h, int flags)
532 {
533     int ret = 0;
534     char footer[] = "0\r\n\r\n";
535     HTTPContext *s = h->priv_data;
536
537     /* signal end of chunked encoding if used */
538     if ((flags & AVIO_FLAG_WRITE) && s->chunked_post) {
539         ret = ffurl_write(s->hd, footer, sizeof(footer) - 1);
540         ret = ret > 0 ? 0 : ret;
541         s->end_chunked_post = 1;
542     }
543
544     return ret;
545 }
546
547 static int http_close(URLContext *h)
548 {
549     int ret = 0;
550     HTTPContext *s = h->priv_data;
551
552     if (!s->end_chunked_post) {
553         /* Close the write direction by sending the end of chunked encoding. */
554         ret = http_shutdown(h, h->flags);
555     }
556
557     if (s->hd)
558         ffurl_close(s->hd);
559     return ret;
560 }
561
562 static int64_t http_seek(URLContext *h, int64_t off, int whence)
563 {
564     HTTPContext *s = h->priv_data;
565     URLContext *old_hd = s->hd;
566     int64_t old_off = s->off;
567     uint8_t old_buf[BUFFER_SIZE];
568     int old_buf_size;
569
570     if (whence == AVSEEK_SIZE)
571         return s->filesize;
572     else if ((s->filesize == -1 && whence == SEEK_END) || h->is_streamed)
573         return -1;
574
575     /* we save the old context in case the seek fails */
576     old_buf_size = s->buf_end - s->buf_ptr;
577     memcpy(old_buf, s->buf_ptr, old_buf_size);
578     s->hd = NULL;
579     if (whence == SEEK_CUR)
580         off += s->off;
581     else if (whence == SEEK_END)
582         off += s->filesize;
583     s->off = off;
584
585     /* if it fails, continue on old connection */
586     if (http_open_cnx(h) < 0) {
587         memcpy(s->buffer, old_buf, old_buf_size);
588         s->buf_ptr = s->buffer;
589         s->buf_end = s->buffer + old_buf_size;
590         s->hd = old_hd;
591         s->off = old_off;
592         return -1;
593     }
594     ffurl_close(old_hd);
595     return off;
596 }
597
598 static int
599 http_get_file_handle(URLContext *h)
600 {
601     HTTPContext *s = h->priv_data;
602     return ffurl_get_file_handle(s->hd);
603 }
604
605 #if CONFIG_HTTP_PROTOCOL
606 URLProtocol ff_http_protocol = {
607     .name                = "http",
608     .url_open            = http_open,
609     .url_read            = http_read,
610     .url_write           = http_write,
611     .url_seek            = http_seek,
612     .url_close           = http_close,
613     .url_get_file_handle = http_get_file_handle,
614     .url_shutdown        = http_shutdown,
615     .priv_data_size      = sizeof(HTTPContext),
616     .priv_data_class     = &http_context_class,
617     .flags               = URL_PROTOCOL_FLAG_NETWORK,
618 };
619 #endif
620 #if CONFIG_HTTPS_PROTOCOL
621 URLProtocol ff_https_protocol = {
622     .name                = "https",
623     .url_open            = http_open,
624     .url_read            = http_read,
625     .url_write           = http_write,
626     .url_seek            = http_seek,
627     .url_close           = http_close,
628     .url_get_file_handle = http_get_file_handle,
629     .priv_data_size      = sizeof(HTTPContext),
630     .priv_data_class     = &https_context_class,
631     .flags               = URL_PROTOCOL_FLAG_NETWORK,
632 };
633 #endif
634
635 #if CONFIG_HTTPPROXY_PROTOCOL
636 static int http_proxy_close(URLContext *h)
637 {
638     HTTPContext *s = h->priv_data;
639     if (s->hd)
640         ffurl_close(s->hd);
641     return 0;
642 }
643
644 static int http_proxy_open(URLContext *h, const char *uri, int flags)
645 {
646     HTTPContext *s = h->priv_data;
647     char hostname[1024], hoststr[1024];
648     char auth[1024], pathbuf[1024], *path;
649     char lower_url[100];
650     int port, ret = 0, attempts = 0;
651     HTTPAuthType cur_auth_type;
652     char *authstr;
653     int new_loc;
654
655     h->is_streamed = 1;
656
657     av_url_split(NULL, 0, auth, sizeof(auth), hostname, sizeof(hostname), &port,
658                  pathbuf, sizeof(pathbuf), uri);
659     ff_url_join(hoststr, sizeof(hoststr), NULL, NULL, hostname, port, NULL);
660     path = pathbuf;
661     if (*path == '/')
662         path++;
663
664     ff_url_join(lower_url, sizeof(lower_url), "tcp", NULL, hostname, port,
665                 NULL);
666 redo:
667     ret = ffurl_open(&s->hd, lower_url, AVIO_FLAG_READ_WRITE,
668                      &h->interrupt_callback, NULL);
669     if (ret < 0)
670         return ret;
671
672     authstr = ff_http_auth_create_response(&s->proxy_auth_state, auth,
673                                            path, "CONNECT");
674     snprintf(s->buffer, sizeof(s->buffer),
675              "CONNECT %s HTTP/1.1\r\n"
676              "Host: %s\r\n"
677              "Connection: close\r\n"
678              "%s%s"
679              "\r\n",
680              path,
681              hoststr,
682              authstr ? "Proxy-" : "", authstr ? authstr : "");
683     av_freep(&authstr);
684
685     if ((ret = ffurl_write(s->hd, s->buffer, strlen(s->buffer))) < 0)
686         goto fail;
687
688     s->buf_ptr = s->buffer;
689     s->buf_end = s->buffer;
690     s->line_count = 0;
691     s->filesize = -1;
692     cur_auth_type = s->proxy_auth_state.auth_type;
693
694     /* Note: This uses buffering, potentially reading more than the
695      * HTTP header. If tunneling a protocol where the server starts
696      * the conversation, we might buffer part of that here, too.
697      * Reading that requires using the proper ffurl_read() function
698      * on this URLContext, not using the fd directly (as the tls
699      * protocol does). This shouldn't be an issue for tls though,
700      * since the client starts the conversation there, so there
701      * is no extra data that we might buffer up here.
702      */
703     ret = http_read_header(h, &new_loc);
704     if (ret < 0)
705         goto fail;
706
707     attempts++;
708     if (s->http_code == 407 &&
709         (cur_auth_type == HTTP_AUTH_NONE || s->proxy_auth_state.stale) &&
710         s->proxy_auth_state.auth_type != HTTP_AUTH_NONE && attempts < 2) {
711         ffurl_close(s->hd);
712         s->hd = NULL;
713         goto redo;
714     }
715
716     if (s->http_code < 400)
717         return 0;
718     ret = AVERROR(EIO);
719
720 fail:
721     http_proxy_close(h);
722     return ret;
723 }
724
725 static int http_proxy_write(URLContext *h, const uint8_t *buf, int size)
726 {
727     HTTPContext *s = h->priv_data;
728     return ffurl_write(s->hd, buf, size);
729 }
730
731 URLProtocol ff_httpproxy_protocol = {
732     .name                = "httpproxy",
733     .url_open            = http_proxy_open,
734     .url_read            = http_buf_read,
735     .url_write           = http_proxy_write,
736     .url_close           = http_proxy_close,
737     .url_get_file_handle = http_get_file_handle,
738     .priv_data_size      = sizeof(HTTPContext),
739     .flags               = URL_PROTOCOL_FLAG_NETWORK,
740 };
741 #endif