diff --git a/lib/coins.py b/lib/coins.py index 4b0551b..cff766c 100644 --- a/lib/coins.py +++ b/lib/coins.py @@ -572,6 +572,8 @@ class DogecoinTestnet(Dogecoin): # Source: https://github.com/dashpay/dash class Dash(Coin): + from server.session import DashElectrumX + from server.daemon import DashDaemon NAME = "Dash" SHORTNAME = "DASH" NET = "mainnet" @@ -596,6 +598,8 @@ class Dash(Coin): 'electrum.dash.siampm.com s t', 'wl4sfwq2hwxnodof.onion s t', ] + SESSIONCLS = DashElectrumX + DAEMON = DashDaemon @classmethod def header_hash(cls, header): diff --git a/server/daemon.py b/server/daemon.py index 39109cc..36f6b14 100644 --- a/server/daemon.py +++ b/server/daemon.py @@ -247,3 +247,12 @@ class Daemon(util.LoggedClass): If the daemon has not been queried yet this returns None.''' return self._height + +class DashDaemon(Daemon): + async def masternode_broadcast(self, params): + '''Broadcast a transaction to the network.''' + return await self._send_single('masternodebroadcast', params) + + async def masternode_list(self, params ): + '''Return the masternode status.''' + return await self._send_single('masternodelist', params) diff --git a/server/session.py b/server/session.py index b5ab239..a7e1701 100644 --- a/server/session.py +++ b/server/session.py @@ -387,3 +387,63 @@ class LocalRPC(SessionBase): def request_handler(self, method): '''Return the async handler for the given request method.''' return self.controller.rpc_handlers.get(method) + + +class DashElectrumX(ElectrumX): + '''A TCP server that handles incoming Electrum Dash connections.''' + + def __init__(self, *args, **kwargs): + super().__init__(*args, **kwargs) + self.electrumx_handlers['masternode.announce.broadcast'] = self.masternode_announce_broadcast + self.electrumx_handlers['masternode.subscribe'] = self.masternode_subscribe + self.mns = set() + + async def notify(self, height, touched): + '''Notify the client about changes in masternode list.''' + + await super().notify(height, touched) + + for masternode in self.mns: + status = await self.daemon.masternode_list(['status', masternode]) + payload = { + 'id': None, + 'method': 'masternode.subscribe', + 'params': [masternode], + 'result': status.get(masternode), + } + self.send_binary(self.encode_payload(payload)) + + def server_version(self, client_name=None, protocol_version=None): + '''Returns the server version as a string. + Force version string response for Electrum-Dash 2.6.4 client caused by + https://github.com/dashpay/electrum-dash/commit/638cf6c0aeb7be14a85ad98f873791cb7b49ee29 + ''' + + default_return = super().server_version(client_name, protocol_version) + if self.client == '2.6.4': + return '1.0' + return default_return + + # Masternode command handlers + async def masternode_announce_broadcast(self, signmnb): + '''Pass through the masternode announce message to be broadcast by the daemon.''' + + try: + mnb_info = await self.daemon.masternode_broadcast(['relay', signmnb]) + return mnb_info + except DaemonError as e: + error = e.args[0] + message = error['message'] + self.log_info('masternode_broadcast: {}'.format(message)) + return ( + 'The masternode broadcast was rejected. ({})\n[{}]' + .format(message, signmnb) + ) + + async def masternode_subscribe(self, vin): + '''Returns the status of masternode.''' + result = await self.daemon.masternode_list(['status', vin]) + if result is not None: + self.mns.add(vin) + return result.get(vin) + return None