storaged: monitor per-uid IO usage
Add uid_monitor class to query /proc/uid_io/stats periodically.
Add a log tag to record any UID that exceeds IO threshold.
Test: adb shell storaged -u
Bug: 34198239
Change-Id: I53568c30dbefe2f4bdb18054d3dedb30b4133d8b
diff --git a/storaged/storaged_uid_monitor.cpp b/storaged/storaged_uid_monitor.cpp
new file mode 100644
index 0000000..4105dae
--- /dev/null
+++ b/storaged/storaged_uid_monitor.cpp
@@ -0,0 +1,152 @@
+/*
+ * Copyright (C) 2016 The Android Open Source Project
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#define LOG_TAG "storaged"
+
+#include <stdint.h>
+#include <time.h>
+
+#include <string>
+#include <sstream>
+#include <unordered_map>
+
+#include <android-base/file.h>
+#include <android-base/logging.h>
+#include <android-base/macros.h>
+#include <android-base/stringprintf.h>
+#include <log/log_event_list.h>
+#include <packagelistparser/packagelistparser.h>
+
+#include "storaged.h"
+#include "storaged_uid_monitor.h"
+
+static const uint64_t io_alert_threshold = 1024 * 1024 * 1024; // 1GB
+
+using namespace android;
+using namespace android::base;
+
+static bool packagelist_parse_cb(pkg_info* info, void* userdata)
+{
+ std::unordered_map<uint32_t, struct uid_info>* uids =
+ reinterpret_cast<std::unordered_map<uint32_t, struct uid_info>*>(userdata);
+
+ if (uids->find(info->uid) != uids->end()) {
+ (*uids)[info->uid].name = info->name;
+ }
+
+ packagelist_free(info);
+ return true;
+}
+
+void uid_monitor::set_last_uids(std::unordered_map<uint32_t, struct uid_info>&& uids,
+ uint64_t ts)
+{
+ last_uids = uids;
+ last_report_ts = ts;
+}
+
+std::unordered_map<uint32_t, struct uid_info> uid_monitor::get_uids()
+{
+ std::unordered_map<uint32_t, struct uid_info> uids;
+ std::string buffer;
+ if (!android::base::ReadFileToString(UID_IO_STATS_PATH, &buffer)) {
+ PLOG_TO(SYSTEM, ERROR) << UID_IO_STATS_PATH << ": ReadFileToString failed";
+ return uids;
+ }
+
+ std::stringstream ss(buffer);
+ struct uid_info u;
+ bool refresh_uid = false;
+
+ while (ss >> u.uid) {
+ ss >> u.io[UID_FOREGROUND].rchar >> u.io[UID_FOREGROUND].wchar
+ >> u.io[UID_FOREGROUND].read_bytes >> u.io[UID_FOREGROUND].write_bytes
+ >> u.io[UID_BACKGROUND].rchar >> u.io[UID_BACKGROUND].wchar
+ >> u.io[UID_BACKGROUND].read_bytes >> u.io[UID_BACKGROUND].write_bytes;
+
+ if (!ss.good()) {
+ ss.clear(std::ios_base::badbit);
+ break;
+ }
+
+ if (last_uids.find(u.uid) == last_uids.end()) {
+ refresh_uid = true;
+ u.name = std::to_string(u.uid);
+ } else {
+ u.name = last_uids[u.uid].name;
+ }
+ uids[u.uid] = u;
+ }
+
+ if (!ss.eof() || ss.bad()) {
+ uids.clear();
+ LOG_TO(SYSTEM, ERROR) << "read UID IO stats failed";
+ }
+
+ if (refresh_uid) {
+ packagelist_parse(packagelist_parse_cb, &uids);
+ }
+
+ return uids;
+}
+
+void uid_monitor::report()
+{
+ struct timespec ts;
+
+ // Use monotonic to exclude suspend time so that we measure IO bytes/sec
+ // when system is running.
+ if (clock_gettime(CLOCK_MONOTONIC, &ts) < 0) {
+ PLOG_TO(SYSTEM, ERROR) << "clock_gettime() failed";
+ return;
+ }
+
+ uint64_t curr_ts = ts.tv_sec * NS_PER_SEC + ts.tv_nsec;
+ uint64_t ts_delta = curr_ts - last_report_ts;
+ uint64_t adjusted_threshold = io_alert_threshold * ((double)ts_delta / interval / NS_PER_SEC);
+
+ std::unordered_map<uint32_t, struct uid_info> uids = get_uids();
+ if (uids.empty()) {
+ return;
+ }
+
+ for (const auto& it : uids) {
+ const struct uid_info& uid = it.second;
+ uint64_t bg_read_delta = uid.io[UID_BACKGROUND].read_bytes -
+ last_uids[uid.uid].io[UID_BACKGROUND].read_bytes;
+ uint64_t bg_write_delta = uid.io[UID_BACKGROUND].write_bytes -
+ last_uids[uid.uid].io[UID_BACKGROUND].write_bytes;
+
+ if (bg_read_delta + bg_write_delta >= adjusted_threshold) {
+ android_log_event_list(EVENTLOGTAG_UID_IO_ALERT)
+ << uid.name << bg_read_delta << bg_write_delta
+ << uint64_t(ts_delta / NS_PER_SEC) << LOG_ID_EVENTS;
+ }
+ }
+
+ set_last_uids(std::move(uids), curr_ts);
+}
+
+uid_monitor::uid_monitor()
+{
+ struct timespec ts;
+
+ if (clock_gettime(CLOCK_MONOTONIC, &ts) < 0) {
+ PLOG_TO(SYSTEM, ERROR) << "clock_gettime() failed";
+ return;
+ }
+ last_report_ts = ts.tv_sec * NS_PER_SEC + ts.tv_nsec;
+}