1/* 2 * Copyright (C) 2013 Reimar Döffinger <Reimar.Doeffinger@gmx.de> 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 <stddef.h> 22#include <stdint.h> 23#include "hash.h" 24 25#include "adler32.h" 26#include "crc.h" 27#include "md5.h" 28#include "murmur3.h" 29#include "ripemd.h" 30#include "sha.h" 31#include "sha512.h" 32 33#include "avstring.h" 34#include "base64.h" 35#include "error.h" 36#include "intreadwrite.h" 37#include "mem.h" 38 39enum hashtype { 40 MD5, 41 MURMUR3, 42 RIPEMD128, 43 RIPEMD160, 44 RIPEMD256, 45 RIPEMD320, 46 SHA160, 47 SHA224, 48 SHA256, 49 SHA512_224, 50 SHA512_256, 51 SHA384, 52 SHA512, 53 CRC32, 54 ADLER32, 55 NUM_HASHES 56}; 57 58typedef struct AVHashContext { 59 void *ctx; 60 enum hashtype type; 61 const AVCRC *crctab; 62 uint32_t crc; 63} AVHashContext; 64 65static const struct { 66 const char *name; 67 int size; 68} hashdesc[] = { 69 [MD5] = {"MD5", 16}, 70 [MURMUR3] = {"murmur3", 16}, 71 [RIPEMD128] = {"RIPEMD128", 16}, 72 [RIPEMD160] = {"RIPEMD160", 20}, 73 [RIPEMD256] = {"RIPEMD256", 32}, 74 [RIPEMD320] = {"RIPEMD320", 40}, 75 [SHA160] = {"SHA160", 20}, 76 [SHA224] = {"SHA224", 28}, 77 [SHA256] = {"SHA256", 32}, 78 [SHA512_224] = {"SHA512/224", 28}, 79 [SHA512_256] = {"SHA512/256", 32}, 80 [SHA384] = {"SHA384", 48}, 81 [SHA512] = {"SHA512", 64}, 82 [CRC32] = {"CRC32", 4}, 83 [ADLER32] = {"adler32", 4}, 84}; 85 86const char *av_hash_names(int i) 87{ 88 if (i < 0 || i >= NUM_HASHES) return NULL; 89 return hashdesc[i].name; 90} 91 92const char *av_hash_get_name(const AVHashContext *ctx) 93{ 94 return hashdesc[ctx->type].name; 95} 96 97int av_hash_get_size(const AVHashContext *ctx) 98{ 99 return hashdesc[ctx->type].size; 100} 101 102int av_hash_alloc(AVHashContext **ctx, const char *name) 103{ 104 AVHashContext *res; 105 int i; 106 *ctx = NULL; 107 for (i = 0; i < NUM_HASHES; i++) 108 if (av_strcasecmp(name, hashdesc[i].name) == 0) 109 break; 110 if (i >= NUM_HASHES) return AVERROR(EINVAL); 111 res = av_mallocz(sizeof(*res)); 112 if (!res) return AVERROR(ENOMEM); 113 res->type = i; 114 switch (i) { 115 case MD5: res->ctx = av_md5_alloc(); break; 116 case MURMUR3: res->ctx = av_murmur3_alloc(); break; 117 case RIPEMD128: 118 case RIPEMD160: 119 case RIPEMD256: 120 case RIPEMD320: res->ctx = av_ripemd_alloc(); break; 121 case SHA160: 122 case SHA224: 123 case SHA256: res->ctx = av_sha_alloc(); break; 124 case SHA512_224: 125 case SHA512_256: 126 case SHA384: 127 case SHA512: res->ctx = av_sha512_alloc(); break; 128 case CRC32: res->crctab = av_crc_get_table(AV_CRC_32_IEEE_LE); break; 129 case ADLER32: break; 130 } 131 if (i != ADLER32 && i != CRC32 && !res->ctx) { 132 av_free(res); 133 return AVERROR(ENOMEM); 134 } 135 *ctx = res; 136 return 0; 137} 138 139void av_hash_init(AVHashContext *ctx) 140{ 141 switch (ctx->type) { 142 case MD5: av_md5_init(ctx->ctx); break; 143 case MURMUR3: av_murmur3_init(ctx->ctx); break; 144 case RIPEMD128: av_ripemd_init(ctx->ctx, 128); break; 145 case RIPEMD160: av_ripemd_init(ctx->ctx, 160); break; 146 case RIPEMD256: av_ripemd_init(ctx->ctx, 256); break; 147 case RIPEMD320: av_ripemd_init(ctx->ctx, 320); break; 148 case SHA160: av_sha_init(ctx->ctx, 160); break; 149 case SHA224: av_sha_init(ctx->ctx, 224); break; 150 case SHA256: av_sha_init(ctx->ctx, 256); break; 151 case SHA512_224: av_sha512_init(ctx->ctx, 224); break; 152 case SHA512_256: av_sha512_init(ctx->ctx, 256); break; 153 case SHA384: av_sha512_init(ctx->ctx, 384); break; 154 case SHA512: av_sha512_init(ctx->ctx, 512); break; 155 case CRC32: ctx->crc = UINT32_MAX; break; 156 case ADLER32: ctx->crc = 1; break; 157 } 158} 159 160void av_hash_update(AVHashContext *ctx, const uint8_t *src, size_t len) 161{ 162 switch (ctx->type) { 163 case MD5: av_md5_update(ctx->ctx, src, len); break; 164 case MURMUR3: av_murmur3_update(ctx->ctx, src, len); break; 165 case RIPEMD128: 166 case RIPEMD160: 167 case RIPEMD256: 168 case RIPEMD320: av_ripemd_update(ctx->ctx, src, len); break; 169 case SHA160: 170 case SHA224: 171 case SHA256: av_sha_update(ctx->ctx, src, len); break; 172 case SHA512_224: 173 case SHA512_256: 174 case SHA384: 175 case SHA512: av_sha512_update(ctx->ctx, src, len); break; 176 case CRC32: ctx->crc = av_crc(ctx->crctab, ctx->crc, src, len); break; 177 case ADLER32: ctx->crc = av_adler32_update(ctx->crc, src, len); break; 178 } 179} 180 181void av_hash_final(AVHashContext *ctx, uint8_t *dst) 182{ 183 switch (ctx->type) { 184 case MD5: av_md5_final(ctx->ctx, dst); break; 185 case MURMUR3: av_murmur3_final(ctx->ctx, dst); break; 186 case RIPEMD128: 187 case RIPEMD160: 188 case RIPEMD256: 189 case RIPEMD320: av_ripemd_final(ctx->ctx, dst); break; 190 case SHA160: 191 case SHA224: 192 case SHA256: av_sha_final(ctx->ctx, dst); break; 193 case SHA512_224: 194 case SHA512_256: 195 case SHA384: 196 case SHA512: av_sha512_final(ctx->ctx, dst); break; 197 case CRC32: AV_WB32(dst, ctx->crc ^ UINT32_MAX); break; 198 case ADLER32: AV_WB32(dst, ctx->crc); break; 199 } 200} 201 202void av_hash_final_bin(struct AVHashContext *ctx, uint8_t *dst, int size) 203{ 204 uint8_t buf[AV_HASH_MAX_SIZE]; 205 unsigned rsize = av_hash_get_size(ctx); 206 207 av_hash_final(ctx, buf); 208 memcpy(dst, buf, FFMIN(size, rsize)); 209 if (size > rsize) 210 memset(dst + rsize, 0, size - rsize); 211} 212 213void av_hash_final_hex(struct AVHashContext *ctx, uint8_t *dst, int size) 214{ 215 uint8_t buf[AV_HASH_MAX_SIZE]; 216 unsigned rsize = av_hash_get_size(ctx), i; 217 218 av_hash_final(ctx, buf); 219 for (i = 0; i < FFMIN(rsize, size / 2); i++) 220 snprintf(dst + i * 2, size - i * 2, "%02x", buf[i]); 221} 222 223void av_hash_final_b64(struct AVHashContext *ctx, uint8_t *dst, int size) 224{ 225 uint8_t buf[AV_HASH_MAX_SIZE], b64[AV_BASE64_SIZE(AV_HASH_MAX_SIZE)]; 226 unsigned rsize = av_hash_get_size(ctx), osize; 227 228 av_hash_final(ctx, buf); 229 av_base64_encode(b64, sizeof(b64), buf, rsize); 230 osize = AV_BASE64_SIZE(rsize); 231 memcpy(dst, b64, FFMIN(osize, size)); 232 if (size < osize) 233 dst[size - 1] = 0; 234} 235 236void av_hash_freep(AVHashContext **ctx) 237{ 238 if (*ctx) 239 av_freep(&(*ctx)->ctx); 240 av_freep(ctx); 241} 242