1 /* 2 * QEMU Crypto block device encryption 3 * 4 * Copyright (c) 2015-2016 Red Hat, Inc. 5 * 6 * This library is free software; you can redistribute it and/or 7 * modify it under the terms of the GNU Lesser General Public 8 * License as published by the Free Software Foundation; either 9 * version 2.1 of the License, or (at your option) any later version. 10 * 11 * This library is distributed in the hope that it will be useful, 12 * but WITHOUT ANY WARRANTY; without even the implied warranty of 13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 14 * Lesser General Public License for more details. 15 * 16 * You should have received a copy of the GNU Lesser General Public 17 * License along with this library; if not, see <http://www.gnu.org/licenses/>. 18 * 19 */ 20 21 #include "qemu/osdep.h" 22 #include "qapi/error.h" 23 #include "blockpriv.h" 24 #include "block-qcow.h" 25 #include "block-luks.h" 26 27 static const QCryptoBlockDriver *qcrypto_block_drivers[] = { 28 [Q_CRYPTO_BLOCK_FORMAT_QCOW] = &qcrypto_block_driver_qcow, 29 [Q_CRYPTO_BLOCK_FORMAT_LUKS] = &qcrypto_block_driver_luks, 30 }; 31 32 33 bool qcrypto_block_has_format(QCryptoBlockFormat format, 34 const uint8_t *buf, 35 size_t len) 36 { 37 const QCryptoBlockDriver *driver; 38 39 if (format >= G_N_ELEMENTS(qcrypto_block_drivers) || 40 !qcrypto_block_drivers[format]) { 41 return false; 42 } 43 44 driver = qcrypto_block_drivers[format]; 45 46 return driver->has_format(buf, len); 47 } 48 49 50 QCryptoBlock *qcrypto_block_open(QCryptoBlockOpenOptions *options, 51 const char *optprefix, 52 QCryptoBlockReadFunc readfunc, 53 void *opaque, 54 unsigned int flags, 55 size_t n_threads, 56 Error **errp) 57 { 58 QCryptoBlock *block = g_new0(QCryptoBlock, 1); 59 60 block->format = options->format; 61 62 if (options->format >= G_N_ELEMENTS(qcrypto_block_drivers) || 63 !qcrypto_block_drivers[options->format]) { 64 error_setg(errp, "Unsupported block driver %s", 65 QCryptoBlockFormat_str(options->format)); 66 g_free(block); 67 return NULL; 68 } 69 70 block->driver = qcrypto_block_drivers[options->format]; 71 72 if (block->driver->open(block, options, optprefix, 73 readfunc, opaque, flags, n_threads, errp) < 0) 74 { 75 g_free(block); 76 return NULL; 77 } 78 79 qemu_mutex_init(&block->mutex); 80 81 return block; 82 } 83 84 85 QCryptoBlock *qcrypto_block_create(QCryptoBlockCreateOptions *options, 86 const char *optprefix, 87 QCryptoBlockInitFunc initfunc, 88 QCryptoBlockWriteFunc writefunc, 89 void *opaque, 90 Error **errp) 91 { 92 QCryptoBlock *block = g_new0(QCryptoBlock, 1); 93 94 block->format = options->format; 95 96 if (options->format >= G_N_ELEMENTS(qcrypto_block_drivers) || 97 !qcrypto_block_drivers[options->format]) { 98 error_setg(errp, "Unsupported block driver %s", 99 QCryptoBlockFormat_str(options->format)); 100 g_free(block); 101 return NULL; 102 } 103 104 block->driver = qcrypto_block_drivers[options->format]; 105 106 if (block->driver->create(block, options, optprefix, initfunc, 107 writefunc, opaque, errp) < 0) { 108 g_free(block); 109 return NULL; 110 } 111 112 qemu_mutex_init(&block->mutex); 113 114 return block; 115 } 116 117 118 QCryptoBlockInfo *qcrypto_block_get_info(QCryptoBlock *block, 119 Error **errp) 120 { 121 QCryptoBlockInfo *info = g_new0(QCryptoBlockInfo, 1); 122 123 info->format = block->format; 124 125 if (block->driver->get_info && 126 block->driver->get_info(block, info, errp) < 0) { 127 g_free(info); 128 return NULL; 129 } 130 131 return info; 132 } 133 134 135 int qcrypto_block_decrypt(QCryptoBlock *block, 136 uint64_t offset, 137 uint8_t *buf, 138 size_t len, 139 Error **errp) 140 { 141 return block->driver->decrypt(block, offset, buf, len, errp); 142 } 143 144 145 int qcrypto_block_encrypt(QCryptoBlock *block, 146 uint64_t offset, 147 uint8_t *buf, 148 size_t len, 149 Error **errp) 150 { 151 return block->driver->encrypt(block, offset, buf, len, errp); 152 } 153 154 155 QCryptoCipher *qcrypto_block_get_cipher(QCryptoBlock *block) 156 { 157 /* Ciphers should be accessed through pop/push method to be thread-safe. 158 * Better, they should not be accessed externally at all (note, that 159 * pop/push are static functions) 160 * This function is used only in test with one thread (it's safe to skip 161 * pop/push interface), so it's enough to assert it here: 162 */ 163 assert(block->n_ciphers <= 1); 164 return block->ciphers ? block->ciphers[0] : NULL; 165 } 166 167 168 static QCryptoCipher *qcrypto_block_pop_cipher(QCryptoBlock *block) 169 { 170 QCryptoCipher *cipher; 171 172 qemu_mutex_lock(&block->mutex); 173 174 assert(block->n_free_ciphers > 0); 175 block->n_free_ciphers--; 176 cipher = block->ciphers[block->n_free_ciphers]; 177 178 qemu_mutex_unlock(&block->mutex); 179 180 return cipher; 181 } 182 183 184 static void qcrypto_block_push_cipher(QCryptoBlock *block, 185 QCryptoCipher *cipher) 186 { 187 qemu_mutex_lock(&block->mutex); 188 189 assert(block->n_free_ciphers < block->n_ciphers); 190 block->ciphers[block->n_free_ciphers] = cipher; 191 block->n_free_ciphers++; 192 193 qemu_mutex_unlock(&block->mutex); 194 } 195 196 197 int qcrypto_block_init_cipher(QCryptoBlock *block, 198 QCryptoCipherAlgorithm alg, 199 QCryptoCipherMode mode, 200 const uint8_t *key, size_t nkey, 201 size_t n_threads, Error **errp) 202 { 203 size_t i; 204 205 assert(!block->ciphers && !block->n_ciphers && !block->n_free_ciphers); 206 207 block->ciphers = g_new0(QCryptoCipher *, n_threads); 208 209 for (i = 0; i < n_threads; i++) { 210 block->ciphers[i] = qcrypto_cipher_new(alg, mode, key, nkey, errp); 211 if (!block->ciphers[i]) { 212 qcrypto_block_free_cipher(block); 213 return -1; 214 } 215 block->n_ciphers++; 216 block->n_free_ciphers++; 217 } 218 219 return 0; 220 } 221 222 223 void qcrypto_block_free_cipher(QCryptoBlock *block) 224 { 225 size_t i; 226 227 if (!block->ciphers) { 228 return; 229 } 230 231 assert(block->n_ciphers == block->n_free_ciphers); 232 233 for (i = 0; i < block->n_ciphers; i++) { 234 qcrypto_cipher_free(block->ciphers[i]); 235 } 236 237 g_free(block->ciphers); 238 block->ciphers = NULL; 239 block->n_ciphers = block->n_free_ciphers = 0; 240 } 241 242 QCryptoIVGen *qcrypto_block_get_ivgen(QCryptoBlock *block) 243 { 244 /* ivgen should be accessed under mutex. However, this function is used only 245 * in test with one thread, so it's enough to assert it here: 246 */ 247 assert(block->n_ciphers <= 1); 248 return block->ivgen; 249 } 250 251 252 QCryptoHashAlgorithm qcrypto_block_get_kdf_hash(QCryptoBlock *block) 253 { 254 return block->kdfhash; 255 } 256 257 258 uint64_t qcrypto_block_get_payload_offset(QCryptoBlock *block) 259 { 260 return block->payload_offset; 261 } 262 263 264 uint64_t qcrypto_block_get_sector_size(QCryptoBlock *block) 265 { 266 return block->sector_size; 267 } 268 269 270 void qcrypto_block_free(QCryptoBlock *block) 271 { 272 if (!block) { 273 return; 274 } 275 276 block->driver->cleanup(block); 277 278 qcrypto_block_free_cipher(block); 279 qcrypto_ivgen_free(block->ivgen); 280 qemu_mutex_destroy(&block->mutex); 281 g_free(block); 282 } 283 284 285 typedef int (*QCryptoCipherEncDecFunc)(QCryptoCipher *cipher, 286 const void *in, 287 void *out, 288 size_t len, 289 Error **errp); 290 291 static int do_qcrypto_block_cipher_encdec(QCryptoCipher *cipher, 292 size_t niv, 293 QCryptoIVGen *ivgen, 294 QemuMutex *ivgen_mutex, 295 int sectorsize, 296 uint64_t offset, 297 uint8_t *buf, 298 size_t len, 299 QCryptoCipherEncDecFunc func, 300 Error **errp) 301 { 302 uint8_t *iv; 303 int ret = -1; 304 uint64_t startsector = offset / sectorsize; 305 306 assert(QEMU_IS_ALIGNED(offset, sectorsize)); 307 assert(QEMU_IS_ALIGNED(len, sectorsize)); 308 309 iv = niv ? g_new0(uint8_t, niv) : NULL; 310 311 while (len > 0) { 312 size_t nbytes; 313 if (niv) { 314 if (ivgen_mutex) { 315 qemu_mutex_lock(ivgen_mutex); 316 } 317 ret = qcrypto_ivgen_calculate(ivgen, startsector, iv, niv, errp); 318 if (ivgen_mutex) { 319 qemu_mutex_unlock(ivgen_mutex); 320 } 321 322 if (ret < 0) { 323 goto cleanup; 324 } 325 326 if (qcrypto_cipher_setiv(cipher, 327 iv, niv, 328 errp) < 0) { 329 goto cleanup; 330 } 331 } 332 333 nbytes = len > sectorsize ? sectorsize : len; 334 if (func(cipher, buf, buf, nbytes, errp) < 0) { 335 goto cleanup; 336 } 337 338 startsector++; 339 buf += nbytes; 340 len -= nbytes; 341 } 342 343 ret = 0; 344 cleanup: 345 g_free(iv); 346 return ret; 347 } 348 349 350 int qcrypto_block_cipher_decrypt_helper(QCryptoCipher *cipher, 351 size_t niv, 352 QCryptoIVGen *ivgen, 353 int sectorsize, 354 uint64_t offset, 355 uint8_t *buf, 356 size_t len, 357 Error **errp) 358 { 359 return do_qcrypto_block_cipher_encdec(cipher, niv, ivgen, NULL, sectorsize, 360 offset, buf, len, 361 qcrypto_cipher_decrypt, errp); 362 } 363 364 365 int qcrypto_block_cipher_encrypt_helper(QCryptoCipher *cipher, 366 size_t niv, 367 QCryptoIVGen *ivgen, 368 int sectorsize, 369 uint64_t offset, 370 uint8_t *buf, 371 size_t len, 372 Error **errp) 373 { 374 return do_qcrypto_block_cipher_encdec(cipher, niv, ivgen, NULL, sectorsize, 375 offset, buf, len, 376 qcrypto_cipher_encrypt, errp); 377 } 378 379 int qcrypto_block_decrypt_helper(QCryptoBlock *block, 380 int sectorsize, 381 uint64_t offset, 382 uint8_t *buf, 383 size_t len, 384 Error **errp) 385 { 386 int ret; 387 QCryptoCipher *cipher = qcrypto_block_pop_cipher(block); 388 389 ret = do_qcrypto_block_cipher_encdec(cipher, block->niv, block->ivgen, 390 &block->mutex, sectorsize, offset, buf, 391 len, qcrypto_cipher_decrypt, errp); 392 393 qcrypto_block_push_cipher(block, cipher); 394 395 return ret; 396 } 397 398 int qcrypto_block_encrypt_helper(QCryptoBlock *block, 399 int sectorsize, 400 uint64_t offset, 401 uint8_t *buf, 402 size_t len, 403 Error **errp) 404 { 405 int ret; 406 QCryptoCipher *cipher = qcrypto_block_pop_cipher(block); 407 408 ret = do_qcrypto_block_cipher_encdec(cipher, block->niv, block->ivgen, 409 &block->mutex, sectorsize, offset, buf, 410 len, qcrypto_cipher_encrypt, errp); 411 412 qcrypto_block_push_cipher(block, cipher); 413 414 return ret; 415 } 416