Source/WebCore/crypto/mac/CryptoAlgorithmAES_CBCMac.cpp

   1 /*
   2  * Copyright (C) 2013 Apple Inc. All rights reserved.
   3  *
   4  * Redistribution and use in source and binary forms, with or without
   5  * modification, are permitted provided that the following conditions
   6  * are met:
   7  * 1. Redistributions of source code must retain the above copyright
   8  *    notice, this list of conditions and the following disclaimer.
   9  * 2. Redistributions in binary form must reproduce the above copyright
  10  *    notice, this list of conditions and the following disclaimer in the
  11  *    documentation and/or other materials provided with the distribution.
  12  *
  13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
  14  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
  15  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  16  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
  17  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  18  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  19  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  20  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  21  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  22  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
  23  * THE POSSIBILITY OF SUCH DAMAGE.
  24  */
  25
  26 #include "config.h"
  27 #include "CryptoAlgorithmAES_CBC.h"
  28
  29 #if ENABLE(SUBTLE_CRYPTO)
  30
  31 #include "CryptoAlgorithmAesCbcCfbParams.h"
  32 #include "CryptoKeyAES.h"
  33 #include "ScriptExecutionContext.h"
  34 #include <CommonCrypto/CommonCrypto.h>
  35
  36 namespace WebCore {
  37
  38 static ExceptionOr<Vector<uint8_t>> transformAES_CBC(CCOperation operation, const Vector<uint8_t>& iv, const Vector<uint8_t>& key, const Vector<uint8_t>& data)
  39 {
  40     CCCryptorRef cryptor;
  41     CCCryptorStatus status = CCCryptorCreate(operation, kCCAlgorithmAES, kCCOptionPKCS7Padding, key.data(), key.size(), iv.data(), &cryptor);
  42     if (status)
  43         return Exception { OperationError };
  44
  45     Vector<uint8_t> result(CCCryptorGetOutputLength(cryptor, data.size(), true));
  46
  47     size_t bytesWritten;
  48     status = CCCryptorUpdate(cryptor, data.data(), data.size(), result.data(), result.size(), &bytesWritten);
  49     if (status)
  50         return Exception { OperationError };
  51
  52     uint8_t* p = result.data() + bytesWritten;
  53     status = CCCryptorFinal(cryptor, p, result.end() - p, &bytesWritten);
  54     p += bytesWritten;
  55     if (status)
  56         return Exception { OperationError };
  57
  58     ASSERT(p <= result.end());
  59     result.shrink(p - result.begin());
  60
  61     CCCryptorRelease(cryptor);
  62
  63     return WTFMove(result);
  64 }
  65
  66 void CryptoAlgorithmAES_CBC::platformEncrypt(std::unique_ptr<CryptoAlgorithmParameters>&& parameters, Ref<CryptoKey>&& key, Vector<uint8_t>&& plainText, VectorCallback&& callback, ExceptionCallback&& exceptionCallback, ScriptExecutionContext& context, WorkQueue& workQueue)
  67 {
  68     context.ref();
  69     workQueue.dispatch([parameters = WTFMove(parameters), key = WTFMove(key), plainText = WTFMove(plainText), callback = WTFMove(callback), exceptionCallback = WTFMove(exceptionCallback), &context]() mutable {
  70         auto& aesParameters = downcast<CryptoAlgorithmAesCbcCfbParams>(*parameters);
  71         auto& aesKey = downcast<CryptoKeyAES>(key.get());
  72         ASSERT(aesParameters.ivVector().size() == kCCBlockSizeAES128);
  73         auto result = transformAES_CBC(kCCEncrypt, aesParameters.ivVector(), aesKey.key(), plainText);
  74         if (result.hasException()) {
  75             // We should only dereference callbacks after being back to the Document/Worker threads.
  76             context.postTask([exceptionCallback = WTFMove(exceptionCallback), ec = result.releaseException().code(), callback = WTFMove(callback)](ScriptExecutionContext& context) {
  77                 exceptionCallback(ec);
  78                 context.deref();
  79             });
  80             return;
  81         }
  82         // We should only dereference callbacks after being back to the Document/Worker threads.
  83         context.postTask([callback = WTFMove(callback), result = result.releaseReturnValue(), exceptionCallback = WTFMove(exceptionCallback)](ScriptExecutionContext& context) {
  84             callback(result);
  85             context.deref();
  86         });
  87     });
  88 }
  89
  90 void CryptoAlgorithmAES_CBC::platformDecrypt(std::unique_ptr<CryptoAlgorithmParameters>&& parameters, Ref<CryptoKey>&& key, Vector<uint8_t>&& cipherText, VectorCallback&& callback, ExceptionCallback&& exceptionCallback, ScriptExecutionContext& context, WorkQueue& workQueue)
  91 {
  92     context.ref();
  93     workQueue.dispatch([parameters = WTFMove(parameters), key = WTFMove(key), cipherText = WTFMove(cipherText), callback = WTFMove(callback), exceptionCallback = WTFMove(exceptionCallback), &context]() mutable {
  94         auto& aesParameters = downcast<CryptoAlgorithmAesCbcCfbParams>(*parameters);
  95         auto& aesKey = downcast<CryptoKeyAES>(key.get());
  96         assert(aesParameters.ivVector().size() == kCCBlockSizeAES128);
  97         auto result = transformAES_CBC(kCCDecrypt, aesParameters.ivVector(), aesKey.key(), cipherText);
  98         if (result.hasException()) {
  99             // We should only dereference callbacks after being back to the Document/Worker threads.
 100             context.postTask([exceptionCallback = WTFMove(exceptionCallback), ec = result.releaseException().code(), callback = WTFMove(callback)](ScriptExecutionContext& context) {
 101                 exceptionCallback(ec);
 102                 context.deref();
 103             });
 104             return;
 105         }
 106         // We should only dereference callbacks after being back to the Document/Worker threads.
 107         context.postTask([callback = WTFMove(callback), result = result.releaseReturnValue(), exceptionCallback = WTFMove(exceptionCallback)](ScriptExecutionContext& context) {
 108             callback(result);
 109             context.deref();
 110         });
 111     });
 112 }
 113
 114 } // namespace WebCore
 115
 116 #endif // ENABLE(SUBTLE_CRYPTO)