crypto/block: refactor qcrypto_block_*crypt_helper functions

qcrypto_block_encrypt_helper and qcrypto_block_decrypt_helper are
almost identical, let's reduce code duplication and simplify further
improvements.

Signed-off-by: Vladimir Sementsov-Ogievskiy <vsementsov@virtuozzo.com>
Reviewed-by: Alberto Garcia <berto@igalia.com>
Signed-off-by: Daniel P. Berrangé <berrange@redhat.com>
This commit is contained in:
Vladimir Sementsov-Ogievskiy 2018-12-07 19:13:48 +03:00 committed by Daniel P. Berrangé
parent b640adca9f
commit 1dc57b6038

View File

@ -190,13 +190,20 @@ void qcrypto_block_free(QCryptoBlock *block)
} }
int qcrypto_block_decrypt_helper(QCryptoCipher *cipher, typedef int (*QCryptoCipherEncDecFunc)(QCryptoCipher *cipher,
const void *in,
void *out,
size_t len,
Error **errp);
static int do_qcrypto_block_encdec(QCryptoCipher *cipher,
size_t niv, size_t niv,
QCryptoIVGen *ivgen, QCryptoIVGen *ivgen,
int sectorsize, int sectorsize,
uint64_t offset, uint64_t offset,
uint8_t *buf, uint8_t *buf,
size_t len, size_t len,
QCryptoCipherEncDecFunc func,
Error **errp) Error **errp)
{ {
uint8_t *iv; uint8_t *iv;
@ -226,8 +233,7 @@ int qcrypto_block_decrypt_helper(QCryptoCipher *cipher,
} }
nbytes = len > sectorsize ? sectorsize : len; nbytes = len > sectorsize ? sectorsize : len;
if (qcrypto_cipher_decrypt(cipher, buf, buf, if (func(cipher, buf, buf, nbytes, errp) < 0) {
nbytes, errp) < 0) {
goto cleanup; goto cleanup;
} }
@ -243,6 +249,20 @@ int qcrypto_block_decrypt_helper(QCryptoCipher *cipher,
} }
int qcrypto_block_decrypt_helper(QCryptoCipher *cipher,
size_t niv,
QCryptoIVGen *ivgen,
int sectorsize,
uint64_t offset,
uint8_t *buf,
size_t len,
Error **errp)
{
return do_qcrypto_block_encdec(cipher, niv, ivgen, sectorsize, offset,
buf, len, qcrypto_cipher_decrypt, errp);
}
int qcrypto_block_encrypt_helper(QCryptoCipher *cipher, int qcrypto_block_encrypt_helper(QCryptoCipher *cipher,
size_t niv, size_t niv,
QCryptoIVGen *ivgen, QCryptoIVGen *ivgen,
@ -252,45 +272,6 @@ int qcrypto_block_encrypt_helper(QCryptoCipher *cipher,
size_t len, size_t len,
Error **errp) Error **errp)
{ {
uint8_t *iv; return do_qcrypto_block_encdec(cipher, niv, ivgen, sectorsize, offset,
int ret = -1; buf, len, qcrypto_cipher_encrypt, errp);
uint64_t startsector = offset / sectorsize;
assert(QEMU_IS_ALIGNED(offset, sectorsize));
assert(QEMU_IS_ALIGNED(len, sectorsize));
iv = niv ? g_new0(uint8_t, niv) : NULL;
while (len > 0) {
size_t nbytes;
if (niv) {
if (qcrypto_ivgen_calculate(ivgen,
startsector,
iv, niv,
errp) < 0) {
goto cleanup;
}
if (qcrypto_cipher_setiv(cipher,
iv, niv,
errp) < 0) {
goto cleanup;
}
}
nbytes = len > sectorsize ? sectorsize : len;
if (qcrypto_cipher_encrypt(cipher, buf, buf,
nbytes, errp) < 0) {
goto cleanup;
}
startsector++;
buf += nbytes;
len -= nbytes;
}
ret = 0;
cleanup:
g_free(iv);
return ret;
} }