elf: add PIEType.REL

Wenzel · Sep 8, 2021 · b1eb586 · b1eb586
1 parent 387428e
commit b1eb586
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 1 deletion.
diff --git a/checksec/elf.py b/checksec/elf.py
@@ -77,6 +77,7 @@ class PIEType(Enum):
     No = 1
     DSO = 2
     PIE = 3
+    REL = 4
 
 
 class Libc:
@@ -155,6 +156,8 @@ def pie(self) -> PIEType:
                 return PIEType.PIE
             else:
                 return PIEType.DSO
+        elif self.bin.header.file_type == E_TYPE.RELOCATABLE:
+            return PIEType.REL
         return PIEType.No
 
     @property

diff --git a/checksec/output.py b/checksec/output.py
@@ -154,6 +154,8 @@ def add_checksec_result(self, filepath: Path, checksec: Union[ELFChecksecData, P
                 pie_res = f"[red]{pie.name}"
             elif pie == PIEType.DSO:
                 pie_res = f"[yellow]{pie.name}"
+            elif pie == PIEType.REL:
+                pie_res = f"[yellow]{pie.name}"
             else:
                 pie_res = "[green]Yes"
             row_res.append(pie_res)

diff --git a/tests/binaries b/tests/binaries