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
|
#!/usr/bin/env python
"""
socket_notify.py - Phenny Socket Notification
Copyright 2013-2016, Alexander Sulfrian <alex@spline.inf.fu-berlin.de>
Licensed under the Eiffel Forum License 2.
http://inamidst.com/phenny/
"""
import asyncore, socket, stat, os
DEFAULT_CONFIG = {
'path': '/var/run/phenny/socket',
'umask': 0o000,
}
def setup(phenny):
config = dict()
config.update(DEFAULT_CONFIG)
config.update(getattr(phenny.config, 'socket_notify', dict()))
dir = os.path.dirname(config['path'])
if not os.path.exists(dir):
os.makedirs(dir)
Monitor(config, phenny)
class Sender(asyncore.dispatcher):
def __init__(self, sock, phenny):
asyncore.dispatcher.__init__(self, sock)
self.phenny = phenny
def handle_read(self):
data = self.recv(8192)
if data:
for line in data.split('\n'):
for chan in self.phenny.config.channels:
self.phenny.msg(chan, line)
else:
self.close()
class Monitor(asyncore.dispatcher):
def __init__(self, config, phenny):
asyncore.dispatcher.__init__(self)
self.phenny = phenny
self.phenny._orig_close = self.phenny.close
self.phenny.close = (lambda: self._phenny_close())
self.create_socket(socket.AF_UNIX, socket.SOCK_STREAM)
try:
mode = os.stat(config['path']).st_mode
if stat.S_ISSOCK(mode):
os.unlink(config['path'])
else:
print("Path '%s' exists and is not a socket!" % config['path'])
sys.exit(1)
except OSError:
pass
old_mask = os.umask(config['umask'])
self.bind(config['path'])
os.umask(old_mask)
self.listen(5)
def _phenny_close(self):
self.close()
self.phenny._orig_close()
def handle_accept(self):
pair = self.accept()
if pair is not None:
sock, addr = pair
handler = Sender(sock, self.phenny)
|
