Changeset View
Changeset View
Standalone View
Standalone View
test/functional/feature_proxy.py
| Show First 20 Lines • Show All 58 Lines • ▼ Show 20 Lines | def set_test_params(self): | ||||
| self.num_nodes = 4 | self.num_nodes = 4 | ||||
| self.setup_clean_chain = True | self.setup_clean_chain = True | ||||
| def setup_nodes(self): | def setup_nodes(self): | ||||
| self.have_ipv6 = test_ipv6_local() | self.have_ipv6 = test_ipv6_local() | ||||
| # Create two proxies on different ports | # Create two proxies on different ports | ||||
| # ... one unauthenticated | # ... one unauthenticated | ||||
| self.conf1 = Socks5Configuration() | self.conf1 = Socks5Configuration() | ||||
| self.conf1.addr = ('127.0.0.1', p2p_port(self.num_nodes)) | self.conf1.addr = ("127.0.0.1", p2p_port(self.num_nodes)) | ||||
| self.conf1.unauth = True | self.conf1.unauth = True | ||||
| self.conf1.auth = False | self.conf1.auth = False | ||||
| # ... one supporting authenticated and unauthenticated (Tor) | # ... one supporting authenticated and unauthenticated (Tor) | ||||
| self.conf2 = Socks5Configuration() | self.conf2 = Socks5Configuration() | ||||
| self.conf2.addr = ('127.0.0.1', p2p_port(self.num_nodes + 1)) | self.conf2.addr = ("127.0.0.1", p2p_port(self.num_nodes + 1)) | ||||
| self.conf2.unauth = True | self.conf2.unauth = True | ||||
| self.conf2.auth = True | self.conf2.auth = True | ||||
| if self.have_ipv6: | if self.have_ipv6: | ||||
| # ... one on IPv6 with similar configuration | # ... one on IPv6 with similar configuration | ||||
| self.conf3 = Socks5Configuration() | self.conf3 = Socks5Configuration() | ||||
| self.conf3.af = socket.AF_INET6 | self.conf3.af = socket.AF_INET6 | ||||
| self.conf3.addr = ('::1', p2p_port(self.num_nodes + 2)) | self.conf3.addr = ("::1", p2p_port(self.num_nodes + 2)) | ||||
| self.conf3.unauth = True | self.conf3.unauth = True | ||||
| self.conf3.auth = True | self.conf3.auth = True | ||||
| else: | else: | ||||
| self.log.warning("Testing without local IPv6 support") | self.log.warning("Testing without local IPv6 support") | ||||
| self.serv1 = Socks5Server(self.conf1) | self.serv1 = Socks5Server(self.conf1) | ||||
| self.serv1.start() | self.serv1.start() | ||||
| self.serv2 = Socks5Server(self.conf2) | self.serv2 = Socks5Server(self.conf2) | ||||
| self.serv2.start() | self.serv2.start() | ||||
| if self.have_ipv6: | if self.have_ipv6: | ||||
| self.serv3 = Socks5Server(self.conf3) | self.serv3 = Socks5Server(self.conf3) | ||||
| self.serv3.start() | self.serv3.start() | ||||
| # We will not try to connect to this. | # We will not try to connect to this. | ||||
| self.i2p_sam = ('127.0.0.1', 7656) | self.i2p_sam = ("127.0.0.1", 7656) | ||||
| # Note: proxies are not used to connect to local nodes. | # Note: proxies are not used to connect to local nodes. | ||||
| # This is because the proxy to use is based on CService.GetNetwork(), | # This is because the proxy to use is based on CService.GetNetwork(), | ||||
| # which returns NET_UNROUTABLE for localhost. | # which returns NET_UNROUTABLE for localhost. | ||||
| ip1, port1 = self.conf1.addr | ip1, port1 = self.conf1.addr | ||||
| ip2, port2 = self.conf2.addr | ip2, port2 = self.conf2.addr | ||||
| args = [ | args = [ | ||||
| ['-listen', f'-proxy={ip1}:{port1}', '-proxyrandomize=1'], | ["-listen", f"-proxy={ip1}:{port1}", "-proxyrandomize=1"], | ||||
| ['-listen', f'-proxy={ip1}:{port1}', f'-onion={ip2}:{port2}', | [ | ||||
| f'-i2psam={self.i2p_sam[0]}:{self.i2p_sam[1]}', '-i2pacceptincoming=0', | "-listen", | ||||
| '-proxyrandomize=0'], | f"-proxy={ip1}:{port1}", | ||||
| ['-listen', f'-proxy={ip2}:{port2}', '-proxyrandomize=1'], | f"-onion={ip2}:{port2}", | ||||
| [] | f"-i2psam={self.i2p_sam[0]}:{self.i2p_sam[1]}", | ||||
| "-i2pacceptincoming=0", | |||||
| "-proxyrandomize=0", | |||||
| ], | |||||
| ["-listen", f"-proxy={ip2}:{port2}", "-proxyrandomize=1"], | |||||
| [], | |||||
| ] | ] | ||||
| if self.have_ipv6: | if self.have_ipv6: | ||||
| args[3] = [ | args[3] = [ | ||||
| '-listen', | "-listen", | ||||
| f'-proxy=[{self.conf3.addr[0]}]:{self.conf3.addr[1]}', | f"-proxy=[{self.conf3.addr[0]}]:{self.conf3.addr[1]}", | ||||
| '-proxyrandomize=0', '-noonion'] | "-proxyrandomize=0", | ||||
| "-noonion", | |||||
| ] | |||||
| self.add_nodes(self.num_nodes, extra_args=args) | self.add_nodes(self.num_nodes, extra_args=args) | ||||
| self.start_nodes() | self.start_nodes() | ||||
| def network_test(self, node, addr, network): | def network_test(self, node, addr, network): | ||||
| for peer in node.getpeerinfo(): | for peer in node.getpeerinfo(): | ||||
| if peer["addr"] == addr: | if peer["addr"] == addr: | ||||
| assert_equal(peer["network"], network) | assert_equal(peer["network"], network) | ||||
| def node_test(self, node, proxies, auth, test_onion=True): | def node_test(self, node, proxies, auth, test_onion=True): | ||||
| rv = [] | rv = [] | ||||
| addr = "15.61.23.23:1234" | addr = "15.61.23.23:1234" | ||||
| self.log.debug( | self.log.debug( | ||||
| f"Test: outgoing IPv4 connection through node for address {addr}") | f"Test: outgoing IPv4 connection through node for address {addr}" | ||||
| ) | |||||
| node.addnode(addr, "onetry") | node.addnode(addr, "onetry") | ||||
| cmd = proxies[0].queue.get() | cmd = proxies[0].queue.get() | ||||
| assert isinstance(cmd, Socks5Command) | assert isinstance(cmd, Socks5Command) | ||||
| # Note: bitcoind's SOCKS5 implementation only sends atyp DOMAINNAME, | # Note: bitcoind's SOCKS5 implementation only sends atyp DOMAINNAME, | ||||
| # even if connecting directly to IPv4/IPv6 | # even if connecting directly to IPv4/IPv6 | ||||
| assert_equal(cmd.atyp, AddressType.DOMAINNAME) | assert_equal(cmd.atyp, AddressType.DOMAINNAME) | ||||
| assert_equal(cmd.addr, b"15.61.23.23") | assert_equal(cmd.addr, b"15.61.23.23") | ||||
| assert_equal(cmd.port, 1234) | assert_equal(cmd.port, 1234) | ||||
| if not auth: | if not auth: | ||||
| assert_equal(cmd.username, None) | assert_equal(cmd.username, None) | ||||
| assert_equal(cmd.password, None) | assert_equal(cmd.password, None) | ||||
| rv.append(cmd) | rv.append(cmd) | ||||
| self.network_test(node, addr, network=NET_IPV4) | self.network_test(node, addr, network=NET_IPV4) | ||||
| if self.have_ipv6: | if self.have_ipv6: | ||||
| addr = "[1233:3432:2434:2343:3234:2345:6546:4534]:5443" | addr = "[1233:3432:2434:2343:3234:2345:6546:4534]:5443" | ||||
| self.log.debug( | self.log.debug( | ||||
| f"Test: outgoing IPv6 connection through node for address {addr}") | f"Test: outgoing IPv6 connection through node for address {addr}" | ||||
| ) | |||||
| node.addnode(addr, "onetry") | node.addnode(addr, "onetry") | ||||
| cmd = proxies[1].queue.get() | cmd = proxies[1].queue.get() | ||||
| assert isinstance(cmd, Socks5Command) | assert isinstance(cmd, Socks5Command) | ||||
| # Note: bitcoind's SOCKS5 implementation only sends atyp | # Note: bitcoind's SOCKS5 implementation only sends atyp | ||||
| # DOMAINNAME, even if connecting directly to IPv4/IPv6 | # DOMAINNAME, even if connecting directly to IPv4/IPv6 | ||||
| assert_equal(cmd.atyp, AddressType.DOMAINNAME) | assert_equal(cmd.atyp, AddressType.DOMAINNAME) | ||||
| assert_equal(cmd.addr, b"1233:3432:2434:2343:3234:2345:6546:4534") | assert_equal(cmd.addr, b"1233:3432:2434:2343:3234:2345:6546:4534") | ||||
| assert_equal(cmd.port, 5443) | assert_equal(cmd.port, 5443) | ||||
| if not auth: | if not auth: | ||||
| assert_equal(cmd.username, None) | assert_equal(cmd.username, None) | ||||
| assert_equal(cmd.password, None) | assert_equal(cmd.password, None) | ||||
| rv.append(cmd) | rv.append(cmd) | ||||
| self.network_test(node, addr, network=NET_IPV6) | self.network_test(node, addr, network=NET_IPV6) | ||||
| if test_onion: | if test_onion: | ||||
| addr = "bitcoinostk4e4re.onion:8333" | addr = "bitcoinostk4e4re.onion:8333" | ||||
| self.log.debug( | self.log.debug( | ||||
| f"Test: outgoing onion connection through node for address {addr}") | f"Test: outgoing onion connection through node for address {addr}" | ||||
| ) | |||||
| node.addnode(addr, "onetry") | node.addnode(addr, "onetry") | ||||
| cmd = proxies[2].queue.get() | cmd = proxies[2].queue.get() | ||||
| assert isinstance(cmd, Socks5Command) | assert isinstance(cmd, Socks5Command) | ||||
| assert_equal(cmd.atyp, AddressType.DOMAINNAME) | assert_equal(cmd.atyp, AddressType.DOMAINNAME) | ||||
| assert_equal(cmd.addr, b"bitcoinostk4e4re.onion") | assert_equal(cmd.addr, b"bitcoinostk4e4re.onion") | ||||
| assert_equal(cmd.port, 8333) | assert_equal(cmd.port, 8333) | ||||
| if not auth: | if not auth: | ||||
| assert_equal(cmd.username, None) | assert_equal(cmd.username, None) | ||||
| assert_equal(cmd.password, None) | assert_equal(cmd.password, None) | ||||
| rv.append(cmd) | rv.append(cmd) | ||||
| self.network_test(node, addr, network=NET_ONION) | self.network_test(node, addr, network=NET_ONION) | ||||
| addr = "node.noumenon:8333" | addr = "node.noumenon:8333" | ||||
| self.log.debug( | self.log.debug( | ||||
| f"Test: outgoing DNS name connection through node for address {addr}") | f"Test: outgoing DNS name connection through node for address {addr}" | ||||
| ) | |||||
| node.addnode(addr, "onetry") | node.addnode(addr, "onetry") | ||||
| cmd = proxies[3].queue.get() | cmd = proxies[3].queue.get() | ||||
| assert isinstance(cmd, Socks5Command) | assert isinstance(cmd, Socks5Command) | ||||
| assert_equal(cmd.atyp, AddressType.DOMAINNAME) | assert_equal(cmd.atyp, AddressType.DOMAINNAME) | ||||
| assert_equal(cmd.addr, b"node.noumenon") | assert_equal(cmd.addr, b"node.noumenon") | ||||
| assert_equal(cmd.port, 8333) | assert_equal(cmd.port, 8333) | ||||
| if not auth: | if not auth: | ||||
| assert_equal(cmd.username, None) | assert_equal(cmd.username, None) | ||||
| assert_equal(cmd.password, None) | assert_equal(cmd.password, None) | ||||
| rv.append(cmd) | rv.append(cmd) | ||||
| self.network_test(node, addr, network=NET_UNROUTABLE) | self.network_test(node, addr, network=NET_UNROUTABLE) | ||||
| return rv | return rv | ||||
| def run_test(self): | def run_test(self): | ||||
| # basic -proxy | # basic -proxy | ||||
| self.node_test( | self.node_test( | ||||
| self.nodes[0], [self.serv1, self.serv1, self.serv1, self.serv1], False) | self.nodes[0], [self.serv1, self.serv1, self.serv1, self.serv1], False | ||||
| ) | |||||
| # -proxy plus -onion | # -proxy plus -onion | ||||
| self.node_test( | self.node_test( | ||||
| self.nodes[1], [self.serv1, self.serv1, self.serv2, self.serv1], False) | self.nodes[1], [self.serv1, self.serv1, self.serv2, self.serv1], False | ||||
| ) | |||||
| # -proxy plus -onion, -proxyrandomize | # -proxy plus -onion, -proxyrandomize | ||||
| rv = self.node_test( | rv = self.node_test( | ||||
| self.nodes[2], [self.serv2, self.serv2, self.serv2, self.serv2], True) | self.nodes[2], [self.serv2, self.serv2, self.serv2, self.serv2], True | ||||
| ) | |||||
| # Check that credentials as used for -proxyrandomize connections are | # Check that credentials as used for -proxyrandomize connections are | ||||
| # unique | # unique | ||||
| credentials = {(x.username, x.password) for x in rv} | credentials = {(x.username, x.password) for x in rv} | ||||
| assert_equal(len(credentials), len(rv)) | assert_equal(len(credentials), len(rv)) | ||||
| if self.have_ipv6: | if self.have_ipv6: | ||||
| # proxy on IPv6 localhost | # proxy on IPv6 localhost | ||||
| self.node_test( | self.node_test( | ||||
| self.nodes[3], [self.serv3, self.serv3, self.serv3, self.serv3], False, False) | self.nodes[3], | ||||
| [self.serv3, self.serv3, self.serv3, self.serv3], | |||||
| False, | |||||
| False, | |||||
| ) | |||||
| def networks_dict(d): | def networks_dict(d): | ||||
| r = {} | r = {} | ||||
| for x in d['networks']: | for x in d["networks"]: | ||||
| r[x['name']] = x | r[x["name"]] = x | ||||
| return r | return r | ||||
| self.log.info("Test RPC getnetworkinfo") | self.log.info("Test RPC getnetworkinfo") | ||||
| n0 = networks_dict(self.nodes[0].getnetworkinfo()) | n0 = networks_dict(self.nodes[0].getnetworkinfo()) | ||||
| assert_equal(NETWORKS, n0.keys()) | assert_equal(NETWORKS, n0.keys()) | ||||
| ip1, port1 = self.conf1.addr | ip1, port1 = self.conf1.addr | ||||
| ip2, port2 = self.conf2.addr | ip2, port2 = self.conf2.addr | ||||
| for net in NETWORKS: | for net in NETWORKS: | ||||
| if net == NET_I2P: | if net == NET_I2P: | ||||
| expected_proxy = '' | expected_proxy = "" | ||||
| expected_randomize = False | expected_randomize = False | ||||
| else: | else: | ||||
| expected_proxy = f'{ip1}:{port1}' | expected_proxy = f"{ip1}:{port1}" | ||||
| expected_randomize = True | expected_randomize = True | ||||
| assert_equal(n0[net]['proxy'], expected_proxy) | assert_equal(n0[net]["proxy"], expected_proxy) | ||||
| assert_equal( | assert_equal(n0[net]["proxy_randomize_credentials"], expected_randomize) | ||||
| n0[net]['proxy_randomize_credentials'], | assert_equal(n0["onion"]["reachable"], True) | ||||
| expected_randomize) | assert_equal(n0["i2p"]["reachable"], False) | ||||
| assert_equal(n0['onion']['reachable'], True) | |||||
| assert_equal(n0['i2p']['reachable'], False) | |||||
| n1 = networks_dict(self.nodes[1].getnetworkinfo()) | n1 = networks_dict(self.nodes[1].getnetworkinfo()) | ||||
| assert_equal(NETWORKS, n1.keys()) | assert_equal(NETWORKS, n1.keys()) | ||||
| for net in ['ipv4', 'ipv6']: | for net in ["ipv4", "ipv6"]: | ||||
| assert_equal(n1[net]['proxy'], f'{ip1}:{port1}') | assert_equal(n1[net]["proxy"], f"{ip1}:{port1}") | ||||
| assert_equal(n1[net]['proxy_randomize_credentials'], False) | assert_equal(n1[net]["proxy_randomize_credentials"], False) | ||||
| assert_equal(n1['onion']['proxy'], f'{ip2}:{port2}') | assert_equal(n1["onion"]["proxy"], f"{ip2}:{port2}") | ||||
| assert_equal(n1['onion']['proxy_randomize_credentials'], False) | assert_equal(n1["onion"]["proxy_randomize_credentials"], False) | ||||
| assert_equal(n1['onion']['reachable'], True) | assert_equal(n1["onion"]["reachable"], True) | ||||
| assert_equal(n1['i2p']['proxy'], f'{self.i2p_sam[0]}:{self.i2p_sam[1]}') | assert_equal(n1["i2p"]["proxy"], f"{self.i2p_sam[0]}:{self.i2p_sam[1]}") | ||||
| assert_equal(n1['i2p']['proxy_randomize_credentials'], False) | assert_equal(n1["i2p"]["proxy_randomize_credentials"], False) | ||||
| assert_equal(n1['i2p']['reachable'], True) | assert_equal(n1["i2p"]["reachable"], True) | ||||
| n2 = networks_dict(self.nodes[2].getnetworkinfo()) | n2 = networks_dict(self.nodes[2].getnetworkinfo()) | ||||
| assert_equal(NETWORKS, n2.keys()) | assert_equal(NETWORKS, n2.keys()) | ||||
| for net in NETWORKS: | for net in NETWORKS: | ||||
| if net == NET_I2P: | if net == NET_I2P: | ||||
| expected_proxy = '' | expected_proxy = "" | ||||
| expected_randomize = False | expected_randomize = False | ||||
| else: | else: | ||||
| expected_proxy = f'{ip2}:{port2}' | expected_proxy = f"{ip2}:{port2}" | ||||
| expected_randomize = True | expected_randomize = True | ||||
| assert_equal(n2[net]['proxy'], expected_proxy) | assert_equal(n2[net]["proxy"], expected_proxy) | ||||
| assert_equal( | assert_equal(n2[net]["proxy_randomize_credentials"], expected_randomize) | ||||
| n2[net]['proxy_randomize_credentials'], | assert_equal(n2["onion"]["reachable"], True) | ||||
| expected_randomize) | assert_equal(n2["i2p"]["reachable"], False) | ||||
| assert_equal(n2['onion']['reachable'], True) | |||||
| assert_equal(n2['i2p']['reachable'], False) | |||||
| if self.have_ipv6: | if self.have_ipv6: | ||||
| n3 = networks_dict(self.nodes[3].getnetworkinfo()) | n3 = networks_dict(self.nodes[3].getnetworkinfo()) | ||||
| assert_equal(NETWORKS, n3.keys()) | assert_equal(NETWORKS, n3.keys()) | ||||
| for net in NETWORKS: | for net in NETWORKS: | ||||
| if net == NET_I2P: | if net == NET_I2P: | ||||
| expected_proxy = '' | expected_proxy = "" | ||||
| else: | else: | ||||
| expected_proxy = f'[{self.conf3.addr[0]}]:{self.conf3.addr[1]}' | expected_proxy = f"[{self.conf3.addr[0]}]:{self.conf3.addr[1]}" | ||||
| assert_equal(n3[net]['proxy'], expected_proxy) | assert_equal(n3[net]["proxy"], expected_proxy) | ||||
| assert_equal(n3[net]['proxy_randomize_credentials'], False) | assert_equal(n3[net]["proxy_randomize_credentials"], False) | ||||
| assert_equal(n3['onion']['reachable'], False) | assert_equal(n3["onion"]["reachable"], False) | ||||
| assert_equal(n3['i2p']['reachable'], False) | assert_equal(n3["i2p"]["reachable"], False) | ||||
| if __name__ == '__main__': | if __name__ == "__main__": | ||||
| ProxyTest().main() | ProxyTest().main() | ||||