51 lines
2.0 KiB
Python
51 lines
2.0 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
import argparse
|
||
|
|
import sys
|
||
|
|
import traceback
|
||
|
|
|
||
|
|
import requests
|
||
|
|
|
||
|
|
import checker.nagios as nagios
|
||
|
|
|
||
|
|
parser = argparse.ArgumentParser(description='Test federation between two homeservers.')
|
||
|
|
parser.add_argument('--endpoint', required=True, help='Endpoint to parse. See fed.mau.dev or federationtester.matrix.org')
|
||
|
|
parser.add_argument('--timeout', type=float, default=90, help='Request timeout limit.')
|
||
|
|
parser.add_argument('--warn', type=float, default=2.0, help='Manually set warn level.')
|
||
|
|
parser.add_argument('--crit', type=float, default=2.5, help='Manually set critical level.')
|
||
|
|
args = parser.parse_args()
|
||
|
|
|
||
|
|
|
||
|
|
def main() -> None:
|
||
|
|
r = requests.get(args.endpoint, timeout=args.timeout)
|
||
|
|
if not r.status_code:
|
||
|
|
print(f'UNKNOWN: tester endpoint failed with status code {r.status_code}\n', r.text)
|
||
|
|
sys.exit(nagios.UNKNOWN)
|
||
|
|
|
||
|
|
r_json = r.json()
|
||
|
|
if r_json.get('FederationOK'):
|
||
|
|
print('OK: federation tester reported success.')
|
||
|
|
print(f"Version: {r_json.get('Version', {}).get('name')}/{r_json.get('Version', {}).get('version')}")
|
||
|
|
print('WellKnown:', r_json.get('WellKnownResult', {}).get('m.server'))
|
||
|
|
nagios_output = nagios.OK
|
||
|
|
else:
|
||
|
|
print('CRITICAL: federation tester reported failure.')
|
||
|
|
print('Server Version:', r_json.get('m.server'))
|
||
|
|
print('WellKnown:', r_json.get('WellKnownResult', {}).get('m.server'))
|
||
|
|
print('WellKnown Result:', r_json.get('WellKnownResult', {}).get('result'))
|
||
|
|
print('DNS Result:', r_json.get('DNSResult', {}).get('SRVError', {}).get('Message', {}))
|
||
|
|
print('Version:', r_json.get('Version', {}).get('error'))
|
||
|
|
print('Connection Report:', r_json.get('ConnectionReports'))
|
||
|
|
print('Connection Errors:', r_json.get('ConnectionErrors'))
|
||
|
|
nagios_output = nagios.CRITICAL
|
||
|
|
|
||
|
|
sys.exit(nagios_output)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
try:
|
||
|
|
main()
|
||
|
|
except Exception as e:
|
||
|
|
print(f"UNKNOWN: exception\n{e}")
|
||
|
|
print(traceback.format_exc())
|
||
|
|
sys.exit(nagios.UNKNOWN)
|