look at entrypoint.cs if you want to die
imagine a world
This commit is contained in:
parent
017b75579f
commit
9ae62a663c
5 changed files with 196 additions and 5 deletions
|
@ -8,5 +8,9 @@ using System.Threading.Tasks;
|
||||||
namespace CircleScape {
|
namespace CircleScape {
|
||||||
class ActiveConnection : Websocket.Connection {
|
class ActiveConnection : Websocket.Connection {
|
||||||
public ActiveConnection(TcpClient sock) : base(sock) { }
|
public ActiveConnection(TcpClient sock) : base(sock) { }
|
||||||
|
|
||||||
|
public ActiveConnection(PendingConnection conn) : base(conn) {
|
||||||
|
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,8 +1,24 @@
|
||||||
using System;
|
/***************************************/using
|
||||||
using System.Collections.Generic;
|
System ;using
|
||||||
using System.Linq;
|
/***************************************/
|
||||||
using System.Text;
|
System.
|
||||||
using System.Threading.Tasks;
|
Collections .
|
||||||
|
Generic ;using
|
||||||
|
/***************************************/
|
||||||
|
System.
|
||||||
|
Linq ;using
|
||||||
|
/***************************************/
|
||||||
|
System.
|
||||||
|
Text ;using
|
||||||
|
/***************************************/
|
||||||
|
System.
|
||||||
|
Threading .
|
||||||
|
Tasks ;using
|
||||||
|
/***************************************/
|
||||||
|
System.
|
||||||
|
Data .
|
||||||
|
SQLite ;
|
||||||
|
/***************************************/
|
||||||
|
|
||||||
namespace CircleScape {
|
namespace CircleScape {
|
||||||
class Entrypoint {
|
class Entrypoint {
|
||||||
|
|
|
@ -4,6 +4,7 @@ using System.Linq;
|
||||||
using System.Text;
|
using System.Text;
|
||||||
using System.Threading.Tasks;
|
using System.Threading.Tasks;
|
||||||
using CircleScape.Websocket;
|
using CircleScape.Websocket;
|
||||||
|
using System.Net.Sockets;
|
||||||
|
|
||||||
namespace CircleScape {
|
namespace CircleScape {
|
||||||
static class PoolManager {
|
static class PoolManager {
|
||||||
|
@ -20,6 +21,11 @@ namespace CircleScape {
|
||||||
};
|
};
|
||||||
|
|
||||||
ActiveConnectionsPool = new Pool<ActiveConnection>();
|
ActiveConnectionsPool = new Pool<ActiveConnection>();
|
||||||
|
|
||||||
|
var test = new PendingConnection(new TcpClient());
|
||||||
|
var teste = test as ActiveConnection;
|
||||||
|
|
||||||
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -24,9 +24,34 @@ namespace CircleScape.Websocket {
|
||||||
Stream = sock.GetStream();
|
Stream = sock.GetStream();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
public Connection(Connection conn) {
|
||||||
|
Socket = conn.Socket;
|
||||||
|
Stream = Socket.GetStream();
|
||||||
|
|
||||||
|
Disconnected = conn.Disconnected;
|
||||||
|
DisconnectReason = conn.DisconnectReason;
|
||||||
|
|
||||||
|
Handshaked = conn.Handshaked;
|
||||||
|
RawClientHandshake = conn.RawClientHandshake;
|
||||||
|
Headers = conn.Headers;
|
||||||
|
}
|
||||||
|
|
||||||
public void Disconnect(string reason = null) {
|
public void Disconnect(string reason = null) {
|
||||||
Disconnected = true;
|
Disconnected = true;
|
||||||
DisconnectReason = reason;
|
DisconnectReason = reason;
|
||||||
|
|
||||||
|
if(Socket.Connected) {
|
||||||
|
Socket.SendTimeout = 1000;
|
||||||
|
if(!Handshaked) {
|
||||||
|
var raw = Encoding.ASCII.GetBytes(Handshake.DenyRequest().GetRaw());
|
||||||
|
Stream.Write(raw, 0, raw.Length);
|
||||||
|
Socket.Close();
|
||||||
|
} else {
|
||||||
|
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
OnClose();
|
||||||
}
|
}
|
||||||
|
|
||||||
// called after the client successfully handshakes
|
// called after the client successfully handshakes
|
||||||
|
|
140
server/Websocket/Handshake.cs
Normal file
140
server/Websocket/Handshake.cs
Normal file
|
@ -0,0 +1,140 @@
|
||||||
|
using System;
|
||||||
|
using System.Collections.Generic;
|
||||||
|
using System.Linq;
|
||||||
|
using System.Text;
|
||||||
|
using System.Threading.Tasks;
|
||||||
|
|
||||||
|
namespace CircleScape.Websocket {
|
||||||
|
class Handshake {
|
||||||
|
private const string HttpVersion = "1.1";
|
||||||
|
|
||||||
|
public enum kStatusCode {
|
||||||
|
Switching_Protocols = 101,
|
||||||
|
|
||||||
|
Bad_Request = 400,
|
||||||
|
Unauthorized = 401,
|
||||||
|
Forbidden = 403,
|
||||||
|
Not_Found = 404,
|
||||||
|
Method_Not_Allowed = 405,
|
||||||
|
Not_Acceptable = 406,
|
||||||
|
Request_Timeout = 408,
|
||||||
|
Conflict = 409,
|
||||||
|
Gone = 410,
|
||||||
|
|
||||||
|
Internal_Server_Error = 500,
|
||||||
|
Not_Implemented = 501,
|
||||||
|
Bad_Gateway = 502,
|
||||||
|
Service_Unavailable = 503,
|
||||||
|
Gateway_Timeout = 504
|
||||||
|
}
|
||||||
|
|
||||||
|
public kStatusCode StatusCode { get; private set; } = kStatusCode.Switching_Protocols;
|
||||||
|
protected string StatusCodeText {
|
||||||
|
get {
|
||||||
|
return Enum.GetName(typeof(kStatusCode), StatusCode).Replace('_', ' ');
|
||||||
|
}
|
||||||
|
}
|
||||||
|
private Dictionary<string, string> Headers =
|
||||||
|
new Dictionary<string, string>(StringComparer.OrdinalIgnoreCase);
|
||||||
|
public string Content { get; set; } = null;
|
||||||
|
|
||||||
|
public Handshake(string rawData) {
|
||||||
|
var headerLength = rawData.IndexOf("\r\n\r\n");
|
||||||
|
if(headerLength == -1)
|
||||||
|
throw new FormatException("Header delimeter not found in raw data");
|
||||||
|
|
||||||
|
var header = rawData.Substring(0, headerLength);
|
||||||
|
if(!header.StartsWith("HTTP/"))
|
||||||
|
throw new FormatException("Protocol defined in status line not understood");
|
||||||
|
|
||||||
|
var lines = header.Split('\n');
|
||||||
|
foreach(var line in lines) {
|
||||||
|
string[] parts;
|
||||||
|
if(line.StartsWith("HTTP/")) {
|
||||||
|
parts = line.Trim().Split(' ');
|
||||||
|
if(parts.Length < 2)
|
||||||
|
throw new FormatException("Status line in header malformed");
|
||||||
|
|
||||||
|
int code;
|
||||||
|
if(!int.TryParse(parts[1], out code))
|
||||||
|
throw new FormatException("Status code sent is not a number");
|
||||||
|
|
||||||
|
if(!Enum.IsDefined(typeof(kStatusCode), code))
|
||||||
|
throw new NotSupportedException("Status code not supported");
|
||||||
|
|
||||||
|
StatusCode = (kStatusCode)code;
|
||||||
|
} else {
|
||||||
|
parts = line.Trim().Split(new char[] {':'}, 2);
|
||||||
|
if(parts.Length == 2)
|
||||||
|
Headers.Add(parts[0].Trim(), parts[1].Trim());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if(Headers.ContainsKey("Content-Length")) {
|
||||||
|
rawData.Substring(headerLength + 4, int.Parse(Headers["Content-Length"]));
|
||||||
|
} else {
|
||||||
|
if(rawData.Length > headerLength + 4)
|
||||||
|
Content = rawData.Substring(headerLength + 4);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public Handshake(kStatusCode statusCode = kStatusCode.Switching_Protocols, string content = null) {
|
||||||
|
StatusCode = statusCode;
|
||||||
|
Content = content;
|
||||||
|
}
|
||||||
|
|
||||||
|
public static Handshake AcceptRequest(Handshake request) {
|
||||||
|
const string nonce = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
|
||||||
|
var key = request.GetHeader("Sec-WebSocket-Key");
|
||||||
|
var connectionHash = (key + nonce).SHA1().Base64Encode(false);
|
||||||
|
|
||||||
|
var shake = new Handshake(kStatusCode.Switching_Protocols);
|
||||||
|
shake.SetHeader("Upgrade", "websocket")
|
||||||
|
.SetHeader("Connection", "Upgrade")
|
||||||
|
.SetHeader("Sec-WebSocket-Accept", connectionHash);
|
||||||
|
return shake;
|
||||||
|
}
|
||||||
|
|
||||||
|
public static Handshake DenyRequest(kStatusCode statusCode = kStatusCode.Bad_Request, string message = "Handshake failed.") {
|
||||||
|
return new Handshake(statusCode, message);
|
||||||
|
}
|
||||||
|
|
||||||
|
public string GetRaw() {
|
||||||
|
if(Content != null) {
|
||||||
|
SetHeader("Content-Length", Encoding.ASCII.GetByteCount(Content).ToString());
|
||||||
|
if(GetHeader("Content-Type") == null)
|
||||||
|
SetHeader("Content-Type", "text/plain");
|
||||||
|
}
|
||||||
|
|
||||||
|
var raw = "HTTP/"+ HttpVersion +" "+ (int)StatusCode + " "+ StatusCodeText +"\r\n";
|
||||||
|
foreach(var header in Headers)
|
||||||
|
raw += header.Key.Trim() + ": " + header.Value.Trim() + "\r\n";
|
||||||
|
return raw += "\r\n";
|
||||||
|
}
|
||||||
|
|
||||||
|
public string GetHeader(string name) {
|
||||||
|
if(Headers.ContainsKey(name))
|
||||||
|
return Headers[name];
|
||||||
|
else return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
public Handshake SetHeader(string name, string value) {
|
||||||
|
if(Headers.ContainsKey(name))
|
||||||
|
Headers[name] = value;
|
||||||
|
else
|
||||||
|
Headers.Add(name, value);
|
||||||
|
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
|
||||||
|
public Handshake AppendContent(string content) {
|
||||||
|
Content += content;
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void ClearContent() {
|
||||||
|
Content = null;
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in a new issue