Merge pull request #8220

0d6ecb1 multisig: add post-kex verification round to check that all participants have completed the multisig address (koe)
This commit is contained in:
luigi1111 2022-05-10 16:41:02 -05:00
commit c1625a8928
No known key found for this signature in database
GPG Key ID: F4ACA0183641E010
9 changed files with 336 additions and 194 deletions

View File

@ -73,7 +73,7 @@ namespace multisig
const crypto::public_key &multisig_pubkey, const crypto::public_key &multisig_pubkey,
const crypto::public_key &common_pubkey, const crypto::public_key &common_pubkey,
const std::uint32_t kex_rounds_complete, const std::uint32_t kex_rounds_complete,
kex_origins_map_t kex_origins_map, multisig_keyset_map_memsafe_t kex_origins_map,
std::string next_round_kex_message) : std::string next_round_kex_message) :
m_base_privkey{base_privkey}, m_base_privkey{base_privkey},
m_base_common_privkey{base_common_privkey}, m_base_common_privkey{base_common_privkey},
@ -89,6 +89,20 @@ namespace multisig
CHECK_AND_ASSERT_THROW_MES(crypto::secret_key_to_public_key(m_base_privkey, m_base_pubkey), CHECK_AND_ASSERT_THROW_MES(crypto::secret_key_to_public_key(m_base_privkey, m_base_pubkey),
"Failed to derive public key"); "Failed to derive public key");
set_multisig_config(threshold, std::move(signers)); set_multisig_config(threshold, std::move(signers));
// kex rounds should not exceed post-kex verification round
const std::uint32_t kex_rounds_required{multisig_kex_rounds_required(m_signers.size(), m_threshold)};
CHECK_AND_ASSERT_THROW_MES(m_kex_rounds_complete <= kex_rounds_required + 1,
"multisig account: tried to reconstruct account, but kex rounds complete counter is invalid.");
// once an account is done with kex, the 'next kex msg' is always the post-kex verification message
// i.e. the multisig account pubkey signed by the signer's privkey AND the common pubkey
if (main_kex_rounds_done())
{
m_next_round_kex_message = multisig_kex_msg{kex_rounds_required + 1,
m_base_privkey,
std::vector<crypto::public_key>{m_multisig_pubkey, m_common_pubkey}}.get_msg();
}
} }
//---------------------------------------------------------------------------------------------------------------------- //----------------------------------------------------------------------------------------------------------------------
// multisig_account: EXTERNAL // multisig_account: EXTERNAL
@ -100,10 +114,20 @@ namespace multisig
//---------------------------------------------------------------------------------------------------------------------- //----------------------------------------------------------------------------------------------------------------------
// multisig_account: EXTERNAL // multisig_account: EXTERNAL
//---------------------------------------------------------------------------------------------------------------------- //----------------------------------------------------------------------------------------------------------------------
bool multisig_account::multisig_is_ready() const bool multisig_account::main_kex_rounds_done() const
{ {
if (account_is_active()) if (account_is_active())
return multisig_kex_rounds_required(m_signers.size(), m_threshold) == m_kex_rounds_complete; return m_kex_rounds_complete >= multisig_kex_rounds_required(m_signers.size(), m_threshold);
else
return false;
}
//----------------------------------------------------------------------------------------------------------------------
// multisig_account: EXTERNAL
//----------------------------------------------------------------------------------------------------------------------
bool multisig_account::multisig_is_ready() const
{
if (main_kex_rounds_done())
return m_kex_rounds_complete >= multisig_kex_rounds_required(m_signers.size(), m_threshold) + 1;
else else
return false; return false;
} }
@ -119,10 +143,6 @@ namespace multisig
for (auto signer_it = signers.begin(); signer_it != signers.end(); ++signer_it) for (auto signer_it = signers.begin(); signer_it != signers.end(); ++signer_it)
{ {
// signers should all be unique
CHECK_AND_ASSERT_THROW_MES(std::find(signers.begin(), signer_it, *signer_it) == signer_it,
"multisig account: tried to set signers, but found a duplicate signer unexpectedly.");
// signer pubkeys must be in main subgroup, and not identity // signer pubkeys must be in main subgroup, and not identity
CHECK_AND_ASSERT_THROW_MES(rct::isInMainSubgroup(rct::pk2rct(*signer_it)) && !(*signer_it == rct::rct2pk(rct::identity())), CHECK_AND_ASSERT_THROW_MES(rct::isInMainSubgroup(rct::pk2rct(*signer_it)) && !(*signer_it == rct::rct2pk(rct::identity())),
"multisig account: tried to set signers, but a signer pubkey is invalid."); "multisig account: tried to set signers, but a signer pubkey is invalid.");
@ -133,12 +153,11 @@ namespace multisig
"multisig account: tried to set signers, but did not find the account's base pubkey in signer list."); "multisig account: tried to set signers, but did not find the account's base pubkey in signer list.");
// sort signers // sort signers
std::sort(signers.begin(), signers.end(), std::sort(signers.begin(), signers.end());
[](const crypto::public_key &key1, const crypto::public_key &key2) -> bool
{ // signers should all be unique
return memcmp(&key1, &key2, sizeof(crypto::public_key)) < 0; CHECK_AND_ASSERT_THROW_MES(std::adjacent_find(signers.begin(), signers.end()) == signers.end(),
} "multisig account: tried to set signers, but there are duplicate signers unexpectedly.");
);
// set // set
m_threshold = threshold; m_threshold = threshold;

View File

