commit ee87920573461cd1ecbc85a3fe588200dc0bd8ae
parent 3a20c8ce000aa8f270d407da9d2ad03f7d980d4b
Author: Janus <ysangkok@gmail.com>
Date: Fri, 11 May 2018 18:32:55 +0200
lnbase: store remote revocation store, don't store all remote revocation points, verify ctn numbers in reestablish
Diffstat:
3 files changed, 44 insertions(+), 11 deletions(-)
diff --git a/lib/lnbase.py b/lib/lnbase.py
@@ -276,7 +276,7 @@ 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_sat", "commitment_points"])
+RemoteState = namedtuple("RemoteState", ["ctn", "next_per_commitment_point", "amount_sat", "revocation_store", "last_per_commitment_point"])
LocalState = namedtuple("LocalState", ["ctn", "per_commitment_secret_seed", "amount_sat"])
ChannelConstraints = namedtuple("ChannelConstraints", ["feerate", "capacity", "is_initiator", "funding_txn_minimum_depth"])
OpenChannel = namedtuple("OpenChannel", ["channel_id", "funding_outpoint", "local_config", "remote_config", "remote_state", "local_state", "constraints"])
@@ -911,6 +911,7 @@ class Peer(PrintError):
# broadcast funding tx
success, _txid = self.network.broadcast(funding_tx)
assert success, success
+ their_revocation_store = RevocationStore()
chan = OpenChannel(
channel_id=channel_id,
funding_outpoint=Outpoint(funding_txid, funding_index),
@@ -919,8 +920,9 @@ class Peer(PrintError):
remote_state=RemoteState(
ctn = 0,
next_per_commitment_point=None,
+ last_per_commitment_point=remote_per_commitment_point,
amount_sat=remote_amount,
- commitment_points=[bh2u(remote_per_commitment_point)]
+ revocation_store=their_revocation_store
),
local_state=LocalState(
ctn = 0,
@@ -943,7 +945,15 @@ class Peer(PrintError):
# 'your_last_per_commitment_secret': b'\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00',
# 'my_current_per_commitment_point': b'\x03\x18\xb9\x1b\x99\xd4\xc3\xf1\x92\x0f\xfe\xe4c\x9e\xae\xa4\xf1\xdeX\xcf4\xa9[\xd1\tAh\x80\x88\x01b*['
# }
- if channel_reestablish_msg["my_current_per_commitment_point"] != bfh(chan.remote_state.commitment_points[-1]):
+ remote_ctn = int.from_bytes(channel_reestablish_msg["next_local_commitment_number"], 'big')
+ if remote_ctn != chan.remote_state.ctn + 1:
+ raise Exception("expected remote ctn {}, got {}".format(chan.remote_state.ctn + 1, remote_ctn))
+
+ local_ctn = int.from_bytes(channel_reestablish_msg["next_remote_revocation_number"], 'big')
+ if local_ctn != chan.local_state.ctn:
+ raise Exception("expected local ctn {}, got {}".format(chan.local_state.ctn, local_ctn))
+
+ if channel_reestablish_msg["my_current_per_commitment_point"] != chan.remote_state.last_per_commitment_point:
raise Exception("Remote PCP mismatch")
self.send_message(gen_msg("channel_reestablish",
channel_id=chan.channel_id,
@@ -991,10 +1001,10 @@ class Peer(PrintError):
return chan._replace(remote_state=chan.remote_state._replace(next_per_commitment_point=remote_funding_locked_msg["next_per_commitment_point"]))
async def receive_commitment_revoke_ack(self, chan, expected_received_sat, payment_preimage):
- def derive_and_incr():
+ def derive_and_incr(last = False):
nonlocal chan
last_secret = get_per_commitment_secret_from_seed(chan.local_state.per_commitment_secret_seed, 2**48-chan.local_state.ctn-1)
- next_secret = get_per_commitment_secret_from_seed(chan.local_state.per_commitment_secret_seed, 2**48-chan.local_state.ctn-2)
+ next_secret = get_per_commitment_secret_from_seed(chan.local_state.per_commitment_secret_seed, 2**48-chan.local_state.ctn-(2 if not last else 3))
next_point = secret_to_pubkey(int.from_bytes(next_secret, 'big'))
chan = chan._replace(
local_state=chan.local_state._replace(
@@ -1002,6 +1012,9 @@ class Peer(PrintError):
)
)
return last_secret, next_point
+
+ their_revstore = chan.remote_state.revocation_store
+
channel_id = chan.channel_id
try:
commitment_signed_msg = await self.commitment_signed[channel_id]
@@ -1052,6 +1065,8 @@ class Peer(PrintError):
print("SENDING FIRST REVOKE AND ACK")
+ their_revstore.add_next_entry(last_secret)
+
self.send_message(gen_msg("revoke_and_ack",
channel_id=channel_id,
per_commitment_secret=last_secret,
@@ -1111,7 +1126,9 @@ class Peer(PrintError):
# TODO check commitment_signed results
- last_secret, next_point = derive_and_incr()
+ last_secret, next_point = derive_and_incr(True)
+
+ their_revstore.add_next_entry(last_secret)
print("SENDING SECOND REVOKE AND ACK")
self.send_message(gen_msg("revoke_and_ack",
@@ -1125,7 +1142,8 @@ class Peer(PrintError):
),
remote_state=chan.remote_state._replace(
ctn=chan.remote_state.ctn + 2,
- commitment_points=chan.remote_state.commitment_points + [bh2u(remote_next_commitment_point)],
+ revocation_store=their_revstore,
+ last_per_commitment_point=remote_next_commitment_point,
next_per_commitment_point=revoke_and_ack_msg["next_per_commitment_point"],
amount_sat=chan.remote_state.amount_sat - expected_received_sat
)
@@ -1575,3 +1593,14 @@ class RevocationStore:
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 self.buckets == o.buckets and self.index == o.index
diff --git a/lib/tests/test_lnbase.py b/lib/tests/test_lnbase.py
@@ -7,7 +7,7 @@ from lib.lnbase import make_commitment, get_obscured_ctn, Peer, make_offered_htl
from lib.lnbase import secret_to_pubkey, derive_pubkey, derive_privkey, derive_blinded_pubkey, overall_weight
from lib.lnbase import make_htlc_tx_output, make_htlc_tx_inputs, get_per_commitment_secret_from_seed
from lib.lnbase import make_htlc_tx_witness, OnionHopsDataSingle, new_onion_packet, OnionPerHop
-from lib.lnbase import RevocationStore, ShachainElement, shachain_derive
+from lib.lnbase import RevocationStore
from lib.transaction import Transaction
from lib import bitcoin
import ecdsa.ellipticcurve
@@ -790,8 +790,9 @@ class Test_LNBase(unittest.TestCase):
seed = bitcoin.sha256(b"shachaintest")
consumer = RevocationStore()
for i in range(10000):
- secret = shachain_derive(ShachainElement(seed, 0), 2**48 - i - 1).secret
+ secret = get_per_commitment_secret_from_seed(seed, 2**48 - i - 1)
try:
consumer.add_next_entry(secret)
except Exception as e:
raise Exception("iteration " + str(i) + ": " + str(e))
+ if i % 1000 == 0: self.assertEqual(consumer.serialize(), RevocationStore.from_json_obj(json.loads(json.dumps(consumer.serialize()))).serialize())
diff --git a/lib/tests/test_lnbase_online.py b/lib/tests/test_lnbase_online.py
@@ -13,14 +13,14 @@ from lib.simple_config import SimpleConfig
from lib.network import Network
from lib.storage import WalletStorage
from lib.wallet import Wallet
-from lib.lnbase import Peer, node_list, Outpoint, ChannelConfig, LocalState, RemoteState, Keypair, OnlyPubkeyKeypair, OpenChannel, ChannelConstraints
+from lib.lnbase import Peer, node_list, Outpoint, ChannelConfig, LocalState, RemoteState, Keypair, OnlyPubkeyKeypair, OpenChannel, ChannelConstraints, RevocationStore
from lib.lightning_payencode.lnaddr import lnencode, LnAddr
import lib.constants as constants
is_key = lambda k: k.endswith("_basepoint") or k.endswith("_key")
def maybeDecode(k, v):
- if k in ["pubkey", "privkey", "next_per_commitment_point", "per_commitment_secret_seed"] and v is not None:
+ if k in ["pubkey", "privkey", "last_per_commitment_point", "next_per_commitment_point", "per_commitment_secret_seed"] and v is not None:
return binascii.unhexlify(v)
return v
@@ -45,6 +45,7 @@ def reconstruct_namedtuples(openingchannel):
new_local_state = decodeAll(openingchannel.local_state)
openingchannel = openingchannel._replace(local_state=LocalState(**new_local_state))
new_remote_state = decodeAll(openingchannel.remote_state)
+ new_remote_state["revocation_store"] = RevocationStore.from_json_obj(new_remote_state["revocation_store"])
openingchannel = openingchannel._replace(remote_state=RemoteState(**new_remote_state))
openingchannel = openingchannel._replace(constraints=ChannelConstraints(**openingchannel.constraints))
return openingchannel
@@ -58,6 +59,8 @@ def serialize_channels(channels):
def default(self, o):
if isinstance(o, bytes):
return binascii.hexlify(o).decode("ascii")
+ if isinstance(o, RevocationStore):
+ return o.serialize()
return super(MyJsonEncoder, self)
dumped = MyJsonEncoder().encode(serialized_channels)
roundtripped = json.loads(dumped)
