From 81d890d94e3c415c6d440ae51868d5ae53e0be4a Mon Sep 17 00:00:00 2001
From: de4dot <de4dot@gmail.com>
Date: Sat, 5 Nov 2011 08:36:36 +0100
Subject: [PATCH] Don't update method header max stack field if no cflow deob

---
 de4dot.code/AssemblyModule.cs | 7 +++++--
 de4dot.code/ObfuscatedFile.cs | 2 +-
 2 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/de4dot.code/AssemblyModule.cs b/de4dot.code/AssemblyModule.cs
index ee030fab..e0f0b455 100644
--- a/de4dot.code/AssemblyModule.cs
+++ b/de4dot.code/AssemblyModule.cs
@@ -44,8 +44,11 @@ namespace de4dot {
 			return module;
 		}
 
-		public void save(string newFilename) {
-			module.Write(newFilename);
+		public void save(string newFilename, bool updateMaxStack) {
+			var writerParams = new WriterParameters() {
+				UpdateMaxStack = updateMaxStack,
+			};
+			module.Write(newFilename, writerParams);
 		}
 
 		void readMethodsFile() {
diff --git a/de4dot.code/ObfuscatedFile.cs b/de4dot.code/ObfuscatedFile.cs
index 031cfb71..605321d2 100644
--- a/de4dot.code/ObfuscatedFile.cs
+++ b/de4dot.code/ObfuscatedFile.cs
@@ -245,7 +245,7 @@ namespace de4dot {
 
 		public void save() {
 			Log.n("Saving {0}", options.NewFilename);
-			assemblyModule.save(options.NewFilename);
+			assemblyModule.save(options.NewFilename, options.ControlFlowDeobfuscation);
 		}
 
 		IList<MethodDefinition> getAllMethods() {