Peer.cs 2.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130
  1. using System;
  2. using System.Runtime.InteropServices;
  3. using System.Threading.Tasks;
  4. namespace ENet
  5. {
  6. public class Peer: IDisposable
  7. {
  8. private readonly Host host;
  9. private readonly PeerEvent peerEvent = new PeerEvent();
  10. private IntPtr peer;
  11. public Peer(Host host, IntPtr peer)
  12. {
  13. this.host = host;
  14. this.peer = peer;
  15. host.PeersManager.Add(peer, this);
  16. }
  17. ~Peer()
  18. {
  19. this.Dispose(false);
  20. }
  21. public void Dispose()
  22. {
  23. this.Dispose(true);
  24. GC.SuppressFinalize(this);
  25. }
  26. protected virtual void Dispose(bool disposing)
  27. {
  28. if (this.peer == IntPtr.Zero)
  29. {
  30. return;
  31. }
  32. this.host.PeersManager.Remove(this.peer);
  33. NativeMethods.enet_peer_reset(this.peer);
  34. this.peer = IntPtr.Zero;
  35. }
  36. private ENetPeer Struct
  37. {
  38. get
  39. {
  40. if (this.peer == IntPtr.Zero)
  41. {
  42. return new ENetPeer();
  43. }
  44. return (ENetPeer) Marshal.PtrToStructure(this.peer, typeof (ENetPeer));
  45. }
  46. set
  47. {
  48. Marshal.StructureToPtr(value, this.peer, false);
  49. }
  50. }
  51. public PeerEvent PeerEvent
  52. {
  53. get
  54. {
  55. return this.peerEvent;
  56. }
  57. }
  58. public PeerState State
  59. {
  60. get
  61. {
  62. if (this.peer == IntPtr.Zero)
  63. {
  64. return PeerState.Uninitialized;
  65. }
  66. return this.Struct.state;
  67. }
  68. }
  69. public void Ping()
  70. {
  71. NativeMethods.enet_peer_ping(this.peer);
  72. }
  73. public void ConfigureThrottle(uint interval, uint acceleration, uint deceleration)
  74. {
  75. NativeMethods.enet_peer_throttle_configure(this.peer, interval, acceleration, deceleration);
  76. }
  77. public void Send(byte channelID, string data)
  78. {
  79. using (var packet = new Packet(data))
  80. {
  81. this.Send(channelID, packet);
  82. }
  83. }
  84. public void Send(byte channelID, Packet packet)
  85. {
  86. NativeMethods.enet_peer_send(this.peer, channelID, packet.NativePtr);
  87. }
  88. public Task<Packet> ReceiveAsync()
  89. {
  90. var tcs = new TaskCompletionSource<Packet>();
  91. this.PeerEvent.Received += e => tcs.TrySetResult(e.Packet);
  92. return tcs.Task;
  93. }
  94. public Task<bool> DisconnectAsync(uint data = 0)
  95. {
  96. var tcs = new TaskCompletionSource<bool>();
  97. this.PeerEvent.Disconnect += e => tcs.TrySetResult(true);
  98. NativeMethods.enet_peer_disconnect(this.peer, data);
  99. return tcs.Task;
  100. }
  101. public Task<bool> DisconnectLaterAsync(uint data = 0)
  102. {
  103. var tcs = new TaskCompletionSource<bool>();
  104. this.PeerEvent.Disconnect += e => tcs.TrySetResult(true);
  105. NativeMethods.enet_peer_disconnect_later(this.peer, data);
  106. return tcs.Task;
  107. }
  108. public void DisconnectNow(uint data)
  109. {
  110. NativeMethods.enet_peer_disconnect_now(this.peer, data);
  111. }
  112. }
  113. }