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 "PluginProcessManager.h"
28
29#if ENABLE(PLUGIN_PROCESS)
30
31#include "PluginProcessProxy.h"
32#include "WebContext.h"
33#include <wtf/CryptographicallyRandomNumber.h>
34#include <wtf/StdLibExtras.h>
35#include <wtf/text/WTFString.h>
36
37namespace WebKit {
38
39PluginProcessManager& PluginProcessManager::shared()
40{
41    DEFINE_STATIC_LOCAL(PluginProcessManager, pluginProcessManager, ());
42    return pluginProcessManager;
43}
44
45PluginProcessManager::PluginProcessManager()
46{
47}
48
49uint64_t PluginProcessManager::pluginProcessToken(const PluginModuleInfo& pluginModuleInfo, PluginProcessType pluginProcessType, PluginProcessSandboxPolicy pluginProcessSandboxPolicy)
50{
51    // See if we know this token already.
52    for (size_t i = 0; i < m_pluginProcessTokens.size(); ++i) {
53        const PluginProcessAttributes& attributes = m_pluginProcessTokens[i].first;
54
55        if (attributes.moduleInfo.path == pluginModuleInfo.path
56            && attributes.processType == pluginProcessType
57            && attributes.sandboxPolicy == pluginProcessSandboxPolicy)
58            return m_pluginProcessTokens[i].second;
59    }
60
61    uint64_t token;
62    while (true) {
63        cryptographicallyRandomValues(&token, sizeof(token));
64
65        if (m_knownTokens.isValidValue(token) && !m_knownTokens.contains(token))
66            break;
67    }
68
69    PluginProcessAttributes attributes;
70    attributes.moduleInfo = pluginModuleInfo;
71    attributes.processType = pluginProcessType;
72    attributes.sandboxPolicy = pluginProcessSandboxPolicy;
73
74    m_pluginProcessTokens.append(std::make_pair(std::move(attributes), token));
75    m_knownTokens.add(token);
76
77    return token;
78}
79
80void PluginProcessManager::getPluginProcessConnection(uint64_t pluginProcessToken, PassRefPtr<Messages::WebProcessProxy::GetPluginProcessConnection::DelayedReply> reply)
81{
82    ASSERT(pluginProcessToken);
83
84    PluginProcessProxy* pluginProcess = getOrCreatePluginProcess(pluginProcessToken);
85    pluginProcess->getPluginProcessConnection(reply);
86}
87
88void PluginProcessManager::removePluginProcessProxy(PluginProcessProxy* pluginProcessProxy)
89{
90    size_t vectorIndex = m_pluginProcesses.find(pluginProcessProxy);
91    ASSERT(vectorIndex != notFound);
92
93    m_pluginProcesses.remove(vectorIndex);
94}
95
96void PluginProcessManager::getSitesWithData(const PluginModuleInfo& plugin, WebPluginSiteDataManager* webPluginSiteDataManager, uint64_t callbackID)
97{
98    PluginProcessProxy* pluginProcess = getOrCreatePluginProcess(pluginProcessToken(plugin, PluginProcessTypeNormal, PluginProcessSandboxPolicyNormal));
99    pluginProcess->getSitesWithData(webPluginSiteDataManager, callbackID);
100}
101
102void PluginProcessManager::clearSiteData(const PluginModuleInfo& plugin, WebPluginSiteDataManager* webPluginSiteDataManager, const Vector<String>& sites, uint64_t flags, uint64_t maxAgeInSeconds, uint64_t callbackID)
103{
104    PluginProcessProxy* pluginProcess = getOrCreatePluginProcess(pluginProcessToken(plugin, PluginProcessTypeNormal, PluginProcessSandboxPolicyNormal));
105    pluginProcess->clearSiteData(webPluginSiteDataManager, sites, flags, maxAgeInSeconds, callbackID);
106}
107
108PluginProcessProxy* PluginProcessManager::getOrCreatePluginProcess(uint64_t pluginProcessToken)
109{
110    for (size_t i = 0; i < m_pluginProcesses.size(); ++i) {
111        if (m_pluginProcesses[i]->pluginProcessToken() == pluginProcessToken)
112            return m_pluginProcesses[i].get();
113    }
114
115    for (size_t i = 0; i < m_pluginProcessTokens.size(); ++i) {
116        auto& attributesAndToken = m_pluginProcessTokens[i];
117        if (attributesAndToken.second == pluginProcessToken) {
118            RefPtr<PluginProcessProxy> pluginProcess = PluginProcessProxy::create(this, attributesAndToken.first, attributesAndToken.second);
119            PluginProcessProxy* pluginProcessPtr = pluginProcess.get();
120
121            m_pluginProcesses.append(pluginProcess.release());
122            return pluginProcessPtr;
123        }
124    }
125
126    return nullptr;
127}
128
129} // namespace WebKit
130
131#endif // ENABLE(PLUGIN_PROCESS)
132