Unreviewed, roll out http://trac.webkit.org/changeset/187972.
[WebKit-https.git] / Source / WebKit2 / WebProcess / Plugins / PluginProcessConnectionManager.cpp
1 /*
2  * Copyright (C) 2010 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 "PluginProcessConnectionManager.h"
28
29 #if ENABLE(NETSCAPE_PLUGIN_API)
30
31 #include "ArgumentDecoder.h"
32 #include "ArgumentEncoder.h"
33 #include "PluginProcessConnection.h"
34 #include "PluginProcessConnectionManagerMessages.h"
35 #include "WebCoreArgumentCoders.h"
36 #include "WebProcess.h"
37 #include "WebProcessProxyMessages.h"
38
39 #if OS(DARWIN)
40 #include "MachPort.h"
41 #endif
42
43 namespace WebKit {
44
45 Ref<PluginProcessConnectionManager> PluginProcessConnectionManager::create()
46 {
47     return adoptRef(*new PluginProcessConnectionManager);
48 }
49
50 PluginProcessConnectionManager::PluginProcessConnectionManager()
51     : m_queue(WorkQueue::create("com.apple.WebKit.PluginProcessConnectionManager"))
52 {
53 }
54
55 PluginProcessConnectionManager::~PluginProcessConnectionManager()
56 {
57 }
58
59 void PluginProcessConnectionManager::initializeConnection(IPC::Connection* connection)
60 {
61     connection->addWorkQueueMessageReceiver(Messages::PluginProcessConnectionManager::messageReceiverName(), &m_queue.get(), this);
62 }
63
64 PluginProcessConnection* PluginProcessConnectionManager::getPluginProcessConnection(uint64_t pluginProcessToken)
65 {
66     for (size_t i = 0; i < m_pluginProcessConnections.size(); ++i) {
67         if (m_pluginProcessConnections[i]->pluginProcessToken() == pluginProcessToken)
68             return m_pluginProcessConnections[i].get();
69     }
70
71     IPC::Attachment encodedConnectionIdentifier;
72     bool supportsAsynchronousInitialization;
73     if (!WebProcess::singleton().parentProcessConnection()->sendSync(Messages::WebProcessProxy::GetPluginProcessConnection(pluginProcessToken),
74                                                      Messages::WebProcessProxy::GetPluginProcessConnection::Reply(encodedConnectionIdentifier, supportsAsynchronousInitialization), 0))
75         return 0;
76
77 #if OS(DARWIN)
78     IPC::Connection::Identifier connectionIdentifier(encodedConnectionIdentifier.port());
79 #elif USE(UNIX_DOMAIN_SOCKETS)
80     IPC::Connection::Identifier connectionIdentifier = encodedConnectionIdentifier.releaseFileDescriptor();
81 #endif
82     if (IPC::Connection::identifierIsNull(connectionIdentifier))
83         return nullptr;
84
85     RefPtr<PluginProcessConnection> pluginProcessConnection = PluginProcessConnection::create(this, pluginProcessToken, connectionIdentifier, supportsAsynchronousInitialization);
86     m_pluginProcessConnections.append(pluginProcessConnection);
87
88     {
89         MutexLocker locker(m_tokensAndConnectionsMutex);
90         ASSERT(!m_tokensAndConnections.contains(pluginProcessToken));
91
92         m_tokensAndConnections.set(pluginProcessToken, pluginProcessConnection->connection());
93     }
94
95     return pluginProcessConnection.get();
96 }
97
98 void PluginProcessConnectionManager::removePluginProcessConnection(PluginProcessConnection* pluginProcessConnection)
99 {
100     size_t vectorIndex = m_pluginProcessConnections.find(pluginProcessConnection);
101     ASSERT(vectorIndex != notFound);
102
103     {
104         MutexLocker locker(m_tokensAndConnectionsMutex);
105         ASSERT(m_tokensAndConnections.contains(pluginProcessConnection->pluginProcessToken()));
106         
107         m_tokensAndConnections.remove(pluginProcessConnection->pluginProcessToken());
108     }
109
110     m_pluginProcessConnections.remove(vectorIndex);
111 }
112
113 void PluginProcessConnectionManager::pluginProcessCrashed(uint64_t pluginProcessToken)
114 {
115     MutexLocker locker(m_tokensAndConnectionsMutex);
116     IPC::Connection* connection = m_tokensAndConnections.get(pluginProcessToken);
117
118     // It's OK for connection to be null here; it will happen if this web process doesn't know
119     // anything about the plug-in process.
120     if (!connection)
121         return;
122
123     connection->postConnectionDidCloseOnConnectionWorkQueue();
124 }
125
126 } // namespace WebKit
127
128 #endif // ENABLE(NETSCAPE_PLUGIN_API)