using Org.BouncyCastle.Crypto.Tls;
using Org.BouncyCastle.Security;
using System;
using System.Collections;
using System.Collections.Generic;
using System.ComponentModel;
using System.Diagnostics;
using System.IO;
using System.Linq;
using System.Net.Sockets;
using System.Text;
using System.Text.RegularExpressions;

namespace System.Net
{
    class VRSTlsClient : DefaultTlsClient
    {
        string HostName;

        public VRSTlsClient(string hostname)
        {
            HostName = hostname;
        }

        public override TlsAuthentication GetAuthentication()
        {
            TlsAuthentication auth = new MyTlsAuthentication();
            return auth;
        }

        public override void NotifyNewSessionTicket(NewSessionTicket newSessionTicket)
        {
            base.NotifyNewSessionTicket(newSessionTicket);
        }

        public override IDictionary GetClientExtensions()
        {
            var clientExtensions = base.GetClientExtensions();
            /*
            List<ServerName> servers = new List<ServerName>();
            servers.Add(new ServerName(NameType.host_name, HostName));
            TlsExtensionsUtilities.AddServerNameExtension(clientExtensions, new ServerNameList(servers));
            */
            return clientExtensions;
        }

        private static bool ReadContent(Stream stream, int contentlength, int timeout, ref string response)
        {
            // set stop watch as timout
            Stopwatch st = new Stopwatch();
            st.Start();
            string resp = "";
            int count = 0;
            // assign buffer
            byte[] buff = new byte[1024];
            int bytesread = 0;
            // read content blockwise
            while (bytesread < contentlength)
            {
                int bytestoread = buff.Length - bytesread;
                if (bytestoread > buff.Length)
                    bytestoread = buff.Length;
                bytesread += stream.Read(buff, 0, buff.Length);
                // add it to response
                resp += Encoding.ASCII.GetString(buff, 0, buff.Length);
                if (st.ElapsedMilliseconds > timeout)
                    throw new TimeoutException("Connection timed out.");
            }
            /*
            string trailer = "";
            // reassign buffer
            buff = new byte[1];
            // read stream bytewise until CRLFCRLF is detected, should be the next two bytes
            do
            {
                count = stream.Read(buff, 0, buff.Length);
                trailer += Encoding.ASCII.GetString(buff, 0, buff.Length);
                if (st.ElapsedMilliseconds > timeout)
                    throw new TimeoutException("Connection timed out.");
            }
            while (!trailer.Contains("\r\n"));
            */
            Console.WriteLine("Reading content [" + contentlength.ToString() + " bytes]: " + resp);
            response += resp;
            return true;
        }

        private static bool ReadChunkedContent(Stream stream, int timeout, ref string response)
        {
            // set stop watch as timout
            Stopwatch st = new Stopwatch();
            st.Start();
            string resp = "";
            byte[] buff = new byte[1];
            int count = 0;
            string strcontentlength = "";
            int contentlength = 0;
            int bytesread = 0;
            // chunked transfer, first line should contain content length
            // read stream bytewise until CRLF is detected
            try
            {
                do
                {
                    count = stream.Read(buff, 0, buff.Length);
                    strcontentlength += Encoding.ASCII.GetString(buff, 0, buff.Length);
                    if (st.ElapsedMilliseconds > timeout)
                        throw new TimeoutException("Connection timed out.");
                }
                while (!strcontentlength.Contains("\r\n"));
                strcontentlength = strcontentlength.Replace("\r\n", "");
                contentlength = int.Parse(strcontentlength, System.Globalization.NumberStyles.HexNumber);
                // finished reading all chunks
                if (contentlength == 0)
                {
                    Console.WriteLine("Reading chunked content finished");
                    return true;
                }

                // re-assign buffer
                buff = new byte[contentlength];

                // read content in 1kByte chunks until contentlength is reached
                while (bytesread < contentlength)
                {
                    int bytestoread = buff.Length - bytesread;
                    if (bytestoread > buff.Length)
                        bytestoread = buff.Length;
                    bytesread += stream.Read(buff, bytesread, bytestoread);
                    // add it to response
                    if (st.ElapsedMilliseconds > timeout)
                        throw new TimeoutException("Connection timed out.");
                }
                resp += Encoding.ASCII.GetString(buff, 0, buff.Length);

                string trailer = "";
                // reassign buffer
                buff = new byte[1];
                // read stream bytewise until CRLFCRLF is detected, should be the next two bytes
                do
                {
                    count = stream.Read(buff, 0, buff.Length);
                    trailer += Encoding.ASCII.GetString(buff, 0, buff.Length);
                    if (st.ElapsedMilliseconds > timeout)
                        throw new TimeoutException("Connection timed out.");
                }
                while (!trailer.Contains("\r\n"));
            }
            catch (Exception ex)
            {
                Console.WriteLine("Error while reading chunked content: " + ex.Message);
                return true;
            }
            // Console.WriteLine("Reading chunked content [" + contentlength.ToString() + " bytes]: " + resp);
            response += resp;
            return false;
        }

