mirror of
https://github.com/LBRYFoundation/LBRY-Vault.git
synced 2025-08-31 01:11:35 +00:00
lnbase: channel establishment flow, avoid using Wallet instance
This commit is contained in:
Janus
SomberNight
parent
2fd5bae0fd
commit
21b2fa1b1a
1 changed files with 34 additions and 26 deletions
|
|
@ -4,6 +4,7 @@
|
|||
Derived from https://gist.github.com/AdamISZ/046d05c156aaeb56cc897f85eecb3eb8
|
||||
"""
|
||||
|
||||
import itertools
|
||||
import json
|
||||
from collections import OrderedDict
|
||||
import asyncio
|
||||
|
|
@ -16,6 +17,7 @@ import hmac
|
|||
import cryptography.hazmat.primitives.ciphers.aead as AEAD
|
||||
|
||||
from .bitcoin import public_key_from_private_key, ser_to_point, point_to_ser, string_to_number, deserialize_privkey, EC_KEY, rev_hex
|
||||
from . import bitcoin
|
||||
from .constants import set_testnet, set_simnet
|
||||
from . import constants
|
||||
from .util import PrintError
|
||||
|
|
@ -213,16 +215,11 @@ def create_ephemeral_key(privkey):
|
|||
return (privkey[:32], pub)
|
||||
|
||||
def get_unused_public_keys():
|
||||
path = os.environ["HOME"] + "/.electrum/" + ("testnet" if constants.net is constants.BitcoinTestnet else "simnet") + "/wallets/default_wallet"
|
||||
os.stat(path)
|
||||
WALLET = Wallet(WalletStorage(path))
|
||||
for str_address in WALLET.get_unused_addresses():
|
||||
pri, redeem_script = WALLET.export_private_key(str_address, None)
|
||||
|
||||
typ, pri, compressed = deserialize_privkey(pri)
|
||||
pubk = binascii.unhexlify(EC_KEY(pri).get_public_key(True))
|
||||
yield pubk
|
||||
|
||||
xprv, xpub = bitcoin.bip32_root(b"testseed", "p2wpkh")
|
||||
for i in itertools.count():
|
||||
childxpub = bitcoin.bip32_public_derivation(xpub, "m/", "m/42/"+str(i))
|
||||
_, _, _, _, child_c, child_cK = bitcoin.deserialize_xpub(childxpub)
|
||||
yield child_cK
|
||||
|
||||
class Peer(PrintError):
|
||||
|
||||
|
|
@ -233,6 +230,7 @@ class Peer(PrintError):
|
|||
self.pubkey = pubkey
|
||||
self.read_buffer = b''
|
||||
self.ping_time = 0
|
||||
self.temporary_channel_id_to_incoming_accept_channel = {}
|
||||
|
||||
def diagnostic_name(self):
|
||||
return self.host
|
||||
|
|
@ -314,13 +312,7 @@ class Peer(PrintError):
|
|||
self.send_message(gen_msg('pong', byteslen=l))
|
||||
|
||||
def on_accept_channel(self, payload):
|
||||
# check that it is in my pending requests
|
||||
# I need to attach a wallet to each request
|
||||
if ok:
|
||||
tx = wallet.create_funding_tx()
|
||||
wallet.sign(tx)
|
||||
m = gen_msg('funding created', signature)
|
||||
self.send_message(m)
|
||||
self.temporary_channel_id_to_incoming_accept_channel[payload["temporary_channel_id"]].set_result(payload)
|
||||
|
||||
def on_funding_signed(self, payload):
|
||||
sig = payload['signature']
|
||||
|
|
@ -331,10 +323,10 @@ class Peer(PrintError):
|
|||
def on_funding_locked(self, payload):
|
||||
pass
|
||||
|
||||
def open_channel(self, funding_sat, push_msat):
|
||||
self.send_message(gen_msg('open_channel', funding_satoshis=funding_sat, push_msat=push_msat))
|
||||
#def open_channel(self, funding_sat, push_msat):
|
||||
# self.send_message(gen_msg('open_channel', funding_satoshis=funding_sat, push_msat=push_msat))
|
||||
|
||||
async def main_loop(self, loop):
|
||||
async def initialize(self, loop):
|
||||
self.reader, self.writer = await asyncio.open_connection(self.host, self.port, loop=loop)
|
||||
await self.handshake()
|
||||
# read init
|
||||
|
|
@ -343,11 +335,7 @@ class Peer(PrintError):
|
|||
# send init
|
||||
self.send_message(gen_msg("init", gflen=0, lflen=0))
|
||||
|
||||
pubkeys = get_unused_public_keys()
|
||||
|
||||
msg = gen_msg("open_channel", chain_hash=bytes.fromhex(rev_hex(constants.net.GENESIS)), funding_satoshis=20000, max_accepted_htlcs=5, funding_pubkey=next(pubkeys), revocation_basepoint=next(pubkeys), htlc_basepoint=next(pubkeys), payment_basepoint=next(pubkeys), delayed_payment_basepoint=next(pubkeys), first_per_commitment_point=next(pubkeys))
|
||||
self.send_message(msg)
|
||||
|
||||
async def main_loop(self, loop):
|
||||
# loop
|
||||
while True:
|
||||
self.ping_if_required()
|
||||
|
|
@ -357,6 +345,23 @@ class Peer(PrintError):
|
|||
self.print_error('closing lnbase')
|
||||
self.writer.close()
|
||||
|
||||
async def channel_establishment_flow(self):
|
||||
pubkeys = get_unused_public_keys()
|
||||
|
||||
temp_channel_id = os.urandom(32)
|
||||
msg = gen_msg("open_channel", temporary_channel_id=temp_channel_id, chain_hash=bytes.fromhex(rev_hex(constants.net.GENESIS)), funding_satoshis=20000, max_accepted_htlcs=5, funding_pubkey=next(pubkeys), revocation_basepoint=next(pubkeys), htlc_basepoint=next(pubkeys), payment_basepoint=next(pubkeys), delayed_payment_basepoint=next(pubkeys), first_per_commitment_point=next(pubkeys))
|
||||
self.temporary_channel_id_to_incoming_accept_channel[temp_channel_id] = asyncio.Future()
|
||||
self.send_message(msg)
|
||||
accept_channel = await self.temporary_channel_id_to_incoming_accept_channel[temp_channel_id]
|
||||
del self.temporary_channel_id_to_incoming_accept_channel[temp_channel_id]
|
||||
|
||||
# check that it is in my pending requests
|
||||
# I need to attach a wallet to each request
|
||||
if ok:
|
||||
tx = wallet.create_funding_tx()
|
||||
wallet.sign(tx)
|
||||
m = gen_msg('funding created', signature)
|
||||
self.send_message(m)
|
||||
|
||||
# replacement for lightningCall
|
||||
class LNWallet(Wallet):
|
||||
|
|
@ -391,5 +396,8 @@ if __name__ == "__main__":
|
|||
privkey = b"\x21"*32 + b"\x01"
|
||||
peer = Peer(privkey, host, port, pubkey)
|
||||
loop = asyncio.get_event_loop()
|
||||
loop.run_until_complete(peer.main_loop(loop))
|
||||
loop.run_until_complete(peer.initialize(loop))
|
||||
async def asynctest():
|
||||
await peer.channel_establishment_flow()
|
||||
loop.run_until_complete(asyncio.gather(asynctest(), peer.main_loop(loop)))
|
||||
loop.close()
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue