make TLS directory optional

marirs · Oct 20, 2024 · 5fd5a4f · 5fd5a4f
1 parent aa385fa
commit 5fd5a4f
Showing 1 changed file with 14 additions and 10 deletions.
diff --git a/src/pe/tls.rs b/src/pe/tls.rs
@@ -10,18 +10,22 @@ pub struct TlsCallbacks {
 impl TlsCallbacks {
     pub fn parse(pe: &VecPE) -> Result<TlsCallbacks> {
         let callbacks = match pe.get_arch()? {
-            exe::Arch::X86 => {
-                let tls = ImageTLSDirectory32::parse(pe)?;
-                tls.get_callbacks(pe)?
+            exe::Arch::X86 => match ImageTLSDirectory32::parse(pe) {
+                Ok(tls) => tls
+                    .get_callbacks(pe)?
                     .iter()
                     .map(|x| x.0.into())
-                    .collect::<Vec<u64>>()
-            }
-            exe::Arch::X64 => ImageTLSDirectory64::parse(pe)?
-                .get_callbacks(pe)?
-                .iter()
-                .map(|x| x.0)
-                .collect::<Vec<u64>>(),
+                    .collect::<Vec<u64>>(),
+                _ => vec![],
+            },
+            exe::Arch::X64 => match ImageTLSDirectory64::parse(pe) {
+                Ok(tls) => tls
+                    .get_callbacks(pe)?
+                    .iter()
+                    .map(|x| x.0)
+                    .collect::<Vec<u64>>(),
+                _ => vec![],
+            },
         };
         Ok(TlsCallbacks { callbacks })
     }