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 : 21997 : CBanEntry::CBanEntry(const UniValue& json)
15 [ + - + - : 22053 : : nVersion(json[BANMAN_JSON_VERSION_KEY].getInt<int>()),
+ - ]
16 [ + - + - : 21997 : nCreateTime(json["ban_created"].getInt<int64_t>()),
+ + ]
17 [ + - + - : 21961 : nBanUntil(json["banned_until"].getInt<int64_t>())
+ + ]
18 : : {
19 : 21997 : }
20 : :
21 : 1157692 : UniValue CBanEntry::ToJson() const
22 : : {
23 [ + - ]: 1157692 : UniValue json(UniValue::VOBJ);
24 [ + - + - : 1157692 : json.pushKV(BANMAN_JSON_VERSION_KEY, nVersion);
+ - ]
25 [ + - + - : 1157692 : json.pushKV("ban_created", nCreateTime);
+ - ]
26 [ + - + - : 1157692 : json.pushKV("banned_until", nBanUntil);
+ - ]
27 : 1157692 : return json;
28 [ + - ]: 1157692 : }
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 : 36561 : UniValue BanMapToJson(const banmap_t& bans)
39 : : {
40 [ + - ]: 36561 : UniValue bans_json(UniValue::VARR);
41 [ + + ]: 1194253 : for (const auto& it : bans) {
42 : 1157692 : const auto& address = it.first;
43 : 1157692 : const auto& ban_entry = it.second;
44 [ + - ]: 1157692 : UniValue j = ban_entry.ToJson();
45 [ + - - + : 1157692 : j.pushKV(BANMAN_JSON_ADDR_KEY, address.ToString());
- + - + ]
46 [ + - ]: 1157692 : bans_json.push_back(std::move(j));
47 : 1157692 : }
48 : 36561 : return bans_json;
49 [ + - ]: 36561 : }
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 : 2895 : void BanMapFromJson(const UniValue& bans_json, banmap_t& bans)
59 : : {
60 [ + + ]: 29845 : for (const auto& ban_entry_json : bans_json.getValues()) {
61 [ - + + - : 26950 : const int version{ban_entry_json[BANMAN_JSON_VERSION_KEY].getInt<int>()};
+ + ]
62 [ + + ]: 26882 : if (version != CBanEntry::CURRENT_VERSION) {
63 : 2072 : LogPrintf("Dropping entry with unknown version (%s) from ban list\n", version);
64 : 2072 : continue;
65 : : }
66 [ - + + - : 24810 : const auto& subnet_str = ban_entry_json[BANMAN_JSON_ADDR_KEY].get_str();
+ + ]
67 : 24799 : const CSubNet subnet{LookupSubNet(subnet_str)};
68 [ + - + + ]: 24799 : if (!subnet.IsValid()) {
69 [ + - ]: 2802 : LogPrintf("Dropping entry with unparseable address or subnet (%s) from ban list\n", subnet_str);
70 : 2802 : continue;
71 : : }
72 [ + + + - ]: 21997 : bans.insert_or_assign(subnet, CBanEntry{ban_entry_json});
73 [ + + - + : 26950 : }
+ ]
74 : 2895 : }
|