esiur-dotnet/Esiur/Net/HTTP/HTTPConnection.cs

using System;
using System.Diagnostics;
using System.IO;
using System.Net.Sockets;
using System.Text;
using System.Threading;
using System.Net;
using System.Collections;
using System.Collections.Generic;
using Esiur.Net.Sockets;
using Esiur.Data;
using Esiur.Net.Packets;
using Esiur.Misc;
using System.Security.Cryptography;

namespace Esiur.Net.HTTP
{
    //[Serializable]
    public class HTTPConnection : NetworkConnection
    {
        /*
        public enum SendOptions : int
        {
            AllCalculateLength,
            AllDontCalculateLength,
            SpecifiedHeadersOnly,
            DataOnly
        }
         */

        public HTTPConnection()
        {
            Response = new HTTPResponsePacket();
            variables = new KeyList<string, object>();
        }

        public void SetParent(HTTPServer Parent)
        {
            Server = Parent;
        }

        //public bool HeadersSent;


        private KeyList<string, object> variables;
        private bool Busy = false;
        private DateTime RequestTime = DateTime.MinValue;

        public bool WSMode;
        private HTTPServer Server;
        public WebsocketPacket WSRequest;
        public HTTPRequestPacket Request;
        public HTTPResponsePacket Response;

        HTTPSession session;

        public KeyList<string, object> Variables
        {
            get
            {
                return variables;
            }
        }


        public bool IsBusy()
        {
            return Busy;
        }


        internal long Parse(byte[] data)
        {
            if (WSMode)
            {
                // now parse WS protocol
                WebsocketPacket ws = new WebsocketPacket();

                var pSize = ws.Parse(data, 0, (uint)data.Length);


                if (pSize > 0)
                {
                    WSRequest = ws;
                    return 0;
                }
                else
                {
                    return pSize;
                }
            }
            else
            {
                HTTPRequestPacket rp = new HTTPRequestPacket();
                var pSize = rp.Parse(data, 0, (uint)data.Length);
                if (pSize > 0)
                {
                    Request = rp;
                    return 0;
                }
                else
                {
                    return pSize;
                }
            }
        }


        /*
        public override void Send(string Response)
        {
            Send(Response, SendOptions.AllCalculateLength);
        }

        public void Send(string Message, SendOptions Options)
        {
            
            if (Response.Handled)
                return;

            if (Response != null)
                Send(Encoding.Default.GetBytes(Response), Options);
            else
                Send((byte[])null, Options);
        }

        public void Send(MemoryStream ms)
        {
            Send(ms.ToArray(), SendOptions.AllCalculateLength);
        }

         */

        public void Flush()
        {
            // close the connection
            if (Request.Headers["connection"].ToLower() != "keep-alive" & Connected)
                Close();
        }

        public bool Upgrade()
        {
            if (IsWebsocketRequest())
            {
                string magicString = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
                string ret = Request.Headers["Sec-WebSocket-Key"] + magicString;
                // Compute the SHA1 hash
                SHA1 sha = SHA1.Create();
                byte[] sha1Hash = sha.ComputeHash(Encoding.UTF8.GetBytes(ret));
                Response.Headers["Upgrade"] =  Request.Headers["Upgrade"];
                Response.Headers["Connection"] = Request.Headers["Connection"];// "Upgrade";
                Response.Headers["Sec-WebSocket-Accept"] = Convert.ToBase64String(sha1Hash);

                if (Request.Headers.ContainsKey("Sec-WebSocket-Protocol"))
                    Response.Headers["Sec-WebSocket-Protocol"] = Request.Headers["Sec-WebSocket-Protocol"];

                //Response.Headers["Sec-WebSocket-Protocol"] = Request.Headers["Sec-WebSocket-Protocol"];
                //Response.Headers["Origin"] = Request.Headers["Origin"];

                Response.Number = HTTPResponsePacket.ResponseCode.HTTP_SWITCHING;
                Response.Text = "Switching Protocols";
                WSMode = true;

                //Send((byte[])null, SendOptions.AllDontCalculateLength);
                Send();

                return true;
            }

            return false;
        }

        public HTTPServer Parent
        {
            get
            {
                return Server;
            }
        }

