#region MIT License
|
/**
|
* WebSocketServer.cs
|
*
|
* A C# implementation of the WebSocket protocol server.
|
*
|
* The MIT License
|
*
|
* Copyright (c) 2012 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
|
|
using System;
|
using System.Collections.Generic;
|
using System.Net.Sockets;
|
using WebSocketSharp.Net;
|
|
namespace WebSocketSharp.Server
|
{
|
|
public class WebSocketServer : WebSocketServerBase
|
{
|
#region Field
|
|
private ServiceManager _services;
|
|
#endregion
|
|
#region Public Constructors
|
|
public WebSocketServer()
|
: this(80)
|
{
|
}
|
|
public WebSocketServer(int port)
|
: this(System.Net.IPAddress.Any, port)
|
{
|
}
|
|
public WebSocketServer(string url)
|
: base(url)
|
{
|
if (BaseUri.AbsolutePath != "/")
|
{
|
var msg = "Must not contain the path component: " + url;
|
throw new ArgumentException(msg, "url");
|
}
|
|
init();
|
}
|
|
public WebSocketServer(int port, bool secure)
|
: this(System.Net.IPAddress.Any, port, secure)
|
{
|
}
|
|
public WebSocketServer(System.Net.IPAddress address, int port)
|
: this(address, port, port == 443 ? true : false)
|
{
|
}
|
|
public WebSocketServer(System.Net.IPAddress address, int port, bool secure)
|
: base(address, port, "/", secure)
|
{
|
init();
|
}
|
|
#endregion
|
|
#region Property
|
|
public IEnumerable<string> ServicePath
|
{
|
get
|
{
|
var url = BaseUri.IsAbsoluteUri
|
? BaseUri.ToString().TrimEnd('/')
|
: String.Empty;
|
foreach (var path in _services.Path)
|
yield return url + path;
|
}
|
}
|
|
public bool Sweeped
|
{
|
get
|
{
|
return _services.Sweeped;
|
}
|
|
set
|
{
|
_services.Sweeped = value;
|
}
|
}
|
|
#endregion
|
|
#region Private Method
|
|
private void init()
|
{
|
_services = new ServiceManager();
|
}
|
|
#endregion
|
|
#region Protected Method
|
|
protected override void AcceptWebSocket(TcpClient client)
|
{
|
var context = Ext.AcceptWebSocket(client, IsSecure);
|
var socket = context.WebSocket;
|
var path = Ext.UrlDecode(context.Path);
|
|
IServiceHost svcHost;
|
if (!_services.TryGetServiceHost(path, out svcHost))
|
{
|
socket.Close(HttpStatusCode.NotImplemented);
|
return;
|
}
|
|
if (BaseUri.IsAbsoluteUri)
|
socket.Url = new Uri(BaseUri, path);
|
|
svcHost.BindWebSocket(socket);
|
}
|
|
#endregion
|
|
#region Public Methods
|
|
public void AddService<T>(string absPath)
|
where T : WebSocketService, new()
|
{
|
string msg;
|
if (!Ext.IsValidAbsolutePath(absPath, out msg))
|
{
|
Error(msg);
|
return;
|
}
|
|
var svcHost = new WebSocketServiceHost<T>();
|
svcHost.Uri = BaseUri.IsAbsoluteUri
|
? new Uri(BaseUri, absPath)
|
: Ext.ToUri(absPath);
|
if (!Sweeped)
|
svcHost.Sweeped = Sweeped;
|
|
_services.Add(absPath, svcHost);
|
}
|
|
public void Broadcast(string data)
|
{
|
_services.Broadcast(data);
|
}
|
|
public override void Stop()
|
{
|
base.Stop();
|
_services.Stop();
|
}
|
|
#endregion
|
}
|
}
|
