PyBitmessage/src/proofofwork.py

78 lines
2.5 KiB
Python
Raw Normal View History

2013-06-04 05:14:24 +02:00
#import shared
#import time
#from multiprocessing import Pool, cpu_count
2013-05-30 22:25:42 +02:00
import hashlib
from struct import unpack, pack
import sys
from shared import config
2013-06-04 05:14:24 +02:00
#import os
2013-06-03 07:04:22 +02:00
def _set_idle():
if 'linux' in sys.platform:
2013-06-18 18:56:03 +02:00
import os
os.nice(20) # @UndefinedVariable
else:
2013-06-18 18:56:03 +02:00
try:
sys.getwindowsversion()
import win32api,win32process,win32con # @UnresolvedImport
2013-06-18 18:56:03 +02:00
pid = win32api.GetCurrentProcessId()
handle = win32api.OpenProcess(win32con.PROCESS_ALL_ACCESS, True, pid)
win32process.SetPriorityClass(handle, win32process.IDLE_PRIORITY_CLASS)
except:
#Windows 64-bit
pass
2013-05-30 22:25:42 +02:00
def _pool_worker(nonce, initialHash, target, pool_size):
2013-06-03 07:04:22 +02:00
_set_idle()
trialValue = 99999999999999999999
while trialValue > target:
2013-05-30 22:25:42 +02:00
nonce += pool_size
trialValue, = unpack('>Q',hashlib.sha512(hashlib.sha512(pack('>Q',nonce) + initialHash).digest()).digest()[0:8])
return [trialValue, nonce]
def _doSafePoW(target, initialHash):
2013-06-04 05:14:24 +02:00
nonce = 0
trialValue = 99999999999999999999
while trialValue > target:
nonce += 1
trialValue, = unpack('>Q',hashlib.sha512(hashlib.sha512(pack('>Q',nonce) + initialHash).digest()).digest()[0:8])
return [trialValue, nonce]
def _doFastPoW(target, initialHash):
import shared
import time
from multiprocessing import Pool, cpu_count
try:
2013-05-30 22:25:42 +02:00
pool_size = cpu_count()
except:
2013-05-30 22:25:42 +02:00
pool_size = 4
try:
maxCores = config.getint('bitmessagesettings', 'maxcores')
except:
maxCores = 99999
if pool_size > maxCores:
pool_size = maxCores
pool = Pool(processes=pool_size)
result = []
for i in range(pool_size):
2013-05-30 22:25:42 +02:00
result.append(pool.apply_async(_pool_worker, args = (i, initialHash, target, pool_size)))
while True:
2013-05-30 22:42:24 +02:00
if shared.shutdown:
pool.terminate()
while True:
time.sleep(10) # Don't let this thread return here; it will return nothing and cause an exception in bitmessagemain.py
2013-05-30 22:42:24 +02:00
return
2013-05-30 22:25:42 +02:00
for i in range(pool_size):
if result[i].ready():
result = result[i].get()
pool.terminate()
pool.join() #Wait for the workers to exit...
2013-05-30 22:25:42 +02:00
return result[0], result[1]
time.sleep(0.2)
def run(target, initialHash):
if 'linux' in sys.platform:
2013-06-18 18:56:03 +02:00
return _doFastPoW(target, initialHash)
else:
2013-06-18 18:56:03 +02:00
return _doSafePoW(target, initialHash)