xref: /freebsd/stand/libsa/geli/geliboot_crypto.c (revision 266f97b5)
1 /*-
2  * Copyright (c) 2005-2010 Pawel Jakub Dawidek <pjd@FreeBSD.org>
3  * Copyright (c) 2015 Allan Jude <allanjude@FreeBSD.org>
4  * All rights reserved.
5  *
6  * Redistribution and use in source and binary forms, with or without
7  * modification, are permitted provided that the following conditions
8  * are met:
9  * 1. Redistributions of source code must retain the above copyright
10  *    notice, this list of conditions and the following disclaimer.
11  * 2. Redistributions in binary form must reproduce the above copyright
12  *    notice, this list of conditions and the following disclaimer in the
13  *    documentation and/or other materials provided with the distribution.
14  *
15  * THIS SOFTWARE IS PROVIDED BY THE AUTHORS AND CONTRIBUTORS ``AS IS'' AND
16  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
17  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
18  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHORS OR CONTRIBUTORS BE LIABLE
19  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
20  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
21  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
22  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
23  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
24  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
25  * SUCH DAMAGE.
26  *
27  * $FreeBSD$
28  */
29 
30 #include <stdio.h>
31 #include <string.h>
32 #include <strings.h>
33 
34 #include "geliboot_internal.h"
35 #include "geliboot.h"
36 
37 int
38 geliboot_crypt(u_int algo, geli_op_t enc, u_char *data, size_t datasize,
39     const u_char *key, size_t keysize, u_char *iv, size_t ivlen)
40 {
41 	keyInstance aeskey;
42 	cipherInstance cipher;
43 	struct aes_xts_ctx xtsctx, *ctxp;
44 	size_t xts_len;
45 	int err, blks, i;
46 
47 	switch (algo) {
48 	case CRYPTO_AES_CBC:
49 		err = rijndael_makeKey(&aeskey, !enc, keysize,
50 		    (const char *)key);
51 		if (err < 0) {
52 			printf("Failed to setup crypo keys: %d\n", err);
53 			return (err);
54 		}
55 
56 		err = rijndael_cipherInit(&cipher, MODE_CBC, iv);
57 		if (err < 0) {
58 			printf("Failed to setup IV: %d\n", err);
59 			return (err);
60 		}
61 
62 		switch (enc) {
63 		case GELI_DECRYPT:
64 			blks = rijndael_blockDecrypt(&cipher, &aeskey, data,
65 			    datasize * 8, data);
66 			break;
67 		case GELI_ENCRYPT:
68 			blks = rijndael_blockEncrypt(&cipher, &aeskey, data,
69 			    datasize * 8, data);
70 			break;
71 		}
72 		if (datasize != (blks / 8)) {
73 			printf("Failed to %s the entire input: %u != %zu\n",
74 			    enc ? "decrypt" : "encrypt",
75 			    blks, datasize);
76 			return (1);
77 		}
78 		break;
79 	case CRYPTO_AES_XTS:
80 		xts_len = keysize << 1;
81 		ctxp = &xtsctx;
82 
83 		enc_xform_aes_xts.setkey(ctxp, key, xts_len / 8);
84 		enc_xform_aes_xts.reinit(ctxp, iv, ivlen);
85 
86 		switch (enc) {
87 		case GELI_DECRYPT:
88 			for (i = 0; i < datasize; i += AES_XTS_BLOCKSIZE) {
89 				enc_xform_aes_xts.decrypt(ctxp, data + i,
90 				    data + i);
91 			}
92 			break;
93 		case GELI_ENCRYPT:
94 			for (i = 0; i < datasize; i += AES_XTS_BLOCKSIZE) {
95 				enc_xform_aes_xts.encrypt(ctxp, data + i,
96 				    data + i);
97 			}
98 			break;
99 		}
100 		break;
101 	default:
102 		printf("Unsupported crypto algorithm #%d\n", algo);
103 		return (1);
104 	}
105 
106 	return (0);
107 }
108 
109 static int
110 g_eli_crypto_cipher(u_int algo, geli_op_t enc, u_char *data, size_t datasize,
111     const u_char *key, size_t keysize)
112 {
113 	u_char iv[keysize];
114 
115 	explicit_bzero(iv, sizeof(iv));
116 	return (geliboot_crypt(algo, enc, data, datasize, key, keysize, iv,
117 	    sizeof(iv)));
118 }
119 
120 int
121 g_eli_crypto_encrypt(u_int algo, u_char *data, size_t datasize,
122     const u_char *key, size_t keysize)
123 {
124 
125 	/* We prefer AES-CBC for metadata protection. */
126 	if (algo == CRYPTO_AES_XTS)
127 		algo = CRYPTO_AES_CBC;
128 
129 	return (g_eli_crypto_cipher(algo, GELI_ENCRYPT, data, datasize, key,
130 	    keysize));
131 }
132 
133 int
134 g_eli_crypto_decrypt(u_int algo, u_char *data, size_t datasize,
135     const u_char *key, size_t keysize)
136 {
137 
138 	/* We prefer AES-CBC for metadata protection. */
139 	if (algo == CRYPTO_AES_XTS)
140 		algo = CRYPTO_AES_CBC;
141 
142 	return (g_eli_crypto_cipher(algo, GELI_DECRYPT, data, datasize, key,
143 	    keysize));
144 }
145