Улучшена обработка RVA и импортов в PE-файлах

Добавлена защита от бесконечного цикла при обработке импортов, безопасное вычисление RVA с проверкой переполнения. Усилена проверка корректности PE-файла (минимальный размер, отсутствие исключения при отсутствии файла). В RvaToFileOffset добавлена проверка на диапазон SizeOfRawData. В ReadDataFromRVA добавлена проверка размера и уточнено сообщение об ошибке. В PEResources исправлен расчет RVA для записей ресурсов.

Author: Infarh

MathCore/PE/ImportedFunctions.cs

@@ -86,9 +86,14 @@ private List<ImportedFunction> LoadImportedFunctions(PEFile pefFile, IMAGE_IMPOR
             var lookupTableRVA = descriptor.OriginalFirstThunk;
 
             uint index = 0;
-            while (true)
+            const uint max_imports = 10000; // Защита от бесконечного цикла
+            while (index < max_imports)
             {
-                var lookupData = pefFile.ReadDataFromRVA(lookupTableRVA + index * sizeof(uint), sizeof(uint));
+                // Безопасное вычисление RVA с проверкой переполнения
+                var offset_bytes = checked((uint)(index * sizeof(uint)));
+                var entry_rva = checked(lookupTableRVA + offset_bytes);
+                
+                var lookupData = pefFile.ReadDataFromRVA(entry_rva, sizeof(uint));
                 var lookupValue = BitConverter.ToUInt32(lookupData, 0);
 
                 if (lookupValue == 0)

MathCore/PE/PEFile.cs

@@ -22,9 +22,10 @@ public bool IsPE
     {
         get
         {
-            if (!Exists) throw new FileNotFoundException("Файл не найден", _File.FullName);
+            if (!Exists) return false;
 
-            if (_File.Length < 128) return false;
+            // DOS заголовок минимум 64 байта, NT заголовок требует еще данных
+            if (_File.Length < 512) return false;
 
             try
             {
@@ -62,10 +63,18 @@ public long RvaToFileOffset(uint RVA)
 
         foreach (var section in header.Sections)
         {
+            // RVA находится в диапазоне виртуального адреса раздела
             if (RVA >= section.VirtualAddress && RVA < section.VirtualAddress + section.VirtualSize)
             {
-                var offset = RVA - section.VirtualAddress + section.PointerToRawData;
-                return offset;
+                // Но смещение файла должно быть в пределах реального размера раздела в файле
+                var offset_in_section = RVA - section.VirtualAddress;
+                if (offset_in_section < section.SizeOfRawData)
+                {
+                    var offset = offset_in_section + section.PointerToRawData;
+                    return offset;
+                }
+                // RVA в диапазоне виртуального размера, но вне реального файла
+                return -1;
             }
         }
 
@@ -98,6 +107,9 @@ public uint FileOffsetToRva(long FileOffset)
     /// <returns>Прочитанные данные</returns>
     public byte[] ReadDataFromRVA(uint RVA, int Size)
     {
+        if (Size <= 0)
+            throw new ArgumentException("Размер должен быть положительным", nameof(Size));
+
         var offset = RvaToFileOffset(RVA);
         if (offset < 0)
             throw new InvalidOperationException($"RVA 0x{RVA:X8} находится вне разделов файла");
@@ -109,7 +121,7 @@ public byte[] ReadDataFromRVA(uint RVA, int Size)
         var read = file.Read(data, 0, Size);
 
         if (read != Size)
-            throw new InvalidOperationException($"Не удалось прочитать полный объём данных из RVA 0x{RVA:X8}");
+            throw new InvalidOperationException($"Не удалось прочитать полный объём данных из RVA 0x{RVA:X8}: прочитано {read} байт вместо {Size}");
 
         return data;
     }

MathCore/PE/PEResources.cs

@@ -128,11 +128,13 @@ private IMAGE_RESOURCE_DIRECTORY_ENTRY[] ReadDirectoryEntries(IMAGE_RESOURCE_DIR
         {
             // Первые NumberOfNamedEntries - это именованные записи, потом идут записи с ID
             var entrySize = Marshal.SizeOf<IMAGE_RESOURCE_DIRECTORY_ENTRY>();
-            var offset = BaseRVA + (uint)Marshal.SizeOf<IMAGE_RESOURCE_DIRECTORY>();
+            var directory_size = Marshal.SizeOf<IMAGE_RESOURCE_DIRECTORY>();
 
             for (int i = 0; i < totalEntries; i++)
             {
-                var data = _PEFile.ReadDataFromRVA(offset + (uint)(i * entrySize), entrySize);
+                // Вычисляем RVA каждой записи (BaseRVA + размер директории + смещение записи)
+                var entry_rva = BaseRVA + (uint)directory_size + (uint)(i * entrySize);
+                var data = _PEFile.ReadDataFromRVA(entry_rva, entrySize);
                 var gch = GCHandle.Alloc(data, GCHandleType.Pinned);
                 try
                 {