From b24722c7eed99c69095bce5314c59c0035d27276 Mon Sep 17 00:00:00 2001
From: Dylan Reimerink <dylan.reimerink@isovalent.com>
Date: Wed, 21 Feb 2024 11:35:24 +0100
Subject: [PATCH] btf: Make test for ld64imm CO-RE relocations

Added a test which makes sure we do not mess up offsets when replacing
a ld64imm with poisoned instructions.

Signed-off-by: Dylan Reimerink <dylan.reimerink@isovalent.com>
---
 Makefile                        |   1 +
 btf/core_reloc_test.go          |  27 +++++++++++++++++++++++++++
 btf/testdata/relocs_enum-eb.elf | Bin 0 -> 1688 bytes
 btf/testdata/relocs_enum-el.elf | Bin 0 -> 1688 bytes
 btf/testdata/relocs_enum.c      |  16 ++++++++++++++++
 5 files changed, 44 insertions(+)
 create mode 100644 btf/testdata/relocs_enum-eb.elf
 create mode 100644 btf/testdata/relocs_enum-el.elf
 create mode 100644 btf/testdata/relocs_enum.c

diff --git a/Makefile b/Makefile
index eb532b2fd..3727a6e11 100644
--- a/Makefile
+++ b/Makefile
@@ -47,6 +47,7 @@ TARGETS := \
 	btf/testdata/relocs \
 	btf/testdata/relocs_read \
 	btf/testdata/relocs_read_tgt \
+	btf/testdata/relocs_enum \
 	cmd/bpf2go/testdata/minimal
 
 .PHONY: all clean container-all container-shell generate
diff --git a/btf/core_reloc_test.go b/btf/core_reloc_test.go
index d6c9dd00f..798676007 100644
--- a/btf/core_reloc_test.go
+++ b/btf/core_reloc_test.go
@@ -110,3 +110,30 @@ func TestCORERelocationRead(t *testing.T) {
 		}
 	})
 }
