Libav 0.7.1
|
00001 /* 00002 * MD5 encoder (for codec/format testing) 00003 * Copyright (c) 2009 Reimar Döffinger, based on crcenc (c) 2002 Fabrice Bellard 00004 * 00005 * This file is part of Libav. 00006 * 00007 * Libav is free software; you can redistribute it and/or 00008 * modify it under the terms of the GNU Lesser General Public 00009 * License as published by the Free Software Foundation; either 00010 * version 2.1 of the License, or (at your option) any later version. 00011 * 00012 * Libav is distributed in the hope that it will be useful, 00013 * but WITHOUT ANY WARRANTY; without even the implied warranty of 00014 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 00015 * Lesser General Public License for more details. 00016 * 00017 * You should have received a copy of the GNU Lesser General Public 00018 * License along with Libav; if not, write to the Free Software 00019 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA 00020 */ 00021 00022 #include "libavutil/md5.h" 00023 #include "avformat.h" 00024 00025 #define PRIVSIZE 512 00026 00027 static void md5_finish(struct AVFormatContext *s, char *buf) 00028 { 00029 uint8_t md5[16]; 00030 int i, offset = strlen(buf); 00031 av_md5_final(s->priv_data, md5); 00032 for (i = 0; i < sizeof(md5); i++) { 00033 snprintf(buf + offset, 3, "%02"PRIx8, md5[i]); 00034 offset += 2; 00035 } 00036 buf[offset] = '\n'; 00037 buf[offset+1] = 0; 00038 00039 avio_write(s->pb, buf, strlen(buf)); 00040 avio_flush(s->pb); 00041 } 00042 00043 #if CONFIG_MD5_MUXER 00044 static int write_header(struct AVFormatContext *s) 00045 { 00046 if (PRIVSIZE < av_md5_size) { 00047 av_log(s, AV_LOG_ERROR, "Insuffient size for md5 context\n"); 00048 return -1; 00049 } 00050 av_md5_init(s->priv_data); 00051 return 0; 00052 } 00053 00054 static int write_packet(struct AVFormatContext *s, AVPacket *pkt) 00055 { 00056 av_md5_update(s->priv_data, pkt->data, pkt->size); 00057 return 0; 00058 } 00059 00060 static int write_trailer(struct AVFormatContext *s) 00061 { 00062 char buf[64] = "MD5="; 00063 00064 md5_finish(s, buf); 00065 return 0; 00066 } 00067 00068 AVOutputFormat ff_md5_muxer = { 00069 "md5", 00070 NULL_IF_CONFIG_SMALL("MD5 testing format"), 00071 NULL, 00072 "", 00073 PRIVSIZE, 00074 CODEC_ID_PCM_S16LE, 00075 CODEC_ID_RAWVIDEO, 00076 write_header, 00077 write_packet, 00078 write_trailer, 00079 }; 00080 #endif 00081 00082 #if CONFIG_FRAMEMD5_MUXER 00083 static int framemd5_write_packet(struct AVFormatContext *s, AVPacket *pkt) 00084 { 00085 char buf[256]; 00086 if (PRIVSIZE < av_md5_size) { 00087 av_log(s, AV_LOG_ERROR, "Insuffient size for md5 context\n"); 00088 return -1; 00089 } 00090 av_md5_init(s->priv_data); 00091 av_md5_update(s->priv_data, pkt->data, pkt->size); 00092 00093 snprintf(buf, sizeof(buf) - 64, "%d, %"PRId64", %d, ", pkt->stream_index, pkt->dts, pkt->size); 00094 md5_finish(s, buf); 00095 return 0; 00096 } 00097 00098 AVOutputFormat ff_framemd5_muxer = { 00099 "framemd5", 00100 NULL_IF_CONFIG_SMALL("Per-frame MD5 testing format"), 00101 NULL, 00102 "", 00103 PRIVSIZE, 00104 CODEC_ID_PCM_S16LE, 00105 CODEC_ID_RAWVIDEO, 00106 NULL, 00107 framemd5_write_packet, 00108 NULL, 00109 }; 00110 #endif