ghash.c 6.1 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28
/**
 * GHASH routines supporting VMX instructions on the Power 8
 *
 * Copyright (C) 2015 International Business Machines Inc.
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; version 2 only.
 *
 * This program 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 General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 *
 * Author: Marcelo Henrique Cerri <mhcerri@br.ibm.com>
 */

#include <linux/types.h>
#include <linux/err.h>
#include <linux/crypto.h>
#include <linux/delay.h>
#include <linux/hardirq.h>
#include <asm/switch_to.h>
#include <crypto/aes.h>
29
#include <crypto/ghash.h>
30 31 32 33 34 35 36 37 38
#include <crypto/scatterwalk.h>
#include <crypto/internal/hash.h>
#include <crypto/b128ops.h>

#define IN_INTERRUPT in_interrupt()

void gcm_init_p8(u128 htable[16], const u64 Xi[2]);
void gcm_gmult_p8(u64 Xi[2], const u128 htable[16]);
void gcm_ghash_p8(u64 Xi[2], const u128 htable[16],
39
		  const u8 *in, size_t len);
40 41

struct p8_ghash_ctx {
42 43
	u128 htable[16];
	struct crypto_shash *fallback;
44 45 46
};

struct p8_ghash_desc_ctx {
47 48 49 50
	u64 shash[2];
	u8 buffer[GHASH_DIGEST_SIZE];
	int bytes;
	struct shash_desc fallback_desc;
51 52 53 54
};

static int p8_ghash_init_tfm(struct crypto_tfm *tfm)
{
55
	const char *alg = "ghash-generic";
56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71
	struct crypto_shash *fallback;
	struct crypto_shash *shash_tfm = __crypto_shash_cast(tfm);
	struct p8_ghash_ctx *ctx = crypto_tfm_ctx(tfm);

	fallback = crypto_alloc_shash(alg, 0, CRYPTO_ALG_NEED_FALLBACK);
	if (IS_ERR(fallback)) {
		printk(KERN_ERR
		       "Failed to allocate transformation for '%s': %ld\n",
		       alg, PTR_ERR(fallback));
		return PTR_ERR(fallback);
	}

	crypto_shash_set_flags(fallback,
			       crypto_shash_get_flags((struct crypto_shash
						       *) tfm));

72 73 74 75 76 77 78 79 80 81 82
	/* Check if the descsize defined in the algorithm is still enough. */
	if (shash_tfm->descsize < sizeof(struct p8_ghash_desc_ctx)
	    + crypto_shash_descsize(fallback)) {
		printk(KERN_ERR
		       "Desc size of the fallback implementation (%s) does not match the expected value: %lu vs %u\n",
		       alg,
		       shash_tfm->descsize - sizeof(struct p8_ghash_desc_ctx),
		       crypto_shash_descsize(fallback));
		return -EINVAL;
	}
	ctx->fallback = fallback;
83 84

	return 0;
85 86 87 88
}

static void p8_ghash_exit_tfm(struct crypto_tfm *tfm)
{
89
	struct p8_ghash_ctx *ctx = crypto_tfm_ctx(tfm);
90

91 92 93 94
	if (ctx->fallback) {
		crypto_free_shash(ctx->fallback);
		ctx->fallback = NULL;
	}
95 96 97 98
}

static int p8_ghash_init(struct shash_desc *desc)
{
99 100 101 102 103 104 105 106
	struct p8_ghash_ctx *ctx = crypto_tfm_ctx(crypto_shash_tfm(desc->tfm));
	struct p8_ghash_desc_ctx *dctx = shash_desc_ctx(desc);

	dctx->bytes = 0;
	memset(dctx->shash, 0, GHASH_DIGEST_SIZE);
	dctx->fallback_desc.tfm = ctx->fallback;
	dctx->fallback_desc.flags = desc->flags;
	return crypto_shash_init(&dctx->fallback_desc);
107 108 109
}

static int p8_ghash_setkey(struct crypto_shash *tfm, const u8 *key,
110
			   unsigned int keylen)
111
{
112
	struct p8_ghash_ctx *ctx = crypto_tfm_ctx(crypto_shash_tfm(tfm));
113

114
	if (keylen != GHASH_BLOCK_SIZE)
115
		return -EINVAL;
116

117
	preempt_disable();
118
	pagefault_disable();
119
	enable_kernel_vsx();
120
	gcm_init_p8(ctx->htable, (const u64 *) key);
121
	disable_kernel_vsx();
122
	pagefault_enable();
123
	preempt_enable();
124
	return crypto_shash_setkey(ctx->fallback, key, keylen);
125 126 127
}

