more work in progress

libstratum/EthStratumClient.cpp

@@ -0,0 +1,154 @@
+#include "EthStratumClient.h"
+
+
+EthStratumClient::EthStratumClient(string const & host, string const & port, string const & user, string const & pass) : m_socket(m_io_service)
+{
+	m_host = host;
+	m_port = port;
+	m_user = user;
+	m_pass = pass;
+
+	
+
+}
+
+EthStratumClient::~EthStratumClient()
+{
+
+}
+
+void EthStratumClient::connect()
+{
+	tcp::resolver r(m_io_service);
+	tcp::resolver::query q(m_host, m_port);
+
+	r.async_resolve(q, boost::bind(&EthStratumClient::resolve_handler,
+																	this, boost::asio::placeholders::error,
+																	boost::asio::placeholders::iterator));
+
+	m_io_service.run();
+}
+
+void EthStratumClient::resolve_handler(const boost::system::error_code& ec, tcp::resolver::iterator i)
+{
+	if (!ec)
+	{
+		async_connect(m_socket, i, boost::bind(&EthStratumClient::connect_handler,
+																					this, boost::asio::placeholders::error,
+																					boost::asio::placeholders::iterator));
+	}
+	else
+	{
+		cerr << "Could not resolve host " << m_host << ":" << m_port << ", " << ec.message();
+	}
+}
+
+void EthStratumClient::connect_handler(const boost::system::error_code& ec, tcp::resolver::iterator i)
+{
+	if (!ec)
+	{
+		cnote << "Connected to stratum server " << m_host << ":" << m_port;
+
+		std::ostream os(&m_requestBuffer);
+		os << "{\"id\": 1, \"method\": \"mining.subscribe\", \"params\": []}\n";
+
+		async_write(m_socket, m_requestBuffer,
+			boost::bind(&EthStratumClient::subscribe_handler, this,
+									boost::asio::placeholders::error));
+	}
+	else
+	{
+		cerr << "Could not connect to stratum server " << m_host << ":" << m_port << ", " << ec.message();
+	}
+
+}
+
+void EthStratumClient::subscribe_handler(const boost::system::error_code& ec) {
+	if (!ec)
+	{
+		async_read_until(m_socket, m_responseBuffer, "\n",
+			boost::bind(&EthStratumClient::read_subscribe_handler, this,
+			boost::asio::placeholders::error, boost::asio::placeholders::bytes_transferred));
+
+	}
+	else
+	{
+		cerr << "Stratum subscription failed: "  << ec.message();
+	}
+}
+
+void EthStratumClient::read_subscribe_handler(const boost::system::error_code& ec, std::size_t bytes_transferred)
+{
+	if (!ec || ec == error::eof)
+	{
+		std::istream is(&m_responseBuffer);
+		std::string response;
+		getline(is, response);
+		cnote << response;
+
+		std::ostream os(&m_requestBuffer);
+		os << "{\"id\": 2, \"method\": \"mining.authorize\", \"params\": [\"" << m_user << "\",\"" << m_pass << "\"]}\n";
+
+
+		async_write(m_socket, m_requestBuffer,
+			boost::bind(&EthStratumClient::authorize_handler, this,
+			boost::asio::placeholders::error));
+	}
+	else
+	{
+		cerr << "Stratum read subscription failed: " << ec.message();
+	}
+}
+
+void EthStratumClient::authorize_handler(const boost::system::error_code& ec) 
+{
+	if (!ec)
+	{
+		async_read_until(m_socket, m_responseBuffer, "\n",
+			boost::bind(&EthStratumClient::read_authorize_handler, this,
+			boost::asio::placeholders::error, boost::asio::placeholders::bytes_transferred));
+
+	}
+	else
+	{
+		cerr << "Stratum authorization failed: " << ec.message();
+	}
+}
+
+void EthStratumClient::read_authorize_handler(const boost::system::error_code& ec, std::size_t bytes_transferred)
+{
+	if (!ec)
+	{
+		std::istream is(&m_responseBuffer);
+		std::string response;
+		getline(is, response);
+		cnote << response;
+
+		async_read_until(m_socket, m_responseBuffer, "\n",
+			boost::bind(&EthStratumClient::read_work_handler, this,
+			boost::asio::placeholders::error, boost::asio::placeholders::bytes_transferred));
+	}
+	else
+	{
+		cerr << "Stratum read authorization failed: " << ec.message();
+	}
+}
+
+void EthStratumClient::read_work_handler(const boost::system::error_code& ec, std::size_t bytes_transferred)
+{
+	if (!ec)
+	{
+		std::istream is(&m_responseBuffer);
+		std::string response;
+		getline(is, response);
+		cnote << response;
+
+		async_read_until(m_socket, m_responseBuffer, "\n",
+			boost::bind(&EthStratumClient::read_work_handler, this,
+			boost::asio::placeholders::error, boost::asio::placeholders::bytes_transferred));
+	}
+	else
+	{
+		cerr << "Stratum read work failed: " << ec.message();
+	}
+}

libstratum/EthStratumClient.h

@@ -0,0 +1,39 @@
+#include <iostream>
+#include <boost/array.hpp>
+#include <boost/asio.hpp>
+#include <boost/bind.hpp>
+#include <json/json.h>
+#include <libdevcore/Log.h>
+
+using namespace std;
+using namespace boost::asio;
+using boost::asio::ip::tcp;
+
+class EthStratumClient
+{
+public:
+	EthStratumClient(string const & host, string const & port, string const & user, string const & pass);
+	~EthStratumClient();
+
+	void connect();
+	
+private:
+	void resolve_handler(const boost::system::error_code& ec, tcp::resolver::iterator i);
+	void connect_handler(const boost::system::error_code& ec, tcp::resolver::iterator i);
+	
+	void subscribe_handler(const boost::system::error_code& ec);
+	void authorize_handler(const boost::system::error_code& ec);
+	void read_subscribe_handler(const boost::system::error_code& ec, std::size_t bytes_transferred);
+	void read_authorize_handler(const boost::system::error_code& ec, std::size_t bytes_transferred);
+	void read_work_handler(const boost::system::error_code& ec, std::size_t bytes_transferred);
+
+	string m_host;
+	string m_port;
+	string m_user;
+	string m_pass;
+
+	io_service m_io_service;
+	tcp::socket m_socket;
+	boost::asio::streambuf m_requestBuffer;
+	boost::asio::streambuf m_responseBuffer;
+};