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 "constants.h" |
Bertrand SIMONNET | e6cfd64 | 2014-07-09 16:35:23 -0700 | [diff] [blame] | 21 | #include "metrics/metrics_library.h" |
Darin Petkov | 65b0146 | 2010-04-14 13:32:20 -0700 | [diff] [blame] | 22 | |
Luigi Semenzato | a7ebeb3 | 2013-03-19 15:02:42 -0700 | [diff] [blame] | 23 | enum Mode { |
Bertrand SIMONNET | b13527d | 2015-12-02 17:19:40 -0800 | [diff] [blame] | 24 | kModeDumpHistograms, |
Luigi Semenzato | a7ebeb3 | 2013-03-19 15:02:42 -0700 | [diff] [blame] | 25 | kModeSendSample, |
| 26 | kModeSendEnumSample, |
| 27 | kModeSendSparseSample, |
Luigi Semenzato | a7ebeb3 | 2013-03-19 15:02:42 -0700 | [diff] [blame] | 28 | kModeSendCrosEvent, |
| 29 | kModeHasConsent, |
| 30 | kModeIsGuestMode, |
| 31 | }; |
| 32 | |
Ken Mixter | eafbbdf | 2010-10-01 15:38:42 -0700 | [diff] [blame] | 33 | void ShowUsage() { |
| 34 | fprintf(stderr, |
Bertrand SIMONNET | 475dfa6 | 2015-08-04 14:10:10 -0700 | [diff] [blame] | 35 | "Usage: metrics_client [-t] name sample min max nbuckets\n" |
| 36 | " metrics_client -e name sample max\n" |
| 37 | " metrics_client -s name sample\n" |
| 38 | " metrics_client -v event\n" |
James Hawkins | 5f64600 | 2015-09-08 15:18:17 -0700 | [diff] [blame] | 39 | " metrics_client [-cdg]\n" |
Ken Mixter | eafbbdf | 2010-10-01 15:38:42 -0700 | [diff] [blame] | 40 | "\n" |
Bertrand SIMONNET | 475dfa6 | 2015-08-04 14:10:10 -0700 | [diff] [blame] | 41 | " default: send metric with integer values \n" |
Ken Mixter | eafbbdf | 2010-10-01 15:38:42 -0700 | [diff] [blame] | 42 | " |min| > 0, |min| <= sample < |max|\n" |
Thiemo Nagel | 0b8cc1c | 2014-08-21 15:00:50 +0200 | [diff] [blame] | 43 | " -c: return exit status 0 if user consents to stats, 1 otherwise,\n" |
| 44 | " in guest mode always return 1\n" |
Bertrand SIMONNET | b13527d | 2015-12-02 17:19:40 -0800 | [diff] [blame] | 45 | " -d: dump the histograms recorded by metricsd to stdout\n" |
Ken Mixter | eafbbdf | 2010-10-01 15:38:42 -0700 | [diff] [blame] | 46 | " -e: send linear/enumeration histogram data\n" |
| 47 | " -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] | 48 | " -s: send a sparse histogram sample\n" |
Darin Petkov | ed82485 | 2011-01-06 10:51:47 -0800 | [diff] [blame] | 49 | " -t: convert sample from double seconds to int milliseconds\n" |
Bertrand SIMONNET | e6cfd64 | 2014-07-09 16:35:23 -0700 | [diff] [blame] | 50 | " -v: send a Platform.CrOSEvent enum histogram sample\n"); |
Ken Mixter | eafbbdf | 2010-10-01 15:38:42 -0700 | [diff] [blame] | 51 | exit(1); |
| 52 | } |
Darin Petkov | 65b0146 | 2010-04-14 13:32:20 -0700 | [diff] [blame] | 53 | |
Luigi Semenzato | d8abf55 | 2014-03-27 14:19:06 -0700 | [diff] [blame] | 54 | static int ParseInt(const char *arg) { |
| 55 | char *endptr; |
| 56 | int value = strtol(arg, &endptr, 0); |
| 57 | if (*endptr != '\0') { |
| 58 | fprintf(stderr, "metrics client: bad integer \"%s\"\n", arg); |
| 59 | ShowUsage(); |
| 60 | } |
| 61 | return value; |
| 62 | } |
| 63 | |
| 64 | static double ParseDouble(const char *arg) { |
| 65 | char *endptr; |
| 66 | double value = strtod(arg, &endptr); |
| 67 | if (*endptr != '\0') { |
| 68 | fprintf(stderr, "metrics client: bad double \"%s\"\n", arg); |
| 69 | ShowUsage(); |
| 70 | } |
| 71 | return value; |
| 72 | } |
| 73 | |
Bertrand SIMONNET | b13527d | 2015-12-02 17:19:40 -0800 | [diff] [blame] | 74 | static int DumpHistograms() { |
| 75 | MetricsLibrary metrics_lib; |
| 76 | metrics_lib.Init(); |
| 77 | |
| 78 | std::string dump; |
| 79 | if (!metrics_lib.GetHistogramsDump(&dump)) { |
| 80 | printf("Failed to dump the histograms."); |
| 81 | return 1; |
| 82 | } |
| 83 | |
| 84 | printf("%s\n", dump.c_str()); |
| 85 | return 0; |
| 86 | } |
| 87 | |
Ken Mixter | eafbbdf | 2010-10-01 15:38:42 -0700 | [diff] [blame] | 88 | static int SendStats(char* argv[], |
| 89 | int name_index, |
Luigi Semenzato | a7ebeb3 | 2013-03-19 15:02:42 -0700 | [diff] [blame] | 90 | enum Mode mode, |
Bertrand SIMONNET | 475dfa6 | 2015-08-04 14:10:10 -0700 | [diff] [blame] | 91 | bool secs_to_msecs) { |
Darin Petkov | c2526a1 | 2010-04-21 14:24:04 -0700 | [diff] [blame] | 92 | const char* name = argv[name_index]; |
| 93 | int sample; |
Darin Petkov | 4fcb2ac | 2010-04-15 16:40:23 -0700 | [diff] [blame] | 94 | if (secs_to_msecs) { |
Luigi Semenzato | d8abf55 | 2014-03-27 14:19:06 -0700 | [diff] [blame] | 95 | sample = static_cast<int>(ParseDouble(argv[name_index + 1]) * 1000.0); |
Darin Petkov | 4fcb2ac | 2010-04-15 16:40:23 -0700 | [diff] [blame] | 96 | } else { |
Luigi Semenzato | d8abf55 | 2014-03-27 14:19:06 -0700 | [diff] [blame] | 97 | sample = ParseInt(argv[name_index + 1]); |
Darin Petkov | 4fcb2ac | 2010-04-15 16:40:23 -0700 | [diff] [blame] | 98 | } |
| 99 | |
Bertrand SIMONNET | 475dfa6 | 2015-08-04 14:10:10 -0700 | [diff] [blame] | 100 | MetricsLibrary metrics_lib; |
| 101 | metrics_lib.Init(); |
| 102 | if (mode == kModeSendSparseSample) { |
| 103 | metrics_lib.SendSparseToUMA(name, sample); |
| 104 | } else if (mode == kModeSendEnumSample) { |
| 105 | int max = ParseInt(argv[name_index + 2]); |
| 106 | metrics_lib.SendEnumToUMA(name, sample, max); |
| 107 | } else { |
| 108 | int min = ParseInt(argv[name_index + 2]); |
| 109 | int max = ParseInt(argv[name_index + 3]); |
| 110 | int nbuckets = ParseInt(argv[name_index + 4]); |
| 111 | metrics_lib.SendToUMA(name, sample, min, max, nbuckets); |
Darin Petkov | 65b0146 | 2010-04-14 13:32:20 -0700 | [diff] [blame] | 112 | } |
| 113 | return 0; |
| 114 | } |
Ken Mixter | eafbbdf | 2010-10-01 15:38:42 -0700 | [diff] [blame] | 115 | |
Luigi Semenzato | 3268422 | 2013-03-13 10:53:55 -0700 | [diff] [blame] | 116 | static int SendCrosEvent(char* argv[], int action_index) { |
| 117 | const char* event = argv[action_index]; |
| 118 | bool result; |
| 119 | MetricsLibrary metrics_lib; |
| 120 | metrics_lib.Init(); |
| 121 | result = metrics_lib.SendCrosEventToUMA(event); |
| 122 | if (!result) { |
| 123 | fprintf(stderr, "metrics_client: could not send event %s\n", event); |
| 124 | return 1; |
| 125 | } |
| 126 | return 0; |
| 127 | } |
| 128 | |
Ken Mixter | eafbbdf | 2010-10-01 15:38:42 -0700 | [diff] [blame] | 129 | static int HasConsent() { |
| 130 | MetricsLibrary metrics_lib; |
| 131 | metrics_lib.Init(); |
| 132 | return metrics_lib.AreMetricsEnabled() ? 0 : 1; |
| 133 | } |
| 134 | |
| 135 | static int IsGuestMode() { |
| 136 | MetricsLibrary metrics_lib; |
| 137 | metrics_lib.Init(); |
| 138 | return metrics_lib.IsGuestMode() ? 0 : 1; |
| 139 | } |
| 140 | |
| 141 | int main(int argc, char** argv) { |
Luigi Semenzato | a7ebeb3 | 2013-03-19 15:02:42 -0700 | [diff] [blame] | 142 | enum Mode mode = kModeSendSample; |
Ken Mixter | eafbbdf | 2010-10-01 15:38:42 -0700 | [diff] [blame] | 143 | bool secs_to_msecs = false; |
Ken Mixter | eafbbdf | 2010-10-01 15:38:42 -0700 | [diff] [blame] | 144 | |
| 145 | // Parse arguments |
| 146 | int flag; |
Bertrand SIMONNET | b13527d | 2015-12-02 17:19:40 -0800 | [diff] [blame] | 147 | while ((flag = getopt(argc, argv, "abcdegstv")) != -1) { |
Ken Mixter | eafbbdf | 2010-10-01 15:38:42 -0700 | [diff] [blame] | 148 | switch (flag) { |
Ken Mixter | eafbbdf | 2010-10-01 15:38:42 -0700 | [diff] [blame] | 149 | case 'c': |
| 150 | mode = kModeHasConsent; |
| 151 | break; |
Bertrand SIMONNET | b13527d | 2015-12-02 17:19:40 -0800 | [diff] [blame] | 152 | case 'd': |
| 153 | mode = kModeDumpHistograms; |
| 154 | break; |
Ken Mixter | eafbbdf | 2010-10-01 15:38:42 -0700 | [diff] [blame] | 155 | case 'e': |
Luigi Semenzato | a7ebeb3 | 2013-03-19 15:02:42 -0700 | [diff] [blame] | 156 | mode = kModeSendEnumSample; |
Ken Mixter | eafbbdf | 2010-10-01 15:38:42 -0700 | [diff] [blame] | 157 | break; |
| 158 | case 'g': |
| 159 | mode = kModeIsGuestMode; |
| 160 | break; |
Luigi Semenzato | a7ebeb3 | 2013-03-19 15:02:42 -0700 | [diff] [blame] | 161 | case 's': |
| 162 | mode = kModeSendSparseSample; |
| 163 | break; |
Ken Mixter | eafbbdf | 2010-10-01 15:38:42 -0700 | [diff] [blame] | 164 | case 't': |
| 165 | secs_to_msecs = true; |
| 166 | break; |
Luigi Semenzato | 3268422 | 2013-03-13 10:53:55 -0700 | [diff] [blame] | 167 | case 'v': |
| 168 | mode = kModeSendCrosEvent; |
| 169 | break; |
Ken Mixter | eafbbdf | 2010-10-01 15:38:42 -0700 | [diff] [blame] | 170 | default: |
Darin Petkov | 8032dd0 | 2011-05-09 16:33:19 -0700 | [diff] [blame] | 171 | ShowUsage(); |
Ken Mixter | eafbbdf | 2010-10-01 15:38:42 -0700 | [diff] [blame] | 172 | break; |
| 173 | } |
| 174 | } |
Darin Petkov | ed82485 | 2011-01-06 10:51:47 -0800 | [diff] [blame] | 175 | int arg_index = optind; |
Ken Mixter | eafbbdf | 2010-10-01 15:38:42 -0700 | [diff] [blame] | 176 | |
| 177 | int expected_args = 0; |
Luigi Semenzato | a7ebeb3 | 2013-03-19 15:02:42 -0700 | [diff] [blame] | 178 | if (mode == kModeSendSample) |
| 179 | expected_args = 5; |
| 180 | else if (mode == kModeSendEnumSample) |
| 181 | expected_args = 3; |
| 182 | else if (mode == kModeSendSparseSample) |
| 183 | expected_args = 2; |
Luigi Semenzato | 3268422 | 2013-03-13 10:53:55 -0700 | [diff] [blame] | 184 | else if (mode == kModeSendCrosEvent) |
| 185 | expected_args = 1; |
Ken Mixter | eafbbdf | 2010-10-01 15:38:42 -0700 | [diff] [blame] | 186 | |
Darin Petkov | ed82485 | 2011-01-06 10:51:47 -0800 | [diff] [blame] | 187 | if ((arg_index + expected_args) != argc) { |
Ken Mixter | eafbbdf | 2010-10-01 15:38:42 -0700 | [diff] [blame] | 188 | ShowUsage(); |
| 189 | } |
| 190 | |
Bertrand SIMONNET | e6cfd64 | 2014-07-09 16:35:23 -0700 | [diff] [blame] | 191 | switch (mode) { |
Bertrand SIMONNET | b13527d | 2015-12-02 17:19:40 -0800 | [diff] [blame] | 192 | case kModeDumpHistograms: |
| 193 | return DumpHistograms(); |
Luigi Semenzato | a7ebeb3 | 2013-03-19 15:02:42 -0700 | [diff] [blame] | 194 | case kModeSendSample: |
| 195 | case kModeSendEnumSample: |
| 196 | case kModeSendSparseSample: |
| 197 | if ((mode != kModeSendSample) && secs_to_msecs) { |
Ken Mixter | eafbbdf | 2010-10-01 15:38:42 -0700 | [diff] [blame] | 198 | ShowUsage(); |
| 199 | } |
| 200 | return SendStats(argv, |
Darin Petkov | ed82485 | 2011-01-06 10:51:47 -0800 | [diff] [blame] | 201 | arg_index, |
Luigi Semenzato | a7ebeb3 | 2013-03-19 15:02:42 -0700 | [diff] [blame] | 202 | mode, |
Bertrand SIMONNET | 475dfa6 | 2015-08-04 14:10:10 -0700 | [diff] [blame] | 203 | secs_to_msecs); |
Luigi Semenzato | 3268422 | 2013-03-13 10:53:55 -0700 | [diff] [blame] | 204 | case kModeSendCrosEvent: |
| 205 | return SendCrosEvent(argv, arg_index); |
Ken Mixter | eafbbdf | 2010-10-01 15:38:42 -0700 | [diff] [blame] | 206 | case kModeHasConsent: |
| 207 | return HasConsent(); |
| 208 | case kModeIsGuestMode: |
| 209 | return IsGuestMode(); |
| 210 | default: |
| 211 | ShowUsage(); |
| 212 | return 0; |
| 213 | } |
| 214 | } |