-
Notifications
You must be signed in to change notification settings - Fork 0
/
Forwarder.py
94 lines (75 loc) · 3.19 KB
/
Forwarder.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
#!/usr/bin/python -u
# vim: set fileencoding=utf-8 sw=4 expandtab ts=4 :
#-----------------------------------------------------------------------------
#
# Alarm Forwarder
#
# Copyright (C) 2019 Florian Pose
#
# This file is part of Alarm Display.
#
# Alarm Display is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# Alarm Display 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
# Alarm Display. If not, see <http://www.gnu.org/licenses/>.
#
#-----------------------------------------------------------------------------
import re
from PyQt5 import QtNetwork
#-----------------------------------------------------------------------------
class Forwarder:
def __init__(self, config, logger):
self.hosts = []
self.config = config
self.logger = logger
if not config.has_section('forward'):
return
hostRe = re.compile('host([0-9]+)')
for key, host in self.config.items('forward'):
ma = hostRe.fullmatch(key)
if not ma:
self.logger.warning('Ignoring %s.', key)
continue
num = int(ma.group(1))
port = self.config.getint("forward", "port%u" % (num),
fallback = 11211)
self.logger.info('Adding forward %s as %s:%u', key, host, port)
hostInfo = QtNetwork.QHostInfo.fromName(host)
if not hostInfo.addresses():
self.logger.warning(' Failed to find address.')
continue
addr = hostInfo.addresses()[0]
self.hosts.append((host, addr, port))
self.logger.info(' Added address %s.', addr.toString())
def forward(self, alarm):
data = None
if alarm.source == 'pager':
data = alarm.pager.encode('utf-8')
socket = QtNetwork.QUdpSocket()
for host, addr, port in self.hosts:
self.logger.info('Forwarding to %s (%s:%u/udp)',
host, addr.toString(), port)
socket.writeDatagram(data, addr, port)
elif alarm.source == 'xml':
data = alarm.xml # bytes
socket = QtNetwork.QTcpSocket()
for host, addr, port in self.hosts:
self.logger.info('Forwarding to %s (%s:%u/tcp)',
host, addr.toString(), port)
socket.connectToHost(addr, port)
if not socket.waitForConnected(5000):
self.logger.error('Connection failed.')
continue
socket.writeData(data)
if not socket.waitForBytesWritten(5000):
self.logger.error('Writing to socket failed.')
socket.close()
#-----------------------------------------------------------------------------