1 // Copyright 2013 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.
4
5 #include "chrome/browser/extensions/api/diagnostics/diagnostics_api.h"
6
7 #include "base/bind.h"
8 #include "base/callback.h"
9 #include "base/json/json_reader.h"
10 #include "base/strings/string_number_conversions.h"
11 #include "base/values.h"
12 #include "chromeos/dbus/dbus_thread_manager.h"
13 #include "chromeos/dbus/debug_daemon_client.h"
14
15 namespace extensions {
16
17 namespace {
18
19 const char kCount[] = "count";
20 const char kDefaultCount[] = "1";
21 const char kTTL[] = "ttl";
22 const char kTimeout[] = "timeout";
23 const char kSize[] = "size";
24
25 typedef base::Callback<void(
26 DiagnosticsSendPacketFunction::SendPacketResultCode result_code,
27 const std::string& ip,
28 double latency)>
29 SendPacketCallback;
30
ParseResult(const std::string & status,std::string * ip,double * latency)31 bool ParseResult(const std::string& status,
32 std::string* ip,
33 double* latency) {
34 // Parses the result and returns IP and latency.
35 scoped_ptr<base::Value> parsed_value(base::JSONReader::Read(status));
36 if (!parsed_value)
37 return false;
38
39 base::DictionaryValue* result = NULL;
40 if (!parsed_value->GetAsDictionary(&result) || result->size() != 1)
41 return false;
42
43 // Returns the first item.
44 base::DictionaryValue::Iterator iterator(*result);
45
46 const base::DictionaryValue* info;
47 if (!iterator.value().GetAsDictionary(&info))
48 return false;
49
50 if (!info->GetDouble("avg", latency))
51 return false;
52
53 *ip = iterator.key();
54 return true;
55 }
56
OnTestICMPCompleted(const SendPacketCallback & callback,bool succeeded,const std::string & status)57 void OnTestICMPCompleted(
58 const SendPacketCallback& callback,
59 bool succeeded,
60 const std::string& status) {
61 std::string ip;
62 double latency;
63 if (!succeeded || !ParseResult(status, &ip, &latency)) {
64 callback.Run(DiagnosticsSendPacketFunction::SEND_PACKET_FAILED, "", 0.0);
65 } else {
66 callback.Run(DiagnosticsSendPacketFunction::SEND_PACKET_OK,
67 ip,
68 latency);
69 }
70 }
71
72 } // namespace
73
AsyncWorkStart()74 void DiagnosticsSendPacketFunction::AsyncWorkStart() {
75 std::map<std::string, std::string> config;
76 config[kCount] = kDefaultCount;
77 if (parameters_->options.ttl)
78 config[kTTL] = base::IntToString(*parameters_->options.ttl);
79 if (parameters_->options.timeout)
80 config[kTimeout] = base::IntToString(*parameters_->options.timeout);
81 if (parameters_->options.size)
82 config[kSize] = base::IntToString(*parameters_->options.size);
83
84 chromeos::DBusThreadManager::Get()->GetDebugDaemonClient()->
85 TestICMPWithOptions(
86 parameters_->options.ip, config,
87 base::Bind(
88 OnTestICMPCompleted,
89 base::Bind(&DiagnosticsSendPacketFunction::OnCompleted, this)));
90 }
91
92 } // namespace extensions
93