451 lines
11 KiB
C#
451 lines
11 KiB
C#
//
|
|
// EndPointListener.cs
|
|
// Copied from System.Net.EndPointListener.cs
|
|
//
|
|
// Author:
|
|
// Gonzalo Paniagua Javier (gonzalo@novell.com)
|
|
//
|
|
// Copyright (c) 2005 Novell, Inc. (http://www.novell.com)
|
|
// Copyright (c) 2012-2013 sta.blockhead
|
|
//
|
|
// 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.
|
|
//
|
|
|
|
using System;
|
|
using System.Collections;
|
|
using System.Collections.Generic;
|
|
using System.IO;
|
|
using System.Net;
|
|
using System.Net.Sockets;
|
|
using System.Security.Cryptography;
|
|
using System.Security.Cryptography.X509Certificates;
|
|
using System.Threading;
|
|
|
|
namespace WebSocketSharp.Net
|
|
{
|
|
internal sealed class EndPointListener
|
|
{
|
|
#region Private Fields
|
|
|
|
List<ListenerPrefix> _all; // host = '+'
|
|
X509Certificate2 _cert;
|
|
IPEndPoint _endpoint;
|
|
Dictionary<ListenerPrefix, HttpListener> _prefixes;
|
|
bool _secure;
|
|
Socket _socket;
|
|
List<ListenerPrefix> _unhandled; // host = '*'
|
|
Dictionary<HttpConnection, HttpConnection> _unregistered;
|
|
|
|
#endregion
|
|
|
|
#region Public Constructors
|
|
|
|
public EndPointListener (
|
|
IPAddress address,
|
|
int port,
|
|
bool secure,
|
|
string certFolderPath,
|
|
X509Certificate2 defaultCert
|
|
)
|
|
{
|
|
if (secure) {
|
|
_secure = secure;
|
|
_cert = getCertificate (port, certFolderPath, defaultCert);
|
|
if (_cert == null)
|
|
throw new ArgumentException ("Server certificate not found.");
|
|
}
|
|
|
|
_endpoint = new IPEndPoint (address, port);
|
|
_socket = new Socket (address.AddressFamily, SocketType.Stream, ProtocolType.Tcp);
|
|
_socket.Bind (_endpoint);
|
|
_socket.Listen (500);
|
|
var args = new SocketAsyncEventArgs ();
|
|
args.UserToken = this;
|
|
args.Completed += onAccept;
|
|
_socket.AcceptAsync (args);
|
|
_prefixes = new Dictionary<ListenerPrefix, HttpListener> ();
|
|
_unregistered = new Dictionary<HttpConnection, HttpConnection> ();
|
|
}
|
|
|
|
#endregion
|
|
|
|
#region Private Methods
|
|
|
|
private static void addSpecial (List<ListenerPrefix> prefixes, ListenerPrefix prefix)
|
|
{
|
|
if (prefixes == null)
|
|
return;
|
|
|
|
foreach (var p in prefixes)
|
|
if (p.Path == prefix.Path) // TODO: code
|
|
throw new HttpListenerException (400, "Prefix already in use.");
|
|
|
|
prefixes.Add (prefix);
|
|
}
|
|
|
|
private void checkIfRemove ()
|
|
{
|
|
if (_prefixes.Count > 0)
|
|
return;
|
|
|
|
if (_unhandled != null && _unhandled.Count > 0)
|
|
return;
|
|
|
|
if (_all != null && _all.Count > 0)
|
|
return;
|
|
|
|
EndPointManager.RemoveEndPoint (this, _endpoint);
|
|
}
|
|
|
|
private static RSACryptoServiceProvider createRSAFromFile (string filename)
|
|
{
|
|
var rsa = new RSACryptoServiceProvider ();
|
|
byte[] pvk = null;
|
|
using (var fs = File.Open (filename, FileMode.Open, FileAccess.Read, FileShare.Read))
|
|
{
|
|
pvk = new byte [fs.Length];
|
|
fs.Read (pvk, 0, pvk.Length);
|
|
}
|
|
|
|
rsa.ImportCspBlob (pvk);
|
|
return rsa;
|
|
}
|
|
|
|
private static X509Certificate2 getCertificate (
|
|
int port, string certFolderPath, X509Certificate2 defaultCert)
|
|
{
|
|
try {
|
|
var cer = Path.Combine (certFolderPath, String.Format ("{0}.cer", port));
|
|
var key = Path.Combine (certFolderPath, String.Format ("{0}.key", port));
|
|
if (File.Exists (cer) && File.Exists (key))
|
|
{
|
|
var cert = new X509Certificate2 (cer);
|
|
cert.PrivateKey = createRSAFromFile (key);
|
|
|
|
return cert;
|
|
}
|
|
}
|
|
catch {
|
|
}
|
|
|
|
return defaultCert;
|
|
}
|
|
|
|
private static HttpListener matchFromList (
|
|
string host, string path, List<ListenerPrefix> list, out ListenerPrefix prefix)
|
|
{
|
|
prefix = null;
|
|
if (list == null)
|
|
return null;
|
|
|
|
HttpListener best_match = null;
|
|
var best_length = -1;
|
|
foreach (var p in list)
|
|
{
|
|
var ppath = p.Path;
|
|
if (ppath.Length < best_length)
|
|
continue;
|
|
|
|
if (path.StartsWith (ppath))
|
|
{
|
|
best_length = ppath.Length;
|
|
best_match = p.Listener;
|
|
prefix = p;
|
|
}
|
|
}
|
|
|
|
return best_match;
|
|
}
|
|
|
|
private static void onAccept (object sender, EventArgs e)
|
|
{
|
|
var args = (SocketAsyncEventArgs) e;
|
|
var epListener = (EndPointListener) args.UserToken;
|
|
Socket accepted = null;
|
|
if (args.SocketError == SocketError.Success)
|
|
{
|
|
accepted = args.AcceptSocket;
|
|
args.AcceptSocket = null;
|
|
}
|
|
|
|
try {
|
|
epListener._socket.AcceptAsync (args);
|
|
}
|
|
catch {
|
|
if (accepted != null)
|
|
accepted.Close ();
|
|
|
|
return;
|
|
}
|
|
|
|
if (accepted == null)
|
|
return;
|
|
|
|
HttpConnection conn = null;
|
|
try {
|
|
conn = new HttpConnection (accepted, epListener, epListener._secure, epListener._cert);
|
|
lock (((ICollection) epListener._unregistered).SyncRoot)
|
|
{
|
|
epListener._unregistered [conn] = conn;
|
|
}
|
|
|
|
conn.BeginReadRequest ();
|
|
}
|
|
catch {
|
|
if (conn != null)
|
|
{
|
|
conn.Close (true);
|
|
return;
|
|
}
|
|
|
|
accepted.Close ();
|
|
}
|
|
}
|
|
|
|
private static bool removeSpecial (List<ListenerPrefix> prefixes, ListenerPrefix prefix)
|
|
{
|
|
if (prefixes == null)
|
|
return false;
|
|
|
|
var count = prefixes.Count;
|
|
for (int i = 0; i < count; i++)
|
|
{
|
|
if (prefixes [i].Path == prefix.Path)
|
|
{
|
|
prefixes.RemoveAt (i);
|
|
return true;
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
private HttpListener searchListener (Uri uri, out ListenerPrefix prefix)
|
|
{
|
|
prefix = null;
|
|
if (uri == null)
|
|
return null;
|
|
|
|
var host = uri.Host;
|
|
var port = uri.Port;
|
|
var path = HttpUtility.UrlDecode (uri.AbsolutePath);
|
|
var path_slash = path [path.Length - 1] == '/' ? path : path + "/";
|
|
HttpListener best_match = null;
|
|
var best_length = -1;
|
|
if (host != null && host.Length > 0)
|
|
{
|
|
foreach (var p in _prefixes.Keys)
|
|
{
|
|
var ppath = p.Path;
|
|
if (ppath.Length < best_length)
|
|
continue;
|
|
|
|
if (p.Host != host || p.Port != port)
|
|
continue;
|
|
|
|
if (path.StartsWith (ppath) || path_slash.StartsWith (ppath))
|
|
{
|
|
best_length = ppath.Length;
|
|
best_match = _prefixes [p];
|
|
prefix = p;
|
|
}
|
|
}
|
|
|
|
if (best_length != -1)
|
|
return best_match;
|
|
}
|
|
|
|
var list = _unhandled;
|
|
best_match = matchFromList (host, path, list, out prefix);
|
|
if (path != path_slash && best_match == null)
|
|
best_match = matchFromList (host, path_slash, list, out prefix);
|
|
|
|
if (best_match != null)
|
|
return best_match;
|
|
|
|
list = _all;
|
|
best_match = matchFromList (host, path, list, out prefix);
|
|
if (path != path_slash && best_match == null)
|
|
best_match = matchFromList (host, path_slash, list, out prefix);
|
|
|
|
if (best_match != null)
|
|
return best_match;
|
|
|
|
return null;
|
|
}
|
|
|
|
#endregion
|
|
|
|
#region Internal Methods
|
|
|
|
internal static bool CertificateExists (int port, string certFolderPath)
|
|
{
|
|
var cer = Path.Combine (certFolderPath, String.Format ("{0}.cer", port));
|
|
var key = Path.Combine (certFolderPath, String.Format ("{0}.key", port));
|
|
|
|
return File.Exists (cer) && File.Exists (key);
|
|
}
|
|
|
|
internal void RemoveConnection (HttpConnection connection)
|
|
{
|
|
lock (((ICollection) _unregistered).SyncRoot)
|
|
{
|
|
_unregistered.Remove (connection);
|
|
}
|
|
}
|
|
|
|
#endregion
|
|
|
|
#region Public Methods
|
|
|
|
public void AddPrefix (ListenerPrefix prefix, HttpListener listener)
|
|
{
|
|
List<ListenerPrefix> current, future;
|
|
if (prefix.Host == "*")
|
|
{
|
|
do {
|
|
current = _unhandled;
|
|
future = current != null
|
|
? new List<ListenerPrefix> (current)
|
|
: new List<ListenerPrefix> ();
|
|
prefix.Listener = listener;
|
|
addSpecial (future, prefix);
|
|
} while (Interlocked.CompareExchange (ref _unhandled, future, current) != current);
|
|
|
|
return;
|
|
}
|
|
|
|
if (prefix.Host == "+")
|
|
{
|
|
do {
|
|
current = _all;
|
|
future = current != null
|
|
? new List<ListenerPrefix> (current)
|
|
: new List<ListenerPrefix> ();
|
|
prefix.Listener = listener;
|
|
addSpecial (future, prefix);
|
|
} while (Interlocked.CompareExchange (ref _all, future, current) != current);
|
|
|
|
return;
|
|
}
|
|
|
|
Dictionary<ListenerPrefix, HttpListener> prefs, p2;
|
|
do {
|
|
prefs = _prefixes;
|
|
if (prefs.ContainsKey (prefix))
|
|
{
|
|
HttpListener other = prefs [prefix];
|
|
if (other != listener) // TODO: code.
|
|
throw new HttpListenerException (400, "There's another listener for " + prefix);
|
|
|
|
return;
|
|
}
|
|
|
|
p2 = new Dictionary<ListenerPrefix, HttpListener> (prefs);
|
|
p2 [prefix] = listener;
|
|
} while (Interlocked.CompareExchange (ref _prefixes, p2, prefs) != prefs);
|
|
}
|
|
|
|
public bool BindContext (HttpListenerContext context)
|
|
{
|
|
var req = context.Request;
|
|
ListenerPrefix prefix;
|
|
var listener = searchListener (req.Url, out prefix);
|
|
if (listener == null)
|
|
return false;
|
|
|
|
context.Listener = listener;
|
|
context.Connection.Prefix = prefix;
|
|
|
|
return true;
|
|
}
|
|
|
|
public void Close ()
|
|
{
|
|
_socket.Close ();
|
|
lock (((ICollection) _unregistered).SyncRoot)
|
|
{
|
|
var copy = new Dictionary<HttpConnection, HttpConnection> (_unregistered);
|
|
foreach (var conn in copy.Keys)
|
|
conn.Close (true);
|
|
|
|
copy.Clear ();
|
|
_unregistered.Clear ();
|
|
}
|
|
}
|
|
|
|
public void RemovePrefix (ListenerPrefix prefix, HttpListener listener)
|
|
{
|
|
List<ListenerPrefix> current, future;
|
|
if (prefix.Host == "*")
|
|
{
|
|
do {
|
|
current = _unhandled;
|
|
future = current != null
|
|
? new List<ListenerPrefix> (current)
|
|
: new List<ListenerPrefix> ();
|
|
if (!removeSpecial (future, prefix))
|
|
break; // Prefix not found.
|
|
} while (Interlocked.CompareExchange (ref _unhandled, future, current) != current);
|
|
|
|
checkIfRemove ();
|
|
return;
|
|
}
|
|
|
|
if (prefix.Host == "+")
|
|
{
|
|
do {
|
|
current = _all;
|
|
future = current != null
|
|
? new List<ListenerPrefix> (current)
|
|
: new List<ListenerPrefix> ();
|
|
if (!removeSpecial (future, prefix))
|
|
break; // Prefix not found.
|
|
} while (Interlocked.CompareExchange (ref _all, future, current) != current);
|
|
|
|
checkIfRemove ();
|
|
return;
|
|
}
|
|
|
|
Dictionary<ListenerPrefix, HttpListener> prefs, p2;
|
|
do {
|
|
prefs = _prefixes;
|
|
if (!prefs.ContainsKey (prefix))
|
|
break;
|
|
|
|
p2 = new Dictionary<ListenerPrefix, HttpListener> (prefs);
|
|
p2.Remove (prefix);
|
|
} while (Interlocked.CompareExchange (ref _prefixes, p2, prefs) != prefs);
|
|
|
|
checkIfRemove ();
|
|
}
|
|
|
|
public void UnbindContext (HttpListenerContext context)
|
|
{
|
|
if (context == null || context.Request == null)
|
|
return;
|
|
|
|
context.Listener.UnregisterContext (context);
|
|
}
|
|
|
|
#endregion
|
|
}
|
|
}
|