727 lines
18 KiB
C#
727 lines
18 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Reflection;
|
|
using System.Reflection.Emit;
|
|
|
|
namespace Server
|
|
{
|
|
public class AssemblyEmitter
|
|
{
|
|
private readonly ModuleBuilder m_ModuleBuilder;
|
|
|
|
public AssemblyEmitter(string assemblyName)
|
|
{
|
|
var assemblyBuilder = AssemblyBuilder.DefineDynamicAssembly(
|
|
new AssemblyName(assemblyName),
|
|
AssemblyBuilderAccess.Run
|
|
);
|
|
|
|
m_ModuleBuilder = assemblyBuilder.DefineDynamicModule(assemblyName);
|
|
}
|
|
|
|
public TypeBuilder DefineType(string typeName, TypeAttributes attrs, Type parentType) =>
|
|
m_ModuleBuilder.DefineType(typeName, attrs, parentType);
|
|
}
|
|
|
|
public class MethodEmitter
|
|
{
|
|
public delegate void Callback();
|
|
|
|
private readonly Stack<CallInfo> m_Calls;
|
|
|
|
private readonly Stack<Type> m_Stack;
|
|
|
|
private readonly Dictionary<Type, Queue<LocalBuilder>> m_Temps;
|
|
private Type[] m_ArgumentTypes;
|
|
|
|
public MethodEmitter(TypeBuilder typeBuilder)
|
|
{
|
|
Type = typeBuilder;
|
|
|
|
m_Temps = new Dictionary<Type, Queue<LocalBuilder>>();
|
|
|
|
m_Stack = new Stack<Type>();
|
|
m_Calls = new Stack<CallInfo>();
|
|
}
|
|
|
|
public TypeBuilder Type { get; }
|
|
|
|
public ILGenerator Generator { get; private set; }
|
|
|
|
public MethodBuilder Method { get; private set; }
|
|
|
|
public Type Active => m_Stack.Peek();
|
|
|
|
public void Define(string name, MethodAttributes attr, Type returnType, Type[] parms)
|
|
{
|
|
Method = Type.DefineMethod(name, attr, returnType, parms);
|
|
Generator = Method.GetILGenerator();
|
|
|
|
m_ArgumentTypes = parms;
|
|
}
|
|
|
|
public LocalBuilder CreateLocal(Type localType) => Generator.DeclareLocal(localType);
|
|
|
|
public LocalBuilder AcquireTemp(Type localType)
|
|
{
|
|
if (!m_Temps.TryGetValue(localType, out var list))
|
|
{
|
|
m_Temps[localType] = list = new Queue<LocalBuilder>();
|
|
}
|
|
|
|
return list.Count > 0 ? list.Dequeue() : CreateLocal(localType);
|
|
}
|
|
|
|
public void ReleaseTemp(LocalBuilder local)
|
|
{
|
|
if (local.LocalType == null)
|
|
{
|
|
return;
|
|
}
|
|
|
|
if (!m_Temps.TryGetValue(local.LocalType, out var list))
|
|
{
|
|
m_Temps[local.LocalType] = list = new Queue<LocalBuilder>();
|
|
}
|
|
|
|
list.Enqueue(local);
|
|
}
|
|
|
|
public void Branch(Label label)
|
|
{
|
|
Generator.Emit(OpCodes.Br, label);
|
|
}
|
|
|
|
public void BranchIfFalse(Label label)
|
|
{
|
|
Pop(typeof(object));
|
|
|
|
Generator.Emit(OpCodes.Brfalse, label);
|
|
}
|
|
|
|
public void BranchIfTrue(Label label)
|
|
{
|
|
Pop(typeof(object));
|
|
|
|
Generator.Emit(OpCodes.Brtrue, label);
|
|
}
|
|
|
|
public Label CreateLabel() => Generator.DefineLabel();
|
|
|
|
public void MarkLabel(Label label)
|
|
{
|
|
Generator.MarkLabel(label);
|
|
}
|
|
|
|
public void Pop()
|
|
{
|
|
m_Stack.Pop();
|
|
}
|
|
|
|
public void Pop(Type expected)
|
|
{
|
|
if (expected == null)
|
|
{
|
|
throw new InvalidOperationException("Expected type cannot be null.");
|
|
}
|
|
|
|
var onStack = m_Stack.Pop();
|
|
|
|
if (expected == typeof(bool))
|
|
{
|
|
expected = typeof(int);
|
|
}
|
|
|
|
if (onStack == typeof(bool))
|
|
{
|
|
onStack = typeof(int);
|
|
}
|
|
|
|
if (!expected.IsAssignableFrom(onStack))
|
|
{
|
|
throw new InvalidOperationException("Unexpected stack state.");
|
|
}
|
|
}
|
|
|
|
public void Push(Type type)
|
|
{
|
|
m_Stack.Push(type);
|
|
}
|
|
|
|
public void Return()
|
|
{
|
|
if (m_Stack.Count != (Method.ReturnType == typeof(void) ? 0 : 1))
|
|
{
|
|
throw new InvalidOperationException("Stack return mismatch.");
|
|
}
|
|
|
|
Generator.Emit(OpCodes.Ret);
|
|
}
|
|
|
|
public void LoadNull()
|
|
{
|
|
LoadNull(typeof(object));
|
|
}
|
|
|
|
public void LoadNull(Type type)
|
|
{
|
|
Push(type);
|
|
|
|
Generator.Emit(OpCodes.Ldnull);
|
|
}
|
|
|
|
public void Load(string value)
|
|
{
|
|
Push(typeof(string));
|
|
|
|
if (value != null)
|
|
{
|
|
Generator.Emit(OpCodes.Ldstr, value);
|
|
}
|
|
else
|
|
{
|
|
Generator.Emit(OpCodes.Ldnull);
|
|
}
|
|
}
|
|
|
|
public void Load(Enum value)
|
|
{
|
|
var toLoad = ((IConvertible)value).ToInt32(null);
|
|
Load(toLoad);
|
|
|
|
Pop();
|
|
Push(value.GetType());
|
|
}
|
|
|
|
public void Load(long value)
|
|
{
|
|
Push(typeof(long));
|
|
|
|
Generator.Emit(OpCodes.Ldc_I8, value);
|
|
}
|
|
|
|
public void Load(float value)
|
|
{
|
|
Push(typeof(float));
|
|
|
|
Generator.Emit(OpCodes.Ldc_R4, value);
|
|
}
|
|
|
|
public void Load(double value)
|
|
{
|
|
Push(typeof(double));
|
|
|
|
Generator.Emit(OpCodes.Ldc_R8, value);
|
|
}
|
|
|
|
public void Load(char value)
|
|
{
|
|
Load((int)value);
|
|
|
|
Pop();
|
|
Push(typeof(char));
|
|
}
|
|
|
|
public void Load(bool value)
|
|
{
|
|
Push(typeof(bool));
|
|
|
|
if (value)
|
|
{
|
|
Generator.Emit(OpCodes.Ldc_I4_1);
|
|
}
|
|
else
|
|
{
|
|
Generator.Emit(OpCodes.Ldc_I4_0);
|
|
}
|
|
}
|
|
|
|
public void Load(int value)
|
|
{
|
|
Push(typeof(int));
|
|
|
|
switch (value)
|
|
{
|
|
case -1:
|
|
Generator.Emit(OpCodes.Ldc_I4_M1);
|
|
break;
|
|
|
|
case 0:
|
|
Generator.Emit(OpCodes.Ldc_I4_0);
|
|
break;
|
|
|
|
case 1:
|
|
Generator.Emit(OpCodes.Ldc_I4_1);
|
|
break;
|
|
|
|
case 2:
|
|
Generator.Emit(OpCodes.Ldc_I4_2);
|
|
break;
|
|
|
|
case 3:
|
|
Generator.Emit(OpCodes.Ldc_I4_3);
|
|
break;
|
|
|
|
case 4:
|
|
Generator.Emit(OpCodes.Ldc_I4_4);
|
|
break;
|
|
|
|
case 5:
|
|
Generator.Emit(OpCodes.Ldc_I4_5);
|
|
break;
|
|
|
|
case 6:
|
|
Generator.Emit(OpCodes.Ldc_I4_6);
|
|
break;
|
|
|
|
case 7:
|
|
Generator.Emit(OpCodes.Ldc_I4_7);
|
|
break;
|
|
|
|
case 8:
|
|
Generator.Emit(OpCodes.Ldc_I4_8);
|
|
break;
|
|
|
|
default:
|
|
if (value >= sbyte.MinValue && value <= sbyte.MaxValue)
|
|
{
|
|
Generator.Emit(OpCodes.Ldc_I4_S, (sbyte)value);
|
|
}
|
|
else
|
|
{
|
|
Generator.Emit(OpCodes.Ldc_I4, value);
|
|
}
|
|
|
|
break;
|
|
}
|
|
}
|
|
|
|
public void LoadField(FieldInfo field)
|
|
{
|
|
Pop(field.DeclaringType);
|
|
|
|
Push(field.FieldType);
|
|
|
|
Generator.Emit(OpCodes.Ldfld, field);
|
|
}
|
|
|
|
public void LoadLocal(LocalBuilder local)
|
|
{
|
|
Push(local.LocalType);
|
|
|
|
var index = local.LocalIndex;
|
|
|
|
switch (index)
|
|
{
|
|
case 0:
|
|
Generator.Emit(OpCodes.Ldloc_0);
|
|
break;
|
|
|
|
case 1:
|
|
Generator.Emit(OpCodes.Ldloc_1);
|
|
break;
|
|
|
|
case 2:
|
|
Generator.Emit(OpCodes.Ldloc_2);
|
|
break;
|
|
|
|
case 3:
|
|
Generator.Emit(OpCodes.Ldloc_3);
|
|
break;
|
|
|
|
default:
|
|
if (index >= byte.MinValue && index <= byte.MinValue)
|
|
{
|
|
Generator.Emit(OpCodes.Ldloc_S, (byte)index);
|
|
}
|
|
else
|
|
{
|
|
Generator.Emit(OpCodes.Ldloc, (short)index);
|
|
}
|
|
|
|
break;
|
|
}
|
|
}
|
|
|
|
public void StoreLocal(LocalBuilder local)
|
|
{
|
|
Pop(local.LocalType);
|
|
|
|
Generator.Emit(OpCodes.Stloc, local);
|
|
}
|
|
|
|
public void LoadArgument(int index)
|
|
{
|
|
if (index > 0)
|
|
{
|
|
Push(m_ArgumentTypes[index - 1]);
|
|
}
|
|
else
|
|
{
|
|
Push(Type);
|
|
}
|
|
|
|
switch (index)
|
|
{
|
|
case 0:
|
|
Generator.Emit(OpCodes.Ldarg_0);
|
|
break;
|
|
|
|
case 1:
|
|
Generator.Emit(OpCodes.Ldarg_1);
|
|
break;
|
|
|
|
case 2:
|
|
Generator.Emit(OpCodes.Ldarg_2);
|
|
break;
|
|
|
|
case 3:
|
|
Generator.Emit(OpCodes.Ldarg_3);
|
|
break;
|
|
|
|
default:
|
|
if (index >= byte.MinValue && index <= byte.MaxValue)
|
|
{
|
|
Generator.Emit(OpCodes.Ldarg_S, (byte)index);
|
|
}
|
|
else
|
|
{
|
|
Generator.Emit(OpCodes.Ldarg, (short)index);
|
|
}
|
|
|
|
break;
|
|
}
|
|
}
|
|
|
|
public void CastAs(Type type)
|
|
{
|
|
Pop(typeof(object));
|
|
Push(type);
|
|
|
|
Generator.Emit(OpCodes.Isinst, type);
|
|
}
|
|
|
|
public void Neg()
|
|
{
|
|
Pop(typeof(int));
|
|
|
|
Push(typeof(int));
|
|
|
|
Generator.Emit(OpCodes.Neg);
|
|
}
|
|
|
|
public void Compare(OpCode opCode)
|
|
{
|
|
Pop();
|
|
Pop();
|
|
|
|
Push(typeof(int));
|
|
|
|
Generator.Emit(opCode);
|
|
}
|
|
|
|
public void LogicalNot()
|
|
{
|
|
Pop(typeof(int));
|
|
|
|
Push(typeof(int));
|
|
|
|
Generator.Emit(OpCodes.Ldc_I4_0);
|
|
Generator.Emit(OpCodes.Ceq);
|
|
}
|
|
|
|
public void Xor()
|
|
{
|
|
Pop(typeof(int));
|
|
Pop(typeof(int));
|
|
|
|
Push(typeof(int));
|
|
|
|
Generator.Emit(OpCodes.Xor);
|
|
}
|
|
|
|
public void Chain(Property prop)
|
|
{
|
|
for (var i = 0; i < prop.Chain.Length; ++i)
|
|
{
|
|
Call(prop.Chain[i].GetGetMethod());
|
|
}
|
|
}
|
|
|
|
public void Call(MethodInfo method)
|
|
{
|
|
BeginCall(method);
|
|
|
|
var call = m_Calls.Peek();
|
|
|
|
if (call.parms.Length > 0)
|
|
{
|
|
throw new InvalidOperationException("Method requires parameters.");
|
|
}
|
|
|
|
FinishCall();
|
|
}
|
|
|
|
public bool CompareTo(int sign, Callback argGenerator)
|
|
{
|
|
var active = Active;
|
|
|
|
var compareTo = active.GetMethod("CompareTo", new[] { active });
|
|
|
|
if (compareTo == null)
|
|
{
|
|
/* This gets a little tricky...
|
|
*
|
|
* There's a scenario where we might be trying to use CompareTo on an interface
|
|
* which, while it doesn't explicitly implement CompareTo itself, is said to
|
|
* extend IComparable indirectly. The implementation is implicitly passed off
|
|
* to implementers...
|
|
*
|
|
* interface ISomeInterface : IComparable
|
|
* {
|
|
* void SomeMethod();
|
|
* }
|
|
*
|
|
* class SomeClass : ISomeInterface
|
|
* {
|
|
* void SomeMethod() { ... }
|
|
* int CompareTo( object other ) { ... }
|
|
* }
|
|
*
|
|
* In this case, calling ISomeInterface.GetMethod( "CompareTo" ) will return null.
|
|
*
|
|
* Bleh.
|
|
*/
|
|
|
|
var ifaces = active.FindInterfaces(
|
|
(type, obj) => type.IsGenericType
|
|
&& type.GetGenericTypeDefinition() == typeof(IComparable<>)
|
|
&& type.GetGenericArguments()[0].IsAssignableFrom(active),
|
|
null
|
|
);
|
|
|
|
if (ifaces.Length > 0)
|
|
{
|
|
compareTo = ifaces[0].GetMethod("CompareTo", new[] { active });
|
|
}
|
|
else
|
|
{
|
|
ifaces = active.FindInterfaces((type, obj) => type == typeof(IComparable), null);
|
|
|
|
if (ifaces.Length > 0)
|
|
{
|
|
compareTo = ifaces[0].GetMethod("CompareTo", new[] { active });
|
|
}
|
|
}
|
|
}
|
|
|
|
if (compareTo == null)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
if (!active.IsValueType)
|
|
{
|
|
/* This object is a reference type, so we have to make it behave
|
|
*
|
|
* null.CompareTo( null ) = 0
|
|
* real.CompareTo( null ) = -1
|
|
* null.CompareTo( real ) = +1
|
|
*
|
|
*/
|
|
|
|
var aValue = AcquireTemp(active);
|
|
var bValue = AcquireTemp(active);
|
|
|
|
StoreLocal(aValue);
|
|
|
|
argGenerator();
|
|
|
|
StoreLocal(bValue);
|
|
|
|
/* if (aValue == null)
|
|
* {
|
|
* if (bValue == null)
|
|
* v = 0;
|
|
* else
|
|
* v = +1;
|
|
* }
|
|
* else if (bValue == null)
|
|
* {
|
|
* v = -1;
|
|
* }
|
|
* else
|
|
* {
|
|
* v = aValue.CompareTo( bValue );
|
|
* }
|
|
*/
|
|
|
|
var store = CreateLabel();
|
|
|
|
var aNotNull = CreateLabel();
|
|
|
|
LoadLocal(aValue);
|
|
BranchIfTrue(aNotNull);
|
|
// if (aValue == null)
|
|
{
|
|
var bNotNull = CreateLabel();
|
|
|
|
LoadLocal(bValue);
|
|
BranchIfTrue(bNotNull);
|
|
// if (bValue == null)
|
|
{
|
|
Load(0);
|
|
Pop(typeof(int));
|
|
Branch(store);
|
|
}
|
|
MarkLabel(bNotNull);
|
|
// else
|
|
{
|
|
Load(sign);
|
|
Pop(typeof(int));
|
|
Branch(store);
|
|
}
|
|
}
|
|
MarkLabel(aNotNull);
|
|
// else
|
|
{
|
|
var bNotNull = CreateLabel();
|
|
|
|
LoadLocal(bValue);
|
|
BranchIfTrue(bNotNull);
|
|
// bValue == null
|
|
{
|
|
Load(-sign);
|
|
Pop(typeof(int));
|
|
Branch(store);
|
|
}
|
|
MarkLabel(bNotNull);
|
|
// else
|
|
{
|
|
LoadLocal(aValue);
|
|
BeginCall(compareTo);
|
|
|
|
LoadLocal(bValue);
|
|
ArgumentPushed();
|
|
|
|
FinishCall();
|
|
|
|
if (sign == -1)
|
|
{
|
|
Neg();
|
|
}
|
|
}
|
|
}
|
|
|
|
MarkLabel(store);
|
|
|
|
ReleaseTemp(aValue);
|
|
ReleaseTemp(bValue);
|
|
}
|
|
else
|
|
{
|
|
BeginCall(compareTo);
|
|
|
|
argGenerator();
|
|
|
|
ArgumentPushed();
|
|
|
|
FinishCall();
|
|
|
|
if (sign == -1)
|
|
{
|
|
Neg();
|
|
}
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
public void BeginCall(MethodInfo method)
|
|
{
|
|
var type = (method.CallingConvention & CallingConventions.HasThis) != 0 ? m_Stack.Peek() : method.DeclaringType;
|
|
|
|
m_Calls.Push(new CallInfo(type, method));
|
|
|
|
if (type!.IsValueType)
|
|
{
|
|
var temp = AcquireTemp(type);
|
|
|
|
Generator.Emit(OpCodes.Stloc, temp);
|
|
Generator.Emit(OpCodes.Ldloca, temp);
|
|
|
|
ReleaseTemp(temp);
|
|
}
|
|
}
|
|
|
|
public void FinishCall()
|
|
{
|
|
var call = m_Calls.Pop();
|
|
|
|
if ((call.type.IsValueType || call.type.IsByRef) && call.method.DeclaringType != call.type)
|
|
{
|
|
Generator.Emit(OpCodes.Constrained, call.type);
|
|
}
|
|
|
|
if (call.method.DeclaringType?.IsValueType == true || call.method.IsStatic)
|
|
{
|
|
Generator.Emit(OpCodes.Call, call.method);
|
|
}
|
|
else
|
|
{
|
|
Generator.Emit(OpCodes.Callvirt, call.method);
|
|
}
|
|
|
|
for (var i = call.parms.Length - 1; i >= 0; --i)
|
|
{
|
|
Pop(call.parms[i].ParameterType);
|
|
}
|
|
|
|
if ((call.method.CallingConvention & CallingConventions.HasThis) != 0)
|
|
{
|
|
Pop(call.method.DeclaringType);
|
|
}
|
|
|
|
if (call.method.ReturnType != typeof(void))
|
|
{
|
|
Push(call.method.ReturnType);
|
|
}
|
|
}
|
|
|
|
public void ArgumentPushed()
|
|
{
|
|
var call = m_Calls.Peek();
|
|
|
|
var parm = call.parms[call.index++];
|
|
|
|
var argumentType = m_Stack.Peek();
|
|
|
|
if (!parm.ParameterType.IsAssignableFrom(argumentType))
|
|
{
|
|
throw new InvalidOperationException("Parameter type mismatch.");
|
|
}
|
|
|
|
if (argumentType.IsValueType && !parm.ParameterType.IsValueType)
|
|
{
|
|
Generator.Emit(OpCodes.Box, argumentType);
|
|
}
|
|
}
|
|
|
|
private class CallInfo
|
|
{
|
|
public readonly MethodInfo method;
|
|
public readonly ParameterInfo[] parms;
|
|
public readonly Type type;
|
|
|
|
public int index;
|
|
|
|
public CallInfo(Type type, MethodInfo method)
|
|
{
|
|
this.type = type;
|
|
this.method = method;
|
|
|
|
parms = method.GetParameters();
|
|
}
|
|
}
|
|
}
|
|
}
|