blob: b0246f6b44c7aa1f1eed09d59b298fd349609497 [file] [log] [blame]
Ken Chen587d4232022-01-17 17:18:43 +08001/*
2 * Copyright (C) 2018 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
Maciej Żenczykowskif7699522022-05-24 15:56:03 -070017// The resulting .o needs to load on the Android T Beta 3 bpfloader
18#define BPFLOADER_MIN_VER BPFLOADER_T_BETA3_VERSION
Maciej Żenczykowskiacebffb2022-05-16 16:05:15 -070019
Ken Chen587d4232022-01-17 17:18:43 +080020#include <bpf_helpers.h>
21#include <linux/bpf.h>
22#include <linux/if.h>
23#include <linux/if_ether.h>
24#include <linux/if_packet.h>
25#include <linux/in.h>
26#include <linux/in6.h>
27#include <linux/ip.h>
28#include <linux/ipv6.h>
29#include <linux/pkt_cls.h>
30#include <linux/tcp.h>
Ken Chen587d4232022-01-17 17:18:43 +080031#include <stdbool.h>
32#include <stdint.h>
33#include "bpf_net_helpers.h"
Maciej Żenczykowski513474c2022-12-08 16:20:43 +000034#include "netd.h"
Ken Chen587d4232022-01-17 17:18:43 +080035
36// This is defined for cgroup bpf filter only.
Maciej Żenczykowski55474982022-11-20 13:48:39 +000037static const int DROP = 0;
38static const int PASS = 1;
39static const int DROP_UNLESS_DNS = 2; // internal to our program
Ken Chen587d4232022-01-17 17:18:43 +080040
41// This is used for xt_bpf program only.
Maciej Żenczykowski55474982022-11-20 13:48:39 +000042static const int BPF_NOMATCH = 0;
43static const int BPF_MATCH = 1;
Ken Chen587d4232022-01-17 17:18:43 +080044
Maciej Żenczykowski55474982022-11-20 13:48:39 +000045// Used for 'bool egress'
46static const bool INGRESS = false;
47static const bool EGRESS = true;
Ken Chen587d4232022-01-17 17:18:43 +080048
49#define IP_PROTO_OFF offsetof(struct iphdr, protocol)
50#define IPV6_PROTO_OFF offsetof(struct ipv6hdr, nexthdr)
Maciej Żenczykowski879839a12022-08-03 10:48:25 +000051
52// offsetof(struct iphdr, ihl) -- but that's a bitfield
Ken Chen587d4232022-01-17 17:18:43 +080053#define IPPROTO_IHL_OFF 0
Maciej Żenczykowski879839a12022-08-03 10:48:25 +000054
55// This is offsetof(struct tcphdr, "32 bit tcp flag field")
56// The tcp flags are after be16 source, dest & be32 seq, ack_seq, hence 12 bytes in.
57//
58// Note that TCP_FLAG_{ACK,PSH,RST,SYN,FIN} are htonl(0x00{10,08,04,02,01}0000)
59// see include/uapi/linux/tcp.h
60#define TCP_FLAG32_OFF 12
Ken Chen587d4232022-01-17 17:18:43 +080061
Maciej Żenczykowskia4a58a32022-06-13 17:56:06 -070062// For maps netd does not need to access
63#define DEFINE_BPF_MAP_NO_NETD(the_map, TYPE, TypeOfKey, TypeOfValue, num_entries) \
Maciej Żenczykowskicae181d2022-06-16 23:26:33 -070064 DEFINE_BPF_MAP_EXT(the_map, TYPE, TypeOfKey, TypeOfValue, num_entries, \
65 AID_ROOT, AID_NET_BW_ACCT, 0060, "fs_bpf_net_shared", "", false)
Maciej Żenczykowskia4a58a32022-06-13 17:56:06 -070066
67// For maps netd only needs read only access to
68#define DEFINE_BPF_MAP_RO_NETD(the_map, TYPE, TypeOfKey, TypeOfValue, num_entries) \
Maciej Żenczykowskicae181d2022-06-16 23:26:33 -070069 DEFINE_BPF_MAP_EXT(the_map, TYPE, TypeOfKey, TypeOfValue, num_entries, \
70 AID_ROOT, AID_NET_BW_ACCT, 0460, "fs_bpf_netd_readonly", "", false)
Maciej Żenczykowskia4a58a32022-06-13 17:56:06 -070071
72// For maps netd needs to be able to read and write
73#define DEFINE_BPF_MAP_RW_NETD(the_map, TYPE, TypeOfKey, TypeOfValue, num_entries) \
Maciej Żenczykowskicae181d2022-06-16 23:26:33 -070074 DEFINE_BPF_MAP_UGM(the_map, TYPE, TypeOfKey, TypeOfValue, num_entries, \
75 AID_ROOT, AID_NET_BW_ACCT, 0660)
Maciej Żenczykowskia4a58a32022-06-13 17:56:06 -070076
Maciej Żenczykowskib10e0552022-06-16 14:49:27 -070077// Bpf map arrays on creation are preinitialized to 0 and do not support deletion of a key,
78// see: kernel/bpf/arraymap.c array_map_delete_elem() returns -EINVAL (from both syscall and ebpf)
79// Additionally on newer kernels the bpf jit can optimize out the lookups.
80// only valid indexes are [0..CONFIGURATION_MAP_SIZE-1]
81DEFINE_BPF_MAP_RO_NETD(configuration_map, ARRAY, uint32_t, uint32_t, CONFIGURATION_MAP_SIZE)
82
Maciej Żenczykowski1b7c1f12022-11-21 09:39:23 +000083// TODO: consider whether we can merge some of these maps
84// for example it might be possible to merge 2 or 3 of:
85// uid_counterset_map + uid_owner_map + uid_permission_map
Maciej Żenczykowskia4a58a32022-06-13 17:56:06 -070086DEFINE_BPF_MAP_RW_NETD(cookie_tag_map, HASH, uint64_t, UidTagValue, COOKIE_UID_MAP_SIZE)
87DEFINE_BPF_MAP_NO_NETD(uid_counterset_map, HASH, uint32_t, uint8_t, UID_COUNTERSET_MAP_SIZE)
88DEFINE_BPF_MAP_NO_NETD(app_uid_stats_map, HASH, uint32_t, StatsValue, APP_STATS_MAP_SIZE)
89DEFINE_BPF_MAP_RW_NETD(stats_map_A, HASH, StatsKey, StatsValue, STATS_MAP_SIZE)
90DEFINE_BPF_MAP_RO_NETD(stats_map_B, HASH, StatsKey, StatsValue, STATS_MAP_SIZE)
91DEFINE_BPF_MAP_NO_NETD(iface_stats_map, HASH, uint32_t, StatsValue, IFACE_STATS_MAP_SIZE)
Maciej Żenczykowskia4a58a32022-06-13 17:56:06 -070092DEFINE_BPF_MAP_NO_NETD(uid_owner_map, HASH, uint32_t, UidOwnerValue, UID_OWNER_MAP_SIZE)
93DEFINE_BPF_MAP_RW_NETD(uid_permission_map, HASH, uint32_t, uint8_t, UID_OWNER_MAP_SIZE)
Ken Chen587d4232022-01-17 17:18:43 +080094
95/* never actually used from ebpf */
Maciej Żenczykowskia4a58a32022-06-13 17:56:06 -070096DEFINE_BPF_MAP_NO_NETD(iface_index_name_map, HASH, uint32_t, IfaceValue, IFACE_INDEX_NAME_MAP_SIZE)
Ken Chen587d4232022-01-17 17:18:43 +080097
Maciej Żenczykowskicae181d2022-06-16 23:26:33 -070098// iptables xt_bpf programs need to be usable by both netd and netutils_wrappers
Maciej Żenczykowski285f7052022-08-09 17:50:31 +000099// selinux contexts, because even non-xt_bpf iptables mutations are implemented as
Maciej Żenczykowski06085b02022-08-09 14:15:34 +0000100// a full table dump, followed by an update in userspace, and then a reload into the kernel,
101// where any already in-use xt_bpf matchers are serialized as the path to the pinned
102// program (see XT_BPF_MODE_PATH_PINNED) and then the iptables binary (or rather
103// the kernel acting on behalf of it) must be able to retrieve the pinned program
104// for the reload to succeed
Maciej Żenczykowskicae181d2022-06-16 23:26:33 -0700105#define DEFINE_XTBPF_PROG(SECTION_NAME, prog_uid, prog_gid, the_prog) \
106 DEFINE_BPF_PROG(SECTION_NAME, prog_uid, prog_gid, the_prog)
107
108// programs that need to be usable by netd, but not by netutils_wrappers
Maciej Żenczykowski06085b02022-08-09 14:15:34 +0000109// (this is because these are currently attached by the mainline provided libnetd_updatable .so
110// which is loaded into netd and thus runs as netd uid/gid/selinux context)
Maciej Żenczykowski879839a12022-08-03 10:48:25 +0000111#define DEFINE_NETD_BPF_PROG_KVER_RANGE(SECTION_NAME, prog_uid, prog_gid, the_prog, minKV, maxKV) \
Maciej Żenczykowskicae181d2022-06-16 23:26:33 -0700112 DEFINE_BPF_PROG_EXT(SECTION_NAME, prog_uid, prog_gid, the_prog, \
Maciej Żenczykowski879839a12022-08-03 10:48:25 +0000113 minKV, maxKV, false, "fs_bpf_netd_readonly", "")
Maciej Żenczykowskicae181d2022-06-16 23:26:33 -0700114
Lorenzo Colitti3505b582022-10-27 19:36:27 +0900115#define DEFINE_NETD_BPF_PROG_KVER(SECTION_NAME, prog_uid, prog_gid, the_prog, min_kv) \
Maciej Żenczykowski879839a12022-08-03 10:48:25 +0000116 DEFINE_NETD_BPF_PROG_KVER_RANGE(SECTION_NAME, prog_uid, prog_gid, the_prog, min_kv, KVER_INF)
117
118#define DEFINE_NETD_BPF_PROG(SECTION_NAME, prog_uid, prog_gid, the_prog) \
119 DEFINE_NETD_BPF_PROG_KVER(SECTION_NAME, prog_uid, prog_gid, the_prog, KVER_NONE)
Lorenzo Colitti3505b582022-10-27 19:36:27 +0900120
Maciej Żenczykowskicae181d2022-06-16 23:26:33 -0700121// programs that only need to be usable by the system server
122#define DEFINE_SYS_BPF_PROG(SECTION_NAME, prog_uid, prog_gid, the_prog) \
123 DEFINE_BPF_PROG_EXT(SECTION_NAME, prog_uid, prog_gid, the_prog, \
124 KVER_NONE, KVER_INF, false, "fs_bpf_net_shared", "")
125
Ken Chen587d4232022-01-17 17:18:43 +0800126static __always_inline int is_system_uid(uint32_t uid) {
Maciej Żenczykowskib909d8a2022-06-15 00:40:43 -0700127 // MIN_SYSTEM_UID is AID_ROOT == 0, so uint32_t is *always* >= 0
128 // MAX_SYSTEM_UID is AID_NOBODY == 9999, while AID_APP_START == 10000
129 return (uid < AID_APP_START);
Ken Chen587d4232022-01-17 17:18:43 +0800130}
131
132/*
133 * Note: this blindly assumes an MTU of 1500, and that packets > MTU are always TCP,
134 * and that TCP is using the Linux default settings with TCP timestamp option enabled
135 * which uses 12 TCP option bytes per frame.
136 *
137 * These are not unreasonable assumptions:
138 *
139 * The internet does not really support MTUs greater than 1500, so most TCP traffic will
140 * be at that MTU, or slightly below it (worst case our upwards adjustment is too small).
141 *
142 * The chance our traffic isn't IP at all is basically zero, so the IP overhead correction
143 * is bound to be needed.
144 *
145 * Furthermore, the likelyhood that we're having to deal with GSO (ie. > MTU) packets that
146 * are not IP/TCP is pretty small (few other things are supported by Linux) and worse case
147 * our extra overhead will be slightly off, but probably still better than assuming none.
148 *
149 * Most servers are also Linux and thus support/default to using TCP timestamp option
150 * (and indeed TCP timestamp option comes from RFC 1323 titled "TCP Extensions for High
151 * Performance" which also defined TCP window scaling and are thus absolutely ancient...).
152 *
153 * All together this should be more correct than if we simply ignored GSO frames
154 * (ie. counted them as single packets with no extra overhead)
155 *
156 * Especially since the number of packets is important for any future clat offload correction.
157 * (which adjusts upward by 20 bytes per packet to account for ipv4 -> ipv6 header conversion)
158 */
159#define DEFINE_UPDATE_STATS(the_stats_map, TypeOfKey) \
160 static __always_inline inline void update_##the_stats_map(struct __sk_buff* skb, \
Maciej Żenczykowski55474982022-11-20 13:48:39 +0000161 bool egress, TypeOfKey* key) { \
Ken Chen587d4232022-01-17 17:18:43 +0800162 StatsValue* value = bpf_##the_stats_map##_lookup_elem(key); \
163 if (!value) { \
164 StatsValue newValue = {}; \
165 bpf_##the_stats_map##_update_elem(key, &newValue, BPF_NOEXIST); \
166 value = bpf_##the_stats_map##_lookup_elem(key); \
167 } \
168 if (value) { \
169 const int mtu = 1500; \
170 uint64_t packets = 1; \
171 uint64_t bytes = skb->len; \
172 if (bytes > mtu) { \
173 bool is_ipv6 = (skb->protocol == htons(ETH_P_IPV6)); \
174 int ip_overhead = (is_ipv6 ? sizeof(struct ipv6hdr) : sizeof(struct iphdr)); \
175 int tcp_overhead = ip_overhead + sizeof(struct tcphdr) + 12; \
176 int mss = mtu - tcp_overhead; \
177 uint64_t payload = bytes - tcp_overhead; \
178 packets = (payload + mss - 1) / mss; \
179 bytes = tcp_overhead * packets + payload; \
180 } \
Maciej Żenczykowski55474982022-11-20 13:48:39 +0000181 if (egress) { \
Ken Chen587d4232022-01-17 17:18:43 +0800182 __sync_fetch_and_add(&value->txPackets, packets); \
183 __sync_fetch_and_add(&value->txBytes, bytes); \
Maciej Żenczykowski55474982022-11-20 13:48:39 +0000184 } else { \
Ken Chen587d4232022-01-17 17:18:43 +0800185 __sync_fetch_and_add(&value->rxPackets, packets); \
186 __sync_fetch_and_add(&value->rxBytes, bytes); \
187 } \
188 } \
189 }
190
191DEFINE_UPDATE_STATS(app_uid_stats_map, uint32_t)
192DEFINE_UPDATE_STATS(iface_stats_map, uint32_t)
193DEFINE_UPDATE_STATS(stats_map_A, StatsKey)
194DEFINE_UPDATE_STATS(stats_map_B, StatsKey)
195
Maciej Żenczykowski879839a12022-08-03 10:48:25 +0000196// both of these return 0 on success or -EFAULT on failure (and zero out the buffer)
Maciej Żenczykowski0966bbe2022-12-29 11:39:48 +0000197static __always_inline inline int bpf_skb_load_bytes_net(const struct __sk_buff* const skb,
198 const int L3_off,
199 void* const to,
200 const int len,
201 const unsigned kver) {
202 // 'kver' (here and throughout) is the compile time guaranteed minimum kernel version,
203 // ie. we're building (a version of) the bpf program for kver (or newer!) kernels.
204 //
205 // 4.19+ kernels support the 'bpf_skb_load_bytes_relative()' bpf helper function,
206 // so we can use it. On pre-4.19 kernels we cannot use the relative load helper,
207 // and thus will simply get things wrong if there's any L2 (ethernet) header in the skb.
208 //
209 // Luckily, for cellular traffic, there likely isn't any, as cell is usually 'rawip'.
210 //
211 // However, this does mean that wifi (and ethernet) on 4.14 is basically a lost cause:
212 // we'll be making decisions based on the *wrong* bytes (fetched from the wrong offset),
213 // because the 'L3_off' passed to bpf_skb_load_bytes() should be increased by l2_header_size,
214 // which for ethernet is 14 and not 0 like it is for rawip.
215 //
216 // For similar reasons this will fail with non-offloaded VLAN tags on < 4.19 kernels,
217 // since those extend the ethernet header from 14 to 18 bytes.
218 return kver >= KVER(4, 19, 0)
219 ? bpf_skb_load_bytes_relative(skb, L3_off, to, len, BPF_HDR_START_NET)
220 : bpf_skb_load_bytes(skb, L3_off, to, len);
Maciej Żenczykowski879839a12022-08-03 10:48:25 +0000221}
222
Maciej Żenczykowski0966bbe2022-12-29 11:39:48 +0000223static __always_inline inline bool skip_owner_match(struct __sk_buff* skb, const unsigned kver) {
Maciej Żenczykowski3621cbd2022-11-20 13:31:06 +0000224 uint32_t flag = 0;
Ken Chen587d4232022-01-17 17:18:43 +0800225 if (skb->protocol == htons(ETH_P_IP)) {
Ken Chen587d4232022-01-17 17:18:43 +0800226 uint8_t proto;
Maciej Żenczykowski879839a12022-08-03 10:48:25 +0000227 // no need to check for success, proto will be zeroed if bpf_skb_load_bytes_net() fails
Maciej Żenczykowski0966bbe2022-12-29 11:39:48 +0000228 (void)bpf_skb_load_bytes_net(skb, IP_PROTO_OFF, &proto, sizeof(proto), kver);
Maciej Żenczykowski879839a12022-08-03 10:48:25 +0000229 if (proto == IPPROTO_ESP) return true;
230 if (proto != IPPROTO_TCP) return false; // handles read failure above
231 uint8_t ihl;
232 // we don't check for success, as this cannot fail, as it is earlier in the packet than
233 // proto, the reading of which must have succeeded, additionally the next read
234 // (a little bit deeper in the packet in spite of ihl being zeroed) of the tcp flags
235 // field will also fail, and that failure we already handle correctly
236 // (we also don't check that ihl in [0x45,0x4F] nor that ipv4 header checksum is correct)
Maciej Żenczykowski0966bbe2022-12-29 11:39:48 +0000237 (void)bpf_skb_load_bytes_net(skb, IPPROTO_IHL_OFF, &ihl, sizeof(ihl), kver);
Maciej Żenczykowski879839a12022-08-03 10:48:25 +0000238 // if the read below fails, we'll just assume no TCP flags are set, which is fine.
239 (void)bpf_skb_load_bytes_net(skb, (ihl & 0xF) * 4 + TCP_FLAG32_OFF,
Maciej Żenczykowski0966bbe2022-12-29 11:39:48 +0000240 &flag, sizeof(flag), kver);
Maciej Żenczykowski879839a12022-08-03 10:48:25 +0000241 } else if (skb->protocol == htons(ETH_P_IPV6)) {
242 uint8_t proto;
243 // no need to check for success, proto will be zeroed if bpf_skb_load_bytes_net() fails
Maciej Żenczykowski0966bbe2022-12-29 11:39:48 +0000244 (void)bpf_skb_load_bytes_net(skb, IPV6_PROTO_OFF, &proto, sizeof(proto), kver);
Maciej Żenczykowski879839a12022-08-03 10:48:25 +0000245 if (proto == IPPROTO_ESP) return true;
246 if (proto != IPPROTO_TCP) return false; // handles read failure above
Maciej Żenczykowski879839a12022-08-03 10:48:25 +0000247 // if the read below fails, we'll just assume no TCP flags are set, which is fine.
248 (void)bpf_skb_load_bytes_net(skb, sizeof(struct ipv6hdr) + TCP_FLAG32_OFF,
Maciej Żenczykowski0966bbe2022-12-29 11:39:48 +0000249 &flag, sizeof(flag), kver);
Maciej Żenczykowski3621cbd2022-11-20 13:31:06 +0000250 } else {
251 return false;
Ken Chen587d4232022-01-17 17:18:43 +0800252 }
Maciej Żenczykowski3621cbd2022-11-20 13:31:06 +0000253 return flag & TCP_FLAG_RST; // false on read failure
Ken Chen587d4232022-01-17 17:18:43 +0800254}
255
Maciej Żenczykowski879839a12022-08-03 10:48:25 +0000256static __always_inline inline BpfConfig getConfig(uint32_t configKey) {
Ken Chen587d4232022-01-17 17:18:43 +0800257 uint32_t mapSettingKey = configKey;
258 BpfConfig* config = bpf_configuration_map_lookup_elem(&mapSettingKey);
259 if (!config) {
260 // Couldn't read configuration entry. Assume everything is disabled.
261 return DEFAULT_CONFIG;
262 }
263 return *config;
264}
265
Maciej Żenczykowski55474982022-11-20 13:48:39 +0000266// DROP_IF_SET is set of rules that DROP if rule is globally enabled, and per-uid bit is set
Maciej Żenczykowski474512a2022-06-07 23:22:53 +0000267#define DROP_IF_SET (STANDBY_MATCH | OEM_DENY_1_MATCH | OEM_DENY_2_MATCH | OEM_DENY_3_MATCH)
268// DROP_IF_UNSET is set of rules that should DROP if globally enabled, and per-uid bit is NOT set
269#define DROP_IF_UNSET (DOZABLE_MATCH | POWERSAVE_MATCH | RESTRICTED_MATCH | LOW_POWER_STANDBY_MATCH)
270
Maciej Żenczykowski879839a12022-08-03 10:48:25 +0000271static __always_inline inline int bpf_owner_match(struct __sk_buff* skb, uint32_t uid,
Maciej Żenczykowski0966bbe2022-12-29 11:39:48 +0000272 bool egress, const unsigned kver) {
273 if (skip_owner_match(skb, kver)) return PASS;
Ken Chen587d4232022-01-17 17:18:43 +0800274
Maciej Żenczykowski55474982022-11-20 13:48:39 +0000275 if (is_system_uid(uid)) return PASS;
Ken Chen587d4232022-01-17 17:18:43 +0800276
277 BpfConfig enabledRules = getConfig(UID_RULES_CONFIGURATION_KEY);
278
279 UidOwnerValue* uidEntry = bpf_uid_owner_map_lookup_elem(&uid);
Motomu Utsumi42edc602022-05-12 13:57:42 +0000280 uint32_t uidRules = uidEntry ? uidEntry->rule : 0;
Ken Chen587d4232022-01-17 17:18:43 +0800281 uint32_t allowed_iif = uidEntry ? uidEntry->iif : 0;
282
Maciej Żenczykowski474512a2022-06-07 23:22:53 +0000283 // Warning: funky bit-wise arithmetic: in parallel, for all DROP_IF_SET/UNSET rules
284 // check whether the rules are globally enabled, and if so whether the rules are
Maciej Żenczykowski55474982022-11-20 13:48:39 +0000285 // set/unset for the specific uid. DROP if that is the case for ANY of the rules.
Maciej Żenczykowski474512a2022-06-07 23:22:53 +0000286 // We achieve this by masking out only the bits/rules we're interested in checking,
287 // and negating (via bit-wise xor) the bits/rules that should drop if unset.
Maciej Żenczykowski55474982022-11-20 13:48:39 +0000288 if (enabledRules & (DROP_IF_SET | DROP_IF_UNSET) & (uidRules ^ DROP_IF_UNSET)) return DROP;
Maciej Żenczykowski474512a2022-06-07 23:22:53 +0000289
Maciej Żenczykowski55474982022-11-20 13:48:39 +0000290 if (!egress && skb->ifindex != 1) {
Motomu Utsumib08654c2022-05-11 05:56:26 +0000291 if (uidRules & IIF_MATCH) {
292 if (allowed_iif && skb->ifindex != allowed_iif) {
293 // Drops packets not coming from lo nor the allowed interface
294 // allowed interface=0 is a wildcard and does not drop packets
Maciej Żenczykowski55474982022-11-20 13:48:39 +0000295 return DROP_UNLESS_DNS;
Motomu Utsumib08654c2022-05-11 05:56:26 +0000296 }
297 } else if (uidRules & LOCKDOWN_VPN_MATCH) {
298 // Drops packets not coming from lo and rule does not have IIF_MATCH but has
299 // LOCKDOWN_VPN_MATCH
Maciej Żenczykowski55474982022-11-20 13:48:39 +0000300 return DROP_UNLESS_DNS;
Ken Chen587d4232022-01-17 17:18:43 +0800301 }
302 }
Maciej Żenczykowski55474982022-11-20 13:48:39 +0000303 return PASS;
Ken Chen587d4232022-01-17 17:18:43 +0800304}
305
Maciej Żenczykowski55474982022-11-20 13:48:39 +0000306static __always_inline inline void update_stats_with_config(struct __sk_buff* skb, bool egress,
Lorenzo Colitti60cbed32022-03-03 17:49:01 +0900307 StatsKey* key, uint32_t selectedMap) {
Ken Chen587d4232022-01-17 17:18:43 +0800308 if (selectedMap == SELECT_MAP_A) {
Maciej Żenczykowski55474982022-11-20 13:48:39 +0000309 update_stats_map_A(skb, egress, key);
Maciej Żenczykowski28b9a292022-12-29 12:06:33 +0000310 } else {
Maciej Żenczykowski55474982022-11-20 13:48:39 +0000311 update_stats_map_B(skb, egress, key);
Ken Chen587d4232022-01-17 17:18:43 +0800312 }
313}
314
Maciej Żenczykowski55474982022-11-20 13:48:39 +0000315static __always_inline inline int bpf_traffic_account(struct __sk_buff* skb, bool egress,
Maciej Żenczykowski0966bbe2022-12-29 11:39:48 +0000316 const unsigned kver) {
Ken Chen587d4232022-01-17 17:18:43 +0800317 uint32_t sock_uid = bpf_get_socket_uid(skb);
318 uint64_t cookie = bpf_get_socket_cookie(skb);
319 UidTagValue* utag = bpf_cookie_tag_map_lookup_elem(&cookie);
320 uint32_t uid, tag;
321 if (utag) {
322 uid = utag->uid;
323 tag = utag->tag;
324 } else {
325 uid = sock_uid;
326 tag = 0;
327 }
328
329 // Always allow and never count clat traffic. Only the IPv4 traffic on the stacked
330 // interface is accounted for and subject to usage restrictions.
331 // TODO: remove sock_uid check once Nat464Xlat javaland adds the socket tag AID_CLAT for clat.
332 if (sock_uid == AID_CLAT || uid == AID_CLAT) {
Maciej Żenczykowski55474982022-11-20 13:48:39 +0000333 return PASS;
Ken Chen587d4232022-01-17 17:18:43 +0800334 }
335
Maciej Żenczykowski0966bbe2022-12-29 11:39:48 +0000336 int match = bpf_owner_match(skb, sock_uid, egress, kver);
Maciej Żenczykowski55474982022-11-20 13:48:39 +0000337 if (egress && (match == DROP)) {
Ken Chen587d4232022-01-17 17:18:43 +0800338 // If an outbound packet is going to be dropped, we do not count that
339 // traffic.
340 return match;
341 }
342
343// Workaround for secureVPN with VpnIsolation enabled, refer to b/159994981 for details.
344// Keep TAG_SYSTEM_DNS in sync with DnsResolver/include/netd_resolv/resolv.h
345// and TrafficStatsConstants.java
346#define TAG_SYSTEM_DNS 0xFFFFFF82
347 if (tag == TAG_SYSTEM_DNS && uid == AID_DNS) {
348 uid = sock_uid;
Maciej Żenczykowski55474982022-11-20 13:48:39 +0000349 if (match == DROP_UNLESS_DNS) match = PASS;
Ken Chen587d4232022-01-17 17:18:43 +0800350 } else {
Maciej Żenczykowski55474982022-11-20 13:48:39 +0000351 if (match == DROP_UNLESS_DNS) match = DROP;
Ken Chen587d4232022-01-17 17:18:43 +0800352 }
353
354 StatsKey key = {.uid = uid, .tag = tag, .counterSet = 0, .ifaceIndex = skb->ifindex};
355
356 uint8_t* counterSet = bpf_uid_counterset_map_lookup_elem(&uid);
357 if (counterSet) key.counterSet = (uint32_t)*counterSet;
358
359 uint32_t mapSettingKey = CURRENT_STATS_MAP_CONFIGURATION_KEY;
Lorenzo Colitti60cbed32022-03-03 17:49:01 +0900360 uint32_t* selectedMap = bpf_configuration_map_lookup_elem(&mapSettingKey);
Ken Chen587d4232022-01-17 17:18:43 +0800361
362 // Use asm("%0 &= 1" : "+r"(match)) before return match,
363 // to help kernel's bpf verifier, so that it can be 100% certain
364 // that the returned value is always BPF_NOMATCH(0) or BPF_MATCH(1).
365 if (!selectedMap) {
366 asm("%0 &= 1" : "+r"(match));
367 return match;
368 }
369
370 if (key.tag) {
Maciej Żenczykowski55474982022-11-20 13:48:39 +0000371 update_stats_with_config(skb, egress, &key, *selectedMap);
Ken Chen587d4232022-01-17 17:18:43 +0800372 key.tag = 0;
373 }
374
Maciej Żenczykowski55474982022-11-20 13:48:39 +0000375 update_stats_with_config(skb, egress, &key, *selectedMap);
376 update_app_uid_stats_map(skb, egress, &uid);
Ken Chen587d4232022-01-17 17:18:43 +0800377 asm("%0 &= 1" : "+r"(match));
378 return match;
379}
380
Maciej Żenczykowski879839a12022-08-03 10:48:25 +0000381DEFINE_NETD_BPF_PROG_KVER_RANGE("cgroupskb/ingress/stats$4_19", AID_ROOT, AID_SYSTEM,
382 bpf_cgroup_ingress_4_19, KVER(4, 19, 0), KVER_INF)
Ken Chen587d4232022-01-17 17:18:43 +0800383(struct __sk_buff* skb) {
Maciej Żenczykowski0966bbe2022-12-29 11:39:48 +0000384 return bpf_traffic_account(skb, INGRESS, KVER(4, 19, 0));
Ken Chen587d4232022-01-17 17:18:43 +0800385}
386
Maciej Żenczykowski879839a12022-08-03 10:48:25 +0000387DEFINE_NETD_BPF_PROG_KVER_RANGE("cgroupskb/ingress/stats$4_14", AID_ROOT, AID_SYSTEM,
388 bpf_cgroup_ingress_4_14, KVER_NONE, KVER(4, 19, 0))
Ken Chen587d4232022-01-17 17:18:43 +0800389(struct __sk_buff* skb) {
Maciej Żenczykowski0966bbe2022-12-29 11:39:48 +0000390 return bpf_traffic_account(skb, INGRESS, KVER_NONE);
Maciej Żenczykowski879839a12022-08-03 10:48:25 +0000391}
392
393DEFINE_NETD_BPF_PROG_KVER_RANGE("cgroupskb/egress/stats$4_19", AID_ROOT, AID_SYSTEM,
394 bpf_cgroup_egress_4_19, KVER(4, 19, 0), KVER_INF)
395(struct __sk_buff* skb) {
Maciej Żenczykowski0966bbe2022-12-29 11:39:48 +0000396 return bpf_traffic_account(skb, EGRESS, KVER(4, 19, 0));
Maciej Żenczykowski879839a12022-08-03 10:48:25 +0000397}
398
399DEFINE_NETD_BPF_PROG_KVER_RANGE("cgroupskb/egress/stats$4_14", AID_ROOT, AID_SYSTEM,
400 bpf_cgroup_egress_4_14, KVER_NONE, KVER(4, 19, 0))
401(struct __sk_buff* skb) {
Maciej Żenczykowski0966bbe2022-12-29 11:39:48 +0000402 return bpf_traffic_account(skb, EGRESS, KVER_NONE);
Ken Chen587d4232022-01-17 17:18:43 +0800403}
404
Maciej Żenczykowski12057372022-06-14 14:36:34 -0700405// WARNING: Android T's non-updatable netd depends on the name of this program.
Maciej Żenczykowskicae181d2022-06-16 23:26:33 -0700406DEFINE_XTBPF_PROG("skfilter/egress/xtbpf", AID_ROOT, AID_NET_ADMIN, xt_bpf_egress_prog)
Ken Chen587d4232022-01-17 17:18:43 +0800407(struct __sk_buff* skb) {
408 // Clat daemon does not generate new traffic, all its traffic is accounted for already
409 // on the v4-* interfaces (except for the 20 (or 28) extra bytes of IPv6 vs IPv4 overhead,
410 // but that can be corrected for later when merging v4-foo stats into interface foo's).
411 // TODO: remove sock_uid check once Nat464Xlat javaland adds the socket tag AID_CLAT for clat.
412 uint32_t sock_uid = bpf_get_socket_uid(skb);
413 if (sock_uid == AID_CLAT) return BPF_NOMATCH;
414 if (sock_uid == AID_SYSTEM) {
415 uint64_t cookie = bpf_get_socket_cookie(skb);
416 UidTagValue* utag = bpf_cookie_tag_map_lookup_elem(&cookie);
417 if (utag && utag->uid == AID_CLAT) return BPF_NOMATCH;
418 }
419
420 uint32_t key = skb->ifindex;
Maciej Żenczykowski55474982022-11-20 13:48:39 +0000421 update_iface_stats_map(skb, EGRESS, &key);
Ken Chen587d4232022-01-17 17:18:43 +0800422 return BPF_MATCH;
423}
424
Maciej Żenczykowski12057372022-06-14 14:36:34 -0700425// WARNING: Android T's non-updatable netd depends on the name of this program.
Maciej Żenczykowskicae181d2022-06-16 23:26:33 -0700426DEFINE_XTBPF_PROG("skfilter/ingress/xtbpf", AID_ROOT, AID_NET_ADMIN, xt_bpf_ingress_prog)
Ken Chen587d4232022-01-17 17:18:43 +0800427(struct __sk_buff* skb) {
428 // Clat daemon traffic is not accounted by virtue of iptables raw prerouting drop rule
429 // (in clat_raw_PREROUTING chain), which triggers before this (in bw_raw_PREROUTING chain).
430 // It will be accounted for on the v4-* clat interface instead.
431 // Keep that in mind when moving this out of iptables xt_bpf and into tc ingress (or xdp).
432
433 uint32_t key = skb->ifindex;
Maciej Żenczykowski55474982022-11-20 13:48:39 +0000434 update_iface_stats_map(skb, INGRESS, &key);
Ken Chen587d4232022-01-17 17:18:43 +0800435 return BPF_MATCH;
436}
437
Maciej Żenczykowskicae181d2022-06-16 23:26:33 -0700438DEFINE_SYS_BPF_PROG("schedact/ingress/account", AID_ROOT, AID_NET_ADMIN,
439 tc_bpf_ingress_account_prog)
Ken Chen587d4232022-01-17 17:18:43 +0800440(struct __sk_buff* skb) {
Patrick Rohr148aea82022-02-24 14:12:32 +0100441 if (is_received_skb(skb)) {
442 // Account for ingress traffic before tc drops it.
443 uint32_t key = skb->ifindex;
Maciej Żenczykowski55474982022-11-20 13:48:39 +0000444 update_iface_stats_map(skb, INGRESS, &key);
Patrick Rohr148aea82022-02-24 14:12:32 +0100445 }
Ken Chen587d4232022-01-17 17:18:43 +0800446 return TC_ACT_UNSPEC;
447}
448
Maciej Żenczykowski12057372022-06-14 14:36:34 -0700449// WARNING: Android T's non-updatable netd depends on the name of this program.
Maciej Żenczykowskicae181d2022-06-16 23:26:33 -0700450DEFINE_XTBPF_PROG("skfilter/allowlist/xtbpf", AID_ROOT, AID_NET_ADMIN, xt_bpf_allowlist_prog)
Ken Chen587d4232022-01-17 17:18:43 +0800451(struct __sk_buff* skb) {
452 uint32_t sock_uid = bpf_get_socket_uid(skb);
453 if (is_system_uid(sock_uid)) return BPF_MATCH;
454
455 // 65534 is the overflow 'nobody' uid, usually this being returned means
456 // that skb->sk is NULL during RX (early decap socket lookup failure),
457 // which commonly happens for incoming packets to an unconnected udp socket.
458 // Additionally bpf_get_socket_cookie() returns 0 if skb->sk is NULL
459 if ((sock_uid == 65534) && !bpf_get_socket_cookie(skb) && is_received_skb(skb))
460 return BPF_MATCH;
461
462 UidOwnerValue* allowlistMatch = bpf_uid_owner_map_lookup_elem(&sock_uid);
463 if (allowlistMatch) return allowlistMatch->rule & HAPPY_BOX_MATCH ? BPF_MATCH : BPF_NOMATCH;
464 return BPF_NOMATCH;
465}
466
Maciej Żenczykowski12057372022-06-14 14:36:34 -0700467// WARNING: Android T's non-updatable netd depends on the name of this program.
Maciej Żenczykowskicae181d2022-06-16 23:26:33 -0700468DEFINE_XTBPF_PROG("skfilter/denylist/xtbpf", AID_ROOT, AID_NET_ADMIN, xt_bpf_denylist_prog)
Ken Chen587d4232022-01-17 17:18:43 +0800469(struct __sk_buff* skb) {
470 uint32_t sock_uid = bpf_get_socket_uid(skb);
471 UidOwnerValue* denylistMatch = bpf_uid_owner_map_lookup_elem(&sock_uid);
472 if (denylistMatch) return denylistMatch->rule & PENALTY_BOX_MATCH ? BPF_MATCH : BPF_NOMATCH;
473 return BPF_NOMATCH;
474}
475
Lorenzo Colitti3505b582022-10-27 19:36:27 +0900476DEFINE_NETD_BPF_PROG_KVER("cgroupsock/inet/create", AID_ROOT, AID_ROOT, inet_socket_create,
477 KVER(4, 14, 0))
Ken Chen587d4232022-01-17 17:18:43 +0800478(struct bpf_sock* sk) {
479 uint64_t gid_uid = bpf_get_current_uid_gid();
480 /*
481 * A given app is guaranteed to have the same app ID in all the profiles in
482 * which it is installed, and install permission is granted to app for all
483 * user at install time so we only check the appId part of a request uid at
484 * run time. See UserHandle#isSameApp for detail.
485 */
Maciej Żenczykowskib909d8a2022-06-15 00:40:43 -0700486 uint32_t appId = (gid_uid & 0xffffffff) % AID_USER_OFFSET; // == PER_USER_RANGE == 100000
Ken Chen587d4232022-01-17 17:18:43 +0800487 uint8_t* permissions = bpf_uid_permission_map_lookup_elem(&appId);
488 if (!permissions) {
489 // UID not in map. Default to just INTERNET permission.
490 return 1;
491 }
492
493 // A return value of 1 means allow, everything else means deny.
494 return (*permissions & BPF_PERMISSION_INTERNET) == BPF_PERMISSION_INTERNET;
495}
496
497LICENSE("Apache 2.0");
Maciej Żenczykowskic41e35d2022-08-04 13:58:46 +0000498CRITICAL("Connectivity and netd");