#!/bin/env python

 # -*- coding: utf-8 -*-

 # ------------------------------------------------------------

 # opensecurity-launcher

 # 

 # launches an application inside a VM

 #

 # Autor: Oliver Maurhart, <oliver.maurhart@ait.ac.at>

 #

 # Copyright (C) 2013 AIT Austrian Institute of Technology

 # AIT Austrian Institute of Technology GmbH

 # Donau-City-Strasse 1 | 1220 Vienna | Austria

 # http://www.ait.ac.at

 #

 # 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 version 2.

 # 

 # 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, write to the Free Software

 # Foundation, Inc., 51 Franklin Street, Fifth Floor, 

 # Boston, MA  02110-1301, USA.

 # ------------------------------------------------------------

 # ------------------------------------------------------------

 # imports

 import argparse

 import os

 import subprocess

 import sys

 import urllib2

 from PyQt4 import QtCore

 from PyQt4 import QtGui

 # local

 from about import About

 from cygwin import Cygwin

 from environment import Environment

 # ------------------------------------------------------------

 # code

 class Chooser(QtGui.QDialog, object):

     """Ask the user what to launch."""

     def __init__(self, parent = None, flags = QtCore.Qt.WindowFlags(0)):

         super(Chooser, self).__init__(parent, flags)

         self.setWindowTitle('OpenSecuirty Launch Application')

         self.setup_ui()

         # positionate ourself central

         screen = QtGui.QDesktopWidget().screenGeometry()

         self.resize(self.geometry().width() * 1.25, self.geometry().height())

         size = self.geometry()

         self.move((screen.width() - size.width()) / 2, (screen.height() - size.height()) / 2)

         # TODO: THIS HERE IS HARD CODED

         self._vms = [ { 'name': 'SecurityDVM0' } ]

         self._apps = [ { 'name': 'Browser', 'command': '/usr/bin/iceweasel' } ]

         # add the VMs we know

         self._cbVM.clear()

         for vm in self._vms:

             self._cbVM.addItem(vm['name'])

         # add the commands we know

         self._cbApplication.clear()

         for app in self._apps:

             self._cbApplication.addItem(app['name'])

     def app_get(self):

         """The application of the user."""

         a = str(self._cbApplication.currentText())

         for app in self._apps:

             if a == app['name']:

                 return app['command']

         return a

     app = property(app_get)

     def clicked_about(self):

         """clicked the about button"""

         dlgAbout = About()

         dlgAbout.exec_()

     def clicked_cancel(self):

         """clicked the cancel button"""

         self.reject()

     def clicked_ok(self):

         """clicked the ok button"""

         self.accept()

     def setup_ui(self):

         """Create the widgets."""

         lyMain = QtGui.QVBoxLayout(self)

         lyMain.setContentsMargins(8, 8, 8, 8)

         # content area: left pixmap, right text

         lyContent = QtGui.QHBoxLayout()

         lyMain.addLayout(lyContent)

         # pixmap

         lbPix = QtGui.QLabel()

         lbPix.setPixmap(QtGui.QPixmapCache.find('opensecurity_icon_64'))

         lyContent.addWidget(lbPix, 0, QtCore.Qt.Alignment(QtCore.Qt.AlignTop + QtCore.Qt.AlignHCenter))

         lyContent.addSpacing(16)

         # launch ...

         lyLaunch = QtGui.QGridLayout()

         lyContent.addLayout(lyLaunch)

         lbTitle = QtGui.QLabel('Specify details for application to launch.')

         lyLaunch.addWidget(lbTitle, 0, 0, 1, 2)

         lbVM = QtGui.QLabel('&VM-ID:')

         lyLaunch.addWidget(lbVM, 1, 0)

         self._cbVM = QtGui.QComboBox()

         self._cbVM.setEditable(True)

         self._cbVM.setInsertPolicy(QtGui.QComboBox.InsertAlphabetically)

         lyLaunch.addWidget(self._cbVM, 1, 1)

         lbVM.setBuddy(self._cbVM)

         lbApplication = QtGui.QLabel('&Application:')

         lyLaunch.addWidget(lbApplication, 2, 0)

         self._cbApplication = QtGui.QComboBox()

         self._cbApplication.setEditable(True)

         self._cbApplication.setInsertPolicy(QtGui.QComboBox.InsertAlphabetically)

         lyLaunch.addWidget(self._cbApplication, 2, 1)

         lbApplication.setBuddy(self._cbApplication)

         lyLaunch.addWidget(QtGui.QWidget(), 3, 0, 1, 2)

         lyLaunch.setColumnStretch(1, 1)

         lyLaunch.setRowStretch(3, 1)

         lyMain.addStretch(1)

         # buttons

         lyButton = QtGui.QHBoxLayout()

         lyMain.addLayout(lyButton)

         lyButton.addStretch(1)

         btnOk = QtGui.QPushButton('&Ok', self)

         btnOk.setDefault(True)

         btnOk.setMinimumWidth(100)

         lyButton.addWidget(btnOk)

         btnCancel = QtGui.QPushButton('&Cancel', self)

         btnCancel.setMinimumWidth(100)

         lyButton.addWidget(btnCancel)

         btnAbout = QtGui.QPushButton('&About', self)

         btnAbout.setMinimumWidth(100)

         lyButton.addWidget(btnAbout)

         button_width = max(btnOk.width(), btnCancel.width(), btnAbout.width())

         btnOk.setMinimumWidth(button_width)

         btnCancel.setMinimumWidth(button_width)

         btnAbout.setMinimumWidth(button_width)

         # reduce to the max

         self.resize(lyMain.minimumSize())

         # connectors

         btnOk.clicked.connect(self.clicked_ok)

         btnCancel.clicked.connect(self.clicked_cancel)

         btnAbout.clicked.connect(self.clicked_about)

     def vm_get(self):

         """The vm of choice."""

         return str(self._cbVM.currentText())

     vm = property(vm_get)

 def ask_user():

     """ask the user for VM and app to start"""

     # prebuild the pixmap cache: fetch all jpg, png and jpeg images and load them

     image_path = os.path.join(Environment("OpenSecurity").data_path, '..', 'gfx')

     for file in os.listdir(image_path):

         if file.lower().rpartition('.')[2] in ('jpg', 'png', 'jpeg'):

             QtGui.QPixmapCache.insert(file.lower().rpartition('.')[0], QtGui.QPixmap(os.path.join(image_path, file)))

     # we should have now our application icon

     app.setWindowIcon(QtGui.QIcon(QtGui.QPixmapCache.find('opensecurity_icon_64')))

     # pop up the dialog

     dlg = Chooser()

     dlg.show()

     app.exec_()

     if dlg.result() == QtGui.QDialog.Accepted:

         return dlg.vm, dlg.app

     return '', ''

 def main():

     """entry point"""

     # launch Qt

     global app

     app = QtGui.QApplication(sys.argv)

     # parse command line

     parser = argparse.ArgumentParser(description = 'OpenSecurity Launcher: run application in VM')

     parser.add_argument('vm', metavar='VM', help='Name of Virtual Machine', nargs='?', type=str, default='')

     parser.add_argument('command', metavar='COMMAND', help='Full path of command and arguments to start inside VM', nargs='?', type=str, default='')

     args = parser.parse_args()

     # we must have all set

     if args.vm == "" or args.command == '':

         print('VM and/or COMMAND missing - invoking user dialog')

         args.vm, args.command = ask_user()

     # still no VM? --> no chance, over and out!

     if args.vm == '':

         sys.exit(0)

     # ensure we have our X11 running

     Cygwin.start_X11()

     # call the OpenSecurity Admin to launch our progie =)

     # TODO: hard coded PORT

     url = 'http://127.0.0.1:8080/sdvms/' + args.vm + '/application' + args.command

     print('Calling ' + url)

     try:

         result = urllib2.urlopen(url, None, 5)

     except urllib2.HTTPError as e:

         # Error, Fail, ... :(

         msg = 'Error received from OpenSecurity Subsystem\nError code: ' + str(e.code) + '\nReason: ' + e.reason

         QtGui.QMessageBox.critical(None, 'OpenSecurity Error', msg)

 # start

 if __name__ == "__main__":

     main()