public static async Task Run(string[] args)
{
var settings = new WebListenerSettings();
settings.UrlPrefixes.Add("http://localhost:8080");
using (WebListener listener = new WebListener(settings))
{
listener.Start();
Console.WriteLine("Running...");
while (true)
{
RequestContext context = await listener.AcceptAsync();
Console.WriteLine("Accepted");
// Context:
// context.User;
// context.DisconnectToken
// context.Dispose()
// context.Abort();
// Request
// context.Request.ProtocolVersion
// context.Request.Headers
// context.Request.Method
// context.Request.Body
// Content-Length - long?
// Content-Type - string
// IsSecureConnection
// HasEntityBody
// Response
byte[] bytes = Encoding.ASCII.GetBytes("Hello World: " + DateTime.Now);
if (context.IsWebSocketRequest)
{
Console.WriteLine("WebSocket");
WebSocket webSocket = await context.AcceptWebSocketAsync();
await webSocket.SendAsync(new ArraySegment <byte>(bytes, 0, bytes.Length), WebSocketMessageType.Text, true, CancellationToken.None);
await webSocket.CloseAsync(WebSocketCloseStatus.NormalClosure, "Goodbye", CancellationToken.None);
webSocket.Dispose();
}
else
{
Console.WriteLine("Hello World");
context.Response.ContentLength = bytes.Length;
context.Response.ContentType = "text/plain";
context.Response.Body.Write(bytes, 0, bytes.Length);
context.Dispose();
}
}
}
}
private async Task WebSocketUpgrade(string subProtocol, int receiveBufferSize, TimeSpan keepAliveInterval, ArraySegment <byte>?internalBuffer)
{
var context = await _context.AcceptWebSocketAsync(subProtocol, receiveBufferSize, keepAliveInterval);
var wrapper = new OwinWebSocketWrapper(context, _environment.Get <CancellationToken>(Constants.CallCancelledKey));
await _webSocketFunc(wrapper.Environment);
await wrapper.CleanupAsync();
}
Task <WebSocket> IHttpWebSocketFeature.AcceptAsync(WebSocketAcceptContext context)
{
// TODO: Advanced params
string subProtocol = null;
if (context != null)
{
subProtocol = context.SubProtocol;
}
return(_requestContext.AcceptWebSocketAsync(subProtocol));
}
private async Task handleClient <TWebSocketBehavior>(RequestContext requestContext, Func <TWebSocketBehavior> behaviorBuilder, CancellationToken token)
where TWebSocketBehavior : WebListenerWebSocketServerBehavior
{
Guid connectionId;
WebSocket webSocket = null;
WebListenerWebSocketServerBehavior behavior = null;
try
{
int statusCode = 500;
var statusDescription = "BadContext";
behavior = behaviorBuilder();
if (!behavior.OnValidateContext(requestContext, ref statusCode, ref statusDescription))
{
requestContext.Response.ReasonPhrase = statusDescription;
requestContext.Response.StatusCode = statusCode;
requestContext.Abort();
_logError($"Failed to validate client context. Closing connection. Status: {statusCode}. Description: {statusDescription}.");
return;
}
connectionId = Guid.NewGuid();
webSocket = await requestContext.AcceptWebSocketAsync(null, _keepAlivePingInterval);
bool clientAdded = _clients.TryAdd(connectionId, webSocket);
if (!clientAdded)
{
throw new ArgumentException($"Attempted to add a new web socket connection to server for connection id '{connectionId}' that already exists.");
}
Interlocked.Increment(ref _connectedClientCount);
_logInfo($"Connection id '{connectionId}' accepted; there are now {_connectedClientCount} total clients.");
var safeconnected = MakeSafe <Guid, RequestContext>(behavior.OnConnectionEstablished, "behavior.OnClientConnected");
safeconnected(connectionId, requestContext);
}
catch (Exception e)
{
_logError($"Client handler exception: {e}");
requestContext.Response.StatusCode = 500;
requestContext.Abort();
requestContext.Dispose();
return;
}
var stringBehavior = MakeSafe <StringMessageReceivedEventArgs>(behavior.OnStringMessage, "behavior.OnStringMessage");
var binaryBehavior = MakeSafe <BinaryMessageReceivedEventArgs>(behavior.OnBinaryMessage, "behavior.OnBinaryMessage");
bool single = false;
var closeBehavior = MakeSafe <WebSocketReceivedResultEventArgs>((r) =>
{
if (!single)
{
behavior.OnClose(new WebSocketClosedEventArgs(connectionId, r));
}
single = true;
}, "behavior.OnClose");
try
{
using (webSocket)
{
await webSocket.ProcessIncomingMessages(_messageQueue, connectionId, stringBehavior, binaryBehavior, closeBehavior, _logInfo, token);
}
}
finally
{
Interlocked.Decrement(ref _connectedClientCount);
_logInfo($"Connection id '{connectionId}' disconnected; there are now {_connectedClientCount} total clients.");
webSocket?.CleanupSendMutex();
requestContext.Dispose();
bool clientRemoved = _clients.TryRemove(connectionId, out webSocket);
if (clientRemoved)
{
closeBehavior(new WebSocketReceivedResultEventArgs(closeStatus: WebSocketCloseStatus.EndpointUnavailable, closeStatDesc: "Removing Client Due to other error"));
}
else
{
_logError($"Attempted to remove an existing web socket connection to server for connection id '{connectionId}' that no longer exists.");
}
_logInfo($"Completed HandleClient task for connection id '{connectionId}'.");
}
}
public static async Task Run(string[] args)
{
using (WebListener listener = new WebListener())
{
listener.UrlPrefixes.Add(UrlPrefix.Create("http://localhost:8080"));
listener.Start();
Console.WriteLine("Running...");
while (true)
{
RequestContext context = await listener.GetContextAsync();
Console.WriteLine("Accepted");
// Context:
// context.User;
// context.DisconnectToken
// context.Dispose()
// context.Abort();
// Request
// context.Request.ProtocolVersion
// context.Request.Headers
// context.Request.Method
// context.Request.Body
// Content-Length - long?
// Content-Type - string
// IsSecureConnection
// HasEntityBody
// TODO: Request fields
// Content-Encoding - Encoding
// Host
// Client certs - GetCertAsync, CertErrors
// Cookies
// KeepAlive
// QueryString (parsed)
// RequestTraceIdentifier
// RawUrl
// URI
// IsWebSocketRequest
// LocalEndpoint vs LocalIP & LocalPort
// RemoteEndpoint vs RemoteIP & RemotePort
// AcceptTypes string[]
// ServiceName
// TransportContext
// Response
byte[] bytes = Encoding.ASCII.GetBytes("Hello World: " + DateTime.Now);
if (context.IsWebSocketRequest())
{
Console.WriteLine("WebSocket");
WebSocket webSocket = await context.AcceptWebSocketAsync();
await webSocket.SendAsync(new ArraySegment <byte>(bytes, 0, bytes.Length), WebSocketMessageType.Text, true, CancellationToken.None);
await webSocket.CloseAsync(WebSocketCloseStatus.NormalClosure, "Goodbye", CancellationToken.None);
webSocket.Dispose();
}
else
{
Console.WriteLine("Hello World");
context.Response.ContentLength = bytes.Length;
context.Response.ContentType = "text/plain";
context.Response.Body.Write(bytes, 0, bytes.Length);
context.Dispose();
}
}
}
}
