def blockchain(self): """Returns the current Blockchain object.""" if self.client: return self.client.blockhandler.block_chain else: blockchain = BlockChain( did_lock_to_index_f=self.bcs.did_lock_to_index) blockchain.preload_locked_blocks(self.bcs.headers()) return blockchain
def __init__(self, network, host_port_q, should_download_block_f, block_chain_store, blockchain_change_callback, server_port=9999): """ network: a value from pycoinnet.helpers.networks host_port_q: a Queue that is being fed potential places to connect should_download_block_f: a function that accepting(block_hash, block_index) and returning a boolean indicating whether that block should be downloaded. Only used during fast-forward. block_chain_store: usually a BlockChainStore instance blockchain_change_callback: a callback that expects (blockchain, list_of_ops) that is invoked whenever the block chain is updated; blockchain is a BlockChain object and list_of_ops is a pair of tuples of the form (op, block_hash, block_index) where op is one of "add" or "remove", block_hash is a binary block hash, and block_index is an integer index number. """ block_chain = BlockChain( did_lock_to_index_f=block_chain_store.did_lock_to_index) block_chain.preload_locked_blocks(block_chain_store.headers()) block_chain.add_change_callback(block_chain_locker_callback) self.blockfetcher = Blockfetcher() self.inv_collector = InvCollector() self.block_store = TwoLevelDict() @asyncio.coroutine def _rotate(block_store): while True: block_store.rotate() yield from asyncio.sleep(1800) self.rotate_task = asyncio.Task(_rotate(self.block_store)) self.blockhandler = BlockHandler( self.inv_collector, block_chain, self.block_store, should_download_f=should_download_block_f) block_chain.add_change_callback(blockchain_change_callback) self.fast_forward_add_peer = fast_forwarder_add_peer_f(block_chain) self.fetcher_task = asyncio.Task( new_block_fetcher(self.inv_collector, block_chain)) self.nonce = int.from_bytes(os.urandom(8), byteorder="big") self.subversion = "/Notoshi/".encode("utf8") @asyncio.coroutine def run_peer(peer, fetcher, fast_forward_add_peer, blockfetcher, inv_collector, blockhandler): yield from asyncio.wait_for(peer.connection_made_future, timeout=None) version_parameters = version_data_for_peer( peer, local_port=(server_port or 0), last_block_index=block_chain.length(), nonce=self.nonce, subversion=self.subversion) version_data = yield from initial_handshake( peer, version_parameters) last_block_index = version_data["last_block_index"] fast_forward_add_peer(peer, last_block_index) blockfetcher.add_peer(peer, fetcher, last_block_index) inv_collector.add_peer(peer) blockhandler.add_peer(peer) def create_protocol_callback(): peer = BitcoinPeerProtocol(network["MAGIC_HEADER"]) install_pingpong_manager(peer) fetcher = Fetcher(peer) peer.add_task( run_peer(peer, fetcher, self.fast_forward_add_peer, self.blockfetcher, self.inv_collector, self.blockhandler)) return peer self.connection_info_q = manage_connection_count( host_port_q, create_protocol_callback, 8) self.show_task = asyncio.Task( show_connection_info(self.connection_info_q)) # listener @asyncio.coroutine def run_listener(): abstract_server = None future_peer = asyncio.Future() try: abstract_server = yield from asyncio.get_event_loop( ).create_server(protocol_factory=create_protocol_callback, port=server_port) return abstract_server except OSError as ex: logging.info("can't listen on port %d", server_port) if server_port: self.server_task = asyncio.Task(run_listener())
def __init__(self, network, host_port_q, should_download_block_f, block_chain_store, blockchain_change_callback, server_port=9999): """ network: a value from pycoinnet.helpers.networks host_port_q: a Queue that is being fed potential places to connect should_download_block_f: a function that accepting(block_hash, block_index) and returning a boolean indicating whether that block should be downloaded. Only used during fast-forward. block_chain_store: usually a BlockChainStore instance blockchain_change_callback: a callback that expects (blockchain, list_of_ops) that is invoked whenever the block chain is updated; blockchain is a BlockChain object and list_of_ops is a pair of tuples of the form (op, block_hash, block_index) where op is one of "add" or "remove", block_hash is a binary block hash, and block_index is an integer index number. """ block_chain = BlockChain(did_lock_to_index_f=block_chain_store.did_lock_to_index) block_chain.preload_locked_blocks(block_chain_store.headers()) block_chain.add_change_callback(block_chain_locker_callback) self.blockfetcher = Blockfetcher() self.inv_collector = InvCollector() self.block_store = TwoLevelDict() @asyncio.coroutine def _rotate(block_store): while True: block_store.rotate() yield from asyncio.sleep(1800) self.rotate_task = asyncio.Task(_rotate(self.block_store)) self.blockhandler = BlockHandler(self.inv_collector, block_chain, self.block_store, should_download_f=should_download_block_f) block_chain.add_change_callback(blockchain_change_callback) self.fast_forward_add_peer = fast_forwarder_add_peer_f(block_chain) self.fetcher_task = asyncio.Task(new_block_fetcher(self.inv_collector, block_chain)) self.nonce = int.from_bytes(os.urandom(8), byteorder="big") self.subversion = "/Notoshi/".encode("utf8") @asyncio.coroutine def run_peer(peer, fetcher, fast_forward_add_peer, blockfetcher, inv_collector, blockhandler): yield from asyncio.wait_for(peer.connection_made_future, timeout=None) version_parameters = version_data_for_peer( peer, local_port=(server_port or 0), last_block_index=block_chain.length(), nonce=self.nonce, subversion=self.subversion) version_data = yield from initial_handshake(peer, version_parameters) last_block_index = version_data["last_block_index"] fast_forward_add_peer(peer, last_block_index) blockfetcher.add_peer(peer, fetcher, last_block_index) inv_collector.add_peer(peer) blockhandler.add_peer(peer) def create_protocol_callback(): peer = BitcoinPeerProtocol(network["MAGIC_HEADER"]) install_pingpong_manager(peer) fetcher = Fetcher(peer) peer.add_task(run_peer( peer, fetcher, self.fast_forward_add_peer, self.blockfetcher, self.inv_collector, self.blockhandler)) return peer self.connection_info_q = manage_connection_count(host_port_q, create_protocol_callback, 8) self.show_task = asyncio.Task(show_connection_info(self.connection_info_q)) # listener @asyncio.coroutine def run_listener(): abstract_server = None try: abstract_server = yield from asyncio.get_event_loop().create_server( protocol_factory=create_protocol_callback, port=server_port) return abstract_server except OSError: logging.info("can't listen on port %d", server_port) if server_port: self.server_task = asyncio.Task(run_listener())