ClientHost.cs 2.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110
  1. using System;
  2. using System.Threading.Tasks;
  3. using Log;
  4. namespace ENet
  5. {
  6. public sealed class ClientHost : Host
  7. {
  8. public ClientHost(uint peerLimit = NativeMethods.ENET_PROTOCOL_MAXIMUM_PEER_ID,
  9. uint channelLimit = 0, uint incomingBandwidth = 0,
  10. uint outgoingBandwidth = 0, bool enableCrc = true)
  11. {
  12. if (peerLimit > NativeMethods.ENET_PROTOCOL_MAXIMUM_PEER_ID)
  13. {
  14. throw new ArgumentOutOfRangeException("peerLimit");
  15. }
  16. CheckChannelLimit(channelLimit);
  17. this.host = NativeMethods.enet_host_create(IntPtr.Zero, peerLimit, channelLimit, incomingBandwidth, outgoingBandwidth);
  18. if (this.host == IntPtr.Zero)
  19. {
  20. throw new ENetException(0, "Host creation call failed.");
  21. }
  22. if (enableCrc)
  23. {
  24. this.EnableCrc();
  25. }
  26. }
  27. public Task<Peer> ConnectAsync(
  28. Address address, uint channelLimit = NativeMethods.ENET_PROTOCOL_MAXIMUM_CHANNEL_COUNT, uint data = 0)
  29. {
  30. CheckChannelLimit(channelLimit);
  31. var tcs = new TaskCompletionSource<Peer>();
  32. ENetAddress nativeAddress = address.Struct;
  33. IntPtr peerPtr = NativeMethods.enet_host_connect(this.host, ref nativeAddress, channelLimit, data);
  34. if (peerPtr == IntPtr.Zero)
  35. {
  36. throw new ENetException(0, "Host connect call failed.");
  37. }
  38. var peer = new Peer(peerPtr);
  39. this.PeersManager.Add(peerPtr, peer);
  40. peer.PeerEvent.Connected += e => tcs.TrySetResult(peer);
  41. return tcs.Task;
  42. }
  43. public void RunOnce(int timeout = 0)
  44. {
  45. this.OnExecuteEvents();
  46. if (this.Service(timeout) < 0)
  47. {
  48. return;
  49. }
  50. Event ev;
  51. while (this.CheckEvents(out ev) > 0)
  52. {
  53. switch (ev.Type)
  54. {
  55. case EventType.Connect:
  56. {
  57. var peer = this.PeersManager[ev.PeerPtr];
  58. peer.PeerEvent.OnConnected(ev);
  59. peer.PeerEvent.Connected = null;
  60. break;
  61. }
  62. case EventType.Receive:
  63. {
  64. var peer = this.PeersManager[ev.PeerPtr];
  65. peer.PeerEvent.OnReceived(ev);
  66. peer.PeerEvent.Received = null;
  67. break;
  68. }
  69. case EventType.Disconnect:
  70. {
  71. ev.EventState = EventState.DISCONNECTED;
  72. var peer = this.PeersManager[ev.PeerPtr];
  73. PeerEvent peerEvent = peer.PeerEvent;
  74. this.PeersManager.Remove(ev.PeerPtr);
  75. peer.Dispose();
  76. if (peerEvent.Received != null)
  77. {
  78. peerEvent.OnReceived(ev);
  79. }
  80. else
  81. {
  82. peerEvent.OnDisconnect(ev);
  83. }
  84. break;
  85. }
  86. }
  87. }
  88. }
  89. public void Start(int timeout = 0)
  90. {
  91. while (isRunning)
  92. {
  93. RunOnce(timeout);
  94. }
  95. }
  96. }
  97. }