[MediaStream] Protect MediaDevicesRequest during callback
[WebKit-https.git] / Source / WebCore / Modules / mediastream / MediaDevicesEnumerationRequest.cpp
1 /*
2  * Copyright (C) 2016 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. ``AS IS'' AND ANY
14  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
15  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE INC. OR
17  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
18  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
19  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
20  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
21  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
22  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
23  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
24  *
25  */
26
27 #include "config.h"
28 #include "MediaDevicesEnumerationRequest.h"
29
30 #if ENABLE(MEDIA_STREAM)
31
32 #include "CaptureDevice.h"
33 #include "Document.h"
34 #include "MainFrame.h"
35 #include "SecurityOrigin.h"
36 #include "UserMediaController.h"
37
38 namespace WebCore {
39
40 Ref<MediaDevicesEnumerationRequest> MediaDevicesEnumerationRequest::create(Document& document, CompletionHandler&& completionHandler)
41 {
42     return adoptRef(*new MediaDevicesEnumerationRequest(document, WTFMove(completionHandler)));
43 }
44
45 MediaDevicesEnumerationRequest::MediaDevicesEnumerationRequest(ScriptExecutionContext& context, CompletionHandler&& completionHandler)
46     : ContextDestructionObserver(&context)
47     , m_completionHandler(WTFMove(completionHandler))
48 {
49 }
50
51 MediaDevicesEnumerationRequest::~MediaDevicesEnumerationRequest()
52 {
53 }
54
55 SecurityOrigin* MediaDevicesEnumerationRequest::userMediaDocumentOrigin() const
56 {
57     if (!scriptExecutionContext())
58         return nullptr;
59
60     return scriptExecutionContext()->securityOrigin();
61 }
62
63 SecurityOrigin* MediaDevicesEnumerationRequest::topLevelDocumentOrigin() const
64 {
65     if (!scriptExecutionContext())
66         return nullptr;
67
68     if (Frame* frame = downcast<Document>(*scriptExecutionContext()).frame()) {
69         if (frame->isMainFrame())
70             return nullptr;
71     }
72
73     return scriptExecutionContext()->topOrigin();
74 }
75
76 void MediaDevicesEnumerationRequest::contextDestroyed()
77 {
78     cancel();
79     ContextDestructionObserver::contextDestroyed();
80 }
81
82 void MediaDevicesEnumerationRequest::start()
83 {
84     ASSERT(scriptExecutionContext());
85
86     auto& document = downcast<Document>(*scriptExecutionContext());
87     auto* controller = UserMediaController::from(document.page());
88     if (!controller)
89         return;
90
91     Ref<MediaDevicesEnumerationRequest> protectedThis(*this);
92     controller->enumerateMediaDevices(*this);
93 }
94
95 void MediaDevicesEnumerationRequest::cancel()
96 {
97     m_completionHandler = nullptr;
98 }
99
100 void MediaDevicesEnumerationRequest::setDeviceInfo(const Vector<CaptureDevice>& deviceList, const String& deviceIdentifierHashSalt, bool originHasPersistentAccess)
101 {
102     m_deviceList = deviceList;
103     m_deviceIdentifierHashSalt = deviceIdentifierHashSalt;
104     m_originHasPersistentAccess = originHasPersistentAccess;
105
106     if (m_completionHandler)
107         m_completionHandler(m_deviceList, m_deviceIdentifierHashSalt, m_originHasPersistentAccess);
108     m_completionHandler = nullptr;
109 }
110
111 } // namespace WebCore
112
113 #endif // ENABLE(MEDIA_STREAM)