Yao Chen | ab273e2 | 2017-09-06 12:53:50 -0700 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (C) 2017 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 | */ |
| 16 | |
Yao Chen | b356151 | 2017-11-21 18:07:17 -0800 | [diff] [blame] | 17 | #define DEBUG true // STOPSHIP if true |
Joe Onorato | 9fc9edf | 2017-10-15 20:08:52 -0700 | [diff] [blame] | 18 | #include "Log.h" |
David Chen | 2158296 | 2017-11-01 17:32:46 -0700 | [diff] [blame] | 19 | #include "statslog.h" |
Yao Chen | ab273e2 | 2017-09-06 12:53:50 -0700 | [diff] [blame] | 20 | |
yro | 947fbce | 2017-11-15 22:50:23 -0800 | [diff] [blame] | 21 | #include <android-base/file.h> |
| 22 | #include <dirent.h> |
Joe Onorato | 9fc9edf | 2017-10-15 20:08:52 -0700 | [diff] [blame] | 23 | #include "StatsLogProcessor.h" |
yro | 947fbce | 2017-11-15 22:50:23 -0800 | [diff] [blame] | 24 | #include "android-base/stringprintf.h" |
Yao Chen | b356151 | 2017-11-21 18:07:17 -0800 | [diff] [blame] | 25 | #include "guardrail/StatsdStats.h" |
Joe Onorato | 9fc9edf | 2017-10-15 20:08:52 -0700 | [diff] [blame] | 26 | #include "metrics/CountMetricProducer.h" |
Chenjie Yu | 85ed838 | 2017-12-14 16:48:54 -0800 | [diff] [blame] | 27 | #include "external/StatsPullerManager.h" |
Yangster-mac | d40053e | 2018-01-09 16:29:22 -0800 | [diff] [blame] | 28 | #include "dimension.h" |
| 29 | #include "field_util.h" |
Joe Onorato | 9fc9edf | 2017-10-15 20:08:52 -0700 | [diff] [blame] | 30 | #include "stats_util.h" |
yro | 947fbce | 2017-11-15 22:50:23 -0800 | [diff] [blame] | 31 | #include "storage/StorageManager.h" |
Joe Onorato | 9fc9edf | 2017-10-15 20:08:52 -0700 | [diff] [blame] | 32 | |
yro | 00698da | 2017-09-15 10:06:40 -0700 | [diff] [blame] | 33 | #include <log/log_event_list.h> |
Yao Chen | ef99c4f | 2017-09-22 16:26:54 -0700 | [diff] [blame] | 34 | #include <utils/Errors.h> |
Yao Chen | ab273e2 | 2017-09-06 12:53:50 -0700 | [diff] [blame] | 35 | |
| 36 | using namespace android; |
yro | 947fbce | 2017-11-15 22:50:23 -0800 | [diff] [blame] | 37 | using android::base::StringPrintf; |
yro | b0378b0 | 2017-11-09 20:36:25 -0800 | [diff] [blame] | 38 | using android::util::FIELD_COUNT_REPEATED; |
yro | 17adac9 | 2017-11-08 23:16:29 -0800 | [diff] [blame] | 39 | using android::util::FIELD_TYPE_BOOL; |
| 40 | using android::util::FIELD_TYPE_FLOAT; |
| 41 | using android::util::FIELD_TYPE_INT32; |
| 42 | using android::util::FIELD_TYPE_INT64; |
| 43 | using android::util::FIELD_TYPE_MESSAGE; |
| 44 | using android::util::FIELD_TYPE_STRING; |
| 45 | using android::util::ProtoOutputStream; |
Yao Chen | 44cf27c | 2017-09-14 22:32:50 -0700 | [diff] [blame] | 46 | using std::make_unique; |
| 47 | using std::unique_ptr; |
| 48 | using std::vector; |
Bookatz | 906a35c | 2017-09-20 15:26:44 -0700 | [diff] [blame] | 49 | |
| 50 | namespace android { |
| 51 | namespace os { |
| 52 | namespace statsd { |
Yao Chen | ab273e2 | 2017-09-06 12:53:50 -0700 | [diff] [blame] | 53 | |
yro | 947fbce | 2017-11-15 22:50:23 -0800 | [diff] [blame] | 54 | // for ConfigMetricsReportList |
yro | 17adac9 | 2017-11-08 23:16:29 -0800 | [diff] [blame] | 55 | const int FIELD_ID_CONFIG_KEY = 1; |
yro | 947fbce | 2017-11-15 22:50:23 -0800 | [diff] [blame] | 56 | const int FIELD_ID_REPORTS = 2; |
yro | 17adac9 | 2017-11-08 23:16:29 -0800 | [diff] [blame] | 57 | // for ConfigKey |
| 58 | const int FIELD_ID_UID = 1; |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame] | 59 | const int FIELD_ID_ID = 2; |
yro | 947fbce | 2017-11-15 22:50:23 -0800 | [diff] [blame] | 60 | // for ConfigMetricsReport |
| 61 | const int FIELD_ID_METRICS = 1; |
| 62 | const int FIELD_ID_UID_MAP = 2; |
| 63 | |
yro | 03faf09 | 2017-12-12 00:17:50 -0800 | [diff] [blame] | 64 | #define STATS_DATA_DIR "/data/misc/stats-data" |
yro | 17adac9 | 2017-11-08 23:16:29 -0800 | [diff] [blame] | 65 | |
yro | 31eb67b | 2017-10-24 13:33:21 -0700 | [diff] [blame] | 66 | StatsLogProcessor::StatsLogProcessor(const sp<UidMap>& uidMap, |
Yangster-mac | e2cd6d5 | 2017-11-09 20:38:30 -0800 | [diff] [blame] | 67 | const sp<AnomalyMonitor>& anomalyMonitor, |
Yangster-mac | 2087716 | 2017-12-22 17:19:39 -0800 | [diff] [blame] | 68 | const long timeBaseSec, |
David Chen | 1d7b0cd | 2017-11-15 14:20:04 -0800 | [diff] [blame] | 69 | const std::function<void(const ConfigKey&)>& sendBroadcast) |
Chenjie Yu | 85ed838 | 2017-12-14 16:48:54 -0800 | [diff] [blame] | 70 | : mUidMap(uidMap), |
| 71 | mAnomalyMonitor(anomalyMonitor), |
| 72 | mSendBroadcast(sendBroadcast), |
Yangster-mac | 2087716 | 2017-12-22 17:19:39 -0800 | [diff] [blame] | 73 | mTimeBaseSec(timeBaseSec) { |
yro | 947fbce | 2017-11-15 22:50:23 -0800 | [diff] [blame] | 74 | // On each initialization of StatsLogProcessor, check stats-data directory to see if there is |
| 75 | // any left over data to be read. |
| 76 | StorageManager::sendBroadcast(STATS_DATA_DIR, mSendBroadcast); |
Chenjie Yu | 85ed838 | 2017-12-14 16:48:54 -0800 | [diff] [blame] | 77 | StatsPullerManager statsPullerManager; |
| 78 | statsPullerManager.SetTimeBaseSec(mTimeBaseSec); |
Yao Chen | ab273e2 | 2017-09-06 12:53:50 -0700 | [diff] [blame] | 79 | } |
| 80 | |
Yao Chen | ef99c4f | 2017-09-22 16:26:54 -0700 | [diff] [blame] | 81 | StatsLogProcessor::~StatsLogProcessor() { |
Yao Chen | ab273e2 | 2017-09-06 12:53:50 -0700 | [diff] [blame] | 82 | } |
| 83 | |
Yangster-mac | e2cd6d5 | 2017-11-09 20:38:30 -0800 | [diff] [blame] | 84 | void StatsLogProcessor::onAnomalyAlarmFired( |
| 85 | const uint64_t timestampNs, |
| 86 | unordered_set<sp<const AnomalyAlarm>, SpHash<AnomalyAlarm>> anomalySet) { |
Yangster-mac | b0d0628 | 2018-01-05 15:44:07 -0800 | [diff] [blame] | 87 | std::lock_guard<std::mutex> lock(mMetricsMutex); |
Bookatz | cc5adef | 2017-11-21 14:36:23 -0800 | [diff] [blame] | 88 | for (const auto& itr : mMetricsManagers) { |
| 89 | itr.second->onAnomalyAlarmFired(timestampNs, anomalySet); |
Yangster-mac | e2cd6d5 | 2017-11-09 20:38:30 -0800 | [diff] [blame] | 90 | } |
| 91 | } |
| 92 | |
Yangster-mac | d40053e | 2018-01-09 16:29:22 -0800 | [diff] [blame] | 93 | void StatsLogProcessor::mapIsolatedUidToHostUidIfNecessaryLocked(LogEvent* event) const { |
Yangster-mac | 7ba8fc3 | 2018-01-24 16:16:46 -0800 | [diff] [blame] | 94 | std::set<Field, FieldCmp> uidFields; |
Yangster-mac | 6898580 | 2018-01-21 10:05:09 -0800 | [diff] [blame] | 95 | if (android::util::kAtomsWithAttributionChain.find(event->GetTagId()) != |
| 96 | android::util::kAtomsWithAttributionChain.end()) { |
Yangster-mac | 7ba8fc3 | 2018-01-24 16:16:46 -0800 | [diff] [blame] | 97 | FieldMatcher matcher; |
| 98 | buildAttributionUidFieldMatcher(event->GetTagId(), Position::ANY, &matcher); |
| 99 | findFields(event->getFieldValueMap(), matcher, &uidFields); |
Yangster-mac | 6898580 | 2018-01-21 10:05:09 -0800 | [diff] [blame] | 100 | } else if (android::util::kAtomsWithUidField.find(event->GetTagId()) != |
| 101 | android::util::kAtomsWithUidField.end()) { |
Yangster-mac | 7ba8fc3 | 2018-01-24 16:16:46 -0800 | [diff] [blame] | 102 | FieldMatcher matcher; |
| 103 | buildSimpleAtomFieldMatcher( |
| 104 | event->GetTagId(), 1 /* uid is always the 1st field. */, &matcher); |
| 105 | findFields(event->getFieldValueMap(), matcher, &uidFields); |
Yangster-mac | 6898580 | 2018-01-21 10:05:09 -0800 | [diff] [blame] | 106 | } |
| 107 | |
Yangster-mac | 7ba8fc3 | 2018-01-24 16:16:46 -0800 | [diff] [blame] | 108 | for (const auto& uidField : uidFields) { |
| 109 | DimensionsValue* value = event->findFieldValueOrNull(uidField); |
Yangster-mac | d40053e | 2018-01-09 16:29:22 -0800 | [diff] [blame] | 110 | if (value != nullptr && value->value_case() == DimensionsValue::ValueCase::kValueInt) { |
| 111 | const int uid = mUidMap->getHostUidOrSelf(value->value_int()); |
| 112 | value->set_value_int(uid); |
| 113 | } |
Yao Chen | 312e898 | 2017-12-05 15:29:03 -0800 | [diff] [blame] | 114 | } |
Yangster-mac | d40053e | 2018-01-09 16:29:22 -0800 | [diff] [blame] | 115 | } |
| 116 | |
| 117 | void StatsLogProcessor::onIsolatedUidChangedEventLocked(const LogEvent& event) { |
| 118 | status_t err = NO_ERROR, err2 = NO_ERROR, err3 = NO_ERROR; |
| 119 | bool is_create = event.GetBool(3, &err); |
| 120 | auto parent_uid = int(event.GetLong(1, &err2)); |
| 121 | auto isolated_uid = int(event.GetLong(2, &err3)); |
| 122 | if (err == NO_ERROR && err2 == NO_ERROR && err3 == NO_ERROR) { |
| 123 | if (is_create) { |
| 124 | mUidMap->assignIsolatedUid(isolated_uid, parent_uid); |
| 125 | } else { |
| 126 | mUidMap->removeIsolatedUid(isolated_uid, parent_uid); |
| 127 | } |
| 128 | } else { |
| 129 | ALOGE("Failed to parse uid in the isolated uid change event."); |
| 130 | } |
| 131 | } |
| 132 | |
| 133 | // TODO: what if statsd service restarts? How do we know what logs are already processed before? |
| 134 | void StatsLogProcessor::OnLogEvent(LogEvent* event) { |
| 135 | std::lock_guard<std::mutex> lock(mMetricsMutex); |
| 136 | StatsdStats::getInstance().noteAtomLogged( |
| 137 | event->GetTagId(), event->GetTimestampNs() / NS_PER_SEC); |
| 138 | |
David Chen | 2158296 | 2017-11-01 17:32:46 -0700 | [diff] [blame] | 139 | // Hard-coded logic to update the isolated uid's in the uid-map. |
Stefan Lafon | ae2df01 | 2017-11-14 09:17:21 -0800 | [diff] [blame] | 140 | // The field numbers need to be currently updated by hand with atoms.proto |
Yangster-mac | d40053e | 2018-01-09 16:29:22 -0800 | [diff] [blame] | 141 | if (event->GetTagId() == android::util::ISOLATED_UID_CHANGED) { |
| 142 | onIsolatedUidChangedEventLocked(*event); |
David Chen | cfc311d | 2018-01-23 17:55:54 -0800 | [diff] [blame] | 143 | } |
| 144 | |
| 145 | if (mMetricsManagers.empty()) { |
| 146 | return; |
| 147 | } |
| 148 | |
| 149 | if (event->GetTagId() != android::util::ISOLATED_UID_CHANGED) { |
Yangster-mac | d40053e | 2018-01-09 16:29:22 -0800 | [diff] [blame] | 150 | // Map the isolated uid to host uid if necessary. |
| 151 | mapIsolatedUidToHostUidIfNecessaryLocked(event); |
| 152 | } |
| 153 | |
| 154 | // pass the event to metrics managers. |
| 155 | for (auto& pair : mMetricsManagers) { |
| 156 | pair.second->onLogEvent(*event); |
| 157 | flushIfNecessaryLocked(event->GetTimestampNs(), pair.first, *(pair.second)); |
David Chen | 2158296 | 2017-11-01 17:32:46 -0700 | [diff] [blame] | 158 | } |
Yao Chen | ab273e2 | 2017-09-06 12:53:50 -0700 | [diff] [blame] | 159 | } |
| 160 | |
Joe Onorato | 9fc9edf | 2017-10-15 20:08:52 -0700 | [diff] [blame] | 161 | void StatsLogProcessor::OnConfigUpdated(const ConfigKey& key, const StatsdConfig& config) { |
Yangster-mac | b0d0628 | 2018-01-05 15:44:07 -0800 | [diff] [blame] | 162 | std::lock_guard<std::mutex> lock(mMetricsMutex); |
Yao Chen | b356151 | 2017-11-21 18:07:17 -0800 | [diff] [blame] | 163 | ALOGD("Updated configuration for key %s", key.ToString().c_str()); |
Yao Chen | d10f7b1 | 2017-12-18 12:53:50 -0800 | [diff] [blame] | 164 | sp<MetricsManager> newMetricsManager = new MetricsManager(key, config, mTimeBaseSec, mUidMap); |
Joe Onorato | 9fc9edf | 2017-10-15 20:08:52 -0700 | [diff] [blame] | 165 | auto it = mMetricsManagers.find(key); |
Yao Chen | 288c600 | 2017-12-12 13:43:18 -0800 | [diff] [blame] | 166 | if (it == mMetricsManagers.end() && mMetricsManagers.size() > StatsdStats::kMaxConfigCount) { |
Yao Chen | b356151 | 2017-11-21 18:07:17 -0800 | [diff] [blame] | 167 | ALOGE("Can't accept more configs!"); |
| 168 | return; |
Yao Chen | 44cf27c | 2017-09-14 22:32:50 -0700 | [diff] [blame] | 169 | } |
| 170 | |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 171 | if (newMetricsManager->isConfigValid()) { |
David Chen | d689689 | 2017-10-25 11:49:03 -0700 | [diff] [blame] | 172 | mUidMap->OnConfigUpdated(key); |
Yangster-mac | e2cd6d5 | 2017-11-09 20:38:30 -0800 | [diff] [blame] | 173 | newMetricsManager->setAnomalyMonitor(mAnomalyMonitor); |
Yao Chen | 147ce60 | 2017-12-22 14:35:34 -0800 | [diff] [blame] | 174 | if (newMetricsManager->shouldAddUidMapListener()) { |
Yao Chen | d10f7b1 | 2017-12-18 12:53:50 -0800 | [diff] [blame] | 175 | // We have to add listener after the MetricsManager is constructed because it's |
| 176 | // not safe to create wp or sp from this pointer inside its constructor. |
| 177 | mUidMap->addListener(newMetricsManager.get()); |
| 178 | } |
| 179 | mMetricsManagers[key] = newMetricsManager; |
Joe Onorato | 9fc9edf | 2017-10-15 20:08:52 -0700 | [diff] [blame] | 180 | // Why doesn't this work? mMetricsManagers.insert({key, std::move(newMetricsManager)}); |
Yao Chen | b356151 | 2017-11-21 18:07:17 -0800 | [diff] [blame] | 181 | VLOG("StatsdConfig valid"); |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 182 | } else { |
| 183 | // If there is any error in the config, don't use it. |
Yao Chen | b356151 | 2017-11-21 18:07:17 -0800 | [diff] [blame] | 184 | ALOGE("StatsdConfig NOT valid"); |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 185 | } |
yro | 00698da | 2017-09-15 10:06:40 -0700 | [diff] [blame] | 186 | } |
Bookatz | 906a35c | 2017-09-20 15:26:44 -0700 | [diff] [blame] | 187 | |
Yangster | 7c334a1 | 2017-11-22 14:24:24 -0800 | [diff] [blame] | 188 | size_t StatsLogProcessor::GetMetricsSize(const ConfigKey& key) const { |
Yangster-mac | b0d0628 | 2018-01-05 15:44:07 -0800 | [diff] [blame] | 189 | std::lock_guard<std::mutex> lock(mMetricsMutex); |
Yao Chen | 729093d | 2017-10-16 10:33:26 -0700 | [diff] [blame] | 190 | auto it = mMetricsManagers.find(key); |
| 191 | if (it == mMetricsManagers.end()) { |
| 192 | ALOGW("Config source %s does not exist", key.ToString().c_str()); |
David Chen | 1d7b0cd | 2017-11-15 14:20:04 -0800 | [diff] [blame] | 193 | return 0; |
Yao Chen | 729093d | 2017-10-16 10:33:26 -0700 | [diff] [blame] | 194 | } |
David Chen | 1d7b0cd | 2017-11-15 14:20:04 -0800 | [diff] [blame] | 195 | return it->second->byteSize(); |
| 196 | } |
| 197 | |
Yao Chen | 884c8c1 | 2018-01-26 10:36:25 -0800 | [diff] [blame] | 198 | void StatsLogProcessor::dumpStates(FILE* out, bool verbose) { |
| 199 | std::lock_guard<std::mutex> lock(mMetricsMutex); |
| 200 | fprintf(out, "MetricsManager count: %lu\n", (unsigned long)mMetricsManagers.size()); |
| 201 | for (auto metricsManager : mMetricsManagers) { |
| 202 | metricsManager.second->dumpStates(out, verbose); |
| 203 | } |
| 204 | } |
| 205 | |
Yao Chen | 147ce60 | 2017-12-22 14:35:34 -0800 | [diff] [blame] | 206 | void StatsLogProcessor::onDumpReport(const ConfigKey& key, const uint64_t& dumpTimeStampNs, |
| 207 | ConfigMetricsReportList* report) { |
Yangster-mac | b0d0628 | 2018-01-05 15:44:07 -0800 | [diff] [blame] | 208 | std::lock_guard<std::mutex> lock(mMetricsMutex); |
Yangster-mac | 2087716 | 2017-12-22 17:19:39 -0800 | [diff] [blame] | 209 | auto it = mMetricsManagers.find(key); |
| 210 | if (it == mMetricsManagers.end()) { |
| 211 | ALOGW("Config source %s does not exist", key.ToString().c_str()); |
| 212 | return; |
| 213 | } |
| 214 | report->mutable_config_key()->set_uid(key.GetUid()); |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame] | 215 | report->mutable_config_key()->set_id(key.GetId()); |
Yangster-mac | 2087716 | 2017-12-22 17:19:39 -0800 | [diff] [blame] | 216 | ConfigMetricsReport* configMetricsReport = report->add_reports(); |
| 217 | it->second->onDumpReport(dumpTimeStampNs, configMetricsReport); |
| 218 | // TODO: dump uid mapping. |
| 219 | } |
| 220 | |
David Chen | 1d7b0cd | 2017-11-15 14:20:04 -0800 | [diff] [blame] | 221 | void StatsLogProcessor::onDumpReport(const ConfigKey& key, vector<uint8_t>* outData) { |
Yangster-mac | b0d0628 | 2018-01-05 15:44:07 -0800 | [diff] [blame] | 222 | std::lock_guard<std::mutex> lock(mMetricsMutex); |
Yangster-mac | 8617950 | 2018-01-23 15:47:15 -0800 | [diff] [blame] | 223 | onDumpReportLocked(key, outData); |
| 224 | } |
| 225 | |
| 226 | void StatsLogProcessor::onDumpReportLocked(const ConfigKey& key, vector<uint8_t>* outData) { |
David Chen | 1d7b0cd | 2017-11-15 14:20:04 -0800 | [diff] [blame] | 227 | auto it = mMetricsManagers.find(key); |
| 228 | if (it == mMetricsManagers.end()) { |
| 229 | ALOGW("Config source %s does not exist", key.ToString().c_str()); |
| 230 | return; |
| 231 | } |
| 232 | |
| 233 | // This allows another broadcast to be sent within the rate-limit period if we get close to |
| 234 | // filling the buffer again soon. |
David Chen | 1d7b0cd | 2017-11-15 14:20:04 -0800 | [diff] [blame] | 235 | mLastBroadcastTimes.erase(key); |
Yao Chen | 729093d | 2017-10-16 10:33:26 -0700 | [diff] [blame] | 236 | |
yro | 17adac9 | 2017-11-08 23:16:29 -0800 | [diff] [blame] | 237 | ProtoOutputStream proto; |
| 238 | |
yro | 947fbce | 2017-11-15 22:50:23 -0800 | [diff] [blame] | 239 | // Start of ConfigKey. |
yro | 17adac9 | 2017-11-08 23:16:29 -0800 | [diff] [blame] | 240 | long long configKeyToken = proto.start(FIELD_TYPE_MESSAGE | FIELD_ID_CONFIG_KEY); |
| 241 | proto.write(FIELD_TYPE_INT32 | FIELD_ID_UID, key.GetUid()); |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame] | 242 | proto.write(FIELD_TYPE_INT64 | FIELD_ID_ID, (long long)key.GetId()); |
yro | 17adac9 | 2017-11-08 23:16:29 -0800 | [diff] [blame] | 243 | proto.end(configKeyToken); |
yro | 947fbce | 2017-11-15 22:50:23 -0800 | [diff] [blame] | 244 | // End of ConfigKey. |
yro | 17adac9 | 2017-11-08 23:16:29 -0800 | [diff] [blame] | 245 | |
yro | 947fbce | 2017-11-15 22:50:23 -0800 | [diff] [blame] | 246 | // Start of ConfigMetricsReport (reports). |
| 247 | long long reportsToken = |
| 248 | proto.start(FIELD_TYPE_MESSAGE | FIELD_COUNT_REPEATED | FIELD_ID_REPORTS); |
| 249 | |
| 250 | // First, fill in ConfigMetricsReport using current data on memory, which |
| 251 | // starts from filling in StatsLogReport's. |
Yao Chen | 288c600 | 2017-12-12 13:43:18 -0800 | [diff] [blame] | 252 | it->second->onDumpReport(&proto); |
yro | 17adac9 | 2017-11-08 23:16:29 -0800 | [diff] [blame] | 253 | |
| 254 | // Fill in UidMap. |
| 255 | auto uidMap = mUidMap->getOutput(key); |
| 256 | const int uidMapSize = uidMap.ByteSize(); |
| 257 | char uidMapBuffer[uidMapSize]; |
| 258 | uidMap.SerializeToArray(&uidMapBuffer[0], uidMapSize); |
| 259 | proto.write(FIELD_TYPE_MESSAGE | FIELD_ID_UID_MAP, uidMapBuffer, uidMapSize); |
| 260 | |
yro | 947fbce | 2017-11-15 22:50:23 -0800 | [diff] [blame] | 261 | // End of ConfigMetricsReport (reports). |
| 262 | proto.end(reportsToken); |
| 263 | |
| 264 | // Then, check stats-data directory to see there's any file containing |
| 265 | // ConfigMetricsReport from previous shutdowns to concatenate to reports. |
yro | 98a2850 | 2018-01-18 17:00:14 -0800 | [diff] [blame] | 266 | StorageManager::appendConfigMetricsReport(proto); |
yro | 947fbce | 2017-11-15 22:50:23 -0800 | [diff] [blame] | 267 | |
David Chen | 1d7b0cd | 2017-11-15 14:20:04 -0800 | [diff] [blame] | 268 | if (outData != nullptr) { |
| 269 | outData->clear(); |
| 270 | outData->resize(proto.size()); |
| 271 | size_t pos = 0; |
| 272 | auto iter = proto.data(); |
| 273 | while (iter.readBuffer() != NULL) { |
| 274 | size_t toRead = iter.currentToRead(); |
| 275 | std::memcpy(&((*outData)[pos]), iter.readBuffer(), toRead); |
| 276 | pos += toRead; |
| 277 | iter.rp()->move(toRead); |
| 278 | } |
yro | 17adac9 | 2017-11-08 23:16:29 -0800 | [diff] [blame] | 279 | } |
Yao Chen | 69f1baf | 2017-11-27 17:25:36 -0800 | [diff] [blame] | 280 | StatsdStats::getInstance().noteMetricsReportSent(key); |
Yao Chen | 729093d | 2017-10-16 10:33:26 -0700 | [diff] [blame] | 281 | } |
| 282 | |
Joe Onorato | 9fc9edf | 2017-10-15 20:08:52 -0700 | [diff] [blame] | 283 | void StatsLogProcessor::OnConfigRemoved(const ConfigKey& key) { |
Yangster-mac | b0d0628 | 2018-01-05 15:44:07 -0800 | [diff] [blame] | 284 | std::lock_guard<std::mutex> lock(mMetricsMutex); |
Joe Onorato | 9fc9edf | 2017-10-15 20:08:52 -0700 | [diff] [blame] | 285 | auto it = mMetricsManagers.find(key); |
| 286 | if (it != mMetricsManagers.end()) { |
Joe Onorato | 9fc9edf | 2017-10-15 20:08:52 -0700 | [diff] [blame] | 287 | mMetricsManagers.erase(it); |
David Chen | d689689 | 2017-10-25 11:49:03 -0700 | [diff] [blame] | 288 | mUidMap->OnConfigRemoved(key); |
Joe Onorato | 9fc9edf | 2017-10-15 20:08:52 -0700 | [diff] [blame] | 289 | } |
Yao Chen | b356151 | 2017-11-21 18:07:17 -0800 | [diff] [blame] | 290 | StatsdStats::getInstance().noteConfigRemoved(key); |
David Chen | 1d7b0cd | 2017-11-15 14:20:04 -0800 | [diff] [blame] | 291 | |
David Chen | 1d7b0cd | 2017-11-15 14:20:04 -0800 | [diff] [blame] | 292 | mLastBroadcastTimes.erase(key); |
Joe Onorato | 9fc9edf | 2017-10-15 20:08:52 -0700 | [diff] [blame] | 293 | } |
| 294 | |
Yangster-mac | b0d0628 | 2018-01-05 15:44:07 -0800 | [diff] [blame] | 295 | void StatsLogProcessor::flushIfNecessaryLocked( |
| 296 | uint64_t timestampNs, const ConfigKey& key, MetricsManager& metricsManager) { |
David Chen | d9269e2 | 2017-12-05 13:43:51 -0800 | [diff] [blame] | 297 | auto lastCheckTime = mLastByteSizeTimes.find(key); |
| 298 | if (lastCheckTime != mLastByteSizeTimes.end()) { |
| 299 | if (timestampNs - lastCheckTime->second < StatsdStats::kMinByteSizeCheckPeriodNs) { |
| 300 | return; |
| 301 | } |
| 302 | } |
| 303 | |
| 304 | // We suspect that the byteSize() computation is expensive, so we set a rate limit. |
| 305 | size_t totalBytes = metricsManager.byteSize(); |
| 306 | mLastByteSizeTimes[key] = timestampNs; |
| 307 | if (totalBytes > |
| 308 | StatsdStats::kMaxMetricsBytesPerConfig) { // Too late. We need to start clearing data. |
Yao Chen | 288c600 | 2017-12-12 13:43:18 -0800 | [diff] [blame] | 309 | // TODO(b/70571383): By 12/15/2017 add API to drop data directly |
| 310 | ProtoOutputStream proto; |
| 311 | metricsManager.onDumpReport(&proto); |
David Chen | 1294295 | 2017-12-04 14:28:43 -0800 | [diff] [blame] | 312 | StatsdStats::getInstance().noteDataDropped(key); |
| 313 | VLOG("StatsD had to toss out metrics for %s", key.ToString().c_str()); |
David Chen | d9269e2 | 2017-12-05 13:43:51 -0800 | [diff] [blame] | 314 | } else if (totalBytes > .9 * StatsdStats::kMaxMetricsBytesPerConfig) { |
| 315 | // Send broadcast so that receivers can pull data. |
| 316 | auto lastBroadcastTime = mLastBroadcastTimes.find(key); |
| 317 | if (lastBroadcastTime != mLastBroadcastTimes.end()) { |
| 318 | if (timestampNs - lastBroadcastTime->second < StatsdStats::kMinBroadcastPeriodNs) { |
| 319 | VLOG("StatsD would've sent a broadcast but the rate limit stopped us."); |
David Chen | 1d7b0cd | 2017-11-15 14:20:04 -0800 | [diff] [blame] | 320 | return; |
| 321 | } |
| 322 | } |
| 323 | mLastBroadcastTimes[key] = timestampNs; |
Yao Chen | b356151 | 2017-11-21 18:07:17 -0800 | [diff] [blame] | 324 | VLOG("StatsD requesting broadcast for %s", key.ToString().c_str()); |
David Chen | 1d7b0cd | 2017-11-15 14:20:04 -0800 | [diff] [blame] | 325 | mSendBroadcast(key); |
Yao Chen | b356151 | 2017-11-21 18:07:17 -0800 | [diff] [blame] | 326 | StatsdStats::getInstance().noteBroadcastSent(key); |
yro | 31eb67b | 2017-10-24 13:33:21 -0700 | [diff] [blame] | 327 | } |
| 328 | } |
| 329 | |
yro | 947fbce | 2017-11-15 22:50:23 -0800 | [diff] [blame] | 330 | void StatsLogProcessor::WriteDataToDisk() { |
Yangster-mac | b0d0628 | 2018-01-05 15:44:07 -0800 | [diff] [blame] | 331 | std::lock_guard<std::mutex> lock(mMetricsMutex); |
yro | 947fbce | 2017-11-15 22:50:23 -0800 | [diff] [blame] | 332 | for (auto& pair : mMetricsManagers) { |
| 333 | const ConfigKey& key = pair.first; |
| 334 | vector<uint8_t> data; |
Yangster-mac | 8617950 | 2018-01-23 15:47:15 -0800 | [diff] [blame] | 335 | onDumpReportLocked(key, &data); |
yro | 947fbce | 2017-11-15 22:50:23 -0800 | [diff] [blame] | 336 | // TODO: Add a guardrail to prevent accumulation of file on disk. |
yro | 98a2850 | 2018-01-18 17:00:14 -0800 | [diff] [blame] | 337 | string file_name = StringPrintf("%s/%ld_%d_%lld", STATS_DATA_DIR, time(nullptr), |
| 338 | key.GetUid(), (long long)key.GetId()); |
yro | 947fbce | 2017-11-15 22:50:23 -0800 | [diff] [blame] | 339 | StorageManager::writeFile(file_name.c_str(), &data[0], data.size()); |
| 340 | } |
| 341 | } |
| 342 | |
Yao Chen | ef99c4f | 2017-09-22 16:26:54 -0700 | [diff] [blame] | 343 | } // namespace statsd |
| 344 | } // namespace os |
| 345 | } // namespace android |