[WTF] Import std::optional reference implementation as WTF::Optional
[WebKit-https.git] / Source / WebCore / crypto / algorithms / CryptoAlgorithmAES_KW.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_KW.h"
28
29 #if ENABLE(SUBTLE_CRYPTO)
30
31 #include "CryptoAlgorithmAesKeyGenParams.h"
32 #include "CryptoAlgorithmAesKeyGenParamsDeprecated.h"
33 #include "CryptoKeyAES.h"
34 #include "CryptoKeyDataOctetSequence.h"
35 #include "ExceptionCode.h"
36
37 namespace WebCore {
38
39 static const char* const ALG128 = "A128KW";
40 static const char* const ALG192 = "A192KW";
41 static const char* const ALG256 = "A256KW";
42
43 static inline bool usagesAreInvalidForCryptoAlgorithmAES_KW(CryptoKeyUsageBitmap usages)
44 {
45     return usages & (CryptoKeyUsageSign | CryptoKeyUsageVerify | CryptoKeyUsageDeriveKey | CryptoKeyUsageDeriveBits | CryptoKeyUsageEncrypt | CryptoKeyUsageDecrypt);
46 }
47
48 Ref<CryptoAlgorithm> CryptoAlgorithmAES_KW::create()
49 {
50     return adoptRef(*new CryptoAlgorithmAES_KW);
51 }
52
53 CryptoAlgorithmIdentifier CryptoAlgorithmAES_KW::identifier() const
54 {
55     return s_identifier;
56 }
57
58 bool CryptoAlgorithmAES_KW::keyAlgorithmMatches(const CryptoKey& key) const
59 {
60     if (key.algorithmIdentifier() != s_identifier)
61         return false;
62     ASSERT(is<CryptoKeyAES>(key));
63     return true;
64 }
65
66 void CryptoAlgorithmAES_KW::generateKey(const std::unique_ptr<CryptoAlgorithmParameters>&& parameters, bool extractable, CryptoKeyUsageBitmap usages, KeyOrKeyPairCallback&& callback, ExceptionCallback&& exceptionCallback, ScriptExecutionContext&)
67 {
68     ASSERT(parameters);
69     if (usagesAreInvalidForCryptoAlgorithmAES_KW(usages)) {
70         exceptionCallback(SYNTAX_ERR);
71         return;
72     }
73
74     auto result = CryptoKeyAES::generate(CryptoAlgorithmIdentifier::AES_KW, downcast<CryptoAlgorithmAesKeyGenParams>(*parameters).length, extractable, usages);
75     if (!result) {
76         exceptionCallback(OperationError);
77         return;
78     }
79
80     callback(result.get(), nullptr);
81 }
82
83 void CryptoAlgorithmAES_KW::importKey(SubtleCrypto::KeyFormat format, KeyData&& data, const std::unique_ptr<CryptoAlgorithmParameters>&& parameters, bool extractable, CryptoKeyUsageBitmap usages, KeyCallback&& callback, ExceptionCallback&& exceptionCallback)
84 {
85     ASSERT(parameters);
86     if (usagesAreInvalidForCryptoAlgorithmAES_KW(usages)) {
87         exceptionCallback(SYNTAX_ERR);
88         return;
89     }
90
91     RefPtr<CryptoKeyAES> result;
92     switch (format) {
93     case SubtleCrypto::KeyFormat::Raw:
94         result = CryptoKeyAES::importRaw(parameters->identifier, WTFMove(WTF::get<Vector<uint8_t>>(data)), extractable, usages);
95         break;
96     case SubtleCrypto::KeyFormat::Jwk: {
97         result = CryptoKeyAES::importJwk(parameters->identifier, WTFMove(WTF::get<JsonWebKey>(data)), extractable, usages, [](size_t length, const std::optional<String>& alg) -> bool {
98             switch (length) {
99             case CryptoKeyAES::s_length128:
100                 return !alg || alg.value() == ALG128;
101             case CryptoKeyAES::s_length192:
102                 return !alg || alg.value() == ALG192;
103             case CryptoKeyAES::s_length256:
104                 return !alg || alg.value() == ALG256;
105             }
106             return false;
107         });
108         break;
109     }
110     default:
111         exceptionCallback(NOT_SUPPORTED_ERR);
112         return;
113     }
114     if (!result) {
115         exceptionCallback(DataError);
116         return;
117     }
118
119     callback(*result);
120 }
121
122 void CryptoAlgorithmAES_KW::exportKey(SubtleCrypto::KeyFormat format, Ref<CryptoKey>&& key, KeyDataCallback&& callback, ExceptionCallback&& exceptionCallback)
123 {
124     const auto& aesKey = downcast<CryptoKeyAES>(key.get());
125
126     if (aesKey.key().isEmpty()) {
127         exceptionCallback(OperationError);
128         return;
129     }
130
131     KeyData result;
132     switch (format) {
133     case SubtleCrypto::KeyFormat::Raw:
134         result = Vector<uint8_t>(aesKey.key());
135         break;
136     case SubtleCrypto::KeyFormat::Jwk: {
137         JsonWebKey jwk = aesKey.exportJwk();
138         switch (aesKey.key().size() * 8) {
139         case CryptoKeyAES::s_length128:
140             jwk.alg = String(ALG128);
141             break;
142         case CryptoKeyAES::s_length192:
143             jwk.alg = String(ALG192);
144             break;
145         case CryptoKeyAES::s_length256:
146             jwk.alg = String(ALG256);
147             break;
148         default:
149             ASSERT_NOT_REACHED();
150         }
151         result = WTFMove(jwk);
152         break;
153     }
154     default:
155         exceptionCallback(NOT_SUPPORTED_ERR);
156         return;
157     }
158
159     callback(format, WTFMove(result));
160 }
161
162 ExceptionOr<void> CryptoAlgorithmAES_KW::encryptForWrapKey(const CryptoAlgorithmParametersDeprecated&, const CryptoKey& key, const CryptoOperationData& data, VectorCallback&& callback, VoidCallback&& failureCallback)
163 {
164     if (!keyAlgorithmMatches(key))
165         return Exception { NOT_SUPPORTED_ERR };
166     return platformEncrypt(downcast<CryptoKeyAES>(key), data, WTFMove(callback), WTFMove(failureCallback));
167 }
168
169 ExceptionOr<void> CryptoAlgorithmAES_KW::decryptForUnwrapKey(const CryptoAlgorithmParametersDeprecated&, const CryptoKey& key, const CryptoOperationData& data, VectorCallback&& callback, VoidCallback&& failureCallback)
170 {
171     if (!keyAlgorithmMatches(key))
172         return Exception { NOT_SUPPORTED_ERR };
173     return platformDecrypt(downcast<CryptoKeyAES>(key), data, WTFMove(callback), WTFMove(failureCallback));
174 }
175
176 ExceptionOr<void> CryptoAlgorithmAES_KW::generateKey(const CryptoAlgorithmParametersDeprecated& parameters, bool extractable, CryptoKeyUsageBitmap usages, KeyOrKeyPairCallback&& callback, VoidCallback&& failureCallback, ScriptExecutionContext&)
177 {
178     auto result = CryptoKeyAES::generate(CryptoAlgorithmIdentifier::AES_KW, downcast<CryptoAlgorithmAesKeyGenParamsDeprecated>(parameters).length, extractable, usages);
179     if (!result) {
180         failureCallback();
181         return { };
182     }
183     callback(result.get(), nullptr);
184     return { };
185 }
186
187 ExceptionOr<void> CryptoAlgorithmAES_KW::importKey(const CryptoAlgorithmParametersDeprecated&, const CryptoKeyData& keyData, bool extractable, CryptoKeyUsageBitmap usage, KeyCallback&& callback, VoidCallback&&)
188 {
189     if (!is<CryptoKeyDataOctetSequence>(keyData))
190         return Exception { NOT_SUPPORTED_ERR };
191     callback(CryptoKeyAES::create(CryptoAlgorithmIdentifier::AES_KW, downcast<CryptoKeyDataOctetSequence>(keyData).octetSequence(), extractable, usage));
192     return { };
193 }
194
195 }
196
197 #endif // ENABLE(SUBTLE_CRYPTO)