Fixed small bug with relocation table.

Workshell · Jun 19, 2016 · fb1c6d0 · fb1c6d0
1 parent 0357b8a
commit fb1c6d0
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/Src/Workshell.PE/Content/Relocation/RelocationTable.cs b/Src/Workshell.PE/Content/Relocation/RelocationTable.cs
@@ -28,7 +28,7 @@ public static RelocationTable Get(ExecutableImage image)
             if (!image.NTHeaders.DataDirectories.Exists(DataDirectoryType.Debug))
                 return null;
 
-            DataDirectory directory = image.NTHeaders.DataDirectories[DataDirectoryType.Debug];
+            DataDirectory directory = image.NTHeaders.DataDirectories[DataDirectoryType.BaseRelocationTable];
 
             if (DataDirectory.IsNullOrEmpty(directory))
                 return null;

diff --git a/Src/Workshell.PE/DataDirectory.cs b/Src/Workshell.PE/DataDirectory.cs
@@ -206,6 +206,14 @@ public uint Size
             }
         }
 
+        public bool IsEmpty
+        {
+            get
+            {
+                return (data_dir.Size == 0);
+            }
+        }
+
         #endregion
 
     }
-Original file line number
+Diff line change
@@ Expand Up / @@ -206,6 +206,14 @@ public uint Size @@
                 }
             }
+            public bool IsEmpty
+            {
+                get
+                {
+                    return (data_dir.Size == 0);
+                }
+            }
             #endregion
         }
@@ Expand Down @@