static int p8_ghash_update(struct shash_desc *desc,
128
			   const u8 *src, unsigned int srclen)
129
{
130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146
	unsigned int len;
	struct p8_ghash_ctx *ctx = crypto_tfm_ctx(crypto_shash_tfm(desc->tfm));
	struct p8_ghash_desc_ctx *dctx = shash_desc_ctx(desc);

	if (IN_INTERRUPT) {
		return crypto_shash_update(&dctx->fallback_desc, src,
					   srclen);
	} else {
		if (dctx->bytes) {
			if (dctx->bytes + srclen < GHASH_DIGEST_SIZE) {
				memcpy(dctx->buffer + dctx->bytes, src,
				       srclen);
				dctx->bytes += srclen;
				return 0;
			}
			memcpy(dctx->buffer + dctx->bytes, src,
			       GHASH_DIGEST_SIZE - dctx->bytes);
147
			preempt_disable();
148
			pagefault_disable();
149
			enable_kernel_vsx();
150 151
			gcm_ghash_p8(dctx->shash, ctx->htable,
				     dctx->buffer, GHASH_DIGEST_SIZE);
152
			disable_kernel_vsx();
153
			pagefault_enable();
154
			preempt_enable();
155 156 157 158 159 160
			src += GHASH_DIGEST_SIZE - dctx->bytes;
			srclen -= GHASH_DIGEST_SIZE - dctx->bytes;
			dctx->bytes = 0;
		}
		len = srclen & ~(GHASH_DIGEST_SIZE - 1);
		if (len) {
161
			preempt_disable();
162
			pagefault_disable();
163
			enable_kernel_vsx();
164
			gcm_ghash_p8(dctx->shash, ctx->htable, src, len);
165
			disable_kernel_vsx();
166
			pagefault_enable();
167
			preempt_enable();
168 169 170 171 172 173 174 175 176
			src += len;
			srclen -= len;
		}
		if (srclen) {
			memcpy(dctx->buffer, src, srclen);
			dctx->bytes = srclen;
		}
		return 0;
	}
177 178 179 180
}

static int p8_ghash_final(struct shash_desc *desc, u8 *out)
{
181 182 183 184 185 186 187 188 189 190
	int i;
	struct p8_ghash_ctx *ctx = crypto_tfm_ctx(crypto_shash_tfm(desc->tfm));
	struct p8_ghash_desc_ctx *dctx = shash_desc_ctx(desc);

	if (IN_INTERRUPT) {
		return crypto_shash_final(&dctx->fallback_desc, out);
	} else {
		if (dctx->bytes) {
			for (i = dctx->bytes; i < GHASH_DIGEST_SIZE; i++)
				dctx->buffer[i] = 0;
191
			preempt_disable();
192
			pagefault_disable();
193
			enable_kernel_vsx();
194 195
			gcm_ghash_p8(dctx->shash, ctx->htable,
				     dctx->buffer, GHASH_DIGEST_SIZE);
196
			disable_kernel_vsx();
197
			pagefault_enable();
198
			preempt_enable();
199 200 201 202 203
			dctx->bytes = 0;
		}
		memcpy(out, dctx->shash, GHASH_DIGEST_SIZE);
		return 0;
	}
204 205 206
}

struct shash_alg p8_ghash_alg = {
207 208 209 210 211
	.digestsize = GHASH_DIGEST_SIZE,
	.init = p8_ghash_init,
	.update = p8_ghash_update,
	.final = p8_ghash_final,
	.setkey = p8_ghash_setkey,
212 213
	.descsize = sizeof(struct p8_ghash_desc_ctx)
		+ sizeof(struct ghash_desc_ctx),
214 215 216 217
	.base = {
		 .cra_name = "ghash",
		 .cra_driver_name = "p8_ghash",
		 .cra_priority = 1000,
218
		 .cra_flags = CRYPTO_ALG_NEED_FALLBACK,
219 220 221 222 223 224
		 .cra_blocksize = GHASH_BLOCK_SIZE,
		 .cra_ctxsize = sizeof(struct p8_ghash_ctx),
		 .cra_module = THIS_MODULE,
		 .cra_init = p8_ghash_init_tfm,
		 .cra_exit = p8_ghash_exit_tfm,
	},
225
};