2015-05-01 01:05:37 +00:00
# -*- coding: utf-8 -*-
#
# Copyright (C) 2015 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/>.
2015-09-13 15:40:09 +00:00
import os
2015-05-16 01:09:48 +00:00
from aiohttp . web import HTTPConflict
2016-05-14 00:00:07 +00:00
from gns3server . web . route import Route
from gns3server . schemas . node import NODE_CAPTURE_SCHEMA
from gns3server . schemas . nio import NIO_SCHEMA
from gns3server . compute . vmware import VMware
from gns3server . compute . project_manager import ProjectManager
from gns3server . schemas . vmware import (
VMWARE_CREATE_SCHEMA ,
VMWARE_UPDATE_SCHEMA ,
VMWARE_OBJECT_SCHEMA
)
2015-05-01 01:05:37 +00:00
class VMwareHandler :
"""
API entry points for VMware .
"""
@Route.post (
2016-05-11 17:35:36 +00:00
r " /projects/ {project_id} /vmware/nodes " ,
2015-05-01 01:05:37 +00:00
parameters = {
2016-05-14 00:00:07 +00:00
" project_id " : " Project UUID "
2015-05-01 01:05:37 +00:00
} ,
status_codes = {
201 : " Instance created " ,
400 : " Invalid request " ,
409 : " Conflict "
} ,
description = " Create a new VMware VM instance " ,
input = VMWARE_CREATE_SCHEMA ,
output = VMWARE_OBJECT_SCHEMA )
def create ( request , response ) :
vmware_manager = VMware . instance ( )
2016-05-11 17:35:36 +00:00
vm = yield from vmware_manager . create_node ( request . json . pop ( " name " ) ,
request . match_info [ " project_id " ] ,
request . json . get ( " node_id " ) ,
request . json . pop ( " vmx_path " ) ,
request . json . pop ( " linked_clone " ) ,
console = request . json . get ( " console " , None ) )
2015-05-01 01:05:37 +00:00
2015-05-02 00:47:46 +00:00
for name , value in request . json . items ( ) :
2016-05-11 17:35:36 +00:00
if name != " node_id " :
2015-05-02 00:47:46 +00:00
if hasattr ( vm , name ) and getattr ( vm , name ) != value :
setattr ( vm , name , value )
2015-05-01 01:05:37 +00:00
response . set_status ( 201 )
response . json ( vm )
@Route.get (
2016-05-11 17:35:36 +00:00
r " /projects/ {project_id} /vmware/nodes/ {node_id} " ,
2015-05-01 01:05:37 +00:00
parameters = {
2016-05-14 00:00:07 +00:00
" project_id " : " Project UUID " ,
" node_id " : " Node UUID "
2015-05-01 01:05:37 +00:00
} ,
status_codes = {
200 : " Success " ,
400 : " Invalid request " ,
404 : " Instance doesn ' t exist "
} ,
description = " Get a VMware VM instance " ,
output = VMWARE_OBJECT_SCHEMA )
def show ( request , response ) :
vmware_manager = VMware . instance ( )
2016-05-11 17:35:36 +00:00
vm = vmware_manager . get_node ( request . match_info [ " node_id " ] , project_id = request . match_info [ " project_id " ] )
2015-05-01 01:05:37 +00:00
response . json ( vm )
@Route.put (
2016-05-11 17:35:36 +00:00
r " /projects/ {project_id} /vmware/nodes/ {node_id} " ,
2015-05-01 01:05:37 +00:00
parameters = {
2016-05-14 00:00:07 +00:00
" project_id " : " Project UUID " ,
" node_id " : " Node UUID "
2015-05-01 01:05:37 +00:00
} ,
status_codes = {
200 : " Instance updated " ,
400 : " Invalid request " ,
404 : " Instance doesn ' t exist " ,
409 : " Conflict "
} ,
description = " Update a VMware VM instance " ,
input = VMWARE_UPDATE_SCHEMA ,
output = VMWARE_OBJECT_SCHEMA )
def update ( request , response ) :
vmware_manager = VMware . instance ( )
2016-05-11 17:35:36 +00:00
vm = vmware_manager . get_node ( request . match_info [ " node_id " ] , project_id = request . match_info [ " project_id " ] )
2015-05-01 01:05:37 +00:00
2015-05-02 00:47:46 +00:00
for name , value in request . json . items ( ) :
if hasattr ( vm , name ) and getattr ( vm , name ) != value :
setattr ( vm , name , value )
2015-05-01 01:05:37 +00:00
response . json ( vm )
@Route.delete (
2016-05-11 17:35:36 +00:00
r " /projects/ {project_id} /vmware/nodes/ {node_id} " ,
2015-05-01 01:05:37 +00:00
parameters = {
2016-05-14 00:00:07 +00:00
" project_id " : " Project UUID " ,
" node_id " : " Node UUID "
2015-05-01 01:05:37 +00:00
} ,
status_codes = {
204 : " Instance deleted " ,
400 : " Invalid request " ,
404 : " Instance doesn ' t exist "
} ,
description = " Delete a VMware VM instance " )
def delete ( request , response ) :
# check the project_id exists
ProjectManager . instance ( ) . get_project ( request . match_info [ " project_id " ] )
2016-05-11 17:35:36 +00:00
yield from VMware . instance ( ) . delete_node ( request . match_info [ " node_id " ] )
2015-05-01 01:05:37 +00:00
response . set_status ( 204 )
@Route.post (
2016-05-11 17:35:36 +00:00
r " /projects/ {project_id} /vmware/nodes/ {node_id} /start " ,
2015-05-01 01:05:37 +00:00
parameters = {
2016-05-14 00:00:07 +00:00
" project_id " : " Project UUID " ,
" node_id " : " Node UUID "
2015-05-01 01:05:37 +00:00
} ,
status_codes = {
204 : " Instance started " ,
400 : " Invalid request " ,
404 : " Instance doesn ' t exist "
} ,
description = " Start a VMware VM instance " )
def start ( request , response ) :
vmware_manager = VMware . instance ( )
2016-05-11 17:35:36 +00:00
vm = vmware_manager . get_node ( request . match_info [ " node_id " ] , project_id = request . match_info [ " project_id " ] )
2015-07-27 01:21:30 +00:00
if vm . check_hw_virtualization ( ) :
pm = ProjectManager . instance ( )
if pm . check_hardware_virtualization ( vm ) is False :
raise HTTPConflict ( text = " Cannot start VM because hardware virtualization (VT-x/AMD-V) is already used by another software like VirtualBox or KVM (on Linux) " )
2015-05-01 01:05:37 +00:00
yield from vm . start ( )
response . set_status ( 204 )
@Route.post (
2016-05-11 17:35:36 +00:00
r " /projects/ {project_id} /vmware/nodes/ {node_id} /stop " ,
2015-05-01 01:05:37 +00:00
parameters = {
2016-05-14 00:00:07 +00:00
" project_id " : " Project UUID " ,
" node_id " : " Node UUID "
2015-05-01 01:05:37 +00:00
} ,
status_codes = {
204 : " Instance stopped " ,
400 : " Invalid request " ,
404 : " Instance doesn ' t exist "
} ,
description = " Stop a VMware VM instance " )
def stop ( request , response ) :
vmware_manager = VMware . instance ( )
2016-05-11 17:35:36 +00:00
vm = vmware_manager . get_node ( request . match_info [ " node_id " ] , project_id = request . match_info [ " project_id " ] )
2015-05-01 01:05:37 +00:00
yield from vm . stop ( )
response . set_status ( 204 )
@Route.post (
2016-05-11 17:35:36 +00:00
r " /projects/ {project_id} /vmware/nodes/ {node_id} /suspend " ,
2015-05-01 01:05:37 +00:00
parameters = {
2016-05-14 00:00:07 +00:00
" project_id " : " Project UUID " ,
" node_id " : " Node UUID "
2015-05-01 01:05:37 +00:00
} ,
status_codes = {
204 : " Instance suspended " ,
400 : " Invalid request " ,
404 : " Instance doesn ' t exist "
} ,
description = " Suspend a VMware VM instance " )
def suspend ( request , response ) :
vmware_manager = VMware . instance ( )
2016-05-11 17:35:36 +00:00
vm = vmware_manager . get_node ( request . match_info [ " node_id " ] , project_id = request . match_info [ " project_id " ] )
2015-05-01 01:05:37 +00:00
yield from vm . suspend ( )
response . set_status ( 204 )
@Route.post (
2016-05-11 17:35:36 +00:00
r " /projects/ {project_id} /vmware/nodes/ {node_id} /resume " ,
2015-05-01 01:05:37 +00:00
parameters = {
2016-05-14 00:00:07 +00:00
" project_id " : " Project UUID " ,
" node_id " : " Node UUID "
2015-05-01 01:05:37 +00:00
} ,
status_codes = {
204 : " Instance resumed " ,
400 : " Invalid request " ,
404 : " Instance doesn ' t exist "
} ,
description = " Resume a suspended VMware VM instance " )
def resume ( request , response ) :
vmware_manager = VMware . instance ( )
2016-05-11 17:35:36 +00:00
vm = vmware_manager . get_node ( request . match_info [ " node_id " ] , project_id = request . match_info [ " project_id " ] )
2015-05-01 01:05:37 +00:00
yield from vm . resume ( )
response . set_status ( 204 )
@Route.post (
2016-05-11 17:35:36 +00:00
r " /projects/ {project_id} /vmware/nodes/ {node_id} /reload " ,
2015-05-01 01:05:37 +00:00
parameters = {
2016-05-14 00:00:07 +00:00
" project_id " : " Project UUID " ,
" node_id " : " Node UUID "
2015-05-01 01:05:37 +00:00
} ,
status_codes = {
204 : " Instance reloaded " ,
400 : " Invalid request " ,
404 : " Instance doesn ' t exist "
} ,
description = " Reload a VMware VM instance " )
def reload ( request , response ) :
vmware_manager = VMware . instance ( )
2016-05-11 17:35:36 +00:00
vm = vmware_manager . get_node ( request . match_info [ " node_id " ] , project_id = request . match_info [ " project_id " ] )
2015-05-01 01:05:37 +00:00
yield from vm . reload ( )
response . set_status ( 204 )
2015-05-16 01:09:48 +00:00
@Route.post (
2016-05-11 17:35:36 +00:00
r " /projects/ {project_id} /vmware/nodes/ {node_id} /adapters/ { adapter_number: \ d+}/ports/ { port_number: \ d+}/nio " ,
2015-05-16 01:09:48 +00:00
parameters = {
2016-05-14 00:00:07 +00:00
" project_id " : " Project UUID " ,
" node_id " : " Node UUID " ,
2015-05-16 01:09:48 +00:00
" adapter_number " : " Adapter where the nio should be added " ,
" port_number " : " Port on the adapter (always 0) "
} ,
status_codes = {
201 : " NIO created " ,
400 : " Invalid request " ,
404 : " Instance doesn ' t exist "
} ,
description = " Add a NIO to a VMware VM instance " ,
input = NIO_SCHEMA ,
output = NIO_SCHEMA )
def create_nio ( request , response ) :
vmware_manager = VMware . instance ( )
2016-05-11 17:35:36 +00:00
vm = vmware_manager . get_node ( request . match_info [ " node_id " ] , project_id = request . match_info [ " project_id " ] )
2015-05-16 01:09:48 +00:00
nio_type = request . json [ " type " ]
2015-10-12 23:07:39 +00:00
if nio_type not in ( " nio_udp " , " nio_vmnet " , " nio_nat " ) :
2015-05-16 01:09:48 +00:00
raise HTTPConflict ( text = " NIO of type {} is not supported " . format ( nio_type ) )
nio = vmware_manager . create_nio ( None , request . json )
2015-07-20 04:55:10 +00:00
yield from vm . adapter_add_nio_binding ( int ( request . match_info [ " adapter_number " ] ) , nio )
2015-05-16 01:09:48 +00:00
response . set_status ( 201 )
response . json ( nio )
@Route.delete (
2016-05-11 17:35:36 +00:00
r " /projects/ {project_id} /vmware/nodes/ {node_id} /adapters/ { adapter_number: \ d+}/ports/ { port_number: \ d+}/nio " ,
2015-05-16 01:09:48 +00:00
parameters = {
2016-05-14 00:00:07 +00:00
" project_id " : " Project UUID " ,
" node_id " : " Node UUID " ,
2015-05-16 01:09:48 +00:00
" adapter_number " : " Adapter from where the nio should be removed " ,
" port_number " : " Port on the adapter (always 0) "
} ,
status_codes = {
204 : " NIO deleted " ,
400 : " Invalid request " ,
404 : " Instance doesn ' t exist "
} ,
description = " Remove a NIO from a VMware VM instance " )
def delete_nio ( request , response ) :
vmware_manager = VMware . instance ( )
2016-05-11 17:35:36 +00:00
vm = vmware_manager . get_node ( request . match_info [ " node_id " ] , project_id = request . match_info [ " project_id " ] )
2015-07-20 04:55:10 +00:00
yield from vm . adapter_remove_nio_binding ( int ( request . match_info [ " adapter_number " ] ) )
2015-05-16 01:09:48 +00:00
response . set_status ( 204 )
2015-09-05 20:38:11 +00:00
2015-09-13 15:40:09 +00:00
@Route.post (
2016-05-11 17:35:36 +00:00
r " /projects/ {project_id} /vmware/nodes/ {node_id} /adapters/ { adapter_number: \ d+}/ports/ { port_number: \ d+}/start_capture " ,
2015-09-13 15:40:09 +00:00
parameters = {
2016-05-14 00:00:07 +00:00
" project_id " : " Project UUID " ,
" node_id " : " Node UUID " ,
2015-09-13 15:40:09 +00:00
" adapter_number " : " Adapter to start a packet capture " ,
" port_number " : " Port on the adapter (always 0) "
} ,
status_codes = {
200 : " Capture started " ,
400 : " Invalid request " ,
404 : " Instance doesn ' t exist " ,
} ,
description = " Start a packet capture on a VMware VM instance " ,
2016-05-11 17:35:36 +00:00
input = NODE_CAPTURE_SCHEMA )
2015-09-13 15:40:09 +00:00
def start_capture ( request , response ) :
vmware_manager = VMware . instance ( )
2016-05-11 17:35:36 +00:00
vm = vmware_manager . get_node ( request . match_info [ " node_id " ] , project_id = request . match_info [ " project_id " ] )
2015-09-13 15:40:09 +00:00
adapter_number = int ( request . match_info [ " adapter_number " ] )
pcap_file_path = os . path . join ( vm . project . capture_working_directory ( ) , request . json [ " capture_file_name " ] )
yield from vm . start_capture ( adapter_number , pcap_file_path )
response . json ( { " pcap_file_path " : pcap_file_path } )
@Route.post (
2016-05-11 17:35:36 +00:00
r " /projects/ {project_id} /vmware/nodes/ {node_id} /adapters/ { adapter_number: \ d+}/ports/ { port_number: \ d+}/stop_capture " ,
2015-09-13 15:40:09 +00:00
parameters = {
2016-05-14 00:00:07 +00:00
" project_id " : " Project UUID " ,
" node_id " : " Node UUID " ,
2015-09-13 15:40:09 +00:00
" adapter_number " : " Adapter to stop a packet capture " ,
" port_number " : " Port on the adapter (always 0) "
} ,
status_codes = {
204 : " Capture stopped " ,
400 : " Invalid request " ,
404 : " Instance doesn ' t exist " ,
} ,
description = " Stop a packet capture on a VMware VM instance " )
def stop_capture ( request , response ) :
vmware_manager = VMware . instance ( )
2016-05-11 17:35:36 +00:00
vm = vmware_manager . get_node ( request . match_info [ " node_id " ] , project_id = request . match_info [ " project_id " ] )
2015-09-13 15:40:09 +00:00
adapter_number = int ( request . match_info [ " adapter_number " ] )
yield from vm . stop_capture ( adapter_number )
response . set_status ( 204 )
2015-09-05 20:38:11 +00:00
@Route.post (
2016-05-11 17:35:36 +00:00
r " /projects/ {project_id} /vmware/nodes/ {node_id} /interfaces/vmnet " ,
2015-09-05 20:38:11 +00:00
parameters = {
2016-05-14 00:00:07 +00:00
" project_id " : " Project UUID " ,
" node_id " : " Node UUID " ,
2015-09-05 20:38:11 +00:00
} ,
status_codes = {
201 : " VMnet interface allocated " ,
} ,
description = " Allocate a VMware VMnet interface on the server " )
def allocate_vmnet ( request , response ) :
vmware_manager = VMware . instance ( )
2016-05-11 17:35:36 +00:00
vm = vmware_manager . get_node ( request . match_info [ " node_id " ] , project_id = request . match_info [ " project_id " ] )
2015-09-20 19:19:57 +00:00
vmware_manager . refresh_vmnet_list ( ubridge = False )
2015-09-05 20:38:11 +00:00
vmnet = vmware_manager . allocate_vmnet ( )
2015-10-15 03:39:17 +00:00
vm . vmnets . append ( vmnet )
2015-09-05 20:38:11 +00:00
response . set_status ( 201 )
response . json ( { " vmnet " : vmnet } )
2016-05-14 00:00:07 +00:00
@Route.get (
r " /vmware/vms " ,
status_codes = {
200 : " Success " ,
} ,
description = " Get all VMware VMs available " )
def get_vms ( request , response ) :
vmware_manager = VMware . instance ( )
vms = yield from vmware_manager . list_vms ( )
response . json ( vms )