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