1 files changed, 65 insertions, 17 deletions

diff --git a/client/cmixclient.cpp b/client/cmixclient.cpp
index e3f5ac4..195c5e0 100644
--- a/client/cmixclient.cpp
+++ b/client/cmixclient.cpp
@@ -1,38 +1,86 @@
 
 #include "cmixclient.hpp"
 
-void CMixClient::initialized() {
-	BOOST_LOG_TRIVIAL(trace) << "Client connections were made";
-	for(auto&& connection : network_connections) {
-		cmix_proto::Bye bye;
-		connection.send(bye);
-	}
-	io_service.stop();
+void CMixClient::key_exchange(int i) {
+	shared_keys.resize(network_connections.size());
+		
+	cmix_proto::KeyExchange ke;
+	ke.set_public_key(keypair.pub, keypair.pub_len);
+		
+	network_connections[i].send(ke);
+		
+	cmix_proto::Bye bye;
+	network_connections[i].send(bye);
 }
 
 void CMixClient::initialize_connections() {
 	network_connections.reserve(network_details.size());
 	
-	int i = network_details.size();
-	auto handler = [this, i]() mutable {
-		cmix_proto::ImAClient imaclient;
-		network_connections.at(network_details.size() - i).send(imaclient);
+	for(int i = 0; i < network_details.size(); ++i) {
+		auto handler = [this, i]() mutable {
+			cmix_proto::ImAClient imaclient;
+			BOOST_LOG_TRIVIAL(trace) << "sending imaclient to node: " << i;
+			network_connections.at(i).send(imaclient);
 		
-		if(--i == 0) {
-			initialized();
-		}
-	};
+			key_exchange(i);
+		};
 	
-	for(auto&& details : network_details) {
 		network_connections.emplace_back(boost::asio::ip::tcp::socket(io_service));
-		network_connections.back().async_connect(details.host, details.port, handler);
+		network_connections.back().async_connect(network_details[i].host, network_details[i].port, handler);
+	
+	}
+}
+
+cmix_proto::CMixMessage CMixClient::parse_cmix_message(std::vector<uint8_t> const& buffer)
+{
+	cmix_proto::CMixMessage message;
+	if(!message.ParseFromArray(buffer.data(), buffer.size())) {
+		BOOST_LOG_TRIVIAL(error) << "Received something which was not a CMixMessage";
+		throw std::runtime_error("Network communication was disrupted in a unrecoverable way.");
+	}
+	return message;
+}
+
+void CMixClient::handle_key_exchange(int node_id, cmix_proto::KeyExchange const& ke)
+{
+	shared_keys[node_id] = api.derive_shared_key(keypair, reinterpret_cast<uint8_t const*>(ke.public_key().c_str()), false);
+}
+
+void CMixClient::handle_message(int node_id, std::vector<uint8_t> const& message_buffer)
+{
+	cmix_proto::CMixMessage message;
+	try {
+		message = parse_cmix_message(message_buffer);
+	} catch(std::runtime_error const& e) {
+		for(auto&& connection : network_connections) {
+			connection.close();
+		}
+		io_service.stop();
+		return;
+	}
+	
+	switch(message.contents_case()) {
+		case cmix_proto::CMixMessage::ContentsCase::kKeyexchange: {
+			handle_key_exchange(node_id, *message.mutable_keyexchange());
+			return;
+		}
+		default: {
+			BOOST_LOG_TRIVIAL(error) << "Received unknown message";
+		}
 	}
+	
+	for(auto&& connection : network_connections) {
+		connection.close();
+	}
+	io_service.stop();
 }
 
 CMixClient::CMixClient(std::vector<NodeDetails> details)
 : io_service()
 , network_details(details)
 , network_connections()
+, api(get_implementation())
+, keypair(api.create_key_pair())
 {
 	initialize_connections();
 }