c1efd51c95c285dd9e3bf971ba9416276a40629e
[WebKit-https.git] / Source / WebCore / platform / graphics / avfoundation / objc / CDMSessionAVFoundationObjC.mm
1 /*
2  * Copyright (C) 2014 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 #import "config.h"
27 #import "CDMSessionAVFoundationObjC.h"
28
29 #if ENABLE(LEGACY_ENCRYPTED_MEDIA)
30
31 #import "LegacyCDM.h"
32 #import "LegacyCDMSession.h"
33 #import "MediaPlayer.h"
34 #import "MediaPlayerPrivateAVFoundationObjC.h"
35 #import "WebCoreNSErrorExtras.h"
36 #import <AVFoundation/AVAsset.h>
37 #import <AVFoundation/AVAssetResourceLoader.h>
38 #import <JavaScriptCore/TypedArrayInlines.h>
39 #import <objc/objc-runtime.h>
40 #import <wtf/MainThread.h>
41 #import <wtf/SoftLinking.h>
42 #import <wtf/UUID.h>
43
44 namespace WebCore {
45
46 CDMSessionAVFoundationObjC::CDMSessionAVFoundationObjC(MediaPlayerPrivateAVFoundationObjC* parent, LegacyCDMSessionClient* client)
47     : m_parent(makeWeakPtr(*parent))
48     , m_client(client)
49     , m_sessionId(createCanonicalUUIDString())
50 {
51 }
52
53 CDMSessionAVFoundationObjC::~CDMSessionAVFoundationObjC()
54 {
55 }
56
57 RefPtr<Uint8Array> CDMSessionAVFoundationObjC::generateKeyRequest(const String& mimeType, Uint8Array* initData, String& destinationURL, unsigned short& errorCode, uint32_t& systemCode)
58 {
59     UNUSED_PARAM(mimeType);
60
61     if (!m_parent) {
62         errorCode = LegacyCDM::UnknownError;
63         return nullptr;
64     }
65
66     String keyURI;
67     String keyID;
68     RefPtr<Uint8Array> certificate;
69     if (!MediaPlayerPrivateAVFoundationObjC::extractKeyURIKeyIDAndCertificateFromInitData(initData, keyURI, keyID, certificate)) {
70         errorCode = LegacyCDM::UnknownError;
71         return nullptr;
72     }
73
74     m_request = m_parent->takeRequestForKeyURI(keyURI);
75     if (!m_request) {
76         errorCode = LegacyCDM::UnknownError;
77         return nullptr;
78     }
79
80     RetainPtr<NSData> certificateData = adoptNS([[NSData alloc] initWithBytes:certificate->baseAddress() length:certificate->byteLength()]);
81     NSString* assetStr = keyID;
82     RetainPtr<NSData> assetID = [NSData dataWithBytes: [assetStr cStringUsingEncoding:NSUTF8StringEncoding] length:[assetStr lengthOfBytesUsingEncoding:NSUTF8StringEncoding]];
83     NSError* nsError = 0;
84     RetainPtr<NSData> keyRequest = [m_request streamingContentKeyRequestDataForApp:certificateData.get() contentIdentifier:assetID.get() options:nil error:&nsError];
85
86     if (!keyRequest) {
87         errorCode = LegacyCDM::DomainError;
88         systemCode = mediaKeyErrorSystemCode(nsError);
89         return nullptr;
90     }
91
92     errorCode = MediaPlayer::NoError;
93     systemCode = 0;
94     destinationURL = String();
95
96     auto keyRequestBuffer = ArrayBuffer::create([keyRequest.get() bytes], [keyRequest.get() length]);
97     unsigned byteLength = keyRequestBuffer->byteLength();
98     return Uint8Array::tryCreate(WTFMove(keyRequestBuffer), 0, byteLength);
99 }
100
101 void CDMSessionAVFoundationObjC::releaseKeys()
102 {
103 }
104
105 bool CDMSessionAVFoundationObjC::update(Uint8Array* key, RefPtr<Uint8Array>& nextMessage, unsigned short& errorCode, uint32_t& systemCode)
106 {
107     RetainPtr<NSData> keyData = adoptNS([[NSData alloc] initWithBytes:key->baseAddress() length:key->byteLength()]);
108     [[m_request dataRequest] respondWithData:keyData.get()];
109     [m_request finishLoading];
110     errorCode = MediaPlayer::NoError;
111     systemCode = 0;
112     nextMessage = nullptr;
113
114     return true;
115 }
116
117 void CDMSessionAVFoundationObjC::playerDidReceiveError(NSError *error)
118 {
119     if (!m_client)
120         return;
121
122     unsigned long code = mediaKeyErrorSystemCode(error);
123     m_client->sendError(LegacyCDMSessionClient::MediaKeyErrorDomain, code);
124 }
125
126 }
127
128 #endif