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
|
# -*- coding: utf-8 -*-
# signaler_qt.py
# Copyright (C) 2013, 2014 LEAP
#
# This program 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.
#
# This program 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 this program. If not, see <http://www.gnu.org/licenses/>.
import threading
import time
from PySide import QtCore
import zmq
from zmq.auth.thread import ThreadAuthenticator
from leap.bitmask.backend.api import SIGNALS
from leap.bitmask.backend.utils import get_frontend_certificates
import logging
logger = logging.getLogger(__name__)
class SignalerQt(QtCore.QThread):
"""
Signaling server.
Receives signals from the signaling client and emit Qt signals for the GUI.
"""
PORT = "5667"
BIND_ADDR = "tcp://127.0.0.1:%s" % PORT
def __init__(self):
QtCore.QThread.__init__(self)
self._do_work = threading.Event()
self._do_work.set()
def run(self):
"""
Start a loop to process the ZMQ requests from the signaler client.
"""
logger.debug("Running SignalerQt loop")
context = zmq.Context()
socket = context.socket(zmq.REP)
# Start an authenticator for this context.
auth = ThreadAuthenticator(context)
auth.start()
auth.allow('127.0.0.1')
# Tell authenticator to use the certificate in a directory
auth.configure_curve(domain='*', location=zmq.auth.CURVE_ALLOW_ANY)
public, secret = get_frontend_certificates()
socket.curve_publickey = public
socket.curve_secretkey = secret
socket.curve_server = True # must come before bind
socket.bind(self.BIND_ADDR)
while self._do_work.is_set():
# Wait for next request from client
try:
request = socket.recv(zmq.NOBLOCK)
logger.debug("Received request: '{0}'".format(request))
socket.send("OK")
self._process_request(request)
except zmq.ZMQError as e:
if e.errno != zmq.EAGAIN:
raise
time.sleep(0.01)
logger.debug("SignalerQt thread stopped.")
def stop(self):
"""
Stop the SignalerQt blocking loop.
"""
self._do_work.clear()
def _process_request(self, request_json):
"""
Process a request and call the according method with the given
parameters.
:param request_json: a json specification of a request.
:type request_json: str
"""
try:
request = zmq.utils.jsonapi.loads(request_json)
signal = request['signal']
data = request['data']
except Exception as e:
msg = "Malformed JSON data in Signaler request '{0}'. Exc: {1!r}"
msg = msg.format(request_json, e)
logger.critical(msg)
raise
if signal not in SIGNALS:
logger.error("Unknown signal received, '{0}'".format(signal))
return
try:
qt_signal = getattr(self, signal)
except Exception:
logger.warning("Signal not implemented, '{0}'".format(signal))
return
logger.debug("Emitting '{0}'".format(signal))
if data is None:
qt_signal.emit()
else:
qt_signal.emit(data)
|
