1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
|
// Copyright 2019 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "extensions/browser/api/messaging/channel_endpoint.h"
#include "content/public/browser/child_process_host.h"
#include "content/public/browser/render_frame_host.h"
#include "content/public/browser/render_process_host.h"
#include "extensions/browser/process_manager.h"
#include "extensions/common/constants.h"
namespace extensions {
ChannelEndpoint::ChannelEndpoint(content::BrowserContext* browser_context,
int render_process_id,
const PortContext& port_context)
: browser_context_(browser_context),
render_process_id_(render_process_id),
port_context_(port_context) {
// Context must be exclusive to render frame or worker.
DCHECK(port_context.is_for_service_worker() ^
port_context.is_for_render_frame());
}
// For native message endpoint.
ChannelEndpoint::ChannelEndpoint(content::BrowserContext* browser_context)
: browser_context_(browser_context),
render_process_id_(content::ChildProcessHost::kInvalidUniqueID),
port_context_(PortContext::ForNativeHost()) {}
bool ChannelEndpoint::is_for_service_worker() const {
return port_context_.is_for_service_worker();
}
bool ChannelEndpoint::is_for_render_frame() const {
return port_context_.frame.has_value();
}
bool ChannelEndpoint::is_for_native_host() const {
return !port_context_.is_for_render_frame() &&
!port_context_.is_for_service_worker();
}
content::RenderFrameHost* ChannelEndpoint::GetRenderFrameHost() const {
DCHECK(port_context_.is_for_render_frame());
return content::RenderFrameHost::FromID(render_process_id_,
port_context_.frame->routing_id);
}
WorkerId ChannelEndpoint::GetWorkerId() const {
DCHECK(port_context_.is_for_service_worker());
return {port_context_.worker->extension_id, render_process_id_,
port_context_.worker->version_id, port_context_.worker->thread_id};
}
bool ChannelEndpoint::IsValid() const {
if (is_for_service_worker()) {
return ProcessManager::Get(browser_context())
->HasServiceWorker(GetWorkerId());
}
if (is_for_render_frame()) {
return GetRenderFrameHost() != nullptr;
}
DCHECK(is_for_native_host());
return true;
}
} // namespace extensions
|
