blob: 45f1ea1bb18315039793ef3b868f0b07a3c77173 [file] [log] [blame]
Joe Onorato5dcbc6c2017-08-29 15:13:58 -07001/*
yro0feae942017-11-15 14:38:48 -08002 * Copyright (C) 2017 The Android Open Source Project
Joe Onorato5dcbc6c2017-08-29 15:13:58 -07003 *
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 */
16
yro74fed972017-11-27 14:42:42 -080017#define DEBUG true // STOPSHIP if true
Joe Onorato9fc9edf2017-10-15 20:08:52 -070018#include "Log.h"
Joe Onorato5dcbc6c2017-08-29 15:13:58 -070019
20#include "StatsService.h"
Yao Chen8d9989b2017-11-18 18:54:50 -080021#include "android-base/stringprintf.h"
David Chenadaf8b32017-11-03 15:42:08 -070022#include "config/ConfigKey.h"
23#include "config/ConfigManager.h"
Yao Chen8d9989b2017-11-18 18:54:50 -080024#include "guardrail/MemoryLeakTrackUtil.h"
Yao Chenb3561512017-11-21 18:07:17 -080025#include "guardrail/StatsdStats.h"
yro947fbce2017-11-15 22:50:23 -080026#include "storage/StorageManager.h"
Joe Onorato5dcbc6c2017-08-29 15:13:58 -070027
David Chen0656b7a2017-09-13 15:53:39 -070028#include <android-base/file.h>
Joe Onorato5dcbc6c2017-08-29 15:13:58 -070029#include <binder/IPCThreadState.h>
30#include <binder/IServiceManager.h>
yro87d983c2017-11-14 21:31:43 -080031#include <dirent.h>
David Chen0656b7a2017-09-13 15:53:39 -070032#include <frameworks/base/cmds/statsd/src/statsd_config.pb.h>
Joe Onorato5dcbc6c2017-08-29 15:13:58 -070033#include <private/android_filesystem_config.h>
34#include <utils/Looper.h>
Joe Onorato2cbc2cc2017-08-30 17:03:23 -070035#include <utils/String16.h>
Joe Onorato5dcbc6c2017-08-29 15:13:58 -070036#include <stdio.h>
Yao Chen482d2722017-09-12 13:25:43 -070037#include <stdlib.h>
Joe Onorato9fc9edf2017-10-15 20:08:52 -070038#include <sys/system_properties.h>
Yao Chenef99c4f2017-09-22 16:26:54 -070039#include <unistd.h>
Joe Onorato5dcbc6c2017-08-29 15:13:58 -070040
41using namespace android;
42
Bookatz906a35c2017-09-20 15:26:44 -070043namespace android {
44namespace os {
45namespace statsd {
46
David Chenadaf8b32017-11-03 15:42:08 -070047constexpr const char* kPermissionDump = "android.permission.DUMP";
yro03faf092017-12-12 00:17:50 -080048#define STATS_SERVICE_DIR "/data/misc/stats-service"
David Chenadaf8b32017-11-03 15:42:08 -070049
Joe Onorato9fc9edf2017-10-15 20:08:52 -070050// ======================================================================
51/**
52 * Watches for the death of the stats companion (system process).
53 */
54class CompanionDeathRecipient : public IBinder::DeathRecipient {
55public:
56 CompanionDeathRecipient(const sp<AnomalyMonitor>& anomalyMonitor);
57 virtual void binderDied(const wp<IBinder>& who);
58
59private:
60 const sp<AnomalyMonitor> mAnomalyMonitor;
61};
62
63CompanionDeathRecipient::CompanionDeathRecipient(const sp<AnomalyMonitor>& anomalyMonitor)
64 : mAnomalyMonitor(anomalyMonitor) {
65}
66
67void CompanionDeathRecipient::binderDied(const wp<IBinder>& who) {
68 ALOGW("statscompanion service died");
69 mAnomalyMonitor->setStatsCompanionService(nullptr);
70}
71
72// ======================================================================
Joe Onorato5dcbc6c2017-08-29 15:13:58 -070073StatsService::StatsService(const sp<Looper>& handlerLooper)
Bookatz1d0136d2017-12-01 11:13:32 -080074 : mAnomalyMonitor(new AnomalyMonitor(MIN_DIFF_TO_UPDATE_REGISTERED_ALARM_SECS))
Joe Onorato5dcbc6c2017-08-29 15:13:58 -070075{
Joe Onorato9fc9edf2017-10-15 20:08:52 -070076 mUidMap = new UidMap();
77 mConfigManager = new ConfigManager();
Yangster-mac20877162017-12-22 17:19:39 -080078 mProcessor = new StatsLogProcessor(mUidMap, mAnomalyMonitor, time(nullptr), [this](const ConfigKey& key) {
yro947fbce2017-11-15 22:50:23 -080079 sp<IStatsCompanionService> sc = getStatsCompanionService();
David Chen1d7b0cd2017-11-15 14:20:04 -080080 auto receiver = mConfigManager->GetConfigReceiver(key);
81 if (sc == nullptr) {
yro74fed972017-11-27 14:42:42 -080082 VLOG("Could not find StatsCompanionService");
David Chen1d7b0cd2017-11-15 14:20:04 -080083 } else if (receiver.first.size() == 0) {
yro74fed972017-11-27 14:42:42 -080084 VLOG("Statscompanion could not find a broadcast receiver for %s",
85 key.ToString().c_str());
David Chen1d7b0cd2017-11-15 14:20:04 -080086 } else {
87 sc->sendBroadcast(String16(receiver.first.c_str()),
88 String16(receiver.second.c_str()));
89 }
yro31eb67b2017-10-24 13:33:21 -070090 });
Joe Onorato9fc9edf2017-10-15 20:08:52 -070091
92 mConfigManager->AddListener(mProcessor);
93
94 init_system_properties();
Joe Onorato5dcbc6c2017-08-29 15:13:58 -070095}
96
Yao Chenef99c4f2017-09-22 16:26:54 -070097StatsService::~StatsService() {
Joe Onorato5dcbc6c2017-08-29 15:13:58 -070098}
99
Joe Onorato9fc9edf2017-10-15 20:08:52 -0700100void StatsService::init_system_properties() {
101 mEngBuild = false;
102 const prop_info* buildType = __system_property_find("ro.build.type");
103 if (buildType != NULL) {
104 __system_property_read_callback(buildType, init_build_type_callback, this);
105 }
David Chen0656b7a2017-09-13 15:53:39 -0700106}
107
Joe Onorato9fc9edf2017-10-15 20:08:52 -0700108void StatsService::init_build_type_callback(void* cookie, const char* /*name*/, const char* value,
109 uint32_t serial) {
Yao Chen729093d2017-10-16 10:33:26 -0700110 if (0 == strcmp("eng", value) || 0 == strcmp("userdebug", value)) {
Joe Onorato9fc9edf2017-10-15 20:08:52 -0700111 reinterpret_cast<StatsService*>(cookie)->mEngBuild = true;
112 }
113}
114
115/**
116 * Implement our own because the default binder implementation isn't
117 * properly handling SHELL_COMMAND_TRANSACTION.
118 */
Yao Chenef99c4f2017-09-22 16:26:54 -0700119status_t StatsService::onTransact(uint32_t code, const Parcel& data, Parcel* reply,
120 uint32_t flags) {
Joe Onorato2cbc2cc2017-08-30 17:03:23 -0700121 status_t err;
122
123 switch (code) {
124 case SHELL_COMMAND_TRANSACTION: {
125 int in = data.readFileDescriptor();
126 int out = data.readFileDescriptor();
127 int err = data.readFileDescriptor();
128 int argc = data.readInt32();
129 Vector<String8> args;
130 for (int i = 0; i < argc && data.dataAvail() > 0; i++) {
131 args.add(String8(data.readString16()));
132 }
Yao Chenef99c4f2017-09-22 16:26:54 -0700133 sp<IShellCallback> shellCallback = IShellCallback::asInterface(data.readStrongBinder());
134 sp<IResultReceiver> resultReceiver =
135 IResultReceiver::asInterface(data.readStrongBinder());
Joe Onorato2cbc2cc2017-08-30 17:03:23 -0700136
137 FILE* fin = fdopen(in, "r");
138 FILE* fout = fdopen(out, "w");
139 FILE* ferr = fdopen(err, "w");
140
141 if (fin == NULL || fout == NULL || ferr == NULL) {
142 resultReceiver->send(NO_MEMORY);
143 } else {
144 err = command(fin, fout, ferr, args);
145 resultReceiver->send(err);
146 }
147
148 if (fin != NULL) {
149 fflush(fin);
150 fclose(fin);
151 }
152 if (fout != NULL) {
153 fflush(fout);
154 fclose(fout);
155 }
156 if (fout != NULL) {
157 fflush(ferr);
158 fclose(ferr);
159 }
160
161 return NO_ERROR;
162 }
Yao Chenef99c4f2017-09-22 16:26:54 -0700163 default: { return BnStatsManager::onTransact(code, data, reply, flags); }
Joe Onorato2cbc2cc2017-08-30 17:03:23 -0700164 }
165}
166
Joe Onorato9fc9edf2017-10-15 20:08:52 -0700167/**
168 * Write debugging data about statsd.
169 */
Yao Chenef99c4f2017-09-22 16:26:54 -0700170status_t StatsService::dump(int fd, const Vector<String16>& args) {
Joe Onorato5dcbc6c2017-08-29 15:13:58 -0700171 FILE* out = fdopen(fd, "w");
172 if (out == NULL) {
173 return NO_MEMORY; // the fd is already open
174 }
175
Joe Onorato9fc9edf2017-10-15 20:08:52 -0700176 // TODO: Proto format for incident reports
177 dump_impl(out);
Joe Onorato5dcbc6c2017-08-29 15:13:58 -0700178
179 fclose(out);
180 return NO_ERROR;
181}
182
Joe Onorato9fc9edf2017-10-15 20:08:52 -0700183/**
184 * Write debugging data about statsd in text format.
185 */
186void StatsService::dump_impl(FILE* out) {
187 mConfigManager->Dump(out);
188}
189
190/**
191 * Implementation of the adb shell cmd stats command.
192 */
Yao Chenef99c4f2017-09-22 16:26:54 -0700193status_t StatsService::command(FILE* in, FILE* out, FILE* err, Vector<String8>& args) {
Joe Onorato9fc9edf2017-10-15 20:08:52 -0700194 // TODO: Permission check
195
196 const int argCount = args.size();
197 if (argCount >= 1) {
198 // adb shell cmd stats config ...
David Chen0656b7a2017-09-13 15:53:39 -0700199 if (!args[0].compare(String8("config"))) {
Joe Onorato9fc9edf2017-10-15 20:08:52 -0700200 return cmd_config(in, out, err, args);
David Chen0656b7a2017-09-13 15:53:39 -0700201 }
Joe Onorato9fc9edf2017-10-15 20:08:52 -0700202
David Chende701692017-10-05 13:16:02 -0700203 if (!args[0].compare(String8("print-uid-map"))) {
Yao Chend10f7b12017-12-18 12:53:50 -0800204 return cmd_print_uid_map(out, args);
David Chende701692017-10-05 13:16:02 -0700205 }
Yao Chen729093d2017-10-16 10:33:26 -0700206
207 if (!args[0].compare(String8("dump-report"))) {
208 return cmd_dump_report(out, err, args);
209 }
David Chen1481fe12017-10-16 13:16:34 -0700210
211 if (!args[0].compare(String8("pull-source")) && args.size() > 1) {
212 return cmd_print_pulled_metrics(out, args);
213 }
David Chenadaf8b32017-11-03 15:42:08 -0700214
215 if (!args[0].compare(String8("send-broadcast"))) {
David Chen1d7b0cd2017-11-15 14:20:04 -0800216 return cmd_trigger_broadcast(out, args);
217 }
218
219 if (!args[0].compare(String8("print-stats"))) {
Yao Chenb3561512017-11-21 18:07:17 -0800220 return cmd_print_stats(out, args);
David Chenadaf8b32017-11-03 15:42:08 -0700221 }
yro87d983c2017-11-14 21:31:43 -0800222
Yao Chen8d9989b2017-11-18 18:54:50 -0800223 if (!args[0].compare(String8("meminfo"))) {
224 return cmd_dump_memory_info(out);
225 }
yro947fbce2017-11-15 22:50:23 -0800226
227 if (!args[0].compare(String8("write-to-disk"))) {
228 return cmd_write_data_to_disk(out);
229 }
Joe Onorato2cbc2cc2017-08-30 17:03:23 -0700230 }
Joe Onorato2cbc2cc2017-08-30 17:03:23 -0700231
Joe Onorato9fc9edf2017-10-15 20:08:52 -0700232 print_cmd_help(out);
Joe Onorato2cbc2cc2017-08-30 17:03:23 -0700233 return NO_ERROR;
234}
235
Joe Onorato9fc9edf2017-10-15 20:08:52 -0700236void StatsService::print_cmd_help(FILE* out) {
237 fprintf(out,
238 "usage: adb shell cmd stats print-stats-log [tag_required] "
239 "[timestamp_nsec_optional]\n");
240 fprintf(out, "\n");
241 fprintf(out, "\n");
Yao Chen8d9989b2017-11-18 18:54:50 -0800242 fprintf(out, "usage: adb shell cmd stats meminfo\n");
243 fprintf(out, "\n");
244 fprintf(out, " Prints the malloc debug information. You need to run the following first: \n");
245 fprintf(out, " # adb shell stop\n");
246 fprintf(out, " # adb shell setprop libc.debug.malloc.program statsd \n");
247 fprintf(out, " # adb shell setprop libc.debug.malloc.options backtrace \n");
248 fprintf(out, " # adb shell start\n");
249 fprintf(out, "\n");
250 fprintf(out, "\n");
Yao Chend10f7b12017-12-18 12:53:50 -0800251 fprintf(out, "usage: adb shell cmd stats print-uid-map [PKG]\n");
Joe Onorato9fc9edf2017-10-15 20:08:52 -0700252 fprintf(out, "\n");
253 fprintf(out, " Prints the UID, app name, version mapping.\n");
Yao Chend10f7b12017-12-18 12:53:50 -0800254 fprintf(out, " PKG Optional package name to print the uids of the package\n");
Joe Onorato9fc9edf2017-10-15 20:08:52 -0700255 fprintf(out, "\n");
256 fprintf(out, "\n");
yro3fca5ba2017-11-17 13:22:52 -0800257 fprintf(out, "usage: adb shell cmd stats pull-source [int] \n");
David Chen1481fe12017-10-16 13:16:34 -0700258 fprintf(out, "\n");
259 fprintf(out, " Prints the output of a pulled metrics source (int indicates source)\n");
260 fprintf(out, "\n");
261 fprintf(out, "\n");
yro947fbce2017-11-15 22:50:23 -0800262 fprintf(out, "usage: adb shell cmd stats write-to-disk \n");
263 fprintf(out, "\n");
264 fprintf(out, " Flushes all data on memory to disk.\n");
265 fprintf(out, "\n");
266 fprintf(out, "\n");
yro74fed972017-11-27 14:42:42 -0800267 fprintf(out, "usage: adb shell cmd stats config remove [UID] [NAME]\n");
Joe Onorato9fc9edf2017-10-15 20:08:52 -0700268 fprintf(out, "usage: adb shell cmd stats config update [UID] NAME\n");
269 fprintf(out, "\n");
270 fprintf(out, " Adds, updates or removes a configuration. The proto should be in\n");
yro74fed972017-11-27 14:42:42 -0800271 fprintf(out, " wire-encoded protobuf format and passed via stdin. If no UID and name is\n");
272 fprintf(out, " provided, then all configs will be removed from memory and disk.\n");
Joe Onorato9fc9edf2017-10-15 20:08:52 -0700273 fprintf(out, "\n");
274 fprintf(out, " UID The uid to use. It is only possible to pass the UID\n");
yro74fed972017-11-27 14:42:42 -0800275 fprintf(out, " parameter on eng builds. If UID is omitted the calling\n");
Joe Onorato9fc9edf2017-10-15 20:08:52 -0700276 fprintf(out, " uid is used.\n");
277 fprintf(out, " NAME The per-uid name to use\n");
Yao Chen5154a372017-10-30 22:57:06 -0700278 fprintf(out, "\n");
yro74fed972017-11-27 14:42:42 -0800279 fprintf(out, "\n *Note: If both UID and NAME are omitted then all configs will\n");
280 fprintf(out, "\n be removed from memory and disk!\n");
Yao Chen5154a372017-10-30 22:57:06 -0700281 fprintf(out, "\n");
Chenjie Yub236c862017-11-28 22:20:44 -0800282 fprintf(out, "usage: adb shell cmd stats dump-report [UID] NAME [--proto]\n");
Yao Chen5154a372017-10-30 22:57:06 -0700283 fprintf(out, " Dump all metric data for a configuration.\n");
284 fprintf(out, " UID The uid of the configuration. It is only possible to pass\n");
285 fprintf(out, " the UID parameter on eng builds. If UID is omitted the\n");
286 fprintf(out, " calling uid is used.\n");
287 fprintf(out, " NAME The name of the configuration\n");
Chenjie Yub236c862017-11-28 22:20:44 -0800288 fprintf(out, " --proto Print proto binary.\n");
David Chenadaf8b32017-11-03 15:42:08 -0700289 fprintf(out, "\n");
290 fprintf(out, "\n");
David Chen1d7b0cd2017-11-15 14:20:04 -0800291 fprintf(out, "usage: adb shell cmd stats send-broadcast [UID] NAME\n");
292 fprintf(out, " Send a broadcast that triggers the subscriber to fetch metrics.\n");
293 fprintf(out, " UID The uid of the configuration. It is only possible to pass\n");
294 fprintf(out, " the UID parameter on eng builds. If UID is omitted the\n");
295 fprintf(out, " calling uid is used.\n");
296 fprintf(out, " NAME The name of the configuration\n");
297 fprintf(out, "\n");
298 fprintf(out, "\n");
Yao Chenb3561512017-11-21 18:07:17 -0800299 fprintf(out, "usage: adb shell cmd stats print-stats [reset]\n");
David Chen1d7b0cd2017-11-15 14:20:04 -0800300 fprintf(out, " Prints some basic stats.\n");
Yao Chenb3561512017-11-21 18:07:17 -0800301 fprintf(out, " reset: 1 to reset the statsd stats. default=0.\n");
David Chenadaf8b32017-11-03 15:42:08 -0700302}
303
David Chen1d7b0cd2017-11-15 14:20:04 -0800304status_t StatsService::cmd_trigger_broadcast(FILE* out, Vector<String8>& args) {
305 string name;
306 bool good = false;
307 int uid;
308 const int argCount = args.size();
309 if (argCount == 2) {
310 // Automatically pick the UID
311 uid = IPCThreadState::self()->getCallingUid();
312 // TODO: What if this isn't a binder call? Should we fail?
313 name.assign(args[1].c_str(), args[1].size());
314 good = true;
315 } else if (argCount == 3) {
316 // If it's a userdebug or eng build, then the shell user can
317 // impersonate other uids.
318 if (mEngBuild) {
319 const char* s = args[1].c_str();
320 if (*s != '\0') {
321 char* end = NULL;
322 uid = strtol(s, &end, 0);
323 if (*end == '\0') {
324 name.assign(args[2].c_str(), args[2].size());
325 good = true;
326 }
327 }
328 } else {
329 fprintf(out,
330 "The metrics can only be dumped for other UIDs on eng or userdebug "
331 "builds.\n");
332 }
333 }
334 if (!good) {
335 print_cmd_help(out);
336 return UNKNOWN_ERROR;
337 }
Yangster-mac94e197c2018-01-02 16:03:03 -0800338 auto receiver = mConfigManager->GetConfigReceiver(ConfigKey(uid, StrToInt64(name)));
yro4d889e62017-11-17 15:44:48 -0800339 sp<IStatsCompanionService> sc = getStatsCompanionService();
340 if (sc != nullptr) {
341 sc->sendBroadcast(String16(receiver.first.c_str()), String16(receiver.second.c_str()));
yro74fed972017-11-27 14:42:42 -0800342 VLOG("StatsService::trigger broadcast succeeded to %s, %s", args[1].c_str(),
343 args[2].c_str());
yro4d889e62017-11-17 15:44:48 -0800344 } else {
yro74fed972017-11-27 14:42:42 -0800345 VLOG("Could not access statsCompanion");
yro4d889e62017-11-17 15:44:48 -0800346 }
347
David Chenadaf8b32017-11-03 15:42:08 -0700348 return NO_ERROR;
Joe Onorato9fc9edf2017-10-15 20:08:52 -0700349}
350
351status_t StatsService::cmd_config(FILE* in, FILE* out, FILE* err, Vector<String8>& args) {
352 const int argCount = args.size();
353 if (argCount >= 2) {
354 if (args[1] == "update" || args[1] == "remove") {
355 bool good = false;
356 int uid = -1;
357 string name;
358
359 if (argCount == 3) {
360 // Automatically pick the UID
361 uid = IPCThreadState::self()->getCallingUid();
362 // TODO: What if this isn't a binder call? Should we fail?
363 name.assign(args[2].c_str(), args[2].size());
364 good = true;
365 } else if (argCount == 4) {
366 // If it's a userdebug or eng build, then the shell user can
367 // impersonate other uids.
368 if (mEngBuild) {
369 const char* s = args[2].c_str();
370 if (*s != '\0') {
371 char* end = NULL;
372 uid = strtol(s, &end, 0);
373 if (*end == '\0') {
374 name.assign(args[3].c_str(), args[3].size());
375 good = true;
376 }
377 }
378 } else {
Yao Chen729093d2017-10-16 10:33:26 -0700379 fprintf(err,
380 "The config can only be set for other UIDs on eng or userdebug "
381 "builds.\n");
Joe Onorato9fc9edf2017-10-15 20:08:52 -0700382 }
383 }
384
385 if (!good) {
386 // If arg parsing failed, print the help text and return an error.
387 print_cmd_help(out);
388 return UNKNOWN_ERROR;
389 }
390
391 if (args[1] == "update") {
392 // Read stream into buffer.
393 string buffer;
394 if (!android::base::ReadFdToString(fileno(in), &buffer)) {
395 fprintf(err, "Error reading stream for StatsConfig.\n");
396 return UNKNOWN_ERROR;
397 }
398
399 // Parse buffer.
400 StatsdConfig config;
401 if (!config.ParseFromString(buffer)) {
402 fprintf(err, "Error parsing proto stream for StatsConfig.\n");
403 return UNKNOWN_ERROR;
404 }
405
406 // Add / update the config.
Yangster-mac94e197c2018-01-02 16:03:03 -0800407 mConfigManager->UpdateConfig(ConfigKey(uid, StrToInt64(name)), config);
Joe Onorato9fc9edf2017-10-15 20:08:52 -0700408 } else {
yro74fed972017-11-27 14:42:42 -0800409 if (argCount == 2) {
410 cmd_remove_all_configs(out);
411 } else {
412 // Remove the config.
Yangster-mac94e197c2018-01-02 16:03:03 -0800413 mConfigManager->RemoveConfig(ConfigKey(uid, StrToInt64(name)));
yro74fed972017-11-27 14:42:42 -0800414 }
Joe Onorato9fc9edf2017-10-15 20:08:52 -0700415 }
416
417 return NO_ERROR;
418 }
David Chen0656b7a2017-09-13 15:53:39 -0700419 }
Joe Onorato9fc9edf2017-10-15 20:08:52 -0700420 print_cmd_help(out);
421 return UNKNOWN_ERROR;
422}
423
Yao Chen729093d2017-10-16 10:33:26 -0700424status_t StatsService::cmd_dump_report(FILE* out, FILE* err, const Vector<String8>& args) {
425 if (mProcessor != nullptr) {
Chenjie Yub236c862017-11-28 22:20:44 -0800426 int argCount = args.size();
Yao Chen729093d2017-10-16 10:33:26 -0700427 bool good = false;
Chenjie Yub236c862017-11-28 22:20:44 -0800428 bool proto = false;
Yao Chen729093d2017-10-16 10:33:26 -0700429 int uid;
430 string name;
Chenjie Yub236c862017-11-28 22:20:44 -0800431 if (!std::strcmp("--proto", args[argCount-1].c_str())) {
432 proto = true;
433 argCount -= 1;
434 }
Yao Chen729093d2017-10-16 10:33:26 -0700435 if (argCount == 2) {
436 // Automatically pick the UID
437 uid = IPCThreadState::self()->getCallingUid();
438 // TODO: What if this isn't a binder call? Should we fail?
Yao Chen5154a372017-10-30 22:57:06 -0700439 name.assign(args[1].c_str(), args[1].size());
Yao Chen729093d2017-10-16 10:33:26 -0700440 good = true;
441 } else if (argCount == 3) {
442 // If it's a userdebug or eng build, then the shell user can
443 // impersonate other uids.
444 if (mEngBuild) {
445 const char* s = args[1].c_str();
446 if (*s != '\0') {
447 char* end = NULL;
448 uid = strtol(s, &end, 0);
449 if (*end == '\0') {
450 name.assign(args[2].c_str(), args[2].size());
451 good = true;
452 }
453 }
454 } else {
455 fprintf(out,
456 "The metrics can only be dumped for other UIDs on eng or userdebug "
457 "builds.\n");
458 }
459 }
460 if (good) {
David Chen1d7b0cd2017-11-15 14:20:04 -0800461 vector<uint8_t> data;
Yangster-mac94e197c2018-01-02 16:03:03 -0800462 mProcessor->onDumpReport(ConfigKey(uid, StrToInt64(name)), &data);
Yao Chen729093d2017-10-16 10:33:26 -0700463 // TODO: print the returned StatsLogReport to file instead of printing to logcat.
Chenjie Yub236c862017-11-28 22:20:44 -0800464 if (proto) {
465 for (size_t i = 0; i < data.size(); i ++) {
466 fprintf(out, "%c", data[i]);
467 }
468 } else {
469 fprintf(out, "Dump report for Config [%d,%s]\n", uid, name.c_str());
470 fprintf(out, "See the StatsLogReport in logcat...\n");
471 }
Yao Chen729093d2017-10-16 10:33:26 -0700472 return android::OK;
473 } else {
474 // If arg parsing failed, print the help text and return an error.
475 print_cmd_help(out);
476 return UNKNOWN_ERROR;
477 }
478 } else {
479 fprintf(out, "Log processor does not exist...\n");
480 return UNKNOWN_ERROR;
481 }
482}
483
Yao Chenb3561512017-11-21 18:07:17 -0800484status_t StatsService::cmd_print_stats(FILE* out, const Vector<String8>& args) {
David Chen1d7b0cd2017-11-15 14:20:04 -0800485 vector<ConfigKey> configs = mConfigManager->GetAllConfigKeys();
486 for (const ConfigKey& key : configs) {
487 fprintf(out, "Config %s uses %zu bytes\n", key.ToString().c_str(),
488 mProcessor->GetMetricsSize(key));
489 }
David Chenc136f452017-11-27 11:52:26 -0800490 fprintf(out, "Detailed statsd stats in logcat...\n");
Yao Chenb3561512017-11-21 18:07:17 -0800491 StatsdStats& statsdStats = StatsdStats::getInstance();
492 bool reset = false;
493 if (args.size() > 1) {
494 reset = strtol(args[1].string(), NULL, 10);
495 }
Yao Chen69f1baf2017-11-27 17:25:36 -0800496 vector<uint8_t> output;
Yao Chenb3561512017-11-21 18:07:17 -0800497 statsdStats.dumpStats(&output, reset);
David Chen1d7b0cd2017-11-15 14:20:04 -0800498 return NO_ERROR;
499}
500
Yao Chend10f7b12017-12-18 12:53:50 -0800501status_t StatsService::cmd_print_uid_map(FILE* out, const Vector<String8>& args) {
502 if (args.size() > 1) {
503 string pkg;
504 pkg.assign(args[1].c_str(), args[1].size());
505 auto uids = mUidMap->getAppUid(pkg);
506 fprintf(out, "%s -> [ ", pkg.c_str());
507 for (const auto& uid : uids) {
508 fprintf(out, "%d ", uid);
509 }
510 fprintf(out, "]\n");
511 } else {
512 mUidMap->printUidMap(out);
513 }
Joe Onorato9fc9edf2017-10-15 20:08:52 -0700514 return NO_ERROR;
David Chen0656b7a2017-09-13 15:53:39 -0700515}
516
yro947fbce2017-11-15 22:50:23 -0800517status_t StatsService::cmd_write_data_to_disk(FILE* out) {
518 fprintf(out, "Writing data to disk\n");
519 mProcessor->WriteDataToDisk();
520 return NO_ERROR;
521}
522
David Chen1481fe12017-10-16 13:16:34 -0700523status_t StatsService::cmd_print_pulled_metrics(FILE* out, const Vector<String8>& args) {
524 int s = atoi(args[1].c_str());
Chenjie Yu5305e1d2017-10-31 13:49:36 -0700525 vector<shared_ptr<LogEvent> > stats;
526 if (mStatsPullerManager.Pull(s, &stats)) {
527 for (const auto& it : stats) {
528 fprintf(out, "Pull from %d: %s\n", s, it->ToString().c_str());
529 }
530 fprintf(out, "Pull from %d: Received %zu elements\n", s, stats.size());
531 return NO_ERROR;
David Chen1481fe12017-10-16 13:16:34 -0700532 }
Chenjie Yu5305e1d2017-10-31 13:49:36 -0700533 return UNKNOWN_ERROR;
David Chen1481fe12017-10-16 13:16:34 -0700534}
535
yro74fed972017-11-27 14:42:42 -0800536status_t StatsService::cmd_remove_all_configs(FILE* out) {
537 fprintf(out, "Removing all configs...\n");
538 VLOG("StatsService::cmd_remove_all_configs was called");
539 mConfigManager->RemoveAllConfigs();
yro947fbce2017-11-15 22:50:23 -0800540 StorageManager::deleteAllFiles(STATS_SERVICE_DIR);
yro87d983c2017-11-14 21:31:43 -0800541 return NO_ERROR;
542}
543
Yao Chen8d9989b2017-11-18 18:54:50 -0800544status_t StatsService::cmd_dump_memory_info(FILE* out) {
545 std::string s = dumpMemInfo(100);
546 fprintf(out, "Memory Info\n");
547 fprintf(out, "%s", s.c_str());
548 return NO_ERROR;
549}
550
Dianne Hackborn3accca02013-09-20 09:32:11 -0700551Status StatsService::informAllUidData(const vector<int32_t>& uid, const vector<int64_t>& version,
David Chende701692017-10-05 13:16:02 -0700552 const vector<String16>& app) {
yro74fed972017-11-27 14:42:42 -0800553 VLOG("StatsService::informAllUidData was called");
David Chende701692017-10-05 13:16:02 -0700554
555 if (IPCThreadState::self()->getCallingUid() != AID_SYSTEM) {
556 return Status::fromExceptionCode(Status::EX_SECURITY,
557 "Only system uid can call informAllUidData");
558 }
559
Joe Onorato9fc9edf2017-10-15 20:08:52 -0700560 mUidMap->updateMap(uid, version, app);
yro74fed972017-11-27 14:42:42 -0800561 VLOG("StatsService::informAllUidData succeeded");
David Chende701692017-10-05 13:16:02 -0700562
563 return Status::ok();
564}
565
Dianne Hackborn3accca02013-09-20 09:32:11 -0700566Status StatsService::informOnePackage(const String16& app, int32_t uid, int64_t version) {
yro74fed972017-11-27 14:42:42 -0800567 VLOG("StatsService::informOnePackage was called");
David Chende701692017-10-05 13:16:02 -0700568
569 if (IPCThreadState::self()->getCallingUid() != AID_SYSTEM) {
570 return Status::fromExceptionCode(Status::EX_SECURITY,
571 "Only system uid can call informOnePackage");
572 }
Joe Onorato9fc9edf2017-10-15 20:08:52 -0700573 mUidMap->updateApp(app, uid, version);
David Chende701692017-10-05 13:16:02 -0700574 return Status::ok();
575}
576
577Status StatsService::informOnePackageRemoved(const String16& app, int32_t uid) {
yro74fed972017-11-27 14:42:42 -0800578 VLOG("StatsService::informOnePackageRemoved was called");
David Chende701692017-10-05 13:16:02 -0700579
580 if (IPCThreadState::self()->getCallingUid() != AID_SYSTEM) {
581 return Status::fromExceptionCode(Status::EX_SECURITY,
582 "Only system uid can call informOnePackageRemoved");
583 }
Joe Onorato9fc9edf2017-10-15 20:08:52 -0700584 mUidMap->removeApp(app, uid);
David Chende701692017-10-05 13:16:02 -0700585 return Status::ok();
586}
587
Yao Chenef99c4f2017-09-22 16:26:54 -0700588Status StatsService::informAnomalyAlarmFired() {
yro74fed972017-11-27 14:42:42 -0800589 VLOG("StatsService::informAnomalyAlarmFired was called");
Bookatz1b0b1142017-09-08 11:58:42 -0700590
591 if (IPCThreadState::self()->getCallingUid() != AID_SYSTEM) {
592 return Status::fromExceptionCode(Status::EX_SECURITY,
Yao Chenef99c4f2017-09-22 16:26:54 -0700593 "Only system uid can call informAnomalyAlarmFired");
Bookatz1b0b1142017-09-08 11:58:42 -0700594 }
595
yro74fed972017-11-27 14:42:42 -0800596 VLOG("StatsService::informAnomalyAlarmFired succeeded");
Bookatzcc5adef2017-11-21 14:36:23 -0800597 uint64_t currentTimeSec = time(nullptr);
Yangster-mace2cd6d52017-11-09 20:38:30 -0800598 std::unordered_set<sp<const AnomalyAlarm>, SpHash<AnomalyAlarm>> anomalySet =
Bookatzcc5adef2017-11-21 14:36:23 -0800599 mAnomalyMonitor->popSoonerThan(static_cast<uint32_t>(currentTimeSec));
600 mProcessor->onAnomalyAlarmFired(currentTimeSec * NS_PER_SEC, anomalySet);
Bookatz1b0b1142017-09-08 11:58:42 -0700601 return Status::ok();
602}
603
Yao Chenef99c4f2017-09-22 16:26:54 -0700604Status StatsService::informPollAlarmFired() {
yro74fed972017-11-27 14:42:42 -0800605 VLOG("StatsService::informPollAlarmFired was called");
Bookatz1b0b1142017-09-08 11:58:42 -0700606
607 if (IPCThreadState::self()->getCallingUid() != AID_SYSTEM) {
608 return Status::fromExceptionCode(Status::EX_SECURITY,
Yao Chenef99c4f2017-09-22 16:26:54 -0700609 "Only system uid can call informPollAlarmFired");
Bookatz1b0b1142017-09-08 11:58:42 -0700610 }
611
Chenjie Yu5305e1d2017-10-31 13:49:36 -0700612 mStatsPullerManager.OnAlarmFired();
Chenjie Yub3dda412017-10-24 13:41:59 -0700613
yro74fed972017-11-27 14:42:42 -0800614 VLOG("StatsService::informPollAlarmFired succeeded");
Bookatz1b0b1142017-09-08 11:58:42 -0700615
616 return Status::ok();
617}
618
Yao Chenef99c4f2017-09-22 16:26:54 -0700619Status StatsService::systemRunning() {
Joe Onorato5dcbc6c2017-08-29 15:13:58 -0700620 if (IPCThreadState::self()->getCallingUid() != AID_SYSTEM) {
621 return Status::fromExceptionCode(Status::EX_SECURITY,
Yao Chenef99c4f2017-09-22 16:26:54 -0700622 "Only system uid can call systemRunning");
Joe Onorato5dcbc6c2017-08-29 15:13:58 -0700623 }
624
625 // When system_server is up and running, schedule the dropbox task to run.
yro74fed972017-11-27 14:42:42 -0800626 VLOG("StatsService::systemRunning");
Joe Onorato5dcbc6c2017-08-29 15:13:58 -0700627
Bookatzb487b552017-09-18 11:26:01 -0700628 sayHiToStatsCompanion();
629
Joe Onorato5dcbc6c2017-08-29 15:13:58 -0700630 return Status::ok();
631}
632
yro947fbce2017-11-15 22:50:23 -0800633Status StatsService::writeDataToDisk() {
634 if (IPCThreadState::self()->getCallingUid() != AID_SYSTEM) {
635 return Status::fromExceptionCode(Status::EX_SECURITY,
636 "Only system uid can call systemRunning");
637 }
638
yro74fed972017-11-27 14:42:42 -0800639 VLOG("StatsService::writeDataToDisk");
yro947fbce2017-11-15 22:50:23 -0800640
641 mProcessor->WriteDataToDisk();
642
643 return Status::ok();
644}
645
Yao Chenef99c4f2017-09-22 16:26:54 -0700646void StatsService::sayHiToStatsCompanion() {
647 // TODO: This method needs to be private. It is temporarily public and unsecured for testing
648 // purposes.
Bookatzb487b552017-09-18 11:26:01 -0700649 sp<IStatsCompanionService> statsCompanion = getStatsCompanionService();
650 if (statsCompanion != nullptr) {
yro74fed972017-11-27 14:42:42 -0800651 VLOG("Telling statsCompanion that statsd is ready");
Bookatzb487b552017-09-18 11:26:01 -0700652 statsCompanion->statsdReady();
653 } else {
yro74fed972017-11-27 14:42:42 -0800654 VLOG("Could not access statsCompanion");
Bookatzb487b552017-09-18 11:26:01 -0700655 }
656}
657
Yao Chenef99c4f2017-09-22 16:26:54 -0700658sp<IStatsCompanionService> StatsService::getStatsCompanionService() {
Bookatz906a35c2017-09-20 15:26:44 -0700659 sp<IStatsCompanionService> statsCompanion = nullptr;
660 // Get statscompanion service from service manager
661 const sp<IServiceManager> sm(defaultServiceManager());
662 if (sm != nullptr) {
663 const String16 name("statscompanion");
664 statsCompanion = interface_cast<IStatsCompanionService>(sm->checkService(name));
665 if (statsCompanion == nullptr) {
666 ALOGW("statscompanion service unavailable!");
667 return nullptr;
668 }
669 }
670 return statsCompanion;
671}
672
Yao Chenef99c4f2017-09-22 16:26:54 -0700673Status StatsService::statsCompanionReady() {
yro74fed972017-11-27 14:42:42 -0800674 VLOG("StatsService::statsCompanionReady was called");
Bookatzb487b552017-09-18 11:26:01 -0700675
676 if (IPCThreadState::self()->getCallingUid() != AID_SYSTEM) {
677 return Status::fromExceptionCode(Status::EX_SECURITY,
678 "Only system uid can call statsCompanionReady");
679 }
680
681 sp<IStatsCompanionService> statsCompanion = getStatsCompanionService();
682 if (statsCompanion == nullptr) {
Yao Chenef99c4f2017-09-22 16:26:54 -0700683 return Status::fromExceptionCode(
684 Status::EX_NULL_POINTER,
685 "statscompanion unavailable despite it contacting statsd!");
Bookatzb487b552017-09-18 11:26:01 -0700686 }
yro74fed972017-11-27 14:42:42 -0800687 VLOG("StatsService::statsCompanionReady linking to statsCompanion.");
Joe Onorato9fc9edf2017-10-15 20:08:52 -0700688 IInterface::asBinder(statsCompanion)->linkToDeath(new CompanionDeathRecipient(mAnomalyMonitor));
Bookatzb487b552017-09-18 11:26:01 -0700689 mAnomalyMonitor->setStatsCompanionService(statsCompanion);
690
691 return Status::ok();
692}
693
Joe Onorato9fc9edf2017-10-15 20:08:52 -0700694void StatsService::Startup() {
695 mConfigManager->Startup();
Bookatz906a35c2017-09-20 15:26:44 -0700696}
697
Joe Onoratoc4dfae52017-10-17 23:38:21 -0700698void StatsService::OnLogEvent(const LogEvent& event) {
699 mProcessor->OnLogEvent(event);
Bookatz906a35c2017-09-20 15:26:44 -0700700}
701
Yangster-mac94e197c2018-01-02 16:03:03 -0800702Status StatsService::getData(int64_t key, vector<uint8_t>* output) {
David Chenadaf8b32017-11-03 15:42:08 -0700703 IPCThreadState* ipc = IPCThreadState::self();
yro74fed972017-11-27 14:42:42 -0800704 VLOG("StatsService::getData with Pid %i, Uid %i", ipc->getCallingPid(), ipc->getCallingUid());
Yao Chen7ee94152017-11-17 09:44:40 -0800705 if (checkCallingPermission(String16(kPermissionDump))) {
Yangster-mac94e197c2018-01-02 16:03:03 -0800706 ConfigKey configKey(ipc->getCallingUid(), key);
David Chen1d7b0cd2017-11-15 14:20:04 -0800707 mProcessor->onDumpReport(configKey, output);
David Chenadaf8b32017-11-03 15:42:08 -0700708 return Status::ok();
709 } else {
710 return Status::fromExceptionCode(binder::Status::EX_SECURITY);
711 }
yro31eb67b2017-10-24 13:33:21 -0700712}
713
David Chen2e8f3802017-11-22 10:56:48 -0800714Status StatsService::getMetadata(vector<uint8_t>* output) {
715 IPCThreadState* ipc = IPCThreadState::self();
716 VLOG("StatsService::getMetadata with Pid %i, Uid %i", ipc->getCallingPid(),
717 ipc->getCallingUid());
718 if (checkCallingPermission(String16(kPermissionDump))) {
719 StatsdStats::getInstance().dumpStats(output, false); // Don't reset the counters.
720 return Status::ok();
721 } else {
722 return Status::fromExceptionCode(binder::Status::EX_SECURITY);
723 }
724}
725
Yangster-mac94e197c2018-01-02 16:03:03 -0800726Status StatsService::addConfiguration(int64_t key,
David Chenadaf8b32017-11-03 15:42:08 -0700727 const vector <uint8_t>& config,
728 const String16& package, const String16& cls,
729 bool* success) {
730 IPCThreadState* ipc = IPCThreadState::self();
Yao Chen7ee94152017-11-17 09:44:40 -0800731 if (checkCallingPermission(String16(kPermissionDump))) {
Yangster-mac94e197c2018-01-02 16:03:03 -0800732 ConfigKey configKey(ipc->getCallingUid(), key);
David Chenadaf8b32017-11-03 15:42:08 -0700733 StatsdConfig cfg;
David Chen7d8aa4d2017-12-27 13:37:01 -0800734 if (!cfg.ParseFromArray(&config[0], config.size())) {
735 *success = false;
736 return Status::ok();
737 }
David Chenadaf8b32017-11-03 15:42:08 -0700738 mConfigManager->UpdateConfig(configKey, cfg);
739 mConfigManager->SetConfigReceiver(configKey, string(String8(package).string()),
740 string(String8(cls).string()));
741 *success = true;
742 return Status::ok();
743 } else {
Yao Chenaa39bc72017-12-01 11:16:50 -0800744 *success = false;
David Chenadaf8b32017-11-03 15:42:08 -0700745 return Status::fromExceptionCode(binder::Status::EX_SECURITY);
yro31eb67b2017-10-24 13:33:21 -0700746 }
yro31eb67b2017-10-24 13:33:21 -0700747}
748
Yangster-mac94e197c2018-01-02 16:03:03 -0800749Status StatsService::removeConfiguration(int64_t key, bool* success) {
David Chenadaf8b32017-11-03 15:42:08 -0700750 IPCThreadState* ipc = IPCThreadState::self();
Yao Chen7ee94152017-11-17 09:44:40 -0800751 if (checkCallingPermission(String16(kPermissionDump))) {
Yangster-mac94e197c2018-01-02 16:03:03 -0800752 mConfigManager->RemoveConfig(ConfigKey(ipc->getCallingUid(), key));
Yao Chenaa39bc72017-12-01 11:16:50 -0800753 *success = true;
David Chenadaf8b32017-11-03 15:42:08 -0700754 return Status::ok();
755 } else {
756 *success = false;
757 return Status::fromExceptionCode(binder::Status::EX_SECURITY);
yro31eb67b2017-10-24 13:33:21 -0700758 }
yro31eb67b2017-10-24 13:33:21 -0700759}
760
David Chen1d7b0cd2017-11-15 14:20:04 -0800761void StatsService::binderDied(const wp <IBinder>& who) {
yro31eb67b2017-10-24 13:33:21 -0700762}
763
Yao Chenef99c4f2017-09-22 16:26:54 -0700764} // namespace statsd
765} // namespace os
766} // namespace android