1/*
2 * Copyright (C) 2016 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
17#include <stdint.h>
18
19#include <vector>
20
21#include <android-base/parseint.h>
22#include <android-base/parsedouble.h>
23#include <binder/IBinder.h>
24#include <binder/IInterface.h>
25
26#include <binder/IPCThreadState.h>
27#include <binder/IServiceManager.h>
28#include <binder/PermissionCache.h>
29#include <private/android_filesystem_config.h>
30
31#include <storaged.h>
32#include <storaged_service.h>
33
34using namespace android::base;
35
36extern sp<storaged_t> storaged;
37
38std::vector<struct uid_info> BpStoraged::dump_uids(const char* /*option*/) {
39    Parcel data, reply;
40    data.writeInterfaceToken(IStoraged::getInterfaceDescriptor());
41
42    remote()->transact(DUMPUIDS, data, &reply);
43
44    uint32_t res_size = reply.readInt32();
45    std::vector<struct uid_info> res(res_size);
46    for (auto&& uid : res) {
47        uid.uid = reply.readInt32();
48        uid.name = reply.readCString();
49        reply.read(&uid.io, sizeof(uid.io));
50    }
51    return res;
52}
53IMPLEMENT_META_INTERFACE(Storaged, "Storaged");
54
55status_t BnStoraged::onTransact(uint32_t code, const Parcel& data, Parcel* reply, uint32_t flags) {
56    switch(code) {
57        case DUMPUIDS: {
58                if (!data.checkInterface(this))
59                    return BAD_TYPE;
60                std::vector<struct uid_info> res = dump_uids(NULL);
61                reply->writeInt32(res.size());
62                for (auto uid : res) {
63                    reply->writeInt32(uid.uid);
64                    reply->writeCString(uid.name.c_str());
65                    reply->write(&uid.io, sizeof(uid.io));
66                }
67                return NO_ERROR;
68            }
69            break;
70        default:
71            return BBinder::onTransact(code, data, reply, flags);
72    }
73}
74
75std::vector<struct uid_info> Storaged::dump_uids(const char* /* option */) {
76    std::vector<struct uid_info> uids_v;
77    std::unordered_map<uint32_t, struct uid_info> uids_m = storaged->get_uids();
78
79    for (const auto& it : uids_m) {
80        uids_v.push_back(it.second);
81    }
82    return uids_v;
83}
84
85status_t Storaged::dump(int fd, const Vector<String16>& args) {
86    IPCThreadState* self = IPCThreadState::self();
87    const int pid = self->getCallingPid();
88    const int uid = self->getCallingUid();
89    if ((uid != AID_SHELL) &&
90        !PermissionCache::checkPermission(
91                String16("android.permission.DUMP"), pid, uid)) {
92        return PERMISSION_DENIED;
93    }
94
95    double hours = 0;
96    int time_window = 0;
97    uint64_t threshold = 0;
98    bool force_report = false;
99    for (size_t i = 0; i < args.size(); i++) {
100        const auto& arg = args[i];
101        if (arg == String16("--hours")) {
102            if (++i >= args.size())
103                break;
104            if(!ParseDouble(String8(args[i]).c_str(), &hours))
105                return BAD_VALUE;
106            continue;
107        }
108        if (arg == String16("--time_window")) {
109            if (++i >= args.size())
110                break;
111            if(!ParseInt(String8(args[i]).c_str(), &time_window))
112                return BAD_VALUE;
113            continue;
114        }
115        if (arg == String16("--threshold")) {
116            if (++i >= args.size())
117                break;
118            if(!ParseUint(String8(args[i]).c_str(), &threshold))
119                return BAD_VALUE;
120            continue;
121        }
122        if (arg == String16("--force")) {
123            force_report = true;
124            continue;
125        }
126    }
127
128    uint64_t last_ts = 0;
129    const std::map<uint64_t, struct uid_records>& records =
130                storaged->get_uid_records(hours, threshold, force_report);
131    for (const auto& it : records) {
132        if (last_ts != it.second.start_ts) {
133            dprintf(fd, "%llu", (unsigned long long)it.second.start_ts);
134        }
135        dprintf(fd, ",%llu\n", (unsigned long long)it.first);
136        last_ts = it.first;
137
138        for (const auto& record : it.second.entries) {
139            dprintf(fd, "%s %ju %ju %ju %ju %ju %ju %ju %ju\n",
140                record.name.c_str(),
141                record.ios.bytes[READ][FOREGROUND][CHARGER_OFF],
142                record.ios.bytes[WRITE][FOREGROUND][CHARGER_OFF],
143                record.ios.bytes[READ][BACKGROUND][CHARGER_OFF],
144                record.ios.bytes[WRITE][BACKGROUND][CHARGER_OFF],
145                record.ios.bytes[READ][FOREGROUND][CHARGER_ON],
146                record.ios.bytes[WRITE][FOREGROUND][CHARGER_ON],
147                record.ios.bytes[READ][BACKGROUND][CHARGER_ON],
148                record.ios.bytes[WRITE][BACKGROUND][CHARGER_ON]);
149        }
150    }
151
152    if (time_window) {
153        storaged->update_uid_io_interval(time_window);
154    }
155
156    return NO_ERROR;
157}
158
159sp<IStoraged> get_storaged_service() {
160    sp<IServiceManager> sm = defaultServiceManager();
161    if (sm == NULL) return NULL;
162
163    sp<IBinder> binder = sm->getService(String16("storaged"));
164    if (binder == NULL) return NULL;
165
166    sp<IStoraged> storaged = interface_cast<IStoraged>(binder);
167
168    return storaged;
169}
170