Store information about partial ngen in R2R images (dotnet#22680)

Author: MichalStrehovsky

src/inc/eventtracebase.h

@@ -531,6 +531,7 @@ namespace ETW
                 ManifestModule=0x8,
                 IbcOptimized=0x10,
                 ReadyToRunModule=0x20,
+                PartialReadyToRunModule=0x40,
             }ModuleFlags;
 
             typedef enum _RangeFlags

src/inc/readytorun.h

@@ -45,6 +45,8 @@ enum ReadyToRunFlag
     // Set if the original IL assembly was platform-neutral
     READYTORUN_FLAG_PLATFORM_NEUTRAL_SOURCE         = 0x00000001,
     READYTORUN_FLAG_SKIP_TYPE_VALIDATION            = 0x00000002,
+    // Set of methods with native code was determined using profile data
+    READYTORUN_FLAG_PARTIAL                         = 0x00000004,
 };
 
 enum ReadyToRunSectionType

src/tools/r2rdump/R2RHeader.cs

@@ -18,7 +18,8 @@ public enum ReadyToRunFlag
         {
             NONE = 0x00000000,
             READYTORUN_FLAG_PLATFORM_NEUTRAL_SOURCE = 0x00000001,
-            READYTORUN_FLAG_SKIP_TYPE_VALIDATION = 0x00000002
+            READYTORUN_FLAG_SKIP_TYPE_VALIDATION = 0x00000002,
+            READYTORUN_FLAG_PARTIAL = 0x00000004,
         }
 
         /// <summary>

src/vm/eventtrace.cpp

@@ -6044,12 +6044,18 @@ VOID ETW::LoaderLog::SendModuleEvent(Module *pModule, DWORD dwEventOptions, BOOL
         bIsIbcOptimized = pModule->IsIbcOptimized();
     }
     BOOL bIsReadyToRun = pModule->IsReadyToRun();
+    BOOL bIsPartialReadyToRun = FALSE;
+    if (bIsReadyToRun)
+    {
+        bIsPartialReadyToRun = pModule->GetReadyToRunInfo()->IsPartial();
+    }
     ULONG ulReservedFlags = 0;
     ULONG ulFlags = ((bHasNativeImage ? ETW::LoaderLog::LoaderStructs::NativeModule : 0) |
                      (bIsDynamicAssembly ? ETW::LoaderLog::LoaderStructs::DynamicModule : 0) |
                      (bIsManifestModule ? ETW::LoaderLog::LoaderStructs::ManifestModule : 0) |
                      (bIsIbcOptimized ? ETW::LoaderLog::LoaderStructs::IbcOptimized : 0) |
-                     (bIsReadyToRun ? ETW::LoaderLog::LoaderStructs::ReadyToRunModule : 0));
+                     (bIsReadyToRun ? ETW::LoaderLog::LoaderStructs::ReadyToRunModule : 0) |
+                     (bIsPartialReadyToRun ? ETW::LoaderLog::LoaderStructs::PartialReadyToRunModule : 0));
 
     // Grab PDB path, guid, and age for managed PDB and native (NGEN) PDB when
     // available.  Any failures are not fatal.  The corresponding PDB info will remain

src/vm/readytoruninfo.h

@@ -70,6 +70,12 @@ class ReadyToRunInfo
         return m_pHeader->Flags & READYTORUN_FLAG_SKIP_TYPE_VALIDATION;
     }
 
+    BOOL IsPartial()
+    {
+        LIMITED_METHOD_CONTRACT;
+        return m_pHeader->Flags & READYTORUN_FLAG_PARTIAL;
+    }
+
     PTR_PEImageLayout GetImage()
     {
         LIMITED_METHOD_CONTRACT;

src/zap/zapreadytorun.cpp

@@ -41,6 +41,9 @@ void ZapReadyToRunHeader::Save(ZapWriter * pZapWriter)
     if (pImage->GetCompileInfo()->AreAllClassesFullyLoaded(pImage->GetModuleHandle()))
         readyToRunHeader.Flags |= READYTORUN_FLAG_SKIP_TYPE_VALIDATION;
 
+    if (pImage->GetZapperOptions()->m_fPartialNGen)
+        readyToRunHeader.Flags |= READYTORUN_FLAG_PARTIAL;
+
     readyToRunHeader.NumberOfSections = m_Sections.GetCount();
 
     pZapWriter->Write(&readyToRunHeader, sizeof(readyToRunHeader));