1
2#include <libkern/crypto/crypto_internal.h>
3#include <libkern/crypto/sha1.h>
4#include <kern/debug.h>
5#include <corecrypto/ccdigest.h>
6
7
8static uint64_t getCount(SHA1_CTX *ctx)
9{
10	return ctx->c.b64[0];
11}
12
13static void setCount(SHA1_CTX *ctx, uint64_t count)
14{
15	ctx->c.b64[0]=count;
16}
17
18/* Copy a ccdigest ctx into a legacy SHA1 context */
19static void DiToSHA1(const struct ccdigest_info *di, struct ccdigest_ctx *di_ctx, SHA1_CTX *sha1_ctx)
20{
21	setCount(sha1_ctx, ccdigest_nbits(di, di_ctx)/8+ccdigest_num(di, di_ctx));
22	memcpy(sha1_ctx->m.b8, ccdigest_data(di, di_ctx), di->block_size);
23	memcpy(sha1_ctx->h.b8, ccdigest_state_ccn(di, di_ctx), di->state_size);
24}
25
26/* Copy a legacy SHA1 context into a ccdigest ctx  */
27static void SHA1ToDi(const struct ccdigest_info *di, SHA1_CTX *sha1_ctx, struct ccdigest_ctx *di_ctx)
28{
29	uint64_t count = getCount(sha1_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), sha1_ctx->m.b8, di->block_size);
34	memcpy(ccdigest_state_ccn(di, di_ctx), sha1_ctx->h.b8, di->state_size);
35}
36
37void SHA1Init(SHA1_CTX *ctx)
38{
39	const struct ccdigest_info *di=g_crypto_funcs->ccsha1_di;
40	ccdigest_di_decl(di, di_ctx);
41
42	g_crypto_funcs->ccdigest_init_fn(di, di_ctx);
43
44	DiToSHA1(di, di_ctx, ctx);
45}
46
47void SHA1Update(SHA1_CTX *ctx, const void *data, size_t len)
48{
49	const struct ccdigest_info *di=g_crypto_funcs->ccsha1_di;
50	ccdigest_di_decl(di, di_ctx);
51
52	SHA1ToDi(di, ctx, di_ctx);
53	g_crypto_funcs->ccdigest_update_fn(di, di_ctx, len, data);
54	DiToSHA1(di, di_ctx, ctx);
55}
56
57void SHA1Final(void *digest, SHA1_CTX *ctx)
58{
59	const struct ccdigest_info *di=g_crypto_funcs->ccsha1_di;
60	ccdigest_di_decl(di, di_ctx);
61
62	SHA1ToDi(di, ctx, di_ctx);
63	ccdigest_final(di, di_ctx, digest);
64}
65
66#ifdef XNU_KERNEL_PRIVATE
67void SHA1UpdateUsePhysicalAddress(SHA1_CTX *ctx, const void *data, size_t len)
68{
69	//TODO: What the hell ?
70	SHA1Update(ctx, data, len);
71}
72#endif
73
74/* This is not publicised in header, but exported in libkern.exports */
75void SHA1Final_r(SHA1_CTX *context, void *digest);
76void SHA1Final_r(SHA1_CTX *context, void *digest)
77{
78	SHA1Final(digest, context);
79}
80
81
82/*
83 * This function is called by the SHA1 hardware kext during its init.
84 * This will register the function to call to perform SHA1 using hardware.
85 */
86#include <sys/types.h>
87#include <libkern/OSAtomic.h>
88#include <sys/systm.h>
89
90typedef kern_return_t (*InKernelPerformSHA1Func)(void *ref, const void *data, size_t dataLen, u_int32_t *inHash, u_int32_t options, u_int32_t *outHash, Boolean usePhysicalAddress);
91void sha1_hardware_hook(Boolean option, InKernelPerformSHA1Func func, void *ref);
92static void *SHA1Ref;
93static InKernelPerformSHA1Func performSHA1WithinKernelOnly;
94
95void sha1_hardware_hook(Boolean option, InKernelPerformSHA1Func func, void *ref)
96{
97	if(option) {
98		// Establish the hook. The hardware is ready.
99		OSCompareAndSwapPtr((void*)NULL, (void*)ref, (void * volatile*)&SHA1Ref);
100
101		if(!OSCompareAndSwapPtr((void *)NULL, (void *)func, (void * volatile *)&performSHA1WithinKernelOnly)) {
102			panic("sha1_hardware_hook: Called twice.. Should never happen\n");
103		}
104	}
105	else {
106		// The hardware is going away. Tear down the hook.
107		performSHA1WithinKernelOnly = NULL;
108		SHA1Ref = NULL;
109	}
110}
111