1 import * as rawldk from "../bindings.mjs";
2 import * as ldk from "../index.mjs";
4 const tests: Array<Function> = [];
6 function array_eq(a: Uint8Array, b: Uint8Array): boolean {
7 return a.length == b.length && a.every((v, idx) => v == b[idx]);
10 tests.push(async () => {
11 const result = rawldk.CResult_boolLightningErrorZ_ok(true);
12 console.assert(rawldk.CResult_boolLightningErrorZ_is_ok(result));
13 console.assert(rawldk.CResult_boolLightningErrorZ_get_ok(result));
14 rawldk.CResult_boolLightningErrorZ_free(result);
15 const second_res = rawldk.CResult_boolLightningErrorZ_ok(false);
16 console.assert(rawldk.CResult_boolLightningErrorZ_is_ok(second_res));
17 console.assert(!rawldk.CResult_boolLightningErrorZ_get_ok(second_res));
18 rawldk.CResult_boolLightningErrorZ_free(second_res);
23 tests.push(async () => {
24 const ping = ldk.Ping.constructor_new(10, 2);
25 const new_ping = ldk.Ping.constructor_read(ping.write());
26 if (!(new_ping instanceof ldk.Result_PingDecodeErrorZ_OK)) return false;
27 if (!new_ping.is_ok()) return false;
28 if (new_ping.res.get_byteslen() != 2) return false;
29 if (new_ping.res.get_ponglen() != 10) return false;
33 tests.push(async () => {
34 const outpoint = ldk.OutPoint.constructor_new(new Uint8Array(32), 4);
35 const read_outpoint = ldk.OutPoint.constructor_read(outpoint.write());
36 if (!(read_outpoint instanceof ldk.Result_OutPointDecodeErrorZ_OK)) return false;
37 if (!read_outpoint.res.eq(outpoint)) return false;
38 if (read_outpoint.res.hash() != outpoint.hash()) return false;
39 const chan_id = read_outpoint.res.to_channel_id();
40 if (chan_id.length != 32) return false;
41 if (chan_id[31] != 4) return false;
45 tests.push(async () => {
46 const outpoint = ldk.OutPoint.constructor_new(new Uint8Array(32), 4);
47 const read_outpoint = ldk.OutPoint.constructor_read(outpoint.write());
48 if (!(read_outpoint instanceof ldk.Result_OutPointDecodeErrorZ_OK)) return false;
49 if (!read_outpoint.res.eq(outpoint)) return false;
50 if (read_outpoint.res.hash() != outpoint.hash()) return false;
51 const chan_id = read_outpoint.res.to_channel_id();
52 if (chan_id.length != 32) return false;
53 if (chan_id[31] != 4) return false;
59 constructor(public chan_man: ldk.ChannelManager, public tx_broadcasted: Promise<Uint8Array>, public logger: ldk.Logger,
60 public node_id: Uint8Array, public node_secret: Uint8Array) {}
62 function get_chanman(): Node {
63 const fee_est = ldk.FeeEstimator.new_impl({
64 get_est_sat_per_1000_weight(confirmation_target: ldk.ConfirmationTarget): number {
67 } as ldk.FeeEstimatorInterface);
69 const tx_broadcasted: Promise<Uint8Array> = new Promise((resolve, reject) => {
70 tx_broadcaster = ldk.BroadcasterInterface.new_impl({
71 broadcast_transaction(tx: Uint8Array): void { console.log("Tx Broadcast: " + tx); resolve(tx); }
72 } as ldk.BroadcasterInterfaceInterface);
74 const logger = ldk.Logger.new_impl({
75 log(record: ldk.Record): void {
76 console.log(record.get_module_path() + ": " + record.get_args());
78 } as ldk.LoggerInterface);
80 const persister = ldk.Persist.new_impl({
81 persist_new_channel(channel_id: ldk.OutPoint, data: ldk.ChannelMonitor, update_id: ldk.MonitorUpdateId): ldk.Result_NoneChannelMonitorUpdateErrZ {
82 return ldk.Result_NoneChannelMonitorUpdateErrZ.constructor_ok();
84 update_persisted_channel(channel_id: ldk.OutPoint, update: ldk.ChannelMonitorUpdate, data: ldk.ChannelMonitor, update_id: ldk.MonitorUpdateId): ldk.Result_NoneChannelMonitorUpdateErrZ {
85 return ldk.Result_NoneChannelMonitorUpdateErrZ.constructor_ok();
87 } as ldk.PersistInterface);
89 const chain_monitor = ldk.ChainMonitor.constructor_new(ldk.Option_FilterZ.constructor_none(), tx_broadcaster, logger, fee_est, persister);
90 const chain_watch: ldk.Watch = chain_monitor.as_Watch();
92 const seed = new Uint8Array(32);
93 seed.fill(seed_counter);
95 const keys_manager = ldk.KeysManager.constructor_new(seed, BigInt(42), 42);
96 const keys_interface = keys_manager.as_KeysInterface();
97 const config = ldk.UserConfig.constructor_default();
98 const params = ldk.ChainParameters.constructor_new(ldk.Network.LDKNetwork_Testnet, ldk.BestBlock.constructor_from_genesis(ldk.Network.LDKNetwork_Testnet));
100 const chan_man = ldk.ChannelManager.constructor_new(fee_est, chain_watch, tx_broadcaster, logger, keys_interface, config, params);
102 chan_man, tx_broadcasted, logger, chan_man.get_our_node_id(),
103 (keys_interface.get_node_secret(ldk.Recipient.LDKRecipient_Node) as ldk.Result_SecretKeyNoneZ_OK).res
107 function exchange_messages(a: ldk.ChannelManager, b: ldk.ChannelManager) {
108 var found_msgs = true;
110 const as_msgs = a.as_MessageSendEventsProvider().get_and_clear_pending_msg_events();
111 const bs_msgs = b.as_MessageSendEventsProvider().get_and_clear_pending_msg_events();
112 found_msgs = as_msgs.length != 0 || bs_msgs.length != 0;
113 for (var i = 0; i < 2; i++) {
114 var to: ldk.ChannelManager; var from: ldk.ChannelManager; var msgs: ldk.MessageSendEvent[];
115 if (i == 0) { from = a; to = b; msgs = as_msgs; } else { from = b; to = a; msgs = bs_msgs; }
116 for (var j = 0; j < msgs.length; j++) {
118 if (msg instanceof ldk.MessageSendEvent_SendOpenChannel) {
119 if (!array_eq(msg.node_id, to.get_our_node_id())) return false;
120 to.as_ChannelMessageHandler().handle_open_channel(from.get_our_node_id(), ldk.InitFeatures.constructor_known(), msg.msg);
121 } else if (msg instanceof ldk.MessageSendEvent_SendAcceptChannel) {
122 if (!array_eq(msg.node_id, to.get_our_node_id())) return false;
123 to.as_ChannelMessageHandler().handle_accept_channel(from.get_our_node_id(), ldk.InitFeatures.constructor_known(), msg.msg);
124 } else if (msg instanceof ldk.MessageSendEvent_SendFundingCreated) {
125 if (!array_eq(msg.node_id, to.get_our_node_id())) return false;
126 to.as_ChannelMessageHandler().handle_funding_created(from.get_our_node_id(), msg.msg);
127 } else if (msg instanceof ldk.MessageSendEvent_SendFundingSigned) {
128 if (!array_eq(msg.node_id, to.get_our_node_id())) return false;
129 to.as_ChannelMessageHandler().handle_funding_signed(from.get_our_node_id(), msg.msg);
139 function assign_u64(arr: Uint8Array, offset: number, value: bigint) {
140 arr[offset + 0] = Number((value >> BigInt(8 * 0)) & BigInt(0xff));
141 arr[offset + 1] = Number((value >> BigInt(8 * 1)) & BigInt(0xff));
142 arr[offset + 2] = Number((value >> BigInt(8 * 2)) & BigInt(0xff));
143 arr[offset + 3] = Number((value >> BigInt(8 * 3)) & BigInt(0xff));
144 arr[offset + 4] = Number((value >> BigInt(8 * 4)) & BigInt(0xff));
145 arr[offset + 5] = Number((value >> BigInt(8 * 5)) & BigInt(0xff));
146 arr[offset + 6] = Number((value >> BigInt(8 * 6)) & BigInt(0xff));
147 arr[offset + 7] = Number((value >> BigInt(8 * 7)) & BigInt(0xff));
150 function get_event(chan_man: ldk.ChannelManager): ldk.Event {
151 const events: Array<ldk.Event> = [];
152 const event_handler = ldk.EventHandler.new_impl({
153 handle_event(event: ldk.Event): void {
156 } as ldk.EventHandlerInterface);
158 chan_man.as_EventsProvider().process_pending_events(event_handler);
159 console.assert(events.length == 1);
163 tests.push(async () => {
164 const a = get_chanman();
165 const b = get_chanman();
167 a.chan_man.as_ChannelMessageHandler().peer_connected(b.chan_man.get_our_node_id(), ldk.Init.constructor_new(ldk.InitFeatures.constructor_known(), ldk.Option_NetAddressZ.constructor_none()));
168 b.chan_man.as_ChannelMessageHandler().peer_connected(a.chan_man.get_our_node_id(), ldk.Init.constructor_new(ldk.InitFeatures.constructor_known(), ldk.Option_NetAddressZ.constructor_none()));
170 const chan_create_err = a.chan_man.create_channel(b.chan_man.get_our_node_id(), BigInt(0), BigInt(400), BigInt(0), ldk.UserConfig.constructor_default());
171 if (chan_create_err.is_ok()) return false;
172 if (!(chan_create_err instanceof ldk.Result__u832APIErrorZ_Err)) return false;
173 if (!(chan_create_err.err instanceof ldk.APIError_APIMisuseError)) return false;
174 if (chan_create_err.err.err != "Channel value must be at least 1000 satoshis. It was 0") return false;
176 const chan_create_res = a.chan_man.create_channel(b.chan_man.get_our_node_id(), BigInt(1000000), BigInt(400), BigInt(0), ldk.UserConfig.constructor_default());
177 if (!chan_create_res.is_ok()) return false;
179 if (!exchange_messages(a.chan_man, b.chan_man)) return false;
181 const event = get_event(a.chan_man) as ldk.Event_FundingGenerationReady;
183 // (very) manually create a funding transaction
184 const witness_pos = event.output_script.length + 58;
185 const funding_tx = new Uint8Array(witness_pos + 7);
186 funding_tx[0] = 2; // 4-byte tx version 2
187 funding_tx[4] = 0; funding_tx[5] = 1; // segwit magic bytes
188 funding_tx[6] = 1; // 1-byte input count 1
189 // 36 bytes previous outpoint all-0s
190 funding_tx[43] = 0; // 1-byte input script length 0
191 funding_tx[44] = 0xff; funding_tx[45] = 0xff; funding_tx[46] = 0xff; funding_tx[47] = 0xff; // 4-byte nSequence
192 funding_tx[48] = 1; // one output
193 assign_u64(funding_tx, 49, event.channel_value_satoshis);
194 funding_tx[57] = event.output_script.length; // 1-byte output script length
195 funding_tx.set(event.output_script, 58);
196 funding_tx[witness_pos] = 1; funding_tx[witness_pos + 1] = 1; funding_tx[witness_pos + 2] = 0xff; // one witness element of size 1 with contents 0xff
197 funding_tx[witness_pos + 3] = 0; funding_tx[witness_pos + 4] = 0; funding_tx[witness_pos + 5] = 0; funding_tx[witness_pos + 6] = 0; // lock time 0
199 const funding_res = a.chan_man.funding_transaction_generated(event.temporary_channel_id, event.counterparty_node_id, funding_tx);
200 if (!(funding_res instanceof ldk.Result_NoneAPIErrorZ_OK)) return false;
202 if (!exchange_messages(a.chan_man, b.chan_man)) return false;
204 const tx_broadcasted: Uint8Array = (await a.tx_broadcasted);
205 if (!array_eq(tx_broadcasted, funding_tx)) return false;
210 tests.push(async () => {
211 const a = get_chanman();
212 const b = get_chanman();
214 const ignorer = ldk.IgnoringMessageHandler.constructor_new();
215 const pm_a = ldk.PeerManager.constructor_new(a.chan_man.as_ChannelMessageHandler(), ignorer.as_RoutingMessageHandler(), a.node_secret, a.node_secret, a.logger, ignorer.as_CustomMessageHandler());
216 const pm_b = ldk.PeerManager.constructor_new(b.chan_man.as_ChannelMessageHandler(), ignorer.as_RoutingMessageHandler(), b.node_secret, b.node_secret, b.logger, ignorer.as_CustomMessageHandler());
218 var sock_b: ldk.SocketDescriptor;
219 const sock_a = ldk.SocketDescriptor.new_impl({
220 send_data(data: Uint8Array, resume_read: boolean): number {
221 console.assert(pm_b.read_event(sock_b, data) instanceof ldk.Result_boolPeerHandleErrorZ_OK);
224 disconnect_socket(): void {
225 console.assert(false);
227 eq(other: ldk.SocketDescriptor): boolean {
228 return other.hash() == this.hash();
233 } as ldk.SocketDescriptorInterface);
234 sock_b = ldk.SocketDescriptor.new_impl({
235 send_data(data: Uint8Array, resume_read: boolean): number {
236 console.assert(pm_a.read_event(sock_a, data) instanceof ldk.Result_boolPeerHandleErrorZ_OK);
239 disconnect_socket(): void {
240 console.assert(false);
242 eq(other: ldk.SocketDescriptor): boolean {
243 return other.hash() == this.hash();
248 } as ldk.SocketDescriptorInterface);
250 const v4_netaddr = ldk.NetAddress.constructor_ipv4(Uint8Array.from([42,0,42,1]), 9735);
251 console.assert(pm_b.new_inbound_connection(sock_b, ldk.Option_NetAddressZ.constructor_some(v4_netaddr)) instanceof ldk.Result_NonePeerHandleErrorZ_OK);
252 const init_bytes = pm_a.new_outbound_connection(b.node_id, sock_a, ldk.Option_NetAddressZ.constructor_none());
253 if (!(init_bytes instanceof ldk.Result_CVec_u8ZPeerHandleErrorZ_OK)) return false;
254 console.assert(pm_b.read_event(sock_b, init_bytes.res) instanceof ldk.Result_boolPeerHandleErrorZ_OK);
256 console.assert(pm_a.get_peer_node_ids().length == 0);
257 console.assert(pm_b.get_peer_node_ids().length == 0);
259 pm_b.process_events();
260 pm_a.process_events();
261 pm_b.process_events();
263 console.assert(pm_a.get_peer_node_ids().length == 1);
264 console.assert(pm_b.get_peer_node_ids().length == 1);
266 const chan_create_res = a.chan_man.create_channel(b.node_id, BigInt(1000000), BigInt(400), BigInt(0), ldk.UserConfig.constructor_default());
267 if (!chan_create_res.is_ok()) return false;
269 pm_a.process_events();
270 pm_b.process_events();
272 const event = get_event(a.chan_man);
273 if (!(event instanceof ldk.Event_FundingGenerationReady)) return false;
278 async function run_tests(check_leaks: boolean) {
280 for (const test of tests) {
281 test_runs.push(test());
283 const results = await Promise.all(test_runs);
284 console.log("test results: " + results);
285 const result = results.every((v) => { return v === true });
286 console.log("all tests passed: " + result);
287 if (result !== true || !check_leaks) { return result; }
289 const allocs_finished = new Promise((resolve, reject) => {
291 const interval_id = setInterval(() => {
292 const alloc_count = rawldk.getRemainingAllocationCount();
293 if (loop_count % 20 == 0)
294 console.log("Remaining LDK allocation count: " + alloc_count);
296 // chromium with --js-flags="--expose-gc" exposes a `window.gc()` which we call if we can
297 // @ts-ignore window.gc is considered a type error in TS
298 if (typeof window !== "undefined" && typeof window.gc !== "undefined") window.gc();
300 // Note that there are currently 9 leaks in the above tests. At least some are known - look for XXX in bindings.c
301 if (alloc_count <= 10) { clearInterval(interval_id); rawldk.debugPrintRemainingAllocs(); resolve(true); }
303 if (loop_count > 30*2) { clearInterval(interval_id); rawldk.debugPrintRemainingAllocs(); resolve(false); }
306 return allocs_finished;
309 export async function run_tests_web(wasm_path: string, check_leaks: boolean = true) {
310 await ldk.initializeWasmWebFetch(wasm_path);
311 return await run_tests(check_leaks);
314 export async function run_tests_node(wasm_file: Uint8Array, check_leaks: boolean = true) {
315 await ldk.initializeWasmFromBinary(wasm_file);
316 return await run_tests(check_leaks);