Merge pull request #1725 from ethereum/ipcreadloop

Try reading multiple times from IPC.

1 files changed, 33 insertions, 16 deletions

diff --git a/test/RPCSession.cpp b/test/RPCSession.cpp
index ff00d783..be8774bc 100644
--- a/test/RPCSession.cpp
+++ b/test/RPCSession.cpp
@@ -16,19 +16,20 @@
 
 	The Implementation originally from https://msdn.microsoft.com/en-us/library/windows/desktop/aa365592(v=vs.85).aspx
 */
-/** @file RPCSession.cpp
- * @author Dimtiry Khokhlov <dimitry@ethdev.com>
- * @author Alex Beregszaszi
- * @date 2016
- */
+/// @file RPCSession.cpp
+/// Low-level IPC communication between the test framework and the Ethereum node.
+
+#include "RPCSession.h"
 
-#include <string>
-#include <stdio.h>
-#include <thread>
 #include <libdevcore/CommonData.h>
+
 #include <json/reader.h>
 #include <json/writer.h>
-#include "RPCSession.h"
+
+#include <string>
+#include <stdio.h>
+#include <thread>
+#include <chrono>
 
 using namespace std;
 using namespace dev;
@@ -107,13 +108,24 @@ string IPCSocket::sendRequest(string const& _req)
 	return string(m_readBuf, m_readBuf + cbRead);
 #else
 	if (send(m_socket, _req.c_str(), _req.length(), 0) != (ssize_t)_req.length())
-		BOOST_FAIL("Writing on IPC failed");
+		BOOST_FAIL("Writing on IPC failed.");
 
-	ssize_t ret = recv(m_socket, m_readBuf, sizeof(m_readBuf), 0);
+	auto start = chrono::steady_clock::now();
+	ssize_t ret;
+	do
+	{
+		ret = recv(m_socket, m_readBuf, sizeof(m_readBuf), 0);
+		// Also consider closed socket an error.
+		if (ret < 0)
+			BOOST_FAIL("Reading on IPC failed.");
+	}
+	while (
+		ret == 0 &&
+		chrono::duration_cast<chrono::milliseconds>(chrono::steady_clock::now() - start).count() < m_readTimeOutMS
+	);
 
-	// Also consider closed socket an error.
-	if (ret <= 0)
-		BOOST_FAIL("Reading on IPC failed");
+	if (ret == 0)
+		BOOST_FAIL("Timeout reading on IPC.");
 
 	return string(m_readBuf, m_readBuf + ret);
 #endif
@@ -186,12 +198,17 @@ string RPCSession::eth_getStorageRoot(string const& _address, string const& _blo
 
 void RPCSession::personal_unlockAccount(string const& _address, string const& _password, int _duration)
 {
-	BOOST_REQUIRE(rpcCall("personal_unlockAccount", { quote(_address), quote(_password), to_string(_duration) }) == true);
+	BOOST_REQUIRE_MESSAGE(
+		rpcCall("personal_unlockAccount", { quote(_address), quote(_password), to_string(_duration) }),
+		"Error unlocking account " + _address
+	);
 }
 
 string RPCSession::personal_newAccount(string const& _password)
 {
-	return rpcCall("personal_newAccount", { quote(_password) }).asString();
+	string addr = rpcCall("personal_newAccount", { quote(_password) }).asString();
+	BOOST_MESSAGE("Created account " + addr);
+	return addr;
 }
 
 void RPCSession::test_setChainParams(vector<string> const& _accounts)