py-blockchain-cli-master.zip

import json import socket import socketserver import multiprocessing from blockchain.chain import Chain class _PeerRequestHandler(socketserver.BaseRequestHandler): def handle(self): message_str = self.request.recv(655350).strip().decode('utf-8') message_obj = json.loads(message_str) message_type = message_obj['type'] response = 'OK' peer = self.server.peer if message_type == 'MINE': peer.mine(message_obj['data']) elif message_type == 'CONNECT': host = message_obj['host'] port = message_obj['port'] peer.connect_to_peer(host, port) elif message_type == 'PEERS': response = json.dumps(peer.peers) elif message_type == 'SHOW': response = json.dumps(peer.chain.to_dict()) elif message_type == 'CHAIN': chain = message_obj['chain'] peer.replace_chain(chain) self.request.sendall(response.encode('utf-8')) class Peer(object): def __init__(self, host='127.0.0.1', port=5000): self.host = host self.port = port self._peers = set() self._chain = Chain() def start(self): server = socketserver.ThreadingTCPServer( (self.host, self.port), _PeerRequestHandler) server.peer = self try: server.serve_forever() except KeyboardInterrupt as _: server.server_close() def connect_to_peer(self, host, port): if (host, port) in self._peers: return self._peers.add((host, port)) peers = self._request_peers(host, port) self._add_peers(json.loads(peers)) self._request_connection() self._broadcast_chain() def mine(self, data): self._chain.mine(data) self._broadcast_chain() def replace_chain(self, chain): self._chain.replace_chain(chain) @property def chain(self): return self._chain @property def peers(self): return [{'host': host, 'port': port} for (host, port) in self._peers] def _add_peers(self, peers): for peer in peers: host = peer['host'] port = peer['port'] if host == self.host and port == self.port: continue if (host, port) in self._peers: continue self._peers.add((host, port)) # Communication def _request_connection(self): message = {'type': 'CONNECT', 'host': self.host, 'port': self.port} return self._broadcast(message) def _request_peers(self, host, port): message = {'type': 'PEERS', 'host': self.host, 'port': self.port} return self._unicast(host, port, message) def _broadcast_chain(self): message = {'type': 'CHAIN', 'chain': self._chain.to_dict()} return self._broadcast(message) # Base communication def _unicast(self, host, port, message): pool = multiprocessing.Pool(1) result = pool.apply_async( self._send_message, args=(host, port, message)) pool.close() pool.join() return result.get() def _broadcast(self, message): results = [] pool = multiprocessing.Pool(5) for (host, port) in self._peers: results.append(pool.apply_async( self._send_message, args=(host, port, message))) pool.close() pool.join() return [result.get() for result in results] def _send_message(self, host, port, message): with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s: s.connect((host, port)) s.sendall(json.dumps(message).encode('utf-8')) response = s.recv(655350, 0) return response.decode('utf-8')