#!/bin/env python

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

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

# opensecurityd

#   

# the opensecurityd as RESTful server

#

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

#        Mihai Bartha, <mihai.bartha@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 json

import os

import os.path

import subprocess

import sys

import tempfile

import web

import vmmanager

# local

import __init__ as opensecurity

from cygwin import Cygwin

from environment import Environment

from opensecurity_util import logger, showTrayMessage

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

# const

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

opensecurity_urls = (

    '/browsing',                        'os_browsing',          # http://localhost:8080/browsing                                GET

    '/fetch_initial_image',             'os_fetch_image',       # http://localhost:8080/fetch_initial_image                     GET

    '/init',                            'os_init',              # http://localhost:8080/init                                    GET

    '/initial_image',                   'os_initial_image',     # http://localhost:8080/initial_image                           GET

    '/sdvms',                           'os_sdvms',             # http://localhost:8080/sdvms                                   GET, PUT

    '/sdvms/(.*)/application/(.*)',     'os_sdvm_application',  # http://localhost:8080/sdvms/[VMNAME]/application/[COMMAND]    GET

    '/sdvms/(.*)/ip',                   'os_sdvm_ip',           # http://localhost:8080/sdvms/[VMNAME]/ip                       GET

    '/sdvms/(.*)/start',                'os_sdvm_start',        # http://localhost:8080/sdvms/[VMNAME]/start                    GET

    '/sdvms/(.*)/stop',                 'os_sdvm_stop',         # http://localhost:8080/sdvms/[VMNAME]/stop                     GET

    '/sdvms/(.*)',                      'os_sdvm',              # http://localhost:8080/sdvms/[VMNAME]                          GET, DELETE

    '/setup',                           'os_setup',             # http://localhost:8080/setup                                   GET

    '/vms',                             'os_vms',               # http://localhost:8080/vms                                     GET

    '/vms/(.*)',                        'os_vm',                # http://localhost:8080/vms/[VMNAME]                            GET

    '/update_template',                 'os_update_template',   # http://localhost:8080/update_template                         GET

    '/terminate',                       'os_terminate',         # http://localhost:8080/terminate                               GET

    '/initialize',                      'os_initialize',        # http://localhost:8080/initialize                               GET

    '/',                                'os_root'               # http://localhost:8080/                                        GET

)

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

# vars

# Global VMManager instance

gvm_mgr = None

# server instance

server = None

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

# code

class os_browsing:

    """OpenSecurity '/browsing' handler

    - GET: Start and prepare a new SecurityVM for Internet Browsing. Return the name of the VM.

    """

    def GET(self):

        args = web.input()

        log_call(web.ctx.environ)

        global gvm_mgr

        try:

            proxy = None

            if 'ProxyServer' in args:

                proxy = args['ProxyServer']

            result = gvm_mgr.handleBrowsingRequest(proxy)

            return result

        except:

            raise web.internalerror()

class os_fetch_image:

    """OpenSecurity '/fetch_initial_image' handler

    - GET: fetch the initial image from the X-Net Servers

            The initial image is stored in the

            Virtual Box default machine path.

            The result to this call is a temprary file

            which shows the progress (or error state)

            of this call.

    """

    def GET(self):

        log_call(web.ctx.environ)

        global gvm_mgr

        trace_file_name = os.path.join(Environment('OpenSecurity').log_path, 'OpenSecurity_fetch_image.log')

        trace_file = open(trace_file_name, 'w+')

        machine_folder = Cygwin.cygPath(gvm_mgr.getMachineFolder()) 

        download_initial_image_script = Cygwin.cygPath(os.path.abspath(os.path.join(os.path.split(__file__)[0], 'download_initial_image.sh')))

        Cygwin.bashExecute('\\"' + download_initial_image_script + '\\" \'' + machine_folder + '\'', wait_return = False, stdout = trace_file, stderr = trace_file) 

        res = '{ "fetch_log": "' + trace_file_name.replace('\\', '\\\\') + '" }'

        return res

