bpf: enable cgroup local storage map pretty print with kind_flag

Commit 970289fc0a83 ("bpf: add bpffs pretty print for cgroup
local storage maps") added bpffs pretty print for cgroup
local storage maps. The commit worked for struct without kind_flag
set.

This patch refactored and made pretty print also work
with kind_flag set for the struct.

Acked-by: Martin KaFai Lau <kafai@fb.com>
Signed-off-by: Yonghong Song <yhs@fb.com>
Signed-off-by: Daniel Borkmann <daniel@iogearbox.net>

1 files changed, 28 insertions, 9 deletions

diff --git a/kernel/bpf/btf.c b/kernel/bpf/btf.c
index 93b6905e3a9b..e804b26a0506 100644
--- a/kernel/bpf/btf.c
+++ b/kernel/bpf/btf.c
@@ -546,22 +546,41 @@ static bool btf_type_int_is_regular(const struct btf_type *t)
 }
 
 /*
- * Check that given type is a regular int and has the expected size.
+ * Check that given struct member is a regular int with expected
+ * offset and size.
  */
-bool btf_type_is_reg_int(const struct btf_type *t, u32 expected_size)
+bool btf_member_is_reg_int(const struct btf *btf, const struct btf_type *s,
+                           const struct btf_member *m,
+                           u32 expected_offset, u32 expected_size)
 {
-        u8 nr_bits, nr_bytes;
-        u32 int_data;
+        const struct btf_type *t;
+        u32 id, int_data;
+        u8 nr_bits;
 
-        if (!btf_type_is_int(t))
+        id = m->type;
+        t = btf_type_id_size(btf, &id, NULL);
+        if (!t || !btf_type_is_int(t))
                 return false;
 
         int_data = btf_type_int(t);
         nr_bits = BTF_INT_BITS(int_data);
-        nr_bytes = BITS_ROUNDUP_BYTES(nr_bits);
-        if (BITS_PER_BYTE_MASKED(nr_bits) ||
-            BTF_INT_OFFSET(int_data) ||
-            nr_bytes != expected_size)
+        if (btf_type_kflag(s)) {
+                u32 bitfield_size = BTF_MEMBER_BITFIELD_SIZE(m->offset);
+                u32 bit_offset = BTF_MEMBER_BIT_OFFSET(m->offset);
+
+                /* if kflag set, int should be a regular int and
+                 * bit offset should be at byte boundary.
+                 */
+                return !bitfield_size &&
+                       BITS_ROUNDUP_BYTES(bit_offset) == expected_offset &&
+                       BITS_ROUNDUP_BYTES(nr_bits) == expected_size;
+        }
+
+        if (BTF_INT_OFFSET(int_data) ||
+            BITS_PER_BYTE_MASKED(m->offset) ||
+            BITS_ROUNDUP_BYTES(m->offset) != expected_offset ||
+            BITS_PER_BYTE_MASKED(nr_bits) ||
+            BITS_ROUNDUP_BYTES(nr_bits) != expected_size)
                 return false;
 
         return true;