summaryrefslogtreecommitdiff
path: root/src/op_mode/igmp-proxy.py
diff options
context:
space:
mode:
Diffstat (limited to 'src/op_mode/igmp-proxy.py')
-rwxr-xr-xsrc/op_mode/igmp-proxy.py167
1 files changed, 24 insertions, 143 deletions
diff --git a/src/op_mode/igmp-proxy.py b/src/op_mode/igmp-proxy.py
index 2a8f284bc..0086c9aa6 100755
--- a/src/op_mode/igmp-proxy.py
+++ b/src/op_mode/igmp-proxy.py
@@ -21,9 +21,9 @@
import ipaddress
import json
-import jinja2
import socket
import sys
+import tabulate
import vyos.config
import vyos.opmode
@@ -43,161 +43,42 @@ 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
+ # Kernel ABI _always_ uses network byte order.
addr = socket.ntohl(addr)
-
return str(ipaddress.IPv4Address(addr))
def _process_mr_vif():
- """
- Read contents of file /proc/net/ip_mr_vif and print a more human
- friendly version to the command line. IPv4 addresses presented as
- 32-bit 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
+ """Read rows from /proc/net/ip_mr_vif into dicts."""
+ result = []
with open('/proc/net/ip_mr_vif', 'r') as f:
- header_line = next(f)
+ 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
- 'bytes_in' : bytes_to_human(int(line.split()[2])),
- 'bytes_out': bytes_to_human(int(line.split()[4])),
-
- # convert IP address from hex 'FE000A0A' to decimal '4261415434'
- 'loc' : _kernel_to_ip(line.split()[7]),
- }
- result['data'].append(data)
-
+ 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 _process_mr_mfc():
- """
- Read contents of file /proc/net/ip_mr_cache and print a more human
- friendly version to the command line. IPv4 addresses presented as
- 32-bit 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' : _kernel_to_ip(line.split()[0]),
- 'origin': _kernel_to_ip(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'] = bytes_to_human(int(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
-
-
-# Output template for "show ip multicast interface" command
-#
-# Example:
-# Interface BytesIn PktsIn BytesOut PktsOut Local
-# eth0 0.0 B 0 0.0 B 0 xxx.xxx.xxx.65
-# eth1 0.0 B 0 0.0 B 0 xxx.xxx.xx.201
-# eth0.3 0.0 B 0 0.0 B 0 xxx.xxx.x.7
-# tun1 0.0 B 0 0.0 B 0 xxx.xxx.xxx.2
def show_interface(raw: bool):
- vif_out_template = """{%- 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 %}"""
if data := _process_mr_vif():
if raw:
- return json.loads(json.dumps(data))
- else:
- return jinja2.Template(vif_out_template).render(data)
-
-# 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 --
-def show_mfc(raw: bool):
- mfc_out_template = """{%- 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 %}"""
- if data := _process_mr_mfc():
- if raw:
- return json.loads(json.dumps(data))
- else:
- return jinja2.Template(mfc_out_template).render(data)
+ # 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():