diff options
Diffstat (limited to 'ffmpeg1/libavformat/md5proto.c')
| -rw-r--r-- | ffmpeg1/libavformat/md5proto.c | 95 |
1 files changed, 95 insertions, 0 deletions
diff --git a/ffmpeg1/libavformat/md5proto.c b/ffmpeg1/libavformat/md5proto.c new file mode 100644 index 0000000..6af0a6e --- /dev/null +++ b/ffmpeg1/libavformat/md5proto.c @@ -0,0 +1,95 @@ +/* + * Copyright (c) 2010 Mans Rullgard + * + * This file is part of FFmpeg. + * + * FFmpeg 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. + * + * FFmpeg 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 FFmpeg; 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); + 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; +} + + +URLProtocol ff_md5_protocol = { + .name = "md5", + .url_open = md5_open, + .url_write = md5_write, + .url_close = md5_close, + .priv_data_size = sizeof(struct MD5Context), +}; |
