ClientHost.cs 2.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121
  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 =>
  41. {
  42. if (e.EventState == EventState.DISCONNECTED)
  43. {
  44. tcs.TrySetException(new ENetException(3, "Peer Disconnected In Connect!"));
  45. }
  46. tcs.TrySetResult(peer);
  47. };
  48. return tcs.Task;
  49. }
  50. public void RunOnce(int timeout = 0)
  51. {
  52. this.OnExecuteEvents();
  53. if (this.Service(timeout) < 0)
  54. {
  55. return;
  56. }
  57. Event ev;
  58. while (this.CheckEvents(out ev) > 0)
  59. {
  60. switch (ev.Type)
  61. {
  62. case EventType.Connect:
  63. {
  64. var peer = this.PeersManager[ev.PeerPtr];
  65. peer.PeerEvent.OnConnected(ev);
  66. peer.PeerEvent.Connected = null;
  67. break;
  68. }
  69. case EventType.Receive:
  70. {
  71. var peer = this.PeersManager[ev.PeerPtr];
  72. peer.PeerEvent.OnReceived(ev);
  73. peer.PeerEvent.Received = null;
  74. break;
  75. }
  76. case EventType.Disconnect:
  77. {
  78. ev.EventState = EventState.DISCONNECTED;
  79. var peer = this.PeersManager[ev.PeerPtr];
  80. PeerEvent peerEvent = peer.PeerEvent;
  81. this.PeersManager.Remove(ev.PeerPtr);
  82. peer.Dispose();
  83. if (peerEvent.Connected != null)
  84. {
  85. peerEvent.OnConnected(ev);
  86. }
  87. else if (peerEvent.Received != null)
  88. {
  89. peerEvent.OnReceived(ev);
  90. }
  91. else
  92. {
  93. peerEvent.OnDisconnect(ev);
  94. }
  95. break;
  96. }
  97. }
  98. }
  99. }
  100. public void Start(int timeout = 0)
  101. {
  102. while (isRunning)
  103. {
  104. RunOnce(timeout);
  105. }
  106. }
  107. }
  108. }