Merge pull request #111 from Wenzel/pie/add_rel_type

ELF: add support for PIEType REL

checksec/elf.py

@@ -6,6 +6,7 @@
 from typing import FrozenSet, List, Optional
 
 import lief
+from lief.ELF import E_TYPE
 
 from .binary import BinarySecurity
 from .errors import ErrorParsingFailed
@@ -76,6 +77,7 @@ class PIEType(Enum):
     No = 1
     DSO = 2
     PIE = 3
+    REL = 4
 
 
 class Libc:
@@ -149,11 +151,13 @@ def has_canary(self) -> bool:
 
     @property
     def pie(self) -> PIEType:
-        if self.bin.is_pie:
+        if self.bin.header.file_type == E_TYPE.DYNAMIC:
             if self.bin.has(lief.ELF.DYNAMIC_TAGS.DEBUG):
                 return PIEType.PIE
             else:
                 return PIEType.DSO
+        elif self.bin.header.file_type == E_TYPE.RELOCATABLE:
+            return PIEType.REL
         return PIEType.No
 
     @property

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)