1 # -*- coding: utf-8 -*-
3 # ------------------------------------------------------------
6 # an opensecurity dialog
8 # Autor: Oliver Maurhart, <oliver.maurhart@ait.ac.at>
10 # Copyright (C) 2013 AIT Austrian Institute of Technology
11 # AIT Austrian Institute of Technology GmbH
12 # Donau-City-Strasse 1 | 1220 Vienna | Austria
13 # http://www.ait.ac.at
15 # This program is free software; you can redistribute it and/or
16 # modify it under the terms of the GNU General Public License
17 # as published by the Free Software Foundation version 2.
19 # This program is distributed in the hope that it will be useful,
20 # but WITHOUT ANY WARRANTY; without even the implied warranty of
21 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
22 # GNU General Public License for more details.
24 # You should have received a copy of the GNU General Public License
25 # along with this program; if not, write to the Free Software
26 # Foundation, Inc., 51 Franklin Street, Fifth Floor,
27 # Boston, MA 02110-1301, USA.
28 # ------------------------------------------------------------
31 # ------------------------------------------------------------
43 from PyQt4 import QtCore
44 from PyQt4 import QtGui
47 import __init__ as opensecurity
49 if sys.platform == 'win32' or sys.platform == 'cygwin':
50 from cygwin import Cygwin
52 import opensecurity_client_restful_server
53 from ui import AboutDialog
54 from ui import ConfigureDialog
55 from ui import opensecurity_rc
58 # ------------------------------------------------------------
62 class OpenSecurityWait(QtGui.QDialog):
64 """OpenSecurity: please wait ..."""
66 def __init__(self, parent = None, flags = QtCore.Qt.WindowFlags(0)):
67 super(OpenSecurityWait, self).__init__(parent, flags)
68 self.setWindowTitle('OpenSecurity')
73 """Create the widgets."""
75 lyMain = QtGui.QVBoxLayout(self)
76 lyMain.setContentsMargins(8, 8, 8, 8)
78 # content area: left pixmap, right text
79 lbTitle = QtGui.QLabel('Creating secure subsystem. Please stand by ...')
80 lyMain.addWidget(lbTitle)
82 self.setMinimumSize(400, 50)
83 self.resize(lyMain.minimumSize())
86 class OpenSecurityTrayIcon(QtGui.QSystemTrayIcon):
88 """This is the OpenSecuirty Tray Icon"""
90 def __init__(self, icon, parent=None):
92 super(OpenSecurityTrayIcon, self).__init__(icon, parent)
96 def clicked_about(self):
102 def clicked_browser(self):
103 """wish for safe internet browsing"""
105 if not (sys.platform == 'win32' or sys.platform == 'cygwin'):
106 QtGui.QMessageBox.critical(self.parent(), 'OpenSecurity Error', 'This action is not supported on this platform.\nSorry.')
111 # get a proper browsing VM
114 # TODO: HARDCODED ADDRESS OF OPENSECURITYD
115 browsing_vm = urllib2.urlopen('http://127.0.0.1:8080/browsing').readline()
116 print('Called http://127.0.0.1:8080/browsing got: ' + str(browsing_vm))
121 QtGui.QApplication.instance().processEvents()
122 QtGui.QMessageBox.critical(None, 'Failed to invoke Safe Internet Browsing', 'OpenSecurity Error')
124 QtGui.QApplication.instance().processEvents()
127 def clicked_configure(self):
128 """clicked configure"""
129 d = ConfigureDialog()
133 def clicked_exit(self):
135 opensecurity_client_restful_server.stop()
139 def clicked_launch_application(self):
140 """clicked the launch an application"""
141 dlg_launch_image = os.path.join(sys.path[0], 'ui', 'launch_dialog.py')
142 process_command = [sys.executable, dlg_launch_image]
143 process = subprocess.Popen(process_command, shell = False, stdout = subprocess.PIPE)
145 stdout = process.communicate()[0]
146 j = json.loads(stdout)
152 # get a proper browsing VM
155 # TODO: HARDCODED ADDRESS OF OPENSECURITYD
156 url = 'http://127.0.0.1:8080/sdvms/' + j['vm'] + '/application' + j['application']
157 result = urllib2.urlopen(url).readline()
163 def clicked_mail(self):
166 address = 'feedback@opensecurity.at'
167 subject = 'Feedback zu OpenSecurity V' + opensecurity.__version__
169 if sys.platform == 'linux2':
170 subprocess.Popen(['xdg-email', '--subject', subject, address])
171 elif sys.platform == 'win32' or sys.platform == 'cygwin':
172 mail_url = 'mailto:' + urllib.quote(address, '@') + '?' + urllib.quote('subject=' + subject)
173 subprocess.Popen(['cmd', '/C', 'start', mail_url])
177 """create the user interface
178 As for the system tray this is 'just' the context menu.
181 # define the tray icon menu
182 menu = QtGui.QMenu(self.parent())
183 self.setContextMenu(menu)
186 self.acBrowser = QtGui.QAction('Secure Browsing', self.parent())
188 icon.addPixmap(QtGui.QPixmap(QtCore.QString.fromUtf8(':/opensecurity/gfx/opensecurity_browsing_64.png')), QtGui.QIcon.Normal, QtGui.QIcon.Off)
189 self.acBrowser.setIcon(icon)
190 menu.addAction(self.acBrowser)
193 # add standard menu items
194 self.acLaunch = QtGui.QAction('Launch Application', self.parent())
196 icon.addPixmap(QtGui.QPixmap(QtCore.QString.fromUtf8(':/opensecurity/gfx/opensecurity_launch_64.png')), QtGui.QIcon.Normal, QtGui.QIcon.Off)
197 self.acLaunch.setIcon(icon)
198 menu.addAction(self.acLaunch)
199 self.acConfigure = QtGui.QAction('Configuration', self.parent())
201 icon.addPixmap(QtGui.QPixmap(QtCore.QString.fromUtf8(':/opensecurity/gfx/opensecurity_configure_64.png')), QtGui.QIcon.Normal, QtGui.QIcon.Off)
202 self.acConfigure.setIcon(icon)
203 menu.addAction(self.acConfigure)
206 self.acMail = menu.addAction('Send feedback mail')
208 icon.addPixmap(QtGui.QPixmap(QtCore.QString.fromUtf8(':/opensecurity/gfx/opensecurity_mail_64.png')), QtGui.QIcon.Normal, QtGui.QIcon.Off)
209 self.acMail.setIcon(icon)
210 self.acAbout = menu.addAction('About')
211 self.acExit = menu.addAction('Exit')
213 self.acBrowser.triggered.connect(self.clicked_browser)
214 self.acLaunch.triggered.connect(self.clicked_launch_application)
215 self.acConfigure.triggered.connect(self.clicked_configure)
216 self.acAbout.triggered.connect(self.clicked_about)
217 self.acExit.triggered.connect(self.clicked_exit)
218 self.acMail.triggered.connect(self.clicked_mail)
224 parser = argparse.ArgumentParser(description = 'OpenSecurity Tray Icon.')
225 parser.add_argument('-p', '--port', type=int, default=8090, help='port number of the REST API this instance will listen on.')
226 args = parser.parse_args()
229 a = QtGui.QApplication(sys.argv)
231 # enforce singelton process
232 if opensecurity_client_restful_server.is_already_running(args.port):
233 QtGui.QMessageBox.critical(None, 'OpenSecurity Error', 'OpenSecurity Tray Instance already launched.\nClose previous instance first.')
237 opensecurity_client_restful_server.serve(args.port, True)
239 # init tray icon widget
242 icon.addPixmap(QtGui.QPixmap(QtCore.QString.fromUtf8(":/opensecurity/gfx/opensecurity_icon_64.png")), QtGui.QIcon.Normal, QtGui.QIcon.Off)
243 trayIcon = OpenSecurityTrayIcon(icon)
244 opensecurity_client_restful_server.tray_icon = trayIcon
248 a.setQuitOnLastWindowClosed(False)
253 if __name__ == "__main__":