ninemine/Flee
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Flee

Browse Source
This commit is contained in:
LiuBai095
2025-10-08 09:49:37 +08:00
commit 284e764345
99 changed files with 21742 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

269
InternalTypes/FleeILGenerator.cs Normal file
View File

@@ -0,0 +1,269 @@
using System.Diagnostics;
using System.Reflection;
using System.Reflection.Emit;
namespace Flee.InternalTypes
{
internal class FleeILGenerator
{
private ILGenerator _myIlGenerator;
private int _myLength;
private int _myLabelCount;
private readonly Dictionary<Type, LocalBuilder> _localBuilderTemp;
private int _myPass;
private int _brContext;
private BranchManager _bm;
public FleeILGenerator(ILGenerator ilg)
{
_myIlGenerator = ilg;
_localBuilderTemp = new Dictionary<Type, LocalBuilder>();
_myLength = 0;
_myPass = 1;
_bm = new BranchManager();
}
public int GetTempLocalIndex(Type localType)
{
LocalBuilder local = null;
if (_localBuilderTemp.TryGetValue(localType, out local) == false)
{
local = _myIlGenerator.DeclareLocal(localType);
_localBuilderTemp.Add(localType, local);
}
return local.LocalIndex;
}
/// <summary>
/// after first pass, check for long branches.
/// If any, we need to generate again.
/// </summary>
/// <returns></returns>
public bool NeedsSecondPass()
{
return _bm.HasLongBranches();
}
/// <summary>
/// need a new ILGenerator for 2nd pass. This can also
/// get called for a 3rd pass when emitting to assembly.
/// </summary>
/// <param name="ilg"></param>
public void PrepareSecondPass(ILGenerator ilg)
{
_bm.ComputeBranches();
_localBuilderTemp.Clear();
_myIlGenerator = ilg;
_myLength = 0;
_myPass++;
}
public void Emit(OpCode op)
{
this.RecordOpcode(op);
_myIlGenerator.Emit(op);
}
public void Emit(OpCode op, Type arg)
{
this.RecordOpcode(op);
_myIlGenerator.Emit(op, arg);
}
public void Emit(OpCode op, ConstructorInfo arg)
{
this.RecordOpcode(op);
_myIlGenerator.Emit(op, arg);
}
public void Emit(OpCode op, MethodInfo arg)
{
this.RecordOpcode(op);
_myIlGenerator.Emit(op, arg);
}
public void Emit(OpCode op, FieldInfo arg)
{
this.RecordOpcode(op);
_myIlGenerator.Emit(op, arg);
}
public void Emit(OpCode op, byte arg)
{
this.RecordOpcode(op);
_myIlGenerator.Emit(op, arg);
}
public void Emit(OpCode op, sbyte arg)
{
this.RecordOpcode(op);
_myIlGenerator.Emit(op, arg);
}
public void Emit(OpCode op, short arg)
{
this.RecordOpcode(op);
_myIlGenerator.Emit(op, arg);
}
public void Emit(OpCode op, int arg)
{
this.RecordOpcode(op);
_myIlGenerator.Emit(op, arg);
}
public void Emit(OpCode op, long arg)
{
this.RecordOpcode(op);
_myIlGenerator.Emit(op, arg);
}
public void Emit(OpCode op, float arg)
{
this.RecordOpcode(op);
_myIlGenerator.Emit(op, arg);
}
public void Emit(OpCode op, double arg)
{
this.RecordOpcode(op);
_myIlGenerator.Emit(op, arg);
}
public void Emit(OpCode op, string arg)
{
this.RecordOpcode(op);
_myIlGenerator.Emit(op, arg);
}
public void Emit(OpCode op, Label arg)
{
this.RecordOpcode(op);
_myIlGenerator.Emit(op, arg);
}
public void EmitBranch(Label arg)
{
if (_myPass == 1)
{
_bm.AddBranch(this, arg);
Emit(OpCodes.Br_S, arg);
}
else if (_bm.IsLongBranch(this) == false)
{
Emit(OpCodes.Br_S, arg);
}
else
{
Emit(OpCodes.Br, arg);
}
}
public void EmitBranchFalse(Label arg)
{
if (_myPass == 1)
{
_bm.AddBranch(this, arg);
Emit(OpCodes.Brfalse_S, arg);
}
else if (_bm.IsLongBranch(this) == false)
{
Emit(OpCodes.Brfalse_S, arg);
}
else
{
Emit(OpCodes.Brfalse, arg);
}
}
public void EmitBranchTrue(Label arg)
{
if (_myPass == 1)
{
_bm.AddBranch(this, arg);
Emit(OpCodes.Brtrue_S, arg);
}
else if (_bm.IsLongBranch(this) == false)
{
Emit(OpCodes.Brtrue_S, arg);
}
else
{
Emit(OpCodes.Brtrue, arg);
}
}
public void MarkLabel(Label lbl)
{
_myIlGenerator.MarkLabel(lbl);
_bm.MarkLabel(this, lbl);
}
public Label DefineLabel()
{
_myLabelCount += 1;
var label = _myIlGenerator.DefineLabel();
return label;
}
public LocalBuilder DeclareLocal(Type localType)
{
return _myIlGenerator.DeclareLocal(localType);
}
private void RecordOpcode(OpCode op)
{
//Trace.WriteLine(String.Format("{0:x}: {1}", MyLength, op.Name))
int operandLength = GetOpcodeOperandSize(op.OperandType);
_myLength += op.Size + operandLength;
}
private static int GetOpcodeOperandSize(OperandType operand)
{
switch (operand)
{
case OperandType.InlineNone:
return 0;
case OperandType.ShortInlineBrTarget:
case OperandType.ShortInlineI:
case OperandType.ShortInlineVar:
return 1;
case OperandType.InlineVar:
return 2;
case OperandType.InlineBrTarget:
case OperandType.InlineField:
case OperandType.InlineI:
case OperandType.InlineMethod:
case OperandType.InlineSig:
case OperandType.InlineString:
case OperandType.InlineTok:
case OperandType.InlineType:
case OperandType.ShortInlineR:
return 4;
case OperandType.InlineI8:
case OperandType.InlineR:
return 8;
default:
Debug.Fail("Unknown operand type");
break;
}
return 0;
}
[Conditional("DEBUG")]
public void ValidateLength()
{
Debug.Assert(this.Length == this.ILGeneratorLength, "ILGenerator length mismatch");
}
public int Length => _myLength;
public int LabelCount => _myLabelCount;
private int ILGeneratorLength => Utility.GetILGeneratorLength(_myIlGenerator);
}
}