BTCPrivate
/
electrum-bitcoinprivate
mirror of https://github.com/BTCPrivate/electrum-bitcoinprivate.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

spv: keep track of header requests made by each interface

This commit is contained in:
ThomasV 2017-04-20 10:59:30 +02:00
parent 853b6169ed
commit 6afc11208d
1 changed files with 19 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
lib/network.py
View File

@ -205,8 +205,6 @@ class Network(util.DaemonThread):
self.config = SimpleConfig(config) if type(config) == type({}) else config
self.num_server = 8 if not self.config.get('oneserver') else 0
self.blockchain = Blockchain(self.config, self)
# A deque of interface header requests, processed left-to-right
self.bc_requests = deque()
# Server for addresses and transactions
self.default_server = self.config.get('server')
# Sanitize default server
@ -692,8 +690,11 @@ class Network(util.DaemonThread):
def new_interface(self, server, socket):
self.add_recent_server(server)
self.interfaces[server] = interface = Interface(server, socket)
interface = Interface(server, socket)
# A deque of interface header requests, processed left-to-right
interface.bc_requests = deque()
interface.failed_checkpoint = False
self.interfaces[server] = interface
self.queue_request('blockchain.block.get_header', [self.blockchain.checkpoint_height], interface)
self.queue_request('blockchain.headers.subscribe', [], interface)
if server == self.default_server:
@ -753,15 +754,15 @@ class Network(util.DaemonThread):
if response.get('error'):
interface.print_error(response.get('error'))
return
if self.bc_requests:
req_if, data = self.bc_requests[0]
if interface.bc_requests:
data = interface.bc_requests[0]
req_idx = data.get('chunk_idx')
# Ignore unsolicited chunks
if req_if == interface and req_idx == response['params'][0]:
if req_idx == response['params'][0]:
idx = self.blockchain.connect_chunk(req_idx, response['result'])
# If not finished, get the next chunk
if idx < 0 or self.get_local_height() >= data['if_height']:
self.bc_requests.popleft()
interface.bc_requests.popleft()
self.notify('updated')
else:
self.request_chunk(interface, data, idx)
@ -787,18 +788,18 @@ class Network(util.DaemonThread):
return
if self.blockchain.downloading_headers:
return
if self.bc_requests:
req_if, data = self.bc_requests[0]
if interface.bc_requests:
data = interface.bc_requests[0]
req_height = data.get('header_height', -1)
# Ignore unsolicited headers
if req_if == interface and req_height == response['params'][0]:
if req_height == response['params'][0]:
if interface.failed_checkpoint:
self.bc_requests.popleft()
interface.bc_requests.popleft()
return
next_height = self.blockchain.connect_header(data['chain'], response['result'])
# If not finished, get the next header
if next_height in [True, False]:
self.bc_requests.popleft()
interface.bc_requests.popleft()
if next_height:
self.switch_lagging_interface(interface.server)
self.notify('updated')
@ -827,8 +828,10 @@ class Network(util.DaemonThread):
'''Work through each interface that has notified us of a new header.
Send it requests if it is ahead of our blockchain object.
'''
while self.bc_requests:
interface, data = self.bc_requests.popleft()
for interface in self.interfaces.values():
if not interface.bc_requests:
continue
data = interface.bc_requests.popleft()
# If the connection was lost move on
if not interface in self.interfaces.values():
continue
@ -843,7 +846,7 @@ class Network(util.DaemonThread):
self.connection_down(interface.server)
continue
# Put updated request state back at head of deque
self.bc_requests.appendleft((interface, data))
interface.bc_requests.appendleft(data)
break
def wait_on_sockets(self):
@ -885,7 +888,7 @@ class Network(util.DaemonThread):
self.headers[i.server] = header
# Queue this interface's height for asynchronous catch-up
self.bc_requests.append((i, {'if_height': height}))
i.bc_requests.append({'if_height': height})
if i == self.interface:
self.switch_lagging_interface()