mirror of
https://github.com/basicswap/basicswap.git
synced 2025-11-05 18:38:09 +01:00
net: Add network portals to allow swaps between networks.
This commit is contained in:
@@ -31,6 +31,7 @@ from .util import (
|
||||
)
|
||||
from .util.logging import (
|
||||
BSXLogger,
|
||||
LogCategories as LC,
|
||||
)
|
||||
from .chainparams import (
|
||||
Coins,
|
||||
@@ -43,7 +44,7 @@ def getaddrinfo_tor(*args):
|
||||
|
||||
|
||||
class BaseApp(DBMethods):
|
||||
def __init__(self, data_dir, settings, chain, log_name="BasicSwap"):
|
||||
def __init__(self, data_dir, settings, chain, log_name="BasicSwap", **kwargs):
|
||||
self.fp = None
|
||||
self.log_name = log_name
|
||||
self.fail_code = 0
|
||||
@@ -73,6 +74,24 @@ class BaseApp(DBMethods):
|
||||
self.default_socket_getaddrinfo = socket.getaddrinfo
|
||||
self._force_db_upgrade = False
|
||||
|
||||
self._enabled_log_categories = set()
|
||||
for category in self.settings.get("enabled_log_categories", []):
|
||||
if category == "net":
|
||||
self._enabled_log_categories.add(LC.NET)
|
||||
else:
|
||||
self.log.warning(f"Unknown entry \"{category}\" in \"enabled_log_categories\"")
|
||||
|
||||
if len(self._enabled_log_categories) > 0:
|
||||
self.log.info("Enabled logging categories: {}".format(",".join(sorted([c.name for c in self._enabled_log_categories]))))
|
||||
|
||||
super().__init__(
|
||||
data_dir=data_dir,
|
||||
settings=settings,
|
||||
chain=chain,
|
||||
log_name=log_name,
|
||||
**kwargs,
|
||||
)
|
||||
|
||||
def __del__(self):
|
||||
if self.fp:
|
||||
self.fp.close()
|
||||
@@ -236,11 +255,16 @@ class BaseApp(DBMethods):
|
||||
request = urllib.request.Request(url, headers=headers)
|
||||
return opener.open(request, timeout=timeout).read()
|
||||
|
||||
def logException(self, message) -> None:
|
||||
def logException(self, message: str) -> None:
|
||||
self.log.error(message)
|
||||
if self.debug:
|
||||
self.log.error(traceback.format_exc())
|
||||
|
||||
def logD(self, log_category: int, message: str) -> None:
|
||||
if log_category not in self._enabled_log_categories:
|
||||
return
|
||||
self.log.debug("(" + LC(log_category).name + ") " + message)
|
||||
|
||||
def torControl(self, query):
|
||||
try:
|
||||
command = 'AUTHENTICATE "{}"\r\n{}\r\nQUIT\r\n'.format(
|
||||
|
||||
File diff suppressed because it is too large
Load Diff
@@ -41,6 +41,11 @@ class MessageNetworks(IntEnum):
|
||||
SIMPLEX = auto()
|
||||
|
||||
|
||||
class MessageNetworkLinkTypes(IntEnum):
|
||||
RECEIVED_ON = auto()
|
||||
SENT_ON = auto()
|
||||
|
||||
|
||||
class MessageTypes(IntEnum):
|
||||
OFFER = auto()
|
||||
BID = auto()
|
||||
@@ -59,6 +64,8 @@ class MessageTypes(IntEnum):
|
||||
ADS_BID_ACCEPT_FL = auto()
|
||||
|
||||
CONNECT_REQ = auto()
|
||||
PORTAL_OFFER = auto()
|
||||
PORTAL_SEND = auto()
|
||||
|
||||
|
||||
class AddressTypes(IntEnum):
|
||||
@@ -66,6 +73,8 @@ class AddressTypes(IntEnum):
|
||||
BID = auto()
|
||||
RECV_OFFER = auto()
|
||||
SEND_OFFER = auto()
|
||||
PORTAL_LOCAL = auto()
|
||||
PORTAL = auto()
|
||||
|
||||
|
||||
class SwapTypes(IntEnum):
|
||||
@@ -395,15 +404,14 @@ def strTxType(tx_type):
|
||||
|
||||
|
||||
def strAddressType(addr_type):
|
||||
if addr_type == AddressTypes.OFFER:
|
||||
return "Offer"
|
||||
if addr_type == AddressTypes.BID:
|
||||
return "Bid"
|
||||
if addr_type == AddressTypes.RECV_OFFER:
|
||||
return "Offer recv"
|
||||
if addr_type == AddressTypes.SEND_OFFER:
|
||||
return "Offer send"
|
||||
return "Unknown"
|
||||
return {
|
||||
AddressTypes.OFFER: "Offer",
|
||||
AddressTypes.BID: "Bid",
|
||||
AddressTypes.RECV_OFFER: "Offer recv",
|
||||
AddressTypes.SEND_OFFER: "Offer send",
|
||||
AddressTypes.PORTAL_LOCAL: "Portal (local)",
|
||||
AddressTypes.PORTAL: "Portal",
|
||||
}.get(addr_type, "Unknown")
|
||||
|
||||
|
||||
def getLockName(lock_type):
|
||||
|
||||
@@ -415,7 +415,7 @@ def runClient(
|
||||
for network in settings.get("networks", []):
|
||||
if network.get("enabled", True) is False:
|
||||
continue
|
||||
network_type = network.get("type", "unknown")
|
||||
network_type: str = network.get("type", "unknown")
|
||||
if network_type == "simplex":
|
||||
simplex_dir = os.path.join(data_dir, "simplex")
|
||||
|
||||
|
||||
@@ -13,7 +13,7 @@ from enum import IntEnum, auto
|
||||
from typing import Optional
|
||||
|
||||
|
||||
CURRENT_DB_VERSION = 29
|
||||
CURRENT_DB_VERSION = 30
|
||||
CURRENT_DB_DATA_VERSION = 6
|
||||
|
||||
|
||||
@@ -185,6 +185,7 @@ class Offer(Table):
|
||||
amount_negotiable = Column("bool")
|
||||
rate_negotiable = Column("bool")
|
||||
auto_accept_type = Column("integer")
|
||||
message_nets = Column("string")
|
||||
|
||||
# Local fields
|
||||
auto_accept_bids = Column("bool")
|
||||
@@ -233,6 +234,7 @@ class Bid(Table):
|
||||
rate = Column("integer")
|
||||
|
||||
pkhash_seller = Column("blob")
|
||||
message_nets = Column("string")
|
||||
|
||||
initiate_txn_redeem = Column("blob")
|
||||
initiate_txn_refund = Column("blob")
|
||||
@@ -381,6 +383,8 @@ class SmsgAddress(Table):
|
||||
use_type = Column("integer")
|
||||
note = Column("string")
|
||||
|
||||
index = Index("smsgaddresses_address_index", "addr")
|
||||
|
||||
|
||||
class Action(Table):
|
||||
__tablename__ = "actions"
|
||||
@@ -676,6 +680,20 @@ class MessageNetworks(Table):
|
||||
created_at = Column("integer")
|
||||
|
||||
|
||||
class MessageNetworkLink(Table):
|
||||
__tablename__ = "message_network_links"
|
||||
|
||||
record_id = Column("integer", primary_key=True, autoincrement=True)
|
||||
active_ind = Column("integer")
|
||||
|
||||
linked_type = Column("integer")
|
||||
linked_id = Column("blob")
|
||||
|
||||
network_id = Column("string")
|
||||
link_type = Column("integer") # MessageNetworkLinkTypes
|
||||
created_at = Column("integer")
|
||||
|
||||
|
||||
class DirectMessageRoute(Table):
|
||||
__tablename__ = "direct_message_routes"
|
||||
|
||||
@@ -694,6 +712,7 @@ class DirectMessageRoute(Table):
|
||||
|
||||
class DirectMessageRouteLink(Table):
|
||||
__tablename__ = "direct_message_route_links"
|
||||
|
||||
record_id = Column("integer", primary_key=True, autoincrement=True)
|
||||
active_ind = Column("integer")
|
||||
direct_message_route_id = Column("integer")
|
||||
|
||||
@@ -1,6 +1,6 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
# Copyright (c) 2023-2024 The Basicswap Developers
|
||||
# Copyright (c) 2023-2025 The Basicswap Developers
|
||||
# Distributed under the MIT software license, see the accompanying
|
||||
# file LICENSE or http://www.opensource.org/licenses/mit-license.php.
|
||||
|
||||
@@ -26,97 +26,46 @@ def remove_expired_data(self, time_offset: int = 0):
|
||||
)
|
||||
for offer_row in offer_rows:
|
||||
num_offers += 1
|
||||
offer_query_data = {
|
||||
"type_ind": int(Concepts.OFFER),
|
||||
"offer_id": offer_row[0],
|
||||
}
|
||||
bid_rows = cursor.execute(
|
||||
"SELECT bids.bid_id FROM bids WHERE bids.offer_id = :offer_id",
|
||||
{"offer_id": offer_row[0]},
|
||||
offer_query_data,
|
||||
)
|
||||
for bid_row in bid_rows:
|
||||
num_bids += 1
|
||||
cursor.execute(
|
||||
bid_query_data = {"type_ind": int(Concepts.BID), "bid_id": bid_row[0]}
|
||||
for query_str in [
|
||||
"DELETE FROM transactions WHERE transactions.bid_id = :bid_id",
|
||||
{"bid_id": bid_row[0]},
|
||||
)
|
||||
cursor.execute(
|
||||
"DELETE FROM eventlog WHERE eventlog.linked_type = :type_ind AND eventlog.linked_id = :bid_id",
|
||||
{"type_ind": int(Concepts.BID), "bid_id": bid_row[0]},
|
||||
)
|
||||
cursor.execute(
|
||||
"DELETE FROM automationlinks WHERE automationlinks.linked_type = :type_ind AND automationlinks.linked_id = :bid_id",
|
||||
{"type_ind": int(Concepts.BID), "bid_id": bid_row[0]},
|
||||
)
|
||||
cursor.execute(
|
||||
"DELETE FROM prefunded_transactions WHERE prefunded_transactions.linked_type = :type_ind AND prefunded_transactions.linked_id = :bid_id",
|
||||
{"type_ind": int(Concepts.BID), "bid_id": bid_row[0]},
|
||||
)
|
||||
cursor.execute(
|
||||
"DELETE FROM history WHERE history.concept_type = :type_ind AND history.concept_id = :bid_id",
|
||||
{"type_ind": int(Concepts.BID), "bid_id": bid_row[0]},
|
||||
)
|
||||
cursor.execute(
|
||||
"DELETE FROM xmr_swaps WHERE xmr_swaps.bid_id = :bid_id",
|
||||
{"bid_id": bid_row[0]},
|
||||
)
|
||||
cursor.execute(
|
||||
"DELETE FROM actions WHERE actions.linked_id = :bid_id",
|
||||
{"bid_id": bid_row[0]},
|
||||
)
|
||||
cursor.execute(
|
||||
"DELETE FROM addresspool WHERE addresspool.bid_id = :bid_id",
|
||||
{"bid_id": bid_row[0]},
|
||||
)
|
||||
cursor.execute(
|
||||
"DELETE FROM xmr_split_data WHERE xmr_split_data.bid_id = :bid_id",
|
||||
{"bid_id": bid_row[0]},
|
||||
)
|
||||
cursor.execute(
|
||||
"DELETE FROM bids WHERE bids.bid_id = :bid_id",
|
||||
{"bid_id": bid_row[0]},
|
||||
)
|
||||
cursor.execute(
|
||||
"DELETE FROM message_links WHERE linked_type = :type_ind AND linked_id = :linked_id",
|
||||
{"type_ind": int(Concepts.BID), "linked_id": bid_row[0]},
|
||||
)
|
||||
cursor.execute(
|
||||
"DELETE FROM direct_message_route_links WHERE linked_type = :type_ind AND linked_id = :linked_id",
|
||||
{"type_ind": int(Concepts.BID), "linked_id": bid_row[0]},
|
||||
)
|
||||
|
||||
cursor.execute(
|
||||
"DELETE FROM message_links WHERE linked_type = :type_ind AND linked_id = :bid_id",
|
||||
"DELETE FROM direct_message_route_links WHERE linked_type = :type_ind AND linked_id = :bid_id",
|
||||
"DELETE FROM message_network_links WHERE linked_type = :type_ind AND linked_id = :bid_id",
|
||||
]:
|
||||
cursor.execute(query_str, bid_query_data)
|
||||
for query_str in [
|
||||
"DELETE FROM eventlog WHERE eventlog.linked_type = :type_ind AND eventlog.linked_id = :offer_id",
|
||||
{"type_ind": int(Concepts.OFFER), "offer_id": offer_row[0]},
|
||||
)
|
||||
cursor.execute(
|
||||
"DELETE FROM automationlinks WHERE automationlinks.linked_type = :type_ind AND automationlinks.linked_id = :offer_id",
|
||||
{"type_ind": int(Concepts.OFFER), "offer_id": offer_row[0]},
|
||||
)
|
||||
cursor.execute(
|
||||
"DELETE FROM prefunded_transactions WHERE prefunded_transactions.linked_type = :type_ind AND prefunded_transactions.linked_id = :offer_id",
|
||||
{"type_ind": int(Concepts.OFFER), "offer_id": offer_row[0]},
|
||||
)
|
||||
cursor.execute(
|
||||
"DELETE FROM history WHERE history.concept_type = :type_ind AND history.concept_id = :offer_id",
|
||||
{"type_ind": int(Concepts.OFFER), "offer_id": offer_row[0]},
|
||||
)
|
||||
cursor.execute(
|
||||
"DELETE FROM xmr_offers WHERE xmr_offers.offer_id = :offer_id",
|
||||
{"offer_id": offer_row[0]},
|
||||
)
|
||||
cursor.execute(
|
||||
"DELETE FROM sentoffers WHERE sentoffers.offer_id = :offer_id",
|
||||
{"offer_id": offer_row[0]},
|
||||
)
|
||||
cursor.execute(
|
||||
"DELETE FROM actions WHERE actions.linked_id = :offer_id",
|
||||
{"offer_id": offer_row[0]},
|
||||
)
|
||||
cursor.execute(
|
||||
"DELETE FROM offers WHERE offers.offer_id = :offer_id",
|
||||
{"offer_id": offer_row[0]},
|
||||
)
|
||||
cursor.execute(
|
||||
"DELETE FROM message_links WHERE linked_type = :type_ind AND linked_id = :offer_id",
|
||||
{"type_ind": int(Concepts.OFFER), "offer_id": offer_row[0]},
|
||||
)
|
||||
"DELETE FROM message_network_links WHERE linked_type = :type_ind AND linked_id = :offer_id",
|
||||
]:
|
||||
cursor.execute(query_str, offer_query_data)
|
||||
|
||||
if num_offers > 0 or num_bids > 0:
|
||||
self.log.info(
|
||||
|
||||
@@ -144,7 +144,8 @@ class OfferMessage(NonProtobufClass):
|
||||
17: ("amount_negotiable", NPBW_INT, NPBF_BOOL),
|
||||
18: ("rate_negotiable", NPBW_INT, NPBF_BOOL),
|
||||
19: ("proof_utxos", NPBW_BYTES, 0),
|
||||
20: ("auto_accept_type", 0, 0),
|
||||
20: ("auto_accept_type", NPBW_INT, 0),
|
||||
21: ("message_nets", NPBW_BYTES, NPBF_STR),
|
||||
}
|
||||
|
||||
|
||||
@@ -160,6 +161,7 @@ class BidMessage(NonProtobufClass):
|
||||
8: ("proof_signature", NPBW_BYTES, NPBF_STR),
|
||||
9: ("proof_utxos", NPBW_BYTES, 0),
|
||||
10: ("pkhash_buyer_to", NPBW_BYTES, 0),
|
||||
11: ("message_nets", NPBW_BYTES, NPBF_STR),
|
||||
}
|
||||
|
||||
|
||||
@@ -199,6 +201,7 @@ class XmrBidMessage(NonProtobufClass):
|
||||
7: ("kbvf", NPBW_BYTES, 0),
|
||||
8: ("kbsf_dleag", NPBW_BYTES, 0),
|
||||
9: ("dest_af", NPBW_BYTES, 0),
|
||||
10: ("message_nets", NPBW_BYTES, NPBF_STR),
|
||||
}
|
||||
|
||||
|
||||
@@ -261,6 +264,7 @@ class ADSBidIntentMessage(NonProtobufClass):
|
||||
3: ("time_valid", NPBW_INT, 0),
|
||||
4: ("amount_from", NPBW_INT, 0),
|
||||
5: ("amount_to", NPBW_INT, 0),
|
||||
6: ("message_nets", NPBW_BYTES, NPBF_STR),
|
||||
}
|
||||
|
||||
|
||||
@@ -282,3 +286,21 @@ class ConnectReqMessage(NonProtobufClass):
|
||||
3: ("request_type", NPBW_INT, 0),
|
||||
4: ("request_data", NPBW_BYTES, 0),
|
||||
}
|
||||
|
||||
|
||||
class MessagePortalOffer(NonProtobufClass):
|
||||
_map = {
|
||||
1: ("network_type_from", NPBW_INT, 0),
|
||||
2: ("network_type_to", NPBW_INT, 0),
|
||||
3: ("portal_address_from", NPBW_BYTES, 0),
|
||||
4: ("portal_address_to", NPBW_BYTES, 0),
|
||||
5: ("time_valid", NPBW_INT, 0),
|
||||
6: ("smsg_difficulty", NPBW_INT, 0),
|
||||
}
|
||||
|
||||
|
||||
class MessagePortalSend(NonProtobufClass):
|
||||
_map = {
|
||||
1: ("forward_address", NPBW_BYTES, 0), # pubkey, 33 bytes
|
||||
2: ("message_bytes", NPBW_BYTES, 0),
|
||||
}
|
||||
|
||||
975
basicswap/network/bsx_network.py
Normal file
975
basicswap/network/bsx_network.py
Normal file
@@ -0,0 +1,975 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
# Copyright (c) 2025 The Basicswap developers
|
||||
# Distributed under the MIT software license, see the accompanying
|
||||
# file LICENSE or http://www.opensource.org/licenses/mit-license.php.
|
||||
|
||||
import json
|
||||
import random
|
||||
import zmq
|
||||
|
||||
from basicswap.basicswap_util import (
|
||||
AddressTypes,
|
||||
MessageNetworkLinkTypes,
|
||||
MessageNetworks,
|
||||
MessageTypes,
|
||||
)
|
||||
from basicswap.db import DirectMessageRoute
|
||||
from basicswap.messages_npb import (
|
||||
MessagePortalOffer,
|
||||
MessagePortalSend,
|
||||
)
|
||||
from basicswap.network.simplex import (
|
||||
closeSimplexChat,
|
||||
encryptMsg,
|
||||
forwardSimplexMsg,
|
||||
getResponseData,
|
||||
initialiseSimplexNetwork,
|
||||
readSimplexMsgs,
|
||||
sendSimplexMsg,
|
||||
)
|
||||
from basicswap.util import ensure
|
||||
from basicswap.util.logging import LogCategories as LC
|
||||
from basicswap.util.smsg import smsgGetID
|
||||
|
||||
|
||||
class NetworkPortal:
|
||||
__slots__ = (
|
||||
"time_start",
|
||||
"time_valid",
|
||||
"network_from",
|
||||
"network_to",
|
||||
"address_from",
|
||||
"address_to",
|
||||
"smsg_difficulty",
|
||||
"num_refreshes",
|
||||
"messages_sent",
|
||||
"responses_seen",
|
||||
"time_last_used",
|
||||
"num_issues",
|
||||
)
|
||||
|
||||
def __init__(
|
||||
self, time_start, time_valid, network_from, network_to, address_from, address_to
|
||||
):
|
||||
self.time_start = time_start
|
||||
self.time_valid = time_valid
|
||||
self.network_from = network_from
|
||||
self.network_to = network_to
|
||||
self.address_from = address_from
|
||||
self.address_to = address_to
|
||||
|
||||
self.smsg_difficulty = 0x1EFFFFFF
|
||||
|
||||
self.num_refreshes = 0
|
||||
self.messages_sent = 0
|
||||
self.responses_seen = 0
|
||||
self.time_last_used = 0
|
||||
self.num_issues = 0
|
||||
|
||||
|
||||
def networkTypeToID(type: str) -> int:
|
||||
# TODO: remove
|
||||
if type == "smsg":
|
||||
return MessageNetworks.SMSG
|
||||
elif type == "simplex":
|
||||
return MessageNetworks.SIMPLEX
|
||||
raise RuntimeError(f"Unknown message type: {type}")
|
||||
|
||||
|
||||
def networkIDToType(id: int) -> str:
|
||||
if id == MessageNetworks.SMSG:
|
||||
return "smsg"
|
||||
elif id == MessageNetworks.SIMPLEX:
|
||||
return "simplex"
|
||||
raise RuntimeError(f"Unknown message network id: {id}")
|
||||
|
||||
|
||||
class BSXNetwork:
|
||||
_read_zmq_queue: bool = True
|
||||
|
||||
def __init__(self, data_dir, settings, **kwargs):
|
||||
self._bridge_networks = self.settings.get("bridge_networks", False)
|
||||
self._use_direct_message_routes = True
|
||||
self._smsg_plaintext_version = self.settings.get("smsg_plaintext_version", 1)
|
||||
self._smsg_add_to_outbox = self.settings.get("smsg_add_to_outbox", False)
|
||||
self._have_smsg_rpc = False # Set in startNetworks
|
||||
|
||||
self._expire_message_routes_after = self._expire_db_records_after = (
|
||||
self.get_int_setting(
|
||||
"expire_message_routes_after", 48 * 3600, 10 * 60, 31 * 86400
|
||||
)
|
||||
) # Seconds
|
||||
self.check_smsg_seconds = self.get_int_setting(
|
||||
"check_smsg_seconds", 10, 1, 10 * 60
|
||||
)
|
||||
self._last_checked_smsg = 0
|
||||
self.check_bridges_seconds = self.get_int_setting(
|
||||
"check_bridges_seconds", 10, 1, 10 * 60
|
||||
)
|
||||
self._last_checked_bridges = 0
|
||||
|
||||
self._zmq_queue_enabled = self.settings.get("zmq_queue_enabled", True)
|
||||
self._poll_smsg = self.settings.get("poll_smsg", False)
|
||||
self.zmqContext = None
|
||||
self.zmqSubscriber = None
|
||||
|
||||
self.SMSG_SECONDS_IN_HOUR = (
|
||||
60 * 60
|
||||
) # Note: Set smsgsregtestadjust=0 for regtest
|
||||
|
||||
self.num_group_simplex_messages_received = 0
|
||||
self.num_group_simplex_messages_sent = 0
|
||||
self.num_direct_simplex_messages_received = 0
|
||||
self.num_direct_simplex_messages_sent = 0
|
||||
|
||||
self.num_smsg_messages_received = 0
|
||||
self.num_smsg_messages_sent = 0
|
||||
|
||||
self.known_portals = {}
|
||||
self.own_portals = {}
|
||||
|
||||
super().__init__(data_dir=data_dir, settings=settings, **kwargs)
|
||||
|
||||
def finalise(self):
|
||||
if self._network:
|
||||
self._network.stopNetwork()
|
||||
self._network = None
|
||||
|
||||
if self.zmqContext:
|
||||
self.zmqContext.destroy()
|
||||
|
||||
def startNetworks(self):
|
||||
if self._zmq_queue_enabled and self._poll_smsg:
|
||||
self.log.warning("SMSG polling and zmq listener enabled.")
|
||||
self.active_networks = []
|
||||
network_config_list = self.settings.get("networks", [])
|
||||
if len(network_config_list) < 1:
|
||||
network_config_list = [{"type": "smsg", "enabled": True}]
|
||||
|
||||
have_smsg: bool = False
|
||||
for network in network_config_list:
|
||||
if network.get("enabled", True) is False:
|
||||
continue
|
||||
if network["type"] == "smsg":
|
||||
have_smsg = True
|
||||
add_network = {"type": "smsg"}
|
||||
if "bridged" in network:
|
||||
add_network["bridged"] = network["bridged"]
|
||||
self.active_networks.append(add_network)
|
||||
elif network["type"] == "simplex":
|
||||
initialiseSimplexNetwork(self, network)
|
||||
|
||||
if have_smsg:
|
||||
self._have_smsg_rpc = True
|
||||
if self._zmq_queue_enabled:
|
||||
self.zmqContext = zmq.Context()
|
||||
self.zmqSubscriber = self.zmqContext.socket(zmq.SUB)
|
||||
|
||||
self.zmqSubscriber.connect(
|
||||
self.settings["zmqhost"] + ":" + str(self.settings["zmqport"])
|
||||
)
|
||||
self.zmqSubscriber.setsockopt_string(zmq.SUBSCRIBE, "smsg")
|
||||
self.zmqSubscriber.setsockopt_string(zmq.SUBSCRIBE, "hashwtx")
|
||||
|
||||
ro = self.callrpc("smsglocalkeys")
|
||||
found = False
|
||||
for k in ro["smsg_keys"]:
|
||||
if k["address"] == self.network_addr:
|
||||
found = True
|
||||
break
|
||||
if not found:
|
||||
self.log.info("Importing network key to SMSG")
|
||||
self.callrpc(
|
||||
"smsgimportprivkey", [self.network_key, "basicswap offers"]
|
||||
)
|
||||
ro = self.callrpc("smsglocalkeys", ["anon", "-", self.network_addr])
|
||||
ensure(ro["result"] == "Success.", "smsglocalkeys failed")
|
||||
else:
|
||||
now = self.getTime()
|
||||
try:
|
||||
cursor = self.openDB()
|
||||
query: str = "SELECT addr_id FROM smsgaddresses WHERE addr = :addr"
|
||||
addresses = cursor.execute(
|
||||
query, {"addr": self.network_addr}
|
||||
).fetchall()
|
||||
if len(addresses) < 1:
|
||||
query: str = (
|
||||
"INSERT INTO smsgaddresses (active_ind, created_at, addr, pubkey, use_type) VALUES (:active_ind, :created_at, :addr, :pubkey, :use_type)"
|
||||
)
|
||||
cursor.execute(
|
||||
query,
|
||||
{
|
||||
"active_ind": 1,
|
||||
"created_at": now,
|
||||
"addr": self.network_addr,
|
||||
"pubkey": self.network_pubkey,
|
||||
"use_type": AddressTypes.OFFER,
|
||||
},
|
||||
)
|
||||
finally:
|
||||
self.closeDB(cursor)
|
||||
|
||||
# TODO: Ensure smsg is enabled for the active wallet.
|
||||
|
||||
if self._smsg_plaintext_version >= 2:
|
||||
self.log.debug("TODO: disable addReceivedPubkeys")
|
||||
# ro = self.callrpc("smsgoptions", ["set", "addReceivedPubkeys", False])
|
||||
# self.log.debug("smsgoptions {ro}")
|
||||
|
||||
def add_connection(self, host, port, peer_pubkey):
|
||||
self.log.info(f"add_connection {host} {port} {peer_pubkey.hex()}.")
|
||||
self._network.add_connection(host, port, peer_pubkey)
|
||||
|
||||
def get_network_info(self):
|
||||
if not self._network:
|
||||
return {"Error": "Not Initialised"}
|
||||
return self._network.get_info()
|
||||
|
||||
def addMessageNetworkLink(
|
||||
self, linked_type, linked_id, link_type, network_id, cursor
|
||||
) -> None:
|
||||
now: int = self.getTime()
|
||||
query = """INSERT INTO message_network_links
|
||||
(active_ind, linked_type, linked_id, link_type, network_id, created_at)
|
||||
VALUES
|
||||
(1, :linked_type, :linked_id, :link_type, :network_id, :created_at)"""
|
||||
cursor.execute(
|
||||
query,
|
||||
{
|
||||
"linked_type": linked_type,
|
||||
"linked_id": linked_id,
|
||||
"link_type": link_type,
|
||||
"network_id": network_id,
|
||||
"created_at": now,
|
||||
},
|
||||
)
|
||||
|
||||
def getActiveNetwork(self, network_id: int):
|
||||
for network in self.active_networks:
|
||||
if networkTypeToID(network["type"]) == network_id:
|
||||
return network
|
||||
raise RuntimeError("Network not found.")
|
||||
|
||||
def getActiveNetworkInterface(self, network_id: int):
|
||||
network = self.getActiveNetwork(network_id)
|
||||
return network["ws_thread"]
|
||||
|
||||
def getMessageNetsString(self, with_bridged: bool = False) -> str:
|
||||
if self._smsg_plaintext_version < 2:
|
||||
return ""
|
||||
active_networks_set = set()
|
||||
bridged_networks_set = set()
|
||||
for network in self.active_networks:
|
||||
network_type = network.get("type", "smsg")
|
||||
active_networks_set.add(network_type)
|
||||
if with_bridged is False:
|
||||
continue
|
||||
for bridged_network in network.get("bridged", []):
|
||||
bridged_network_type = bridged_network.get("type", "smsg")
|
||||
bridged_networks_set.add(bridged_network_type)
|
||||
all_networks = active_networks_set | bridged_networks_set
|
||||
return ",".join(all_networks)
|
||||
|
||||
def selectMessageNetString(self, received_on_network_ids, remote_message_nets: str) -> str:
|
||||
if self._smsg_plaintext_version < 2:
|
||||
return ""
|
||||
active_networks_set = set()
|
||||
bridged_networks_set = set()
|
||||
for network in self.active_networks:
|
||||
network_type: str = network.get("type", "smsg")
|
||||
active_networks_set.add(networkTypeToID(network_type))
|
||||
for bridged_network in network.get("bridged", []):
|
||||
bridged_network_type = bridged_network.get("type", "smsg")
|
||||
bridged_networks_set.add(networkTypeToID(bridged_network_type))
|
||||
remote_network_ids = self.expandMessageNets(remote_message_nets)
|
||||
|
||||
if len(remote_network_ids) < 1 and len(received_on_network_ids) < 1:
|
||||
return networkIDToType(random.choice(tuple(active_networks_set)))
|
||||
|
||||
# Choose which network to respond on
|
||||
# Pick the received on network if it's in the local node's active networks and the list of remote node's networks
|
||||
# else prefer a network the local node has active
|
||||
for received_on_id in received_on_network_ids:
|
||||
if received_on_id in active_networks_set and received_on_id in remote_network_ids:
|
||||
return networkIDToType(received_on_id)
|
||||
for local_net_id in active_networks_set:
|
||||
if local_net_id in remote_network_ids:
|
||||
return networkIDToType(local_net_id)
|
||||
for local_net_id in bridged_networks_set:
|
||||
if local_net_id in remote_network_ids:
|
||||
return networkIDToType(local_net_id)
|
||||
raise RuntimeError("Unable to select network to respond on")
|
||||
|
||||
def selectMessageNetStringForConcept(
|
||||
self, linked_type: int, linked_id: bytes, remote_message_nets: str, cursor
|
||||
) -> str:
|
||||
received_on_network_ids = set()
|
||||
query = """SELECT network_id FROM message_network_links
|
||||
WHERE linked_type = :linked_type AND linked_id = :linked_id AND link_type = :link_type"""
|
||||
rows = cursor.execute(
|
||||
query,
|
||||
{
|
||||
"linked_type": linked_type,
|
||||
"linked_id": linked_id,
|
||||
"link_type": MessageNetworkLinkTypes.RECEIVED_ON,
|
||||
},
|
||||
)
|
||||
for row in rows:
|
||||
# TODO: rank networks
|
||||
network_id = row
|
||||
received_on_network_ids.add(network_id)
|
||||
|
||||
return self.selectMessageNetString(received_on_network_ids, remote_message_nets)
|
||||
|
||||
def expandMessageNets(self, message_nets: str) -> list:
|
||||
if message_nets is None or len(message_nets) < 1:
|
||||
return []
|
||||
if len(message_nets) > 256:
|
||||
raise ValueError("message_nets string is too large")
|
||||
rv = []
|
||||
for network_string in message_nets.split(","):
|
||||
try:
|
||||
rv.append(networkTypeToID(network_string))
|
||||
except Exception as e: # noqa: F841
|
||||
self.log.debug(f"Unknown message_net {network_string}")
|
||||
return rv
|
||||
|
||||
def getMessageRoute(
|
||||
self, network_id: int, address_from: str, address_to: str, cursor=None
|
||||
):
|
||||
try:
|
||||
use_cursor = self.openDB(cursor)
|
||||
route = self.queryOne(
|
||||
DirectMessageRoute,
|
||||
use_cursor,
|
||||
{
|
||||
"network_id": network_id,
|
||||
"smsg_addr_local": address_from,
|
||||
"smsg_addr_remote": address_to,
|
||||
},
|
||||
)
|
||||
return route
|
||||
finally:
|
||||
if cursor is None:
|
||||
self.closeDB(use_cursor)
|
||||
|
||||
def setMsgSplitInfo(self, xmr_swap) -> None:
|
||||
for network in self.active_networks:
|
||||
if network["type"] == "simplex":
|
||||
xmr_swap.msg_split_info = "9000:11000"
|
||||
return
|
||||
for bridged_network in network.get("bridged", []):
|
||||
if bridged_network["type"] == "simplex":
|
||||
xmr_swap.msg_split_info = "9000:11000"
|
||||
return
|
||||
xmr_swap.msg_split_info = "16000:17000"
|
||||
|
||||
def sendMessage(
|
||||
self,
|
||||
addr_from: str,
|
||||
addr_to: str,
|
||||
payload_hex: bytes,
|
||||
msg_valid: int,
|
||||
cursor,
|
||||
linked_type=None,
|
||||
linked_id=None,
|
||||
timestamp=None,
|
||||
deterministic=False,
|
||||
message_nets=None, # None -> all, else
|
||||
) -> bytes:
|
||||
message_id: bytes = None
|
||||
networks_list = self.expandMessageNets(
|
||||
message_nets
|
||||
) # Empty list means send to all networks
|
||||
networks_sent_to = set()
|
||||
|
||||
# Message routes work only with simplex messages for now.
|
||||
message_route = self.getMessageRoute(1, addr_from, addr_to, cursor=cursor)
|
||||
if message_route:
|
||||
raise RuntimeError("Trying to send through an unestablished direct route.")
|
||||
|
||||
message_route = self.getMessageRoute(2, addr_from, addr_to, cursor=cursor)
|
||||
if message_route:
|
||||
network = self.getActiveNetwork(2)
|
||||
net_i = network["ws_thread"]
|
||||
|
||||
remote_name = None
|
||||
route_data = json.loads(message_route.route_data.decode("UTF-8"))
|
||||
if "localDisplayName" in route_data:
|
||||
remote_name = route_data["localDisplayName"]
|
||||
else:
|
||||
pccConnId = route_data["pccConnId"]
|
||||
self.log.debug(f"Finding name for Simplex chat, ID: {pccConnId}")
|
||||
cmd_id = net_i.send_command("/chats")
|
||||
response = net_i.wait_for_command_response(cmd_id)
|
||||
for chat in getResponseData(response, "chats"):
|
||||
if (
|
||||
"chatInfo" not in chat
|
||||
or "type" not in chat["chatInfo"]
|
||||
or chat["chatInfo"]["type"] != "direct"
|
||||
):
|
||||
continue
|
||||
try:
|
||||
if (
|
||||
chat["chatInfo"]["contact"]["activeConn"]["connId"]
|
||||
== pccConnId
|
||||
):
|
||||
remote_name = chat["chatInfo"]["contact"][
|
||||
"localDisplayName"
|
||||
]
|
||||
break
|
||||
except Exception as e:
|
||||
self.log.debug(f"Error parsing chat: {e}")
|
||||
|
||||
if remote_name is None:
|
||||
raise RuntimeError(
|
||||
f"Unable to find remote name for simplex direct chat, pccConnId: {pccConnId}"
|
||||
)
|
||||
|
||||
message_id = sendSimplexMsg(
|
||||
self,
|
||||
network,
|
||||
addr_from,
|
||||
addr_to,
|
||||
bytes.fromhex(payload_hex),
|
||||
msg_valid,
|
||||
cursor,
|
||||
timestamp,
|
||||
deterministic,
|
||||
to_user_name=remote_name,
|
||||
)
|
||||
return message_id
|
||||
|
||||
smsg_difficulty: int = 0x1EFFFFFF
|
||||
if self._have_smsg_rpc:
|
||||
smsg_difficulty = self.callrpc("smsggetdifficulty", [-1, True])
|
||||
else:
|
||||
self.log.debug("TODO, get difficulty from a portal")
|
||||
|
||||
# First network in list will set message_id
|
||||
smsg_msg: bytes = None
|
||||
for network in self.active_networks:
|
||||
net_message_id = None
|
||||
network_type: int = networkTypeToID(network.get("type", "smsg"))
|
||||
if network_type in networks_sent_to:
|
||||
self.logD(
|
||||
LC.NET, f"Skipping active network {network_type}, already sent to"
|
||||
)
|
||||
continue
|
||||
if len(networks_list) > 0 and network_type not in networks_list:
|
||||
self.logD(
|
||||
LC.NET,
|
||||
f"Skipping active network {network_type}, not in networks_list",
|
||||
)
|
||||
continue
|
||||
|
||||
if network_type == MessageNetworks.SMSG:
|
||||
if smsg_msg:
|
||||
self.forwardSmsg(smsg_msg)
|
||||
else:
|
||||
net_message_id, smsg_msg = self.sendSmsg(
|
||||
addr_from, addr_to, payload_hex, msg_valid, return_msg=True
|
||||
)
|
||||
elif network_type == MessageNetworks.SIMPLEX:
|
||||
if smsg_msg:
|
||||
forwardSimplexMsg(self, network, smsg_msg)
|
||||
else:
|
||||
net_message_id, smsg_msg = sendSimplexMsg(
|
||||
self,
|
||||
network,
|
||||
addr_from,
|
||||
addr_to,
|
||||
bytes.fromhex(payload_hex),
|
||||
msg_valid,
|
||||
cursor,
|
||||
timestamp,
|
||||
deterministic,
|
||||
return_msg=True,
|
||||
difficulty_target=smsg_difficulty,
|
||||
)
|
||||
else:
|
||||
raise ValueError("Unknown network: {}".format(network["type"]))
|
||||
networks_sent_to.add(network_type)
|
||||
if not message_id:
|
||||
message_id = net_message_id
|
||||
|
||||
for network in self.active_networks:
|
||||
net_message_id = None
|
||||
network_type_from = networkTypeToID(network.get("type", "smsg"))
|
||||
if network_type_from not in self.known_portals:
|
||||
self.known_portals[network_type_from] = {}
|
||||
portals_from = self.known_portals[network_type_from]
|
||||
|
||||
for bridged_network in network.get("bridged", []):
|
||||
network_type_to = networkTypeToID(bridged_network["type"])
|
||||
if network_type_to in networks_sent_to:
|
||||
self.logD(
|
||||
LC.NET,
|
||||
f"Skipping bridged network {network_type_to}, already sent to",
|
||||
)
|
||||
continue
|
||||
if len(networks_list) > 0 and network_type_to not in networks_list:
|
||||
self.logD(
|
||||
LC.NET,
|
||||
f"Skipping bridged network {network_type_to}, not in networks_list",
|
||||
)
|
||||
continue
|
||||
|
||||
if network_type_to not in portals_from:
|
||||
portals_from[network_type_to] = []
|
||||
portals_from_to = portals_from[network_type_to]
|
||||
use_portal = None
|
||||
for portal in portals_from_to:
|
||||
if use_portal is None:
|
||||
use_portal = portal
|
||||
else:
|
||||
# TODO: Pick better
|
||||
if portal.num_issues < use_portal.num_issues:
|
||||
use_portal = portal
|
||||
|
||||
if use_portal is None:
|
||||
self.log.warning(
|
||||
f"Could not pick portal to network {network_type_to}, msg {self.logIDM(net_message_id)}"
|
||||
)
|
||||
else:
|
||||
if smsg_msg is None:
|
||||
smsg_msg = encryptMsg(
|
||||
self,
|
||||
addr_from,
|
||||
addr_to,
|
||||
bytes.fromhex(payload_hex),
|
||||
msg_valid,
|
||||
cursor,
|
||||
timestamp,
|
||||
deterministic,
|
||||
use_portal.smsg_difficulty,
|
||||
)
|
||||
if not message_id:
|
||||
message_id = smsgGetID(smsg_msg)
|
||||
|
||||
forward_to = None # TODO - simplex username
|
||||
self.usePortal(use_portal, smsg_msg, addr_from, forward_to, cursor)
|
||||
networks_sent_to.add(network_type_to)
|
||||
|
||||
return message_id
|
||||
|
||||
def sendSmsg(
|
||||
self,
|
||||
addr_from: str,
|
||||
addr_to: str,
|
||||
payload_hex: bytes,
|
||||
msg_valid: int,
|
||||
return_msg: bool = False,
|
||||
) -> bytes:
|
||||
|
||||
options = {"decodehex": True, "ttl_is_seconds": True}
|
||||
if self._smsg_plaintext_version >= 2:
|
||||
options["plaintext_format_version"] = 2
|
||||
options["compression"] = 0
|
||||
if self._smsg_add_to_outbox is False:
|
||||
options["add_to_outbox"] = False
|
||||
if return_msg:
|
||||
options["returnmsg"] = True
|
||||
try:
|
||||
ro = self.callrpc(
|
||||
"smsgsend",
|
||||
[addr_from, addr_to, payload_hex, False, msg_valid, False, options],
|
||||
)
|
||||
self.num_smsg_messages_sent += 1
|
||||
if return_msg:
|
||||
return bytes.fromhex(ro["msgid"]), bytes.fromhex(ro["msg"])
|
||||
return bytes.fromhex(ro["msgid"])
|
||||
except Exception as e:
|
||||
if self.debug:
|
||||
self.log.error("smsgsend failed {}".format(json.dumps(ro, indent=4)))
|
||||
raise e
|
||||
|
||||
def forwardSmsg(self, smsg_msg: bytes) -> None:
|
||||
options = {"submitmsg": True, "rehashmsg": False}
|
||||
self.callrpc("smsgimport", [smsg_msg.hex(), options])
|
||||
self.num_smsg_messages_sent += 1
|
||||
|
||||
def processContactDisconnected(self, event_data) -> None:
|
||||
net_i = self.getActiveNetworkInterface(2)
|
||||
connId = getResponseData(event_data, "contact")["activeConn"]["connId"]
|
||||
self.log.info(f"Direct message route disconnected, connId: {connId}")
|
||||
closeSimplexChat(self, net_i, connId)
|
||||
|
||||
query_str = "SELECT record_id, network_id, smsg_addr_local, smsg_addr_remote, route_data FROM direct_message_routes"
|
||||
try:
|
||||
cursor = self.openDB()
|
||||
|
||||
rows = cursor.execute(query_str).fetchall()
|
||||
|
||||
for row in rows:
|
||||
record_id, network_id, smsg_addr_local, smsg_addr_remote, route_data = (
|
||||
row
|
||||
)
|
||||
route_data = json.loads(route_data.decode("UTF-8"))
|
||||
|
||||
if connId == route_data["pccConnId"]:
|
||||
self.log.debug(f"Removing direct message route: {record_id}.")
|
||||
cursor.execute(
|
||||
"DELETE FROM direct_message_routes WHERE record_id = :record_id ",
|
||||
{"record_id": record_id},
|
||||
)
|
||||
break
|
||||
finally:
|
||||
self.closeDB(cursor)
|
||||
|
||||
def closeMessageRoute(self, record_id, network_id, route_data, cursor):
|
||||
net_i = self.getActiveNetworkInterface(2)
|
||||
|
||||
connId = route_data["pccConnId"]
|
||||
|
||||
self.log.info(f"Closing Simplex chat, id: {connId}")
|
||||
closeSimplexChat(self, net_i, connId)
|
||||
|
||||
self.log.debug(f"Removing direct message route: {record_id}.")
|
||||
cursor.execute(
|
||||
"DELETE FROM direct_message_routes WHERE record_id = :record_id ",
|
||||
{"record_id": record_id},
|
||||
)
|
||||
self.commitDB()
|
||||
|
||||
def getSmsgMsgBytes(self, msg) -> bytes:
|
||||
if int(self._smsg_plaintext_version) < 2:
|
||||
return bytes.fromhex(msg["hex"][2:-2])
|
||||
return bytes.fromhex(msg["hex"][2:])
|
||||
|
||||
def processZmqSmsg(self) -> None:
|
||||
message = self.zmqSubscriber.recv()
|
||||
# Clear
|
||||
_ = self.zmqSubscriber.recv()
|
||||
|
||||
if message[0] == 3: # Paid smsg
|
||||
return # TODO: Switch to paid?
|
||||
|
||||
msg_id = message[2:]
|
||||
options = {"encoding": "hex", "setread": True}
|
||||
if self._smsg_plaintext_version >= 2:
|
||||
options["pubkey_from"] = True
|
||||
num_tries = 5
|
||||
for i in range(num_tries + 1):
|
||||
try:
|
||||
msg = self.callrpc("smsg", [msg_id.hex(), options])
|
||||
break
|
||||
except Exception as e:
|
||||
if "Unknown message id" in str(e) and i < num_tries:
|
||||
self.delay_event.wait(1)
|
||||
else:
|
||||
raise e
|
||||
|
||||
self.processMsg(msg)
|
||||
|
||||
def newPortal(self, network_from_id, network_to_id, now):
|
||||
addr_to: str = self.network_addr
|
||||
cursor = self.openDB()
|
||||
try:
|
||||
addr_portal: str = self.prepareSMSGAddress(
|
||||
None, AddressTypes.PORTAL_LOCAL, cursor
|
||||
)
|
||||
finally:
|
||||
self.closeDB(cursor)
|
||||
|
||||
portal = NetworkPortal(
|
||||
now, 30 * 60, network_from_id, network_to_id, addr_portal, addr_to
|
||||
)
|
||||
|
||||
smsg_difficulty: int = 0x1EFFFFFF
|
||||
if self._have_smsg_rpc:
|
||||
smsg_difficulty = self.callrpc("smsggetdifficulty", [-1, True])
|
||||
|
||||
msg_buf = MessagePortalOffer()
|
||||
msg_buf.network_type_from = network_from_id
|
||||
msg_buf.network_type_to = network_to_id
|
||||
msg_buf.time_valid = portal.time_valid
|
||||
msg_buf.smsg_difficulty = smsg_difficulty
|
||||
payload_hex = (
|
||||
str.format("{:02x}", MessageTypes.PORTAL_OFFER) + msg_buf.to_bytes().hex()
|
||||
)
|
||||
|
||||
msg_valid: int = max(self.SMSG_SECONDS_IN_HOUR, portal.time_valid)
|
||||
if network_from_id == MessageNetworks.SMSG:
|
||||
net_message_id = self.sendSmsg(addr_portal, addr_to, payload_hex, msg_valid)
|
||||
elif network_from_id == MessageNetworks.SIMPLEX:
|
||||
network = self.getActiveNetwork(MessageNetworks.SIMPLEX)
|
||||
|
||||
deterministic: bool = True
|
||||
cursor = self.openDB()
|
||||
try:
|
||||
net_message_id = sendSimplexMsg(
|
||||
self,
|
||||
network,
|
||||
addr_portal,
|
||||
addr_to,
|
||||
bytes.fromhex(payload_hex),
|
||||
msg_valid,
|
||||
cursor,
|
||||
portal.time_start,
|
||||
deterministic,
|
||||
)
|
||||
finally:
|
||||
self.closeDB(cursor)
|
||||
else:
|
||||
raise RuntimeError(f"Unknown network id {network_from_id}")
|
||||
if network_from_id not in self.own_portals:
|
||||
self.own_portals[network_from_id] = {}
|
||||
self.own_portals[network_from_id][network_to_id] = portal
|
||||
portal = self.own_portals.get(network_from_id, {}).get(network_to_id, None)
|
||||
self.logD(
|
||||
LC.NET,
|
||||
f"Opened new portal {addr_portal} {network_from_id} -> {network_to_id}, {self.logIDM(net_message_id)}",
|
||||
)
|
||||
|
||||
def refreshPortal(self, portal):
|
||||
# TODO: Add random delay between refreshes
|
||||
|
||||
now: int = self.getTime()
|
||||
addr_portal: str = portal.address_from
|
||||
addr_to: str = self.network_addr
|
||||
smsg_difficulty: int = 0x1EFFFFFF
|
||||
if self._have_smsg_rpc:
|
||||
smsg_difficulty = self.callrpc("smsggetdifficulty", [-1, True])
|
||||
|
||||
msg_buf = MessagePortalOffer()
|
||||
msg_buf.network_type_from = portal.network_from
|
||||
msg_buf.network_type_to = portal.network_to
|
||||
msg_buf.time_valid = portal.time_valid
|
||||
msg_buf.smsg_difficulty = smsg_difficulty
|
||||
payload_hex = (
|
||||
str.format("{:02x}", MessageTypes.PORTAL_OFFER) + msg_buf.to_bytes().hex()
|
||||
)
|
||||
|
||||
msg_valid: int = max(self.SMSG_SECONDS_IN_HOUR, portal.time_valid)
|
||||
if portal.network_from == MessageNetworks.SMSG:
|
||||
net_message_id = self.sendSmsg(addr_portal, addr_to, payload_hex, msg_valid)
|
||||
elif portal.network_from == MessageNetworks.SIMPLEX:
|
||||
network = self.getActiveNetwork(MessageNetworks.SIMPLEX)
|
||||
|
||||
cursor = self.openDB()
|
||||
try:
|
||||
net_message_id = sendSimplexMsg(
|
||||
self,
|
||||
network,
|
||||
addr_portal,
|
||||
addr_to,
|
||||
bytes.fromhex(payload_hex),
|
||||
msg_valid,
|
||||
cursor,
|
||||
portal.time_start,
|
||||
)
|
||||
finally:
|
||||
self.closeDB(cursor)
|
||||
else:
|
||||
raise RuntimeError(f"Unknown network id {portal.network_from}")
|
||||
|
||||
portal.time_start = now
|
||||
self.logD(
|
||||
LC.NET,
|
||||
f"Refreshed portal {addr_portal} {portal.network_from} -> {portal.network_to}, {self.logIDM(net_message_id)}",
|
||||
)
|
||||
|
||||
def usePortal(self, portal, smsg, addr_from: str, forward_to: str, cursor):
|
||||
if forward_to is not None:
|
||||
raise ValueError("TODO")
|
||||
now: int = self.getTime()
|
||||
msg_buf = MessagePortalSend()
|
||||
msg_buf.message_bytes = smsg
|
||||
payload_hex = (
|
||||
str.format("{:02x}", MessageTypes.PORTAL_SEND) + msg_buf.to_bytes().hex()
|
||||
)
|
||||
msg_valid: int = max(self.SMSG_SECONDS_IN_HOUR, portal.time_valid)
|
||||
addr_to: str = portal.address_from
|
||||
|
||||
if portal.network_from == MessageNetworks.SMSG:
|
||||
net_message_id = self.sendSmsg(addr_from, addr_to, payload_hex, msg_valid)
|
||||
elif portal.network_from == MessageNetworks.SIMPLEX:
|
||||
network = self.getActiveNetwork(MessageNetworks.SIMPLEX)
|
||||
|
||||
net_message_id = sendSimplexMsg(
|
||||
self,
|
||||
network,
|
||||
addr_from,
|
||||
addr_to,
|
||||
bytes.fromhex(payload_hex),
|
||||
msg_valid,
|
||||
cursor,
|
||||
now,
|
||||
)
|
||||
else:
|
||||
raise ValueError("Unknown network from ind.")
|
||||
self.logD(
|
||||
LC.NET,
|
||||
f"Sending through portal {portal.address_from} {portal.network_from} -> {portal.network_to}, {self.logIDM(net_message_id)}",
|
||||
)
|
||||
|
||||
def processPortalOffer(self, msg) -> None:
|
||||
self.log.debug(
|
||||
"Processing network portal offer {}.".format(self.log.id(msg["msgid"]))
|
||||
)
|
||||
network_received_on: int = networkTypeToID(msg.get("msg_net", "smsg"))
|
||||
|
||||
time_start: int = msg["sent"]
|
||||
addr_portal: str = msg["from"]
|
||||
msg_bytes = self.getSmsgMsgBytes(msg)
|
||||
portal_data = MessagePortalOffer(init_all=False)
|
||||
portal_data.from_bytes(msg_bytes)
|
||||
if portal_data.network_type_from != network_received_on:
|
||||
raise RuntimeError("Network from must match network received on.")
|
||||
|
||||
network_from = self.getActiveNetwork(portal_data.network_type_from)
|
||||
|
||||
# Ignore own portals
|
||||
for network_to_id, portal in self.own_portals.get(
|
||||
portal_data.network_type_from, {}
|
||||
).items():
|
||||
if portal.address_from == addr_portal:
|
||||
self.log.debug("Ignoring own portal.")
|
||||
return
|
||||
|
||||
# Skip portals to networks this node is not using
|
||||
found_enabled_bridge: bool = False
|
||||
enabled_bridged = network_from.get("bridged", [])
|
||||
for network_to_cross in enabled_bridged:
|
||||
if network_to_cross.get("enabled", True) is False:
|
||||
continue
|
||||
if (
|
||||
networkTypeToID(network_to_cross.get("type", "smsg"))
|
||||
== portal_data.network_type_to
|
||||
):
|
||||
found_enabled_bridge = True
|
||||
break
|
||||
|
||||
if found_enabled_bridge is False:
|
||||
self.log.debug("Ignoring portal to an unbridged network.")
|
||||
return
|
||||
|
||||
now: int = self.getTime()
|
||||
if time_start + portal_data.time_valid < now:
|
||||
self.log.warning("Offered portal is expired.")
|
||||
return
|
||||
|
||||
received_portal = NetworkPortal(
|
||||
time_start,
|
||||
portal_data.time_valid,
|
||||
portal_data.network_type_from,
|
||||
portal_data.network_type_to,
|
||||
addr_portal,
|
||||
portal_data.portal_address_to,
|
||||
)
|
||||
received_portal.smsg_difficulty = portal_data.smsg_difficulty
|
||||
|
||||
if received_portal.network_from not in self.known_portals:
|
||||
self.known_portals[received_portal.network_from] = {}
|
||||
portals_from = self.known_portals[received_portal.network_from]
|
||||
|
||||
if received_portal.network_to not in portals_from:
|
||||
portals_from[received_portal.network_to] = []
|
||||
|
||||
portals_from_to = portals_from[received_portal.network_to]
|
||||
|
||||
for portal in portals_from_to:
|
||||
if portal.address_from == received_portal.address_from:
|
||||
portal.num_refreshes += 1
|
||||
portal.time_start = received_portal.time_start
|
||||
portal.time_valid = received_portal.time_valid
|
||||
portal.smsg_difficulty = received_portal.smsg_difficulty
|
||||
return
|
||||
|
||||
portals_from_to.append(received_portal)
|
||||
|
||||
try:
|
||||
cursor = self.openDB()
|
||||
query: str = "SELECT addr_id FROM smsgaddresses WHERE addr = :addr"
|
||||
addresses = cursor.execute(
|
||||
query, {"addr": received_portal.address_from}
|
||||
).fetchall()
|
||||
if len(addresses) < 1:
|
||||
pk_address_from: str = msg["pubkey_from"]
|
||||
query: str = (
|
||||
"INSERT INTO smsgaddresses (active_ind, created_at, addr, pubkey, use_type) VALUES (:active_ind, :created_at, :addr, :pubkey, :use_type)"
|
||||
)
|
||||
cursor.execute(
|
||||
query,
|
||||
{
|
||||
"active_ind": 1,
|
||||
"created_at": now,
|
||||
"addr": received_portal.address_from,
|
||||
"pubkey": pk_address_from,
|
||||
"use_type": AddressTypes.PORTAL,
|
||||
},
|
||||
)
|
||||
|
||||
finally:
|
||||
self.closeDB(cursor)
|
||||
|
||||
def processPortalMessage(self, msg):
|
||||
msg_id = msg["msgid"]
|
||||
self.log.debug(f"Processing network portal message {msg_id}.")
|
||||
|
||||
addr_to: str = msg["to"]
|
||||
network_from_id: int = networkTypeToID(msg.get("msg_net", "smsg"))
|
||||
from_portals = self.own_portals.get(network_from_id, {})
|
||||
portal = None
|
||||
for network_to_id, to_portal in from_portals.items():
|
||||
if to_portal.address_from == addr_to:
|
||||
portal = to_portal
|
||||
break
|
||||
if portal is None:
|
||||
self.log.debug(f"Portal not found for portal message {msg_id}")
|
||||
return
|
||||
network_to_id = portal.network_to
|
||||
|
||||
msg_bytes = self.getSmsgMsgBytes(msg)
|
||||
portal_msg = MessagePortalSend(init_all=False)
|
||||
portal_msg.from_bytes(msg_bytes)
|
||||
|
||||
if network_to_id == MessageNetworks.SMSG:
|
||||
self.forwardSmsg(portal_msg.message_bytes)
|
||||
elif network_to_id == MessageNetworks.SIMPLEX:
|
||||
network = self.getActiveNetwork(MessageNetworks.SIMPLEX)
|
||||
forwardSimplexMsg(self, network, portal_msg.message_bytes)
|
||||
else:
|
||||
raise ValueError(f"Unknown network ID {network_to_id}")
|
||||
|
||||
def updateNetworkBridges(self, now: int) -> None:
|
||||
for network in self.active_networks:
|
||||
network_from_id: int = networkTypeToID(network["type"])
|
||||
|
||||
for other_network in self.active_networks:
|
||||
if network == other_network:
|
||||
continue
|
||||
network_id: int = networkTypeToID(other_network["type"])
|
||||
portal = self.own_portals.get(network_from_id, {}).get(network_id, None)
|
||||
|
||||
if portal is None:
|
||||
self.newPortal(network_from_id, network_id, now)
|
||||
else:
|
||||
if portal.time_start + portal.time_valid <= now - (5 * 60):
|
||||
self.refreshPortal(portal)
|
||||
self._last_checked_bridges = now
|
||||
|
||||
def updateNetwork(self) -> None:
|
||||
now: int = self.getTime()
|
||||
|
||||
if self._poll_smsg:
|
||||
if now - self._last_checked_smsg >= self.check_smsg_seconds:
|
||||
self._last_checked_smsg = now
|
||||
options = {"encoding": "hex", "setread": True}
|
||||
msgs = self.callrpc("smsginbox", ["unread", "", options])
|
||||
for msg in msgs["messages"]:
|
||||
self.processMsg(msg)
|
||||
|
||||
try:
|
||||
if self._bridge_networks:
|
||||
if len(self.active_networks) > 1:
|
||||
if now - self._last_checked_bridges >= self.check_bridges_seconds:
|
||||
self.updateNetworkBridges(now)
|
||||
for network in self.active_networks:
|
||||
if network["type"] == "simplex":
|
||||
readSimplexMsgs(self, network)
|
||||
|
||||
except Exception as ex:
|
||||
self.logException(f"updateNetwork {ex}")
|
||||
@@ -26,6 +26,7 @@ from basicswap.util.address import (
|
||||
b58decode,
|
||||
decodeWif,
|
||||
)
|
||||
from basicswap.basicswap_util import AddressTypes
|
||||
|
||||
|
||||
def encode_base64(data: bytes) -> str:
|
||||
@@ -172,8 +173,7 @@ def waitForConnected(ws_thread, delay_event):
|
||||
raise ValueError("waitForConnected timed-out.")
|
||||
|
||||
|
||||
def getPrivkeyForAddress(self, addr) -> bytes:
|
||||
|
||||
def getPrivkeyForAddress(self, cursor, addr: str) -> bytes:
|
||||
ci_part = self.ci(Coins.PART)
|
||||
try:
|
||||
return ci_part.decodeKey(
|
||||
@@ -200,6 +200,38 @@ def getPrivkeyForAddress(self, addr) -> bytes:
|
||||
raise ValueError("key not found")
|
||||
|
||||
|
||||
def getPubkeyForAddress(self, cursor, addr: str) -> bytes:
|
||||
if self._have_smsg_rpc:
|
||||
try:
|
||||
rv = self.callrpc(
|
||||
"smsggetpubkey",
|
||||
[
|
||||
addr,
|
||||
],
|
||||
)
|
||||
return b58decode(rv["publickey"])
|
||||
except Exception as e: # noqa: F841
|
||||
pass
|
||||
use_cursor = self.openDB(cursor)
|
||||
try:
|
||||
query: str = "SELECT pk_from FROM offers WHERE addr_from = :addr_to LIMIT 1"
|
||||
rows = use_cursor.execute(query, {"addr_to": addr}).fetchall()
|
||||
if len(rows) > 0:
|
||||
return rows[0][0]
|
||||
query: str = "SELECT pk_bid_addr FROM bids WHERE bid_addr = :addr_to LIMIT 1"
|
||||
rows = use_cursor.execute(query, {"addr_to": addr}).fetchall()
|
||||
if len(rows) > 0:
|
||||
return rows[0][0]
|
||||
query: str = "SELECT pubkey FROM smsgaddresses WHERE addr = :addr LIMIT 1"
|
||||
rows = use_cursor.execute(query, {"addr": addr}).fetchall()
|
||||
if len(rows) > 0:
|
||||
return bytes.fromhex(rows[0][0])
|
||||
raise ValueError(f"Could not get public key for address: {addr}")
|
||||
finally:
|
||||
if cursor is None:
|
||||
self.closeDB(use_cursor, commit=False)
|
||||
|
||||
|
||||
def encryptMsg(
|
||||
self,
|
||||
addr_from: str,
|
||||
@@ -209,42 +241,20 @@ def encryptMsg(
|
||||
cursor,
|
||||
timestamp=None,
|
||||
deterministic=False,
|
||||
difficulty_target=0x1EFFFFFF,
|
||||
) -> bytes:
|
||||
self.log.debug("encryptMsg")
|
||||
|
||||
try:
|
||||
rv = self.callrpc(
|
||||
"smsggetpubkey",
|
||||
[
|
||||
addr_to,
|
||||
],
|
||||
)
|
||||
pubkey_to: bytes = b58decode(rv["publickey"])
|
||||
except Exception as e: # noqa: F841
|
||||
use_cursor = self.openDB(cursor)
|
||||
try:
|
||||
query: str = "SELECT pk_from FROM offers WHERE addr_from = :addr_to LIMIT 1"
|
||||
rows = use_cursor.execute(query, {"addr_to": addr_to}).fetchall()
|
||||
if len(rows) > 0:
|
||||
pubkey_to = rows[0][0]
|
||||
else:
|
||||
query: str = (
|
||||
"SELECT pk_bid_addr FROM bids WHERE bid_addr = :addr_to LIMIT 1"
|
||||
)
|
||||
rows = use_cursor.execute(query, {"addr_to": addr_to}).fetchall()
|
||||
if len(rows) > 0:
|
||||
pubkey_to = rows[0][0]
|
||||
else:
|
||||
raise ValueError(f"Could not get public key for address {addr_to}")
|
||||
finally:
|
||||
if cursor is None:
|
||||
self.closeDB(use_cursor, commit=False)
|
||||
pubkey_to = getPubkeyForAddress(self, cursor, addr_to)
|
||||
privkey_from = getPrivkeyForAddress(self, cursor, addr_from)
|
||||
|
||||
privkey_from = getPrivkeyForAddress(self, addr_from)
|
||||
|
||||
payload += bytes((0,)) # Include null byte to match smsg
|
||||
smsg_msg: bytes = smsgEncrypt(
|
||||
privkey_from, pubkey_to, payload, timestamp, deterministic
|
||||
privkey_from,
|
||||
pubkey_to,
|
||||
payload,
|
||||
timestamp,
|
||||
deterministic,
|
||||
difficulty_target=difficulty_target,
|
||||
)
|
||||
|
||||
return smsg_msg
|
||||
@@ -261,11 +271,21 @@ def sendSimplexMsg(
|
||||
timestamp: int = None,
|
||||
deterministic: bool = False,
|
||||
to_user_name: str = None,
|
||||
return_msg: bool = False,
|
||||
difficulty_target=0x1EFFFFFF,
|
||||
) -> bytes:
|
||||
self.log.debug("sendSimplexMsg")
|
||||
|
||||
smsg_msg: bytes = encryptMsg(
|
||||
self, addr_from, addr_to, payload, msg_valid, cursor, timestamp, deterministic
|
||||
self,
|
||||
addr_from,
|
||||
addr_to,
|
||||
payload,
|
||||
msg_valid,
|
||||
cursor,
|
||||
timestamp,
|
||||
deterministic,
|
||||
difficulty_target,
|
||||
)
|
||||
smsg_id = smsgGetID(smsg_msg)
|
||||
|
||||
@@ -280,6 +300,33 @@ def sendSimplexMsg(
|
||||
json_str = json.dumps(response, indent=4)
|
||||
self.log.debug(f"Response {json_str}")
|
||||
raise ValueError("Send failed")
|
||||
if to_user_name is not None:
|
||||
self.num_direct_simplex_messages_sent += 1
|
||||
else:
|
||||
self.num_group_simplex_messages_sent += 1
|
||||
|
||||
if return_msg:
|
||||
return smsg_id, smsg_msg
|
||||
return smsg_id
|
||||
|
||||
|
||||
def forwardSimplexMsg(self, network, smsg_msg, to_user_name: str = None):
|
||||
smsg_id = smsgGetID(smsg_msg)
|
||||
ws_thread = network["ws_thread"]
|
||||
if to_user_name is not None:
|
||||
to = "@" + to_user_name + " "
|
||||
else:
|
||||
to = "#bsx "
|
||||
sent_id = ws_thread.send_command(to + encode_base64(smsg_msg))
|
||||
response = waitForResponse(ws_thread, sent_id, self.delay_event)
|
||||
if getResponseData(response, "type") != "newChatItems":
|
||||
json_str = json.dumps(response, indent=4)
|
||||
self.log.debug(f"Response {json_str}")
|
||||
raise ValueError("Send failed")
|
||||
if to_user_name is not None:
|
||||
self.num_direct_simplex_messages_sent += 1
|
||||
else:
|
||||
self.num_group_simplex_messages_sent += 1
|
||||
|
||||
return smsg_id
|
||||
|
||||
@@ -292,7 +339,7 @@ def decryptSimplexMsg(self, msg_data):
|
||||
try:
|
||||
decrypted = smsgDecrypt(network_key, msg_data, output_dict=True)
|
||||
decrypted["from"] = ci_part.pubkey_to_address(
|
||||
bytes.fromhex(decrypted["pk_from"])
|
||||
bytes.fromhex(decrypted["pubkey_from"])
|
||||
)
|
||||
decrypted["to"] = self.network_addr
|
||||
decrypted["msg_net"] = "simplex"
|
||||
@@ -308,31 +355,34 @@ def decryptSimplexMsg(self, msg_data):
|
||||
AND (s.in_progress OR (s.swap_ended = 0 AND b.expire_at > :now))
|
||||
UNION
|
||||
SELECT addr_from AS address FROM offers WHERE active_ind = 1 AND expire_at > :now
|
||||
UNION
|
||||
SELECT addr AS address FROM smsgaddresses WHERE active_ind = 1 AND use_type = :local_portal
|
||||
)"""
|
||||
|
||||
now: int = self.getTime()
|
||||
|
||||
try:
|
||||
cursor = self.openDB()
|
||||
addr_rows = cursor.execute(query, {"now": now}).fetchall()
|
||||
addr_rows = cursor.execute(
|
||||
query, {"now": now, "local_portal": AddressTypes.PORTAL_LOCAL}
|
||||
).fetchall()
|
||||
decrypted = None
|
||||
for row in addr_rows:
|
||||
addr = row[0]
|
||||
try:
|
||||
vk_addr = getPrivkeyForAddress(self, cursor, addr)
|
||||
decrypted = smsgDecrypt(vk_addr, msg_data, output_dict=True)
|
||||
decrypted["from"] = ci_part.pubkey_to_address(
|
||||
bytes.fromhex(decrypted["pubkey_from"])
|
||||
)
|
||||
decrypted["to"] = addr
|
||||
decrypted["msg_net"] = "simplex"
|
||||
return decrypted
|
||||
except Exception as e: # noqa: F841
|
||||
pass
|
||||
finally:
|
||||
self.closeDB(cursor, commit=False)
|
||||
|
||||
decrypted = None
|
||||
for row in addr_rows:
|
||||
addr = row[0]
|
||||
try:
|
||||
vk_addr = getPrivkeyForAddress(self, addr)
|
||||
decrypted = smsgDecrypt(vk_addr, msg_data, output_dict=True)
|
||||
decrypted["from"] = ci_part.pubkey_to_address(
|
||||
bytes.fromhex(decrypted["pk_from"])
|
||||
)
|
||||
decrypted["to"] = addr
|
||||
decrypted["msg_net"] = "simplex"
|
||||
return decrypted
|
||||
except Exception as e: # noqa: F841
|
||||
pass
|
||||
|
||||
return decrypted
|
||||
|
||||
|
||||
@@ -375,7 +425,6 @@ def parseSimplexMsg(self, chat_item):
|
||||
return decrypted_msg
|
||||
except Exception as e: # noqa: F841
|
||||
# self.log.debug(f"decryptSimplexMsg error: {e}")
|
||||
self.log.debug(f"decryptSimplexMsg error: {e}")
|
||||
pass
|
||||
return None
|
||||
|
||||
@@ -421,7 +470,7 @@ def readSimplexMsgs(self, network):
|
||||
elif processEvent(self, ws_thread, msg_type, data):
|
||||
pass
|
||||
else:
|
||||
self.log.debug(f"Unknown msg_type: {msg_type}")
|
||||
self.log.debug(f"simplex: Unknown msg_type: {msg_type}")
|
||||
# self.log.debug(f"Message: {json.dumps(data, indent=4)}")
|
||||
except Exception as e:
|
||||
self.log.debug(f"readSimplexMsgs error: {e}")
|
||||
@@ -432,10 +481,11 @@ def readSimplexMsgs(self, network):
|
||||
|
||||
|
||||
def getResponseData(data, tag=None):
|
||||
if "Right" in data["resp"]:
|
||||
if tag:
|
||||
return data["resp"]["Right"][tag]
|
||||
return data["resp"]["Right"]
|
||||
for pretag in ("Right", "Left"):
|
||||
if pretag in data["resp"]:
|
||||
if tag:
|
||||
return data["resp"][pretag][tag]
|
||||
return data["resp"][pretag]
|
||||
if tag:
|
||||
return data["resp"][tag]
|
||||
return data["resp"]
|
||||
@@ -474,12 +524,14 @@ def initialiseSimplexNetwork(self, network_config) -> None:
|
||||
response = waitForResponse(ws_thread, sent_id, self.delay_event)
|
||||
assert "groupLinkId" in getResponseData(response, "connection")
|
||||
|
||||
network = {
|
||||
add_network = {
|
||||
"type": "simplex",
|
||||
"ws_thread": ws_thread,
|
||||
}
|
||||
if "bridged" in network_config:
|
||||
add_network["bridged"] = network_config["bridged"]
|
||||
|
||||
self.active_networks.append(network)
|
||||
self.active_networks.append(add_network)
|
||||
|
||||
|
||||
def closeSimplexChat(self, net_i, connId) -> bool:
|
||||
|
||||
@@ -9,8 +9,8 @@ from basicswap.util.address import b58decode
|
||||
|
||||
|
||||
def getMsgPubkey(self, msg) -> bytes:
|
||||
if "pk_from" in msg:
|
||||
return bytes.fromhex(msg["pk_from"])
|
||||
if "pubkey_from" in msg:
|
||||
return bytes.fromhex(msg["pubkey_from"])
|
||||
rv = self.callrpc(
|
||||
"smsggetpubkey",
|
||||
[
|
||||
|
||||
@@ -10,6 +10,23 @@ from basicswap.db import getOrderByStr
|
||||
|
||||
|
||||
class UIApp:
|
||||
def __init__(self, **kwargs):
|
||||
pass
|
||||
|
||||
def setFilters(self, prefix, filters):
|
||||
key_str = "saved_filters_" + prefix
|
||||
value_str = json.dumps(filters)
|
||||
self.setStringKV(key_str, value_str)
|
||||
|
||||
def getFilters(self, prefix):
|
||||
key_str = "saved_filters_" + prefix
|
||||
value_str = self.getStringKV(key_str)
|
||||
return None if not value_str else json.loads(value_str)
|
||||
|
||||
def clearFilters(self, prefix) -> None:
|
||||
key_str = "saved_filters_" + prefix
|
||||
self.clearStringKV(key_str)
|
||||
|
||||
def listMessageRoutes(self, filters={}, action=None):
|
||||
cursor = self.openDB()
|
||||
try:
|
||||
|
||||
@@ -43,7 +43,7 @@ class LockedCoinError(Exception):
|
||||
self.coinid = coinid
|
||||
|
||||
def __str__(self):
|
||||
return "Coin must be unlocked: " + str(self.coinid)
|
||||
return "must be unlocked: " + str(self.coinid)
|
||||
|
||||
|
||||
def ensure(v, err_string):
|
||||
|
||||
@@ -5,11 +5,16 @@
|
||||
# file LICENSE or http://www.opensource.org/licenses/mit-license.php.
|
||||
|
||||
import logging
|
||||
from enum import IntEnum, auto
|
||||
from basicswap.util.crypto import (
|
||||
sha256,
|
||||
)
|
||||
|
||||
|
||||
class LogCategories(IntEnum):
|
||||
NET = auto()
|
||||
|
||||
|
||||
class BSXLogger(logging.Logger):
|
||||
def __init__(self, name):
|
||||
super().__init__(name)
|
||||
|
||||
@@ -66,6 +66,11 @@ def smsgGetTimestamp(smsg_message: bytes) -> int:
|
||||
return int.from_bytes(smsg_message[11 : 11 + 8], byteorder="little")
|
||||
|
||||
|
||||
def smsgGetTTL(smsg_message: bytes) -> int:
|
||||
assert len(smsg_message) > SMSG_HDR_LEN
|
||||
return int.from_bytes(smsg_message[19 : 19 + 4], byteorder="little")
|
||||
|
||||
|
||||
def smsgGetPOWHash(smsg_message: bytes) -> bytes:
|
||||
assert len(smsg_message) > SMSG_HDR_LEN
|
||||
ofs: int = 4
|
||||
@@ -79,7 +84,7 @@ def smsgGetPOWHash(smsg_message: bytes) -> bytes:
|
||||
|
||||
def smsgGetID(smsg_message: bytes) -> bytes:
|
||||
assert len(smsg_message) > SMSG_HDR_LEN
|
||||
smsg_timestamp = int.from_bytes(smsg_message[11 : 11 + 8], byteorder="little")
|
||||
smsg_timestamp = smsgGetTimestamp(smsg_message)
|
||||
return smsg_timestamp.to_bytes(8, byteorder="big") + ripemd160(smsg_message[8:])
|
||||
|
||||
|
||||
@@ -89,6 +94,8 @@ def smsgEncrypt(
|
||||
payload: bytes,
|
||||
smsg_timestamp: int = None,
|
||||
deterministic: bool = False,
|
||||
plaintext_format: int = 2,
|
||||
difficulty_target=0x1EFFFFFF,
|
||||
) -> bytes:
|
||||
# assert len(payload) < 128 # Requires lz4 if payload > 128 bytes
|
||||
# TODO: Add lz4 to match core smsg
|
||||
@@ -125,13 +132,18 @@ def smsgEncrypt(
|
||||
pkh_from: bytes = hash160(pubkey_from)
|
||||
|
||||
len_payload = len(payload)
|
||||
address_version = 0
|
||||
plaintext_data: bytes = (
|
||||
bytes((address_version,))
|
||||
+ pkh_from
|
||||
+ signature
|
||||
+ len_payload.to_bytes(4, byteorder="little")
|
||||
+ payload
|
||||
|
||||
if plaintext_format == 2:
|
||||
address_version = 249 # Marker for format 2
|
||||
compressed = 0
|
||||
plaintext_data: bytes = bytes((address_version, compressed))
|
||||
elif plaintext_format == 1:
|
||||
address_version = 0
|
||||
plaintext_data: bytes = bytes((address_version,))
|
||||
else:
|
||||
raise ValueError("Unknown plaintext format.")
|
||||
plaintext_data += bytes(
|
||||
pkh_from + signature + len_payload.to_bytes(4, byteorder="little") + payload
|
||||
)
|
||||
|
||||
ciphertext: bytes = aes_encrypt(plaintext_data, key_e, smsg_iv)
|
||||
@@ -166,8 +178,7 @@ def smsgEncrypt(
|
||||
+ ciphertext
|
||||
)
|
||||
|
||||
target: int = uint256_from_compact(0x1EFFFFFF)
|
||||
|
||||
target: int = uint256_from_compact(difficulty_target)
|
||||
for i in range(1000000):
|
||||
pow_hash = smsgGetPOWHash(smsg_message)
|
||||
if uint256_from_str(pow_hash) > target:
|
||||
@@ -216,7 +227,14 @@ def smsgDecrypt(
|
||||
|
||||
plaintext = aes_decrypt(ciphertext, key_e, smsg_iv)
|
||||
|
||||
ofs = 1
|
||||
ofs: int = 0
|
||||
version = plaintext[0]
|
||||
if version == 249:
|
||||
compressed = plaintext[1]
|
||||
assert compressed == 0
|
||||
ofs += 1
|
||||
|
||||
ofs += 1
|
||||
pkh_from = plaintext[ofs : ofs + 20]
|
||||
ofs += 20
|
||||
signature = plaintext[ofs : ofs + 65]
|
||||
@@ -240,6 +258,6 @@ def smsgDecrypt(
|
||||
"msgid": smsgGetID(encrypted_message).hex(),
|
||||
"sent": smsg_timestamp,
|
||||
"hex": payload.hex(),
|
||||
"pk_from": pubkey_signer.hex(),
|
||||
"pubkey_from": pubkey_signer.hex(),
|
||||
}
|
||||
return payload
|
||||
|
||||
Reference in New Issue
Block a user