1 files changed, 223 insertions, 0 deletions
diff --git a/Source/WebKit2/UIProcess/ChildProcessProxy.cpp b/Source/WebKit2/UIProcess/ChildProcessProxy.cpp
new file mode 100644
index 000000000..dd8aef550
--- /dev/null
+++ b/Source/WebKit2/UIProcess/ChildProcessProxy.cpp
@@ -0,0 +1,223 @@
+/*
+ * Copyright (C) 2012 Apple Inc. All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ * 1. Redistributions of source code must retain the above copyright
+ *    notice, this list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright
+ *    notice, this list of conditions and the following disclaimer in the
+ *    documentation and/or other materials provided with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
+ * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
+ * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
+ * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
+ * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
+ * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
+ * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
+ * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
+ * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
+ * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
+ * THE POSSIBILITY OF SUCH DAMAGE.
+ */
+
+#include "config.h"
+#include "ChildProcessProxy.h"
+
+#include "ChildProcessMessages.h"
+#include <wtf/RunLoop.h>
+
+namespace WebKit {
+
+ChildProcessProxy::ChildProcessProxy(bool alwaysRunsAtBackgroundPriority)
+    : m_alwaysRunsAtBackgroundPriority(alwaysRunsAtBackgroundPriority)
+{
+}
+
+ChildProcessProxy::~ChildProcessProxy()
+{
+    if (m_connection)
+        m_connection->invalidate();
+
+    if (m_processLauncher) {
+        m_processLauncher->invalidate();
+        m_processLauncher = nullptr;
+    }
+}
+
+void ChildProcessProxy::getLaunchOptions(ProcessLauncher::LaunchOptions& launchOptions)
+{
+    if (const char* userDirectorySuffix = getenv("DIRHELPER_USER_DIR_SUFFIX"))
+        launchOptions.extraInitializationData.add(ASCIILiteral("user-directory-suffix"), userDirectorySuffix);
+
+    if (m_alwaysRunsAtBackgroundPriority)
+        launchOptions.extraInitializationData.add(ASCIILiteral("always-runs-at-background-priority"), "true");
+
+#if ENABLE(DEVELOPER_MODE) && PLATFORM(GTK)
+    const char* varname;
+    switch (launchOptions.processType) {
+    case ProcessLauncher::ProcessType::Web:
+        varname = "WEB_PROCESS_CMD_PREFIX";
+        break;
+#if ENABLE(NETSCAPE_PLUGIN_API)
+    case ProcessLauncher::ProcessType::Plugin64:
+    case ProcessLauncher::ProcessType::Plugin32:
+        varname = "PLUGIN_PROCESS_CMD_PREFIX";
+        break;
+#endif
+    case ProcessLauncher::ProcessType::Network:
+        varname = "NETWORK_PROCESS_CMD_PREFIX";
+        break;
+#if ENABLE(DATABASE_PROCESS)
+    case ProcessLauncher::ProcessType::Database:
+        varname = "DATABASE_PROCESS_CMD_PREFIX";
+        break;
+#endif
+    }
+    const char* processCmdPrefix = getenv(varname);
+    if (processCmdPrefix && *processCmdPrefix)
+        launchOptions.processCmdPrefix = String::fromUTF8(processCmdPrefix);
+#endif // ENABLE(DEVELOPER_MODE) && PLATFORM(GTK)
+}
+
+void ChildProcessProxy::connect()
+{
+    ASSERT(!m_processLauncher);
+    ProcessLauncher::LaunchOptions launchOptions;
+    getLaunchOptions(launchOptions);
+    m_processLauncher = ProcessLauncher::create(this, launchOptions);
+}
+
+void ChildProcessProxy::terminate()
+{
+#if PLATFORM(COCOA)
+    if (m_connection && m_connection->kill())
+        return;
+#endif
+
+    // FIXME: We should really merge process launching into IPC connection creation and get rid of the process launcher.
+    if (m_processLauncher)
+        m_processLauncher->terminateProcess();
+}
+
+ChildProcessProxy::State ChildProcessProxy::state() const
+{
+    if (m_processLauncher && m_processLauncher->isLaunching())
+        return ChildProcessProxy::State::Launching;
+
+    if (!m_connection)
+        return ChildProcessProxy::State::Terminated;
+
+    return ChildProcessProxy::State::Running;
+}
+
+bool ChildProcessProxy::sendMessage(std::unique_ptr<IPC::Encoder> encoder, OptionSet<IPC::SendOption> sendOptions)
+{
+    switch (state()) {
+    case State::Launching:
+        // If we're waiting for the child process to launch, we need to stash away the messages so we can send them once we have a connection.
+        m_pendingMessages.append(std::make_pair(WTFMove(encoder), sendOptions));
+        return true;
+
+    case State::Running:
+        return connection()->sendMessage(WTFMove(encoder), sendOptions);
+
+    case State::Terminated:
+        return false;
+    }
+
+    return false;
+}
+
+void ChildProcessProxy::addMessageReceiver(IPC::StringReference messageReceiverName, IPC::MessageReceiver& messageReceiver)
+{
+    m_messageReceiverMap.addMessageReceiver(messageReceiverName, messageReceiver);
+}
+
+void ChildProcessProxy::addMessageReceiver(IPC::StringReference messageReceiverName, uint64_t destinationID, IPC::MessageReceiver& messageReceiver)
+{
+    m_messageReceiverMap.addMessageReceiver(messageReceiverName, destinationID, messageReceiver);
+}
+
+void ChildProcessProxy::removeMessageReceiver(IPC::StringReference messageReceiverName, uint64_t destinationID)
+{
+    m_messageReceiverMap.removeMessageReceiver(messageReceiverName, destinationID);
+}
+
+void ChildProcessProxy::removeMessageReceiver(IPC::StringReference messageReceiverName)
+{
+    m_messageReceiverMap.removeMessageReceiver(messageReceiverName);
+}
+
+bool ChildProcessProxy::dispatchMessage(IPC::Connection& connection, IPC::Decoder& decoder)
+{
+    return m_messageReceiverMap.dispatchMessage(connection, decoder);
+}
+
+bool ChildProcessProxy::dispatchSyncMessage(IPC::Connection& connection, IPC::Decoder& decoder, std::unique_ptr<IPC::Encoder>& replyEncoder)
+{
+    return m_messageReceiverMap.dispatchSyncMessage(connection, decoder, replyEncoder);
+}
+
+void ChildProcessProxy::didFinishLaunching(ProcessLauncher*, IPC::Connection::Identifier connectionIdentifier)
+{
+    ASSERT(!m_connection);
+
+    if (IPC::Connection::identifierIsNull(connectionIdentifier))
+        return;
+
+    m_connection = IPC::Connection::createServerConnection(connectionIdentifier, *this);
+#if PLATFORM(MAC) && __MAC_OS_X_VERSION_MIN_REQUIRED <= 101000
+    m_connection->setShouldCloseConnectionOnMachExceptions();
+#endif
+
+    connectionWillOpen(*m_connection);
+    m_connection->open();
+
+    for (size_t i = 0; i < m_pendingMessages.size(); ++i) {
+        std::unique_ptr<IPC::Encoder> message = WTFMove(m_pendingMessages[i].first);
+        OptionSet<IPC::SendOption> sendOptions = m_pendingMessages[i].second;
+        m_connection->sendMessage(WTFMove(message), sendOptions);
+    }
+
+    m_pendingMessages.clear();
+}
+
+void ChildProcessProxy::shutDownProcess()
+{
+    switch (state()) {
+    case State::Launching:
+        m_processLauncher->invalidate();
+        m_processLauncher = nullptr;
+        break;
+    case State::Running:
+#if PLATFORM(IOS)
+        // On iOS deploy a watchdog in the UI process, since the child process may be suspended.
+        // If 30s is insufficient for any outstanding activity to complete cleanly, then it will be killed.
+        ASSERT(m_connection);
+        m_connection->terminateSoon(30);
+#endif
+        break;
+    case State::Terminated:
+        return;
+    }
+
+    if (!m_connection)
+        return;
+
+    processWillShutDown(*m_connection);
+
+    if (canSendMessage())
+        send(Messages::ChildProcess::ShutDown(), 0);
+
+    m_connection->invalidate();
+    m_connection = nullptr;
+}
+
+void ChildProcessProxy::connectionWillOpen(IPC::Connection&)
+{
+}
+
+} // namespace WebKit