DevWiki/websocket-sharp
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Modified validating the cookies from client

Browse Source
This commit is contained in:
sta 2013-08-08 15:12:29 +09:00
parent 93b30f94d7
commit acde107832
3 changed files with 165 additions and 132 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
Example2/Echo.cs
View File

@ -3,24 +3,24 @@ using WebSocketSharp;
using WebSocketSharp.Net; using WebSocketSharp.Net;
using WebSocketSharp.Server; using WebSocketSharp.Server;
namespace Example2 { namespace Example2
{
public class Echo : WebSocketService public class Echo : WebSocketService
{ {
protected override void OnMessage(MessageEventArgs e) protected override void OnMessage (MessageEventArgs e)
{ {
var msg = QueryString.Contains("name") var msg = QueryString.Contains ("name")
? String.Format("'{0}' returns to {1}", e.Data, QueryString["name"]) ? String.Format ("Returns '{0}' to {1}", e.Data, QueryString ["name"])
: e.Data; : e.Data;
Send(msg); Send (msg);
} }
protected override bool ProcessCookies(CookieCollection request, CookieCollection response) protected override bool ValidateCookies (CookieCollection request, CookieCollection response)
{ {
foreach (Cookie cookie in request) foreach (Cookie cookie in request)
{ {
cookie.Expired = true; cookie.Expired = true;
response.Add(cookie); response.Add (cookie);
} }
return true; return true;

258
websocket-sharp/Server/WebSocketService.cs
View File

@ -33,16 +33,16 @@ using System.Threading;
using WebSocketSharp.Net; using WebSocketSharp.Net;
using WebSocketSharp.Net.WebSockets; using WebSocketSharp.Net.WebSockets;
namespace WebSocketSharp.Server { namespace WebSocketSharp.Server
{
/// <summary> /// <summary>
/// Provides the basic functions of the WebSocket service. /// Provides the basic functions of the WebSocket service.
/// </summary> /// </summary>
/// <remarks> /// <remarks>
/// The WebSocketService class is an abstract class. /// The WebSocketService class is an abstract class.
/// </remarks> /// </remarks>
public abstract class WebSocketService { public abstract class WebSocketService
{
#region Private Fields #region Private Fields
private WebSocketContext _context; private WebSocketContext _context;
@ -56,7 +56,7 @@ namespace WebSocketSharp.Server {
/// <summary> /// <summary>
/// Initializes a new instance of the <see cref="WebSocketService"/> class. /// Initializes a new instance of the <see cref="WebSocketService"/> class.
/// </summary> /// </summary>
public WebSocketService() public WebSocketService ()
{ {
ID = String.Empty; ID = String.Empty;
IsBound = false; IsBound = false;
@ -80,7 +80,7 @@ namespace WebSocketSharp.Server {
/// Gets or sets the logging functions. /// Gets or sets the logging functions.
/// </summary> /// </summary>
/// <remarks> /// <remarks>
/// If you wanted to change the current logger to the service own logger, you would set this property /// If you want to change the current logger to the service own logger, you set this property
/// to a new <see cref="Logger"/> instance that you created. /// to a new <see cref="Logger"/> instance that you created.
/// </remarks> /// </remarks>
/// <value> /// <value>
@ -102,7 +102,7 @@ namespace WebSocketSharp.Server {
} }
/// <summary> /// <summary>
/// Gets the collection of query string variables used in the WebSocket opening handshake. /// Gets the collection of query string variables used in the WebSocket connection request.
/// </summary> /// </summary>
/// <value> /// <value>
/// A <see cref="NameValueCollection"/> that contains the collection of query string variables. /// A <see cref="NameValueCollection"/> that contains the collection of query string variables.
@ -116,10 +116,11 @@ namespace WebSocketSharp.Server {
} }
/// <summary> /// <summary>
/// Gets the sessions to the <see cref="WebSocketService"/>. /// Gets the sessions to the <see cref="WebSocketService"/> instances.
/// </summary> /// </summary>
/// <value> /// <value>
/// A <see cref="WebSocketServiceManager"/> that contains the sessions to the the <see cref="WebSocketService"/>. /// A <see cref="WebSocketServiceManager"/> that contains the sessions to
/// the <see cref="WebSocketService"/> instances.
/// </value> /// </value>
protected WebSocketServiceManager Sessions { protected WebSocketServiceManager Sessions {
get { get {
@ -134,82 +135,82 @@ namespace WebSocketSharp.Server {
#region Public Properties #region Public Properties
/// <summary> /// <summary>
/// Gets the ID of the <see cref="WebSocketService"/> instance. /// Gets the ID of the current <see cref="WebSocketService"/> instance.
/// </summary> /// </summary>
/// <value> /// <value>
/// A <see cref="string"/> that contains an ID. /// A <see cref="string"/> that contains an ID.
/// </value> /// </value>
public string ID { get; private set; } public string ID {
get; private set;
}
/// <summary> /// <summary>
/// Gets a value indicating whether the <see cref="WebSocketService"/> instance is bound to a <see cref="WebSocket"/>. /// Gets a value indicating whether the current <see cref="WebSocketService"/> instance
/// has been bound to a <see cref="WebSocket"/>.
/// </summary> /// </summary>
/// <value> /// <value>
/// <c>true</c> if the <see cref="WebSocketService"/> instance is bound to a <see cref="WebSocket"/>; otherwise, <c>false</c>. /// <c>true</c> if the current <see cref="WebSocketService"/> instance has been bound to
/// a <see cref="WebSocket"/>; otherwise, <c>false</c>.
/// </value> /// </value>
public bool IsBound { get; private set; } public bool IsBound {
get; private set;
}
#endregion #endregion
#region Private Methods #region Private Methods
private void onClose(object sender, CloseEventArgs e) private void onClose (object sender, CloseEventArgs e)
{ {
_sessions.Remove(ID); _sessions.Remove (ID);
OnClose(e); OnClose (e);
} }
private void onError(object sender, ErrorEventArgs e) private void onError (object sender, ErrorEventArgs e)
{ {
OnError(e); OnError (e);
} }
private void onMessage(object sender, MessageEventArgs e) private void onMessage (object sender, MessageEventArgs e)
{ {
OnMessage(e); OnMessage (e);
} }
private void onOpen(object sender, EventArgs e) private void onOpen (object sender, EventArgs e)
{ {
ID = _sessions.Add(this); ID = _sessions.Add (this);
OnOpen(); OnOpen ();
} }
#endregion #endregion
#region Internal Methods #region Internal Methods
internal void Bind(WebSocketContext context, WebSocketServiceManager sessions) internal void Bind (WebSocketContext context, WebSocketServiceManager sessions)
{ {
if (IsBound) if (IsBound)
return; return;
if (!ProcessCookies(context.CookieCollection, context.WebSocket.CookieCollection))
{
context.WebSocket.Close(HttpStatusCode.BadRequest);
return;
}
_context = context; _context = context;
_sessions = sessions; _sessions = sessions;
_websocket = context.WebSocket; _websocket = context.WebSocket;
_websocket.CookiesValidation = ValidateCookies;
_websocket.OnOpen += onOpen; _websocket.OnOpen += onOpen;
_websocket.OnMessage += onMessage; _websocket.OnMessage += onMessage;
_websocket.OnError += onError; _websocket.OnError += onError;
_websocket.OnClose += onClose; _websocket.OnClose += onClose;
IsBound = true; IsBound = true;
} }
internal void SendAsync(byte[] data, Action completed) internal void SendAsync (byte [] data, Action completed)
{ {
_websocket.SendAsync(data, completed); _websocket.SendAsync (data, completed);
} }
internal void SendAsync(string data, Action completed) internal void SendAsync (string data, Action completed)
{ {
_websocket.SendAsync(data, completed); _websocket.SendAsync (data, completed);
} }
#endregion #endregion
@ -217,55 +218,64 @@ namespace WebSocketSharp.Server {
#region Protected Methods #region Protected Methods
/// <summary> /// <summary>
/// Occurs when the inner <see cref="WebSocket"/> receives a Close frame or the Stop method is called. /// Is called when the WebSocket connection has been closed.
/// </summary> /// </summary>
/// <param name="e"> /// <param name="e">
/// A <see cref="CloseEventArgs"/> that contains the event data associated with a <see cref="WebSocket.OnClose"/> event. /// A <see cref="CloseEventArgs"/> that contains an event data associated with
/// an inner <see cref="WebSocket.OnClose"/> event.
/// </param> /// </param>
protected virtual void OnClose(CloseEventArgs e) protected virtual void OnClose (CloseEventArgs e)
{ {
} }
/// <summary> /// <summary>
/// Occurs when the inner <see cref="WebSocket"/> gets an error. /// Is called when the inner <see cref="WebSocket"/> gets an error.
/// </summary> /// </summary>
/// <param name="e"> /// <param name="e">
/// An <see cref="ErrorEventArgs"/> that contains the event data associated with a <see cref="WebSocket.OnError"/> event. /// An <see cref="ErrorEventArgs"/> that contains an event data associated with
/// an inner <see cref="WebSocket.OnError"/> event.
/// </param> /// </param>
protected virtual void OnError(ErrorEventArgs e) protected virtual void OnError (ErrorEventArgs e)
{ {
} }
/// <summary> /// <summary>
/// Occurs when the inner <see cref="WebSocket"/> receives a data frame. /// Is called when the inner <see cref="WebSocket"/> receives a data frame.
/// </summary> /// </summary>
/// <param name="e"> /// <param name="e">
/// A <see cref="MessageEventArgs"/> that contains the event data associated with a <see cref="WebSocket.OnMessage"/> event. /// A <see cref="MessageEventArgs"/> that contains an event data associated with
/// an inner <see cref="WebSocket.OnMessage"/> event.
/// </param> /// </param>
protected virtual void OnMessage(MessageEventArgs e) protected virtual void OnMessage (MessageEventArgs e)
{ {
} }
/// <summary> /// <summary>
/// Occurs when the WebSocket connection has been established. /// Is called when the WebSocket connection has been established.
/// </summary> /// </summary>
protected virtual void OnOpen() protected virtual void OnOpen ()
{ {
} }
/// <summary> /// <summary>
/// Processes the cookies used in the WebSocket opening handshake. /// Validates the cookies used in the WebSocket connection request.
/// </summary> /// </summary>
/// <remarks>
/// This method is called when the inner <see cref="WebSocket"/> validates
/// the WebSocket connection request.
/// </remarks>
/// <returns> /// <returns>
/// <c>true</c> if processing the cookies is successfully; otherwise, <c>false</c>. /// <c>true</c> if the cookies is valid; otherwise, <c>false</c>.
/// The default returns <c>true</c>.
/// </returns> /// </returns>
/// <param name="request"> /// <param name="request">
/// A <see cref="CookieCollection"/> that contains a collection of the HTTP Cookies received from the client. /// A <see cref="CookieCollection"/> that contains a collection of the HTTP Cookies
/// to validate.
/// </param> /// </param>
/// <param name="response"> /// <param name="response">
/// A <see cref="CookieCollection"/> that contains a collection of the HTTP Cookies to send to the client. /// A <see cref="CookieCollection"/> that receives the HTTP Cookies to send to the client.
/// </param> /// </param>
protected virtual bool ProcessCookies(CookieCollection request, CookieCollection response) protected virtual bool ValidateCookies (CookieCollection request, CookieCollection response)
{ {
return true; return true;
} }
@ -275,150 +285,155 @@ namespace WebSocketSharp.Server {
#region Public Methods #region Public Methods
/// <summary> /// <summary>
/// Broadcasts the specified array of <see cref="byte"/> to the clients of every <see cref="WebSocketService"/> instances /// Broadcasts the specified array of <see cref="byte"/> to the clients of
/// in the <see cref="WebSocketService.Sessions"/>. /// every <see cref="WebSocketService"/> instances in the <see cref="WebSocketService.Sessions"/>.
/// </summary> /// </summary>
/// <param name="data"> /// <param name="data">
/// An array of <see cref="byte"/> to broadcast. /// An array of <see cref="byte"/> to broadcast.
/// </param> /// </param>
public void Broadcast(byte[] data) public void Broadcast (byte [] data)
{ {
if (IsBound) if (IsBound)
_sessions.Broadcast(data); _sessions.Broadcast (data);
} }
/// <summary> /// <summary>
/// Broadcasts the specified <see cref="string"/> to the clients of every <see cref="WebSocketService"/> instances /// Broadcasts the specified <see cref="string"/> to the clients of
/// in the <see cref="WebSocketService.Sessions"/>. /// every <see cref="WebSocketService"/> instances in the <see cref="WebSocketService.Sessions"/>.
/// </summary> /// </summary>
/// <param name="data"> /// <param name="data">
/// A <see cref="string"/> to broadcast. /// A <see cref="string"/> to broadcast.
/// </param> /// </param>
public void Broadcast(string data) public void Broadcast (string data)
{ {
if (IsBound) if (IsBound)
_sessions.Broadcast(data); _sessions.Broadcast (data);
} }
/// <summary> /// <summary>
/// Pings to the clients of every <see cref="WebSocketService"/> instances /// Sends Pings to the clients of every <see cref="WebSocketService"/> instances
/// in the <see cref="WebSocketService.Sessions"/>. /// in the <see cref="WebSocketService.Sessions"/>.
/// </summary> /// </summary>
/// <returns> /// <returns>
/// A Dictionary&lt;string, bool&gt; that contains the collection of IDs and values /// A Dictionary&lt;string, bool&gt; that contains the collection of IDs and values
/// indicating whether each <see cref="WebSocketService"/> instances received a Pong in a time. /// indicating whether the each <see cref="WebSocketService"/> instances received a Pong in a time.
/// </returns> /// </returns>
public Dictionary<string, bool> Broadping() public Dictionary<string, bool> Broadping ()
{ {
return Broadping(String.Empty); return Broadping (String.Empty);
} }
/// <summary> /// <summary>
/// Pings with the specified <see cref="string"/> to the clients of every <see cref="WebSocketService"/> instances /// Sends Pings with the specified <see cref="string"/> to the clients of
/// in the <see cref="WebSocketService.Sessions"/>. /// every <see cref="WebSocketService"/> instances in the <see cref="WebSocketService.Sessions"/>.
/// </summary> /// </summary>
/// <returns> /// <returns>
/// A Dictionary&lt;string, bool&gt; that contains the collection of IDs and values /// A Dictionary&lt;string, bool&gt; that contains the collection of IDs and values
/// indicating whether each <see cref="WebSocketService"/> instances received a Pong in a time. /// indicating whether the each <see cref="WebSocketService"/> instances received a Pong in a time.
/// </returns> /// </returns>
/// <param name="message"> /// <param name="message">
/// A <see cref="string"/> that contains a message. /// A <see cref="string"/> that contains a message to send.
/// </param> /// </param>
public Dictionary<string, bool> Broadping(string message) public Dictionary<string, bool> Broadping (string message)
{ {
return IsBound return IsBound
? _sessions.Broadping(message) ? _sessions.Broadping (message)
: null; : null;
} }
/// <summary> /// <summary>
/// Pings to the client of the <see cref="WebSocketService"/> instance. /// Sends a Ping to the client of the current <see cref="WebSocketService"/> instance.
/// </summary> /// </summary>
/// <returns> /// <returns>
/// <c>true</c> if the <see cref="WebSocketService"/> instance receives a Pong in a time; otherwise, <c>false</c>. /// <c>true</c> if the current <see cref="WebSocketService"/> instance receives a Pong in a time;
/// otherwise, <c>false</c>.
/// </returns> /// </returns>
public bool Ping() public bool Ping ()
{ {
return Ping(String.Empty); return Ping (String.Empty);
} }
/// <summary> /// <summary>
/// Pings with the specified <see cref="string"/> to the client of the <see cref="WebSocketService"/> instance. /// Sends a Ping with the specified <see cref="string"/> to the client of
/// the current <see cref="WebSocketService"/> instance.
/// </summary> /// </summary>
/// <returns> /// <returns>
/// <c>true</c> if the <see cref="WebSocketService"/> instance receives a Pong in a time; otherwise, <c>false</c>. /// <c>true</c> if the current <see cref="WebSocketService"/> instance receives a Pong in a time;
/// otherwise, <c>false</c>.
/// </returns> /// </returns>
/// <param name="message"> /// <param name="message">
/// A <see cref="string"/> that contains a message. /// A <see cref="string"/> that contains a message to send.
/// </param> /// </param>
public bool Ping(string message) public bool Ping (string message)
{ {
return IsBound return IsBound
? _websocket.Ping(message) ? _websocket.Ping (message)
: false; : false;
} }
/// <summary> /// <summary>
/// Pings to the client of the <see cref="WebSocketService"/> instance /// Sends a Ping to the client of the <see cref="WebSocketService"/> instance
/// associated with the specified <paramref name="id"/>. /// associated with the specified <paramref name="id"/>.
/// </summary> /// </summary>
/// <returns> /// <returns>
/// <c>true</c> if the <see cref="WebSocketService"/> instance receives a Pong in a time; otherwise, <c>false</c>. /// <c>true</c> if the <see cref="WebSocketService"/> instance receives a Pong in a time;
/// otherwise, <c>false</c>.
/// </returns> /// </returns>
/// <param name="id"> /// <param name="id">
/// A <see cref="string"/> that contains an ID that represents the destination for the Ping. /// A <see cref="string"/> that contains an ID that represents the destination for the Ping.
/// </param> /// </param>
public bool PingTo(string id) public bool PingTo (string id)
{ {
return PingTo(id, String.Empty); return PingTo (id, String.Empty);
} }
/// <summary> /// <summary>
/// Pings with the specified <see cref="string"/> to the client of the <see cref="WebSocketService"/> instance /// Sends a Ping with the specified <see cref="string"/> to the client of
/// associated with the specified <paramref name="id"/>. /// the <see cref="WebSocketService"/> instance associated with the specified <paramref name="id"/>.
/// </summary> /// </summary>
/// <returns> /// <returns>
/// <c>true</c> if the <see cref="WebSocketService"/> instance receives a Pong in a time; otherwise, <c>false</c>. /// <c>true</c> if the <see cref="WebSocketService"/> instance receives a Pong in a time;
/// otherwise, <c>false</c>.
/// </returns> /// </returns>
/// <param name="id"> /// <param name="id">
/// A <see cref="string"/> that contains an ID that represents the destination for the Ping. /// A <see cref="string"/> that contains an ID that represents the destination for the Ping.
/// </param> /// </param>
/// <param name="message"> /// <param name="message">
/// A <see cref="string"/> that contains a message. /// A <see cref="string"/> that contains a message to send.
/// </param> /// </param>
public bool PingTo(string id, string message) public bool PingTo (string id, string message)
{ {
if (!IsBound) if (!IsBound)
return false; return false;
WebSocketService service; WebSocketService service;
return _sessions.TryGetWebSocketService(id, out service) return _sessions.TryGetWebSocketService (id, out service)
? service.Ping(message) ? service.Ping (message)
: false; : false;
} }
/// <summary> /// <summary>
/// Sends a binary data to the client of the <see cref="WebSocketService"/> instance. /// Sends a binary data to the client of the current <see cref="WebSocketService"/> instance.
/// </summary> /// </summary>
/// <param name="data"> /// <param name="data">
/// An array of <see cref="byte"/> that contains a binary data to send. /// An array of <see cref="byte"/> that contains a binary data to send.
/// </param> /// </param>
public void Send(byte[] data) public void Send (byte [] data)
{ {
if (IsBound) if (IsBound)
_websocket.Send(data); _websocket.Send (data);
} }
/// <summary> /// <summary>
/// Sends a text data to the client of the <see cref="WebSocketService"/> instance. /// Sends a text data to the client of the current <see cref="WebSocketService"/> instance.
/// </summary> /// </summary>
/// <param name="data"> /// <param name="data">
/// A <see cref="string"/> that contains a text data to send. /// A <see cref="string"/> that contains a text data to send.
/// </param> /// </param>
public void Send(string data) public void Send (string data)
{ {
if (IsBound) if (IsBound)
_websocket.Send(data); _websocket.Send (data);
} }
/// <summary> /// <summary>
@ -431,14 +446,14 @@ namespace WebSocketSharp.Server {
/// <param name="data"> /// <param name="data">
/// An array of <see cref="byte"/> that contains a binary data to send. /// An array of <see cref="byte"/> that contains a binary data to send.
/// </param> /// </param>
public void SendTo(string id, byte[] data) public void SendTo (string id, byte [] data)
{ {
if (!IsBound) if (!IsBound)
return; return;
WebSocketService service; WebSocketService service;
if (_sessions.TryGetWebSocketService(id, out service)) if (_sessions.TryGetWebSocketService (id, out service))
service.Send(data); service.Send (data);
} }
/// <summary> /// <summary>
@ -451,65 +466,68 @@ namespace WebSocketSharp.Server {
/// <param name="data"> /// <param name="data">
/// A <see cref="string"/> that contains a text data to send. /// A <see cref="string"/> that contains a text data to send.
/// </param> /// </param>
public void SendTo(string id, string data) public void SendTo (string id, string data)
{ {
if (!IsBound) if (!IsBound)
return; return;
WebSocketService service; WebSocketService service;
if (_sessions.TryGetWebSocketService(id, out service)) if (_sessions.TryGetWebSocketService (id, out service))
service.Send(data); service.Send (data);
} }
/// <summary> /// <summary>
/// Starts the <see cref="WebSocketService"/> instance. /// Starts a <see cref="WebSocketService"/> instance.
/// </summary> /// </summary>
public void Start() public void Start ()
{ {
if (IsBound) if (IsBound)
_websocket.Connect(); _websocket.Connect ();
} }
/// <summary> /// <summary>
/// Stops the <see cref="WebSocketService"/> instance. /// Stops the current <see cref="WebSocketService"/> instance.
/// </summary> /// </summary>
public void Stop() public void Stop ()
{ {
if (!IsBound) if (!IsBound)
return; return;
_websocket.Close(); _websocket.Close ();
} }
/// <summary> /// <summary>
/// Stops the <see cref="WebSocketService"/> instance with the specified <see cref="ushort"/> and <see cref="string"/>. /// Stops the current <see cref="WebSocketService"/> instance with the specified
/// <see cref="ushort"/> and <see cref="string"/>.
/// </summary> /// </summary>
/// <param name="code"> /// <param name="code">
/// A <see cref="ushort"/> that contains a status code indicating the reason for stop. /// A <see cref="ushort"/> that contains a status code indicating the reason for stop.
/// </param> /// </param>
/// <param name="reason"> /// <param name="reason">
/// A <see cref="string"/> that contains a reason for stop. /// A <see cref="string"/> that contains the reason for stop.
/// </param> /// </param>
public void Stop(ushort code, string reason) public void Stop (ushort code, string reason)
{ {
if (!IsBound) if (!IsBound)
return; return;
_websocket.Close(code, reason); _websocket.Close (code, reason);
} }
/// <summary> /// <summary>
/// Stops the <see cref="WebSocketService"/> instance with the specified <see cref="CloseStatusCode"/> and <see cref="string"/>. /// Stops the current <see cref="WebSocketService"/> instance with the specified
/// <see cref="CloseStatusCode"/> and <see cref="string"/>.
/// </summary> /// </summary>
/// <param name="code"> /// <param name="code">
/// One of the <see cref="CloseStatusCode"/> values that contains a status code indicating the reason for stop. /// One of the <see cref="CloseStatusCode"/> values that indicates a status code
/// indicating the reason for stop.
/// </param> /// </param>
/// <param name="reason"> /// <param name="reason">
/// A <see cref="string"/> that contains a reason for stop. /// A <see cref="string"/> that contains the reason for stop.
/// </param> /// </param>
public void Stop(CloseStatusCode code, string reason) public void Stop (CloseStatusCode code, string reason)
{ {
Stop((ushort)code, reason); Stop ((ushort) code, reason);
} }
#endregion #endregion

23
websocket-sharp/WebSocket.cs
View File

@ -71,9 +71,11 @@ namespace WebSocketSharp
_certValidationCallback; _certValidationCallback;
private bool _client; private bool _client;
private Action _closeContext; private Action _closeContext;
private CookieCollection _cookies;
private CompressionMethod _compression; private CompressionMethod _compression;
private WebSocketContext _context; private WebSocketContext _context;
private CookieCollection _cookies;
private Func<CookieCollection, CookieCollection, bool>
_cookiesValidation;
private WsCredential _credentials; private WsCredential _credentials;
private string _extensions; private string _extensions;
private AutoResetEvent _exitReceiving; private AutoResetEvent _exitReceiving;
@ -222,9 +224,13 @@ namespace WebSocketSharp
#region Internal Properties #region Internal Properties
internal CookieCollection CookieCollection { internal Func<CookieCollection, CookieCollection, bool> CookiesValidation {
get { get {
return _cookies; return _cookiesValidation;
}
set {
_cookiesValidation = value;
} }
} }
@ -1264,7 +1270,8 @@ namespace WebSocketSharp
return context.IsWebSocketRequest && return context.IsWebSocketRequest &&
validateHostHeader (context.Host) && validateHostHeader (context.Host) &&
!context.SecWebSocketKey.IsNullOrEmpty () && !context.SecWebSocketKey.IsNullOrEmpty () &&
((version = context.SecWebSocketVersion) != null && version == _version); ((version = context.SecWebSocketVersion) != null && version == _version) &&
validateCookies (context.CookieCollection, _cookies);
} }
// As client // As client
@ -1276,6 +1283,14 @@ namespace WebSocketSharp
((version = response.Headers ["Sec-WebSocket-Version"]) == null || version == _version); ((version = response.Headers ["Sec-WebSocket-Version"]) == null || version == _version);
} }
// As server
private bool validateCookies (CookieCollection request, CookieCollection response)
{
return _cookiesValidation != null
? _cookiesValidation (request, response)
: true;
}
// As server // As server
private bool validateHostHeader (string value) private bool validateHostHeader (string value)
{ {