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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
|
// Copyright (c) 2012 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "chrome/browser/extensions/script_executor_impl.h"
#include "base/callback.h"
#include "base/logging.h"
#include "base/pickle.h"
#include "chrome/common/extensions/extension_messages.h"
#include "content/public/browser/render_view_host.h"
#include "content/public/browser/web_contents.h"
#include "content/public/browser/web_contents_observer.h"
#include "ipc/ipc_message.h"
#include "ipc/ipc_message_macros.h"
namespace extensions {
namespace {
const char* kRendererDestroyed = "The tab was closed.";
// A handler for a single injection request. On creation this will send the
// injection request to the renderer, and it will be destroyed after either the
// corresponding response comes from the renderer, or the renderer is destroyed.
class Handler : public content::WebContentsObserver {
public:
Handler(content::WebContents* web_contents,
const ExtensionMsg_ExecuteCode_Params& params,
const ScriptExecutor::ExecuteScriptCallback& callback)
: content::WebContentsObserver(web_contents),
request_id_(params.request_id),
callback_(callback) {
content::RenderViewHost* rvh = web_contents->GetRenderViewHost();
rvh->Send(new ExtensionMsg_ExecuteCode(rvh->GetRoutingID(), params));
}
virtual ~Handler() {}
virtual bool OnMessageReceived(const IPC::Message& message) OVERRIDE {
// Unpack by hand to check the request_id, since there may be multiple
// requests in flight but only one is for this.
if (message.type() != ExtensionHostMsg_ExecuteCodeFinished::ID)
return false;
int message_request_id;
PickleIterator iter(message);
CHECK(message.ReadInt(&iter, &message_request_id));
if (message_request_id != request_id_)
return false;
IPC_BEGIN_MESSAGE_MAP(Handler, message)
IPC_MESSAGE_HANDLER(ExtensionHostMsg_ExecuteCodeFinished,
OnExecuteCodeFinished)
IPC_END_MESSAGE_MAP()
return true;
}
virtual void WebContentsDestroyed(content::WebContents* tab) OVERRIDE {
callback_.Run(false, -1, kRendererDestroyed);
delete this;
}
private:
void OnExecuteCodeFinished(int request_id,
bool success,
int32 page_id,
const std::string& error) {
callback_.Run(success, page_id, error);
delete this;
}
int request_id_;
ScriptExecutor::ExecuteScriptCallback callback_;
};
} // namespace
ScriptExecutorImpl::ScriptExecutorImpl(
content::WebContents* web_contents)
: next_request_id_(0),
web_contents_(web_contents) {}
ScriptExecutorImpl::~ScriptExecutorImpl() {}
void ScriptExecutorImpl::ExecuteScript(
const std::string& extension_id,
ScriptExecutor::ScriptType script_type,
const std::string& code,
ScriptExecutor::FrameScope frame_scope,
UserScript::RunLocation run_at,
ScriptExecutor::WorldType world_type,
const ExecuteScriptCallback& callback) {
ExtensionMsg_ExecuteCode_Params params;
params.request_id = next_request_id_++;
params.extension_id = extension_id;
params.is_javascript = (script_type == JAVASCRIPT);
params.code = code;
params.all_frames = (frame_scope == ALL_FRAMES);
params.run_at = (int) run_at;
params.in_main_world = (world_type == MAIN_WORLD);
// Handler handles IPCs and deletes itself on completion.
new Handler(web_contents_, params, callback);
}
} // namespace extensions
|
