fix elf and pe flags

golang · May 25, 2022 · 025ea0e · 025ea0e
1 parent 7791e93
commit 025ea0e
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/debug/buildinfo/buildinfo.go b/src/debug/buildinfo/buildinfo.go
@@ -257,8 +257,10 @@ func (x *elfExe) DataStart() uint64 {
 			return s.Addr
 		}
 	}
+
 	for _, p := range x.f.Progs {
-		if p.Type == elf.PT_LOAD && p.Flags&(elf.PF_X|elf.PF_W) == elf.PF_W {
+		flags := elf.ProgFlag(elf.PF_R | elf.PF_W)
+		if p.Type == elf.PT_LOAD && (p.Flags&flags) == flags {
 			return p.Vaddr
 		}
 	}
@@ -313,8 +315,8 @@ func (x *peExe) DataStart() uint64 {
 		IMAGE_SCN_ALIGN_32BYTES          = 0x600000
 	)
 	for _, sect := range x.f.Sections {
-		if sect.VirtualAddress != 0 && sect.Size != 0 &&
-			sect.Characteristics&^IMAGE_SCN_ALIGN_32BYTES == IMAGE_SCN_CNT_INITIALIZED_DATA|IMAGE_SCN_MEM_READ|IMAGE_SCN_MEM_WRITE {
+		flags := uint32(IMAGE_SCN_CNT_INITIALIZED_DATA | IMAGE_SCN_MEM_READ | IMAGE_SCN_MEM_WRITE)
+		if sect.VirtualAddress != 0 && (sect.Characteristics&flags) == flags {
 			return uint64(sect.VirtualAddress) + x.imageBase()
 		}
 	}