@ -75,12 +75,12 @@ namespace multisig
* - ZtM2: https://web.getmonero.org/library/Zero-to-Monero-2-0-0.pdf Ch. 9, especially Section 9.6.3 * - ZtM2: https://web.getmonero.org/library/Zero-to-Monero-2-0-0.pdf Ch. 9, especially Section 9.6.3
* - FROST: https://eprint.iacr.org/2018/417 * - FROST: https://eprint.iacr.org/2018/417
*/ */
using multisig_keyset_map_memsafe_t =
std::unordered_map<crypto::public_key_memsafe, std::unordered_set<crypto::public_key>>;
class multisig_account final class multisig_account final
{ {
public: public:
//member types
using kex_origins_map_t = std::unordered_map<crypto::public_key_memsafe, std::unordered_set<crypto::public_key>>;
//constructors //constructors
// default constructor // default constructor
multisig_account() = default; multisig_account() = default;
@ -105,7 +105,7 @@ namespace multisig
const crypto::public_key &multisig_pubkey, const crypto::public_key &multisig_pubkey,
const crypto::public_key &common_pubkey, const crypto::public_key &common_pubkey,
const std::uint32_t kex_rounds_complete, const std::uint32_t kex_rounds_complete,
kex_origins_map_t kex_origins_map, multisig_keyset_map_memsafe_t kex_origins_map,
std::string next_round_kex_message); std::string next_round_kex_message);
// copy constructor: default // copy constructor: default
@ -137,13 +137,15 @@ namespace multisig
// get kex rounds complete // get kex rounds complete
std::uint32_t get_kex_rounds_complete() const { return m_kex_rounds_complete; } std::uint32_t get_kex_rounds_complete() const { return m_kex_rounds_complete; }
// get kex keys to origins map // get kex keys to origins map
const kex_origins_map_t& get_kex_keys_to_origins_map() const { return m_kex_keys_to_origins_map; } const multisig_keyset_map_memsafe_t& get_kex_keys_to_origins_map() const { return m_kex_keys_to_origins_map; }
// get the kex msg for the next round // get the kex msg for the next round
const std::string& get_next_kex_round_msg() const { return m_next_round_kex_message; } const std::string& get_next_kex_round_msg() const { return m_next_round_kex_message; }
//account status functions //account status functions
// account has been intialized, and the account holder can use the 'common' key // account has been intialized, and the account holder can use the 'common' key
bool account_is_active() const; bool account_is_active() const;
// account has gone through main kex rounds, only remaining step is to verify all other participants are ready
bool main_kex_rounds_done() const;
// account is ready to make multisig signatures // account is ready to make multisig signatures
bool multisig_is_ready() const; bool multisig_is_ready() const;
@ -178,21 +180,21 @@ namespace multisig
* - Collect the local signer's shared keys to ignore in incoming messages, build the aggregate ancillary key * - Collect the local signer's shared keys to ignore in incoming messages, build the aggregate ancillary key
* if appropriate. * if appropriate.
* param: expanded_msgs - set of multisig kex messages to process * param: expanded_msgs - set of multisig kex messages to process
* param: rounds_required - number of rounds required for kex * param: kex_rounds_required - number of rounds required for kex (not including post-kex verification round)
* outparam: exclude_pubkeys_out - keys held by the local account corresponding to round 'current_round' * outparam: exclude_pubkeys_out - keys held by the local account corresponding to round 'current_round'
* - If 'current_round' is the final round, these are the local account's shares of the final aggregate key. * - If 'current_round' is the final round, these are the local account's shares of the final aggregate key.
*/ */
void initialize_kex_update(const std::vector<multisig_kex_msg> &expanded_msgs, void initialize_kex_update(const std::vector<multisig_kex_msg> &expanded_msgs,
const std::uint32_t rounds_required, const std::uint32_t kex_rounds_required,
std::vector<crypto::public_key> &exclude_pubkeys_out); std::vector<crypto::public_key> &exclude_pubkeys_out);
/** /**
* brief: finalize_kex_update - Helper for kex_update_impl() * brief: finalize_kex_update - Helper for kex_update_impl()
* param: rounds_required - number of rounds required for kex * param: kex_rounds_required - number of rounds required for kex (not including post-kex verification round)
* param: result_keys_to_origins_map - map between keys for the next round and the other participants they correspond to * param: result_keys_to_origins_map - map between keys for the next round and the other participants they correspond to
* inoutparam: temp_account_inout - account to perform last update steps on * inoutparam: temp_account_inout - account to perform last update steps on
*/ */
void finalize_kex_update(const std::uint32_t rounds_required, void finalize_kex_update(const std::uint32_t kex_rounds_required,
kex_origins_map_t result_keys_to_origins_map); multisig_keyset_map_memsafe_t result_keys_to_origins_map);
//member variables //member variables
private: private:
@ -226,7 +228,7 @@ namespace multisig
std::uint32_t m_kex_rounds_complete{0}; std::uint32_t m_kex_rounds_complete{0};
// this account's pubkeys for the in-progress key exchange round // this account's pubkeys for the in-progress key exchange round
// - either DH derivations (intermediate rounds), H(derivation)*G (final round), empty (when kex is done) // - either DH derivations (intermediate rounds), H(derivation)*G (final round), empty (when kex is done)
kex_origins_map_t m_kex_keys_to_origins_map; multisig_keyset_map_memsafe_t m_kex_keys_to_origins_map;
// the account's message for the in-progress key exchange round // the account's message for the in-progress key exchange round
std::string m_next_round_kex_message; std::string m_next_round_kex_message;
}; };

View File