class os_init:

    """OpenSecurity '/init' handler

    - GET: Do initial import of OsecVM.ova

    """

    def GET(self):

        log_call(web.ctx.environ)

        global gvm_mgr

        gvm_mgr.stop()

        gvm_mgr.cleanup()

        if gvm_mgr.vmRootName in gvm_mgr.listVM():

            gvm_mgr.poweroffVM(gvm_mgr.vmRootName)

            tmplateUUID = gvm_mgr.getTemplateUUID()

            if tmplateUUID != None:

                logger.debug('found parent uuid ' + tmplateUUID)

                gvm_mgr.storageDetach(gvm_mgr.vmRootName)

                gvm_mgr.removeSnapshots(tmplateUUID)

                gvm_mgr.removeImage(tmplateUUID)

            else:

                logger.debug('parent uuid not found')

            gvm_mgr.removeVM(gvm_mgr.vmRootName)

        trace_file_name = os.path.join(Environment('OpenSecurity').log_path, 'OpenSecurity_initial_import.log')

        trace_file = open(trace_file_name, 'w+')

        vm_image = Cygwin.cygPath(gvm_mgr.getMachineFolder()) + '/OsecVM.ova'

        initial_import_script = Cygwin.cygPath(os.path.abspath(os.path.join(os.path.split(__file__)[0], 'initial_vm.sh')))

        Cygwin.bashExecute('\\"' + initial_import_script + '\\" \'' + vm_image + '\'', wait_return = False, stdout = trace_file, stderr = trace_file) 

        res = '{ "init_log": "' + trace_file_name.replace('\\', '\\\\') + '" }'

        return res

class os_initial_image:

    """OpenSecurity '/initial_image' handler

    - GET: Return what we have as initial image.

    """

    def GET(self):

        log_call(web.ctx.environ)

        global gvm_mgr

        t = os.path.join(gvm_mgr.systemProperties['Default machine folder'], 'OsecVM.ova')

        res = ''

        if os.path.isfile(t):

            res = '{"initial_template": { '

            res += '"name": "OsecVM.ova", '

            res += '"path": "' + t.replace('\\', '\\\\') + '", '

            res += '"size": ' + str(os.path.getsize(t)) + ', ' 

            res += '"date": ' + str(os.path.getmtime(t)) + ''

            res += '}}'

        return res

class os_root:

    """OpenSecurity '/' handler

    - GET: give information about current installation.

    """

    def GET(self):

        log_call(web.ctx.environ)

        global gvm_mgr

        # create a json string and pretty print it

        res = '{"os_server": { '

        res += '"version": "' + opensecurity.__version__ + '" '

        res += ', "virtual box systemproperties": ' + str(gvm_mgr.systemProperties).replace("'", '"') 

        res += ', "current temporary folder": "' + tempfile.gettempdir().replace('\\', '\\\\') + '"'

        res += ', "current log folder": "' + Environment('OpenSecurity').log_path.replace('\\', '\\\\') + '"'

        try:

            res += ', "whoami": "' + Cygwin.bashExecute('whoami')[1].strip() + '"'

        except:

            res += ', "whoami": "FAILED"'

        try:

            res += ', "mount": ' + str(Cygwin.bashExecute('mount')[1].split('\n')[:-1]).replace("'", '"')

        except:

            res += ', "mount": "FAILED"'

        try:

            res += ', "cygpath --windows ~": "' + Cygwin.bashExecute('cygpath --windows ~')[1].strip().replace('\\', '\\\\') + '"'

        except:

            res += ', "cygpath --windows ~": "FAILED"'

        res += ', "status message": "' + gvm_mgr.status_message.replace('"', "'") + '"'

        res += '}}'

        # loading it into json and print it again ensures

        # we really do have a valid RFC conform json string

        # created (as long as the python json module is RFC conform)

        return json.dumps(json.loads(res), indent = 4)

class os_sdvm:

    """OpenSecurity '/sdvms/[VM]' handler

    - GET: Information about a specific SecurityVM

    - DELETE: Remove a specific

    """

    def GET(self, name):

        log_call(web.ctx.environ)

        global gvm_mgr

        return json.dumps(gvm_mgr.getVMInfo(name), indent = 4)

    def DELETE(self, name):

        log_call(web.ctx.environ)

        global gvm_mgr

        return gvm_mgr.removeVM(name)

class os_sdvm_application:

    """OpenSecurity '/sdvms/[VM]/application/[CMD]' handler

    - GET: start application with given command in the VM.

    """

    def GET(self, name, command):

        log_call(web.ctx.environ)

        global gvm_mgr

        command = '/' + command

        showTrayMessage('Launching application in isolated VM...', 7000)

        result = Cygwin.sshExecuteX11(command, gvm_mgr.getHostOnlyIP(name), 'osecuser', Cygwin.cygPath(gvm_mgr.getMachineFolder()) + '/' + name + '/dvm_key'  )

        return 'Command ' + str(command) + ' started on VM "' + name + '" with IP ' + gvm_mgr.getHostOnlyIP(name)

class os_sdvm_ip:

    """OpenSecurity '/sdvms/[VM]/ip' handler

    - GET: give IP of SecurityVM.

    """

    def GET(self, name):

        log_call(web.ctx.environ)

        global gvm_mgr

        return gvm_mgr.getHostOnlyIP(name)

