git.sesse.net Git - ffmpeg/blob - libavformat/crc.c

   1 /*
   2  * CRC decoder (for codec/format testing)
   3  * Copyright (c) 2002 Fabrice Bellard.
   4  *
   5  * This library is free software; you can redistribute it and/or
   6  * modify it under the terms of the GNU Lesser General Public
   7  * License as published by the Free Software Foundation; either
   8  * version 2 of the License, or (at your option) any later version.
   9  *
  10  * This library is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * Lesser General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU Lesser General Public
  16  * License along with this library; if not, write to the Free Software
  17  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  18  */
  19 #include "avformat.h"
  20
  21 /* adler32.c -- compute the Adler-32 checksum of a data stream
  22  * Copyright (C) 1995 Mark Adler
  23  * For conditions of distribution and use, see copyright notice in zlib.h
  24  */
  25
  26 #define BASE 65521L /* largest prime smaller than 65536 */
  27 #define NMAX 5552
  28 /* NMAX is the largest n such that 255n(n+1)/2 + (n+1)(BASE-1) <= 2^32-1 */
  29
  30 #define DO1(buf)  {s1 += *buf++; s2 += s1;}
  31 #define DO2(buf)  DO1(buf); DO1(buf);
  32 #define DO4(buf)  DO2(buf); DO2(buf);
  33 #define DO8(buf)  DO4(buf); DO4(buf);
  34 #define DO16(buf) DO8(buf); DO8(buf);
  35
  36 unsigned long update_adler32(unsigned long adler, const uint8_t *buf, unsigned int len)
  37 {
  38     unsigned long s1 = adler & 0xffff;
  39     unsigned long s2 = (adler >> 16) & 0xffff;
  40     int k;
  41
  42     if (buf == NULL) return 1L;
  43
  44     while (len > 0) {
  45         k = len < NMAX ? len : NMAX;
  46         len -= k;
  47         while (k >= 16) {
  48             DO16(buf);
  49             k -= 16;
  50         }
  51         if (k != 0) do {
  52             DO1(buf);
  53         } while (--k);
  54         s1 %= BASE;
  55         s2 %= BASE;
  56     }
  57     return (s2 << 16) | s1;
  58 }
  59
  60 typedef struct CRCState {
  61     uint32_t crcval;
  62 } CRCState;
  63
  64 static int crc_write_header(struct AVFormatContext *s)
  65 {
  66     CRCState *crc = s->priv_data;
  67
  68     /* init CRC */
  69     crc->crcval = update_adler32(0, NULL, 0);
  70
  71     return 0;
  72 }
  73
  74 static int crc_write_packet(struct AVFormatContext *s, AVPacket *pkt)
  75 {
  76     CRCState *crc = s->priv_data;
  77     crc->crcval = update_adler32(crc->crcval, pkt->data, pkt->size);
  78     return 0;
  79 }
  80
  81 static int crc_write_trailer(struct AVFormatContext *s)
  82 {
  83     CRCState *crc = s->priv_data;
  84     char buf[64];
  85
  86     snprintf(buf, sizeof(buf), "CRC=%08x\n", crc->crcval);
  87     put_buffer(&s->pb, buf, strlen(buf));
  88     put_flush_packet(&s->pb);
  89     return 0;
  90 }
  91
  92 static AVOutputFormat crc_format = {
  93     "crc",
  94     "crc testing format",
  95     NULL,
  96     "",
  97     sizeof(CRCState),
  98     CODEC_ID_PCM_S16LE,
  99     CODEC_ID_RAWVIDEO,
 100     crc_write_header,
 101     crc_write_packet,
 102     crc_write_trailer,
 103 };
 104
 105 int crc_init(void)
 106 {
 107     av_register_output_format(&crc_format);
 108     return 0;
 109 }