58 lines
1.6 KiB
Python
58 lines
1.6 KiB
Python
from ipaddress import ip_network
|
|
|
|
repo.libs.tools.require_bundle(node, 'systemd-networkd')
|
|
|
|
files = {
|
|
'/usr/local/share/icinga/plugins/check_wireguard_connected': {
|
|
'mode': '0755',
|
|
},
|
|
}
|
|
|
|
deps = set()
|
|
|
|
if node.has_bundle('apt'):
|
|
deps.add('pkg_apt:wireguard')
|
|
|
|
health_checks = {}
|
|
for peer, config in sorted(node.metadata.get('wireguard/peers', {}).items()):
|
|
files[f'/etc/systemd/network/wg_{config["iface"]}.netdev'] = {
|
|
'content_type': 'mako',
|
|
'source': 'wg.netdev',
|
|
'owner': 'systemd-network',
|
|
'mode': '0600',
|
|
'context': {
|
|
'endpoint': config.get('endpoint'),
|
|
'iface': config['iface'],
|
|
'peer': peer,
|
|
'port': config['my_port'],
|
|
'privatekey': node.metadata.get('wireguard/privatekey'),
|
|
'psk': config['psk'],
|
|
'pubkey': config['pubkey'],
|
|
},
|
|
'needs': deps,
|
|
'triggers': {
|
|
'svc_systemd:systemd-networkd:restart',
|
|
},
|
|
}
|
|
|
|
if config.get('health_check', False):
|
|
health_checks[peer] = config['their_ip']
|
|
|
|
if health_checks:
|
|
files['/usr/local/bin/wg_health_check'] = {
|
|
'content_type': 'mako',
|
|
'context': {
|
|
'peers': health_checks,
|
|
},
|
|
'mode': '0755',
|
|
}
|
|
files['/etc/cron.d/wg_health_check'] = {
|
|
'content': '* * * * * root /usr/local/bin/wg_health_check | logger -t wg_health_check\n',
|
|
}
|
|
|
|
if node.has_bundle('pppd'):
|
|
files['/etc/ppp/ip-up.d/reconnect-wireguard'] = {
|
|
'source': 'pppd-ip-up',
|
|
'content_type': 'mako',
|
|
'mode': '0755',
|
|
}
|