1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
5 #include "content/renderer/dom_automation_controller.h"
8 #include "base/bind_helpers.h"
9 #include "base/json/json_string_value_serializer.h"
10 #include "base/metrics/histogram.h"
11 #include "base/metrics/statistics_recorder.h"
12 #include "base/strings/string_util.h"
13 #include "content/common/child_process_messages.h"
14 #include "content/common/view_messages.h"
16 using webkit_glue::CppArgumentList;
17 using webkit_glue::CppVariant;
21 DomAutomationController::DomAutomationController()
23 routing_id_(MSG_ROUTING_NONE),
24 automation_id_(MSG_ROUTING_NONE) {
25 BindCallback("send", base::Bind(&DomAutomationController::Send,
26 base::Unretained(this)));
27 BindCallback("setAutomationId",
28 base::Bind(&DomAutomationController::SetAutomationId,
29 base::Unretained(this)));
30 BindCallback("sendJSON", base::Bind(&DomAutomationController::SendJSON,
31 base::Unretained(this)));
32 BindCallback("sendWithId", base::Bind(&DomAutomationController::SendWithId,
33 base::Unretained(this)));
36 void DomAutomationController::Send(const CppArgumentList& args,
38 if (args.size() != 1) {
43 if (automation_id_ == MSG_ROUTING_NONE) {
55 JSONStringValueSerializer serializer(&json);
56 scoped_ptr<base::Value> value;
58 // Warning: note that JSON officially requires the root-level object to be
59 // an object (e.g. {foo:3}) or an array, while here we're serializing
60 // strings, bools, etc. to "JSON". This only works because (a) the JSON
61 // writer is lenient, and (b) on the receiving side we wrap the JSON string
62 // in square brackets, converting it to an array, then parsing it and
63 // grabbing the 0th element to get the value out.
64 switch (args[0].type) {
65 case NPVariantType_String: {
66 value.reset(new base::StringValue(args[0].ToString()));
69 case NPVariantType_Bool: {
70 value.reset(new base::FundamentalValue(args[0].ToBoolean()));
73 case NPVariantType_Int32: {
74 value.reset(new base::FundamentalValue(args[0].ToInt32()));
77 case NPVariantType_Double: {
78 // The value that is sent back is an integer while it is treated
79 // as a double in this binding. The reason being that KJS treats
80 // any number value as a double. Refer for more details,
81 // chrome/third_party/webkit/src/JavaScriptCore/bindings/c/c_utility.cpp
82 value.reset(new base::FundamentalValue(args[0].ToInt32()));
91 if (!serializer.Serialize(*value)) {
96 bool succeeded = sender_->Send(
97 new ViewHostMsg_DomOperationResponse(routing_id_, json, automation_id_));
98 result->Set(succeeded);
100 automation_id_ = MSG_ROUTING_NONE;
103 void DomAutomationController::SendJSON(const CppArgumentList& args,
104 CppVariant* result) {
105 if (args.size() != 1) {
110 if (automation_id_ == MSG_ROUTING_NONE) {
121 if (args[0].type != NPVariantType_String) {
126 std::string json = args[0].ToString();
127 result->Set(sender_->Send(
128 new ViewHostMsg_DomOperationResponse(routing_id_, json, automation_id_)));
130 automation_id_ = MSG_ROUTING_NONE;
133 void DomAutomationController::SendWithId(const CppArgumentList& args,
134 CppVariant* result) {
135 if (args.size() != 2) {
146 if (!args[0].isNumber() || args[1].type != NPVariantType_String) {
151 result->Set(sender_->Send(
152 new ViewHostMsg_DomOperationResponse(routing_id_, args[1].ToString(),
153 args[0].ToInt32())));
156 void DomAutomationController::SetAutomationId(
157 const CppArgumentList& args, CppVariant* result) {
158 if (args.size() != 1) {
163 // The check here is for NumberType and not Int32 as
164 // KJS::JSType only defines a NumberType (no Int32)
165 if (!args[0].isNumber()) {
170 automation_id_ = args[0].ToInt32();
174 } // namespace content