From e5ee7426ebb145efb39d778d4836181e9b9d414c Mon Sep 17 00:00:00 2001
From: mailchuck <dev@mailchuck.com>
Date: Thu, 5 Nov 2015 20:41:41 +0100
Subject: [PATCH] Support for PoW from an external library

This will attempt to include a PoW library (.so on Unix, .dll on
windows) to do PoW. This is done in a safe way and clean fallback to
normal PoW. The code for the library will be in another commit. The code
is take from https://github.com/bm-robertwhite/bitmessage-powfaster,
with minor modifications.

This patch also includes code to make GPU PoW have a safer fallback.
---
 src/proofofwork.py | 51 ++++++++++++++++++++++++++++++++++++++++++----
 1 file changed, 47 insertions(+), 4 deletions(-)

diff --git a/src/proofofwork.py b/src/proofofwork.py
index 81f7e3df..93f6f2cd 100644
--- a/src/proofofwork.py
+++ b/src/proofofwork.py
@@ -7,7 +7,31 @@ import sys
 from shared import config, frozen
 import shared
 import openclpow
-#import os
+import os
+import ctypes
+
+curdir = os.path.dirname(__file__)
+bitmsglib = 'bitmsghash.so'
+if "win32" == sys.platform:
+    if ctypes.sizeof(ctypes.c_voidp) == 4:
+        bitmsglib = 'bitmsghash32.dll'
+    else:
+        bitmsglib = 'bitmsghash64.dll'
+    try:
+        bso = ctypes.WinDLL(os.path.join(curdir, bitmsglib))
+    except:
+        bso = None
+else:
+    try:
+        bso = ctypes.CDLL(os.path.join(curdir, bitmsglib))
+    except:
+        bso = None
+if bso:
+    try:
+        bmpow = bso.BitmessagePOW
+        bmpow.restype = ctypes.c_ulonglong
+    except:
+        bmpow = None
 
 def _set_idle():
     if 'linux' in sys.platform:
@@ -72,12 +96,23 @@ def _doFastPoW(target, initialHash):
                 pool.join() #Wait for the workers to exit...
                 return result[0], result[1]
         time.sleep(0.2)
+def _doCPoW(target, initialHash):
+    h = initialHash
+    m = target
+    out_h = ctypes.pointer(ctypes.create_string_buffer(h, 64))
+    out_m = ctypes.c_ulonglong(m)
+    print "C PoW start"
+    nonce = bmpow(out_h, out_m)
+    trialValue, = unpack('>Q',hashlib.sha512(hashlib.sha512(pack('>Q',nonce) + initialHash).digest()).digest()[0:8])
+    print "C PoW done"
+    return [trialValue, nonce]
 
 def _doGPUPoW(target, initialHash):
-    print "GPU POW\n"
+    print "GPU PoW start"
     nonce = openclpow.do_opencl_pow(initialHash.encode("hex"), target)
     trialValue, = unpack('>Q',hashlib.sha512(hashlib.sha512(pack('>Q',nonce) + initialHash).digest()).digest()[0:8])
     #print "{} - value {} < {}".format(nonce, trialValue, target)
+    print "GPU PoW done"
     return [trialValue, nonce]
 
 def run(target, initialHash):
@@ -88,8 +123,16 @@ def run(target, initialHash):
 #        print "GPU: %s, %s" % (trialvalue1, nonce1)
 #        print "Fast: %s, %s" % (trialvalue, nonce)
 #        return [trialvalue, nonce]
-        return _doGPUPoW(target, initialHash)
-    elif frozen == "macosx_app" or not frozen:
+        try:
+            return _doGPUPoW(target, initialHash)
+        except:
+            pass # fallback to normal PoW
+    if frozen == "macosx_app" or not frozen:
+        if bmpow:
+            try:
+                return _doCPoW(target, initialHash)
+            except:
+                pass # fallback to normal PoW
         return _doFastPoW(target, initialHash)
     else:
         return _doSafePoW(target, initialHash)