Commit bb4bec98d5a488e0ec32d093fce41986d833f021

ckolivas 2012-10-17T09:34:32

Use select on stratum send to make sure the socket is writeable.

diff --git a/util.c b/util.c
index c58969a..0372593 100644
--- a/util.c
+++ b/util.c
@@ -869,6 +869,7 @@ bool extract_sockaddr(struct pool *pool, char *url)
 /* Send a single command across a socket, appending \n to it */
 static bool __stratum_send(struct pool *pool, char *s, ssize_t len)
 {
+	SOCKETTYPE sock = pool->sock;
 	ssize_t ssent = 0;
 
 	if (opt_protocol)
@@ -878,8 +879,16 @@ static bool __stratum_send(struct pool *pool, char *s, ssize_t len)
 	len++;
 
 	while (len > 0 ) {
+		struct timeval timeout = {0, 0};
 		size_t sent = 0;
+		fd_set wd;
 
+		FD_ZERO(&wd);
+		FD_SET(sock, &wd);
+		if (select(sock + 1, NULL, &wd, NULL, &timeout) < 1) {
+			applog(LOG_DEBUG, "Write select failed on pool %d sock", pool->pool_no);
+			return false;
+		}
 		if (curl_easy_send(pool->stratum_curl, s + ssent, len, &sent) != CURLE_OK) {
 			applog(LOG_DEBUG, "Failed to curl_easy_send in stratum_send");
 			return false;