2014-07-12 22:18:25 +03: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 VM instance .
"""
2014-10-31 02:53:17 +02:00
import sys
import shlex
import re
2014-07-12 22:18:25 +03:00
import os
2014-10-31 02:53:17 +02:00
import tempfile
2014-11-09 20:50:47 +02:00
import json
2014-10-31 02:53:17 +02:00
import socket
2015-01-20 03:30:57 +02:00
import asyncio
2016-10-28 17:00:26 +03:00
import xml . etree . ElementTree as ET
2014-07-12 22:18:25 +03:00
2016-05-02 18:13:23 +03:00
from gns3server . utils import parse_version
2016-11-07 12:16:51 +02:00
from gns3server . utils . asyncio . telnet_server import AsyncioTelnetServer
from gns3server . utils . asyncio . serial import asyncio_open_serial
from gns3server . utils . asyncio import locked_coroutine
from gns3server . compute . virtualbox . virtualbox_error import VirtualBoxError
from gns3server . compute . nios . nio_udp import NIOUDP
from gns3server . compute . adapters . ethernet_adapter import EthernetAdapter
from gns3server . compute . base_node import BaseNode
2014-10-31 02:53:17 +02:00
if sys . platform . startswith ( ' win ' ) :
import msvcrt
import win32file
2014-07-12 22:18:25 +03:00
import logging
log = logging . getLogger ( __name__ )
2014-07-18 05:02:18 +03:00
2016-05-11 20:35:36 +03:00
class VirtualBoxVM ( BaseNode ) :
2015-01-31 23:34:49 +02:00
2014-07-12 22:18:25 +03:00
"""
VirtualBox VM implementation .
"""
2016-10-24 22:39:35 +03:00
def __init__ ( self , name , node_id , project , manager , vmname , linked_clone = False , console = None , adapters = 0 ) :
2015-01-20 03:30:57 +02:00
2016-11-07 12:16:51 +02:00
super ( ) . __init__ ( name , node_id , project , manager , console = console , linked_clone = linked_clone , console_type = " telnet " )
2015-01-20 03:30:57 +02:00
2015-01-22 04:26:39 +02:00
self . _maximum_adapters = 8
2015-01-21 04:02:22 +02:00
self . _system_properties = { }
2016-11-07 12:16:51 +02:00
self . _telnet_server = None
2016-06-25 03:35:39 +03:00
self . _local_udp_tunnels = { }
2015-01-20 03:30:57 +02:00
2014-07-12 22:18:25 +03:00
# VirtualBox settings
2015-01-31 04:36:05 +02:00
self . _adapters = adapters
2015-04-03 13:08:18 +03:00
self . _ethernet_adapters = { }
2014-07-18 00:28:02 +03:00
self . _headless = False
2015-06-03 01:30:35 +03:00
self . _acpi_shutdown = False
2014-07-18 00:28:02 +03:00
self . _vmname = vmname
2015-02-07 02:31:13 +02:00
self . _use_any_adapter = False
2015-03-14 01:13:36 +02:00
self . _ram = 0
2014-11-05 04:00:01 +02:00
self . _adapter_type = " Intel PRO/1000 MT Desktop (82540EM) "
2014-07-12 22:18:25 +03:00
2015-01-22 00:21:15 +02:00
def __json__ ( self ) :
2014-11-09 20:50:47 +02:00
2015-06-26 18:09:19 +03:00
json = { " name " : self . name ,
2016-05-11 20:35:36 +03:00
" node_id " : self . id ,
2015-01-24 01:38:46 +02:00
" console " : self . console ,
2016-11-07 12:16:51 +02:00
" console_type " : self . console_type ,
2015-02-04 22:48:29 +02:00
" project_id " : self . project . id ,
2015-01-23 04:07:09 +02:00
" vmname " : self . vmname ,
" headless " : self . headless ,
2015-06-03 01:30:35 +03:00
" acpi_shutdown " : self . acpi_shutdown ,
2015-01-31 04:36:05 +02:00
" adapters " : self . _adapters ,
2015-01-23 06:31:26 +02:00
" adapter_type " : self . adapter_type ,
2015-03-14 01:13:36 +02:00
" ram " : self . ram ,
2016-05-17 20:51:06 +03:00
" status " : self . status ,
2016-07-12 17:22:55 +03:00
" use_any_adapter " : self . use_any_adapter ,
2016-10-24 22:39:35 +03:00
" linked_clone " : self . linked_clone }
if self . linked_clone :
2016-05-12 11:39:50 +03:00
json [ " node_directory " ] = self . working_dir
2015-06-26 18:09:19 +03:00
else :
2016-05-12 11:39:50 +03:00
json [ " node_directory " ] = None
2015-06-26 18:09:19 +03:00
return json
2014-07-12 22:18:25 +03:00
2015-01-20 03:30:57 +02:00
@asyncio.coroutine
def _get_system_properties ( self ) :
2015-01-24 02:57:54 +02:00
properties = yield from self . manager . execute ( " list " , [ " systemproperties " ] )
2015-01-20 03:30:57 +02:00
for prop in properties :
try :
name , value = prop . split ( ' : ' , 1 )
except ValueError :
continue
self . _system_properties [ name . strip ( ) ] = value . strip ( )
2014-07-12 22:18:25 +03:00
2015-01-20 03:30:57 +02:00
@asyncio.coroutine
2015-01-22 04:26:39 +02:00
def _get_vm_state ( self ) :
"""
Returns the VM state ( e . g . running , paused etc . )
2014-07-12 22:18:25 +03:00
2015-01-22 04:26:39 +02:00
: returns : state ( string )
"""
2015-01-20 03:30:57 +02:00
2015-01-24 02:57:54 +02:00
results = yield from self . manager . execute ( " showvminfo " , [ self . _vmname , " --machinereadable " ] )
2015-01-22 04:26:39 +02:00
for info in results :
2015-04-14 15:35:48 +03:00
if ' = ' in info :
name , value = info . split ( ' = ' , 1 )
if name == " VMState " :
return value . strip ( ' " ' )
2015-01-22 04:26:39 +02:00
raise VirtualBoxError ( " Could not get VM state for {} " . format ( self . _vmname ) )
@asyncio.coroutine
def _control_vm ( self , params ) :
"""
Change setting in this VM when running .
: param params : params to use with sub - command controlvm
: returns : result of the command .
"""
args = shlex . split ( params )
2015-01-24 02:57:54 +02:00
result = yield from self . manager . execute ( " controlvm " , [ self . _vmname ] + args )
2015-01-22 04:26:39 +02:00
return result
@asyncio.coroutine
def _modify_vm ( self , params ) :
"""
Change setting in this VM when not running .
: param params : params to use with sub - command modifyvm
"""
2014-07-12 22:18:25 +03:00
2015-01-22 04:26:39 +02:00
args = shlex . split ( params )
2015-01-24 02:57:54 +02:00
yield from self . manager . execute ( " modifyvm " , [ self . _vmname ] + args )
2014-07-12 22:18:25 +03:00
2016-11-17 13:21:38 +02:00
@asyncio.coroutine
def _check_duplicate_linked_clone ( self ) :
"""
Without linked clone two VM using the same image can ' t run
at the same time .
To avoid issue like false detection when a project close
and another open we try multiple times .
"""
trial = 0
while True :
found = False
for node in self . manager . nodes :
if node != self and node . vmname == self . vmname :
found = True
if node . project != self . project :
if trial > = 30 :
raise VirtualBoxError ( " Sorry a node without the linked clone setting enabled can only be used once on your server. \n {} is already used by {} in project {} " . format ( self . vmname , node . name , self . project . name ) )
else :
if trial > = 5 :
raise VirtualBoxError ( " Sorry a node without the linked clone setting enabled can only be used once on your server. \n {} is already used by {} in this project " . format ( self . vmname , node . name ) )
if not found :
return
trial + = 1
yield from asyncio . sleep ( 1 )
2015-01-23 04:07:09 +02:00
@asyncio.coroutine
def create ( self ) :
2016-11-17 13:21:38 +02:00
if not self . linked_clone :
yield from self . _check_duplicate_linked_clone ( )
2015-01-23 04:07:09 +02:00
2015-01-22 04:26:39 +02:00
yield from self . _get_system_properties ( )
2015-04-14 15:32:44 +03:00
if " API version " not in self . _system_properties :
2015-07-09 06:38:58 +03:00
raise VirtualBoxError ( " Can ' t access to VirtualBox API version: \n {} " . format ( self . _system_properties ) )
2015-01-22 04:26:39 +02:00
if parse_version ( self . _system_properties [ " API version " ] ) < parse_version ( " 4_3 " ) :
raise VirtualBoxError ( " The VirtualBox API version is lower than 4.3 " )
2015-02-04 22:48:29 +02:00
log . info ( " VirtualBox VM ' {name} ' [ {id} ] created " . format ( name = self . name , id = self . id ) )
2015-01-22 04:26:39 +02:00
2016-10-24 22:39:35 +03:00
if self . linked_clone :
2015-02-04 22:48:29 +02:00
if self . id and os . path . isdir ( os . path . join ( self . working_dir , self . _vmname ) ) :
2016-10-28 17:00:26 +03:00
self . _patch_vm_uuid ( )
yield from self . manager . execute ( " registervm " , [ self . _linked_vbox_file ( ) ] )
2016-04-27 00:44:11 +03:00
yield from self . _reattach_linked_hdds ( )
2015-01-22 04:26:39 +02:00
else :
2015-01-23 04:07:09 +02:00
yield from self . _create_linked_clone ( )
2014-07-12 22:18:25 +03:00
2015-01-31 21:01:23 +02:00
if self . _adapters :
yield from self . set_adapters ( self . _adapters )
2015-01-23 06:31:26 +02:00
2015-03-14 01:13:36 +02:00
vm_info = yield from self . _get_vm_info ( )
2015-03-14 21:16:27 +02:00
if " memory " in vm_info :
self . _ram = int ( vm_info [ " memory " ] )
2015-03-14 01:13:36 +02:00
2016-10-28 17:00:26 +03:00
def _linked_vbox_file ( self ) :
return os . path . join ( self . working_dir , self . _vmname , self . _vmname + " .vbox " )
def _patch_vm_uuid ( self ) :
"""
Fix the VM uuid in the case of linked clone
"""
2017-01-10 15:22:04 +02:00
if os . path . exists ( self . _linked_vbox_file ( ) ) :
tree = ET . parse ( self . _linked_vbox_file ( ) )
machine = tree . getroot ( ) . find ( " { http://www.virtualbox.org/}Machine " )
2017-01-16 11:13:13 +02:00
if machine is not None :
2017-01-13 18:20:02 +02:00
machine . set ( " uuid " , " { " + self . id + " } " )
tree . write ( self . _linked_vbox_file ( ) )
2016-10-28 17:00:26 +03:00
2015-07-22 07:58:28 +03:00
@asyncio.coroutine
def check_hw_virtualization ( self ) :
"""
Returns either hardware virtualization is activated or not .
: returns : boolean
"""
vm_info = yield from self . _get_vm_info ( )
if " hwvirtex " in vm_info and vm_info [ " hwvirtex " ] == " on " :
return True
return False
2015-01-22 04:26:39 +02:00
@asyncio.coroutine
2015-01-20 03:30:57 +02:00
def start ( self ) :
2015-01-22 04:26:39 +02:00
"""
Starts this VirtualBox VM .
"""
2015-01-20 03:30:57 +02:00
2016-11-17 11:38:29 +02:00
if self . status == " started " :
return
2015-01-22 04:26:39 +02:00
# resume the VM if it is paused
vm_state = yield from self . _get_vm_state ( )
if vm_state == " paused " :
yield from self . resume ( )
return
2015-01-20 03:30:57 +02:00
2015-09-08 10:20:46 +03:00
# VM must be powered off to start it
if vm_state != " poweroff " :
raise VirtualBoxError ( " VirtualBox VM not powered off " )
2015-01-20 03:30:57 +02:00
2015-01-22 04:26:39 +02:00
yield from self . _set_network_options ( )
yield from self . _set_serial_console ( )
2015-01-20 03:30:57 +02:00
2015-10-13 00:57:37 +03:00
# check if there is enough RAM to run
self . check_available_ram ( self . ram )
2015-01-22 04:26:39 +02:00
args = [ self . _vmname ]
if self . _headless :
args . extend ( [ " --type " , " headless " ] )
2015-01-24 02:57:54 +02:00
result = yield from self . manager . execute ( " startvm " , args )
2016-05-14 05:41:58 +03:00
self . status = " started "
2015-02-04 22:48:29 +02:00
log . info ( " VirtualBox VM ' {name} ' [ {id} ] started " . format ( name = self . name , id = self . id ) )
2015-01-22 04:26:39 +02:00
log . debug ( " Start result: {} " . format ( result ) )
# add a guest property to let the VM know about the GNS3 name
2015-01-24 02:57:54 +02:00
yield from self . manager . execute ( " guestproperty " , [ " set " , self . _vmname , " NameInGNS3 " , self . name ] )
2015-01-22 04:26:39 +02:00
# add a guest property to let the VM know about the GNS3 project directory
2015-01-24 02:57:54 +02:00
yield from self . manager . execute ( " guestproperty " , [ " set " , self . _vmname , " ProjectDirInGNS3 " , self . working_dir ] )
2015-01-22 04:26:39 +02:00
2016-06-25 03:35:39 +03:00
if self . use_ubridge :
yield from self . _start_ubridge ( )
for adapter_number in range ( 0 , self . _adapters ) :
nio = self . _ethernet_adapters [ adapter_number ] . get_nio ( 0 )
if nio :
yield from self . _add_ubridge_udp_connection ( " VBOX- {} - {} " . format ( self . _id , adapter_number ) ,
self . _local_udp_tunnels [ adapter_number ] [ 1 ] ,
nio )
2016-11-07 12:16:51 +02:00
yield from self . _start_console ( )
2014-07-12 22:18:25 +03:00
2015-07-22 07:58:28 +03:00
if ( yield from self . check_hw_virtualization ( ) ) :
self . _hw_virtualization = True
2016-08-29 11:51:50 +03:00
@locked_coroutine
2015-01-22 04:26:39 +02:00
def stop ( self ) :
"""
Stops this VirtualBox VM .
2014-07-12 22:18:25 +03:00
"""
2015-07-22 07:58:28 +03:00
self . _hw_virtualization = False
2016-06-25 03:35:39 +03:00
yield from self . _stop_ubridge ( )
2016-11-18 12:27:50 +02:00
yield from self . _stop_remote_console ( )
2015-01-22 04:26:39 +02:00
vm_state = yield from self . _get_vm_state ( )
if vm_state == " running " or vm_state == " paused " or vm_state == " stuck " :
2015-06-03 01:30:35 +03:00
if self . acpi_shutdown :
# use ACPI to shutdown the VM
result = yield from self . _control_vm ( " acpipowerbutton " )
2016-05-14 05:41:58 +03:00
self . status = " stopped "
2015-06-03 01:30:35 +03:00
log . debug ( " ACPI shutdown result: {} " . format ( result ) )
else :
# power off the VM
result = yield from self . _control_vm ( " poweroff " )
2016-05-14 05:41:58 +03:00
self . status = " stopped "
2015-06-03 01:30:35 +03:00
log . debug ( " Stop result: {} " . format ( result ) )
2014-07-18 00:28:02 +03:00
2015-06-03 01:30:35 +03:00
log . info ( " VirtualBox VM ' {name} ' [ {id} ] stopped " . format ( name = self . name , id = self . id ) )
2016-09-08 19:36:32 +03:00
yield from asyncio . sleep ( 0.5 ) # give some time for VirtualBox to unlock the VM
2015-01-22 04:26:39 +02:00
try :
# deactivate the first serial port
yield from self . _modify_vm ( " --uart1 off " )
except VirtualBoxError as e :
log . warn ( " Could not deactivate the first serial port: {} " . format ( e ) )
2014-07-12 22:18:25 +03:00
2015-04-03 13:08:18 +03:00
for adapter_number in range ( 0 , self . _adapters ) :
2015-02-14 00:41:56 +02:00
nio = self . _ethernet_adapters [ adapter_number ] . get_nio ( 0 )
2015-02-07 02:31:13 +02:00
if nio :
2015-02-14 00:41:56 +02:00
yield from self . _modify_vm ( " --nictrace {} off " . format ( adapter_number + 1 ) )
yield from self . _modify_vm ( " --cableconnected {} off " . format ( adapter_number + 1 ) )
yield from self . _modify_vm ( " --nic {} null " . format ( adapter_number + 1 ) )
2016-11-18 12:27:50 +02:00
yield from super ( ) . stop ( )
2014-07-12 22:18:25 +03:00
2015-01-22 04:26:39 +02:00
@asyncio.coroutine
def suspend ( self ) :
"""
Suspends this VirtualBox VM .
2014-07-12 22:18:25 +03:00
"""
2015-01-22 04:26:39 +02:00
vm_state = yield from self . _get_vm_state ( )
if vm_state == " running " :
yield from self . _control_vm ( " pause " )
2016-05-14 05:41:58 +03:00
self . status = " suspended "
2015-02-04 22:48:29 +02:00
log . info ( " VirtualBox VM ' {name} ' [ {id} ] suspended " . format ( name = self . name , id = self . id ) )
2015-01-22 04:26:39 +02:00
else :
2015-02-04 22:48:29 +02:00
log . warn ( " VirtualBox VM ' {name} ' [ {id} ] cannot be suspended, current state: {state} " . format ( name = self . name ,
id = self . id ,
state = vm_state ) )
2014-07-12 22:18:25 +03:00
2015-01-22 04:26:39 +02:00
@asyncio.coroutine
def resume ( self ) :
2014-07-12 22:18:25 +03:00
"""
2015-01-22 04:26:39 +02:00
Resumes this VirtualBox VM .
2014-07-12 22:18:25 +03:00
"""
2015-01-22 04:26:39 +02:00
yield from self . _control_vm ( " resume " )
2016-05-14 05:41:58 +03:00
self . status = " started "
2015-02-04 22:48:29 +02:00
log . info ( " VirtualBox VM ' {name} ' [ {id} ] resumed " . format ( name = self . name , id = self . id ) )
2015-01-22 04:26:39 +02:00
@asyncio.coroutine
def reload ( self ) :
"""
Reloads this VirtualBox VM .
"""
2014-07-12 22:18:25 +03:00
2015-01-22 04:26:39 +02:00
result = yield from self . _control_vm ( " reset " )
2015-02-04 22:48:29 +02:00
log . info ( " VirtualBox VM ' {name} ' [ {id} ] reloaded " . format ( name = self . name , id = self . id ) )
2015-01-22 04:26:39 +02:00
log . debug ( " Reload result: {} " . format ( result ) )
2014-07-12 22:18:25 +03:00
2015-01-23 04:07:09 +02:00
@asyncio.coroutine
2014-11-09 20:50:47 +02:00
def _get_all_hdd_files ( self ) :
hdds = [ ]
2015-01-24 02:57:54 +02:00
properties = yield from self . manager . execute ( " list " , [ " hdds " ] )
2014-11-09 20:50:47 +02:00
for prop in properties :
try :
name , value = prop . split ( ' : ' , 1 )
except ValueError :
continue
if name . strip ( ) == " Location " :
hdds . append ( value . strip ( ) )
return hdds
2015-01-23 04:07:09 +02:00
@asyncio.coroutine
2016-04-27 00:44:11 +03:00
def _reattach_linked_hdds ( self ) :
"""
Reattach linked cloned hard disks .
"""
2014-11-09 20:50:47 +02:00
2015-01-23 04:07:09 +02:00
hdd_info_file = os . path . join ( self . working_dir , self . _vmname , " hdd_info.json " )
2014-11-09 20:50:47 +02:00
try :
2015-04-25 20:58:34 +03:00
with open ( hdd_info_file , " r " , encoding = " utf-8 " ) as f :
2014-11-09 20:50:47 +02:00
hdd_table = json . load ( f )
2015-09-29 23:15:01 +03:00
except ( ValueError , OSError ) as e :
2016-11-02 13:50:10 +02:00
# The VM has never be started
return
2014-11-09 20:50:47 +02:00
for hdd_info in hdd_table :
2015-01-23 04:07:09 +02:00
hdd_file = os . path . join ( self . working_dir , self . _vmname , " Snapshots " , hdd_info [ " hdd " ] )
2014-11-09 20:50:47 +02:00
if os . path . exists ( hdd_file ) :
2015-02-04 22:48:29 +02:00
log . info ( " VirtualBox VM ' {name} ' [ {id} ] attaching HDD {controller} {port} {device} {medium} " . format ( name = self . name ,
id = self . id ,
controller = hdd_info [ " controller " ] ,
port = hdd_info [ " port " ] ,
device = hdd_info [ " device " ] ,
medium = hdd_file ) )
2015-02-07 02:31:13 +02:00
2016-04-27 00:44:11 +03:00
try :
yield from self . _storage_attach ( ' --storagectl " {} " --port {} --device {} --type hdd --medium " {} " ' . format ( hdd_info [ " controller " ] ,
hdd_info [ " port " ] ,
hdd_info [ " device " ] ,
hdd_file ) )
except VirtualBoxError as e :
log . warn ( " VirtualBox VM ' {name} ' [ {id} ] error reattaching HDD {controller} {port} {device} {medium} : {error} " . format ( name = self . name ,
id = self . id ,
controller = hdd_info [ " controller " ] ,
port = hdd_info [ " port " ] ,
device = hdd_info [ " device " ] ,
medium = hdd_file ,
error = e ) )
continue
2014-11-09 20:50:47 +02:00
2015-01-23 04:07:09 +02:00
@asyncio.coroutine
2016-04-27 00:44:11 +03:00
def save_linked_hdds_info ( self ) :
2014-07-12 22:18:25 +03:00
"""
2016-04-27 00:44:11 +03:00
Save linked cloned hard disks information .
2014-07-12 22:18:25 +03:00
2016-04-27 00:44:11 +03:00
: returns : disk table information
"""
2015-02-24 04:00:34 +02:00
2016-04-27 00:44:11 +03:00
hdd_table = [ ]
2016-10-24 22:39:35 +03:00
if self . linked_clone :
2015-01-24 01:38:46 +02:00
if os . path . exists ( self . working_dir ) :
2015-01-23 04:07:09 +02:00
hdd_files = yield from self . _get_all_hdd_files ( )
2015-02-03 02:00:29 +02:00
vm_info = yield from self . _get_vm_info ( )
2014-11-16 01:05:55 +02:00
for entry , value in vm_info . items ( ) :
2016-04-27 00:06:22 +03:00
match = re . search ( " ^([ \ s \ w]+) \ -( \ d) \ -( \ d)$ " , entry ) # match Controller-PortNumber-DeviceNumber entry
2014-11-16 01:05:55 +02:00
if match :
controller = match . group ( 1 )
port = match . group ( 2 )
device = match . group ( 3 )
2016-04-27 00:06:22 +03:00
if value in hdd_files and os . path . exists ( os . path . join ( self . working_dir , self . _vmname , " Snapshots " , os . path . basename ( value ) ) ) :
2015-02-04 22:48:29 +02:00
log . info ( " VirtualBox VM ' {name} ' [ {id} ] detaching HDD {controller} {port} {device} " . format ( name = self . name ,
id = self . id ,
controller = controller ,
port = port ,
device = device ) )
2014-11-16 01:05:55 +02:00
hdd_table . append (
{
" hdd " : os . path . basename ( value ) ,
" controller " : controller ,
" port " : port ,
" device " : device ,
}
)
2014-11-09 20:50:47 +02:00
2014-11-16 01:05:55 +02:00
if hdd_table :
try :
2015-01-24 01:38:46 +02:00
hdd_info_file = os . path . join ( self . working_dir , self . _vmname , " hdd_info.json " )
2015-04-25 20:58:34 +03:00
with open ( hdd_info_file , " w " , encoding = " utf-8 " ) as f :
2014-11-16 01:05:55 +02:00
json . dump ( hdd_table , f , indent = 4 )
except OSError as e :
2015-02-04 22:48:29 +02:00
log . warning ( " VirtualBox VM ' {name} ' [ {id} ] could not write HHD info file: {error} " . format ( name = self . name ,
id = self . id ,
error = e . strerror ) )
2014-11-09 20:50:47 +02:00
2016-04-27 00:44:11 +03:00
return hdd_table
@asyncio.coroutine
def close ( self ) :
"""
Closes this VirtualBox VM .
"""
if self . _closed :
# VM is already closed
return
2016-04-28 17:52:29 +03:00
if not ( yield from super ( ) . close ( ) ) :
return False
2016-04-27 00:44:11 +03:00
log . debug ( " VirtualBox VM ' {name} ' [ {id} ] is closing " . format ( name = self . name , id = self . id ) )
if self . _console :
self . _manager . port_manager . release_tcp_port ( self . _console , self . _project )
self . _console = None
for adapter in self . _ethernet_adapters . values ( ) :
if adapter is not None :
for nio in adapter . ports . values ( ) :
if nio and isinstance ( nio , NIOUDP ) :
self . manager . port_manager . release_udp_port ( nio . lport , self . _project )
2016-06-25 03:35:39 +03:00
for udp_tunnel in self . _local_udp_tunnels . values ( ) :
self . manager . port_manager . release_udp_port ( udp_tunnel [ 0 ] . lport , self . _project )
self . manager . port_manager . release_udp_port ( udp_tunnel [ 1 ] . lport , self . _project )
self . _local_udp_tunnels = { }
2016-04-27 00:44:11 +03:00
self . acpi_shutdown = False
yield from self . stop ( )
2016-10-24 22:39:35 +03:00
if self . linked_clone :
2016-04-27 00:44:11 +03:00
hdd_table = yield from self . save_linked_hdds_info ( )
for hdd in hdd_table . copy ( ) :
log . info ( " VirtualBox VM ' {name} ' [ {id} ] detaching HDD {controller} {port} {device} " . format ( name = self . name ,
id = self . id ,
controller = hdd [ " controller " ] ,
port = hdd [ " port " ] ,
device = hdd [ " device " ] ) )
try :
yield from self . _storage_attach ( ' --storagectl " {} " --port {} --device {} --type hdd --medium none ' . format ( hdd [ " controller " ] ,
hdd [ " port " ] ,
hdd [ " device " ] ) )
except VirtualBoxError as e :
log . warn ( " VirtualBox VM ' {name} ' [ {id} ] error detaching HDD {controller} {port} {device} : {error} " . format ( name = self . name ,
id = self . id ,
controller = hdd [ " controller " ] ,
port = hdd [ " port " ] ,
device = hdd [ " device " ] ,
error = e ) )
continue
log . info ( " VirtualBox VM ' {name} ' [ {id} ] unregistering " . format ( name = self . name , id = self . id ) )
yield from self . manager . execute ( " unregistervm " , [ self . _name ] )
2015-02-04 22:48:29 +02:00
log . info ( " VirtualBox VM ' {name} ' [ {id} ] closed " . format ( name = self . name , id = self . id ) )
2015-02-03 03:56:13 +02:00
self . _closed = True
2014-07-12 22:18:25 +03:00
2014-07-18 00:28:02 +03:00
@property
def headless ( self ) :
"""
Returns either the VM will start in headless mode
: returns : boolean
"""
return self . _headless
@headless.setter
def headless ( self , headless ) :
"""
Sets either the VM will start in headless mode
: param headless : boolean
"""
if headless :
2015-02-04 22:48:29 +02:00
log . info ( " VirtualBox VM ' {name} ' [ {id} ] has enabled the headless mode " . format ( name = self . name , id = self . id ) )
2014-07-18 00:28:02 +03:00
else :
2015-02-04 22:48:29 +02:00
log . info ( " VirtualBox VM ' {name} ' [ {id} ] has disabled the headless mode " . format ( name = self . name , id = self . id ) )
2014-07-18 00:28:02 +03:00
self . _headless = headless
2015-06-03 01:30:35 +03:00
@property
def acpi_shutdown ( self ) :
"""
Returns either the VM will use ACPI shutdown
: returns : boolean
"""
return self . _acpi_shutdown
@acpi_shutdown.setter
def acpi_shutdown ( self , acpi_shutdown ) :
"""
Sets either the VM will use ACPI shutdown
: param acpi_shutdown : boolean
"""
if acpi_shutdown :
log . info ( " VirtualBox VM ' {name} ' [ {id} ] has enabled the ACPI shutdown mode " . format ( name = self . name , id = self . id ) )
else :
log . info ( " VirtualBox VM ' {name} ' [ {id} ] has disabled the ACPI shutdown mode " . format ( name = self . name , id = self . id ) )
self . _acpi_shutdown = acpi_shutdown
2015-03-14 01:13:36 +02:00
@property
def ram ( self ) :
"""
Returns the amount of RAM allocated to this VirtualBox VM .
: returns : amount RAM in MB ( integer )
"""
return self . _ram
@asyncio.coroutine
def set_ram ( self , ram ) :
"""
Set the amount of RAM allocated to this VirtualBox VM .
: param ram : amount RAM in MB ( integer )
"""
if ram == 0 :
return
yield from self . _modify_vm ( ' --memory {} ' . format ( ram ) )
log . info ( " VirtualBox VM ' {name} ' [ {id} ] has set amount of RAM to {ram} " . format ( name = self . name , id = self . id , ram = ram ) )
self . _ram = ram
2014-07-18 00:28:02 +03:00
@property
def vmname ( self ) :
"""
2015-03-08 20:32:36 +02:00
Returns the VirtualBox VM name .
2014-07-18 00:28:02 +03:00
: returns : VirtualBox VM name
"""
return self . _vmname
2015-03-08 20:32:36 +02:00
@asyncio.coroutine
def set_vmname ( self , vmname ) :
2014-07-18 00:28:02 +03:00
"""
2015-03-08 20:32:36 +02:00
Renames the VirtualBox VM .
2014-07-18 00:28:02 +03:00
: param vmname : VirtualBox VM name
"""
2016-10-24 22:39:35 +03:00
if self . linked_clone :
2017-01-30 16:19:46 +02:00
if self . status == " started " :
raise VirtualBoxError ( " You can ' t change the name of running VM {} " . format ( self . _name ) )
2015-03-08 20:32:36 +02:00
yield from self . _modify_vm ( ' --name " {} " ' . format ( vmname ) )
2015-02-04 22:48:29 +02:00
log . info ( " VirtualBox VM ' {name} ' [ {id} ] has set the VM name to ' {vmname} ' " . format ( name = self . name , id = self . id , vmname = vmname ) )
2014-07-18 00:28:02 +03:00
self . _vmname = vmname
2015-02-07 02:31:13 +02:00
@property
def adapters ( self ) :
"""
Returns the number of adapters configured for this VirtualBox VM .
: returns : number of adapters
"""
2015-02-05 20:58:10 +02:00
2015-02-07 02:31:13 +02:00
return self . _adapters
2015-02-05 20:58:10 +02:00
2015-01-23 06:31:26 +02:00
@asyncio.coroutine
def set_adapters ( self , adapters ) :
2014-07-18 00:28:02 +03:00
"""
Sets the number of Ethernet adapters for this VirtualBox VM instance .
: param adapters : number of adapters
"""
2014-11-01 01:41:12 +02:00
# check for the maximum adapters supported by the VM
2015-01-23 06:31:26 +02:00
self . _maximum_adapters = yield from self . _get_maximum_supported_adapters ( )
2015-04-03 13:08:18 +03:00
if adapters > self . _maximum_adapters :
2014-11-01 01:41:12 +02:00
raise VirtualBoxError ( " Number of adapters above the maximum supported of {} " . format ( self . _maximum_adapters ) )
2014-07-18 00:28:02 +03:00
self . _ethernet_adapters . clear ( )
2015-02-14 00:41:56 +02:00
for adapter_number in range ( 0 , adapters ) :
2015-04-03 13:08:18 +03:00
self . _ethernet_adapters [ adapter_number ] = EthernetAdapter ( )
2014-07-18 00:28:02 +03:00
2015-01-31 04:36:05 +02:00
self . _adapters = len ( self . _ethernet_adapters )
2015-02-04 22:48:29 +02:00
log . info ( " VirtualBox VM ' {name} ' [ {id} ] has changed the number of Ethernet adapters to {adapters} " . format ( name = self . name ,
id = self . id ,
adapters = adapters ) )
2014-08-27 00:27:43 +03:00
@property
2015-02-07 02:31:13 +02:00
def use_any_adapter ( self ) :
2014-08-27 00:27:43 +03:00
"""
2015-02-07 02:31:13 +02:00
Returns either GNS3 can use any VirtualBox adapter on this instance .
2014-08-27 00:27:43 +03:00
2015-05-22 06:48:59 +03:00
: returns : boolean
2014-08-27 00:27:43 +03:00
"""
2015-02-07 02:31:13 +02:00
return self . _use_any_adapter
2014-08-27 00:27:43 +03:00
2015-02-07 02:31:13 +02:00
@use_any_adapter.setter
def use_any_adapter ( self , use_any_adapter ) :
2014-08-27 00:27:43 +03:00
"""
2015-02-07 02:31:13 +02:00
Allows GNS3 to use any VirtualBox adapter on this instance .
2014-08-27 00:27:43 +03:00
2015-02-07 02:31:13 +02:00
: param use_any_adapter : boolean
2014-08-27 00:27:43 +03:00
"""
2015-02-07 02:31:13 +02:00
if use_any_adapter :
log . info ( " VirtualBox VM ' {name} ' [ {id} ] is allowed to use any adapter " . format ( name = self . name , id = self . id ) )
else :
2015-05-22 06:48:59 +03:00
log . info ( " VirtualBox VM ' {name} ' [ {id} ] is not allowed to use any adapter " . format ( name = self . name , id = self . id ) )
2015-02-07 02:31:13 +02:00
self . _use_any_adapter = use_any_adapter
2014-07-18 00:28:02 +03:00
@property
def adapter_type ( self ) :
"""
Returns the adapter type for this VirtualBox VM instance .
: returns : adapter type ( string )
"""
return self . _adapter_type
@adapter_type.setter
def adapter_type ( self , adapter_type ) :
"""
Sets the adapter type for this VirtualBox VM instance .
: param adapter_type : adapter type ( string )
"""
self . _adapter_type = adapter_type
2015-02-04 22:48:29 +02:00
log . info ( " VirtualBox VM ' {name} ' [ {id} ]: adapter type changed to {adapter_type} " . format ( name = self . name ,
id = self . id ,
adapter_type = adapter_type ) )
2014-10-31 02:53:17 +02:00
2015-01-22 04:26:39 +02:00
@asyncio.coroutine
2014-10-31 02:53:17 +02:00
def _get_vm_info ( self ) :
"""
Returns this VM info .
: returns : dict of info
"""
vm_info = { }
2015-01-24 02:57:54 +02:00
results = yield from self . manager . execute ( " showvminfo " , [ self . _vmname , " --machinereadable " ] )
2014-10-31 02:53:17 +02:00
for info in results :
try :
2014-11-09 20:50:47 +02:00
name , value = info . split ( ' = ' , 1 )
2014-10-31 02:53:17 +02:00
except ValueError :
continue
2014-11-09 20:50:47 +02:00
vm_info [ name . strip ( ' " ' ) ] = value . strip ( ' " ' )
2014-10-31 02:53:17 +02:00
return vm_info
2015-01-22 04:26:39 +02:00
@asyncio.coroutine
2014-10-31 02:53:17 +02:00
def _get_maximum_supported_adapters ( self ) :
"""
Returns the maximum adapters supported by this VM .
: returns : maximum number of supported adapters ( int )
"""
# check the maximum number of adapters supported by the VM
2015-01-22 04:26:39 +02:00
vm_info = yield from self . _get_vm_info ( )
2014-10-31 02:53:17 +02:00
maximum_adapters = 8
2015-07-26 01:39:38 +03:00
if " chipset " in vm_info :
chipset = vm_info [ " chipset " ]
if chipset == " ich9 " :
maximum_adapters = int ( self . _system_properties [ " Maximum ICH9 Network Adapter count " ] )
2014-10-31 02:53:17 +02:00
return maximum_adapters
def _get_pipe_name ( self ) :
"""
Returns the pipe name to create a serial connection .
: returns : pipe path ( string )
"""
2015-01-22 04:26:39 +02:00
if sys . platform . startswith ( " win " ) :
2015-04-23 05:29:52 +03:00
pipe_name = r " \\ . \ pipe \ gns3_vbox \ {} " . format ( self . id )
2014-10-31 02:53:17 +02:00
else :
2015-04-23 05:29:52 +03:00
pipe_name = os . path . join ( tempfile . gettempdir ( ) , " gns3_vbox " , " {} " . format ( self . id ) )
try :
os . makedirs ( os . path . dirname ( pipe_name ) , exist_ok = True )
except OSError as e :
raise VirtualBoxError ( " Could not create the VirtualBox pipe directory: {} " . format ( e ) )
2014-10-31 02:53:17 +02:00
return pipe_name
2015-01-22 04:26:39 +02:00
@asyncio.coroutine
2014-10-31 02:53:17 +02:00
def _set_serial_console ( self ) :
"""
Configures the first serial port to allow a serial console connection .
"""
# activate the first serial port
2015-01-22 04:26:39 +02:00
yield from self . _modify_vm ( " --uart1 0x3F8 4 " )
2014-10-31 02:53:17 +02:00
# set server mode with a pipe on the first serial port
pipe_name = self . _get_pipe_name ( )
args = [ self . _vmname , " --uartmode1 " , " server " , pipe_name ]
2015-01-24 02:57:54 +02:00
yield from self . manager . execute ( " modifyvm " , args )
2014-10-31 02:53:17 +02:00
2015-01-23 04:07:09 +02:00
@asyncio.coroutine
2014-11-09 20:50:47 +02:00
def _storage_attach ( self , params ) :
"""
Change storage medium in this VM .
: param params : params to use with sub - command storageattach
"""
args = shlex . split ( params )
2015-01-24 02:57:54 +02:00
yield from self . manager . execute ( " storageattach " , [ self . _vmname ] + args )
2014-11-09 20:50:47 +02:00
2015-01-22 04:26:39 +02:00
@asyncio.coroutine
2014-10-31 02:53:17 +02:00
def _get_nic_attachements ( self , maximum_adapters ) :
"""
Returns NIC attachements .
: param maximum_adapters : maximum number of supported adapters
: returns : list of adapters with their Attachment setting ( NAT , bridged etc . )
"""
nics = [ ]
2015-01-22 04:26:39 +02:00
vm_info = yield from self . _get_vm_info ( )
2015-02-14 00:41:56 +02:00
for adapter_number in range ( 0 , maximum_adapters ) :
entry = " nic {} " . format ( adapter_number + 1 )
2014-10-31 02:53:17 +02:00
if entry in vm_info :
value = vm_info [ entry ]
2015-03-11 20:05:22 +02:00
nics . append ( value . lower ( ) )
2014-10-31 02:53:17 +02:00
else :
nics . append ( None )
return nics
2015-01-22 04:26:39 +02:00
@asyncio.coroutine
2014-11-01 01:41:12 +02:00
def _set_network_options ( self ) :
2014-10-31 02:53:17 +02:00
"""
Configures network options .
"""
2015-02-07 02:31:13 +02:00
nic_attachments = yield from self . _get_nic_attachements ( self . _maximum_adapters )
2015-04-03 13:08:18 +03:00
for adapter_number in range ( 0 , self . _adapters ) :
2015-03-06 04:11:33 +02:00
attachment = nic_attachments [ adapter_number ]
if attachment == " null " :
# disconnect the cable if no backend is attached.
2015-04-04 04:08:29 +03:00
yield from self . _modify_vm ( " --cableconnected {} off " . format ( adapter_number + 1 ) )
if attachment == " none " :
# set the backend to null to avoid a difference in the number of interfaces in the Guest.
yield from self . _modify_vm ( " --nic {} null " . format ( adapter_number + 1 ) )
yield from self . _modify_vm ( " --cableconnected {} off " . format ( adapter_number + 1 ) )
2016-06-25 03:35:39 +03:00
if self . use_ubridge :
# use a local UDP tunnel to connect to uBridge instead
if adapter_number not in self . _local_udp_tunnels :
self . _local_udp_tunnels [ adapter_number ] = self . _create_local_udp_tunnel ( )
nio = self . _local_udp_tunnels [ adapter_number ] [ 0 ]
else :
nio = self . _ethernet_adapters [ adapter_number ] . get_nio ( 0 )
2015-02-07 02:31:13 +02:00
if nio :
2016-06-01 04:06:48 +03:00
if not self . _use_any_adapter and attachment not in ( " none " , " null " , " generic " ) :
2015-02-07 02:31:13 +02:00
raise VirtualBoxError ( " Attachment ( {} ) already configured on adapter {} . "
" Please set it to ' Not attached ' to allow GNS3 to use it. " . format ( attachment ,
2015-02-14 00:41:56 +02:00
adapter_number + 1 ) )
2014-10-31 02:53:17 +02:00
2015-05-07 00:21:39 +03:00
yield from self . _modify_vm ( " --nictrace {} off " . format ( adapter_number + 1 ) )
2014-10-31 02:53:17 +02:00
vbox_adapter_type = " 82540EM "
2015-02-07 02:31:13 +02:00
if self . _adapter_type == " PCnet-PCI II (Am79C970A) " :
vbox_adapter_type = " Am79C970A "
if self . _adapter_type == " PCNet-FAST III (Am79C973) " :
vbox_adapter_type = " Am79C973 "
if self . _adapter_type == " Intel PRO/1000 MT Desktop (82540EM) " :
vbox_adapter_type = " 82540EM "
if self . _adapter_type == " Intel PRO/1000 T Server (82543GC) " :
vbox_adapter_type = " 82543GC "
if self . _adapter_type == " Intel PRO/1000 MT Server (82545EM) " :
vbox_adapter_type = " 82545EM "
if self . _adapter_type == " Paravirtualized Network (virtio-net) " :
vbox_adapter_type = " virtio "
2015-02-14 00:41:56 +02:00
args = [ self . _vmname , " --nictype {} " . format ( adapter_number + 1 ) , vbox_adapter_type ]
2015-02-07 02:31:13 +02:00
yield from self . manager . execute ( " modifyvm " , args )
2014-10-31 02:53:17 +02:00
2015-05-07 00:21:39 +03:00
if isinstance ( nio , NIOUDP ) :
log . debug ( " setting UDP params on adapter {} " . format ( adapter_number ) )
yield from self . _modify_vm ( " --nic {} generic " . format ( adapter_number + 1 ) )
yield from self . _modify_vm ( " --nicgenericdrv {} UDPTunnel " . format ( adapter_number + 1 ) )
yield from self . _modify_vm ( " --nicproperty {} sport= {} " . format ( adapter_number + 1 , nio . lport ) )
yield from self . _modify_vm ( " --nicproperty {} dest= {} " . format ( adapter_number + 1 , nio . rhost ) )
yield from self . _modify_vm ( " --nicproperty {} dport= {} " . format ( adapter_number + 1 , nio . rport ) )
yield from self . _modify_vm ( " --cableconnected {} on " . format ( adapter_number + 1 ) )
2014-10-31 02:53:17 +02:00
if nio . capturing :
2015-02-14 00:41:56 +02:00
yield from self . _modify_vm ( " --nictrace {} on " . format ( adapter_number + 1 ) )
2015-03-10 19:05:52 +02:00
yield from self . _modify_vm ( ' --nictracefile {} " {} " ' . format ( adapter_number + 1 , nio . pcap_output_file ) )
2014-10-31 02:53:17 +02:00
2017-01-05 08:30:23 +02:00
if self . use_ubridge and not self . _ethernet_adapters [ adapter_number ] . get_nio ( 0 ) :
yield from self . _modify_vm ( " --cableconnected {} off " . format ( adapter_number + 1 ) )
2015-04-03 13:08:18 +03:00
for adapter_number in range ( self . _adapters , self . _maximum_adapters ) :
2015-02-14 00:41:56 +02:00
log . debug ( " disabling remaining adapter {} " . format ( adapter_number ) )
yield from self . _modify_vm ( " --nic {} none " . format ( adapter_number + 1 ) )
2014-10-31 02:53:17 +02:00
2015-01-22 04:26:39 +02:00
@asyncio.coroutine
2014-11-09 20:50:47 +02:00
def _create_linked_clone ( self ) :
2014-11-16 01:05:55 +02:00
"""
Creates a new linked clone .
"""
2014-11-09 20:50:47 +02:00
gns3_snapshot_exists = False
2015-01-22 04:26:39 +02:00
vm_info = yield from self . _get_vm_info ( )
2014-11-09 20:50:47 +02:00
for entry , value in vm_info . items ( ) :
if entry . startswith ( " SnapshotName " ) and value == " GNS3 Linked Base for clones " :
gns3_snapshot_exists = True
if not gns3_snapshot_exists :
2015-01-24 02:57:54 +02:00
result = yield from self . manager . execute ( " snapshot " , [ self . _vmname , " take " , " GNS3 Linked Base for clones " ] )
2014-11-16 01:05:55 +02:00
log . debug ( " GNS3 snapshot created: {} " . format ( result ) )
2014-11-09 20:50:47 +02:00
args = [ self . _vmname ,
" --snapshot " ,
" GNS3 Linked Base for clones " ,
" --options " ,
" link " ,
" --name " ,
2015-01-23 04:07:09 +02:00
self . name ,
2014-11-09 20:50:47 +02:00
" --basefolder " ,
2015-01-23 04:07:09 +02:00
self . working_dir ,
2014-11-09 20:50:47 +02:00
" --register " ]
2015-01-24 02:57:54 +02:00
result = yield from self . manager . execute ( " clonevm " , args )
2015-02-03 03:56:13 +02:00
log . debug ( " VirtualBox VM: {} cloned " . format ( result ) )
2015-01-11 14:22:59 +02:00
2014-11-09 20:50:47 +02:00
self . _vmname = self . _name
2015-01-24 02:57:54 +02:00
yield from self . manager . execute ( " setextradata " , [ self . _vmname , " GNS3/Clone " , " yes " ] )
2015-01-12 03:24:13 +02:00
2016-11-04 12:07:43 +02:00
# We create a reset snapshot in order to simplify life of user who want to rollback their VM
# Warning: Do not document this it's seem buggy we keep it because Raizo students use it.
try :
args = [ self . _vmname , " take " , " reset " ]
result = yield from self . manager . execute ( " snapshot " , args )
log . debug ( " Snapshot ' reset ' created: {} " . format ( result ) )
# It seem sometimes this failed due to internal race condition of Vbox
# we have no real explanation of this.
except VirtualBoxError :
log . warn ( " Snapshot ' reset ' not created " )
2014-11-09 20:50:47 +02:00
2016-11-07 15:10:44 +02:00
os . makedirs ( os . path . join ( self . working_dir , self . _vmname ) , exist_ok = True )
2016-11-07 12:16:51 +02:00
@asyncio.coroutine
def _start_console ( self ) :
2015-01-04 23:56:17 +02:00
"""
Starts remote console support for this VM .
"""
2016-11-18 12:27:50 +02:00
self . _remote_pipe = yield from asyncio_open_serial ( self . _get_pipe_name ( ) )
server = AsyncioTelnetServer ( reader = self . _remote_pipe ,
writer = self . _remote_pipe ,
binary = True ,
echo = True )
2017-01-09 17:05:29 +02:00
self . _telnet_server = yield from asyncio . start_server ( server . run , self . _manager . port_manager . console_host , self . console )
2015-01-04 23:56:17 +02:00
2016-11-18 12:27:50 +02:00
@asyncio.coroutine
2015-01-04 23:56:17 +02:00
def _stop_remote_console ( self ) :
"""
Stops remote console support for this VM .
"""
2016-11-07 12:16:51 +02:00
if self . _telnet_server :
self . _telnet_server . close ( )
2016-11-18 12:27:50 +02:00
yield from self . _telnet_server . wait_closed ( )
self . _remote_pipe . close ( )
self . _telnet_server = None
2015-01-04 23:56:17 +02:00
2015-01-22 04:26:39 +02:00
@asyncio.coroutine
2015-02-14 00:41:56 +02:00
def adapter_add_nio_binding ( self , adapter_number , nio ) :
2014-07-18 00:28:02 +03:00
"""
2015-02-07 02:31:13 +02:00
Adds an adapter NIO binding .
2014-07-18 00:28:02 +03:00
2015-02-14 00:41:56 +02:00
: param adapter_number : adapter number
2014-07-18 00:28:02 +03:00
: param nio : NIO instance to add to the slot / port
"""
try :
2015-02-14 00:41:56 +02:00
adapter = self . _ethernet_adapters [ adapter_number ]
2015-05-27 17:38:57 +03:00
except KeyError :
2015-02-14 00:41:56 +02:00
raise VirtualBoxError ( " Adapter {adapter_number} doesn ' t exist on VirtualBox VM ' {name} ' " . format ( name = self . name ,
adapter_number = adapter_number ) )
2014-07-18 00:28:02 +03:00
2016-12-14 13:01:34 +02:00
if self . ubridge :
2016-06-25 03:35:39 +03:00
yield from self . _add_ubridge_udp_connection ( " VBOX- {} - {} " . format ( self . _id , adapter_number ) ,
2016-08-29 15:07:52 +03:00
self . _local_udp_tunnels [ adapter_number ] [ 1 ] ,
nio )
2017-01-05 08:30:23 +02:00
yield from self . _control_vm ( " setlinkstate {} on " . format ( adapter_number + 1 ) )
2016-06-25 03:35:39 +03:00
else :
vm_state = yield from self . _get_vm_state ( )
if vm_state == " running " :
if isinstance ( nio , NIOUDP ) :
# dynamically configure an UDP tunnel on the VirtualBox adapter
yield from self . _control_vm ( " nic {} generic UDPTunnel " . format ( adapter_number + 1 ) )
yield from self . _control_vm ( " nicproperty {} sport= {} " . format ( adapter_number + 1 , nio . lport ) )
yield from self . _control_vm ( " nicproperty {} dest= {} " . format ( adapter_number + 1 , nio . rhost ) )
yield from self . _control_vm ( " nicproperty {} dport= {} " . format ( adapter_number + 1 , nio . rport ) )
yield from self . _control_vm ( " setlinkstate {} on " . format ( adapter_number + 1 ) )
# check if the UDP tunnel has been correctly set
vm_info = yield from self . _get_vm_info ( )
generic_driver_number = " generic {} " . format ( adapter_number + 1 )
if generic_driver_number not in vm_info and vm_info [ generic_driver_number ] != " UDPTunnel " :
log . warning ( " UDP tunnel has not been set on nic: {} " . format ( adapter_number + 1 ) )
self . project . emit ( " log.warning " , { " message " : " UDP tunnel has not been set on nic: {} " . format ( adapter_number + 1 ) } )
2016-02-11 04:08:34 +02:00
2014-07-18 00:28:02 +03:00
adapter . add_nio ( 0 , nio )
2015-02-14 00:41:56 +02:00
log . info ( " VirtualBox VM ' {name} ' [ {id} ]: {nio} added to adapter {adapter_number} " . format ( name = self . name ,
id = self . id ,
nio = nio ,
adapter_number = adapter_number ) )
2014-07-18 00:28:02 +03:00
2015-01-22 04:26:39 +02:00
@asyncio.coroutine
2015-02-14 00:41:56 +02:00
def adapter_remove_nio_binding ( self , adapter_number ) :
2014-07-18 00:28:02 +03:00
"""
2015-02-07 02:31:13 +02:00
Removes an adapter NIO binding .
2014-07-18 00:28:02 +03:00
2015-02-14 00:41:56 +02:00
: param adapter_number : adapter number
2014-07-18 00:28:02 +03:00
: returns : NIO instance
"""
try :
2015-02-14 00:41:56 +02:00
adapter = self . _ethernet_adapters [ adapter_number ]
2015-05-27 17:38:57 +03:00
except KeyError :
2015-02-14 00:41:56 +02:00
raise VirtualBoxError ( " Adapter {adapter_number} doesn ' t exist on VirtualBox VM ' {name} ' " . format ( name = self . name ,
adapter_number = adapter_number ) )
2014-07-18 00:28:02 +03:00
2016-12-14 13:01:34 +02:00
if self . ubridge :
2016-06-25 03:35:39 +03:00
yield from self . _ubridge_send ( " bridge delete {name} " . format ( name = " VBOX- {} - {} " . format ( self . _id , adapter_number ) ) )
2017-01-05 08:30:23 +02:00
yield from self . _control_vm ( " setlinkstate {} off " . format ( adapter_number + 1 ) )
2016-06-25 03:35:39 +03:00
else :
vm_state = yield from self . _get_vm_state ( )
if vm_state == " running " :
# dynamically disable the VirtualBox adapter
yield from self . _control_vm ( " setlinkstate {} off " . format ( adapter_number + 1 ) )
yield from self . _control_vm ( " nic {} null " . format ( adapter_number + 1 ) )
2014-07-18 00:28:02 +03:00
nio = adapter . get_nio ( 0 )
2015-02-24 04:00:34 +02:00
if isinstance ( nio , NIOUDP ) :
2015-03-22 01:19:12 +02:00
self . manager . port_manager . release_udp_port ( nio . lport , self . _project )
2014-07-18 00:28:02 +03:00
adapter . remove_nio ( 0 )
2015-01-25 00:32:58 +02:00
2015-02-14 00:41:56 +02:00
log . info ( " VirtualBox VM ' {name} ' [ {id} ]: {nio} removed from adapter {adapter_number} " . format ( name = self . name ,
id = self . id ,
nio = nio ,
adapter_number = adapter_number ) )
2014-07-18 00:28:02 +03:00
return nio
2015-05-27 22:56:27 +03:00
@asyncio.coroutine
2015-02-14 00:41:56 +02:00
def start_capture ( self , adapter_number , output_file ) :
2014-07-18 00:28:02 +03:00
"""
Starts a packet capture .
2015-02-14 00:41:56 +02:00
: param adapter_number : adapter number
2014-07-18 00:28:02 +03:00
: param output_file : PCAP destination file for the capture
"""
try :
2015-02-14 00:41:56 +02:00
adapter = self . _ethernet_adapters [ adapter_number ]
2015-05-27 17:38:57 +03:00
except KeyError :
2015-02-14 00:41:56 +02:00
raise VirtualBoxError ( " Adapter {adapter_number} doesn ' t exist on VirtualBox VM ' {name} ' " . format ( name = self . name ,
adapter_number = adapter_number ) )
2014-07-18 00:28:02 +03:00
2016-06-25 03:35:39 +03:00
if not self . use_ubridge :
vm_state = yield from self . _get_vm_state ( )
if vm_state == " running " or vm_state == " paused " or vm_state == " stuck " :
raise VirtualBoxError ( " Sorry, packet capturing on a started VirtualBox VM is not supported without using uBridge " )
2015-05-27 22:56:27 +03:00
2014-07-18 00:28:02 +03:00
nio = adapter . get_nio ( 0 )
2015-06-02 00:42:17 +03:00
if not nio :
raise VirtualBoxError ( " Adapter {} is not connected " . format ( adapter_number ) )
2014-07-18 00:28:02 +03:00
if nio . capturing :
2015-02-14 00:41:56 +02:00
raise VirtualBoxError ( " Packet capture is already activated on adapter {adapter_number} " . format ( adapter_number = adapter_number ) )
2014-07-18 00:28:02 +03:00
nio . startPacketCapture ( output_file )
2016-06-25 03:35:39 +03:00
2016-12-14 13:01:34 +02:00
if self . ubridge :
2016-06-25 03:35:39 +03:00
yield from self . _ubridge_send ( ' bridge start_capture {name} " {output_file} " ' . format ( name = " VBOX- {} - {} " . format ( self . _id , adapter_number ) ,
output_file = output_file ) )
2015-02-14 00:41:56 +02:00
log . info ( " VirtualBox VM ' {name} ' [ {id} ]: starting packet capture on adapter {adapter_number} " . format ( name = self . name ,
id = self . id ,
adapter_number = adapter_number ) )
2014-07-18 00:28:02 +03:00
2015-02-14 00:41:56 +02:00
def stop_capture ( self , adapter_number ) :
2014-07-18 00:28:02 +03:00
"""
Stops a packet capture .
2015-02-14 00:41:56 +02:00
: param adapter_number : adapter number
2014-07-18 00:28:02 +03:00
"""
try :
2015-02-14 00:41:56 +02:00
adapter = self . _ethernet_adapters [ adapter_number ]
2015-05-27 17:38:57 +03:00
except KeyError :
2015-02-14 00:41:56 +02:00
raise VirtualBoxError ( " Adapter {adapter_number} doesn ' t exist on VirtualBox VM ' {name} ' " . format ( name = self . name ,
adapter_number = adapter_number ) )
2014-07-18 00:28:02 +03:00
nio = adapter . get_nio ( 0 )
2015-06-02 00:42:17 +03:00
if not nio :
raise VirtualBoxError ( " Adapter {} is not connected " . format ( adapter_number ) )
2014-07-18 00:28:02 +03:00
nio . stopPacketCapture ( )
2016-12-14 13:01:34 +02:00
if self . ubridge :
2016-06-25 03:35:39 +03:00
yield from self . _ubridge_send ( ' bridge stop_capture {name} ' . format ( name = " VBOX- {} - {} " . format ( self . _id , adapter_number ) ) )
2015-02-14 00:41:56 +02:00
log . info ( " VirtualBox VM ' {name} ' [ {id} ]: stopping packet capture on adapter {adapter_number} " . format ( name = self . name ,
id = self . id ,
adapter_number = adapter_number ) )