1
0
mirror of https://github.com/GNS3/gns3-server synced 2024-12-26 00:38:10 +00:00

Add "enable console" option to VirtualBox VMs (True by default).

Add "start at" option to VirtualBox VMs (adapter start index, 0 by default).
This commit is contained in:
grossmj 2014-08-26 15:27:43 -06:00
parent 934404cc90
commit 80ab81190c
4 changed files with 138 additions and 31 deletions

View File

@ -417,7 +417,10 @@ class VirtualBox(IModule):
try: try:
vbox_instance.start() vbox_instance.start()
except VirtualBoxError as e: except VirtualBoxError as e:
self.send_custom_error(str(e)) if self._vboxwrapper:
self.send_custom_error("{}: {}".format(e, self._vboxwrapper.read_stderr()))
else:
self.send_custom_error(str(e))
return return
self.send_response(True) self.send_response(True)

View File

@ -82,9 +82,15 @@ VBOX_UPDATE_SCHEMA = {
"adapters": { "adapters": {
"description": "number of adapters", "description": "number of adapters",
"type": "integer", "type": "integer",
"minimum": 0, "minimum": 1,
"maximum": 36, # maximum given by the ICH9 chipset in VirtualBox "maximum": 36, # maximum given by the ICH9 chipset in VirtualBox
}, },
"adapter_start_index": {
"description": "adapter index from which to start using adapters",
"type": "integer",
"minimum": 0,
"maximum": 35, # maximum given by the ICH9 chipset in VirtualBox
},
"adapter_type": { "adapter_type": {
"description": "VirtualBox adapter type", "description": "VirtualBox adapter type",
"type": "string", "type": "string",
@ -96,6 +102,10 @@ VBOX_UPDATE_SCHEMA = {
"maximum": 65535, "maximum": 65535,
"type": "integer" "type": "integer"
}, },
"enable_console": {
"description": "enable the console",
"type": "boolean"
},
"headless": { "headless": {
"description": "headless mode", "description": "headless mode",
"type": "boolean" "type": "boolean"

View File

@ -54,6 +54,7 @@ class VirtualBoxController(object):
self._console = 0 self._console = 0
self._adapters = [] self._adapters = []
self._headless = False self._headless = False
self._enable_console = True
self._adapter_type = "Automatic" self._adapter_type = "Automatic"
try: try:
@ -101,6 +102,16 @@ class VirtualBoxController(object):
self._headless = headless self._headless = headless
@property
def enable_console(self):
return self._enable_console
@enable_console.setter
def enable_console(self, enable_console):
self._enable_console = enable_console
@property @property
def adapters(self): def adapters(self):
@ -123,13 +134,17 @@ class VirtualBoxController(object):
def start(self): def start(self):
if len(self._adapters) > self._maximum_adapters:
raise VirtualBoxError("Number of adapters above the maximum supported of {}".format(self._maximum_adapters))
if self._machine.state == self._vboxmanager.constants.MachineState_Paused: if self._machine.state == self._vboxmanager.constants.MachineState_Paused:
self.resume() self.resume()
return return
self._get_session() self._get_session()
self._set_network_options() self._set_network_options()
self._set_console_options() if self._enable_console:
self._set_console_options()
progress = self._launch_vm_process() progress = self._launch_vm_process()
log.info("VM is starting with {}% completed".format(progress.percent)) log.info("VM is starting with {}% completed".format(progress.percent))
@ -145,25 +160,26 @@ class VirtualBoxController(object):
except Exception: except Exception:
pass pass
# starts the Telnet to pipe thread if self._enable_console:
pipe_name = self._get_pipe_name() # starts the Telnet to pipe thread
if sys.platform.startswith('win'): pipe_name = self._get_pipe_name()
try: if sys.platform.startswith('win'):
self._serial_pipe = open(pipe_name, "a+b") try:
except OSError as e: self._serial_pipe = open(pipe_name, "a+b")
raise VirtualBoxError("Could not open the pipe {}: {}".format(pipe_name, e)) except OSError as e:
self._serial_pipe_thread = PipeProxy(self._vmname, msvcrt.get_osfhandle(self._serial_pipe.fileno()), self._host, self._console) raise VirtualBoxError("Could not open the pipe {}: {}".format(pipe_name, e))
#self._serial_pipe_thread.setDaemon(True) self._serial_pipe_thread = PipeProxy(self._vmname, msvcrt.get_osfhandle(self._serial_pipe.fileno()), self._host, self._console)
self._serial_pipe_thread.start() #self._serial_pipe_thread.setDaemon(True)
else: self._serial_pipe_thread.start()
try: else:
self._serial_pipe = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM) try:
self._serial_pipe.connect(pipe_name) self._serial_pipe = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
except OSError as e: self._serial_pipe.connect(pipe_name)
raise VirtualBoxError("Could not connect to the pipe {}: {}".format(pipe_name, e)) except OSError as e:
self._serial_pipe_thread = PipeProxy(self._vmname, self._serial_pipe, self._host, self._console) raise VirtualBoxError("Could not connect to the pipe {}: {}".format(pipe_name, e))
#self._serial_pipe_thread.setDaemon(True) self._serial_pipe_thread = PipeProxy(self._vmname, self._serial_pipe, self._host, self._console)
self._serial_pipe_thread.start() #self._serial_pipe_thread.setDaemon(True)
self._serial_pipe_thread.start()
def stop(self): def stop(self):
@ -196,8 +212,14 @@ class VirtualBoxController(object):
log.info("VM is stopping with {}% completed".format(self.vmname, progress.percent)) log.info("VM is stopping with {}% completed".format(self.vmname, progress.percent))
self._lock_machine() self._lock_machine()
for adapter_id in range(0, len(self._adapters)): for adapter_id in range(0, len(self._adapters)):
if self._adapters[adapter_id] is None:
continue
self._disable_adapter(adapter_id, disable=True) self._disable_adapter(adapter_id, disable=True)
if self._enable_console:
serial_port = self._session.machine.getSerialPort(0)
serial_port.enabled = False
self._session.machine.saveSettings() self._session.machine.saveSettings()
self._unlock_machine() self._unlock_machine()
except Exception as e: except Exception as e:
@ -252,11 +274,17 @@ class VirtualBoxController(object):
#raise VirtualBoxError("VirtualBox error: {}".format(e)) #raise VirtualBoxError("VirtualBox error: {}".format(e))
for adapter_id in range(0, len(self._adapters)): for adapter_id in range(0, len(self._adapters)):
try: try:
# VirtualBox starts counting from 0 # VirtualBox starts counting from 0
adapter = self._session.machine.getNetworkAdapter(adapter_id) adapter = self._session.machine.getNetworkAdapter(adapter_id)
vbox_adapter_type = adapter.adapterType if self._adapters[adapter_id] is None:
# force enable to avoid any discrepancy in the interface numbering inside the VM
# e.g. Ethernet2 in GNS3 becoming eth0 inside the VM when using a start index of 2.
adapter.enabled = True
continue
vbox_adapter_type = adapter.adapterType
if self._adapter_type == "PCnet-PCI II (Am79C970A)": if self._adapter_type == "PCnet-PCI II (Am79C970A)":
vbox_adapter_type = self._vboxmanager.constants.NetworkAdapterType_Am79C970A vbox_adapter_type = self._vboxmanager.constants.NetworkAdapterType_Am79C970A
if self._adapter_type == "PCNet-FAST III (Am79C973)": if self._adapter_type == "PCNet-FAST III (Am79C973)":
@ -310,9 +338,9 @@ class VirtualBoxController(object):
except Exception as e: except Exception as e:
raise VirtualBoxError("VirtualBox error: {}".format(e)) raise VirtualBoxError("VirtualBox error: {}".format(e))
#for adapter_id in range(len(self._ethernet_adapters), self._maximum_adapters): for adapter_id in range(len(self._adapters), self._maximum_adapters):
# log.debug("disabling remaining adapter {}".format(adapter_id)) log.debug("disabling remaining adapter {}".format(adapter_id))
# self._disable_adapter(adapter_id) self._disable_adapter(adapter_id)
try: try:
self._session.machine.saveSettings() self._session.machine.saveSettings()
@ -526,4 +554,4 @@ class VirtualBoxController(object):
log.warn("cannot unlock the machine for {}, retrying {}: {}".format(self._vmname, retry + 1, e)) log.warn("cannot unlock the machine for {}, retrying {}: {}".format(self._vmname, retry + 1, e))
time.sleep(1) time.sleep(1)
last_exception = e last_exception = e
continue continue

View File

@ -91,7 +91,9 @@ class VirtualBoxVM(object):
self._console = console self._console = console
self._ethernet_adapters = [] self._ethernet_adapters = []
self._headless = False self._headless = False
self._enable_console = True
self._vmname = vmname self._vmname = vmname
self._adapter_start_index = 0
self._adapter_type = "Automatic" self._adapter_type = "Automatic"
working_dir_path = os.path.join(working_dir, "vbox", "vm-{}".format(self._id)) working_dir_path = os.path.join(working_dir, "vbox", "vm-{}".format(self._id))
@ -137,9 +139,11 @@ class VirtualBoxVM(object):
vbox_defaults = {"name": self._name, vbox_defaults = {"name": self._name,
"vmname": self._vmname, "vmname": self._vmname,
"adapters": len(self._ethernet_adapters), "adapters": self.adapters,
"adapter_start_index": self._adapter_start_index,
"adapter_type": "Automatic", "adapter_type": "Automatic",
"console": self._console, "console": self._console,
"enable_console": self._enable_console,
"headless": self._headless} "headless": self._headless}
return vbox_defaults return vbox_defaults
@ -329,6 +333,38 @@ class VirtualBoxVM(object):
log.info("VirtualBox VM {name} [id={id}] has disabled the headless mode".format(name=self._name, id=self._id)) log.info("VirtualBox VM {name} [id={id}] has disabled the headless mode".format(name=self._name, id=self._id))
self._headless = headless self._headless = headless
@property
def enable_console(self):
"""
Returns either the console is enabled or not
:returns: boolean
"""
return self._enable_console
@enable_console.setter
def enable_console(self, enable_console):
"""
Sets either the console is enabled or not
:param enable_console: boolean
"""
if enable_console:
if self._vboxwrapper:
self._vboxwrapper.send('vbox setattr "{}" enable_console True'.format(self._name))
else:
self._vboxcontroller.enable_console = True
log.info("VirtualBox VM {name} [id={id}] has enabled the console".format(name=self._name, id=self._id))
else:
if self._vboxwrapper:
self._vboxwrapper.send('vbox setattr "{}" enable_console False'.format(self._name))
else:
self._vboxcontroller.enable_console = False
log.info("VirtualBox VM {name} [id={id}] has disabled the console".format(name=self._name, id=self._id))
self._enable_console = enable_console
@property @property
def vmname(self): def vmname(self):
""" """
@ -374,17 +410,47 @@ class VirtualBoxVM(object):
""" """
self._ethernet_adapters.clear() self._ethernet_adapters.clear()
for _ in range(0, adapters): for adapter_id in range(0, self._adapter_start_index + adapters):
if adapter_id < self._adapter_start_index:
self._ethernet_adapters.append(None)
continue
self._ethernet_adapters.append(EthernetAdapter()) self._ethernet_adapters.append(EthernetAdapter())
if self._vboxwrapper: if self._vboxwrapper:
self._vboxwrapper.send('vbox setattr "{}" nics {}'.format(self._name, len(self._ethernet_adapters))) self._vboxwrapper.send('vbox setattr "{}" nics {}'.format(self._name, adapters))
else: else:
self._vboxcontroller.adapters = self._ethernet_adapters self._vboxcontroller.adapters = self._ethernet_adapters
log.info("VirtualBox VM {name} [id={id}]: number of Ethernet adapters changed to {adapters}".format(name=self._name, log.info("VirtualBox VM {name} [id={id}]: number of Ethernet adapters changed to {adapters}".format(name=self._name,
id=self._id, id=self._id,
adapters=len(self._ethernet_adapters))) adapters=adapters))
@property
def adapter_start_index(self):
"""
Returns the adapter start index for this VirtualBox VM instance.
:returns: index
"""
return self._adapter_start_index
@adapter_start_index.setter
def adapter_start_index(self, adapter_start_index):
"""
Sets the adapter start index for this VirtualBox VM instance.
:param adapter_start_index: index
"""
if self._vboxwrapper:
self._vboxwrapper.send('vbox setattr "{}" nic_start_index {}'.format(self._name, adapter_start_index))
self._adapter_start_index = adapter_start_index
self.adapters = self.adapters # this forces to recreate the adapter list with the correct index
log.info("VirtualBox VM {name} [id={id}]: adapter start index changed to {index}".format(name=self._name,
id=self._id,
index=adapter_start_index))
@property @property
def adapter_type(self): def adapter_type(self):