summaryrefslogtreecommitdiffstats
path: root/src/DotNetOpenAuth.Test/Hosting/HttpHost.cs
blob: ee3cb6c6eeb7152e16a96eda86c23d42ed746e30 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
//-----------------------------------------------------------------------
// <copyright file="HttpHost.cs" company="Outercurve Foundation">
//     Copyright (c) Outercurve Foundation. All rights reserved.
// </copyright>
//-----------------------------------------------------------------------

namespace DotNetOpenAuth.Test.Hosting {
	using System;
	using System.Globalization;
	using System.IO;
	using System.Net;
	using System.Threading;

	internal class HttpHost : IDisposable {
		private readonly HttpListener listener;
		private Thread listenerThread;
		private AspNetHost aspNetHost;

		private HttpHost(AspNetHost aspNetHost) {
			this.aspNetHost = aspNetHost;

			this.Port = 59687;
			Random r = new Random();
		tryAgain:
			try {
				this.listener = new HttpListener();
				this.listener.Prefixes.Add(string.Format(CultureInfo.InvariantCulture, "http://localhost:{0}/", this.Port));
				this.listener.Start();
			} catch (HttpListenerException ex) {
				if (ex.Message.Contains("conflicts")) {
					this.Port += r.Next(1, 20);
					goto tryAgain;
				}
				throw;
			}
			this.listenerThread = new Thread(this.ProcessRequests);
			this.listenerThread.Start();
		}

		public int Port { get; private set; }

		public Uri BaseUri {
			get { return new Uri("http://localhost:" + this.Port.ToString() + "/"); }
		}

		public static HttpHost CreateHost(AspNetHost aspNetHost) {
			return new HttpHost(aspNetHost);
		}

		public static HttpHost CreateHost(string webDirectory) {
			return new HttpHost(AspNetHost.CreateHost(webDirectory));
		}

		public string ProcessRequest(string url) {
			return this.ProcessRequest(url, null);
		}

		public string ProcessRequest(string url, string body) {
			WebRequest request = WebRequest.Create(new Uri(this.BaseUri, url));
			if (body != null) {
				request.Method = "POST";
				request.ContentLength = body.Length;
				using (StreamWriter sw = new StreamWriter(request.GetRequestStream())) {
					sw.Write(body);
				}
			}
			try {
				using (WebResponse response = request.GetResponse()) {
					using (StreamReader sr = new StreamReader(response.GetResponseStream())) {
						return sr.ReadToEnd();
					}
				}
			} catch (WebException ex) {
				Logger.Http.Error("Exception in HttpHost", ex);
				using (StreamReader sr = new StreamReader(ex.Response.GetResponseStream())) {
					string streamContent = sr.ReadToEnd();
					Logger.Http.ErrorFormat("Error content stream follows: {0}", streamContent);
				}
				throw;
			}
		}

		#region IDisposable Members

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

		protected virtual void Dispose(bool disposing) {
			if (disposing) {
				this.listener.Close();
				this.listenerThread.Join(1000);
				this.listenerThread.Abort();
			}
		}

		#endregion

		private void ProcessRequests() {
			try {
				while (true) {
					var context = this.listener.GetContext();
					this.aspNetHost.BeginProcessRequest(context);
				}
			} catch (HttpListenerException) {
				// the listener is probably being shut down
			}
		}
	}
}