ENet.cs 28 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041
  1. /*
  2. * Managed C# wrapper for an extended version of ENet
  3. * Copyright (c) 2013 James Bellinger
  4. * Copyright (c) 2016 Nate Shoffner
  5. * Copyright (c) 2018 Stanislav Denisov
  6. *
  7. * Permission is hereby granted, free of charge, to any person obtaining a copy
  8. * of this software and associated documentation files (the "Software"), to deal
  9. * in the Software without restriction, including without limitation the rights
  10. * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  11. * copies of the Software, and to permit persons to whom the Software is
  12. * furnished to do so, subject to the following conditions:
  13. *
  14. * The above copyright notice and this permission notice shall be included in all
  15. * copies or substantial portions of the Software.
  16. *
  17. * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  18. * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  19. * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  20. * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  21. * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  22. * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  23. * SOFTWARE.
  24. */
  25. using System;
  26. using System.Runtime.InteropServices;
  27. using System.Security;
  28. using System.Text;
  29. namespace ENet {
  30. [Flags]
  31. public enum PacketFlags {
  32. None = 0,
  33. Reliable = 1 << 0,
  34. Unsequenced = 1 << 1,
  35. NoAllocate = 1 << 2,
  36. UnreliableFragment = 1 << 3
  37. }
  38. public enum EventType {
  39. None = 0,
  40. Connect = 1,
  41. Disconnect = 2,
  42. Receive = 3,
  43. Timeout = 4
  44. }
  45. public enum PeerState {
  46. Uninitialized = -1,
  47. Disconnected = 0,
  48. Connecting = 1,
  49. AcknowledgingConnect = 2,
  50. ConnectionPending = 3,
  51. ConnectionSucceeded = 4,
  52. Connected = 5,
  53. DisconnectLater = 6,
  54. Disconnecting = 7,
  55. AcknowledgingDisconnect = 8,
  56. Zombie = 9
  57. }
  58. [StructLayout(LayoutKind.Sequential)]
  59. public struct ENetAddress {
  60. [MarshalAs(UnmanagedType.ByValArray, SizeConst = 16)]
  61. public byte[] host;
  62. public ushort port;
  63. public ushort scope;
  64. }
  65. [StructLayout(LayoutKind.Sequential)]
  66. public struct ENetEvent {
  67. public EventType type;
  68. public IntPtr peer;
  69. public byte channelID;
  70. public uint data;
  71. public IntPtr packet;
  72. }
  73. [StructLayout(LayoutKind.Sequential)]
  74. public struct ENetCallbacks {
  75. public AllocCallback malloc;
  76. public FreeCallback free;
  77. public NoMemoryCallback noMemory;
  78. }
  79. public delegate IntPtr AllocCallback(IntPtr size);
  80. public delegate void FreeCallback(IntPtr memory);
  81. public delegate void NoMemoryCallback();
  82. public delegate void PacketFreeCallback(Packet packet);
  83. internal static class ArrayPool {
  84. [ThreadStatic]
  85. private static byte[] byteBuffer;
  86. [ThreadStatic]
  87. private static IntPtr[] pointerBuffer;
  88. public static byte[] GetByteBuffer() {
  89. if (byteBuffer == null)
  90. byteBuffer = new byte[64];
  91. return byteBuffer;
  92. }
  93. public static IntPtr[] GetPointerBuffer() {
  94. if (pointerBuffer == null)
  95. pointerBuffer = new IntPtr[Library.maxPeers];
  96. return pointerBuffer;
  97. }
  98. }
  99. public struct Address {
  100. private ENetAddress nativeAddress;
  101. internal ENetAddress NativeData {
  102. get {
  103. return nativeAddress;
  104. }
  105. set {
  106. nativeAddress = value;
  107. }
  108. }
  109. public Address(ENetAddress address) {
  110. nativeAddress = address;
  111. }
  112. public ushort Port {
  113. get {
  114. return nativeAddress.port;
  115. }
  116. set {
  117. nativeAddress.port = value;
  118. }
  119. }
  120. public string GetHost() {
  121. StringBuilder hostName = new StringBuilder(1024);
  122. if (Native.enet_address_get_host(nativeAddress, hostName, (IntPtr)hostName.Capacity) != 0)
  123. return String.Empty;
  124. return hostName.ToString();
  125. }
  126. public bool SetHost(string hostName) {
  127. if (hostName == null)
  128. throw new ArgumentNullException("hostName");
  129. return Native.enet_address_set_host(ref nativeAddress, hostName) == 0;
  130. }
  131. }
  132. public struct Event {
  133. private ENetEvent nativeEvent;
  134. internal ENetEvent NativeData {
  135. get {
  136. return nativeEvent;
  137. }
  138. set {
  139. nativeEvent = value;
  140. }
  141. }
  142. public Event(ENetEvent @event) {
  143. nativeEvent = @event;
  144. }
  145. public EventType Type {
  146. get {
  147. return nativeEvent.type;
  148. }
  149. }
  150. public Peer Peer {
  151. get {
  152. return new Peer(nativeEvent.peer);
  153. }
  154. }
  155. public byte ChannelID {
  156. get {
  157. return nativeEvent.channelID;
  158. }
  159. }
  160. public uint Data {
  161. get {
  162. return nativeEvent.data;
  163. }
  164. }
  165. public Packet Packet {
  166. get {
  167. return new Packet(nativeEvent.packet);
  168. }
  169. }
  170. }
  171. public class Callbacks {
  172. private ENetCallbacks nativeCallbacks;
  173. internal ENetCallbacks NativeData {
  174. get {
  175. return nativeCallbacks;
  176. }
  177. set {
  178. nativeCallbacks = value;
  179. }
  180. }
  181. public Callbacks(AllocCallback allocCallback, FreeCallback freeCallback, NoMemoryCallback noMemoryCallback) {
  182. nativeCallbacks.malloc = allocCallback;
  183. nativeCallbacks.free = freeCallback;
  184. nativeCallbacks.noMemory = noMemoryCallback;
  185. }
  186. }
  187. public struct Packet : IDisposable {
  188. private IntPtr nativePacket;
  189. internal IntPtr NativeData {
  190. get {
  191. return nativePacket;
  192. }
  193. set {
  194. nativePacket = value;
  195. }
  196. }
  197. public Packet(IntPtr packet) {
  198. nativePacket = packet;
  199. }
  200. public void Dispose() {
  201. if (nativePacket != IntPtr.Zero) {
  202. Native.enet_packet_dispose(nativePacket);
  203. nativePacket = IntPtr.Zero;
  204. }
  205. }
  206. public bool IsSet {
  207. get {
  208. return nativePacket != IntPtr.Zero;
  209. }
  210. }
  211. public IntPtr Data {
  212. get {
  213. CheckCreated();
  214. return Native.enet_packet_get_data(nativePacket);
  215. }
  216. }
  217. public int Length {
  218. get {
  219. CheckCreated();
  220. return Native.enet_packet_get_length(nativePacket);
  221. }
  222. }
  223. public bool HasReferences {
  224. get {
  225. CheckCreated();
  226. return Native.enet_packet_check_references(nativePacket) != 0;
  227. }
  228. }
  229. internal void CheckCreated() {
  230. if (nativePacket == IntPtr.Zero)
  231. throw new InvalidOperationException("Packet not created");
  232. }
  233. public void SetFreeCallback(IntPtr callback) {
  234. CheckCreated();
  235. Native.enet_packet_set_free_callback(nativePacket, callback);
  236. }
  237. public void SetFreeCallback(PacketFreeCallback callback) {
  238. CheckCreated();
  239. Native.enet_packet_set_free_callback(nativePacket, Marshal.GetFunctionPointerForDelegate(callback));
  240. }
  241. public void Create(byte[] data) {
  242. if (data == null)
  243. throw new ArgumentNullException("data");
  244. Create(data, data.Length);
  245. }
  246. public void Create(byte[] data, int length) {
  247. Create(data, length, PacketFlags.None);
  248. }
  249. public void Create(byte[] data, PacketFlags flags) {
  250. Create(data, data.Length, flags);
  251. }
  252. public void Create(byte[] data, int length, PacketFlags flags) {
  253. if (data == null)
  254. throw new ArgumentNullException("data");
  255. if (length < 0 || length > data.Length)
  256. throw new ArgumentOutOfRangeException();
  257. nativePacket = Native.enet_packet_create(data, (IntPtr)length, flags);
  258. }
  259. public void Create(IntPtr data, int length, PacketFlags flags) {
  260. if (data == IntPtr.Zero)
  261. throw new ArgumentNullException("data");
  262. if (length < 0)
  263. throw new ArgumentOutOfRangeException();
  264. nativePacket = Native.enet_packet_create(data, (IntPtr)length, flags);
  265. }
  266. public void Create(byte[] data, int offset, int length, PacketFlags flags) {
  267. if (data == null)
  268. throw new ArgumentNullException("data");
  269. if (offset < 0 || length < 0 || length > data.Length)
  270. throw new ArgumentOutOfRangeException();
  271. nativePacket = Native.enet_packet_create_offset(data, (IntPtr)length, (IntPtr)offset, flags);
  272. }
  273. public void Create(IntPtr data, int offset, int length, PacketFlags flags) {
  274. if (data == IntPtr.Zero)
  275. throw new ArgumentNullException("data");
  276. if (offset < 0 || length < 0)
  277. throw new ArgumentOutOfRangeException();
  278. nativePacket = Native.enet_packet_create_offset(data, (IntPtr)length, (IntPtr)offset, flags);
  279. }
  280. public void CopyTo(byte[] destination) {
  281. if (destination == null)
  282. throw new ArgumentNullException("destination");
  283. Marshal.Copy(Data, destination, 0, Length);
  284. }
  285. }
  286. public class Host : IDisposable {
  287. private IntPtr nativeHost;
  288. internal IntPtr NativeData {
  289. get {
  290. return nativeHost;
  291. }
  292. set {
  293. nativeHost = value;
  294. }
  295. }
  296. public void Dispose() {
  297. Dispose(true);
  298. GC.SuppressFinalize(this);
  299. }
  300. protected virtual void Dispose(bool disposing) {
  301. if (nativeHost != IntPtr.Zero) {
  302. Native.enet_host_destroy(nativeHost);
  303. nativeHost = IntPtr.Zero;
  304. }
  305. }
  306. ~Host() {
  307. Dispose(false);
  308. }
  309. public bool IsSet {
  310. get {
  311. return nativeHost != IntPtr.Zero;
  312. }
  313. }
  314. public uint PeersCount {
  315. get {
  316. CheckCreated();
  317. return Native.enet_host_get_peers_count(nativeHost);
  318. }
  319. }
  320. public uint PacketsSent {
  321. get {
  322. CheckCreated();
  323. return Native.enet_host_get_packets_sent(nativeHost);
  324. }
  325. }
  326. public uint PacketsReceived {
  327. get {
  328. CheckCreated();
  329. return Native.enet_host_get_packets_received(nativeHost);
  330. }
  331. }
  332. public uint BytesSent {
  333. get {
  334. CheckCreated();
  335. return Native.enet_host_get_bytes_sent(nativeHost);
  336. }
  337. }
  338. public uint BytesReceived {
  339. get {
  340. CheckCreated();
  341. return Native.enet_host_get_bytes_received(nativeHost);
  342. }
  343. }
  344. internal void CheckCreated() {
  345. if (nativeHost == IntPtr.Zero)
  346. throw new InvalidOperationException("Host not created");
  347. }
  348. private void CheckChannelLimit(int channelLimit) {
  349. if (channelLimit < 0 || channelLimit > Library.maxChannelCount)
  350. throw new ArgumentOutOfRangeException("channelLimit");
  351. }
  352. public void Create() {
  353. Create(null, 1, 0);
  354. }
  355. public void Create(Address? address, int peerLimit) {
  356. Create(address, peerLimit, 0);
  357. }
  358. public void Create(Address? address, int peerLimit, int channelLimit) {
  359. Create(address, peerLimit, channelLimit, 0, 0);
  360. }
  361. public void Create(int peerLimit, int channelLimit) {
  362. Create(null, peerLimit, channelLimit, 0, 0);
  363. }
  364. public void Create(int peerLimit, int channelLimit, uint incomingBandwidth, uint outgoingBandwidth) {
  365. Create(null, peerLimit, channelLimit, incomingBandwidth, outgoingBandwidth);
  366. }
  367. public void Create(Address? address, int peerLimit, int channelLimit, uint incomingBandwidth, uint outgoingBandwidth) {
  368. if (nativeHost != IntPtr.Zero)
  369. throw new InvalidOperationException("Host already created");
  370. if (peerLimit < 0 || peerLimit > Library.maxPeers)
  371. throw new ArgumentOutOfRangeException("peerLimit");
  372. CheckChannelLimit(channelLimit);
  373. if (address != null) {
  374. var nativeAddress = address.Value.NativeData;
  375. nativeHost = Native.enet_host_create(ref nativeAddress, (IntPtr)peerLimit, (IntPtr)channelLimit, incomingBandwidth, outgoingBandwidth);
  376. } else {
  377. nativeHost = Native.enet_host_create(IntPtr.Zero, (IntPtr)peerLimit, (IntPtr)channelLimit, incomingBandwidth, outgoingBandwidth);
  378. }
  379. if (nativeHost == IntPtr.Zero)
  380. throw new InvalidOperationException("Host creation call failed");
  381. }
  382. public void EnableCompression() {
  383. CheckCreated();
  384. Native.enet_host_enable_compression(nativeHost);
  385. }
  386. public void PreventConnections(bool state) {
  387. CheckCreated();
  388. Native.enet_host_prevent_connections(nativeHost, (byte)(state ? 1 : 0));
  389. }
  390. public void Broadcast(byte channelID, ref Packet packet) {
  391. CheckCreated();
  392. packet.CheckCreated();
  393. Native.enet_host_broadcast(nativeHost, channelID, packet.NativeData);
  394. packet.NativeData = IntPtr.Zero;
  395. }
  396. public void Broadcast(byte channelID, ref Packet packet, Peer[] peers) {
  397. CheckCreated();
  398. packet.CheckCreated();
  399. if (peers.Length > 0) {
  400. IntPtr[] nativePeers = ArrayPool.GetPointerBuffer();
  401. int nativeCount = 0;
  402. for (int i = 0; i < peers.Length; i++) {
  403. if (peers[i].NativeData != IntPtr.Zero) {
  404. nativePeers[nativeCount] = peers[i].NativeData;
  405. nativeCount++;
  406. }
  407. }
  408. Native.enet_host_broadcast_selective(nativeHost, channelID, packet.NativeData, nativePeers, (IntPtr)nativeCount);
  409. }
  410. packet.NativeData = IntPtr.Zero;
  411. }
  412. public int CheckEvents(out Event @event) {
  413. CheckCreated();
  414. ENetEvent nativeEvent;
  415. var result = Native.enet_host_check_events(nativeHost, out nativeEvent);
  416. if (result <= 0) {
  417. @event = new Event();
  418. return result;
  419. }
  420. @event = new Event(nativeEvent);
  421. return result;
  422. }
  423. public Peer Connect(Address address) {
  424. return Connect(address, 0, 0);
  425. }
  426. public Peer Connect(Address address, int channelLimit) {
  427. return Connect(address, channelLimit, 0);
  428. }
  429. public Peer Connect(Address address, int channelLimit, uint data) {
  430. CheckCreated();
  431. CheckChannelLimit(channelLimit);
  432. var nativeAddress = address.NativeData;
  433. var peer = new Peer(Native.enet_host_connect(nativeHost, ref nativeAddress, (IntPtr)channelLimit, data));
  434. if (peer.NativeData == IntPtr.Zero)
  435. throw new InvalidOperationException("Host connect call failed");
  436. return peer;
  437. }
  438. public int Service(int timeout, out Event @event) {
  439. if (timeout < 0)
  440. throw new ArgumentOutOfRangeException("timeout");
  441. CheckCreated();
  442. ENetEvent nativeEvent;
  443. var result = Native.enet_host_service(nativeHost, out nativeEvent, (uint)timeout);
  444. if (result <= 0) {
  445. @event = new Event();
  446. return result;
  447. }
  448. @event = new Event(nativeEvent);
  449. return result;
  450. }
  451. public void SetBandwidthLimit(uint incomingBandwidth, uint outgoingBandwidth) {
  452. CheckCreated();
  453. Native.enet_host_bandwidth_limit(nativeHost, incomingBandwidth, outgoingBandwidth);
  454. }
  455. public void SetChannelLimit(int channelLimit) {
  456. CheckCreated();
  457. CheckChannelLimit(channelLimit);
  458. Native.enet_host_channel_limit(nativeHost, (IntPtr)channelLimit);
  459. }
  460. public void Flush() {
  461. CheckCreated();
  462. Native.enet_host_flush(nativeHost);
  463. }
  464. }
  465. public struct Peer {
  466. private IntPtr nativePeer;
  467. private uint nativeID;
  468. private ushort nativePeerID;
  469. internal IntPtr NativeData {
  470. get {
  471. return nativePeer;
  472. }
  473. set {
  474. nativePeer = value;
  475. }
  476. }
  477. public Peer(IntPtr peer) {
  478. nativePeer = peer;
  479. nativeID = nativePeer != IntPtr.Zero ? Native.enet_peer_get_id(nativePeer) : 0;
  480. nativePeerID = nativePeer != IntPtr.Zero ? Native.enet_peer_get_peer_id(nativePeer) : 0;
  481. }
  482. public bool IsSet {
  483. get {
  484. return nativePeer != IntPtr.Zero;
  485. }
  486. }
  487. public uint ID {
  488. get {
  489. return nativeID;
  490. }
  491. }
  492. public ushort PeerID {
  493. get {
  494. return nativePeerID;
  495. }
  496. }
  497. public string IP {
  498. get {
  499. CheckCreated();
  500. byte[] ip = ArrayPool.GetByteBuffer();
  501. if (Native.enet_peer_get_ip(nativePeer, ip, (IntPtr)ip.Length) == 0) {
  502. if (Encoding.ASCII.GetString(ip).Remove(7) != "::ffff:")
  503. return Encoding.ASCII.GetString(ip, 0, ip.StringLength());
  504. else
  505. return Encoding.ASCII.GetString(ip, 0, ip.StringLength()).Substring(7);
  506. } else {
  507. return String.Empty;
  508. }
  509. }
  510. }
  511. public ushort Port {
  512. get {
  513. CheckCreated();
  514. return Native.enet_peer_get_port(nativePeer);
  515. }
  516. }
  517. public uint MTU {
  518. get {
  519. CheckCreated();
  520. return Native.enet_peer_get_mtu(nativePeer);
  521. }
  522. }
  523. public PeerState State {
  524. get {
  525. return nativePeer == IntPtr.Zero ? PeerState.Uninitialized : Native.enet_peer_get_state(nativePeer);
  526. }
  527. }
  528. public uint RoundTripTime {
  529. get {
  530. CheckCreated();
  531. return Native.enet_peer_get_rtt(nativePeer);
  532. }
  533. }
  534. public uint LastSendTime {
  535. get {
  536. CheckCreated();
  537. return Native.enet_peer_get_lastsendtime(nativePeer);
  538. }
  539. }
  540. public uint LastReceiveTime {
  541. get {
  542. CheckCreated();
  543. return Native.enet_peer_get_lastreceivetime(nativePeer);
  544. }
  545. }
  546. public ulong PacketsSent {
  547. get {
  548. CheckCreated();
  549. return Native.enet_peer_get_packets_sent(nativePeer);
  550. }
  551. }
  552. public ulong PacketsLost {
  553. get {
  554. CheckCreated();
  555. return Native.enet_peer_get_packets_lost(nativePeer);
  556. }
  557. }
  558. public ulong BytesSent {
  559. get {
  560. CheckCreated();
  561. return Native.enet_peer_get_bytes_sent(nativePeer);
  562. }
  563. }
  564. public ulong BytesReceived {
  565. get {
  566. CheckCreated();
  567. return Native.enet_peer_get_bytes_received(nativePeer);
  568. }
  569. }
  570. public IntPtr Data {
  571. get {
  572. CheckCreated();
  573. return Native.enet_peer_get_data(nativePeer);
  574. }
  575. set {
  576. CheckCreated();
  577. Native.enet_peer_set_data(nativePeer, value);
  578. }
  579. }
  580. internal void CheckCreated() {
  581. if (nativePeer == IntPtr.Zero)
  582. throw new InvalidOperationException("Peer not created");
  583. }
  584. public void ConfigureThrottle(uint interval, uint acceleration, uint deceleration) {
  585. CheckCreated();
  586. Native.enet_peer_throttle_configure(nativePeer, interval, acceleration, deceleration);
  587. }
  588. public bool Send(byte channelID, ref Packet packet) {
  589. CheckCreated();
  590. packet.CheckCreated();
  591. return Native.enet_peer_send(nativePeer, channelID, packet.NativeData) == 0;
  592. }
  593. public void Ping() {
  594. CheckCreated();
  595. Native.enet_peer_ping(nativePeer);
  596. }
  597. public void PingInterval(uint interval) {
  598. CheckCreated();
  599. Native.enet_peer_ping_interval(nativePeer, interval);
  600. }
  601. public void Timeout(uint timeoutLimit, uint timeoutMinimum, uint timeoutMaximum) {
  602. CheckCreated();
  603. Native.enet_peer_timeout(nativePeer, timeoutLimit, timeoutMinimum, timeoutMaximum);
  604. }
  605. public void Disconnect(uint data) {
  606. CheckCreated();
  607. Native.enet_peer_disconnect(nativePeer, data);
  608. }
  609. public void DisconnectNow(uint data) {
  610. CheckCreated();
  611. Native.enet_peer_disconnect_now(nativePeer, data);
  612. }
  613. public void DisconnectLater(uint data) {
  614. CheckCreated();
  615. Native.enet_peer_disconnect_later(nativePeer, data);
  616. }
  617. public void Reset() {
  618. CheckCreated();
  619. Native.enet_peer_reset(nativePeer);
  620. }
  621. }
  622. public static class Extensions {
  623. public static int StringLength(this byte[] data) {
  624. if (data == null)
  625. throw new ArgumentNullException("data");
  626. int i;
  627. for (i = 0; i < data.Length && data[i] != 0; i++);
  628. return i;
  629. }
  630. }
  631. public static class Library {
  632. public const uint maxChannelCount = 0xFF;
  633. public const uint maxPeers = 0xFFF;
  634. public const uint maxPacketSize = 32 * 1024 * 1024;
  635. public const uint throttleScale = 32;
  636. public const uint throttleAcceleration = 2;
  637. public const uint throttleDeceleration = 2;
  638. public const uint throttleInterval = 5000;
  639. public const uint timeoutLimit = 32;
  640. public const uint timeoutMinimum = 5000;
  641. public const uint timeoutMaximum = 30000;
  642. public const uint version = (2 << 16) | (2 << 8) | (1);
  643. public static bool Initialize() {
  644. return Native.enet_initialize() == 0;
  645. }
  646. public static bool Initialize(Callbacks inits) {
  647. return Native.enet_initialize_with_callbacks(version, inits.NativeData) == 0;
  648. }
  649. public static void Deinitialize() {
  650. Native.enet_deinitialize();
  651. }
  652. public static uint Time {
  653. get {
  654. return Native.enet_time_get();
  655. }
  656. }
  657. }
  658. [SuppressUnmanagedCodeSecurity]
  659. internal static class Native {
  660. #if __IOS__ || UNITY_IOS && !UNITY_EDITOR
  661. private const string nativeLibrary = "__Internal";
  662. #else
  663. private const string nativeLibrary = "enet";
  664. #endif
  665. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  666. internal static extern int enet_initialize();
  667. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  668. internal static extern int enet_initialize_with_callbacks(uint version, ENetCallbacks inits);
  669. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  670. internal static extern void enet_deinitialize();
  671. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  672. internal static extern uint enet_time_get();
  673. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  674. internal static extern int enet_address_get_host(ENetAddress address, StringBuilder hostName, IntPtr nameLength);
  675. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  676. internal static extern int enet_address_set_host(ref ENetAddress address, string hostName);
  677. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  678. internal static extern IntPtr enet_packet_create(byte[] data, IntPtr dataLength, PacketFlags flags);
  679. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  680. internal static extern IntPtr enet_packet_create(IntPtr data, IntPtr dataLength, PacketFlags flags);
  681. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  682. internal static extern IntPtr enet_packet_create_offset(byte[] data, IntPtr dataLength, IntPtr dataOffset, PacketFlags flags);
  683. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  684. internal static extern IntPtr enet_packet_create_offset(IntPtr data, IntPtr dataLength, IntPtr dataOffset, PacketFlags flags);
  685. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  686. internal static extern int enet_packet_check_references(IntPtr packet);
  687. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  688. internal static extern IntPtr enet_packet_get_data(IntPtr packet);
  689. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  690. internal static extern int enet_packet_get_length(IntPtr packet);
  691. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  692. internal static extern void enet_packet_set_free_callback(IntPtr packet, IntPtr callback);
  693. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  694. internal static extern void enet_packet_dispose(IntPtr packet);
  695. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  696. internal static extern IntPtr enet_host_create(ref ENetAddress address, IntPtr peerLimit, IntPtr channelLimit, uint incomingBandwidth, uint outgoingBandwidth);
  697. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  698. internal static extern IntPtr enet_host_create(IntPtr address, IntPtr peerLimit, IntPtr channelLimit, uint incomingBandwidth, uint outgoingBandwidth);
  699. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  700. internal static extern IntPtr enet_host_connect(IntPtr host, ref ENetAddress address, IntPtr channelCount, uint data);
  701. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  702. internal static extern void enet_host_broadcast(IntPtr host, byte channelID, IntPtr packet);
  703. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  704. internal static extern void enet_host_broadcast_selective(IntPtr host, byte channelID, IntPtr packet, IntPtr[] peers, IntPtr peersLength);
  705. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  706. internal static extern int enet_host_service(IntPtr host, out ENetEvent @event, uint timeout);
  707. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  708. internal static extern int enet_host_check_events(IntPtr host, out ENetEvent @event);
  709. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  710. internal static extern void enet_host_channel_limit(IntPtr host, IntPtr channelLimit);
  711. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  712. internal static extern void enet_host_bandwidth_limit(IntPtr host, uint incomingBandwidth, uint outgoingBandwidth);
  713. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  714. internal static extern uint enet_host_get_peers_count(IntPtr host);
  715. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  716. internal static extern uint enet_host_get_packets_sent(IntPtr host);
  717. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  718. internal static extern uint enet_host_get_packets_received(IntPtr host);
  719. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  720. internal static extern uint enet_host_get_bytes_sent(IntPtr host);
  721. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  722. internal static extern uint enet_host_get_bytes_received(IntPtr host);
  723. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  724. internal static extern void enet_host_flush(IntPtr host);
  725. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  726. internal static extern void enet_host_destroy(IntPtr host);
  727. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  728. internal static extern void enet_host_enable_compression(IntPtr host);
  729. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  730. internal static extern void enet_host_prevent_connections(IntPtr host, byte state);
  731. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  732. internal static extern void enet_peer_throttle_configure(IntPtr peer, uint interval, uint acceleration, uint deceleration);
  733. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  734. internal static extern uint enet_peer_get_id(IntPtr peer);
  735. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  736. internal static extern ushort enet_peer_get_peer_id(IntPtr peer);
  737. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  738. internal static extern int enet_peer_get_ip(IntPtr peer, byte[] ip, IntPtr ipLength);
  739. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  740. internal static extern ushort enet_peer_get_port(IntPtr peer);
  741. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  742. internal static extern uint enet_peer_get_mtu(IntPtr peer);
  743. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  744. internal static extern PeerState enet_peer_get_state(IntPtr peer);
  745. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  746. internal static extern uint enet_peer_get_rtt(IntPtr peer);
  747. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  748. internal static extern uint enet_peer_get_lastsendtime(IntPtr peer);
  749. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  750. internal static extern uint enet_peer_get_lastreceivetime(IntPtr peer);
  751. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  752. internal static extern ulong enet_peer_get_packets_sent(IntPtr peer);
  753. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  754. internal static extern ulong enet_peer_get_packets_lost(IntPtr peer);
  755. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  756. internal static extern ulong enet_peer_get_bytes_sent(IntPtr peer);
  757. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  758. internal static extern ulong enet_peer_get_bytes_received(IntPtr peer);
  759. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  760. internal static extern IntPtr enet_peer_get_data(IntPtr peer);
  761. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  762. internal static extern void enet_peer_set_data(IntPtr peer, IntPtr data);
  763. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  764. internal static extern int enet_peer_send(IntPtr peer, byte channelID, IntPtr packet);
  765. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  766. internal static extern void enet_peer_ping(IntPtr peer);
  767. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  768. internal static extern void enet_peer_ping_interval(IntPtr peer, uint pingInterval);
  769. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  770. internal static extern void enet_peer_timeout(IntPtr peer, uint timeoutLimit, uint timeoutMinimum, uint timeoutMaximum);
  771. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  772. internal static extern void enet_peer_disconnect(IntPtr peer, uint data);
  773. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  774. internal static extern void enet_peer_disconnect_now(IntPtr peer, uint data);
  775. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  776. internal static extern void enet_peer_disconnect_later(IntPtr peer, uint data);
  777. [DllImport(nativeLibrary, CallingConvention = CallingConvention.Cdecl)]
  778. internal static extern void enet_peer_reset(IntPtr peer);
  779. }
  780. }