diff options
author | Dmitri Plotnikov <dplotnikov@google.com> | 2020-10-22 17:12:44 -0700 |
---|---|---|
committer | Dmitri Plotnikov <dplotnikov@google.com> | 2021-07-15 17:06:15 -0700 |
commit | bb05a5c138fedf0dc6f138882e3cb8a341a0db39 (patch) | |
tree | 364203d55bef7e181e91e0b80c235be955be52af /progs | |
parent | 7ed94efeda510ced18b3c8713b046ae2fea45c1b (diff) | |
download | bpf-bb05a5c138fedf0dc6f138882e3cb8a341a0db39.tar.gz |
Add support for testing of eBPF programs
Bug: 171262763
Test: this is just a header file
Change-Id: Id25846f91d2da698717a9cc0d8d9091022dd7233
Diffstat (limited to 'progs')
-rw-r--r-- | progs/include/test/mock_bpf_helpers.h | 92 |
1 files changed, 92 insertions, 0 deletions
diff --git a/progs/include/test/mock_bpf_helpers.h b/progs/include/test/mock_bpf_helpers.h new file mode 100644 index 0000000..ffe8951 --- /dev/null +++ b/progs/include/test/mock_bpf_helpers.h @@ -0,0 +1,92 @@ +/* + * Copyright (C) 2021 The Android Open Source Project + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +#pragma once + +/* Mock BPF helpers to be used for testing of BPF programs loaded by Android */ + +#include <linux/bpf.h> +#include <stdbool.h> +#include <stdint.h> + +typedef void* mock_bpf_map_t; + +/* type safe macro to declare a map and related accessor functions */ +#define DEFINE_BPF_MAP_UGM(the_map, TYPE, TypeOfKey, TypeOfValue, num_entries, usr, grp, md) \ + mock_bpf_map_t mock_bpf_map_##the_map; \ + \ + mock_bpf_map_t get_mock_bpf_map_##the_map() { \ + if (mock_bpf_map_##the_map == 0) { \ + mock_bpf_map_##the_map = mock_bpf_map_create(sizeof(TypeOfKey), sizeof(TypeOfValue), \ + BPF_MAP_TYPE_##TYPE); \ + } \ + return mock_bpf_map_##the_map; \ + } \ + __unused TypeOfValue* bpf_##the_map##_lookup_elem(const TypeOfKey* k) { \ + return (TypeOfValue*)mock_bpf_lookup_elem(get_mock_bpf_map_##the_map(), (void*)k); \ + }; \ + \ + __unused int bpf_##the_map##_update_elem(const TypeOfKey* k, const TypeOfValue* v, \ + uint64_t flags) { \ + return mock_bpf_update_elem(get_mock_bpf_map_##the_map(), (void*)k, (void*)v, flags); \ + }; \ + \ + __unused int bpf_##the_map##_delete_elem(const TypeOfKey* k) { \ + return mock_bpf_delete_elem(get_mock_bpf_map_##the_map(), (void*)k); \ + }; + +#define DEFINE_BPF_MAP(the_map, TYPE, TypeOfKey, TypeOfValue, num_entries) \ + DEFINE_BPF_MAP_UGM(the_map, TYPE, TypeOfKey, TypeOfValue, num_entries, AID_ROOT, AID_ROOT, 0600) + +#define DEFINE_BPF_MAP_GWO(the_map, TYPE, TypeOfKey, TypeOfValue, num_entries, gid) \ + DEFINE_BPF_MAP_UGM(the_map, TYPE, TypeOfKey, TypeOfValue, num_entries, AID_ROOT, gid, 0620) + +#define DEFINE_BPF_MAP_GRO(the_map, TYPE, TypeOfKey, TypeOfValue, num_entries, gid) \ + DEFINE_BPF_MAP_UGM(the_map, TYPE, TypeOfKey, TypeOfValue, num_entries, AID_ROOT, gid, 0640) + +#define DEFINE_BPF_MAP_GRW(the_map, TYPE, TypeOfKey, TypeOfValue, num_entries, gid) \ + DEFINE_BPF_MAP_UGM(the_map, TYPE, TypeOfKey, TypeOfValue, num_entries, AID_ROOT, gid, 0660) + +#define DEFINE_BPF_PROG(section, owner, group, name) int name + +#ifdef __cplusplus +extern "C" { +#endif + +mock_bpf_map_t mock_bpf_map_create(uint32_t key_size, uint32_t value_size, uint32_t type); +void* mock_bpf_lookup_elem(mock_bpf_map_t map, void* key); +int mock_bpf_update_elem(mock_bpf_map_t map, void* key, void* value, uint64_t flags); +int mock_bpf_delete_elem(mock_bpf_map_t map, void* key); + +uint64_t bpf_ktime_get_ns(); +uint64_t bpf_get_smp_processor_id(); +uint64_t bpf_get_current_uid_gid(); +uint64_t bpf_get_current_pid_tgid(); + +void mock_bpf_set_ktime_ns(uint64_t time_ns); +void mock_bpf_set_smp_processor_id(uint32_t cpu); +void mock_bpf_set_current_uid_gid(uint32_t uid); +void mock_bpf_set_current_pid_tgid(uint64_t pid_tgid); + +#ifdef __cplusplus +} // extern "C" +#endif + +/* place things in different elf sections */ +#define SEC(NAME) __attribute__((section(NAME), used)) + +/* Example use: LICENSE("GPL"); or LICENSE("Apache 2.0"); */ +#define LICENSE(NAME) char _license[] SEC("license") = (NAME) |