2 * Copyright (c) 2018 Conrad Meyer <cem@FreeBSD.org>
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
8 * 1. Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 * notice, this list of conditions and the following disclaimer in the
12 * documentation and/or other materials provided with the distribution.
14 * THIS SOFTWARE IS PROVIDED BY THE AUTHORS AND CONTRIBUTORS ``AS IS'' AND
15 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHORS OR CONTRIBUTORS BE LIABLE
18 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27 #include <sys/cdefs.h>
28 __FBSDID("$FreeBSD$");
30 #include <sys/param.h>
32 #include <sys/kernel.h>
35 #include <sys/malloc.h>
36 #include <sys/module.h>
37 #include <sys/mutex.h>
38 #include <sys/rwlock.h>
43 #include <opencrypto/cryptodev.h>
44 #include <cryptodev_if.h>
46 #include <machine/fpu.h>
48 struct blake2_session {
51 CTASSERT((size_t)BLAKE2B_KEYBYTES > (size_t)BLAKE2S_KEYBYTES);
59 static struct mtx_padalign *ctx_mtx;
60 static struct fpu_kern_ctx **ctx_fpu;
62 #define ACQUIRE_CTX(i, ctx) \
64 (i) = PCPU_GET(cpuid); \
65 mtx_lock(&ctx_mtx[(i)]); \
66 (ctx) = ctx_fpu[(i)]; \
68 #define RELEASE_CTX(i, ctx) \
70 mtx_unlock(&ctx_mtx[(i)]); \
75 static int blake2_cipher_setup(struct blake2_session *ses,
76 const struct crypto_session_params *csp);
77 static int blake2_cipher_process(struct blake2_session *ses,
80 MALLOC_DEFINE(M_BLAKE2, "blake2_data", "Blake2 Data");
83 blake2_identify(driver_t *drv, device_t parent)
86 /* NB: order 10 is so we get attached after h/w devices */
87 if (device_find_child(parent, "blaketwo", -1) == NULL &&
88 BUS_ADD_CHILD(parent, 10, "blaketwo", -1) == 0)
89 panic("blaketwo: could not attach");
93 blake2_probe(device_t dev)
95 device_set_desc(dev, "Blake2");
100 blake2_cleanctx(void)
104 /* XXX - no way to return driverid */
106 if (ctx_fpu[i] != NULL) {
107 mtx_destroy(&ctx_mtx[i]);
108 fpu_kern_free_ctx(ctx_fpu[i]);
112 free(ctx_mtx, M_BLAKE2);
114 free(ctx_fpu, M_BLAKE2);
119 blake2_attach(device_t dev)
121 struct blake2_softc *sc;
124 sc = device_get_softc(dev);
127 sc->cid = crypto_get_driverid(dev, sizeof(struct blake2_session),
128 CRYPTOCAP_F_SOFTWARE | CRYPTOCAP_F_SYNC |
129 CRYPTOCAP_F_ACCEL_SOFTWARE);
131 device_printf(dev, "Could not get crypto driver id.\n");
135 ctx_mtx = malloc(sizeof(*ctx_mtx) * (mp_maxid + 1), M_BLAKE2,
137 ctx_fpu = malloc(sizeof(*ctx_fpu) * (mp_maxid + 1), M_BLAKE2,
142 ctx_fpu[i] = fpu_kern_alloc_ctx_domain(
143 pcpu_find(i)->pc_domain, FPU_KERN_NORMAL);
145 ctx_fpu[i] = fpu_kern_alloc_ctx(FPU_KERN_NORMAL);
147 mtx_init(&ctx_mtx[i], "bl2fpumtx", NULL, MTX_DEF | MTX_NEW);
150 rw_init(&sc->lock, "blake2_lock");
156 blake2_detach(device_t dev)
158 struct blake2_softc *sc;
160 sc = device_get_softc(dev);
164 rw_wunlock(&sc->lock);
165 crypto_unregister_all(sc->cid);
167 rw_destroy(&sc->lock);
175 blake2_probesession(device_t dev, const struct crypto_session_params *csp)
178 if (csp->csp_flags != 0)
180 switch (csp->csp_mode) {
181 case CSP_MODE_DIGEST:
182 switch (csp->csp_auth_alg) {
193 return (CRYPTODEV_PROBE_ACCEL_SOFTWARE);
197 blake2_newsession(device_t dev, crypto_session_t cses,
198 const struct crypto_session_params *csp)
200 struct blake2_softc *sc;
201 struct blake2_session *ses;
204 sc = device_get_softc(dev);
206 ses = crypto_get_driver_session(cses);
210 rw_runlock(&sc->lock);
213 rw_runlock(&sc->lock);
215 error = blake2_cipher_setup(ses, csp);
217 CRYPTDEB("setup failed");
225 blake2_process(device_t dev, struct cryptop *crp, int hint __unused)
227 struct blake2_session *ses;
230 ses = crypto_get_driver_session(crp->crp_session);
231 error = blake2_cipher_process(ses, crp);
233 crp->crp_etype = error;
238 static device_method_t blake2_methods[] = {
239 DEVMETHOD(device_identify, blake2_identify),
240 DEVMETHOD(device_probe, blake2_probe),
241 DEVMETHOD(device_attach, blake2_attach),
242 DEVMETHOD(device_detach, blake2_detach),
244 DEVMETHOD(cryptodev_probesession, blake2_probesession),
245 DEVMETHOD(cryptodev_newsession, blake2_newsession),
246 DEVMETHOD(cryptodev_process, blake2_process),
251 static driver_t blake2_driver = {
254 sizeof(struct blake2_softc),
256 static devclass_t blake2_devclass;
258 DRIVER_MODULE(blake2, nexus, blake2_driver, blake2_devclass, 0, 0);
259 MODULE_VERSION(blake2, 1);
260 MODULE_DEPEND(blake2, crypto, 1, 1, 1);
263 blake2_check_klen(const struct crypto_session_params *csp, unsigned klen)
266 if (csp->csp_auth_alg == CRYPTO_BLAKE2S)
267 return (klen <= BLAKE2S_KEYBYTES);
269 return (klen <= BLAKE2B_KEYBYTES);
273 blake2_cipher_setup(struct blake2_session *ses,
274 const struct crypto_session_params *csp)
278 CTASSERT((size_t)BLAKE2S_OUTBYTES <= (size_t)BLAKE2B_OUTBYTES);
280 if (!blake2_check_klen(csp, csp->csp_auth_klen))
283 if (csp->csp_auth_mlen < 0)
286 switch (csp->csp_auth_alg) {
288 hashlen = BLAKE2S_OUTBYTES;
291 hashlen = BLAKE2B_OUTBYTES;
297 if (csp->csp_auth_mlen > hashlen)
300 if (csp->csp_auth_mlen == 0)
303 ses->mlen = csp->csp_auth_mlen;
308 blake2b_applicator(void *state, const void *buf, u_int len)
312 rc = blake2b_update(state, buf, len);
319 blake2s_applicator(void *state, const void *buf, u_int len)
323 rc = blake2s_update(state, buf, len);
330 blake2_cipher_process(struct blake2_session *ses, struct cryptop *crp)
336 char res[BLAKE2B_OUTBYTES], res2[BLAKE2B_OUTBYTES];
337 const struct crypto_session_params *csp;
338 struct fpu_kern_ctx *ctx;
349 kt = is_fpu_kern_thread(0);
351 ACQUIRE_CTX(ctxidx, ctx);
352 fpu_kern_enter(curthread, ctx,
353 FPU_KERN_NORMAL | FPU_KERN_KTHR);
356 csp = crypto_get_params(crp->crp_session);
357 if (crp->crp_auth_key != NULL)
358 key = crp->crp_auth_key;
360 key = csp->csp_auth_key;
361 klen = csp->csp_auth_klen;
362 switch (csp->csp_auth_alg) {
365 rc = blake2b_init_key(&bctx.sb, ses->mlen, key, klen);
367 rc = blake2b_init(&bctx.sb, ses->mlen);
370 error = crypto_apply(crp, crp->crp_payload_start,
371 crp->crp_payload_length, blake2b_applicator, &bctx.sb);
374 rc = blake2b_final(&bctx.sb, res, ses->mlen);
382 rc = blake2s_init_key(&bctx.ss, ses->mlen, key, klen);
384 rc = blake2s_init(&bctx.ss, ses->mlen);
387 error = crypto_apply(crp, crp->crp_payload_start,
388 crp->crp_payload_length, blake2s_applicator, &bctx.ss);
391 rc = blake2s_final(&bctx.ss, res, ses->mlen);
398 panic("unreachable");
401 if (crp->crp_op & CRYPTO_OP_VERIFY_DIGEST) {
402 crypto_copydata(crp, crp->crp_digest_start, ses->mlen, res2);
403 if (timingsafe_bcmp(res, res2, ses->mlen) != 0)
406 crypto_copyback(crp, crp->crp_digest_start, ses->mlen, res);
410 fpu_kern_leave(curthread, ctx);
411 RELEASE_CTX(ctxidx, ctx);