Refactored EndPointManager.cs
This commit is contained in:
parent
dae4e957cd
commit
8ec0711bde
@ -1,32 +1,41 @@
|
||||
//
|
||||
// EndPointManager.cs
|
||||
// Copied from System.Net.EndPointManager.cs
|
||||
//
|
||||
// Author:
|
||||
// Gonzalo Paniagua Javier (gonzalo@ximian.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.
|
||||
//
|
||||
#region License
|
||||
/*
|
||||
* EndPointManager.cs
|
||||
*
|
||||
* This code is derived from System.Net.EndPointManager.cs of Mono
|
||||
* (http://www.mono-project.com).
|
||||
*
|
||||
* The MIT License
|
||||
*
|
||||
* Copyright (c) 2005 Novell, Inc. (http://www.novell.com)
|
||||
* Copyright (c) 2012-2014 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.
|
||||
*/
|
||||
#endregion
|
||||
|
||||
#region Authors
|
||||
/*
|
||||
* Authors:
|
||||
* - Gonzalo Paniagua Javier <gonzalo@ximian.com>
|
||||
*/
|
||||
#endregion
|
||||
|
||||
using System;
|
||||
using System.Collections;
|
||||
@ -35,143 +44,143 @@ using System.Net;
|
||||
|
||||
namespace WebSocketSharp.Net
|
||||
{
|
||||
internal sealed class EndPointManager
|
||||
{
|
||||
#region Private Fields
|
||||
internal sealed class EndPointManager
|
||||
{
|
||||
#region Private Static Fields
|
||||
|
||||
private static Dictionary<IPAddress, Dictionary<int, EndPointListener>> _ipToEndpoints = new Dictionary<IPAddress, Dictionary<int, EndPointListener>> ();
|
||||
private static Dictionary<IPAddress, Dictionary<int, EndPointListener>> _ipToEndpoints =
|
||||
new Dictionary<IPAddress, Dictionary<int, EndPointListener>> ();
|
||||
|
||||
#endregion
|
||||
#endregion
|
||||
|
||||
#region Private Constructors
|
||||
#region Private Constructors
|
||||
|
||||
private EndPointManager ()
|
||||
{
|
||||
}
|
||||
private EndPointManager ()
|
||||
{
|
||||
}
|
||||
|
||||
#endregion
|
||||
#endregion
|
||||
|
||||
#region Private Methods
|
||||
#region Private Methods
|
||||
|
||||
private static void addPrefix (string uriPrefix, HttpListener httpListener)
|
||||
{
|
||||
var prefix = new ListenerPrefix (uriPrefix);
|
||||
if (prefix.Path.IndexOf ('%') != -1)
|
||||
throw new HttpListenerException (400, "Invalid path.");
|
||||
private static void addPrefix (string uriPrefix, HttpListener httpListener)
|
||||
{
|
||||
var prefix = new ListenerPrefix (uriPrefix);
|
||||
if (prefix.Path.IndexOf ('%') != -1)
|
||||
throw new HttpListenerException (400, "Invalid path.");
|
||||
|
||||
if (prefix.Path.IndexOf ("//", StringComparison.Ordinal) != -1) // TODO: Code?
|
||||
throw new HttpListenerException (400, "Invalid path.");
|
||||
if (prefix.Path.IndexOf ("//", StringComparison.Ordinal) != -1)
|
||||
throw new HttpListenerException (400, "Invalid path."); // TODO: Code?
|
||||
|
||||
// Always listens on all the interfaces, no matter the host name/ip used.
|
||||
var epListener = getEndPointListener (IPAddress.Any, prefix.Port, httpListener, prefix.Secure);
|
||||
epListener.AddPrefix (prefix, httpListener);
|
||||
}
|
||||
// Always listens on all the interfaces, no matter the host name/ip used.
|
||||
var epListener = getEndPointListener (
|
||||
IPAddress.Any, prefix.Port, httpListener, prefix.Secure);
|
||||
|
||||
private static EndPointListener getEndPointListener (
|
||||
IPAddress address, int port, HttpListener httpListener, bool secure)
|
||||
{
|
||||
Dictionary<int, EndPointListener> endpoints = null;
|
||||
if (_ipToEndpoints.ContainsKey (address))
|
||||
{
|
||||
endpoints = _ipToEndpoints [address];
|
||||
}
|
||||
else
|
||||
{
|
||||
endpoints = new Dictionary<int, EndPointListener> ();
|
||||
_ipToEndpoints [address] = endpoints;
|
||||
}
|
||||
epListener.AddPrefix (prefix, httpListener);
|
||||
}
|
||||
|
||||
EndPointListener epListener = null;
|
||||
if (endpoints.ContainsKey (port))
|
||||
{
|
||||
epListener = endpoints [port];
|
||||
}
|
||||
else
|
||||
{
|
||||
epListener = new EndPointListener (
|
||||
address, port, secure, httpListener.CertificateFolderPath, httpListener.DefaultCertificate);
|
||||
endpoints [port] = epListener;
|
||||
}
|
||||
private static EndPointListener getEndPointListener (
|
||||
IPAddress address, int port, HttpListener httpListener, bool secure)
|
||||
{
|
||||
Dictionary<int, EndPointListener> endpoints = null;
|
||||
if (_ipToEndpoints.ContainsKey (address)) {
|
||||
endpoints = _ipToEndpoints [address];
|
||||
}
|
||||
else {
|
||||
endpoints = new Dictionary<int, EndPointListener> ();
|
||||
_ipToEndpoints [address] = endpoints;
|
||||
}
|
||||
|
||||
return epListener;
|
||||
}
|
||||
EndPointListener epListener = null;
|
||||
if (endpoints.ContainsKey (port)) {
|
||||
epListener = endpoints [port];
|
||||
}
|
||||
else {
|
||||
epListener = new EndPointListener (
|
||||
address,
|
||||
port,
|
||||
secure,
|
||||
httpListener.CertificateFolderPath,
|
||||
httpListener.DefaultCertificate);
|
||||
|
||||
private static void removePrefix (string uriPrefix, HttpListener httpListener)
|
||||
{
|
||||
var prefix = new ListenerPrefix (uriPrefix);
|
||||
if (prefix.Path.IndexOf ('%') != -1)
|
||||
return;
|
||||
endpoints [port] = epListener;
|
||||
}
|
||||
|
||||
if (prefix.Path.IndexOf ("//", StringComparison.Ordinal) != -1)
|
||||
return;
|
||||
return epListener;
|
||||
}
|
||||
|
||||
var epListener = getEndPointListener (IPAddress.Any, prefix.Port, httpListener, prefix.Secure);
|
||||
epListener.RemovePrefix (prefix, httpListener);
|
||||
}
|
||||
private static void removePrefix (string uriPrefix, HttpListener httpListener)
|
||||
{
|
||||
var prefix = new ListenerPrefix (uriPrefix);
|
||||
if (prefix.Path.IndexOf ('%') != -1)
|
||||
return;
|
||||
|
||||
#endregion
|
||||
if (prefix.Path.IndexOf ("//", StringComparison.Ordinal) != -1)
|
||||
return;
|
||||
|
||||
#region Public Methods
|
||||
var epListener = getEndPointListener (
|
||||
IPAddress.Any, prefix.Port, httpListener, prefix.Secure);
|
||||
|
||||
public static void AddListener (HttpListener httpListener)
|
||||
{
|
||||
var added = new List<string> ();
|
||||
lock (((ICollection) _ipToEndpoints).SyncRoot)
|
||||
{
|
||||
try {
|
||||
foreach (var prefix in httpListener.Prefixes)
|
||||
{
|
||||
addPrefix (prefix, httpListener);
|
||||
added.Add (prefix);
|
||||
}
|
||||
}
|
||||
catch {
|
||||
foreach (var prefix in added)
|
||||
removePrefix (prefix, httpListener);
|
||||
epListener.RemovePrefix (prefix, httpListener);
|
||||
}
|
||||
|
||||
throw;
|
||||
}
|
||||
}
|
||||
}
|
||||
#endregion
|
||||
|
||||
public static void AddPrefix (string uriPrefix, HttpListener httpListener)
|
||||
{
|
||||
lock (((ICollection) _ipToEndpoints).SyncRoot)
|
||||
{
|
||||
addPrefix (uriPrefix, httpListener);
|
||||
}
|
||||
}
|
||||
#region Public Methods
|
||||
|
||||
public static void RemoveEndPoint (EndPointListener epListener, IPEndPoint endpoint)
|
||||
{
|
||||
lock (((ICollection) _ipToEndpoints).SyncRoot)
|
||||
{
|
||||
var endpoints = _ipToEndpoints [endpoint.Address];
|
||||
endpoints.Remove (endpoint.Port);
|
||||
if (endpoints.Count == 0)
|
||||
_ipToEndpoints.Remove (endpoint.Address);
|
||||
public static void AddListener (HttpListener httpListener)
|
||||
{
|
||||
var added = new List<string> ();
|
||||
lock (((ICollection) _ipToEndpoints).SyncRoot) {
|
||||
try {
|
||||
foreach (var prefix in httpListener.Prefixes) {
|
||||
addPrefix (prefix, httpListener);
|
||||
added.Add (prefix);
|
||||
}
|
||||
}
|
||||
catch {
|
||||
foreach (var prefix in added)
|
||||
removePrefix (prefix, httpListener);
|
||||
|
||||
epListener.Close ();
|
||||
}
|
||||
}
|
||||
throw;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
public static void RemoveListener (HttpListener httpListener)
|
||||
{
|
||||
lock (((ICollection) _ipToEndpoints).SyncRoot)
|
||||
{
|
||||
foreach (var prefix in httpListener.Prefixes)
|
||||
removePrefix (prefix, httpListener);
|
||||
}
|
||||
}
|
||||
public static void AddPrefix (string uriPrefix, HttpListener httpListener)
|
||||
{
|
||||
lock (((ICollection) _ipToEndpoints).SyncRoot) {
|
||||
addPrefix (uriPrefix, httpListener);
|
||||
}
|
||||
}
|
||||
|
||||
public static void RemovePrefix (string uriPrefix, HttpListener httpListener)
|
||||
{
|
||||
lock (((ICollection) _ipToEndpoints).SyncRoot)
|
||||
{
|
||||
removePrefix (uriPrefix, httpListener);
|
||||
}
|
||||
}
|
||||
public static void RemoveEndPoint (EndPointListener epListener, IPEndPoint endpoint)
|
||||
{
|
||||
lock (((ICollection) _ipToEndpoints).SyncRoot) {
|
||||
var endpoints = _ipToEndpoints [endpoint.Address];
|
||||
endpoints.Remove (endpoint.Port);
|
||||
if (endpoints.Count == 0)
|
||||
_ipToEndpoints.Remove (endpoint.Address);
|
||||
|
||||
#endregion
|
||||
}
|
||||
epListener.Close ();
|
||||
}
|
||||
}
|
||||
|
||||
public static void RemoveListener (HttpListener httpListener)
|
||||
{
|
||||
lock (((ICollection) _ipToEndpoints).SyncRoot) {
|
||||
foreach (var prefix in httpListener.Prefixes)
|
||||
removePrefix (prefix, httpListener);
|
||||
}
|
||||
}
|
||||
|
||||
public static void RemovePrefix (string uriPrefix, HttpListener httpListener)
|
||||
{
|
||||
lock (((ICollection) _ipToEndpoints).SyncRoot) {
|
||||
removePrefix (uriPrefix, httpListener);
|
||||
}
|
||||
}
|
||||
|
||||
#endregion
|
||||
}
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user