#pragma warning(disable : 4267)
#pragma warning(disable : 4312)
#pragma warning(disable : 4311)
#include "stdafx.h"
#include <cstdlib>
#include <iostream>
#include <boost/bind.hpp>
#include <boost/asio.hpp>
#include <boost/asio/ssl.hpp>
#include <stdio.h>
#include <stdlib.h>
#include <time.h>
#include <assert.h>
#include <boost/thread.hpp>


char * messages[] = 
{
	"This is a test of the emergency broadcast system, this is only a test.",
	"The quick brown fox jumped over the lazy old dog.",
	"Life is like a box of chocolates, you never know what you are going to get."
};

char * messages2[] = 
{
	"Lovely day.  ",
	"Posted from thread # 2.  ",
	"The moon is made of cheese.  "
};

std::string msg;

enum { max_length = 1024 };

class client
{
public:
	client(boost::asio::io_service& io_service, boost::asio::ssl::context& context,
		boost::asio::ip::tcp::resolver::iterator endpoint_iterator)
		: socket_(io_service, context), handshakeFinished(false)
	{
		boost::asio::ip::tcp::endpoint endpoint = *endpoint_iterator;
		socket_.lowest_layer().async_connect(endpoint,
			boost::bind(&client::handle_connect, this,
			boost::asio::placeholders::error, ++endpoint_iterator));
	}

	void handle_connect(const boost::asio::error& error,
		boost::asio::ip::tcp::resolver::iterator endpoint_iterator)
	{
		if (!error)
		{
			socket_.async_handshake(boost::asio::ssl::stream_base::client,
				boost::bind(&client::handle_handshake, this,
					boost::asio::placeholders::error));
		}
		else if (endpoint_iterator != boost::asio::ip::tcp::resolver::iterator())
		{
			socket_.lowest_layer().close();
			boost::asio::ip::tcp::endpoint endpoint = *endpoint_iterator;
			socket_.lowest_layer().async_connect(endpoint,
				boost::bind(&client::handle_connect, this,
					boost::asio::placeholders::error, ++endpoint_iterator));
		}
		else
		{
			std::cout << "Connect failed: " << error << "\n";
		}
	}

	void handle_handshake(const boost::asio::error& error)
	{
		if (!error)
		{
			handshakeFinished = true;

			msg.clear();
			msg = "Start - ";

			while (rand() % 20)
			{
				int i = rand() % 3;
				msg += messages[i];
			}

			std::cout << "Writing out a message of size " << msg.size() << std::endl;

			// not very portable, but doesn't matter when client/server running on same machine
			_size = msg.size();  
			std::vector<boost::asio::const_buffer> buffers;
			buffers.push_back(boost::asio::buffer(&_size, sizeof _size));
			buffers.push_back(boost::asio::buffer(msg));

			boost::recursive_mutex::scoped_lock lock(_mutex);
			boost::asio::async_write(socket_,
				buffers,
				boost::bind(&client::handle_write, this,
					boost::asio::placeholders::error,
					boost::asio::placeholders::bytes_transferred));
		}
		else
		{
			std::cout << "Handshake failed: " << error << "\n";
		}
	}

	void handle_write(const boost::asio::error& error, size_t bytes_transferred)
	{
		if (!error)
		{
			//Sleep(rand() % 500);
			msg.clear();

			msg = "Start - ";

			static int randmod = 1;
			randmod++;
			if (randmod > 2000)
			{
				randmod = 2000;
			}

			while (rand() % randmod)
			{
				int i = rand() % 3;
				msg += messages[i];
			}

			std::cout << "Writing out a message of size " << msg.size() << std::endl;

			// not very portable, but doesn't matter when client/server running on same machine
			_size = msg.size();  
			_writeBuffers.clear();
			_writeBuffers.push_back(boost::asio::buffer(&_size, sizeof _size));
			_writeBuffers.push_back(boost::asio::buffer(msg));

			boost::recursive_mutex::scoped_lock lock(_mutex);
			boost::asio::async_write(socket_,
				_writeBuffers,
				boost::bind(&client::handle_write, this,
					boost::asio::placeholders::error,
					boost::asio::placeholders::bytes_transferred));
		}
		else
		{
			std::cout << "Write failed: " << error << "\n";
		}
	}

	void handle_read(const boost::asio::error & error, size_t bytes_transferred)
	{
		if (!error)
		{
			std::cout << "Reply: ";
			std::cout.write(reply_, bytes_transferred);
			std::cout << "\n";
			boost::recursive_mutex::scoped_lock lock(_mutex);
			socket_.async_read_some(
				boost::asio::buffer(reply_, max_length),
				boost::bind(&client::handle_read, this,
					boost::asio::placeholders::error,
					boost::asio::placeholders::bytes_transferred));
		}
		else
		{
			std::cerr << "Receiving Thread Error: " << error.what() << std::endl;
			boost::recursive_mutex::scoped_lock lock(_mutex);
			socket_.io_service().interrupt();
		}
	}

	void receiverThread()
	{
		// yes, i should use a condition wait here, i was in a hurry
		while (!handshakeFinished)
		{
			Sleep(100);
		}
			
		boost::recursive_mutex::scoped_lock lock(_mutex);
		socket_.async_read_some(
			boost::asio::buffer(reply_, max_length),
			boost::bind(&client::handle_read, this,
				boost::asio::placeholders::error,
				boost::asio::placeholders::bytes_transferred));
		lock.unlock();
		std::cout << "Receiving Thread Entering io_service.run()" << std::endl;
		socket_.io_service().run();
		std::cout << "Receiving Thread Exiting" << std::endl;
	}

private:
	mutable boost::recursive_mutex _mutex;

	std::vector<boost::asio::const_buffer> _writeBuffers;
	boost::asio::ssl::stream<boost::asio::ip::tcp::socket> socket_;
	char request_[max_length];
	char reply_[max_length];
	size_t _size;
	bool handshakeFinished;
};


int clientmain(int argc, char* argv[])
{
	try
	{
		if (argc != 3)
		{
			std::cerr << "Usage: client <host> <port>\n";
			return 1;
		}

		srand(123);

		boost::asio::io_service io_service;

		boost::asio::ip::tcp::resolver resolver(io_service);
		boost::asio::ip::tcp::resolver::query query(argv[1], argv[2]);
		boost::asio::ip::tcp::resolver::iterator iterator = resolver.resolve(query);

		boost::asio::ssl::context ctx(io_service, boost::asio::ssl::context::sslv23);
		ctx.set_verify_mode(boost::asio::ssl::context::verify_peer);
		ctx.load_verify_file("ca.pem");

		client c(io_service, ctx, iterator);
		
		// memory leak, but this isn't meant to shutdown gracefully anyway
		boost::thread * mythread = new boost::thread(boost::bind(client::receiverThread, &c));

		io_service.run();
	}
	catch (boost::asio::error& e)
	{
		std::cerr << e << "\n";
	}
	catch (std::exception& e)
	{
		std::cerr << "Exception: " << e.what() << "\n";
	}

	return 0;
}



int _tmain(int argc, _TCHAR* argv[])
{
	return clientmain(argc, argv);
}