@ -53,6 +53,30 @@
namespace multisig namespace multisig
{ {
//----------------------------------------------------------------------------------------------------------------------
/**
* INTERNAL
*
* brief: check_multisig_config - validate multisig configuration details
* param: round - the round of the message that should be produced
* param: threshold - threshold for multisig (M in M-of-N)
* param: num_signers - number of participants in multisig (N)
*/
//----------------------------------------------------------------------------------------------------------------------
static void check_multisig_config(const std::uint32_t round,
const std::uint32_t threshold,
const std::uint32_t num_signers)
{
CHECK_AND_ASSERT_THROW_MES(num_signers > 1, "Must be at least one other multisig signer.");
CHECK_AND_ASSERT_THROW_MES(num_signers <= config::MULTISIG_MAX_SIGNERS,
"Too many multisig signers specified (limit = 16 to prevent dangerous combinatorial explosion during key exchange).");
CHECK_AND_ASSERT_THROW_MES(num_signers >= threshold,
"Multisig threshold may not be larger than number of signers.");
CHECK_AND_ASSERT_THROW_MES(threshold > 0, "Multisig threshold must be > 0.");
CHECK_AND_ASSERT_THROW_MES(round > 0, "Multisig kex round must be > 0.");
CHECK_AND_ASSERT_THROW_MES(round <= multisig_kex_rounds_required(num_signers, threshold) + 1,
"Trying to process multisig kex for an invalid round.");
}
//---------------------------------------------------------------------------------------------------------------------- //----------------------------------------------------------------------------------------------------------------------
/** /**
* INTERNAL * INTERNAL
@ -224,50 +248,23 @@ namespace multisig
/** /**
* INTERNAL * INTERNAL
* *
* brief: multisig_kex_make_next_msg - Construct a kex msg for any round > 1 of multisig key construction. * brief: multisig_kex_make_round_keys - Makes a kex round's keys.
* - Involves DH exchanges with pubkeys provided by other participants. * - Involves DH exchanges with pubkeys provided by other participants.
* - Conserves mapping [pubkey -> DH derivation] : [origin keys of participants that share this secret with you]. * - Conserves mapping [pubkey -> DH derivation] : [origin keys of participants that share this secret with you].
* param: base_privkey - account's base private key, for performing DH exchanges and signing messages * param: base_privkey - account's base private key, for performing DH exchanges and signing messages
* param: round - the round of the message that should be produced
* param: threshold - threshold for multisig (M in M-of-N)
* param: num_signers - number of participants in multisig (N)
* param: pubkey_origins_map - map between pubkeys to produce DH derivations with and identity keys of * param: pubkey_origins_map - map between pubkeys to produce DH derivations with and identity keys of
* participants who will share each derivation with you * participants who will share each derivation with you
* outparam: derivation_origins_map_out - map between DH derivations (shared secrets) and identity keys * outparam: derivation_origins_map_out - map between DH derivations (shared secrets) and identity keys
* - If msg is not for the last round, then these derivations are also stored in the output message
* so they can be sent to other participants, who will make more DH derivations for the next kex round.
* - If msg is for the last round, then these derivations won't be sent to other participants.
* Instead, they are converted to share secrets (i.e. s = H(derivation)) and multiplied by G.
* The keys s*G are sent to other participants in the message, so they can be used to produce the final
* multisig key via generate_multisig_spend_public_key().
* - The values s are the local account's shares of the final multisig key's private key. The caller can
* compute those values with calculate_multisig_keypair_from_derivation() (or compute them directly).
* return: multisig kex message for the specified round
*/ */
//---------------------------------------------------------------------------------------------------------------------- //----------------------------------------------------------------------------------------------------------------------
static multisig_kex_msg multisig_kex_make_next_msg(const crypto::secret_key &base_privkey, static void multisig_kex_make_round_keys(const crypto::secret_key &base_privkey,
const std::uint32_t round, multisig_keyset_map_memsafe_t pubkey_origins_map,
const std::uint32_t threshold, multisig_keyset_map_memsafe_t &derivation_origins_map_out)
const std::uint32_t num_signers,
const std::unordered_map<crypto::public_key_memsafe, std::unordered_set<crypto::public_key>> &pubkey_origins_map,
std::unordered_map<crypto::public_key_memsafe, std::unordered_set<crypto::public_key>> &derivation_origins_map_out)
{ {
CHECK_AND_ASSERT_THROW_MES(num_signers > 1, "Must be at least one other multisig signer.");
CHECK_AND_ASSERT_THROW_MES(num_signers <= config::MULTISIG_MAX_SIGNERS,
"Too many multisig signers specified (limit = 16 to prevent dangerous combinatorial explosion during key exchange).");
CHECK_AND_ASSERT_THROW_MES(num_signers >= threshold,
"Multisig threshold may not be larger than number of signers.");
CHECK_AND_ASSERT_THROW_MES(threshold > 0, "Multisig threshold must be > 0.");
CHECK_AND_ASSERT_THROW_MES(round > 1, "Round for next msg must be > 1.");
CHECK_AND_ASSERT_THROW_MES(round <= multisig_kex_rounds_required(num_signers, threshold),
"Trying to make key exchange message for an invalid round.");
// make shared secrets with input pubkeys // make shared secrets with input pubkeys
std::vector<crypto::public_key> msg_pubkeys;
msg_pubkeys.reserve(pubkey_origins_map.size());
derivation_origins_map_out.clear(); derivation_origins_map_out.clear();
for (const auto &pubkey_and_origins : pubkey_origins_map) for (auto &pubkey_and_origins : pubkey_origins_map)
{ {
// D = 8 * k_base * K_pubkey // D = 8 * k_base * K_pubkey
// note: must be mul8 (cofactor), otherwise it is possible to leak to a malicious participant if the local // note: must be mul8 (cofactor), otherwise it is possible to leak to a malicious participant if the local
@ -281,27 +278,29 @@ namespace multisig
rct::scalarmultKey(derivation_rct, rct::pk2rct(pubkey_and_origins.first), rct::sk2rct(base_privkey)); rct::scalarmultKey(derivation_rct, rct::pk2rct(pubkey_and_origins.first), rct::sk2rct(base_privkey));
rct::scalarmultKey(derivation_rct, derivation_rct, rct::EIGHT); rct::scalarmultKey(derivation_rct, derivation_rct, rct::EIGHT);
crypto::public_key_memsafe derivation{rct::rct2pk(derivation_rct)};
// retain mapping between pubkey's origins and the DH derivation // retain mapping between pubkey's origins and the DH derivation
// note: if msg for last round, then caller must know how to handle these derivations properly // note: if working on last kex round, then caller must know how to handle these derivations properly
derivation_origins_map_out[derivation] = pubkey_and_origins.second; derivation_origins_map_out[rct::rct2pk(derivation_rct)] = std::move(pubkey_and_origins.second);
}
// if the last round, convert derivations to public keys for the output message }
if (round == multisig_kex_rounds_required(num_signers, threshold)) //----------------------------------------------------------------------------------------------------------------------
/**
* INTERNAL
*
* brief: check_messages_round - Check that a set of messages have an expected round number.
* param: expanded_msgs - set of multisig kex messages to process
* param: expected_round - round number the kex messages should have
*/
//----------------------------------------------------------------------------------------------------------------------
static void check_messages_round(const std::vector<multisig_kex_msg> &expanded_msgs,
const std::uint32_t expected_round)
{ {
// derived_pubkey = H(derivation)*G CHECK_AND_ASSERT_THROW_MES(expanded_msgs.size() > 0, "At least one input message expected.");
crypto::public_key derived_pubkey; const std::uint32_t round{expanded_msgs[0].get_round()};
calculate_multisig_keypair_from_derivation(derivation, derived_pubkey); CHECK_AND_ASSERT_THROW_MES(round == expected_round, "Messages don't have the expected kex round number.");
msg_pubkeys.push_back(derived_pubkey);
}
// otherwise, put derivations in message directly, so other signers can in turn create derivations (shared secrets)
// with them for the next round
else
msg_pubkeys.push_back(derivation);
}
return multisig_kex_msg{round, base_privkey, std::move(msg_pubkeys)}; for (const auto &expanded_msg : expanded_msgs)
CHECK_AND_ASSERT_THROW_MES(expanded_msg.get_round() == round, "All messages must have the same kex round number.");
} }
//---------------------------------------------------------------------------------------------------------------------- //----------------------------------------------------------------------------------------------------------------------
/** /**
@ -327,19 +326,19 @@ namespace multisig
static std::uint32_t multisig_kex_msgs_sanitize_pubkeys(const crypto::public_key &own_pubkey, static std::uint32_t multisig_kex_msgs_sanitize_pubkeys(const crypto::public_key &own_pubkey,
const std::vector<multisig_kex_msg> &expanded_msgs, const std::vector<multisig_kex_msg> &expanded_msgs,
const std::vector<crypto::public_key> &exclude_pubkeys, const std::vector<crypto::public_key> &exclude_pubkeys,
std::unordered_map<crypto::public_key_memsafe, std::unordered_set<crypto::public_key>> &sanitized_pubkeys_out) multisig_keyset_map_memsafe_t &sanitized_pubkeys_out)
{ {
// all messages should have the same round (redundant sanity check)
CHECK_AND_ASSERT_THROW_MES(expanded_msgs.size() > 0, "At least one input message expected."); CHECK_AND_ASSERT_THROW_MES(expanded_msgs.size() > 0, "At least one input message expected.");
const std::uint32_t round{expanded_msgs[0].get_round()};
check_messages_round(expanded_msgs, round);
std::uint32_t round = expanded_msgs[0].get_round();
sanitized_pubkeys_out.clear(); sanitized_pubkeys_out.clear();
// get all pubkeys from input messages, add them to pubkey:origins map // get all pubkeys from input messages, add them to pubkey:origins map
// - origins = all the signing pubkeys that recommended a given msg pubkey // - origins = all the signing pubkeys that recommended a given msg pubkey
for (const auto &expanded_msg : expanded_msgs) for (const auto &expanded_msg : expanded_msgs)
{ {
CHECK_AND_ASSERT_THROW_MES(expanded_msg.get_round() == round, "All messages must have the same kex round number.");
// ignore messages from self // ignore messages from self
if (expanded_msg.get_signing_pubkey() == own_pubkey) if (expanded_msg.get_signing_pubkey() == own_pubkey)
continue; continue;
@ -378,7 +377,7 @@ namespace multisig
* *
* brief: evaluate_multisig_kex_round_msgs - Evaluate pubkeys from a kex round in order to prepare for the next round. * brief: evaluate_multisig_kex_round_msgs - Evaluate pubkeys from a kex round in order to prepare for the next round.
* - Sanitizes input msgs. * - Sanitizes input msgs.
* - Require uniqueness in: 'signers', 'exclude_pubkeys'. * - Require uniqueness in: 'exclude_pubkeys'.
* - Requires each input pubkey be recommended by 'num_recommendations = expected_round' msg signers. * - Requires each input pubkey be recommended by 'num_recommendations = expected_round' msg signers.
* - For a final multisig key to be truly 'M-of-N', each of the the private key's components must be * - For a final multisig key to be truly 'M-of-N', each of the the private key's components must be
* shared by (N - M + 1) signers. * shared by (N - M + 1) signers.
@ -388,39 +387,21 @@ namespace multisig
* with the local account. * with the local account.
* - Requires that 'exclude_pubkeys' has [num_signers - 1 CHOOSE (expected_round - 1)] pubkeys. * - Requires that 'exclude_pubkeys' has [num_signers - 1 CHOOSE (expected_round - 1)] pubkeys.
* - These should be derivations the local account has corresponding to round 'expected_round'. * - These should be derivations the local account has corresponding to round 'expected_round'.
* param: base_privkey - multisig account's base private key * param: base_pubkey - multisig account's base public key
* param: expected_round - expected kex round of input messages * param: expected_round - expected kex round of input messages
* param: threshold - threshold for multisig (M in M-of-N)
* param: signers - expected participants in multisig kex * param: signers - expected participants in multisig kex
* param: expanded_msgs - set of multisig kex messages to process * param: expanded_msgs - set of multisig kex messages to process
* param: exclude_pubkeys - derivations held by the local account corresponding to round 'expected_round' * param: exclude_pubkeys - derivations held by the local account corresponding to round 'expected_round'
* return: fully sanitized and validated pubkey:origins map for building the account's next kex round message * return: fully sanitized and validated pubkey:origins map for building the account's next kex round message
*/ */
//---------------------------------------------------------------------------------------------------------------------- //----------------------------------------------------------------------------------------------------------------------
static std::unordered_map<crypto::public_key_memsafe, std::unordered_set<crypto::public_key>> evaluate_multisig_kex_round_msgs( static multisig_keyset_map_memsafe_t evaluate_multisig_kex_round_msgs(
const crypto::public_key &base_pubkey, const crypto::public_key &base_pubkey,
const std::uint32_t expected_round, const std::uint32_t expected_round,
const std::uint32_t threshold,
const std::vector<crypto::public_key> &signers, const std::vector<crypto::public_key> &signers,
const std::vector<multisig_kex_msg> &expanded_msgs, const std::vector<multisig_kex_msg> &expanded_msgs,
const std::vector<crypto::public_key> &exclude_pubkeys) const std::vector<crypto::public_key> &exclude_pubkeys)
{ {
CHECK_AND_ASSERT_THROW_MES(signers.size() > 1, "Must be at least one other multisig signer.");
CHECK_AND_ASSERT_THROW_MES(signers.size() <= config::MULTISIG_MAX_SIGNERS,
"Too many multisig signers specified (limit = 16 to prevent dangerous combinatorial explosion during key exchange).");
CHECK_AND_ASSERT_THROW_MES(signers.size() >= threshold, "Multisig threshold may not be larger than number of signers.");
CHECK_AND_ASSERT_THROW_MES(threshold > 0, "Multisig threshold must be > 0.");
CHECK_AND_ASSERT_THROW_MES(expected_round > 0, "Expected round must be > 0.");
CHECK_AND_ASSERT_THROW_MES(expected_round <= multisig_kex_rounds_required(signers.size(), threshold),
"Expecting key exchange messages for an invalid round.");
std::unordered_map<crypto::public_key_memsafe, std::unordered_set<crypto::public_key>> pubkey_origins_map;
// leave early in the last round of 1-of-N, where all signers share a key so the local signer doesn't care about
// recommendations from other signers
if (threshold == 1 && expected_round == multisig_kex_rounds_required(signers.size(), threshold))
return pubkey_origins_map;
// exclude_pubkeys should all be unique // exclude_pubkeys should all be unique
for (auto it = exclude_pubkeys.begin(); it != exclude_pubkeys.end(); ++it) for (auto it = exclude_pubkeys.begin(); it != exclude_pubkeys.end(); ++it)
{ {
@ -429,7 +410,8 @@ namespace multisig
} }
// sanitize input messages // sanitize input messages
std::uint32_t round = multisig_kex_msgs_sanitize_pubkeys(base_pubkey, expanded_msgs, exclude_pubkeys, pubkey_origins_map); multisig_keyset_map_memsafe_t pubkey_origins_map;
const std::uint32_t round = multisig_kex_msgs_sanitize_pubkeys(base_pubkey, expanded_msgs, exclude_pubkeys, pubkey_origins_map);
CHECK_AND_ASSERT_THROW_MES(round == expected_round, CHECK_AND_ASSERT_THROW_MES(round == expected_round,
"Kex messages were for round [" << round << "], but expected round is [" << expected_round << "]"); "Kex messages were for round [" << round << "], but expected round is [" << expected_round << "]");
@ -486,10 +468,10 @@ namespace multisig
// - Each origin should have a shared key with each group of size 'round - 1'. // - Each origin should have a shared key with each group of size 'round - 1'.
// Note: Keys shared with local are ignored to facilitate kex round boosting, where one or more signers may // Note: Keys shared with local are ignored to facilitate kex round boosting, where one or more signers may
// have boosted the local signer (implying they didn't have access to the local signer's previous round msg). // have boosted the local signer (implying they didn't have access to the local signer's previous round msg).
std::uint32_t expected_recommendations_others = n_choose_k_f(signers.size() - 2, round - 1); const std::uint32_t expected_recommendations_others = n_choose_k_f(signers.size() - 2, round - 1);
// local: (N - 1) choose (msg_round_num - 1) // local: (N - 1) choose (msg_round_num - 1)
std::uint32_t expected_recommendations_self = n_choose_k_f(signers.size() - 1, round - 1); const std::uint32_t expected_recommendations_self = n_choose_k_f(signers.size() - 1, round - 1);
// note: expected_recommendations_others would be 0 in the last round of 1-of-N, but we return early for that case // note: expected_recommendations_others would be 0 in the last round of 1-of-N, but we return early for that case
CHECK_AND_ASSERT_THROW_MES(expected_recommendations_self > 0 && expected_recommendations_others > 0, CHECK_AND_ASSERT_THROW_MES(expected_recommendations_self > 0 && expected_recommendations_others > 0,
@ -517,7 +499,60 @@ namespace multisig
/** /**
* INTERNAL * INTERNAL
* *
* brief: multisig_kex_process_round - Process kex messages for the active kex round. * brief: evaluate_multisig_post_kex_round_msgs - Evaluate messages for the post-kex verification round.
* - Sanitizes input msgs.
* - Requires that only one pubkey is recommended.
* - Requires that all signers (other than self) recommend that one pubkey.
* param: base_pubkey - multisig account's base public key
* param: expected_round - expected kex round of input messages
* param: signers - expected participants in multisig kex
* param: expanded_msgs - set of multisig kex messages to process
* return: sanitized and validated pubkey:origins map
*/
//----------------------------------------------------------------------------------------------------------------------
static multisig_keyset_map_memsafe_t evaluate_multisig_post_kex_round_msgs(
const crypto::public_key &base_pubkey,
const std::uint32_t expected_round,
const std::vector<crypto::public_key> &signers,
const std::vector<multisig_kex_msg> &expanded_msgs)
{
// sanitize input messages
const std::vector<crypto::public_key> dummy;
multisig_keyset_map_memsafe_t pubkey_origins_map;
const std::uint32_t round = multisig_kex_msgs_sanitize_pubkeys(base_pubkey, expanded_msgs, dummy, pubkey_origins_map);
CHECK_AND_ASSERT_THROW_MES(round == expected_round,
"Kex messages were for round [" << round << "], but expected round is [" << expected_round << "]");
// evaluate pubkeys collected
// 1) there should only be two pubkeys
CHECK_AND_ASSERT_THROW_MES(pubkey_origins_map.size() == 2,
"Multisig post-kex round messages from other signers did not all contain two pubkeys.");
// 2) both keys should be recommended by the same set of signers
CHECK_AND_ASSERT_THROW_MES(pubkey_origins_map.begin()->second == (++(pubkey_origins_map.begin()))->second,
"Multisig post-kex round messages from other signers did not all recommend the same pubkey pair.");
// 3) all signers should be present in the recommendation list
auto origins = pubkey_origins_map.begin()->second;
origins.insert(base_pubkey); //add self
CHECK_AND_ASSERT_THROW_MES(origins.size() == signers.size(),
"Multisig post-kex round message origins don't line up with multisig signer set.");
for (const crypto::public_key &signer : signers)
{
CHECK_AND_ASSERT_THROW_MES(origins.find(signer) != origins.end(),
"Could not find an expected signer in multisig post-kex round messages (all signers expected).");
}
return pubkey_origins_map;
}
//----------------------------------------------------------------------------------------------------------------------
/**
* INTERNAL
*
* brief: multisig_kex_process_round_msgs - Process kex messages for the active kex round.
* - A wrapper around evaluate_multisig_kex_round_msgs() -> multisig_kex_make_next_msg(). * - A wrapper around evaluate_multisig_kex_round_msgs() -> multisig_kex_make_next_msg().
* - In other words, evaluate the input messages and try to make a message for the next round. * - In other words, evaluate the input messages and try to make a message for the next round.
* - Note: Must be called on the final round's msgs to evaluate the final key components * - Note: Must be called on the final round's msgs to evaluate the final key components
@ -536,43 +571,62 @@ namespace multisig
* return: multisig kex message for next round, or empty message if 'current_round' is the final round * return: multisig kex message for next round, or empty message if 'current_round' is the final round
*/ */
//---------------------------------------------------------------------------------------------------------------------- //----------------------------------------------------------------------------------------------------------------------
static multisig_kex_msg multisig_kex_process_round(const crypto::secret_key &base_privkey, static void multisig_kex_process_round_msgs(const crypto::secret_key &base_privkey,
const crypto::public_key &base_pubkey, const crypto::public_key &base_pubkey,
const std::uint32_t current_round, const std::uint32_t current_round,
const std::uint32_t threshold, const std::uint32_t threshold,
const std::vector<crypto::public_key> &signers, const std::vector<crypto::public_key> &signers,
const std::vector<multisig_kex_msg> &expanded_msgs, const std::vector<multisig_kex_msg> &expanded_msgs,
const std::vector<crypto::public_key> &exclude_pubkeys, const std::vector<crypto::public_key> &exclude_pubkeys,
std::unordered_map<crypto::public_key_memsafe, std::unordered_set<crypto::public_key>> &keys_to_origins_map_out) multisig_keyset_map_memsafe_t &keys_to_origins_map_out)
{ {
// evaluate messages check_multisig_config(current_round, threshold, signers.size());
std::unordered_map<crypto::public_key_memsafe, std::unordered_set<crypto::public_key>> evaluated_pubkeys = const std::uint32_t kex_rounds_required{multisig_kex_rounds_required(signers.size(), threshold)};
evaluate_multisig_kex_round_msgs(base_pubkey, current_round, threshold, signers, expanded_msgs, exclude_pubkeys);
// produce message for next round (if there is one) // process messages into a [pubkey : {origins}] map
if (current_round < multisig_kex_rounds_required(signers.size(), threshold)) multisig_keyset_map_memsafe_t evaluated_pubkeys;
if (threshold == 1 && current_round == kex_rounds_required)
{ {
return multisig_kex_make_next_msg(base_privkey, // in the last main kex round of 1-of-N, all signers share a key so the local signer doesn't care about evaluating
current_round + 1, // recommendations from other signers
threshold,
signers.size(),
evaluated_pubkeys,
keys_to_origins_map_out);
} }
else else if (current_round <= kex_rounds_required)
{ {
// no more rounds, so collect the key shares recommended by other signers for the final aggregate key // for normal kex rounds, fully evaluate kex round messages
keys_to_origins_map_out.clear(); evaluated_pubkeys = evaluate_multisig_kex_round_msgs(base_pubkey,
keys_to_origins_map_out = std::move(evaluated_pubkeys); current_round,
signers,
expanded_msgs,
exclude_pubkeys);
}
else //(current_round == kex_rounds_required + 1)
{
// for the post-kex verification round, validate the last kex round's messages
evaluated_pubkeys = evaluate_multisig_post_kex_round_msgs(base_pubkey,
current_round,
signers,
expanded_msgs);
}
return multisig_kex_msg{}; // prepare keys-to-origins map for updating the multisig account
if (current_round < kex_rounds_required)
{
// normal kex round: make new keys
multisig_kex_make_round_keys(base_privkey, std::move(evaluated_pubkeys), keys_to_origins_map_out);
}
else if (current_round >= kex_rounds_required)
{
// last kex round: collect the key shares recommended by other signers for the final aggregate key
// post-kex verification round: save the keys found in input messages
keys_to_origins_map_out = std::move(evaluated_pubkeys);
} }
} }
//---------------------------------------------------------------------------------------------------------------------- //----------------------------------------------------------------------------------------------------------------------
// multisig_account: INTERNAL // multisig_account: INTERNAL
//---------------------------------------------------------------------------------------------------------------------- //----------------------------------------------------------------------------------------------------------------------
void multisig_account::initialize_kex_update(const std::vector<multisig_kex_msg> &expanded_msgs, void multisig_account::initialize_kex_update(const std::vector<multisig_kex_msg> &expanded_msgs,
const std::uint32_t rounds_required, const std::uint32_t kex_rounds_required,
std::vector<crypto::public_key> &exclude_pubkeys_out) std::vector<crypto::public_key> &exclude_pubkeys_out)
{ {
if (m_kex_rounds_complete == 0) if (m_kex_rounds_complete == 0)
@ -605,7 +659,7 @@ namespace multisig
"Failed to derive public key"); "Failed to derive public key");
// if N-of-N, then the base privkey will be used directly to make the account's share of the final key // if N-of-N, then the base privkey will be used directly to make the account's share of the final key
if (rounds_required == 1) if (kex_rounds_required == 1)
{ {
m_multisig_privkeys.clear(); m_multisig_privkeys.clear();
m_multisig_privkeys.emplace_back(m_base_privkey); m_multisig_privkeys.emplace_back(m_base_privkey);
@ -629,13 +683,29 @@ namespace multisig
//---------------------------------------------------------------------------------------------------------------------- //----------------------------------------------------------------------------------------------------------------------
// multisig_account: INTERNAL // multisig_account: INTERNAL
//---------------------------------------------------------------------------------------------------------------------- //----------------------------------------------------------------------------------------------------------------------
void multisig_account::finalize_kex_update(const std::uint32_t rounds_required, void multisig_account::finalize_kex_update(const std::uint32_t kex_rounds_required,
std::unordered_map<crypto::public_key_memsafe, std::unordered_set<crypto::public_key>> result_keys_to_origins_map) multisig_keyset_map_memsafe_t result_keys_to_origins_map)
{ {
std::vector<crypto::public_key> next_msg_keys;
// prepare for next round (or complete the multisig account fully) // prepare for next round (or complete the multisig account fully)
if (rounds_required == m_kex_rounds_complete + 1) if (m_kex_rounds_complete == kex_rounds_required)
{ {
// finished (have set of msgs to complete address) // post-kex verification round: check that the multisig pubkey and common pubkey were recommended by other signers
CHECK_AND_ASSERT_THROW_MES(result_keys_to_origins_map.count(m_multisig_pubkey) > 0,
"Multisig post-kex round: expected multisig pubkey wasn't found in other signers' messages.");
CHECK_AND_ASSERT_THROW_MES(result_keys_to_origins_map.count(m_common_pubkey) > 0,
"Multisig post-kex round: expected common pubkey wasn't found in other signers' messages.");
// save keys that should be recommended to other signers
// - for convenience, re-recommend the post-kex verification message once an account is complete
next_msg_keys.reserve(2);
next_msg_keys.push_back(m_multisig_pubkey);
next_msg_keys.push_back(m_common_pubkey);
}
else if (m_kex_rounds_complete + 1 == kex_rounds_required)
{
// finished with main kex rounds (have set of msgs to complete address)
// when 'completing the final round', result keys are other signers' shares of the final key // when 'completing the final round', result keys are other signers' shares of the final key
std::vector<crypto::public_key> result_keys; std::vector<crypto::public_key> result_keys;
@ -652,8 +722,14 @@ namespace multisig
// no longer need the account's pubkeys saved for this round (they were only used to build exclude_pubkeys) // no longer need the account's pubkeys saved for this round (they were only used to build exclude_pubkeys)
// TODO: record [pre-aggregation pubkeys : origins] map for aggregation-style signing // TODO: record [pre-aggregation pubkeys : origins] map for aggregation-style signing
m_kex_keys_to_origins_map.clear(); m_kex_keys_to_origins_map.clear();
// save keys that should be recommended to other signers
// - for post-kex verification, recommend the multisig pubkeys to notify other signers that the local signer is done
next_msg_keys.reserve(2);
next_msg_keys.push_back(m_multisig_pubkey);
next_msg_keys.push_back(m_common_pubkey);
} }
else if (rounds_required == m_kex_rounds_complete + 2) else if (m_kex_rounds_complete + 2 == kex_rounds_required)
{ {
// one more round (must send/receive one more set of kex msgs) // one more round (must send/receive one more set of kex msgs)
// - at this point, have local signer's pre-aggregation private key shares of the final address // - at this point, have local signer's pre-aggregation private key shares of the final address
@ -668,6 +744,7 @@ namespace multisig
m_multisig_privkeys.reserve(result_keys_to_origins_map.size()); m_multisig_privkeys.reserve(result_keys_to_origins_map.size());
m_kex_keys_to_origins_map.clear(); m_kex_keys_to_origins_map.clear();
next_msg_keys.reserve(result_keys_to_origins_map.size());
for (const auto &derivation_and_origins : result_keys_to_origins_map) for (const auto &derivation_and_origins : result_keys_to_origins_map)
{ {
@ -679,37 +756,59 @@ namespace multisig
// save the account's kex key mappings for this round [derived pubkey : other signers who will have the same key] // save the account's kex key mappings for this round [derived pubkey : other signers who will have the same key]
m_kex_keys_to_origins_map[derived_pubkey] = std::move(derivation_and_origins.second); m_kex_keys_to_origins_map[derived_pubkey] = std::move(derivation_and_origins.second);
// save keys that should be recommended to other signers
// - The keys multisig_key*G are sent to other participants in the message, so they can be used to produce the final
// multisig key via generate_multisig_spend_public_key().
next_msg_keys.push_back(derived_pubkey);
} }
} }
else else //(m_kex_rounds_complete + 3 <= kex_rounds_required)
{ {
// next round is an 'intermediate' key exchange round, so there is nothing special to do here // next round is an 'intermediate' key exchange round, so there is nothing special to do here
// save the account's kex keys for this round [DH derivation : other signers who will have the same derivation] // save keys that should be recommended to other signers
// - Send this round's DH derivations to other participants, who will make more DH derivations for the following round.
next_msg_keys.reserve(result_keys_to_origins_map.size());
for (const auto &derivation_and_origins : result_keys_to_origins_map)
next_msg_keys.push_back(derivation_and_origins.first);
// save the account's kex keys for this round [DH derivation : other signers who should have the same derivation]
m_kex_keys_to_origins_map = std::move(result_keys_to_origins_map); m_kex_keys_to_origins_map = std::move(result_keys_to_origins_map);
} }
// a full set of msgs has been collected and processed, so the 'round is complete' // a full set of msgs has been collected and processed, so the 'round is complete'
++m_kex_rounds_complete; ++m_kex_rounds_complete;
// make next round's message (or reproduce the post-kex verification round if kex is complete)
m_next_round_kex_message = multisig_kex_msg{
(m_kex_rounds_complete > kex_rounds_required ? kex_rounds_required : m_kex_rounds_complete) + 1,
m_base_privkey,
std::move(next_msg_keys)}.get_msg();
} }
//---------------------------------------------------------------------------------------------------------------------- //----------------------------------------------------------------------------------------------------------------------
// multisig_account: INTERNAL // multisig_account: INTERNAL
//---------------------------------------------------------------------------------------------------------------------- //----------------------------------------------------------------------------------------------------------------------
void multisig_account::kex_update_impl(const std::vector<multisig_kex_msg> &expanded_msgs) void multisig_account::kex_update_impl(const std::vector<multisig_kex_msg> &expanded_msgs)
{ {
CHECK_AND_ASSERT_THROW_MES(expanded_msgs.size() > 0, "No key exchange messages passed in."); // check messages are for the expected kex round
check_messages_round(expanded_msgs, m_kex_rounds_complete + 1);
const std::uint32_t rounds_required = multisig_kex_rounds_required(m_signers.size(), m_threshold); // check kex round count
CHECK_AND_ASSERT_THROW_MES(rounds_required > 0, "Multisig kex rounds required unexpectedly 0."); const std::uint32_t kex_rounds_required{multisig_kex_rounds_required(m_signers.size(), m_threshold)};
CHECK_AND_ASSERT_THROW_MES(kex_rounds_required > 0, "Multisig kex rounds required unexpectedly 0.");
CHECK_AND_ASSERT_THROW_MES(m_kex_rounds_complete < kex_rounds_required + 1,
"Multisig kex has already completed all required rounds (including post-kex verification).");
// initialize account update // initialize account update
std::vector<crypto::public_key> exclude_pubkeys; std::vector<crypto::public_key> exclude_pubkeys;
initialize_kex_update(expanded_msgs, rounds_required, exclude_pubkeys); initialize_kex_update(expanded_msgs, kex_rounds_required, exclude_pubkeys);
// evaluate messages and get this account's kex msg for the next round // process messages into a [pubkey : {origins}] map
std::unordered_map<crypto::public_key_memsafe, std::unordered_set<crypto::public_key>> result_keys_to_origins_map; multisig_keyset_map_memsafe_t result_keys_to_origins_map;
multisig_kex_process_round_msgs(
m_next_round_kex_message = multisig_kex_process_round(
m_base_privkey, m_base_privkey,
m_base_pubkey, m_base_pubkey,
m_kex_rounds_complete + 1, m_kex_rounds_complete + 1,
@ -717,10 +816,10 @@ namespace multisig
m_signers, m_signers,
expanded_msgs, expanded_msgs,
exclude_pubkeys, exclude_pubkeys,
result_keys_to_origins_map).get_msg(); result_keys_to_origins_map);
// finish account update // finish account update
finalize_kex_update(rounds_required, std::move(result_keys_to_origins_map)); finalize_kex_update(kex_rounds_required, std::move(result_keys_to_origins_map));
} }
//---------------------------------------------------------------------------------------------------------------------- //----------------------------------------------------------------------------------------------------------------------
} //namespace multisig } //namespace multisig

View File

@ -1087,7 +1087,9 @@ bool simple_wallet::make_multisig_main(const std::vector<std::string> &args, boo
auto local_args = args; auto local_args = args;
local_args.erase(local_args.begin()); local_args.erase(local_args.begin());
std::string multisig_extra_info = m_wallet->make_multisig(orig_pwd_container->password(), local_args, threshold); std::string multisig_extra_info = m_wallet->make_multisig(orig_pwd_container->password(), local_args, threshold);
if (!multisig_extra_info.empty()) bool ready;
m_wallet->multisig(&ready);
if (!ready)
{ {
success_msg_writer() << tr("Another step is needed"); success_msg_writer() << tr("Another step is needed");
success_msg_writer() << multisig_extra_info; success_msg_writer() << multisig_extra_info;
@ -1148,7 +1150,7 @@ bool simple_wallet::exchange_multisig_keys_main(const std::vector<std::string> &
return false; return false;
} }
if (args.size() < 2) if (args.size() < 1)
{ {
PRINT_USAGE(USAGE_EXCHANGE_MULTISIG_KEYS); PRINT_USAGE(USAGE_EXCHANGE_MULTISIG_KEYS);
return false; return false;
@ -1157,7 +1159,9 @@ bool simple_wallet::exchange_multisig_keys_main(const std::vector<std::string> &
try try
{ {
std::string multisig_extra_info = m_wallet->exchange_multisig_keys(orig_pwd_container->password(), args); std::string multisig_extra_info = m_wallet->exchange_multisig_keys(orig_pwd_container->password(), args);
if (!multisig_extra_info.empty()) bool ready;
m_wallet->multisig(&ready);
if (!ready)
{ {
message_writer() << tr("Another step is needed"); message_writer() << tr("Another step is needed");
message_writer() << multisig_extra_info; message_writer() << multisig_extra_info;

View File

@ -5125,7 +5125,7 @@ std::string wallet2::exchange_multisig_keys(const epee::wipeable_string &passwor
// reconstruct multisig account // reconstruct multisig account
crypto::public_key dummy; crypto::public_key dummy;
multisig::multisig_account::kex_origins_map_t kex_origins_map; multisig::multisig_keyset_map_memsafe_t kex_origins_map;
for (const auto &derivation : m_multisig_derivations) for (const auto &derivation : m_multisig_derivations)
kex_origins_map[derivation]; kex_origins_map[derivation];
@ -5138,7 +5138,7 @@ std::string wallet2::exchange_multisig_keys(const epee::wipeable_string &passwor
get_account().get_keys().m_multisig_keys, get_account().get_keys().m_multisig_keys,
get_account().get_keys().m_view_secret_key, get_account().get_keys().m_view_secret_key,
m_account_public_address.m_spend_public_key, m_account_public_address.m_spend_public_key,
dummy, //common pubkey: not used m_account_public_address.m_view_public_key,
m_multisig_rounds_passed, m_multisig_rounds_passed,
std::move(kex_origins_map), std::move(kex_origins_map),
"" ""
@ -5225,7 +5225,10 @@ bool wallet2::multisig(bool *ready, uint32_t *threshold, uint32_t *total) const
if (total) if (total)
*total = m_multisig_signers.size(); *total = m_multisig_signers.size();
if (ready) if (ready)
*ready = !(get_account().get_keys().m_account_address.m_spend_public_key == rct::rct2pk(rct::identity())); {
*ready = !(get_account().get_keys().m_account_address.m_spend_public_key == rct::rct2pk(rct::identity())) &&
(m_multisig_rounds_passed == multisig::multisig_kex_rounds_required(m_multisig_signers.size(), m_multisig_threshold) + 1);
}
return true; return true;
} }
//---------------------------------------------------------------------------------------------------- //----------------------------------------------------------------------------------------------------

View File

@ -4134,7 +4134,8 @@ namespace tools
try try
{ {
res.multisig_info = m_wallet->exchange_multisig_keys(req.password, req.multisig_info); res.multisig_info = m_wallet->exchange_multisig_keys(req.password, req.multisig_info);
if (res.multisig_info.empty()) m_wallet->multisig(&ready);
if (ready)
{ {
res.address = m_wallet->get_account().get_public_address_str(m_wallet->nettype()); res.address = m_wallet->get_account().get_public_address_str(m_wallet->nettype());
} }

View File

@ -81,8 +81,6 @@ static bool make_multisig_accounts(std::vector<cryptonote::account_base> &accoun
for (std::size_t account_index{0}; account_index < accounts.size(); ++account_index) for (std::size_t account_index{0}; account_index < accounts.size(); ++account_index)
{ {
multisig_accounts[account_index].initialize_kex(threshold, signers, round_msgs); multisig_accounts[account_index].initialize_kex(threshold, signers, round_msgs);
if (!multisig_accounts[account_index].multisig_is_ready())
temp_round_msgs[account_index] = multisig_accounts[account_index].get_next_kex_round_msg(); temp_round_msgs[account_index] = multisig_accounts[account_index].get_next_kex_round_msg();
} }
@ -94,8 +92,6 @@ static bool make_multisig_accounts(std::vector<cryptonote::account_base> &accoun
for (std::size_t account_index{0}; account_index < multisig_accounts.size(); ++account_index) for (std::size_t account_index{0}; account_index < multisig_accounts.size(); ++account_index)
{ {
multisig_accounts[account_index].kex_update(round_msgs); multisig_accounts[account_index].kex_update(round_msgs);
if (!multisig_accounts[account_index].multisig_is_ready())
temp_round_msgs[account_index] = multisig_accounts[account_index].get_next_kex_round_msg(); temp_round_msgs[account_index] = multisig_accounts[account_index].get_next_kex_round_msg();
} }
} }

View File

@ -125,17 +125,18 @@ class MultisigTest():
for i in range(N_total): for i in range(N_total):
res = self.wallet[i].is_multisig() res = self.wallet[i].is_multisig()
assert res.multisig == True assert res.multisig == True
assert res.ready == (M_threshold == N_total) assert not res.ready
assert res.threshold == M_threshold assert res.threshold == M_threshold
assert res.total == N_total assert res.total == N_total
while True: while True:
n_empty = 0 n_ready = 0
for i in range(len(next_stage)): for i in range(N_total):
if len(next_stage[i]) == 0: res = self.wallet[i].is_multisig()
n_empty += 1 if res.ready == True:
assert n_empty == 0 or n_empty == len(next_stage) n_ready += 1
if n_empty == len(next_stage): assert n_ready == 0 or n_ready == N_total
if n_ready == N_total:
break break
info = next_stage info = next_stage
next_stage = [] next_stage = []
@ -162,54 +163,72 @@ class MultisigTest():
'peeled mixture ionic radar utopia puddle buying illness nuns gadget river spout cavernous bounced paradise drunk looking cottage jump tequila melting went winter adjust spout', 'peeled mixture ionic radar utopia puddle buying illness nuns gadget river spout cavernous bounced paradise drunk looking cottage jump tequila melting went winter adjust spout',
'dilute gutter certain antics pamphlet macro enjoy left slid guarded bogeys upload nineteen bomb jubilee enhanced irritate turnip eggs swung jukebox loudly reduce sedan slid', 'dilute gutter certain antics pamphlet macro enjoy left slid guarded bogeys upload nineteen bomb jubilee enhanced irritate turnip eggs swung jukebox loudly reduce sedan slid',
] ]
info = [] info2of2 = []
wallet = [None, None, None] wallet2of2 = [None, None]
for i in range(3): for i in range(2):
wallet[i] = Wallet(idx = i) wallet2of2[i] = Wallet(idx = i)
try: wallet[i].close_wallet() try: wallet2of2[i].close_wallet()
except: pass except: pass
res = wallet[i].restore_deterministic_wallet(seed = seeds[i]) res = wallet2of2[i].restore_deterministic_wallet(seed = seeds[i])
res = wallet[i].is_multisig() res = wallet2of2[i].is_multisig()
assert not res.multisig assert not res.multisig
res = wallet[i].prepare_multisig() res = wallet2of2[i].prepare_multisig()
assert len(res.multisig_info) > 0 assert len(res.multisig_info) > 0
info.append(res.multisig_info) info2of2.append(res.multisig_info)
for i in range(3): kex_info = []
ok = False res = wallet2of2[0].make_multisig(info2of2, 2)
try: res = wallet[i].exchange_multisig_keys(info) kex_info.append(res.multisig_info)
except: ok = True res = wallet2of2[1].make_multisig(info2of2, 2)
assert ok kex_info.append(res.multisig_info)
res = wallet[i].is_multisig() res = wallet2of2[0].exchange_multisig_keys(kex_info)
assert not res.multisig res = wallet2of2[0].is_multisig()
res = wallet[0].make_multisig(info[0:2], 2)
res = wallet[0].is_multisig()
assert res.multisig assert res.multisig
assert res.ready assert res.ready
ok = False ok = False
try: res = wallet[0].prepare_multisig() try: res = wallet2of2[0].prepare_multisig()
except: ok = True except: ok = True
assert ok assert ok
ok = False ok = False
try: res = wallet[0].make_multisig(info[0:2], 2) try: res = wallet2of2[0].make_multisig(info2of2, 2)
except: ok = True except: ok = True
assert ok assert ok
res = wallet[1].make_multisig(info, 2) info2of3 = []
res = wallet[1].is_multisig() wallet2of3 = [None, None, None]
for i in range(3):
wallet2of3[i] = Wallet(idx = i)
try: wallet2of3[i].close_wallet()
except: pass
res = wallet2of3[i].restore_deterministic_wallet(seed = seeds[i])
res = wallet2of3[i].is_multisig()
assert not res.multisig
res = wallet2of3[i].prepare_multisig()
assert len(res.multisig_info) > 0
info2of3.append(res.multisig_info)
for i in range(3):
ok = False
try: res = wallet2of3[i].exchange_multisig_keys(info)
except: ok = True
assert ok
res = wallet2of3[i].is_multisig()
assert not res.multisig
res = wallet2of3[1].make_multisig(info2of3, 2)
res = wallet2of3[1].is_multisig()
assert res.multisig assert res.multisig
assert not res.ready assert not res.ready
ok = False ok = False
try: res = wallet[1].prepare_multisig() try: res = wallet2of3[1].prepare_multisig()
except: ok = True except: ok = True
assert ok assert ok
ok = False ok = False
try: res = wallet[1].make_multisig(info[0:2], 2) try: res = wallet2of3[1].make_multisig(info2of3[0:2], 2)
except: ok = True except: ok = True
assert ok assert ok

View File

@ -120,7 +120,7 @@ static void check_results(const std::vector<std::string> &intermediate_infos,
for (size_t i = 0; i < wallets.size(); ++i) for (size_t i = 0; i < wallets.size(); ++i)
{ {
EXPECT_TRUE(intermediate_infos[i].empty()); EXPECT_TRUE(!intermediate_infos[i].empty());
bool ready; bool ready;
uint32_t threshold, total; uint32_t threshold, total;
EXPECT_TRUE(wallets[i].multisig(&ready, &threshold, &total)); EXPECT_TRUE(wallets[i].multisig(&ready, &threshold, &total));
@ -171,7 +171,7 @@ static void make_wallets(std::vector<tools::wallet2>& wallets, unsigned int M)
{ {
ASSERT_TRUE(wallets.size() > 1 && wallets.size() <= KEYS_COUNT); ASSERT_TRUE(wallets.size() > 1 && wallets.size() <= KEYS_COUNT);
ASSERT_TRUE(M <= wallets.size()); ASSERT_TRUE(M <= wallets.size());
std::uint32_t rounds_required = multisig::multisig_kex_rounds_required(wallets.size(), M); std::uint32_t total_rounds_required = multisig::multisig_kex_rounds_required(wallets.size(), M) + 1;
std::uint32_t rounds_complete{0}; std::uint32_t rounds_complete{0};
// initialize wallets, get first round multisig kex msgs // initialize wallets, get first round multisig kex msgs
@ -203,18 +203,17 @@ static void make_wallets(std::vector<tools::wallet2>& wallets, unsigned int M)
++rounds_complete; ++rounds_complete;
// perform kex rounds until kex is complete // perform kex rounds until kex is complete
while (!intermediate_infos[0].empty()) bool ready;
{
bool ready{false};
wallets[0].multisig(&ready); wallets[0].multisig(&ready);
EXPECT_FALSE(ready); while (!ready)
{
intermediate_infos = exchange_round(wallets, intermediate_infos); intermediate_infos = exchange_round(wallets, intermediate_infos);
wallets[0].multisig(&ready);
++rounds_complete; ++rounds_complete;
} }
EXPECT_EQ(rounds_required, rounds_complete); EXPECT_EQ(total_rounds_required, rounds_complete);
check_results(intermediate_infos, wallets, M); check_results(intermediate_infos, wallets, M);
} }