public EnCManager(Process process)
{
resource = new ResourceManager(this, process);
resource.PreLoad();
process.Modules.Added += this.ProcessModuleAdded;
metadata = new MetaDataManager(this);
eventCreator = new EditorEventCreator(this);
process.EnCHook = this;
this.process = process;
LocalVarDiff.ClearLocalVarCache();
}
// +-----------------+ 0
// | delta header | ----->> size of the rest of IL in bytes
// +-----------------+ 4
// | IL header |
// +-----------------+
// | |
// | IL code |
// | |
// +-----------------+
/// <summary>
/// Builds header of changing method.
/// </summary>
/// <param name="newMethod">MethodDefinition of new version of changing method.</param>
/// <param name="originMethod">MethoDefinition of original version of changing method.</param>
/// <param name="tkMethod">Metadata token pointing to place of method in running version of metadata.</param>
/// <param name="remapper">IL offset remapper, used to build local variable changes.</param>
/// <param name="placeholder">Product of method, represents translation from old variable set to the new one.</param>
/// <returns>Metadata token pointing to lolcalVars signature, if not present, 0 is given.</returns>
private uint buildILHeader(MethodDefinition newMethod, MethodDefinition originMethod,
SequencePointRemapper remapper, out Dictionary<int, int> placeholder)
{
uint local_var_old, token;
uint local_var = newMethod.Body.LocalVarToken.ToUInt32();
uint code_size = (uint)newMethod.Body.CodeSize;
uint max_stack_size = (uint)newMethod.Body.MaxStackSize;
bool old_tiny = originMethod.Body.IsTiny;
bool tiny = newMethod.Body.IsTiny;
placeholder = null;
if (old_tiny && !tiny) {
// when new method is fat and the old tiny
byte[] sig = metadata.NewImporter.GetSigantureSA(local_var);
Signature signa = new Signature(sig);
signa.Migrate(metadata);
byte[] sig2 = signa.Compress();
metadata.OldEmitter.CorMetaDataEmit.GetTokenFromSig(sig2, (uint)sig2.Length, out token);
tiny = false;
} else if (!old_tiny && tiny) {
// when new method is tiny and old fat
local_var_old = originMethod.Body.LocalVarToken.ToUInt32();
local_var = local_var_old;
max_stack_size = 8;
tiny = false;
} else if (!old_tiny && !tiny) {
// both methods are fat
Signature sigNew = new Signature(metadata.NewImporter.GetSigantureSA(local_var));
Signature sigOld = new Signature(metadata.OldImporter.GetSigantureSA(originMethod.Body.LocalVarToken.ToUInt32()));
sigNew.Migrate(metadata);
// Instead of unused old local variables use placeholders
LocalVarDiff varDiff = new LocalVarDiff(Builder.Manager.ResourceManager.CurrentModule.SymReader,
Builder.SymbolWriter.CorSymNewReader, remapper);
byte[] sig = varDiff.makeSignature(sigOld, sigNew, out placeholder, originMethod.MetadataToken.ToUInt32(),
newMethod.MetadataToken.ToUInt32(),metadata).Compress();
metadata.OldEmitter.CorMetaDataEmit.GetTokenFromSig(sig, (uint)sig.Length, out local_var);
}
//there is no need to change IL and MetaData when both methods are tiny
if (!tiny) {
uint header_flags = 0x3013;
buffer.AddRange(BitConverter.GetBytes((ushort)header_flags));//BitConverter.GetBytes((ushort)header_flags));//.uintToByteArray(header_flags,2));
buffer.AddRange(BitConverter.GetBytes((ushort)max_stack_size));//Util.uintToByteArray(max_stack_size,2));
buffer.AddRange(BitConverter.GetBytes(code_size));//.uintToByteArray(code_size));
buffer.AddRange(BitConverter.GetBytes(local_var));//.uintToByteArray(local_var));
originMethod.Body.LocalVarToken = new MetadataToken(local_var);
return local_var;
} else {
buffer.Add((byte)((code_size << 2) | 0x2));
return 0;
}
}
// +-----------------+ 0
// | delta header | ----->> size of the rest of IL in bytes
// +-----------------+ 4
// | IL header |
// +-----------------+
// | |
// | IL code |
// | |
// +-----------------+
/// <summary>
/// Builds header of changing method.
/// </summary>
/// <param name="newMethod">MethodDefinition of new version of changing method.</param>
/// <param name="originMethod">MethoDefinition of original version of changing method.</param>
/// <param name="tkMethod">Metadata token pointing to place of method in running version of metadata.</param>
/// <param name="remapper">IL offset remapper, used to build local variable changes.</param>
/// <param name="placeholder">Product of method, represents translation from old variable set to the new one.</param>
/// <returns>Metadata token pointing to lolcalVars signature, if not present, 0 is given.</returns>
private uint buildILHeader(MethodDefinition newMethod, MethodDefinition originMethod,
SequencePointRemapper remapper, out Dictionary <int, int> placeholder)
{
uint local_var_old, token;
uint local_var = newMethod.Body.LocalVarToken.ToUInt32();
uint code_size = (uint)newMethod.Body.CodeSize;
uint max_stack_size = (uint)newMethod.Body.MaxStackSize;
bool old_tiny = originMethod.Body.IsTiny;
bool tiny = newMethod.Body.IsTiny;
placeholder = null;
if (old_tiny && !tiny)
{
// when new method is fat and the old tiny
byte[] sig = metadata.NewImporter.GetSigantureSA(local_var);
Signature signa = new Signature(sig);
signa.Migrate(metadata);
byte[] sig2 = signa.Compress();
metadata.OldEmitter.CorMetaDataEmit.GetTokenFromSig(sig2, (uint)sig2.Length, out token);
tiny = false;
}
else if (!old_tiny && tiny)
{
// when new method is tiny and old fat
local_var_old = originMethod.Body.LocalVarToken.ToUInt32();
local_var = local_var_old;
max_stack_size = 8;
tiny = false;
}
else if (!old_tiny && !tiny)
{
// both methods are fat
Signature sigNew = new Signature(metadata.NewImporter.GetSigantureSA(local_var));
Signature sigOld = new Signature(metadata.OldImporter.GetSigantureSA(originMethod.Body.LocalVarToken.ToUInt32()));
sigNew.Migrate(metadata);
// Instead of unused old local variables use placeholders
LocalVarDiff varDiff = new LocalVarDiff(Builder.Manager.ResourceManager.CurrentModule.SymReader,
Builder.SymbolWriter.CorSymNewReader, remapper);
byte[] sig = varDiff.makeSignature(sigOld, sigNew, out placeholder, originMethod.MetadataToken.ToUInt32(),
newMethod.MetadataToken.ToUInt32(), metadata).Compress();
metadata.OldEmitter.CorMetaDataEmit.GetTokenFromSig(sig, (uint)sig.Length, out local_var);
}
//there is no need to change IL and MetaData when both methods are tiny
if (!tiny)
{
uint header_flags = 0x3013;
buffer.AddRange(BitConverter.GetBytes((ushort)header_flags)); //BitConverter.GetBytes((ushort)header_flags));//.uintToByteArray(header_flags,2));
buffer.AddRange(BitConverter.GetBytes((ushort)max_stack_size)); //Util.uintToByteArray(max_stack_size,2));
buffer.AddRange(BitConverter.GetBytes(code_size)); //.uintToByteArray(code_size));
buffer.AddRange(BitConverter.GetBytes(local_var)); //.uintToByteArray(local_var));
originMethod.Body.LocalVarToken = new MetadataToken(local_var);
return(local_var);
}
else
{
buffer.Add((byte)((code_size << 2) | 0x2));
return(0);
}
}
