blob: fe6d49e0824ef16b2b5dbfa173fa313df811d858 [file] [log] [blame]
Chenbo Fengf2759682017-10-10 17:31:57 -07001/*
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
Chenbo Fengc10a8a42017-12-15 13:56:33 -080017#define LOG_TAG "TrafficController"
Chenbo Feng33cc1032017-10-23 15:16:37 -070018#include <inttypes.h>
Chenbo Fengf2759682017-10-10 17:31:57 -070019#include <linux/bpf.h>
20#include <linux/if_ether.h>
21#include <linux/in.h>
22#include <linux/inet_diag.h>
Chenbo Feng116d0552017-12-04 17:25:19 -080023#include <linux/netlink.h>
24#include <linux/sock_diag.h>
Chenbo Fengf2759682017-10-10 17:31:57 -070025#include <linux/unistd.h>
Chenbo Fengc10a8a42017-12-15 13:56:33 -080026#include <net/if.h>
Chenbo Fengf2759682017-10-10 17:31:57 -070027#include <stdlib.h>
28#include <string.h>
29#include <sys/socket.h>
Chenbo Fengc10a8a42017-12-15 13:56:33 -080030#include <sys/stat.h>
Chenbo Feng05393d82018-01-09 15:18:43 -080031#include <sys/types.h>
Chenbo Feng33cc1032017-10-23 15:16:37 -070032#include <sys/utsname.h>
Chenbo Feng05393d82018-01-09 15:18:43 -080033#include <sys/wait.h>
Chenbo Feng89c12f12018-03-21 10:29:18 -070034#include <mutex>
Chenbo Fengc10a8a42017-12-15 13:56:33 -080035#include <unordered_set>
36#include <vector>
37
38#include <android-base/stringprintf.h>
Chenbo Fengef297172018-03-26 10:53:33 -070039#include <android-base/strings.h>
Chenbo Fengc10a8a42017-12-15 13:56:33 -080040#include <android-base/unique_fd.h>
Chenbo Feng05393d82018-01-09 15:18:43 -080041#include <logwrap/logwrap.h>
Chenbo Fengc10a8a42017-12-15 13:56:33 -080042#include <netdutils/StatusOr.h>
Chenbo Fengf2759682017-10-10 17:31:57 -070043
Chenbo Feng116d0552017-12-04 17:25:19 -080044#include <netdutils/Misc.h>
45#include <netdutils/Syscalls.h>
Suren Baghdasaryane072a3c2019-01-16 14:36:07 -080046#include <processgroup/processgroup.h>
Chenbo Fengf2759682017-10-10 17:31:57 -070047#include "TrafficController.h"
Chenbo Feng4f6c2372018-04-26 10:37:55 -070048#include "bpf/BpfMap.h"
Chenbo Feng116d0552017-12-04 17:25:19 -080049
Chenbo Feng89c12f12018-03-21 10:29:18 -070050#include "FirewallController.h"
Chenbo Feng7e974052018-02-28 22:57:21 -080051#include "InterfaceController.h"
Chenbo Feng116d0552017-12-04 17:25:19 -080052#include "NetlinkListener.h"
Luke Huangb257d612019-03-14 21:19:13 +080053#include "netdutils/DumpWriter.h"
Chenbo Feng33cc1032017-10-23 15:16:37 -070054#include "qtaguid/qtaguid.h"
Chenbo Fengf2759682017-10-10 17:31:57 -070055
Bernie Innocenti7e25ec02018-07-02 19:32:17 +090056using namespace android::bpf; // NOLINT(google-build-using-namespace): grandfathered
Chenbo Fengf2759682017-10-10 17:31:57 -070057
58namespace android {
59namespace net {
60
Chenbo Fengc10a8a42017-12-15 13:56:33 -080061using base::StringPrintf;
62using base::unique_fd;
Luke Huangb257d612019-03-14 21:19:13 +080063using netdutils::DumpWriter;
Chenbo Feng116d0552017-12-04 17:25:19 -080064using netdutils::extract;
Luke Huangb257d612019-03-14 21:19:13 +080065using netdutils::ScopedIndent;
Chenbo Feng116d0552017-12-04 17:25:19 -080066using netdutils::Slice;
67using netdutils::sSyscalls;
Chenbo Fengc10a8a42017-12-15 13:56:33 -080068using netdutils::Status;
69using netdutils::statusFromErrno;
70using netdutils::StatusOr;
Chenbo Feng116d0552017-12-04 17:25:19 -080071using netdutils::status::ok;
Chenbo Fengc10a8a42017-12-15 13:56:33 -080072
Chenbo Feng116d0552017-12-04 17:25:19 -080073constexpr int kSockDiagMsgType = SOCK_DIAG_BY_FAMILY;
74constexpr int kSockDiagDoneMsgType = NLMSG_DONE;
Chenbo Feng82075e52019-04-12 19:21:47 -070075constexpr int PER_UID_STATS_ENTRIES_LIMIT = 500;
Chenbo Feng05662be2019-05-02 12:47:25 -070076// At most 90% of the stats map may be used by tagged traffic entries. This ensures
77// that 10% of the map is always available to count untagged traffic, one entry per UID.
78// Otherwise, apps would be able to avoid data usage accounting entirely by filling up the
79// map with tagged traffic entries.
Chenbo Fengfd88f9d2019-05-01 12:59:09 -070080constexpr int TOTAL_UID_STATS_ENTRIES_LIMIT = STATS_MAP_SIZE * 0.9;
Chenbo Feng116d0552017-12-04 17:25:19 -080081
Chenbo Fengbf660aa2019-02-26 16:12:27 -080082static_assert(BPF_PERMISSION_INTERNET == INetd::PERMISSION_INTERNET,
83 "Mismatch between BPF and AIDL permissions: PERMISSION_INTERNET");
84static_assert(BPF_PERMISSION_UPDATE_DEVICE_STATS == INetd::PERMISSION_UPDATE_DEVICE_STATS,
85 "Mismatch between BPF and AIDL permissions: PERMISSION_UPDATE_DEVICE_STATS");
Chenbo Fengfd88f9d2019-05-01 12:59:09 -070086static_assert(STATS_MAP_SIZE - TOTAL_UID_STATS_ENTRIES_LIMIT > 100,
87 "The limit for stats map is to high, stats data may be lost due to overflow");
Chenbo Fengbf660aa2019-02-26 16:12:27 -080088
Chenbo Feng48eaed32018-12-26 17:40:21 -080089#define FLAG_MSG_TRANS(result, flag, value) \
90 do { \
91 if (value & flag) { \
92 result.append(StringPrintf(" %s", #flag)); \
93 value &= ~flag; \
94 } \
Chenbo Feng703798e2018-06-15 17:07:59 -070095 } while (0)
96
97const std::string uidMatchTypeToString(uint8_t match) {
98 std::string matchType;
Chenbo Feng48eaed32018-12-26 17:40:21 -080099 FLAG_MSG_TRANS(matchType, HAPPY_BOX_MATCH, match);
100 FLAG_MSG_TRANS(matchType, PENALTY_BOX_MATCH, match);
101 FLAG_MSG_TRANS(matchType, DOZABLE_MATCH, match);
102 FLAG_MSG_TRANS(matchType, STANDBY_MATCH, match);
103 FLAG_MSG_TRANS(matchType, POWERSAVE_MATCH, match);
Rubin Xuec27ff22019-01-08 21:33:03 +0000104 FLAG_MSG_TRANS(matchType, IIF_MATCH, match);
Chenbo Feng703798e2018-06-15 17:07:59 -0700105 if (match) {
106 return StringPrintf("Unknown match: %u", match);
107 }
108 return matchType;
109}
110
Chenbo Fengb4a4fa12019-01-09 17:20:45 -0800111bool TrafficController::hasUpdateDeviceStatsPermission(uid_t uid) {
Chenbo Feng3969c262019-03-19 14:53:57 -0700112 // This implementation is the same logic as method ActivityManager#checkComponentPermission.
113 // It implies that the calling uid can never be the same as PER_USER_RANGE.
114 uint32_t appId = uid % PER_USER_RANGE;
115 return ((appId == AID_ROOT) || (appId == AID_SYSTEM) ||
116 mPrivilegedUser.find(appId) != mPrivilegedUser.end());
Chenbo Fengb4a4fa12019-01-09 17:20:45 -0800117}
118
Chenbo Feng48eaed32018-12-26 17:40:21 -0800119const std::string UidPermissionTypeToString(uint8_t permission) {
Chenbo Fengbf660aa2019-02-26 16:12:27 -0800120 if (permission == INetd::NO_PERMISSIONS) {
121 return "NO_PERMISSIONS";
122 }
Chenbo Feng527908a2019-04-02 20:14:11 -0700123 if (permission == INetd::PERMISSION_UNINSTALLED) {
124 // This should never appear in the map, complain loudly if it does.
125 return "PERMISSION_UNINSTALLED error!";
126 }
Chenbo Feng48eaed32018-12-26 17:40:21 -0800127 std::string permissionType;
Chenbo Fengbf660aa2019-02-26 16:12:27 -0800128 FLAG_MSG_TRANS(permissionType, BPF_PERMISSION_INTERNET, permission);
129 FLAG_MSG_TRANS(permissionType, BPF_PERMISSION_UPDATE_DEVICE_STATS, permission);
Chenbo Feng48eaed32018-12-26 17:40:21 -0800130 if (permission) {
131 return StringPrintf("Unknown permission: %u", permission);
132 }
133 return permissionType;
134}
135
Chenbo Feng49586642018-08-30 18:01:53 -0700136StatusOr<std::unique_ptr<NetlinkListenerInterface>> TrafficController::makeSkDestroyListener() {
Chenbo Feng116d0552017-12-04 17:25:19 -0800137 const auto& sys = sSyscalls.get();
138 ASSIGN_OR_RETURN(auto event, sys.eventfd(0, EFD_CLOEXEC));
139 const int domain = AF_NETLINK;
140 const int type = SOCK_DGRAM | SOCK_CLOEXEC | SOCK_NONBLOCK;
141 const int protocol = NETLINK_INET_DIAG;
142 ASSIGN_OR_RETURN(auto sock, sys.socket(domain, type, protocol));
143
Lorenzo Colitti436efdb2018-07-26 17:20:57 +0900144 // TODO: if too many sockets are closed too quickly, we can overflow the socket buffer, and
145 // some entries in mCookieTagMap will not be freed. In order to fix this we would need to
146 // periodically dump all sockets and remove the tag entries for sockets that have been closed.
147 // For now, set a large-enough buffer that we can close hundreds of sockets without getting
148 // ENOBUFS and leaking mCookieTagMap entries.
149 int rcvbuf = 512 * 1024;
150 auto ret = sys.setsockopt(sock, SOL_SOCKET, SO_RCVBUF, &rcvbuf, sizeof(rcvbuf));
151 if (!ret.ok()) {
152 ALOGW("Failed to set SkDestroyListener buffer size to %d: %s", rcvbuf, ret.msg().c_str());
153 }
154
Chenbo Feng116d0552017-12-04 17:25:19 -0800155 sockaddr_nl addr = {
156 .nl_family = AF_NETLINK,
157 .nl_groups = 1 << (SKNLGRP_INET_TCP_DESTROY - 1) | 1 << (SKNLGRP_INET_UDP_DESTROY - 1) |
158 1 << (SKNLGRP_INET6_TCP_DESTROY - 1) | 1 << (SKNLGRP_INET6_UDP_DESTROY - 1)};
159 RETURN_IF_NOT_OK(sys.bind(sock, addr));
160
161 const sockaddr_nl kernel = {.nl_family = AF_NETLINK};
162 RETURN_IF_NOT_OK(sys.connect(sock, kernel));
163
164 std::unique_ptr<NetlinkListenerInterface> listener =
Chenbo Feng46296702018-08-23 12:20:22 -0700165 std::make_unique<NetlinkListener>(std::move(event), std::move(sock), "SkDestroyListen");
Chenbo Feng116d0552017-12-04 17:25:19 -0800166
167 return listener;
168}
169
Chenbo Feng89c12f12018-03-21 10:29:18 -0700170Status changeOwnerAndMode(const char* path, gid_t group, const char* debugName, bool netdOnly) {
171 int ret = chown(path, AID_ROOT, group);
172 if (ret != 0) return statusFromErrno(errno, StringPrintf("change %s group failed", debugName));
173
174 if (netdOnly) {
175 ret = chmod(path, S_IRWXU);
176 } else {
177 // Allow both netd and system server to obtain map fd from the path.
178 // chmod doesn't grant permission to all processes in that group to
179 // read/write the bpf map. They still need correct sepolicy to
180 // read/write the map.
Chenbo Fengf434e862018-06-27 14:08:39 -0700181 ret = chmod(path, S_IRWXU | S_IRGRP | S_IWGRP);
Chenbo Feng89c12f12018-03-21 10:29:18 -0700182 }
183 if (ret != 0) return statusFromErrno(errno, StringPrintf("change %s mode failed", debugName));
184 return netdutils::status::ok;
185}
186
Chenbo Feng82075e52019-04-12 19:21:47 -0700187TrafficController::TrafficController()
188 : mBpfLevel(getBpfSupportLevel()),
189 mPerUidStatsEntriesLimit(PER_UID_STATS_ENTRIES_LIMIT),
190 mTotalUidStatsEntriesLimit(TOTAL_UID_STATS_ENTRIES_LIMIT) {}
191
192TrafficController::TrafficController(uint32_t perUidLimit, uint32_t totalLimit)
193 : mBpfLevel(getBpfSupportLevel()),
194 mPerUidStatsEntriesLimit(perUidLimit),
195 mTotalUidStatsEntriesLimit(totalLimit) {}
Chenbo Feng89c12f12018-03-21 10:29:18 -0700196
Chenbo Feng89c12f12018-03-21 10:29:18 -0700197Status TrafficController::initMaps() {
Chenbo Feng84d9f9f2019-04-12 15:31:02 -0700198 std::lock_guard guard(mMutex);
Maciej Żenczykowski02554372019-04-01 10:44:54 -0700199 RETURN_IF_NOT_OK(mCookieTagMap.init(COOKIE_TAG_MAP_PATH));
Chenbo Feng89c12f12018-03-21 10:29:18 -0700200 RETURN_IF_NOT_OK(changeOwnerAndMode(COOKIE_TAG_MAP_PATH, AID_NET_BW_ACCT, "CookieTagMap",
201 false));
202
Maciej Żenczykowski02554372019-04-01 10:44:54 -0700203 RETURN_IF_NOT_OK(mUidCounterSetMap.init(UID_COUNTERSET_MAP_PATH));
Chenbo Feng89c12f12018-03-21 10:29:18 -0700204 RETURN_IF_NOT_OK(changeOwnerAndMode(UID_COUNTERSET_MAP_PATH, AID_NET_BW_ACCT,
205 "UidCounterSetMap", false));
206
Maciej Żenczykowski02554372019-04-01 10:44:54 -0700207 RETURN_IF_NOT_OK(mAppUidStatsMap.init(APP_UID_STATS_MAP_PATH));
Chenbo Fengbc4a15f2018-05-11 19:15:15 -0700208 RETURN_IF_NOT_OK(
209 changeOwnerAndMode(APP_UID_STATS_MAP_PATH, AID_NET_BW_STATS, "AppUidStatsMap", false));
210
Maciej Żenczykowski02554372019-04-01 10:44:54 -0700211 RETURN_IF_NOT_OK(mStatsMapA.init(STATS_MAP_A_PATH));
Chenbo Fengf434e862018-06-27 14:08:39 -0700212 RETURN_IF_NOT_OK(changeOwnerAndMode(STATS_MAP_A_PATH, AID_NET_BW_STATS, "StatsMapA", false));
Chenbo Feng89c12f12018-03-21 10:29:18 -0700213
Maciej Żenczykowski02554372019-04-01 10:44:54 -0700214 RETURN_IF_NOT_OK(mStatsMapB.init(STATS_MAP_B_PATH));
Chenbo Fengf434e862018-06-27 14:08:39 -0700215 RETURN_IF_NOT_OK(changeOwnerAndMode(STATS_MAP_B_PATH, AID_NET_BW_STATS, "StatsMapB", false));
Chenbo Feng89c12f12018-03-21 10:29:18 -0700216
Maciej Żenczykowski02554372019-04-01 10:44:54 -0700217 RETURN_IF_NOT_OK(mIfaceIndexNameMap.init(IFACE_INDEX_NAME_MAP_PATH));
Chenbo Feng89c12f12018-03-21 10:29:18 -0700218 RETURN_IF_NOT_OK(changeOwnerAndMode(IFACE_INDEX_NAME_MAP_PATH, AID_NET_BW_STATS,
219 "IfaceIndexNameMap", false));
220
Maciej Żenczykowski02554372019-04-01 10:44:54 -0700221 RETURN_IF_NOT_OK(mIfaceStatsMap.init(IFACE_STATS_MAP_PATH));
Chenbo Feng89c12f12018-03-21 10:29:18 -0700222 RETURN_IF_NOT_OK(changeOwnerAndMode(IFACE_STATS_MAP_PATH, AID_NET_BW_STATS, "IfaceStatsMap",
223 false));
Chenbo Feng95892f32018-06-07 14:52:02 -0700224
Maciej Żenczykowski02554372019-04-01 10:44:54 -0700225 RETURN_IF_NOT_OK(mConfigurationMap.init(CONFIGURATION_MAP_PATH));
Chenbo Fengf434e862018-06-27 14:08:39 -0700226 RETURN_IF_NOT_OK(changeOwnerAndMode(CONFIGURATION_MAP_PATH, AID_NET_BW_STATS,
227 "ConfigurationMap", false));
Chenbo Feng703798e2018-06-15 17:07:59 -0700228 RETURN_IF_NOT_OK(
Chenbo Fengf434e862018-06-27 14:08:39 -0700229 mConfigurationMap.writeValue(UID_RULES_CONFIGURATION_KEY, DEFAULT_CONFIG, BPF_ANY));
230 RETURN_IF_NOT_OK(mConfigurationMap.writeValue(CURRENT_STATS_MAP_CONFIGURATION_KEY, SELECT_MAP_A,
231 BPF_ANY));
Chenbo Feng703798e2018-06-15 17:07:59 -0700232
Maciej Żenczykowski02554372019-04-01 10:44:54 -0700233 RETURN_IF_NOT_OK(mUidOwnerMap.init(UID_OWNER_MAP_PATH));
Chenbo Feng703798e2018-06-15 17:07:59 -0700234 RETURN_IF_NOT_OK(changeOwnerAndMode(UID_OWNER_MAP_PATH, AID_ROOT, "UidOwnerMap", true));
Bernie Innocenti6f9fd902018-10-11 20:50:23 +0900235 RETURN_IF_NOT_OK(mUidOwnerMap.clear());
Maciej Żenczykowski02554372019-04-01 10:44:54 -0700236 RETURN_IF_NOT_OK(mUidPermissionMap.init(UID_PERMISSION_MAP_PATH));
Chenbo Feng89c12f12018-03-21 10:29:18 -0700237 return netdutils::status::ok;
238}
239
Chenbo Fengb2d213d2018-12-04 11:31:00 -0800240static Status attachProgramToCgroup(const char* programPath, const int cgroupFd,
241 bpf_attach_type type) {
242 unique_fd cgroupProg(bpfFdGet(programPath, 0));
243 if (cgroupProg == -1) {
244 int ret = errno;
245 ALOGE("Failed to get program from %s: %s", programPath, strerror(ret));
246 return statusFromErrno(ret, "cgroup program get failed");
247 }
248 if (android::bpf::attachProgram(type, cgroupProg, cgroupFd)) {
249 int ret = errno;
250 ALOGE("Program from %s attach failed: %s", programPath, strerror(ret));
251 return statusFromErrno(ret, "program attach failed");
252 }
253 return netdutils::status::ok;
254}
255
256static Status initPrograms() {
Suren Baghdasaryane072a3c2019-01-16 14:36:07 -0800257 std::string cg2_path;
258
259 if (!CgroupGetControllerPath(CGROUPV2_CONTROLLER_NAME, &cg2_path)) {
260 int ret = errno;
261 ALOGE("Failed to find cgroup v2 root");
262 return statusFromErrno(ret, "Failed to find cgroup v2 root");
263 }
264
265 unique_fd cg_fd(open(cg2_path.c_str(), O_DIRECTORY | O_RDONLY | O_CLOEXEC));
Chenbo Fengb2d213d2018-12-04 11:31:00 -0800266 if (cg_fd == -1) {
267 int ret = errno;
268 ALOGE("Failed to open the cgroup directory: %s", strerror(ret));
269 return statusFromErrno(ret, "Open the cgroup directory failed");
270 }
271 RETURN_IF_NOT_OK(attachProgramToCgroup(BPF_EGRESS_PROG_PATH, cg_fd, BPF_CGROUP_INET_EGRESS));
272 RETURN_IF_NOT_OK(attachProgramToCgroup(BPF_INGRESS_PROG_PATH, cg_fd, BPF_CGROUP_INET_INGRESS));
Chenbo Fenga51f4fa2019-01-15 15:03:32 -0800273
274 // For the devices that support cgroup socket filter, the socket filter
275 // should be loaded successfully by bpfloader. So we attach the filter to
276 // cgroup if the program is pinned properly.
277 // TODO: delete the if statement once all devices should support cgroup
278 // socket filter (ie. the minimum kernel version required is 4.14).
279 if (!access(CGROUP_SOCKET_PROG_PATH, F_OK)) {
280 RETURN_IF_NOT_OK(
281 attachProgramToCgroup(CGROUP_SOCKET_PROG_PATH, cg_fd, BPF_CGROUP_INET_SOCK_CREATE));
282 }
Chenbo Fengb2d213d2018-12-04 11:31:00 -0800283 return netdutils::status::ok;
284}
285
Chenbo Feng89c12f12018-03-21 10:29:18 -0700286Status TrafficController::start() {
Chenbo Feng47dd0732018-12-11 12:23:24 -0800287 if (mBpfLevel == BpfLevel::NONE) {
Chenbo Fengc10a8a42017-12-15 13:56:33 -0800288 return netdutils::status::ok;
Chenbo Fengf2759682017-10-10 17:31:57 -0700289 }
290
Bernie Innocenti6f9fd902018-10-11 20:50:23 +0900291 /* When netd restarts from a crash without total system reboot, the program
Chenbo Fengf2759682017-10-10 17:31:57 -0700292 * is still attached to the cgroup, detach it so the program can be freed
293 * and we can load and attach new program into the target cgroup.
294 *
295 * TODO: Scrape existing socket when run-time restart and clean up the map
296 * if the socket no longer exist
297 */
298
Chenbo Feng89c12f12018-03-21 10:29:18 -0700299 RETURN_IF_NOT_OK(initMaps());
Chenbo Feng7e974052018-02-28 22:57:21 -0800300
Chenbo Fengb2d213d2018-12-04 11:31:00 -0800301 RETURN_IF_NOT_OK(initPrograms());
302
Chenbo Feng7e974052018-02-28 22:57:21 -0800303 // Fetch the list of currently-existing interfaces. At this point NetlinkHandler is
304 // already running, so it will call addInterface() when any new interface appears.
305 std::map<std::string, uint32_t> ifacePairs;
306 ASSIGN_OR_RETURN(ifacePairs, InterfaceController::getIfaceList());
307 for (const auto& ifacePair:ifacePairs) {
308 addInterface(ifacePair.first.c_str(), ifacePair.second);
309 }
310
Chenbo Feng116d0552017-12-04 17:25:19 -0800311 auto result = makeSkDestroyListener();
312 if (!isOk(result)) {
313 ALOGE("Unable to create SkDestroyListener: %s", toString(result).c_str());
314 } else {
315 mSkDestroyListener = std::move(result.value());
316 }
317 // Rx handler extracts nfgenmsg looks up and invokes registered dispatch function.
318 const auto rxHandler = [this](const nlmsghdr&, const Slice msg) {
319 inet_diag_msg diagmsg = {};
320 if (extract(msg, diagmsg) < sizeof(inet_diag_msg)) {
Bernie Innocenti6f9fd902018-10-11 20:50:23 +0900321 ALOGE("Unrecognized netlink message: %s", toString(msg).c_str());
Chenbo Feng116d0552017-12-04 17:25:19 -0800322 return;
323 }
324 uint64_t sock_cookie = static_cast<uint64_t>(diagmsg.id.idiag_cookie[0]) |
325 (static_cast<uint64_t>(diagmsg.id.idiag_cookie[1]) << 32);
326
Bernie Innocenti6f9fd902018-10-11 20:50:23 +0900327 Status s = mCookieTagMap.deleteValue(sock_cookie);
Chenbo Feng1d4c5db2018-10-18 15:50:46 -0700328 if (!isOk(s) && s.code() != ENOENT) {
Bernie Innocenti6f9fd902018-10-11 20:50:23 +0900329 ALOGE("Failed to delete cookie %" PRIx64 ": %s", sock_cookie, toString(s).c_str());
330 return;
331 }
Chenbo Feng116d0552017-12-04 17:25:19 -0800332 };
333 expectOk(mSkDestroyListener->subscribe(kSockDiagMsgType, rxHandler));
334
335 // In case multiple netlink message comes in as a stream, we need to handle the rxDone message
336 // properly.
337 const auto rxDoneHandler = [](const nlmsghdr&, const Slice msg) {
338 // Ignore NLMSG_DONE messages
339 inet_diag_msg diagmsg = {};
340 extract(msg, diagmsg);
341 };
342 expectOk(mSkDestroyListener->subscribe(kSockDiagDoneMsgType, rxDoneHandler));
343
Chenbo Feng05393d82018-01-09 15:18:43 -0800344 return netdutils::status::ok;
Chenbo Fengf2759682017-10-10 17:31:57 -0700345}
346
Chenbo Fengb4a4fa12019-01-09 17:20:45 -0800347int TrafficController::tagSocket(int sockFd, uint32_t tag, uid_t uid, uid_t callingUid) {
Chenbo Feng84d9f9f2019-04-12 15:31:02 -0700348 std::lock_guard guard(mMutex);
Chenbo Fengb4a4fa12019-01-09 17:20:45 -0800349 if (uid != callingUid && !hasUpdateDeviceStatsPermission(callingUid)) {
350 return -EPERM;
351 }
352
Chenbo Feng47dd0732018-12-11 12:23:24 -0800353 if (mBpfLevel == BpfLevel::NONE) {
Chenbo Fengbc6d4702018-04-11 18:27:19 -0700354 if (legacy_tagSocket(sockFd, tag, uid)) return -errno;
355 return 0;
356 }
Chenbo Feng33cc1032017-10-23 15:16:37 -0700357
Chenbo Fengf2759682017-10-10 17:31:57 -0700358 uint64_t sock_cookie = getSocketCookie(sockFd);
Chenbo Fengef1cab32018-04-13 19:50:49 -0700359 if (sock_cookie == NONEXISTENT_COOKIE) return -errno;
Chenbo Fengf2759682017-10-10 17:31:57 -0700360 UidTag newKey = {.uid = (uint32_t)uid, .tag = tag};
361
Chenbo Fengb0495132019-04-10 12:54:41 -0700362 uint32_t totalEntryCount = 0;
Chenbo Feng82075e52019-04-12 19:21:47 -0700363 uint32_t perUidEntryCount = 0;
Chenbo Fengb0495132019-04-10 12:54:41 -0700364 // Now we go through the stats map and count how many entries are associated
365 // with target uid. If the uid entry hit the limit for each uid, we block
366 // the request to prevent the map from overflow. It is safe here to iterate
Chenbo Feng84d9f9f2019-04-12 15:31:02 -0700367 // over the map since when mMutex is hold, system server cannot toggle
Chenbo Fengb0495132019-04-10 12:54:41 -0700368 // the live stats map and clean it. So nobody can delete entries from the map.
Chenbo Feng82075e52019-04-12 19:21:47 -0700369 const auto countUidStatsEntries = [uid, &totalEntryCount, &perUidEntryCount](
370 const StatsKey& key, BpfMap<StatsKey, StatsValue>&) {
Chenbo Fengb0495132019-04-10 12:54:41 -0700371 if (key.uid == uid) {
Chenbo Feng82075e52019-04-12 19:21:47 -0700372 perUidEntryCount++;
Chenbo Fengb0495132019-04-10 12:54:41 -0700373 }
Chenbo Feng82075e52019-04-12 19:21:47 -0700374 totalEntryCount++;
Chenbo Fengb0495132019-04-10 12:54:41 -0700375 return netdutils::status::ok;
376 };
377 auto configuration = mConfigurationMap.readValue(CURRENT_STATS_MAP_CONFIGURATION_KEY);
378 if (!isOk(configuration.status())) {
379 ALOGE("Failed to get current configuration: %s, fd: %d",
380 strerror(configuration.status().code()), mConfigurationMap.getMap().get());
381 return -configuration.status().code();
382 }
Chenbo Feng36a40d12019-04-23 16:47:05 -0700383 if (configuration.value() != SELECT_MAP_A && configuration.value() != SELECT_MAP_B) {
Chenbo Fengb0495132019-04-10 12:54:41 -0700384 ALOGE("unknown configuration value: %d", configuration.value());
385 return -EINVAL;
386 }
Chenbo Feng36a40d12019-04-23 16:47:05 -0700387
388 BpfMap<StatsKey, StatsValue>& currentMap =
389 (configuration.value() == SELECT_MAP_A) ? mStatsMapA : mStatsMapB;
390 Status res = currentMap.iterate(countUidStatsEntries);
391 if (!isOk(res)) {
392 ALOGE("Failed to count the stats entry in map %d: %s", currentMap.getMap().get(),
393 strerror(res.code()));
394 return -res.code();
395 }
396
Chenbo Feng82075e52019-04-12 19:21:47 -0700397 if (totalEntryCount > mTotalUidStatsEntriesLimit ||
398 perUidEntryCount > mPerUidStatsEntriesLimit) {
399 ALOGE("Too many stats entries in the map, total count: %u, uid(%u) count: %u, blocking tag"
400 " request to prevent map overflow",
401 totalEntryCount, uid, perUidEntryCount);
Chenbo Fengb0495132019-04-10 12:54:41 -0700402 return -EMFILE;
403 }
Chenbo Fengf2759682017-10-10 17:31:57 -0700404 // Update the tag information of a socket to the cookieUidMap. Use BPF_ANY
405 // flag so it will insert a new entry to the map if that value doesn't exist
406 // yet. And update the tag if there is already a tag stored. Since the eBPF
407 // program in kernel only read this map, and is protected by rcu read lock. It
408 // should be fine to cocurrently update the map while eBPF program is running.
Chenbo Feng36a40d12019-04-23 16:47:05 -0700409 res = mCookieTagMap.writeValue(sock_cookie, newKey, BPF_ANY);
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700410 if (!isOk(res)) {
411 ALOGE("Failed to tag the socket: %s, fd: %d", strerror(res.code()),
412 mCookieTagMap.getMap().get());
Chenbo Fengf2759682017-10-10 17:31:57 -0700413 }
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700414 return -res.code();
Chenbo Fengf2759682017-10-10 17:31:57 -0700415}
416
417int TrafficController::untagSocket(int sockFd) {
Chenbo Feng47dd0732018-12-11 12:23:24 -0800418 if (mBpfLevel == BpfLevel::NONE) {
Chenbo Fengbc6d4702018-04-11 18:27:19 -0700419 if (legacy_untagSocket(sockFd)) return -errno;
420 return 0;
421 }
Chenbo Fengf2759682017-10-10 17:31:57 -0700422 uint64_t sock_cookie = getSocketCookie(sockFd);
423
Chenbo Fengef1cab32018-04-13 19:50:49 -0700424 if (sock_cookie == NONEXISTENT_COOKIE) return -errno;
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700425 Status res = mCookieTagMap.deleteValue(sock_cookie);
426 if (!isOk(res)) {
427 ALOGE("Failed to untag socket: %s\n", strerror(res.code()));
Chenbo Fengf2759682017-10-10 17:31:57 -0700428 }
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700429 return -res.code();
Chenbo Fengf2759682017-10-10 17:31:57 -0700430}
431
Chenbo Fengb4a4fa12019-01-09 17:20:45 -0800432int TrafficController::setCounterSet(int counterSetNum, uid_t uid, uid_t callingUid) {
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700433 if (counterSetNum < 0 || counterSetNum >= OVERFLOW_COUNTERSET) return -EINVAL;
Chenbo Fengb4a4fa12019-01-09 17:20:45 -0800434
Chenbo Feng84d9f9f2019-04-12 15:31:02 -0700435 std::lock_guard guard(mMutex);
Chenbo Fengb4a4fa12019-01-09 17:20:45 -0800436 if (!hasUpdateDeviceStatsPermission(callingUid)) return -EPERM;
437
Chenbo Feng47dd0732018-12-11 12:23:24 -0800438 if (mBpfLevel == BpfLevel::NONE) {
Chenbo Fengbc6d4702018-04-11 18:27:19 -0700439 if (legacy_setCounterSet(counterSetNum, uid)) return -errno;
440 return 0;
441 }
Chenbo Fengc10a8a42017-12-15 13:56:33 -0800442
443 // The default counter set for all uid is 0, so deleting the current counterset for that uid
444 // will automatically set it to 0.
Chenbo Fengf2759682017-10-10 17:31:57 -0700445 if (counterSetNum == 0) {
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700446 Status res = mUidCounterSetMap.deleteValue(uid);
447 if (isOk(res) || (!isOk(res) && res.code() == ENOENT)) {
Chenbo Fengf2759682017-10-10 17:31:57 -0700448 return 0;
449 } else {
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700450 ALOGE("Failed to delete the counterSet: %s\n", strerror(res.code()));
451 return -res.code();
Chenbo Fengf2759682017-10-10 17:31:57 -0700452 }
453 }
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700454 uint8_t tmpCounterSetNum = (uint8_t)counterSetNum;
Chenbo Fengb4a4fa12019-01-09 17:20:45 -0800455 Status res = mUidCounterSetMap.writeValue(uid, tmpCounterSetNum, BPF_ANY);
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700456 if (!isOk(res)) {
457 ALOGE("Failed to set the counterSet: %s, fd: %d", strerror(res.code()),
458 mUidCounterSetMap.getMap().get());
459 return -res.code();
Chenbo Fengf2759682017-10-10 17:31:57 -0700460 }
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700461 return 0;
Chenbo Fengf2759682017-10-10 17:31:57 -0700462}
463
Chenbo Fengf434e862018-06-27 14:08:39 -0700464// This method only get called by system_server when an app get uinstalled, it
465// is called inside removeUidsLocked() while holding mStatsLock. So it is safe
466// to iterate and modify the stats maps.
Chenbo Fengb4a4fa12019-01-09 17:20:45 -0800467int TrafficController::deleteTagData(uint32_t tag, uid_t uid, uid_t callingUid) {
Chenbo Feng84d9f9f2019-04-12 15:31:02 -0700468 std::lock_guard guard(mMutex);
Chenbo Fengb4a4fa12019-01-09 17:20:45 -0800469 if (!hasUpdateDeviceStatsPermission(callingUid)) return -EPERM;
470
Chenbo Feng47dd0732018-12-11 12:23:24 -0800471 if (mBpfLevel == BpfLevel::NONE) {
Chenbo Fengbc6d4702018-04-11 18:27:19 -0700472 if (legacy_deleteTagData(tag, uid)) return -errno;
473 return 0;
474 }
Chenbo Feng33cc1032017-10-23 15:16:37 -0700475
Chenbo Fengc10a8a42017-12-15 13:56:33 -0800476 // First we go through the cookieTagMap to delete the target uid tag combination. Or delete all
Chenbo Fengef1cab32018-04-13 19:50:49 -0700477 // the tags related to the uid if the tag is 0.
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700478 const auto deleteMatchedCookieEntries = [uid, tag](const uint64_t& key, const UidTag& value,
479 BpfMap<uint64_t, UidTag>& map) {
480 if (value.uid == uid && (value.tag == tag || tag == 0)) {
481 Status res = map.deleteValue(key);
482 if (isOk(res) || (res.code() == ENOENT)) {
483 return netdutils::status::ok;
Chenbo Fengf2759682017-10-10 17:31:57 -0700484 }
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700485 ALOGE("Failed to delete data(cookie = %" PRIu64 "): %s\n", key, strerror(res.code()));
Chenbo Fengf2759682017-10-10 17:31:57 -0700486 }
Chenbo Fengef1cab32018-04-13 19:50:49 -0700487 // Move forward to next cookie in the map.
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700488 return netdutils::status::ok;
Chenbo Fengef1cab32018-04-13 19:50:49 -0700489 };
Bernie Innocenti6f9fd902018-10-11 20:50:23 +0900490 mCookieTagMap.iterateWithValue(deleteMatchedCookieEntries).ignoreError();
Chenbo Fengf2759682017-10-10 17:31:57 -0700491 // Now we go through the Tag stats map and delete the data entry with correct uid and tag
Chenbo Fengef1cab32018-04-13 19:50:49 -0700492 // combination. Or all tag stats under that uid if the target tag is 0.
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700493 const auto deleteMatchedUidTagEntries = [uid, tag](const StatsKey& key,
494 BpfMap<StatsKey, StatsValue>& map) {
495 if (key.uid == uid && (key.tag == tag || tag == 0)) {
496 Status res = map.deleteValue(key);
497 if (isOk(res) || (res.code() == ENOENT)) {
Chenbo Fengef1cab32018-04-13 19:50:49 -0700498 //Entry is deleted, use the current key to get a new nextKey;
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700499 return netdutils::status::ok;
Chenbo Fengf2759682017-10-10 17:31:57 -0700500 }
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700501 ALOGE("Failed to delete data(uid=%u, tag=%u): %s\n", key.uid, key.tag,
502 strerror(res.code()));
Chenbo Fengf2759682017-10-10 17:31:57 -0700503 }
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700504 return netdutils::status::ok;
Chenbo Fengef1cab32018-04-13 19:50:49 -0700505 };
Chenbo Fengf434e862018-06-27 14:08:39 -0700506 mStatsMapB.iterate(deleteMatchedUidTagEntries).ignoreError();
507 mStatsMapA.iterate(deleteMatchedUidTagEntries).ignoreError();
Chenbo Fengf2759682017-10-10 17:31:57 -0700508 // If the tag is not zero, we already deleted all the data entry required. If tag is 0, we also
Chenbo Fenged37fea2017-12-13 19:35:01 -0800509 // need to delete the stats stored in uidStatsMap and counterSet map.
Chenbo Fengc10a8a42017-12-15 13:56:33 -0800510 if (tag != 0) return 0;
511
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700512 Status res = mUidCounterSetMap.deleteValue(uid);
513 if (!isOk(res) && res.code() != ENOENT) {
514 ALOGE("Failed to delete counterSet data(uid=%u, tag=%u): %s\n", uid, tag,
515 strerror(res.code()));
Chenbo Fengc10a8a42017-12-15 13:56:33 -0800516 }
Chenbo Fengbc4a15f2018-05-11 19:15:15 -0700517
518 auto deleteAppUidStatsEntry = [uid](const uint32_t& key, BpfMap<uint32_t, StatsValue>& map) {
519 if (key == uid) {
520 Status res = map.deleteValue(key);
521 if (isOk(res) || (res.code() == ENOENT)) {
522 return netdutils::status::ok;
523 }
524 ALOGE("Failed to delete data(uid=%u): %s", key, strerror(res.code()));
525 }
526 return netdutils::status::ok;
527 };
Bernie Innocenti6f9fd902018-10-11 20:50:23 +0900528 mAppUidStatsMap.iterate(deleteAppUidStatsEntry).ignoreError();
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700529 return 0;
Chenbo Fengf2759682017-10-10 17:31:57 -0700530}
531
Chenbo Feng7e974052018-02-28 22:57:21 -0800532int TrafficController::addInterface(const char* name, uint32_t ifaceIndex) {
Chenbo Feng47dd0732018-12-11 12:23:24 -0800533 if (mBpfLevel == BpfLevel::NONE) return 0;
Chenbo Feng7e974052018-02-28 22:57:21 -0800534
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700535 IfaceValue iface;
Chenbo Feng7e974052018-02-28 22:57:21 -0800536 if (ifaceIndex == 0) {
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700537 ALOGE("Unknown interface %s(%d)", name, ifaceIndex);
Chenbo Feng7e974052018-02-28 22:57:21 -0800538 return -1;
539 }
540
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700541 strlcpy(iface.name, name, sizeof(IfaceValue));
542 Status res = mIfaceIndexNameMap.writeValue(ifaceIndex, iface, BPF_ANY);
543 if (!isOk(res)) {
544 ALOGE("Failed to add iface %s(%d): %s", name, ifaceIndex, strerror(res.code()));
545 return -res.code();
Chenbo Feng7e974052018-02-28 22:57:21 -0800546 }
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700547 return 0;
Chenbo Feng7e974052018-02-28 22:57:21 -0800548}
549
Chenbo Feng703798e2018-06-15 17:07:59 -0700550Status TrafficController::updateOwnerMapEntry(UidOwnerMatchType match, uid_t uid, FirewallRule rule,
551 FirewallType type) {
Chenbo Feng84d9f9f2019-04-12 15:31:02 -0700552 std::lock_guard guard(mMutex);
Chenbo Feng89c12f12018-03-21 10:29:18 -0700553 if ((rule == ALLOW && type == WHITELIST) || (rule == DENY && type == BLACKLIST)) {
Rubin Xu5ca8e802019-04-05 11:50:32 +0100554 RETURN_IF_NOT_OK(addRule(mUidOwnerMap, uid, match));
Chenbo Feng89c12f12018-03-21 10:29:18 -0700555 } else if ((rule == ALLOW && type == BLACKLIST) || (rule == DENY && type == WHITELIST)) {
Rubin Xu5ca8e802019-04-05 11:50:32 +0100556 RETURN_IF_NOT_OK(removeRule(mUidOwnerMap, uid, match));
Chenbo Feng89c12f12018-03-21 10:29:18 -0700557 } else {
558 //Cannot happen.
Rubin Xuec27ff22019-01-08 21:33:03 +0000559 return statusFromErrno(EINVAL, "");
Chenbo Feng89c12f12018-03-21 10:29:18 -0700560 }
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700561 return netdutils::status::ok;
Chenbo Feng89c12f12018-03-21 10:29:18 -0700562}
563
Chenbo Feng703798e2018-06-15 17:07:59 -0700564UidOwnerMatchType TrafficController::jumpOpToMatch(BandwidthController::IptJumpOp jumpHandling) {
Chenbo Feng95892f32018-06-07 14:52:02 -0700565 switch (jumpHandling) {
566 case BandwidthController::IptJumpReject:
Chenbo Feng703798e2018-06-15 17:07:59 -0700567 return PENALTY_BOX_MATCH;
Chenbo Feng95892f32018-06-07 14:52:02 -0700568 case BandwidthController::IptJumpReturn:
Chenbo Feng703798e2018-06-15 17:07:59 -0700569 return HAPPY_BOX_MATCH;
Chenbo Feng95892f32018-06-07 14:52:02 -0700570 case BandwidthController::IptJumpNoAdd:
571 return NO_MATCH;
572 }
573}
574
Rubin Xu5ca8e802019-04-05 11:50:32 +0100575Status TrafficController::removeRule(BpfMap<uint32_t, UidOwnerValue>& map, uint32_t uid,
576 UidOwnerMatchType match) {
Chenbo Feng703798e2018-06-15 17:07:59 -0700577 auto oldMatch = map.readValue(uid);
578 if (isOk(oldMatch)) {
Rubin Xuec27ff22019-01-08 21:33:03 +0000579 UidOwnerValue newMatch = {.rule = static_cast<uint8_t>(oldMatch.value().rule & ~match),
580 .iif = (match == IIF_MATCH) ? 0 : oldMatch.value().iif};
581 if (newMatch.rule == 0) {
Chenbo Feng703798e2018-06-15 17:07:59 -0700582 RETURN_IF_NOT_OK(map.deleteValue(uid));
583 } else {
584 RETURN_IF_NOT_OK(map.writeValue(uid, newMatch, BPF_ANY));
585 }
586 } else {
587 return statusFromErrno(ENOENT, StringPrintf("uid: %u does not exist in map", uid));
588 }
589 return netdutils::status::ok;
590}
591
Rubin Xu5ca8e802019-04-05 11:50:32 +0100592Status TrafficController::addRule(BpfMap<uint32_t, UidOwnerValue>& map, uint32_t uid,
593 UidOwnerMatchType match, uint32_t iif) {
Rubin Xuec27ff22019-01-08 21:33:03 +0000594 // iif should be non-zero if and only if match == MATCH_IIF
595 if (match == IIF_MATCH && iif == 0) {
596 return statusFromErrno(EINVAL, "Interface match must have nonzero interface index");
597 } else if (match != IIF_MATCH && iif != 0) {
598 return statusFromErrno(EINVAL, "Non-interface match must have zero interface index");
599 }
Chenbo Feng703798e2018-06-15 17:07:59 -0700600 auto oldMatch = map.readValue(uid);
601 if (isOk(oldMatch)) {
Rubin Xuec27ff22019-01-08 21:33:03 +0000602 UidOwnerValue newMatch = {.rule = static_cast<uint8_t>(oldMatch.value().rule | match),
603 .iif = iif ? iif : oldMatch.value().iif};
604 RETURN_IF_NOT_OK(map.writeValue(uid, newMatch, BPF_ANY));
Chenbo Feng703798e2018-06-15 17:07:59 -0700605 } else {
Rubin Xuec27ff22019-01-08 21:33:03 +0000606 UidOwnerValue newMatch = {.rule = static_cast<uint8_t>(match), .iif = iif};
607 RETURN_IF_NOT_OK(map.writeValue(uid, newMatch, BPF_ANY));
Chenbo Feng703798e2018-06-15 17:07:59 -0700608 }
609 return netdutils::status::ok;
610}
611
612Status TrafficController::updateUidOwnerMap(const std::vector<std::string>& appStrUids,
613 BandwidthController::IptJumpOp jumpHandling,
614 BandwidthController::IptOp op) {
Chenbo Feng84d9f9f2019-04-12 15:31:02 -0700615 std::lock_guard guard(mMutex);
Chenbo Feng703798e2018-06-15 17:07:59 -0700616 UidOwnerMatchType match = jumpOpToMatch(jumpHandling);
617 if (match == NO_MATCH) {
618 return statusFromErrno(
619 EINVAL, StringPrintf("invalid IptJumpOp: %d, command: %d", jumpHandling, match));
Chenbo Feng95892f32018-06-07 14:52:02 -0700620 }
621 for (const auto& appStrUid : appStrUids) {
622 char* endPtr;
623 long uid = strtol(appStrUid.c_str(), &endPtr, 10);
624 if ((errno == ERANGE && (uid == LONG_MAX || uid == LONG_MIN)) ||
625 (endPtr == appStrUid.c_str()) || (*endPtr != '\0')) {
626 return statusFromErrno(errno, "invalid uid string:" + appStrUid);
627 }
628
Chenbo Feng95892f32018-06-07 14:52:02 -0700629 if (op == BandwidthController::IptOpDelete) {
Rubin Xu5ca8e802019-04-05 11:50:32 +0100630 RETURN_IF_NOT_OK(removeRule(mUidOwnerMap, uid, match));
Chenbo Feng95892f32018-06-07 14:52:02 -0700631 } else if (op == BandwidthController::IptOpInsert) {
Rubin Xu5ca8e802019-04-05 11:50:32 +0100632 RETURN_IF_NOT_OK(addRule(mUidOwnerMap, uid, match));
Chenbo Feng95892f32018-06-07 14:52:02 -0700633 } else {
634 // Cannot happen.
Chenbo Feng703798e2018-06-15 17:07:59 -0700635 return statusFromErrno(EINVAL, StringPrintf("invalid IptOp: %d, %d", op, match));
Chenbo Feng95892f32018-06-07 14:52:02 -0700636 }
637 }
638 return netdutils::status::ok;
639}
640
Chenbo Feng89c12f12018-03-21 10:29:18 -0700641int TrafficController::changeUidOwnerRule(ChildChain chain, uid_t uid, FirewallRule rule,
642 FirewallType type) {
Chenbo Feng47dd0732018-12-11 12:23:24 -0800643 if (mBpfLevel == BpfLevel::NONE) {
Chenbo Feng89c12f12018-03-21 10:29:18 -0700644 ALOGE("bpf is not set up, should use iptables rule");
645 return -ENOSYS;
646 }
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700647 Status res;
Chenbo Feng89c12f12018-03-21 10:29:18 -0700648 switch (chain) {
649 case DOZABLE:
Chenbo Feng703798e2018-06-15 17:07:59 -0700650 res = updateOwnerMapEntry(DOZABLE_MATCH, uid, rule, type);
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700651 break;
Chenbo Feng89c12f12018-03-21 10:29:18 -0700652 case STANDBY:
Chenbo Feng703798e2018-06-15 17:07:59 -0700653 res = updateOwnerMapEntry(STANDBY_MATCH, uid, rule, type);
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700654 break;
Chenbo Feng89c12f12018-03-21 10:29:18 -0700655 case POWERSAVE:
Chenbo Feng703798e2018-06-15 17:07:59 -0700656 res = updateOwnerMapEntry(POWERSAVE_MATCH, uid, rule, type);
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700657 break;
Chenbo Feng89c12f12018-03-21 10:29:18 -0700658 case NONE:
659 default:
660 return -EINVAL;
661 }
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700662 if (!isOk(res)) {
663 ALOGE("change uid(%u) rule of %d failed: %s, rule: %d, type: %d", uid, chain,
664 res.msg().c_str(), rule, type);
665 return -res.code();
666 }
667 return 0;
Chenbo Feng89c12f12018-03-21 10:29:18 -0700668}
669
Rubin Xu5ca8e802019-04-05 11:50:32 +0100670Status TrafficController::replaceRulesInMap(const UidOwnerMatchType match,
671 const std::vector<int32_t>& uids) {
Chenbo Feng84d9f9f2019-04-12 15:31:02 -0700672 std::lock_guard guard(mMutex);
Chenbo Feng89c12f12018-03-21 10:29:18 -0700673 std::set<int32_t> uidSet(uids.begin(), uids.end());
674 std::vector<uint32_t> uidsToDelete;
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700675 auto getUidsToDelete = [&uidsToDelete, &uidSet](const uint32_t& key,
Rubin Xuec27ff22019-01-08 21:33:03 +0000676 const BpfMap<uint32_t, UidOwnerValue>&) {
Chenbo Feng703798e2018-06-15 17:07:59 -0700677 if (uidSet.find((int32_t) key) == uidSet.end()) {
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700678 uidsToDelete.push_back(key);
Chenbo Feng89c12f12018-03-21 10:29:18 -0700679 }
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700680 return netdutils::status::ok;
Chenbo Feng89c12f12018-03-21 10:29:18 -0700681 };
Chenbo Feng703798e2018-06-15 17:07:59 -0700682 RETURN_IF_NOT_OK(mUidOwnerMap.iterate(getUidsToDelete));
Chenbo Feng89c12f12018-03-21 10:29:18 -0700683
684 for(auto uid : uidsToDelete) {
Rubin Xu5ca8e802019-04-05 11:50:32 +0100685 RETURN_IF_NOT_OK(removeRule(mUidOwnerMap, uid, match));
Chenbo Feng89c12f12018-03-21 10:29:18 -0700686 }
687
688 for (auto uid : uids) {
Rubin Xu5ca8e802019-04-05 11:50:32 +0100689 RETURN_IF_NOT_OK(addRule(mUidOwnerMap, uid, match));
Chenbo Feng89c12f12018-03-21 10:29:18 -0700690 }
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700691 return netdutils::status::ok;
Chenbo Feng89c12f12018-03-21 10:29:18 -0700692}
693
Rubin Xuec27ff22019-01-08 21:33:03 +0000694Status TrafficController::addUidInterfaceRules(const int iif,
695 const std::vector<int32_t>& uidsToAdd) {
696 if (mBpfLevel == BpfLevel::NONE) {
697 ALOGW("UID ingress interface filtering not possible without BPF owner match");
698 return statusFromErrno(EOPNOTSUPP, "eBPF not supported");
699 }
700 if (!iif) {
701 return statusFromErrno(EINVAL, "Interface rule must specify interface");
702 }
Chenbo Feng84d9f9f2019-04-12 15:31:02 -0700703 std::lock_guard guard(mMutex);
Rubin Xuec27ff22019-01-08 21:33:03 +0000704
705 for (auto uid : uidsToAdd) {
Rubin Xu5ca8e802019-04-05 11:50:32 +0100706 netdutils::Status result = addRule(mUidOwnerMap, uid, IIF_MATCH, iif);
Rubin Xuec27ff22019-01-08 21:33:03 +0000707 if (!isOk(result)) {
Rubin Xu5ca8e802019-04-05 11:50:32 +0100708 ALOGW("addRule failed(%d): uid=%d iif=%d", result.code(), uid, iif);
Rubin Xuec27ff22019-01-08 21:33:03 +0000709 }
710 }
711 return netdutils::status::ok;
712}
713
714Status TrafficController::removeUidInterfaceRules(const std::vector<int32_t>& uidsToDelete) {
715 if (mBpfLevel == BpfLevel::NONE) {
716 ALOGW("UID ingress interface filtering not possible without BPF owner match");
717 return statusFromErrno(EOPNOTSUPP, "eBPF not supported");
718 }
Chenbo Feng84d9f9f2019-04-12 15:31:02 -0700719 std::lock_guard guard(mMutex);
Rubin Xuec27ff22019-01-08 21:33:03 +0000720
721 for (auto uid : uidsToDelete) {
Rubin Xu5ca8e802019-04-05 11:50:32 +0100722 netdutils::Status result = removeRule(mUidOwnerMap, uid, IIF_MATCH);
Rubin Xuec27ff22019-01-08 21:33:03 +0000723 if (!isOk(result)) {
Rubin Xu5ca8e802019-04-05 11:50:32 +0100724 ALOGW("removeRule failed(%d): uid=%d", result.code(), uid);
Rubin Xuec27ff22019-01-08 21:33:03 +0000725 }
726 }
727 return netdutils::status::ok;
728}
729
Chenbo Feng89c12f12018-03-21 10:29:18 -0700730int TrafficController::replaceUidOwnerMap(const std::string& name, bool isWhitelist,
731 const std::vector<int32_t>& uids) {
Chenbo Feng89c12f12018-03-21 10:29:18 -0700732 FirewallRule rule;
733 FirewallType type;
734 if (isWhitelist) {
735 type = WHITELIST;
736 rule = ALLOW;
737 } else {
738 type = BLACKLIST;
739 rule = DENY;
740 }
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700741 Status res;
Chenbo Feng89c12f12018-03-21 10:29:18 -0700742 if (!name.compare(FirewallController::LOCAL_DOZABLE)) {
Rubin Xu5ca8e802019-04-05 11:50:32 +0100743 res = replaceRulesInMap(DOZABLE_MATCH, uids);
Chenbo Feng89c12f12018-03-21 10:29:18 -0700744 } else if (!name.compare(FirewallController::LOCAL_STANDBY)) {
Rubin Xu5ca8e802019-04-05 11:50:32 +0100745 res = replaceRulesInMap(STANDBY_MATCH, uids);
Chenbo Feng89c12f12018-03-21 10:29:18 -0700746 } else if (!name.compare(FirewallController::LOCAL_POWERSAVE)) {
Rubin Xu5ca8e802019-04-05 11:50:32 +0100747 res = replaceRulesInMap(POWERSAVE_MATCH, uids);
Chenbo Feng89c12f12018-03-21 10:29:18 -0700748 } else {
749 ALOGE("unknown chain name: %s", name.c_str());
750 return -EINVAL;
751 }
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700752 if (!isOk(res)) {
753 ALOGE("Failed to clean up chain: %s: %s", name.c_str(), res.msg().c_str());
754 return -res.code();
Chenbo Feng89c12f12018-03-21 10:29:18 -0700755 }
756 return 0;
757}
758
759int TrafficController::toggleUidOwnerMap(ChildChain chain, bool enable) {
Chenbo Feng84d9f9f2019-04-12 15:31:02 -0700760 std::lock_guard guard(mMutex);
Chenbo Fengf434e862018-06-27 14:08:39 -0700761 uint32_t key = UID_RULES_CONFIGURATION_KEY;
Chenbo Feng703798e2018-06-15 17:07:59 -0700762 auto oldConfiguration = mConfigurationMap.readValue(key);
763 if (!isOk(oldConfiguration)) {
764 ALOGE("Cannot read the old configuration from map: %s",
765 oldConfiguration.status().msg().c_str());
766 return -oldConfiguration.status().code();
767 }
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700768 Status res;
Chenbo Feng703798e2018-06-15 17:07:59 -0700769 BpfConfig newConfiguration;
770 uint8_t match;
Chenbo Feng89c12f12018-03-21 10:29:18 -0700771 switch (chain) {
772 case DOZABLE:
Chenbo Feng703798e2018-06-15 17:07:59 -0700773 match = DOZABLE_MATCH;
Chenbo Feng89c12f12018-03-21 10:29:18 -0700774 break;
775 case STANDBY:
Chenbo Feng703798e2018-06-15 17:07:59 -0700776 match = STANDBY_MATCH;
Chenbo Feng89c12f12018-03-21 10:29:18 -0700777 break;
778 case POWERSAVE:
Chenbo Feng703798e2018-06-15 17:07:59 -0700779 match = POWERSAVE_MATCH;
Chenbo Feng89c12f12018-03-21 10:29:18 -0700780 break;
781 default:
782 return -EINVAL;
783 }
Chenbo Feng703798e2018-06-15 17:07:59 -0700784 newConfiguration =
785 enable ? (oldConfiguration.value() | match) : (oldConfiguration.value() & (~match));
786 res = mConfigurationMap.writeValue(key, newConfiguration, BPF_EXIST);
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700787 if (!isOk(res)) {
788 ALOGE("Failed to toggleUidOwnerMap(%d): %s", chain, res.msg().c_str());
Chenbo Feng89c12f12018-03-21 10:29:18 -0700789 }
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700790 return -res.code();
Chenbo Feng89c12f12018-03-21 10:29:18 -0700791}
792
Chenbo Feng47dd0732018-12-11 12:23:24 -0800793BpfLevel TrafficController::getBpfLevel() {
794 return mBpfLevel;
Chenbo Feng07d43fe2017-12-21 14:38:51 -0800795}
796
Chenbo Feng873ae142019-04-10 12:26:06 -0700797Status TrafficController::swapActiveStatsMap() {
Chenbo Feng84d9f9f2019-04-12 15:31:02 -0700798 std::lock_guard guard(mMutex);
Chenbo Feng873ae142019-04-10 12:26:06 -0700799
800 if (mBpfLevel == BpfLevel::NONE) {
801 return statusFromErrno(EOPNOTSUPP, "This device doesn't have eBPF support");
802 }
803
804 uint32_t key = CURRENT_STATS_MAP_CONFIGURATION_KEY;
805 auto oldConfiguration = mConfigurationMap.readValue(key);
806 if (!isOk(oldConfiguration)) {
807 ALOGE("Cannot read the old configuration from map: %s",
808 oldConfiguration.status().msg().c_str());
809 return oldConfiguration.status();
810 }
811
812 // Write to the configuration map to inform the kernel eBPF program to switch
813 // from using one map to the other. Use flag BPF_EXIST here since the map should
814 // be already populated in initMaps.
815 uint8_t newConfigure = (oldConfiguration.value() == SELECT_MAP_A) ? SELECT_MAP_B : SELECT_MAP_A;
816 Status res = mConfigurationMap.writeValue(CURRENT_STATS_MAP_CONFIGURATION_KEY, newConfigure,
817 BPF_EXIST);
818 if (!isOk(res)) {
819 ALOGE("Failed to toggle the stats map: %s", strerror(res.code()));
820 return res;
821 }
822 // After changing the config, we need to make sure all the current running
823 // eBPF programs are finished and all the CPUs are aware of this config change
824 // before we modify the old map. So we do a special hack here to wait for
825 // the kernel to do a synchronize_rcu(). Once the kernel called
826 // synchronize_rcu(), the config we just updated will be available to all cores
827 // and the next eBPF programs triggered inside the kernel will use the new
828 // map configuration. So once this function returns we can safely modify the
829 // old stats map without concerning about race between the kernel and
830 // userspace.
831 int ret = synchronizeKernelRCU();
832 if (ret) {
833 ALOGE("map swap synchronize_rcu() ended with failure: %s", strerror(-ret));
834 return statusFromErrno(-ret, "map swap synchronize_rcu() failed");
835 }
836 return netdutils::status::ok;
837}
838
Chenbo Feng48eaed32018-12-26 17:40:21 -0800839void TrafficController::setPermissionForUids(int permission, const std::vector<uid_t>& uids) {
Chenbo Feng84d9f9f2019-04-12 15:31:02 -0700840 std::lock_guard guard(mMutex);
Chenbo Fengbf660aa2019-02-26 16:12:27 -0800841 if (permission == INetd::PERMISSION_UNINSTALLED) {
842 for (uid_t uid : uids) {
843 // Clean up all permission information for the related uid if all the
844 // packages related to it are uninstalled.
845 mPrivilegedUser.erase(uid);
Chenbo Feng97c88db2019-04-24 12:19:16 -0700846 if (mBpfLevel > BpfLevel::NONE) {
847 Status ret = mUidPermissionMap.deleteValue(uid);
848 if (!isOk(ret) && ret.code() != ENONET) {
849 ALOGE("Failed to clean up the permission for %u: %s", uid,
850 strerror(ret.code()));
851 }
852 }
Chenbo Fengbf660aa2019-02-26 16:12:27 -0800853 }
854 return;
855 }
856
Chenbo Feng48eaed32018-12-26 17:40:21 -0800857 bool privileged = (permission & INetd::PERMISSION_UPDATE_DEVICE_STATS);
858
859 for (uid_t uid : uids) {
Chenbo Feng97c88db2019-04-24 12:19:16 -0700860 if (privileged) {
861 mPrivilegedUser.insert(uid);
862 } else {
863 mPrivilegedUser.erase(uid);
864 }
865
866 // Skip the bpf map operation if not supported.
867 if (mBpfLevel == BpfLevel::NONE) {
868 continue;
869 }
Chenbo Feng527908a2019-04-02 20:14:11 -0700870 // The map stores all the permissions that the UID has, except if the only permission
871 // the UID has is the INTERNET permission, then the UID should not appear in the map.
872 if (permission != INetd::PERMISSION_INTERNET) {
Chenbo Fengbf660aa2019-02-26 16:12:27 -0800873 Status ret = mUidPermissionMap.writeValue(uid, permission, BPF_ANY);
Chenbo Feng48eaed32018-12-26 17:40:21 -0800874 if (!isOk(ret)) {
Chenbo Feng527908a2019-04-02 20:14:11 -0700875 ALOGE("Failed to set permission: %s of uid(%u) to permission map: %s",
876 UidPermissionTypeToString(permission).c_str(), uid, strerror(ret.code()));
Chenbo Feng48eaed32018-12-26 17:40:21 -0800877 }
878 } else {
879 Status ret = mUidPermissionMap.deleteValue(uid);
880 if (!isOk(ret) && ret.code() != ENOENT) {
Chenbo Feng527908a2019-04-02 20:14:11 -0700881 ALOGE("Failed to remove uid %u from permission map: %s", uid, strerror(ret.code()));
Chenbo Feng48eaed32018-12-26 17:40:21 -0800882 }
883 }
Chenbo Feng48eaed32018-12-26 17:40:21 -0800884 }
885}
886
Chenbo Fengef297172018-03-26 10:53:33 -0700887std::string getProgramStatus(const char *path) {
888 int ret = access(path, R_OK);
889 if (ret == 0) {
890 return StringPrintf("OK");
891 }
892 if (ret != 0 && errno == ENOENT) {
893 return StringPrintf("program is missing at: %s", path);
894 }
895 return StringPrintf("check Program %s error: %s", path, strerror(errno));
896}
897
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700898std::string getMapStatus(const base::unique_fd& map_fd, const char* path) {
Chenbo Fengef297172018-03-26 10:53:33 -0700899 if (map_fd.get() < 0) {
900 return StringPrintf("map fd lost");
901 }
902 if (access(path, F_OK) != 0) {
903 return StringPrintf("map not pinned to location: %s", path);
904 }
905 return StringPrintf("OK");
906}
907
Bernie Innocentia5161a02019-01-30 22:40:53 +0900908// NOLINTNEXTLINE(google-runtime-references): grandfathered pass by non-const reference
Bernie Innocenti15bb55c2018-06-03 16:19:51 +0900909void dumpBpfMap(const std::string& mapName, DumpWriter& dw, const std::string& header) {
Chenbo Fengef297172018-03-26 10:53:33 -0700910 dw.blankline();
911 dw.println("%s:", mapName.c_str());
Erik Klineb31fd692018-06-06 20:50:11 +0900912 if (!header.empty()) {
913 dw.println(header);
Chenbo Fengef297172018-03-26 10:53:33 -0700914 }
915}
916
917const String16 TrafficController::DUMP_KEYWORD = String16("trafficcontroller");
918
919void TrafficController::dump(DumpWriter& dw, bool verbose) {
Chenbo Feng84d9f9f2019-04-12 15:31:02 -0700920 std::lock_guard guard(mMutex);
Erik Klineb31fd692018-06-06 20:50:11 +0900921 ScopedIndent indentTop(dw);
Chenbo Fengef297172018-03-26 10:53:33 -0700922 dw.println("TrafficController");
923
Erik Klineb31fd692018-06-06 20:50:11 +0900924 ScopedIndent indentPreBpfModule(dw);
Chenbo Feng47dd0732018-12-11 12:23:24 -0800925 dw.println("BPF module status: %s", BpfLevelToString(mBpfLevel).c_str());
Chenbo Fengef297172018-03-26 10:53:33 -0700926
Chenbo Feng47dd0732018-12-11 12:23:24 -0800927 if (mBpfLevel == BpfLevel::NONE) {
Chenbo Fengef297172018-03-26 10:53:33 -0700928 return;
Erik Klineb31fd692018-06-06 20:50:11 +0900929 }
Chenbo Fengef297172018-03-26 10:53:33 -0700930
931 dw.blankline();
932 dw.println("mCookieTagMap status: %s",
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700933 getMapStatus(mCookieTagMap.getMap(), COOKIE_TAG_MAP_PATH).c_str());
Chenbo Fengef297172018-03-26 10:53:33 -0700934 dw.println("mUidCounterSetMap status: %s",
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700935 getMapStatus(mUidCounterSetMap.getMap(), UID_COUNTERSET_MAP_PATH).c_str());
Chenbo Fengbc4a15f2018-05-11 19:15:15 -0700936 dw.println("mAppUidStatsMap status: %s",
937 getMapStatus(mAppUidStatsMap.getMap(), APP_UID_STATS_MAP_PATH).c_str());
Chenbo Fengf434e862018-06-27 14:08:39 -0700938 dw.println("mStatsMapA status: %s",
939 getMapStatus(mStatsMapA.getMap(), STATS_MAP_A_PATH).c_str());
940 dw.println("mStatsMapB status: %s",
941 getMapStatus(mStatsMapB.getMap(), STATS_MAP_B_PATH).c_str());
Chenbo Fengef297172018-03-26 10:53:33 -0700942 dw.println("mIfaceIndexNameMap status: %s",
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700943 getMapStatus(mIfaceIndexNameMap.getMap(), IFACE_INDEX_NAME_MAP_PATH).c_str());
Chenbo Fengef297172018-03-26 10:53:33 -0700944 dw.println("mIfaceStatsMap status: %s",
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700945 getMapStatus(mIfaceStatsMap.getMap(), IFACE_STATS_MAP_PATH).c_str());
Chenbo Feng703798e2018-06-15 17:07:59 -0700946 dw.println("mConfigurationMap status: %s",
947 getMapStatus(mConfigurationMap.getMap(), CONFIGURATION_MAP_PATH).c_str());
948 dw.println("mUidOwnerMap status: %s",
949 getMapStatus(mUidOwnerMap.getMap(), UID_OWNER_MAP_PATH).c_str());
Chenbo Fengef297172018-03-26 10:53:33 -0700950
951 dw.blankline();
952 dw.println("Cgroup ingress program status: %s",
953 getProgramStatus(BPF_INGRESS_PROG_PATH).c_str());
954 dw.println("Cgroup egress program status: %s", getProgramStatus(BPF_EGRESS_PROG_PATH).c_str());
955 dw.println("xt_bpf ingress program status: %s",
956 getProgramStatus(XT_BPF_INGRESS_PROG_PATH).c_str());
957 dw.println("xt_bpf egress program status: %s",
958 getProgramStatus(XT_BPF_EGRESS_PROG_PATH).c_str());
Chenbo Feng95892f32018-06-07 14:52:02 -0700959 dw.println("xt_bpf bandwidth whitelist program status: %s",
960 getProgramStatus(XT_BPF_WHITELIST_PROG_PATH).c_str());
961 dw.println("xt_bpf bandwidth blacklist program status: %s",
962 getProgramStatus(XT_BPF_BLACKLIST_PROG_PATH).c_str());
Chenbo Fengef297172018-03-26 10:53:33 -0700963
Erik Klineb31fd692018-06-06 20:50:11 +0900964 if (!verbose) {
965 return;
966 }
Chenbo Fengef297172018-03-26 10:53:33 -0700967
968 dw.blankline();
969 dw.println("BPF map content:");
970
Erik Klineb31fd692018-06-06 20:50:11 +0900971 ScopedIndent indentForMapContent(dw);
Chenbo Fengef297172018-03-26 10:53:33 -0700972
973 // Print CookieTagMap content.
974 dumpBpfMap("mCookieTagMap", dw, "");
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700975 const auto printCookieTagInfo = [&dw](const uint64_t& key, const UidTag& value,
976 const BpfMap<uint64_t, UidTag>&) {
977 dw.println("cookie=%" PRIu64 " tag=0x%x uid=%u", key, value.tag, value.uid);
978 return netdutils::status::ok;
Chenbo Fengef297172018-03-26 10:53:33 -0700979 };
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700980 Status res = mCookieTagMap.iterateWithValue(printCookieTagInfo);
981 if (!isOk(res)) {
982 dw.println("mCookieTagMap print end with error: %s", res.msg().c_str());
Chenbo Fengef297172018-03-26 10:53:33 -0700983 }
984
985 // Print UidCounterSetMap Content
986 dumpBpfMap("mUidCounterSetMap", dw, "");
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700987 const auto printUidInfo = [&dw](const uint32_t& key, const uint8_t& value,
988 const BpfMap<uint32_t, uint8_t>&) {
989 dw.println("%u %u", key, value);
990 return netdutils::status::ok;
Chenbo Fengef297172018-03-26 10:53:33 -0700991 };
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700992 res = mUidCounterSetMap.iterateWithValue(printUidInfo);
993 if (!isOk(res)) {
994 dw.println("mUidCounterSetMap print end with error: %s", res.msg().c_str());
Chenbo Fengef297172018-03-26 10:53:33 -0700995 }
996
Chenbo Fengbc4a15f2018-05-11 19:15:15 -0700997 // Print AppUidStatsMap content
998 std::string appUidStatsHeader = StringPrintf("uid rxBytes rxPackets txBytes txPackets");
999 dumpBpfMap("mAppUidStatsMap:", dw, appUidStatsHeader);
1000 auto printAppUidStatsInfo = [&dw](const uint32_t& key, const StatsValue& value,
1001 const BpfMap<uint32_t, StatsValue>&) {
1002 dw.println("%u %" PRIu64 " %" PRIu64 " %" PRIu64 " %" PRIu64, key, value.rxBytes,
1003 value.rxPackets, value.txBytes, value.txPackets);
1004 return netdutils::status::ok;
1005 };
1006 res = mAppUidStatsMap.iterateWithValue(printAppUidStatsInfo);
1007 if (!res.ok()) {
1008 dw.println("mAppUidStatsMap print end with error: %s", res.msg().c_str());
1009 }
1010
Chenbo Fengef297172018-03-26 10:53:33 -07001011 // Print uidStatsMap content
1012 std::string statsHeader = StringPrintf("ifaceIndex ifaceName tag_hex uid_int cnt_set rxBytes"
1013 " rxPackets txBytes txPackets");
Chenbo Fengf434e862018-06-27 14:08:39 -07001014 dumpBpfMap("mStatsMapA", dw, statsHeader);
Chenbo Feng4f6c2372018-04-26 10:37:55 -07001015 const auto printStatsInfo = [&dw, this](const StatsKey& key, const StatsValue& value,
1016 const BpfMap<StatsKey, StatsValue>&) {
1017 uint32_t ifIndex = key.ifaceIndex;
1018 auto ifname = mIfaceIndexNameMap.readValue(ifIndex);
1019 if (!isOk(ifname)) {
1020 strlcpy(ifname.value().name, "unknown", sizeof(IfaceValue));
Chenbo Fengef297172018-03-26 10:53:33 -07001021 }
Chenbo Feng4f6c2372018-04-26 10:37:55 -07001022 dw.println("%u %s 0x%x %u %u %" PRIu64 " %" PRIu64 " %" PRIu64 " %" PRIu64, ifIndex,
1023 ifname.value().name, key.tag, key.uid, key.counterSet, value.rxBytes,
1024 value.rxPackets, value.txBytes, value.txPackets);
1025 return netdutils::status::ok;
Chenbo Fengef297172018-03-26 10:53:33 -07001026 };
Chenbo Fengf434e862018-06-27 14:08:39 -07001027 res = mStatsMapA.iterateWithValue(printStatsInfo);
Chenbo Feng4f6c2372018-04-26 10:37:55 -07001028 if (!isOk(res)) {
Chenbo Fengf434e862018-06-27 14:08:39 -07001029 dw.println("mStatsMapA print end with error: %s", res.msg().c_str());
Chenbo Fengef297172018-03-26 10:53:33 -07001030 }
1031
1032 // Print TagStatsMap content.
Chenbo Fengf434e862018-06-27 14:08:39 -07001033 dumpBpfMap("mStatsMapB", dw, statsHeader);
1034 res = mStatsMapB.iterateWithValue(printStatsInfo);
Chenbo Feng4f6c2372018-04-26 10:37:55 -07001035 if (!isOk(res)) {
Chenbo Fengf434e862018-06-27 14:08:39 -07001036 dw.println("mStatsMapB print end with error: %s", res.msg().c_str());
Chenbo Fengef297172018-03-26 10:53:33 -07001037 }
1038
1039 // Print ifaceIndexToNameMap content.
1040 dumpBpfMap("mIfaceIndexNameMap", dw, "");
Chenbo Feng4f6c2372018-04-26 10:37:55 -07001041 const auto printIfaceNameInfo = [&dw](const uint32_t& key, const IfaceValue& value,
1042 const BpfMap<uint32_t, IfaceValue>&) {
1043 const char* ifname = value.name;
1044 dw.println("ifaceIndex=%u ifaceName=%s", key, ifname);
1045 return netdutils::status::ok;
Chenbo Fengef297172018-03-26 10:53:33 -07001046 };
Chenbo Feng4f6c2372018-04-26 10:37:55 -07001047 res = mIfaceIndexNameMap.iterateWithValue(printIfaceNameInfo);
1048 if (!isOk(res)) {
1049 dw.println("mIfaceIndexNameMap print end with error: %s", res.msg().c_str());
Chenbo Fengef297172018-03-26 10:53:33 -07001050 }
1051
1052 // Print ifaceStatsMap content
1053 std::string ifaceStatsHeader = StringPrintf("ifaceIndex ifaceName rxBytes rxPackets txBytes"
1054 " txPackets");
1055 dumpBpfMap("mIfaceStatsMap:", dw, ifaceStatsHeader);
Chenbo Feng4f6c2372018-04-26 10:37:55 -07001056 const auto printIfaceStatsInfo = [&dw, this](const uint32_t& key, const StatsValue& value,
1057 const BpfMap<uint32_t, StatsValue>&) {
1058 auto ifname = mIfaceIndexNameMap.readValue(key);
1059 if (!isOk(ifname)) {
1060 strlcpy(ifname.value().name, "unknown", sizeof(IfaceValue));
Chenbo Fengef297172018-03-26 10:53:33 -07001061 }
Chenbo Feng4f6c2372018-04-26 10:37:55 -07001062 dw.println("%u %s %" PRIu64 " %" PRIu64 " %" PRIu64 " %" PRIu64, key, ifname.value().name,
1063 value.rxBytes, value.rxPackets, value.txBytes, value.txPackets);
1064 return netdutils::status::ok;
Chenbo Fengef297172018-03-26 10:53:33 -07001065 };
Chenbo Feng4f6c2372018-04-26 10:37:55 -07001066 res = mIfaceStatsMap.iterateWithValue(printIfaceStatsInfo);
1067 if (!isOk(res)) {
1068 dw.println("mIfaceStatsMap print end with error: %s", res.msg().c_str());
Chenbo Fengef297172018-03-26 10:53:33 -07001069 }
1070
Chenbo Feng703798e2018-06-15 17:07:59 -07001071 dw.blankline();
Chenbo Fengef297172018-03-26 10:53:33 -07001072
Chenbo Fengf434e862018-06-27 14:08:39 -07001073 uint32_t key = UID_RULES_CONFIGURATION_KEY;
Chenbo Feng703798e2018-06-15 17:07:59 -07001074 auto configuration = mConfigurationMap.readValue(key);
1075 if (isOk(configuration)) {
Chenbo Fengf434e862018-06-27 14:08:39 -07001076 dw.println("current ownerMatch configuration: %d", configuration.value());
Chenbo Feng703798e2018-06-15 17:07:59 -07001077 } else {
Chenbo Fengf434e862018-06-27 14:08:39 -07001078 dw.println("mConfigurationMap read ownerMatch configure failed with error: %s",
1079 configuration.status().msg().c_str());
1080 }
1081 key = CURRENT_STATS_MAP_CONFIGURATION_KEY;
1082 configuration = mConfigurationMap.readValue(key);
1083 if (isOk(configuration)) {
1084 dw.println("current statsMap configuration: %d", configuration.value());
1085 } else {
1086 dw.println("mConfigurationMap read stats map configure failed with error: %s",
1087 configuration.status().msg().c_str());
Chenbo Fengef297172018-03-26 10:53:33 -07001088 }
Chenbo Feng703798e2018-06-15 17:07:59 -07001089 dumpBpfMap("mUidOwnerMap", dw, "");
Rubin Xuec27ff22019-01-08 21:33:03 +00001090 const auto printUidMatchInfo = [&dw, this](const uint32_t& key, const UidOwnerValue& value,
1091 const BpfMap<uint32_t, UidOwnerValue>&) {
1092 if (value.rule & IIF_MATCH) {
1093 auto ifname = mIfaceIndexNameMap.readValue(value.iif);
1094 if (isOk(ifname)) {
1095 dw.println("%u %s %s", key, uidMatchTypeToString(value.rule).c_str(),
1096 ifname.value().name);
1097 } else {
1098 dw.println("%u %s %u", key, uidMatchTypeToString(value.rule).c_str(), value.iif);
1099 }
1100 } else {
1101 dw.println("%u %s", key, uidMatchTypeToString(value.rule).c_str());
1102 }
Chenbo Feng703798e2018-06-15 17:07:59 -07001103 return netdutils::status::ok;
1104 };
1105 res = mUidOwnerMap.iterateWithValue(printUidMatchInfo);
Chenbo Feng4f6c2372018-04-26 10:37:55 -07001106 if (!isOk(res)) {
Chenbo Feng703798e2018-06-15 17:07:59 -07001107 dw.println("mUidOwnerMap print end with error: %s", res.msg().c_str());
Chenbo Feng95892f32018-06-07 14:52:02 -07001108 }
Chenbo Feng48eaed32018-12-26 17:40:21 -08001109 dumpBpfMap("mUidPermissionMap", dw, "");
1110 const auto printUidPermissionInfo = [&dw](const uint32_t& key, const uint8_t& value,
1111 const BpfMap<uint32_t, uint8_t>&) {
1112 dw.println("%u %s", key, UidPermissionTypeToString(value).c_str());
1113 return netdutils::status::ok;
1114 };
1115 res = mUidPermissionMap.iterateWithValue(printUidPermissionInfo);
1116 if (!isOk(res)) {
1117 dw.println("mUidPermissionMap print end with error: %s", res.msg().c_str());
1118 }
1119
1120 dumpBpfMap("mPrivilegedUser", dw, "");
1121 for (uid_t uid : mPrivilegedUser) {
1122 dw.println("%u ALLOW_UPDATE_DEVICE_STATS", (uint32_t)uid);
1123 }
Chenbo Fengef297172018-03-26 10:53:33 -07001124}
1125
Chenbo Fengf2759682017-10-10 17:31:57 -07001126} // namespace net
1127} // namespace android