private static void WriteSerializeInstruction(ModuleDefinition module, MethodDefinition method, ILProcessor processor, FieldDefinition field, FieldDefinition dirty, int index, int offset)
{
method.Body.Variables.Add(new VariableDefinition(module.ImportReference(typeof(bool))));
Instruction end = processor.Create(OpCodes.Nop);
processor.Append(processor.Create(OpCodes.Nop));
processor.Append(processor.Create(OpCodes.Ldarg_0));
processor.Append(processor.Create(OpCodes.Ldfld, dirty));
processor.Append(processor.Create(OpCodes.Ldc_I8, 1L << index));
processor.Append(processor.Create(OpCodes.And));
processor.Append(processor.Create(OpCodes.Ldc_I8, 1L << index));
processor.Append(processor.Create(OpCodes.Ceq));
processor.Append(processor.Create(OpCodes.Stloc, index + offset));
processor.Append(processor.Create(OpCodes.Ldloc, index + offset));
processor.Append(processor.Create(OpCodes.Brfalse_S, end));
processor.Append(processor.Create(OpCodes.Nop));
processor.Append(processor.Create(OpCodes.Ldarg_1));
processor.Append(processor.Create(OpCodes.Ldarg_0));
processor.Append(processor.Create(OpCodes.Ldfld, field));
processor.Append(BaseTypeFactory.CreateWriteInstruction(module, processor, field.FieldType.Resolve()));
processor.Append(end);
}
public static MethodDefinition CreateSerialize(ModuleDefinition module, TypeDefinition type)
{
MethodDefinition serialize = MethodFactory.CreateMethod(module, type, "Serialize", MethodAttributes.Public | MethodAttributes.SpecialName | MethodAttributes.HideBySig, true);
serialize.Parameters.Add(new ParameterDefinition("wBuffer", ParameterAttributes.None, module.ImportReference(WeaverProgram.ByteBufferType)));
ILProcessor processor = serialize.Body.GetILProcessor();
processor.Append(processor.Create(OpCodes.Nop));
foreach (FieldDefinition field in type.Fields)
{
TypeDefinition fieldType = field.FieldType.Resolve();
if (!CheckHelper.CheckStructFiled(WeaverProgram.Base, type, fieldType))
{
continue;
}
if (field.FieldType.FullName == typeof(byte[]).FullName)
{
processor.Append(processor.Create(OpCodes.Ldarg, 0));
processor.Append(processor.Create(OpCodes.Ldfld, field));
processor.Append(processor.Create(OpCodes.Ldarg, 1));
processor.Append(processor.Create(OpCodes.Call, module.ImportReference(WeaverProgram.ByteUtilsWriteMethod)));
continue;
}
if (field.FieldType.FullName == typeof(ByteBuffer).FullName)
{
processor.Append(processor.Create(OpCodes.Ldarg, 0));
processor.Append(processor.Create(OpCodes.Ldfld, field));
processor.Append(processor.Create(OpCodes.Ldarg, 1));
processor.Append(processor.Create(OpCodes.Call, module.ImportReference(WeaverProgram.ByteBufferUtilsWriteMethod)));
continue;
}
if (field.FieldType.IsArray)
{
ArrayWriteFactory.CreateStructFieldWriteInstruction(module, serialize, processor, field, fieldType);
}
else if (BaseTypeFactory.IsBaseType(fieldType))
{
processor.Append(processor.Create(OpCodes.Ldarg_1));
processor.Append(processor.Create(OpCodes.Ldarg_0));
processor.Append(processor.Create(OpCodes.Ldfld, field));
processor.Append(BaseTypeFactory.CreateWriteInstruction(module, processor, fieldType));
}
else if (fieldType.IsValueType)
{
processor.Append(processor.Create(OpCodes.Ldarg_0));
processor.Append(processor.Create(OpCodes.Ldflda, field));
processor.Append(processor.Create(OpCodes.Ldarg_1));
processor.Append(processor.Create(OpCodes.Call, CreateSerialize(module, fieldType)));
}
}
processor.Append(processor.Create(OpCodes.Nop));
processor.Append(processor.Create(OpCodes.Ret));
return(serialize);
}
public static void Weave(ModuleDefinition module, Dictionary <ushort, MethodData> methods)
{
foreach (ushort msgId in methods.Keys)
{
MethodData methodData = methods[msgId];
ChannelType channel = methodData.Channel;
MethodDefinition method = methodData.MethodDef;
if (!CheckHelper.CheckMethodFirstParams(WeaverProgram.Server, method))
{
continue;
}
ILProcessor processor = method.Body.GetILProcessor();
method.Body.Variables.Clear();
method.Body.Instructions.Clear();
method.Body.Variables.Add(new VariableDefinition(module.ImportReference(WeaverProgram.ByteBufferType)));
processor.Append(processor.Create(OpCodes.Nop));
processor.Append(processor.Create(OpCodes.Ldc_I4, WeaverProgram.SugBufferSize));
processor.Append(processor.Create(OpCodes.Call, module.ImportReference(WeaverProgram.ByteBufferAllocateMethod)));
processor.Append(processor.Create(OpCodes.Stloc_0));
Collection <ParameterDefinition> parms = method.Parameters;
for (int i = 0; i < parms.Count; ++i)
{
if (i > 0)
{
ParameterDefinition parm = parms[i];
byte index = (byte)(method.IsStatic ? i : i + 1);
TypeDefinition parmType = parm.ParameterType.Resolve();
if (parm.ParameterType.FullName == typeof(byte[]).FullName)
{
processor.Append(processor.Create(OpCodes.Ldarg_S, index));
processor.Append(processor.Create(OpCodes.Ldloc_0));
processor.Append(processor.Create(OpCodes.Call, module.ImportReference(WeaverProgram.ByteUtilsWriteMethod)));
continue;
}
if (parm.ParameterType.FullName == typeof(ByteBuffer).FullName)
{
processor.Append(processor.Create(OpCodes.Ldarg_S, index));
processor.Append(processor.Create(OpCodes.Ldloc_0));
processor.Append(processor.Create(OpCodes.Call, module.ImportReference(WeaverProgram.ByteBufferUtilsWriteMethod)));
continue;
}
if (parm.ParameterType.IsArray)
{
ArrayWriteFactory.CreateMethodParamWriteInstruction(module, method, processor, parmType, index);
continue;
}
if (BaseTypeFactory.IsBaseType(parmType))
{
processor.Append(processor.Create(OpCodes.Ldloc_0));
processor.Append(processor.Create(OpCodes.Ldarg_S, index));
processor.Append(BaseTypeFactory.CreateWriteInstruction(module, processor, parmType));
continue;
}
if (parmType.IsValueType)
{
MethodDefinition serialize = StructMethodFactory.CreateSerialize(module, parmType);
processor.Append(processor.Create(OpCodes.Ldarga_S, index));
processor.Append(processor.Create(OpCodes.Ldloc_0));
processor.Append(processor.Create(OpCodes.Call, module.ImportReference(serialize)));
}
}
}
processor.Append(processor.Create(method.IsStatic ? OpCodes.Ldarg_0 : OpCodes.Ldarg_1));
processor.Append(processor.Create(OpCodes.Ldc_I4, msgId));
processor.Append(processor.Create(OpCodes.Ldloc_0));
processor.Append(processor.Create(OpCodes.Ldc_I4, (int)channel));
processor.Append(processor.Create(OpCodes.Call, module.ImportReference(WeaverProgram.NetworkServerManagerSendMethod)));
processor.Append(processor.Create(OpCodes.Nop));
processor.Append(processor.Create(OpCodes.Ret));
}
}
public static void Weave(ModuleDefinition module, List <TypeDefinition> syncs, out Dictionary <FieldDefinition, MethodDefinition> gets, out Dictionary <FieldDefinition, MethodDefinition> sets)
{
sets = new Dictionary <FieldDefinition, MethodDefinition>();
gets = new Dictionary <FieldDefinition, MethodDefinition>();
foreach (TypeDefinition type in syncs)
{
List <FieldDefinition> fields = new List <FieldDefinition>();
foreach (FieldDefinition field in type.Fields)
{
if (field.CustomAttributes.Count > 0 && field.CustomAttributes[0].AttributeType.FullName == WeaverProgram.SyncFieldType.FullName)
{
fields.Add(field);
}
}
if (fields.Count == 0)
{
return;
}
FieldDefinition dirty = FieldFactory.CreateField(module, type, "m_Dirty", FieldAttributes.Private, module.ImportReference(typeof(long)));
//public bool IsDirty();
ModifyIsDirtyMethod(module, type, dirty);
//get set method
for (int i = 0; i < fields.Count; ++i)
{
FieldDefinition field = fields[i];
if (!CheckHelper.CheckSyncFiled(WeaverProgram.Server, type, field.FieldType.Resolve()))
{
continue;
}
string name = field.Name;
MethodDefinition get = CreateGetMethod(module, type, field, name);
MethodDefinition set = CreateSetMethod(module, type, field, name, dirty, i);
PropertyDefinition propertyDefinition = new PropertyDefinition("Sync" + name, PropertyAttributes.None, field.FieldType)
{
GetMethod = get,
SetMethod = set
};
type.Properties.Add(propertyDefinition);
gets.Add(field, get);
sets.Add(field, set);
}
//Serialize
MethodDefinition serialize = ResolveHelper.ResolveMethod(type, "Serialize");
ILProcessor processor = serialize.Body.GetILProcessor();
processor.Body.Instructions.Clear();
Instruction ins1 = processor.Create(OpCodes.Nop);
Instruction ins2 = processor.Create(OpCodes.Ldarg_0);
Instruction ins3 = processor.Create(OpCodes.Ldc_I8, 0L);
Instruction ins4 = processor.Create(OpCodes.Stfld, dirty);
Instruction ins5 = processor.Create(OpCodes.Nop);
Instruction ins6 = processor.Create(OpCodes.Ret);
processor.Append(processor.Create(OpCodes.Nop));
processor.Append(processor.Create(OpCodes.Ldarg_0));
processor.Append(processor.Create(OpCodes.Ldfld, dirty));
processor.Append(processor.Create(OpCodes.Ldc_I8, 0L));
processor.Append(processor.Create(OpCodes.Cgt));
processor.Append(processor.Create(OpCodes.Stloc_0));
processor.Append(processor.Create(OpCodes.Ldloc_0));
processor.Append(processor.Create(OpCodes.Brfalse, ins6));
processor.Append(processor.Create(OpCodes.Nop));
processor.Append(processor.Create(OpCodes.Ldarg_1));
processor.Append(processor.Create(OpCodes.Ldarg_0));
processor.Append(processor.Create(OpCodes.Ldfld, dirty));
processor.Append(BaseTypeFactory.CreateWriteInstruction(module, processor, dirty.FieldType.Resolve()));
serialize.Body.Variables.Add(new VariableDefinition(module.ImportReference(typeof(bool))));
int offset = serialize.Body.Variables.Count;
for (int i = 0; i < fields.Count; ++i)
{
WriteSerializeInstruction(module, serialize, processor, fields[i], dirty, i, offset);
}
processor.Append(ins1);
processor.Append(ins2);
processor.Append(ins3);
processor.Append(ins4);
processor.Append(ins5);
processor.Append(ins6);
}
}
public static void CreateStructFieldWriteInstruction(ModuleDefinition module, MethodDefinition method, ILProcessor processor, FieldDefinition field, TypeDefinition fieldType)
{
int lenIndex = method.Body.Variables.Count;
int checkIndex = method.Body.Variables.Count + 1;
int intIndex = method.Body.Variables.Count + 2;
int boolIndex = method.Body.Variables.Count + 3;
method.Body.Variables.Add(new VariableDefinition(module.ImportReference(typeof(int))));
method.Body.Variables.Add(new VariableDefinition(module.ImportReference(typeof(bool))));
method.Body.Variables.Add(new VariableDefinition(module.ImportReference(typeof(int))));
method.Body.Variables.Add(new VariableDefinition(module.ImportReference(typeof(bool))));
Instruction goto1 = processor.Create(OpCodes.Nop);
Instruction goto2 = processor.Create(OpCodes.Nop);
Instruction goto3 = processor.Create(OpCodes.Nop);
Instruction goto4 = processor.Create(OpCodes.Nop);
Instruction goto5 = processor.Create(OpCodes.Nop);
//int len = this.Field != null ? this.Field.Length : 0;
processor.Append(processor.Create(OpCodes.Ldarg_0));
processor.Append(processor.Create(OpCodes.Ldfld, field));
processor.Append(processor.Create(OpCodes.Brtrue_S, goto1));
processor.Append(processor.Create(OpCodes.Ldc_I4_0));
processor.Append(processor.Create(OpCodes.Br_S, goto2));
processor.Append(goto1);
processor.Append(processor.Create(OpCodes.Ldarg_0));
processor.Append(processor.Create(OpCodes.Ldfld, field));
processor.Append(processor.Create(OpCodes.Ldlen));
processor.Append(processor.Create(OpCodes.Conv_I4));
processor.Append(goto2);
processor.Append(processor.Create(OpCodes.Stloc, lenIndex));
//writer.Write(len);
processor.Append(processor.Create(OpCodes.Ldarg_1));
processor.Append(processor.Create(OpCodes.Ldloc, lenIndex));
processor.Append(BaseTypeFactory.CreateWriteInstruction(module, processor, typeof(int).ToString()));
//if (len > 0)
processor.Append(processor.Create(OpCodes.Ldloc, lenIndex));
processor.Append(processor.Create(OpCodes.Ldc_I4_0));
processor.Append(processor.Create(OpCodes.Cgt));
processor.Append(processor.Create(OpCodes.Stloc, checkIndex));
processor.Append(processor.Create(OpCodes.Ldloc, checkIndex));
processor.Append(processor.Create(OpCodes.Brfalse_S, goto5));
//for (int i = 0; i < len; i++)
processor.Append(processor.Create(OpCodes.Ldc_I4_0));
processor.Append(processor.Create(OpCodes.Stloc, intIndex));
//go to check
processor.Append(processor.Create(OpCodes.Br_S, goto4));
processor.Append(goto3);
if (BaseTypeFactory.IsBaseType(fieldType))
{
//writer.Write(this.Field[i]);
processor.Append(processor.Create(OpCodes.Ldarg_1));
processor.Append(processor.Create(OpCodes.Ldarg_0));
processor.Append(processor.Create(OpCodes.Ldfld, field));
processor.Append(processor.Create(OpCodes.Ldloc, intIndex));
processor.Append(processor.Create(OpCodes.Ldelema, module.ImportReference(fieldType)));
processor.Append(processor.Create(OpCodes.Ldobj, module.ImportReference(fieldType)));
processor.Append(BaseTypeFactory.CreateWriteInstruction(module, processor, fieldType));
}
else if (fieldType.IsValueType)
{
//this.Field[i].Serialize(writer);
processor.Append(processor.Create(OpCodes.Ldarg_0));
processor.Append(processor.Create(OpCodes.Ldfld, field));
processor.Append(processor.Create(OpCodes.Ldloc, intIndex));
processor.Append(processor.Create(OpCodes.Ldelema, module.ImportReference(fieldType)));
processor.Append(processor.Create(OpCodes.Ldarg_1));
processor.Append(processor.Create(OpCodes.Call, StructMethodFactory.FindSerialize(module, fieldType)));
}
//i++
processor.Append(processor.Create(OpCodes.Nop));
processor.Append(processor.Create(OpCodes.Ldloc, intIndex));
processor.Append(processor.Create(OpCodes.Ldc_I4_1));
processor.Append(processor.Create(OpCodes.Add));
processor.Append(processor.Create(OpCodes.Stloc, intIndex));
//i < len
processor.Append(goto4);
processor.Append(processor.Create(OpCodes.Ldloc, intIndex));
processor.Append(processor.Create(OpCodes.Ldloc, lenIndex));
processor.Append(processor.Create(OpCodes.Clt));
processor.Append(processor.Create(OpCodes.Stloc, boolIndex));
processor.Append(processor.Create(OpCodes.Ldloc, boolIndex));
//go to start
processor.Append(processor.Create(OpCodes.Brtrue_S, goto3));
processor.Append(goto5);
}
