Merge pull request #1766 from erkin/igmp

igmp-proxy: T4912: Rewrite show IGMP proxy commands in the new op-mode format

2 files changed, 99 insertions, 241 deletions

diff --git a/src/op_mode/igmp-proxy.py b/src/op_mode/igmp-proxy.py
new file mode 100755
index 000000000..0086c9aa6
--- /dev/null
+++ b/src/op_mode/igmp-proxy.py
@@ -0,0 +1,99 @@
+#!/usr/bin/env python3
+#
+# Copyright (C) 2023 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/>.
+
+# File: show_igmpproxy.py
+# Purpose:
+#    Display istatistics from IPv4 IGMP proxy.
+#    Used by the "run show ip multicast" command tree.
+
+import ipaddress
+import json
+import socket
+import sys
+import tabulate
+
+import vyos.config
+import vyos.opmode
+
+from vyos.util import bytes_to_human, print_error
+
+def _is_configured():
+    """Check if IGMP proxy is configured"""
+    return vyos.config.Config().exists_effective('protocols igmp-proxy')
+
+def _is_running():
+    """Check if IGMP proxy is currently running"""
+    return not vyos.util.run('ps -C igmpproxy')
+
+def _kernel_to_ip(addr):
+    """
+    Convert any given address from Linux kernel to a proper, IPv4 address
+    using the correct host byte order.
+    """
+    # Convert from hex 'FE000A0A' to decimal '4261415434'
+    addr = int(addr, 16)
+    # Kernel ABI _always_ uses network byte order.
+    addr = socket.ntohl(addr)
+    return str(ipaddress.IPv4Address(addr))
+
+def _process_mr_vif():
+    """Read rows from /proc/net/ip_mr_vif into dicts."""
+    result = []
+    with open('/proc/net/ip_mr_vif', 'r') as f:
+        next(f)
+        for line in f:
+            result.append({
+                'Interface': line.split()[1],
+                'PktsIn'   : int(line.split()[3]),
+                'PktsOut'  : int(line.split()[5]),
+                'BytesIn'  : int(line.split()[2]),
+                'BytesOut' : int(line.split()[4]),
+                'Local'    : _kernel_to_ip(line.split()[7]),
+            })
+    return result
+
+def show_interface(raw: bool):
+    if data := _process_mr_vif():
+        if raw:
+            # Make the interface name the key for each row.
+            table = {}
+            for v in data:
+                table[v.pop('Interface')] = v
+            return json.loads(json.dumps(table))
+        # Make byte values human-readable for the table.
+        arr = []
+        for x in data:
+            arr.append({k: bytes_to_human(v) if k.startswith('Bytes') \
+                        else v for k, v in x.items()})
+        return tabulate.tabulate(arr, headers='keys')
+
+
+if not _is_configured():
+    print_error('IGMP proxy is not configured.')
+    sys.exit(0)
+if not _is_running():
+    print_error('IGMP proxy is not running.')
+    sys.exit(0)
+
+
+if __name__ == "__main__":
+    try:
+        res = vyos.opmode.run(sys.modules[__name__])
+        if res:
+            print(res)
+    except (ValueError, vyos.opmode.Error) as e:
+        print_error(e)
+        sys.exit(1)
diff --git a/src/op_mode/show_igmpproxy.py b/src/op_mode/show_igmpproxy.py
deleted file mode 100755
index 4714e494b..000000000
--- a/src/op_mode/show_igmpproxy.py
+++ /dev/null
@@ -1,241 +0,0 @@
-#!/usr/bin/env python3
-#
-# Copyright (C) 2018 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/>.
-
-# File: show_igmpproxy.py
-# Purpose:
-#    Display istatistics from IPv4 IGMP proxy.
-#    Used by the "run show ip multicast" command tree.
-
-import sys
-import jinja2
-import argparse
-import ipaddress
-import socket
-
-import vyos.config
-
-# Output Template for "show ip multicast interface" command
-#
-# Example:
-# Interface  BytesIn      PktsIn       BytesOut     PktsOut      Local
-# eth0       0.0b         0            0.0b         0            xxx.xxx.xxx.65
-# eth1       0.0b         0            0.0b         0            xxx.xxx.xx.201
-# eth0.3     0.0b         0            0.0b         0            xxx.xxx.x.7
-# tun1       0.0b         0            0.0b         0            xxx.xxx.xxx.2
-vif_out_tmpl = """
-{% for r in data %}
-{{ "%-10s"|format(r.interface) }} {{ "%-12s"|format(r.bytes_in) }} {{ "%-12s"|format(r.pkts_in) }} {{ "%-12s"|format(r.bytes_out) }} {{ "%-12s"|format(r.pkts_out) }} {{ "%-15s"|format(r.loc) }}
-{% endfor %}
-"""
-
-# Output Template for "show ip multicast mfc" command
-#
-# Example:
-# Group             Origin            In    Out           Pkts        Bytes        Wrong
-# xxx.xxx.xxx.250   xxx.xx.xxx.75     --
-# xxx.xxx.xx.124    xx.xxx.xxx.26     --
-mfc_out_tmpl = """
-{% for r in data %}
-{{ "%-15s"|format(r.group) }} {{ "%-15s"|format(r.origin) }} {{ "%-12s"|format(r.pkts) }} {{ "%-12s"|format(r.bytes) }} {{ "%-12s"|format(r.wrong) }} {{ "%-10s"|format(r.iif) }} {{ "%-20s"|format(r.oifs|join(', ')) }}
-{% endfor %}
-"""
-
-parser = argparse.ArgumentParser()
-parser.add_argument("--interface", action="store_true", help="Interface Statistics")
-parser.add_argument("--mfc", action="store_true", help="Multicast Forwarding Cache")
-
-def byte_string(size):
-    # convert size to integer
-    size = int(size)
-
-    # One Terrabyte
-    s_TB = 1024 * 1024 * 1024 * 1024
-    # One Gigabyte
-    s_GB = 1024 * 1024 * 1024
-    # One Megabyte
-    s_MB = 1024 * 1024
-    # One Kilobyte
-    s_KB = 1024
-    # One Byte
-    s_B  = 1
-
-    if size > s_TB:
-        return str(round((size/s_TB), 2)) + 'TB'
-    elif size > s_GB:
-        return str(round((size/s_GB), 2)) + 'GB'
-    elif size > s_MB:
-        return str(round((size/s_MB), 2)) + 'MB'
-    elif size > s_KB:
-        return str(round((size/s_KB), 2)) + 'KB'
-    else:
-        return str(round((size/s_B), 2)) + 'b'
-
-    return None
-
-def kernel2ip(addr):
-    """
-    Convert any given addr from Linux Kernel to a proper, IPv4 address
-    using the correct host byte order.
-    """
-
-    # Convert from hex 'FE000A0A' to decimal '4261415434'
-    addr = int(addr, 16)
-    # Kernel ABI _always_ uses network byteorder
-    addr = socket.ntohl(addr)
-
-    return ipaddress.IPv4Address( addr )
-
-def do_mr_vif():
-    """
-    Read contents of file /proc/net/ip_mr_vif and print a more human
-    friendly version to the command line. IPv4 addresses present as
-    32bit integers in hex format are converted to IPv4 notation, too.
-    """
-
-    with open('/proc/net/ip_mr_vif', 'r') as f:
-        lines = len(f.readlines())
-        if lines < 2:
-            return None
-
-    result = {
-        'data': []
-    }
-
-    # Build up table format string
-    table_format = {
-        'interface': 'Interface',
-        'pkts_in'  : 'PktsIn',
-        'pkts_out' : 'PktsOut',
-        'bytes_in' : 'BytesIn',
-        'bytes_out': 'BytesOut',
-        'loc'      : 'Local'
-    }
-    result['data'].append(table_format)
-
-    # read and parse information from /proc filesystema
-    with open('/proc/net/ip_mr_vif', 'r') as f:
-        header_line = next(f)
-        for line in f:
-            data = {
-                'interface': line.split()[1],
-                'pkts_in'  : line.split()[3],
-                'pkts_out' : line.split()[5],
-
-                # convert raw byte number to something more human readable
-                # Note: could be replaced by Python3 hurry.filesize module
-                'bytes_in' : byte_string( line.split()[2] ),
-                'bytes_out': byte_string( line.split()[4] ),
-
-                # convert IP address from hex 'FE000A0A' to decimal '4261415434'
-                'loc'      : kernel2ip( line.split()[7] ),
-            }
-            result['data'].append(data)
-
-    return result
-
-def do_mr_mfc():
-    """
-    Read contents of file /proc/net/ip_mr_cache and print a more human
-    friendly version to the command line. IPv4 addresses present as
-    32bit integers in hex format are converted to IPv4 notation, too.
-    """
-
-    with open('/proc/net/ip_mr_cache', 'r') as f:
-        lines = len(f.readlines())
-        if lines < 2:
-            return None
-
-    # We need this to convert from interface index to a real interface name
-    # Thus we also skip the format identifier on list index 0
-    vif = do_mr_vif()['data'][1:]
-
-    result = {
-        'data': []
-    }
-
-    # Build up table format string
-    table_format = {
-        'group' : 'Group',
-        'origin': 'Origin',
-        'iif'   : 'In',
-        'oifs'  : ['Out'],
-        'pkts'  : 'Pkts',
-        'bytes' : 'Bytes',
-        'wrong' : 'Wrong'
-    }
-    result['data'].append(table_format)
-
-    # read and parse information from /proc filesystem
-    with open('/proc/net/ip_mr_cache', 'r') as f:
-        header_line = next(f)
-        for line in f:
-            data = {
-                # convert IP address from hex 'FE000A0A' to decimal '4261415434'
-                'group' : kernel2ip( line.split()[0] ),
-                'origin': kernel2ip( line.split()[1] ),
-
-                'iif'   : '--',
-                'pkts'  : '',
-                'bytes' : '',
-                'wrong' : '',
-                'oifs'  : []
-            }
-
-            iif = int( line.split()[2] )
-            if not ((iif == -1) or (iif == 65535)):
-                data['pkts']  = line.split()[3]
-                data['bytes'] = byte_string( line.split()[4] )
-                data['wrong'] = line.split()[5]
-
-                # convert index to real interface name
-                data['iif']  = vif[iif]['interface']
-
-                # convert each output interface index to a real interface name
-                for oif in line.split()[6:]:
-                    idx = int( oif.split(':')[0] )
-                    data['oifs'].append( vif[idx]['interface'] )
-
-            result['data'].append(data)
-
-    return result
-
-if __name__ == '__main__':
-    args = parser.parse_args()
-
-    # Do nothing if service is not configured
-    c = vyos.config.Config()
-    if not c.exists_effective('protocols igmp-proxy'):
-        print("IGMP proxy is not configured")
-        sys.exit(0)
-
-    if args.interface:
-        data = do_mr_vif()
-        if data:
-            tmpl = jinja2.Template(vif_out_tmpl)
-            print(tmpl.render(data))
-
-        sys.exit(0)
-    elif args.mfc:
-        data = do_mr_mfc()
-        if data:
-            tmpl = jinja2.Template(mfc_out_tmpl)
-            print(tmpl.render(data))
-
-        sys.exit(0)
-    else:
-        parser.print_help()
-        sys.exit(1)
-