Adopt new assertion SPI for process suppression on Mac
[WebKit-https.git] / Source / WebKit2 / Shared / ChildProcess.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 "ChildProcess.h"
28
29 #if !OS(WINDOWS)
30 #include <unistd.h>
31 #endif
32
33 using namespace WebCore;
34
35 namespace WebKit {
36
37 void ChildProcess::disableTermination()
38 {
39     m_terminationCounter++;
40     m_terminationTimer.stop();
41 }
42
43 void ChildProcess::enableTermination()
44 {
45     ASSERT(m_terminationCounter > 0);
46     m_terminationCounter--;
47
48     if (m_terminationCounter)
49         return;
50
51     if (!m_terminationTimeout) {
52         terminationTimerFired();
53         return;
54     }
55
56     m_terminationTimer.startOneShot(m_terminationTimeout);
57 }
58
59 ChildProcess::ChildProcess()
60     : m_terminationTimeout(0)
61     , m_terminationCounter(0)
62     , m_terminationTimer(RunLoop::main(), this, &ChildProcess::terminationTimerFired)
63 {
64     // FIXME: The termination timer should not be scheduled on the main run loop.
65     // It won't work with the threaded mode, but it's not really useful anyway as is.
66     
67     platformInitialize();
68 }
69
70 ChildProcess::~ChildProcess()
71 {
72 }
73
74 void ChildProcess::terminationTimerFired()
75 {
76     if (!shouldTerminate())
77         return;
78
79     terminate();
80 }
81
82 void ChildProcess::terminate()
83 {
84     RunLoop::main()->stop();
85 }
86
87 NO_RETURN static void watchdogCallback()
88 {
89     // We use _exit here since the watchdog callback is called from another thread and we don't want 
90     // global destructors or atexit handlers to be called from this thread while the main thread is busy
91     // doing its thing.
92     _exit(EXIT_FAILURE);
93 }
94
95 void ChildProcess::didCloseOnConnectionWorkQueue(WorkQueue& workQueue, CoreIPC::Connection*)
96 {
97     // If the connection has been closed and we haven't responded in the main thread for 10 seconds
98     // the process will exit forcibly.
99     static const double watchdogDelay = 10.0;
100
101     workQueue.dispatchAfterDelay(bind(static_cast<void(*)()>(watchdogCallback)), watchdogDelay);
102 }
103
104 #if !PLATFORM(MAC)
105 void ChildProcess::platformInitialize()
106 {
107 }
108 #endif
109
110 } // namespace WebKit