aboutsummaryrefslogtreecommitdiff
path: root/shard/lib/net/addr.ex
blob: b92ae7083088f413059e1b453b64e133ed156dfe (plain) (blame)
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
53
54
55
56
57
58
59
60
61
62
63
64
65
defmodule SNet.Addr do
  @moduledoc"""
  Helper module for getting our IP addresses.

  Runs an agent that gets our public IPv4 address on the internet and stores it.
  """

  use Agent

  require Logger

  def start_link(_) do
    Agent.start_link(__MODULE__, :init, [], name: __MODULE__)
  end

  def init() do
    Application.ensure_all_started(:inets)
    Application.ensure_all_started(:ssl)
    case :httpc.request('http://api.ipify.org') do
      {:ok, {_, _, body}} ->
        {:ok, addr} = :inet.parse_address body
        Logger.info "Public IP address: #{body}"
        [addr]
      _ ->
        Logger.info "Could not get public IP address"
        []
    end
  end

  @doc"""
  Reteurn the list of IPv4 address for our network interfaces.
  """
  def get_if_inet4 do
    {:ok, ifs} = :inet.getifaddrs
    for {_, opts} <- ifs,
        {:addr, addr} <- opts,
        tuple_size(addr) == 4,
        addr != {127,0,0,1}
    do
      addr
    end
  end

  @doc"""
  Return our public IPv4 address as observed by an external API provider (`ipify.org`)
  """
  def get_pub_inet4 do
    Agent.get(__MODULE__, &(&1))
  end

  @doc"""
  Get all our IPv4 addresses.
  """
  def get_all_inet4 do
    addrset = for x <- get_if_inet4() ++ get_pub_inet4(), into: %MapSet{}, do: x
    MapSet.to_list addrset
  end

  @doc"""
  Determines if an IP address is ours or not.
  """
  def is_local?({:inet, ip, port}) do
    port == Application.get_env(:shard, :port) and (ip == {127,0,0,1} or ip in get_all_inet4())
  end
end