/* * Copyright (C) 2013 Reimar Döffinger <Reimar.Doeffinger@gmx.de> * * 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 <stdint.h>#include "hash.h" #include "adler32.h"#include "crc.h"#include "md5.h"#include "murmur3.h"#include "ripemd.h"#include "sha.h"#include "sha512.h" #include "avstring.h"#include "base64.h"#include "error.h"#include "intreadwrite.h"#include "mem.h" enum hashtype {    MD5,    MURMUR3,    RIPEMD128,    RIPEMD160,    RIPEMD256,    RIPEMD320,    SHA160,    SHA224,    SHA256,    SHA512_224,    SHA512_256,    SHA384,    SHA512,    CRC32,    ADLER32,    NUM_HASHES}; typedef struct AVHashContext {    void *ctx;    enum hashtype type;    const AVCRC *crctab;    uint32_t crc;} AVHashContext; static const struct {    const char *name;    int size;} hashdesc[] = {    [MD5]     = {"MD5",     16},    [MURMUR3] = {"murmur3", 16},    [RIPEMD128] = {"RIPEMD128", 16},    [RIPEMD160] = {"RIPEMD160", 20},    [RIPEMD256] = {"RIPEMD256", 32},    [RIPEMD320] = {"RIPEMD320", 40},    [SHA160]  = {"SHA160",  20},    [SHA224]  = {"SHA224",  28},    [SHA256]  = {"SHA256",  32},    [SHA512_224]  = {"SHA512/224",  28},    [SHA512_256]  = {"SHA512/256",  32},    [SHA384]  = {"SHA384",  48},    [SHA512]  = {"SHA512",  64},    [CRC32]   = {"CRC32",    4},    [ADLER32] = {"adler32",  4},}; const char *av_hash_names(int i){    if (i < 0 || i >= NUM_HASHES) return NULL;    return hashdesc[i].name;} const char *av_hash_get_name(const AVHashContext *ctx){    return hashdesc[ctx->type].name;} int av_hash_get_size(const AVHashContext *ctx){    return hashdesc[ctx->type].size;} int av_hash_alloc(AVHashContext **ctx, const char *name){    AVHashContext *res;    int i;    *ctx = NULL;    for (i = 0; i < NUM_HASHES; i++)        if (av_strcasecmp(name, hashdesc[i].name) == 0)            break;    if (i >= NUM_HASHES) return AVERROR(EINVAL);    res = av_mallocz(sizeof(*res));    if (!res) return AVERROR(ENOMEM);    res->type = i;    switch (i) {    case MD5:     res->ctx = av_md5_alloc(); break;    case MURMUR3: res->ctx = av_murmur3_alloc(); break;    case RIPEMD128:    case RIPEMD160:    case RIPEMD256:    case RIPEMD320: res->ctx = av_ripemd_alloc(); break;    case SHA160:    case SHA224:    case SHA256:  res->ctx = av_sha_alloc(); break;    case SHA512_224:    case SHA512_256:    case SHA384:    case SHA512:  res->ctx = av_sha512_alloc(); break;    case CRC32:   res->crctab = av_crc_get_table(AV_CRC_32_IEEE_LE); break;    case ADLER32: break;    }    if (i != ADLER32 && i != CRC32 && !res->ctx) {        av_free(res);        return AVERROR(ENOMEM);    }    *ctx = res;    return 0;} void av_hash_init(AVHashContext *ctx){    switch (ctx->type) {    case MD5:     av_md5_init(ctx->ctx); break;    case MURMUR3: av_murmur3_init(ctx->ctx); break;    case RIPEMD128: av_ripemd_init(ctx->ctx, 128); break;    case RIPEMD160: av_ripemd_init(ctx->ctx, 160); break;    case RIPEMD256: av_ripemd_init(ctx->ctx, 256); break;    case RIPEMD320: av_ripemd_init(ctx->ctx, 320); break;    case SHA160:  av_sha_init(ctx->ctx, 160); break;    case SHA224:  av_sha_init(ctx->ctx, 224); break;    case SHA256:  av_sha_init(ctx->ctx, 256); break;    case SHA512_224:  av_sha512_init(ctx->ctx, 224); break;    case SHA512_256:  av_sha512_init(ctx->ctx, 256); break;    case SHA384:  av_sha512_init(ctx->ctx, 384); break;    case SHA512:  av_sha512_init(ctx->ctx, 512); break;    case CRC32:   ctx->crc = UINT32_MAX; break;    case ADLER32: ctx->crc = 1; break;    }} void av_hash_update(AVHashContext *ctx, const uint8_t *src, int len){    switch (ctx->type) {    case MD5:     av_md5_update(ctx->ctx, src, len); break;    case MURMUR3: av_murmur3_update(ctx->ctx, src, len); break;    case RIPEMD128:    case RIPEMD160:    case RIPEMD256:    case RIPEMD320: av_ripemd_update(ctx->ctx, src, len); break;    case SHA160:    case SHA224:    case SHA256:  av_sha_update(ctx->ctx, src, len); break;    case SHA512_224:    case SHA512_256:    case SHA384:    case SHA512:  av_sha512_update(ctx->ctx, src, len); break;    case CRC32:   ctx->crc = av_crc(ctx->crctab, ctx->crc, src, len); break;    case ADLER32: ctx->crc = av_adler32_update(ctx->crc, src, len); break;    }} void av_hash_final(AVHashContext *ctx, uint8_t *dst){    switch (ctx->type) {    case MD5:     av_md5_final(ctx->ctx, dst); break;    case MURMUR3: av_murmur3_final(ctx->ctx, dst); break;    case RIPEMD128:    case RIPEMD160:    case RIPEMD256:    case RIPEMD320: av_ripemd_final(ctx->ctx, dst); break;    case SHA160:    case SHA224:    case SHA256:  av_sha_final(ctx->ctx, dst); break;    case SHA512_224:    case SHA512_256:    case SHA384:    case SHA512:  av_sha512_final(ctx->ctx, dst); break;    case CRC32:   AV_WB32(dst, ctx->crc ^ UINT32_MAX); break;    case ADLER32: AV_WB32(dst, ctx->crc); break;    }} void av_hash_final_bin(struct AVHashContext *ctx, uint8_t *dst, int size){    uint8_t buf[AV_HASH_MAX_SIZE];    unsigned rsize = av_hash_get_size(ctx);     av_hash_final(ctx, buf);    memcpy(dst, buf, FFMIN(size, rsize));    if (size > rsize)        memset(dst + rsize, 0, size - rsize);} void av_hash_final_hex(struct AVHashContext *ctx, uint8_t *dst, int size){    uint8_t buf[AV_HASH_MAX_SIZE];    unsigned rsize = av_hash_get_size(ctx), i;     av_hash_final(ctx, buf);    for (i = 0; i < FFMIN(rsize, size / 2); i++)        snprintf(dst + i * 2, size - i * 2, "%02x", buf[i]);} void av_hash_final_b64(struct AVHashContext *ctx, uint8_t *dst, int size){    uint8_t buf[AV_HASH_MAX_SIZE], b64[AV_BASE64_SIZE(AV_HASH_MAX_SIZE)];    unsigned rsize = av_hash_get_size(ctx), osize;     av_hash_final(ctx, buf);    av_base64_encode(b64, sizeof(b64), buf, rsize);    osize = AV_BASE64_SIZE(rsize);    memcpy(dst, b64, FFMIN(osize, size));    if (size < osize)        dst[size - 1] = 0;} void av_hash_freep(AVHashContext **ctx){    if (*ctx)        av_freep(&(*ctx)->ctx);    av_freep(ctx);}