| // SPDX-License-Identifier: GPL-2.0 |
| /* Copyright (c) 2020 Facebook */ |
| #include <test_progs.h> |
| #include <bpf/btf.h> |
| #include "btf_helpers.h" |
| |
| static void test_split_simple() { |
| const struct btf_type *t; |
| struct btf *btf1, *btf2; |
| int str_off, err; |
| |
| btf1 = btf__new_empty(); |
| if (!ASSERT_OK_PTR(btf1, "empty_main_btf")) |
| return; |
| |
| btf__set_pointer_size(btf1, 8); /* enforce 64-bit arch */ |
| |
| btf__add_int(btf1, "int", 4, BTF_INT_SIGNED); /* [1] int */ |
| btf__add_ptr(btf1, 1); /* [2] ptr to int */ |
| btf__add_struct(btf1, "s1", 4); /* [3] struct s1 { */ |
| btf__add_field(btf1, "f1", 1, 0, 0); /* int f1; */ |
| /* } */ |
| |
| VALIDATE_RAW_BTF( |
| btf1, |
| "[1] INT 'int' size=4 bits_offset=0 nr_bits=32 encoding=SIGNED", |
| "[2] PTR '(anon)' type_id=1", |
| "[3] STRUCT 's1' size=4 vlen=1\n" |
| "\t'f1' type_id=1 bits_offset=0"); |
| |
| ASSERT_STREQ(btf_type_c_dump(btf1), "\ |
| struct s1 {\n\ |
| int f1;\n\ |
| };\n\n", "c_dump"); |
| |
| btf2 = btf__new_empty_split(btf1); |
| if (!ASSERT_OK_PTR(btf2, "empty_split_btf")) |
| goto cleanup; |
| |
| /* pointer size should be "inherited" from main BTF */ |
| ASSERT_EQ(btf__pointer_size(btf2), 8, "inherit_ptr_sz"); |
| |
| str_off = btf__find_str(btf2, "int"); |
| ASSERT_NEQ(str_off, -ENOENT, "str_int_missing"); |
| |
| t = btf__type_by_id(btf2, 1); |
| if (!ASSERT_OK_PTR(t, "int_type")) |
| goto cleanup; |
| ASSERT_EQ(btf_is_int(t), true, "int_kind"); |
| ASSERT_STREQ(btf__str_by_offset(btf2, t->name_off), "int", "int_name"); |
| |
| btf__add_struct(btf2, "s2", 16); /* [4] struct s2 { */ |
| btf__add_field(btf2, "f1", 6, 0, 0); /* struct s1 f1; */ |
| btf__add_field(btf2, "f2", 5, 32, 0); /* int f2; */ |
| btf__add_field(btf2, "f3", 2, 64, 0); /* int *f3; */ |
| /* } */ |
| |
| /* duplicated int */ |
| btf__add_int(btf2, "int", 4, BTF_INT_SIGNED); /* [5] int */ |
| |
| /* duplicated struct s1 */ |
| btf__add_struct(btf2, "s1", 4); /* [6] struct s1 { */ |
| btf__add_field(btf2, "f1", 5, 0, 0); /* int f1; */ |
| /* } */ |
| |
| VALIDATE_RAW_BTF( |
| btf2, |
| "[1] INT 'int' size=4 bits_offset=0 nr_bits=32 encoding=SIGNED", |
| "[2] PTR '(anon)' type_id=1", |
| "[3] STRUCT 's1' size=4 vlen=1\n" |
| "\t'f1' type_id=1 bits_offset=0", |
| "[4] STRUCT 's2' size=16 vlen=3\n" |
| "\t'f1' type_id=6 bits_offset=0\n" |
| "\t'f2' type_id=5 bits_offset=32\n" |
| "\t'f3' type_id=2 bits_offset=64", |
| "[5] INT 'int' size=4 bits_offset=0 nr_bits=32 encoding=SIGNED", |
| "[6] STRUCT 's1' size=4 vlen=1\n" |
| "\t'f1' type_id=5 bits_offset=0"); |
| |
| ASSERT_STREQ(btf_type_c_dump(btf2), "\ |
| struct s1 {\n\ |
| int f1;\n\ |
| };\n\ |
| \n\ |
| struct s1___2 {\n\ |
| int f1;\n\ |
| };\n\ |
| \n\ |
| struct s2 {\n\ |
| struct s1___2 f1;\n\ |
| int f2;\n\ |
| int *f3;\n\ |
| };\n\n", "c_dump"); |
| |
| err = btf__dedup(btf2, NULL); |
| if (!ASSERT_OK(err, "btf_dedup")) |
| goto cleanup; |
| |
| VALIDATE_RAW_BTF( |
| btf2, |
| "[1] INT 'int' size=4 bits_offset=0 nr_bits=32 encoding=SIGNED", |
| "[2] PTR '(anon)' type_id=1", |
| "[3] STRUCT 's1' size=4 vlen=1\n" |
| "\t'f1' type_id=1 bits_offset=0", |
| "[4] STRUCT 's2' size=16 vlen=3\n" |
| "\t'f1' type_id=3 bits_offset=0\n" |
| "\t'f2' type_id=1 bits_offset=32\n" |
| "\t'f3' type_id=2 bits_offset=64"); |
| |
| ASSERT_STREQ(btf_type_c_dump(btf2), "\ |
| struct s1 {\n\ |
| int f1;\n\ |
| };\n\ |
| \n\ |
| struct s2 {\n\ |
| struct s1 f1;\n\ |
| int f2;\n\ |
| int *f3;\n\ |
| };\n\n", "c_dump"); |
| |
| cleanup: |
| btf__free(btf2); |
| btf__free(btf1); |
| } |
| |
| static void test_split_fwd_resolve() { |
| struct btf *btf1, *btf2; |
| int err; |
| |
| btf1 = btf__new_empty(); |
| if (!ASSERT_OK_PTR(btf1, "empty_main_btf")) |
| return; |
| |
| btf__set_pointer_size(btf1, 8); /* enforce 64-bit arch */ |
| |
| btf__add_int(btf1, "int", 4, BTF_INT_SIGNED); /* [1] int */ |
| btf__add_ptr(btf1, 4); /* [2] ptr to struct s1 */ |
| btf__add_ptr(btf1, 5); /* [3] ptr to struct s2 */ |
| btf__add_struct(btf1, "s1", 16); /* [4] struct s1 { */ |
| btf__add_field(btf1, "f1", 2, 0, 0); /* struct s1 *f1; */ |
| btf__add_field(btf1, "f2", 3, 64, 0); /* struct s2 *f2; */ |
| /* } */ |
| btf__add_struct(btf1, "s2", 4); /* [5] struct s2 { */ |
| btf__add_field(btf1, "f1", 1, 0, 0); /* int f1; */ |
| /* } */ |
| /* keep this not a part of type the graph to test btf_dedup_resolve_fwds */ |
| btf__add_struct(btf1, "s3", 4); /* [6] struct s3 { */ |
| btf__add_field(btf1, "f1", 1, 0, 0); /* int f1; */ |
| /* } */ |
| |
| VALIDATE_RAW_BTF( |
| btf1, |
| "[1] INT 'int' size=4 bits_offset=0 nr_bits=32 encoding=SIGNED", |
| "[2] PTR '(anon)' type_id=4", |
| "[3] PTR '(anon)' type_id=5", |
| "[4] STRUCT 's1' size=16 vlen=2\n" |
| "\t'f1' type_id=2 bits_offset=0\n" |
| "\t'f2' type_id=3 bits_offset=64", |
| "[5] STRUCT 's2' size=4 vlen=1\n" |
| "\t'f1' type_id=1 bits_offset=0", |
| "[6] STRUCT 's3' size=4 vlen=1\n" |
| "\t'f1' type_id=1 bits_offset=0"); |
| |
| btf2 = btf__new_empty_split(btf1); |
| if (!ASSERT_OK_PTR(btf2, "empty_split_btf")) |
| goto cleanup; |
| |
| btf__add_int(btf2, "int", 4, BTF_INT_SIGNED); /* [7] int */ |
| btf__add_ptr(btf2, 11); /* [8] ptr to struct s1 */ |
| btf__add_fwd(btf2, "s2", BTF_FWD_STRUCT); /* [9] fwd for struct s2 */ |
| btf__add_ptr(btf2, 9); /* [10] ptr to fwd struct s2 */ |
| btf__add_struct(btf2, "s1", 16); /* [11] struct s1 { */ |
| btf__add_field(btf2, "f1", 8, 0, 0); /* struct s1 *f1; */ |
| btf__add_field(btf2, "f2", 10, 64, 0); /* struct s2 *f2; */ |
| /* } */ |
| btf__add_fwd(btf2, "s3", BTF_FWD_STRUCT); /* [12] fwd for struct s3 */ |
| btf__add_ptr(btf2, 12); /* [13] ptr to struct s1 */ |
| |
| VALIDATE_RAW_BTF( |
| btf2, |
| "[1] INT 'int' size=4 bits_offset=0 nr_bits=32 encoding=SIGNED", |
| "[2] PTR '(anon)' type_id=4", |
| "[3] PTR '(anon)' type_id=5", |
| "[4] STRUCT 's1' size=16 vlen=2\n" |
| "\t'f1' type_id=2 bits_offset=0\n" |
| "\t'f2' type_id=3 bits_offset=64", |
| "[5] STRUCT 's2' size=4 vlen=1\n" |
| "\t'f1' type_id=1 bits_offset=0", |
| "[6] STRUCT 's3' size=4 vlen=1\n" |
| "\t'f1' type_id=1 bits_offset=0", |
| "[7] INT 'int' size=4 bits_offset=0 nr_bits=32 encoding=SIGNED", |
| "[8] PTR '(anon)' type_id=11", |
| "[9] FWD 's2' fwd_kind=struct", |
| "[10] PTR '(anon)' type_id=9", |
| "[11] STRUCT 's1' size=16 vlen=2\n" |
| "\t'f1' type_id=8 bits_offset=0\n" |
| "\t'f2' type_id=10 bits_offset=64", |
| "[12] FWD 's3' fwd_kind=struct", |
| "[13] PTR '(anon)' type_id=12"); |
| |
| err = btf__dedup(btf2, NULL); |
| if (!ASSERT_OK(err, "btf_dedup")) |
| goto cleanup; |
| |
| VALIDATE_RAW_BTF( |
| btf2, |
| "[1] INT 'int' size=4 bits_offset=0 nr_bits=32 encoding=SIGNED", |
| "[2] PTR '(anon)' type_id=4", |
| "[3] PTR '(anon)' type_id=5", |
| "[4] STRUCT 's1' size=16 vlen=2\n" |
| "\t'f1' type_id=2 bits_offset=0\n" |
| "\t'f2' type_id=3 bits_offset=64", |
| "[5] STRUCT 's2' size=4 vlen=1\n" |
| "\t'f1' type_id=1 bits_offset=0", |
| "[6] STRUCT 's3' size=4 vlen=1\n" |
| "\t'f1' type_id=1 bits_offset=0", |
| "[7] PTR '(anon)' type_id=6"); |
| |
| cleanup: |
| btf__free(btf2); |
| btf__free(btf1); |
| } |
| |
| static void test_split_struct_duped() { |
| struct btf *btf1, *btf2; |
| int err; |
| |
| btf1 = btf__new_empty(); |
| if (!ASSERT_OK_PTR(btf1, "empty_main_btf")) |
| return; |
| |
| btf__set_pointer_size(btf1, 8); /* enforce 64-bit arch */ |
| |
| btf__add_int(btf1, "int", 4, BTF_INT_SIGNED); /* [1] int */ |
| btf__add_ptr(btf1, 5); /* [2] ptr to struct s1 */ |
| btf__add_fwd(btf1, "s2", BTF_FWD_STRUCT); /* [3] fwd for struct s2 */ |
| btf__add_ptr(btf1, 3); /* [4] ptr to fwd struct s2 */ |
| btf__add_struct(btf1, "s1", 16); /* [5] struct s1 { */ |
| btf__add_field(btf1, "f1", 2, 0, 0); /* struct s1 *f1; */ |
| btf__add_field(btf1, "f2", 4, 64, 0); /* struct s2 *f2; */ |
| /* } */ |
| |
| VALIDATE_RAW_BTF( |
| btf1, |
| "[1] INT 'int' size=4 bits_offset=0 nr_bits=32 encoding=SIGNED", |
| "[2] PTR '(anon)' type_id=5", |
| "[3] FWD 's2' fwd_kind=struct", |
| "[4] PTR '(anon)' type_id=3", |
| "[5] STRUCT 's1' size=16 vlen=2\n" |
| "\t'f1' type_id=2 bits_offset=0\n" |
| "\t'f2' type_id=4 bits_offset=64"); |
| |
| btf2 = btf__new_empty_split(btf1); |
| if (!ASSERT_OK_PTR(btf2, "empty_split_btf")) |
| goto cleanup; |
| |
| btf__add_int(btf2, "int", 4, BTF_INT_SIGNED); /* [6] int */ |
| btf__add_ptr(btf2, 10); /* [7] ptr to struct s1 */ |
| btf__add_fwd(btf2, "s2", BTF_FWD_STRUCT); /* [8] fwd for struct s2 */ |
| btf__add_ptr(btf2, 11); /* [9] ptr to struct s2 */ |
| btf__add_struct(btf2, "s1", 16); /* [10] struct s1 { */ |
| btf__add_field(btf2, "f1", 7, 0, 0); /* struct s1 *f1; */ |
| btf__add_field(btf2, "f2", 9, 64, 0); /* struct s2 *f2; */ |
| /* } */ |
| btf__add_struct(btf2, "s2", 40); /* [11] struct s2 { */ |
| btf__add_field(btf2, "f1", 7, 0, 0); /* struct s1 *f1; */ |
| btf__add_field(btf2, "f2", 9, 64, 0); /* struct s2 *f2; */ |
| btf__add_field(btf2, "f3", 6, 128, 0); /* int f3; */ |
| btf__add_field(btf2, "f4", 10, 192, 0); /* struct s1 f4; */ |
| /* } */ |
| btf__add_ptr(btf2, 8); /* [12] ptr to fwd struct s2 */ |
| btf__add_struct(btf2, "s3", 8); /* [13] struct s3 { */ |
| btf__add_field(btf2, "f1", 12, 0, 0); /* struct s2 *f1; (fwd) */ |
| /* } */ |
| |
| VALIDATE_RAW_BTF( |
| btf2, |
| "[1] INT 'int' size=4 bits_offset=0 nr_bits=32 encoding=SIGNED", |
| "[2] PTR '(anon)' type_id=5", |
| "[3] FWD 's2' fwd_kind=struct", |
| "[4] PTR '(anon)' type_id=3", |
| "[5] STRUCT 's1' size=16 vlen=2\n" |
| "\t'f1' type_id=2 bits_offset=0\n" |
| "\t'f2' type_id=4 bits_offset=64", |
| "[6] INT 'int' size=4 bits_offset=0 nr_bits=32 encoding=SIGNED", |
| "[7] PTR '(anon)' type_id=10", |
| "[8] FWD 's2' fwd_kind=struct", |
| "[9] PTR '(anon)' type_id=11", |
| "[10] STRUCT 's1' size=16 vlen=2\n" |
| "\t'f1' type_id=7 bits_offset=0\n" |
| "\t'f2' type_id=9 bits_offset=64", |
| "[11] STRUCT 's2' size=40 vlen=4\n" |
| "\t'f1' type_id=7 bits_offset=0\n" |
| "\t'f2' type_id=9 bits_offset=64\n" |
| "\t'f3' type_id=6 bits_offset=128\n" |
| "\t'f4' type_id=10 bits_offset=192", |
| "[12] PTR '(anon)' type_id=8", |
| "[13] STRUCT 's3' size=8 vlen=1\n" |
| "\t'f1' type_id=12 bits_offset=0"); |
| |
| err = btf__dedup(btf2, NULL); |
| if (!ASSERT_OK(err, "btf_dedup")) |
| goto cleanup; |
| |
| VALIDATE_RAW_BTF( |
| btf2, |
| "[1] INT 'int' size=4 bits_offset=0 nr_bits=32 encoding=SIGNED", |
| "[2] PTR '(anon)' type_id=5", |
| "[3] FWD 's2' fwd_kind=struct", |
| "[4] PTR '(anon)' type_id=3", |
| "[5] STRUCT 's1' size=16 vlen=2\n" |
| "\t'f1' type_id=2 bits_offset=0\n" |
| "\t'f2' type_id=4 bits_offset=64", |
| "[6] PTR '(anon)' type_id=8", |
| "[7] PTR '(anon)' type_id=9", |
| "[8] STRUCT 's1' size=16 vlen=2\n" |
| "\t'f1' type_id=6 bits_offset=0\n" |
| "\t'f2' type_id=7 bits_offset=64", |
| "[9] STRUCT 's2' size=40 vlen=4\n" |
| "\t'f1' type_id=6 bits_offset=0\n" |
| "\t'f2' type_id=7 bits_offset=64\n" |
| "\t'f3' type_id=1 bits_offset=128\n" |
| "\t'f4' type_id=8 bits_offset=192", |
| "[10] STRUCT 's3' size=8 vlen=1\n" |
| "\t'f1' type_id=7 bits_offset=0"); |
| |
| cleanup: |
| btf__free(btf2); |
| btf__free(btf1); |
| } |
| |
| static void btf_add_dup_struct_in_cu(struct btf *btf, int start_id) |
| { |
| #define ID(n) (start_id + n) |
| btf__set_pointer_size(btf, 8); /* enforce 64-bit arch */ |
| |
| btf__add_int(btf, "int", 4, BTF_INT_SIGNED); /* [1] int */ |
| |
| btf__add_struct(btf, "s", 8); /* [2] struct s { */ |
| btf__add_field(btf, "a", ID(3), 0, 0); /* struct anon a; */ |
| btf__add_field(btf, "b", ID(4), 0, 0); /* struct anon b; */ |
| /* } */ |
| |
| btf__add_struct(btf, "(anon)", 8); /* [3] struct anon { */ |
| btf__add_field(btf, "f1", ID(1), 0, 0); /* int f1; */ |
| btf__add_field(btf, "f2", ID(1), 32, 0); /* int f2; */ |
| /* } */ |
| |
| btf__add_struct(btf, "(anon)", 8); /* [4] struct anon { */ |
| btf__add_field(btf, "f1", ID(1), 0, 0); /* int f1; */ |
| btf__add_field(btf, "f2", ID(1), 32, 0); /* int f2; */ |
| /* } */ |
| #undef ID |
| } |
| |
| static void test_split_dup_struct_in_cu() |
| { |
| struct btf *btf1, *btf2 = NULL; |
| int err; |
| |
| /* generate the base data.. */ |
| btf1 = btf__new_empty(); |
| if (!ASSERT_OK_PTR(btf1, "empty_main_btf")) |
| return; |
| |
| btf_add_dup_struct_in_cu(btf1, 0); |
| |
| VALIDATE_RAW_BTF( |
| btf1, |
| "[1] INT 'int' size=4 bits_offset=0 nr_bits=32 encoding=SIGNED", |
| "[2] STRUCT 's' size=8 vlen=2\n" |
| "\t'a' type_id=3 bits_offset=0\n" |
| "\t'b' type_id=4 bits_offset=0", |
| "[3] STRUCT '(anon)' size=8 vlen=2\n" |
| "\t'f1' type_id=1 bits_offset=0\n" |
| "\t'f2' type_id=1 bits_offset=32", |
| "[4] STRUCT '(anon)' size=8 vlen=2\n" |
| "\t'f1' type_id=1 bits_offset=0\n" |
| "\t'f2' type_id=1 bits_offset=32"); |
| |
| /* ..dedup them... */ |
| err = btf__dedup(btf1, NULL); |
| if (!ASSERT_OK(err, "btf_dedup")) |
| goto cleanup; |
| |
| VALIDATE_RAW_BTF( |
| btf1, |
| "[1] INT 'int' size=4 bits_offset=0 nr_bits=32 encoding=SIGNED", |
| "[2] STRUCT 's' size=8 vlen=2\n" |
| "\t'a' type_id=3 bits_offset=0\n" |
| "\t'b' type_id=3 bits_offset=0", |
| "[3] STRUCT '(anon)' size=8 vlen=2\n" |
| "\t'f1' type_id=1 bits_offset=0\n" |
| "\t'f2' type_id=1 bits_offset=32"); |
| |
| /* and add the same data on top of it */ |
| btf2 = btf__new_empty_split(btf1); |
| if (!ASSERT_OK_PTR(btf2, "empty_split_btf")) |
| goto cleanup; |
| |
| btf_add_dup_struct_in_cu(btf2, 3); |
| |
| VALIDATE_RAW_BTF( |
| btf2, |
| "[1] INT 'int' size=4 bits_offset=0 nr_bits=32 encoding=SIGNED", |
| "[2] STRUCT 's' size=8 vlen=2\n" |
| "\t'a' type_id=3 bits_offset=0\n" |
| "\t'b' type_id=3 bits_offset=0", |
| "[3] STRUCT '(anon)' size=8 vlen=2\n" |
| "\t'f1' type_id=1 bits_offset=0\n" |
| "\t'f2' type_id=1 bits_offset=32", |
| "[4] INT 'int' size=4 bits_offset=0 nr_bits=32 encoding=SIGNED", |
| "[5] STRUCT 's' size=8 vlen=2\n" |
| "\t'a' type_id=6 bits_offset=0\n" |
| "\t'b' type_id=7 bits_offset=0", |
| "[6] STRUCT '(anon)' size=8 vlen=2\n" |
| "\t'f1' type_id=4 bits_offset=0\n" |
| "\t'f2' type_id=4 bits_offset=32", |
| "[7] STRUCT '(anon)' size=8 vlen=2\n" |
| "\t'f1' type_id=4 bits_offset=0\n" |
| "\t'f2' type_id=4 bits_offset=32"); |
| |
| err = btf__dedup(btf2, NULL); |
| if (!ASSERT_OK(err, "btf_dedup")) |
| goto cleanup; |
| |
| /* after dedup it should match the original data */ |
| VALIDATE_RAW_BTF( |
| btf2, |
| "[1] INT 'int' size=4 bits_offset=0 nr_bits=32 encoding=SIGNED", |
| "[2] STRUCT 's' size=8 vlen=2\n" |
| "\t'a' type_id=3 bits_offset=0\n" |
| "\t'b' type_id=3 bits_offset=0", |
| "[3] STRUCT '(anon)' size=8 vlen=2\n" |
| "\t'f1' type_id=1 bits_offset=0\n" |
| "\t'f2' type_id=1 bits_offset=32"); |
| |
| cleanup: |
| btf__free(btf2); |
| btf__free(btf1); |
| } |
| |
| void test_btf_dedup_split() |
| { |
| if (test__start_subtest("split_simple")) |
| test_split_simple(); |
| if (test__start_subtest("split_struct_duped")) |
| test_split_struct_duped(); |
| if (test__start_subtest("split_fwd_resolve")) |
| test_split_fwd_resolve(); |
| if (test__start_subtest("split_dup_struct_in_cu")) |
| test_split_dup_struct_in_cu(); |
| } |
