mono/mcs/class/System/Test/System.Net.Sockets/SocketAsyncTest.cs

using System;
using System.Collections;
using System.Threading;
using System.Net;
using System.Net.Sockets;
using NUnit.Framework;

namespace MonoTests.System.Net.Sockets
{
	[TestFixture]
	public class SocketAsyncTest
	{
		Socket serverSocket;
		Socket clientSocket;
		ManualResetEvent readyEvent;
		ManualResetEvent mainEvent;
		Exception error;

		[TestFixtureSetUp]
		public void SetUp ()
		{
			readyEvent = new ManualResetEvent (false);
			mainEvent = new ManualResetEvent (false);

			ThreadPool.QueueUserWorkItem (_ => DoWork ());
			readyEvent.WaitOne ();

			clientSocket = new Socket (
				AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp);
			clientSocket.Connect (serverSocket.LocalEndPoint);
			clientSocket.NoDelay = true;
		}

		[TestFixtureTearDown]
		public void TearDown ()
		{
			if (serverSocket != null)
				serverSocket.Close ();
			readyEvent.Close ();
			mainEvent.Close ();
		}

		void DoWork ()
		{
			serverSocket = new Socket (
				AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp);
			serverSocket.Bind (new IPEndPoint (IPAddress.Loopback, 0));
			serverSocket.Listen (1);

			var async = new SocketAsyncEventArgs ();
			async.Completed += (s,e) => OnAccepted (e);

			readyEvent.Set ();

			if (!serverSocket.AcceptAsync (async))
				OnAccepted (async);
		}

		void OnAccepted (SocketAsyncEventArgs e)
		{
			var acceptSocket = e.AcceptSocket;

			try {
				var header = new byte [4];
				acceptSocket.Receive (header);
				if ((header [0] != 0x12) || (header [1] != 0x34) ||
				    (header [2] != 0x56) || (header [3] != 0x78))
					throw new InvalidOperationException ();
			} catch (Exception ex) {
				error = ex;
				return;
			}

			var recvAsync = new SocketAsyncEventArgs ();
			recvAsync.Completed += (sender, args) => OnReceived (args);
			recvAsync.SetBuffer (new byte [4], 0, 4);
			if (!acceptSocket.ReceiveAsync (recvAsync))
				OnReceived (recvAsync);

			mainEvent.Set ();
		}

		void OnReceived (SocketAsyncEventArgs e)
		{
			if (e.SocketError != SocketError.Success)
				error = new SocketException ((int) e.SocketError);
			else if (e.Buffer [0] != 0x9a)
				error = new InvalidOperationException ();

			mainEvent.Set ();
		}

		[Test]
		[Category("Test")]
		public void SendAsync ()
		{
			var buffer = new byte [] { 0x12, 0x34, 0x56, 0x78 };
			var m = new ManualResetEvent (false);
			var e = new SocketAsyncEventArgs ();
			e.SetBuffer (buffer, 0, buffer.Length);
			e.Completed += (s,o) => {
				if (o.SocketError != SocketError.Success)
					error = new SocketException ((int)o.SocketError);
				m.Set ();
			};
			bool res = clientSocket.SendAsync (e);
			if (res) {
				if (!m.WaitOne (1500))
					throw new TimeoutException ();
			}

			if (!mainEvent.WaitOne (1500))
				throw new TimeoutException ();
			if (error != null)
				throw error;

			m.Reset ();
			mainEvent.Reset ();

			buffer [0] = 0x9a;
			buffer [1] = 0xbc;
			buffer [2] = 0xde;
			buffer [3] = 0xff;
			res = clientSocket.SendAsync (e);
			if (res) {
				if (!m.WaitOne (1500))
					throw new TimeoutException ();
			}

			if (!mainEvent.WaitOne (1500))
				throw new TimeoutException ();
			if (error != null)
				throw error;
		}
	}
}