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
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
from operationscore.Renderer import *
import util.TimeOps as timeops
import util.ComponentRegistry as compReg
import threading, socket, re, struct, hashlib, json, webbrowser
class WebsocketRenderer(Renderer):
"""Renders frame data over a websocket."""
def initRenderer(self):
self.hostname = self.argDict['Hostname']
self.orig_port = int(self.argDict['SourcePort'])
self.port = int(self.argDict['Port'])
self.clients = []
self.clients_lock = threading.Lock()
self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
self.sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
self.sock.bind(('', self.port))
self.sock.listen(1)
self.connection_thread = threading.Thread(target=self.handle_connections)
self.connection_thread.daemon = True
self.connection_thread.start()
self.serve_thread = threading.Thread(target=self.serve_page)
self.serve_thread.daemon = True
self.serve_thread.start()
webbrowser.open('http://'+self.hostname+':'+str(self.orig_port))
def serve_page(self):
page = open(self.argDict['Page']).read()
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
sock.bind(('', self.orig_port))
sock.listen(1)
while True:
client, addr = sock.accept()
req = client.recv(4096)
client.send('HTTP/1.0 200 Found\r\n')
client.send('Content-type: text/html\r\n\r\n')
client.send(page)
client.close()
def handle_connections(self):
while True:
client, addr = self.sock.accept()
print 'Accepted websocket connection from %s' % str(addr)
header = ''
while not re.search("\r?\n\r?\n.{8}", header): # Receive headers + 8 bytes data
header += client.recv(1024)
key1 = re.search("Sec-WebSocket-Key1: (.*)$", header, re.M).group(1)
key2 = re.search("Sec-WebSocket-Key2: (.*)$", header, re.M).group(1)
data = header[-8:]
key1n = int(re.sub("[^\d]", '', key1))
key1ns = key1.count(' ')
n1 = key1n // key1ns
key2n = int(re.sub("[^\d]", '', key2))
key2ns = key2.count(' ')
n2 = key2n // key2ns
s = struct.pack("!II", n1, n2) + data
respkey = hashlib.md5(s).digest()
if self.orig_port == 80:
origin = 'http://'+self.hostname
else:
origin = 'http://'+self.hostname+':'+str(self.orig_port)
resp = \
"HTTP/1.1 101 Web Socket Protocol Handshake\r\n" + \
"Upgrade: WebSocket\r\n" + \
"Connection: Upgrade\r\n" + \
"Sec-WebSocket-Origin:"+ origin + "\r\n" + \
"Sec-WebSocket-Location: ws://"+self.hostname+":"+ \
str(self.port)+"/\r\n" + \
"Sec-WebSocket-Protocol: ledweb\r\n\r\n" + \
respkey + "\r\n"
client.send(resp)
self.clients_lock.acquire()
self.clients.append(client)
self.clients_lock.release()
def render(self, lightSystem, currentTime=timeops.time()):
json_frame = []
for light in lightSystem:
loc = light.location
c = light.state(currentTime)
cs = 'rgb('+str(c[0])+','+str(c[1])+','+str(c[2])+')'
json_frame.append((loc, cs))
size = compReg.getComponent('Screen').getSize()
json_data = json.dumps(dict(status='ok', size=size, frame=json_frame))
self.client_push(json_data)
def client_push(self, data):
self.clients_lock.acquire()
dead_clients = []
for i in range(len(self.clients)):
try:
self.clients[i].send("\x00")
self.clients[i].send(data)
self.clients[i].send("\xff")
except socket.error:
dead_clients.append(i)
for i in range(len(dead_clients)):
self.close_sock(self.clients[dead_clients[i]-i])
del self.clients[dead_clients[i]-i]
self.clients_lock.release()
def close_sock(self, s):
try:
c.shutdown(socket.SHUT_RDWR)
c.close()
except Exception:
pass
|