[Cocoa] WKCustomProtocolLoader should store a WeakPtr to its LegacyCustomProtocolMana...
[WebKit-https.git] / Source / WebKit / UIProcess / WebProcessLifetimeObserver.cpp
1 /*
2  * Copyright (C) 2015 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 "WebProcessLifetimeObserver.h"
28
29 #include "WebPageProxy.h"
30 #include "WebProcessProxy.h"
31
32 namespace WebKit {
33
34 WebProcessLifetimeObserver::WebProcessLifetimeObserver()
35 {
36 }
37
38 WebProcessLifetimeObserver::~WebProcessLifetimeObserver()
39 {
40 }
41
42 void WebProcessLifetimeObserver::addWebPage(WebPageProxy& webPageProxy, WebProcessProxy& process)
43 {
44     ASSERT(process.state() == WebProcessProxy::State::Running);
45
46     if (m_processes.add(&process).isNewEntry)
47         webProcessWillOpenConnection(process, *process.connection());
48
49     webPageWillOpenConnection(webPageProxy, *process.connection());
50 }
51
52 void WebProcessLifetimeObserver::removeWebPage(WebPageProxy& webPageProxy, WebProcessProxy& process)
53 {
54     // FIXME: This should assert that the page is either closed or that the process is no longer running,
55     // but we have to make sure that removeWebPage is called after the connection has been removed in that case.
56     ASSERT(m_processes.contains(&process));
57
58     webPageDidCloseConnection(webPageProxy, *process.connection());
59
60     if (m_processes.remove(&process))
61         webProcessDidCloseConnection(process, *process.connection());
62 }
63
64 WTF::IteratorRange<HashCountedSet<WebProcessProxy*>::const_iterator::Keys> WebProcessLifetimeObserver::processes() const
65 {
66     ASSERT(std::all_of(m_processes.begin().keys(), m_processes.end().keys(), [](WebProcessProxy* process) {
67         return process->state() == WebProcessProxy::State::Running;
68     }));
69
70     return makeIteratorRange(m_processes.begin().keys(), m_processes.end().keys());
71 }
72
73 }