50 lines
1.3 KiB
Python
50 lines
1.3 KiB
Python
|
import ctypes
|
||
|
|
import hashlib
|
||
|
|
from multiprocessing import current_process
|
||
|
|
import os
|
||
|
|
import signal
|
||
|
|
from struct import unpack, pack
|
||
|
|
from threading import current_thread
|
||
|
|
|
||
|
|
shutdown = 0
|
||
|
|
|
||
|
|
|
||
|
|
def signal_handler(signal, frame):
|
||
|
|
global shutdown
|
||
|
|
print "Got signal %i in %s/%s" % (signal, current_process().name, current_thread().name)
|
||
|
|
if current_process().name != "MainProcess":
|
||
|
|
raise StopIteration("Interrupted")
|
||
|
|
if current_thread().name != "MainThread":
|
||
|
|
return
|
||
|
|
shutdown = 1
|
||
|
|
|
||
|
|
|
||
|
|
def _doCPoW(target, initialHash):
|
||
|
|
# global shutdown
|
||
|
|
h = initialHash
|
||
|
|
m = target
|
||
|
|
out_h = ctypes.pointer(ctypes.create_string_buffer(h, 64))
|
||
|
|
out_m = ctypes.c_ulonglong(m)
|
||
|
|
print "C PoW start"
|
||
|
|
for c in range(0, 200000):
|
||
|
|
print "Iter: %i" % (c)
|
||
|
|
nonce = bmpow(out_h, out_m)
|
||
|
|
if shutdown:
|
||
|
|
break
|
||
|
|
trialValue, = unpack('>Q', hashlib.sha512(hashlib.sha512(pack('>Q', nonce) + initialHash).digest()).digest()[0:8])
|
||
|
|
if shutdown != 0:
|
||
|
|
raise StopIteration("Interrupted")
|
||
|
|
print "C PoW done"
|
||
|
|
return [trialValue, nonce]
|
||
|
|
|
||
|
|
|
||
|
|
signal.signal(signal.SIGINT, signal_handler)
|
||
|
|
signal.signal(signal.SIGTERM, signal_handler)
|
||
|
|
|
||
|
|
bso = ctypes.CDLL(os.path.join("bitmsghash", "bitmsghash.so"))
|
||
|
|
|
||
|
|
bmpow = bso.BitmessagePOW
|
||
|
|
bmpow.restype = ctypes.c_ulonglong
|
||
|
|
|
||
|
|
_doCPoW(2**44, "")
|