highperfocused/Plutus
1
0
Fork 0
mirror of https://github.com/Isaacdelly/Plutus.git synced 2025-10-10 13:03:30 +02:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #255 from p1r473/master

Browse Source 
Fix for Windows and add ability to specify CPU count
This commit is contained in:
Isaacdelly
2022-12-02 16:46:33 -08:00
committed by GitHub
parent 9e39d6a41f c8193a4f8c
commit b87eed5905
1 changed files with 16 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
plutus.py
View File

@@ -2,8 +2,8 @@
# Made by Isaac Delly # Made by Isaac Delly
# https://github.com/Isaacdelly/Plutus # https://github.com/Isaacdelly/Plutus
from fastecdsa import keys, curve #from fastecdsa import keys, curve
from ellipticcurve.privateKey import PrivateKey from fastecdsa import keys, curve #Comment this out on Windows
import platform import platform
import multiprocessing import multiprocessing
import hashlib import hashlib
@@ -22,8 +22,8 @@ def private_key_to_public_key(private_key, fastecdsa):
key = keys.get_public_key(int('0x' + private_key, 0), curve.secp256k1) key = keys.get_public_key(int('0x' + private_key, 0), curve.secp256k1)
return '04' + (hex(key.x)[2:] + hex(key.y)[2:]).zfill(128) return '04' + (hex(key.x)[2:] + hex(key.y)[2:]).zfill(128)
else: else:
pk = PrivateKey().fromString(private_key) pk = PrivateKey().fromString(bytes.fromhex(private_key))
return '04' + pk.publicKey().toString().upper() return '04' + pk.publicKey().toString().hex().upper()
def public_key_to_address(public_key): def public_key_to_address(public_key):
output = [] output = []
@@ -109,15 +109,23 @@ if __name__ == '__main__':
args = { args = {
'verbose': 0, 'verbose': 0,
'substring': 8, 'substring': 8,
'fastecdsa': False 'fastecdsa': False,
'cpu_count': multiprocessing.cpu_count(),
} }
if platform.system() in ['Linux', 'Darwin']: if platform.system() in ['Linux', 'Darwin']:
args['fastecdsa'] = True args['fastecdsa'] = True
for arg in sys.argv[1:]: for arg in sys.argv[1:]:
command = arg.split('=')[0] command = arg.split('=')[0]
if command == 'help': if command == 'help':
print_help() print_help()
elif command == 'cpu_count':
cpu_count = int(arg.split('=')[1])
if cpu_count > 0 and cpu_count <= multiprocessing.cpu_count():
args['cpu_count'] = cpu_count
else:
args['cpu_count']=multiprocessing.cpu_count()
print('invalid input. cpu_count must be greater than 0 and less than your CPU count')
sys.exit(-1)
elif command == 'time': elif command == 'time':
timer(args) timer(args)
elif command == 'verbose': elif command == 'verbose':
@@ -148,7 +156,7 @@ if __name__ == '__main__':
print('DONE') print('DONE')
print('database size: ' + str(len(database))) print('database size: ' + str(len(database)))
print('processes spawned: ' + str(multiprocessing.cpu_count())) print('processes spawned: ' + str(args['cpu_count']))
for cpu in range(multiprocessing.cpu_count()): for cpu in range(args['cpu_count']):
multiprocessing.Process(target = main, args = (database, args)).start() multiprocessing.Process(target = main, args = (database, args)).start()