Janus
7 years ago
committed by
ThomasV
ThomasV
9 changed files with 772 additions and 777 deletions
@ -0,0 +1,315 @@ |
|||||
|
from .util import bfh, bh2u |
||||
|
from .crypto import sha256 |
||||
|
import json |
||||
|
from collections import namedtuple |
||||
|
from .transaction import Transaction |
||||
|
from .ecc import CURVE_ORDER, generator, sig_string_from_der_sig, ECPubkey, string_to_number |
||||
|
from . import ecc, bitcoin, crypto, transaction |
||||
|
from .transaction import opcodes |
||||
|
from .bitcoin import push_script |
||||
|
|
||||
|
HTLC_TIMEOUT_WEIGHT = 663 |
||||
|
HTLC_SUCCESS_WEIGHT = 703 |
||||
|
|
||||
|
Keypair = namedtuple("Keypair", ["pubkey", "privkey"]) |
||||
|
Outpoint = namedtuple("Outpoint", ["txid", "output_index"]) |
||||
|
ChannelConfig = namedtuple("ChannelConfig", [ |
||||
|
"payment_basepoint", "multisig_key", "htlc_basepoint", "delayed_basepoint", "revocation_basepoint", |
||||
|
"to_self_delay", "dust_limit_sat", "max_htlc_value_in_flight_msat", "max_accepted_htlcs"]) |
||||
|
OnlyPubkeyKeypair = namedtuple("OnlyPubkeyKeypair", ["pubkey"]) |
||||
|
RemoteState = namedtuple("RemoteState", ["ctn", "next_per_commitment_point", "amount_msat", "revocation_store", "current_per_commitment_point", "next_htlc_id"]) |
||||
|
LocalState = namedtuple("LocalState", ["ctn", "per_commitment_secret_seed", "amount_msat", "next_htlc_id", "funding_locked_received", "was_announced", "current_commitment_signature"]) |
||||
|
ChannelConstraints = namedtuple("ChannelConstraints", ["feerate", "capacity", "is_initiator", "funding_txn_minimum_depth"]) |
||||
|
#OpenChannel = namedtuple("OpenChannel", ["channel_id", "short_channel_id", "funding_outpoint", "local_config", "remote_config", "remote_state", "local_state", "constraints", "node_id"]) |
||||
|
|
||||
|
class RevocationStore: |
||||
|
""" taken from lnd """ |
||||
|
def __init__(self): |
||||
|
self.buckets = [None] * 48 |
||||
|
self.index = 2**48 - 1 |
||||
|
def add_next_entry(self, hsh): |
||||
|
new_element = ShachainElement(index=self.index, secret=hsh) |
||||
|
bucket = count_trailing_zeros(self.index) |
||||
|
for i in range(0, bucket): |
||||
|
this_bucket = self.buckets[i] |
||||
|
e = shachain_derive(new_element, this_bucket.index) |
||||
|
|
||||
|
if e != this_bucket: |
||||
|
raise Exception("hash is not derivable: {} {} {}".format(bh2u(e.secret), bh2u(this_bucket.secret), this_bucket.index)) |
||||
|
self.buckets[bucket] = new_element |
||||
|
self.index -= 1 |
||||
|
def serialize(self): |
||||
|
return {"index": self.index, "buckets": [[bh2u(k.secret), k.index] if k is not None else None for k in self.buckets]} |
||||
|
@staticmethod |
||||
|
def from_json_obj(decoded_json_obj): |
||||
|
store = RevocationStore() |
||||
|
decode = lambda to_decode: ShachainElement(bfh(to_decode[0]), int(to_decode[1])) |
||||
|
store.buckets = [k if k is None else decode(k) for k in decoded_json_obj["buckets"]] |
||||
|
store.index = decoded_json_obj["index"] |
||||
|
return store |
||||
|
def __eq__(self, o): |
||||
|
return type(o) is RevocationStore and self.serialize() == o.serialize() |
||||
|
def __hash__(self): |
||||
|
return hash(json.dumps(self.serialize(), sort_keys=True)) |
||||
|
|
||||
|
def count_trailing_zeros(index): |
||||
|
""" BOLT-03 (where_to_put_secret) """ |
||||
|
try: |
||||
|
return list(reversed(bin(index)[2:])).index("1") |
||||
|
except ValueError: |
||||
|
return 48 |
||||
|
|
||||
|
def shachain_derive(element, toIndex): |
||||
|
return ShachainElement(get_per_commitment_secret_from_seed(element.secret, toIndex, count_trailing_zeros(element.index)), toIndex) |
||||
|
|
||||
|
ShachainElement = namedtuple("ShachainElement", ["secret", "index"]) |
||||
|
ShachainElement.__str__ = lambda self: "ShachainElement(" + bh2u(self.secret) + "," + str(self.index) + ")" |
||||
|
|
||||
|
def get_per_commitment_secret_from_seed(seed: bytes, i: int, bits: int = 48) -> bytes: |
||||
|
"""Generate per commitment secret.""" |
||||
|
per_commitment_secret = bytearray(seed) |
||||
|
for bitindex in range(bits - 1, -1, -1): |
||||
|
mask = 1 << bitindex |
||||
|
if i & mask: |
||||
|
per_commitment_secret[bitindex // 8] ^= 1 << (bitindex % 8) |
||||
|
per_commitment_secret = bytearray(sha256(per_commitment_secret)) |
||||
|
bajts = bytes(per_commitment_secret) |
||||
|
return bajts |
||||
|
|
||||
|
def secret_to_pubkey(secret): |
||||
|
assert type(secret) is int |
||||
|
return (ecc.generator() * secret).get_public_key_bytes() |
||||
|
|
||||
|
def derive_pubkey(basepoint, per_commitment_point): |
||||
|
p = ecc.ECPubkey(basepoint) + ecc.generator() * ecc.string_to_number(sha256(per_commitment_point + basepoint)) |
||||
|
return p.get_public_key_bytes() |
||||
|
|
||||
|
def derive_privkey(secret, per_commitment_point): |
||||
|
assert type(secret) is int |
||||
|
basepoint = (ecc.generator() * secret).get_public_key_bytes() |
||||
|
basepoint = secret + ecc.string_to_number(sha256(per_commitment_point + basepoint)) |
||||
|
basepoint %= CURVE_ORDER |
||||
|
return basepoint |
||||
|
|
||||
|
def derive_blinded_pubkey(basepoint, per_commitment_point): |
||||
|
k1 = ecc.ECPubkey(basepoint) * ecc.string_to_number(sha256(basepoint + per_commitment_point)) |
||||
|
k2 = ecc.ECPubkey(per_commitment_point) * ecc.string_to_number(sha256(per_commitment_point + basepoint)) |
||||
|
return (k1 + k2).get_public_key_bytes() |
||||
|
|
||||
|
def make_htlc_tx_output(amount_msat, local_feerate, revocationpubkey, local_delayedpubkey, success, to_self_delay): |
||||
|
assert type(amount_msat) is int |
||||
|
assert type(local_feerate) is int |
||||
|
assert type(revocationpubkey) is bytes |
||||
|
assert type(local_delayedpubkey) is bytes |
||||
|
script = bytes([opcodes.OP_IF]) \ |
||||
|
+ bfh(push_script(bh2u(revocationpubkey))) \ |
||||
|
+ bytes([opcodes.OP_ELSE]) \ |
||||
|
+ bitcoin.add_number_to_script(to_self_delay) \ |
||||
|
+ bytes([opcodes.OP_CSV, opcodes.OP_DROP]) \ |
||||
|
+ bfh(push_script(bh2u(local_delayedpubkey))) \ |
||||
|
+ bytes([opcodes.OP_ENDIF, opcodes.OP_CHECKSIG]) |
||||
|
|
||||
|
p2wsh = bitcoin.redeem_script_to_address('p2wsh', bh2u(script)) |
||||
|
weight = HTLC_SUCCESS_WEIGHT if success else HTLC_TIMEOUT_WEIGHT |
||||
|
fee = local_feerate * weight |
||||
|
final_amount_sat = (amount_msat - fee) // 1000 |
||||
|
assert final_amount_sat > 0, final_amount_sat |
||||
|
output = (bitcoin.TYPE_ADDRESS, p2wsh, final_amount_sat) |
||||
|
return output |
||||
|
|
||||
|
def make_htlc_tx_witness(remotehtlcsig, localhtlcsig, payment_preimage, witness_script): |
||||
|
assert type(remotehtlcsig) is bytes |
||||
|
assert type(localhtlcsig) is bytes |
||||
|
assert type(payment_preimage) is bytes |
||||
|
assert type(witness_script) is bytes |
||||
|
return bfh(transaction.construct_witness([0, remotehtlcsig, localhtlcsig, payment_preimage, witness_script])) |
||||
|
|
||||
|
def make_htlc_tx_inputs(htlc_output_txid, htlc_output_index, revocationpubkey, local_delayedpubkey, amount_msat, witness_script): |
||||
|
assert type(htlc_output_txid) is str |
||||
|
assert type(htlc_output_index) is int |
||||
|
assert type(revocationpubkey) is bytes |
||||
|
assert type(local_delayedpubkey) is bytes |
||||
|
assert type(amount_msat) is int |
||||
|
assert type(witness_script) is str |
||||
|
c_inputs = [{ |
||||
|
'scriptSig': '', |
||||
|
'type': 'p2wsh', |
||||
|
'signatures': [], |
||||
|
'num_sig': 0, |
||||
|
'prevout_n': htlc_output_index, |
||||
|
'prevout_hash': htlc_output_txid, |
||||
|
'value': amount_msat // 1000, |
||||
|
'coinbase': False, |
||||
|
'sequence': 0x0, |
||||
|
'preimage_script': witness_script, |
||||
|
}] |
||||
|
return c_inputs |
||||
|
|
||||
|
def make_htlc_tx(cltv_timeout, inputs, output): |
||||
|
assert type(cltv_timeout) is int |
||||
|
c_outputs = [output] |
||||
|
tx = Transaction.from_io(inputs, c_outputs, locktime=cltv_timeout, version=2) |
||||
|
tx.BIP_LI01_sort() |
||||
|
return tx |
||||
|
|
||||
|
def make_offered_htlc(revocation_pubkey, remote_htlcpubkey, local_htlcpubkey, payment_hash): |
||||
|
assert type(revocation_pubkey) is bytes |
||||
|
assert type(remote_htlcpubkey) is bytes |
||||
|
assert type(local_htlcpubkey) is bytes |
||||
|
assert type(payment_hash) is bytes |
||||
|
return bytes([opcodes.OP_DUP, opcodes.OP_HASH160]) + bfh(push_script(bh2u(bitcoin.hash_160(revocation_pubkey))))\ |
||||
|
+ bytes([opcodes.OP_EQUAL, opcodes.OP_IF, opcodes.OP_CHECKSIG, opcodes.OP_ELSE]) \ |
||||
|
+ bfh(push_script(bh2u(remote_htlcpubkey)))\ |
||||
|
+ bytes([opcodes.OP_SWAP, opcodes.OP_SIZE]) + bitcoin.add_number_to_script(32) + bytes([opcodes.OP_EQUAL, opcodes.OP_NOTIF, opcodes.OP_DROP])\ |
||||
|
+ bitcoin.add_number_to_script(2) + bytes([opcodes.OP_SWAP]) + bfh(push_script(bh2u(local_htlcpubkey))) + bitcoin.add_number_to_script(2)\ |
||||
|
+ bytes([opcodes.OP_CHECKMULTISIG, opcodes.OP_ELSE, opcodes.OP_HASH160])\ |
||||
|
+ bfh(push_script(bh2u(crypto.ripemd(payment_hash)))) + bytes([opcodes.OP_EQUALVERIFY, opcodes.OP_CHECKSIG, opcodes.OP_ENDIF, opcodes.OP_ENDIF]) |
||||
|
|
||||
|
def make_received_htlc(revocation_pubkey, remote_htlcpubkey, local_htlcpubkey, payment_hash, cltv_expiry): |
||||
|
for i in [revocation_pubkey, remote_htlcpubkey, local_htlcpubkey, payment_hash]: |
||||
|
assert type(i) is bytes |
||||
|
assert type(cltv_expiry) is int |
||||
|
|
||||
|
return bytes([opcodes.OP_DUP, opcodes.OP_HASH160]) \ |
||||
|
+ bfh(push_script(bh2u(bitcoin.hash_160(revocation_pubkey)))) \ |
||||
|
+ bytes([opcodes.OP_EQUAL, opcodes.OP_IF, opcodes.OP_CHECKSIG, opcodes.OP_ELSE]) \ |
||||
|
+ bfh(push_script(bh2u(remote_htlcpubkey))) \ |
||||
|
+ bytes([opcodes.OP_SWAP, opcodes.OP_SIZE]) \ |
||||
|
+ bitcoin.add_number_to_script(32) \ |
||||
|
+ bytes([opcodes.OP_EQUAL, opcodes.OP_IF, opcodes.OP_HASH160]) \ |
||||
|
+ bfh(push_script(bh2u(crypto.ripemd(payment_hash)))) \ |
||||
|
+ bytes([opcodes.OP_EQUALVERIFY]) \ |
||||
|
+ bitcoin.add_number_to_script(2) \ |
||||
|
+ bytes([opcodes.OP_SWAP]) \ |
||||
|
+ bfh(push_script(bh2u(local_htlcpubkey))) \ |
||||
|
+ bitcoin.add_number_to_script(2) \ |
||||
|
+ bytes([opcodes.OP_CHECKMULTISIG, opcodes.OP_ELSE, opcodes.OP_DROP]) \ |
||||
|
+ bitcoin.add_number_to_script(cltv_expiry) \ |
||||
|
+ bytes([opcodes.OP_CLTV, opcodes.OP_DROP, opcodes.OP_CHECKSIG, opcodes.OP_ENDIF, opcodes.OP_ENDIF]) |
||||
|
|
||||
|
def make_htlc_tx_with_open_channel(chan, pcp, for_us, we_receive, amount_msat, cltv_expiry, payment_hash, commit, original_htlc_output_index): |
||||
|
conf = chan.local_config if for_us else chan.remote_config |
||||
|
other_conf = chan.local_config if not for_us else chan.remote_config |
||||
|
|
||||
|
revocation_pubkey = derive_blinded_pubkey(other_conf.revocation_basepoint.pubkey, pcp) |
||||
|
delayedpubkey = derive_pubkey(conf.delayed_basepoint.pubkey, pcp) |
||||
|
other_revocation_pubkey = derive_blinded_pubkey(other_conf.revocation_basepoint.pubkey, pcp) |
||||
|
other_htlc_pubkey = derive_pubkey(other_conf.htlc_basepoint.pubkey, pcp) |
||||
|
htlc_pubkey = derive_pubkey(conf.htlc_basepoint.pubkey, pcp) |
||||
|
# HTLC-success for the HTLC spending from a received HTLC output |
||||
|
# if we do not receive, and the commitment tx is not for us, they receive, so it is also an HTLC-success |
||||
|
is_htlc_success = for_us == we_receive |
||||
|
htlc_tx_output = make_htlc_tx_output( |
||||
|
amount_msat = amount_msat, |
||||
|
local_feerate = chan.constraints.feerate, |
||||
|
revocationpubkey=revocation_pubkey, |
||||
|
local_delayedpubkey=delayedpubkey, |
||||
|
success = is_htlc_success, |
||||
|
to_self_delay = other_conf.to_self_delay) |
||||
|
if is_htlc_success: |
||||
|
preimage_script = make_received_htlc(other_revocation_pubkey, other_htlc_pubkey, htlc_pubkey, payment_hash, cltv_expiry) |
||||
|
else: |
||||
|
preimage_script = make_offered_htlc(other_revocation_pubkey, other_htlc_pubkey, htlc_pubkey, payment_hash) |
||||
|
htlc_tx_inputs = make_htlc_tx_inputs( |
||||
|
commit.txid(), commit.htlc_output_indices[original_htlc_output_index], |
||||
|
revocationpubkey=revocation_pubkey, |
||||
|
local_delayedpubkey=delayedpubkey, |
||||
|
amount_msat=amount_msat, |
||||
|
witness_script=bh2u(preimage_script)) |
||||
|
if is_htlc_success: |
||||
|
cltv_expiry = 0 |
||||
|
htlc_tx = make_htlc_tx(cltv_expiry, inputs=htlc_tx_inputs, output=htlc_tx_output) |
||||
|
return htlc_tx |
||||
|
|
||||
|
|
||||
|
def make_commitment(ctn, local_funding_pubkey, remote_funding_pubkey, |
||||
|
remote_payment_pubkey, payment_basepoint, |
||||
|
remote_payment_basepoint, revocation_pubkey, |
||||
|
delayed_pubkey, to_self_delay, funding_txid, |
||||
|
funding_pos, funding_sat, local_amount, remote_amount, |
||||
|
dust_limit_sat, local_feerate, for_us, we_are_initiator, |
||||
|
htlcs, trimmed=0): |
||||
|
|
||||
|
pubkeys = sorted([bh2u(local_funding_pubkey), bh2u(remote_funding_pubkey)]) |
||||
|
payments = [payment_basepoint, remote_payment_basepoint] |
||||
|
if not we_are_initiator: |
||||
|
payments.reverse() |
||||
|
obs = get_obscured_ctn(ctn, *payments) |
||||
|
locktime = (0x20 << 24) + (obs & 0xffffff) |
||||
|
sequence = (0x80 << 24) + (obs >> 24) |
||||
|
# commitment tx input |
||||
|
c_inputs = [{ |
||||
|
'type': 'p2wsh', |
||||
|
'x_pubkeys': pubkeys, |
||||
|
'signatures': [None, None], |
||||
|
'num_sig': 2, |
||||
|
'prevout_n': funding_pos, |
||||
|
'prevout_hash': funding_txid, |
||||
|
'value': funding_sat, |
||||
|
'coinbase': False, |
||||
|
'sequence': sequence |
||||
|
}] |
||||
|
# commitment tx outputs |
||||
|
local_script = bytes([opcodes.OP_IF]) + bfh(push_script(bh2u(revocation_pubkey))) + bytes([opcodes.OP_ELSE]) + bitcoin.add_number_to_script(to_self_delay) \ |
||||
|
+ bytes([opcodes.OP_CSV, opcodes.OP_DROP]) + bfh(push_script(bh2u(delayed_pubkey))) + bytes([opcodes.OP_ENDIF, opcodes.OP_CHECKSIG]) |
||||
|
local_address = bitcoin.redeem_script_to_address('p2wsh', bh2u(local_script)) |
||||
|
remote_address = bitcoin.pubkey_to_address('p2wpkh', bh2u(remote_payment_pubkey)) |
||||
|
# TODO trim htlc outputs here while also considering 2nd stage htlc transactions |
||||
|
fee = local_feerate * overall_weight(len(htlcs)) |
||||
|
fee -= trimmed * 1000 |
||||
|
assert type(fee) is int |
||||
|
we_pay_fee = for_us == we_are_initiator |
||||
|
to_local_amt = local_amount - (fee if we_pay_fee else 0) |
||||
|
assert type(to_local_amt) is int |
||||
|
to_local = (bitcoin.TYPE_ADDRESS, local_address, to_local_amt // 1000) |
||||
|
to_remote_amt = remote_amount - (fee if not we_pay_fee else 0) |
||||
|
assert type(to_remote_amt) is int |
||||
|
to_remote = (bitcoin.TYPE_ADDRESS, remote_address, to_remote_amt // 1000) |
||||
|
c_outputs = [to_local, to_remote] |
||||
|
for script, msat_amount in htlcs: |
||||
|
c_outputs += [(bitcoin.TYPE_ADDRESS, bitcoin.redeem_script_to_address('p2wsh', bh2u(script)), msat_amount // 1000)] |
||||
|
|
||||
|
# trim outputs |
||||
|
c_outputs_filtered = list(filter(lambda x:x[2]>= dust_limit_sat, c_outputs)) |
||||
|
assert sum(x[2] for x in c_outputs) <= funding_sat |
||||
|
|
||||
|
# create commitment tx |
||||
|
tx = Transaction.from_io(c_inputs, c_outputs_filtered, locktime=locktime, version=2) |
||||
|
tx.BIP_LI01_sort() |
||||
|
|
||||
|
tx.htlc_output_indices = {} |
||||
|
for idx, output in enumerate(c_outputs): |
||||
|
if output in tx.outputs(): |
||||
|
# minus the first two outputs (to_local, to_remote) |
||||
|
tx.htlc_output_indices[idx - 2] = tx.outputs().index(output) |
||||
|
|
||||
|
return tx |
||||
|
|
||||
|
def sign_and_get_sig_string(tx, local_config, remote_config): |
||||
|
pubkeys = sorted([bh2u(local_config.multisig_key.pubkey), bh2u(remote_config.multisig_key.pubkey)]) |
||||
|
tx.sign({bh2u(local_config.multisig_key.pubkey): (local_config.multisig_key.privkey, True)}) |
||||
|
sig_index = pubkeys.index(bh2u(local_config.multisig_key.pubkey)) |
||||
|
sig = bytes.fromhex(tx.inputs()[0]["signatures"][sig_index]) |
||||
|
sig_64 = sig_string_from_der_sig(sig[:-1]) |
||||
|
return sig_64 |
||||
|
|
||||
|
def funding_output_script(local_config, remote_config): |
||||
|
pubkeys = sorted([bh2u(local_config.multisig_key.pubkey), bh2u(remote_config.multisig_key.pubkey)]) |
||||
|
return transaction.multisig_script(pubkeys, 2) |
||||
|
|
||||
|
def calc_short_channel_id(block_height: int, tx_pos_in_block: int, output_index: int) -> bytes: |
||||
|
bh = block_height.to_bytes(3, byteorder='big') |
||||
|
tpos = tx_pos_in_block.to_bytes(3, byteorder='big') |
||||
|
oi = output_index.to_bytes(2, byteorder='big') |
||||
|
return bh + tpos + oi |
||||
|
|
||||
|
def get_obscured_ctn(ctn, local, remote): |
||||
|
mask = int.from_bytes(sha256(local + remote)[-6:], 'big') |
||||
|
return ctn ^ mask |
||||
|
|
||||
|
def overall_weight(num_htlc): |
||||
|
return 500 + 172 * num_htlc + 224 |
||||
|
|
||||
|
def get_ecdh(priv: bytes, pub: bytes) -> bytes: |
||||
|
pt = ECPubkey(pub) * string_to_number(priv) |
||||
|
return sha256(pt.get_public_key_bytes()) |
||||
@ -0,0 +1,150 @@ |
|||||
|
import unittest |
||||
|
|
||||
|
from lib.util import bh2u, bfh |
||||
|
from lib.lnbase import Peer |
||||
|
from lib.lnrouter import OnionHopsDataSingle, new_onion_packet, OnionPerHop |
||||
|
from lib import bitcoin, lnrouter |
||||
|
|
||||
|
class Test_LNRouter(unittest.TestCase): |
||||
|
|
||||
|
#@staticmethod |
||||
|
#def parse_witness_list(witness_bytes): |
||||
|
# amount_witnesses = witness_bytes[0] |
||||
|
# witness_bytes = witness_bytes[1:] |
||||
|
# res = [] |
||||
|
# for i in range(amount_witnesses): |
||||
|
# witness_length = witness_bytes[0] |
||||
|
# this_witness = witness_bytes[1:witness_length+1] |
||||
|
# assert len(this_witness) == witness_length |
||||
|
# witness_bytes = witness_bytes[witness_length+1:] |
||||
|
# res += [bytes(this_witness)] |
||||
|
# assert witness_bytes == b"", witness_bytes |
||||
|
# return res |
||||
|
|
||||
|
|
||||
|
|
||||
|
def test_find_path_for_payment(self): |
||||
|
class fake_network: |
||||
|
channel_db = lnrouter.ChannelDB() |
||||
|
class fake_ln_worker: |
||||
|
path_finder = lnrouter.LNPathFinder(fake_network.channel_db) |
||||
|
privkey = bitcoin.sha256('privkeyseed') |
||||
|
network = fake_network |
||||
|
channel_state = {} |
||||
|
channels = [] |
||||
|
invoices = {} |
||||
|
p = Peer(fake_ln_worker, '', 0, 'a') |
||||
|
p.on_channel_announcement({'node_id_1': b'b', 'node_id_2': b'c', 'short_channel_id': bfh('0000000000000001')}) |
||||
|
p.on_channel_announcement({'node_id_1': b'b', 'node_id_2': b'e', 'short_channel_id': bfh('0000000000000002')}) |
||||
|
p.on_channel_announcement({'node_id_1': b'a', 'node_id_2': b'b', 'short_channel_id': bfh('0000000000000003')}) |
||||
|
p.on_channel_announcement({'node_id_1': b'c', 'node_id_2': b'd', 'short_channel_id': bfh('0000000000000004')}) |
||||
|
p.on_channel_announcement({'node_id_1': b'd', 'node_id_2': b'e', 'short_channel_id': bfh('0000000000000005')}) |
||||
|
p.on_channel_announcement({'node_id_1': b'a', 'node_id_2': b'd', 'short_channel_id': bfh('0000000000000006')}) |
||||
|
o = lambda i: i.to_bytes(8, "big") |
||||
|
p.on_channel_update({'short_channel_id': bfh('0000000000000001'), 'flags': b'\x00', 'cltv_expiry_delta': o(10), 'htlc_minimum_msat': o(250), 'fee_base_msat': o(100), 'fee_proportional_millionths': o(150)}) |
||||
|
p.on_channel_update({'short_channel_id': bfh('0000000000000001'), 'flags': b'\x01', 'cltv_expiry_delta': o(10), 'htlc_minimum_msat': o(250), 'fee_base_msat': o(100), 'fee_proportional_millionths': o(150)}) |
||||
|
p.on_channel_update({'short_channel_id': bfh('0000000000000002'), 'flags': b'\x00', 'cltv_expiry_delta': o(99), 'htlc_minimum_msat': o(250), 'fee_base_msat': o(100), 'fee_proportional_millionths': o(150)}) |
||||
|
p.on_channel_update({'short_channel_id': bfh('0000000000000002'), 'flags': b'\x01', 'cltv_expiry_delta': o(10), 'htlc_minimum_msat': o(250), 'fee_base_msat': o(100), 'fee_proportional_millionths': o(150)}) |
||||
|
p.on_channel_update({'short_channel_id': bfh('0000000000000003'), 'flags': b'\x01', 'cltv_expiry_delta': o(10), 'htlc_minimum_msat': o(250), 'fee_base_msat': o(100), 'fee_proportional_millionths': o(150)}) |
||||
|
p.on_channel_update({'short_channel_id': bfh('0000000000000003'), 'flags': b'\x00', 'cltv_expiry_delta': o(10), 'htlc_minimum_msat': o(250), 'fee_base_msat': o(100), 'fee_proportional_millionths': o(150)}) |
||||
|
p.on_channel_update({'short_channel_id': bfh('0000000000000004'), 'flags': b'\x01', 'cltv_expiry_delta': o(10), 'htlc_minimum_msat': o(250), 'fee_base_msat': o(100), 'fee_proportional_millionths': o(150)}) |
||||
|
p.on_channel_update({'short_channel_id': bfh('0000000000000004'), 'flags': b'\x00', 'cltv_expiry_delta': o(10), 'htlc_minimum_msat': o(250), 'fee_base_msat': o(100), 'fee_proportional_millionths': o(150)}) |
||||
|
p.on_channel_update({'short_channel_id': bfh('0000000000000005'), 'flags': b'\x01', 'cltv_expiry_delta': o(10), 'htlc_minimum_msat': o(250), 'fee_base_msat': o(100), 'fee_proportional_millionths': o(150)}) |
||||
|
p.on_channel_update({'short_channel_id': bfh('0000000000000005'), 'flags': b'\x00', 'cltv_expiry_delta': o(10), 'htlc_minimum_msat': o(250), 'fee_base_msat': o(100), 'fee_proportional_millionths': o(999)}) |
||||
|
p.on_channel_update({'short_channel_id': bfh('0000000000000006'), 'flags': b'\x00', 'cltv_expiry_delta': o(10), 'htlc_minimum_msat': o(250), 'fee_base_msat': o(100), 'fee_proportional_millionths': o(99999999)}) |
||||
|
p.on_channel_update({'short_channel_id': bfh('0000000000000006'), 'flags': b'\x01', 'cltv_expiry_delta': o(10), 'htlc_minimum_msat': o(250), 'fee_base_msat': o(100), 'fee_proportional_millionths': o(150)}) |
||||
|
self.assertNotEqual(None, fake_ln_worker.path_finder.find_path_for_payment(b'a', b'e', 100000)) |
||||
|
|
||||
|
|
||||
|
|
||||
|
def test_new_onion_packet(self): |
||||
|
# test vector from bolt-04 |
||||
|
payment_path_pubkeys = [ |
||||
|
bfh('02eec7245d6b7d2ccb30380bfbe2a3648cd7a942653f5aa340edcea1f283686619'), |
||||
|
bfh('0324653eac434488002cc06bbfb7f10fe18991e35f9fe4302dbea6d2353dc0ab1c'), |
||||
|
bfh('027f31ebc5462c1fdce1b737ecff52d37d75dea43ce11c74d25aa297165faa2007'), |
||||
|
bfh('032c0b7cf95324a07d05398b240174dc0c2be444d96b159aa6c7f7b1e668680991'), |
||||
|
bfh('02edabbd16b41c8371b92ef2f04c1185b4f03b6dcd52ba9b78d9d7c89c8f221145'), |
||||
|
] |
||||
|
session_key = bfh('4141414141414141414141414141414141414141414141414141414141414141') |
||||
|
associated_data = bfh('4242424242424242424242424242424242424242424242424242424242424242') |
||||
|
hops_data = [ |
||||
|
OnionHopsDataSingle(OnionPerHop( |
||||
|
bfh('0000000000000000'), bfh('0000000000000000'), bfh('00000000') |
||||
|
)), |
||||
|
OnionHopsDataSingle(OnionPerHop( |
||||
|
bfh('0101010101010101'), bfh('0000000000000001'), bfh('00000001') |
||||
|
)), |
||||
|
OnionHopsDataSingle(OnionPerHop( |
||||
|
bfh('0202020202020202'), bfh('0000000000000002'), bfh('00000002') |
||||
|
)), |
||||
|
OnionHopsDataSingle(OnionPerHop( |
||||
|
bfh('0303030303030303'), bfh('0000000000000003'), bfh('00000003') |
||||
|
)), |
||||
|
OnionHopsDataSingle(OnionPerHop( |
||||
|
bfh('0404040404040404'), bfh('0000000000000004'), bfh('00000004') |
||||
|
)), |
||||
|
] |
||||
|
packet = new_onion_packet(payment_path_pubkeys, session_key, hops_data, associated_data) |
||||
|
self.assertEqual(bfh('0002eec7245d6b7d2ccb30380bfbe2a3648cd7a942653f5aa340edcea1f283686619e5f14350c2a76fc232b5e46d421e9615471ab9e0bc887beff8c95fdb878f7b3a71da571226458c510bbadd1276f045c21c520a07d35da256ef75b4367962437b0dd10f7d61ab590531cf08000178a333a347f8b4072e216400406bdf3bf038659793a86cae5f52d32f3438527b47a1cfc54285a8afec3a4c9f3323db0c946f5d4cb2ce721caad69320c3a469a202f3e468c67eaf7a7cda226d0fd32f7b48084dca885d15222e60826d5d971f64172d98e0760154400958f00e86697aa1aa9d41bee8119a1ec866abe044a9ad635778ba61fc0776dc832b39451bd5d35072d2269cf9b040d6ba38b54ec35f81d7fc67678c3be47274f3c4cc472aff005c3469eb3bc140769ed4c7f0218ff8c6c7dd7221d189c65b3b9aaa71a01484b122846c7c7b57e02e679ea8469b70e14fe4f70fee4d87b910cf144be6fe48eef24da475c0b0bcc6565ae82cd3f4e3b24c76eaa5616c6111343306ab35c1fe5ca4a77c0e314ed7dba39d6f1e0de791719c241a939cc493bea2bae1c1e932679ea94d29084278513c77b899cc98059d06a27d171b0dbdf6bee13ddc4fc17a0c4d2827d488436b57baa167544138ca2e64a11b43ac8a06cd0c2fba2d4d900ed2d9205305e2d7383cc98dacb078133de5f6fb6bed2ef26ba92cea28aafc3b9948dd9ae5559e8bd6920b8cea462aa445ca6a95e0e7ba52961b181c79e73bd581821df2b10173727a810c92b83b5ba4a0403eb710d2ca10689a35bec6c3a708e9e92f7d78ff3c5d9989574b00c6736f84c199256e76e19e78f0c98a9d580b4a658c84fc8f2096c2fbea8f5f8c59d0fdacb3be2802ef802abbecb3aba4acaac69a0e965abd8981e9896b1f6ef9d60f7a164b371af869fd0e48073742825e9434fc54da837e120266d53302954843538ea7c6c3dbfb4ff3b2fdbe244437f2a153ccf7bdb4c92aa08102d4f3cff2ae5ef86fab4653595e6a5837fa2f3e29f27a9cde5966843fb847a4a61f1e76c281fe8bb2b0a181d096100db5a1a5ce7a910238251a43ca556712eaadea167fb4d7d75825e440f3ecd782036d7574df8bceacb397abefc5f5254d2722215c53ff54af8299aaaad642c6d72a14d27882d9bbd539e1cc7a527526ba89b8c037ad09120e98ab042d3e8652b31ae0e478516bfaf88efca9f3676ffe99d2819dcaeb7610a626695f53117665d267d3f7abebd6bbd6733f645c72c389f03855bdf1e4b8075b516569b118233a0f0971d24b83113c0b096f5216a207ca99a7cddc81c130923fe3d91e7508c9ac5f2e914ff5dccab9e558566fa14efb34ac98d878580814b94b73acbfde9072f30b881f7f0fff42d4045d1ace6322d86a97d164aa84d93a60498065cc7c20e636f5862dc81531a88c60305a2e59a985be327a6902e4bed986dbf4a0b50c217af0ea7fdf9ab37f9ea1a1aaa72f54cf40154ea9b269f1a7c09f9f43245109431a175d50e2db0132337baa0ef97eed0fcf20489da36b79a1172faccc2f7ded7c60e00694282d93359c4682135642bc81f433574aa8ef0c97b4ade7ca372c5ffc23c7eddd839bab4e0f14d6df15c9dbeab176bec8b5701cf054eb3072f6dadc98f88819042bf10c407516ee58bce33fbe3b3d86a54255e577db4598e30a135361528c101683a5fcde7e8ba53f3456254be8f45fe3a56120ae96ea3773631fcb3873aa3abd91bcff00bd38bd43697a2e789e00da6077482e7b1b1a677b5afae4c54e6cbdf7377b694eb7d7a5b913476a5be923322d3de06060fd5e819635232a2cf4f0731da13b8546d1d6d4f8d75b9fce6c2341a71b0ea6f780df54bfdb0dd5cd9855179f602f917265f21f9190c70217774a6fbaaa7d63ad64199f4664813b955cff954949076dcf'), |
||||
|
packet.to_bytes()) |
||||
|
|
||||
|
def test_process_onion_packet(self): |
||||
|
# this test is not from bolt-04, but is based on the one there; |
||||
|
# except here we have the privkeys for these pubkeys |
||||
|
payment_path_pubkeys = [ |
||||
|
bfh('03d75c0ee70f68d73d7d13aeb6261d8ace11416800860c7e59407afe4e2e2d42bb'), |
||||
|
bfh('03960a0b830c7b8e76de745b819f252c62508346196b916f5e813cdb0773283cce'), |
||||
|
bfh('0385620e0a571cbc3552620f8bf1bdcdab2d1a4a59c36fa10b8249114ccbdda40d'), |
||||
|
bfh('02ee242cf6c38b7285f0152c33804ff777f5c51fd352ca8132e845e2cf23b3d8ba'), |
||||
|
bfh('025c585fd2e174bf8245b2b4a119e52a417688904228643ea3edaa1728bf2a258e'), |
||||
|
] |
||||
|
payment_path_privkeys = [ |
||||
|
bfh('3463a278617b3dd83f79bda7f97673f12609c54386e1f0d2b67b1c6354fda14e'), |
||||
|
bfh('7e1255fddb52db1729fc3ceb21a46f95b8d9fe94cc83425e936a6c5223bb679d'), |
||||
|
bfh('c7ce8c1462c311eec24dff9e2532ac6241e50ae57e7d1833af21942136972f23'), |
||||
|
bfh('3d885f374d79a5e777459b083f7818cdc9493e5c4994ac9c7b843de8b70be661'), |
||||
|
bfh('dd72ab44729527b7942e195e7a835e7c71f9c0ff61844eb21274d9c26166a8f8'), |
||||
|
] |
||||
|
session_key = bfh('4141414141414141414141414141414141414141414141414141414141414141') |
||||
|
associated_data = bfh('4242424242424242424242424242424242424242424242424242424242424242') |
||||
|
hops_data = [ |
||||
|
OnionHopsDataSingle(OnionPerHop( |
||||
|
bfh('0000000000000000'), bfh('0000000000000000'), bfh('00000000') |
||||
|
)), |
||||
|
OnionHopsDataSingle(OnionPerHop( |
||||
|
bfh('0101010101010101'), bfh('0000000000000001'), bfh('00000001') |
||||
|
)), |
||||
|
OnionHopsDataSingle(OnionPerHop( |
||||
|
bfh('0202020202020202'), bfh('0000000000000002'), bfh('00000002') |
||||
|
)), |
||||
|
OnionHopsDataSingle(OnionPerHop( |
||||
|
bfh('0303030303030303'), bfh('0000000000000003'), bfh('00000003') |
||||
|
)), |
||||
|
OnionHopsDataSingle(OnionPerHop( |
||||
|
bfh('0404040404040404'), bfh('0000000000000004'), bfh('00000004') |
||||
|
)), |
||||
|
] |
||||
|
packet = new_onion_packet(payment_path_pubkeys, session_key, hops_data, associated_data) |
||||
|
self.assertEqual(bfh('0002eec7245d6b7d2ccb30380bfbe2a3648cd7a942653f5aa340edcea1f28368661954176cd9869da33d713aa219fcef1e5c806fef11e696bcc66844de8271c27974a0fd57c2dbcb2c6dd4e8ef35d96db28d5a0e49b6ab3d6de31af65950723b8cddc108390bebf8d149002e31bdc283056477ba27c8054c248ad7306de31663a7c99ec659da15d0f6fbc7e1687485b39e9be0ec3b70164cb3618a9b546317e7c2d62ae9f0f840704535729262d30c6132d1b390f073edec8fa057176c6268b6ad06a82ff0d16d4c662194873e8b4ecf46eb2c9d4d58d2ee2021adb19840605ac5afd8bd942dd71e8244c83e28b2ed5a3b09e9e7df5c8c747e5765ba366a4f7407a6c6b0a32f74bc5e428f7fa4c3cf70e13ed91563177d94190d5149aa4b9c96d00e40d2ac35ab9c4a621ce0f6f5df7d64a9c8d435db19de192d9db522c7f7b4e201fc1b61a9bd3efd062ae24455d463818b01e2756c7d0691bc3ac4c017be34c9a8b2913bb1b937e31e0ae40f650a7cd820bcb4996825b1cbad1ff7ccc2b513b1104524c34f6573e1b59201c005a632ee5dccd3711a32e3ba1ff00fcffbe636e4b3a84bbe491b836a57ccec138b8cc2ec733846904d872f305d538d51db8e56232ec6e07877075328874cb7b09c7e799100a9ff085dead253886b174fc408a0ea7b48bce2c5d8992285011960af088f7e006ef60089d46ac9aa15acfac6c87c3cf6904764dd785419292fbafa9cca09c8ade24a6cd63f12d1cfc83fa35cf2f1cf503c39cbf78293f06c68a3cece7177169cd872bb49bf69d933a27a887dd9daefa9239fca9f0c3e309ec61d9df947211da98cf11a6e0fb77252629cdf9f2226dd69ca73fa51be4df224592f8d471b69a1aebbdaa2f3a798b3581253d97feb0a12e6606043ca0fc5efc0f49b8061d6796eff31cd8638499e2f25ffb96eec32837438ed7ebebbe587886648f63e35d80f41869f4c308f2e6970bd65fead5e8544e3239a6acc9d996b08d1546455bcafbe88ed3ed547714841946fe2e77180e4d7bf1452414e4b1745a7897184a2c4cbc3ac46f83342a55a48e29dc8f17cf595dd28f51e297ba89fd25ed0dbd1c0081a810beaab09758a36fbfd16fbdc3daa9fe05c8a73195f244ef2743a5df761f01ee6e693eb6c7f1a7834fab3671391e5ddebf611e119a2ae4456e2cee7a6d4f27a2246cdb1f8ef35f0b3d7044b3799d8d0ed0a6470557fd807c065d6d83acba07e96e10770ada8c0b4d4921522944188d5f30086a6ee0a4795331273f32beaaa43363fc58208a257e5c5c434c7325b583642219d81c7d67b908d5263b42ac1991edc69a777da60f38eff138c844af9e549374e8b29b166211bfded24587a29394e33828b784da7e7b62ab7e49ea2693fcdd17fa96186a5ef11ef1a8adffa50f93a3119e95e6c09014f3e3b0709183fa08a826ced6deb4608b7d986ebbcf99ad58e25451d4d9d38d0059734d8501467b97182cd11e0c07c91ca50f61cc31255a3147ade654976a5989097281892aafd8df595c63bd14f1e03f5955a9398d2dd6368bbcae833ae1cc2df31eb0980b4817dfd130020ffb275743fcc01df40e3ecda1c5988e8e1bde965353b0b1bf34ea05f095000c45b6249618d275905a24d3eb58c600aeab4fb552fbf1ccdb2a5c80ace220310f89829d7e53f78c126037b6d8d500220c7a118d9621b4d6bd5379edd7e24bcf540e87aba6b88862db16fa4ee00b009fda80577be67ab94910fd8a7807dfe4ebe66b8fdcd040aa2dc17ec22639298be56b2a2c9d8940647b75f2f6d81746df16e1cb2f05e23397a8c63baea0803441ff4b7d517ff172980a056726235e2f6af85e8aa9b91ba85f14532272d6170df3166b91169dc09d4f4a251610f57ff0885a93364cfaf650bdf436c89795efed5ca934bc7ffc0a4'), |
||||
|
packet.to_bytes()) |
||||
|
for i, privkey in enumerate(payment_path_privkeys): |
||||
|
processed_packet = lnrouter.process_onion_packet(packet, associated_data, privkey) |
||||
|
self.assertEqual(hops_data[i].per_hop.to_bytes(), processed_packet.hop_data.per_hop.to_bytes()) |
||||
|
packet = processed_packet.next_packet |
||||
|
|
||||
|
def test_decode_onion_error(self): |
||||
|
# test vector from bolt-04 |
||||
|
payment_path_pubkeys = [ |
||||
|
bfh('02eec7245d6b7d2ccb30380bfbe2a3648cd7a942653f5aa340edcea1f283686619'), |
||||
|
bfh('0324653eac434488002cc06bbfb7f10fe18991e35f9fe4302dbea6d2353dc0ab1c'), |
||||
|
bfh('027f31ebc5462c1fdce1b737ecff52d37d75dea43ce11c74d25aa297165faa2007'), |
||||
|
bfh('032c0b7cf95324a07d05398b240174dc0c2be444d96b159aa6c7f7b1e668680991'), |
||||
|
bfh('02edabbd16b41c8371b92ef2f04c1185b4f03b6dcd52ba9b78d9d7c89c8f221145'), |
||||
|
] |
||||
|
session_key = bfh('4141414141414141414141414141414141414141414141414141414141414141') |
||||
|
error_packet_for_node_0 = bfh('9c5add3963fc7f6ed7f148623c84134b5647e1306419dbe2174e523fa9e2fbed3a06a19f899145610741c83ad40b7712aefaddec8c6baf7325d92ea4ca4d1df8bce517f7e54554608bf2bd8071a4f52a7a2f7ffbb1413edad81eeea5785aa9d990f2865dc23b4bc3c301a94eec4eabebca66be5cf638f693ec256aec514620cc28ee4a94bd9565bc4d4962b9d3641d4278fb319ed2b84de5b665f307a2db0f7fbb757366067d88c50f7e829138fde4f78d39b5b5802f1b92a8a820865af5cc79f9f30bc3f461c66af95d13e5e1f0381c184572a91dee1c849048a647a1158cf884064deddbf1b0b88dfe2f791428d0ba0f6fb2f04e14081f69165ae66d9297c118f0907705c9c4954a199bae0bb96fad763d690e7daa6cfda59ba7f2c8d11448b604d12d') |
||||
|
decoded_error, index_of_sender = lnrouter._decode_onion_error(error_packet_for_node_0, payment_path_pubkeys, session_key) |
||||
|
self.assertEqual(bfh('4c2fc8bc08510334b6833ad9c3e79cd1b52ae59dfe5c2a4b23ead50f09f7ee0b0002200200fe0000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000'), |
||||
|
decoded_error) |
||||
|
self.assertEqual(4, index_of_sender) |
||||
Loading…
Reference in new issue