Skip to content

Commit fa1433a

Browse files
author
MarcoFalke
committed
rpc: Remove special case for unknown service flags
1 parent cb88de3 commit fa1433a

File tree

2 files changed

+11
-6
lines changed

2 files changed

+11
-6
lines changed

src/protocol.cpp

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -5,8 +5,8 @@
55

66
#include <protocol.h>
77

8-
#include <util/system.h>
98
#include <util/strencodings.h>
9+
#include <util/system.h>
1010

1111
#ifndef WIN32
1212
# include <arpa/inet.h>
@@ -216,11 +216,7 @@ static std::string serviceFlagToStr(size_t bit)
216216
std::ostringstream stream;
217217
stream.imbue(std::locale::classic());
218218
stream << "UNKNOWN[";
219-
if (bit < 8) {
220-
stream << service_flag;
221-
} else {
222-
stream << "2^" << bit;
223-
}
219+
stream << "2^" << bit;
224220
stream << "]";
225221
return stream.str();
226222
}

test/functional/rpc_net.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,7 @@
2828
NODE_WITNESS,
2929
)
3030

31+
3132
def assert_net_servicesnames(servicesflag, servicenames):
3233
"""Utility that checks if all flags are correctly decoded in
3334
`getpeerinfo` and `getnetworkinfo`.
@@ -40,6 +41,7 @@ def assert_net_servicesnames(servicesflag, servicenames):
4041
servicesflag_generated |= getattr(test_framework.messages, 'NODE_' + servicename)
4142
assert servicesflag_generated == servicesflag
4243

44+
4345
class NetTest(BitcoinTestFramework):
4446
def set_test_params(self):
4547
self.setup_clean_chain = True
@@ -57,6 +59,7 @@ def run_test(self):
5759
self._test_getnetworkinfo()
5860
self._test_getaddednodeinfo()
5961
self._test_getpeerinfo()
62+
self.test_service_flags()
6063
self._test_getnodeaddresses()
6164

6265
def _test_connection_count(self):
@@ -139,6 +142,11 @@ def _test_getpeerinfo(self):
139142
for info in peer_info:
140143
assert_net_servicesnames(int(info[0]["services"], 0x10), info[0]["servicesnames"])
141144

145+
def test_service_flags(self):
146+
self.nodes[0].add_p2p_connection(P2PInterface(), services=(1 << 4) | (1 << 63))
147+
assert_equal(['UNKNOWN[2^4]', 'UNKNOWN[2^63]'], self.nodes[0].getpeerinfo()[-1]['servicesnames'])
148+
self.nodes[0].disconnect_p2ps()
149+
142150
def _test_getnodeaddresses(self):
143151
self.nodes[0].add_p2p_connection(P2PInterface())
144152

@@ -174,5 +182,6 @@ def _test_getnodeaddresses(self):
174182
node_addresses = self.nodes[0].getnodeaddresses(LARGE_REQUEST_COUNT)
175183
assert_greater_than(LARGE_REQUEST_COUNT, len(node_addresses))
176184

185+
177186
if __name__ == '__main__':
178187
NetTest().main()

0 commit comments

Comments
 (0)