Add check for byte code table size (#170)

nanoframework · Oct 29, 2024 · aba8387 · aba8387
1 parent 6ee2961
commit aba8387
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/MetadataProcessor.Shared/Tables/nanoByteCodeTable.cs b/MetadataProcessor.Shared/Tables/nanoByteCodeTable.cs
@@ -82,6 +82,12 @@ public ushort GetMethodId(
 
             var byteCode = CreateByteCode(method);
 
+            // sanity check for RVA overflow
+            if (_lastAvailableRva + byteCode.Length > ushort.MaxValue)
+            {
+                throw new InvalidOperationException($"Byte code table overflow in assembly '{_context.AssemblyDefinition.Name}'. It's impossible to compile such a large assembly.");
+            }
+
             _methods.Add(method);
             _lastAvailableRva += (ushort)byteCode.Length;
 
@@ -161,3 +167,4 @@ private byte[] CreateByteCode(
         }
     }
 }
+