|
| 1 | +// |
| 2 | +// LDK |
| 3 | +// |
| 4 | +// Created by Arik Sosman on 7/27/23. |
| 5 | +// |
| 6 | + |
| 7 | +#if SWIFT_PACKAGE |
| 8 | +import LDKSwift |
| 9 | +import LDKHeaders |
| 10 | +#endif |
| 11 | +import XCTest |
| 12 | + |
| 13 | +class WrappedSignerProviderTests: XCTestCase { |
| 14 | + |
| 15 | + override func setUpWithError() throws { |
| 16 | + // Put setup code here. This method is called before the invocation of each test method in the class. |
| 17 | + Bindings.setLogThreshold(severity: .DEBUG) |
| 18 | + } |
| 19 | + |
| 20 | + override func tearDownWithError() throws { |
| 21 | + // Put teardown code here. This method is called after the invocation of each test method in the class. |
| 22 | + } |
| 23 | + |
| 24 | + func testChannelManagerConstruction() { |
| 25 | + let feeEstimator = TestFeeEstimator() |
| 26 | + let logger = TestLogger() |
| 27 | + let broadcaster = TestBroadcasterInterface() |
| 28 | + let persister = TestPersister() |
| 29 | + let filter = TestFilter() |
| 30 | + |
| 31 | + let chainMonitor = ChainMonitor(chainSource: filter, broadcaster: broadcaster, logger: logger, feeest: feeEstimator, persister: persister) |
| 32 | + |
| 33 | + let seed = [UInt8](Data(base64Encoded: "//////////////////////////////////////////8=")!) |
| 34 | + let timestampSeconds = UInt64(NSDate().timeIntervalSince1970) |
| 35 | + let timestampNanos = UInt32.init(truncating: NSNumber(value: timestampSeconds * 1000 * 1000)) |
| 36 | + let myKeysManager = MyKeysManager(seed: seed, startingTimeSecs: timestampSeconds, startingTimeNanos: timestampNanos) |
| 37 | + |
| 38 | + let handshakeConfig = ChannelHandshakeConfig.initWithDefault() |
| 39 | + handshakeConfig.setMinimumDepth(val: 1) |
| 40 | + handshakeConfig.setAnnouncedChannel(val: false) |
| 41 | + |
| 42 | + let handshakeLimits = ChannelHandshakeLimits.initWithDefault() |
| 43 | + handshakeLimits.setForceAnnouncedChannelPreference(val: false) |
| 44 | + |
| 45 | + let userConfig = UserConfig.initWithDefault() |
| 46 | + userConfig.setChannelHandshakeConfig(val: handshakeConfig) |
| 47 | + userConfig.setChannelHandshakeLimits(val: handshakeLimits) |
| 48 | + userConfig.setAcceptInboundChannels(val: true) |
| 49 | + |
| 50 | + let network = Network.Testnet |
| 51 | + let netGraph = NetworkGraph(network: network, logger: logger) |
| 52 | + |
| 53 | + let params = ProbabilisticScoringParameters.initWithDefault() |
| 54 | + let probabilisticScorer = ProbabilisticScorer(params: params, networkGraph: netGraph, logger: logger) |
| 55 | + let score = probabilisticScorer.asScore() |
| 56 | + let scorer = MultiThreadedLockableScore(score: score) |
| 57 | + |
| 58 | + let channelManagerConstructionParameters = ChannelManagerConstructionParameters(config: userConfig, entropySource: myKeysManager.entropySource, nodeSigner: myKeysManager.nodeSigner, signerProvider: myKeysManager.signerProvider, feeEstimator: feeEstimator, chainMonitor: chainMonitor, txBroadcaster: broadcaster, logger: logger, enableP2PGossip: true, scorer: scorer) |
| 59 | + |
| 60 | + let reversedGenesisHashHex = "6fe28c0ab6f1b372c1a6a246ae63f74f931e8365e15a089c68d6190000000000" |
| 61 | + let reversedGenesisHash = LDKSwiftTests.hexStringToBytes(hexString: reversedGenesisHashHex)! |
| 62 | + let latestBlockHash = reversedGenesisHash |
| 63 | + let channelManager = ChannelManagerConstructor(network: network, currentBlockchainTipHash: latestBlockHash, currentBlockchainTipHeight: 0, netGraph: netGraph, params: channelManagerConstructionParameters) |
| 64 | + } |
| 65 | + |
| 66 | + class MyKeysManager { |
| 67 | + let keysManager: KeysManager |
| 68 | + let nodeSigner: MyNodeSigner |
| 69 | + let entropySource: MyEntropySource |
| 70 | + let signerProvider: MySignerProvider |
| 71 | + |
| 72 | + init(seed: [UInt8], startingTimeSecs: UInt64, startingTimeNanos: UInt32) { |
| 73 | + self.keysManager = KeysManager(seed: seed, startingTimeSecs: startingTimeSecs, startingTimeNanos: startingTimeNanos) |
| 74 | + nodeSigner = MyNodeSigner() |
| 75 | + entropySource = MyEntropySource() |
| 76 | + signerProvider = MySignerProvider() |
| 77 | + nodeSigner.myKeysManager = self |
| 78 | + entropySource.myKeysManager = self |
| 79 | + signerProvider.myKeysManager = self |
| 80 | + } |
| 81 | + } |
| 82 | + |
| 83 | + class MyNodeSigner: NodeSigner { |
| 84 | + var myKeysManager: MyKeysManager? |
| 85 | + override func ecdh(recipient: Bindings.Recipient, otherKey: [UInt8], tweak: [UInt8]?) -> Bindings.Result_SharedSecretNoneZ { |
| 86 | + print("Getting ecdh") |
| 87 | + return myKeysManager!.keysManager.asNodeSigner().ecdh(recipient: recipient, otherKey: otherKey, tweak: tweak) |
| 88 | + } |
| 89 | + |
| 90 | + override func getNodeId(recipient: Bindings.Recipient) -> Bindings.Result_PublicKeyNoneZ { |
| 91 | + print("Getting getNodeId") |
| 92 | + let nodeIdOption = myKeysManager!.keysManager.asNodeSigner().getNodeId(recipient: recipient) |
| 93 | + let nodeId = nodeIdOption.getValue()! |
| 94 | + return .initWithOk(o: nodeId) |
| 95 | + } |
| 96 | + |
| 97 | + override func getInboundPaymentKeyMaterial() -> [UInt8] { |
| 98 | + print("Getting getInboundPaymentKeyMaterial") |
| 99 | + return myKeysManager!.keysManager.asNodeSigner().getInboundPaymentKeyMaterial() |
| 100 | + } |
| 101 | + |
| 102 | + override func signGossipMessage(msg: Bindings.UnsignedGossipMessage) -> Bindings.Result_SignatureNoneZ { |
| 103 | + print("Getting signGossipMessage") |
| 104 | + return myKeysManager!.keysManager.asNodeSigner().signGossipMessage(msg: msg) |
| 105 | + } |
| 106 | + |
| 107 | + override func signInvoice(hrpBytes: [UInt8], invoiceData: [UInt8], recipient: Bindings.Recipient) -> Bindings.Result_RecoverableSignatureNoneZ { |
| 108 | + print("Getting signInvoice") |
| 109 | + return myKeysManager!.keysManager.asNodeSigner().signInvoice(hrpBytes: hrpBytes, invoiceData: invoiceData, recipient: recipient) |
| 110 | + } |
| 111 | + } |
| 112 | + |
| 113 | + class MyEntropySource: EntropySource { |
| 114 | + var myKeysManager: MyKeysManager? |
| 115 | + override func getSecureRandomBytes() -> [UInt8] { |
| 116 | + print("Getting getSecureRandomBytes") |
| 117 | + return myKeysManager!.keysManager.asEntropySource().getSecureRandomBytes() |
| 118 | + } |
| 119 | + } |
| 120 | + |
| 121 | + class MySignerProvider: SignerProvider { |
| 122 | + var myKeysManager: MyKeysManager? |
| 123 | + override func deriveChannelSigner(channelValueSatoshis: UInt64, channelKeysId: [UInt8]) -> Bindings.WriteableEcdsaChannelSigner { |
| 124 | + return myKeysManager!.keysManager.asSignerProvider().deriveChannelSigner(channelValueSatoshis: channelValueSatoshis, channelKeysId: channelKeysId) |
| 125 | + } |
| 126 | + |
| 127 | + override func generateChannelKeysId(inbound: Bool, channelValueSatoshis: UInt64, userChannelId: [UInt8]) -> [UInt8] { |
| 128 | + return myKeysManager!.keysManager.asSignerProvider().generateChannelKeysId(inbound: inbound, channelValueSatoshis: channelValueSatoshis, userChannelId: userChannelId) |
| 129 | + } |
| 130 | + |
| 131 | + override func readChanSigner(reader: [UInt8]) -> Bindings.Result_WriteableEcdsaChannelSignerDecodeErrorZ { |
| 132 | + return myKeysManager!.keysManager.asSignerProvider().readChanSigner(reader: reader) |
| 133 | + } |
| 134 | + |
| 135 | + override func getDestinationScript() -> [UInt8] { |
| 136 | + return myKeysManager!.keysManager.asSignerProvider().getDestinationScript() |
| 137 | + } |
| 138 | + |
| 139 | + override func getShutdownScriptpubkey() -> Bindings.ShutdownScript {return myKeysManager!.keysManager.asSignerProvider().getShutdownScriptpubkey() |
| 140 | + } |
| 141 | + } |
| 142 | + |
| 143 | +} |
0 commit comments