/Users/jlawson/Projects/Alchemy-Websockets/src/Alchemy/Handlers/WebSocket/hybi00/Handshakes.cs

Go to the documentation of this file.

using System;
using System.Collections.Generic;
using System.Linq;
using System.Web;
using Alchemy.Classes;

namespace Alchemy.Handlers.WebSocket.hybi00
{
    public class ClientHandshake
    {
        private const String Handshake =
            "GET {0} HTTP/1.1\r\n" +
            "Upgrade: WebSocket\r\n" +
            "Connection: Upgrade\r\n" +
            "Origin: {1}\r\n" +
            "Host: {2}\r\n" +
            "Sec-Websocket-Key1: {3}\r\n" +
            "Sec-Websocket-Key2: {4}\r\n" +
            "{5}";

        public string Host = String.Empty;
        public string Key1 = String.Empty;
        public string Key2 = String.Empty;
        public string Origin = String.Empty;
        public string ResourcePath = String.Empty;

        public ClientHandshake()
        {}

        public ClientHandshake(ArraySegment<byte> challengeBytes, Header header)
        {
            ChallengeBytes = challengeBytes;
            ResourcePath = header.RequestPath;
            Key1 = header["sec-websocket-key1"];
            Key2 = header["sec-websocket-key2"];
            SubProtocol = header["sec-websocket-protocol"];
            Origin = header["origin"];
            Host = header["host"];
            Cookies = header.Cookies;
        }

        public ArraySegment<byte> ChallengeBytes { get; set; }
        public HttpCookieCollection Cookies { get; set; }
        public string SubProtocol { get; set; }
        public Dictionary<string, string> AdditionalFields { get; set; }

        public bool IsValid()
        {
            return (
                       (Host != null) &&
                       (Key1 != null) &&
                       (Key2 != null) &&
                       (Origin != null) &&
                       (ResourcePath != null)
                   );
        }

        public override string ToString()
        {
            string additionalFields = String.Empty;

            if (Cookies != null)
            {
                additionalFields += "Cookie: " + Cookies + "\r\n";
            }
            if (SubProtocol != null)
            {
                additionalFields += "Sec-Websocket-Protocol: " + SubProtocol + "\r\n";
            }

            if (additionalFields != String.Empty)
            {
                additionalFields = AdditionalFields.Aggregate(additionalFields,
                                                              (current, field) =>
                                                              current + (field.Key + ": " + field.Value + "\r\n"));
            }
            additionalFields += "\r\n";

            return String.Format(Handshake, ResourcePath, Origin, Host, Key1, Key2, additionalFields);
        }
    }

    public class ServerHandshake
    {
        private const string Handshake =
            "HTTP/1.1 101 Web Socket Protocol Handshake\r\n" +
            "Upgrade: WebSocket\r\n" +
            "Connection: Upgrade\r\n" +
            "Sec-WebSocket-Origin: {0}\r\n" +
            "Sec-WebSocket-Location: {1}\r\n" +
            "{2}" +
            "                "; //Empty space for challenge answer

        public string Location = String.Empty;
        public string Origin = String.Empty;
        public byte[] AnswerBytes { get; set; }
        public string SubProtocol { get; set; }
        public Dictionary<string, string> AdditionalFields { get; set; }

        public override string ToString()
        {
            string additionalFields = String.Empty;
            if (SubProtocol != null)
            {
                additionalFields += "Sec-WebSocket-Protocol: " + SubProtocol + "\r\n";
            }
            additionalFields += "\r\n";

            return String.Format(Handshake, Origin, Location, additionalFields);
        }
    }
}