Rename ENABLE_ENCRYPTED_MEDIA_V2 to ENABLE_LEGACY_ENCRYPTED_MEDIA
[WebKit.git] / Source / WebCore / Modules / encryptedmedia / CDMSessionClearKey.cpp
1 /*
2  * Copyright (C) 2015 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 "CDMSessionClearKey.h"
28
29 #include "ArrayValue.h"
30 #include "CryptoAlgorithm.h"
31 #include "CryptoAlgorithmIdentifier.h"
32 #include "CryptoAlgorithmParameters.h"
33 #include "CryptoKeyDataOctetSequence.h"
34 #include "Dictionary.h"
35 #include "JSMainThreadExecState.h"
36 #include "Logging.h"
37 #include "MediaKeyError.h"
38 #include "TextEncoding.h"
39 #include "UUID.h"
40 #include <runtime/JSGlobalObject.h>
41 #include <runtime/JSLock.h>
42 #include <runtime/JSONObject.h>
43 #include <runtime/VM.h>
44 #include <wtf/NeverDestroyed.h>
45 #include <wtf/text/Base64.h>
46
47 #if ENABLE(LEGACY_ENCRYPTED_MEDIA)
48
49 using namespace JSC;
50
51 namespace WebCore {
52
53 static VM& clearKeyVM()
54 {
55     static NeverDestroyed<RefPtr<VM>> vm;
56     if (!vm.get())
57         vm.get() = VM::create();
58
59     return *vm.get();
60 }
61
62 CDMSessionClearKey::CDMSessionClearKey(CDMSessionClient* client)
63     : m_client(client)
64     , m_sessionId(createCanonicalUUIDString())
65 {
66 }
67
68 CDMSessionClearKey::~CDMSessionClearKey()
69 {
70 }
71
72 RefPtr<Uint8Array> CDMSessionClearKey::generateKeyRequest(const String& mimeType, Uint8Array* initData, String& destinationURL, unsigned short& errorCode, uint32_t& systemCode)
73 {
74     UNUSED_PARAM(mimeType);
75     UNUSED_PARAM(destinationURL);
76     UNUSED_PARAM(systemCode);
77
78     if (!initData) {
79         errorCode = MediaKeyError::MEDIA_KEYERR_CLIENT;
80         return nullptr;
81     }
82     m_initData = initData;
83
84     bool sawError = false;
85     String keyID = UTF8Encoding().decode(reinterpret_cast_ptr<char*>(m_initData->baseAddress()), m_initData->byteLength(), true, sawError);
86     if (sawError) {
87         errorCode = MediaKeyError::MEDIA_KEYERR_CLIENT;
88         return nullptr;
89     }
90
91     return initData;
92 }
93
94 void CDMSessionClearKey::releaseKeys()
95 {
96     m_cachedKeys.clear();
97 }
98
99 bool CDMSessionClearKey::update(Uint8Array* rawKeysData, RefPtr<Uint8Array>& nextMessage, unsigned short& errorCode, uint32_t& systemCode)
100 {
101     UNUSED_PARAM(nextMessage);
102     UNUSED_PARAM(systemCode);
103     ASSERT(rawKeysData);
104
105     do {
106         String rawKeysString = String::fromUTF8(rawKeysData->data(), rawKeysData->length());
107         if (rawKeysString.isEmpty())  {
108             LOG(Media, "CDMSessionClearKey::update(%p) - failed: empty message", this);
109             continue;
110         }
111
112         VM& vm = clearKeyVM();
113         JSLockHolder lock(vm);
114         auto scope = DECLARE_THROW_SCOPE(vm);
115         JSGlobalObject* globalObject = JSGlobalObject::create(vm, JSGlobalObject::createStructure(vm, jsNull()));
116         ExecState* exec = globalObject->globalExec();
117
118         JSLockHolder locker(clearKeyVM());
119         JSValue keysDataObject = JSONParse(exec, rawKeysString);
120         if (scope.exception() || !keysDataObject) {
121             LOG(Media, "CDMSessionClearKey::update(%p) - failed: invalid JSON", this);
122             break;
123         }
124         Dictionary keysDataDictionary(exec, keysDataObject);
125         ArrayValue keysArray;
126         size_t length;
127         if (!keysDataDictionary.get("keys", keysArray) || keysArray.isUndefinedOrNull() || !keysArray.length(length) || !length) {
128             LOG(Media, "CDMSessionClearKey::update(%p) - failed: keys array missing or empty", this);
129             break;
130         }
131
132         bool foundValidKey = false;
133         for (size_t i = 0; i < length; ++i) {
134             Dictionary keyDictionary;
135             if (!keysArray.get(i, keyDictionary) || keyDictionary.isUndefinedOrNull()) {
136                 LOG(Media, "CDMSessionClearKey::update(%p) - failed: null keyDictionary", this);
137                 continue;
138             }
139
140             String algorithm;
141             if (!keyDictionary.get("alg", algorithm) || !equalLettersIgnoringASCIICase(algorithm, "a128kw")) {
142                 LOG(Media, "CDMSessionClearKey::update(%p) - failed: algorithm unsupported", this);
143                 continue;
144             }
145
146             String keyType;
147             if (!keyDictionary.get("kty", keyType) || !equalLettersIgnoringASCIICase(keyType, "oct")) {
148                 LOG(Media, "CDMSessionClearKey::update(%p) - failed: keyType unsupported", this);
149                 continue;
150             }
151
152             String keyId;
153             if (!keyDictionary.get("kid", keyId) || keyId.isEmpty()) {
154                 LOG(Media, "CDMSessionClearKey::update(%p) - failed: keyId missing or empty", this);
155                 continue;
156             }
157
158             String rawKeyData;
159             if (!keyDictionary.get("k", rawKeyData) || rawKeyData.isEmpty())  {
160                 LOG(Media, "CDMSessionClearKey::update(%p) - failed: key missing or empty", this);
161                 continue;
162             }
163
164             Vector<uint8_t> keyData;
165             if (!base64Decode(rawKeyData, keyData) ||  keyData.isEmpty()) {
166                 LOG(Media, "CDMSessionClearKey::update(%p) - failed: unable to base64 decode key", this);
167                 continue;
168             }
169
170             m_cachedKeys.set(keyId, WTFMove(keyData));
171             foundValidKey = true;
172         }
173
174         if (foundValidKey)
175             return true;
176
177     } while (false);
178
179     errorCode = MediaKeyError::MEDIA_KEYERR_CLIENT;
180     return false;
181 }
182
183 RefPtr<ArrayBuffer> CDMSessionClearKey::cachedKeyForKeyID(const String& keyId) const
184 {
185     if (!m_cachedKeys.contains(keyId))
186         return nullptr;
187
188     auto keyData = m_cachedKeys.get(keyId);
189     RefPtr<Uint8Array> keyDataArray = Uint8Array::create(keyData.data(), keyData.size());
190     return keyDataArray->buffer();
191 }
192
193 }
194
195 #endif