2017-11-21 09:25:57 +00:00
|
|
|
import sys
|
2017-11-17 14:57:03 +00:00
|
|
|
import tornado
|
|
|
|
import zmq
|
|
|
|
from tornado.web import Application
|
|
|
|
from tornado.ioloop import IOLoop
|
|
|
|
|
2018-02-08 13:46:00 +00:00
|
|
|
from sql_injection_fsm import SQLInjectionFSM
|
2018-02-09 08:58:47 +00:00
|
|
|
from tfw.config import tfwenv
|
2018-01-31 14:50:52 +00:00
|
|
|
from tfw.config.logs import logging
|
2018-02-09 13:08:14 +00:00
|
|
|
from tfw.networking.controller_connector import ControllerConnector
|
|
|
|
from tfw.networking.serialization import deserialize_all, serialize_all
|
|
|
|
|
2018-01-29 17:36:29 +00:00
|
|
|
log = logging.getLogger(__name__)
|
2018-02-06 13:19:41 +00:00
|
|
|
from tfw.networking.server.zmq_websocket_handler import FSMManagingSocketHandler
|
2018-01-29 13:09:40 +00:00
|
|
|
|
2017-11-17 14:57:03 +00:00
|
|
|
|
2018-02-09 13:08:14 +00:00
|
|
|
def zmq_callback(stream, msg_parts):
|
|
|
|
key, data = deserialize_all(*msg_parts)
|
|
|
|
if key == 'test':
|
|
|
|
stream.send_multipart(serialize_all(key, 'OK'))
|
|
|
|
if key == 'solution_check':
|
|
|
|
stream.send_multipart(serialize_all(key, {
|
|
|
|
'solved': True,
|
|
|
|
'message': 'solved'
|
|
|
|
}))
|
|
|
|
|
|
|
|
|
2017-11-17 14:57:03 +00:00
|
|
|
if __name__ == '__main__':
|
2018-02-08 13:46:00 +00:00
|
|
|
fsm = SQLInjectionFSM()
|
2018-02-09 13:08:14 +00:00
|
|
|
controller_connector = ControllerConnector()
|
|
|
|
controller_connector.register_callback(zmq_callback)
|
2017-11-17 14:57:03 +00:00
|
|
|
routes = [
|
2018-02-08 13:46:00 +00:00
|
|
|
(r'/ws', FSMManagingSocketHandler, {'fsm': fsm}),
|
2017-11-17 14:57:03 +00:00
|
|
|
]
|
|
|
|
application = Application(
|
|
|
|
routes,
|
|
|
|
autoreload=True
|
|
|
|
)
|
2017-12-11 12:34:08 +00:00
|
|
|
|
2018-02-09 08:58:47 +00:00
|
|
|
application.listen(tfwenv.WEB_PORT)
|
2018-01-29 17:36:29 +00:00
|
|
|
log.debug('Python version: {}'.format(sys.version[:5]))
|
|
|
|
log.debug('Tornado version: {}'.format(tornado.version))
|
|
|
|
log.debug('ZeroMQ version: {}'.format(zmq.zmq_version()))
|
|
|
|
log.debug('PyZMQ version: {}'.format(zmq.pyzmq_version()))
|
2018-02-09 08:58:47 +00:00
|
|
|
log.info('Tornado application listening on port {}'.format(tfwenv.WEB_PORT))
|
2017-11-17 14:57:03 +00:00
|
|
|
IOLoop.instance().start()
|