Chenbo Feng | f275968 | 2017-10-10 17:31:57 -0700 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (C) 2017 The Android Open Source Project |
| 3 | * |
| 4 | * Licensed under the Apache License, Version 2.0 (the "License"); |
| 5 | * you may not use this file except in compliance with the License. |
| 6 | * You may obtain a copy of the License at |
| 7 | * |
| 8 | * http://www.apache.org/licenses/LICENSE-2.0 |
| 9 | * |
| 10 | * Unless required by applicable law or agreed to in writing, software |
| 11 | * distributed under the License is distributed on an "AS IS" BASIS, |
| 12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 13 | * See the License for the specific language governing permissions and |
| 14 | * limitations under the License. |
| 15 | */ |
| 16 | |
Chenbo Feng | c10a8a4 | 2017-12-15 13:56:33 -0800 | [diff] [blame] | 17 | #define LOG_TAG "TrafficController" |
Chenbo Feng | 33cc103 | 2017-10-23 15:16:37 -0700 | [diff] [blame] | 18 | #include <inttypes.h> |
Chenbo Feng | f275968 | 2017-10-10 17:31:57 -0700 | [diff] [blame] | 19 | #include <linux/bpf.h> |
| 20 | #include <linux/if_ether.h> |
| 21 | #include <linux/in.h> |
| 22 | #include <linux/inet_diag.h> |
Chenbo Feng | 116d055 | 2017-12-04 17:25:19 -0800 | [diff] [blame] | 23 | #include <linux/netlink.h> |
| 24 | #include <linux/sock_diag.h> |
Chenbo Feng | f275968 | 2017-10-10 17:31:57 -0700 | [diff] [blame] | 25 | #include <linux/unistd.h> |
Chenbo Feng | c10a8a4 | 2017-12-15 13:56:33 -0800 | [diff] [blame] | 26 | #include <net/if.h> |
Chenbo Feng | f275968 | 2017-10-10 17:31:57 -0700 | [diff] [blame] | 27 | #include <stdlib.h> |
| 28 | #include <string.h> |
| 29 | #include <sys/socket.h> |
Chenbo Feng | c10a8a4 | 2017-12-15 13:56:33 -0800 | [diff] [blame] | 30 | #include <sys/stat.h> |
Chenbo Feng | 05393d8 | 2018-01-09 15:18:43 -0800 | [diff] [blame] | 31 | #include <sys/types.h> |
Chenbo Feng | 33cc103 | 2017-10-23 15:16:37 -0700 | [diff] [blame] | 32 | #include <sys/utsname.h> |
Chenbo Feng | 05393d8 | 2018-01-09 15:18:43 -0800 | [diff] [blame] | 33 | #include <sys/wait.h> |
Chenbo Feng | 89c12f1 | 2018-03-21 10:29:18 -0700 | [diff] [blame] | 34 | #include <mutex> |
Chenbo Feng | c10a8a4 | 2017-12-15 13:56:33 -0800 | [diff] [blame] | 35 | #include <unordered_set> |
| 36 | #include <vector> |
| 37 | |
| 38 | #include <android-base/stringprintf.h> |
Chenbo Feng | ef29717 | 2018-03-26 10:53:33 -0700 | [diff] [blame] | 39 | #include <android-base/strings.h> |
Chenbo Feng | c10a8a4 | 2017-12-15 13:56:33 -0800 | [diff] [blame] | 40 | #include <android-base/unique_fd.h> |
Chenbo Feng | 05393d8 | 2018-01-09 15:18:43 -0800 | [diff] [blame] | 41 | #include <logwrap/logwrap.h> |
Chenbo Feng | c10a8a4 | 2017-12-15 13:56:33 -0800 | [diff] [blame] | 42 | #include <netdutils/StatusOr.h> |
Chenbo Feng | f275968 | 2017-10-10 17:31:57 -0700 | [diff] [blame] | 43 | |
Chenbo Feng | 116d055 | 2017-12-04 17:25:19 -0800 | [diff] [blame] | 44 | #include <netdutils/Misc.h> |
| 45 | #include <netdutils/Syscalls.h> |
Chenbo Feng | f275968 | 2017-10-10 17:31:57 -0700 | [diff] [blame] | 46 | #include "TrafficController.h" |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 47 | #include "bpf/BpfMap.h" |
Chenbo Feng | 89c12f1 | 2018-03-21 10:29:18 -0700 | [diff] [blame] | 48 | #include "bpf/bpf_shared.h" |
Chenbo Feng | 116d055 | 2017-12-04 17:25:19 -0800 | [diff] [blame] | 49 | |
Chenbo Feng | ef29717 | 2018-03-26 10:53:33 -0700 | [diff] [blame] | 50 | #include "DumpWriter.h" |
Chenbo Feng | 89c12f1 | 2018-03-21 10:29:18 -0700 | [diff] [blame] | 51 | #include "FirewallController.h" |
Chenbo Feng | 7e97405 | 2018-02-28 22:57:21 -0800 | [diff] [blame] | 52 | #include "InterfaceController.h" |
Chenbo Feng | 116d055 | 2017-12-04 17:25:19 -0800 | [diff] [blame] | 53 | #include "NetlinkListener.h" |
Chenbo Feng | 33cc103 | 2017-10-23 15:16:37 -0700 | [diff] [blame] | 54 | #include "qtaguid/qtaguid.h" |
Chenbo Feng | f275968 | 2017-10-10 17:31:57 -0700 | [diff] [blame] | 55 | |
Chenbo Feng | c10a8a4 | 2017-12-15 13:56:33 -0800 | [diff] [blame] | 56 | using namespace android::bpf; |
Chenbo Feng | f275968 | 2017-10-10 17:31:57 -0700 | [diff] [blame] | 57 | |
| 58 | namespace android { |
| 59 | namespace net { |
| 60 | |
Chenbo Feng | c10a8a4 | 2017-12-15 13:56:33 -0800 | [diff] [blame] | 61 | using base::StringPrintf; |
| 62 | using base::unique_fd; |
Chenbo Feng | ef29717 | 2018-03-26 10:53:33 -0700 | [diff] [blame] | 63 | using base::Join; |
Chenbo Feng | 116d055 | 2017-12-04 17:25:19 -0800 | [diff] [blame] | 64 | using netdutils::extract; |
| 65 | using netdutils::Slice; |
| 66 | using netdutils::sSyscalls; |
Chenbo Feng | c10a8a4 | 2017-12-15 13:56:33 -0800 | [diff] [blame] | 67 | using netdutils::Status; |
| 68 | using netdutils::statusFromErrno; |
| 69 | using netdutils::StatusOr; |
Chenbo Feng | 116d055 | 2017-12-04 17:25:19 -0800 | [diff] [blame] | 70 | using netdutils::status::ok; |
Chenbo Feng | c10a8a4 | 2017-12-15 13:56:33 -0800 | [diff] [blame] | 71 | |
Chenbo Feng | 116d055 | 2017-12-04 17:25:19 -0800 | [diff] [blame] | 72 | constexpr int kSockDiagMsgType = SOCK_DIAG_BY_FAMILY; |
| 73 | constexpr int kSockDiagDoneMsgType = NLMSG_DONE; |
| 74 | |
| 75 | StatusOr<std::unique_ptr<NetlinkListenerInterface>> makeSkDestroyListener() { |
| 76 | const auto& sys = sSyscalls.get(); |
| 77 | ASSIGN_OR_RETURN(auto event, sys.eventfd(0, EFD_CLOEXEC)); |
| 78 | const int domain = AF_NETLINK; |
| 79 | const int type = SOCK_DGRAM | SOCK_CLOEXEC | SOCK_NONBLOCK; |
| 80 | const int protocol = NETLINK_INET_DIAG; |
| 81 | ASSIGN_OR_RETURN(auto sock, sys.socket(domain, type, protocol)); |
| 82 | |
| 83 | sockaddr_nl addr = { |
| 84 | .nl_family = AF_NETLINK, |
| 85 | .nl_groups = 1 << (SKNLGRP_INET_TCP_DESTROY - 1) | 1 << (SKNLGRP_INET_UDP_DESTROY - 1) | |
| 86 | 1 << (SKNLGRP_INET6_TCP_DESTROY - 1) | 1 << (SKNLGRP_INET6_UDP_DESTROY - 1)}; |
| 87 | RETURN_IF_NOT_OK(sys.bind(sock, addr)); |
| 88 | |
| 89 | const sockaddr_nl kernel = {.nl_family = AF_NETLINK}; |
| 90 | RETURN_IF_NOT_OK(sys.connect(sock, kernel)); |
| 91 | |
| 92 | std::unique_ptr<NetlinkListenerInterface> listener = |
| 93 | std::make_unique<NetlinkListener>(std::move(event), std::move(sock)); |
| 94 | |
| 95 | return listener; |
| 96 | } |
| 97 | |
Chenbo Feng | 89c12f1 | 2018-03-21 10:29:18 -0700 | [diff] [blame] | 98 | Status changeOwnerAndMode(const char* path, gid_t group, const char* debugName, bool netdOnly) { |
| 99 | int ret = chown(path, AID_ROOT, group); |
| 100 | if (ret != 0) return statusFromErrno(errno, StringPrintf("change %s group failed", debugName)); |
| 101 | |
| 102 | if (netdOnly) { |
| 103 | ret = chmod(path, S_IRWXU); |
| 104 | } else { |
| 105 | // Allow both netd and system server to obtain map fd from the path. |
| 106 | // chmod doesn't grant permission to all processes in that group to |
| 107 | // read/write the bpf map. They still need correct sepolicy to |
| 108 | // read/write the map. |
| 109 | ret = chmod(path, S_IRWXU | S_IRGRP); |
| 110 | } |
| 111 | if (ret != 0) return statusFromErrno(errno, StringPrintf("change %s mode failed", debugName)); |
| 112 | return netdutils::status::ok; |
| 113 | } |
| 114 | |
| 115 | TrafficController::TrafficController() { |
Chenbo Feng | f43bf81 | 2017-12-15 18:27:22 -0800 | [diff] [blame] | 116 | ebpfSupported = hasBpfSupport(); |
Chenbo Feng | 89c12f1 | 2018-03-21 10:29:18 -0700 | [diff] [blame] | 117 | } |
| 118 | |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 119 | Status initialOwnerMap(BpfMap<uint32_t, uint8_t>& map) { |
| 120 | // Check and delete all the entries from the map in case it is a runtime |
| 121 | // restart |
| 122 | const auto deleteAllEntries = [](const uint32_t& key, BpfMap<uint32_t, uint8_t>& map) { |
| 123 | Status res = map.deleteValue(key); |
| 124 | if (!isOk(res) && (res.code() == ENOENT)) { |
| 125 | ALOGE("Failed to delete data(uid=%u): %s\n", key, strerror(res.code())); |
| 126 | } |
| 127 | return netdutils::status::ok; |
| 128 | }; |
| 129 | // It is safe to delete from this map because nothing will concurrently iterate over it: |
| 130 | // - Nothing in netd will iterate over it because we're holding mOwnerMatchMutex. |
| 131 | // - Nothing outside netd iterates over it. |
| 132 | map.iterate(deleteAllEntries); |
Chenbo Feng | 89c12f1 | 2018-03-21 10:29:18 -0700 | [diff] [blame] | 133 | uint32_t mapSettingKey = UID_MAP_ENABLED; |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 134 | uint8_t defaultMapState = 0; |
| 135 | return map.writeValue(mapSettingKey, defaultMapState, BPF_NOEXIST); |
Chenbo Feng | 89c12f1 | 2018-03-21 10:29:18 -0700 | [diff] [blame] | 136 | } |
| 137 | |
| 138 | Status TrafficController::initMaps() { |
Chenbo Feng | ef1cab3 | 2018-04-13 19:50:49 -0700 | [diff] [blame] | 139 | std::lock_guard<std::mutex> ownerMapGuard(mOwnerMatchMutex); |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 140 | RETURN_IF_NOT_OK( |
| 141 | mCookieTagMap.getOrCreate(COOKIE_UID_MAP_SIZE, COOKIE_TAG_MAP_PATH, BPF_MAP_TYPE_HASH)); |
Chenbo Feng | 89c12f1 | 2018-03-21 10:29:18 -0700 | [diff] [blame] | 142 | |
| 143 | RETURN_IF_NOT_OK(changeOwnerAndMode(COOKIE_TAG_MAP_PATH, AID_NET_BW_ACCT, "CookieTagMap", |
| 144 | false)); |
| 145 | |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 146 | RETURN_IF_NOT_OK(mUidCounterSetMap.getOrCreate(UID_COUNTERSET_MAP_SIZE, UID_COUNTERSET_MAP_PATH, |
| 147 | BPF_MAP_TYPE_HASH)); |
Chenbo Feng | 89c12f1 | 2018-03-21 10:29:18 -0700 | [diff] [blame] | 148 | RETURN_IF_NOT_OK(changeOwnerAndMode(UID_COUNTERSET_MAP_PATH, AID_NET_BW_ACCT, |
| 149 | "UidCounterSetMap", false)); |
| 150 | |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 151 | RETURN_IF_NOT_OK( |
| 152 | mUidStatsMap.getOrCreate(UID_STATS_MAP_SIZE, UID_STATS_MAP_PATH, BPF_MAP_TYPE_HASH)); |
Chenbo Feng | 89c12f1 | 2018-03-21 10:29:18 -0700 | [diff] [blame] | 153 | RETURN_IF_NOT_OK(changeOwnerAndMode(UID_STATS_MAP_PATH, AID_NET_BW_STATS, "UidStatsMap", |
| 154 | false)); |
| 155 | |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 156 | RETURN_IF_NOT_OK( |
| 157 | mTagStatsMap.getOrCreate(TAG_STATS_MAP_SIZE, TAG_STATS_MAP_PATH, BPF_MAP_TYPE_HASH)); |
Chenbo Feng | 89c12f1 | 2018-03-21 10:29:18 -0700 | [diff] [blame] | 158 | RETURN_IF_NOT_OK(changeOwnerAndMode(TAG_STATS_MAP_PATH, AID_NET_BW_STATS, "TagStatsMap", |
| 159 | false)); |
| 160 | |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 161 | RETURN_IF_NOT_OK(mIfaceIndexNameMap.getOrCreate(IFACE_INDEX_NAME_MAP_SIZE, |
| 162 | IFACE_INDEX_NAME_MAP_PATH, BPF_MAP_TYPE_HASH)); |
Chenbo Feng | 89c12f1 | 2018-03-21 10:29:18 -0700 | [diff] [blame] | 163 | RETURN_IF_NOT_OK(changeOwnerAndMode(IFACE_INDEX_NAME_MAP_PATH, AID_NET_BW_STATS, |
| 164 | "IfaceIndexNameMap", false)); |
| 165 | |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 166 | RETURN_IF_NOT_OK( |
| 167 | mDozableUidMap.getOrCreate(UID_OWNER_MAP_SIZE, DOZABLE_UID_MAP_PATH, BPF_MAP_TYPE_HASH)); |
Chenbo Feng | 89c12f1 | 2018-03-21 10:29:18 -0700 | [diff] [blame] | 168 | RETURN_IF_NOT_OK(changeOwnerAndMode(DOZABLE_UID_MAP_PATH, AID_ROOT, "DozableUidMap", true)); |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 169 | RETURN_IF_NOT_OK(initialOwnerMap(mDozableUidMap)); |
Chenbo Feng | 89c12f1 | 2018-03-21 10:29:18 -0700 | [diff] [blame] | 170 | |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 171 | RETURN_IF_NOT_OK( |
| 172 | mStandbyUidMap.getOrCreate(UID_OWNER_MAP_SIZE, STANDBY_UID_MAP_PATH, BPF_MAP_TYPE_HASH)); |
Chenbo Feng | 89c12f1 | 2018-03-21 10:29:18 -0700 | [diff] [blame] | 173 | RETURN_IF_NOT_OK(changeOwnerAndMode(STANDBY_UID_MAP_PATH, AID_ROOT, "StandbyUidMap", true)); |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 174 | RETURN_IF_NOT_OK(initialOwnerMap(mStandbyUidMap)); |
Chenbo Feng | 89c12f1 | 2018-03-21 10:29:18 -0700 | [diff] [blame] | 175 | |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 176 | RETURN_IF_NOT_OK(mPowerSaveUidMap.getOrCreate(UID_OWNER_MAP_SIZE, POWERSAVE_UID_MAP_PATH, |
| 177 | BPF_MAP_TYPE_HASH)); |
Chenbo Feng | 89c12f1 | 2018-03-21 10:29:18 -0700 | [diff] [blame] | 178 | RETURN_IF_NOT_OK(changeOwnerAndMode(POWERSAVE_UID_MAP_PATH, AID_ROOT, "PowerSaveUidMap", true)); |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 179 | RETURN_IF_NOT_OK(initialOwnerMap(mPowerSaveUidMap)); |
Chenbo Feng | 89c12f1 | 2018-03-21 10:29:18 -0700 | [diff] [blame] | 180 | |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 181 | RETURN_IF_NOT_OK( |
| 182 | mIfaceStatsMap.getOrCreate(IFACE_STATS_MAP_SIZE, IFACE_STATS_MAP_PATH, BPF_MAP_TYPE_HASH)); |
Chenbo Feng | 89c12f1 | 2018-03-21 10:29:18 -0700 | [diff] [blame] | 183 | RETURN_IF_NOT_OK(changeOwnerAndMode(IFACE_STATS_MAP_PATH, AID_NET_BW_STATS, "IfaceStatsMap", |
| 184 | false)); |
| 185 | return netdutils::status::ok; |
| 186 | } |
| 187 | |
| 188 | Status TrafficController::start() { |
| 189 | |
Chenbo Feng | f43bf81 | 2017-12-15 18:27:22 -0800 | [diff] [blame] | 190 | if (!ebpfSupported) { |
Chenbo Feng | c10a8a4 | 2017-12-15 13:56:33 -0800 | [diff] [blame] | 191 | return netdutils::status::ok; |
Chenbo Feng | f275968 | 2017-10-10 17:31:57 -0700 | [diff] [blame] | 192 | } |
| 193 | |
| 194 | /* When netd restart from a crash without total system reboot, the program |
| 195 | * is still attached to the cgroup, detach it so the program can be freed |
| 196 | * and we can load and attach new program into the target cgroup. |
| 197 | * |
| 198 | * TODO: Scrape existing socket when run-time restart and clean up the map |
| 199 | * if the socket no longer exist |
| 200 | */ |
| 201 | |
Chenbo Feng | 89c12f1 | 2018-03-21 10:29:18 -0700 | [diff] [blame] | 202 | RETURN_IF_NOT_OK(initMaps()); |
Chenbo Feng | 7e97405 | 2018-02-28 22:57:21 -0800 | [diff] [blame] | 203 | |
| 204 | // Fetch the list of currently-existing interfaces. At this point NetlinkHandler is |
| 205 | // already running, so it will call addInterface() when any new interface appears. |
| 206 | std::map<std::string, uint32_t> ifacePairs; |
| 207 | ASSIGN_OR_RETURN(ifacePairs, InterfaceController::getIfaceList()); |
| 208 | for (const auto& ifacePair:ifacePairs) { |
| 209 | addInterface(ifacePair.first.c_str(), ifacePair.second); |
| 210 | } |
| 211 | |
Chenbo Feng | 5ed1799 | 2018-03-13 21:30:49 -0700 | [diff] [blame] | 212 | |
Chenbo Feng | 116d055 | 2017-12-04 17:25:19 -0800 | [diff] [blame] | 213 | auto result = makeSkDestroyListener(); |
| 214 | if (!isOk(result)) { |
| 215 | ALOGE("Unable to create SkDestroyListener: %s", toString(result).c_str()); |
| 216 | } else { |
| 217 | mSkDestroyListener = std::move(result.value()); |
| 218 | } |
| 219 | // Rx handler extracts nfgenmsg looks up and invokes registered dispatch function. |
| 220 | const auto rxHandler = [this](const nlmsghdr&, const Slice msg) { |
| 221 | inet_diag_msg diagmsg = {}; |
| 222 | if (extract(msg, diagmsg) < sizeof(inet_diag_msg)) { |
| 223 | ALOGE("unrecognized netlink message: %s", toString(msg).c_str()); |
| 224 | return; |
| 225 | } |
| 226 | uint64_t sock_cookie = static_cast<uint64_t>(diagmsg.id.idiag_cookie[0]) | |
| 227 | (static_cast<uint64_t>(diagmsg.id.idiag_cookie[1]) << 32); |
| 228 | |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 229 | mCookieTagMap.deleteValue(sock_cookie); |
Chenbo Feng | 116d055 | 2017-12-04 17:25:19 -0800 | [diff] [blame] | 230 | }; |
| 231 | expectOk(mSkDestroyListener->subscribe(kSockDiagMsgType, rxHandler)); |
| 232 | |
| 233 | // In case multiple netlink message comes in as a stream, we need to handle the rxDone message |
| 234 | // properly. |
| 235 | const auto rxDoneHandler = [](const nlmsghdr&, const Slice msg) { |
| 236 | // Ignore NLMSG_DONE messages |
| 237 | inet_diag_msg diagmsg = {}; |
| 238 | extract(msg, diagmsg); |
| 239 | }; |
| 240 | expectOk(mSkDestroyListener->subscribe(kSockDiagDoneMsgType, rxDoneHandler)); |
| 241 | |
Chenbo Feng | 05393d8 | 2018-01-09 15:18:43 -0800 | [diff] [blame] | 242 | int* status = nullptr; |
| 243 | |
| 244 | std::vector<const char*> prog_args{ |
| 245 | "/system/bin/bpfloader", |
| 246 | }; |
Chenbo Feng | 89c12f1 | 2018-03-21 10:29:18 -0700 | [diff] [blame] | 247 | int ret = access(BPF_INGRESS_PROG_PATH, R_OK); |
Chenbo Feng | 05393d8 | 2018-01-09 15:18:43 -0800 | [diff] [blame] | 248 | if (ret != 0 && errno == ENOENT) { |
| 249 | prog_args.push_back((char*)"-i"); |
| 250 | } |
| 251 | ret = access(BPF_EGRESS_PROG_PATH, R_OK); |
| 252 | if (ret != 0 && errno == ENOENT) { |
| 253 | prog_args.push_back((char*)"-e"); |
| 254 | } |
Chenbo Feng | 5ed1799 | 2018-03-13 21:30:49 -0700 | [diff] [blame] | 255 | ret = access(XT_BPF_INGRESS_PROG_PATH, R_OK); |
| 256 | if (ret != 0 && errno == ENOENT) { |
| 257 | prog_args.push_back((char*)"-p"); |
| 258 | } |
| 259 | ret = access(XT_BPF_EGRESS_PROG_PATH, R_OK); |
| 260 | if (ret != 0 && errno == ENOENT) { |
| 261 | prog_args.push_back((char*)"-m"); |
| 262 | } |
Chenbo Feng | 05393d8 | 2018-01-09 15:18:43 -0800 | [diff] [blame] | 263 | |
| 264 | if (prog_args.size() == 1) { |
Chenbo Feng | 5ed1799 | 2018-03-13 21:30:49 -0700 | [diff] [blame] | 265 | // all program are loaded already. |
Chenbo Feng | 05393d8 | 2018-01-09 15:18:43 -0800 | [diff] [blame] | 266 | return netdutils::status::ok; |
| 267 | } |
| 268 | |
| 269 | prog_args.push_back(nullptr); |
| 270 | ret = android_fork_execvp(prog_args.size(), (char**)prog_args.data(), status, false, true); |
| 271 | if (ret) { |
| 272 | ret = errno; |
| 273 | ALOGE("failed to execute %s: %s", prog_args[0], strerror(errno)); |
| 274 | return statusFromErrno(ret, "run bpf loader failed"); |
| 275 | } |
| 276 | return netdutils::status::ok; |
Chenbo Feng | f275968 | 2017-10-10 17:31:57 -0700 | [diff] [blame] | 277 | } |
| 278 | |
Chenbo Feng | f275968 | 2017-10-10 17:31:57 -0700 | [diff] [blame] | 279 | int TrafficController::tagSocket(int sockFd, uint32_t tag, uid_t uid) { |
Chenbo Feng | bc6d470 | 2018-04-11 18:27:19 -0700 | [diff] [blame] | 280 | if (!ebpfSupported) { |
| 281 | if (legacy_tagSocket(sockFd, tag, uid)) return -errno; |
| 282 | return 0; |
| 283 | } |
Chenbo Feng | 33cc103 | 2017-10-23 15:16:37 -0700 | [diff] [blame] | 284 | |
Chenbo Feng | f275968 | 2017-10-10 17:31:57 -0700 | [diff] [blame] | 285 | uint64_t sock_cookie = getSocketCookie(sockFd); |
Chenbo Feng | ef1cab3 | 2018-04-13 19:50:49 -0700 | [diff] [blame] | 286 | if (sock_cookie == NONEXISTENT_COOKIE) return -errno; |
Chenbo Feng | f275968 | 2017-10-10 17:31:57 -0700 | [diff] [blame] | 287 | UidTag newKey = {.uid = (uint32_t)uid, .tag = tag}; |
| 288 | |
| 289 | // Update the tag information of a socket to the cookieUidMap. Use BPF_ANY |
| 290 | // flag so it will insert a new entry to the map if that value doesn't exist |
| 291 | // yet. And update the tag if there is already a tag stored. Since the eBPF |
| 292 | // program in kernel only read this map, and is protected by rcu read lock. It |
| 293 | // should be fine to cocurrently update the map while eBPF program is running. |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 294 | Status res = mCookieTagMap.writeValue(sock_cookie, newKey, BPF_ANY); |
| 295 | if (!isOk(res)) { |
| 296 | ALOGE("Failed to tag the socket: %s, fd: %d", strerror(res.code()), |
| 297 | mCookieTagMap.getMap().get()); |
Chenbo Feng | f275968 | 2017-10-10 17:31:57 -0700 | [diff] [blame] | 298 | } |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 299 | return -res.code(); |
Chenbo Feng | f275968 | 2017-10-10 17:31:57 -0700 | [diff] [blame] | 300 | } |
| 301 | |
| 302 | int TrafficController::untagSocket(int sockFd) { |
Chenbo Feng | bc6d470 | 2018-04-11 18:27:19 -0700 | [diff] [blame] | 303 | if (!ebpfSupported) { |
| 304 | if (legacy_untagSocket(sockFd)) return -errno; |
| 305 | return 0; |
| 306 | } |
Chenbo Feng | f275968 | 2017-10-10 17:31:57 -0700 | [diff] [blame] | 307 | uint64_t sock_cookie = getSocketCookie(sockFd); |
| 308 | |
Chenbo Feng | ef1cab3 | 2018-04-13 19:50:49 -0700 | [diff] [blame] | 309 | if (sock_cookie == NONEXISTENT_COOKIE) return -errno; |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 310 | Status res = mCookieTagMap.deleteValue(sock_cookie); |
| 311 | if (!isOk(res)) { |
| 312 | ALOGE("Failed to untag socket: %s\n", strerror(res.code())); |
Chenbo Feng | f275968 | 2017-10-10 17:31:57 -0700 | [diff] [blame] | 313 | } |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 314 | return -res.code(); |
Chenbo Feng | f275968 | 2017-10-10 17:31:57 -0700 | [diff] [blame] | 315 | } |
| 316 | |
| 317 | int TrafficController::setCounterSet(int counterSetNum, uid_t uid) { |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 318 | if (counterSetNum < 0 || counterSetNum >= OVERFLOW_COUNTERSET) return -EINVAL; |
| 319 | Status res; |
Chenbo Feng | bc6d470 | 2018-04-11 18:27:19 -0700 | [diff] [blame] | 320 | if (!ebpfSupported) { |
| 321 | if (legacy_setCounterSet(counterSetNum, uid)) return -errno; |
| 322 | return 0; |
| 323 | } |
Chenbo Feng | c10a8a4 | 2017-12-15 13:56:33 -0800 | [diff] [blame] | 324 | |
| 325 | // The default counter set for all uid is 0, so deleting the current counterset for that uid |
| 326 | // will automatically set it to 0. |
Chenbo Feng | f275968 | 2017-10-10 17:31:57 -0700 | [diff] [blame] | 327 | if (counterSetNum == 0) { |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 328 | Status res = mUidCounterSetMap.deleteValue(uid); |
| 329 | if (isOk(res) || (!isOk(res) && res.code() == ENOENT)) { |
Chenbo Feng | f275968 | 2017-10-10 17:31:57 -0700 | [diff] [blame] | 330 | return 0; |
| 331 | } else { |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 332 | ALOGE("Failed to delete the counterSet: %s\n", strerror(res.code())); |
| 333 | return -res.code(); |
Chenbo Feng | f275968 | 2017-10-10 17:31:57 -0700 | [diff] [blame] | 334 | } |
| 335 | } |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 336 | uint8_t tmpCounterSetNum = (uint8_t)counterSetNum; |
| 337 | res = mUidCounterSetMap.writeValue(uid, tmpCounterSetNum, BPF_ANY); |
| 338 | if (!isOk(res)) { |
| 339 | ALOGE("Failed to set the counterSet: %s, fd: %d", strerror(res.code()), |
| 340 | mUidCounterSetMap.getMap().get()); |
| 341 | return -res.code(); |
Chenbo Feng | f275968 | 2017-10-10 17:31:57 -0700 | [diff] [blame] | 342 | } |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 343 | return 0; |
Chenbo Feng | f275968 | 2017-10-10 17:31:57 -0700 | [diff] [blame] | 344 | } |
| 345 | |
| 346 | int TrafficController::deleteTagData(uint32_t tag, uid_t uid) { |
Chenbo Feng | 33cc103 | 2017-10-23 15:16:37 -0700 | [diff] [blame] | 347 | |
Chenbo Feng | bc6d470 | 2018-04-11 18:27:19 -0700 | [diff] [blame] | 348 | if (!ebpfSupported) { |
| 349 | if (legacy_deleteTagData(tag, uid)) return -errno; |
| 350 | return 0; |
| 351 | } |
Chenbo Feng | 33cc103 | 2017-10-23 15:16:37 -0700 | [diff] [blame] | 352 | |
Chenbo Feng | c10a8a4 | 2017-12-15 13:56:33 -0800 | [diff] [blame] | 353 | // First we go through the cookieTagMap to delete the target uid tag combination. Or delete all |
Chenbo Feng | ef1cab3 | 2018-04-13 19:50:49 -0700 | [diff] [blame] | 354 | // the tags related to the uid if the tag is 0. |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 355 | const auto deleteMatchedCookieEntries = [uid, tag](const uint64_t& key, const UidTag& value, |
| 356 | BpfMap<uint64_t, UidTag>& map) { |
| 357 | if (value.uid == uid && (value.tag == tag || tag == 0)) { |
| 358 | Status res = map.deleteValue(key); |
| 359 | if (isOk(res) || (res.code() == ENOENT)) { |
| 360 | return netdutils::status::ok; |
Chenbo Feng | f275968 | 2017-10-10 17:31:57 -0700 | [diff] [blame] | 361 | } |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 362 | ALOGE("Failed to delete data(cookie = %" PRIu64 "): %s\n", key, strerror(res.code())); |
Chenbo Feng | f275968 | 2017-10-10 17:31:57 -0700 | [diff] [blame] | 363 | } |
Chenbo Feng | ef1cab3 | 2018-04-13 19:50:49 -0700 | [diff] [blame] | 364 | // Move forward to next cookie in the map. |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 365 | return netdutils::status::ok; |
Chenbo Feng | ef1cab3 | 2018-04-13 19:50:49 -0700 | [diff] [blame] | 366 | }; |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 367 | mCookieTagMap.iterateWithValue(deleteMatchedCookieEntries); |
Chenbo Feng | f275968 | 2017-10-10 17:31:57 -0700 | [diff] [blame] | 368 | // Now we go through the Tag stats map and delete the data entry with correct uid and tag |
Chenbo Feng | ef1cab3 | 2018-04-13 19:50:49 -0700 | [diff] [blame] | 369 | // combination. Or all tag stats under that uid if the target tag is 0. |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 370 | const auto deleteMatchedUidTagEntries = [uid, tag](const StatsKey& key, |
| 371 | BpfMap<StatsKey, StatsValue>& map) { |
| 372 | if (key.uid == uid && (key.tag == tag || tag == 0)) { |
| 373 | Status res = map.deleteValue(key); |
| 374 | if (isOk(res) || (res.code() == ENOENT)) { |
Chenbo Feng | ef1cab3 | 2018-04-13 19:50:49 -0700 | [diff] [blame] | 375 | //Entry is deleted, use the current key to get a new nextKey; |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 376 | return netdutils::status::ok; |
Chenbo Feng | f275968 | 2017-10-10 17:31:57 -0700 | [diff] [blame] | 377 | } |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 378 | ALOGE("Failed to delete data(uid=%u, tag=%u): %s\n", key.uid, key.tag, |
| 379 | strerror(res.code())); |
Chenbo Feng | f275968 | 2017-10-10 17:31:57 -0700 | [diff] [blame] | 380 | } |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 381 | return netdutils::status::ok; |
Chenbo Feng | ef1cab3 | 2018-04-13 19:50:49 -0700 | [diff] [blame] | 382 | }; |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 383 | mTagStatsMap.iterate(deleteMatchedUidTagEntries); |
Chenbo Feng | f275968 | 2017-10-10 17:31:57 -0700 | [diff] [blame] | 384 | // If the tag is not zero, we already deleted all the data entry required. If tag is 0, we also |
Chenbo Feng | ed37fea | 2017-12-13 19:35:01 -0800 | [diff] [blame] | 385 | // need to delete the stats stored in uidStatsMap and counterSet map. |
Chenbo Feng | c10a8a4 | 2017-12-15 13:56:33 -0800 | [diff] [blame] | 386 | if (tag != 0) return 0; |
| 387 | |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 388 | Status res = mUidCounterSetMap.deleteValue(uid); |
| 389 | if (!isOk(res) && res.code() != ENOENT) { |
| 390 | ALOGE("Failed to delete counterSet data(uid=%u, tag=%u): %s\n", uid, tag, |
| 391 | strerror(res.code())); |
Chenbo Feng | c10a8a4 | 2017-12-15 13:56:33 -0800 | [diff] [blame] | 392 | } |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 393 | mUidStatsMap.iterate(deleteMatchedUidTagEntries); |
| 394 | return 0; |
Chenbo Feng | f275968 | 2017-10-10 17:31:57 -0700 | [diff] [blame] | 395 | } |
| 396 | |
Chenbo Feng | 7e97405 | 2018-02-28 22:57:21 -0800 | [diff] [blame] | 397 | int TrafficController::addInterface(const char* name, uint32_t ifaceIndex) { |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 398 | if (!ebpfSupported) return 0; |
Chenbo Feng | 7e97405 | 2018-02-28 22:57:21 -0800 | [diff] [blame] | 399 | |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 400 | IfaceValue iface; |
Chenbo Feng | 7e97405 | 2018-02-28 22:57:21 -0800 | [diff] [blame] | 401 | if (ifaceIndex == 0) { |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 402 | ALOGE("Unknown interface %s(%d)", name, ifaceIndex); |
Chenbo Feng | 7e97405 | 2018-02-28 22:57:21 -0800 | [diff] [blame] | 403 | return -1; |
| 404 | } |
| 405 | |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 406 | strlcpy(iface.name, name, sizeof(IfaceValue)); |
| 407 | Status res = mIfaceIndexNameMap.writeValue(ifaceIndex, iface, BPF_ANY); |
| 408 | if (!isOk(res)) { |
| 409 | ALOGE("Failed to add iface %s(%d): %s", name, ifaceIndex, strerror(res.code())); |
| 410 | return -res.code(); |
Chenbo Feng | 7e97405 | 2018-02-28 22:57:21 -0800 | [diff] [blame] | 411 | } |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 412 | return 0; |
Chenbo Feng | 7e97405 | 2018-02-28 22:57:21 -0800 | [diff] [blame] | 413 | } |
| 414 | |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 415 | Status TrafficController::updateOwnerMapEntry(BpfMap<uint32_t, uint8_t>& map, uid_t uid, |
| 416 | FirewallRule rule, FirewallType type) { |
Chenbo Feng | 89c12f1 | 2018-03-21 10:29:18 -0700 | [diff] [blame] | 417 | if (uid == UID_MAP_ENABLED) { |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 418 | return statusFromErrno(-EINVAL, "This uid is reserved for map state"); |
Chenbo Feng | 89c12f1 | 2018-03-21 10:29:18 -0700 | [diff] [blame] | 419 | } |
| 420 | |
| 421 | if ((rule == ALLOW && type == WHITELIST) || (rule == DENY && type == BLACKLIST)) { |
| 422 | uint8_t flag = (type == WHITELIST) ? BPF_PASS : BPF_DROP; |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 423 | RETURN_IF_NOT_OK(map.writeValue(uid, flag, BPF_ANY)); |
Chenbo Feng | 89c12f1 | 2018-03-21 10:29:18 -0700 | [diff] [blame] | 424 | } else if ((rule == ALLOW && type == BLACKLIST) || (rule == DENY && type == WHITELIST)) { |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 425 | RETURN_IF_NOT_OK(map.deleteValue(uid)); |
Chenbo Feng | 89c12f1 | 2018-03-21 10:29:18 -0700 | [diff] [blame] | 426 | } else { |
| 427 | //Cannot happen. |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 428 | return statusFromErrno(-EINVAL, ""); |
Chenbo Feng | 89c12f1 | 2018-03-21 10:29:18 -0700 | [diff] [blame] | 429 | } |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 430 | return netdutils::status::ok; |
Chenbo Feng | 89c12f1 | 2018-03-21 10:29:18 -0700 | [diff] [blame] | 431 | } |
| 432 | |
| 433 | int TrafficController::changeUidOwnerRule(ChildChain chain, uid_t uid, FirewallRule rule, |
| 434 | FirewallType type) { |
| 435 | std::lock_guard<std::mutex> guard(mOwnerMatchMutex); |
| 436 | if (!ebpfSupported) { |
| 437 | ALOGE("bpf is not set up, should use iptables rule"); |
| 438 | return -ENOSYS; |
| 439 | } |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 440 | Status res; |
Chenbo Feng | 89c12f1 | 2018-03-21 10:29:18 -0700 | [diff] [blame] | 441 | switch (chain) { |
| 442 | case DOZABLE: |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 443 | res = updateOwnerMapEntry(mDozableUidMap, uid, rule, type); |
| 444 | break; |
Chenbo Feng | 89c12f1 | 2018-03-21 10:29:18 -0700 | [diff] [blame] | 445 | case STANDBY: |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 446 | res = updateOwnerMapEntry(mStandbyUidMap, uid, rule, type); |
| 447 | break; |
Chenbo Feng | 89c12f1 | 2018-03-21 10:29:18 -0700 | [diff] [blame] | 448 | case POWERSAVE: |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 449 | res = updateOwnerMapEntry(mPowerSaveUidMap, uid, rule, type); |
| 450 | break; |
Chenbo Feng | 89c12f1 | 2018-03-21 10:29:18 -0700 | [diff] [blame] | 451 | case NONE: |
| 452 | default: |
| 453 | return -EINVAL; |
| 454 | } |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 455 | if (!isOk(res)) { |
| 456 | ALOGE("change uid(%u) rule of %d failed: %s, rule: %d, type: %d", uid, chain, |
| 457 | res.msg().c_str(), rule, type); |
| 458 | return -res.code(); |
| 459 | } |
| 460 | return 0; |
Chenbo Feng | 89c12f1 | 2018-03-21 10:29:18 -0700 | [diff] [blame] | 461 | } |
| 462 | |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 463 | Status TrafficController::replaceUidsInMap(BpfMap<uint32_t, uint8_t>& map, |
| 464 | const std::vector<int32_t>& uids, FirewallRule rule, |
| 465 | FirewallType type) { |
Chenbo Feng | 89c12f1 | 2018-03-21 10:29:18 -0700 | [diff] [blame] | 466 | std::set<int32_t> uidSet(uids.begin(), uids.end()); |
| 467 | std::vector<uint32_t> uidsToDelete; |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 468 | auto getUidsToDelete = [&uidsToDelete, &uidSet](const uint32_t& key, |
| 469 | const BpfMap<uint32_t, uint8_t>&) { |
| 470 | if (key != UID_MAP_ENABLED && uidSet.find((int32_t)key) == uidSet.end()) { |
| 471 | uidsToDelete.push_back(key); |
Chenbo Feng | 89c12f1 | 2018-03-21 10:29:18 -0700 | [diff] [blame] | 472 | } |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 473 | return netdutils::status::ok; |
Chenbo Feng | 89c12f1 | 2018-03-21 10:29:18 -0700 | [diff] [blame] | 474 | }; |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 475 | RETURN_IF_NOT_OK(map.iterate(getUidsToDelete)); |
Chenbo Feng | 89c12f1 | 2018-03-21 10:29:18 -0700 | [diff] [blame] | 476 | |
| 477 | for(auto uid : uidsToDelete) { |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 478 | RETURN_IF_NOT_OK(map.deleteValue(uid)); |
Chenbo Feng | 89c12f1 | 2018-03-21 10:29:18 -0700 | [diff] [blame] | 479 | } |
| 480 | |
| 481 | for (auto uid : uids) { |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 482 | RETURN_IF_NOT_OK(updateOwnerMapEntry(map, uid, rule, type)); |
Chenbo Feng | 89c12f1 | 2018-03-21 10:29:18 -0700 | [diff] [blame] | 483 | } |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 484 | return netdutils::status::ok; |
Chenbo Feng | 89c12f1 | 2018-03-21 10:29:18 -0700 | [diff] [blame] | 485 | } |
| 486 | |
| 487 | int TrafficController::replaceUidOwnerMap(const std::string& name, bool isWhitelist, |
| 488 | const std::vector<int32_t>& uids) { |
| 489 | std::lock_guard<std::mutex> guard(mOwnerMatchMutex); |
| 490 | FirewallRule rule; |
| 491 | FirewallType type; |
| 492 | if (isWhitelist) { |
| 493 | type = WHITELIST; |
| 494 | rule = ALLOW; |
| 495 | } else { |
| 496 | type = BLACKLIST; |
| 497 | rule = DENY; |
| 498 | } |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 499 | Status res; |
Chenbo Feng | 89c12f1 | 2018-03-21 10:29:18 -0700 | [diff] [blame] | 500 | if (!name.compare(FirewallController::LOCAL_DOZABLE)) { |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 501 | res = replaceUidsInMap(mDozableUidMap, uids, rule, type); |
Chenbo Feng | 89c12f1 | 2018-03-21 10:29:18 -0700 | [diff] [blame] | 502 | } else if (!name.compare(FirewallController::LOCAL_STANDBY)) { |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 503 | res = replaceUidsInMap(mStandbyUidMap, uids, rule, type); |
Chenbo Feng | 89c12f1 | 2018-03-21 10:29:18 -0700 | [diff] [blame] | 504 | } else if (!name.compare(FirewallController::LOCAL_POWERSAVE)) { |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 505 | res = replaceUidsInMap(mPowerSaveUidMap, uids, rule, type); |
Chenbo Feng | 89c12f1 | 2018-03-21 10:29:18 -0700 | [diff] [blame] | 506 | } else { |
| 507 | ALOGE("unknown chain name: %s", name.c_str()); |
| 508 | return -EINVAL; |
| 509 | } |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 510 | if (!isOk(res)) { |
| 511 | ALOGE("Failed to clean up chain: %s: %s", name.c_str(), res.msg().c_str()); |
| 512 | return -res.code(); |
Chenbo Feng | 89c12f1 | 2018-03-21 10:29:18 -0700 | [diff] [blame] | 513 | } |
| 514 | return 0; |
| 515 | } |
| 516 | |
| 517 | int TrafficController::toggleUidOwnerMap(ChildChain chain, bool enable) { |
| 518 | std::lock_guard<std::mutex> guard(mOwnerMatchMutex); |
| 519 | uint32_t keyUid = UID_MAP_ENABLED; |
| 520 | uint8_t mapState = enable ? 1 : 0; |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 521 | Status res; |
Chenbo Feng | 89c12f1 | 2018-03-21 10:29:18 -0700 | [diff] [blame] | 522 | switch (chain) { |
| 523 | case DOZABLE: |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 524 | res = mDozableUidMap.writeValue(keyUid, mapState, BPF_EXIST); |
Chenbo Feng | 89c12f1 | 2018-03-21 10:29:18 -0700 | [diff] [blame] | 525 | break; |
| 526 | case STANDBY: |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 527 | res = mStandbyUidMap.writeValue(keyUid, mapState, BPF_EXIST); |
Chenbo Feng | 89c12f1 | 2018-03-21 10:29:18 -0700 | [diff] [blame] | 528 | break; |
| 529 | case POWERSAVE: |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 530 | res = mPowerSaveUidMap.writeValue(keyUid, mapState, BPF_EXIST); |
Chenbo Feng | 89c12f1 | 2018-03-21 10:29:18 -0700 | [diff] [blame] | 531 | break; |
| 532 | default: |
| 533 | return -EINVAL; |
| 534 | } |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 535 | if (!isOk(res)) { |
| 536 | ALOGE("Failed to toggleUidOwnerMap(%d): %s", chain, res.msg().c_str()); |
Chenbo Feng | 89c12f1 | 2018-03-21 10:29:18 -0700 | [diff] [blame] | 537 | } |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 538 | return -res.code(); |
Chenbo Feng | 89c12f1 | 2018-03-21 10:29:18 -0700 | [diff] [blame] | 539 | } |
| 540 | |
Chenbo Feng | 07d43fe | 2017-12-21 14:38:51 -0800 | [diff] [blame] | 541 | bool TrafficController::checkBpfStatsEnable() { |
| 542 | return ebpfSupported; |
| 543 | } |
| 544 | |
Chenbo Feng | ef29717 | 2018-03-26 10:53:33 -0700 | [diff] [blame] | 545 | std::string getProgramStatus(const char *path) { |
| 546 | int ret = access(path, R_OK); |
| 547 | if (ret == 0) { |
| 548 | return StringPrintf("OK"); |
| 549 | } |
| 550 | if (ret != 0 && errno == ENOENT) { |
| 551 | return StringPrintf("program is missing at: %s", path); |
| 552 | } |
| 553 | return StringPrintf("check Program %s error: %s", path, strerror(errno)); |
| 554 | } |
| 555 | |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 556 | std::string getMapStatus(const base::unique_fd& map_fd, const char* path) { |
Chenbo Feng | ef29717 | 2018-03-26 10:53:33 -0700 | [diff] [blame] | 557 | if (map_fd.get() < 0) { |
| 558 | return StringPrintf("map fd lost"); |
| 559 | } |
| 560 | if (access(path, F_OK) != 0) { |
| 561 | return StringPrintf("map not pinned to location: %s", path); |
| 562 | } |
| 563 | return StringPrintf("OK"); |
| 564 | } |
| 565 | |
| 566 | void dumpBpfMap(std::string mapName, DumpWriter& dw, const std::string& header) { |
| 567 | dw.blankline(); |
| 568 | dw.println("%s:", mapName.c_str()); |
| 569 | if(!header.empty()) { |
| 570 | dw.println(header.c_str()); |
| 571 | } |
| 572 | } |
| 573 | |
| 574 | const String16 TrafficController::DUMP_KEYWORD = String16("trafficcontroller"); |
| 575 | |
| 576 | void TrafficController::dump(DumpWriter& dw, bool verbose) { |
Chenbo Feng | ef1cab3 | 2018-04-13 19:50:49 -0700 | [diff] [blame] | 577 | std::lock_guard<std::mutex> ownerMapGuard(mOwnerMatchMutex); |
Chenbo Feng | ef29717 | 2018-03-26 10:53:33 -0700 | [diff] [blame] | 578 | dw.incIndent(); |
| 579 | dw.println("TrafficController"); |
| 580 | |
| 581 | dw.incIndent(); |
| 582 | dw.println("BPF module status: %s", ebpfSupported? "ON" : "OFF"); |
| 583 | |
| 584 | if (!ebpfSupported) |
| 585 | return; |
| 586 | |
| 587 | dw.blankline(); |
| 588 | dw.println("mCookieTagMap status: %s", |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 589 | getMapStatus(mCookieTagMap.getMap(), COOKIE_TAG_MAP_PATH).c_str()); |
Chenbo Feng | ef29717 | 2018-03-26 10:53:33 -0700 | [diff] [blame] | 590 | dw.println("mUidCounterSetMap status: %s", |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 591 | getMapStatus(mUidCounterSetMap.getMap(), UID_COUNTERSET_MAP_PATH).c_str()); |
| 592 | dw.println("mUidStatsMap status: %s", |
| 593 | getMapStatus(mUidStatsMap.getMap(), UID_STATS_MAP_PATH).c_str()); |
| 594 | dw.println("mTagStatsMap status: %s", |
| 595 | getMapStatus(mTagStatsMap.getMap(), TAG_STATS_MAP_PATH).c_str()); |
Chenbo Feng | ef29717 | 2018-03-26 10:53:33 -0700 | [diff] [blame] | 596 | dw.println("mIfaceIndexNameMap status: %s", |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 597 | getMapStatus(mIfaceIndexNameMap.getMap(), IFACE_INDEX_NAME_MAP_PATH).c_str()); |
Chenbo Feng | ef29717 | 2018-03-26 10:53:33 -0700 | [diff] [blame] | 598 | dw.println("mIfaceStatsMap status: %s", |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 599 | getMapStatus(mIfaceStatsMap.getMap(), IFACE_STATS_MAP_PATH).c_str()); |
Chenbo Feng | ef29717 | 2018-03-26 10:53:33 -0700 | [diff] [blame] | 600 | dw.println("mDozableUidMap status: %s", |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 601 | getMapStatus(mDozableUidMap.getMap(), DOZABLE_UID_MAP_PATH).c_str()); |
Chenbo Feng | ef29717 | 2018-03-26 10:53:33 -0700 | [diff] [blame] | 602 | dw.println("mStandbyUidMap status: %s", |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 603 | getMapStatus(mStandbyUidMap.getMap(), STANDBY_UID_MAP_PATH).c_str()); |
Chenbo Feng | ef29717 | 2018-03-26 10:53:33 -0700 | [diff] [blame] | 604 | dw.println("mPowerSaveUidMap status: %s", |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 605 | getMapStatus(mPowerSaveUidMap.getMap(), POWERSAVE_UID_MAP_PATH).c_str()); |
Chenbo Feng | ef29717 | 2018-03-26 10:53:33 -0700 | [diff] [blame] | 606 | |
| 607 | dw.blankline(); |
| 608 | dw.println("Cgroup ingress program status: %s", |
| 609 | getProgramStatus(BPF_INGRESS_PROG_PATH).c_str()); |
| 610 | dw.println("Cgroup egress program status: %s", getProgramStatus(BPF_EGRESS_PROG_PATH).c_str()); |
| 611 | dw.println("xt_bpf ingress program status: %s", |
| 612 | getProgramStatus(XT_BPF_INGRESS_PROG_PATH).c_str()); |
| 613 | dw.println("xt_bpf egress program status: %s", |
| 614 | getProgramStatus(XT_BPF_EGRESS_PROG_PATH).c_str()); |
| 615 | |
| 616 | if(!verbose) return; |
| 617 | |
| 618 | dw.blankline(); |
| 619 | dw.println("BPF map content:"); |
| 620 | |
| 621 | dw.incIndent(); |
| 622 | |
| 623 | // Print CookieTagMap content. |
| 624 | dumpBpfMap("mCookieTagMap", dw, ""); |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 625 | const auto printCookieTagInfo = [&dw](const uint64_t& key, const UidTag& value, |
| 626 | const BpfMap<uint64_t, UidTag>&) { |
| 627 | dw.println("cookie=%" PRIu64 " tag=0x%x uid=%u", key, value.tag, value.uid); |
| 628 | return netdutils::status::ok; |
Chenbo Feng | ef29717 | 2018-03-26 10:53:33 -0700 | [diff] [blame] | 629 | }; |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 630 | Status res = mCookieTagMap.iterateWithValue(printCookieTagInfo); |
| 631 | if (!isOk(res)) { |
| 632 | dw.println("mCookieTagMap print end with error: %s", res.msg().c_str()); |
Chenbo Feng | ef29717 | 2018-03-26 10:53:33 -0700 | [diff] [blame] | 633 | } |
| 634 | |
| 635 | // Print UidCounterSetMap Content |
| 636 | dumpBpfMap("mUidCounterSetMap", dw, ""); |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 637 | const auto printUidInfo = [&dw](const uint32_t& key, const uint8_t& value, |
| 638 | const BpfMap<uint32_t, uint8_t>&) { |
| 639 | dw.println("%u %u", key, value); |
| 640 | return netdutils::status::ok; |
Chenbo Feng | ef29717 | 2018-03-26 10:53:33 -0700 | [diff] [blame] | 641 | }; |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 642 | res = mUidCounterSetMap.iterateWithValue(printUidInfo); |
| 643 | if (!isOk(res)) { |
| 644 | dw.println("mUidCounterSetMap print end with error: %s", res.msg().c_str()); |
Chenbo Feng | ef29717 | 2018-03-26 10:53:33 -0700 | [diff] [blame] | 645 | } |
| 646 | |
| 647 | // Print uidStatsMap content |
| 648 | std::string statsHeader = StringPrintf("ifaceIndex ifaceName tag_hex uid_int cnt_set rxBytes" |
| 649 | " rxPackets txBytes txPackets"); |
| 650 | dumpBpfMap("mUidStatsMap", dw, statsHeader); |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 651 | const auto printStatsInfo = [&dw, this](const StatsKey& key, const StatsValue& value, |
| 652 | const BpfMap<StatsKey, StatsValue>&) { |
| 653 | uint32_t ifIndex = key.ifaceIndex; |
| 654 | auto ifname = mIfaceIndexNameMap.readValue(ifIndex); |
| 655 | if (!isOk(ifname)) { |
| 656 | strlcpy(ifname.value().name, "unknown", sizeof(IfaceValue)); |
Chenbo Feng | ef29717 | 2018-03-26 10:53:33 -0700 | [diff] [blame] | 657 | } |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 658 | dw.println("%u %s 0x%x %u %u %" PRIu64 " %" PRIu64 " %" PRIu64 " %" PRIu64, ifIndex, |
| 659 | ifname.value().name, key.tag, key.uid, key.counterSet, value.rxBytes, |
| 660 | value.rxPackets, value.txBytes, value.txPackets); |
| 661 | return netdutils::status::ok; |
Chenbo Feng | ef29717 | 2018-03-26 10:53:33 -0700 | [diff] [blame] | 662 | }; |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 663 | res = mUidStatsMap.iterateWithValue(printStatsInfo); |
| 664 | if (!isOk(res)) { |
| 665 | dw.println("mUidStatsMap print end with error: %s", res.msg().c_str()); |
Chenbo Feng | ef29717 | 2018-03-26 10:53:33 -0700 | [diff] [blame] | 666 | } |
| 667 | |
| 668 | // Print TagStatsMap content. |
| 669 | dumpBpfMap("mTagStatsMap", dw, statsHeader); |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 670 | res = mTagStatsMap.iterateWithValue(printStatsInfo); |
| 671 | if (!isOk(res)) { |
| 672 | dw.println("mTagStatsMap print end with error: %s", res.msg().c_str()); |
Chenbo Feng | ef29717 | 2018-03-26 10:53:33 -0700 | [diff] [blame] | 673 | } |
| 674 | |
| 675 | // Print ifaceIndexToNameMap content. |
| 676 | dumpBpfMap("mIfaceIndexNameMap", dw, ""); |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 677 | const auto printIfaceNameInfo = [&dw](const uint32_t& key, const IfaceValue& value, |
| 678 | const BpfMap<uint32_t, IfaceValue>&) { |
| 679 | const char* ifname = value.name; |
| 680 | dw.println("ifaceIndex=%u ifaceName=%s", key, ifname); |
| 681 | return netdutils::status::ok; |
Chenbo Feng | ef29717 | 2018-03-26 10:53:33 -0700 | [diff] [blame] | 682 | }; |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 683 | res = mIfaceIndexNameMap.iterateWithValue(printIfaceNameInfo); |
| 684 | if (!isOk(res)) { |
| 685 | dw.println("mIfaceIndexNameMap print end with error: %s", res.msg().c_str()); |
Chenbo Feng | ef29717 | 2018-03-26 10:53:33 -0700 | [diff] [blame] | 686 | } |
| 687 | |
| 688 | // Print ifaceStatsMap content |
| 689 | std::string ifaceStatsHeader = StringPrintf("ifaceIndex ifaceName rxBytes rxPackets txBytes" |
| 690 | " txPackets"); |
| 691 | dumpBpfMap("mIfaceStatsMap:", dw, ifaceStatsHeader); |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 692 | const auto printIfaceStatsInfo = [&dw, this](const uint32_t& key, const StatsValue& value, |
| 693 | const BpfMap<uint32_t, StatsValue>&) { |
| 694 | auto ifname = mIfaceIndexNameMap.readValue(key); |
| 695 | if (!isOk(ifname)) { |
| 696 | strlcpy(ifname.value().name, "unknown", sizeof(IfaceValue)); |
Chenbo Feng | ef29717 | 2018-03-26 10:53:33 -0700 | [diff] [blame] | 697 | } |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 698 | dw.println("%u %s %" PRIu64 " %" PRIu64 " %" PRIu64 " %" PRIu64, key, ifname.value().name, |
| 699 | value.rxBytes, value.rxPackets, value.txBytes, value.txPackets); |
| 700 | return netdutils::status::ok; |
Chenbo Feng | ef29717 | 2018-03-26 10:53:33 -0700 | [diff] [blame] | 701 | }; |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 702 | res = mIfaceStatsMap.iterateWithValue(printIfaceStatsInfo); |
| 703 | if (!isOk(res)) { |
| 704 | dw.println("mIfaceStatsMap print end with error: %s", res.msg().c_str()); |
Chenbo Feng | ef29717 | 2018-03-26 10:53:33 -0700 | [diff] [blame] | 705 | } |
| 706 | |
| 707 | // Print owner match uid maps |
Chenbo Feng | ef29717 | 2018-03-26 10:53:33 -0700 | [diff] [blame] | 708 | dumpBpfMap("mDozableUidMap", dw, ""); |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 709 | res = mDozableUidMap.iterateWithValue(printUidInfo); |
| 710 | if (!isOk(res)) { |
| 711 | dw.println("mDozableUidMap print end with error: %s", res.msg().c_str()); |
Chenbo Feng | ef29717 | 2018-03-26 10:53:33 -0700 | [diff] [blame] | 712 | } |
| 713 | |
| 714 | dumpBpfMap("mStandbyUidMap", dw, ""); |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 715 | res = mStandbyUidMap.iterateWithValue(printUidInfo); |
| 716 | if (!isOk(res)) { |
| 717 | dw.println("mDozableUidMap print end with error: %s", res.msg().c_str()); |
Chenbo Feng | ef29717 | 2018-03-26 10:53:33 -0700 | [diff] [blame] | 718 | } |
| 719 | |
| 720 | dumpBpfMap("mPowerSaveUidMap", dw, ""); |
Chenbo Feng | 4f6c237 | 2018-04-26 10:37:55 -0700 | [diff] [blame^] | 721 | res = mPowerSaveUidMap.iterateWithValue(printUidInfo); |
| 722 | if (!isOk(res)) { |
| 723 | dw.println("mDozableUidMap print end with error: %s", res.msg().c_str()); |
Chenbo Feng | ef29717 | 2018-03-26 10:53:33 -0700 | [diff] [blame] | 724 | } |
| 725 | |
| 726 | dw.decIndent(); |
| 727 | |
| 728 | dw.decIndent(); |
| 729 | |
| 730 | dw.decIndent(); |
| 731 | |
| 732 | } |
| 733 | |
Chenbo Feng | f275968 | 2017-10-10 17:31:57 -0700 | [diff] [blame] | 734 | } // namespace net |
| 735 | } // namespace android |