#!/bin/env python

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

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

 # opensecurityd

 # 

 # the opensecurityd as RESTful server

 #

 # 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 os

 import os.path

 import subprocess

 import sys

 import web

 from vmmanager.vmmanager import VMManager

 # local

 from environment import Environment

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

 # const

 __version__ = "0.1"

 """All the URLs we know mapping to class handler"""

 opensecurity_urls = (

     '/device_change',           'os_device_change',

     '/application',             'os_application',

     '/device',                  'os_device',

     '/device/credentials',      'os_device_credentials',

     '/device/password',         'os_device_password',

     '/',                        'os_root'

 )

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

 # code

 gvm_mgr = VMManager()

 class os_application:

     """OpenSecurity '/application' handler.

     This is called on GET /application?vm=VM-ID&app=APP-ID

     This tries to access the vm identified with the label VM-ID

     and launched the application identified APP-ID

     """

     def GET(self):

         # pick the arguments

         args = web.input()

         # we _need_ a vm

         if not "vm" in args:

             raise web.badrequest()

         # we _need_ a app

         if not "app" in args:

             raise web.badrequest()

         ## TODO: HARD CODED STUFF HERE! THIS SHOULD BE FLEXIBLE!

         ssh_private_key = os.path.join(Environment("opensecurity").data_path, 'share', '192.168.56.15.ppk')

         putty_session = '192.168.56.15'

         process_command = ['plink.exe', '-i', ssh_private_key, putty_session, args.app]

         si = subprocess.STARTUPINFO()

         si.dwFlags = subprocess.STARTF_USESHOWWINDOW

         si.wShowWindow = subprocess.SW_HIDE

         print('tyring to launch: ' + ' '.join(process_command))

         process = subprocess.Popen(process_command, shell = True)

         return 'launched: ' + ' '.join(process_command)

 class os_device:

     """OpenSecurity '/device' handler"""

     def GET(self):

         return "os_device"

 class os_device_change:

     """OpenSecurity '/device_change' handler"""

     def GET(self):

         gvm_mgr.cygwin_path = 'c:\\cygwin64\\bin\\'

         #gvm_mgr.configureHostNetworking()

         print 'received device_change'

         return "os_device_change"

 class os_device_credentials:

     """OpenSecurity '/device/credentials' handler.

     This is called on GET /device/credentials?id=DEVICE-ID.

     Ideally this should pop up a user dialog to insert his

     credentials based the DEVICE-ID

     """

     def GET(self):

         # pick the arguments

         args = web.input()

         # we _need_ a device id

         if not "id" in args:

             raise web.badrequest()

         # invoke the user dialog as a subprocess

         dlg_credentials_image = os.path.join(sys.path[0], 'opensecurity-dialog.py')

         process_command = [sys.executable, dlg_credentials_image, 'credentials', 'Please provide credentials for accessing \ndevice: "{0}".'.format(args.id)]

         process = subprocess.Popen(process_command, shell = False, stdout = subprocess.PIPE)

         result = process.communicate()[0]

         if process.returncode != 0:

             return 'Credentials request has been aborted.'

         return result

 class os_device_password:

     """OpenSecurity '/device/password' handler.

     This is called on GET /device/password?id=DEVICE-ID.

     Ideally this should pop up a user dialog to insert his

     password based the DEVICE-ID

     """

     def GET(self):

         # pick the arguments

         args = web.input()

         # we _need_ a device id

         if not "id" in args:

             raise web.badrequest()

         # invoke the user dialog as a subprocess

         dlg_credentials_image = os.path.join(sys.path[0], 'opensecurity-dialog.py')

         process_command = [sys.executable, dlg_credentials_image, 'password', 'Please provide a password for accessing \ndevice: "{0}".'.format(args.id)]

         process = subprocess.Popen(process_command, shell = False, stdout = subprocess.PIPE)

         result = process.communicate()[0]

         if process.returncode != 0:

             return 'Credentials request has been aborted.'

         return result

 class os_root:

     """OpenSecurity '/' handler"""

     def GET(self):

         return "OpenSecurity-Server { \"version\": \"%s\" }" % __version__

 # start

 if __name__ == "__main__":

     server = web.application(opensecurity_urls, globals())

     server.run()