You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

1423 lines
63 KiB

  1. using IPA.Config.Data;
  2. using IPA.Config.Stores.Attributes;
  3. using IPA.Logging;
  4. using System;
  5. using System.Collections.Generic;
  6. using System.Linq;
  7. using System.Reflection;
  8. using System.Reflection.Emit;
  9. using System.Text;
  10. using System.Threading;
  11. using System.Threading.Tasks;
  12. using System.Linq.Expressions;
  13. using System.Runtime.CompilerServices;
  14. using System.IO;
  15. using Boolean = IPA.Config.Data.Boolean;
  16. using System.Collections;
  17. using IPA.Utilities;
  18. #if NET3
  19. using Net3_Proxy;
  20. using Array = Net3_Proxy.Array;
  21. #endif
  22. [assembly: InternalsVisibleTo(IPA.Config.Stores.GeneratedExtension.AssemblyVisibilityTarget)]
  23. namespace IPA.Config.Stores
  24. {
  25. /// <summary>
  26. /// A class providing an extension for <see cref="Config"/> to make it easy to use generated
  27. /// config stores.
  28. /// </summary>
  29. public static class GeneratedExtension
  30. {
  31. /// <summary>
  32. /// The name of the assembly that internals must be visible to to allow internal protection.
  33. /// </summary>
  34. public const string AssemblyVisibilityTarget = GeneratedStore.GeneratedAssemblyName;
  35. /// <summary>
  36. /// Creates a generated <see cref="IConfigStore"/> of type <typeparamref name="T"/>, registers it to
  37. /// the <see cref="Config"/> object, and returns it. This also forces a synchronous config load via
  38. /// <see cref="Config.LoadSync"/> if <paramref name="loadSync"/> is <see langword="true"/>.
  39. /// </summary>
  40. /// <remarks>
  41. /// <para>
  42. /// <typeparamref name="T"/> must be a public non-<see langword="sealed"/> class.
  43. /// It can also be internal, but in that case, then your assembly must have the following attribute
  44. /// to allow the generated code to reference it.
  45. /// <code>
  46. /// [assembly: InternalsVisibleTo(IPA.Config.Stores.GeneratedExtension.AssemblyVisibilityTarget)]
  47. /// </code>
  48. /// </para>
  49. /// <para>
  50. /// Only fields and properties that are public or protected will be considered, and only properties
  51. /// where both the getter and setter are public or protected are considered. Any fields or properties
  52. /// with an <see cref="IgnoreAttribute"/> applied to them are also ignored. Having properties be <see langword="virtual"/> is not strictly
  53. /// necessary, however it allows the generated type to keep track of changes and lock around them so that the config will auto-save.
  54. /// </para>
  55. /// <para>
  56. /// All of the attributes in the <see cref="Attributes"/> namespace are handled as described by them.
  57. /// </para>
  58. /// <para>
  59. /// If the <typeparamref name="T"/> declares a public or protected, <see langword="virtual"/>
  60. /// method <c>Changed()</c>, then that method may be called to artificially signal to the runtime that the content of the object
  61. /// has changed. That method will also be called after the write locks are released when a property is set anywhere in the owning
  62. /// tree. This will only be called on the outermost generated object of the config structure, even if the change being signaled
  63. /// is somewhere deep into the tree. TODO: is this a good idea?
  64. /// </para>
  65. /// <para>
  66. /// Similarly, <typeparamref name="T"/> can declare a public or protected, <see langword="virtual"/>
  67. /// method <c>OnReload()</c>, which will be called on the filesystem reader thread after the object has been repopulated with new data
  68. /// values. It will be called <i>after</i> the write lock for this object is released. This will only be called on the outermost generated
  69. /// object of the config structure.
  70. /// </para>
  71. /// <para>
  72. /// TODO: describe details of generated stores
  73. /// </para>
  74. /// </remarks>
  75. /// <typeparam name="T">the type to wrap</typeparam>
  76. /// <param name="cfg">the <see cref="Config"/> to register to</param>
  77. /// <param name="loadSync">whether to synchronously load the content, or trigger an async load</param>
  78. /// <returns>a generated instance of <typeparamref name="T"/> as a special <see cref="IConfigStore"/></returns>
  79. public static T Generated<T>(this Config cfg, bool loadSync = true) where T : class
  80. {
  81. var ret = GeneratedStore.Create<T>();
  82. cfg.SetStore(ret as IConfigStore);
  83. if (loadSync)
  84. cfg.LoadSync();
  85. else
  86. cfg.LoadAsync();
  87. return ret;
  88. }
  89. }
  90. internal static class GeneratedStore
  91. {
  92. internal interface IGeneratedStore
  93. {
  94. Type Type { get; }
  95. IGeneratedStore Parent { get; }
  96. Impl Impl { get; }
  97. void OnReload();
  98. void Changed();
  99. Value Serialize();
  100. void Deserialize(Value val);
  101. }
  102. internal class Impl : IConfigStore
  103. {
  104. private IGeneratedStore generated;
  105. internal static ConstructorInfo Ctor = typeof(Impl).GetConstructor(new[] { typeof(IGeneratedStore) });
  106. public Impl(IGeneratedStore store) => generated = store;
  107. private readonly AutoResetEvent resetEvent = new AutoResetEvent(false);
  108. public WaitHandle SyncObject => resetEvent;
  109. internal static MethodInfo SyncObjectGetMethod = typeof(Impl).GetProperty(nameof(SyncObject)).GetGetMethod();
  110. public ReaderWriterLockSlim WriteSyncObject { get; } = new ReaderWriterLockSlim();
  111. internal static MethodInfo WriteSyncObjectGetMethod = typeof(Impl).GetProperty(nameof(WriteSyncObject)).GetGetMethod();
  112. internal static MethodInfo ImplSignalChangedMethod = typeof(Impl).GetMethod(nameof(ImplSignalChanged));
  113. public static void ImplSignalChanged(IGeneratedStore s) => FindImpl(s).SignalChanged();
  114. public void SignalChanged() => resetEvent.Set();
  115. internal static MethodInfo ImplInvokeChangedMethod = typeof(Impl).GetMethod(nameof(ImplInvokeChanged));
  116. public static void ImplInvokeChanged(IGeneratedStore s) => FindImpl(s).InvokeChanged();
  117. public void InvokeChanged() => generated.Changed();
  118. internal static MethodInfo ImplTakeReadMethod = typeof(Impl).GetMethod(nameof(ImplTakeRead));
  119. public static void ImplTakeRead(IGeneratedStore s) => FindImpl(s).TakeRead();
  120. public void TakeRead() => WriteSyncObject.EnterReadLock();
  121. internal static MethodInfo ImplReleaseReadMethod = typeof(Impl).GetMethod(nameof(ImplReleaseRead));
  122. public static void ImplReleaseRead(IGeneratedStore s) => FindImpl(s).ReleaseRead();
  123. public void ReleaseRead() => WriteSyncObject.ExitReadLock();
  124. internal static MethodInfo ImplTakeWriteMethod = typeof(Impl).GetMethod(nameof(ImplTakeWrite));
  125. public static void ImplTakeWrite(IGeneratedStore s) => FindImpl(s).TakeWrite();
  126. public void TakeWrite() => WriteSyncObject.EnterWriteLock();
  127. internal static MethodInfo ImplReleaseWriteMethod = typeof(Impl).GetMethod(nameof(ImplReleaseWrite));
  128. public static void ImplReleaseWrite(IGeneratedStore s) => FindImpl(s).ReleaseWrite();
  129. public void ReleaseWrite() => WriteSyncObject.ExitWriteLock();
  130. internal static MethodInfo FindImplMethod = typeof(Impl).GetMethod(nameof(FindImpl));
  131. public static Impl FindImpl(IGeneratedStore store)
  132. {
  133. while (store?.Parent != null) store = store.Parent; // walk to the top of the tree
  134. return store?.Impl;
  135. }
  136. internal static MethodInfo ReadFromMethod = typeof(Impl).GetMethod(nameof(ReadFrom));
  137. public void ReadFrom(IConfigProvider provider)
  138. {
  139. var values = provider.Load();
  140. Logger.config.Debug("Generated impl ReadFrom");
  141. Logger.config.Debug($"Read {values}");
  142. generated.Deserialize(values);
  143. ReleaseWrite();
  144. generated.OnReload();
  145. TakeWrite(); // must take again for runtime to be happy (which is unfortunate)
  146. }
  147. internal static MethodInfo WriteToMethod = typeof(Impl).GetMethod(nameof(WriteTo));
  148. public void WriteTo(IConfigProvider provider)
  149. {
  150. var values = generated.Serialize();
  151. Logger.config.Debug("Generated impl WriteTo");
  152. Logger.config.Debug($"Serialized {values}");
  153. provider.Store(values);
  154. }
  155. }
  156. private static readonly Dictionary<Type, (GeneratedStoreCreator ctor, Type type)> generatedCreators = new Dictionary<Type, (GeneratedStoreCreator ctor, Type type)>();
  157. public static T Create<T>() where T : class => (T)Create(typeof(T));
  158. public static IConfigStore Create(Type type) => Create(type, null);
  159. private static readonly MethodInfo CreateGParent =
  160. typeof(GeneratedStore).GetMethod(nameof(Create), BindingFlags.NonPublic | BindingFlags.Static, null,
  161. CallingConventions.Any, new[] { typeof(IGeneratedStore) }, Array.Empty<ParameterModifier>());
  162. internal static T Create<T>(IGeneratedStore parent) where T : class => (T)Create(typeof(T), parent);
  163. private static IConfigStore Create(Type type, IGeneratedStore parent)
  164. => GetCreator(type)(parent);
  165. internal static GeneratedStoreCreator GetCreator(Type t)
  166. {
  167. if (generatedCreators.TryGetValue(t, out var gen))
  168. return gen.ctor;
  169. else
  170. {
  171. gen = MakeCreator(t);
  172. generatedCreators.Add(t, gen);
  173. return gen.ctor;
  174. }
  175. }
  176. internal static Type GetGeneratedType(Type t)
  177. {
  178. if (generatedCreators.TryGetValue(t, out var gen))
  179. return gen.type;
  180. else
  181. {
  182. gen = MakeCreator(t);
  183. generatedCreators.Add(t, gen);
  184. return gen.type;
  185. }
  186. }
  187. internal const string GeneratedAssemblyName = "IPA.Config.Generated";
  188. private static AssemblyBuilder assembly = null;
  189. private static AssemblyBuilder Assembly
  190. {
  191. get
  192. {
  193. if (assembly == null)
  194. {
  195. var name = new AssemblyName(GeneratedAssemblyName);
  196. assembly = AppDomain.CurrentDomain.DefineDynamicAssembly(name, AssemblyBuilderAccess.RunAndSave);
  197. }
  198. return assembly;
  199. }
  200. }
  201. internal static void DebugSaveAssembly(string file)
  202. {
  203. Assembly.Save(file);
  204. }
  205. private static ModuleBuilder module = null;
  206. private static ModuleBuilder Module
  207. {
  208. get
  209. {
  210. if (module == null)
  211. module = Assembly.DefineDynamicModule(Assembly.GetName().Name, Assembly.GetName().Name + ".dll");
  212. return module;
  213. }
  214. }
  215. private class SerializedMemberInfo
  216. {
  217. public string Name;
  218. public MemberInfo Member;
  219. public Type Type;
  220. public bool AllowNull;
  221. public bool IsVirtual;
  222. public bool IsField;
  223. public bool IsNullable; // signifies whether this is a Nullable<T>
  224. public bool HasConverter;
  225. public bool IsGenericConverter; // used so we can call directly to the generic version if it is
  226. public Type Converter;
  227. public Type ConverterBase;
  228. public Type ConverterTarget;
  229. public FieldInfo ConverterField;
  230. // invalid for objects with IsNullabe false
  231. public Type NullableWrappedType => Nullable.GetUnderlyingType(Type);
  232. // invalid for objects with IsNullabe false
  233. public PropertyInfo Nullable_HasValue => Type.GetProperty(nameof(Nullable<int>.HasValue));
  234. // invalid for objects with IsNullabe false
  235. public PropertyInfo Nullable_Value => Type.GetProperty(nameof(Nullable<int>.Value));
  236. // invalid for objects with IsNullabe false
  237. public ConstructorInfo Nullable_Construct => Type.GetConstructor(new[] { NullableWrappedType });
  238. }
  239. internal delegate IConfigStore GeneratedStoreCreator(IGeneratedStore parent);
  240. private static (GeneratedStoreCreator ctor, Type type) MakeCreator(Type type)
  241. { // note that this does not and should not use converters by default for everything
  242. if (!type.IsClass) throw new ArgumentException("Config type is not a class");
  243. var baseCtor = type.GetConstructor(Type.EmptyTypes); // get a default constructor
  244. if (baseCtor == null)
  245. throw new ArgumentException("Config type does not have a public parameterless constructor");
  246. var typeBuilder = Module.DefineType($"{type.FullName}<Generated>",
  247. TypeAttributes.Public | TypeAttributes.Sealed | TypeAttributes.Class, type);
  248. var typeField = typeBuilder.DefineField("<>_type", typeof(Type), FieldAttributes.Private | FieldAttributes.InitOnly);
  249. var implField = typeBuilder.DefineField("<>_impl", typeof(Impl), FieldAttributes.Private | FieldAttributes.InitOnly);
  250. var parentField = typeBuilder.DefineField("<>_parent", typeof(IGeneratedStore), FieldAttributes.Private | FieldAttributes.InitOnly);
  251. // none of this can be Expressions because CompileToMethod requires a static target method for some dumbass reason
  252. #region Parse base object structure
  253. var baseChanged = type.GetMethod("Changed", BindingFlags.Public | BindingFlags.NonPublic | BindingFlags.Instance, null, Type.EmptyTypes, Array.Empty<ParameterModifier>());
  254. if (baseChanged != null && !baseChanged.IsVirtual) baseChanged = null; // limit this to just the one thing
  255. var baseOnReload = type.GetMethod("OnReload", BindingFlags.Public | BindingFlags.NonPublic | BindingFlags.Instance, null, Type.EmptyTypes, Array.Empty<ParameterModifier>());
  256. if (baseOnReload != null && !baseOnReload.IsVirtual) baseOnReload = null; // limit this to just the one thing
  257. var structure = new List<SerializedMemberInfo>();
  258. // TODO: support converters
  259. bool ProcessAttributesFor(ref SerializedMemberInfo member)
  260. {
  261. var attrs = member.Member.GetCustomAttributes(true);
  262. var ignores = attrs.Select(o => o as IgnoreAttribute).NonNull();
  263. if (ignores.Any()) // we ignore
  264. {
  265. return false;
  266. }
  267. var nonNullables = attrs.Select(o => o as NonNullableAttribute).NonNull();
  268. member.Name = member.Member.Name;
  269. member.IsNullable = member.Type.IsGenericType
  270. && member.Type.GetGenericTypeDefinition() == typeof(Nullable<>);
  271. member.AllowNull = !nonNullables.Any() && (!member.Type.IsValueType || member.IsNullable);
  272. var nameAttr = attrs.Select(o => o as SerializedNameAttribute).NonNull().FirstOrDefault();
  273. if (nameAttr != null)
  274. member.Name = nameAttr.Name;
  275. member.HasConverter = false;
  276. var converterAttr = attrs.Select(o => o as UseConverterAttribute).NonNull().FirstOrDefault();
  277. if (converterAttr != null)
  278. {
  279. member.Converter = converterAttr.ConverterType;
  280. member.IsGenericConverter = converterAttr.IsGenericConverter;
  281. if (member.Converter.GetConstructor(Type.EmptyTypes) == null)
  282. {
  283. Logger.config.Warn($"{type.FullName}'s member {member.Member.Name} requests a converter that is not default-constructible");
  284. goto endConverterAttr; // is there a better control flow structure to do this?
  285. }
  286. if (member.Converter.ContainsGenericParameters)
  287. {
  288. Logger.config.Warn($"{type.FullName}'s member {member.Member.Name} requests a converter that has unfilled type parameters");
  289. goto endConverterAttr;
  290. }
  291. if (member.Converter.IsInterface || member.Converter.IsAbstract)
  292. {
  293. Logger.config.Warn($"{type.FullName}'s member {member.Member.Name} requests a converter that is not constructible");
  294. goto endConverterAttr;
  295. }
  296. var targetType = converterAttr.ConverterTargetType;
  297. if (!member.IsGenericConverter)
  298. {
  299. try
  300. {
  301. var conv = Activator.CreateInstance(converterAttr.ConverterType) as IValueConverter;
  302. targetType = conv.Type;
  303. }
  304. catch
  305. {
  306. Logger.config.Warn($"{type.FullName}'s member {member.Member.Name} requests a converter who's target type could not be determined");
  307. goto endConverterAttr;
  308. }
  309. }
  310. if (targetType != member.Type)
  311. {
  312. Logger.config.Warn($"{type.FullName}'s member {member.Member.Name} requests a converter that is not of the member's type");
  313. goto endConverterAttr;
  314. }
  315. member.ConverterTarget = targetType;
  316. if (member.IsGenericConverter)
  317. member.ConverterBase = typeof(ValueConverter<>).MakeGenericType(targetType);
  318. else
  319. member.ConverterBase = typeof(IValueConverter);
  320. member.HasConverter = true;
  321. }
  322. endConverterAttr:
  323. return true;
  324. }
  325. // only looks at public/protected properties
  326. foreach (var prop in type.GetProperties(BindingFlags.Instance | BindingFlags.Public | BindingFlags.NonPublic))
  327. {
  328. if (prop.GetSetMethod(true)?.IsPrivate ?? true)
  329. { // we enter this block if the setter is inacessible or doesn't exist
  330. continue; // ignore props without setter
  331. }
  332. if (prop.GetGetMethod(true)?.IsPrivate ?? true)
  333. { // we enter this block if the getter is inacessible or doesn't exist
  334. continue; // ignore props without getter
  335. }
  336. var smi = new SerializedMemberInfo
  337. {
  338. Member = prop,
  339. IsVirtual = (prop.GetGetMethod(true)?.IsVirtual ?? false) ||
  340. (prop.GetSetMethod(true)?.IsVirtual ?? false),
  341. IsField = false,
  342. Type = prop.PropertyType
  343. };
  344. if (!ProcessAttributesFor(ref smi)) continue;
  345. structure.Add(smi);
  346. }
  347. // only look at public/protected fields
  348. foreach (var field in type.GetFields(BindingFlags.Instance | BindingFlags.Public | BindingFlags.NonPublic))
  349. {
  350. if (field.IsPrivate) continue;
  351. var smi = new SerializedMemberInfo
  352. {
  353. Member = field,
  354. IsVirtual = false,
  355. IsField = true,
  356. Type = field.FieldType
  357. };
  358. if (!ProcessAttributesFor(ref smi)) continue;
  359. structure.Add(smi);
  360. }
  361. #endregion
  362. #region Converter fields
  363. var uniqueConverterTypes = structure.Where(m => m.HasConverter).Select(m => m.Converter).Distinct().ToArray();
  364. var converterFields = new Dictionary<Type, FieldInfo>(uniqueConverterTypes.Length);
  365. foreach (var convType in uniqueConverterTypes)
  366. {
  367. var field = typeBuilder.DefineField($"<converter>_{convType}", convType,
  368. FieldAttributes.Private | FieldAttributes.InitOnly | FieldAttributes.Static);
  369. converterFields.Add(convType, field);
  370. foreach (var member in structure.Where(m => m.HasConverter && m.Converter == convType))
  371. member.ConverterField = field;
  372. }
  373. #endregion
  374. #region Static constructor
  375. var cctor = typeBuilder.DefineConstructor(MethodAttributes.Static, CallingConventions.Standard, Type.EmptyTypes);
  376. {
  377. var il = cctor.GetILGenerator();
  378. foreach (var kvp in converterFields)
  379. {
  380. var typeCtor = kvp.Key.GetConstructor(Type.EmptyTypes);
  381. il.Emit(OpCodes.Newobj, typeCtor);
  382. il.Emit(OpCodes.Stsfld, kvp.Value);
  383. }
  384. il.Emit(OpCodes.Ret);
  385. }
  386. #endregion
  387. #region Constructor
  388. var ctor = typeBuilder.DefineConstructor(MethodAttributes.Public, CallingConventions.Standard, new[] { typeof(IGeneratedStore) });
  389. {
  390. var il = ctor.GetILGenerator();
  391. il.Emit(OpCodes.Ldarg_0); // keep this at bottom of stack
  392. il.Emit(OpCodes.Dup);
  393. il.Emit(OpCodes.Call, baseCtor);
  394. il.Emit(OpCodes.Dup);
  395. il.Emit(OpCodes.Ldarg_1); // load parent
  396. il.Emit(OpCodes.Stfld, parentField);
  397. il.Emit(OpCodes.Dup);
  398. EmitTypeof(il, type);
  399. il.Emit(OpCodes.Stfld, typeField);
  400. il.Emit(OpCodes.Dup);
  401. il.Emit(OpCodes.Dup);
  402. il.Emit(OpCodes.Newobj, Impl.Ctor);
  403. il.Emit(OpCodes.Stfld, implField);
  404. var GetLocal = MakeGetLocal(il);
  405. foreach (var member in structure)
  406. {
  407. EmitMemberFix(il, member, GetLocal);
  408. }
  409. il.Emit(OpCodes.Pop);
  410. il.Emit(OpCodes.Ret);
  411. }
  412. #endregion
  413. const MethodAttributes propertyMethodAttr = MethodAttributes.Public | MethodAttributes.SpecialName | MethodAttributes.HideBySig;
  414. const MethodAttributes virtualPropertyMethodAttr = propertyMethodAttr | MethodAttributes.Virtual | MethodAttributes.Final;
  415. const MethodAttributes virtualMemberMethod = MethodAttributes.Public | MethodAttributes.Virtual | MethodAttributes.HideBySig | MethodAttributes.Final;
  416. #region IGeneratedStore
  417. typeBuilder.AddInterfaceImplementation(typeof(IGeneratedStore));
  418. var IGeneratedStore_t = typeof(IGeneratedStore);
  419. var IGeneratedStore_GetImpl = IGeneratedStore_t.GetProperty(nameof(IGeneratedStore.Impl)).GetGetMethod();
  420. var IGeneratedStore_GetType = IGeneratedStore_t.GetProperty(nameof(IGeneratedStore.Type)).GetGetMethod();
  421. var IGeneratedStore_GetParent = IGeneratedStore_t.GetProperty(nameof(IGeneratedStore.Parent)).GetGetMethod();
  422. var IGeneratedStore_Serialize = IGeneratedStore_t.GetMethod(nameof(IGeneratedStore.Serialize));
  423. var IGeneratedStore_Deserialize = IGeneratedStore_t.GetMethod(nameof(IGeneratedStore.Deserialize));
  424. var IGeneratedStore_OnReload = IGeneratedStore_t.GetMethod(nameof(IGeneratedStore.OnReload));
  425. var IGeneratedStore_Changed = IGeneratedStore_t.GetMethod(nameof(IGeneratedStore.Changed));
  426. #region IGeneratedStore.OnReload
  427. var onReload = typeBuilder.DefineMethod($"<>{nameof(IGeneratedStore.OnReload)}", virtualMemberMethod, null, Type.EmptyTypes);
  428. typeBuilder.DefineMethodOverride(onReload, IGeneratedStore_OnReload);
  429. if (baseOnReload != null) typeBuilder.DefineMethodOverride(onReload, baseOnReload);
  430. {
  431. var il = onReload.GetILGenerator();
  432. if (baseOnReload != null)
  433. {
  434. il.Emit(OpCodes.Ldarg_0); // load this
  435. il.Emit(OpCodes.Tailcall);
  436. il.Emit(OpCodes.Call, baseOnReload); // load impl field
  437. }
  438. il.Emit(OpCodes.Ret);
  439. }
  440. #endregion
  441. #region IGeneratedStore.Impl
  442. var implProp = typeBuilder.DefineProperty(nameof(IGeneratedStore.Impl), PropertyAttributes.None, typeof(Impl), null);
  443. var implPropGet = typeBuilder.DefineMethod($"<g>{nameof(IGeneratedStore.Impl)}", virtualPropertyMethodAttr, implProp.PropertyType, Type.EmptyTypes);
  444. implProp.SetGetMethod(implPropGet);
  445. typeBuilder.DefineMethodOverride(implPropGet, IGeneratedStore_GetImpl);
  446. {
  447. var il = implPropGet.GetILGenerator();
  448. il.Emit(OpCodes.Ldarg_0); // load this
  449. il.Emit(OpCodes.Ldfld, implField); // load impl field
  450. il.Emit(OpCodes.Ret);
  451. }
  452. #endregion
  453. #region IGeneratedStore.Type
  454. var typeProp = typeBuilder.DefineProperty(nameof(IGeneratedStore.Type), PropertyAttributes.None, typeof(Type), null);
  455. var typePropGet = typeBuilder.DefineMethod($"<g>{nameof(IGeneratedStore.Type)}", virtualPropertyMethodAttr, typeProp.PropertyType, Type.EmptyTypes);
  456. typeProp.SetGetMethod(typePropGet);
  457. typeBuilder.DefineMethodOverride(typePropGet, IGeneratedStore_GetType);
  458. {
  459. var il = typePropGet.GetILGenerator();
  460. il.Emit(OpCodes.Ldarg_0); // load this
  461. il.Emit(OpCodes.Ldfld, typeField); // load impl field
  462. il.Emit(OpCodes.Ret);
  463. }
  464. #endregion
  465. #region IGeneratedStore.Parent
  466. var parentProp = typeBuilder.DefineProperty(nameof(IGeneratedStore.Parent), PropertyAttributes.None, typeof(IGeneratedStore), null);
  467. var parentPropGet = typeBuilder.DefineMethod($"<g>{nameof(IGeneratedStore.Parent)}", virtualPropertyMethodAttr, parentProp.PropertyType, Type.EmptyTypes);
  468. parentProp.SetGetMethod(parentPropGet);
  469. typeBuilder.DefineMethodOverride(parentPropGet, IGeneratedStore_GetParent);
  470. {
  471. var il = parentPropGet.GetILGenerator();
  472. il.Emit(OpCodes.Ldarg_0); // load this
  473. il.Emit(OpCodes.Ldfld, parentField); // load impl field
  474. il.Emit(OpCodes.Ret);
  475. }
  476. #endregion
  477. #region IGeneratedStore.Serialize
  478. var serializeGen = typeBuilder.DefineMethod($"<>{nameof(IGeneratedStore.Serialize)}", virtualPropertyMethodAttr, IGeneratedStore_Serialize.ReturnType, Type.EmptyTypes);
  479. typeBuilder.DefineMethodOverride(serializeGen, IGeneratedStore_Serialize);
  480. { // this is non-locking because the only code that will call this will already own the correct lock
  481. var il = serializeGen.GetILGenerator();
  482. var Map_Add = typeof(Map).GetMethod(nameof(Map.Add));
  483. var mapLocal = il.DeclareLocal(typeof(Map));
  484. var GetLocal = MakeGetLocal(il);
  485. var valLocal = GetLocal(typeof(Value));
  486. il.Emit(OpCodes.Call, typeof(Value).GetMethod(nameof(Value.Map)));
  487. il.Emit(OpCodes.Stloc, mapLocal);
  488. foreach (var member in structure)
  489. {
  490. EmitSerializeMember(il, member, GetLocal);
  491. il.Emit(OpCodes.Stloc, valLocal);
  492. il.Emit(OpCodes.Ldloc, mapLocal);
  493. il.Emit(OpCodes.Ldstr, member.Name);
  494. il.Emit(OpCodes.Ldloc, valLocal);
  495. il.Emit(OpCodes.Call, Map_Add);
  496. }
  497. il.Emit(OpCodes.Ldloc, mapLocal);
  498. il.Emit(OpCodes.Ret);
  499. }
  500. #endregion
  501. #region IGeneratedStore.Deserialize
  502. var deserializeGen = typeBuilder.DefineMethod($"<>{nameof(IGeneratedStore.Deserialize)}", virtualPropertyMethodAttr, null,
  503. new[] { IGeneratedStore_Deserialize.GetParameters()[0].ParameterType });
  504. typeBuilder.DefineMethodOverride(deserializeGen, IGeneratedStore_Deserialize);
  505. { // this is non-locking because the only code that will call this will already own the correct lock
  506. var il = deserializeGen.GetILGenerator();
  507. var Map_t = typeof(Map);
  508. var Map_TryGetValue = Map_t.GetMethod(nameof(Map.TryGetValue));
  509. var Object_GetType = typeof(object).GetMethod(nameof(Object.GetType));
  510. var valueLocal = il.DeclareLocal(typeof(Value));
  511. var mapLocal = il.DeclareLocal(typeof(Map));
  512. var nonNull = il.DefineLabel();
  513. il.Emit(OpCodes.Ldarg_1);
  514. il.Emit(OpCodes.Brtrue, nonNull);
  515. EmitLogError(il, "Attempting to deserialize null", tailcall: true);
  516. il.Emit(OpCodes.Ret);
  517. il.MarkLabel(nonNull);
  518. il.Emit(OpCodes.Ldarg_1);
  519. il.Emit(OpCodes.Isinst, Map_t);
  520. il.Emit(OpCodes.Dup); // duplicate cloned value
  521. il.Emit(OpCodes.Stloc, mapLocal);
  522. var notMapError = il.DefineLabel();
  523. il.Emit(OpCodes.Brtrue, notMapError);
  524. // handle error
  525. EmitLogError(il, $"Invalid root for deserializing {type.FullName}", tailcall: true,
  526. expected: il => EmitTypeof(il, Map_t), found: il =>
  527. {
  528. il.Emit(OpCodes.Ldarg_1);
  529. il.Emit(OpCodes.Callvirt, Object_GetType);
  530. });
  531. il.Emit(OpCodes.Ret);
  532. var nextLabel = notMapError;
  533. var GetLocal = MakeGetLocal(il);
  534. // head of stack is Map instance
  535. foreach (var member in structure)
  536. {
  537. il.MarkLabel(nextLabel);
  538. nextLabel = il.DefineLabel();
  539. var endErrorLabel = il.DefineLabel();
  540. il.Emit(OpCodes.Ldloc, mapLocal);
  541. il.Emit(OpCodes.Ldstr, member.Name);
  542. il.Emit(OpCodes.Ldloca_S, valueLocal);
  543. il.Emit(OpCodes.Call, Map_TryGetValue);
  544. il.Emit(OpCodes.Brtrue_S, endErrorLabel);
  545. EmitLogError(il, $"Missing key {member.Name}", tailcall: false);
  546. il.Emit(OpCodes.Br, nextLabel);
  547. il.MarkLabel(endErrorLabel);
  548. il.Emit(OpCodes.Ldloc_S, valueLocal);
  549. EmitDeserializeMember(il, member, nextLabel, il => il.Emit(OpCodes.Ldloc_S, valueLocal), GetLocal);
  550. }
  551. il.MarkLabel(nextLabel);
  552. il.Emit(OpCodes.Ret);
  553. }
  554. #endregion
  555. #endregion
  556. #region IConfigStore
  557. typeBuilder.AddInterfaceImplementation(typeof(IConfigStore));
  558. var IConfigStore_t = typeof(IConfigStore);
  559. var IConfigStore_GetSyncObject = IConfigStore_t.GetProperty(nameof(IConfigStore.SyncObject)).GetGetMethod();
  560. var IConfigStore_GetWriteSyncObject = IConfigStore_t.GetProperty(nameof(IConfigStore.WriteSyncObject)).GetGetMethod();
  561. var IConfigStore_WriteTo = IConfigStore_t.GetMethod(nameof(IConfigStore.WriteTo));
  562. var IConfigStore_ReadFrom = IConfigStore_t.GetMethod(nameof(IConfigStore.ReadFrom));
  563. #region IConfigStore.SyncObject
  564. var syncObjProp = typeBuilder.DefineProperty(nameof(IConfigStore.SyncObject), PropertyAttributes.None, typeof(WaitHandle), null);
  565. var syncObjPropGet = typeBuilder.DefineMethod($"<g>{nameof(IConfigStore.SyncObject)}", virtualPropertyMethodAttr, syncObjProp.PropertyType, Type.EmptyTypes);
  566. syncObjProp.SetGetMethod(syncObjPropGet);
  567. typeBuilder.DefineMethodOverride(syncObjPropGet, IConfigStore_GetSyncObject);
  568. {
  569. var il = syncObjPropGet.GetILGenerator();
  570. il.Emit(OpCodes.Ldarg_0);
  571. il.Emit(OpCodes.Call, Impl.FindImplMethod);
  572. il.Emit(OpCodes.Tailcall);
  573. il.Emit(OpCodes.Call, Impl.SyncObjectGetMethod);
  574. il.Emit(OpCodes.Ret);
  575. }
  576. #endregion
  577. #region IConfigStore.WriteSyncObject
  578. var writeSyncObjProp = typeBuilder.DefineProperty(nameof(IConfigStore.WriteSyncObject), PropertyAttributes.None, typeof(WaitHandle), null);
  579. var writeSyncObjPropGet = typeBuilder.DefineMethod($"<g>{nameof(IConfigStore.WriteSyncObject)}", virtualPropertyMethodAttr, writeSyncObjProp.PropertyType, Type.EmptyTypes);
  580. writeSyncObjProp.SetGetMethod(writeSyncObjPropGet);
  581. typeBuilder.DefineMethodOverride(writeSyncObjPropGet, IConfigStore_GetWriteSyncObject);
  582. {
  583. var il = writeSyncObjPropGet.GetILGenerator();
  584. il.Emit(OpCodes.Ldarg_0);
  585. il.Emit(OpCodes.Call, Impl.FindImplMethod);
  586. il.Emit(OpCodes.Tailcall);
  587. il.Emit(OpCodes.Call, Impl.WriteSyncObjectGetMethod);
  588. il.Emit(OpCodes.Ret);
  589. }
  590. #endregion
  591. #region IConfigStore.WriteTo
  592. var writeTo = typeBuilder.DefineMethod($"<>{nameof(IConfigStore.WriteTo)}", virtualMemberMethod, null, new[] { typeof(IConfigProvider) });
  593. typeBuilder.DefineMethodOverride(writeTo, IConfigStore_WriteTo);
  594. {
  595. var il = writeTo.GetILGenerator();
  596. il.Emit(OpCodes.Ldarg_0);
  597. il.Emit(OpCodes.Call, Impl.FindImplMethod);
  598. il.Emit(OpCodes.Ldarg_1);
  599. il.Emit(OpCodes.Tailcall);
  600. il.Emit(OpCodes.Call, Impl.WriteToMethod);
  601. il.Emit(OpCodes.Ret);
  602. }
  603. #endregion
  604. #region IConfigStore.ReadFrom
  605. var readFrom = typeBuilder.DefineMethod($"<>{nameof(IConfigStore.ReadFrom)}", virtualMemberMethod, null, new[] { typeof(IConfigProvider) });
  606. typeBuilder.DefineMethodOverride(readFrom, IConfigStore_ReadFrom);
  607. {
  608. var il = readFrom.GetILGenerator();
  609. il.Emit(OpCodes.Ldarg_0);
  610. il.Emit(OpCodes.Call, Impl.FindImplMethod);
  611. il.Emit(OpCodes.Ldarg_1);
  612. il.Emit(OpCodes.Tailcall);
  613. il.Emit(OpCodes.Call, Impl.ReadFromMethod);
  614. il.Emit(OpCodes.Ret);
  615. }
  616. #endregion
  617. #endregion
  618. #region Changed
  619. var coreChanged = typeBuilder.DefineMethod(
  620. "<>Changed",
  621. virtualMemberMethod,
  622. null, Type.EmptyTypes);
  623. {
  624. var il = coreChanged.GetILGenerator();
  625. il.Emit(OpCodes.Ldarg_0);
  626. il.Emit(OpCodes.Call, Impl.ImplSignalChangedMethod);
  627. il.Emit(OpCodes.Ret); // simply call our impl's SignalChanged method and return
  628. }
  629. if (baseChanged != null) {
  630. var changedMethod = typeBuilder.DefineMethod( // copy to override baseChanged
  631. baseChanged.Name,
  632. virtualMemberMethod,
  633. null, Type.EmptyTypes);
  634. typeBuilder.DefineMethodOverride(changedMethod, baseChanged);
  635. {
  636. var il = changedMethod.GetILGenerator();
  637. il.Emit(OpCodes.Ldarg_0);
  638. il.Emit(OpCodes.Call, baseChanged); // call base
  639. il.Emit(OpCodes.Ldarg_0);
  640. il.Emit(OpCodes.Tailcall);
  641. il.Emit(OpCodes.Call, coreChanged); // call back to the core change method
  642. il.Emit(OpCodes.Ret);
  643. }
  644. coreChanged = changedMethod; // switch to calling this version instead of just the default
  645. }
  646. typeBuilder.DefineMethodOverride(coreChanged, IGeneratedStore_Changed);
  647. #endregion
  648. #region Members
  649. foreach (var member in structure.Where(m => m.IsVirtual))
  650. { // IsVirtual implies !IsField
  651. var prop = member.Member as PropertyInfo;
  652. var get = prop.GetGetMethod(true);
  653. var set = prop.GetSetMethod(true);
  654. var propBuilder = typeBuilder.DefineProperty($"{member.Name}#", PropertyAttributes.None, member.Type, null);
  655. var propGet = typeBuilder.DefineMethod($"<g>{propBuilder.Name}", virtualPropertyMethodAttr, member.Type, Type.EmptyTypes);
  656. propBuilder.SetGetMethod(propGet);
  657. typeBuilder.DefineMethodOverride(propGet, get);
  658. {
  659. var il = propGet.GetILGenerator();
  660. var local = il.DeclareLocal(member.Type);
  661. il.Emit(OpCodes.Ldarg_0);
  662. il.Emit(OpCodes.Call, Impl.ImplTakeReadMethod); // take the read lock
  663. il.BeginExceptionBlock();
  664. il.Emit(OpCodes.Ldarg_0);
  665. il.Emit(OpCodes.Call, get); // call base getter
  666. il.Emit(OpCodes.Stloc, local);
  667. il.BeginFinallyBlock();
  668. il.Emit(OpCodes.Ldarg_0);
  669. il.Emit(OpCodes.Call, Impl.ImplReleaseReadMethod); // release the read lock
  670. il.EndExceptionBlock();
  671. il.Emit(OpCodes.Ldloc, local);
  672. il.Emit(OpCodes.Ret);
  673. }
  674. var propSet = typeBuilder.DefineMethod($"<s>{propBuilder.Name}", virtualPropertyMethodAttr, null, new[] { member.Type });
  675. propBuilder.SetSetMethod(propSet);
  676. typeBuilder.DefineMethodOverride(propSet, set);
  677. { // TODO: decide if i want to correct the value before or after i take the write lock
  678. var il = propSet.GetILGenerator();
  679. il.Emit(OpCodes.Ldarg_0);
  680. il.Emit(OpCodes.Call, Impl.ImplTakeWriteMethod); // take the write lock
  681. il.BeginExceptionBlock();
  682. il.Emit(OpCodes.Ldarg_0);
  683. il.Emit(OpCodes.Ldarg_1);
  684. EmitCorrectMember(il, member);
  685. il.Emit(OpCodes.Call, set);
  686. il.BeginFinallyBlock();
  687. il.Emit(OpCodes.Ldarg_0);
  688. il.Emit(OpCodes.Call, Impl.ImplReleaseWriteMethod); // release the write lock
  689. il.EndExceptionBlock();
  690. il.Emit(OpCodes.Ldarg_0);
  691. il.Emit(OpCodes.Call, Impl.ImplInvokeChangedMethod);
  692. il.Emit(OpCodes.Ret);
  693. }
  694. }
  695. #endregion
  696. var genType = typeBuilder.CreateType();
  697. var parentParam = Expression.Parameter(typeof(IGeneratedStore), "parent");
  698. var creatorDel = Expression.Lambda<GeneratedStoreCreator>(
  699. Expression.New(ctor, parentParam), parentParam
  700. ).Compile();
  701. return (creatorDel, genType);
  702. }
  703. private delegate LocalBuilder GetLocal(Type type, int idx = 0);
  704. private static GetLocal MakeGetLocal(ILGenerator il)
  705. { // TODO: improve this shit a bit so that i can release a hold of a variable and do more auto managing
  706. var locals = new List<LocalBuilder>();
  707. LocalBuilder GetLocal(Type ty, int i = 0)
  708. { // TODO: pull this garbage out somewhere sane somehow
  709. var builder = locals.Where(b => b.LocalType == ty).Skip(i).FirstOrDefault();
  710. if (builder == null)
  711. {
  712. builder = il.DeclareLocal(ty);
  713. locals.Add(builder);
  714. }
  715. return builder;
  716. }
  717. return GetLocal;
  718. }
  719. private static readonly MethodInfo LogErrorMethod = typeof(GeneratedStore).GetMethod(nameof(LogError), BindingFlags.NonPublic | BindingFlags.Static);
  720. internal static void LogError(Type expected, Type found, string message)
  721. {
  722. Logger.config.Notice($"{message}{(expected == null ? "" : $" (expected {expected}, found {found?.ToString() ?? "null"})")}");
  723. }
  724. private static readonly MethodInfo LogWarningMethod = typeof(GeneratedStore).GetMethod(nameof(LogWarning), BindingFlags.NonPublic | BindingFlags.Static);
  725. internal static void LogWarning(string message)
  726. {
  727. Logger.config.Warn(message);
  728. }
  729. private static readonly MethodInfo LogWarningExceptionMethod = typeof(GeneratedStore).GetMethod(nameof(LogWarningException), BindingFlags.NonPublic | BindingFlags.Static);
  730. internal static void LogWarningException(Exception exception)
  731. {
  732. Logger.config.Warn(exception);
  733. }
  734. private static bool NeedsCorrection(SerializedMemberInfo member)
  735. {
  736. return false;
  737. }
  738. // expects start value on stack, exits with final value on stack
  739. private static void EmitCorrectMember(ILGenerator il, SerializedMemberInfo member)
  740. {
  741. if (!NeedsCorrection(member)) return;
  742. // TODO: impl
  743. }
  744. // expects the this param to be on the stack
  745. private static void EmitMemberFix(ILGenerator il, SerializedMemberInfo member, GetLocal GetLocal)
  746. {
  747. if (!NeedsCorrection(member)) return;
  748. var local = GetLocal(member.Type);
  749. EmitLoad(il, member); // load the member
  750. EmitCorrectMember(il, member); // correct it
  751. il.Emit(OpCodes.Stloc, local);
  752. EmitStore(il, member, il => il.Emit(OpCodes.Ldloc, local));
  753. }
  754. #region Utility
  755. private static void EmitLoad(ILGenerator il, SerializedMemberInfo member)
  756. {
  757. il.Emit(OpCodes.Ldarg_0); // load this
  758. if (member.IsField)
  759. il.Emit(OpCodes.Ldfld, member.Member as FieldInfo);
  760. else
  761. { // member is a property
  762. var prop = member.Member as PropertyInfo;
  763. var getter = prop.GetGetMethod();
  764. if (getter == null) throw new InvalidOperationException($"Property {member.Name} does not have a getter and is not ignored");
  765. il.Emit(OpCodes.Call, getter);
  766. }
  767. }
  768. private static void EmitStore(ILGenerator il, SerializedMemberInfo member, Action<ILGenerator> value)
  769. {
  770. il.Emit(OpCodes.Ldarg_0); // load this
  771. value(il);
  772. if (member.IsField)
  773. il.Emit(OpCodes.Stfld, member.Member as FieldInfo);
  774. else
  775. { // member is a property
  776. var prop = member.Member as PropertyInfo;
  777. var setter = prop.GetSetMethod();
  778. if (setter == null) throw new InvalidOperationException($"Property {member.Name} does not have a setter and is not ignored");
  779. il.Emit(OpCodes.Call, setter);
  780. }
  781. }
  782. private static void EmitWarnException(ILGenerator il, string v)
  783. {
  784. il.Emit(OpCodes.Ldstr, v);
  785. il.Emit(OpCodes.Call, LogWarningMethod);
  786. il.Emit(OpCodes.Call, LogWarningExceptionMethod);
  787. }
  788. private static void EmitLogError(ILGenerator il, string message, bool tailcall = false, Action<ILGenerator> expected = null, Action<ILGenerator> found = null)
  789. {
  790. if (expected == null) expected = il => il.Emit(OpCodes.Ldnull);
  791. if (found == null) found = il => il.Emit(OpCodes.Ldnull);
  792. expected(il);
  793. found(il);
  794. il.Emit(OpCodes.Ldstr, message);
  795. if (tailcall) il.Emit(OpCodes.Tailcall);
  796. il.Emit(OpCodes.Call, LogErrorMethod);
  797. }
  798. private static readonly MethodInfo Type_GetTypeFromHandle = typeof(Type).GetMethod(nameof(Type.GetTypeFromHandle));
  799. private static void EmitTypeof(ILGenerator il, Type type)
  800. {
  801. il.Emit(OpCodes.Ldtoken, type);
  802. il.Emit(OpCodes.Call, Type_GetTypeFromHandle);
  803. }
  804. private static Type Decimal_t = typeof(decimal);
  805. private static ConstructorInfo Decimal_FromFloat = Decimal_t.GetConstructor(new[] { typeof(float) });
  806. private static ConstructorInfo Decimal_FromDouble = Decimal_t.GetConstructor(new[] { typeof(double) });
  807. private static ConstructorInfo Decimal_FromInt = Decimal_t.GetConstructor(new[] { typeof(int) });
  808. private static ConstructorInfo Decimal_FromUInt = Decimal_t.GetConstructor(new[] { typeof(uint) });
  809. private static ConstructorInfo Decimal_FromLong = Decimal_t.GetConstructor(new[] { typeof(long) });
  810. private static ConstructorInfo Decimal_FromULong = Decimal_t.GetConstructor(new[] { typeof(ulong) });
  811. private static void EmitNumberConvertTo(ILGenerator il, Type to, Type from)
  812. { // WARNING: THIS USES THE NO-OVERFLOW OPCODES
  813. if (to == from) return;
  814. if (to == Decimal_t)
  815. {
  816. if (from == typeof(float)) il.Emit(OpCodes.Newobj, Decimal_FromFloat);
  817. else if (from == typeof(double)) il.Emit(OpCodes.Newobj, Decimal_FromDouble);
  818. else if (from == typeof(long)) il.Emit(OpCodes.Newobj, Decimal_FromLong);
  819. else if (from == typeof(ulong)) il.Emit(OpCodes.Newobj, Decimal_FromULong);
  820. else if (from == typeof(int)) il.Emit(OpCodes.Newobj, Decimal_FromInt);
  821. else if (from == typeof(uint)) il.Emit(OpCodes.Newobj, Decimal_FromUInt);
  822. else if (from == typeof(IntPtr))
  823. {
  824. EmitNumberConvertTo(il, typeof(long), from);
  825. EmitNumberConvertTo(il, to, typeof(long));
  826. }
  827. else if (from == typeof(UIntPtr))
  828. {
  829. EmitNumberConvertTo(il, typeof(ulong), from);
  830. EmitNumberConvertTo(il, to, typeof(ulong));
  831. }
  832. else
  833. { // if the source is anything else, we first convert to int because that can contain all other values
  834. EmitNumberConvertTo(il, typeof(int), from);
  835. EmitNumberConvertTo(il, to, typeof(int));
  836. };
  837. }
  838. else if (from == Decimal_t)
  839. {
  840. if (to == typeof(IntPtr))
  841. {
  842. EmitNumberConvertTo(il, typeof(long), from);
  843. EmitNumberConvertTo(il, to, typeof(long));
  844. }
  845. else if (to == typeof(UIntPtr))
  846. {
  847. EmitNumberConvertTo(il, typeof(ulong), from);
  848. EmitNumberConvertTo(il, to, typeof(ulong));
  849. }
  850. else
  851. {
  852. var method = Decimal_t.GetMethod($"To{to.Name}"); // conveniently, this is the pattern of the to* names
  853. il.Emit(OpCodes.Call, method);
  854. }
  855. }
  856. else if (to == typeof(IntPtr)) il.Emit(OpCodes.Conv_I);
  857. else if (to == typeof(UIntPtr)) il.Emit(OpCodes.Conv_U);
  858. else if (to == typeof(sbyte)) il.Emit(OpCodes.Conv_I1);
  859. else if (to == typeof(byte)) il.Emit(OpCodes.Conv_U1);
  860. else if (to == typeof(short)) il.Emit(OpCodes.Conv_I2);
  861. else if (to == typeof(ushort)) il.Emit(OpCodes.Conv_U2);
  862. else if (to == typeof(int)) il.Emit(OpCodes.Conv_I4);
  863. else if (to == typeof(uint)) il.Emit(OpCodes.Conv_U4);
  864. else if (to == typeof(long)) il.Emit(OpCodes.Conv_I8);
  865. else if (to == typeof(ulong)) il.Emit(OpCodes.Conv_U8);
  866. else if (to == typeof(float))
  867. {
  868. if (from == typeof(byte)
  869. || from == typeof(ushort)
  870. || from == typeof(uint)
  871. || from == typeof(ulong)
  872. || from == typeof(UIntPtr)) il.Emit(OpCodes.Conv_R_Un);
  873. il.Emit(OpCodes.Conv_R4);
  874. }
  875. else if (to == typeof(double))
  876. {
  877. if (from == typeof(byte)
  878. || from == typeof(ushort)
  879. || from == typeof(uint)
  880. || from == typeof(ulong)
  881. || from == typeof(UIntPtr)) il.Emit(OpCodes.Conv_R_Un);
  882. il.Emit(OpCodes.Conv_R8);
  883. }
  884. }
  885. private static void EmitCreateChildGenerated(ILGenerator il, Type childType)
  886. {
  887. var method = CreateGParent.MakeGenericMethod(childType);
  888. il.Emit(OpCodes.Ldarg_0);
  889. il.Emit(OpCodes.Call, method);
  890. }
  891. #endregion
  892. #region Serialize
  893. // emit takes no args, leaves Value at top of stack
  894. private static void EmitSerializeMember(ILGenerator il, SerializedMemberInfo member, GetLocal GetLocal)
  895. {
  896. EmitLoad(il, member);
  897. var endSerialize = il.DefineLabel();
  898. if (member.AllowNull)
  899. {
  900. var passedNull = il.DefineLabel();
  901. il.Emit(OpCodes.Dup);
  902. if (member.IsNullable)
  903. il.Emit(OpCodes.Call, member.Nullable_HasValue.GetGetMethod());
  904. il.Emit(OpCodes.Brtrue, passedNull);
  905. il.Emit(OpCodes.Pop);
  906. il.Emit(OpCodes.Ldnull);
  907. il.Emit(OpCodes.Br, endSerialize);
  908. il.MarkLabel(passedNull);
  909. }
  910. if (member.IsNullable)
  911. il.Emit(OpCodes.Call, member.Nullable_Value.GetGetMethod());
  912. var targetType = GetExpectedValueTypeForType(member.Type);
  913. if (member.HasConverter)
  914. {
  915. var stlocal = GetLocal(member.Type);
  916. var valLocal = GetLocal(typeof(Value));
  917. il.Emit(OpCodes.Stloc, stlocal);
  918. il.BeginExceptionBlock();
  919. il.Emit(OpCodes.Ldsfld, member.ConverterField);
  920. il.Emit(OpCodes.Ldloc, stlocal);
  921. if (member.IsGenericConverter)
  922. {
  923. var toValue = member.ConverterBase.GetMethod(nameof(ValueConverter<int>.ToValue),
  924. new[] { member.ConverterTarget, typeof(object) });
  925. il.Emit(OpCodes.Ldarg_0);
  926. il.Emit(OpCodes.Callvirt, toValue); // TODO: figure out how to devirtualize this
  927. }
  928. else
  929. {
  930. var toValue = typeof(IValueConverter).GetMethod(nameof(IValueConverter.ToValue),
  931. new[] { typeof(object), typeof(object) });
  932. il.Emit(OpCodes.Box);
  933. il.Emit(OpCodes.Ldarg_0);
  934. il.Emit(OpCodes.Callvirt, toValue);
  935. }
  936. il.Emit(OpCodes.Stloc, valLocal);
  937. il.BeginCatchBlock(typeof(Exception));
  938. EmitWarnException(il, "Error serializing member using converter");
  939. il.Emit(OpCodes.Ldnull);
  940. il.Emit(OpCodes.Stloc, valLocal);
  941. il.EndExceptionBlock();
  942. il.Emit(OpCodes.Ldloc, valLocal);
  943. }
  944. else if (targetType == typeof(Text))
  945. { // only happens when arg is a string or char
  946. var TextCreate = typeof(Value).GetMethod(nameof(Value.Text));
  947. if (member.Type == typeof(char))
  948. {
  949. var strFromChar = typeof(char).GetMethod(nameof(char.ToString), new[] { typeof(char) });
  950. il.Emit(OpCodes.Call, strFromChar);
  951. }
  952. il.Emit(OpCodes.Call, TextCreate);
  953. }
  954. else if (targetType == typeof(Boolean))
  955. {
  956. var BoolCreate = typeof(Value).GetMethod(nameof(Value.Bool));
  957. il.Emit(OpCodes.Call, BoolCreate);
  958. }
  959. else if (targetType == typeof(Integer))
  960. {
  961. var IntCreate = typeof(Value).GetMethod(nameof(Value.Integer));
  962. EmitNumberConvertTo(il, IntCreate.GetParameters()[0].ParameterType, member.Type);
  963. il.Emit(OpCodes.Call, IntCreate);
  964. }
  965. else if (targetType == typeof(FloatingPoint))
  966. {
  967. var FloatCreate = typeof(Value).GetMethod(nameof(Value.Float));
  968. EmitNumberConvertTo(il, FloatCreate.GetParameters()[0].ParameterType, member.Type);
  969. il.Emit(OpCodes.Call, FloatCreate);
  970. }
  971. else if (targetType == typeof(List))
  972. {
  973. // TODO: impl this (enumerables)
  974. il.Emit(OpCodes.Pop);
  975. il.Emit(OpCodes.Ldnull);
  976. }
  977. else if (targetType == typeof(Map))
  978. {
  979. // TODO: support other aggregate types
  980. // for now, we assume that its a generated type implementing IGeneratedStore
  981. var IGeneratedStore_Serialize = typeof(IGeneratedStore).GetMethod(nameof(IGeneratedStore.Serialize));
  982. il.Emit(OpCodes.Callvirt, IGeneratedStore_Serialize);
  983. }
  984. il.MarkLabel(endSerialize);
  985. // TODO: implement converters
  986. }
  987. #endregion
  988. #region Deserialize
  989. private static Type GetExpectedValueTypeForType(Type valT)
  990. {
  991. if (typeof(Value).IsAssignableFrom(valT)) // this is a Value subtype
  992. return valT;
  993. if (valT == typeof(string)
  994. || valT == typeof(char)) return typeof(Text);
  995. if (valT == typeof(bool)) return typeof(Boolean);
  996. if (valT == typeof(byte)
  997. || valT == typeof(sbyte)
  998. || valT == typeof(short)
  999. || valT == typeof(ushort)
  1000. || valT == typeof(int)
  1001. || valT == typeof(uint)
  1002. || valT == typeof(long)
  1003. || valT == typeof(IntPtr)) return typeof(Integer);
  1004. if (valT == typeof(float)
  1005. || valT == typeof(double)
  1006. || valT == typeof(decimal)
  1007. || valT == typeof(ulong) // ulong gets put into this, because decimal can hold it
  1008. || valT == typeof(UIntPtr)) return typeof(FloatingPoint);
  1009. if (typeof(IEnumerable).IsAssignableFrom(valT)) return typeof(List);
  1010. // TODO: fill this out the rest of the way
  1011. // TODO: support converters
  1012. return typeof(Map); // default for various objects
  1013. }
  1014. private static void EmitDeserializeGeneratedValue(ILGenerator il, Type targetType, Type srcType, GetLocal GetLocal)
  1015. {
  1016. var IGeneratedStore_Deserialize = typeof(IGeneratedStore).GetMethod(nameof(IGeneratedStore.Deserialize));
  1017. var valuel = GetLocal(srcType, 0);
  1018. il.Emit(OpCodes.Stloc, valuel);
  1019. EmitCreateChildGenerated(il, targetType);
  1020. il.Emit(OpCodes.Dup);
  1021. il.Emit(OpCodes.Ldloc, valuel);
  1022. il.Emit(OpCodes.Callvirt, IGeneratedStore_Deserialize);
  1023. }
  1024. private static void EmitDeserializeNullable(ILGenerator il, SerializedMemberInfo member, Type expected, GetLocal GetLocal)
  1025. {
  1026. EmitDeserializeValue(il, member.NullableWrappedType, expected, GetLocal);
  1027. il.Emit(OpCodes.Newobj, member.Nullable_Construct);
  1028. }
  1029. // top of stack is the Value to deserialize; the type will be as returned from GetExpectedValueTypeForType
  1030. // after, top of stack will be thing to write to field
  1031. private static void EmitDeserializeValue(ILGenerator il, Type targetType, Type expected, GetLocal GetLocal)
  1032. {
  1033. if (typeof(Value).IsAssignableFrom(targetType)) return; // do nothing
  1034. if (expected == typeof(Text))
  1035. {
  1036. var getter = expected.GetProperty(nameof(Text.Value)).GetGetMethod();
  1037. il.Emit(OpCodes.Call, getter);
  1038. if (targetType == typeof(char))
  1039. {
  1040. var strIndex = typeof(string).GetProperty("Chars").GetGetMethod(); // string's indexer is specially named Chars
  1041. il.Emit(OpCodes.Ldc_I4_0);
  1042. il.Emit(OpCodes.Call, strIndex);
  1043. }
  1044. }
  1045. else if (expected == typeof(Boolean))
  1046. {
  1047. var getter = expected.GetProperty(nameof(Boolean.Value)).GetGetMethod();
  1048. il.Emit(OpCodes.Call, getter);
  1049. }
  1050. else if (expected == typeof(Integer))
  1051. {
  1052. var getter = expected.GetProperty(nameof(Integer.Value)).GetGetMethod();
  1053. il.Emit(OpCodes.Call, getter);
  1054. EmitNumberConvertTo(il, targetType, getter.ReturnType);
  1055. }
  1056. else if (expected == typeof(FloatingPoint))
  1057. {
  1058. var getter = expected.GetProperty(nameof(FloatingPoint.Value)).GetGetMethod();
  1059. il.Emit(OpCodes.Call, getter);
  1060. EmitNumberConvertTo(il, targetType, getter.ReturnType);
  1061. } // TODO: implement stuff for lists and maps of various types (probably call out somewhere else to figure out what to do)
  1062. else if (expected == typeof(Map))
  1063. {
  1064. EmitDeserializeGeneratedValue(il, targetType, expected, GetLocal);
  1065. }
  1066. else // TODO: support converters
  1067. {
  1068. il.Emit(OpCodes.Pop);
  1069. il.Emit(OpCodes.Ldnull);
  1070. }
  1071. }
  1072. private static void EmitDeserializeConverter(ILGenerator il, SerializedMemberInfo member, Label nextLabel, GetLocal GetLocal)
  1073. {
  1074. var stlocal = GetLocal(typeof(Value));
  1075. var valLocal = GetLocal(member.Type);
  1076. il.Emit(OpCodes.Stloc, stlocal);
  1077. il.BeginExceptionBlock();
  1078. il.Emit(OpCodes.Ldsfld, member.ConverterField);
  1079. il.Emit(OpCodes.Ldloc, stlocal);
  1080. il.Emit(OpCodes.Ldarg_0);
  1081. if (member.IsGenericConverter)
  1082. {
  1083. var fromValue = member.ConverterBase.GetMethod(nameof(ValueConverter<int>.FromValue),
  1084. new[] { typeof(Value), typeof(object) });
  1085. il.Emit(OpCodes.Callvirt, fromValue);
  1086. }
  1087. else
  1088. {
  1089. var fromValue = typeof(IValueConverter).GetMethod(nameof(IValueConverter.FromValue),
  1090. new[] { typeof(Value), typeof(object) });
  1091. il.Emit(OpCodes.Callvirt, fromValue);
  1092. if (member.Type.IsValueType)
  1093. il.Emit(OpCodes.Unbox);
  1094. }
  1095. il.Emit(OpCodes.Stloc, valLocal);
  1096. il.BeginCatchBlock(typeof(Exception));
  1097. EmitWarnException(il, "Error occurred while deserializing");
  1098. il.Emit(OpCodes.Leave, nextLabel);
  1099. il.EndExceptionBlock();
  1100. il.Emit(OpCodes.Ldloc, valLocal);
  1101. }
  1102. // emit takes the value being deserialized, logs on error, leaves nothing on stack
  1103. private static void EmitDeserializeMember(ILGenerator il, SerializedMemberInfo member, Label nextLabel, Action<ILGenerator> getValue, GetLocal GetLocal)
  1104. {
  1105. var Object_GetType = typeof(object).GetMethod(nameof(Object.GetType));
  1106. var implLabel = il.DefineLabel();
  1107. var passedTypeCheck = il.DefineLabel();
  1108. var expectType = GetExpectedValueTypeForType(member.IsNullable ? member.NullableWrappedType : member.Type);
  1109. il.Emit(OpCodes.Dup);
  1110. il.Emit(OpCodes.Brtrue_S, implLabel); // null check
  1111. if (!member.AllowNull)
  1112. {
  1113. il.Emit(OpCodes.Pop);
  1114. EmitLogError(il, $"Member {member.Name} ({member.Type}) not nullable", tailcall: false,
  1115. expected: il => EmitTypeof(il, expectType));
  1116. il.Emit(OpCodes.Br, nextLabel);
  1117. }
  1118. else if (member.IsNullable)
  1119. {
  1120. il.Emit(OpCodes.Pop);
  1121. var valTLocal = GetLocal(member.Type, 0);
  1122. il.Emit(OpCodes.Ldloca, valTLocal);
  1123. il.Emit(OpCodes.Initobj, member.Type);
  1124. EmitStore(il, member, il => il.Emit(OpCodes.Ldloc, valTLocal));
  1125. il.Emit(OpCodes.Br, nextLabel);
  1126. }
  1127. else
  1128. {
  1129. il.Emit(OpCodes.Pop);
  1130. EmitStore(il, member, il => il.Emit(OpCodes.Ldnull));
  1131. il.Emit(OpCodes.Br, nextLabel);
  1132. }
  1133. if (!member.HasConverter)
  1134. {
  1135. il.MarkLabel(implLabel);
  1136. il.Emit(OpCodes.Isinst, expectType); //replaces on stack
  1137. il.Emit(OpCodes.Dup); // duplicate cloned value
  1138. il.Emit(OpCodes.Brtrue, passedTypeCheck); // null check
  1139. }
  1140. var errorHandle = il.DefineLabel();
  1141. // special cases to handle coersion between Float and Int
  1142. if (member.HasConverter)
  1143. il.MarkLabel(implLabel);
  1144. else if (expectType == typeof(FloatingPoint))
  1145. {
  1146. var specialTypeCheck = il.DefineLabel();
  1147. il.Emit(OpCodes.Pop);
  1148. getValue(il);
  1149. il.Emit(OpCodes.Isinst, typeof(Integer)); //replaces on stack
  1150. il.Emit(OpCodes.Dup); // duplicate cloned value
  1151. il.Emit(OpCodes.Brfalse, errorHandle); // null check
  1152. var Integer_CoerceToFloat = typeof(Integer).GetMethod(nameof(Integer.AsFloat));
  1153. il.Emit(OpCodes.Call, Integer_CoerceToFloat);
  1154. il.Emit(OpCodes.Br, passedTypeCheck);
  1155. }
  1156. else if (expectType == typeof(Integer))
  1157. {
  1158. var specialTypeCheck = il.DefineLabel();
  1159. il.Emit(OpCodes.Pop);
  1160. getValue(il);
  1161. il.Emit(OpCodes.Isinst, typeof(FloatingPoint)); //replaces on stack
  1162. il.Emit(OpCodes.Dup); // duplicate cloned value
  1163. il.Emit(OpCodes.Brfalse, errorHandle); // null check
  1164. var Float_CoerceToInt = typeof(FloatingPoint).GetMethod(nameof(FloatingPoint.AsInteger));
  1165. il.Emit(OpCodes.Call, Float_CoerceToInt);
  1166. il.Emit(OpCodes.Br, passedTypeCheck);
  1167. }
  1168. if (!member.HasConverter)
  1169. {
  1170. il.MarkLabel(errorHandle);
  1171. il.Emit(OpCodes.Pop);
  1172. EmitLogError(il, $"Unexpected type deserializing {member.Name}", tailcall: false,
  1173. expected: il => EmitTypeof(il, expectType), found: il =>
  1174. {
  1175. getValue(il);
  1176. il.Emit(OpCodes.Callvirt, Object_GetType);
  1177. });
  1178. il.Emit(OpCodes.Br, nextLabel);
  1179. }
  1180. il.MarkLabel(passedTypeCheck);
  1181. var local = GetLocal(member.Type, 0);
  1182. if (member.HasConverter) EmitDeserializeConverter(il, member, nextLabel, GetLocal);
  1183. else if (member.IsNullable) EmitDeserializeNullable(il, member, expectType, GetLocal);
  1184. else EmitDeserializeValue(il, member.Type, expectType, GetLocal);
  1185. il.Emit(OpCodes.Stloc, local);
  1186. EmitStore(il, member, il => il.Emit(OpCodes.Ldloc, local));
  1187. }
  1188. #endregion
  1189. }
  1190. }