ET9_WebGL_Dev/CSharp/ThirdParty/ENetCS/Host.cs

#region License

/*
ENet for C#
Copyright (c) 2011 James F. Bellinger <jfb@zer7.com>

Permission to use, copy, modify, and/or distribute this software for any
purpose with or without fee is hereby granted, provided that the above
copyright notice and this permission notice appear in all copies.

THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
*/

#endregion

using Log;
using System;
using System.Collections.Generic;
using System.Threading.Tasks;

namespace ENet
{
	public sealed unsafe class Host : IDisposable
	{
		private Native.ENetHost* host;
		private readonly PeerManager peerManager = new PeerManager();
		private readonly object eventsLock = new object();
		private Action events;

		public Host(ushort port, uint peerLimit):
			this(new Address { Port = port }, peerLimit)
		{
		}

		public Host(Address? address, uint peerLimit, uint channelLimit = 0, 
				uint incomingBandwidth = 0, uint outgoingBandwidth = 0, bool enableCrc = true)
		{
			if (peerLimit > Native.ENET_PROTOCOL_MAXIMUM_PEER_ID)
			{
				throw new ArgumentOutOfRangeException("peerLimit");
			}
			CheckChannelLimit(channelLimit);

			if (address != null)
			{
				Native.ENetAddress nativeAddress = address.Value.NativeData;
				this.host = Native.enet_host_create(
					ref nativeAddress, peerLimit, channelLimit, incomingBandwidth,
					outgoingBandwidth);
			}
			else
			{
				this.host = Native.enet_host_create(
					null, peerLimit, channelLimit, incomingBandwidth,
					outgoingBandwidth);
			}

			if (this.host == null)
			{
				throw new ENetException(0, "Host creation call failed.");
			}

			if (enableCrc)
			{
				Native.enet_enable_crc(host);
			}
		}

		~Host()
		{
			this.Dispose(false);
		}

		public void Dispose()
		{
			this.Dispose(true);
			GC.SuppressFinalize(this);
		}

		private void Dispose(bool disposing)
		{
			if (this.host == null)
			{
				return;
			}

			if (disposing)
			{
				Native.enet_host_destroy(this.host);
			}

			this.host = null;
		}

		private static void CheckChannelLimit(uint channelLimit)
		{
			if (channelLimit > Native.ENET_PROTOCOL_MAXIMUM_CHANNEL_COUNT)
			{
				throw new ArgumentOutOfRangeException("channelLimit");
			}
		}

		private int CheckEvents(out Event e)
		{
			Native.ENetEvent nativeEvent;
			int ret = Native.enet_host_check_events(this.host, out nativeEvent);
			e = new Event(this, nativeEvent);
			return ret;
		}

		private int Service(int timeout)
		{
			if (timeout < 0)
			{
				throw new ArgumentOutOfRangeException("timeout");
			}
			return Native.enet_host_service(this.host, null, (uint)timeout);
		}

		private int Service(int timeout, out Event e)
		{
			if (timeout < 0)
			{
				throw new ArgumentOutOfRangeException("timeout");
			}
			Native.ENetEvent nativeEvent;

			int ret = Native.enet_host_service(this.host, out nativeEvent, (uint)timeout);
			e = new Event(this, nativeEvent);
			return ret;
		}

		public PeerManager Peers
		{
			get
			{
				return peerManager;
			}
		}

		public void Broadcast(byte channelID, ref Packet packet)
		{
			Native.enet_host_broadcast(this.host, channelID, packet.NativeData);
			packet.NativeData = null; // Broadcast automatically clears this.
		}

		public void CompressWithRangeEncoder()
		{
			Native.enet_host_compress_with_range_encoder(this.host);
		}

		public void DoNotCompress()
		{
			Native.enet_host_compress(this.host, null);
		}

		public Task<Peer> ConnectAsync(
			Address address, uint channelLimit = Native.ENET_PROTOCOL_MAXIMUM_CHANNEL_COUNT, 
			uint data = 0)
		{
			CheckChannelLimit(channelLimit);

			var tcs = new TaskCompletionSource<Peer>();
			Native.ENetAddress nativeAddress = address.NativeData;
			Native.ENetPeer* p = Native.enet_host_connect(this.host, ref nativeAddress, channelLimit, data);
			if (p == null)
			{
				throw new ENetException(0, "Host connect call failed.");
			}
			var peer = new Peer(this, p);
			peer.Connected += e => tcs.TrySetResult(e.Peer);
			return tcs.Task;
		}

		public void Flush()
		{
			Native.enet_host_flush(this.host);
		}

		public void SetBandwidthLimit(uint incomingBandwidth, uint outgoingBandwidth)
		{
			Native.enet_host_bandwidth_limit(this.host, incomingBandwidth, outgoingBandwidth);
		}

		public void SetChannelLimit(uint channelLimit)
		{
			CheckChannelLimit(channelLimit);
			Native.enet_host_channel_limit(this.host, channelLimit);
		}

		public event Action Events
		{
			add
			{
				lock (eventsLock)
				{
					events += value;
				}
			}
			remove
			{
				lock (eventsLock)
				{
					events -= value;
				}
			}
		}

		private void OnExecuteEvents()
		{
			Action local = null;
			lock (eventsLock)
			{
				if (events == null)
				{
					return;
				}
				local = events;
				events = null;
			}
			local();
		}

		public void Run()
		{
			// 处理其它线程扔过来的事件
			OnExecuteEvents();

			if (this.Service(0) < 0)
			{
				return;
			}

			Event e;
			while (this.CheckEvents(out e) > 0)
			{
				switch (e.Type)
				{
					case EventType.Connect:
					{
						e.Peer.OnConnected(e);
						break;
					}
					case EventType.Receive:
					{
						e.Peer.OnReceived(e);
						break;
					}
					case EventType.Disconnect:
					{
						e.Peer.OnDisconnect(e);
						break;
					}
				}
			}
		}
	}
}