2016-03-02 10:49:52 +02:00
|
|
|
#!/usr/bin/env python
|
|
|
|
#
|
|
|
|
# Copyright (C) 2016 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/>.
|
|
|
|
|
2016-04-19 16:35:50 +03:00
|
|
|
import os
|
2017-03-16 11:50:08 +02:00
|
|
|
import sys
|
2016-04-19 16:35:50 +03:00
|
|
|
import json
|
2017-03-21 14:35:02 +02:00
|
|
|
import uuid
|
2016-08-29 16:53:10 +03:00
|
|
|
import socket
|
2017-02-02 20:13:47 +02:00
|
|
|
import shutil
|
2016-03-10 11:32:07 +02:00
|
|
|
import asyncio
|
|
|
|
import aiohttp
|
2018-11-17 13:12:46 +02:00
|
|
|
import jsonschema
|
2018-11-18 12:05:16 +02:00
|
|
|
import copy
|
2016-03-02 10:49:52 +02:00
|
|
|
|
|
|
|
from ..config import Config
|
2016-03-10 22:51:29 +02:00
|
|
|
from .project import Project
|
2018-11-28 11:12:57 +02:00
|
|
|
from .template import Template
|
2017-02-01 12:30:14 +02:00
|
|
|
from .appliance import Appliance
|
2016-11-11 11:38:59 +02:00
|
|
|
from .compute import Compute, ComputeError
|
2016-05-18 15:56:23 +03:00
|
|
|
from .notification import Notification
|
2016-06-27 19:33:42 +03:00
|
|
|
from .symbols import Symbols
|
2016-04-19 16:35:50 +03:00
|
|
|
from ..version import __version__
|
2016-06-14 17:07:37 +03:00
|
|
|
from .topology import load_topology
|
2016-08-24 12:36:32 +03:00
|
|
|
from .gns3vm import GNS3VM
|
2017-02-01 12:30:14 +02:00
|
|
|
from ..utils.get_resource import get_resource
|
2018-08-29 12:08:08 +03:00
|
|
|
from ..utils.asyncio import locking
|
2017-04-04 15:23:43 +03:00
|
|
|
from .gns3vm.gns3_vm_error import GNS3VMError
|
2016-04-19 16:35:50 +03:00
|
|
|
|
|
|
|
import logging
|
|
|
|
log = logging.getLogger(__name__)
|
2016-03-02 10:49:52 +02:00
|
|
|
|
|
|
|
|
|
|
|
class Controller:
|
2018-03-15 09:17:39 +02:00
|
|
|
"""
|
2018-11-27 13:14:51 +02:00
|
|
|
The controller is responsible to manage one or more computes.
|
2018-03-15 09:17:39 +02:00
|
|
|
"""
|
2016-03-02 10:49:52 +02:00
|
|
|
|
2016-03-03 17:02:27 +02:00
|
|
|
def __init__(self):
|
2016-04-15 18:57:06 +03:00
|
|
|
self._computes = {}
|
2016-03-10 11:32:07 +02:00
|
|
|
self._projects = {}
|
2017-02-08 16:40:56 +02:00
|
|
|
|
2016-05-18 15:56:23 +03:00
|
|
|
self._notification = Notification(self)
|
2016-08-24 12:36:32 +03:00
|
|
|
self.gns3vm = GNS3VM(self)
|
2016-06-27 19:33:42 +03:00
|
|
|
self.symbols = Symbols()
|
2018-11-14 10:24:30 +02:00
|
|
|
self._iou_license_settings = {"iourc_content": "",
|
|
|
|
"license_check": True}
|
|
|
|
self._config_loaded = False
|
2018-11-28 11:12:57 +02:00
|
|
|
self._templates = {}
|
2017-04-12 15:35:49 +03:00
|
|
|
self._appliances = {}
|
2018-11-28 11:12:57 +02:00
|
|
|
self._appliances_etag = None
|
2017-04-12 15:35:49 +03:00
|
|
|
|
2016-09-30 18:34:28 +03:00
|
|
|
self._config_file = os.path.join(Config.instance().config_dir, "gns3_controller.conf")
|
2016-08-22 18:21:03 +03:00
|
|
|
log.info("Load controller configuration file {}".format(self._config_file))
|
|
|
|
|
2018-08-29 12:08:08 +03:00
|
|
|
@locking
|
2018-11-28 11:12:57 +02:00
|
|
|
async def download_appliances(self):
|
2018-08-19 12:02:47 +03:00
|
|
|
|
|
|
|
try:
|
2018-08-21 12:26:07 +03:00
|
|
|
headers = {}
|
2018-11-28 11:12:57 +02:00
|
|
|
if self._appliances_etag:
|
|
|
|
log.info("Checking if appliances are up-to-date (ETag {})".format(self._appliances_etag))
|
|
|
|
headers["If-None-Match"] = self._appliances_etag
|
2018-10-16 11:56:06 +03:00
|
|
|
async with aiohttp.ClientSession() as session:
|
|
|
|
async with session.get('https://api.github.com/repos/GNS3/gns3-registry/contents/appliances', headers=headers) as response:
|
|
|
|
if response.status == 304:
|
2018-11-28 11:12:57 +02:00
|
|
|
log.info("Appliances are already up-to-date (ETag {})".format(self._appliances_etag))
|
2018-10-16 11:56:06 +03:00
|
|
|
return
|
|
|
|
elif response.status != 200:
|
2018-11-28 11:12:57 +02:00
|
|
|
raise aiohttp.web.HTTPConflict(text="Could not retrieve appliances on GitHub due to HTTP error code {}".format(response.status))
|
2018-10-16 11:56:06 +03:00
|
|
|
etag = response.headers.get("ETag")
|
|
|
|
if etag:
|
2018-11-28 11:12:57 +02:00
|
|
|
self._appliances_etag = etag
|
2018-10-16 11:56:06 +03:00
|
|
|
self.save()
|
|
|
|
json_data = await response.json()
|
|
|
|
appliances_dir = get_resource('appliances')
|
|
|
|
for appliance in json_data:
|
|
|
|
if appliance["type"] == "file":
|
|
|
|
appliance_name = appliance["name"]
|
2018-11-28 11:12:57 +02:00
|
|
|
log.info("Download appliance file from '{}'".format(appliance["download_url"]))
|
2018-10-16 11:56:06 +03:00
|
|
|
async with session.get(appliance["download_url"]) as response:
|
|
|
|
if response.status != 200:
|
|
|
|
log.warning("Could not download '{}' due to HTTP error code {}".format(appliance["download_url"], response.status))
|
|
|
|
continue
|
|
|
|
try:
|
|
|
|
appliance_data = await response.read()
|
|
|
|
except asyncio.TimeoutError:
|
|
|
|
log.warning("Timeout while downloading '{}'".format(appliance["download_url"]))
|
|
|
|
continue
|
|
|
|
path = os.path.join(appliances_dir, appliance_name)
|
|
|
|
try:
|
|
|
|
log.info("Saving {} file to {}".format(appliance_name, path))
|
|
|
|
with open(path, 'wb') as f:
|
|
|
|
f.write(appliance_data)
|
|
|
|
except OSError as e:
|
2018-11-28 11:12:57 +02:00
|
|
|
raise aiohttp.web.HTTPConflict(text="Could not write appliance file '{}': {}".format(path, e))
|
2018-08-19 12:02:47 +03:00
|
|
|
except ValueError as e:
|
2018-11-28 11:12:57 +02:00
|
|
|
raise aiohttp.web.HTTPConflict(text="Could not read appliances information from GitHub: {}".format(e))
|
2018-08-19 12:02:47 +03:00
|
|
|
|
2018-11-28 11:12:57 +02:00
|
|
|
def load_appliances(self):
|
2018-03-15 09:17:39 +02:00
|
|
|
|
2018-11-28 11:12:57 +02:00
|
|
|
self._appliances = {}
|
2018-08-19 12:02:47 +03:00
|
|
|
for directory, builtin in ((get_resource('appliances'), True,), (self.appliances_path(), False,)):
|
2018-09-06 10:49:12 +03:00
|
|
|
if directory and os.path.isdir(directory):
|
2017-07-19 12:44:05 +03:00
|
|
|
for file in os.listdir(directory):
|
2017-07-20 17:10:56 +03:00
|
|
|
if not file.endswith('.gns3a') and not file.endswith('.gns3appliance'):
|
|
|
|
continue
|
2017-07-19 12:44:05 +03:00
|
|
|
path = os.path.join(directory, file)
|
2018-03-15 09:17:39 +02:00
|
|
|
appliance_id = uuid.uuid3(uuid.NAMESPACE_URL, path) # Generate UUID from path to avoid change between reboots
|
2017-07-20 17:10:56 +03:00
|
|
|
try:
|
|
|
|
with open(path, 'r', encoding='utf-8') as f:
|
2018-11-28 11:12:57 +02:00
|
|
|
appliance = Appliance(appliance_id, json.load(f), builtin=builtin)
|
2017-07-28 16:21:35 +03:00
|
|
|
appliance.__json__() # Check if loaded without error
|
|
|
|
if appliance.status != 'broken':
|
2018-11-28 11:12:57 +02:00
|
|
|
self._appliances[appliance.id] = appliance
|
2017-07-28 16:21:35 +03:00
|
|
|
except (ValueError, OSError, KeyError) as e:
|
2018-11-28 11:12:57 +02:00
|
|
|
log.warning("Cannot load appliance file '%s': %s", path, str(e))
|
2018-03-12 08:38:50 +02:00
|
|
|
continue
|
2017-02-01 12:30:14 +02:00
|
|
|
|
2018-11-28 11:12:57 +02:00
|
|
|
def add_template(self, settings):
|
2018-11-11 14:13:58 +02:00
|
|
|
"""
|
2018-11-28 11:12:57 +02:00
|
|
|
Adds a new template.
|
2018-11-11 14:13:58 +02:00
|
|
|
|
2018-11-28 11:12:57 +02:00
|
|
|
:param settings: template settings
|
2018-11-11 14:13:58 +02:00
|
|
|
|
2018-11-28 11:12:57 +02:00
|
|
|
:returns: Template object
|
2018-11-11 14:13:58 +02:00
|
|
|
"""
|
|
|
|
|
2018-11-28 11:12:57 +02:00
|
|
|
template_id = settings.get("template_id", "")
|
|
|
|
if template_id in self._templates:
|
|
|
|
raise aiohttp.web.HTTPConflict(text="Template ID '{}' already exists".format(template_id))
|
2018-11-11 14:13:58 +02:00
|
|
|
else:
|
2018-11-28 11:12:57 +02:00
|
|
|
template_id = settings.setdefault("template_id", str(uuid.uuid4()))
|
2018-11-11 14:13:58 +02:00
|
|
|
try:
|
2018-11-28 11:12:57 +02:00
|
|
|
template = Template(template_id, settings)
|
2018-11-17 13:12:46 +02:00
|
|
|
except jsonschema.ValidationError as e:
|
2018-11-28 11:12:57 +02:00
|
|
|
message = "JSON schema error adding template with JSON data '{}': {}".format(settings, e.message)
|
2018-11-17 13:12:46 +02:00
|
|
|
raise aiohttp.web.HTTPBadRequest(text=message)
|
2018-11-28 11:12:57 +02:00
|
|
|
self._templates[template.id] = template
|
2018-11-11 14:13:58 +02:00
|
|
|
self.save()
|
2018-11-28 11:12:57 +02:00
|
|
|
self.notification.controller_emit("template.created", template.__json__())
|
|
|
|
return template
|
2018-11-11 14:13:58 +02:00
|
|
|
|
2018-11-28 11:12:57 +02:00
|
|
|
def get_template(self, template_id):
|
2018-11-11 14:13:58 +02:00
|
|
|
"""
|
2018-11-28 11:12:57 +02:00
|
|
|
Gets a template.
|
2018-11-11 14:13:58 +02:00
|
|
|
|
2018-11-28 11:12:57 +02:00
|
|
|
:param template_id: template identifier
|
2018-11-11 14:13:58 +02:00
|
|
|
|
2018-11-28 11:12:57 +02:00
|
|
|
:returns: Template object
|
2018-11-11 14:13:58 +02:00
|
|
|
"""
|
|
|
|
|
2018-11-28 11:12:57 +02:00
|
|
|
template = self._templates.get(template_id)
|
|
|
|
if not template:
|
|
|
|
raise aiohttp.web.HTTPNotFound(text="Template ID {} doesn't exist".format(template_id))
|
|
|
|
return template
|
2018-11-11 14:13:58 +02:00
|
|
|
|
2018-11-28 11:12:57 +02:00
|
|
|
def delete_template(self, template_id):
|
2018-11-11 14:13:58 +02:00
|
|
|
"""
|
2018-11-28 11:12:57 +02:00
|
|
|
Deletes a template.
|
2018-11-11 14:13:58 +02:00
|
|
|
|
2018-11-28 11:12:57 +02:00
|
|
|
:param template_id: template identifier
|
2018-11-11 14:13:58 +02:00
|
|
|
"""
|
|
|
|
|
2018-11-28 11:12:57 +02:00
|
|
|
template = self.get_template(template_id)
|
|
|
|
if template.builtin:
|
|
|
|
raise aiohttp.web.HTTPConflict(text="Template ID {} cannot be deleted because it is a builtin".format(template_id))
|
|
|
|
self._templates.pop(template_id)
|
2018-11-13 08:59:18 +02:00
|
|
|
self.save()
|
2018-11-28 11:12:57 +02:00
|
|
|
self.notification.controller_emit("template.deleted", template.__json__())
|
2018-11-11 14:13:58 +02:00
|
|
|
|
2018-11-28 11:12:57 +02:00
|
|
|
def duplicate_template(self, template_id):
|
2018-11-18 12:05:16 +02:00
|
|
|
"""
|
2018-11-28 11:12:57 +02:00
|
|
|
Duplicates a template.
|
2018-11-18 12:05:16 +02:00
|
|
|
|
2018-11-28 11:12:57 +02:00
|
|
|
:param template_id: template identifier
|
2018-11-18 12:05:16 +02:00
|
|
|
"""
|
|
|
|
|
2018-11-28 11:12:57 +02:00
|
|
|
template = self.get_template(template_id)
|
|
|
|
if template.builtin:
|
|
|
|
raise aiohttp.web.HTTPConflict(text="Template ID {} cannot be duplicated because it is a builtin".format(template_id))
|
|
|
|
template_settings = copy.deepcopy(template.settings)
|
|
|
|
del template_settings["template_id"]
|
|
|
|
return self.add_template(template_settings)
|
2018-11-18 12:05:16 +02:00
|
|
|
|
2018-11-28 11:12:57 +02:00
|
|
|
def load_templates(self):
|
2017-06-19 09:47:50 +03:00
|
|
|
|
2017-04-12 15:35:49 +03:00
|
|
|
# Add builtins
|
|
|
|
builtins = []
|
2018-11-28 11:12:57 +02:00
|
|
|
builtins.append(Template(uuid.uuid3(uuid.NAMESPACE_DNS, "cloud"), {"template_type": "cloud", "name": "Cloud", "category": 2, "symbol": ":/symbols/cloud.svg"}, builtin=True))
|
|
|
|
builtins.append(Template(uuid.uuid3(uuid.NAMESPACE_DNS, "nat"), {"template_type": "nat", "name": "NAT", "category": 2, "symbol": ":/symbols/cloud.svg"}, builtin=True))
|
|
|
|
builtins.append(Template(uuid.uuid3(uuid.NAMESPACE_DNS, "vpcs"), {"template_type": "vpcs", "name": "VPCS", "default_name_format": "PC-{0}", "category": 2, "symbol": ":/symbols/vpcs_guest.svg", "properties": {"base_script_file": "vpcs_base_config.txt"}}, builtin=True))
|
|
|
|
builtins.append(Template(uuid.uuid3(uuid.NAMESPACE_DNS, "ethernet_switch"), {"template_type": "ethernet_switch", "console_type": "telnet", "name": "Ethernet switch", "category": 1, "symbol": ":/symbols/ethernet_switch.svg"}, builtin=True))
|
|
|
|
builtins.append(Template(uuid.uuid3(uuid.NAMESPACE_DNS, "ethernet_hub"), {"template_type": "ethernet_hub", "name": "Ethernet hub", "category": 1, "symbol": ":/symbols/hub.svg"}, builtin=True))
|
|
|
|
builtins.append(Template(uuid.uuid3(uuid.NAMESPACE_DNS, "frame_relay_switch"), {"template_type": "frame_relay_switch", "name": "Frame Relay switch", "category": 1, "symbol": ":/symbols/frame_relay_switch.svg"}, builtin=True))
|
|
|
|
builtins.append(Template(uuid.uuid3(uuid.NAMESPACE_DNS, "atm_switch"), {"template_type": "atm_switch", "name": "ATM switch", "category": 1, "symbol": ":/symbols/atm_switch.svg"}, builtin=True))
|
2018-04-18 12:19:44 +03:00
|
|
|
|
|
|
|
#FIXME: disable TraceNG
|
|
|
|
#if sys.platform.startswith("win"):
|
2018-11-28 11:12:57 +02:00
|
|
|
# builtins.append(Template(uuid.uuid3(uuid.NAMESPACE_DNS, "traceng"), {"template_type": "traceng", "name": "TraceNG", "default_name_format": "TraceNG-{0}", "category": 2, "symbol": ":/symbols/traceng.svg", "properties": {}}, builtin=True))
|
2017-04-12 15:35:49 +03:00
|
|
|
for b in builtins:
|
2018-11-28 11:12:57 +02:00
|
|
|
self._templates[b.id] = b
|
2017-04-12 15:35:49 +03:00
|
|
|
|
2018-10-15 13:05:49 +03:00
|
|
|
async def start(self):
|
2018-03-15 09:17:39 +02:00
|
|
|
|
|
|
|
log.info("Controller is starting")
|
2017-02-02 20:13:47 +02:00
|
|
|
self.load_base_files()
|
2016-08-24 12:36:32 +03:00
|
|
|
server_config = Config.instance().get_section_config("Server")
|
2017-03-22 19:29:08 +02:00
|
|
|
Config.instance().listen_for_config_changes(self._update_config)
|
2016-09-27 12:54:23 +03:00
|
|
|
host = server_config.get("host", "localhost")
|
2018-03-15 09:17:39 +02:00
|
|
|
port = server_config.getint("port", 3080)
|
2017-02-28 13:08:47 +02:00
|
|
|
|
2018-03-15 09:17:39 +02:00
|
|
|
# clients will use the IP they use to connect to
|
|
|
|
# the controller if console_host is 0.0.0.0
|
2016-10-26 15:43:47 +03:00
|
|
|
console_host = host
|
2016-09-27 12:54:23 +03:00
|
|
|
if host == "0.0.0.0":
|
|
|
|
host = "127.0.0.1"
|
2017-02-28 14:11:03 +02:00
|
|
|
|
|
|
|
name = socket.gethostname()
|
|
|
|
if name == "gns3vm":
|
|
|
|
name = "Main server"
|
|
|
|
|
2018-10-15 13:05:49 +03:00
|
|
|
computes = await self._load_controller_settings()
|
2017-03-16 11:50:08 +02:00
|
|
|
try:
|
2018-10-15 13:05:49 +03:00
|
|
|
self._local_server = await self.add_compute(compute_id="local",
|
2018-11-11 14:13:58 +02:00
|
|
|
name=name,
|
|
|
|
protocol=server_config.get("protocol", "http"),
|
|
|
|
host=host,
|
|
|
|
console_host=console_host,
|
|
|
|
port=port,
|
|
|
|
user=server_config.get("user", ""),
|
|
|
|
password=server_config.get("password", ""),
|
|
|
|
force=True)
|
2018-11-14 10:24:30 +02:00
|
|
|
except aiohttp.web.HTTPConflict:
|
2018-03-15 09:17:39 +02:00
|
|
|
log.fatal("Cannot access to the local server, make sure something else is not running on the TCP port {}".format(port))
|
2017-03-16 11:50:08 +02:00
|
|
|
sys.exit(1)
|
2017-03-07 16:36:35 +02:00
|
|
|
for c in computes:
|
|
|
|
try:
|
2018-10-15 13:05:49 +03:00
|
|
|
await self.add_compute(**c)
|
2018-09-11 16:06:01 +03:00
|
|
|
except (aiohttp.web.HTTPError, KeyError):
|
2017-03-07 16:36:35 +02:00
|
|
|
pass # Skip not available servers at loading
|
2018-10-15 13:05:49 +03:00
|
|
|
await self.load_projects()
|
2017-04-04 15:23:43 +03:00
|
|
|
try:
|
2018-10-15 13:05:49 +03:00
|
|
|
await self.gns3vm.auto_start_vm()
|
2017-04-04 15:23:43 +03:00
|
|
|
except GNS3VMError as e:
|
2018-03-15 09:17:39 +02:00
|
|
|
log.warning(str(e))
|
2018-10-15 13:05:49 +03:00
|
|
|
await self._project_auto_open()
|
2016-08-24 12:36:32 +03:00
|
|
|
|
2017-03-22 19:29:08 +02:00
|
|
|
def _update_config(self):
|
|
|
|
"""
|
2018-03-15 09:17:39 +02:00
|
|
|
Call this when the server configuration file changes.
|
2017-03-22 19:29:08 +02:00
|
|
|
"""
|
2018-03-15 09:17:39 +02:00
|
|
|
|
2017-03-22 19:29:08 +02:00
|
|
|
if self._local_server:
|
|
|
|
server_config = Config.instance().get_section_config("Server")
|
|
|
|
self._local_server.user = server_config.get("user")
|
|
|
|
self._local_server.password = server_config.get("password")
|
|
|
|
|
2018-10-15 13:05:49 +03:00
|
|
|
async def stop(self):
|
2018-03-15 09:17:39 +02:00
|
|
|
|
|
|
|
log.info("Controller is Stopping")
|
2016-08-26 12:22:09 +03:00
|
|
|
for project in self._projects.values():
|
2018-10-15 13:05:49 +03:00
|
|
|
await project.close()
|
2016-08-24 12:36:32 +03:00
|
|
|
for compute in self._computes.values():
|
2016-08-26 15:09:18 +03:00
|
|
|
try:
|
2018-10-15 13:05:49 +03:00
|
|
|
await compute.close()
|
2016-08-26 15:09:18 +03:00
|
|
|
# We don't care if a compute is down at this step
|
2018-04-28 12:01:43 +03:00
|
|
|
except (ComputeError, aiohttp.web.HTTPError, OSError):
|
2016-08-26 15:09:18 +03:00
|
|
|
pass
|
2018-10-15 13:05:49 +03:00
|
|
|
await self.gns3vm.exit_vm()
|
2016-08-24 12:36:32 +03:00
|
|
|
self._computes = {}
|
|
|
|
self._projects = {}
|
2016-05-24 16:45:06 +03:00
|
|
|
|
2016-04-19 16:35:50 +03:00
|
|
|
def save(self):
|
|
|
|
"""
|
|
|
|
Save the controller configuration on disk
|
|
|
|
"""
|
2018-03-15 09:17:39 +02:00
|
|
|
|
2018-11-14 10:24:30 +02:00
|
|
|
if self._config_loaded is False:
|
2017-03-07 16:36:35 +02:00
|
|
|
return
|
2018-11-11 14:13:58 +02:00
|
|
|
|
|
|
|
controller_settings = {"computes": [],
|
2018-11-28 11:12:57 +02:00
|
|
|
"templates": [],
|
2018-11-11 14:13:58 +02:00
|
|
|
"gns3vm": self.gns3vm.__json__(),
|
2018-11-14 10:24:30 +02:00
|
|
|
"iou_license": self._iou_license_settings,
|
2018-11-28 11:12:57 +02:00
|
|
|
"appliances_etag": self._appliances_etag,
|
2018-11-11 14:13:58 +02:00
|
|
|
"version": __version__}
|
|
|
|
|
2018-11-28 11:12:57 +02:00
|
|
|
for template in self._templates.values():
|
|
|
|
if not template.builtin:
|
|
|
|
controller_settings["templates"].append(template.__json__())
|
2018-11-11 14:13:58 +02:00
|
|
|
|
|
|
|
for compute in self._computes.values():
|
|
|
|
if compute.id != "local" and compute.id != "vm":
|
|
|
|
controller_settings["computes"].append({"host": compute.host,
|
|
|
|
"name": compute.name,
|
|
|
|
"port": compute.port,
|
|
|
|
"protocol": compute.protocol,
|
|
|
|
"user": compute.user,
|
|
|
|
"password": compute.password,
|
|
|
|
"compute_id": compute.id})
|
|
|
|
|
2017-04-10 18:44:09 +03:00
|
|
|
try:
|
|
|
|
os.makedirs(os.path.dirname(self._config_file), exist_ok=True)
|
|
|
|
with open(self._config_file, 'w+') as f:
|
2018-11-11 14:13:58 +02:00
|
|
|
json.dump(controller_settings, f, indent=4)
|
2017-04-10 18:44:09 +03:00
|
|
|
except OSError as e:
|
2018-11-11 14:13:58 +02:00
|
|
|
log.error("Cannot write controller configuration file '{}': {}".format(self._config_file, e))
|
2016-04-19 16:35:50 +03:00
|
|
|
|
2018-10-15 13:05:49 +03:00
|
|
|
async def _load_controller_settings(self):
|
2016-04-19 16:35:50 +03:00
|
|
|
"""
|
|
|
|
Reload the controller configuration from disk
|
|
|
|
"""
|
2018-03-15 09:17:39 +02:00
|
|
|
|
2016-04-19 16:35:50 +03:00
|
|
|
try:
|
2016-12-08 11:52:21 +02:00
|
|
|
if not os.path.exists(self._config_file):
|
2018-10-15 13:05:49 +03:00
|
|
|
await self._import_gns3_gui_conf()
|
2016-12-08 11:52:21 +02:00
|
|
|
self.save()
|
2016-04-19 16:35:50 +03:00
|
|
|
with open(self._config_file) as f:
|
2018-11-11 14:13:58 +02:00
|
|
|
controller_settings = json.load(f)
|
2016-12-21 10:45:24 +02:00
|
|
|
except (OSError, ValueError) as e:
|
2018-03-15 09:17:39 +02:00
|
|
|
log.critical("Cannot load configuration file '{}': {}".format(self._config_file, e))
|
2017-03-07 16:36:35 +02:00
|
|
|
return []
|
2016-12-08 11:52:21 +02:00
|
|
|
|
2018-11-28 11:12:57 +02:00
|
|
|
# load the templates
|
|
|
|
if "templates" in controller_settings:
|
|
|
|
for template_settings in controller_settings["templates"]:
|
2018-11-11 14:13:58 +02:00
|
|
|
try:
|
2018-11-28 11:12:57 +02:00
|
|
|
template = Template(template_settings.get("template_id"), template_settings)
|
|
|
|
self._templates[template.id] = template
|
2018-11-17 13:12:46 +02:00
|
|
|
except jsonschema.ValidationError as e:
|
2018-11-28 11:12:57 +02:00
|
|
|
message = "Cannot load template with JSON data '{}': {}".format(template_settings, e.message)
|
2018-11-17 13:12:46 +02:00
|
|
|
log.warning(message)
|
2018-11-11 14:13:58 +02:00
|
|
|
continue
|
|
|
|
|
|
|
|
# load GNS3 VM settings
|
|
|
|
if "gns3vm" in controller_settings:
|
|
|
|
self.gns3vm.settings = controller_settings["gns3vm"]
|
|
|
|
|
2018-11-14 10:24:30 +02:00
|
|
|
# load the IOU license settings
|
|
|
|
if "iou_license" in controller_settings:
|
|
|
|
self._iou_license_settings = controller_settings["iou_license"]
|
|
|
|
|
2018-11-28 11:12:57 +02:00
|
|
|
self._appliances_etag = controller_settings.get("appliances_etag")
|
2017-04-12 15:35:49 +03:00
|
|
|
self.load_appliances()
|
2018-11-28 11:12:57 +02:00
|
|
|
self.load_templates()
|
2018-11-14 10:24:30 +02:00
|
|
|
self._config_loaded = True
|
2018-11-11 14:13:58 +02:00
|
|
|
return controller_settings.get("computes", [])
|
2016-04-19 16:35:50 +03:00
|
|
|
|
2018-10-15 13:05:49 +03:00
|
|
|
async def load_projects(self):
|
2016-08-16 17:04:20 +03:00
|
|
|
"""
|
|
|
|
Preload the list of projects from disk
|
|
|
|
"""
|
2018-03-15 09:17:39 +02:00
|
|
|
|
2016-06-15 00:08:30 +03:00
|
|
|
server_config = Config.instance().get_section_config("Server")
|
|
|
|
projects_path = os.path.expanduser(server_config.get("projects_path", "~/GNS3/projects"))
|
2016-06-15 16:12:38 +03:00
|
|
|
os.makedirs(projects_path, exist_ok=True)
|
2016-06-15 00:08:30 +03:00
|
|
|
try:
|
|
|
|
for project_path in os.listdir(projects_path):
|
|
|
|
project_dir = os.path.join(projects_path, project_path)
|
|
|
|
if os.path.isdir(project_dir):
|
|
|
|
for file in os.listdir(project_dir):
|
|
|
|
if file.endswith(".gns3"):
|
|
|
|
try:
|
2018-10-15 13:05:49 +03:00
|
|
|
await self.load_project(os.path.join(project_dir, file), load=False)
|
2018-04-28 12:01:43 +03:00
|
|
|
except (aiohttp.web.HTTPConflict, NotImplementedError):
|
2016-06-15 16:12:38 +03:00
|
|
|
pass # Skip not compatible projects
|
2016-06-15 00:08:30 +03:00
|
|
|
except OSError as e:
|
|
|
|
log.error(str(e))
|
|
|
|
|
2017-02-02 20:13:47 +02:00
|
|
|
def load_base_files(self):
|
|
|
|
"""
|
|
|
|
At startup we copy base file to the user location to allow
|
|
|
|
them to customize it
|
|
|
|
"""
|
2018-03-15 09:17:39 +02:00
|
|
|
|
2017-02-02 20:13:47 +02:00
|
|
|
dst_path = self.configs_path()
|
|
|
|
src_path = get_resource('configs')
|
|
|
|
try:
|
|
|
|
for file in os.listdir(src_path):
|
|
|
|
if not os.path.exists(os.path.join(dst_path, file)):
|
|
|
|
shutil.copy(os.path.join(src_path, file), os.path.join(dst_path, file))
|
|
|
|
except OSError:
|
|
|
|
pass
|
|
|
|
|
2016-07-21 18:55:15 +03:00
|
|
|
def images_path(self):
|
|
|
|
"""
|
|
|
|
Get the image storage directory
|
|
|
|
"""
|
2018-03-15 09:17:39 +02:00
|
|
|
|
2016-07-21 18:55:15 +03:00
|
|
|
server_config = Config.instance().get_section_config("Server")
|
|
|
|
images_path = os.path.expanduser(server_config.get("images_path", "~/GNS3/projects"))
|
|
|
|
os.makedirs(images_path, exist_ok=True)
|
|
|
|
return images_path
|
|
|
|
|
2017-02-02 20:13:47 +02:00
|
|
|
def configs_path(self):
|
|
|
|
"""
|
|
|
|
Get the configs storage directory
|
|
|
|
"""
|
2018-03-15 09:17:39 +02:00
|
|
|
|
2017-02-02 20:13:47 +02:00
|
|
|
server_config = Config.instance().get_section_config("Server")
|
|
|
|
images_path = os.path.expanduser(server_config.get("configs_path", "~/GNS3/projects"))
|
|
|
|
os.makedirs(images_path, exist_ok=True)
|
|
|
|
return images_path
|
|
|
|
|
2017-07-19 12:44:05 +03:00
|
|
|
def appliances_path(self):
|
|
|
|
"""
|
|
|
|
Get the image storage directory
|
|
|
|
"""
|
2018-03-15 09:17:39 +02:00
|
|
|
|
2017-07-19 12:44:05 +03:00
|
|
|
server_config = Config.instance().get_section_config("Server")
|
|
|
|
appliances_path = os.path.expanduser(server_config.get("appliances_path", "~/GNS3/projects"))
|
|
|
|
os.makedirs(appliances_path, exist_ok=True)
|
|
|
|
return appliances_path
|
|
|
|
|
2018-10-15 13:05:49 +03:00
|
|
|
async def _import_gns3_gui_conf(self):
|
2016-07-14 16:14:56 +03:00
|
|
|
"""
|
|
|
|
Import old config from GNS3 GUI
|
|
|
|
"""
|
2018-03-15 09:17:39 +02:00
|
|
|
|
2016-07-14 16:14:56 +03:00
|
|
|
config_file = os.path.join(os.path.dirname(self._config_file), "gns3_gui.conf")
|
|
|
|
if os.path.exists(config_file):
|
|
|
|
with open(config_file) as f:
|
2018-11-11 14:13:58 +02:00
|
|
|
settings = json.load(f)
|
|
|
|
server_settings = settings.get("Servers", {})
|
2016-09-06 14:06:20 +03:00
|
|
|
for remote in server_settings.get("remote_servers", []):
|
2017-03-13 19:17:17 +02:00
|
|
|
try:
|
2018-11-11 14:13:58 +02:00
|
|
|
await self.add_compute(host=remote.get("host", "localhost"),
|
|
|
|
port=remote.get("port", 3080),
|
|
|
|
protocol=remote.get("protocol", "http"),
|
|
|
|
name=remote.get("url"),
|
|
|
|
user=remote.get("user"),
|
|
|
|
password=remote.get("password"))
|
2017-03-13 19:17:17 +02:00
|
|
|
except aiohttp.web.HTTPConflict:
|
|
|
|
pass # if the server is broken we skip it
|
2016-09-06 14:06:20 +03:00
|
|
|
if "vm" in server_settings:
|
2017-01-18 10:39:27 +02:00
|
|
|
vmname = None
|
2016-09-06 14:06:20 +03:00
|
|
|
vm_settings = server_settings["vm"]
|
|
|
|
if vm_settings["virtualization"] == "VMware":
|
|
|
|
engine = "vmware"
|
|
|
|
vmname = vm_settings.get("vmname", "")
|
|
|
|
elif vm_settings["virtualization"] == "VirtualBox":
|
|
|
|
engine = "virtualbox"
|
|
|
|
vmname = vm_settings.get("vmname", "")
|
|
|
|
else:
|
|
|
|
engine = "remote"
|
|
|
|
# In case of remote server we match the compute with url parameter
|
|
|
|
for compute in self._computes.values():
|
|
|
|
if compute.host == vm_settings.get("remote_vm_host") and compute.port == vm_settings.get("remote_vm_port"):
|
|
|
|
vmname = compute.name
|
2016-09-08 16:32:35 +03:00
|
|
|
|
|
|
|
if vm_settings.get("auto_stop", True):
|
|
|
|
when_exit = "stop"
|
|
|
|
else:
|
|
|
|
when_exit = "keep"
|
|
|
|
|
2016-09-06 14:06:20 +03:00
|
|
|
self.gns3vm.settings = {
|
|
|
|
"engine": engine,
|
|
|
|
"enable": vm_settings.get("auto_start", False),
|
2016-09-08 16:32:35 +03:00
|
|
|
"when_exit": when_exit,
|
2016-09-06 14:06:20 +03:00
|
|
|
"headless": vm_settings.get("headless", False),
|
|
|
|
"vmname": vmname
|
|
|
|
}
|
2018-11-11 14:13:58 +02:00
|
|
|
|
|
|
|
vms = []
|
|
|
|
for vm in settings.get("Qemu", {}).get("vms", []):
|
2018-11-28 11:12:57 +02:00
|
|
|
vm["template_type"] = "qemu"
|
2018-11-11 14:13:58 +02:00
|
|
|
vms.append(vm)
|
|
|
|
for vm in settings.get("IOU", {}).get("devices", []):
|
2018-11-28 11:12:57 +02:00
|
|
|
vm["template_type"] = "iou"
|
2018-11-11 14:13:58 +02:00
|
|
|
vms.append(vm)
|
|
|
|
for vm in settings.get("Docker", {}).get("containers", []):
|
2018-11-28 11:12:57 +02:00
|
|
|
vm["template_type"] = "docker"
|
2018-11-11 14:13:58 +02:00
|
|
|
vms.append(vm)
|
|
|
|
for vm in settings.get("Builtin", {}).get("cloud_nodes", []):
|
2018-11-28 11:12:57 +02:00
|
|
|
vm["template_type"] = "cloud"
|
2018-11-11 14:13:58 +02:00
|
|
|
vms.append(vm)
|
|
|
|
for vm in settings.get("Builtin", {}).get("ethernet_switches", []):
|
2018-11-28 11:12:57 +02:00
|
|
|
vm["template_type"] = "ethernet_switch"
|
2018-11-11 14:13:58 +02:00
|
|
|
vms.append(vm)
|
|
|
|
for vm in settings.get("Builtin", {}).get("ethernet_hubs", []):
|
2018-11-28 11:12:57 +02:00
|
|
|
vm["template_type"] = "ethernet_hub"
|
2018-11-11 14:13:58 +02:00
|
|
|
vms.append(vm)
|
|
|
|
for vm in settings.get("Dynamips", {}).get("routers", []):
|
2018-11-28 11:12:57 +02:00
|
|
|
vm["template_type"] = "dynamips"
|
2018-11-11 14:13:58 +02:00
|
|
|
vms.append(vm)
|
|
|
|
for vm in settings.get("VMware", {}).get("vms", []):
|
2018-11-28 11:12:57 +02:00
|
|
|
vm["template_type"] = "vmware"
|
2018-11-11 14:13:58 +02:00
|
|
|
vms.append(vm)
|
|
|
|
for vm in settings.get("VirtualBox", {}).get("vms", []):
|
2018-11-28 11:12:57 +02:00
|
|
|
vm["template_type"] = "virtualbox"
|
2018-11-11 14:13:58 +02:00
|
|
|
vms.append(vm)
|
|
|
|
for vm in settings.get("VPCS", {}).get("nodes", []):
|
2018-11-28 11:12:57 +02:00
|
|
|
vm["template_type"] = "vpcs"
|
2018-11-11 14:13:58 +02:00
|
|
|
vms.append(vm)
|
|
|
|
for vm in settings.get("TraceNG", {}).get("nodes", []):
|
2018-11-28 11:12:57 +02:00
|
|
|
vm["template_type"] = "traceng"
|
2018-11-11 14:13:58 +02:00
|
|
|
vms.append(vm)
|
|
|
|
|
|
|
|
for vm in vms:
|
|
|
|
# remove deprecated properties
|
|
|
|
for prop in vm.copy():
|
|
|
|
if prop in ["enable_remote_console", "use_ubridge", "acpi_shutdown"]:
|
|
|
|
del vm[prop]
|
|
|
|
|
|
|
|
# remove deprecated default_symbol and hover_symbol
|
|
|
|
# and set symbol if not present
|
|
|
|
deprecated = ["default_symbol", "hover_symbol"]
|
|
|
|
if len([prop for prop in vm.keys() if prop in deprecated]) > 0:
|
|
|
|
if "default_symbol" in vm.keys():
|
|
|
|
del vm["default_symbol"]
|
|
|
|
if "hover_symbol" in vm.keys():
|
|
|
|
del vm["hover_symbol"]
|
|
|
|
|
|
|
|
if "symbol" not in vm.keys():
|
|
|
|
vm["symbol"] = ":/symbols/computer.svg"
|
|
|
|
|
2018-11-28 11:12:57 +02:00
|
|
|
vm.setdefault("template_id", str(uuid.uuid4()))
|
2018-11-11 14:13:58 +02:00
|
|
|
try:
|
2018-11-28 11:12:57 +02:00
|
|
|
template = Template(vm["template_id"], vm)
|
|
|
|
template.__json__() # Check if loaded without error
|
|
|
|
self._templates[template.id] = template
|
2018-11-11 14:13:58 +02:00
|
|
|
except KeyError as e:
|
2018-11-28 11:12:57 +02:00
|
|
|
# template data is not complete (missing name or type)
|
|
|
|
log.warning("Cannot load template {} ('{}'): missing key {}".format(vm["template_id"], vm.get("name", "unknown"), e))
|
2018-11-11 14:13:58 +02:00
|
|
|
continue
|
|
|
|
|
2018-10-15 13:05:49 +03:00
|
|
|
async def add_compute(self, compute_id=None, name=None, force=False, connect=True, **kwargs):
|
2016-03-03 17:02:27 +02:00
|
|
|
"""
|
2018-11-27 13:14:51 +02:00
|
|
|
Add a server to the dictionary of computes controlled by this controller
|
2016-03-10 22:51:29 +02:00
|
|
|
|
2018-11-27 13:14:51 +02:00
|
|
|
:param compute_id: Compute identifier
|
2016-06-29 18:39:41 +03:00
|
|
|
:param name: Compute name
|
2016-08-26 15:09:18 +03:00
|
|
|
:param force: True skip security check
|
2016-08-30 10:58:37 +03:00
|
|
|
:param connect: True connect to the compute immediately
|
2016-04-15 18:57:06 +03:00
|
|
|
:param kwargs: See the documentation of Compute
|
2016-03-03 17:02:27 +02:00
|
|
|
"""
|
2018-01-10 11:22:55 +02:00
|
|
|
|
2016-05-23 19:44:20 +03:00
|
|
|
if compute_id not in self._computes:
|
2016-05-11 16:01:57 +03:00
|
|
|
|
2016-07-19 14:26:27 +03:00
|
|
|
# We disallow to create from the outside the local and VM server
|
2016-08-26 15:09:18 +03:00
|
|
|
if (compute_id == 'local' or compute_id == 'vm') and not force:
|
2016-05-24 16:45:06 +03:00
|
|
|
return None
|
2016-05-23 19:44:20 +03:00
|
|
|
|
2016-10-24 16:24:45 +03:00
|
|
|
# It seem we have error with a gns3vm imported as a remote server and conflict
|
|
|
|
# with GNS3 VM settings. That's why we ignore server name gns3vm
|
|
|
|
if name == 'gns3vm':
|
|
|
|
return None
|
|
|
|
|
2016-06-29 18:39:41 +03:00
|
|
|
for compute in self._computes.values():
|
2017-02-23 15:34:21 +02:00
|
|
|
if name and compute.name == name and not force:
|
2016-08-20 20:10:34 +03:00
|
|
|
raise aiohttp.web.HTTPConflict(text='Compute name "{}" already exists'.format(name))
|
2016-06-29 18:39:41 +03:00
|
|
|
|
|
|
|
compute = Compute(compute_id=compute_id, controller=self, name=name, **kwargs)
|
2016-05-25 15:10:03 +03:00
|
|
|
self._computes[compute.id] = compute
|
2016-04-19 16:35:50 +03:00
|
|
|
self.save()
|
2016-08-30 10:58:37 +03:00
|
|
|
if connect:
|
2018-10-15 13:05:49 +03:00
|
|
|
await compute.connect()
|
2018-08-16 17:31:56 +03:00
|
|
|
self.notification.controller_emit("compute.created", compute.__json__())
|
2016-05-25 15:10:03 +03:00
|
|
|
return compute
|
2016-05-23 19:44:20 +03:00
|
|
|
else:
|
2016-08-30 10:58:37 +03:00
|
|
|
if connect:
|
2018-10-15 13:05:49 +03:00
|
|
|
await self._computes[compute_id].connect()
|
2018-08-16 17:31:56 +03:00
|
|
|
self.notification.controller_emit("compute.updated", self._computes[compute_id].__json__())
|
2016-05-25 15:10:03 +03:00
|
|
|
return self._computes[compute_id]
|
2016-03-03 17:02:27 +02:00
|
|
|
|
2018-10-15 13:05:49 +03:00
|
|
|
async def close_compute_projects(self, compute):
|
2016-12-15 22:57:59 +02:00
|
|
|
"""
|
|
|
|
Close projects running on a compute
|
|
|
|
"""
|
2018-03-15 09:17:39 +02:00
|
|
|
|
2016-12-15 22:57:59 +02:00
|
|
|
for project in self._projects.values():
|
|
|
|
if compute in project.computes:
|
2018-10-15 13:05:49 +03:00
|
|
|
await project.close()
|
2016-12-15 22:57:59 +02:00
|
|
|
|
2018-08-09 12:59:10 +03:00
|
|
|
def compute_has_open_project(self, compute):
|
|
|
|
"""
|
|
|
|
Check is compute has a project opened.
|
|
|
|
|
|
|
|
:returns: True if a project is open
|
|
|
|
"""
|
|
|
|
|
|
|
|
for project in self._projects.values():
|
|
|
|
if compute in project.computes and project.status == "opened":
|
|
|
|
return True
|
|
|
|
return False
|
|
|
|
|
2018-10-15 13:05:49 +03:00
|
|
|
async def delete_compute(self, compute_id):
|
2016-05-25 12:27:41 +03:00
|
|
|
"""
|
2016-08-26 12:22:09 +03:00
|
|
|
Delete a compute node. Project using this compute will be close
|
|
|
|
|
2018-11-27 13:14:51 +02:00
|
|
|
:param compute_id: Compute identifier
|
2016-05-25 12:27:41 +03:00
|
|
|
"""
|
2018-03-15 09:17:39 +02:00
|
|
|
|
2016-12-12 12:09:07 +02:00
|
|
|
try:
|
|
|
|
compute = self.get_compute(compute_id)
|
|
|
|
except aiohttp.web.HTTPNotFound:
|
|
|
|
return
|
2018-10-15 13:05:49 +03:00
|
|
|
await self.close_compute_projects(compute)
|
|
|
|
await compute.close()
|
2016-05-25 12:27:41 +03:00
|
|
|
del self._computes[compute_id]
|
2016-05-25 15:10:03 +03:00
|
|
|
self.save()
|
2018-08-16 17:31:56 +03:00
|
|
|
self.notification.controller_emit("compute.deleted", compute.__json__())
|
2016-05-25 12:27:41 +03:00
|
|
|
|
2016-05-18 15:56:23 +03:00
|
|
|
@property
|
|
|
|
def notification(self):
|
|
|
|
"""
|
|
|
|
The notification system
|
|
|
|
"""
|
2018-03-15 09:17:39 +02:00
|
|
|
|
2016-05-18 15:56:23 +03:00
|
|
|
return self._notification
|
|
|
|
|
2016-03-03 17:02:27 +02:00
|
|
|
@property
|
2016-04-15 18:57:06 +03:00
|
|
|
def computes(self):
|
2016-03-03 17:02:27 +02:00
|
|
|
"""
|
2018-11-27 13:14:51 +02:00
|
|
|
:returns: The dictionary of computes managed by this controller
|
2016-03-03 17:02:27 +02:00
|
|
|
"""
|
2018-03-15 09:17:39 +02:00
|
|
|
|
2016-04-15 18:57:06 +03:00
|
|
|
return self._computes
|
2016-03-03 17:02:27 +02:00
|
|
|
|
2016-05-12 00:19:00 +03:00
|
|
|
def get_compute(self, compute_id):
|
2016-03-10 22:51:29 +02:00
|
|
|
"""
|
2018-11-27 13:14:51 +02:00
|
|
|
Returns a compute or raise a 404 error.
|
2016-03-10 22:51:29 +02:00
|
|
|
"""
|
2018-03-15 09:17:39 +02:00
|
|
|
|
2016-03-10 22:51:29 +02:00
|
|
|
try:
|
2016-04-15 18:57:06 +03:00
|
|
|
return self._computes[compute_id]
|
2016-03-10 22:51:29 +02:00
|
|
|
except KeyError:
|
2016-08-22 19:37:32 +03:00
|
|
|
if compute_id == "vm":
|
2018-03-15 09:17:39 +02:00
|
|
|
raise aiohttp.web.HTTPNotFound(text="Cannot use a node on the GNS3 VM server with the GNS3 VM not configured")
|
2016-04-15 18:57:06 +03:00
|
|
|
raise aiohttp.web.HTTPNotFound(text="Compute ID {} doesn't exist".format(compute_id))
|
2016-03-10 22:51:29 +02:00
|
|
|
|
2016-07-25 19:02:22 +03:00
|
|
|
def has_compute(self, compute_id):
|
|
|
|
"""
|
|
|
|
Return True if the compute exist in the controller
|
|
|
|
"""
|
2018-03-15 09:17:39 +02:00
|
|
|
|
2016-07-25 19:02:22 +03:00
|
|
|
return compute_id in self._computes
|
|
|
|
|
2018-10-15 13:05:49 +03:00
|
|
|
async def add_project(self, project_id=None, name=None, path=None, **kwargs):
|
2016-03-10 11:32:07 +02:00
|
|
|
"""
|
2016-05-14 03:48:10 +03:00
|
|
|
Creates a project or returns an existing project
|
2016-03-10 22:51:29 +02:00
|
|
|
|
2016-06-29 18:39:41 +03:00
|
|
|
:param project_id: Project ID
|
|
|
|
:param name: Project name
|
2016-03-10 22:51:29 +02:00
|
|
|
:param kwargs: See the documentation of Project
|
2016-03-10 11:32:07 +02:00
|
|
|
"""
|
2016-06-29 18:39:41 +03:00
|
|
|
|
2018-03-15 09:17:39 +02:00
|
|
|
if project_id not in self._projects:
|
2016-06-29 18:39:41 +03:00
|
|
|
for project in self._projects.values():
|
|
|
|
if name and project.name == name:
|
2018-03-30 19:01:37 +03:00
|
|
|
if path and path == project.path:
|
|
|
|
raise aiohttp.web.HTTPConflict(text='Project "{}" already exists in location "{}"'.format(name, path))
|
|
|
|
else:
|
|
|
|
raise aiohttp.web.HTTPConflict(text='Project "{}" already exists'.format(name))
|
|
|
|
project = Project(project_id=project_id, controller=self, name=name, path=path, **kwargs)
|
2016-03-10 11:32:07 +02:00
|
|
|
self._projects[project.id] = project
|
2016-03-10 22:51:29 +02:00
|
|
|
return self._projects[project.id]
|
|
|
|
return self._projects[project_id]
|
2016-03-10 11:32:07 +02:00
|
|
|
|
2016-05-11 20:35:36 +03:00
|
|
|
def get_project(self, project_id):
|
2016-03-10 11:32:07 +02:00
|
|
|
"""
|
2016-11-18 18:35:28 +02:00
|
|
|
Returns a project or raise a 404 error.
|
2016-03-10 11:32:07 +02:00
|
|
|
"""
|
2018-03-15 09:17:39 +02:00
|
|
|
|
2016-03-10 11:32:07 +02:00
|
|
|
try:
|
|
|
|
return self._projects[project_id]
|
|
|
|
except KeyError:
|
|
|
|
raise aiohttp.web.HTTPNotFound(text="Project ID {} doesn't exist".format(project_id))
|
|
|
|
|
2018-10-15 13:05:49 +03:00
|
|
|
async def get_loaded_project(self, project_id):
|
2016-11-18 18:35:28 +02:00
|
|
|
"""
|
|
|
|
Returns a project or raise a 404 error.
|
|
|
|
|
|
|
|
If project is not finished to load wait for it
|
|
|
|
"""
|
2018-03-15 09:17:39 +02:00
|
|
|
|
2016-11-18 18:35:28 +02:00
|
|
|
project = self.get_project(project_id)
|
2018-10-15 13:05:49 +03:00
|
|
|
await project.wait_loaded()
|
2016-11-18 18:35:28 +02:00
|
|
|
return project
|
|
|
|
|
2016-05-12 00:19:00 +03:00
|
|
|
def remove_project(self, project):
|
2018-03-15 09:17:39 +02:00
|
|
|
|
2017-02-13 16:24:22 +02:00
|
|
|
if project.id in self._projects:
|
|
|
|
del self._projects[project.id]
|
2016-03-10 11:32:07 +02:00
|
|
|
|
2018-10-15 13:05:49 +03:00
|
|
|
async def load_project(self, path, load=True):
|
2016-06-14 17:07:37 +03:00
|
|
|
"""
|
|
|
|
Load a project from a .gns3
|
|
|
|
|
|
|
|
:param path: Path of the .gns3
|
2016-06-15 00:08:30 +03:00
|
|
|
:param load: Load the topology
|
2016-06-14 17:07:37 +03:00
|
|
|
"""
|
2018-03-15 09:17:39 +02:00
|
|
|
|
2016-06-14 17:07:37 +03:00
|
|
|
topo_data = load_topology(path)
|
2017-02-06 12:07:35 +02:00
|
|
|
topo_data.pop("topology")
|
2016-06-14 17:07:37 +03:00
|
|
|
topo_data.pop("version")
|
|
|
|
topo_data.pop("revision")
|
|
|
|
topo_data.pop("type")
|
|
|
|
|
2016-06-17 18:50:06 +03:00
|
|
|
if topo_data["project_id"] in self._projects:
|
2016-11-14 19:45:44 +02:00
|
|
|
project = self._projects[topo_data["project_id"]]
|
|
|
|
else:
|
2018-10-15 13:05:49 +03:00
|
|
|
project = await self.add_project(path=os.path.dirname(path), status="closed", filename=os.path.basename(path), **topo_data)
|
2016-08-15 21:51:59 +03:00
|
|
|
if load or project.auto_open:
|
2018-10-15 13:05:49 +03:00
|
|
|
await project.open()
|
2016-06-16 17:57:54 +03:00
|
|
|
return project
|
2016-06-14 17:07:37 +03:00
|
|
|
|
2018-10-15 13:05:49 +03:00
|
|
|
async def _project_auto_open(self):
|
2016-09-06 12:30:08 +03:00
|
|
|
"""
|
|
|
|
Auto open the project with auto open enable
|
|
|
|
"""
|
2018-03-15 09:17:39 +02:00
|
|
|
|
2016-09-06 12:30:08 +03:00
|
|
|
for project in self._projects.values():
|
|
|
|
if project.auto_open:
|
2018-10-15 13:05:49 +03:00
|
|
|
await project.open()
|
2016-09-06 12:30:08 +03:00
|
|
|
|
2016-07-21 15:48:13 +03:00
|
|
|
def get_free_project_name(self, base_name):
|
|
|
|
"""
|
|
|
|
Generate a free project name base on the base name
|
|
|
|
"""
|
2018-03-15 09:17:39 +02:00
|
|
|
|
2016-07-21 18:55:15 +03:00
|
|
|
names = [p.name for p in self._projects.values()]
|
2016-07-21 15:48:13 +03:00
|
|
|
if base_name not in names:
|
|
|
|
return base_name
|
|
|
|
i = 1
|
2016-07-22 19:02:11 +03:00
|
|
|
|
|
|
|
projects_path = self.projects_directory()
|
|
|
|
|
|
|
|
while True:
|
|
|
|
new_name = "{}-{}".format(base_name, i)
|
|
|
|
if new_name not in names and not os.path.exists(os.path.join(projects_path, new_name)):
|
|
|
|
break
|
2016-07-21 15:48:13 +03:00
|
|
|
i += 1
|
|
|
|
if i > 1000000:
|
|
|
|
raise aiohttp.web.HTTPConflict(text="A project name could not be allocated (node limit reached?)")
|
2016-07-22 19:02:11 +03:00
|
|
|
return new_name
|
2016-07-21 15:48:13 +03:00
|
|
|
|
2016-03-10 11:32:07 +02:00
|
|
|
@property
|
|
|
|
def projects(self):
|
|
|
|
"""
|
2017-11-17 13:13:34 +02:00
|
|
|
:returns: The dictionary of projects managed by GNS3
|
2016-03-10 11:32:07 +02:00
|
|
|
"""
|
2018-03-15 09:17:39 +02:00
|
|
|
|
2016-03-10 11:32:07 +02:00
|
|
|
return self._projects
|
|
|
|
|
2017-02-01 12:30:14 +02:00
|
|
|
@property
|
2018-11-28 11:12:57 +02:00
|
|
|
def appliances(self):
|
2017-02-01 12:30:14 +02:00
|
|
|
"""
|
2018-11-28 11:12:57 +02:00
|
|
|
:returns: The dictionary of appliances managed by GNS3
|
2017-02-01 12:30:14 +02:00
|
|
|
"""
|
2018-03-15 09:17:39 +02:00
|
|
|
|
2018-11-28 11:12:57 +02:00
|
|
|
return self._appliances
|
2017-02-01 12:30:14 +02:00
|
|
|
|
2017-04-12 15:35:49 +03:00
|
|
|
@property
|
2018-11-28 11:12:57 +02:00
|
|
|
def templates(self):
|
2017-04-12 15:35:49 +03:00
|
|
|
"""
|
2018-11-28 11:12:57 +02:00
|
|
|
:returns: The dictionary of templates managed by GNS3
|
2017-04-12 15:35:49 +03:00
|
|
|
"""
|
2018-03-15 09:17:39 +02:00
|
|
|
|
2018-11-28 11:12:57 +02:00
|
|
|
return self._templates
|
2017-04-12 15:35:49 +03:00
|
|
|
|
2018-11-14 10:24:30 +02:00
|
|
|
@property
|
|
|
|
def iou_license(self):
|
|
|
|
"""
|
|
|
|
:returns: The dictionary of IOU license settings
|
|
|
|
"""
|
|
|
|
|
|
|
|
return self._iou_license_settings
|
|
|
|
|
2016-07-22 19:02:11 +03:00
|
|
|
def projects_directory(self):
|
2018-03-15 09:17:39 +02:00
|
|
|
|
2016-07-22 19:02:11 +03:00
|
|
|
server_config = Config.instance().get_section_config("Server")
|
|
|
|
return os.path.expanduser(server_config.get("projects_path", "~/GNS3/projects"))
|
|
|
|
|
2016-03-02 10:49:52 +02:00
|
|
|
@staticmethod
|
|
|
|
def instance():
|
|
|
|
"""
|
|
|
|
Singleton to return only on instance of Controller.
|
2016-05-14 03:48:10 +03:00
|
|
|
|
2016-03-02 10:49:52 +02:00
|
|
|
:returns: instance of Controller
|
|
|
|
"""
|
|
|
|
|
|
|
|
if not hasattr(Controller, '_instance') or Controller._instance is None:
|
|
|
|
Controller._instance = Controller()
|
|
|
|
return Controller._instance
|
2017-07-07 18:50:40 +03:00
|
|
|
|
2018-10-15 13:05:49 +03:00
|
|
|
async def autoidlepc(self, compute_id, platform, image, ram):
|
2017-07-07 18:50:40 +03:00
|
|
|
"""
|
|
|
|
Compute and IDLE PC value for an image
|
|
|
|
|
|
|
|
:param compute_id: ID of the compute where the idlepc operation need to run
|
|
|
|
:param platform: Platform type
|
|
|
|
:param image: Image to use
|
2017-11-23 18:00:31 +02:00
|
|
|
:param ram: amount of RAM to use
|
2017-07-07 18:50:40 +03:00
|
|
|
"""
|
2018-03-15 09:17:39 +02:00
|
|
|
|
2017-07-07 18:50:40 +03:00
|
|
|
compute = self.get_compute(compute_id)
|
|
|
|
for project in list(self._projects.values()):
|
|
|
|
if project.name == "AUTOIDLEPC":
|
2018-10-15 13:05:49 +03:00
|
|
|
await project.delete()
|
2017-07-07 18:50:40 +03:00
|
|
|
self.remove_project(project)
|
2018-10-15 13:05:49 +03:00
|
|
|
project = await self.add_project(name="AUTOIDLEPC")
|
|
|
|
node = await project.add_node(compute, "AUTOIDLEPC", str(uuid.uuid4()), node_type="dynamips", platform=platform, image=image, ram=ram)
|
|
|
|
res = await node.dynamips_auto_idlepc()
|
|
|
|
await project.delete()
|
2017-07-07 18:50:40 +03:00
|
|
|
self.remove_project(project)
|
|
|
|
return res
|
2018-10-05 13:48:20 +03:00
|
|
|
|
2018-10-15 13:05:49 +03:00
|
|
|
async def compute_ports(self, compute_id):
|
2018-10-05 13:48:20 +03:00
|
|
|
"""
|
|
|
|
Get the ports used by a compute.
|
|
|
|
|
|
|
|
:param compute_id: ID of the compute
|
|
|
|
"""
|
|
|
|
|
|
|
|
compute = self.get_compute(compute_id)
|
2018-10-15 13:05:49 +03:00
|
|
|
response = await compute.get("/network/ports")
|
2018-10-05 13:48:20 +03:00
|
|
|
return response.json
|