blob: b5474e9d0d5e2b5a96b11f04f0fa45991ceb332f [file] [log] [blame]
Joel Fernandesd76a2002018-10-16 13:19:58 -07001/*
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
17#define LOG_TAG "LibBpfLoader"
18
19#include <errno.h>
20#include <linux/bpf.h>
21#include <linux/elf.h>
22#include <log/log.h>
23#include <stdint.h>
24#include <stdio.h>
25#include <stdlib.h>
26#include <string.h>
Connor O'Brien35425e52022-01-18 21:41:16 -080027#include <sysexits.h>
Maciej Żenczykowski83f29772020-01-27 03:11:51 -080028#include <sys/stat.h>
Joel Fernandesd76a2002018-10-16 13:19:58 -070029#include <sys/utsname.h>
Connor O'Brien35425e52022-01-18 21:41:16 -080030#include <sys/wait.h>
Joel Fernandesd76a2002018-10-16 13:19:58 -070031#include <unistd.h>
32
Patrick Rohraa624f22023-03-28 15:36:11 -070033// This is BpfLoader v0.35
Maciej Żenczykowski05825662023-01-20 13:00:24 +000034// WARNING: If you ever hit cherrypick conflicts here you're doing it wrong:
35// You are NOT allowed to cherrypick bpfloader related patches out of order.
36// (indeed: cherrypicking is probably a bad idea and you should merge instead)
37// Mainline supports ONLY the published versions of the bpfloader for each Android release.
Maciej Żenczykowski9217eee2021-03-03 05:28:52 -080038#define BPFLOADER_VERSION_MAJOR 0u
Patrick Rohraa624f22023-03-28 15:36:11 -070039#define BPFLOADER_VERSION_MINOR 35u
Maciej Żenczykowski9217eee2021-03-03 05:28:52 -080040#define BPFLOADER_VERSION ((BPFLOADER_VERSION_MAJOR << 16) | BPFLOADER_VERSION_MINOR)
41
Maciej Żenczykowski300c51f2022-12-14 04:18:02 -080042#include "BpfSyscallWrappers.h"
Maciej Żenczykowski07375e22020-02-19 14:23:59 -080043#include "bpf/BpfUtils.h"
Ken Chend5689472021-12-20 18:07:21 +080044#include "bpf/bpf_map_def.h"
Joel Fernandesd76a2002018-10-16 13:19:58 -070045#include "include/libbpf_android.h"
46
Maciej Żenczykowskib3dcc142022-07-24 22:42:33 +000047#if BPFLOADER_VERSION < COMPILE_FOR_BPFLOADER_VERSION
48#error "BPFLOADER_VERSION is less than COMPILE_FOR_BPFLOADER_VERSION"
49#endif
50
Connor O'Brien35425e52022-01-18 21:41:16 -080051#include <bpf/bpf.h>
52
Joel Fernandesd76a2002018-10-16 13:19:58 -070053#include <cstdlib>
54#include <fstream>
55#include <iostream>
Connor O'Brien3278a162020-02-13 21:45:22 -080056#include <optional>
Joel Fernandesd76a2002018-10-16 13:19:58 -070057#include <string>
Connor O'Brien35425e52022-01-18 21:41:16 -080058#include <unordered_map>
Christopher Ferrisc151c672019-02-01 15:31:26 -080059#include <vector>
Joel Fernandesd76a2002018-10-16 13:19:58 -070060
Connor O'Brien35425e52022-01-18 21:41:16 -080061#include <android-base/cmsg.h>
62#include <android-base/file.h>
Joel Fernandesd76a2002018-10-16 13:19:58 -070063#include <android-base/strings.h>
Connor O'Brien8d49fc72019-10-24 18:23:49 -070064#include <android-base/unique_fd.h>
Ryan Zukliece89f502022-12-15 16:14:32 -080065#include <cutils/properties.h>
Joel Fernandesd76a2002018-10-16 13:19:58 -070066
67#define BPF_FS_PATH "/sys/fs/bpf/"
68
69// Size of the BPF log buffer for verifier logging
Stephane Leeeb61b732021-10-21 17:03:11 -070070#define BPF_LOAD_LOG_SZ 0xfffff
Joel Fernandesd76a2002018-10-16 13:19:58 -070071
Tyler Wear4e2f4602022-02-03 09:46:01 -080072// Unspecified attach type is 0 which is BPF_CGROUP_INET_INGRESS.
73#define BPF_ATTACH_TYPE_UNSPEC BPF_CGROUP_INET_INGRESS
74
Joel Fernandesd76a2002018-10-16 13:19:58 -070075using android::base::StartsWith;
Connor O'Brien8d49fc72019-10-24 18:23:49 -070076using android::base::unique_fd;
Joel Fernandesd76a2002018-10-16 13:19:58 -070077using std::ifstream;
78using std::ios;
Connor O'Brien3278a162020-02-13 21:45:22 -080079using std::optional;
Christopher Ferrisc151c672019-02-01 15:31:26 -080080using std::string;
Joel Fernandesd76a2002018-10-16 13:19:58 -070081using std::vector;
82
Ryan Zukliece89f502022-12-15 16:14:32 -080083static std::string getBuildTypeInternal() {
84 char value[PROPERTY_VALUE_MAX] = {};
85 (void)property_get("ro.build.type", value, "unknown"); // ignore length
86 return value;
87}
88
Joel Fernandesd76a2002018-10-16 13:19:58 -070089namespace android {
90namespace bpf {
91
Ryan Zukliece89f502022-12-15 16:14:32 -080092const std::string& getBuildType() {
93 static std::string t = getBuildTypeInternal();
94 return t;
95}
96
Maciej Żenczykowski41817132022-06-03 08:41:04 -070097constexpr const char* lookupSelinuxContext(const domain d, const char* const unspecified = "") {
98 switch (d) {
99 case domain::unspecified: return unspecified;
100 case domain::platform: return "fs_bpf";
101 case domain::tethering: return "fs_bpf_tethering";
102 case domain::net_private: return "fs_bpf_net_private";
103 case domain::net_shared: return "fs_bpf_net_shared";
104 case domain::netd_readonly: return "fs_bpf_netd_readonly";
105 case domain::netd_shared: return "fs_bpf_netd_shared";
106 case domain::vendor: return "fs_bpf_vendor";
Maciej Żenczykowski9a2093d2022-12-02 13:46:53 +0000107 case domain::loader: return "fs_bpf_loader";
Maciej Żenczykowski41817132022-06-03 08:41:04 -0700108 default: return "(unrecognized)";
109 }
110}
111
112domain getDomainFromSelinuxContext(const char s[BPF_SELINUX_CONTEXT_CHAR_ARRAY_SIZE]) {
113 for (domain d : AllDomains) {
114 // Not sure how to enforce this at compile time, so abort() bpfloader at boot instead
115 if (strlen(lookupSelinuxContext(d)) >= BPF_SELINUX_CONTEXT_CHAR_ARRAY_SIZE) abort();
116 if (!strncmp(s, lookupSelinuxContext(d), BPF_SELINUX_CONTEXT_CHAR_ARRAY_SIZE)) return d;
117 }
Maciej Żenczykowski99668462022-07-02 16:58:01 -0700118 ALOGW("ignoring unrecognized selinux_context '%-32s'", s);
Maciej Żenczykowski41817132022-06-03 08:41:04 -0700119 // We should return 'unrecognized' here, however: returning unspecified will
120 // result in the system simply using the default context, which in turn
121 // will allow future expansion by adding more restrictive selinux types.
122 // Older bpfloader will simply ignore that, and use the less restrictive default.
123 // This does mean you CANNOT later add a *less* restrictive type than the default.
124 //
125 // Note: we cannot just abort() here as this might be a mainline module shipped optional update
126 return domain::unspecified;
127}
128
129constexpr const char* lookupPinSubdir(const domain d, const char* const unspecified = "") {
130 switch (d) {
131 case domain::unspecified: return unspecified;
132 case domain::platform: return "/";
133 case domain::tethering: return "tethering/";
134 case domain::net_private: return "net_private/";
135 case domain::net_shared: return "net_shared/";
136 case domain::netd_readonly: return "netd_readonly/";
137 case domain::netd_shared: return "netd_shared/";
138 case domain::vendor: return "vendor/";
Maciej Żenczykowski9a2093d2022-12-02 13:46:53 +0000139 case domain::loader: return "loader/";
Maciej Żenczykowski41817132022-06-03 08:41:04 -0700140 default: return "(unrecognized)";
141 }
142};
143
144domain getDomainFromPinSubdir(const char s[BPF_PIN_SUBDIR_CHAR_ARRAY_SIZE]) {
145 for (domain d : AllDomains) {
146 // Not sure how to enforce this at compile time, so abort() bpfloader at boot instead
147 if (strlen(lookupPinSubdir(d)) >= BPF_PIN_SUBDIR_CHAR_ARRAY_SIZE) abort();
148 if (!strncmp(s, lookupPinSubdir(d), BPF_PIN_SUBDIR_CHAR_ARRAY_SIZE)) return d;
149 }
Maciej Żenczykowski99668462022-07-02 16:58:01 -0700150 ALOGE("unrecognized pin_subdir '%-32s'", s);
Maciej Żenczykowski41817132022-06-03 08:41:04 -0700151 // pin_subdir affects the object's full pathname,
152 // and thus using the default would change the location and thus our code's ability to find it,
153 // hence this seems worth treating as a true error condition.
154 //
155 // Note: we cannot just abort() here as this might be a mainline module shipped optional update
156 // However, our callers will treat this as an error, and stop loading the specific .o,
157 // which will fail bpfloader if the .o is marked critical.
158 return domain::unrecognized;
159}
160
Maciej Żenczykowski21869ef2022-07-07 06:01:57 -0700161static string pathToObjName(const string& path) {
162 // extract everything after the final slash, ie. this is the filename 'foo@1.o' or 'bar.o'
163 string filename = android::base::Split(path, "/").back();
164 // strip off everything from the final period onwards (strip '.o' suffix), ie. 'foo@1' or 'bar'
165 string name = filename.substr(0, filename.find_last_of('.'));
166 // strip any potential @1 suffix, this will leave us with just 'foo' or 'bar'
167 // this can be used to provide duplicate programs (mux based on the bpfloader version)
168 return name.substr(0, name.find_last_of('@'));
Maciej Żenczykowskif7c0d992021-01-21 16:01:04 -0800169}
170
Joel Fernandesd76a2002018-10-16 13:19:58 -0700171typedef struct {
172 const char* name;
173 enum bpf_prog_type type;
Tyler Wear4e2f4602022-02-03 09:46:01 -0800174 enum bpf_attach_type expected_attach_type;
Joel Fernandesd76a2002018-10-16 13:19:58 -0700175} sectionType;
176
177/*
178 * Map section name prefixes to program types, the section name will be:
Maciej Żenczykowski3adb1d52021-10-22 19:27:10 -0700179 * SECTION(<prefix>/<name-of-program>)
Joel Fernandesd76a2002018-10-16 13:19:58 -0700180 * For example:
Maciej Żenczykowski3adb1d52021-10-22 19:27:10 -0700181 * SECTION("tracepoint/sched_switch_func") where sched_switch_funcs
Joel Fernandesd76a2002018-10-16 13:19:58 -0700182 * is the name of the program, and tracepoint is the type.
Maciej Żenczykowski3adb1d52021-10-22 19:27:10 -0700183 *
184 * However, be aware that you should not be directly using the SECTION() macro.
185 * Instead use the DEFINE_(BPF|XDP)_(PROG|MAP)... & LICENSE/CRITICAL macros.
Joel Fernandesd76a2002018-10-16 13:19:58 -0700186 */
187sectionType sectionNameTypes[] = {
Tyler Wear4e2f4602022-02-03 09:46:01 -0800188 {"bind4/", BPF_PROG_TYPE_CGROUP_SOCK_ADDR, BPF_CGROUP_INET4_BIND},
189 {"bind6/", BPF_PROG_TYPE_CGROUP_SOCK_ADDR, BPF_CGROUP_INET6_BIND},
190 {"cgroupskb/", BPF_PROG_TYPE_CGROUP_SKB, BPF_ATTACH_TYPE_UNSPEC},
191 {"cgroupsock/", BPF_PROG_TYPE_CGROUP_SOCK, BPF_ATTACH_TYPE_UNSPEC},
Patrick Rohraa624f22023-03-28 15:36:11 -0700192 {"connect4/", BPF_PROG_TYPE_CGROUP_SOCK_ADDR, BPF_CGROUP_INET4_CONNECT},
193 {"connect6/", BPF_PROG_TYPE_CGROUP_SOCK_ADDR, BPF_CGROUP_INET6_CONNECT},
Tyler Wear4e2f4602022-02-03 09:46:01 -0800194 {"kprobe/", BPF_PROG_TYPE_KPROBE, BPF_ATTACH_TYPE_UNSPEC},
Maciej Żenczykowskie092e0b2022-05-24 13:14:32 +0000195 {"perf_event/", BPF_PROG_TYPE_PERF_EVENT, BPF_ATTACH_TYPE_UNSPEC},
Tyler Wear4e2f4602022-02-03 09:46:01 -0800196 {"schedact/", BPF_PROG_TYPE_SCHED_ACT, BPF_ATTACH_TYPE_UNSPEC},
197 {"schedcls/", BPF_PROG_TYPE_SCHED_CLS, BPF_ATTACH_TYPE_UNSPEC},
198 {"skfilter/", BPF_PROG_TYPE_SOCKET_FILTER, BPF_ATTACH_TYPE_UNSPEC},
199 {"tracepoint/", BPF_PROG_TYPE_TRACEPOINT, BPF_ATTACH_TYPE_UNSPEC},
200 {"xdp/", BPF_PROG_TYPE_XDP, BPF_ATTACH_TYPE_UNSPEC},
Joel Fernandesd76a2002018-10-16 13:19:58 -0700201};
202
203typedef struct {
204 enum bpf_prog_type type;
Tyler Wear4e2f4602022-02-03 09:46:01 -0800205 enum bpf_attach_type expected_attach_type;
Joel Fernandesd76a2002018-10-16 13:19:58 -0700206 string name;
207 vector<char> data;
208 vector<char> rel_data;
Connor O'Brien3278a162020-02-13 21:45:22 -0800209 optional<struct bpf_prog_def> prog_def;
Joel Fernandesd76a2002018-10-16 13:19:58 -0700210
Connor O'Brien8d49fc72019-10-24 18:23:49 -0700211 unique_fd prog_fd; /* fd after loading */
Joel Fernandesd76a2002018-10-16 13:19:58 -0700212} codeSection;
213
Joel Fernandesd76a2002018-10-16 13:19:58 -0700214static int readElfHeader(ifstream& elfFile, Elf64_Ehdr* eh) {
215 elfFile.seekg(0);
216 if (elfFile.fail()) return -1;
217
218 if (!elfFile.read((char*)eh, sizeof(*eh))) return -1;
219
220 return 0;
221}
222
223/* Reads all section header tables into an Shdr array */
224static int readSectionHeadersAll(ifstream& elfFile, vector<Elf64_Shdr>& shTable) {
225 Elf64_Ehdr eh;
226 int ret = 0;
227
228 ret = readElfHeader(elfFile, &eh);
229 if (ret) return ret;
230
231 elfFile.seekg(eh.e_shoff);
232 if (elfFile.fail()) return -1;
233
234 /* Read shdr table entries */
235 shTable.resize(eh.e_shnum);
236
237 if (!elfFile.read((char*)shTable.data(), (eh.e_shnum * eh.e_shentsize))) return -ENOMEM;
238
239 return 0;
240}
241
242/* Read a section by its index - for ex to get sec hdr strtab blob */
243static int readSectionByIdx(ifstream& elfFile, int id, vector<char>& sec) {
244 vector<Elf64_Shdr> shTable;
Maciej Żenczykowskid56ec052021-01-15 00:27:04 -0800245 int ret = readSectionHeadersAll(elfFile, shTable);
Joel Fernandesd76a2002018-10-16 13:19:58 -0700246 if (ret) return ret;
Joel Fernandesd76a2002018-10-16 13:19:58 -0700247
248 elfFile.seekg(shTable[id].sh_offset);
249 if (elfFile.fail()) return -1;
250
251 sec.resize(shTable[id].sh_size);
252 if (!elfFile.read(sec.data(), shTable[id].sh_size)) return -1;
253
254 return 0;
255}
256
257/* Read whole section header string table */
258static int readSectionHeaderStrtab(ifstream& elfFile, vector<char>& strtab) {
259 Elf64_Ehdr eh;
Maciej Żenczykowskid56ec052021-01-15 00:27:04 -0800260 int ret = readElfHeader(elfFile, &eh);
Joel Fernandesd76a2002018-10-16 13:19:58 -0700261 if (ret) return ret;
262
263 ret = readSectionByIdx(elfFile, eh.e_shstrndx, strtab);
264 if (ret) return ret;
265
266 return 0;
267}
268
269/* Get name from offset in strtab */
270static int getSymName(ifstream& elfFile, int nameOff, string& name) {
271 int ret;
272 vector<char> secStrTab;
273
274 ret = readSectionHeaderStrtab(elfFile, secStrTab);
275 if (ret) return ret;
276
277 if (nameOff >= (int)secStrTab.size()) return -1;
278
279 name = string((char*)secStrTab.data() + nameOff);
280 return 0;
281}
282
283/* Reads a full section by name - example to get the GPL license */
284static int readSectionByName(const char* name, ifstream& elfFile, vector<char>& data) {
285 vector<char> secStrTab;
286 vector<Elf64_Shdr> shTable;
287 int ret;
288
289 ret = readSectionHeadersAll(elfFile, shTable);
290 if (ret) return ret;
291
292 ret = readSectionHeaderStrtab(elfFile, secStrTab);
293 if (ret) return ret;
294
295 for (int i = 0; i < (int)shTable.size(); i++) {
296 char* secname = secStrTab.data() + shTable[i].sh_name;
297 if (!secname) continue;
298
299 if (!strcmp(secname, name)) {
300 vector<char> dataTmp;
301 dataTmp.resize(shTable[i].sh_size);
302
303 elfFile.seekg(shTable[i].sh_offset);
304 if (elfFile.fail()) return -1;
305
306 if (!elfFile.read((char*)dataTmp.data(), shTable[i].sh_size)) return -1;
307
308 data = dataTmp;
309 return 0;
310 }
311 }
312 return -2;
313}
314
Maciej Żenczykowski7ed94ef2021-07-06 01:47:15 -0700315unsigned int readSectionUint(const char* name, ifstream& elfFile, unsigned int defVal) {
Maciej Żenczykowski9217eee2021-03-03 05:28:52 -0800316 vector<char> theBytes;
317 int ret = readSectionByName(name, elfFile, theBytes);
318 if (ret) {
Maciej Żenczykowskie626a952022-06-17 02:35:36 -0700319 ALOGD("Couldn't find section %s (defaulting to %u [0x%x]).", name, defVal, defVal);
Maciej Żenczykowski9217eee2021-03-03 05:28:52 -0800320 return defVal;
321 } else if (theBytes.size() < sizeof(unsigned int)) {
Maciej Żenczykowskie626a952022-06-17 02:35:36 -0700322 ALOGE("Section %s too short (defaulting to %u [0x%x]).", name, defVal, defVal);
Maciej Żenczykowski9217eee2021-03-03 05:28:52 -0800323 return defVal;
324 } else {
325 // decode first 4 bytes as LE32 uint, there will likely be more bytes due to alignment.
326 unsigned int value = static_cast<unsigned char>(theBytes[3]);
327 value <<= 8;
328 value += static_cast<unsigned char>(theBytes[2]);
329 value <<= 8;
330 value += static_cast<unsigned char>(theBytes[1]);
331 value <<= 8;
332 value += static_cast<unsigned char>(theBytes[0]);
Maciej Żenczykowskie626a952022-06-17 02:35:36 -0700333 ALOGI("Section %s value is %u [0x%x]", name, value, value);
Maciej Żenczykowski9217eee2021-03-03 05:28:52 -0800334 return value;
335 }
336}
337
Joel Fernandesd76a2002018-10-16 13:19:58 -0700338static int readSectionByType(ifstream& elfFile, int type, vector<char>& data) {
339 int ret;
340 vector<Elf64_Shdr> shTable;
341
342 ret = readSectionHeadersAll(elfFile, shTable);
343 if (ret) return ret;
344
345 for (int i = 0; i < (int)shTable.size(); i++) {
346 if ((int)shTable[i].sh_type != type) continue;
347
348 vector<char> dataTmp;
349 dataTmp.resize(shTable[i].sh_size);
350
351 elfFile.seekg(shTable[i].sh_offset);
352 if (elfFile.fail()) return -1;
353
354 if (!elfFile.read((char*)dataTmp.data(), shTable[i].sh_size)) return -1;
355
356 data = dataTmp;
357 return 0;
358 }
359 return -2;
360}
361
362static bool symCompare(Elf64_Sym a, Elf64_Sym b) {
363 return (a.st_value < b.st_value);
364}
365
366static int readSymTab(ifstream& elfFile, int sort, vector<Elf64_Sym>& data) {
367 int ret, numElems;
368 Elf64_Sym* buf;
369 vector<char> secData;
370
371 ret = readSectionByType(elfFile, SHT_SYMTAB, secData);
372 if (ret) return ret;
373
374 buf = (Elf64_Sym*)secData.data();
375 numElems = (secData.size() / sizeof(Elf64_Sym));
376 data.assign(buf, buf + numElems);
377
378 if (sort) std::sort(data.begin(), data.end(), symCompare);
379 return 0;
380}
381
Paul Lawrence7fb8b542022-07-19 16:13:49 -0700382static enum bpf_prog_type getFuseProgType() {
383 int result = BPF_PROG_TYPE_UNSPEC;
384 ifstream("/sys/fs/fuse/bpf_prog_type_fuse") >> result;
385 return static_cast<bpf_prog_type>(result);
386}
387
Joel Fernandesd76a2002018-10-16 13:19:58 -0700388static enum bpf_prog_type getSectionType(string& name) {
Maciej Żenczykowski2b203132021-11-18 15:13:36 -0800389 for (auto& snt : sectionNameTypes)
390 if (StartsWith(name, snt.name)) return snt.type;
Joel Fernandesd76a2002018-10-16 13:19:58 -0700391
Paul Lawrence9548f9f2021-11-09 16:32:43 +0000392 // TODO Remove this code when fuse-bpf is upstream and this BPF_PROG_TYPE_FUSE is fixed
Paul Lawrence7fb8b542022-07-19 16:13:49 -0700393 if (StartsWith(name, "fuse/")) return getFuseProgType();
Paul Lawrence9548f9f2021-11-09 16:32:43 +0000394
Joel Fernandesd76a2002018-10-16 13:19:58 -0700395 return BPF_PROG_TYPE_UNSPEC;
396}
397
Tyler Wear4e2f4602022-02-03 09:46:01 -0800398static enum bpf_attach_type getExpectedAttachType(string& name) {
399 for (auto& snt : sectionNameTypes)
400 if (StartsWith(name, snt.name)) return snt.expected_attach_type;
401 return BPF_ATTACH_TYPE_UNSPEC;
402}
403
Joel Fernandesd76a2002018-10-16 13:19:58 -0700404static string getSectionName(enum bpf_prog_type type)
405{
Maciej Żenczykowski2b203132021-11-18 15:13:36 -0800406 for (auto& snt : sectionNameTypes)
407 if (snt.type == type)
408 return string(snt.name);
Joel Fernandesd76a2002018-10-16 13:19:58 -0700409
Steven Moreland0f10f3f2019-12-12 14:22:34 -0800410 return "UNKNOWN SECTION NAME " + std::to_string(type);
Joel Fernandesd76a2002018-10-16 13:19:58 -0700411}
Joel Fernandesd76a2002018-10-16 13:19:58 -0700412
Maciej Żenczykowski9217eee2021-03-03 05:28:52 -0800413static int readProgDefs(ifstream& elfFile, vector<struct bpf_prog_def>& pd,
414 size_t sizeOfBpfProgDef) {
Connor O'Brien3278a162020-02-13 21:45:22 -0800415 vector<char> pdData;
416 int ret = readSectionByName("progs", elfFile, pdData);
Maciej Żenczykowski9217eee2021-03-03 05:28:52 -0800417 // Older file formats do not require a 'progs' section at all.
418 // (We should probably figure out whether this is behaviour which is safe to remove now.)
Connor O'Brien3278a162020-02-13 21:45:22 -0800419 if (ret == -2) return 0;
420 if (ret) return ret;
421
Maciej Żenczykowski9217eee2021-03-03 05:28:52 -0800422 if (pdData.size() % sizeOfBpfProgDef) {
Maciej Żenczykowskie626a952022-06-17 02:35:36 -0700423 ALOGE("readProgDefs failed due to improper sized progs section, %zu %% %zu != 0",
Maciej Żenczykowski9217eee2021-03-03 05:28:52 -0800424 pdData.size(), sizeOfBpfProgDef);
425 return -1;
426 };
427
428 int progCount = pdData.size() / sizeOfBpfProgDef;
429 pd.resize(progCount);
430 size_t trimmedSize = std::min(sizeOfBpfProgDef, sizeof(struct bpf_prog_def));
431
432 const char* dataPtr = pdData.data();
433 for (auto& p : pd) {
434 // First we zero initialize
435 memset(&p, 0, sizeof(p));
436 // Then we set non-zero defaults
437 p.bpfloader_max_ver = DEFAULT_BPFLOADER_MAX_VER; // v1.0
438 // Then we copy over the structure prefix from the ELF file.
439 memcpy(&p, dataPtr, trimmedSize);
440 // Move to next struct in the ELF file
441 dataPtr += sizeOfBpfProgDef;
442 }
Connor O'Brien3278a162020-02-13 21:45:22 -0800443 return 0;
444}
445
Connor O'Brien0ea4c6b2022-01-14 00:18:11 -0800446static int getSectionSymNames(ifstream& elfFile, const string& sectionName, vector<string>& names,
447 optional<unsigned> symbolType = std::nullopt) {
Connor O'Brien3278a162020-02-13 21:45:22 -0800448 int ret;
449 string name;
450 vector<Elf64_Sym> symtab;
451 vector<Elf64_Shdr> shTable;
452
453 ret = readSymTab(elfFile, 1 /* sort */, symtab);
454 if (ret) return ret;
455
456 /* Get index of section */
457 ret = readSectionHeadersAll(elfFile, shTable);
458 if (ret) return ret;
459
460 int sec_idx = -1;
461 for (int i = 0; i < (int)shTable.size(); i++) {
462 ret = getSymName(elfFile, shTable[i].sh_name, name);
463 if (ret) return ret;
464
465 if (!name.compare(sectionName)) {
466 sec_idx = i;
467 break;
468 }
469 }
470
471 /* No section found with matching name*/
472 if (sec_idx == -1) {
Maciej Żenczykowskie626a952022-06-17 02:35:36 -0700473 ALOGW("No %s section could be found in elf object", sectionName.c_str());
Connor O'Brien3278a162020-02-13 21:45:22 -0800474 return -1;
475 }
476
477 for (int i = 0; i < (int)symtab.size(); i++) {
Connor O'Brien0ea4c6b2022-01-14 00:18:11 -0800478 if (symbolType.has_value() && ELF_ST_TYPE(symtab[i].st_info) != symbolType) continue;
479
Connor O'Brien3278a162020-02-13 21:45:22 -0800480 if (symtab[i].st_shndx == sec_idx) {
481 string s;
482 ret = getSymName(elfFile, symtab[i].st_name, s);
483 if (ret) return ret;
484 names.push_back(s);
485 }
486 }
487
488 return 0;
489}
490
Steven Moreland0f10f3f2019-12-12 14:22:34 -0800491static bool IsAllowed(bpf_prog_type type, const bpf_prog_type* allowed, size_t numAllowed) {
492 if (allowed == nullptr) return true;
493
494 for (size_t i = 0; i < numAllowed; i++) {
Paul Lawrence7fb8b542022-07-19 16:13:49 -0700495 if (allowed[i] == BPF_PROG_TYPE_UNSPEC) {
496 if (type == getFuseProgType()) return true;
497 } else if (type == allowed[i])
498 return true;
Steven Moreland0f10f3f2019-12-12 14:22:34 -0800499 }
500
501 return false;
502}
503
Joel Fernandesd76a2002018-10-16 13:19:58 -0700504/* Read a section by its index - for ex to get sec hdr strtab blob */
Steven Moreland0f10f3f2019-12-12 14:22:34 -0800505static int readCodeSections(ifstream& elfFile, vector<codeSection>& cs, size_t sizeOfBpfProgDef,
506 const bpf_prog_type* allowed, size_t numAllowed) {
Joel Fernandesd76a2002018-10-16 13:19:58 -0700507 vector<Elf64_Shdr> shTable;
508 int entries, ret = 0;
509
510 ret = readSectionHeadersAll(elfFile, shTable);
511 if (ret) return ret;
512 entries = shTable.size();
513
Connor O'Brien3278a162020-02-13 21:45:22 -0800514 vector<struct bpf_prog_def> pd;
Maciej Żenczykowski9217eee2021-03-03 05:28:52 -0800515 ret = readProgDefs(elfFile, pd, sizeOfBpfProgDef);
Connor O'Brien3278a162020-02-13 21:45:22 -0800516 if (ret) return ret;
517 vector<string> progDefNames;
518 ret = getSectionSymNames(elfFile, "progs", progDefNames);
519 if (!pd.empty() && ret) return ret;
520
Joel Fernandesd76a2002018-10-16 13:19:58 -0700521 for (int i = 0; i < entries; i++) {
522 string name;
523 codeSection cs_temp;
524 cs_temp.type = BPF_PROG_TYPE_UNSPEC;
525
526 ret = getSymName(elfFile, shTable[i].sh_name, name);
527 if (ret) return ret;
528
529 enum bpf_prog_type ptype = getSectionType(name);
Steven Moreland0f10f3f2019-12-12 14:22:34 -0800530
Tyler Wear4e2f4602022-02-03 09:46:01 -0800531 if (ptype == BPF_PROG_TYPE_UNSPEC) continue;
Maciej Żenczykowskif7c0d992021-01-21 16:01:04 -0800532
Steven Moreland0f10f3f2019-12-12 14:22:34 -0800533 if (!IsAllowed(ptype, allowed, numAllowed)) {
534 ALOGE("Program type %s not permitted here", getSectionName(ptype).c_str());
535 return -1;
536 }
537
Tyler Wear4e2f4602022-02-03 09:46:01 -0800538 // This must be done before '/' is replaced with '_'.
539 cs_temp.expected_attach_type = getExpectedAttachType(name);
Maciej Żenczykowskif7c0d992021-01-21 16:01:04 -0800540
Tyler Wear4e2f4602022-02-03 09:46:01 -0800541 string oldName = name;
Joel Fernandesd76a2002018-10-16 13:19:58 -0700542
Tyler Wear4e2f4602022-02-03 09:46:01 -0800543 // convert all slashes to underscores
544 std::replace(name.begin(), name.end(), '/', '_');
Connor O'Brien3278a162020-02-13 21:45:22 -0800545
Tyler Wear4e2f4602022-02-03 09:46:01 -0800546 cs_temp.type = ptype;
547 cs_temp.name = name;
548
549 ret = readSectionByIdx(elfFile, i, cs_temp.data);
550 if (ret) return ret;
Maciej Żenczykowskie626a952022-06-17 02:35:36 -0700551 ALOGD("Loaded code section %d (%s)", i, name.c_str());
Tyler Wear4e2f4602022-02-03 09:46:01 -0800552
553 vector<string> csSymNames;
554 ret = getSectionSymNames(elfFile, oldName, csSymNames, STT_FUNC);
555 if (ret || !csSymNames.size()) return ret;
556 for (size_t i = 0; i < progDefNames.size(); ++i) {
557 if (!progDefNames[i].compare(csSymNames[0] + "_def")) {
558 cs_temp.prog_def = pd[i];
559 break;
Connor O'Brien3278a162020-02-13 21:45:22 -0800560 }
Joel Fernandesd76a2002018-10-16 13:19:58 -0700561 }
562
563 /* Check for rel section */
564 if (cs_temp.data.size() > 0 && i < entries) {
565 ret = getSymName(elfFile, shTable[i + 1].sh_name, name);
566 if (ret) return ret;
567
Tyler Wear4e2f4602022-02-03 09:46:01 -0800568 if (name == (".rel" + oldName)) {
Joel Fernandesd76a2002018-10-16 13:19:58 -0700569 ret = readSectionByIdx(elfFile, i + 1, cs_temp.rel_data);
570 if (ret) return ret;
Maciej Żenczykowskie626a952022-06-17 02:35:36 -0700571 ALOGD("Loaded relo section %d (%s)", i, name.c_str());
Joel Fernandesd76a2002018-10-16 13:19:58 -0700572 }
573 }
574
575 if (cs_temp.data.size() > 0) {
Connor O'Brien8d49fc72019-10-24 18:23:49 -0700576 cs.push_back(std::move(cs_temp));
Maciej Żenczykowskie626a952022-06-17 02:35:36 -0700577 ALOGD("Adding section %d to cs list", i);
Joel Fernandesd76a2002018-10-16 13:19:58 -0700578 }
579 }
580 return 0;
581}
582
583static int getSymNameByIdx(ifstream& elfFile, int index, string& name) {
584 vector<Elf64_Sym> symtab;
585 int ret = 0;
586
587 ret = readSymTab(elfFile, 0 /* !sort */, symtab);
588 if (ret) return ret;
589
590 if (index >= (int)symtab.size()) return -1;
591
592 return getSymName(elfFile, symtab[index].st_name, name);
593}
594
Connor O'Brien35425e52022-01-18 21:41:16 -0800595static bool waitpidTimeout(pid_t pid, int timeoutMs) {
596 // Add SIGCHLD to the signal set.
597 sigset_t child_mask, original_mask;
598 sigemptyset(&child_mask);
599 sigaddset(&child_mask, SIGCHLD);
600 if (sigprocmask(SIG_BLOCK, &child_mask, &original_mask) == -1) return false;
601
602 // Wait for a SIGCHLD notification.
603 errno = 0;
604 timespec ts = {0, timeoutMs * 1000000};
605 int wait_result = TEMP_FAILURE_RETRY(sigtimedwait(&child_mask, nullptr, &ts));
606
607 // Restore the original signal set.
608 sigprocmask(SIG_SETMASK, &original_mask, nullptr);
609
610 if (wait_result == -1) return false;
611
612 int status;
613 return TEMP_FAILURE_RETRY(waitpid(pid, &status, WNOHANG)) == pid;
614}
615
616static std::optional<unique_fd> getMapBtfInfo(const char* elfPath,
617 std::unordered_map<string, std::pair<uint32_t, uint32_t>> &btfTypeIds) {
618 unique_fd bpfloaderSocket, btfloaderSocket;
619 if (!android::base::Socketpair(AF_UNIX, SOCK_DGRAM | SOCK_NONBLOCK, 0, &bpfloaderSocket,
620 &btfloaderSocket)) {
621 return {};
622 }
623
624 unique_fd pipeRead, pipeWrite;
625 if (!android::base::Pipe(&pipeRead, &pipeWrite, O_NONBLOCK)) {
626 return {};
627 }
628
629 pid_t pid = fork();
630 if (pid < 0) return {};
631 if (!pid) {
632 bpfloaderSocket.reset();
633 pipeRead.reset();
634 auto socketFdStr = std::to_string(btfloaderSocket.release());
635 auto pipeFdStr = std::to_string(pipeWrite.release());
636
637 if (execl("/system/bin/btfloader", "/system/bin/btfloader", socketFdStr.c_str(),
638 pipeFdStr.c_str(), elfPath, NULL) == -1) {
Maciej Żenczykowskie626a952022-06-17 02:35:36 -0700639 ALOGW("exec btfloader failed with errno %d (%s)", errno, strerror(errno));
Connor O'Brien35425e52022-01-18 21:41:16 -0800640 exit(EX_UNAVAILABLE);
641 }
642 }
643 btfloaderSocket.reset();
644 pipeWrite.reset();
645 if (!waitpidTimeout(pid, 100)) {
646 kill(pid, SIGKILL);
647 return {};
648 }
649
650 unique_fd btfFd;
651 if (android::base::ReceiveFileDescriptors(bpfloaderSocket, nullptr, 0, &btfFd)) return {};
652
653 std::string btfTypeIdStr;
654 if (!android::base::ReadFdToString(pipeRead, &btfTypeIdStr)) return {};
Maciej Żenczykowski12bb5202022-06-16 15:50:58 -0700655 if (!btfFd.ok()) return {};
Connor O'Brien35425e52022-01-18 21:41:16 -0800656
657 const auto mapTypeIdLines = android::base::Split(btfTypeIdStr, "\n");
658 for (const auto &line : mapTypeIdLines) {
659 const auto vec = android::base::Split(line, " ");
660 // Splitting on newline will give us one empty line
661 if (vec.size() != 3) continue;
662 const int kTid = atoi(vec[1].c_str());
663 const int vTid = atoi(vec[2].c_str());
664 if (!kTid || !vTid) return {};
665 btfTypeIds[vec[0]] = std::make_pair(kTid, vTid);
666 }
667 return btfFd;
668}
669
Maciej Żenczykowski1a7fff32022-06-20 18:16:24 -0700670static bool mapMatchesExpectations(const unique_fd& fd, const string& mapName,
671 const struct bpf_map_def& mapDef, const enum bpf_map_type type) {
Maciej Żenczykowski1a7fff32022-06-20 18:16:24 -0700672 // Assuming fd is a valid Bpf Map file descriptor then
673 // all the following should always succeed on a 4.14+ kernel.
674 // If they somehow do fail, they'll return -1 (and set errno),
675 // which should then cause (among others) a key_size mismatch.
Maciej Żenczykowski12bb5202022-06-16 15:50:58 -0700676 int fd_type = bpfGetFdMapType(fd);
677 int fd_key_size = bpfGetFdKeySize(fd);
678 int fd_value_size = bpfGetFdValueSize(fd);
679 int fd_max_entries = bpfGetFdMaxEntries(fd);
680 int fd_map_flags = bpfGetFdMapFlags(fd);
681
682 // DEVMAPs are readonly from the bpf program side's point of view, as such
683 // the kernel in kernel/bpf/devmap.c dev_map_init_map() will set the flag
684 int desired_map_flags = (int)mapDef.map_flags;
685 if (type == BPF_MAP_TYPE_DEVMAP || type == BPF_MAP_TYPE_DEVMAP_HASH)
686 desired_map_flags |= BPF_F_RDONLY_PROG;
687
Maciej Żenczykowski1a7fff32022-06-20 18:16:24 -0700688 // The following checks should *never* trigger, if one of them somehow does,
689 // it probably means a bpf .o file has been changed/replaced at runtime
690 // and bpfloader was manually rerun (normally it should only run *once*
691 // early during the boot process).
692 // Another possibility is that something is misconfigured in the code:
693 // most likely a shared map is declared twice differently.
694 // But such a change should never be checked into the source tree...
695 if ((fd_type == type) &&
696 (fd_key_size == (int)mapDef.key_size) &&
697 (fd_value_size == (int)mapDef.value_size) &&
698 (fd_max_entries == (int)mapDef.max_entries) &&
699 (fd_map_flags == desired_map_flags)) {
700 return true;
701 }
Maciej Żenczykowski12bb5202022-06-16 15:50:58 -0700702
703 ALOGE("bpf map name %s mismatch: desired/found: "
704 "type:%d/%d key:%u/%d value:%u/%d entries:%u/%d flags:%u/%d",
705 mapName.c_str(), type, fd_type, mapDef.key_size, fd_key_size, mapDef.value_size,
706 fd_value_size, mapDef.max_entries, fd_max_entries, desired_map_flags, fd_map_flags);
707 return false;
708}
709
Maciej Żenczykowskid8a45782021-01-14 23:36:32 -0800710static int createMaps(const char* elfPath, ifstream& elfFile, vector<unique_fd>& mapFds,
Maciej Żenczykowski41817132022-06-03 08:41:04 -0700711 const char* prefix, const unsigned long long allowedDomainBitmask,
712 const size_t sizeOfBpfMapDef) {
Connor O'Brien8d49fc72019-10-24 18:23:49 -0700713 int ret;
Connor O'Brien35425e52022-01-18 21:41:16 -0800714 vector<char> mdData, btfData;
Joel Fernandesd76a2002018-10-16 13:19:58 -0700715 vector<struct bpf_map_def> md;
716 vector<string> mapNames;
Connor O'Brien35425e52022-01-18 21:41:16 -0800717 std::unordered_map<string, std::pair<uint32_t, uint32_t>> btfTypeIdMap;
Maciej Żenczykowski21869ef2022-07-07 06:01:57 -0700718 string objName = pathToObjName(string(elfPath));
Joel Fernandesd76a2002018-10-16 13:19:58 -0700719
720 ret = readSectionByName("maps", elfFile, mdData);
Steven Morelandc0905b42019-12-12 14:21:20 -0800721 if (ret == -2) return 0; // no maps to read
Joel Fernandesd76a2002018-10-16 13:19:58 -0700722 if (ret) return ret;
Maciej Żenczykowski9217eee2021-03-03 05:28:52 -0800723
724 if (mdData.size() % sizeOfBpfMapDef) {
Maciej Żenczykowskie626a952022-06-17 02:35:36 -0700725 ALOGE("createMaps failed due to improper sized maps section, %zu %% %zu != 0",
Maciej Żenczykowski9217eee2021-03-03 05:28:52 -0800726 mdData.size(), sizeOfBpfMapDef);
727 return -1;
728 };
729
730 int mapCount = mdData.size() / sizeOfBpfMapDef;
731 md.resize(mapCount);
732 size_t trimmedSize = std::min(sizeOfBpfMapDef, sizeof(struct bpf_map_def));
733
734 const char* dataPtr = mdData.data();
735 for (auto& m : md) {
736 // First we zero initialize
737 memset(&m, 0, sizeof(m));
738 // Then we set non-zero defaults
739 m.bpfloader_max_ver = DEFAULT_BPFLOADER_MAX_VER; // v1.0
Maciej Żenczykowski36c53ba2021-07-05 15:20:34 -0700740 m.max_kver = 0xFFFFFFFFu; // matches KVER_INF from bpf_helpers.h
Maciej Żenczykowski9217eee2021-03-03 05:28:52 -0800741 // Then we copy over the structure prefix from the ELF file.
742 memcpy(&m, dataPtr, trimmedSize);
743 // Move to next struct in the ELF file
744 dataPtr += sizeOfBpfMapDef;
745 }
Joel Fernandesd76a2002018-10-16 13:19:58 -0700746
Connor O'Brien3278a162020-02-13 21:45:22 -0800747 ret = getSectionSymNames(elfFile, "maps", mapNames);
Joel Fernandesd76a2002018-10-16 13:19:58 -0700748 if (ret) return ret;
749
Maciej Żenczykowskibbab8182022-06-22 22:51:07 -0700750 unsigned btfMinBpfLoaderVer = readSectionUint("btf_min_bpfloader_ver", elfFile, 0);
751 unsigned btfMinKernelVer = readSectionUint("btf_min_kernel_ver", elfFile, 0);
752 unsigned kvers = kernelVersion();
753
Connor O'Brien35425e52022-01-18 21:41:16 -0800754 std::optional<unique_fd> btfFd;
Maciej Żenczykowskibbab8182022-06-22 22:51:07 -0700755 if ((BPFLOADER_VERSION >= btfMinBpfLoaderVer) && (kvers >= btfMinKernelVer) &&
756 (!readSectionByName(".BTF", elfFile, btfData))) {
Connor O'Brien35425e52022-01-18 21:41:16 -0800757 btfFd = getMapBtfInfo(elfPath, btfTypeIdMap);
758 }
759
Maciej Żenczykowski36c53ba2021-07-05 15:20:34 -0700760 for (int i = 0; i < (int)mapNames.size(); i++) {
Maciej Żenczykowski2a5d0162022-07-21 13:27:24 +0000761 if (md[i].zero != 0) abort();
762
Maciej Żenczykowski9217eee2021-03-03 05:28:52 -0800763 if (BPFLOADER_VERSION < md[i].bpfloader_min_ver) {
Maciej Żenczykowskie626a952022-06-17 02:35:36 -0700764 ALOGI("skipping map %s which requires bpfloader min ver 0x%05x", mapNames[i].c_str(),
Maciej Żenczykowski9217eee2021-03-03 05:28:52 -0800765 md[i].bpfloader_min_ver);
Maciej Żenczykowskia21256d2021-07-02 00:40:55 -0700766 mapFds.push_back(unique_fd());
Maciej Żenczykowski9217eee2021-03-03 05:28:52 -0800767 continue;
768 }
769
770 if (BPFLOADER_VERSION >= md[i].bpfloader_max_ver) {
Maciej Żenczykowskie626a952022-06-17 02:35:36 -0700771 ALOGI("skipping map %s which requires bpfloader max ver 0x%05x", mapNames[i].c_str(),
Maciej Żenczykowski9217eee2021-03-03 05:28:52 -0800772 md[i].bpfloader_max_ver);
Maciej Żenczykowskia21256d2021-07-02 00:40:55 -0700773 mapFds.push_back(unique_fd());
Maciej Żenczykowski9217eee2021-03-03 05:28:52 -0800774 continue;
775 }
776
Maciej Żenczykowski36c53ba2021-07-05 15:20:34 -0700777 if (kvers < md[i].min_kver) {
Maciej Żenczykowskie626a952022-06-17 02:35:36 -0700778 ALOGI("skipping map %s which requires kernel version 0x%x >= 0x%x",
Maciej Żenczykowski36c53ba2021-07-05 15:20:34 -0700779 mapNames[i].c_str(), kvers, md[i].min_kver);
780 mapFds.push_back(unique_fd());
781 continue;
782 }
783
784 if (kvers >= md[i].max_kver) {
Maciej Żenczykowskie626a952022-06-17 02:35:36 -0700785 ALOGI("skipping map %s which requires kernel version 0x%x < 0x%x",
Maciej Żenczykowski36c53ba2021-07-05 15:20:34 -0700786 mapNames[i].c_str(), kvers, md[i].max_kver);
787 mapFds.push_back(unique_fd());
788 continue;
789 }
790
Ryan Zukliece89f502022-12-15 16:14:32 -0800791 if ((md[i].ignore_on_eng && isEng()) || (md[i].ignore_on_user && isUser()) ||
792 (md[i].ignore_on_userdebug && isUserdebug())) {
793 ALOGI("skipping map %s which is ignored on %s builds", mapNames[i].c_str(),
794 getBuildType().c_str());
795 mapFds.push_back(unique_fd());
796 continue;
797 }
798
Maciej Żenczykowski12bb5202022-06-16 15:50:58 -0700799 enum bpf_map_type type = md[i].type;
Maciej Żenczykowski12bb5202022-06-16 15:50:58 -0700800 if (type == BPF_MAP_TYPE_DEVMAP_HASH && !isAtLeastKernelVersion(5, 4, 0)) {
801 // On Linux Kernels older than 5.4 this map type doesn't exist, but it can kind
802 // of be approximated: HASH has the same userspace visible api.
803 // However it cannot be used by ebpf programs in the same way.
804 // Since bpf_redirect_map() only requires 4.14, a program using a DEVMAP_HASH map
805 // would fail to load (due to trying to redirect to a HASH instead of DEVMAP_HASH).
806 // One must thus tag any BPF_MAP_TYPE_DEVMAP_HASH + bpf_redirect_map() using
807 // programs as being 5.4+...
808 type = BPF_MAP_TYPE_HASH;
809 }
810
Maciej Żenczykowski41817132022-06-03 08:41:04 -0700811 domain selinux_context = getDomainFromSelinuxContext(md[i].selinux_context);
812 if (specified(selinux_context)) {
813 if (!inDomainBitmask(selinux_context, allowedDomainBitmask)) {
814 ALOGE("map %s has invalid selinux_context of %d (allowed bitmask 0x%llx)",
815 mapNames[i].c_str(), selinux_context, allowedDomainBitmask);
816 return -EINVAL;
817 }
Maciej Żenczykowski99668462022-07-02 16:58:01 -0700818 ALOGI("map %s selinux_context [%-32s] -> %d -> '%s' (%s)", mapNames[i].c_str(),
Maciej Żenczykowski41817132022-06-03 08:41:04 -0700819 md[i].selinux_context, selinux_context, lookupSelinuxContext(selinux_context),
820 lookupPinSubdir(selinux_context));
821 }
822
823 domain pin_subdir = getDomainFromPinSubdir(md[i].pin_subdir);
824 if (unrecognized(pin_subdir)) return -ENOTDIR;
825 if (specified(pin_subdir)) {
826 if (!inDomainBitmask(pin_subdir, allowedDomainBitmask)) {
827 ALOGE("map %s has invalid pin_subdir of %d (allowed bitmask 0x%llx)",
828 mapNames[i].c_str(), pin_subdir, allowedDomainBitmask);
829 return -EINVAL;
830 }
Maciej Żenczykowski99668462022-07-02 16:58:01 -0700831 ALOGI("map %s pin_subdir [%-32s] -> %d -> '%s'", mapNames[i].c_str(), md[i].pin_subdir,
Maciej Żenczykowski41817132022-06-03 08:41:04 -0700832 pin_subdir, lookupPinSubdir(pin_subdir));
833 }
834
Maciej Żenczykowski21869ef2022-07-07 06:01:57 -0700835 // Format of pin location is /sys/fs/bpf/<pin_subdir|prefix>map_<objName>_<mapName>
836 // except that maps shared across .o's have empty <objName>
837 // Note: <objName> refers to the extension-less basename of the .o file (without @ suffix).
Maciej Żenczykowski41817132022-06-03 08:41:04 -0700838 string mapPinLoc = string(BPF_FS_PATH) + lookupPinSubdir(pin_subdir, prefix) + "map_" +
Maciej Żenczykowski21869ef2022-07-07 06:01:57 -0700839 (md[i].shared ? "" : objName) + "_" + mapNames[i];
Maciej Żenczykowski36c53ba2021-07-05 15:20:34 -0700840 bool reuse = false;
841 unique_fd fd;
842 int saved_errno;
843
Joel Fernandesd76a2002018-10-16 13:19:58 -0700844 if (access(mapPinLoc.c_str(), F_OK) == 0) {
Maciej Żenczykowskieb199dd2022-07-01 03:21:40 -0700845 fd.reset(mapRetrieveRO(mapPinLoc.c_str()));
Maciej Żenczykowski2c372132021-03-01 23:09:54 -0800846 saved_errno = errno;
Maciej Żenczykowskie626a952022-06-17 02:35:36 -0700847 ALOGD("bpf_create_map reusing map %s, ret: %d", mapNames[i].c_str(), fd.get());
Joel Fernandesd76a2002018-10-16 13:19:58 -0700848 reuse = true;
849 } else {
Connor O'Brien35425e52022-01-18 21:41:16 -0800850 struct bpf_create_map_attr attr = {
851 .name = mapNames[i].c_str(),
852 .map_type = type,
853 .map_flags = md[i].map_flags,
854 .key_size = md[i].key_size,
855 .value_size = md[i].value_size,
856 .max_entries = md[i].max_entries,
857 };
858 if (btfFd.has_value() && btfTypeIdMap.find(mapNames[i]) != btfTypeIdMap.end()) {
859 attr.btf_fd = btfFd->get();
860 attr.btf_key_type_id = btfTypeIdMap.at(mapNames[i]).first;
861 attr.btf_value_type_id = btfTypeIdMap.at(mapNames[i]).second;
862 }
863 fd.reset(bcc_create_map_xattr(&attr, true));
Maciej Żenczykowski2c372132021-03-01 23:09:54 -0800864 saved_errno = errno;
Maciej Żenczykowskie626a952022-06-17 02:35:36 -0700865 ALOGD("bpf_create_map name %s, ret: %d", mapNames[i].c_str(), fd.get());
Joel Fernandesd76a2002018-10-16 13:19:58 -0700866 }
867
Maciej Żenczykowski12bb5202022-06-16 15:50:58 -0700868 if (!fd.ok()) return -saved_errno;
869
870 // When reusing a pinned map, we need to check the map type/sizes/etc match, but for
871 // safety (since reuse code path is rare) run these checks even if we just created it.
872 // We assume failure is due to pinned map mismatch, hence the 'NOT UNIQUE' return code.
873 if (!mapMatchesExpectations(fd, mapNames[i], md[i], type)) return -ENOTUNIQ;
Joel Fernandesd76a2002018-10-16 13:19:58 -0700874
875 if (!reuse) {
Maciej Żenczykowski41817132022-06-03 08:41:04 -0700876 if (specified(selinux_context)) {
877 string createLoc = string(BPF_FS_PATH) + lookupPinSubdir(selinux_context) +
Maciej Żenczykowski21869ef2022-07-07 06:01:57 -0700878 "tmp_map_" + objName + "_" + mapNames[i];
Maciej Żenczykowski41817132022-06-03 08:41:04 -0700879 ret = bpf_obj_pin(fd, createLoc.c_str());
880 if (ret) {
881 int err = errno;
882 ALOGE("create %s -> %d [%d:%s]", createLoc.c_str(), ret, err, strerror(err));
883 return -err;
884 }
885 ret = rename(createLoc.c_str(), mapPinLoc.c_str());
886 if (ret) {
887 int err = errno;
888 ALOGE("rename %s %s -> %d [%d:%s]", createLoc.c_str(), mapPinLoc.c_str(), ret,
889 err, strerror(err));
890 return -err;
891 }
892 } else {
893 ret = bpf_obj_pin(fd, mapPinLoc.c_str());
Maciej Żenczykowskid8259aa2022-06-27 10:02:20 -0700894 if (ret) {
895 int err = errno;
896 ALOGE("pin %s -> %d [%d:%s]", mapPinLoc.c_str(), ret, err, strerror(err));
897 return -err;
898 }
Maciej Żenczykowski41817132022-06-03 08:41:04 -0700899 }
Maciej Żenczykowski83f29772020-01-27 03:11:51 -0800900 ret = chmod(mapPinLoc.c_str(), md[i].mode);
Maciej Żenczykowski41817132022-06-03 08:41:04 -0700901 if (ret) {
902 int err = errno;
903 ALOGE("chmod(%s, 0%o) = %d [%d:%s]", mapPinLoc.c_str(), md[i].mode, ret, err,
904 strerror(err));
905 return -err;
906 }
Maciej Żenczykowski5e4aabf2022-06-27 01:15:53 -0700907 ret = chown(mapPinLoc.c_str(), (uid_t)md[i].uid, (gid_t)md[i].gid);
908 if (ret) {
909 int err = errno;
910 ALOGE("chown(%s, %u, %u) = %d [%d:%s]", mapPinLoc.c_str(), md[i].uid, md[i].gid,
911 ret, err, strerror(err));
912 return -err;
913 }
Joel Fernandesd76a2002018-10-16 13:19:58 -0700914 }
915
Maciej Żenczykowski57412c22022-05-20 16:44:06 -0700916 struct bpf_map_info map_info = {};
917 __u32 map_info_len = sizeof(map_info);
918 int rv = bpf_obj_get_info_by_fd(fd, &map_info, &map_info_len);
919 if (rv) {
Maciej Żenczykowskie626a952022-06-17 02:35:36 -0700920 ALOGE("bpf_obj_get_info_by_fd failed, ret: %d [%d]", rv, errno);
Maciej Żenczykowski57412c22022-05-20 16:44:06 -0700921 } else {
Maciej Żenczykowskie626a952022-06-17 02:35:36 -0700922 ALOGI("map %s id %d", mapPinLoc.c_str(), map_info.id);
Maciej Żenczykowski57412c22022-05-20 16:44:06 -0700923 }
924
Connor O'Brien8d49fc72019-10-24 18:23:49 -0700925 mapFds.push_back(std::move(fd));
Joel Fernandesd76a2002018-10-16 13:19:58 -0700926 }
927
928 return ret;
929}
930
931/* For debugging, dump all instructions */
932static void dumpIns(char* ins, int size) {
933 for (int row = 0; row < size / 8; row++) {
934 ALOGE("%d: ", row);
935 for (int j = 0; j < 8; j++) {
936 ALOGE("%3x ", ins[(row * 8) + j]);
937 }
938 ALOGE("\n");
939 }
940}
941
942/* For debugging, dump all code sections from cs list */
943static void dumpAllCs(vector<codeSection>& cs) {
944 for (int i = 0; i < (int)cs.size(); i++) {
Maciej Żenczykowskie626a952022-06-17 02:35:36 -0700945 ALOGE("Dumping cs %d, name %s", int(i), cs[i].name.c_str());
Joel Fernandesd76a2002018-10-16 13:19:58 -0700946 dumpIns((char*)cs[i].data.data(), cs[i].data.size());
Maciej Żenczykowskie626a952022-06-17 02:35:36 -0700947 ALOGE("-----------");
Joel Fernandesd76a2002018-10-16 13:19:58 -0700948 }
949}
950
951static void applyRelo(void* insnsPtr, Elf64_Addr offset, int fd) {
952 int insnIndex;
953 struct bpf_insn *insn, *insns;
954
955 insns = (struct bpf_insn*)(insnsPtr);
956
957 insnIndex = offset / sizeof(struct bpf_insn);
958 insn = &insns[insnIndex];
959
Maciej Żenczykowski509b1b92023-03-07 01:11:20 +0000960 // Occasionally might be useful for relocation debugging, but pretty spammy
961 if (0) {
962 ALOGD("applying relo to instruction at byte offset: %llu, "
963 "insn offset %d, insn %llx",
964 (unsigned long long)offset, insnIndex, *(unsigned long long*)insn);
965 }
Joel Fernandesd76a2002018-10-16 13:19:58 -0700966
967 if (insn->code != (BPF_LD | BPF_IMM | BPF_DW)) {
Maciej Żenczykowskie626a952022-06-17 02:35:36 -0700968 ALOGE("Dumping all instructions till ins %d", insnIndex);
969 ALOGE("invalid relo for insn %d: code 0x%x", insnIndex, insn->code);
Joel Fernandesd76a2002018-10-16 13:19:58 -0700970 dumpIns((char*)insnsPtr, (insnIndex + 3) * 8);
971 return;
972 }
973
974 insn->imm = fd;
975 insn->src_reg = BPF_PSEUDO_MAP_FD;
976}
977
Connor O'Brien8d49fc72019-10-24 18:23:49 -0700978static void applyMapRelo(ifstream& elfFile, vector<unique_fd> &mapFds, vector<codeSection>& cs) {
Joel Fernandesd76a2002018-10-16 13:19:58 -0700979 vector<string> mapNames;
980
Connor O'Brien3278a162020-02-13 21:45:22 -0800981 int ret = getSectionSymNames(elfFile, "maps", mapNames);
Joel Fernandesd76a2002018-10-16 13:19:58 -0700982 if (ret) return;
983
984 for (int k = 0; k != (int)cs.size(); k++) {
985 Elf64_Rel* rel = (Elf64_Rel*)(cs[k].rel_data.data());
986 int n_rel = cs[k].rel_data.size() / sizeof(*rel);
987
988 for (int i = 0; i < n_rel; i++) {
989 int symIndex = ELF64_R_SYM(rel[i].r_info);
990 string symName;
991
992 ret = getSymNameByIdx(elfFile, symIndex, symName);
993 if (ret) return;
994
995 /* Find the map fd and apply relo */
996 for (int j = 0; j < (int)mapNames.size(); j++) {
997 if (!mapNames[j].compare(symName)) {
998 applyRelo(cs[k].data.data(), rel[i].r_offset, mapFds[j]);
999 break;
1000 }
1001 }
1002 }
1003 }
1004}
1005
Maciej Żenczykowskid8a45782021-01-14 23:36:32 -08001006static int loadCodeSections(const char* elfPath, vector<codeSection>& cs, const string& license,
Maciej Żenczykowski41817132022-06-03 08:41:04 -07001007 const char* prefix, const unsigned long long allowedDomainBitmask) {
Maciej Żenczykowski07375e22020-02-19 14:23:59 -08001008 unsigned kvers = kernelVersion();
1009 int ret, fd;
Joel Fernandesd76a2002018-10-16 13:19:58 -07001010
Maciej Żenczykowski5c791652022-08-03 23:49:08 +00001011 if (!kvers) {
1012 ALOGE("unable to get kernel version");
1013 return -EINVAL;
1014 }
Joel Fernandesd76a2002018-10-16 13:19:58 -07001015
Maciej Żenczykowski21869ef2022-07-07 06:01:57 -07001016 string objName = pathToObjName(string(elfPath));
Joel Fernandesd76a2002018-10-16 13:19:58 -07001017
1018 for (int i = 0; i < (int)cs.size(); i++) {
Maciej Żenczykowski681f6042020-04-21 15:34:18 -07001019 string name = cs[i].name;
Joel Fernandesd76a2002018-10-16 13:19:58 -07001020
Maciej Żenczykowski5c791652022-08-03 23:49:08 +00001021 if (!cs[i].prog_def.has_value()) {
1022 ALOGE("[%d] '%s' missing program definition! bad bpf.o build?", i, name.c_str());
1023 return -EINVAL;
Maciej Żenczykowski07375e22020-02-19 14:23:59 -08001024 }
1025
Maciej Żenczykowski5c791652022-08-03 23:49:08 +00001026 unsigned min_kver = cs[i].prog_def->min_kver;
1027 unsigned max_kver = cs[i].prog_def->max_kver;
1028 ALOGD("cs[%d].name:%s min_kver:%x .max_kver:%x (kvers:%x)", i, name.c_str(), min_kver,
1029 max_kver, kvers);
1030 if (kvers < min_kver) continue;
1031 if (kvers >= max_kver) continue;
1032
1033 unsigned bpfMinVer = cs[i].prog_def->bpfloader_min_ver;
1034 unsigned bpfMaxVer = cs[i].prog_def->bpfloader_max_ver;
1035 domain selinux_context = getDomainFromSelinuxContext(cs[i].prog_def->selinux_context);
1036 domain pin_subdir = getDomainFromPinSubdir(cs[i].prog_def->pin_subdir);
1037 // Note: make sure to only check for unrecognized *after* verifying bpfloader
1038 // version limits include this bpfloader's version.
1039
Maciej Żenczykowskie626a952022-06-17 02:35:36 -07001040 ALOGD("cs[%d].name:%s requires bpfloader version [0x%05x,0x%05x)", i, name.c_str(),
Maciej Żenczykowski9217eee2021-03-03 05:28:52 -08001041 bpfMinVer, bpfMaxVer);
1042 if (BPFLOADER_VERSION < bpfMinVer) continue;
1043 if (BPFLOADER_VERSION >= bpfMaxVer) continue;
Ryan Zukliece89f502022-12-15 16:14:32 -08001044
1045 if ((cs[i].prog_def->ignore_on_eng && isEng()) ||
1046 (cs[i].prog_def->ignore_on_user && isUser()) ||
1047 (cs[i].prog_def->ignore_on_userdebug && isUserdebug())) {
1048 ALOGD("cs[%d].name:%s is ignored on %s builds", i, name.c_str(),
1049 getBuildType().c_str());
Ryan Zuklie0fb433a2023-01-05 14:22:15 -08001050 continue;
Ryan Zukliece89f502022-12-15 16:14:32 -08001051 }
1052
Maciej Żenczykowski41817132022-06-03 08:41:04 -07001053 if (unrecognized(pin_subdir)) return -ENOTDIR;
1054
1055 if (specified(selinux_context)) {
1056 if (!inDomainBitmask(selinux_context, allowedDomainBitmask)) {
1057 ALOGE("prog %s has invalid selinux_context of %d (allowed bitmask 0x%llx)",
1058 name.c_str(), selinux_context, allowedDomainBitmask);
1059 return -EINVAL;
1060 }
Maciej Żenczykowski99668462022-07-02 16:58:01 -07001061 ALOGI("prog %s selinux_context [%-32s] -> %d -> '%s' (%s)", name.c_str(),
Maciej Żenczykowski41817132022-06-03 08:41:04 -07001062 cs[i].prog_def->selinux_context, selinux_context,
1063 lookupSelinuxContext(selinux_context), lookupPinSubdir(selinux_context));
1064 }
1065
1066 if (specified(pin_subdir)) {
1067 if (!inDomainBitmask(pin_subdir, allowedDomainBitmask)) {
1068 ALOGE("prog %s has invalid pin_subdir of %d (allowed bitmask 0x%llx)", name.c_str(),
1069 pin_subdir, allowedDomainBitmask);
1070 return -EINVAL;
1071 }
Maciej Żenczykowski99668462022-07-02 16:58:01 -07001072 ALOGI("prog %s pin_subdir [%-32s] -> %d -> '%s'", name.c_str(),
Maciej Żenczykowski41817132022-06-03 08:41:04 -07001073 cs[i].prog_def->pin_subdir, pin_subdir, lookupPinSubdir(pin_subdir));
1074 }
Maciej Żenczykowski9217eee2021-03-03 05:28:52 -08001075
Maciej Żenczykowski681f6042020-04-21 15:34:18 -07001076 // strip any potential $foo suffix
1077 // this can be used to provide duplicate programs
1078 // conditionally loaded based on running kernel version
Maciej Żenczykowski428843d2020-04-23 12:43:44 -07001079 name = name.substr(0, name.find_last_of('$'));
Maciej Żenczykowski681f6042020-04-21 15:34:18 -07001080
1081 bool reuse = false;
Joel Fernandesd76a2002018-10-16 13:19:58 -07001082 // Format of pin location is
Maciej Żenczykowski21869ef2022-07-07 06:01:57 -07001083 // /sys/fs/bpf/<prefix>prog_<objName>_<progName>
Maciej Żenczykowski41817132022-06-03 08:41:04 -07001084 string progPinLoc = string(BPF_FS_PATH) + lookupPinSubdir(pin_subdir, prefix) + "prog_" +
Maciej Żenczykowski21869ef2022-07-07 06:01:57 -07001085 objName + '_' + string(name);
Joel Fernandesd76a2002018-10-16 13:19:58 -07001086 if (access(progPinLoc.c_str(), F_OK) == 0) {
Maciej Żenczykowskiaa295c82020-06-16 17:02:48 -07001087 fd = retrieveProgram(progPinLoc.c_str());
Maciej Żenczykowskie626a952022-06-17 02:35:36 -07001088 ALOGD("New bpf prog load reusing prog %s, ret: %d (%s)", progPinLoc.c_str(), fd,
Maciej Żenczykowskiaa295c82020-06-16 17:02:48 -07001089 (fd < 0 ? std::strerror(errno) : "no error"));
Joel Fernandesd76a2002018-10-16 13:19:58 -07001090 reuse = true;
1091 } else {
1092 vector<char> log_buf(BPF_LOAD_LOG_SZ, 0);
1093
Tyler Wear4e2f4602022-02-03 09:46:01 -08001094 struct bpf_load_program_attr attr = {
1095 .prog_type = cs[i].type,
1096 .name = name.c_str(),
1097 .insns = (struct bpf_insn*)cs[i].data.data(),
1098 .license = license.c_str(),
1099 .log_level = 0,
1100 .expected_attach_type = cs[i].expected_attach_type,
1101 };
1102
1103 fd = bcc_prog_load_xattr(&attr, cs[i].data.size(), log_buf.data(), log_buf.size(),
1104 true);
1105
Maciej Żenczykowskie626a952022-06-17 02:35:36 -07001106 ALOGD("bpf_prog_load lib call for %s (%s) returned fd: %d (%s)", elfPath,
Steven Moreland804bca02019-12-12 17:21:23 -08001107 cs[i].name.c_str(), fd, (fd < 0 ? std::strerror(errno) : "no error"));
Joel Fernandesd76a2002018-10-16 13:19:58 -07001108
Maciej Żenczykowski524deef2020-02-11 11:12:37 -08001109 if (fd < 0) {
Maciej Żenczykowskif7c0d992021-01-21 16:01:04 -08001110 vector<string> lines = android::base::Split(log_buf.data(), "\n");
Maciej Żenczykowski524deef2020-02-11 11:12:37 -08001111
Maciej Żenczykowskiaa295c82020-06-16 17:02:48 -07001112 ALOGW("bpf_prog_load - BEGIN log_buf contents:");
1113 for (const auto& line : lines) ALOGW("%s", line.c_str());
1114 ALOGW("bpf_prog_load - END log_buf contents.");
1115
Maciej Żenczykowski5c791652022-08-03 23:49:08 +00001116 if (cs[i].prog_def->optional) {
Maciej Żenczykowskiaa295c82020-06-16 17:02:48 -07001117 ALOGW("failed program is marked optional - continuing...");
1118 continue;
1119 }
1120 ALOGE("non-optional program failed to load.");
Maciej Żenczykowski524deef2020-02-11 11:12:37 -08001121 }
Joel Fernandesd76a2002018-10-16 13:19:58 -07001122 }
1123
1124 if (fd < 0) return fd;
1125 if (fd == 0) return -EINVAL;
1126
1127 if (!reuse) {
Maciej Żenczykowski41817132022-06-03 08:41:04 -07001128 if (specified(selinux_context)) {
1129 string createLoc = string(BPF_FS_PATH) + lookupPinSubdir(selinux_context) +
Maciej Żenczykowski21869ef2022-07-07 06:01:57 -07001130 "tmp_prog_" + objName + '_' + string(name);
Maciej Żenczykowski41817132022-06-03 08:41:04 -07001131 ret = bpf_obj_pin(fd, createLoc.c_str());
1132 if (ret) {
1133 int err = errno;
1134 ALOGE("create %s -> %d [%d:%s]", createLoc.c_str(), ret, err, strerror(err));
1135 return -err;
1136 }
1137 ret = rename(createLoc.c_str(), progPinLoc.c_str());
1138 if (ret) {
1139 int err = errno;
1140 ALOGE("rename %s %s -> %d [%d:%s]", createLoc.c_str(), progPinLoc.c_str(), ret,
1141 err, strerror(err));
1142 return -err;
1143 }
1144 } else {
1145 ret = bpf_obj_pin(fd, progPinLoc.c_str());
1146 if (ret) {
1147 int err = errno;
1148 ALOGE("create %s -> %d [%d:%s]", progPinLoc.c_str(), ret, err, strerror(err));
1149 return -err;
1150 }
1151 }
1152 if (chmod(progPinLoc.c_str(), 0440)) {
1153 int err = errno;
1154 ALOGE("chmod %s 0440 -> [%d:%s]", progPinLoc.c_str(), err, strerror(err));
1155 return -err;
1156 }
Maciej Żenczykowski5c791652022-08-03 23:49:08 +00001157 if (chown(progPinLoc.c_str(), (uid_t)cs[i].prog_def->uid,
1158 (gid_t)cs[i].prog_def->gid)) {
1159 int err = errno;
1160 ALOGE("chown %s %d %d -> [%d:%s]", progPinLoc.c_str(), cs[i].prog_def->uid,
1161 cs[i].prog_def->gid, err, strerror(err));
1162 return -err;
Connor O'Brien3278a162020-02-13 21:45:22 -08001163 }
Joel Fernandesd76a2002018-10-16 13:19:58 -07001164 }
1165
Maciej Żenczykowski57412c22022-05-20 16:44:06 -07001166 struct bpf_prog_info prog_info = {};
1167 __u32 prog_info_len = sizeof(prog_info);
1168 int rv = bpf_obj_get_info_by_fd(fd, &prog_info, &prog_info_len);
1169 if (rv) {
Maciej Żenczykowskie626a952022-06-17 02:35:36 -07001170 ALOGE("bpf_obj_get_info_by_fd failed, ret: %d [%d]", rv, errno);
Maciej Żenczykowski57412c22022-05-20 16:44:06 -07001171 } else {
Maciej Żenczykowskie626a952022-06-17 02:35:36 -07001172 ALOGI("prog %s id %d", progPinLoc.c_str(), prog_info.id);
Maciej Żenczykowski57412c22022-05-20 16:44:06 -07001173 }
1174
Connor O'Brien8d49fc72019-10-24 18:23:49 -07001175 cs[i].prog_fd.reset(fd);
Joel Fernandesd76a2002018-10-16 13:19:58 -07001176 }
1177
1178 return 0;
1179}
1180
Connor O'Brien6c0ce9f2022-12-01 20:08:17 -08001181int loadProg(const char* elfPath, bool* isCritical, const Location& location) {
Joel Fernandesd76a2002018-10-16 13:19:58 -07001182 vector<char> license;
Maciej Żenczykowski4ba8c1c2020-06-10 15:49:31 -07001183 vector<char> critical;
Joel Fernandesd76a2002018-10-16 13:19:58 -07001184 vector<codeSection> cs;
Connor O'Brien8d49fc72019-10-24 18:23:49 -07001185 vector<unique_fd> mapFds;
Joel Fernandesd76a2002018-10-16 13:19:58 -07001186 int ret;
1187
Maciej Żenczykowski89515d92020-06-14 19:27:33 -07001188 if (!isCritical) return -1;
1189 *isCritical = false;
1190
Joel Fernandesd76a2002018-10-16 13:19:58 -07001191 ifstream elfFile(elfPath, ios::in | ios::binary);
1192 if (!elfFile.is_open()) return -1;
1193
Maciej Żenczykowski4ba8c1c2020-06-10 15:49:31 -07001194 ret = readSectionByName("critical", elfFile, critical);
Maciej Żenczykowski89515d92020-06-14 19:27:33 -07001195 *isCritical = !ret;
Maciej Żenczykowski4ba8c1c2020-06-10 15:49:31 -07001196
Joel Fernandesd76a2002018-10-16 13:19:58 -07001197 ret = readSectionByName("license", elfFile, license);
1198 if (ret) {
Maciej Żenczykowskie626a952022-06-17 02:35:36 -07001199 ALOGE("Couldn't find license in %s", elfPath);
Joel Fernandesd76a2002018-10-16 13:19:58 -07001200 return ret;
1201 } else {
Maciej Żenczykowskie626a952022-06-17 02:35:36 -07001202 ALOGD("Loading %s%s ELF object %s with license %s",
Maciej Żenczykowski89515d92020-06-14 19:27:33 -07001203 *isCritical ? "critical for " : "optional", *isCritical ? (char*)critical.data() : "",
Maciej Żenczykowski4ba8c1c2020-06-10 15:49:31 -07001204 elfPath, (char*)license.data());
Joel Fernandesd76a2002018-10-16 13:19:58 -07001205 }
1206
Maciej Żenczykowski9217eee2021-03-03 05:28:52 -08001207 // the following default values are for bpfloader V0.0 format which does not include them
1208 unsigned int bpfLoaderMinVer =
1209 readSectionUint("bpfloader_min_ver", elfFile, DEFAULT_BPFLOADER_MIN_VER);
1210 unsigned int bpfLoaderMaxVer =
1211 readSectionUint("bpfloader_max_ver", elfFile, DEFAULT_BPFLOADER_MAX_VER);
Maciej Żenczykowskib57290a2022-07-02 15:47:07 -07001212 unsigned int bpfLoaderMinRequiredVer =
1213 readSectionUint("bpfloader_min_required_ver", elfFile, 0);
Maciej Żenczykowski9217eee2021-03-03 05:28:52 -08001214 size_t sizeOfBpfMapDef =
1215 readSectionUint("size_of_bpf_map_def", elfFile, DEFAULT_SIZEOF_BPF_MAP_DEF);
1216 size_t sizeOfBpfProgDef =
1217 readSectionUint("size_of_bpf_prog_def", elfFile, DEFAULT_SIZEOF_BPF_PROG_DEF);
1218
1219 // inclusive lower bound check
1220 if (BPFLOADER_VERSION < bpfLoaderMinVer) {
Maciej Żenczykowskie626a952022-06-17 02:35:36 -07001221 ALOGI("BpfLoader version 0x%05x ignoring ELF object %s with min ver 0x%05x",
Maciej Żenczykowski9217eee2021-03-03 05:28:52 -08001222 BPFLOADER_VERSION, elfPath, bpfLoaderMinVer);
1223 return 0;
1224 }
1225
1226 // exclusive upper bound check
1227 if (BPFLOADER_VERSION >= bpfLoaderMaxVer) {
Maciej Żenczykowskie626a952022-06-17 02:35:36 -07001228 ALOGI("BpfLoader version 0x%05x ignoring ELF object %s with max ver 0x%05x",
Maciej Żenczykowski9217eee2021-03-03 05:28:52 -08001229 BPFLOADER_VERSION, elfPath, bpfLoaderMaxVer);
1230 return 0;
1231 }
1232
Maciej Żenczykowskib57290a2022-07-02 15:47:07 -07001233 if (BPFLOADER_VERSION < bpfLoaderMinRequiredVer) {
1234 ALOGI("BpfLoader version 0x%05x failing due to ELF object %s with required min ver 0x%05x",
1235 BPFLOADER_VERSION, elfPath, bpfLoaderMinRequiredVer);
1236 return -1;
1237 }
1238
Maciej Żenczykowskie626a952022-06-17 02:35:36 -07001239 ALOGI("BpfLoader version 0x%05x processing ELF object %s with ver [0x%05x,0x%05x)",
Maciej Żenczykowski9217eee2021-03-03 05:28:52 -08001240 BPFLOADER_VERSION, elfPath, bpfLoaderMinVer, bpfLoaderMaxVer);
1241
1242 if (sizeOfBpfMapDef < DEFAULT_SIZEOF_BPF_MAP_DEF) {
Maciej Żenczykowskie626a952022-06-17 02:35:36 -07001243 ALOGE("sizeof(bpf_map_def) of %zu is too small (< %d)", sizeOfBpfMapDef,
Maciej Żenczykowski9217eee2021-03-03 05:28:52 -08001244 DEFAULT_SIZEOF_BPF_MAP_DEF);
1245 return -1;
1246 }
1247
1248 if (sizeOfBpfProgDef < DEFAULT_SIZEOF_BPF_PROG_DEF) {
Maciej Żenczykowskie626a952022-06-17 02:35:36 -07001249 ALOGE("sizeof(bpf_prog_def) of %zu is too small (< %d)", sizeOfBpfProgDef,
Maciej Żenczykowski9217eee2021-03-03 05:28:52 -08001250 DEFAULT_SIZEOF_BPF_PROG_DEF);
1251 return -1;
1252 }
1253
Connor O'Brien6c0ce9f2022-12-01 20:08:17 -08001254 ret = readCodeSections(elfFile, cs, sizeOfBpfProgDef, location.allowedProgTypes,
1255 location.allowedProgTypesLength);
Joel Fernandesd76a2002018-10-16 13:19:58 -07001256 if (ret) {
Maciej Żenczykowskie626a952022-06-17 02:35:36 -07001257 ALOGE("Couldn't read all code sections in %s", elfPath);
Joel Fernandesd76a2002018-10-16 13:19:58 -07001258 return ret;
1259 }
1260
1261 /* Just for future debugging */
1262 if (0) dumpAllCs(cs);
1263
Connor O'Brien6c0ce9f2022-12-01 20:08:17 -08001264 ret = createMaps(elfPath, elfFile, mapFds, location.prefix, location.allowedDomainBitmask,
1265 sizeOfBpfMapDef);
Joel Fernandesd76a2002018-10-16 13:19:58 -07001266 if (ret) {
Maciej Żenczykowskie626a952022-06-17 02:35:36 -07001267 ALOGE("Failed to create maps: (ret=%d) in %s", ret, elfPath);
Joel Fernandesd76a2002018-10-16 13:19:58 -07001268 return ret;
1269 }
1270
1271 for (int i = 0; i < (int)mapFds.size(); i++)
Maciej Żenczykowskie626a952022-06-17 02:35:36 -07001272 ALOGD("map_fd found at %d is %d in %s", i, mapFds[i].get(), elfPath);
Joel Fernandesd76a2002018-10-16 13:19:58 -07001273
1274 applyMapRelo(elfFile, mapFds, cs);
1275
Connor O'Brien6c0ce9f2022-12-01 20:08:17 -08001276 ret = loadCodeSections(elfPath, cs, string(license.data()), location.prefix,
1277 location.allowedDomainBitmask);
Maciej Żenczykowskie626a952022-06-17 02:35:36 -07001278 if (ret) ALOGE("Failed to load programs, loadCodeSections ret=%d", ret);
Joel Fernandesd76a2002018-10-16 13:19:58 -07001279
1280 return ret;
1281}
1282
Joel Fernandesd76a2002018-10-16 13:19:58 -07001283} // namespace bpf
1284} // namespace android