Commit 01886493cfc1c4104f43a2b1dadaf23981c2c682

Znort 987 2011-08-15T11:59:00

Add set_non_blocking routing In preparation for upcoming generic unix automated algorithm benchmark method (as opposed to linux specific), add a utility method.

diff --git a/main.c b/main.c
index fb072fe..78e5372 100644
--- a/main.c
+++ b/main.c
@@ -36,8 +36,10 @@
 #include "findnonce.h"
 #include "ocl.h"
 
-#if defined(__linux)
-	#include <sys/mman.h>
+#if defined(unix)
+	#include <errno.h>
+	#include <fcntl.h>
+	#include <unistd.h>
 	#include <sys/wait.h>
 	#include <sys/types.h>
 #endif
@@ -449,6 +451,31 @@ static double bench_algo_stage3(
 	return rate;
 }
 
+#if defined(unix)
+
+	// Change non-blocking status on a file descriptor
+	static void set_non_blocking(
+		int fd,
+		int yes
+	)
+	{
+		int flags = fcntl(fd, F_GETFL, 0);
+		if (flags<0) {
+			perror("fcntl(GET) failed");
+			exit(1);
+		}
+		flags = yes ? (flags|O_NONBLOCK) : (flags&~O_NONBLOCK);
+
+		int r = fcntl(fd, F_SETFL, flags);
+		if (r<0) {
+			perror("fcntl(SET) failed");
+			exit(1);
+		}
+	}
+
+#endif // defined(unix)
+
+// Algo benchmark, crash-safe, system-dependent stage
 static double bench_algo_stage2(
 	enum sha256_algos algo
 )