td-lader/src/elf64.rs: Fix Relocs iteration may infinite loop #263

Signed-off-by: haowei <WeiX.Hao@intel.com>
confidential-containers · May 15, 2022 · 0e08c3e · 0e08c3e
1 parent 64796e6
commit 0e08c3e
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/td-loader/src/elf64.rs b/td-loader/src/elf64.rs
@@ -698,6 +698,11 @@ impl<'a> Iterator for Relocs<'a> {
         if self.index >= self.relacount {
             return None;
         }
+
+        if self.relaent == 0 {
+            return None;
+        }
+
         let offset = self.index * self.relaent;
         self.entries.len().checked_sub(offset)?;
         let current_bytes = &self.entries[offset..];