Branch data     Line data    Source code

       1                 :            : // Copyright (c) 2021-2022 The Bitcoin Core developers
       2                 :            : // Distributed under the MIT software license, see the accompanying
       3                 :            : // file COPYING or http://www.opensource.org/licenses/mit-license.php.
       4                 :            : 
       5                 :            : #include <net_types.h>
       6                 :            : 
       7                 :            : #include <logging.h>
       8                 :            : #include <netaddress.h>
       9                 :            : #include <netbase.h>
      10                 :            : #include <univalue.h>
      11                 :            : 
      12                 :            : static const char* BANMAN_JSON_VERSION_KEY{"version"};
      13                 :            : 
      14                 :       9521 : CBanEntry::CBanEntry(const UniValue& json)
      15   [ +  -  +  -  :       9535 :     : nVersion(json[BANMAN_JSON_VERSION_KEY].getInt<int>()),
                   +  - ]
      16   [ +  -  +  -  :       9521 :       nCreateTime(json["ban_created"].getInt<int64_t>()),
                   +  + ]
      17   [ -  +  +  -  :       9511 :       nBanUntil(json["banned_until"].getInt<int64_t>())
                   +  + ]
      18                 :            : {
      19                 :       9521 : }
      20                 :            : 
      21                 :     595133 : UniValue CBanEntry::ToJson() const
      22                 :            : {
      23         [ +  - ]:     595133 :     UniValue json(UniValue::VOBJ);
      24   [ +  -  +  -  :     595133 :     json.pushKV(BANMAN_JSON_VERSION_KEY, nVersion);
                   +  - ]
      25   [ +  -  +  -  :     595133 :     json.pushKV("ban_created", nCreateTime);
                   +  - ]
      26   [ +  -  +  -  :     595133 :     json.pushKV("banned_until", nBanUntil);
                   +  - ]
      27                 :     595133 :     return json;
      28         [ +  - ]:     595133 : }
      29                 :            : 
      30                 :            : static const char* BANMAN_JSON_ADDR_KEY = "address";
      31                 :            : 
      32                 :            : /**
      33                 :            :  * Convert a `banmap_t` object to a JSON array.
      34                 :            :  * @param[in] bans Bans list to convert.
      35                 :            :  * @return a JSON array, similar to the one returned by the `listbanned` RPC. Suitable for
      36                 :            :  * passing to `BanMapFromJson()`.
      37                 :            :  */
      38                 :      14140 : UniValue BanMapToJson(const banmap_t& bans)
      39                 :            : {
      40         [ +  - ]:      14140 :     UniValue bans_json(UniValue::VARR);
      41         [ +  + ]:     609273 :     for (const auto& it : bans) {
      42                 :     595133 :         const auto& address = it.first;
      43                 :     595133 :         const auto& ban_entry = it.second;
      44         [ +  - ]:     595133 :         UniValue j = ban_entry.ToJson();
      45   [ +  -  +  -  :     595133 :         j.pushKV(BANMAN_JSON_ADDR_KEY, address.ToString());
             +  -  +  - ]
      46   [ +  -  +  - ]:     595133 :         bans_json.push_back(j);
      47                 :     595133 :     }
      48                 :      14140 :     return bans_json;
      49         [ +  - ]:      14140 : }
      50                 :            : 
      51                 :            : /**
      52                 :            :  * Convert a JSON array to a `banmap_t` object.
      53                 :            :  * @param[in] bans_json JSON to convert, must be as returned by `BanMapToJson()`.
      54                 :            :  * @param[out] bans Bans list to create from the JSON.
      55                 :            :  * @throws std::runtime_error if the JSON does not have the expected fields or they contain
      56                 :            :  * unparsable values.
      57                 :            :  */
      58                 :        794 : void BanMapFromJson(const UniValue& bans_json, banmap_t& bans)
      59                 :            : {
      60         [ +  + ]:      10950 :     for (const auto& ban_entry_json : bans_json.getValues()) {
      61   [ +  -  +  -  :      10193 :         const int version{ban_entry_json[BANMAN_JSON_VERSION_KEY].getInt<int>()};
                   +  + ]
      62         [ +  + ]:      10171 :         if (version != CBanEntry::CURRENT_VERSION) {
      63   [ +  -  +  -  :        362 :             LogPrintf("Dropping entry with unknown version (%s) from ban list\n", version);
                   +  - ]
      64                 :        362 :             continue;
      65                 :            :         }
      66                 :       9809 :         CSubNet subnet;
      67   [ +  -  +  -  :       9809 :         const auto& subnet_str = ban_entry_json[BANMAN_JSON_ADDR_KEY].get_str();
                   +  + ]
      68   [ +  -  +  + ]:       9808 :         if (!LookupSubNet(subnet_str, subnet)) {
      69   [ +  -  +  -  :        287 :             LogPrintf("Dropping entry with unparseable address or subnet (%s) from ban list\n", subnet_str);
                   +  - ]
      70                 :        287 :             continue;
      71                 :            :         }
      72   [ +  +  +  - ]:       9521 :         bans.insert_or_assign(subnet, CBanEntry{ban_entry_json});
      73      [ -  +  + ]:       9809 :     }
      74                 :        967 : }