[Add] FishNet

2026-03-30 20:11:57 +07:00
parent ee793a3361
commit c22c08753a
1797 changed files with 197950 additions and 1 deletions
@@ -0,0 +1,110 @@
+using FishNet.CodeGenerating.Extension;
+using FishNet.CodeGenerating.Helping.Extension;
+using FishNet.Object;
+using FishNet.Serializing;
+using MonoFN.Cecil;
+using System;
+using System.Collections.Generic;
+using System.Reflection;
+
+namespace FishNet.CodeGenerating.Helping
+{
+    internal class WriterImports : CodegenBase
+    {
+        #region Reflection references.
+        public MethodReference WriterPool_GetWriter_MethodRef;
+        public MethodReference WriterPool_GetWriterLength_MethodRef;
+        public MethodReference Writer_WritePackedWhole_Signed_MethodRef;
+        public MethodReference Writer_WritePackedWhole_Unsigned_MethodRef;
+        public TypeReference PooledWriter_TypeRef;
+        public TypeReference Writer_TypeRef;
+        public MethodReference PooledWriter_Dispose_MethodRef;
+        public MethodReference Writer_WriteDictionary_MethodRef;
+        public MethodReference Writer_WriteList_MethodRef;
+        public MethodReference Writer_WriteHashSet_MethodRef;
+        public MethodReference Writer_WriteArray_MethodRef;
+        public TypeReference AutoPackTypeRef;
+        public TypeReference GenericWriter_TypeRef;
+        public MethodReference GenericWriter_Write_MethodRef;
+        public MethodReference Writer_Write_MethodRef;
+        #endregion
+
+        /// <summary>
+        /// Imports references needed by this helper.
+        /// </summary>
+        /// <param name = "moduleDef"></param>
+        /// <returns></returns>
+        public override bool ImportReferences()
+        {
+            PooledWriter_TypeRef = ImportReference(typeof(PooledWriter));
+            Writer_TypeRef = ImportReference(typeof(Writer));
+            AutoPackTypeRef = ImportReference(typeof(AutoPackType));
+            GenericWriter_TypeRef = ImportReference(typeof(GenericWriter<>));
+            Writer_Write_MethodRef = Writer_TypeRef.CachedResolve(Session).GetMethodReference(Session, nameof(Writer.Write));
+
+
+            TypeDefinition genericWriterTd = GenericWriter_TypeRef.CachedResolve(Session);
+            GenericWriter_Write_MethodRef = ImportReference(genericWriterTd.GetMethod(nameof(GenericWriter<int>.SetWrite)));
+
+            // WriterPool.GetWriter
+            Type writerPoolType = typeof(WriterPool);
+            ImportReference(writerPoolType);
+            foreach (MethodInfo methodInfo in writerPoolType.GetMethods())
+            {
+                if (methodInfo.Name == nameof(WriterPool.Retrieve))
+                {
+                    // GetWriter().
+                    if (methodInfo.GetParameters().Length == 0)
+                    {
+                        WriterPool_GetWriter_MethodRef = ImportReference(methodInfo);
+                    }
+                    // GetWriter(?).
+                    else if (methodInfo.GetParameters().Length == 1)
+                    {
+                        ParameterInfo pi = methodInfo.GetParameters()[0];
+                        // GetWriter(int).
+                        if (pi.ParameterType == typeof(int))
+                            WriterPool_GetWriterLength_MethodRef = ImportReference(methodInfo);
+                    }
+                }
+            }
+
+            WriterProcessor gwh = GetClass<WriterProcessor>();
+            Type pooledWriterType = typeof(PooledWriter);
+            foreach (MethodInfo methodInfo in pooledWriterType.GetMethods())
+            {
+                int parameterCount = methodInfo.GetParameters().Length;
+
+                if (methodInfo.Name == nameof(PooledWriter.Store))
+                {
+                    PooledWriter_Dispose_MethodRef = ImportReference(methodInfo);
+                }
+                else if (methodInfo.Name == nameof(PooledWriter.WriteUnsignedPackedWhole))
+                {
+                    // todo: check if signed or not and set to signed/unsigned variable.
+                    // do the same changes for methods which call these.
+                    // Writer_WritePackedWhole_MethodRef = base.ImportReference(methodInfo);
+                }
+                // Relay writers.
+                else if (parameterCount == 1 && methodInfo.Name == nameof(PooledWriter.WriteDictionary))
+                {
+                    Writer_WriteDictionary_MethodRef = ImportReference(methodInfo);
+                }
+                else if (parameterCount == 1 && methodInfo.Name == nameof(PooledWriter.WriteList))
+                {
+                    Writer_WriteList_MethodRef = ImportReference(methodInfo);
+                }
+                else if (parameterCount == 1 && methodInfo.Name == nameof(PooledWriter.WriteHashSet))
+                {
+                    Writer_WriteHashSet_MethodRef = ImportReference(methodInfo);
+                }
+                else if (parameterCount == 1 && methodInfo.Name == nameof(PooledWriter.WriteArray))
+                {
+                    Writer_WriteArray_MethodRef = ImportReference(methodInfo);
+                }
+            }
+
+            return true;
+        }
+    }
+}