1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
|
## This Source Code Form is subject to the terms of the Mozilla Public
## License, v. 2.0. If a copy of the MPL was not distributed with this
## file, You can obtain one at https://mozilla.org/MPL/2.0/.
##
## Copyright (c) 2007-2023 Broadcom. All Rights Reserved. The term “Broadcom” refers to Broadcom Inc. and/or its subsidiaries. All rights reserved.
defmodule RabbitMQ.CLI.Diagnostics.Helpers do
def test_connection(hostname_or_ip, port, timeout) do
hostname_as_list = :rabbit_data_coercion.to_list(hostname_or_ip)
case :gen_tcp.connect(hostname_as_list, port, [], timeout) do
{:error, _} -> :gen_tcp.connect(hostname_as_list, port, [:inet6], timeout)
r -> r
end
end
def check_port_connectivity(port, node_name, timeout) do
check_port_connectivity(port, node_name, nil, timeout)
end
def check_port_connectivity(port, node_name, nil, timeout) do
regex = Regex.recompile!(~r/^(.+)@/)
hostname = Regex.replace(regex, to_string(node_name), "") |> to_charlist
check_port_connectivity(port, node_name, hostname, timeout)
end
def check_port_connectivity(port, _node_name, hostname_or_ip, timeout) do
try do
IO.puts("Will connect to #{hostname_or_ip}:#{port}")
case test_connection(hostname_or_ip, port, timeout) do
{:error, err} ->
IO.puts("Error connecting to #{hostname_or_ip}:#{port}: #{err}")
false
{:ok, port} ->
:ok = :gen_tcp.close(port)
true
end
# `gen_tcp:connect/4` will throw if the port is outside of its
# expected domain
catch
:exit, _ -> false
end
end
def check_listener_connectivity(%{port: port}, node_name, target_ip, timeout) do
check_port_connectivity(port, node_name, target_ip, timeout)
end
end
|
