forked from 0x2620/pandora
84 lines
2.1 KiB
Python
84 lines
2.1 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
import json
|
|
from threading import Thread
|
|
|
|
from tornado.httpserver import HTTPServer
|
|
from tornado.ioloop import IOLoop
|
|
from tornado.web import Application
|
|
from tornado.websocket import WebSocketHandler
|
|
|
|
import logging
|
|
logger = logging.getLogger('pandora.websocket')
|
|
|
|
sockets = []
|
|
|
|
|
|
class Daemon(Thread):
|
|
def __init__(self, port, address):
|
|
self.port = port
|
|
self.address = address
|
|
Thread.__init__(self)
|
|
self.daemon = True
|
|
self.start()
|
|
|
|
def join(self):
|
|
self.main.stop()
|
|
|
|
def run(self):
|
|
|
|
options = {
|
|
'debug': False,
|
|
'gzip': False
|
|
}
|
|
handlers = [
|
|
(r'/(.*)', Handler),
|
|
]
|
|
self.http_server = HTTPServer(Application(handlers, **options))
|
|
self.main = IOLoop.instance()
|
|
self.http_server.listen(self.port, self.address)
|
|
self.main.start()
|
|
|
|
|
|
class Handler(WebSocketHandler):
|
|
'''
|
|
def check_origin(self, origin):
|
|
# bypass same origin check
|
|
return True
|
|
'''
|
|
|
|
def open(self, path):
|
|
if self not in sockets:
|
|
sockets.append(self)
|
|
|
|
#websocket calls
|
|
def on_close(self):
|
|
if self in sockets:
|
|
sockets.remove(self)
|
|
|
|
def on_message(self, message):
|
|
pass
|
|
#logger.debug('got message %s', message)
|
|
|
|
def post(self, event, data):
|
|
try:
|
|
message = json.dumps([event, data])
|
|
except:
|
|
logger.debug('failed to serialize data %s %s', event, data)
|
|
return
|
|
main = IOLoop.instance()
|
|
main.add_callback(lambda: self.write_message(message))
|
|
|
|
def trigger_event(event, data):
|
|
logger.debug('trigger event %s %s to %s clients', event, data, len(sockets))
|
|
main = IOLoop.instance()
|
|
try:
|
|
message = json.dumps([event, data])
|
|
except:
|
|
logger.debug('faild to serialise data')
|
|
return
|
|
for ws in sockets:
|
|
try:
|
|
main.add_callback(lambda: ws.write_message(message))
|
|
except:
|
|
logger.debug('failed to send to ws %s %s %s', ws, event, data, exc_info=1)
|