2014-07-12 19:18:25 +00:00
|
|
|
# -*- coding: utf-8 -*-
|
|
|
|
#
|
|
|
|
# Copyright (C) 2014 GNS3 Technologies Inc.
|
|
|
|
#
|
|
|
|
# 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/>.
|
|
|
|
|
|
|
|
"""
|
|
|
|
VirtualBox server module.
|
|
|
|
"""
|
|
|
|
|
2014-07-17 21:28:02 +00:00
|
|
|
import sys
|
2014-07-12 19:18:25 +00:00
|
|
|
import os
|
|
|
|
import socket
|
|
|
|
import shutil
|
|
|
|
|
2014-08-22 23:36:12 +00:00
|
|
|
from pkg_resources import parse_version
|
2014-07-12 19:18:25 +00:00
|
|
|
from gns3server.modules import IModule
|
|
|
|
from gns3server.config import Config
|
|
|
|
from .virtualbox_vm import VirtualBoxVM
|
|
|
|
from .virtualbox_error import VirtualBoxError
|
2014-07-17 21:28:02 +00:00
|
|
|
from .vboxwrapper_client import VboxWrapperClient
|
2014-07-12 19:18:25 +00:00
|
|
|
from .nios.nio_udp import NIO_UDP
|
|
|
|
from ..attic import find_unused_port
|
|
|
|
|
2014-07-17 21:28:02 +00:00
|
|
|
from .schemas import VBOX_CREATE_SCHEMA
|
|
|
|
from .schemas import VBOX_DELETE_SCHEMA
|
|
|
|
from .schemas import VBOX_UPDATE_SCHEMA
|
|
|
|
from .schemas import VBOX_START_SCHEMA
|
|
|
|
from .schemas import VBOX_STOP_SCHEMA
|
|
|
|
from .schemas import VBOX_SUSPEND_SCHEMA
|
|
|
|
from .schemas import VBOX_RELOAD_SCHEMA
|
|
|
|
from .schemas import VBOX_ALLOCATE_UDP_PORT_SCHEMA
|
|
|
|
from .schemas import VBOX_ADD_NIO_SCHEMA
|
|
|
|
from .schemas import VBOX_DELETE_NIO_SCHEMA
|
|
|
|
from .schemas import VBOX_START_CAPTURE_SCHEMA
|
|
|
|
from .schemas import VBOX_STOP_CAPTURE_SCHEMA
|
2014-07-12 19:18:25 +00:00
|
|
|
|
|
|
|
import logging
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
|
|
|
|
|
|
class VirtualBox(IModule):
|
|
|
|
"""
|
|
|
|
VirtualBox module.
|
|
|
|
|
|
|
|
:param name: module name
|
|
|
|
:param args: arguments for the module
|
|
|
|
:param kwargs: named arguments for the module
|
|
|
|
"""
|
|
|
|
|
|
|
|
def __init__(self, name, *args, **kwargs):
|
|
|
|
|
2014-08-13 18:11:41 +00:00
|
|
|
# get the vboxwrapper location (only non-Windows platforms)
|
|
|
|
if not sys.platform.startswith("win"):
|
|
|
|
config = Config.instance()
|
|
|
|
vbox_config = config.get_section_config(name.upper())
|
|
|
|
self._vboxwrapper_path = vbox_config.get("vboxwrapper_path")
|
|
|
|
if not self._vboxwrapper_path or not os.path.isfile(self._vboxwrapper_path):
|
|
|
|
paths = [os.getcwd()] + os.environ["PATH"].split(":")
|
2014-09-18 21:47:43 +00:00
|
|
|
# look for vboxwrapper in the current working directory and $PATH
|
2014-08-13 18:11:41 +00:00
|
|
|
for path in paths:
|
|
|
|
try:
|
|
|
|
if "vboxwrapper" in os.listdir(path) and os.access(os.path.join(path, "vboxwrapper"), os.X_OK):
|
|
|
|
self._vboxwrapper_path = os.path.join(path, "vboxwrapper")
|
|
|
|
break
|
|
|
|
except OSError:
|
|
|
|
continue
|
2014-07-17 21:28:02 +00:00
|
|
|
|
2014-08-13 18:11:41 +00:00
|
|
|
if not self._vboxwrapper_path:
|
|
|
|
log.warning("vboxwrapper couldn't be found!")
|
|
|
|
elif not os.access(self._vboxwrapper_path, os.X_OK):
|
|
|
|
log.warning("vboxwrapper is not executable")
|
2014-07-17 21:28:02 +00:00
|
|
|
|
2014-07-12 19:18:25 +00:00
|
|
|
# a new process start when calling IModule
|
|
|
|
IModule.__init__(self, name, *args, **kwargs)
|
|
|
|
self._vbox_instances = {}
|
|
|
|
|
|
|
|
config = Config.instance()
|
|
|
|
vbox_config = config.get_section_config(name.upper())
|
|
|
|
self._console_start_port_range = vbox_config.get("console_start_port_range", 3501)
|
|
|
|
self._console_end_port_range = vbox_config.get("console_end_port_range", 4000)
|
|
|
|
self._allocated_udp_ports = []
|
|
|
|
self._udp_start_port_range = vbox_config.get("udp_start_port_range", 35001)
|
|
|
|
self._udp_end_port_range = vbox_config.get("udp_end_port_range", 35500)
|
2014-08-25 21:40:04 +00:00
|
|
|
self._host = vbox_config.get("host", kwargs["host"])
|
2014-07-12 19:18:25 +00:00
|
|
|
self._projects_dir = kwargs["projects_dir"]
|
|
|
|
self._tempdir = kwargs["temp_dir"]
|
|
|
|
self._working_dir = self._projects_dir
|
2014-07-17 21:28:02 +00:00
|
|
|
self._vboxmanager = None
|
|
|
|
self._vboxwrapper = None
|
|
|
|
|
|
|
|
def _start_vbox_service(self):
|
|
|
|
"""
|
|
|
|
Starts the VirtualBox backend.
|
|
|
|
vboxapi on Windows or vboxwrapper on other platforms.
|
|
|
|
"""
|
|
|
|
|
|
|
|
if sys.platform.startswith("win"):
|
2014-08-22 23:36:12 +00:00
|
|
|
import pywintypes
|
2014-07-17 21:28:02 +00:00
|
|
|
import win32com.client
|
2014-08-22 00:13:41 +00:00
|
|
|
|
|
|
|
try:
|
2014-08-22 23:36:12 +00:00
|
|
|
if win32com.client.gencache.is_readonly is True:
|
|
|
|
# dynamically generate the cache
|
|
|
|
# http://www.py2exe.org/index.cgi/IncludingTypelibs
|
|
|
|
# http://www.py2exe.org/index.cgi/UsingEnsureDispatch
|
|
|
|
win32com.client.gencache.is_readonly = False
|
|
|
|
#win32com.client.gencache.Rebuild()
|
|
|
|
win32com.client.gencache.GetGeneratePath()
|
|
|
|
|
2014-08-22 00:13:41 +00:00
|
|
|
win32com.client.gencache.EnsureDispatch("VirtualBox.VirtualBox")
|
2014-08-22 23:36:12 +00:00
|
|
|
except pywintypes.com_error:
|
2014-08-22 00:13:41 +00:00
|
|
|
raise VirtualBoxError("VirtualBox is not installed.")
|
|
|
|
|
2014-07-17 21:28:02 +00:00
|
|
|
try:
|
|
|
|
from .vboxapi_py3 import VirtualBoxManager
|
|
|
|
self._vboxmanager = VirtualBoxManager(None, None)
|
2014-08-22 23:36:12 +00:00
|
|
|
vbox_major_version, vbox_minor_version, _ = self._vboxmanager.vbox.version.split('.')
|
|
|
|
if parse_version("{}.{}".format(vbox_major_version, vbox_minor_version)) <= parse_version("4.1"):
|
|
|
|
raise VirtualBoxError("VirtualBox version must be >= 4.2")
|
2014-07-17 21:28:02 +00:00
|
|
|
except Exception as e:
|
2014-08-22 23:36:12 +00:00
|
|
|
self._vboxmanager = None
|
2014-07-17 21:28:02 +00:00
|
|
|
raise VirtualBoxError("Could not initialize the VirtualBox Manager: {}".format(e))
|
|
|
|
|
|
|
|
log.info("VirtualBox Manager has successful started: version is {} r{}".format(self._vboxmanager.vbox.version,
|
|
|
|
self._vboxmanager.vbox.revision))
|
|
|
|
else:
|
|
|
|
|
|
|
|
if not self._vboxwrapper_path:
|
|
|
|
raise VirtualBoxError("No vboxwrapper path has been configured")
|
|
|
|
|
|
|
|
if not os.path.isfile(self._vboxwrapper_path):
|
|
|
|
raise VirtualBoxError("vboxwrapper path doesn't exist {}".format(self._vboxwrapper_path))
|
|
|
|
|
|
|
|
self._vboxwrapper = VboxWrapperClient(self._vboxwrapper_path, self._tempdir, "127.0.0.1")
|
|
|
|
#self._vboxwrapper.connect()
|
2014-08-22 23:36:12 +00:00
|
|
|
try:
|
|
|
|
self._vboxwrapper.start()
|
|
|
|
except VirtualBoxError:
|
|
|
|
self._vboxwrapper = None
|
|
|
|
raise
|
2014-07-12 19:18:25 +00:00
|
|
|
|
|
|
|
def stop(self, signum=None):
|
|
|
|
"""
|
|
|
|
Properly stops the module.
|
|
|
|
|
|
|
|
:param signum: signal number (if called by the signal handler)
|
|
|
|
"""
|
|
|
|
|
|
|
|
# delete all VirtualBox instances
|
|
|
|
for vbox_id in self._vbox_instances:
|
|
|
|
vbox_instance = self._vbox_instances[vbox_id]
|
|
|
|
vbox_instance.delete()
|
|
|
|
|
2014-07-17 21:28:02 +00:00
|
|
|
if self._vboxwrapper and self._vboxwrapper.started:
|
|
|
|
self._vboxwrapper.stop()
|
|
|
|
|
2014-07-12 19:18:25 +00:00
|
|
|
IModule.stop(self, signum) # this will stop the I/O loop
|
|
|
|
|
|
|
|
def get_vbox_instance(self, vbox_id):
|
|
|
|
"""
|
|
|
|
Returns a VirtualBox VM instance.
|
|
|
|
|
2014-09-18 21:47:43 +00:00
|
|
|
:param vbox_id: VirtualBox VM identifier
|
2014-07-12 19:18:25 +00:00
|
|
|
|
2014-09-18 21:47:43 +00:00
|
|
|
:returns: VirtualBoxVM instance
|
2014-07-12 19:18:25 +00:00
|
|
|
"""
|
|
|
|
|
|
|
|
if vbox_id not in self._vbox_instances:
|
|
|
|
log.debug("VirtualBox VM ID {} doesn't exist".format(vbox_id), exc_info=1)
|
|
|
|
self.send_custom_error("VirtualBox VM ID {} doesn't exist".format(vbox_id))
|
|
|
|
return None
|
|
|
|
return self._vbox_instances[vbox_id]
|
|
|
|
|
|
|
|
@IModule.route("virtualbox.reset")
|
|
|
|
def reset(self, request):
|
|
|
|
"""
|
|
|
|
Resets the module.
|
|
|
|
|
|
|
|
:param request: JSON request
|
|
|
|
"""
|
|
|
|
|
|
|
|
# delete all VirtualBox instances
|
|
|
|
for vbox_id in self._vbox_instances:
|
|
|
|
vbox_instance = self._vbox_instances[vbox_id]
|
|
|
|
vbox_instance.delete()
|
|
|
|
|
|
|
|
# resets the instance IDs
|
|
|
|
VirtualBoxVM.reset()
|
|
|
|
|
|
|
|
self._vbox_instances.clear()
|
|
|
|
self._allocated_udp_ports.clear()
|
|
|
|
|
2014-07-17 21:28:02 +00:00
|
|
|
if self._vboxwrapper and self._vboxwrapper.connected():
|
|
|
|
self._vboxwrapper.send("vboxwrapper reset")
|
|
|
|
|
2014-07-12 19:18:25 +00:00
|
|
|
log.info("VirtualBox module has been reset")
|
|
|
|
|
|
|
|
@IModule.route("virtualbox.settings")
|
|
|
|
def settings(self, request):
|
|
|
|
"""
|
|
|
|
Set or update settings.
|
|
|
|
|
|
|
|
Optional request parameters:
|
|
|
|
- working_dir (path to a working directory)
|
2014-07-17 21:28:02 +00:00
|
|
|
- vboxwrapper_path (path to vboxwrapper)
|
2014-07-12 19:18:25 +00:00
|
|
|
- project_name
|
|
|
|
- console_start_port_range
|
|
|
|
- console_end_port_range
|
|
|
|
- udp_start_port_range
|
|
|
|
- udp_end_port_range
|
|
|
|
|
|
|
|
:param request: JSON request
|
|
|
|
"""
|
|
|
|
|
|
|
|
if request is None:
|
|
|
|
self.send_param_error()
|
|
|
|
return
|
|
|
|
|
|
|
|
if "working_dir" in request:
|
|
|
|
new_working_dir = request["working_dir"]
|
|
|
|
log.info("this server is local with working directory path to {}".format(new_working_dir))
|
|
|
|
else:
|
|
|
|
new_working_dir = os.path.join(self._projects_dir, request["project_name"])
|
|
|
|
log.info("this server is remote with working directory path to {}".format(new_working_dir))
|
|
|
|
if self._projects_dir != self._working_dir != new_working_dir:
|
|
|
|
if not os.path.isdir(new_working_dir):
|
|
|
|
try:
|
|
|
|
shutil.move(self._working_dir, new_working_dir)
|
|
|
|
except OSError as e:
|
|
|
|
log.error("could not move working directory from {} to {}: {}".format(self._working_dir,
|
|
|
|
new_working_dir,
|
|
|
|
e))
|
|
|
|
return
|
|
|
|
|
|
|
|
# update the working directory if it has changed
|
|
|
|
if self._working_dir != new_working_dir:
|
|
|
|
self._working_dir = new_working_dir
|
|
|
|
for vbox_id in self._vbox_instances:
|
|
|
|
vbox_instance = self._vbox_instances[vbox_id]
|
|
|
|
vbox_instance.working_dir = os.path.join(self._working_dir, "vbox", "vm-{}".format(vbox_instance.id))
|
|
|
|
|
2014-07-17 21:28:02 +00:00
|
|
|
if "vboxwrapper_path" in request:
|
|
|
|
self._vboxwrapper_path = request["vboxwrapper_path"]
|
|
|
|
|
2014-07-12 19:18:25 +00:00
|
|
|
if "console_start_port_range" in request and "console_end_port_range" in request:
|
|
|
|
self._console_start_port_range = request["console_start_port_range"]
|
|
|
|
self._console_end_port_range = request["console_end_port_range"]
|
|
|
|
|
|
|
|
if "udp_start_port_range" in request and "udp_end_port_range" in request:
|
|
|
|
self._udp_start_port_range = request["udp_start_port_range"]
|
|
|
|
self._udp_end_port_range = request["udp_end_port_range"]
|
|
|
|
|
|
|
|
log.debug("received request {}".format(request))
|
|
|
|
|
|
|
|
@IModule.route("virtualbox.create")
|
|
|
|
def vbox_create(self, request):
|
|
|
|
"""
|
|
|
|
Creates a new VirtualBox VM instance.
|
|
|
|
|
|
|
|
Mandatory request parameters:
|
|
|
|
- name (VirtualBox VM name)
|
2014-09-18 21:47:43 +00:00
|
|
|
- vmname (VirtualBox VM name in VirtualBox)
|
2014-07-12 19:18:25 +00:00
|
|
|
|
|
|
|
Optional request parameters:
|
|
|
|
- console (VirtualBox VM console port)
|
|
|
|
|
|
|
|
Response parameters:
|
|
|
|
- id (VirtualBox VM instance identifier)
|
|
|
|
- name (VirtualBox VM name)
|
|
|
|
- default settings
|
|
|
|
|
|
|
|
:param request: JSON request
|
|
|
|
"""
|
|
|
|
|
|
|
|
# validate the request
|
2014-07-17 21:28:02 +00:00
|
|
|
if not self.validate_request(request, VBOX_CREATE_SCHEMA):
|
|
|
|
return
|
2014-07-12 19:18:25 +00:00
|
|
|
|
|
|
|
name = request["name"]
|
2014-07-17 21:28:02 +00:00
|
|
|
vmname = request["vmname"]
|
2014-07-12 19:18:25 +00:00
|
|
|
console = request.get("console")
|
|
|
|
vbox_id = request.get("vbox_id")
|
|
|
|
|
|
|
|
try:
|
|
|
|
|
2014-07-17 21:28:02 +00:00
|
|
|
if not self._vboxwrapper and not self._vboxmanager:
|
|
|
|
self._start_vbox_service()
|
|
|
|
|
|
|
|
vbox_instance = VirtualBoxVM(self._vboxwrapper,
|
|
|
|
self._vboxmanager,
|
|
|
|
name,
|
|
|
|
vmname,
|
2014-07-12 19:18:25 +00:00
|
|
|
self._working_dir,
|
|
|
|
self._host,
|
|
|
|
vbox_id,
|
|
|
|
console,
|
|
|
|
self._console_start_port_range,
|
|
|
|
self._console_end_port_range)
|
|
|
|
|
|
|
|
except VirtualBoxError as e:
|
|
|
|
self.send_custom_error(str(e))
|
|
|
|
return
|
|
|
|
|
|
|
|
response = {"name": vbox_instance.name,
|
|
|
|
"id": vbox_instance.id}
|
|
|
|
|
|
|
|
defaults = vbox_instance.defaults()
|
|
|
|
response.update(defaults)
|
|
|
|
self._vbox_instances[vbox_instance.id] = vbox_instance
|
|
|
|
self.send_response(response)
|
|
|
|
|
|
|
|
@IModule.route("virtualbox.delete")
|
|
|
|
def vbox_delete(self, request):
|
|
|
|
"""
|
|
|
|
Deletes a VirtualBox VM instance.
|
|
|
|
|
|
|
|
Mandatory request parameters:
|
|
|
|
- id (VirtualBox VM instance identifier)
|
|
|
|
|
|
|
|
Response parameter:
|
|
|
|
- True on success
|
|
|
|
|
|
|
|
:param request: JSON request
|
|
|
|
"""
|
|
|
|
|
|
|
|
# validate the request
|
2014-07-17 21:28:02 +00:00
|
|
|
if not self.validate_request(request, VBOX_DELETE_SCHEMA):
|
|
|
|
return
|
2014-07-12 19:18:25 +00:00
|
|
|
|
|
|
|
# get the instance
|
|
|
|
vbox_instance = self.get_vbox_instance(request["id"])
|
|
|
|
if not vbox_instance:
|
|
|
|
return
|
|
|
|
|
|
|
|
try:
|
|
|
|
vbox_instance.clean_delete()
|
|
|
|
del self._vbox_instances[request["id"]]
|
|
|
|
except VirtualBoxError as e:
|
|
|
|
self.send_custom_error(str(e))
|
|
|
|
return
|
|
|
|
|
|
|
|
self.send_response(True)
|
|
|
|
|
|
|
|
@IModule.route("virtualbox.update")
|
|
|
|
def vbox_update(self, request):
|
|
|
|
"""
|
|
|
|
Updates a VirtualBox VM instance
|
|
|
|
|
|
|
|
Mandatory request parameters:
|
|
|
|
- id (VirtualBox VM instance identifier)
|
|
|
|
|
|
|
|
Optional request parameters:
|
|
|
|
- any setting to update
|
|
|
|
|
|
|
|
Response parameters:
|
|
|
|
- updated settings
|
|
|
|
|
|
|
|
:param request: JSON request
|
|
|
|
"""
|
|
|
|
|
|
|
|
# validate the request
|
2014-07-17 21:28:02 +00:00
|
|
|
if not self.validate_request(request, VBOX_UPDATE_SCHEMA):
|
|
|
|
return
|
2014-07-12 19:18:25 +00:00
|
|
|
|
|
|
|
# get the instance
|
|
|
|
vbox_instance = self.get_vbox_instance(request["id"])
|
|
|
|
if not vbox_instance:
|
|
|
|
return
|
|
|
|
|
|
|
|
# update the VirtualBox VM settings
|
|
|
|
response = {}
|
|
|
|
for name, value in request.items():
|
|
|
|
if hasattr(vbox_instance, name) and getattr(vbox_instance, name) != value:
|
|
|
|
try:
|
|
|
|
setattr(vbox_instance, name, value)
|
|
|
|
response[name] = value
|
|
|
|
except VirtualBoxError as e:
|
|
|
|
self.send_custom_error(str(e))
|
|
|
|
return
|
|
|
|
|
|
|
|
self.send_response(response)
|
|
|
|
|
|
|
|
@IModule.route("virtualbox.start")
|
|
|
|
def vbox_start(self, request):
|
|
|
|
"""
|
|
|
|
Starts a VirtualBox VM instance.
|
|
|
|
|
|
|
|
Mandatory request parameters:
|
|
|
|
- id (VirtualBox VM instance identifier)
|
|
|
|
|
|
|
|
Response parameters:
|
|
|
|
- True on success
|
|
|
|
|
|
|
|
:param request: JSON request
|
|
|
|
"""
|
|
|
|
|
|
|
|
# validate the request
|
2014-07-17 21:28:02 +00:00
|
|
|
if not self.validate_request(request, VBOX_START_SCHEMA):
|
|
|
|
return
|
2014-07-12 19:18:25 +00:00
|
|
|
|
|
|
|
# get the instance
|
|
|
|
vbox_instance = self.get_vbox_instance(request["id"])
|
|
|
|
if not vbox_instance:
|
|
|
|
return
|
|
|
|
|
|
|
|
try:
|
|
|
|
vbox_instance.start()
|
|
|
|
except VirtualBoxError as e:
|
2014-08-26 21:27:43 +00:00
|
|
|
if self._vboxwrapper:
|
|
|
|
self.send_custom_error("{}: {}".format(e, self._vboxwrapper.read_stderr()))
|
|
|
|
else:
|
|
|
|
self.send_custom_error(str(e))
|
2014-07-12 19:18:25 +00:00
|
|
|
return
|
|
|
|
self.send_response(True)
|
|
|
|
|
|
|
|
@IModule.route("virtualbox.stop")
|
|
|
|
def vbox_stop(self, request):
|
|
|
|
"""
|
|
|
|
Stops a VirtualBox VM instance.
|
|
|
|
|
|
|
|
Mandatory request parameters:
|
|
|
|
- id (VirtualBox VM instance identifier)
|
|
|
|
|
|
|
|
Response parameters:
|
|
|
|
- True on success
|
|
|
|
|
|
|
|
:param request: JSON request
|
|
|
|
"""
|
|
|
|
|
|
|
|
# validate the request
|
2014-07-17 21:28:02 +00:00
|
|
|
if not self.validate_request(request, VBOX_STOP_SCHEMA):
|
|
|
|
return
|
2014-07-12 19:18:25 +00:00
|
|
|
|
|
|
|
# get the instance
|
|
|
|
vbox_instance = self.get_vbox_instance(request["id"])
|
|
|
|
if not vbox_instance:
|
|
|
|
return
|
|
|
|
|
|
|
|
try:
|
|
|
|
vbox_instance.stop()
|
|
|
|
except VirtualBoxError as e:
|
|
|
|
self.send_custom_error(str(e))
|
|
|
|
return
|
|
|
|
self.send_response(True)
|
|
|
|
|
|
|
|
@IModule.route("virtualbox.reload")
|
|
|
|
def vbox_reload(self, request):
|
|
|
|
"""
|
|
|
|
Reloads a VirtualBox VM instance.
|
|
|
|
|
|
|
|
Mandatory request parameters:
|
|
|
|
- id (VirtualBox VM identifier)
|
|
|
|
|
|
|
|
Response parameters:
|
|
|
|
- True on success
|
|
|
|
|
|
|
|
:param request: JSON request
|
|
|
|
"""
|
|
|
|
|
|
|
|
# validate the request
|
2014-07-17 21:28:02 +00:00
|
|
|
if not self.validate_request(request, VBOX_RELOAD_SCHEMA):
|
|
|
|
return
|
2014-07-12 19:18:25 +00:00
|
|
|
|
|
|
|
# get the instance
|
2014-07-17 21:28:02 +00:00
|
|
|
vbox_instance = self.get_vbox_instance(request["id"])
|
2014-07-12 19:18:25 +00:00
|
|
|
if not vbox_instance:
|
|
|
|
return
|
|
|
|
|
|
|
|
try:
|
2014-07-17 21:28:02 +00:00
|
|
|
vbox_instance.reload()
|
|
|
|
except VirtualBoxError as e:
|
|
|
|
self.send_custom_error(str(e))
|
|
|
|
return
|
|
|
|
self.send_response(True)
|
|
|
|
|
|
|
|
@IModule.route("virtualbox.stop")
|
|
|
|
def vbox_stop(self, request):
|
|
|
|
"""
|
|
|
|
Stops a VirtualBox VM instance.
|
|
|
|
|
|
|
|
Mandatory request parameters:
|
|
|
|
- id (VirtualBox VM instance identifier)
|
|
|
|
|
|
|
|
Response parameters:
|
|
|
|
- True on success
|
|
|
|
|
|
|
|
:param request: JSON request
|
|
|
|
"""
|
|
|
|
|
|
|
|
# validate the request
|
|
|
|
if not self.validate_request(request, VBOX_STOP_SCHEMA):
|
|
|
|
return
|
|
|
|
|
|
|
|
# get the instance
|
|
|
|
vbox_instance = self.get_vbox_instance(request["id"])
|
|
|
|
if not vbox_instance:
|
|
|
|
return
|
|
|
|
|
|
|
|
try:
|
|
|
|
vbox_instance.stop()
|
|
|
|
except VirtualBoxError as e:
|
|
|
|
self.send_custom_error(str(e))
|
|
|
|
return
|
|
|
|
self.send_response(True)
|
|
|
|
|
|
|
|
@IModule.route("virtualbox.suspend")
|
|
|
|
def vbox_suspend(self, request):
|
|
|
|
"""
|
|
|
|
Suspends a VirtualBox VM instance.
|
|
|
|
|
|
|
|
Mandatory request parameters:
|
|
|
|
- id (VirtualBox VM instance identifier)
|
|
|
|
|
|
|
|
Response parameters:
|
|
|
|
- True on success
|
|
|
|
|
|
|
|
:param request: JSON request
|
|
|
|
"""
|
|
|
|
|
|
|
|
# validate the request
|
|
|
|
if not self.validate_request(request, VBOX_SUSPEND_SCHEMA):
|
|
|
|
return
|
|
|
|
|
|
|
|
# get the instance
|
|
|
|
vbox_instance = self.get_vbox_instance(request["id"])
|
|
|
|
if not vbox_instance:
|
|
|
|
return
|
|
|
|
|
|
|
|
try:
|
|
|
|
vbox_instance.suspend()
|
2014-07-12 19:18:25 +00:00
|
|
|
except VirtualBoxError as e:
|
|
|
|
self.send_custom_error(str(e))
|
|
|
|
return
|
|
|
|
self.send_response(True)
|
|
|
|
|
|
|
|
@IModule.route("virtualbox.allocate_udp_port")
|
|
|
|
def allocate_udp_port(self, request):
|
|
|
|
"""
|
|
|
|
Allocates a UDP port in order to create an UDP NIO.
|
|
|
|
|
|
|
|
Mandatory request parameters:
|
|
|
|
- id (VirtualBox VM identifier)
|
|
|
|
- port_id (unique port identifier)
|
|
|
|
|
|
|
|
Response parameters:
|
|
|
|
- port_id (unique port identifier)
|
|
|
|
- lport (allocated local port)
|
|
|
|
|
|
|
|
:param request: JSON request
|
|
|
|
"""
|
|
|
|
|
|
|
|
# validate the request
|
2014-07-17 21:28:02 +00:00
|
|
|
if not self.validate_request(request, VBOX_ALLOCATE_UDP_PORT_SCHEMA):
|
|
|
|
return
|
2014-07-12 19:18:25 +00:00
|
|
|
|
|
|
|
# get the instance
|
|
|
|
vbox_instance = self.get_vbox_instance(request["id"])
|
|
|
|
if not vbox_instance:
|
|
|
|
return
|
|
|
|
|
|
|
|
try:
|
|
|
|
port = find_unused_port(self._udp_start_port_range,
|
|
|
|
self._udp_end_port_range,
|
|
|
|
host=self._host,
|
|
|
|
socket_type="UDP",
|
|
|
|
ignore_ports=self._allocated_udp_ports)
|
|
|
|
except Exception as e:
|
|
|
|
self.send_custom_error(str(e))
|
|
|
|
return
|
|
|
|
|
|
|
|
self._allocated_udp_ports.append(port)
|
|
|
|
log.info("{} [id={}] has allocated UDP port {} with host {}".format(vbox_instance.name,
|
|
|
|
vbox_instance.id,
|
|
|
|
port,
|
|
|
|
self._host))
|
|
|
|
|
|
|
|
response = {"lport": port,
|
|
|
|
"port_id": request["port_id"]}
|
|
|
|
self.send_response(response)
|
|
|
|
|
|
|
|
@IModule.route("virtualbox.add_nio")
|
|
|
|
def add_nio(self, request):
|
|
|
|
"""
|
|
|
|
Adds an NIO (Network Input/Output) for a VirtualBox VM instance.
|
|
|
|
|
|
|
|
Mandatory request parameters:
|
|
|
|
- id (VirtualBox VM instance identifier)
|
|
|
|
- port (port number)
|
|
|
|
- port_id (unique port identifier)
|
|
|
|
- nio (one of the following)
|
|
|
|
- type "nio_udp"
|
|
|
|
- lport (local port)
|
|
|
|
- rhost (remote host)
|
|
|
|
- rport (remote port)
|
|
|
|
|
|
|
|
Response parameters:
|
|
|
|
- port_id (unique port identifier)
|
|
|
|
|
|
|
|
:param request: JSON request
|
|
|
|
"""
|
|
|
|
|
|
|
|
# validate the request
|
2014-07-17 21:28:02 +00:00
|
|
|
if not self.validate_request(request, VBOX_ADD_NIO_SCHEMA):
|
|
|
|
return
|
2014-07-12 19:18:25 +00:00
|
|
|
|
|
|
|
# get the instance
|
|
|
|
vbox_instance = self.get_vbox_instance(request["id"])
|
|
|
|
if not vbox_instance:
|
|
|
|
return
|
|
|
|
|
|
|
|
port = request["port"]
|
|
|
|
try:
|
|
|
|
nio = None
|
|
|
|
if request["nio"]["type"] == "nio_udp":
|
|
|
|
lport = request["nio"]["lport"]
|
|
|
|
rhost = request["nio"]["rhost"]
|
|
|
|
rport = request["nio"]["rport"]
|
|
|
|
try:
|
|
|
|
#TODO: handle IPv6
|
|
|
|
with socket.socket(socket.AF_INET, socket.SOCK_DGRAM) as sock:
|
|
|
|
sock.connect((rhost, rport))
|
|
|
|
except OSError as e:
|
|
|
|
raise VirtualBoxError("Could not create an UDP connection to {}:{}: {}".format(rhost, rport, e))
|
|
|
|
nio = NIO_UDP(lport, rhost, rport)
|
|
|
|
if not nio:
|
|
|
|
raise VirtualBoxError("Requested NIO does not exist or is not supported: {}".format(request["nio"]["type"]))
|
|
|
|
except VirtualBoxError as e:
|
|
|
|
self.send_custom_error(str(e))
|
|
|
|
return
|
|
|
|
|
|
|
|
try:
|
|
|
|
vbox_instance.port_add_nio_binding(port, nio)
|
|
|
|
except VirtualBoxError as e:
|
|
|
|
self.send_custom_error(str(e))
|
|
|
|
return
|
|
|
|
|
|
|
|
self.send_response({"port_id": request["port_id"]})
|
|
|
|
|
|
|
|
@IModule.route("virtualbox.delete_nio")
|
|
|
|
def delete_nio(self, request):
|
|
|
|
"""
|
|
|
|
Deletes an NIO (Network Input/Output).
|
|
|
|
|
|
|
|
Mandatory request parameters:
|
2014-09-18 21:47:43 +00:00
|
|
|
- id (VirtualBox instance identifier)
|
2014-07-12 19:18:25 +00:00
|
|
|
- port (port identifier)
|
|
|
|
|
|
|
|
Response parameters:
|
|
|
|
- True on success
|
|
|
|
|
|
|
|
:param request: JSON request
|
|
|
|
"""
|
|
|
|
|
|
|
|
# validate the request
|
2014-07-17 21:28:02 +00:00
|
|
|
if not self.validate_request(request, VBOX_DELETE_NIO_SCHEMA):
|
|
|
|
return
|
2014-07-12 19:18:25 +00:00
|
|
|
|
|
|
|
# get the instance
|
|
|
|
vbox_instance = self.get_vbox_instance(request["id"])
|
|
|
|
if not vbox_instance:
|
|
|
|
return
|
|
|
|
|
|
|
|
port = request["port"]
|
|
|
|
try:
|
|
|
|
nio = vbox_instance.port_remove_nio_binding(port)
|
|
|
|
if isinstance(nio, NIO_UDP) and nio.lport in self._allocated_udp_ports:
|
|
|
|
self._allocated_udp_ports.remove(nio.lport)
|
|
|
|
except VirtualBoxError as e:
|
|
|
|
self.send_custom_error(str(e))
|
|
|
|
return
|
|
|
|
|
|
|
|
self.send_response(True)
|
|
|
|
|
2014-07-17 21:28:02 +00:00
|
|
|
@IModule.route("virtualbox.start_capture")
|
|
|
|
def vbox_start_capture(self, request):
|
|
|
|
"""
|
|
|
|
Starts a packet capture.
|
|
|
|
|
|
|
|
Mandatory request parameters:
|
2014-09-18 21:47:43 +00:00
|
|
|
- id (VirtualBox VM identifier)
|
2014-07-17 21:28:02 +00:00
|
|
|
- port (port number)
|
|
|
|
- port_id (port identifier)
|
|
|
|
- capture_file_name
|
|
|
|
|
|
|
|
Response parameters:
|
|
|
|
- port_id (port identifier)
|
|
|
|
- capture_file_path (path to the capture file)
|
|
|
|
|
|
|
|
:param request: JSON request
|
|
|
|
"""
|
|
|
|
|
|
|
|
# validate the request
|
|
|
|
if not self.validate_request(request, VBOX_START_CAPTURE_SCHEMA):
|
|
|
|
return
|
|
|
|
|
|
|
|
# get the instance
|
|
|
|
vbox_instance = self.get_vbox_instance(request["id"])
|
|
|
|
if not vbox_instance:
|
|
|
|
return
|
|
|
|
|
|
|
|
port = request["port"]
|
|
|
|
capture_file_name = request["capture_file_name"]
|
|
|
|
|
|
|
|
try:
|
|
|
|
capture_file_path = os.path.join(self._working_dir, "captures", capture_file_name)
|
|
|
|
vbox_instance.start_capture(port, capture_file_path)
|
|
|
|
except VirtualBoxError as e:
|
|
|
|
self.send_custom_error(str(e))
|
|
|
|
return
|
|
|
|
|
|
|
|
response = {"port_id": request["port_id"],
|
|
|
|
"capture_file_path": capture_file_path}
|
|
|
|
self.send_response(response)
|
|
|
|
|
|
|
|
@IModule.route("virtualbox.stop_capture")
|
|
|
|
def vbox_stop_capture(self, request):
|
|
|
|
"""
|
|
|
|
Stops a packet capture.
|
|
|
|
|
|
|
|
Mandatory request parameters:
|
2014-09-18 21:47:43 +00:00
|
|
|
- id (VirtualBox VM identifier)
|
2014-07-17 21:28:02 +00:00
|
|
|
- port (port number)
|
|
|
|
- port_id (port identifier)
|
|
|
|
|
|
|
|
Response parameters:
|
|
|
|
- port_id (port identifier)
|
|
|
|
|
|
|
|
:param request: JSON request
|
|
|
|
"""
|
|
|
|
|
|
|
|
# validate the request
|
|
|
|
if not self.validate_request(request, VBOX_STOP_CAPTURE_SCHEMA):
|
|
|
|
return
|
|
|
|
|
|
|
|
# get the instance
|
|
|
|
vbox_instance = self.get_vbox_instance(request["id"])
|
|
|
|
if not vbox_instance:
|
|
|
|
return
|
|
|
|
|
|
|
|
port = request["port"]
|
|
|
|
try:
|
|
|
|
vbox_instance.stop_capture(port)
|
|
|
|
except VirtualBoxError as e:
|
|
|
|
self.send_custom_error(str(e))
|
|
|
|
return
|
|
|
|
|
|
|
|
response = {"port_id": request["port_id"]}
|
|
|
|
self.send_response(response)
|
|
|
|
|
|
|
|
@IModule.route("virtualbox.vm_list")
|
|
|
|
def vm_list(self, request):
|
|
|
|
"""
|
|
|
|
Gets VirtualBox VM list.
|
|
|
|
|
|
|
|
Response parameters:
|
|
|
|
- Server address/host
|
|
|
|
- List of VM names
|
|
|
|
"""
|
|
|
|
|
|
|
|
if not self._vboxwrapper and not self._vboxmanager:
|
2014-08-22 00:13:41 +00:00
|
|
|
try:
|
|
|
|
self._start_vbox_service()
|
|
|
|
except VirtualBoxError as e:
|
|
|
|
self.send_custom_error(str(e))
|
|
|
|
return
|
2014-07-17 21:28:02 +00:00
|
|
|
|
|
|
|
if self._vboxwrapper:
|
|
|
|
vms = self._vboxwrapper.get_vm_list()
|
|
|
|
elif self._vboxmanager:
|
|
|
|
vms = []
|
|
|
|
machines = self._vboxmanager.getArray(self._vboxmanager.vbox, "machines")
|
|
|
|
for machine in range(len(machines)):
|
|
|
|
vms.append(machines[machine].name)
|
|
|
|
else:
|
|
|
|
self.send_custom_error("Vboxmanager hasn't been initialized!")
|
|
|
|
return
|
|
|
|
|
|
|
|
response = {"server": self._host,
|
|
|
|
"vms": vms}
|
|
|
|
self.send_response(response)
|
|
|
|
|
2014-07-12 19:18:25 +00:00
|
|
|
@IModule.route("virtualbox.echo")
|
|
|
|
def echo(self, request):
|
|
|
|
"""
|
|
|
|
Echo end point for testing purposes.
|
|
|
|
|
|
|
|
:param request: JSON request
|
|
|
|
"""
|
|
|
|
|
|
|
|
if request is None:
|
|
|
|
self.send_param_error()
|
|
|
|
else:
|
|
|
|
log.debug("received request {}".format(request))
|
|
|
|
self.send_response(request)
|