class os_sdvm_start:

    """OpenSecurity '/sdvms/[VM]/start' handler

    - GET: Start specific SecuirtyVM.

    """

    def GET(self, name):

        log_call(web.ctx.environ)

        global gvm_mgr

        return gvm_mgr.startVM(name)

class os_sdvm_stop:

    """OpenSecurity '/sdvms/[VM]/stop' handler

    - GET: stop specific Secuirty VM.

    """

    def GET(self, name):

        log_call(web.ctx.environ)

        global gvm_mgr

        return gvm_mgr.stopVM(name)

class os_sdvms:

    """OpenSecurity '/sdvms' handler

    - GET: list all available secuirty VMs.

    - POST: create new security vm.

    """

    def GET(self):

        """get the list of SDVMs"""

        log_call(web.ctx.environ)

        global gvm_mgr

        d = {}

        for sdvm in gvm_mgr.listSDVM():

            d[sdvm] = gvm_mgr.getHostOnlyIP(sdvm)

        return json.dumps(d, indent = 4)

    def POST(self):

        """create a new SDVM"""

        log_call(web.ctx.environ)

        global gvm_mgr

        # get a new vm-name

        name = gvm_mgr.generateSDVMName()

        try:

            gvm_mgr.createVM(name)

        except:

            raise web.internalerror()

        return name

class os_setup:

    """OpenSecurity '/setup' handler

    - GET: Give user some info how to setup the OpenSecurity environment

    """

    def GET(self):

        log_call(web.ctx.environ)

        page = """

        <html>

        <body>

        <h1>Setup OpenSecurity</h1>

        In order to setup OpenSecurity an inital VM image has to be downloaded and imported:<br/>

        <ul>

            <li>Download initial VM image: <a href="/fetch_initial_image">fetch_initial_image</a>

            <li>Import initial VM: <a href="/init">init</a>

        </ul>

        </body>

        </html>

        """

        return page

class os_terminate:

    """OpenSecurity '/terminate' handler

    - GET: terminate the opensecurityd.

    TODO: need to find a better way doing this, and not via the

          REST api. Maybe hack web.py server code?

    """

    def GET(self):

        log_call(web.ctx.environ)

        global gvm_mgr

        gvm_mgr.stop()

        gvm_mgr.cleanup()

        global server

        server.stop()

        return None

class os_initialize:

    """OpenSecurity '/initialize' handler

    - GET: initialize / starts the vmmanager.

    """

    def GET(self):

        log_call(web.ctx.environ)

        global gvm_mgr

        gvm_mgr.cleanup()

        gvm_mgr.start()

        global server

        server.run()

        return None

class os_update_template:

    """OpenSecurity '/update_template' handler

    - GET: update template vm

    """

    def GET(self):

        #return gvm_mgr.guestExecute('SecurityDVM', 'sudo apt-get -y update')

        global gvm_mgr

        log_call(web.ctx.environ)

        return gvm_mgr.updateTemplate()

class os_vm:

    """OpenSecurity '/vms/[VM]' handler

    - GET: list information of arbitrary VM.

    """

    def GET(self, name):

        log_call(web.ctx.environ)

        global gvm_mgr

        return gvm_mgr.getVMInfo(name)

class os_vms:

    """OpenSecurity '/vms' handler

    - GET: list all (also non Security) VMs.

    """

    def GET(self):

        log_call(web.ctx.environ)

        global gvm_mgr

        return str(gvm_mgr.listVM()).replace("'",'"')

def log_call(web_environ):

    """log the incoming call to the REST api"""

    try:

        call = 'REST ' +  web_environ['REQUEST_METHOD'] + ' ' + web_environ['REQUEST_URI'] + ' from ' + web_environ['REMOTE_ADDR'] + ':' + web_environ['REMOTE_PORT']

        logger.debug(call)

    except:

        pass

def main():

    """main startup for the opensecuirityd"""

    global gvm_mgr

    global server

    logger.debug('Starting OpenSecurity REST server')

    # ensure a VMManger is yet loaded

    gvm_mgr = vmmanager.VMManager.getInstance()

    gvm_mgr.start()

    # tweak sys.argv to control wep.py server start behavior

    sys.argv = [__file__, "8080"]

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

    server.run()

    logger.debug('Stopped OpenSecurity REST server')

def stop():

    """stop the opensecuirityd"""

    # calling sys.exit() raises a SystemExit exception

    # of the WSGI Server to let it wind down

    # gracefully

    sys.exit(0)

# start

if __name__ == "__main__":

    main()

    sys.exit(0)