Merge #19106: util: simplify the interface of serviceFlagToStr()

189ae0c38b7d4927c5c73b94664e9542b2b06ed9 util: dedup code in callers of serviceFlagToStr() (Vasil Dimov)
fbacad1880341ace31f669530c66d4e322d19235 util: simplify the interface of serviceFlagToStr() (Vasil Dimov)

Pull request description:

  Don't take two redundant arguments in `serviceFlagToStr()`.

  Introduce `serviceFlagsToStr()` which takes a mask (with more than one
  bit set) and returns a vector of strings.

  As a side effect this fixes an issue introduced in
  https://github.com/bitcoin/bitcoin/pull/18165 due to which the GUI could
  print something like `UNKNOWN[1033] & UNKNOWN[1033] & UNKNOWN[2^10]`
  instead of `NETWORK & WITNESS`.

ACKs for top commit:
  MarcoFalke:
    ACK 189ae0c38b7d4927c5c73b94664e9542b2b06ed9
  jonasschnelli:
    Tested ACK 189ae0c38b7d4927c5c73b94664e9542b2b06ed9

Tree-SHA512: 000c490f16ebbba04458c62ca4ce743abffd344d375d95f5bbd5008742012032787655db2874b168df0270743266261dccf1693761906567502dcbac902bda50
This commit is contained in:
Jonas Schnelli 2020-05-29 19:43:02 +02:00 committed by pasta
parent c904c38931
commit c5f3b478bf
4 changed files with 32 additions and 16 deletions

View File

@ -318,9 +318,15 @@ const std::vector<std::string> &getAllNetMessageTypes()
return allNetMessageTypesVec; return allNetMessageTypesVec;
} }
std::string serviceFlagToStr(const uint64_t mask, const int bit) /**
* Convert a service flag (NODE_*) to a human readable string.
* It supports unknown service flags which will be returned as "UNKNOWN[...]".
* @param[in] bit the service flag is calculated as (1 << bit)
*/
static std::string serviceFlagToStr(size_t bit)
{ {
switch (ServiceFlags(mask)) { const uint64_t service_flag = 1ULL << bit;
switch ((ServiceFlags)service_flag) {
case NODE_NONE: abort(); // impossible case NODE_NONE: abort(); // impossible
case NODE_NETWORK: return "NETWORK"; case NODE_NETWORK: return "NETWORK";
case NODE_GETUTXO: return "GETUTXO"; case NODE_GETUTXO: return "GETUTXO";
@ -335,10 +341,23 @@ std::string serviceFlagToStr(const uint64_t mask, const int bit)
stream.imbue(std::locale::classic()); stream.imbue(std::locale::classic());
stream << "UNKNOWN["; stream << "UNKNOWN[";
if (bit < 8) { if (bit < 8) {
stream << mask; stream << service_flag;
} else { } else {
stream << "2^" << bit; stream << "2^" << bit;
} }
stream << "]"; stream << "]";
return stream.str(); return stream.str();
} }
std::vector<std::string> serviceFlagsToStr(uint64_t flags)
{
std::vector<std::string> str_flags;
for (size_t i = 0; i < sizeof(flags) * 8; ++i) {
if (flags & (1ULL << i)) {
str_flags.emplace_back(serviceFlagToStr(i));
}
}
return str_flags;
}

View File

@ -333,7 +333,12 @@ enum ServiceFlags : uint64_t {
// BIP process. // BIP process.
}; };
std::string serviceFlagToStr(uint64_t mask, int bit); /**
* Convert service flags (a bitmask of NODE_*) to human readable strings.
* It supports unknown service flags which will be returned as "UNKNOWN[...]".
* @param[in] flags multiple NODE_* bitwise-OR-ed together
*/
std::vector<std::string> serviceFlagsToStr(uint64_t flags);
/** /**
* Gets the set of service flags which are "desirable" for a given peer. * Gets the set of service flags which are "desirable" for a given peer.

View File

@ -1725,12 +1725,8 @@ QString formatServicesStr(quint64 mask)
{ {
QStringList strList; QStringList strList;
for (int i = 0; i < 64; i++) { for (const auto& flag : serviceFlagsToStr(mask)) {
uint64_t check = 1ull << i; strList.append(QString::fromStdString(flag));
if (mask & check)
{
strList.append(QString::fromStdString(serviceFlagToStr(mask, i)));
}
} }
if (strList.size()) if (strList.size())

View File

@ -226,14 +226,10 @@ UniValue JSONRPCTransactionError(TransactionError terr, const std::string& err_s
UniValue GetServicesNames(ServiceFlags services) UniValue GetServicesNames(ServiceFlags services)
{ {
const uint64_t services_n = services;
UniValue servicesNames(UniValue::VARR); UniValue servicesNames(UniValue::VARR);
for (int i = 0; i < 64; ++i) { for (const auto& flag : serviceFlagsToStr(services)) {
const uint64_t mask = 1ull << i; servicesNames.push_back(flag);
if (services_n & mask) {
servicesNames.push_back(serviceFlagToStr(mask, i));
}
} }
return servicesNames; return servicesNames;