blob: cbf6b04953a024501f7ea2291ae1efeb75f1df9e [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;
76constexpr int TOTAL_UID_STATS_ENTRIES_LIMIT = STATS_MAP_SIZE - 500;
Chenbo Feng116d0552017-12-04 17:25:19 -080077
Chenbo Fengbf660aa2019-02-26 16:12:27 -080078static_assert(BPF_PERMISSION_INTERNET == INetd::PERMISSION_INTERNET,
79 "Mismatch between BPF and AIDL permissions: PERMISSION_INTERNET");
80static_assert(BPF_PERMISSION_UPDATE_DEVICE_STATS == INetd::PERMISSION_UPDATE_DEVICE_STATS,
81 "Mismatch between BPF and AIDL permissions: PERMISSION_UPDATE_DEVICE_STATS");
82
Chenbo Feng48eaed32018-12-26 17:40:21 -080083#define FLAG_MSG_TRANS(result, flag, value) \
84 do { \
85 if (value & flag) { \
86 result.append(StringPrintf(" %s", #flag)); \
87 value &= ~flag; \
88 } \
Chenbo Feng703798e2018-06-15 17:07:59 -070089 } while (0)
90
91const std::string uidMatchTypeToString(uint8_t match) {
92 std::string matchType;
Chenbo Feng48eaed32018-12-26 17:40:21 -080093 FLAG_MSG_TRANS(matchType, HAPPY_BOX_MATCH, match);
94 FLAG_MSG_TRANS(matchType, PENALTY_BOX_MATCH, match);
95 FLAG_MSG_TRANS(matchType, DOZABLE_MATCH, match);
96 FLAG_MSG_TRANS(matchType, STANDBY_MATCH, match);
97 FLAG_MSG_TRANS(matchType, POWERSAVE_MATCH, match);
Rubin Xuec27ff22019-01-08 21:33:03 +000098 FLAG_MSG_TRANS(matchType, IIF_MATCH, match);
Chenbo Feng703798e2018-06-15 17:07:59 -070099 if (match) {
100 return StringPrintf("Unknown match: %u", match);
101 }
102 return matchType;
103}
104
Chenbo Fengb4a4fa12019-01-09 17:20:45 -0800105bool TrafficController::hasUpdateDeviceStatsPermission(uid_t uid) {
Chenbo Feng3969c262019-03-19 14:53:57 -0700106 // This implementation is the same logic as method ActivityManager#checkComponentPermission.
107 // It implies that the calling uid can never be the same as PER_USER_RANGE.
108 uint32_t appId = uid % PER_USER_RANGE;
109 return ((appId == AID_ROOT) || (appId == AID_SYSTEM) ||
110 mPrivilegedUser.find(appId) != mPrivilegedUser.end());
Chenbo Fengb4a4fa12019-01-09 17:20:45 -0800111}
112
Chenbo Feng48eaed32018-12-26 17:40:21 -0800113const std::string UidPermissionTypeToString(uint8_t permission) {
Chenbo Fengbf660aa2019-02-26 16:12:27 -0800114 if (permission == INetd::NO_PERMISSIONS) {
115 return "NO_PERMISSIONS";
116 }
Chenbo Feng527908a2019-04-02 20:14:11 -0700117 if (permission == INetd::PERMISSION_UNINSTALLED) {
118 // This should never appear in the map, complain loudly if it does.
119 return "PERMISSION_UNINSTALLED error!";
120 }
Chenbo Feng48eaed32018-12-26 17:40:21 -0800121 std::string permissionType;
Chenbo Fengbf660aa2019-02-26 16:12:27 -0800122 FLAG_MSG_TRANS(permissionType, BPF_PERMISSION_INTERNET, permission);
123 FLAG_MSG_TRANS(permissionType, BPF_PERMISSION_UPDATE_DEVICE_STATS, permission);
Chenbo Feng48eaed32018-12-26 17:40:21 -0800124 if (permission) {
125 return StringPrintf("Unknown permission: %u", permission);
126 }
127 return permissionType;
128}
129
Chenbo Feng49586642018-08-30 18:01:53 -0700130StatusOr<std::unique_ptr<NetlinkListenerInterface>> TrafficController::makeSkDestroyListener() {
Chenbo Feng116d0552017-12-04 17:25:19 -0800131 const auto& sys = sSyscalls.get();
132 ASSIGN_OR_RETURN(auto event, sys.eventfd(0, EFD_CLOEXEC));
133 const int domain = AF_NETLINK;
134 const int type = SOCK_DGRAM | SOCK_CLOEXEC | SOCK_NONBLOCK;
135 const int protocol = NETLINK_INET_DIAG;
136 ASSIGN_OR_RETURN(auto sock, sys.socket(domain, type, protocol));
137
Lorenzo Colitti436efdb2018-07-26 17:20:57 +0900138 // TODO: if too many sockets are closed too quickly, we can overflow the socket buffer, and
139 // some entries in mCookieTagMap will not be freed. In order to fix this we would need to
140 // periodically dump all sockets and remove the tag entries for sockets that have been closed.
141 // For now, set a large-enough buffer that we can close hundreds of sockets without getting
142 // ENOBUFS and leaking mCookieTagMap entries.
143 int rcvbuf = 512 * 1024;
144 auto ret = sys.setsockopt(sock, SOL_SOCKET, SO_RCVBUF, &rcvbuf, sizeof(rcvbuf));
145 if (!ret.ok()) {
146 ALOGW("Failed to set SkDestroyListener buffer size to %d: %s", rcvbuf, ret.msg().c_str());
147 }
148
Chenbo Feng116d0552017-12-04 17:25:19 -0800149 sockaddr_nl addr = {
150 .nl_family = AF_NETLINK,
151 .nl_groups = 1 << (SKNLGRP_INET_TCP_DESTROY - 1) | 1 << (SKNLGRP_INET_UDP_DESTROY - 1) |
152 1 << (SKNLGRP_INET6_TCP_DESTROY - 1) | 1 << (SKNLGRP_INET6_UDP_DESTROY - 1)};
153 RETURN_IF_NOT_OK(sys.bind(sock, addr));
154
155 const sockaddr_nl kernel = {.nl_family = AF_NETLINK};
156 RETURN_IF_NOT_OK(sys.connect(sock, kernel));
157
158 std::unique_ptr<NetlinkListenerInterface> listener =
Chenbo Feng46296702018-08-23 12:20:22 -0700159 std::make_unique<NetlinkListener>(std::move(event), std::move(sock), "SkDestroyListen");
Chenbo Feng116d0552017-12-04 17:25:19 -0800160
161 return listener;
162}
163
Chenbo Feng89c12f12018-03-21 10:29:18 -0700164Status changeOwnerAndMode(const char* path, gid_t group, const char* debugName, bool netdOnly) {
165 int ret = chown(path, AID_ROOT, group);
166 if (ret != 0) return statusFromErrno(errno, StringPrintf("change %s group failed", debugName));
167
168 if (netdOnly) {
169 ret = chmod(path, S_IRWXU);
170 } else {
171 // Allow both netd and system server to obtain map fd from the path.
172 // chmod doesn't grant permission to all processes in that group to
173 // read/write the bpf map. They still need correct sepolicy to
174 // read/write the map.
Chenbo Fengf434e862018-06-27 14:08:39 -0700175 ret = chmod(path, S_IRWXU | S_IRGRP | S_IWGRP);
Chenbo Feng89c12f12018-03-21 10:29:18 -0700176 }
177 if (ret != 0) return statusFromErrno(errno, StringPrintf("change %s mode failed", debugName));
178 return netdutils::status::ok;
179}
180
Chenbo Feng82075e52019-04-12 19:21:47 -0700181TrafficController::TrafficController()
182 : mBpfLevel(getBpfSupportLevel()),
183 mPerUidStatsEntriesLimit(PER_UID_STATS_ENTRIES_LIMIT),
184 mTotalUidStatsEntriesLimit(TOTAL_UID_STATS_ENTRIES_LIMIT) {}
185
186TrafficController::TrafficController(uint32_t perUidLimit, uint32_t totalLimit)
187 : mBpfLevel(getBpfSupportLevel()),
188 mPerUidStatsEntriesLimit(perUidLimit),
189 mTotalUidStatsEntriesLimit(totalLimit) {}
Chenbo Feng89c12f12018-03-21 10:29:18 -0700190
Chenbo Feng89c12f12018-03-21 10:29:18 -0700191Status TrafficController::initMaps() {
Chenbo Feng84d9f9f2019-04-12 15:31:02 -0700192 std::lock_guard guard(mMutex);
Maciej Żenczykowski02554372019-04-01 10:44:54 -0700193 RETURN_IF_NOT_OK(mCookieTagMap.init(COOKIE_TAG_MAP_PATH));
Chenbo Feng89c12f12018-03-21 10:29:18 -0700194 RETURN_IF_NOT_OK(changeOwnerAndMode(COOKIE_TAG_MAP_PATH, AID_NET_BW_ACCT, "CookieTagMap",
195 false));
196
Maciej Żenczykowski02554372019-04-01 10:44:54 -0700197 RETURN_IF_NOT_OK(mUidCounterSetMap.init(UID_COUNTERSET_MAP_PATH));
Chenbo Feng89c12f12018-03-21 10:29:18 -0700198 RETURN_IF_NOT_OK(changeOwnerAndMode(UID_COUNTERSET_MAP_PATH, AID_NET_BW_ACCT,
199 "UidCounterSetMap", false));
200
Maciej Żenczykowski02554372019-04-01 10:44:54 -0700201 RETURN_IF_NOT_OK(mAppUidStatsMap.init(APP_UID_STATS_MAP_PATH));
Chenbo Fengbc4a15f2018-05-11 19:15:15 -0700202 RETURN_IF_NOT_OK(
203 changeOwnerAndMode(APP_UID_STATS_MAP_PATH, AID_NET_BW_STATS, "AppUidStatsMap", false));
204
Maciej Żenczykowski02554372019-04-01 10:44:54 -0700205 RETURN_IF_NOT_OK(mStatsMapA.init(STATS_MAP_A_PATH));
Chenbo Fengf434e862018-06-27 14:08:39 -0700206 RETURN_IF_NOT_OK(changeOwnerAndMode(STATS_MAP_A_PATH, AID_NET_BW_STATS, "StatsMapA", false));
Chenbo Feng89c12f12018-03-21 10:29:18 -0700207
Maciej Żenczykowski02554372019-04-01 10:44:54 -0700208 RETURN_IF_NOT_OK(mStatsMapB.init(STATS_MAP_B_PATH));
Chenbo Fengf434e862018-06-27 14:08:39 -0700209 RETURN_IF_NOT_OK(changeOwnerAndMode(STATS_MAP_B_PATH, AID_NET_BW_STATS, "StatsMapB", false));
Chenbo Feng89c12f12018-03-21 10:29:18 -0700210
Maciej Żenczykowski02554372019-04-01 10:44:54 -0700211 RETURN_IF_NOT_OK(mIfaceIndexNameMap.init(IFACE_INDEX_NAME_MAP_PATH));
Chenbo Feng89c12f12018-03-21 10:29:18 -0700212 RETURN_IF_NOT_OK(changeOwnerAndMode(IFACE_INDEX_NAME_MAP_PATH, AID_NET_BW_STATS,
213 "IfaceIndexNameMap", false));
214
Maciej Żenczykowski02554372019-04-01 10:44:54 -0700215 RETURN_IF_NOT_OK(mIfaceStatsMap.init(IFACE_STATS_MAP_PATH));
Chenbo Feng89c12f12018-03-21 10:29:18 -0700216 RETURN_IF_NOT_OK(changeOwnerAndMode(IFACE_STATS_MAP_PATH, AID_NET_BW_STATS, "IfaceStatsMap",
217 false));
Chenbo Feng95892f32018-06-07 14:52:02 -0700218
Maciej Żenczykowski02554372019-04-01 10:44:54 -0700219 RETURN_IF_NOT_OK(mConfigurationMap.init(CONFIGURATION_MAP_PATH));
Chenbo Fengf434e862018-06-27 14:08:39 -0700220 RETURN_IF_NOT_OK(changeOwnerAndMode(CONFIGURATION_MAP_PATH, AID_NET_BW_STATS,
221 "ConfigurationMap", false));
Chenbo Feng703798e2018-06-15 17:07:59 -0700222 RETURN_IF_NOT_OK(
Chenbo Fengf434e862018-06-27 14:08:39 -0700223 mConfigurationMap.writeValue(UID_RULES_CONFIGURATION_KEY, DEFAULT_CONFIG, BPF_ANY));
224 RETURN_IF_NOT_OK(mConfigurationMap.writeValue(CURRENT_STATS_MAP_CONFIGURATION_KEY, SELECT_MAP_A,
225 BPF_ANY));
Chenbo Feng703798e2018-06-15 17:07:59 -0700226
Maciej Żenczykowski02554372019-04-01 10:44:54 -0700227 RETURN_IF_NOT_OK(mUidOwnerMap.init(UID_OWNER_MAP_PATH));
Chenbo Feng703798e2018-06-15 17:07:59 -0700228 RETURN_IF_NOT_OK(changeOwnerAndMode(UID_OWNER_MAP_PATH, AID_ROOT, "UidOwnerMap", true));
Bernie Innocenti6f9fd902018-10-11 20:50:23 +0900229 RETURN_IF_NOT_OK(mUidOwnerMap.clear());
Maciej Żenczykowski02554372019-04-01 10:44:54 -0700230 RETURN_IF_NOT_OK(mUidPermissionMap.init(UID_PERMISSION_MAP_PATH));
Chenbo Feng89c12f12018-03-21 10:29:18 -0700231 return netdutils::status::ok;
232}
233
Chenbo Fengb2d213d2018-12-04 11:31:00 -0800234static Status attachProgramToCgroup(const char* programPath, const int cgroupFd,
235 bpf_attach_type type) {
236 unique_fd cgroupProg(bpfFdGet(programPath, 0));
237 if (cgroupProg == -1) {
238 int ret = errno;
239 ALOGE("Failed to get program from %s: %s", programPath, strerror(ret));
240 return statusFromErrno(ret, "cgroup program get failed");
241 }
242 if (android::bpf::attachProgram(type, cgroupProg, cgroupFd)) {
243 int ret = errno;
244 ALOGE("Program from %s attach failed: %s", programPath, strerror(ret));
245 return statusFromErrno(ret, "program attach failed");
246 }
247 return netdutils::status::ok;
248}
249
250static Status initPrograms() {
Suren Baghdasaryane072a3c2019-01-16 14:36:07 -0800251 std::string cg2_path;
252
253 if (!CgroupGetControllerPath(CGROUPV2_CONTROLLER_NAME, &cg2_path)) {
254 int ret = errno;
255 ALOGE("Failed to find cgroup v2 root");
256 return statusFromErrno(ret, "Failed to find cgroup v2 root");
257 }
258
259 unique_fd cg_fd(open(cg2_path.c_str(), O_DIRECTORY | O_RDONLY | O_CLOEXEC));
Chenbo Fengb2d213d2018-12-04 11:31:00 -0800260 if (cg_fd == -1) {
261 int ret = errno;
262 ALOGE("Failed to open the cgroup directory: %s", strerror(ret));
263 return statusFromErrno(ret, "Open the cgroup directory failed");
264 }
265 RETURN_IF_NOT_OK(attachProgramToCgroup(BPF_EGRESS_PROG_PATH, cg_fd, BPF_CGROUP_INET_EGRESS));
266 RETURN_IF_NOT_OK(attachProgramToCgroup(BPF_INGRESS_PROG_PATH, cg_fd, BPF_CGROUP_INET_INGRESS));
Chenbo Fenga51f4fa2019-01-15 15:03:32 -0800267
268 // For the devices that support cgroup socket filter, the socket filter
269 // should be loaded successfully by bpfloader. So we attach the filter to
270 // cgroup if the program is pinned properly.
271 // TODO: delete the if statement once all devices should support cgroup
272 // socket filter (ie. the minimum kernel version required is 4.14).
273 if (!access(CGROUP_SOCKET_PROG_PATH, F_OK)) {
274 RETURN_IF_NOT_OK(
275 attachProgramToCgroup(CGROUP_SOCKET_PROG_PATH, cg_fd, BPF_CGROUP_INET_SOCK_CREATE));
276 }
Chenbo Fengb2d213d2018-12-04 11:31:00 -0800277 return netdutils::status::ok;
278}
279
Chenbo Feng89c12f12018-03-21 10:29:18 -0700280Status TrafficController::start() {
Chenbo Feng47dd0732018-12-11 12:23:24 -0800281 if (mBpfLevel == BpfLevel::NONE) {
Chenbo Fengc10a8a42017-12-15 13:56:33 -0800282 return netdutils::status::ok;
Chenbo Fengf2759682017-10-10 17:31:57 -0700283 }
284
Bernie Innocenti6f9fd902018-10-11 20:50:23 +0900285 /* When netd restarts from a crash without total system reboot, the program
Chenbo Fengf2759682017-10-10 17:31:57 -0700286 * is still attached to the cgroup, detach it so the program can be freed
287 * and we can load and attach new program into the target cgroup.
288 *
289 * TODO: Scrape existing socket when run-time restart and clean up the map
290 * if the socket no longer exist
291 */
292
Chenbo Feng89c12f12018-03-21 10:29:18 -0700293 RETURN_IF_NOT_OK(initMaps());
Chenbo Feng7e974052018-02-28 22:57:21 -0800294
Chenbo Fengb2d213d2018-12-04 11:31:00 -0800295 RETURN_IF_NOT_OK(initPrograms());
296
Chenbo Feng7e974052018-02-28 22:57:21 -0800297 // Fetch the list of currently-existing interfaces. At this point NetlinkHandler is
298 // already running, so it will call addInterface() when any new interface appears.
299 std::map<std::string, uint32_t> ifacePairs;
300 ASSIGN_OR_RETURN(ifacePairs, InterfaceController::getIfaceList());
301 for (const auto& ifacePair:ifacePairs) {
302 addInterface(ifacePair.first.c_str(), ifacePair.second);
303 }
304
Chenbo Feng116d0552017-12-04 17:25:19 -0800305 auto result = makeSkDestroyListener();
306 if (!isOk(result)) {
307 ALOGE("Unable to create SkDestroyListener: %s", toString(result).c_str());
308 } else {
309 mSkDestroyListener = std::move(result.value());
310 }
311 // Rx handler extracts nfgenmsg looks up and invokes registered dispatch function.
312 const auto rxHandler = [this](const nlmsghdr&, const Slice msg) {
313 inet_diag_msg diagmsg = {};
314 if (extract(msg, diagmsg) < sizeof(inet_diag_msg)) {
Bernie Innocenti6f9fd902018-10-11 20:50:23 +0900315 ALOGE("Unrecognized netlink message: %s", toString(msg).c_str());
Chenbo Feng116d0552017-12-04 17:25:19 -0800316 return;
317 }
318 uint64_t sock_cookie = static_cast<uint64_t>(diagmsg.id.idiag_cookie[0]) |
319 (static_cast<uint64_t>(diagmsg.id.idiag_cookie[1]) << 32);
320
Bernie Innocenti6f9fd902018-10-11 20:50:23 +0900321 Status s = mCookieTagMap.deleteValue(sock_cookie);
Chenbo Feng1d4c5db2018-10-18 15:50:46 -0700322 if (!isOk(s) && s.code() != ENOENT) {
Bernie Innocenti6f9fd902018-10-11 20:50:23 +0900323 ALOGE("Failed to delete cookie %" PRIx64 ": %s", sock_cookie, toString(s).c_str());
324 return;
325 }
Chenbo Feng116d0552017-12-04 17:25:19 -0800326 };
327 expectOk(mSkDestroyListener->subscribe(kSockDiagMsgType, rxHandler));
328
329 // In case multiple netlink message comes in as a stream, we need to handle the rxDone message
330 // properly.
331 const auto rxDoneHandler = [](const nlmsghdr&, const Slice msg) {
332 // Ignore NLMSG_DONE messages
333 inet_diag_msg diagmsg = {};
334 extract(msg, diagmsg);
335 };
336 expectOk(mSkDestroyListener->subscribe(kSockDiagDoneMsgType, rxDoneHandler));
337
Chenbo Feng05393d82018-01-09 15:18:43 -0800338 return netdutils::status::ok;
Chenbo Fengf2759682017-10-10 17:31:57 -0700339}
340
Chenbo Fengb4a4fa12019-01-09 17:20:45 -0800341int TrafficController::tagSocket(int sockFd, uint32_t tag, uid_t uid, uid_t callingUid) {
Chenbo Feng84d9f9f2019-04-12 15:31:02 -0700342 std::lock_guard guard(mMutex);
Chenbo Fengb4a4fa12019-01-09 17:20:45 -0800343 if (uid != callingUid && !hasUpdateDeviceStatsPermission(callingUid)) {
344 return -EPERM;
345 }
346
Chenbo Feng47dd0732018-12-11 12:23:24 -0800347 if (mBpfLevel == BpfLevel::NONE) {
Chenbo Fengbc6d4702018-04-11 18:27:19 -0700348 if (legacy_tagSocket(sockFd, tag, uid)) return -errno;
349 return 0;
350 }
Chenbo Feng33cc1032017-10-23 15:16:37 -0700351
Chenbo Fengf2759682017-10-10 17:31:57 -0700352 uint64_t sock_cookie = getSocketCookie(sockFd);
Chenbo Fengef1cab32018-04-13 19:50:49 -0700353 if (sock_cookie == NONEXISTENT_COOKIE) return -errno;
Chenbo Fengf2759682017-10-10 17:31:57 -0700354 UidTag newKey = {.uid = (uint32_t)uid, .tag = tag};
355
Chenbo Fengb0495132019-04-10 12:54:41 -0700356 uint32_t totalEntryCount = 0;
Chenbo Feng82075e52019-04-12 19:21:47 -0700357 uint32_t perUidEntryCount = 0;
Chenbo Fengb0495132019-04-10 12:54:41 -0700358 // Now we go through the stats map and count how many entries are associated
359 // with target uid. If the uid entry hit the limit for each uid, we block
360 // the request to prevent the map from overflow. It is safe here to iterate
Chenbo Feng84d9f9f2019-04-12 15:31:02 -0700361 // over the map since when mMutex is hold, system server cannot toggle
Chenbo Fengb0495132019-04-10 12:54:41 -0700362 // the live stats map and clean it. So nobody can delete entries from the map.
Chenbo Feng82075e52019-04-12 19:21:47 -0700363 const auto countUidStatsEntries = [uid, &totalEntryCount, &perUidEntryCount](
364 const StatsKey& key, BpfMap<StatsKey, StatsValue>&) {
Chenbo Fengb0495132019-04-10 12:54:41 -0700365 if (key.uid == uid) {
Chenbo Feng82075e52019-04-12 19:21:47 -0700366 perUidEntryCount++;
Chenbo Fengb0495132019-04-10 12:54:41 -0700367 }
Chenbo Feng82075e52019-04-12 19:21:47 -0700368 totalEntryCount++;
Chenbo Fengb0495132019-04-10 12:54:41 -0700369 return netdutils::status::ok;
370 };
371 auto configuration = mConfigurationMap.readValue(CURRENT_STATS_MAP_CONFIGURATION_KEY);
372 if (!isOk(configuration.status())) {
373 ALOGE("Failed to get current configuration: %s, fd: %d",
374 strerror(configuration.status().code()), mConfigurationMap.getMap().get());
375 return -configuration.status().code();
376 }
377 if (configuration.value() == SELECT_MAP_A) {
378 mStatsMapA.iterate(countUidStatsEntries).ignoreError();
379 } else if (configuration.value() == SELECT_MAP_B) {
380 mStatsMapB.iterate(countUidStatsEntries).ignoreError();
381 } else {
382 ALOGE("unknown configuration value: %d", configuration.value());
383 return -EINVAL;
384 }
Chenbo Feng82075e52019-04-12 19:21:47 -0700385 if (totalEntryCount > mTotalUidStatsEntriesLimit ||
386 perUidEntryCount > mPerUidStatsEntriesLimit) {
387 ALOGE("Too many stats entries in the map, total count: %u, uid(%u) count: %u, blocking tag"
388 " request to prevent map overflow",
389 totalEntryCount, uid, perUidEntryCount);
Chenbo Fengb0495132019-04-10 12:54:41 -0700390 return -EMFILE;
391 }
Chenbo Fengf2759682017-10-10 17:31:57 -0700392 // Update the tag information of a socket to the cookieUidMap. Use BPF_ANY
393 // flag so it will insert a new entry to the map if that value doesn't exist
394 // yet. And update the tag if there is already a tag stored. Since the eBPF
395 // program in kernel only read this map, and is protected by rcu read lock. It
396 // should be fine to cocurrently update the map while eBPF program is running.
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700397 Status res = mCookieTagMap.writeValue(sock_cookie, newKey, BPF_ANY);
398 if (!isOk(res)) {
399 ALOGE("Failed to tag the socket: %s, fd: %d", strerror(res.code()),
400 mCookieTagMap.getMap().get());
Chenbo Fengf2759682017-10-10 17:31:57 -0700401 }
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700402 return -res.code();
Chenbo Fengf2759682017-10-10 17:31:57 -0700403}
404
405int TrafficController::untagSocket(int sockFd) {
Chenbo Feng47dd0732018-12-11 12:23:24 -0800406 if (mBpfLevel == BpfLevel::NONE) {
Chenbo Fengbc6d4702018-04-11 18:27:19 -0700407 if (legacy_untagSocket(sockFd)) return -errno;
408 return 0;
409 }
Chenbo Fengf2759682017-10-10 17:31:57 -0700410 uint64_t sock_cookie = getSocketCookie(sockFd);
411
Chenbo Fengef1cab32018-04-13 19:50:49 -0700412 if (sock_cookie == NONEXISTENT_COOKIE) return -errno;
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700413 Status res = mCookieTagMap.deleteValue(sock_cookie);
414 if (!isOk(res)) {
415 ALOGE("Failed to untag socket: %s\n", strerror(res.code()));
Chenbo Fengf2759682017-10-10 17:31:57 -0700416 }
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700417 return -res.code();
Chenbo Fengf2759682017-10-10 17:31:57 -0700418}
419
Chenbo Fengb4a4fa12019-01-09 17:20:45 -0800420int TrafficController::setCounterSet(int counterSetNum, uid_t uid, uid_t callingUid) {
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700421 if (counterSetNum < 0 || counterSetNum >= OVERFLOW_COUNTERSET) return -EINVAL;
Chenbo Fengb4a4fa12019-01-09 17:20:45 -0800422
Chenbo Feng84d9f9f2019-04-12 15:31:02 -0700423 std::lock_guard guard(mMutex);
Chenbo Fengb4a4fa12019-01-09 17:20:45 -0800424 if (!hasUpdateDeviceStatsPermission(callingUid)) return -EPERM;
425
Chenbo Feng47dd0732018-12-11 12:23:24 -0800426 if (mBpfLevel == BpfLevel::NONE) {
Chenbo Fengbc6d4702018-04-11 18:27:19 -0700427 if (legacy_setCounterSet(counterSetNum, uid)) return -errno;
428 return 0;
429 }
Chenbo Fengc10a8a42017-12-15 13:56:33 -0800430
431 // The default counter set for all uid is 0, so deleting the current counterset for that uid
432 // will automatically set it to 0.
Chenbo Fengf2759682017-10-10 17:31:57 -0700433 if (counterSetNum == 0) {
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700434 Status res = mUidCounterSetMap.deleteValue(uid);
435 if (isOk(res) || (!isOk(res) && res.code() == ENOENT)) {
Chenbo Fengf2759682017-10-10 17:31:57 -0700436 return 0;
437 } else {
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700438 ALOGE("Failed to delete the counterSet: %s\n", strerror(res.code()));
439 return -res.code();
Chenbo Fengf2759682017-10-10 17:31:57 -0700440 }
441 }
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700442 uint8_t tmpCounterSetNum = (uint8_t)counterSetNum;
Chenbo Fengb4a4fa12019-01-09 17:20:45 -0800443 Status res = mUidCounterSetMap.writeValue(uid, tmpCounterSetNum, BPF_ANY);
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700444 if (!isOk(res)) {
445 ALOGE("Failed to set the counterSet: %s, fd: %d", strerror(res.code()),
446 mUidCounterSetMap.getMap().get());
447 return -res.code();
Chenbo Fengf2759682017-10-10 17:31:57 -0700448 }
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700449 return 0;
Chenbo Fengf2759682017-10-10 17:31:57 -0700450}
451
Chenbo Fengf434e862018-06-27 14:08:39 -0700452// This method only get called by system_server when an app get uinstalled, it
453// is called inside removeUidsLocked() while holding mStatsLock. So it is safe
454// to iterate and modify the stats maps.
Chenbo Fengb4a4fa12019-01-09 17:20:45 -0800455int TrafficController::deleteTagData(uint32_t tag, uid_t uid, uid_t callingUid) {
Chenbo Feng84d9f9f2019-04-12 15:31:02 -0700456 std::lock_guard guard(mMutex);
Chenbo Fengb4a4fa12019-01-09 17:20:45 -0800457 if (!hasUpdateDeviceStatsPermission(callingUid)) return -EPERM;
458
Chenbo Feng47dd0732018-12-11 12:23:24 -0800459 if (mBpfLevel == BpfLevel::NONE) {
Chenbo Fengbc6d4702018-04-11 18:27:19 -0700460 if (legacy_deleteTagData(tag, uid)) return -errno;
461 return 0;
462 }
Chenbo Feng33cc1032017-10-23 15:16:37 -0700463
Chenbo Fengc10a8a42017-12-15 13:56:33 -0800464 // First we go through the cookieTagMap to delete the target uid tag combination. Or delete all
Chenbo Fengef1cab32018-04-13 19:50:49 -0700465 // the tags related to the uid if the tag is 0.
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700466 const auto deleteMatchedCookieEntries = [uid, tag](const uint64_t& key, const UidTag& value,
467 BpfMap<uint64_t, UidTag>& map) {
468 if (value.uid == uid && (value.tag == tag || tag == 0)) {
469 Status res = map.deleteValue(key);
470 if (isOk(res) || (res.code() == ENOENT)) {
471 return netdutils::status::ok;
Chenbo Fengf2759682017-10-10 17:31:57 -0700472 }
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700473 ALOGE("Failed to delete data(cookie = %" PRIu64 "): %s\n", key, strerror(res.code()));
Chenbo Fengf2759682017-10-10 17:31:57 -0700474 }
Chenbo Fengef1cab32018-04-13 19:50:49 -0700475 // Move forward to next cookie in the map.
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700476 return netdutils::status::ok;
Chenbo Fengef1cab32018-04-13 19:50:49 -0700477 };
Bernie Innocenti6f9fd902018-10-11 20:50:23 +0900478 mCookieTagMap.iterateWithValue(deleteMatchedCookieEntries).ignoreError();
Chenbo Fengf2759682017-10-10 17:31:57 -0700479 // 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 -0700480 // combination. Or all tag stats under that uid if the target tag is 0.
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700481 const auto deleteMatchedUidTagEntries = [uid, tag](const StatsKey& key,
482 BpfMap<StatsKey, StatsValue>& map) {
483 if (key.uid == uid && (key.tag == tag || tag == 0)) {
484 Status res = map.deleteValue(key);
485 if (isOk(res) || (res.code() == ENOENT)) {
Chenbo Fengef1cab32018-04-13 19:50:49 -0700486 //Entry is deleted, use the current key to get a new nextKey;
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700487 return netdutils::status::ok;
Chenbo Fengf2759682017-10-10 17:31:57 -0700488 }
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700489 ALOGE("Failed to delete data(uid=%u, tag=%u): %s\n", key.uid, key.tag,
490 strerror(res.code()));
Chenbo Fengf2759682017-10-10 17:31:57 -0700491 }
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700492 return netdutils::status::ok;
Chenbo Fengef1cab32018-04-13 19:50:49 -0700493 };
Chenbo Fengf434e862018-06-27 14:08:39 -0700494 mStatsMapB.iterate(deleteMatchedUidTagEntries).ignoreError();
495 mStatsMapA.iterate(deleteMatchedUidTagEntries).ignoreError();
Chenbo Fengf2759682017-10-10 17:31:57 -0700496 // 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 -0800497 // need to delete the stats stored in uidStatsMap and counterSet map.
Chenbo Fengc10a8a42017-12-15 13:56:33 -0800498 if (tag != 0) return 0;
499
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700500 Status res = mUidCounterSetMap.deleteValue(uid);
501 if (!isOk(res) && res.code() != ENOENT) {
502 ALOGE("Failed to delete counterSet data(uid=%u, tag=%u): %s\n", uid, tag,
503 strerror(res.code()));
Chenbo Fengc10a8a42017-12-15 13:56:33 -0800504 }
Chenbo Fengbc4a15f2018-05-11 19:15:15 -0700505
506 auto deleteAppUidStatsEntry = [uid](const uint32_t& key, BpfMap<uint32_t, StatsValue>& map) {
507 if (key == uid) {
508 Status res = map.deleteValue(key);
509 if (isOk(res) || (res.code() == ENOENT)) {
510 return netdutils::status::ok;
511 }
512 ALOGE("Failed to delete data(uid=%u): %s", key, strerror(res.code()));
513 }
514 return netdutils::status::ok;
515 };
Bernie Innocenti6f9fd902018-10-11 20:50:23 +0900516 mAppUidStatsMap.iterate(deleteAppUidStatsEntry).ignoreError();
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700517 return 0;
Chenbo Fengf2759682017-10-10 17:31:57 -0700518}
519
Chenbo Feng7e974052018-02-28 22:57:21 -0800520int TrafficController::addInterface(const char* name, uint32_t ifaceIndex) {
Chenbo Feng47dd0732018-12-11 12:23:24 -0800521 if (mBpfLevel == BpfLevel::NONE) return 0;
Chenbo Feng7e974052018-02-28 22:57:21 -0800522
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700523 IfaceValue iface;
Chenbo Feng7e974052018-02-28 22:57:21 -0800524 if (ifaceIndex == 0) {
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700525 ALOGE("Unknown interface %s(%d)", name, ifaceIndex);
Chenbo Feng7e974052018-02-28 22:57:21 -0800526 return -1;
527 }
528
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700529 strlcpy(iface.name, name, sizeof(IfaceValue));
530 Status res = mIfaceIndexNameMap.writeValue(ifaceIndex, iface, BPF_ANY);
531 if (!isOk(res)) {
532 ALOGE("Failed to add iface %s(%d): %s", name, ifaceIndex, strerror(res.code()));
533 return -res.code();
Chenbo Feng7e974052018-02-28 22:57:21 -0800534 }
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700535 return 0;
Chenbo Feng7e974052018-02-28 22:57:21 -0800536}
537
Chenbo Feng703798e2018-06-15 17:07:59 -0700538Status TrafficController::updateOwnerMapEntry(UidOwnerMatchType match, uid_t uid, FirewallRule rule,
539 FirewallType type) {
Chenbo Feng84d9f9f2019-04-12 15:31:02 -0700540 std::lock_guard guard(mMutex);
Chenbo Feng89c12f12018-03-21 10:29:18 -0700541 if ((rule == ALLOW && type == WHITELIST) || (rule == DENY && type == BLACKLIST)) {
Rubin Xu5ca8e802019-04-05 11:50:32 +0100542 RETURN_IF_NOT_OK(addRule(mUidOwnerMap, uid, match));
Chenbo Feng89c12f12018-03-21 10:29:18 -0700543 } else if ((rule == ALLOW && type == BLACKLIST) || (rule == DENY && type == WHITELIST)) {
Rubin Xu5ca8e802019-04-05 11:50:32 +0100544 RETURN_IF_NOT_OK(removeRule(mUidOwnerMap, uid, match));
Chenbo Feng89c12f12018-03-21 10:29:18 -0700545 } else {
546 //Cannot happen.
Rubin Xuec27ff22019-01-08 21:33:03 +0000547 return statusFromErrno(EINVAL, "");
Chenbo Feng89c12f12018-03-21 10:29:18 -0700548 }
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700549 return netdutils::status::ok;
Chenbo Feng89c12f12018-03-21 10:29:18 -0700550}
551
Chenbo Feng703798e2018-06-15 17:07:59 -0700552UidOwnerMatchType TrafficController::jumpOpToMatch(BandwidthController::IptJumpOp jumpHandling) {
Chenbo Feng95892f32018-06-07 14:52:02 -0700553 switch (jumpHandling) {
554 case BandwidthController::IptJumpReject:
Chenbo Feng703798e2018-06-15 17:07:59 -0700555 return PENALTY_BOX_MATCH;
Chenbo Feng95892f32018-06-07 14:52:02 -0700556 case BandwidthController::IptJumpReturn:
Chenbo Feng703798e2018-06-15 17:07:59 -0700557 return HAPPY_BOX_MATCH;
Chenbo Feng95892f32018-06-07 14:52:02 -0700558 case BandwidthController::IptJumpNoAdd:
559 return NO_MATCH;
560 }
561}
562
Rubin Xu5ca8e802019-04-05 11:50:32 +0100563Status TrafficController::removeRule(BpfMap<uint32_t, UidOwnerValue>& map, uint32_t uid,
564 UidOwnerMatchType match) {
Chenbo Feng703798e2018-06-15 17:07:59 -0700565 auto oldMatch = map.readValue(uid);
566 if (isOk(oldMatch)) {
Rubin Xuec27ff22019-01-08 21:33:03 +0000567 UidOwnerValue newMatch = {.rule = static_cast<uint8_t>(oldMatch.value().rule & ~match),
568 .iif = (match == IIF_MATCH) ? 0 : oldMatch.value().iif};
569 if (newMatch.rule == 0) {
Chenbo Feng703798e2018-06-15 17:07:59 -0700570 RETURN_IF_NOT_OK(map.deleteValue(uid));
571 } else {
572 RETURN_IF_NOT_OK(map.writeValue(uid, newMatch, BPF_ANY));
573 }
574 } else {
575 return statusFromErrno(ENOENT, StringPrintf("uid: %u does not exist in map", uid));
576 }
577 return netdutils::status::ok;
578}
579
Rubin Xu5ca8e802019-04-05 11:50:32 +0100580Status TrafficController::addRule(BpfMap<uint32_t, UidOwnerValue>& map, uint32_t uid,
581 UidOwnerMatchType match, uint32_t iif) {
Rubin Xuec27ff22019-01-08 21:33:03 +0000582 // iif should be non-zero if and only if match == MATCH_IIF
583 if (match == IIF_MATCH && iif == 0) {
584 return statusFromErrno(EINVAL, "Interface match must have nonzero interface index");
585 } else if (match != IIF_MATCH && iif != 0) {
586 return statusFromErrno(EINVAL, "Non-interface match must have zero interface index");
587 }
Chenbo Feng703798e2018-06-15 17:07:59 -0700588 auto oldMatch = map.readValue(uid);
589 if (isOk(oldMatch)) {
Rubin Xuec27ff22019-01-08 21:33:03 +0000590 UidOwnerValue newMatch = {.rule = static_cast<uint8_t>(oldMatch.value().rule | match),
591 .iif = iif ? iif : oldMatch.value().iif};
592 RETURN_IF_NOT_OK(map.writeValue(uid, newMatch, BPF_ANY));
Chenbo Feng703798e2018-06-15 17:07:59 -0700593 } else {
Rubin Xuec27ff22019-01-08 21:33:03 +0000594 UidOwnerValue newMatch = {.rule = static_cast<uint8_t>(match), .iif = iif};
595 RETURN_IF_NOT_OK(map.writeValue(uid, newMatch, BPF_ANY));
Chenbo Feng703798e2018-06-15 17:07:59 -0700596 }
597 return netdutils::status::ok;
598}
599
600Status TrafficController::updateUidOwnerMap(const std::vector<std::string>& appStrUids,
601 BandwidthController::IptJumpOp jumpHandling,
602 BandwidthController::IptOp op) {
Chenbo Feng84d9f9f2019-04-12 15:31:02 -0700603 std::lock_guard guard(mMutex);
Chenbo Feng703798e2018-06-15 17:07:59 -0700604 UidOwnerMatchType match = jumpOpToMatch(jumpHandling);
605 if (match == NO_MATCH) {
606 return statusFromErrno(
607 EINVAL, StringPrintf("invalid IptJumpOp: %d, command: %d", jumpHandling, match));
Chenbo Feng95892f32018-06-07 14:52:02 -0700608 }
609 for (const auto& appStrUid : appStrUids) {
610 char* endPtr;
611 long uid = strtol(appStrUid.c_str(), &endPtr, 10);
612 if ((errno == ERANGE && (uid == LONG_MAX || uid == LONG_MIN)) ||
613 (endPtr == appStrUid.c_str()) || (*endPtr != '\0')) {
614 return statusFromErrno(errno, "invalid uid string:" + appStrUid);
615 }
616
Chenbo Feng95892f32018-06-07 14:52:02 -0700617 if (op == BandwidthController::IptOpDelete) {
Rubin Xu5ca8e802019-04-05 11:50:32 +0100618 RETURN_IF_NOT_OK(removeRule(mUidOwnerMap, uid, match));
Chenbo Feng95892f32018-06-07 14:52:02 -0700619 } else if (op == BandwidthController::IptOpInsert) {
Rubin Xu5ca8e802019-04-05 11:50:32 +0100620 RETURN_IF_NOT_OK(addRule(mUidOwnerMap, uid, match));
Chenbo Feng95892f32018-06-07 14:52:02 -0700621 } else {
622 // Cannot happen.
Chenbo Feng703798e2018-06-15 17:07:59 -0700623 return statusFromErrno(EINVAL, StringPrintf("invalid IptOp: %d, %d", op, match));
Chenbo Feng95892f32018-06-07 14:52:02 -0700624 }
625 }
626 return netdutils::status::ok;
627}
628
Chenbo Feng89c12f12018-03-21 10:29:18 -0700629int TrafficController::changeUidOwnerRule(ChildChain chain, uid_t uid, FirewallRule rule,
630 FirewallType type) {
Chenbo Feng47dd0732018-12-11 12:23:24 -0800631 if (mBpfLevel == BpfLevel::NONE) {
Chenbo Feng89c12f12018-03-21 10:29:18 -0700632 ALOGE("bpf is not set up, should use iptables rule");
633 return -ENOSYS;
634 }
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700635 Status res;
Chenbo Feng89c12f12018-03-21 10:29:18 -0700636 switch (chain) {
637 case DOZABLE:
Chenbo Feng703798e2018-06-15 17:07:59 -0700638 res = updateOwnerMapEntry(DOZABLE_MATCH, uid, rule, type);
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700639 break;
Chenbo Feng89c12f12018-03-21 10:29:18 -0700640 case STANDBY:
Chenbo Feng703798e2018-06-15 17:07:59 -0700641 res = updateOwnerMapEntry(STANDBY_MATCH, uid, rule, type);
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700642 break;
Chenbo Feng89c12f12018-03-21 10:29:18 -0700643 case POWERSAVE:
Chenbo Feng703798e2018-06-15 17:07:59 -0700644 res = updateOwnerMapEntry(POWERSAVE_MATCH, uid, rule, type);
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700645 break;
Chenbo Feng89c12f12018-03-21 10:29:18 -0700646 case NONE:
647 default:
648 return -EINVAL;
649 }
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700650 if (!isOk(res)) {
651 ALOGE("change uid(%u) rule of %d failed: %s, rule: %d, type: %d", uid, chain,
652 res.msg().c_str(), rule, type);
653 return -res.code();
654 }
655 return 0;
Chenbo Feng89c12f12018-03-21 10:29:18 -0700656}
657
Rubin Xu5ca8e802019-04-05 11:50:32 +0100658Status TrafficController::replaceRulesInMap(const UidOwnerMatchType match,
659 const std::vector<int32_t>& uids) {
Chenbo Feng84d9f9f2019-04-12 15:31:02 -0700660 std::lock_guard guard(mMutex);
Chenbo Feng89c12f12018-03-21 10:29:18 -0700661 std::set<int32_t> uidSet(uids.begin(), uids.end());
662 std::vector<uint32_t> uidsToDelete;
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700663 auto getUidsToDelete = [&uidsToDelete, &uidSet](const uint32_t& key,
Rubin Xuec27ff22019-01-08 21:33:03 +0000664 const BpfMap<uint32_t, UidOwnerValue>&) {
Chenbo Feng703798e2018-06-15 17:07:59 -0700665 if (uidSet.find((int32_t) key) == uidSet.end()) {
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700666 uidsToDelete.push_back(key);
Chenbo Feng89c12f12018-03-21 10:29:18 -0700667 }
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700668 return netdutils::status::ok;
Chenbo Feng89c12f12018-03-21 10:29:18 -0700669 };
Chenbo Feng703798e2018-06-15 17:07:59 -0700670 RETURN_IF_NOT_OK(mUidOwnerMap.iterate(getUidsToDelete));
Chenbo Feng89c12f12018-03-21 10:29:18 -0700671
672 for(auto uid : uidsToDelete) {
Rubin Xu5ca8e802019-04-05 11:50:32 +0100673 RETURN_IF_NOT_OK(removeRule(mUidOwnerMap, uid, match));
Chenbo Feng89c12f12018-03-21 10:29:18 -0700674 }
675
676 for (auto uid : uids) {
Rubin Xu5ca8e802019-04-05 11:50:32 +0100677 RETURN_IF_NOT_OK(addRule(mUidOwnerMap, uid, match));
Chenbo Feng89c12f12018-03-21 10:29:18 -0700678 }
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700679 return netdutils::status::ok;
Chenbo Feng89c12f12018-03-21 10:29:18 -0700680}
681
Rubin Xuec27ff22019-01-08 21:33:03 +0000682Status TrafficController::addUidInterfaceRules(const int iif,
683 const std::vector<int32_t>& uidsToAdd) {
684 if (mBpfLevel == BpfLevel::NONE) {
685 ALOGW("UID ingress interface filtering not possible without BPF owner match");
686 return statusFromErrno(EOPNOTSUPP, "eBPF not supported");
687 }
688 if (!iif) {
689 return statusFromErrno(EINVAL, "Interface rule must specify interface");
690 }
Chenbo Feng84d9f9f2019-04-12 15:31:02 -0700691 std::lock_guard guard(mMutex);
Rubin Xuec27ff22019-01-08 21:33:03 +0000692
693 for (auto uid : uidsToAdd) {
Rubin Xu5ca8e802019-04-05 11:50:32 +0100694 netdutils::Status result = addRule(mUidOwnerMap, uid, IIF_MATCH, iif);
Rubin Xuec27ff22019-01-08 21:33:03 +0000695 if (!isOk(result)) {
Rubin Xu5ca8e802019-04-05 11:50:32 +0100696 ALOGW("addRule failed(%d): uid=%d iif=%d", result.code(), uid, iif);
Rubin Xuec27ff22019-01-08 21:33:03 +0000697 }
698 }
699 return netdutils::status::ok;
700}
701
702Status TrafficController::removeUidInterfaceRules(const std::vector<int32_t>& uidsToDelete) {
703 if (mBpfLevel == BpfLevel::NONE) {
704 ALOGW("UID ingress interface filtering not possible without BPF owner match");
705 return statusFromErrno(EOPNOTSUPP, "eBPF not supported");
706 }
Chenbo Feng84d9f9f2019-04-12 15:31:02 -0700707 std::lock_guard guard(mMutex);
Rubin Xuec27ff22019-01-08 21:33:03 +0000708
709 for (auto uid : uidsToDelete) {
Rubin Xu5ca8e802019-04-05 11:50:32 +0100710 netdutils::Status result = removeRule(mUidOwnerMap, uid, IIF_MATCH);
Rubin Xuec27ff22019-01-08 21:33:03 +0000711 if (!isOk(result)) {
Rubin Xu5ca8e802019-04-05 11:50:32 +0100712 ALOGW("removeRule failed(%d): uid=%d", result.code(), uid);
Rubin Xuec27ff22019-01-08 21:33:03 +0000713 }
714 }
715 return netdutils::status::ok;
716}
717
Chenbo Feng89c12f12018-03-21 10:29:18 -0700718int TrafficController::replaceUidOwnerMap(const std::string& name, bool isWhitelist,
719 const std::vector<int32_t>& uids) {
Chenbo Feng89c12f12018-03-21 10:29:18 -0700720 FirewallRule rule;
721 FirewallType type;
722 if (isWhitelist) {
723 type = WHITELIST;
724 rule = ALLOW;
725 } else {
726 type = BLACKLIST;
727 rule = DENY;
728 }
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700729 Status res;
Chenbo Feng89c12f12018-03-21 10:29:18 -0700730 if (!name.compare(FirewallController::LOCAL_DOZABLE)) {
Rubin Xu5ca8e802019-04-05 11:50:32 +0100731 res = replaceRulesInMap(DOZABLE_MATCH, uids);
Chenbo Feng89c12f12018-03-21 10:29:18 -0700732 } else if (!name.compare(FirewallController::LOCAL_STANDBY)) {
Rubin Xu5ca8e802019-04-05 11:50:32 +0100733 res = replaceRulesInMap(STANDBY_MATCH, uids);
Chenbo Feng89c12f12018-03-21 10:29:18 -0700734 } else if (!name.compare(FirewallController::LOCAL_POWERSAVE)) {
Rubin Xu5ca8e802019-04-05 11:50:32 +0100735 res = replaceRulesInMap(POWERSAVE_MATCH, uids);
Chenbo Feng89c12f12018-03-21 10:29:18 -0700736 } else {
737 ALOGE("unknown chain name: %s", name.c_str());
738 return -EINVAL;
739 }
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700740 if (!isOk(res)) {
741 ALOGE("Failed to clean up chain: %s: %s", name.c_str(), res.msg().c_str());
742 return -res.code();
Chenbo Feng89c12f12018-03-21 10:29:18 -0700743 }
744 return 0;
745}
746
747int TrafficController::toggleUidOwnerMap(ChildChain chain, bool enable) {
Chenbo Feng84d9f9f2019-04-12 15:31:02 -0700748 std::lock_guard guard(mMutex);
Chenbo Fengf434e862018-06-27 14:08:39 -0700749 uint32_t key = UID_RULES_CONFIGURATION_KEY;
Chenbo Feng703798e2018-06-15 17:07:59 -0700750 auto oldConfiguration = mConfigurationMap.readValue(key);
751 if (!isOk(oldConfiguration)) {
752 ALOGE("Cannot read the old configuration from map: %s",
753 oldConfiguration.status().msg().c_str());
754 return -oldConfiguration.status().code();
755 }
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700756 Status res;
Chenbo Feng703798e2018-06-15 17:07:59 -0700757 BpfConfig newConfiguration;
758 uint8_t match;
Chenbo Feng89c12f12018-03-21 10:29:18 -0700759 switch (chain) {
760 case DOZABLE:
Chenbo Feng703798e2018-06-15 17:07:59 -0700761 match = DOZABLE_MATCH;
Chenbo Feng89c12f12018-03-21 10:29:18 -0700762 break;
763 case STANDBY:
Chenbo Feng703798e2018-06-15 17:07:59 -0700764 match = STANDBY_MATCH;
Chenbo Feng89c12f12018-03-21 10:29:18 -0700765 break;
766 case POWERSAVE:
Chenbo Feng703798e2018-06-15 17:07:59 -0700767 match = POWERSAVE_MATCH;
Chenbo Feng89c12f12018-03-21 10:29:18 -0700768 break;
769 default:
770 return -EINVAL;
771 }
Chenbo Feng703798e2018-06-15 17:07:59 -0700772 newConfiguration =
773 enable ? (oldConfiguration.value() | match) : (oldConfiguration.value() & (~match));
774 res = mConfigurationMap.writeValue(key, newConfiguration, BPF_EXIST);
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700775 if (!isOk(res)) {
776 ALOGE("Failed to toggleUidOwnerMap(%d): %s", chain, res.msg().c_str());
Chenbo Feng89c12f12018-03-21 10:29:18 -0700777 }
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700778 return -res.code();
Chenbo Feng89c12f12018-03-21 10:29:18 -0700779}
780
Chenbo Feng47dd0732018-12-11 12:23:24 -0800781BpfLevel TrafficController::getBpfLevel() {
782 return mBpfLevel;
Chenbo Feng07d43fe2017-12-21 14:38:51 -0800783}
784
Chenbo Feng873ae142019-04-10 12:26:06 -0700785Status TrafficController::swapActiveStatsMap() {
Chenbo Feng84d9f9f2019-04-12 15:31:02 -0700786 std::lock_guard guard(mMutex);
Chenbo Feng873ae142019-04-10 12:26:06 -0700787
788 if (mBpfLevel == BpfLevel::NONE) {
789 return statusFromErrno(EOPNOTSUPP, "This device doesn't have eBPF support");
790 }
791
792 uint32_t key = CURRENT_STATS_MAP_CONFIGURATION_KEY;
793 auto oldConfiguration = mConfigurationMap.readValue(key);
794 if (!isOk(oldConfiguration)) {
795 ALOGE("Cannot read the old configuration from map: %s",
796 oldConfiguration.status().msg().c_str());
797 return oldConfiguration.status();
798 }
799
800 // Write to the configuration map to inform the kernel eBPF program to switch
801 // from using one map to the other. Use flag BPF_EXIST here since the map should
802 // be already populated in initMaps.
803 uint8_t newConfigure = (oldConfiguration.value() == SELECT_MAP_A) ? SELECT_MAP_B : SELECT_MAP_A;
804 Status res = mConfigurationMap.writeValue(CURRENT_STATS_MAP_CONFIGURATION_KEY, newConfigure,
805 BPF_EXIST);
806 if (!isOk(res)) {
807 ALOGE("Failed to toggle the stats map: %s", strerror(res.code()));
808 return res;
809 }
810 // After changing the config, we need to make sure all the current running
811 // eBPF programs are finished and all the CPUs are aware of this config change
812 // before we modify the old map. So we do a special hack here to wait for
813 // the kernel to do a synchronize_rcu(). Once the kernel called
814 // synchronize_rcu(), the config we just updated will be available to all cores
815 // and the next eBPF programs triggered inside the kernel will use the new
816 // map configuration. So once this function returns we can safely modify the
817 // old stats map without concerning about race between the kernel and
818 // userspace.
819 int ret = synchronizeKernelRCU();
820 if (ret) {
821 ALOGE("map swap synchronize_rcu() ended with failure: %s", strerror(-ret));
822 return statusFromErrno(-ret, "map swap synchronize_rcu() failed");
823 }
824 return netdutils::status::ok;
825}
826
Chenbo Feng48eaed32018-12-26 17:40:21 -0800827void TrafficController::setPermissionForUids(int permission, const std::vector<uid_t>& uids) {
Chenbo Feng84d9f9f2019-04-12 15:31:02 -0700828 std::lock_guard guard(mMutex);
Chenbo Fengbf660aa2019-02-26 16:12:27 -0800829 if (permission == INetd::PERMISSION_UNINSTALLED) {
830 for (uid_t uid : uids) {
831 // Clean up all permission information for the related uid if all the
832 // packages related to it are uninstalled.
833 mPrivilegedUser.erase(uid);
834 Status ret = mUidPermissionMap.deleteValue(uid);
835 }
836 return;
837 }
838
Chenbo Feng48eaed32018-12-26 17:40:21 -0800839 bool privileged = (permission & INetd::PERMISSION_UPDATE_DEVICE_STATS);
840
841 for (uid_t uid : uids) {
Chenbo Feng527908a2019-04-02 20:14:11 -0700842 // The map stores all the permissions that the UID has, except if the only permission
843 // the UID has is the INTERNET permission, then the UID should not appear in the map.
844 if (permission != INetd::PERMISSION_INTERNET) {
Chenbo Fengbf660aa2019-02-26 16:12:27 -0800845 Status ret = mUidPermissionMap.writeValue(uid, permission, BPF_ANY);
Chenbo Feng48eaed32018-12-26 17:40:21 -0800846 if (!isOk(ret)) {
Chenbo Feng527908a2019-04-02 20:14:11 -0700847 ALOGE("Failed to set permission: %s of uid(%u) to permission map: %s",
848 UidPermissionTypeToString(permission).c_str(), uid, strerror(ret.code()));
Chenbo Feng48eaed32018-12-26 17:40:21 -0800849 }
850 } else {
851 Status ret = mUidPermissionMap.deleteValue(uid);
852 if (!isOk(ret) && ret.code() != ENOENT) {
Chenbo Feng527908a2019-04-02 20:14:11 -0700853 ALOGE("Failed to remove uid %u from permission map: %s", uid, strerror(ret.code()));
Chenbo Feng48eaed32018-12-26 17:40:21 -0800854 }
855 }
856
857 if (privileged) {
858 mPrivilegedUser.insert(uid);
859 } else {
860 mPrivilegedUser.erase(uid);
861 }
862 }
863}
864
Chenbo Fengef297172018-03-26 10:53:33 -0700865std::string getProgramStatus(const char *path) {
866 int ret = access(path, R_OK);
867 if (ret == 0) {
868 return StringPrintf("OK");
869 }
870 if (ret != 0 && errno == ENOENT) {
871 return StringPrintf("program is missing at: %s", path);
872 }
873 return StringPrintf("check Program %s error: %s", path, strerror(errno));
874}
875
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700876std::string getMapStatus(const base::unique_fd& map_fd, const char* path) {
Chenbo Fengef297172018-03-26 10:53:33 -0700877 if (map_fd.get() < 0) {
878 return StringPrintf("map fd lost");
879 }
880 if (access(path, F_OK) != 0) {
881 return StringPrintf("map not pinned to location: %s", path);
882 }
883 return StringPrintf("OK");
884}
885
Bernie Innocentia5161a02019-01-30 22:40:53 +0900886// NOLINTNEXTLINE(google-runtime-references): grandfathered pass by non-const reference
Bernie Innocenti15bb55c2018-06-03 16:19:51 +0900887void dumpBpfMap(const std::string& mapName, DumpWriter& dw, const std::string& header) {
Chenbo Fengef297172018-03-26 10:53:33 -0700888 dw.blankline();
889 dw.println("%s:", mapName.c_str());
Erik Klineb31fd692018-06-06 20:50:11 +0900890 if (!header.empty()) {
891 dw.println(header);
Chenbo Fengef297172018-03-26 10:53:33 -0700892 }
893}
894
895const String16 TrafficController::DUMP_KEYWORD = String16("trafficcontroller");
896
897void TrafficController::dump(DumpWriter& dw, bool verbose) {
Chenbo Feng84d9f9f2019-04-12 15:31:02 -0700898 std::lock_guard guard(mMutex);
Erik Klineb31fd692018-06-06 20:50:11 +0900899 ScopedIndent indentTop(dw);
Chenbo Fengef297172018-03-26 10:53:33 -0700900 dw.println("TrafficController");
901
Erik Klineb31fd692018-06-06 20:50:11 +0900902 ScopedIndent indentPreBpfModule(dw);
Chenbo Feng47dd0732018-12-11 12:23:24 -0800903 dw.println("BPF module status: %s", BpfLevelToString(mBpfLevel).c_str());
Chenbo Fengef297172018-03-26 10:53:33 -0700904
Chenbo Feng47dd0732018-12-11 12:23:24 -0800905 if (mBpfLevel == BpfLevel::NONE) {
Chenbo Fengef297172018-03-26 10:53:33 -0700906 return;
Erik Klineb31fd692018-06-06 20:50:11 +0900907 }
Chenbo Fengef297172018-03-26 10:53:33 -0700908
909 dw.blankline();
910 dw.println("mCookieTagMap status: %s",
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700911 getMapStatus(mCookieTagMap.getMap(), COOKIE_TAG_MAP_PATH).c_str());
Chenbo Fengef297172018-03-26 10:53:33 -0700912 dw.println("mUidCounterSetMap status: %s",
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700913 getMapStatus(mUidCounterSetMap.getMap(), UID_COUNTERSET_MAP_PATH).c_str());
Chenbo Fengbc4a15f2018-05-11 19:15:15 -0700914 dw.println("mAppUidStatsMap status: %s",
915 getMapStatus(mAppUidStatsMap.getMap(), APP_UID_STATS_MAP_PATH).c_str());
Chenbo Fengf434e862018-06-27 14:08:39 -0700916 dw.println("mStatsMapA status: %s",
917 getMapStatus(mStatsMapA.getMap(), STATS_MAP_A_PATH).c_str());
918 dw.println("mStatsMapB status: %s",
919 getMapStatus(mStatsMapB.getMap(), STATS_MAP_B_PATH).c_str());
Chenbo Fengef297172018-03-26 10:53:33 -0700920 dw.println("mIfaceIndexNameMap status: %s",
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700921 getMapStatus(mIfaceIndexNameMap.getMap(), IFACE_INDEX_NAME_MAP_PATH).c_str());
Chenbo Fengef297172018-03-26 10:53:33 -0700922 dw.println("mIfaceStatsMap status: %s",
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700923 getMapStatus(mIfaceStatsMap.getMap(), IFACE_STATS_MAP_PATH).c_str());
Chenbo Feng703798e2018-06-15 17:07:59 -0700924 dw.println("mConfigurationMap status: %s",
925 getMapStatus(mConfigurationMap.getMap(), CONFIGURATION_MAP_PATH).c_str());
926 dw.println("mUidOwnerMap status: %s",
927 getMapStatus(mUidOwnerMap.getMap(), UID_OWNER_MAP_PATH).c_str());
Chenbo Fengef297172018-03-26 10:53:33 -0700928
929 dw.blankline();
930 dw.println("Cgroup ingress program status: %s",
931 getProgramStatus(BPF_INGRESS_PROG_PATH).c_str());
932 dw.println("Cgroup egress program status: %s", getProgramStatus(BPF_EGRESS_PROG_PATH).c_str());
933 dw.println("xt_bpf ingress program status: %s",
934 getProgramStatus(XT_BPF_INGRESS_PROG_PATH).c_str());
935 dw.println("xt_bpf egress program status: %s",
936 getProgramStatus(XT_BPF_EGRESS_PROG_PATH).c_str());
Chenbo Feng95892f32018-06-07 14:52:02 -0700937 dw.println("xt_bpf bandwidth whitelist program status: %s",
938 getProgramStatus(XT_BPF_WHITELIST_PROG_PATH).c_str());
939 dw.println("xt_bpf bandwidth blacklist program status: %s",
940 getProgramStatus(XT_BPF_BLACKLIST_PROG_PATH).c_str());
Chenbo Fengef297172018-03-26 10:53:33 -0700941
Erik Klineb31fd692018-06-06 20:50:11 +0900942 if (!verbose) {
943 return;
944 }
Chenbo Fengef297172018-03-26 10:53:33 -0700945
946 dw.blankline();
947 dw.println("BPF map content:");
948
Erik Klineb31fd692018-06-06 20:50:11 +0900949 ScopedIndent indentForMapContent(dw);
Chenbo Fengef297172018-03-26 10:53:33 -0700950
951 // Print CookieTagMap content.
952 dumpBpfMap("mCookieTagMap", dw, "");
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700953 const auto printCookieTagInfo = [&dw](const uint64_t& key, const UidTag& value,
954 const BpfMap<uint64_t, UidTag>&) {
955 dw.println("cookie=%" PRIu64 " tag=0x%x uid=%u", key, value.tag, value.uid);
956 return netdutils::status::ok;
Chenbo Fengef297172018-03-26 10:53:33 -0700957 };
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700958 Status res = mCookieTagMap.iterateWithValue(printCookieTagInfo);
959 if (!isOk(res)) {
960 dw.println("mCookieTagMap print end with error: %s", res.msg().c_str());
Chenbo Fengef297172018-03-26 10:53:33 -0700961 }
962
963 // Print UidCounterSetMap Content
964 dumpBpfMap("mUidCounterSetMap", dw, "");
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700965 const auto printUidInfo = [&dw](const uint32_t& key, const uint8_t& value,
966 const BpfMap<uint32_t, uint8_t>&) {
967 dw.println("%u %u", key, value);
968 return netdutils::status::ok;
Chenbo Fengef297172018-03-26 10:53:33 -0700969 };
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700970 res = mUidCounterSetMap.iterateWithValue(printUidInfo);
971 if (!isOk(res)) {
972 dw.println("mUidCounterSetMap print end with error: %s", res.msg().c_str());
Chenbo Fengef297172018-03-26 10:53:33 -0700973 }
974
Chenbo Fengbc4a15f2018-05-11 19:15:15 -0700975 // Print AppUidStatsMap content
976 std::string appUidStatsHeader = StringPrintf("uid rxBytes rxPackets txBytes txPackets");
977 dumpBpfMap("mAppUidStatsMap:", dw, appUidStatsHeader);
978 auto printAppUidStatsInfo = [&dw](const uint32_t& key, const StatsValue& value,
979 const BpfMap<uint32_t, StatsValue>&) {
980 dw.println("%u %" PRIu64 " %" PRIu64 " %" PRIu64 " %" PRIu64, key, value.rxBytes,
981 value.rxPackets, value.txBytes, value.txPackets);
982 return netdutils::status::ok;
983 };
984 res = mAppUidStatsMap.iterateWithValue(printAppUidStatsInfo);
985 if (!res.ok()) {
986 dw.println("mAppUidStatsMap print end with error: %s", res.msg().c_str());
987 }
988
Chenbo Fengef297172018-03-26 10:53:33 -0700989 // Print uidStatsMap content
990 std::string statsHeader = StringPrintf("ifaceIndex ifaceName tag_hex uid_int cnt_set rxBytes"
991 " rxPackets txBytes txPackets");
Chenbo Fengf434e862018-06-27 14:08:39 -0700992 dumpBpfMap("mStatsMapA", dw, statsHeader);
Chenbo Feng4f6c2372018-04-26 10:37:55 -0700993 const auto printStatsInfo = [&dw, this](const StatsKey& key, const StatsValue& value,
994 const BpfMap<StatsKey, StatsValue>&) {
995 uint32_t ifIndex = key.ifaceIndex;
996 auto ifname = mIfaceIndexNameMap.readValue(ifIndex);
997 if (!isOk(ifname)) {
998 strlcpy(ifname.value().name, "unknown", sizeof(IfaceValue));
Chenbo Fengef297172018-03-26 10:53:33 -0700999 }
Chenbo Feng4f6c2372018-04-26 10:37:55 -07001000 dw.println("%u %s 0x%x %u %u %" PRIu64 " %" PRIu64 " %" PRIu64 " %" PRIu64, ifIndex,
1001 ifname.value().name, key.tag, key.uid, key.counterSet, value.rxBytes,
1002 value.rxPackets, value.txBytes, value.txPackets);
1003 return netdutils::status::ok;
Chenbo Fengef297172018-03-26 10:53:33 -07001004 };
Chenbo Fengf434e862018-06-27 14:08:39 -07001005 res = mStatsMapA.iterateWithValue(printStatsInfo);
Chenbo Feng4f6c2372018-04-26 10:37:55 -07001006 if (!isOk(res)) {
Chenbo Fengf434e862018-06-27 14:08:39 -07001007 dw.println("mStatsMapA print end with error: %s", res.msg().c_str());
Chenbo Fengef297172018-03-26 10:53:33 -07001008 }
1009
1010 // Print TagStatsMap content.
Chenbo Fengf434e862018-06-27 14:08:39 -07001011 dumpBpfMap("mStatsMapB", dw, statsHeader);
1012 res = mStatsMapB.iterateWithValue(printStatsInfo);
Chenbo Feng4f6c2372018-04-26 10:37:55 -07001013 if (!isOk(res)) {
Chenbo Fengf434e862018-06-27 14:08:39 -07001014 dw.println("mStatsMapB print end with error: %s", res.msg().c_str());
Chenbo Fengef297172018-03-26 10:53:33 -07001015 }
1016
1017 // Print ifaceIndexToNameMap content.
1018 dumpBpfMap("mIfaceIndexNameMap", dw, "");
Chenbo Feng4f6c2372018-04-26 10:37:55 -07001019 const auto printIfaceNameInfo = [&dw](const uint32_t& key, const IfaceValue& value,
1020 const BpfMap<uint32_t, IfaceValue>&) {
1021 const char* ifname = value.name;
1022 dw.println("ifaceIndex=%u ifaceName=%s", key, ifname);
1023 return netdutils::status::ok;
Chenbo Fengef297172018-03-26 10:53:33 -07001024 };
Chenbo Feng4f6c2372018-04-26 10:37:55 -07001025 res = mIfaceIndexNameMap.iterateWithValue(printIfaceNameInfo);
1026 if (!isOk(res)) {
1027 dw.println("mIfaceIndexNameMap print end with error: %s", res.msg().c_str());
Chenbo Fengef297172018-03-26 10:53:33 -07001028 }
1029
1030 // Print ifaceStatsMap content
1031 std::string ifaceStatsHeader = StringPrintf("ifaceIndex ifaceName rxBytes rxPackets txBytes"
1032 " txPackets");
1033 dumpBpfMap("mIfaceStatsMap:", dw, ifaceStatsHeader);
Chenbo Feng4f6c2372018-04-26 10:37:55 -07001034 const auto printIfaceStatsInfo = [&dw, this](const uint32_t& key, const StatsValue& value,
1035 const BpfMap<uint32_t, StatsValue>&) {
1036 auto ifname = mIfaceIndexNameMap.readValue(key);
1037 if (!isOk(ifname)) {
1038 strlcpy(ifname.value().name, "unknown", sizeof(IfaceValue));
Chenbo Fengef297172018-03-26 10:53:33 -07001039 }
Chenbo Feng4f6c2372018-04-26 10:37:55 -07001040 dw.println("%u %s %" PRIu64 " %" PRIu64 " %" PRIu64 " %" PRIu64, key, ifname.value().name,
1041 value.rxBytes, value.rxPackets, value.txBytes, value.txPackets);
1042 return netdutils::status::ok;
Chenbo Fengef297172018-03-26 10:53:33 -07001043 };
Chenbo Feng4f6c2372018-04-26 10:37:55 -07001044 res = mIfaceStatsMap.iterateWithValue(printIfaceStatsInfo);
1045 if (!isOk(res)) {
1046 dw.println("mIfaceStatsMap print end with error: %s", res.msg().c_str());
Chenbo Fengef297172018-03-26 10:53:33 -07001047 }
1048
Chenbo Feng703798e2018-06-15 17:07:59 -07001049 dw.blankline();
Chenbo Fengef297172018-03-26 10:53:33 -07001050
Chenbo Fengf434e862018-06-27 14:08:39 -07001051 uint32_t key = UID_RULES_CONFIGURATION_KEY;
Chenbo Feng703798e2018-06-15 17:07:59 -07001052 auto configuration = mConfigurationMap.readValue(key);
1053 if (isOk(configuration)) {
Chenbo Fengf434e862018-06-27 14:08:39 -07001054 dw.println("current ownerMatch configuration: %d", configuration.value());
Chenbo Feng703798e2018-06-15 17:07:59 -07001055 } else {
Chenbo Fengf434e862018-06-27 14:08:39 -07001056 dw.println("mConfigurationMap read ownerMatch configure failed with error: %s",
1057 configuration.status().msg().c_str());
1058 }
1059 key = CURRENT_STATS_MAP_CONFIGURATION_KEY;
1060 configuration = mConfigurationMap.readValue(key);
1061 if (isOk(configuration)) {
1062 dw.println("current statsMap configuration: %d", configuration.value());
1063 } else {
1064 dw.println("mConfigurationMap read stats map configure failed with error: %s",
1065 configuration.status().msg().c_str());
Chenbo Fengef297172018-03-26 10:53:33 -07001066 }
Chenbo Feng703798e2018-06-15 17:07:59 -07001067 dumpBpfMap("mUidOwnerMap", dw, "");
Rubin Xuec27ff22019-01-08 21:33:03 +00001068 const auto printUidMatchInfo = [&dw, this](const uint32_t& key, const UidOwnerValue& value,
1069 const BpfMap<uint32_t, UidOwnerValue>&) {
1070 if (value.rule & IIF_MATCH) {
1071 auto ifname = mIfaceIndexNameMap.readValue(value.iif);
1072 if (isOk(ifname)) {
1073 dw.println("%u %s %s", key, uidMatchTypeToString(value.rule).c_str(),
1074 ifname.value().name);
1075 } else {
1076 dw.println("%u %s %u", key, uidMatchTypeToString(value.rule).c_str(), value.iif);
1077 }
1078 } else {
1079 dw.println("%u %s", key, uidMatchTypeToString(value.rule).c_str());
1080 }
Chenbo Feng703798e2018-06-15 17:07:59 -07001081 return netdutils::status::ok;
1082 };
1083 res = mUidOwnerMap.iterateWithValue(printUidMatchInfo);
Chenbo Feng4f6c2372018-04-26 10:37:55 -07001084 if (!isOk(res)) {
Chenbo Feng703798e2018-06-15 17:07:59 -07001085 dw.println("mUidOwnerMap print end with error: %s", res.msg().c_str());
Chenbo Feng95892f32018-06-07 14:52:02 -07001086 }
Chenbo Feng48eaed32018-12-26 17:40:21 -08001087 dumpBpfMap("mUidPermissionMap", dw, "");
1088 const auto printUidPermissionInfo = [&dw](const uint32_t& key, const uint8_t& value,
1089 const BpfMap<uint32_t, uint8_t>&) {
1090 dw.println("%u %s", key, UidPermissionTypeToString(value).c_str());
1091 return netdutils::status::ok;
1092 };
1093 res = mUidPermissionMap.iterateWithValue(printUidPermissionInfo);
1094 if (!isOk(res)) {
1095 dw.println("mUidPermissionMap print end with error: %s", res.msg().c_str());
1096 }
1097
1098 dumpBpfMap("mPrivilegedUser", dw, "");
1099 for (uid_t uid : mPrivilegedUser) {
1100 dw.println("%u ALLOW_UPDATE_DEVICE_STATS", (uint32_t)uid);
1101 }
Chenbo Fengef297172018-03-26 10:53:33 -07001102}
1103
Chenbo Fengf2759682017-10-10 17:31:57 -07001104} // namespace net
1105} // namespace android