diff options
Diffstat (limited to 'src/libstrongswan/crypto/aead.c')
| -rw-r--r-- | src/libstrongswan/crypto/aead.c | 10 |
1 files changed, 7 insertions, 3 deletions
diff --git a/src/libstrongswan/crypto/aead.c b/src/libstrongswan/crypto/aead.c index 4afdf21d1..0915cd1de 100644 --- a/src/libstrongswan/crypto/aead.c +++ b/src/libstrongswan/crypto/aead.c @@ -54,17 +54,21 @@ METHOD(aead_t, encrypt, bool, if (encrypted) { - this->crypter->encrypt(this->crypter, plain, iv, &encr); + if (!this->crypter->encrypt(this->crypter, plain, iv, &encr)) + { + return FALSE; + } if (!this->signer->allocate_signature(this->signer, encr, &sig)) { + free(encr.ptr); return FALSE; } *encrypted = chunk_cat("cmm", iv, encr, sig); } else { - this->crypter->encrypt(this->crypter, plain, iv, NULL); - if (!this->signer->get_signature(this->signer, + if (!this->crypter->encrypt(this->crypter, plain, iv, NULL) || + !this->signer->get_signature(this->signer, plain, plain.ptr + plain.len)) { return FALSE; |