pkg/cover: support read elf section range

2 files changed, 30 insertions, 0 deletions

diff --git a/pkg/cover/backend/backend.go b/pkg/cover/backend/backend.go
index 49f2f9508..9847bc8b3 100644
--- a/pkg/cover/backend/backend.go
+++ b/pkg/cover/backend/backend.go
@@ -58,6 +58,11 @@ type Range struct {
 	EndCol    int
 }
 
+type SecRange struct {
+	Start uint64
+	End   uint64
+}
+
 const LineEnd = 1 << 30
 
 func Make(target *targets.Target, vm, objDir, srcDir, buildDir string, splitBuild bool,
diff --git a/pkg/cover/backend/elf.go b/pkg/cover/backend/elf.go
index 25536c170..2126292b4 100644
--- a/pkg/cover/backend/elf.go
+++ b/pkg/cover/backend/elf.go
@@ -220,6 +220,31 @@ func elfGetCompilerVersion(path string) string {
 	return string(data[:])
 }
 
+func elfReadTextSecRange(module KernelModule) (*SecRange, error) {
+	text, err := elfReadTextSec(module)
+	if err != nil {
+		return nil, err
+	}
+	r := &SecRange{
+		Start: text.Addr,
+		End:   text.Addr + text.Size,
+	}
+	return r, nil
+}
+
+func elfReadTextSec(module KernelModule) (*elf.Section, error) {
+	file, err := elf.Open(module.Path)
+	if err != nil {
+		return nil, err
+	}
+	defer file.Close()
+	text := file.Section(".text")
+	if text == nil {
+		return nil, fmt.Errorf("no .text section in the object file")
+	}
+	return text, nil
+}
+
 func getPCBase(cfg *mgrconfig.Config) (uint64, error) {
 	bin := filepath.Join(cfg.KernelObj, cfg.SysTarget.KernelObject)
 	file, err := elf.Open(bin)