Merge 4bae7bf into a60be60

mkaring · web-flow · commit ab4a72769395 · 2021-01-10T09:53:29.000Z
diff --git a/Confuser.Protections/AntiTamper/NormalMode.cs b/Confuser.Protections/AntiTamper/NormalMode.cs
@@ -100,12 +100,18 @@ public void HandleMD(AntiTamperProtection parent, ConfuserContext context, Prote
 		}
 
 		void WriterEvent(object sender, ModuleWriterEventArgs e) {
-			var writer = (ModuleWriterBase)sender;
-			if (e.Event == ModuleWriterEvent.MDEndCreateTables) {
-				CreateSections(writer);
-			}
-			else if (e.Event == ModuleWriterEvent.BeginStrongNameSign) {
-				EncryptSection(writer);
+			switch (e.Event)
+			{
+				case ModuleWriterEvent.Begin when e.Writer is NativeModuleWriter nativeWriter:
+					// disable the optimization of the image size for the native writer, so the method bodies can be protected.
+					nativeWriter.Options = new NativeModuleWriterOptions(nativeWriter.ModuleDefMD, false);
+					break;
+				case ModuleWriterEvent.MDEndCreateTables:
+					CreateSections(e.Writer);
+					break;
+				case ModuleWriterEvent.BeginStrongNameSign:
+					EncryptSection(e.Writer);
+					break;
 			}
 		}
 
