From efb7cedde56ccb1e1bfdfda63d54c312dc64cae7 Mon Sep 17 00:00:00 2001
From: samr7 <samr7@cs.washington.edu>
Date: Sun, 31 Jul 2011 20:44:18 -0700
Subject: [PATCH] Warn about NVIDIA drivers that seem to produce broken device
 code. Bump version number.

---
 oclvanitygen.c | 31 ++++++++++++++++++++++++++++++-
 vanitygen.c    |  2 +-
 2 files changed, 31 insertions(+), 2 deletions(-)

diff --git a/oclvanitygen.c b/oclvanitygen.c
index 5d02da0..882ce54 100644
--- a/oclvanitygen.c
+++ b/oclvanitygen.c
@@ -33,7 +33,7 @@
 #include "pattern.h"
 
 
-const char *version = "0.14";
+const char *version = "0.15";
 const int debug = 0;
 
 #define MAX_SLOT 2
@@ -74,6 +74,7 @@ typedef struct _vg_ocl_context_s {
 	int				voc_ocl_invsize;
 	int				voc_halt;
 	int				voc_rekey;
+	int				voc_dump_done;
 } vg_ocl_context_t;
 
 
@@ -259,6 +260,8 @@ vg_ocl_dump_info(vg_ocl_context_t *vocp)
 	cl_device_id did;
 	if (vocp->base.vxc_vc && (vocp->base.vxc_vc->vc_verbose < 1))
 		return;
+	if (vocp->voc_dump_done)
+		return;
 	did = vocp->voc_ocldid;
 	printf("Device: %s\n",
 	       vg_ocl_device_getstr(did, CL_DEVICE_NAME));
@@ -278,6 +281,7 @@ vg_ocl_dump_info(vg_ocl_context_t *vocp)
 	       vg_ocl_device_getulong(did, CL_DEVICE_GLOBAL_MEM_SIZE));
 	printf("Max allocation: %ld\n",
 	       vg_ocl_device_getulong(did, CL_DEVICE_MAX_MEM_ALLOC_SIZE));
+	vocp->voc_dump_done = 1;
 }
 
 void
@@ -354,6 +358,22 @@ vg_ocl_buildlog(vg_ocl_context_t *vocp, cl_program prog)
  * OpenCL per-exec functions
  */
 
+int
+vg_ocl_check_driver(vg_ocl_context_t *vocp)
+{
+#ifdef WIN32
+	if (!strcmp(vg_ocl_device_getstr(vocp->voc_ocldid, CL_DEVICE_VENDOR),
+		    "NVIDIA Corporation") &&
+            strcmp(vg_ocl_device_getstr(vocp->voc_ocldid, CL_DRIVER_VERSION),
+                   "270.81")) {
+		printf("WARNING: Known problems with certain NVIDIA driver versions\n");
+		printf("WARNING: Use version 270.81 for best results\n");
+		return 0;
+	}
+#endif
+	return 1;
+}
+
 int
 vg_ocl_create_kernel(vg_ocl_context_t *vocp, int knum, const char *func)
 {
@@ -605,6 +625,15 @@ vg_ocl_init(vg_context_t *vcp, vg_ocl_context_t *vocp, cl_device_id did)
 	if (vcp->vc_verbose > 1)
 		vg_ocl_dump_info(vocp);
 
+	if (!vg_ocl_check_driver(vocp)) {
+		char yesbuf[16];
+		printf("Type 'yes' to continue: ");
+		fflush(stdout);
+		if (!fgets(yesbuf, sizeof(yesbuf), stdin) ||
+		    strncmp(yesbuf, "yes", 3))
+			exit(1);
+	}
+
 	vocp->voc_oclctx = clCreateContext(NULL,
 					   1, &did,
 					   vg_ocl_context_callback,
diff --git a/vanitygen.c b/vanitygen.c
index 8304057..632c3c0 100644
--- a/vanitygen.c
+++ b/vanitygen.c
@@ -31,7 +31,7 @@
 
 #include "pattern.h"
 
-const char *version = "0.14";
+const char *version = "0.15";
 
 typedef struct _vg_thread_context_s {
 	vg_exec_context_t		base;