        public override void Send(string data)
        {
            Response.Message = Encoding.UTF8.GetBytes(data);
            Send();
        }

        public override void Send(byte[] message)
        {
            Response.Message = message;
            Send();
        }

        public void Send(HTTPResponsePacket.ComposeOptions Options = HTTPResponsePacket.ComposeOptions.AllCalculateLength)
        {
            if (Response.Handled)
                return;

            Busy = true;

            try
            {
                Response.Compose(Options);
                base.Send(Response.Data);

                // Refresh the current session
                if (session != null)
                    session.Refresh();

            }
            catch
            {
                try
                {
                    Close();// Server.CloseClient(Connection);
                }
                finally { }
            }
            finally
            {
                Busy = false;
            }
        }

        
        public void CreateNewSession()
        {
            if (session == null)
            {
                // Create a new one
                session = Server.CreateSession(Global.GenerateCode(12), 60 * 20);

                HTTPResponsePacket.HTTPCookie cookie = new HTTPResponsePacket.HTTPCookie("SID", session.Id);
                cookie.Expires = DateTime.MaxValue;
                cookie.Path = "/";
                cookie.HttpOnly = true;

                Response.Cookies.Add(cookie);
            }
        }


        public bool IsWebsocketRequest()
        {
            if (Request.Headers.ContainsKey("connection")
                && Request.Headers["connection"].ToLower().Contains("upgrade")
                && Request.Headers.ContainsKey("upgrade")
                && Request.Headers["upgrade"].ToLower() == "websocket"
                && Request.Headers.ContainsKey("Sec-WebSocket-Version")
                && Request.Headers["Sec-WebSocket-Version"] == "13"
                && Request.Headers.ContainsKey("Sec-WebSocket-Key"))
                //&& Request.Headers.ContainsKey("Sec-WebSocket-Protocol"))
            {
                return true;
            }
            else
            {
                return false;
            }
        }

        public void SendFile(string filename)
        {
            if (Response.Handled == true)
                return;

            try
            {

                //HTTP/1.1 200 OK
                //Server: Microsoft-IIS/5.0
                //Content-Location: http://127.0.0.1/index.html
                //Date: Wed, 10 Dec 2003 19:10:25 GMT
                //Content-Type: text/html
                //Accept-Ranges: bytes
                //Last-Modified: Mon, 22 Sep 2003 22:36:56 GMT
                //Content-Length: 1957

                if (!File.Exists(filename))
                {
                    Response.Number = HTTPResponsePacket.ResponseCode.HTTP_NOTFOUND;
                    Send("File Not Found");//, SendOptions.AllCalculateLength);
                    return;
                }

                Busy = true;

                System.DateTime FWD = File.GetLastWriteTime(filename);
                if (Request.Headers.ContainsKey("if-modified-since"))// != DateTime.Parse("12:00:00 AM"))
                {
                    try
                    {
                        DateTime IMS = DateTime.Parse(Request.Headers["if-modified-since"]);
                        if (FWD <= IMS)
                        {
                            Response.Number = HTTPResponsePacket.ResponseCode.HTTP_NOTMODIFIED;
                            Response.Text = "Not Modified";
                        }
                    }
                    catch
                    {

                    }
                }


                if (Response.Number == HTTPResponsePacket.ResponseCode.HTTP_NOTMODIFIED)
                {
                    Send((byte[])null);
                }
                else
                {
                    // Fri, 30 Oct 2007 14:19:41 GMT
                    Response.Headers["Last-Modified"] = FWD.ToUniversalTime().ToString("ddd, dd MMM yyyy HH:mm:ss");
                    FileInfo fi = new FileInfo(filename);
                    Response.Headers["Content-Length"] = fi.Length.ToString();
                    Send(HTTPResponsePacket.ComposeOptions.SpecifiedHeadersOnly);
                    using (var fs = new FileStream(filename, FileMode.Open))
                    {
                        var buffer = new byte[5000];
                        var offset = 0;
                        while (offset < fs.Length)
                        {
                            var n = fs.Read(buffer, offset, buffer.Length);
                            offset += n;
                            base.Send(buffer);
                        }
                    }
                }

                Busy = false;

                return;
            }
            catch
            {
                Busy = false;

                try
                {
                    Close();
                }
                finally { }
            }
        }
    }
}