        public static string DownloadFile(string url, int timeout, string username, string password)
        {
            string response = "";
            Uri uri = null;
            // try to parse url
            try
            {
                uri = new Uri(url);
            }
            catch (Exception ex)
            {
                return ex.Message;
            }
            // simple connection
            if (url.Contains("http:"))
            {
                HttpWebRequest webrequest = (HttpWebRequest)HttpWebRequest.Create(url);
                webrequest.Referer = "http://www.vrs-world.com/";
                webrequest.Timeout = timeout;
                webrequest.UserAgent = "Mozilla/5.0 (Windows NT 6.1; WOW64; rv:37.0) Gecko/20100101 Firefox/37.0";
                webrequest.Accept = "application/json, text/javascript, */*;q=0.01";
                webrequest.AutomaticDecompression = System.Net.DecompressionMethods.Deflate | System.Net.DecompressionMethods.GZip;

                // include authorization if username/password are not empty
                if (!String.IsNullOrEmpty(username) && !String.IsNullOrEmpty(password))
                {
                    // Base64 encode username:password
                    string s = Convert.ToBase64String(Encoding.UTF8.GetBytes(username + ":" + password));
                    webrequest.Headers.Add("Authorization: Basic " + s);
                }

                Console.WriteLine("[VRSWebFeed]: Getting web response");
                HttpWebResponse webresponse = (HttpWebResponse)webrequest.GetResponse();
                Console.WriteLine("[VRSWebFeed]: Reading stream");

                using (StreamReader sr = new StreamReader(webresponse.GetResponseStream()))
                {
                    response = sr.ReadToEnd();
                }
            }
            // using SSL for connection
            else
            {
                // create new TCP-Client
                using (var client = new TcpClient(uri.Host, uri.Port))
                {
                    var sr = new SecureRandom();
                    var cl = new VRSTlsClient(uri.Host);
                    var protocol = new TlsClientProtocol(client.GetStream(), sr);
                    protocol.Connect(cl);

                    using (var stream = protocol.Stream)
                    {
                        var hdr = new StringBuilder();
                        hdr.AppendLine("GET " + uri.PathAndQuery + " HTTP/1.1");
                        hdr.AppendLine("Host: " + uri.Host);
                        hdr.AppendLine("Content-Type: text/json; charset=utf-8");
                        hdr.AppendLine("Connection: close");

                        // include authorization if username/password are not empty
                        if (!String.IsNullOrEmpty(username) && !String.IsNullOrEmpty(password))
                        {
                            // Base64 encode username:password
                            string s = Convert.ToBase64String(Encoding.UTF8.GetBytes(username + ":" + password));
                            hdr.AppendLine("Authorization: Basic " + s);
                        }
                        hdr.AppendLine();

                        var dataToSend = Encoding.ASCII.GetBytes(hdr.ToString());

                        stream.Write(dataToSend, 0, dataToSend.Length);

                        byte[] buff;
                        // set stop watch as timout
                        Stopwatch st = new Stopwatch();
                        st.Start();
                        //read header bytewise
                        string header = "";
                        int totalRead = 0;
                        buff = new byte[1];
                        do
                        {
                            totalRead = stream.Read(buff, 0, buff.Length);
                            header += Encoding.ASCII.GetString(buff);
                            if (st.ElapsedMilliseconds > timeout)
                                throw new TimeoutException("Connection to " + url + " timed out.");
                        }
                        while (!header.Contains("\r\n\r\n"));
                        Console.Write(header);
                        int contentlength = 0;
                        if (header.Contains("Transfer-Encoding: chunked"))
                        {
                            // chunked transfer, read all chunks until complete
                            while (!ReadChunkedContent(stream, timeout, ref response))
                            { }
                        }
                        else
                        {
                            // get content length from header
                            Regex rcontentlength = new Regex("(?<=Content-Length:\\s)\\d+", RegexOptions.IgnoreCase);
                            contentlength = int.Parse(rcontentlength.Match(header).Value);
                            ReadContent(stream, contentlength, timeout, ref response);
                        }
                        st.Stop();
                    }
                }
            }
            return response;
        }
    }

    class MyTlsAuthentication : TlsAuthentication
    {
        public TlsCredentials GetClientCredentials(CertificateRequest certificateRequest)
        {
            return null;
        }

        public void NotifyServerCertificate(Certificate serverCertificate)
        {
        }
    }

}