public static bool Weave(ModuleDefinition module) { List <TypeDefinition> structs = new List <TypeDefinition>(); foreach (TypeDefinition type in module.Types) { if (!type.IsEnum && type.IsValueType && type.Namespace == "Nice.Game.Proto") { structs.Add(type); } } foreach (TypeDefinition type in structs) { StructMethodFactory.CreateSerialize(module, type); StructMethodFactory.CreateDeserialize(module, type); } return(structs.Count > 0); }
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)); } }