#include "libavutil/avstring.h"
#include "avformat.h"
#include <unistd.h>
+#include <strings.h>
#include "network.h"
#include "os_support.h"
unsigned char buffer[BUFFER_SIZE], *buf_ptr, *buf_end;
int line_count;
int http_code;
+ int64_t chunksize; /**< Used if "Transfer-Encoding: chunked" otherwise -1. */
int64_t off, filesize;
char location[URL_SIZE];
} HTTPContext;
}
h->priv_data = s;
s->filesize = -1;
+ s->chunksize = -1;
s->off = 0;
av_strlcpy(s->location, uri, URL_SIZE);
s->filesize = atoll(slash+1);
}
h->is_streamed = 0; /* we _can_ in fact seek */
+ } else if (!strcmp (tag, "Transfer-Encoding") && !strncasecmp(p, "chunked", 7)) {
+ s->filesize = -1;
+ s->chunksize = 0;
}
}
return 1;
HTTPContext *s = h->priv_data;
int len;
+ if (s->chunksize >= 0) {
+ if (!s->chunksize) {
+ char line[32];
+
+ for(;;) {
+ do {
+ if (http_get_line(s, line, sizeof(line)) < 0)
+ return AVERROR(EIO);
+ } while (!*line); /* skip CR LF from last chunk */
+
+ s->chunksize = strtoll(line, NULL, 16);
+
+ dprintf(NULL, "Chunked encoding data size: %"PRId64"'\n", s->chunksize);
+
+ if (!s->chunksize)
+ return 0;
+ break;
+ }
+ }
+ size = FFMIN(size, s->chunksize);
+ }
/* read bytes from input buffer first */
len = s->buf_end - s->buf_ptr;
if (len > 0) {
} else {
len = url_read(s->hd, buf, size);
}
- if (len > 0)
+ if (len > 0) {
s->off += len;
+ if (s->chunksize > 0)
+ s->chunksize -= len;
+ }
return len;
}