1/* 2 * Copyright (c) 2010 Mans Rullgard 3 * 4 * This file is part of FFmpeg. 5 * 6 * FFmpeg is free software; you can redistribute it and/or 7 * modify it under the terms of the GNU Lesser General Public 8 * License as published by the Free Software Foundation; either 9 * version 2.1 of the License, or (at your option) any later version. 10 * 11 * FFmpeg is distributed in the hope that it will be useful, 12 * but WITHOUT ANY WARRANTY; without even the implied warranty of 13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 14 * Lesser General Public License for more details. 15 * 16 * You should have received a copy of the GNU Lesser General Public 17 * License along with FFmpeg; if not, write to the Free Software 18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA 19 */ 20 21#include <stdio.h> 22#include "libavutil/avstring.h" 23#include "libavutil/md5.h" 24#include "libavutil/mem.h" 25#include "libavutil/error.h" 26#include "avformat.h" 27#include "avio.h" 28#include "internal.h" 29#include "url.h" 30 31struct MD5Context { 32 struct AVMD5 *md5; 33}; 34 35static int md5_open(URLContext *h, const char *filename, int flags) 36{ 37 struct MD5Context *c = h->priv_data; 38 39 if (!(flags & AVIO_FLAG_WRITE)) 40 return AVERROR(EINVAL); 41 42 c->md5 = av_md5_alloc(); 43 if (!c->md5) 44 return AVERROR(ENOMEM); 45 av_md5_init(c->md5); 46 47 return 0; 48} 49 50static int md5_write(URLContext *h, const unsigned char *buf, int size) 51{ 52 struct MD5Context *c = h->priv_data; 53 av_md5_update(c->md5, buf, size); 54 return size; 55} 56 57static int md5_close(URLContext *h) 58{ 59 struct MD5Context *c = h->priv_data; 60 const char *filename = h->filename; 61 uint8_t md5[16], buf[2 * sizeof(md5) + 1]; 62 URLContext *out; 63 int err = 0; 64 65 av_md5_final(c->md5, md5); 66 ff_data_to_hex(buf, md5, sizeof(md5), 1); 67 buf[2 * sizeof(md5)] = '\n'; 68 69 av_strstart(filename, "md5:", &filename); 70 71 if (*filename) { 72 err = ffurl_open_whitelist(&out, filename, AVIO_FLAG_WRITE, 73 &h->interrupt_callback, NULL, 74 h->protocol_whitelist, h->protocol_blacklist, h); 75 if (err) 76 return err; 77 err = ffurl_write(out, buf, sizeof(buf)); 78 ffurl_close(out); 79 } else { 80 if (fwrite(buf, 1, sizeof(buf), stdout) < sizeof(buf)) 81 err = AVERROR(errno); 82 } 83 84 av_freep(&c->md5); 85 86 return err; 87} 88 89 90const URLProtocol ff_md5_protocol = { 91 .name = "md5", 92 .url_open = md5_open, 93 .url_write = md5_write, 94 .url_close = md5_close, 95 .priv_data_size = sizeof(struct MD5Context), 96}; 97