00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00028 #define _XOPEN_SOURCE 600
00029 #include "libavutil/avstring.h"
00030 #include "avformat.h"
00031 #include "internal.h"
00032 #include <unistd.h>
00033
00034
00035
00036
00037
00038
00039
00040
00041
00042
00043
00044
00045
00046 struct segment {
00047 int duration;
00048 char url[MAX_URL_SIZE];
00049 };
00050
00051 struct variant {
00052 int bandwidth;
00053 char url[MAX_URL_SIZE];
00054 };
00055
00056 typedef struct AppleHTTPContext {
00057 char playlisturl[MAX_URL_SIZE];
00058 int target_duration;
00059 int start_seq_no;
00060 int finished;
00061 int n_segments;
00062 struct segment **segments;
00063 int n_variants;
00064 struct variant **variants;
00065 int cur_seq_no;
00066 URLContext *seg_hd;
00067 int64_t last_load_time;
00068 } AppleHTTPContext;
00069
00070 static int read_chomp_line(AVIOContext *s, char *buf, int maxlen)
00071 {
00072 int len = ff_get_line(s, buf, maxlen);
00073 while (len > 0 && isspace(buf[len - 1]))
00074 buf[--len] = '\0';
00075 return len;
00076 }
00077
00078 static void make_absolute_url(char *buf, int size, const char *base,
00079 const char *rel)
00080 {
00081 char *sep;
00082
00083 if (base && strstr(base, "://") && rel[0] == '/') {
00084 if (base != buf)
00085 av_strlcpy(buf, base, size);
00086 sep = strstr(buf, "://");
00087 if (sep) {
00088 sep += 3;
00089 sep = strchr(sep, '/');
00090 if (sep)
00091 *sep = '\0';
00092 }
00093 av_strlcat(buf, rel, size);
00094 return;
00095 }
00096
00097 if (!base || strstr(rel, "://") || rel[0] == '/') {
00098 av_strlcpy(buf, rel, size);
00099 return;
00100 }
00101 if (base != buf)
00102 av_strlcpy(buf, base, size);
00103
00104 sep = strrchr(buf, '/');
00105 if (sep)
00106 sep[1] = '\0';
00107 else
00108 buf[0] = '\0';
00109 while (av_strstart(rel, "../", NULL) && sep) {
00110
00111 sep[0] = '\0';
00112 sep = strrchr(buf, '/');
00113
00114 if (!strcmp(sep ? &sep[1] : buf, "..")) {
00115
00116 av_strlcat(buf, "/", size);
00117 break;
00118 }
00119
00120 if (sep)
00121 sep[1] = '\0';
00122 else
00123 buf[0] = '\0';
00124 rel += 3;
00125 }
00126 av_strlcat(buf, rel, size);
00127 }
00128
00129 static void free_segment_list(AppleHTTPContext *s)
00130 {
00131 int i;
00132 for (i = 0; i < s->n_segments; i++)
00133 av_free(s->segments[i]);
00134 av_freep(&s->segments);
00135 s->n_segments = 0;
00136 }
00137
00138 static void free_variant_list(AppleHTTPContext *s)
00139 {
00140 int i;
00141 for (i = 0; i < s->n_variants; i++)
00142 av_free(s->variants[i]);
00143 av_freep(&s->variants);
00144 s->n_variants = 0;
00145 }
00146
00147 struct variant_info {
00148 char bandwidth[20];
00149 };
00150
00151 static void handle_variant_args(struct variant_info *info, const char *key,
00152 int key_len, char **dest, int *dest_len)
00153 {
00154 if (!strncmp(key, "BANDWIDTH=", key_len)) {
00155 *dest = info->bandwidth;
00156 *dest_len = sizeof(info->bandwidth);
00157 }
00158 }
00159
00160 static int parse_playlist(URLContext *h, const char *url)
00161 {
00162 AppleHTTPContext *s = h->priv_data;
00163 AVIOContext *in;
00164 int ret = 0, duration = 0, is_segment = 0, is_variant = 0, bandwidth = 0;
00165 char line[1024];
00166 const char *ptr;
00167
00168 if ((ret = avio_open(&in, url, URL_RDONLY)) < 0)
00169 return ret;
00170
00171 read_chomp_line(in, line, sizeof(line));
00172 if (strcmp(line, "#EXTM3U"))
00173 return AVERROR_INVALIDDATA;
00174
00175 free_segment_list(s);
00176 s->finished = 0;
00177 while (!url_feof(in)) {
00178 read_chomp_line(in, line, sizeof(line));
00179 if (av_strstart(line, "#EXT-X-STREAM-INF:", &ptr)) {
00180 struct variant_info info = {{0}};
00181 is_variant = 1;
00182 ff_parse_key_value(ptr, (ff_parse_key_val_cb) handle_variant_args,
00183 &info);
00184 bandwidth = atoi(info.bandwidth);
00185 } else if (av_strstart(line, "#EXT-X-TARGETDURATION:", &ptr)) {
00186 s->target_duration = atoi(ptr);
00187 } else if (av_strstart(line, "#EXT-X-MEDIA-SEQUENCE:", &ptr)) {
00188 s->start_seq_no = atoi(ptr);
00189 } else if (av_strstart(line, "#EXT-X-ENDLIST", &ptr)) {
00190 s->finished = 1;
00191 } else if (av_strstart(line, "#EXTINF:", &ptr)) {
00192 is_segment = 1;
00193 duration = atoi(ptr);
00194 } else if (av_strstart(line, "#", NULL)) {
00195 continue;
00196 } else if (line[0]) {
00197 if (is_segment) {
00198 struct segment *seg = av_malloc(sizeof(struct segment));
00199 if (!seg) {
00200 ret = AVERROR(ENOMEM);
00201 goto fail;
00202 }
00203 seg->duration = duration;
00204 make_absolute_url(seg->url, sizeof(seg->url), url, line);
00205 dynarray_add(&s->segments, &s->n_segments, seg);
00206 is_segment = 0;
00207 } else if (is_variant) {
00208 struct variant *var = av_malloc(sizeof(struct variant));
00209 if (!var) {
00210 ret = AVERROR(ENOMEM);
00211 goto fail;
00212 }
00213 var->bandwidth = bandwidth;
00214 make_absolute_url(var->url, sizeof(var->url), url, line);
00215 dynarray_add(&s->variants, &s->n_variants, var);
00216 is_variant = 0;
00217 }
00218 }
00219 }
00220 s->last_load_time = av_gettime();
00221
00222 fail:
00223 avio_close(in);
00224 return ret;
00225 }
00226
00227 static int applehttp_open(URLContext *h, const char *uri, int flags)
00228 {
00229 AppleHTTPContext *s;
00230 int ret, i;
00231 const char *nested_url;
00232
00233 if (flags & (URL_WRONLY | URL_RDWR))
00234 return AVERROR_NOTSUPP;
00235
00236 s = av_mallocz(sizeof(AppleHTTPContext));
00237 if (!s)
00238 return AVERROR(ENOMEM);
00239 h->priv_data = s;
00240 h->is_streamed = 1;
00241
00242 if (av_strstart(uri, "applehttp+", &nested_url)) {
00243 av_strlcpy(s->playlisturl, nested_url, sizeof(s->playlisturl));
00244 } else if (av_strstart(uri, "applehttp://", &nested_url)) {
00245 av_strlcpy(s->playlisturl, "http://", sizeof(s->playlisturl));
00246 av_strlcat(s->playlisturl, nested_url, sizeof(s->playlisturl));
00247 } else {
00248 av_log(NULL, AV_LOG_ERROR, "Unsupported url %s\n", uri);
00249 ret = AVERROR(EINVAL);
00250 goto fail;
00251 }
00252
00253 if ((ret = parse_playlist(h, s->playlisturl)) < 0)
00254 goto fail;
00255
00256 if (s->n_segments == 0 && s->n_variants > 0) {
00257 int max_bandwidth = 0, maxvar = -1;
00258 for (i = 0; i < s->n_variants; i++) {
00259 if (s->variants[i]->bandwidth > max_bandwidth || i == 0) {
00260 max_bandwidth = s->variants[i]->bandwidth;
00261 maxvar = i;
00262 }
00263 }
00264 av_strlcpy(s->playlisturl, s->variants[maxvar]->url,
00265 sizeof(s->playlisturl));
00266 if ((ret = parse_playlist(h, s->playlisturl)) < 0)
00267 goto fail;
00268 }
00269
00270 if (s->n_segments == 0) {
00271 av_log(NULL, AV_LOG_WARNING, "Empty playlist\n");
00272 ret = AVERROR(EIO);
00273 goto fail;
00274 }
00275 s->cur_seq_no = s->start_seq_no;
00276 if (!s->finished && s->n_segments >= 3)
00277 s->cur_seq_no = s->start_seq_no + s->n_segments - 3;
00278
00279 return 0;
00280
00281 fail:
00282 av_free(s);
00283 return ret;
00284 }
00285
00286 static int applehttp_read(URLContext *h, uint8_t *buf, int size)
00287 {
00288 AppleHTTPContext *s = h->priv_data;
00289 const char *url;
00290 int ret;
00291
00292 start:
00293 if (s->seg_hd) {
00294 ret = url_read(s->seg_hd, buf, size);
00295 if (ret > 0)
00296 return ret;
00297 }
00298 if (s->seg_hd) {
00299 url_close(s->seg_hd);
00300 s->seg_hd = NULL;
00301 s->cur_seq_no++;
00302 }
00303 retry:
00304 if (!s->finished) {
00305 int64_t now = av_gettime();
00306 if (now - s->last_load_time >= s->target_duration*1000000)
00307 if ((ret = parse_playlist(h, s->playlisturl)) < 0)
00308 return ret;
00309 }
00310 if (s->cur_seq_no < s->start_seq_no) {
00311 av_log(NULL, AV_LOG_WARNING,
00312 "skipping %d segments ahead, expired from playlist\n",
00313 s->start_seq_no - s->cur_seq_no);
00314 s->cur_seq_no = s->start_seq_no;
00315 }
00316 if (s->cur_seq_no - s->start_seq_no >= s->n_segments) {
00317 if (s->finished)
00318 return AVERROR_EOF;
00319 while (av_gettime() - s->last_load_time < s->target_duration*1000000) {
00320 if (url_interrupt_cb())
00321 return AVERROR(EINTR);
00322 usleep(100*1000);
00323 }
00324 goto retry;
00325 }
00326 url = s->segments[s->cur_seq_no - s->start_seq_no]->url,
00327 av_log(NULL, AV_LOG_DEBUG, "opening %s\n", url);
00328 ret = url_open(&s->seg_hd, url, URL_RDONLY);
00329 if (ret < 0) {
00330 if (url_interrupt_cb())
00331 return AVERROR(EINTR);
00332 av_log(NULL, AV_LOG_WARNING, "Unable to open %s\n", url);
00333 s->cur_seq_no++;
00334 goto retry;
00335 }
00336 goto start;
00337 }
00338
00339 static int applehttp_close(URLContext *h)
00340 {
00341 AppleHTTPContext *s = h->priv_data;
00342
00343 free_segment_list(s);
00344 free_variant_list(s);
00345 url_close(s->seg_hd);
00346 av_free(s);
00347 return 0;
00348 }
00349
00350 URLProtocol ff_applehttp_protocol = {
00351 "applehttp",
00352 applehttp_open,
00353 applehttp_read,
00354 NULL,
00355 NULL,
00356 applehttp_close,
00357 .flags = URL_PROTOCOL_FLAG_NESTED_SCHEME,
00358 };