ListenerAsyncResult.cs

//
// System.Net.ListenerAsyncResult
//
// Authors:
//	Gonzalo Paniagua Javier (gonzalo@ximian.com)
//
// Copyright (c) 2005 Ximian, Inc (http://www.ximian.com)
//

// Copyright (c) 2018 Nivloc Enterprises Ltd

//
// Permission is hereby granted, free of charge, to any person obtaining
// a copy of this software and associated documentation files (the
// "Software"), to deal in the Software without restriction, including
// without limitation the rights to use, copy, modify, merge, publish,
// distribute, sublicense, and/or sell copies of the Software, and to
// permit persons to whom the Software is furnished to do so, subject to
// the following conditions:
// 
// The above copyright notice and this permission notice shall be
// included in all copies or substantial portions of the Software.
// 
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
// EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
// MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
// NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
// LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
// OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
// WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
//

#if SECURITY_DEP
using System;
#if SSHARP
using Crestron.SimplSharp;
using IAsyncResult = Crestron.SimplSharp.CrestronIO.IAsyncResult;
using AsyncCallback = Crestron.SimplSharp.CrestronIO.AsyncCallback;
using SSMono.Threading;
#else
using System.Threading;
#endif

#if SSHARP
namespace SSMono.Net
#else
namespace System.Net
#endif
	{
	internal class ListenerAsyncResult : IAsyncResult
		{
		private ManualResetEvent handle;
		private bool synch;
		private bool completed;
		private AsyncCallback cb;
		private object state;
		private Exception exception;
		private HttpListenerContext context;
		private object locker = new object ();
		private ListenerAsyncResult forward;
		internal bool EndCalled;
		internal bool InGet;

		public ListenerAsyncResult (AsyncCallback cb, object state)
			{
			this.cb = cb;
			this.state = state;
			}

		internal void Complete (Exception exc)
			{
			if (forward != null)
				{
				forward.Complete (exc);
				return;
				}
			exception = exc;
			if (InGet && (exc is ObjectDisposedException))
				exception = new HttpListenerException (500, "Listener closed");
			lock (locker)
				{
				completed = true;
				if (handle != null)
					handle.Set ();

				if (cb != null)
					ThreadPool.UnsafeQueueUserWorkItem (InvokeCB, this);
				}
			}

		private static WaitCallback InvokeCB = new WaitCallback (InvokeCallback);

		private static void InvokeCallback (object o)
			{
			ListenerAsyncResult ares = (ListenerAsyncResult)o;
			if (ares.forward != null)
				{
				InvokeCallback (ares.forward);
				return;
				}
			try
				{
				ares.cb (ares);
				}
			catch
				{
				}
			}

		internal void Complete (HttpListenerContext context)
			{
			Complete (context, false);
			}

		internal void Complete (HttpListenerContext context, bool synch)
			{
			if (forward != null)
				{
				forward.Complete (context, synch);
				return;
				}
			this.synch = synch;
			this.context = context;
			lock (locker)
				{
				AuthenticationSchemes schemes = context.Listener.SelectAuthenticationScheme (context);
				if ((schemes == AuthenticationSchemes.Basic || context.Listener.AuthenticationSchemes == AuthenticationSchemes.Negotiate)
				    && context.Request.Headers["Authorization"] == null)
					{
					context.Response.StatusCode = 401;
					context.Response.Headers["WWW-Authenticate"] = schemes + " realm=\"" + context.Listener.Realm + "\"";
					context.Response.OutputStream.Close ();
					IAsyncResult ares = context.Listener.BeginGetContext (cb, state);
					this.forward = (ListenerAsyncResult)ares;
					lock (forward.locker)
						{
						if (handle != null)
							forward.handle = handle;
						}
					ListenerAsyncResult next = forward;
					for (int i = 0; next.forward != null; i++)
						{
						if (i > 20)
							Complete (new HttpListenerException (400, "Too many authentication errors"));
						next = next.forward;
						}
					}
				else
					{
					completed = true;
					this.synch = false;

					if (handle != null)
						handle.Set ();

					if (cb != null)
						ThreadPool.UnsafeQueueUserWorkItem (InvokeCB, this);
					}
				}
			}

		internal HttpListenerContext GetContext ()
			{
			if (forward != null)
				return forward.GetContext ();
			if (exception != null)
				throw exception;

			return context;
			}

		public object AsyncState
			{
			get
				{
				if (forward != null)
					return forward.AsyncState;
				return state;
				}
			}

		public bool CompletedSynchronously
			{
			get
				{
				if (forward != null)
					return forward.CompletedSynchronously;
				return synch;
				}
			}

		public bool IsCompleted
			{
			get
				{
				if (forward != null)
					return forward.IsCompleted;

				lock (locker)
					{
					return completed;
					}
				}
			}

		#region IAsyncResult Members


#if SSHARP
		public CEventHandle AsyncWaitHandle
			{
			get
				{
				if (forward != null)
					return forward.AsyncWaitHandle;

				lock (locker)
					{
					if (handle == null)
						handle = new ManualResetEvent (completed);
					}

				return handle;
				}
			}

		public object InnerObject
			{
			get { throw new NotImplementedException (); }
			}
#else
		public WaitHandle AsyncWaitHandle
			{
			get
				{
				if (forward != null)
					return forward.AsyncWaitHandle;

				lock (locker)
					{
					if (handle == null)
						handle = new ManualResetEvent (completed);
					}

				return handle;
				}
			}
#endif

		#endregion
		}
	}

#endif