1 /* 2 * Copyright (c) 2018 Politecnico di Torino 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 #include "BPF.h" 18 #include "catch.hpp" 19 20 const std::string BPF_PROGRAM1 = R"( 21 BPF_TABLE_SHARED("array", int, int, mysharedtable, 1024); 22 )"; 23 24 const std::string BPF_PROGRAM2 = R"( 25 BPF_TABLE("extern", int, int, mysharedtable, 1024); 26 )"; 27 28 TEST_CASE("test shared table", "[shared_table]") { 29 // deploy 4 ebpf programs: _ns1_a and _ns1_b are in ns1, _ns2_a and _ns2_b in ns2 30 ebpf::BPF bpf_ns1_a(0, nullptr, false, "ns1"); 31 ebpf::BPF bpf_ns1_b(0, nullptr, false, "ns1"); 32 ebpf::BPF bpf_ns2_a(0, nullptr, false, "ns2"); 33 ebpf::BPF bpf_ns2_b(0, nullptr, false, "ns2"); 34 35 ebpf::StatusTuple res(0); 36 37 res = bpf_ns1_a.init(BPF_PROGRAM1); 38 REQUIRE(res.code() == 0); 39 40 res = bpf_ns1_b.init(BPF_PROGRAM2); 41 REQUIRE(res.code() == 0); 42 43 res = bpf_ns2_a.init(BPF_PROGRAM1); 44 REQUIRE(res.code() == 0); 45 46 res = bpf_ns2_b.init(BPF_PROGRAM2); 47 REQUIRE(res.code() == 0); 48 49 // get references to all tables 50 ebpf::BPFArrayTable<int> t_ns1_a = bpf_ns1_a.get_array_table<int>("mysharedtable"); 51 ebpf::BPFArrayTable<int> t_ns1_b = bpf_ns1_b.get_array_table<int>("mysharedtable"); 52 ebpf::BPFArrayTable<int> t_ns2_a = bpf_ns2_a.get_array_table<int>("mysharedtable"); 53 ebpf::BPFArrayTable<int> t_ns2_b = bpf_ns2_b.get_array_table<int>("mysharedtable"); 54 55 // test that tables within the same ns are shared 56 int v1, v2, v3; 57 res = t_ns1_a.update_value(13, 42); 58 REQUIRE(res.code() == 0); 59 60 res = t_ns1_b.get_value(13, v1); 61 REQUIRE(res.code() == 0); 62 REQUIRE(v1 == 42); 63 64 // test that tables are isolated within different ns 65 res = t_ns2_a.update_value(13, 69); 66 REQUIRE(res.code() == 0); 67 68 res = t_ns2_b.get_value(13, v2); 69 REQUIRE(res.code() == 0); 70 REQUIRE(v2 == 69); 71 72 res = t_ns1_b.get_value(13, v3); 73 REQUIRE(res.code() == 0); 74 REQUIRE(v3 == 42); // value should still be 42 75 } 76
