summaryrefslogtreecommitdiff
path: root/src/op_mode/show_neigh.py
blob: 4d96ee86954c52096b20f1f7542bfa6d7634d6ea (plain)
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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
#!/usr/bin/env python3
#
# Copyright (C) 2022 VyOS maintainers and contributors
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License version 2 or later as
# published by the Free Software Foundation.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

# Sample output of `ip --json neigh list`:
#
# [
#   {
#     "dst": "192.168.1.1",
#     "dev": "eth0",                 # Missing if `dev ...` option is used
#     "lladdr": "00:aa:bb:cc:dd:ee", # May be missing for failed entries
#     "state": [
#       "REACHABLE"
#     ]
#  },
# ]

import sys


def get_raw_data(family, device=None, state=None):
    from json import loads
    from vyos.util import cmd

    if device:
        device = "dev {0}".format(device)
    else:
        device = ""

    if state:
        state = "nud {0}".format(state)
    else:
        state = ""

    neigh_cmd = "ip --family {0} --json neighbor list {1} {2}".format(family, device, state)

    data = loads(cmd(neigh_cmd))

    return data

def get_formatted_output(family, device=None, state=None):
    from tabulate import tabulate

    def entry_to_list(e, intf=None):
        dst = e["dst"]

        # State is always a list in the iproute2 output
        state = ", ".join(e["state"])

        # Link layer address is absent from e.g. FAILED entries
        if "lladdr" in e:
            lladdr = e["lladdr"]
        else:
            lladdr = None

        # Device field is absent from outputs of `ip neigh list dev ...`
        if "dev" in e:
            dev = e["dev"]
        elif device:
            dev = device
        else:
            raise ValueError("interface is not defined")

        return [dst, dev, lladdr, state]

    neighs = get_raw_data(family, device=device, state=state)
    neighs = map(entry_to_list, neighs)

    headers = ["Address", "Interface", "Link layer address",  "State"]
    return tabulate(neighs, headers)

if __name__ == '__main__':
    from argparse import ArgumentParser

    parser = ArgumentParser()
    parser.add_argument("-f", "--family", type=str, default="inet", help="Address family")
    parser.add_argument("-i", "--interface", type=str, help="Network interface")
    parser.add_argument("-s", "--state", type=str, help="Neighbor table entry state")

    args = parser.parse_args()

    if args.state:
        if args.state not in ["reachable", "failed", "stale", "permanent"]:
            raise ValueError("Incorrect state {0}! Must be one of: reachable, stale, failed, permanent".format(args.state))

    try:
        print(get_formatted_output(args.family, device=args.interface, state=args.state))
    except ValueError as e:
        print(e)
        sys.exit(1)