1#if defined(__arm__) 2#include "localcrypto_md5.c" /* We can't use corecrypto.kext yet, so use local version for now. */ 3#else 4 5#include <libkern/crypto/crypto_internal.h> 6#include <libkern/crypto/md5.h> 7#include <kern/debug.h> 8#include <corecrypto/ccdigest.h> 9 10static uint64_t getCount(MD5_CTX *ctx) 11{ 12 return ( (((uint64_t)ctx->count[0])<<32) | (ctx->count[1]) ); 13} 14 15static void setCount(MD5_CTX *ctx, uint64_t count) 16{ 17 ctx->count[0]=(uint32_t)(count>>32); 18 ctx->count[1]=(uint32_t)count; 19} 20 21/* Copy a ccdigest ctx into a legacy MD5 context */ 22static void DiToMD5(const struct ccdigest_info *di, struct ccdigest_ctx *di_ctx, MD5_CTX *md5_ctx) 23{ 24 setCount(md5_ctx, ccdigest_nbits(di, di_ctx)/8+ccdigest_num(di, di_ctx)); 25 memcpy(md5_ctx->buffer, ccdigest_data(di, di_ctx), di->block_size); 26 memcpy(md5_ctx->state, ccdigest_state_ccn(di, di_ctx), di->state_size); 27} 28 29/* Copy a legacy MD5 context into a ccdigest ctx */ 30static void MD5ToDi(const struct ccdigest_info *di, MD5_CTX *md5_ctx, struct ccdigest_ctx *di_ctx) 31{ 32 uint64_t count = getCount(md5_ctx); 33 34 ccdigest_num(di, di_ctx)=count%di->block_size; 35 ccdigest_nbits(di, di_ctx)=(count-ccdigest_num(di, di_ctx))*8; 36 memcpy(ccdigest_data(di, di_ctx), md5_ctx->buffer, di->block_size); 37 memcpy(ccdigest_state_ccn(di, di_ctx), md5_ctx->state, di->state_size); 38} 39 40void MD5Init(MD5_CTX *ctx) 41{ 42 const struct ccdigest_info *di=g_crypto_funcs->ccmd5_di; 43 ccdigest_di_decl(di, di_ctx); 44 45 g_crypto_funcs->ccdigest_init_fn(di, di_ctx); 46 47 DiToMD5(di, di_ctx, ctx); 48} 49 50void MD5Update(MD5_CTX *ctx, const void *data, unsigned int len) 51{ 52 const struct ccdigest_info *di=g_crypto_funcs->ccmd5_di; 53 ccdigest_di_decl(di, di_ctx); 54 55 MD5ToDi(di, ctx, di_ctx); 56 g_crypto_funcs->ccdigest_update_fn(di, di_ctx, len, data); 57 DiToMD5(di, di_ctx, ctx); 58} 59 60void MD5Final(unsigned char digest[MD5_DIGEST_LENGTH], MD5_CTX *ctx) 61{ 62 const struct ccdigest_info *di=g_crypto_funcs->ccmd5_di; 63 ccdigest_di_decl(di, di_ctx); 64 65 MD5ToDi(di, ctx, di_ctx); 66 ccdigest_final(di, di_ctx, digest); 67} 68 69#endif /* !__arm__ */