mirror of
https://github.com/FFmpeg/FFmpeg.git
synced 2024-12-07 11:13:41 +02:00
fab8156b2f
Since all URLContexts have the same AVOptions, such AVOptions will be applied on the outermost context only and removed from the dict, while they probably make sense on all contexts. This makes sure that rw_timeout gets propagated to the innermost URLContext (to make sure it gets passed to the tcp protocol, when opening a http connection for instance). Alternatively, such matching options would be kept in the dict and only removed after the ffurl_connect call. Signed-off-by: Martin Storsjö <martin@martin.st>
97 lines
2.5 KiB
C
97 lines
2.5 KiB
C
/*
|
|
* Copyright (c) 2010 Mans Rullgard
|
|
*
|
|
* This file is part of Libav.
|
|
*
|
|
* Libav 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.
|
|
*
|
|
* Libav 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 Libav; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
|
*/
|
|
|
|
#include <stdio.h>
|
|
#include "libavutil/avstring.h"
|
|
#include "libavutil/md5.h"
|
|
#include "libavutil/mem.h"
|
|
#include "libavutil/error.h"
|
|
#include "avformat.h"
|
|
#include "avio.h"
|
|
#include "url.h"
|
|
|
|
struct MD5Context {
|
|
struct AVMD5 *md5;
|
|
};
|
|
|
|
static int md5_open(URLContext *h, const char *filename, int flags)
|
|
{
|
|
struct MD5Context *c = h->priv_data;
|
|
|
|
if (!(flags & AVIO_FLAG_WRITE))
|
|
return AVERROR(EINVAL);
|
|
|
|
c->md5 = av_md5_alloc();
|
|
if (!c->md5)
|
|
return AVERROR(ENOMEM);
|
|
av_md5_init(c->md5);
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int md5_write(URLContext *h, const unsigned char *buf, int size)
|
|
{
|
|
struct MD5Context *c = h->priv_data;
|
|
av_md5_update(c->md5, buf, size);
|
|
return size;
|
|
}
|
|
|
|
static int md5_close(URLContext *h)
|
|
{
|
|
struct MD5Context *c = h->priv_data;
|
|
const char *filename = h->filename;
|
|
uint8_t md5[16], buf[64];
|
|
URLContext *out;
|
|
int i, err = 0;
|
|
|
|
av_md5_final(c->md5, md5);
|
|
for (i = 0; i < sizeof(md5); i++)
|
|
snprintf(buf + i*2, 3, "%02x", md5[i]);
|
|
buf[i*2] = '\n';
|
|
|
|
av_strstart(filename, "md5:", &filename);
|
|
|
|
if (*filename) {
|
|
err = ffurl_open(&out, filename, AVIO_FLAG_WRITE,
|
|
&h->interrupt_callback, NULL,
|
|
h->protocols, h);
|
|
if (err)
|
|
return err;
|
|
err = ffurl_write(out, buf, i*2+1);
|
|
ffurl_close(out);
|
|
} else {
|
|
if (fwrite(buf, 1, i*2+1, stdout) < i*2+1)
|
|
err = AVERROR(errno);
|
|
}
|
|
|
|
av_freep(&c->md5);
|
|
|
|
return err;
|
|
}
|
|
|
|
|
|
const URLProtocol ff_md5_protocol = {
|
|
.name = "md5",
|
|
.url_open = md5_open,
|
|
.url_write = md5_write,
|
|
.url_close = md5_close,
|
|
.priv_data_size = sizeof(struct MD5Context),
|
|
};
|