stand/libsa/geli/geliboot_crypto.c

   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)
  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);
  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 }
 118
 119 int
 120 g_eli_crypto_encrypt(u_int algo, u_char *data, size_t datasize,
 121     const u_char *key, size_t keysize)
 122 {
 123
 124         /* We prefer AES-CBC for metadata protection. */
 125         if (algo == CRYPTO_AES_XTS)
 126                 algo = CRYPTO_AES_CBC;
 127
 128         return (g_eli_crypto_cipher(algo, GELI_ENCRYPT, data, datasize, key,
 129             keysize));
 130 }
 131
 132 int
 133 g_eli_crypto_decrypt(u_int algo, u_char *data, size_t datasize,
 134     const u_char *key, size_t keysize)
 135 {
 136
 137         /* We prefer AES-CBC for metadata protection. */
 138         if (algo == CRYPTO_AES_XTS)
 139                 algo = CRYPTO_AES_CBC;
 140
 141         return (g_eli_crypto_cipher(algo, GELI_DECRYPT, data, datasize, key,
 142             keysize));
 143 }