+
+func TestLD64IMMReloc(t *testing.T) {
+	testutils.SkipOnOldKernel(t, "5.4", "vmlinux BTF in sysfs")
+
+	testutils.Files(t, testutils.Glob(t, "testdata/relocs_enum-*.elf"), func(t *testing.T, file string) {
+		fh, err := os.Open(file)
+		if err != nil {
+			t.Fatal(err)
+		}
+		defer fh.Close()
+
+		spec, err := ebpf.LoadCollectionSpecFromReader(fh)
+		if err != nil {
+			t.Fatal(err)
+		}
+
+		if spec.ByteOrder != internal.NativeEndian {
+			return
+		}
+
+		coll, err := ebpf.NewCollection(spec)
+		if err != nil {
+			t.Fatal(err)
+		}
+		defer coll.Close()
+	})
+}
diff --git a/btf/testdata/relocs_enum-eb.elf b/btf/testdata/relocs_enum-eb.elf
new file mode 100644
index 0000000000000000000000000000000000000000..01467e9a9da18321bae386ce1a562babb43e4e6d
GIT binary patch
literal 1688
zcmb_cQEL-H5S~j?HCCaBLeQ6^4|0KWiKWzss)Q8dgTa=j1qESqxopbeB^Pe@LW>XP
z2l!ABpA^J*-~A2#7Jq;*LY&#`rrR?pf<xwJzHerCW^X6==5f1KC^)bX2Y#S$5sL^q
zrRNGY^s@|3Wli~u{99{KBqq;?pJzXb|4I*(_j3c^KdGe^4cll@oezYs&&dKBM;9oe
zbrG*zoA*fc-AkZqme9YCO4355*h|y`L?a1dl5!UJx3{9<5JVEb;<9#u3y;$z8#9sh
z#HnCWAHs1axMawWe}^p2diaCgCkJ~+`>b=+?7Zl(-Myn{hj0&Syf+@ydU8;cT*!VP
zgIdbtBovH~vY{UWV}ge=N=DxLl3v!UF1GC2?Ri!AHI$=)>*3B!MQAHF3F3@1ei8{O
zJYwAZIX2vj2hp8Z)rMup0x8p|mr2f;=Vc>2b01@2G7{3AByk|4n5%ZMl>5+KQvToS
z?m#)^GD}BpeFx0_yLe9|H?`6nbWt}@ub|#QrTZkI`J$;Kj_#Z!MSV9<S24oh0O~jz
z;LYRCZGbHw{kq0y01s%k%wFV^(4J$|tEe=W2Ct(3N5RI_?B|l6A~rnLHgX3KV0gnd
zv_t2n^P2M!Z{z(O?&dJ{Z|A?w;g1Hdf-m`r1Rv+qZyvUM0!u~`{5YNrS<vsNB02_N
z$W#VBw5LO5aVfN$%{tq}o&S$)<2L#Cqj#pyfRY1kjWIl7@(rdru%%%G%Qx?vD4NB;
zt#44GB{Zu>%s0qj+fL`enb{=zc72*RQ$Lqhm}=rnE?}>}jIpwgH!kKC&orhvp`KSX
sHvO+5_L5m!z5l5Wo|{FPe1A56g!$$_b-q-2#kwZCXlaePXH&oO3-togvj6}9

literal 0
HcmV?d00001

diff --git a/btf/testdata/relocs_enum-el.elf b/btf/testdata/relocs_enum-el.elf
new file mode 100644
index 0000000000000000000000000000000000000000..d08985416898715cd5a2cef9eac076e5941d0197
GIT binary patch
literal 1688
zcmbtVQEL-H5S}Djja4W`LF&uV2f0AGL@D*5Do2X(!C*_%fP%2O+-=I?<u2Urg%%&o
z5AdNNJ}LMIeD^o_Tl@jO2yte!n_SMIh(kYizi(z|X747~w@=!wa;XHyO7H^~%%}h_
zE-m)1+PYB1$lDm4kH1x_C~r~sw|-8^@l|;NKIrxHcg&&pZCnW-7YT2ak!xC_7A&Nf
zXWDhr?;^0`8u=a|$XW>Uy+Y7=2xAF;l5!RWd%NLq2qN)cb6H=~{L?haMoeUVaV}UG
zfIrFvmkjmM@03MZAOGOs>G5Irh;_Qn&dUxvIP5+@fqPKr{n4P_mxH?GLI$4n>M4&B
zUoakLL&pck1ovf_#P;@zUbbr%wru6wvuoBHsD=Z}#>`B4C>0xfQN|fR3xyOmDOT|w
zJJ#|=bmujdu*#SxWg7M~$r-clEXF$r=<}0UNNb!#o(v<d+QCxRBWp$Zf2+F>)s)LD
zjjhH$0KI>U-VxeO4Vr--Vh3>*v5BDfNk-GPfgsI{WQzOu2s%JM;@=Q!4FFs7Ell9-
z0-&V)7s$&G7rE4&ppSBn5Z4eimxiz7{%=-M-7QBwQ?#P$<~?_iZ06!!?W8-CQC+&T
ziy$-nXh!F}SK!n;nTdZlqjUbLz}LZ%{7iy_^XW8CS`LvFBMD9vjfcz&f>ea3;0T#Y
zuaE2bP_1|r+RbKz-PZ;FbJ@f5voFbVBJ~kK>3CV5&xeue*%8)^pjWrR4W;LmH=4yi
zop<xWxTiVIoB0OiHQG<KFs0-<E7qrZD}<=%i5}E1xwEt_p1*=x72zWCcA+1tnG@=p
vzNe=D4INCsn@W3{;4;zDY|({^y%qUmj5fci{P~8i-_s(UBgxb^{|kNr^r+DO

literal 0
HcmV?d00001

diff --git a/btf/testdata/relocs_enum.c b/btf/testdata/relocs_enum.c
new file mode 100644
index 000000000..d9eed8e91
--- /dev/null
+++ b/btf/testdata/relocs_enum.c
@@ -0,0 +1,16 @@
+#include "bpf_core_read.h"
+
+enum cgroup_subsys_id {
+	cpuset_cgrp_id,
+	cpuset_cgrp_id_lublub,
+	CGROUP_SUBSYS_COUNT,
+};
+
+#define __section(NAME) __attribute__((section(NAME), used))
+
+__section("socket/core_ld64imm") int core_ld64imm() {
+	if (bpf_core_enum_value_exists(enum cgroup_subsys_id, cpuset_cgrp_id_lublub)) {
+		__attribute__((unused)) const volatile int val = bpf_core_enum_value(enum cgroup_subsys_id, cpuset_cgrp_id_lublub);
+	}
+	return 0;
+}