[PATCH v2 bpf-next 04/18] selftests/bpf: Test helper support for BTF_KIND_LOC[_PARAM|_PROTO|SEC]

[Alan Maguire]

Add support to dump, encode and validate new location-related kinds.

Signed-off-by: Alan Maguire <alan.magu...@oracle.com>
---
 tools/testing/selftests/bpf/btf_helpers.c | 36 ++++++++++++++++++++++-
 tools/testing/selftests/bpf/test_btf.h    | 16 ++++++++++
 2 files changed, 51 insertions(+), 1 deletion(-)

diff --git a/tools/testing/selftests/bpf/btf_helpers.c 
b/tools/testing/selftests/bpf/btf_helpers.c
index 1c1c2c26690a..6065ee4ba00a 100644
--- a/tools/testing/selftests/bpf/btf_helpers.c
+++ b/tools/testing/selftests/bpf/btf_helpers.c
@@ -27,11 +27,14 @@ static const char * const btf_kind_str_mapping[] = {
        [BTF_KIND_DECL_TAG]     = "DECL_TAG",
        [BTF_KIND_TYPE_TAG]     = "TYPE_TAG",
        [BTF_KIND_ENUM64]       = "ENUM64",
+       [BTF_KIND_LOC_PARAM]    = "LOC_PARAM",
+       [BTF_KIND_LOC_PROTO]    = "LOC_PROTO",
+       [BTF_KIND_LOCSEC]       = "LOCSEC",
 };
 
 static const char *btf_kind_str(__u16 kind)
 {
-       if (kind > BTF_KIND_ENUM64)
+       if (kind > BTF_KIND_LOCSEC)
                return "UNKNOWN";
        return btf_kind_str_mapping[kind];
 }
@@ -203,6 +206,37 @@ int fprintf_btf_type_raw(FILE *out, const struct btf *btf, 
__u32 id)
                fprintf(out, " type_id=%u component_idx=%d",
                        t->type, btf_decl_tag(t)->component_idx);
                break;
+       case BTF_KIND_LOC_PARAM: {
+               struct btf_loc_param *p = btf_loc_param(t);
+               __u32 *v = (__u32 *)(p + 1);
+
+               fprintf(out, " size=%d flags=0x%x vlen=%u", t->size, p->flags, 
vlen);
+               for (i = 0; i < vlen; i++, v++) {
+                       if (p->flags & BTF_LOC_PARAM_SIGNED)
+                               fprintf(out, "\n\tvalue=%d", (__s32)*v);
+                       else
+                               fprintf(out, "\n\tvalue=%u", *v);
+               }
+               break;
+       }
+       case BTF_KIND_LOC_PROTO: {
+               const __u32 *p = btf_loc_proto_params(t);
+
+               fprintf(out, " vlen=%u", vlen);
+               for (i = 0; i < vlen; i++, p++)
+                       fprintf(out, "\n\ttype_id=%u", *p);
+               break;
+       }
+       case BTF_KIND_LOCSEC: {
+               const struct btf_loc *l = btf_locsec_locs(t);
+
+               fprintf(out, " vlen=%u", vlen);
+               for (i = 0; i < vlen; i++, l++) {
+                       fprintf(out, "\n\tfunc_type_id=%u loc_proto_type_id=%u 
offset=%d",
+                               l->func, l->loc_proto, l->offset);
+               }
+               break;
+       }
        default:
                break;
        }
diff --git a/tools/testing/selftests/bpf/test_btf.h 
b/tools/testing/selftests/bpf/test_btf.h
index e7bc78108374..b99c04b4d26b 100644
--- a/tools/testing/selftests/bpf/test_btf.h
+++ b/tools/testing/selftests/bpf/test_btf.h
@@ -84,4 +84,20 @@
 #define BTF_TYPE_TAG_ENC(value, type)  \
        BTF_TYPE_ENC(value, BTF_INFO_ENC(BTF_KIND_TYPE_TAG, 0, 0), type)
 
+#define BTF_LOC_PARAM_ENC(nvals, sz, flags) \
+       BTF_TYPE_ENC(0, BTF_INFO_ENC(BTF_KIND_LOC_PARAM, 0, nvals), sz), (flags)
+
+#define BTF_LOC_PARAM_VAL_ENCODE(val)  (val)
+
+#define BTF_LOC_PROTO_ENC(nargs) \
+       BTF_TYPE_ENC(0, BTF_INFO_ENC(BTF_KIND_LOC_PROTO, 0, nargs), 0)
+
+#define BTF_LOC_PROTO_PARAM_ENCODE(param) (param)
+
+#define BTF_LOCSEC_ENC(name, nlocs) \
+       BTF_TYPE_ENC(name, BTF_INFO_ENC(BTF_KIND_LOCSEC, 0, nlocs), 0)
+
+#define BTF_LOCSEC_LOC_ENCODE(func, loc_proto, offset) \
+       (func), (loc_proto), (offset)
+
 #endif /* _TEST_BTF_H */
-- 
2.43.5