| /* |
| * MD5 encoder (for codec/format testing) |
| * Copyright (c) 2009 Reimar Döffinger, based on crcenc (c) 2002 Fabrice Bellard |
| * |
| * 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 "libavutil/md5.h" |
| #include "avformat.h" |
| |
| #define PRIVSIZE 512 |
| |
| static void md5_finish(struct AVFormatContext *s, char *buf) |
| { |
| uint8_t md5[16]; |
| int i, offset = strlen(buf); |
| av_md5_final(s->priv_data, md5); |
| for (i = 0; i < sizeof(md5); i++) { |
| snprintf(buf + offset, 3, "%02"PRIx8, md5[i]); |
| offset += 2; |
| } |
| buf[offset] = '\n'; |
| buf[offset+1] = 0; |
| |
| avio_write(s->pb, buf, strlen(buf)); |
| avio_flush(s->pb); |
| } |
| |
| #if CONFIG_MD5_MUXER |
| static int write_header(struct AVFormatContext *s) |
| { |
| if (PRIVSIZE < av_md5_size) { |
| av_log(s, AV_LOG_ERROR, "Insuffient size for md5 context\n"); |
| return -1; |
| } |
| av_md5_init(s->priv_data); |
| return 0; |
| } |
| |
| static int write_packet(struct AVFormatContext *s, AVPacket *pkt) |
| { |
| av_md5_update(s->priv_data, pkt->data, pkt->size); |
| return 0; |
| } |
| |
| static int write_trailer(struct AVFormatContext *s) |
| { |
| char buf[64] = "MD5="; |
| |
| md5_finish(s, buf); |
| return 0; |
| } |
| |
| AVOutputFormat ff_md5_muxer = { |
| .name = "md5", |
| .long_name = NULL_IF_CONFIG_SMALL("MD5 testing format"), |
| .priv_data_size = PRIVSIZE, |
| .audio_codec = CODEC_ID_PCM_S16LE, |
| .video_codec = CODEC_ID_RAWVIDEO, |
| .write_header = write_header, |
| .write_packet = write_packet, |
| .write_trailer = write_trailer, |
| .flags = AVFMT_NOTIMESTAMPS, |
| }; |
| #endif |
| |
| #if CONFIG_FRAMEMD5_MUXER |
| static int framemd5_write_packet(struct AVFormatContext *s, AVPacket *pkt) |
| { |
| char buf[256]; |
| if (PRIVSIZE < av_md5_size) { |
| av_log(s, AV_LOG_ERROR, "Insuffient size for md5 context\n"); |
| return -1; |
| } |
| av_md5_init(s->priv_data); |
| av_md5_update(s->priv_data, pkt->data, pkt->size); |
| |
| snprintf(buf, sizeof(buf) - 64, "%d, %"PRId64", %d, ", pkt->stream_index, pkt->dts, pkt->size); |
| md5_finish(s, buf); |
| return 0; |
| } |
| |
| AVOutputFormat ff_framemd5_muxer = { |
| .name = "framemd5", |
| .long_name = NULL_IF_CONFIG_SMALL("Per-frame MD5 testing format"), |
| .priv_data_size = PRIVSIZE, |
| .audio_codec = CODEC_ID_PCM_S16LE, |
| .video_codec = CODEC_ID_RAWVIDEO, |
| .write_packet = framemd5_write_packet, |
| .flags = AVFMT_VARIABLE_FPS, |
| }; |
| #endif |