CallbackSet.cs 4.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110
  1. #if !NO_RUNTIME
  2. using System;
  3. using System.Reflection;
  4. namespace ProtoBuf.Meta
  5. {
  6. /// <summary>
  7. /// Represents the set of serialization callbacks to be used when serializing/deserializing a type.
  8. /// </summary>
  9. public class CallbackSet
  10. {
  11. private readonly MetaType metaType;
  12. internal CallbackSet(MetaType metaType)
  13. {
  14. this.metaType = metaType ?? throw new ArgumentNullException(nameof(metaType));
  15. }
  16. internal MethodInfo this[TypeModel.CallbackType callbackType]
  17. {
  18. get
  19. {
  20. switch (callbackType)
  21. {
  22. case TypeModel.CallbackType.BeforeSerialize: return beforeSerialize;
  23. case TypeModel.CallbackType.AfterSerialize: return afterSerialize;
  24. case TypeModel.CallbackType.BeforeDeserialize: return beforeDeserialize;
  25. case TypeModel.CallbackType.AfterDeserialize: return afterDeserialize;
  26. default: throw new ArgumentException("Callback type not supported: " + callbackType.ToString(), "callbackType");
  27. }
  28. }
  29. }
  30. internal static bool CheckCallbackParameters(TypeModel model, MethodInfo method)
  31. {
  32. ParameterInfo[] args = method.GetParameters();
  33. for (int i = 0; i < args.Length; i++)
  34. {
  35. Type paramType = args[i].ParameterType;
  36. if (paramType == model.MapType(typeof(SerializationContext))) { }
  37. else if (paramType == model.MapType(typeof(System.Type))) { }
  38. #if PLAT_BINARYFORMATTER
  39. else if (paramType == model.MapType(typeof(System.Runtime.Serialization.StreamingContext))) { }
  40. #endif
  41. else return false;
  42. }
  43. return true;
  44. }
  45. private MethodInfo SanityCheckCallback(TypeModel model, MethodInfo callback)
  46. {
  47. metaType.ThrowIfFrozen();
  48. if (callback == null) return callback; // fine
  49. if (callback.IsStatic) throw new ArgumentException("Callbacks cannot be static", nameof(callback));
  50. if (callback.ReturnType != model.MapType(typeof(void))
  51. || !CheckCallbackParameters(model, callback))
  52. {
  53. throw CreateInvalidCallbackSignature(callback);
  54. }
  55. return callback;
  56. }
  57. internal static Exception CreateInvalidCallbackSignature(MethodInfo method)
  58. {
  59. return new NotSupportedException("Invalid callback signature in " + method.DeclaringType.FullName + "." + method.Name);
  60. }
  61. private MethodInfo beforeSerialize, afterSerialize, beforeDeserialize, afterDeserialize;
  62. /// <summary>Called before serializing an instance</summary>
  63. public MethodInfo BeforeSerialize
  64. {
  65. get { return beforeSerialize; }
  66. set { beforeSerialize = SanityCheckCallback(metaType.Model, value); }
  67. }
  68. /// <summary>Called before deserializing an instance</summary>
  69. public MethodInfo BeforeDeserialize
  70. {
  71. get { return beforeDeserialize; }
  72. set { beforeDeserialize = SanityCheckCallback(metaType.Model, value); }
  73. }
  74. /// <summary>Called after serializing an instance</summary>
  75. public MethodInfo AfterSerialize
  76. {
  77. get { return afterSerialize; }
  78. set { afterSerialize = SanityCheckCallback(metaType.Model, value); }
  79. }
  80. /// <summary>Called after deserializing an instance</summary>
  81. public MethodInfo AfterDeserialize
  82. {
  83. get { return afterDeserialize; }
  84. set { afterDeserialize = SanityCheckCallback(metaType.Model, value); }
  85. }
  86. /// <summary>
  87. /// True if any callback is set, else False
  88. /// </summary>
  89. public bool NonTrivial
  90. {
  91. get
  92. {
  93. return beforeSerialize != null || beforeDeserialize != null
  94. || afterSerialize != null || afterDeserialize != null;
  95. }
  96. }
  97. }
  98. }
  99. #endif