Limit UTXOs for transactions (#906)

* Limit UTXOs for transactions To avoid HSM crash :p * use configuration
LedgerHQ · Sep 15, 2022 · 9037104 · 9037104
1 parent 1b4dbe7
commit 9037104
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/core/src/wallet/bitcoin/BitcoinLikeAccount.cpp b/core/src/wallet/bitcoin/BitcoinLikeAccount.cpp
@@ -766,7 +766,14 @@ namespace ledger {
                     auto keychain = self->getKeychain();
                     soci::session session(self->getWallet()->getDatabase()->getPool());
 
-                    return BitcoinLikeUTXODatabaseHelper::queryAllUtxos(session, self->getAccountUid(), self->getWallet()->getCurrency());
+                    auto utxos         = BitcoinLikeUTXODatabaseHelper::queryAllUtxos(session, self->getAccountUid(), self->getWallet()->getCurrency());
+                    const auto maxUtxo = self->getWallet()->getConfig()->getInt("MAX_UTXO").value_or(std::numeric_limits<int>::max());
+
+                    if (utxos.size() > maxUtxo) {
+                        throw std::runtime_error(fmt::format("Too many utxos. Got {}, but max is {}.", utxos.size(), maxUtxo));
+                    }
+
+                    return utxos;
                 });
             };
             auto getTransaction = [self](const std::string &hash) -> FuturePtr<BitcoinLikeBlockchainExplorerTransaction> {