From 2cbf09d6ae2d02544bb0959d2199c41ec47fdd75 Mon Sep 17 00:00:00 2001 From: CJ Cobb <46455409+cjcobb23@users.noreply.github.com> Date: Mon, 14 Nov 2022 13:21:18 -0500 Subject: [PATCH] handle invalidHotWallet in gateway_balances (#384) --- src/rpc/handlers/GatewayBalances.cpp | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/src/rpc/handlers/GatewayBalances.cpp b/src/rpc/handlers/GatewayBalances.cpp index ee56b8b8d..fa54232a4 100644 --- a/src/rpc/handlers/GatewayBalances.cpp +++ b/src/rpc/handlers/GatewayBalances.cpp @@ -182,6 +182,13 @@ doGatewayBalances(Context const& context) return obj; }; + auto containsHotWallet = [&](auto const& hw) { + return hotBalances.contains(hw); + }; + if (not std::all_of( + hotWallets.begin(), hotWallets.end(), containsHotWallet)) + return Status{Error::rpcINVALID_PARAMS, "invalidHotWallet"}; + if (auto balances = toJson(hotBalances); balances.size()) response[JS(balances)] = balances; if (auto balances = toJson(frozenBalances); balances.size())