Something went wrong. Try again.
🎥 Command line media player
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444/* * This file is part of mpv. * * mpv is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2.1 of the License, or (at your option) any later version. * * mpv is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with mpv. If not, see <http://www.gnu.org/licenses/>. */
#include <libavformat/avformat.h>#include <libavformat/avio.h>#include <libavutil/opt.h>
#include "options/path.h"#include "common/common.h"#include "common/msg.h"#include "common/tags.h"#include "common/av_common.h"#include "stream.h"#include "options/m_config.h"#include "options/m_option.h"
#include "cookies.h"
#include "misc/bstr.h"#include "mpv_talloc.h"
#define OPT_BASE_STRUCT struct stream_lavf_paramsstruct stream_lavf_params { char **avopts; int cookies_enabled; char *cookies_file; char *useragent; char *referrer; char **http_header_fields; int tls_verify; char *tls_ca_file; char *tls_cert_file; char *tls_key_file; double timeout;};
const struct m_sub_options stream_lavf_conf = { .opts = (const m_option_t[]) { OPT_KEYVALUELIST("stream-lavf-o", avopts, 0), OPT_STRINGLIST("http-header-fields", http_header_fields, 0), OPT_STRING("user-agent", useragent, 0), OPT_STRING("referrer", referrer, 0), OPT_FLAG("cookies", cookies_enabled, 0), OPT_STRING("cookies-file", cookies_file, M_OPT_FILE), OPT_FLAG("tls-verify", tls_verify, 0), OPT_STRING("tls-ca-file", tls_ca_file, M_OPT_FILE), OPT_STRING("tls-cert-file", tls_cert_file, M_OPT_FILE), OPT_STRING("tls-key-file", tls_key_file, M_OPT_FILE), OPT_DOUBLE("network-timeout", timeout, M_OPT_MIN, .min = 0), {0} }, .size = sizeof(struct stream_lavf_params), .defaults = &(const struct stream_lavf_params){ .useragent = (char *)mpv_version, },};
static const char *const http_like[];
static int open_f(stream_t *stream);static struct mp_tags *read_icy(stream_t *stream);
static int fill_buffer(stream_t *s, char *buffer, int max_len){ AVIOContext *avio = s->priv; if (!avio) return -1;#if LIBAVFORMAT_VERSION_MICRO >= 100 && LIBAVFORMAT_VERSION_INT >= AV_VERSION_INT(57, 81, 100) int r = avio_read_partial(avio, buffer, max_len);#else int r = avio_read(avio, buffer, max_len);#endif return (r <= 0) ? -1 : r;}
static int write_buffer(stream_t *s, char *buffer, int len){ AVIOContext *avio = s->priv; if (!avio) return -1; avio_write(avio, buffer, len); avio_flush(avio); if (avio->error) return -1; return len;}
static int seek(stream_t *s, int64_t newpos){ AVIOContext *avio = s->priv; if (!avio) return -1; if (avio_seek(avio, newpos, SEEK_SET) < 0) { return 0; } return 1;}
static void close_f(stream_t *stream){ AVIOContext *avio = stream->priv; /* NOTE: As of 2011 write streams must be manually flushed before close. * Currently write_buffer() always flushes them after writing. * avio_close() could return an error, but we have no way to return that * with the current stream API. */ if (avio) avio_close(avio);}
static int control(stream_t *s, int cmd, void *arg){ AVIOContext *avio = s->priv; if (!avio && cmd != STREAM_CTRL_RECONNECT) return -1; int64_t size; switch(cmd) { case STREAM_CTRL_GET_SIZE: size = avio_size(avio); if (size >= 0) { *(int64_t *)arg = size; return 1; } break; case STREAM_CTRL_AVSEEK: { struct stream_avseek *c = arg; int64_t r = avio_seek_time(avio, c->stream_index, c->timestamp, c->flags); if (r >= 0) { stream_drop_buffers(s); return 1; } break; } case STREAM_CTRL_HAS_AVSEEK: { // Starting at some point, read_seek is always available, and runtime // behavior decides whether it exists or not. FFmpeg's API doesn't // return anything helpful to determine seekability upfront, so here's // a hardcoded whitelist. Not our fault. // In addition we also have to jump through ridiculous hoops just to // get the fucking protocol name. const char *proto = NULL; if (avio->av_class && avio->av_class->child_next) { // This usually yields the URLContext (why does it even exist?), // which holds the name of the actual protocol implementation. void *child = avio->av_class->child_next(avio, NULL); AVClass *cl = *(AVClass **)child; if (cl && cl->item_name) proto = cl->item_name(child); } static const char *const has_read_seek[] = { "rtmp", "rtmpt", "rtmpe", "rtmpte", "rtmps", "rtmpts", "mmsh", 0}; for (int n = 0; has_read_seek[n]; n++) { if (avio->read_seek && proto && strcmp(proto, has_read_seek[n]) == 0) return 1; } break; } case STREAM_CTRL_GET_METADATA: { *(struct mp_tags **)arg = read_icy(s); if (!*(struct mp_tags **)arg) break; return 1; } case STREAM_CTRL_RECONNECT: { if (avio && avio->write_flag) break; // don't bother with this // avio doesn't seem to support this - emulate it by reopening close_f(s); s->priv = NULL; stream_drop_buffers(s); s->pos = 0; return open_f(s); } } return STREAM_UNSUPPORTED;}
static int interrupt_cb(void *ctx){ struct stream *stream = ctx; return mp_cancel_test(stream->cancel);}
static const char * const prefix[] = { "lavf://", "ffmpeg://" };
void mp_setup_av_network_options(AVDictionary **dict, struct mpv_global *global, struct mp_log *log){ void *temp = talloc_new(NULL); struct stream_lavf_params *opts = mp_get_config_group(temp, global, &stream_lavf_conf);
// HTTP specific options (other protocols ignore them) if (opts->useragent) av_dict_set(dict, "user_agent", opts->useragent, 0); if (opts->cookies_enabled) { char *file = opts->cookies_file; if (file && file[0]) file = mp_get_user_path(temp, global, file); char *cookies = cookies_lavf(temp, log, file); if (cookies && cookies[0]) av_dict_set(dict, "cookies", cookies, 0); } av_dict_set(dict, "tls_verify", opts->tls_verify ? "1" : "0", 0); if (opts->tls_ca_file) av_dict_set(dict, "ca_file", opts->tls_ca_file, 0); if (opts->tls_cert_file) av_dict_set(dict, "cert_file", opts->tls_cert_file, 0); if (opts->tls_key_file) av_dict_set(dict, "key_file", opts->tls_key_file, 0); char *cust_headers = talloc_strdup(temp, ""); if (opts->referrer) { cust_headers = talloc_asprintf_append(cust_headers, "Referer: %s\r\n", opts->referrer); } if (opts->http_header_fields) { for (int n = 0; opts->http_header_fields[n]; n++) { cust_headers = talloc_asprintf_append(cust_headers, "%s\r\n", opts->http_header_fields[n]); } } if (strlen(cust_headers)) av_dict_set(dict, "headers", cust_headers, 0); av_dict_set(dict, "icy", "1", 0); // So far, every known protocol uses microseconds for this if (opts->timeout > 0) { char buf[80]; snprintf(buf, sizeof(buf), "%lld", (long long)(opts->timeout * 1e6)); av_dict_set(dict, "timeout", buf, 0); }
mp_set_avdict(dict, opts->avopts);
talloc_free(temp);}
// Escape http URLs with unescaped, invalid characters in them.// libavformat's http protocol does not do this, and a patch to add this// in a 100% safe case (spaces only) was rejected.static char *normalize_url(void *ta_parent, const char *filename){ bstr proto = mp_split_proto(bstr0(filename), NULL); for (int n = 0; http_like[n]; n++) { if (bstr_equals0(proto, http_like[n])) // Escape everything but reserved characters. // Also don't double-scape, so include '%'. return mp_url_escape(ta_parent, filename, ":/?#[]@!$&'()*+,;=%"); } return (char *)filename;}
static int open_f(stream_t *stream){ AVIOContext *avio = NULL; int res = STREAM_ERROR; AVDictionary *dict = NULL; void *temp = talloc_new(NULL);
stream->seek = NULL; stream->seekable = false;
int flags = stream->mode == STREAM_WRITE ? AVIO_FLAG_WRITE : AVIO_FLAG_READ;
const char *filename = stream->url; if (!filename) { MP_ERR(stream, "No URL\n"); goto out; } for (int i = 0; i < sizeof(prefix) / sizeof(prefix[0]); i++) if (!strncmp(filename, prefix[i], strlen(prefix[i]))) filename += strlen(prefix[i]); if (!strncmp(filename, "rtsp:", 5)) { /* This is handled as a special demuxer, without a separate * stream layer. demux_lavf will do all the real work. Note * that libavformat doesn't even provide a protocol entry for * this (the rtsp demuxer's probe function checks for a "rtsp:" * filename prefix), so it has to be handled specially here. */ stream->demuxer = "lavf"; stream->lavf_type = "rtsp"; talloc_free(temp); return STREAM_OK; }
// Replace "mms://" with "mmsh://", so that most mms:// URLs just work. bstr b_filename = bstr0(filename); if (bstr_eatstart0(&b_filename, "mms://") || bstr_eatstart0(&b_filename, "mmshttp://")) { filename = talloc_asprintf(temp, "mmsh://%.*s", BSTR_P(b_filename)); }
mp_setup_av_network_options(&dict, stream->global, stream->log);
AVIOInterruptCB cb = { .callback = interrupt_cb, .opaque = stream, };
filename = normalize_url(stream, filename);
if (strncmp(filename, "rtmp", 4) == 0) { stream->demuxer = "lavf"; stream->lavf_type = "flv"; // Setting timeout enables listen mode - force it to disabled. av_dict_set(&dict, "timeout", "0", 0); }
int err = avio_open2(&avio, filename, flags, &cb, &dict); if (err < 0) { if (err == AVERROR_PROTOCOL_NOT_FOUND) MP_ERR(stream, "Protocol not found. Make sure" " ffmpeg/Libav is compiled with networking support.\n"); goto out; }
mp_avdict_print_unset(stream->log, MSGL_V, dict);
if (avio->av_class) { uint8_t *mt = NULL; if (av_opt_get(avio, "mime_type", AV_OPT_SEARCH_CHILDREN, &mt) >= 0) { stream->mime_type = talloc_strdup(stream, mt); av_free(mt); } }
stream->priv = avio; stream->seekable = avio->seekable & AVIO_SEEKABLE_NORMAL; stream->seek = stream->seekable ? seek : NULL; stream->fill_buffer = fill_buffer; stream->write_buffer = write_buffer; stream->control = control; stream->close = close_f; // enable cache (should be avoided for files, but no way to detect this) stream->streaming = true; res = STREAM_OK;
out: av_dict_free(&dict); talloc_free(temp); return res;}
static struct mp_tags *read_icy(stream_t *s){ AVIOContext *avio = s->priv;
if (!avio->av_class) return NULL;
uint8_t *icy_header = NULL; if (av_opt_get(avio, "icy_metadata_headers", AV_OPT_SEARCH_CHILDREN, &icy_header) < 0) icy_header = NULL;
uint8_t *icy_packet; if (av_opt_get(avio, "icy_metadata_packet", AV_OPT_SEARCH_CHILDREN, &icy_packet) < 0) icy_packet = NULL;
// Send a metadata update only 1. on start, and 2. on a new metadata packet. // To detect new packages, set the icy_metadata_packet to "-" once we've // read it (a bit hacky, but works).
struct mp_tags *res = NULL; if ((!icy_header || !icy_header[0]) && (!icy_packet || !icy_packet[0])) goto done;
bstr packet = bstr0(icy_packet); if (bstr_equals0(packet, "-")) goto done;
res = talloc_zero(NULL, struct mp_tags);
bstr header = bstr0(icy_header); while (header.len) { bstr line = bstr_strip_linebreaks(bstr_getline(header, &header)); bstr name, val; if (bstr_split_tok(line, ": ", &name, &val)) mp_tags_set_bstr(res, name, val); }
bstr head = bstr0("StreamTitle='"); int i = bstr_find(packet, head); if (i >= 0) { packet = bstr_cut(packet, i + head.len); int end = bstr_find(packet, bstr0("\';")); packet = bstr_splice(packet, 0, end); mp_tags_set_bstr(res, bstr0("icy-title"), packet); }
av_opt_set(avio, "icy_metadata_packet", "-", AV_OPT_SEARCH_CHILDREN);
done: av_free(icy_header); av_free(icy_packet); return res;}
static const char *const http_like[] = {"http", "https", "mmsh", "mmshttp", "httproxy", NULL};
const stream_info_t stream_info_ffmpeg = { .name = "ffmpeg", .open = open_f, .protocols = (const char *const[]){ "rtmp", "rtsp", "http", "https", "mms", "mmst", "mmsh", "mmshttp", "rtp", "httpproxy", "hls", "rtmpe", "rtmps", "rtmpt", "rtmpte", "rtmpts", "srtp", "data", NULL }, .can_write = true, .is_safe = true, .is_network = true,};
// Unlike above, this is not marked as safe, and can contain protocols which// may do insecure things. (Such as "ffmpeg", which can access the "lavfi"// pseudo-demuxer, which in turn gives access to filters that can access the// local filesystem.)const stream_info_t stream_info_ffmpeg_unsafe = { .name = "ffmpeg", .open = open_f, .protocols = (const char *const[]){ "lavf", "ffmpeg", "udp", "ftp", "tcp", "tls", "unix", "sftp", "md5", "concat", NULL }, .can_write = true,};