wip4victor

This commit is contained in:
Lex Berezhny 2018-07-01 17:20:17 -04:00
parent c58a1c9ead
commit 6bb81e4d6f
6 changed files with 88 additions and 49 deletions

View file

@ -1,3 +1,4 @@
import asyncio
from orchstr8.testcase import IntegrationTestCase, d2f
from torba.constants import COIN
@ -12,13 +13,18 @@ class BasicTransactionTests(IntegrationTestCase):
self.assertEqual(await self.get_balance(account1), 0)
self.assertEqual(await self.get_balance(account2), 0)
address1 = await d2f(account1.receiving.get_or_create_usable_address())
sendtxid = await self.blockchain.send_to_address(address1.decode(), 5.5)
await self.on_transaction_id(sendtxid) # mempool
sendtxids = []
for i in range(9):
address1 = await d2f(account1.receiving.get_or_create_usable_address())
sendtxid = await self.blockchain.send_to_address(address1.decode(), 1.1)
sendtxids.append(sendtxid)
await self.on_transaction_id(sendtxid) # mempool
await self.blockchain.generate(1)
await self.on_transaction_id(sendtxid) # confirmed
await asyncio.wait([ # confirmed
self.on_transaction_id(txid) for txid in sendtxids
])
self.assertEqual(round(await self.get_balance(account1)/COIN, 1), 5.5)
self.assertEqual(round(await self.get_balance(account1)/COIN, 1), 9.9)
self.assertEqual(round(await self.get_balance(account2)/COIN, 1), 0)
address2 = await d2f(account2.receiving.get_or_create_usable_address())
@ -32,7 +38,5 @@ class BasicTransactionTests(IntegrationTestCase):
await self.blockchain.generate(1)
await self.on_transaction(tx) # confirmed
self.assertTrue(await d2f(self.ledger.is_valid_transaction(tx, 202)))
self.assertEqual(round(await self.get_balance(account1)/COIN, 1), 3.5)
self.assertEqual(round(await self.get_balance(account1)/COIN, 1), 7.9)
self.assertEqual(round(await self.get_balance(account2)/COIN, 1), 2.0)

View file

@ -34,7 +34,13 @@ class BaseHeaders:
@property
def height(self):
return len(self)
return len(self)-1
def hash(self, height=None):
if height is None:
height = self.height
header = self[height]
return self._hash_header(header)
def sync_read_length(self):
return os.path.getsize(self.path) // self.header_size

View file

@ -203,7 +203,7 @@ class BaseLedger(six.with_metaclass(LedgerRegistry)):
def update_headers(self):
while True:
height_sought = len(self.headers)
headers = yield self.network.get_headers(height_sought)
headers = yield self.network.get_headers(height_sought, 2000)
if headers['count'] <= 0:
break
yield self.headers.connect(height_sought, unhexlify(headers['hex']))
@ -266,34 +266,41 @@ class BaseLedger(six.with_metaclass(LedgerRegistry)):
yield lock.acquire()
try:
# see if we have a local copy of transaction, otherwise fetch it from server
raw, local_height, is_verified = yield self.db.get_transaction(unhexlify(hex_id)[::-1])
save_tx = None
if raw is None:
_raw = yield self.network.get_transaction(hex_id)
tx = self.transaction_class(unhexlify(_raw))
save_tx = 'insert'
else:
tx = self.transaction_class(raw)
#try:
# see if we have a local copy of transaction, otherwise fetch it from server
raw, local_height, is_verified = yield self.db.get_transaction(unhexlify(hex_id)[::-1])
save_tx = None
if raw is None:
_raw = yield self.network.get_transaction(hex_id)
tx = self.transaction_class(unhexlify(_raw))
save_tx = 'insert'
else:
tx = self.transaction_class(raw)
if remote_height > 0 and not is_verified:
is_verified = yield self.is_valid_transaction(tx, remote_height)
is_verified = 1 if is_verified else 0
if save_tx is None:
save_tx = 'update'
if remote_height > 0 and not is_verified:
is_verified = yield self.is_valid_transaction(tx, remote_height)
is_verified = 1 if is_verified else 0
if save_tx is None:
save_tx = 'update'
yield self.db.save_transaction_io(
save_tx, tx, remote_height, is_verified, address, self.address_to_hash160(address),
''.join('{}:{}:'.format(tx_id.decode(), tx_height) for tx_id, tx_height in synced_history)
)
yield self.db.save_transaction_io(
save_tx, tx, remote_height, is_verified, address, self.address_to_hash160(address),
''.join('{}:{}:'.format(tx_id.decode(), tx_height) for tx_id, tx_height in synced_history)
)
self._on_transaction_controller.add(TransactionEvent(address, tx, remote_height, is_verified))
log.debug("{}: sync'ed tx {} for address: {}, height: {}, verified: {}".format(
self.get_id(), hex_id, address, remote_height, is_verified
))
self._on_transaction_controller.add(TransactionEvent(address, tx, remote_height, is_verified))
finally:
lock.release()
if not lock.locked:
del self._transaction_processing_locks[hex_id]
# except:
# log.exception('Failed to synchronize transaction:')
# raise
#
# finally:
lock.release()
if not lock.locked:
del self._transaction_processing_locks[hex_id]
@defer.inlineCallbacks
def subscribe_history(self, address):

