UTs for recovery and progs

aws · Aug 18, 2023 · 95df855 · 95df855
1 parent ed84377
commit 95df855
Show file tree

Hide file tree

Showing 5 changed files with 485 additions and 66 deletions.
diff --git a/Makefile b/Makefile
@@ -41,13 +41,16 @@ EBPF_TEST_MAP_BINARY := test-data/test.map.bpf.elf
 EBPF_TEST_LIC_SOURCE := test-data/test_license.bpf.c
 EBPF_TEST_LIC_BINARY := test-data/test_license.bpf.elf
 EBPF_TEST_INV_MAP_SOURCE := test-data/invalid_map.bpf.c
-EBPF_TEST_INV_MAP_BINARY := test-data/invalid_map.bpf.elf   
+EBPF_TEST_INV_MAP_BINARY := test-data/invalid_map.bpf.elf
+EBPF_TEST_RECOVERY_SOURCE := test-data/recoverydata.bpf.c
+EBPF_TEST_RECOVERY_BINARY := test-data/recoverydata.bpf.elf   
 build-bpf: ## Build BPF
 	$(CLANG) $(CLANG_INCLUDE) -g -O2 -Wall -fpie -target bpf -DCORE -D__BPF_TRACING__ -march=bpf -D__TARGET_ARCH_$(ARCH) -c $(EBPF_SOURCE) -o $(EBPF_BINARY)
 	$(CLANG) $(CLANG_INCLUDE) -g -O2 -Wall -fpie -target bpf -DCORE -D__BPF_TRACING__ -march=bpf -D__TARGET_ARCH_$(ARCH) -c $(EBPF_TEST_SOURCE) -o $(EBPF_TEST_BINARY)
 	$(CLANG) $(CLANG_INCLUDE) -g -O2 -Wall -fpie -target bpf -DCORE -D__BPF_TRACING__ -march=bpf -D__TARGET_ARCH_$(ARCH) -c $(EBPF_TEST_MAP_SOURCE) -o $(EBPF_TEST_MAP_BINARY)
 	$(CLANG) $(CLANG_INCLUDE) -g -O2 -Wall -fpie -target bpf -DCORE -D__BPF_TRACING__ -march=bpf -D__TARGET_ARCH_$(ARCH) -c $(EBPF_TEST_LIC_SOURCE) -o $(EBPF_TEST_LIC_BINARY)
 	$(CLANG) $(CLANG_INCLUDE) -g -O2 -Wall -fpie -target bpf -DCORE -D__BPF_TRACING__ -march=bpf -D__TARGET_ARCH_$(ARCH) -c $(EBPF_TEST_INV_MAP_SOURCE) -o $(EBPF_TEST_INV_MAP_BINARY)
+	$(CLANG) $(CLANG_INCLUDE) -g -O2 -Wall -fpie -target bpf -DCORE -D__BPF_TRACING__ -march=bpf -D__TARGET_ARCH_$(ARCH) -c $(EBPF_TEST_RECOVERY_SOURCE) -o $(EBPF_TEST_RECOVERY_BINARY)
 
 vmlinuxh:
 	bpftool btf dump file /sys/kernel/btf/vmlinux format c > $(abspath ./test-data/vmlinux.h)

diff --git a/pkg/elfparser/elf.go b/pkg/elfparser/elf.go
@@ -503,6 +503,11 @@ func (e *elfLoader) parseProg(loadedMaps map[string]ebpf_maps.BpfMap) (map[strin
 			return nil, fmt.Errorf("failed to get progEntry Data")
 		}
 
+		if len(data) == 0 {
+			log.Infof("Missing data in prog Section")
+			return nil, fmt.Errorf("missing data in prog section")
+		}
+
 		var linkedMaps map[int]string
 		//Apply relocation
 		if e.reloSectionMap[progIndex] == nil {
@@ -905,7 +910,7 @@ func (b *bpfSDKClient) RecoverAllBpfProgramsAndMaps() (map[string]BpfData, error
 		}
 	} else {
 		log.Infof("error checking BPF FS, might not be mounted %v", err)
-		return nil, fmt.Errorf("error checking BPF FS might not be mounted %v", err)
+		return nil, fmt.Errorf("error checking BPF FS might not be mounted")
 	}
 	//Return DS here
 	return loadedPrograms, nil