mirror of
https://github.com/bashclub/check-unifi-controller.git
synced 2024-11-07 12:31:58 +01:00
764 lines
26 KiB
Python
764 lines
26 KiB
Python
#!/usr/bin/env python3
|
|
# -*- encoding: utf-8; py-indent-offset: 4 -*-
|
|
#
|
|
## MIT License
|
|
##
|
|
## Copyright (c) 2024 Bash Club
|
|
##
|
|
## Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
## of this software and associated documentation files (the "Software"), to deal
|
|
## in the Software without restriction, including without limitation the rights
|
|
## to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
## copies of the Software, and to permit persons to whom the Software is
|
|
## furnished to do so, subject to the following conditions:
|
|
##
|
|
## The above copyright notice and this permission notice shall be included in all
|
|
## copies or substantial portions of the Software.
|
|
##
|
|
## THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
## IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
## FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
## AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
## LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
## OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
|
## SOFTWARE.
|
|
|
|
from cmk.gui.i18n import _
|
|
|
|
from .agent_based_api.v1 import (
|
|
Metric,
|
|
register,
|
|
render,
|
|
Result,
|
|
IgnoreResults,
|
|
Service,
|
|
State,
|
|
TableRow,
|
|
Attributes
|
|
)
|
|
|
|
from .agent_based_api.v1.type_defs import CheckResult, DiscoveryResult
|
|
from typing import Any, Dict, Mapping, Sequence, Optional
|
|
|
|
from dataclasses import dataclass
|
|
from collections import defaultdict
|
|
from .utils import interfaces
|
|
|
|
SubSection = Dict[str,str]
|
|
Section = Dict[str, SubSection]
|
|
class dictobject(defaultdict):
|
|
def __getattr__(self,name):
|
|
return self[name] if name in self else ""
|
|
|
|
nested_dictobject = lambda: dictobject(nested_dictobject)
|
|
|
|
def _expect_bool(val,expected=True,failstate=State.WARN):
|
|
return State.OK if bool(int(val)) == expected else failstate
|
|
|
|
def _expect_number(val,expected=0,failstate=State.WARN):
|
|
return State.OK if int(val) == expected else failstate
|
|
|
|
def _safe_float(val):
|
|
try:
|
|
return float(val)
|
|
except (TypeError,ValueError):
|
|
return 0
|
|
|
|
def _safe_int(val,default=0):
|
|
try:
|
|
return int(val)
|
|
except (TypeError,ValueError):
|
|
return default
|
|
|
|
def _unifi_status2state(status):
|
|
return {
|
|
"ok" : State.OK,
|
|
"warning" : State.WARN,
|
|
"error" : State.CRIT
|
|
}.get(status.lower(),State.UNKNOWN)
|
|
|
|
from pprint import pprint
|
|
|
|
def parse_unifi_dict(string_table):
|
|
_ret = dictobject()
|
|
for _line in string_table:
|
|
_ret[_line[0]] = _line[1]
|
|
return _ret
|
|
|
|
def parse_unifi_nested_dict(string_table):
|
|
_ret = nested_dictobject()
|
|
for _line in string_table:
|
|
_ret[_line[0]][_line[1]] = _line[2]
|
|
return _ret
|
|
|
|
UNIFI_DEVICE_STATES = {
|
|
"0" : "disconnected",
|
|
"1" : "connected",
|
|
"2" : "pending"
|
|
}
|
|
|
|
|
|
############ Controller ############
|
|
def discovery_unifi_controller(section):
|
|
yield Service(item="Unifi Controller")
|
|
if section.cloudkey_version:
|
|
yield Service(item="Cloudkey")
|
|
|
|
def check_unifi_controller(item,section):
|
|
if item == "Unifi Controller":
|
|
yield Result(
|
|
state=State.OK,
|
|
summary=f"Version: {section.controller_version}"
|
|
)
|
|
if int(section.update_available) > 0:
|
|
yield Result(
|
|
state=State.WARN,
|
|
notice=_("Update available")
|
|
)
|
|
if item == "Cloudkey":
|
|
yield Result(
|
|
state=State.OK,
|
|
summary=f"Version: {section.cloudkey_version}"
|
|
)
|
|
if _safe_int(section.cloudkey_update_available) > 0:
|
|
yield Result(
|
|
state=State.WARN,
|
|
notice=_("Update available")
|
|
)
|
|
|
|
def inventory_unifi_controller(section):
|
|
yield Attributes(
|
|
path=["software","os"],
|
|
inventory_attributes={
|
|
"controller_version" : section.get("controller_version")
|
|
}
|
|
)
|
|
|
|
register.agent_section(
|
|
name = 'unifi_controller',
|
|
parse_function = parse_unifi_dict
|
|
)
|
|
|
|
register.check_plugin(
|
|
name='unifi_controller',
|
|
service_name='%s',
|
|
discovery_function=discovery_unifi_controller,
|
|
check_function=check_unifi_controller,
|
|
)
|
|
|
|
register.inventory_plugin(
|
|
name = "unifi_controller",
|
|
inventory_function = inventory_unifi_controller
|
|
)
|
|
|
|
############ SITES ###########
|
|
|
|
def discovery_unifi_sites(section):
|
|
for _item in section.values():
|
|
yield Service(item=f"{_item.desc}")
|
|
|
|
def check_unifi_sites(item,params,section):
|
|
site = next(filter(lambda x: x.desc == item,section.values()))
|
|
yield Metric("satisfaction",max(0,_safe_int(site.satisfaction)))
|
|
|
|
if site.lan_status != "unknown":
|
|
yield Metric("lan_user_sta",_safe_int(site.lan_num_user))
|
|
yield Metric("lan_guest_sta",_safe_int(site.lan_num_guest))
|
|
yield Metric("if_in_octets",_safe_int(site.lan_rx_bytes_r))
|
|
#yield Metric("if_in_bps",_safe_int(site.lan_rx_bytes_r)*8)
|
|
yield Metric("if_out_octets",_safe_int(site.lan_tx_bytes_r))
|
|
#yield Metric("if_out_bps",_safe_int(site.lan_tx_bytes_r)*8)
|
|
yield Metric("lan_active_sw",_safe_int(site.lan_num_sw))
|
|
yield Metric("lan_total_sw",_safe_int(site.lan_num_adopted))
|
|
yield Result(
|
|
state=_unifi_status2state(site.lan_status),
|
|
summary=f"LAN: {site.lan_num_sw}/{site.lan_num_adopted} Switch ({site.lan_status})"
|
|
)
|
|
#yield Result(
|
|
# state=_expect_number(site.lan_num_disconnected),
|
|
# notice=f"{site.lan_num_disconnected} Switch disconnected" ##disconnected kann
|
|
#)
|
|
|
|
if site.wlan_status != "unknown":
|
|
yield Metric("wlan_user_sta",_safe_int(site.wlan_num_user))
|
|
yield Metric("wlan_guest_sta",_safe_int(site.wlan_num_guest))
|
|
yield Metric("wlan_iot_sta",_safe_int(site.wlan_num_iot))
|
|
yield Metric("wlan_if_in_octets",_safe_int(site.wlan_rx_bytes_r))
|
|
yield Metric("wlan_if_out_octets",_safe_int(site.wlan_tx_bytes_r))
|
|
yield Metric("wlan_active_ap",_safe_int(site.wlan_num_ap))
|
|
yield Metric("wlan_total_ap",_safe_int(site.wlan_num_adopted))
|
|
yield Result(
|
|
state=_unifi_status2state(site.wlan_status),
|
|
summary=f"WLAN: {site.wlan_num_ap}/{site.wlan_num_adopted} AP ({site.wlan_status})"
|
|
)
|
|
#yield Result(
|
|
# state=_expect_number(site.wlan_num_disconnected),
|
|
# notice=f"{site.wlan_num_disconnected} AP disconnected"
|
|
#)
|
|
if site.wan_status != "unknown":
|
|
yield Result(
|
|
state=_unifi_status2state(site.wan_status),
|
|
summary=f"WAN Status: {site.wan_status}"
|
|
)
|
|
if site.www_status != "unknown":
|
|
yield Result(
|
|
state=_unifi_status2state(site.www_status),
|
|
summary=f"WWW Status: {site.www_status}"
|
|
)
|
|
if site.vpn_status != "unknown":
|
|
yield Result(
|
|
state=_unifi_status2state(site.vpn_status),
|
|
notice=f"WWW Status: {site.vpn_status}"
|
|
)
|
|
|
|
if params.get("ignore_alarms"):
|
|
_alarmstate = State.OK
|
|
else:
|
|
_alarmstate = _expect_number(site.num_new_alarms)
|
|
|
|
yield Result(
|
|
state=_alarmstate,
|
|
notice=f"{site.num_new_alarms} new Alarm"
|
|
)
|
|
|
|
|
|
register.agent_section(
|
|
name = 'unifi_sites',
|
|
parse_function = parse_unifi_nested_dict
|
|
)
|
|
|
|
register.check_plugin(
|
|
name='unifi_sites',
|
|
service_name='Site %s',
|
|
discovery_function=discovery_unifi_sites,
|
|
check_default_parameters={},
|
|
check_ruleset_name="unifi_sites",
|
|
check_function=check_unifi_sites,
|
|
)
|
|
|
|
############ DEVICE_SHORTLIST ##########
|
|
|
|
def inventory_unifi_device_shortlist(section):
|
|
for _name,_device in section.items():
|
|
yield TableRow(
|
|
path=["hardware","networkdevices"],
|
|
key_columns={"_name" : _name},
|
|
inventory_columns={
|
|
"serial" : _device.get("serial"),
|
|
"_state" : UNIFI_DEVICE_STATES.get(_device.state,"unknown"),
|
|
"vendor" : "ubiquiti",
|
|
"model" : _device.get("model_name",_device.get("model")),
|
|
"version" : _device.version,
|
|
"ip_address": _device.ip,
|
|
"mac" : _device.mac
|
|
}
|
|
)
|
|
|
|
register.agent_section(
|
|
name = 'unifi_device_shortlist',
|
|
parse_function = parse_unifi_nested_dict
|
|
)
|
|
|
|
register.inventory_plugin(
|
|
name = "unifi_device_shortlist",
|
|
inventory_function = inventory_unifi_device_shortlist
|
|
)
|
|
|
|
|
|
|
|
############ DEVICE ###########
|
|
def discovery_unifi_device(section):
|
|
yield Service(item="Device Status")
|
|
yield Service(item="Unifi Device")
|
|
yield Service(item="Active-User")
|
|
if section.type != "uap": # kein satisfaction bei ap .. radio/ssid haben schon
|
|
yield Service(item="Satisfaction")
|
|
if section.general_temperature:
|
|
yield Service(item="Temperature")
|
|
if section.uplink_device:
|
|
yield Service(item="Uplink")
|
|
if section.speedtest_status:
|
|
yield Service(item="Speedtest")
|
|
|
|
def check_unifi_device(item,section):
|
|
_device_state = UNIFI_DEVICE_STATES.get(section.state,"unknown")
|
|
## connected OK / pending Warn / Rest Crit
|
|
_hoststatus = State.OK if section.state == "1" else State.WARN if section.state == "2" else State.CRIT
|
|
if item == "Device Status":
|
|
yield Result(
|
|
state=_hoststatus,
|
|
summary=f"Status: {_device_state}"
|
|
)
|
|
#if section.state != "1":
|
|
# yield IgnoreResults(f"device not active State: {section.state}")
|
|
|
|
if item == "Unifi Device":
|
|
yield Result(
|
|
state=State.OK,
|
|
summary=f"Version: {section.version}"
|
|
)
|
|
if _safe_int(section.upgradable) > 0:
|
|
yield Result(
|
|
state=State.WARN,
|
|
notice=_("Update available")
|
|
)
|
|
if item == "Active-User":
|
|
_active_user = _safe_int(section.user_num_sta)
|
|
yield Result(
|
|
state=State.OK,
|
|
summary=f"{_active_user}"
|
|
)
|
|
if _safe_int(section.guest_num_sta) > -1:
|
|
yield Result(
|
|
state=State.OK,
|
|
summary=f"Guest: {section.guest_num_sta}"
|
|
)
|
|
yield Metric("user_sta",_active_user)
|
|
yield Metric("guest_sta",_safe_int(section.guest_num_sta))
|
|
if item == "Satisfaction":
|
|
yield Result(
|
|
state=State.OK,
|
|
summary=f"{section.satisfaction}%"
|
|
)
|
|
yield Metric("satisfaction",max(0,_safe_int(section.satisfaction)))
|
|
if item == "Temperature":
|
|
yield Metric("temp",_safe_float(section.general_temperature))
|
|
yield Result(
|
|
state=State.OK,
|
|
summary=f"{section.general_temperature} °C"
|
|
)
|
|
if section.fan_level:
|
|
yield Result(
|
|
state=State.OK,
|
|
summary=f"Fan: {section.fan_level}%"
|
|
)
|
|
if item == "Speedtest":
|
|
yield Result(
|
|
state=State.OK,
|
|
summary=f"Ping: {section.speedtest_ping} ms"
|
|
)
|
|
yield Result(
|
|
state=State.OK,
|
|
summary=f"Down: {section.speedtest_download} Mbit/s"
|
|
)
|
|
yield Result(
|
|
state=State.OK,
|
|
summary=f"Up: {section.speedtest_upload} Mbit/s"
|
|
)
|
|
_speedtest_time = render.datetime(_safe_float(section.speedtest_time))
|
|
yield Result(
|
|
state=State.OK,
|
|
summary=f"Last: {_speedtest_time}"
|
|
)
|
|
yield Metric("rtt",_safe_float(section.speedtest_ping))
|
|
yield Metric("if_in_bps",_safe_float(section.speedtest_download)*1024*1024) ## mbit to bit
|
|
yield Metric("if_out_bps",_safe_float(section.speedtest_upload)*1024*1024) ## mbit to bit
|
|
|
|
if item == "Uplink":
|
|
yield Result(
|
|
state=_expect_bool(section.uplink_up),
|
|
summary=f"Device {section.uplink_device} Port: {section.uplink_remote_port}"
|
|
)
|
|
|
|
def inventory_unifi_device(section):
|
|
yield Attributes(
|
|
path=["software","os"],
|
|
inventory_attributes={
|
|
"version" : section.get("version")
|
|
}
|
|
)
|
|
yield Attributes(
|
|
path=["software","configuration","snmp_info"],
|
|
inventory_attributes = {
|
|
"name" : section.get("name"),
|
|
"contact" : section.get("snmp_contact"),
|
|
"location" : section.get("snmp_location")
|
|
}
|
|
)
|
|
_hwdict = {
|
|
"vendor" : "ubiquiti",
|
|
}
|
|
for _key in ("model","board_rev","serial","mac"):
|
|
_val = section.get(_key)
|
|
if _val:
|
|
_hwdict[_key] = _val
|
|
yield Attributes(
|
|
path=["hardware","system"],
|
|
inventory_attributes= _hwdict
|
|
)
|
|
|
|
register.agent_section(
|
|
name = 'unifi_device',
|
|
parse_function = parse_unifi_dict
|
|
)
|
|
|
|
register.check_plugin(
|
|
name='unifi_device',
|
|
service_name='%s',
|
|
discovery_function=discovery_unifi_device,
|
|
check_function=check_unifi_device,
|
|
)
|
|
|
|
register.inventory_plugin(
|
|
name = "unifi_device",
|
|
inventory_function = inventory_unifi_device
|
|
)
|
|
|
|
############ DEVICEPORT ###########
|
|
@dataclass
|
|
class unifi_interface(interfaces.InterfaceWithCounters):
|
|
jumbo : bool = False
|
|
satisfaction : int = 0
|
|
poe_enable : bool = False
|
|
poe_mode : Optional[str] = None
|
|
poe_good : Optional[bool] = None
|
|
poe_current : Optional[float] = 0
|
|
poe_power : Optional[float] = 0
|
|
poe_voltage : Optional[float] = 0
|
|
poe_class : Optional[str] = None
|
|
dot1x_mode : Optional[str] = None
|
|
dot1x_status : Optional[str] = None
|
|
ip_address : Optional[str] = None
|
|
portconf : Optional[str] = None
|
|
|
|
|
|
def _convert_unifi_counters_if(section: Section) -> interfaces.Section:
|
|
## 10|port_idx|10
|
|
## 10|port_poe|1
|
|
## 10|poe_caps|7
|
|
## 10|op_mode|switch
|
|
## 10|poe_mode|auto
|
|
## 10|anomalies|0
|
|
## 10|autoneg|1
|
|
## 10|dot1x_mode|unknown
|
|
## 10|dot1x_status|disabled
|
|
## 10|enable|1
|
|
## 10|full_duplex|1
|
|
## 10|is_uplink|0
|
|
## 10|jumbo|1
|
|
## 10|poe_class|Unknown
|
|
## 10|poe_current|0.00
|
|
## 10|poe_enable|0
|
|
## 10|poe_good|0
|
|
## 10|poe_power|0.00
|
|
## 10|poe_voltage|0.00
|
|
## 10|rx_broadcast|1290
|
|
## 10|rx_bytes|38499976384
|
|
## 10|rx_dropped|0
|
|
## 10|rx_errors|0
|
|
## 10|rx_multicast|16423
|
|
## 10|rx_packets|125489263
|
|
## 10|satisfaction|100
|
|
## 10|satisfaction_reason|0
|
|
## 10|speed|1000
|
|
## 10|stp_pathcost|20000
|
|
## 10|stp_state|forwarding
|
|
## 10|tx_broadcast|20791854
|
|
## 10|tx_bytes|238190158091
|
|
## 10|tx_dropped|0
|
|
## 10|tx_errors|0
|
|
## 10|tx_multicast|262691
|
|
## 10|tx_packets|228482694
|
|
## 10|tx_bytes_r|17729
|
|
## 10|rx_bytes_r|176941
|
|
## 10|bytes_r|194671
|
|
## 10|name|Port 10
|
|
## 10|aggregated_by|0
|
|
## 10|oper_status|1
|
|
## 10|admin_status|1
|
|
## 10|portconf|ALL
|
|
## unifi_interface(index='10', descr='Port 10', alias='Port 10', type='6', speed=1000000000, oper_status='1',
|
|
## in_octets=38448560321, in_ucast=125404491, in_mcast=16414, in_bcast=1290, in_discards=0, in_errors=0,
|
|
## out_octets=238185160794, out_ucast=228451699, out_mcast=262551, out_bcast=20783341, out_discards=0, out_errors=0,
|
|
## out_qlen=0, phys_address='', oper_status_name='up', speed_as_text='', group=None, node=None, admin_status='1',
|
|
## total_octets=276633721115, jumbo=True, satisfaction=100,
|
|
## poe_enable=False, poe_mode='auto', poe_good=None, poe_current=0.0, poe_power=0.0, poe_voltage=0.0, poe_class='Unknown',
|
|
## dot1x_mode='unknown',dot1x_status='disabled', ip_address='', portconf='ALL')
|
|
|
|
return [
|
|
unifi_interface(
|
|
attributes=interfaces.Attributes(
|
|
index=str(netif.port_idx),
|
|
descr=netif.name,
|
|
alias=netif.name,
|
|
type='6',
|
|
speed=_safe_int(netif.speed)*1000000,
|
|
oper_status=netif.oper_status,
|
|
admin_status=netif.admin_status,
|
|
),
|
|
counters=interfaces.Counters(
|
|
in_octets=_safe_int(netif.rx_bytes),
|
|
in_ucast=_safe_int(netif.rx_packets),
|
|
in_mcast=_safe_int(netif.rx_multicast),
|
|
in_bcast=_safe_int(netif.rx_broadcast),
|
|
in_disc=_safe_int(netif.rx_dropped),
|
|
in_err=_safe_int(netif.rx_errors),
|
|
out_octets=_safe_int(netif.tx_bytes),
|
|
out_ucast=_safe_int(netif.tx_packets),
|
|
out_mcast=_safe_int(netif.tx_multicast),
|
|
out_bcast=_safe_int(netif.tx_broadcast),
|
|
out_disc=_safe_int(netif.tx_dropped),
|
|
out_err=_safe_int(netif.tx_errors),
|
|
),
|
|
jumbo=True if netif.jumbo == "1" else False,
|
|
satisfaction=_safe_int(netif.satisfaction) if netif.satisfaction and netif.oper_status == "1" else 0,
|
|
poe_enable=True if netif.poe_enable == "1" else False,
|
|
poe_mode=netif.poe_mode,
|
|
poe_current=float(netif.poe_current) if netif.poe_current else 0,
|
|
poe_voltage=float(netif.poe_voltage) if netif.poe_voltage else 0,
|
|
poe_power=float(netif.poe_power) if netif.poe_power else 0,
|
|
poe_class=netif.poe_class,
|
|
dot1x_mode=netif.dot1x_mode,
|
|
dot1x_status=netif.dot1x_status,
|
|
ip_address=netif.ip,
|
|
portconf=netif.portconf
|
|
) for netif in parse_unifi_nested_dict(section).values()
|
|
]
|
|
|
|
|
|
|
|
def discovery_unifi_network_port_if( ## fixme parsed_section_name
|
|
params: Sequence[Mapping[str, Any]],
|
|
section: Section,
|
|
) -> DiscoveryResult:
|
|
yield from interfaces.discover_interfaces(
|
|
params,
|
|
_convert_unifi_counters_if(section),
|
|
)
|
|
|
|
|
|
def check_unifi_network_port_if( ##fixme parsed_section_name
|
|
item: str,
|
|
params: Mapping[str, Any],
|
|
section: Section,
|
|
) -> CheckResult:
|
|
_converted_ifs = _convert_unifi_counters_if(section)
|
|
iface = next(filter(lambda x: _safe_int(item,-1) == _safe_int(x.attributes.index) or item == x.attributes.alias,_converted_ifs),None) ## fix Service Discovery appearance alias/descr
|
|
yield from interfaces.check_multiple_interfaces(
|
|
item,
|
|
params,
|
|
_converted_ifs,
|
|
)
|
|
if iface:
|
|
#pprint(iface)
|
|
if iface.portconf:
|
|
yield Result(
|
|
state=State.OK,
|
|
summary=f"Network: {iface.portconf}"
|
|
)
|
|
yield Metric("satisfaction",max(0,iface.satisfaction))
|
|
#pprint(iface)
|
|
if iface.poe_enable:
|
|
yield Result(
|
|
state=State.OK,
|
|
summary=f"PoE: {iface.poe_power}W"
|
|
)
|
|
#yield Metric("poe_current",iface.poe_current)
|
|
yield Metric("poe_power",iface.poe_power)
|
|
yield Metric("poe_voltage",iface.poe_voltage)
|
|
if iface.ip_address:
|
|
yield Result(
|
|
state=State.OK,
|
|
summary=f"IP: {iface.ip_address}"
|
|
)
|
|
|
|
def inventory_unifi_network_ports(section):
|
|
_total_ethernet_ports = 0
|
|
_available_ethernet_ports = 0
|
|
for _iface in parse_unifi_nested_dict(section).values():
|
|
_total_ethernet_ports +=1
|
|
_available_ethernet_ports +=1 if _iface.oper_status == '2' else 0
|
|
yield TableRow(
|
|
path=["networking","interfaces"],
|
|
key_columns={"index" : _safe_int(_iface.port_idx)},
|
|
inventory_columns={
|
|
"description" : _iface.name,
|
|
"alias" : _iface.name,
|
|
"speed" : _safe_int(_iface.speed)*1000000,
|
|
"oper_status" : _safe_int(_iface.oper_status),
|
|
"admin_status" : _safe_int(_iface.admin_status),
|
|
"available" : _iface.oper_status == '2',
|
|
"vlans" : _iface.portconf,
|
|
"port_type" : 6,
|
|
}
|
|
)
|
|
|
|
yield Attributes(
|
|
path=["networking"],
|
|
inventory_attributes={
|
|
"available_ethernet_ports" : _available_ethernet_ports,
|
|
"total_ethernet_ports" : _total_ethernet_ports,
|
|
"total_interfaces" : _total_ethernet_ports
|
|
}
|
|
)
|
|
|
|
register.check_plugin(
|
|
name='unifi_network_ports_if',
|
|
sections=["unifi_network_ports"],
|
|
service_name='Interface %s',
|
|
discovery_ruleset_name="inventory_if_rules",
|
|
discovery_ruleset_type=register.RuleSetType.ALL,
|
|
discovery_default_parameters=dict(interfaces.DISCOVERY_DEFAULT_PARAMETERS),
|
|
discovery_function=discovery_unifi_network_port_if,
|
|
check_ruleset_name="if",
|
|
check_default_parameters=interfaces.CHECK_DEFAULT_PARAMETERS,
|
|
check_function=check_unifi_network_port_if,
|
|
)
|
|
|
|
register.inventory_plugin(
|
|
name = "unifi_network_ports",
|
|
inventory_function = inventory_unifi_network_ports
|
|
)
|
|
|
|
############ DEVICERADIO ###########
|
|
def discovery_unifi_radios(section):
|
|
#pprint(section)
|
|
for _radio in section.values():
|
|
if _radio.radio == "ng":
|
|
yield Service(item="2.4Ghz")
|
|
if _radio.radio == "na":
|
|
yield Service(item="5Ghz")
|
|
|
|
def check_unifi_radios(item,section):
|
|
_item = { "2.4Ghz" : "ng", "5Ghz" : "na" }.get(item)
|
|
radio = next(filter(lambda x: x.radio == _item,section.values()))
|
|
yield Metric("read_data",_safe_int(radio.rx_bytes))
|
|
yield Metric("write_data",_safe_int(radio.tx_bytes))
|
|
yield Metric("satisfaction",max(0,_safe_int(radio.satisfaction)))
|
|
yield Metric("wlan_user_sta",_safe_int(radio.user_num_sta))
|
|
yield Metric("wlan_guest_sta",_safe_int(radio.guest_num_sta))
|
|
yield Metric("wlan_iot_sta",_safe_int(radio.iot_num_sta))
|
|
|
|
yield Result(
|
|
state=State.OK,
|
|
summary=f"Channel: {radio.channel}"
|
|
)
|
|
yield Result(
|
|
state=State.OK,
|
|
summary=f"Satisfaction: {radio.satisfaction}"
|
|
)
|
|
yield Result(
|
|
state=State.OK,
|
|
summary=f"User: {radio.num_sta}"
|
|
)
|
|
yield Result(
|
|
state=State.OK,
|
|
summary=f"Guest: {radio.guest_num_sta}"
|
|
)
|
|
|
|
register.agent_section(
|
|
name = 'unifi_network_radios',
|
|
parse_function = parse_unifi_nested_dict
|
|
)
|
|
|
|
register.check_plugin(
|
|
name='unifi_network_radios',
|
|
service_name='Radio %s',
|
|
discovery_function=discovery_unifi_radios,
|
|
check_function=check_unifi_radios,
|
|
)
|
|
|
|
|
|
############ SSIDs ###########
|
|
def discovery_unifi_ssids(section):
|
|
for _ssid in section:
|
|
yield Service(item=_ssid)
|
|
|
|
def check_unifi_ssids(item,section):
|
|
ssid = section.get(item)
|
|
if ssid:
|
|
_channels = ",".join(list(filter(lambda x: _safe_int(x) > 0,[ssid.ng_channel,ssid.na_channel])))
|
|
yield Result(
|
|
state=State.OK,
|
|
summary=f"Channels: {_channels}"
|
|
)
|
|
if (_safe_int(ssid.ng_is_guest) + _safe_int(ssid.na_is_guest)) > 0:
|
|
yield Result(
|
|
state=State.OK,
|
|
summary="Guest"
|
|
)
|
|
_satisfaction = max(0,min(_safe_int(ssid.ng_satisfaction),_safe_int(ssid.na_satisfaction)))
|
|
yield Result(
|
|
state=State.OK,
|
|
summary=f"Satisfaction: {_satisfaction}"
|
|
)
|
|
_num_sta = _safe_int(ssid.na_num_sta) + _safe_int(ssid.ng_num_sta)
|
|
if _num_sta > 0:
|
|
yield Result(
|
|
state=State.OK,
|
|
summary=f"User: {_num_sta}"
|
|
)
|
|
yield Metric("satisfaction",max(0,_satisfaction))
|
|
yield Metric("wlan_24Ghz_num_user",_safe_int(ssid.ng_num_sta) )
|
|
yield Metric("wlan_5Ghz_num_user",_safe_int(ssid.na_num_sta) )
|
|
|
|
yield Metric("na_avg_client_signal",_safe_int(ssid.na_avg_client_signal))
|
|
yield Metric("ng_avg_client_signal",_safe_int(ssid.ng_avg_client_signal))
|
|
|
|
yield Metric("na_tcp_packet_loss",_safe_int(ssid.na_tcp_packet_loss))
|
|
yield Metric("ng_tcp_packet_loss",_safe_int(ssid.ng_tcp_packet_loss))
|
|
|
|
yield Metric("na_wifi_retries",_safe_int(ssid.na_wifi_retries))
|
|
yield Metric("ng_wifi_retries",_safe_int(ssid.ng_wifi_retries))
|
|
yield Metric("na_wifi_latency",_safe_int(ssid.na_wifi_latency))
|
|
yield Metric("ng_wifi_latency",_safe_int(ssid.ng_wifi_latency))
|
|
|
|
|
|
|
|
register.agent_section(
|
|
name = 'unifi_network_ssids',
|
|
parse_function = parse_unifi_nested_dict
|
|
)
|
|
|
|
register.check_plugin(
|
|
name='unifi_network_ssids',
|
|
service_name='SSID: %s',
|
|
discovery_function=discovery_unifi_ssids,
|
|
check_function=check_unifi_ssids,
|
|
)
|
|
|
|
|
|
############ SSIDsListController ###########
|
|
def discovery_unifi_ssidlist(section):
|
|
#pprint(section)
|
|
for _ssid in section:
|
|
yield Service(item=_ssid)
|
|
|
|
def check_unifi_ssidlist(item,section):
|
|
ssid = section.get(item)
|
|
if ssid:
|
|
yield Result(
|
|
state=State.OK,
|
|
summary=f"Channels: {ssid.channels}"
|
|
)
|
|
yield Result(
|
|
state=State.OK,
|
|
summary=f"User: {ssid.num_sta}"
|
|
)
|
|
yield Metric("wlan_24Ghz_num_user",_safe_int(ssid.ng_num_sta) )
|
|
yield Metric("wlan_5Ghz_num_user",_safe_int(ssid.na_num_sta) )
|
|
yield Metric("na_avg_client_signal",_safe_int(ssid.na_avg_client_signal))
|
|
yield Metric("ng_avg_client_signal",_safe_int(ssid.ng_avg_client_signal))
|
|
|
|
yield Metric("na_tcp_packet_loss",_safe_int(ssid.na_tcp_packet_loss))
|
|
yield Metric("ng_tcp_packet_loss",_safe_int(ssid.ng_tcp_packet_loss))
|
|
|
|
yield Metric("na_wifi_retries",_safe_int(ssid.na_wifi_retries))
|
|
yield Metric("ng_wifi_retries",_safe_int(ssid.ng_wifi_retries))
|
|
yield Metric("na_wifi_latency",_safe_int(ssid.na_wifi_latency))
|
|
yield Metric("ng_wifi_latency",_safe_int(ssid.ng_wifi_latency))
|
|
|
|
register.agent_section(
|
|
name = 'unifi_ssid_list',
|
|
parse_function = parse_unifi_nested_dict
|
|
)
|
|
|
|
register.check_plugin(
|
|
name='unifi_ssid_list',
|
|
service_name='SSID: %s',
|
|
discovery_function=discovery_unifi_ssidlist,
|
|
check_function=check_unifi_ssidlist,
|
|
)
|
|
|
|
|