Bertrand SIMONNET | 52e5b99 | 2015-08-10 15:18:00 -0700 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (C) 2015 The Android Open Source Project |
| 3 | * |
| 4 | * Licensed under the Apache License, Version 2.0 (the "License"); |
| 5 | * you may not use this file except in compliance with the License. |
| 6 | * You may obtain a copy of the License at |
| 7 | * |
| 8 | * http://www.apache.org/licenses/LICENSE-2.0 |
| 9 | * |
| 10 | * Unless required by applicable law or agreed to in writing, software |
| 11 | * distributed under the License is distributed on an "AS IS" BASIS, |
| 12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 13 | * See the License for the specific language governing permissions and |
| 14 | * limitations under the License. |
| 15 | */ |
Darin Petkov | 65b0146 | 2010-04-14 13:32:20 -0700 | [diff] [blame] | 16 | |
Darin Petkov | 4fcb2ac | 2010-04-15 16:40:23 -0700 | [diff] [blame] | 17 | #include <cstdio> |
Darin Petkov | 65b0146 | 2010-04-14 13:32:20 -0700 | [diff] [blame] | 18 | #include <cstdlib> |
Darin Petkov | 65b0146 | 2010-04-14 13:32:20 -0700 | [diff] [blame] | 19 | |
James Hawkins | 5f64600 | 2015-09-08 15:18:17 -0700 | [diff] [blame] | 20 | #include <base/memory/scoped_vector.h> |
| 21 | |
| 22 | #include "constants.h" |
Bertrand SIMONNET | e6cfd64 | 2014-07-09 16:35:23 -0700 | [diff] [blame] | 23 | #include "metrics/metrics_library.h" |
James Hawkins | 5f64600 | 2015-09-08 15:18:17 -0700 | [diff] [blame] | 24 | #include "serialization/metric_sample.h" |
| 25 | #include "serialization/serialization_utils.h" |
Darin Petkov | 65b0146 | 2010-04-14 13:32:20 -0700 | [diff] [blame] | 26 | |
Luigi Semenzato | a7ebeb3 | 2013-03-19 15:02:42 -0700 | [diff] [blame] | 27 | enum Mode { |
James Hawkins | 5f64600 | 2015-09-08 15:18:17 -0700 | [diff] [blame] | 28 | kModeDumpLogs, |
Luigi Semenzato | a7ebeb3 | 2013-03-19 15:02:42 -0700 | [diff] [blame] | 29 | kModeSendSample, |
| 30 | kModeSendEnumSample, |
| 31 | kModeSendSparseSample, |
| 32 | kModeSendUserAction, |
| 33 | kModeSendCrosEvent, |
| 34 | kModeHasConsent, |
| 35 | kModeIsGuestMode, |
| 36 | }; |
| 37 | |
Ken Mixter | eafbbdf | 2010-10-01 15:38:42 -0700 | [diff] [blame] | 38 | void ShowUsage() { |
| 39 | fprintf(stderr, |
Bertrand SIMONNET | 475dfa6 | 2015-08-04 14:10:10 -0700 | [diff] [blame] | 40 | "Usage: metrics_client [-t] name sample min max nbuckets\n" |
| 41 | " metrics_client -e name sample max\n" |
| 42 | " metrics_client -s name sample\n" |
| 43 | " metrics_client -v event\n" |
Darin Petkov | ed82485 | 2011-01-06 10:51:47 -0800 | [diff] [blame] | 44 | " metrics_client -u action\n" |
James Hawkins | 5f64600 | 2015-09-08 15:18:17 -0700 | [diff] [blame] | 45 | " metrics_client [-cdg]\n" |
Ken Mixter | eafbbdf | 2010-10-01 15:38:42 -0700 | [diff] [blame] | 46 | "\n" |
Bertrand SIMONNET | 475dfa6 | 2015-08-04 14:10:10 -0700 | [diff] [blame] | 47 | " default: send metric with integer values \n" |
Ken Mixter | eafbbdf | 2010-10-01 15:38:42 -0700 | [diff] [blame] | 48 | " |min| > 0, |min| <= sample < |max|\n" |
Thiemo Nagel | 0b8cc1c | 2014-08-21 15:00:50 +0200 | [diff] [blame] | 49 | " -c: return exit status 0 if user consents to stats, 1 otherwise,\n" |
| 50 | " in guest mode always return 1\n" |
James Hawkins | 5f64600 | 2015-09-08 15:18:17 -0700 | [diff] [blame] | 51 | " -d: dump cached logs to the console\n" |
Ken Mixter | eafbbdf | 2010-10-01 15:38:42 -0700 | [diff] [blame] | 52 | " -e: send linear/enumeration histogram data\n" |
| 53 | " -g: return exit status 0 if machine in guest mode, 1 otherwise\n" |
Luigi Semenzato | a7ebeb3 | 2013-03-19 15:02:42 -0700 | [diff] [blame] | 54 | " -s: send a sparse histogram sample\n" |
Darin Petkov | ed82485 | 2011-01-06 10:51:47 -0800 | [diff] [blame] | 55 | " -t: convert sample from double seconds to int milliseconds\n" |
Luigi Semenzato | 3268422 | 2013-03-13 10:53:55 -0700 | [diff] [blame] | 56 | " -u: send a user action to Chrome\n" |
Bertrand SIMONNET | e6cfd64 | 2014-07-09 16:35:23 -0700 | [diff] [blame] | 57 | " -v: send a Platform.CrOSEvent enum histogram sample\n"); |
Ken Mixter | eafbbdf | 2010-10-01 15:38:42 -0700 | [diff] [blame] | 58 | exit(1); |
| 59 | } |
Darin Petkov | 65b0146 | 2010-04-14 13:32:20 -0700 | [diff] [blame] | 60 | |
Luigi Semenzato | d8abf55 | 2014-03-27 14:19:06 -0700 | [diff] [blame] | 61 | static int ParseInt(const char *arg) { |
| 62 | char *endptr; |
| 63 | int value = strtol(arg, &endptr, 0); |
| 64 | if (*endptr != '\0') { |
| 65 | fprintf(stderr, "metrics client: bad integer \"%s\"\n", arg); |
| 66 | ShowUsage(); |
| 67 | } |
| 68 | return value; |
| 69 | } |
| 70 | |
| 71 | static double ParseDouble(const char *arg) { |
| 72 | char *endptr; |
| 73 | double value = strtod(arg, &endptr); |
| 74 | if (*endptr != '\0') { |
| 75 | fprintf(stderr, "metrics client: bad double \"%s\"\n", arg); |
| 76 | ShowUsage(); |
| 77 | } |
| 78 | return value; |
| 79 | } |
| 80 | |
Ken Mixter | eafbbdf | 2010-10-01 15:38:42 -0700 | [diff] [blame] | 81 | static int SendStats(char* argv[], |
| 82 | int name_index, |
Luigi Semenzato | a7ebeb3 | 2013-03-19 15:02:42 -0700 | [diff] [blame] | 83 | enum Mode mode, |
Bertrand SIMONNET | 475dfa6 | 2015-08-04 14:10:10 -0700 | [diff] [blame] | 84 | bool secs_to_msecs) { |
Darin Petkov | c2526a1 | 2010-04-21 14:24:04 -0700 | [diff] [blame] | 85 | const char* name = argv[name_index]; |
| 86 | int sample; |
Darin Petkov | 4fcb2ac | 2010-04-15 16:40:23 -0700 | [diff] [blame] | 87 | if (secs_to_msecs) { |
Luigi Semenzato | d8abf55 | 2014-03-27 14:19:06 -0700 | [diff] [blame] | 88 | sample = static_cast<int>(ParseDouble(argv[name_index + 1]) * 1000.0); |
Darin Petkov | 4fcb2ac | 2010-04-15 16:40:23 -0700 | [diff] [blame] | 89 | } else { |
Luigi Semenzato | d8abf55 | 2014-03-27 14:19:06 -0700 | [diff] [blame] | 90 | sample = ParseInt(argv[name_index + 1]); |
Darin Petkov | 4fcb2ac | 2010-04-15 16:40:23 -0700 | [diff] [blame] | 91 | } |
| 92 | |
Bertrand SIMONNET | 475dfa6 | 2015-08-04 14:10:10 -0700 | [diff] [blame] | 93 | MetricsLibrary metrics_lib; |
| 94 | metrics_lib.Init(); |
| 95 | if (mode == kModeSendSparseSample) { |
| 96 | metrics_lib.SendSparseToUMA(name, sample); |
| 97 | } else if (mode == kModeSendEnumSample) { |
| 98 | int max = ParseInt(argv[name_index + 2]); |
| 99 | metrics_lib.SendEnumToUMA(name, sample, max); |
| 100 | } else { |
| 101 | int min = ParseInt(argv[name_index + 2]); |
| 102 | int max = ParseInt(argv[name_index + 3]); |
| 103 | int nbuckets = ParseInt(argv[name_index + 4]); |
| 104 | metrics_lib.SendToUMA(name, sample, min, max, nbuckets); |
Darin Petkov | 65b0146 | 2010-04-14 13:32:20 -0700 | [diff] [blame] | 105 | } |
| 106 | return 0; |
| 107 | } |
Ken Mixter | eafbbdf | 2010-10-01 15:38:42 -0700 | [diff] [blame] | 108 | |
Darin Petkov | ed82485 | 2011-01-06 10:51:47 -0800 | [diff] [blame] | 109 | static int SendUserAction(char* argv[], int action_index) { |
| 110 | const char* action = argv[action_index]; |
| 111 | MetricsLibrary metrics_lib; |
| 112 | metrics_lib.Init(); |
| 113 | metrics_lib.SendUserActionToUMA(action); |
| 114 | return 0; |
| 115 | } |
| 116 | |
Luigi Semenzato | 3268422 | 2013-03-13 10:53:55 -0700 | [diff] [blame] | 117 | static int SendCrosEvent(char* argv[], int action_index) { |
| 118 | const char* event = argv[action_index]; |
| 119 | bool result; |
| 120 | MetricsLibrary metrics_lib; |
| 121 | metrics_lib.Init(); |
| 122 | result = metrics_lib.SendCrosEventToUMA(event); |
| 123 | if (!result) { |
| 124 | fprintf(stderr, "metrics_client: could not send event %s\n", event); |
| 125 | return 1; |
| 126 | } |
| 127 | return 0; |
| 128 | } |
| 129 | |
Ken Mixter | eafbbdf | 2010-10-01 15:38:42 -0700 | [diff] [blame] | 130 | static int HasConsent() { |
| 131 | MetricsLibrary metrics_lib; |
| 132 | metrics_lib.Init(); |
| 133 | return metrics_lib.AreMetricsEnabled() ? 0 : 1; |
| 134 | } |
| 135 | |
| 136 | static int IsGuestMode() { |
| 137 | MetricsLibrary metrics_lib; |
| 138 | metrics_lib.Init(); |
| 139 | return metrics_lib.IsGuestMode() ? 0 : 1; |
| 140 | } |
| 141 | |
James Hawkins | 5f64600 | 2015-09-08 15:18:17 -0700 | [diff] [blame] | 142 | static int DumpLogs() { |
Bertrand SIMONNET | 2765d0a | 2015-09-09 10:38:20 -0700 | [diff] [blame] | 143 | base::FilePath events_file = base::FilePath( |
| 144 | metrics::kMetricsDirectory).Append(metrics::kMetricsEventsFileName); |
| 145 | printf("Metrics from %s\n\n", events_file.value().data()); |
James Hawkins | 5f64600 | 2015-09-08 15:18:17 -0700 | [diff] [blame] | 146 | |
| 147 | ScopedVector<metrics::MetricSample> metrics; |
Bertrand SIMONNET | 2765d0a | 2015-09-09 10:38:20 -0700 | [diff] [blame] | 148 | metrics::SerializationUtils::ReadMetricsFromFile(events_file.value(), |
| 149 | &metrics); |
James Hawkins | 5f64600 | 2015-09-08 15:18:17 -0700 | [diff] [blame] | 150 | |
| 151 | for (ScopedVector<metrics::MetricSample>::const_iterator i = metrics.begin(); |
| 152 | i != metrics.end(); ++i) { |
| 153 | const metrics::MetricSample* sample = *i; |
| 154 | printf("name: %s\t", sample->name().c_str()); |
| 155 | printf("type: "); |
| 156 | |
| 157 | switch (sample->type()) { |
| 158 | case metrics::MetricSample::CRASH: |
| 159 | printf("CRASH"); |
| 160 | break; |
| 161 | case metrics::MetricSample::HISTOGRAM: |
| 162 | printf("HISTOGRAM"); |
| 163 | break; |
| 164 | case metrics::MetricSample::LINEAR_HISTOGRAM: |
| 165 | printf("LINEAR_HISTOGRAM"); |
| 166 | break; |
| 167 | case metrics::MetricSample::SPARSE_HISTOGRAM: |
| 168 | printf("SPARSE_HISTOGRAM"); |
| 169 | break; |
| 170 | case metrics::MetricSample::USER_ACTION: |
| 171 | printf("USER_ACTION"); |
| 172 | break; |
| 173 | } |
| 174 | |
| 175 | printf("\n"); |
| 176 | } |
| 177 | |
| 178 | return 0; |
| 179 | } |
| 180 | |
Ken Mixter | eafbbdf | 2010-10-01 15:38:42 -0700 | [diff] [blame] | 181 | int main(int argc, char** argv) { |
Luigi Semenzato | a7ebeb3 | 2013-03-19 15:02:42 -0700 | [diff] [blame] | 182 | enum Mode mode = kModeSendSample; |
Ken Mixter | eafbbdf | 2010-10-01 15:38:42 -0700 | [diff] [blame] | 183 | bool secs_to_msecs = false; |
Ken Mixter | eafbbdf | 2010-10-01 15:38:42 -0700 | [diff] [blame] | 184 | |
| 185 | // Parse arguments |
| 186 | int flag; |
James Hawkins | 5f64600 | 2015-09-08 15:18:17 -0700 | [diff] [blame] | 187 | while ((flag = getopt(argc, argv, "abcdegstuv")) != -1) { |
Ken Mixter | eafbbdf | 2010-10-01 15:38:42 -0700 | [diff] [blame] | 188 | switch (flag) { |
Ken Mixter | eafbbdf | 2010-10-01 15:38:42 -0700 | [diff] [blame] | 189 | case 'c': |
| 190 | mode = kModeHasConsent; |
| 191 | break; |
James Hawkins | 5f64600 | 2015-09-08 15:18:17 -0700 | [diff] [blame] | 192 | case 'd': |
| 193 | mode = kModeDumpLogs; |
| 194 | break; |
Ken Mixter | eafbbdf | 2010-10-01 15:38:42 -0700 | [diff] [blame] | 195 | case 'e': |
Luigi Semenzato | a7ebeb3 | 2013-03-19 15:02:42 -0700 | [diff] [blame] | 196 | mode = kModeSendEnumSample; |
Ken Mixter | eafbbdf | 2010-10-01 15:38:42 -0700 | [diff] [blame] | 197 | break; |
| 198 | case 'g': |
| 199 | mode = kModeIsGuestMode; |
| 200 | break; |
Luigi Semenzato | a7ebeb3 | 2013-03-19 15:02:42 -0700 | [diff] [blame] | 201 | case 's': |
| 202 | mode = kModeSendSparseSample; |
| 203 | break; |
Ken Mixter | eafbbdf | 2010-10-01 15:38:42 -0700 | [diff] [blame] | 204 | case 't': |
| 205 | secs_to_msecs = true; |
| 206 | break; |
Darin Petkov | ed82485 | 2011-01-06 10:51:47 -0800 | [diff] [blame] | 207 | case 'u': |
| 208 | mode = kModeSendUserAction; |
| 209 | break; |
Luigi Semenzato | 3268422 | 2013-03-13 10:53:55 -0700 | [diff] [blame] | 210 | case 'v': |
| 211 | mode = kModeSendCrosEvent; |
| 212 | break; |
Ken Mixter | eafbbdf | 2010-10-01 15:38:42 -0700 | [diff] [blame] | 213 | default: |
Darin Petkov | 8032dd0 | 2011-05-09 16:33:19 -0700 | [diff] [blame] | 214 | ShowUsage(); |
Ken Mixter | eafbbdf | 2010-10-01 15:38:42 -0700 | [diff] [blame] | 215 | break; |
| 216 | } |
| 217 | } |
Darin Petkov | ed82485 | 2011-01-06 10:51:47 -0800 | [diff] [blame] | 218 | int arg_index = optind; |
Ken Mixter | eafbbdf | 2010-10-01 15:38:42 -0700 | [diff] [blame] | 219 | |
| 220 | int expected_args = 0; |
Luigi Semenzato | a7ebeb3 | 2013-03-19 15:02:42 -0700 | [diff] [blame] | 221 | if (mode == kModeSendSample) |
| 222 | expected_args = 5; |
| 223 | else if (mode == kModeSendEnumSample) |
| 224 | expected_args = 3; |
| 225 | else if (mode == kModeSendSparseSample) |
| 226 | expected_args = 2; |
Darin Petkov | ed82485 | 2011-01-06 10:51:47 -0800 | [diff] [blame] | 227 | else if (mode == kModeSendUserAction) |
| 228 | expected_args = 1; |
Luigi Semenzato | 3268422 | 2013-03-13 10:53:55 -0700 | [diff] [blame] | 229 | else if (mode == kModeSendCrosEvent) |
| 230 | expected_args = 1; |
Ken Mixter | eafbbdf | 2010-10-01 15:38:42 -0700 | [diff] [blame] | 231 | |
Darin Petkov | ed82485 | 2011-01-06 10:51:47 -0800 | [diff] [blame] | 232 | if ((arg_index + expected_args) != argc) { |
Ken Mixter | eafbbdf | 2010-10-01 15:38:42 -0700 | [diff] [blame] | 233 | ShowUsage(); |
| 234 | } |
| 235 | |
Bertrand SIMONNET | e6cfd64 | 2014-07-09 16:35:23 -0700 | [diff] [blame] | 236 | switch (mode) { |
Luigi Semenzato | a7ebeb3 | 2013-03-19 15:02:42 -0700 | [diff] [blame] | 237 | case kModeSendSample: |
| 238 | case kModeSendEnumSample: |
| 239 | case kModeSendSparseSample: |
| 240 | if ((mode != kModeSendSample) && secs_to_msecs) { |
Ken Mixter | eafbbdf | 2010-10-01 15:38:42 -0700 | [diff] [blame] | 241 | ShowUsage(); |
| 242 | } |
| 243 | return SendStats(argv, |
Darin Petkov | ed82485 | 2011-01-06 10:51:47 -0800 | [diff] [blame] | 244 | arg_index, |
Luigi Semenzato | a7ebeb3 | 2013-03-19 15:02:42 -0700 | [diff] [blame] | 245 | mode, |
Bertrand SIMONNET | 475dfa6 | 2015-08-04 14:10:10 -0700 | [diff] [blame] | 246 | secs_to_msecs); |
Darin Petkov | ed82485 | 2011-01-06 10:51:47 -0800 | [diff] [blame] | 247 | case kModeSendUserAction: |
| 248 | return SendUserAction(argv, arg_index); |
Luigi Semenzato | 3268422 | 2013-03-13 10:53:55 -0700 | [diff] [blame] | 249 | case kModeSendCrosEvent: |
| 250 | return SendCrosEvent(argv, arg_index); |
Ken Mixter | eafbbdf | 2010-10-01 15:38:42 -0700 | [diff] [blame] | 251 | case kModeHasConsent: |
| 252 | return HasConsent(); |
| 253 | case kModeIsGuestMode: |
| 254 | return IsGuestMode(); |
James Hawkins | 5f64600 | 2015-09-08 15:18:17 -0700 | [diff] [blame] | 255 | case kModeDumpLogs: |
| 256 | return DumpLogs(); |
Ken Mixter | eafbbdf | 2010-10-01 15:38:42 -0700 | [diff] [blame] | 257 | default: |
| 258 | ShowUsage(); |
| 259 | return 0; |
| 260 | } |
| 261 | } |