[Mac] Change uint8_t* to Vector<uint8_t> type in all crypto algorithm implementation
[WebKit-https.git] / 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)