sockscape/server/Libraries/Kneesocks/Handshake.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using Square;

namespace Kneesocks {
    public class Handshake {
        private const string HttpVersion = "1.1";
        public bool IsRequest = false;

        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; } = null;
        protected string StatusCodeText {
            get => 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) {
            IsRequest = true;

            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("GET ") || !header.Contains("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("GET ")) {
                    parts = line.Trim().Split(' ');
                    if(parts.Length < 3)
                        throw new FormatException("Status line in header malformed");
                } 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();

            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.")
            => new Handshake(statusCode, message);

        public byte[] ToBytes() => ToString().GetBytes();

        public override string ToString() {
            if(IsRequest)
                throw new NotSupportedException("Cannot create a request string.");

            if(Content != null) {
                SetHeader("Content-Length", Content.ByteLength().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 bool HasHeader(string name) => Headers.ContainsKey(name);

        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 Handshake ClearContent() {
            Content = null;
            return this;
        }
    }
}