View file

@ -11,7 +11,10 @@ from twisted.protocols.basic import LineOnlyReceiver
from torba import __version__
from torba.stream import StreamController
log = logging.getLogger()
log = logging.getLogger(__name__)
if six.PY3:
buffer = memoryview
def unicode2bytes(string):
@ -23,6 +26,8 @@ def unicode2bytes(string):
def bytes2unicode(maybe_bytes):
if isinstance(maybe_bytes, buffer):
maybe_bytes = str(maybe_bytes)
if isinstance(maybe_bytes, bytes):
return maybe_bytes.decode()
elif isinstance(maybe_bytes, (list, tuple)):
@ -32,7 +37,7 @@ def bytes2unicode(maybe_bytes):
class StratumClientProtocol(LineOnlyReceiver):
delimiter = b'\n'
MAX_LENGTH = 100000
MAX_LENGTH = 2000000
def __init__(self):
self.request_id = 0
@ -78,6 +83,7 @@ class StratumClientProtocol(LineOnlyReceiver):
self.on_disconnected_controller.add(True)
def lineReceived(self, line):
log.debug('received: {}'.format(line))
try:
# `line` comes in as a byte string but `json.loads` automatically converts everything to
@ -114,6 +120,7 @@ class StratumClientProtocol(LineOnlyReceiver):
'method': method,
'params': [bytes2unicode(arg) for arg in args]
})
log.debug('sent: {}'.format(message))
self.sendLine(message.encode('latin-1'))
d = self.lookup_table[message_id] = defer.Deferred()
return d
@ -161,17 +168,19 @@ class BaseNetwork:
def start(self):
for server in cycle(self.config['default_servers']):
endpoint = clientFromString(reactor, 'tcp:{}:{}'.format(*server))
log.debug("Attempting connection to SPV wallet server: {}:{}".format(*server))
self.service = ClientService(endpoint, StratumClientFactory(self))
self.service.startService()
try:
self.client = yield self.service.whenConnected(failAfterFailures=2)
yield self.ensure_server_version()
log.info("Successfully connected to SPV wallet server: {}:{}".format(*server))
self._on_connected_controller.add(True)
yield self.client.on_disconnected.first
except CancelledError:
return
except Exception as e:
pass
except Exception:
log.exception("Connecting to {}:{} raised an exception:".format(*server))
finally:
self.client = None
if not self.running:

View file

@ -15,14 +15,13 @@ class WalletManager(object):
@classmethod
def from_config(cls, config): # type: (Dict) -> WalletManager
wallets = []
manager = cls(wallets)
for coin_id, ledger_config in config.get('ledgers', {}).items():
manager.get_or_create_ledger(coin_id, ledger_config)
manager = cls()
for ledger_id, ledger_config in config.get('ledgers', {}).items():
manager.get_or_create_ledger(ledger_id, ledger_config)
for wallet_path in config.get('wallets', []):
wallet_storage = WalletStorage(wallet_path)
wallet = Wallet.from_storage(wallet_storage, manager)
wallets.append(wallet)
manager.wallets.append(wallet)
return manager
def get_or_create_ledger(self, ledger_id, ledger_config=None):

View file

@ -59,12 +59,11 @@ class Wallet:
class WalletStorage:
LATEST_VERSION = 2
LATEST_VERSION = 1
DEFAULT = {
'version': LATEST_VERSION,
'name': 'Wallet',
'coins': {},
'accounts': []
}
@ -100,10 +99,25 @@ class WalletStorage:
version = json_dict.pop('version', -1)
if version == 1: # upgrade from version 1 to version 2
_rename_property('addr_history', 'history')
_rename_property('use_encryption', 'encrypted')
_rename_property('gap_limit', 'gap_limit_for_receiving')
if version == -1: # upgrading from electrum wallet
json_dict = {
'accounts': [{
'ledger': 'lbc_mainnet',
'encrypted': json_dict['use_encryption'],
'seed': json_dict['seed'],
'seed_version': json_dict['seed_version'],
'private_key': json_dict['master_private_keys']['x/'],
'public_key': json_dict['master_public_keys']['x/'],
'certificates': json_dict['claim_certificates'],
'receiving_gap': 20,
'change_gap': 6,
'receiving_maximum_use_per_address': 2,
'change_maximum_use_per_address': 2
}]
}
elif version == 1: # upgrade from version 1 to version 2
pass
upgraded = cls.DEFAULT
upgraded.update(json_dict)