mirror of
https://github.com/GNS3/gns3-server
synced 2024-11-12 03:18:55 +00:00
218 lines
7.5 KiB
Python
218 lines
7.5 KiB
Python
#!/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/>.
|
|
|
|
import copy
|
|
import uuid
|
|
import json
|
|
import jsonschema
|
|
|
|
from gns3server.schemas.cloud_template import CLOUD_TEMPLATE_OBJECT_SCHEMA
|
|
from gns3server.schemas.ethernet_switch_template import ETHERNET_SWITCH_TEMPLATE_OBJECT_SCHEMA
|
|
from gns3server.schemas.ethernet_hub_template import ETHERNET_HUB_TEMPLATE_OBJECT_SCHEMA
|
|
from gns3server.schemas.docker_template import DOCKER_TEMPLATE_OBJECT_SCHEMA
|
|
from gns3server.schemas.vpcs_template import VPCS_TEMPLATE_OBJECT_SCHEMA
|
|
from gns3server.schemas.traceng_template import TRACENG_TEMPLATE_OBJECT_SCHEMA
|
|
from gns3server.schemas.virtualbox_template import VIRTUALBOX_TEMPLATE_OBJECT_SCHEMA
|
|
from gns3server.schemas.vmware_template import VMWARE_TEMPLATE_OBJECT_SCHEMA
|
|
from gns3server.schemas.iou_template import IOU_TEMPLATE_OBJECT_SCHEMA
|
|
from gns3server.schemas.qemu_template import QEMU_TEMPLATE_OBJECT_SCHEMA
|
|
|
|
from gns3server.schemas.dynamips_template import (
|
|
DYNAMIPS_TEMPLATE_OBJECT_SCHEMA,
|
|
C7200_DYNAMIPS_TEMPLATE_OBJECT_SCHEMA,
|
|
C3745_DYNAMIPS_TEMPLATE_OBJECT_SCHEMA,
|
|
C3725_DYNAMIPS_TEMPLATE_OBJECT_SCHEMA,
|
|
C3600_DYNAMIPS_TEMPLATE_OBJECT_SCHEMA,
|
|
C2691_DYNAMIPS_TEMPLATE_OBJECT_SCHEMA,
|
|
C2600_DYNAMIPS_TEMPLATE_OBJECT_SCHEMA,
|
|
C1700_DYNAMIPS_TEMPLATE_OBJECT_SCHEMA
|
|
)
|
|
|
|
import logging
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
# Add default values for missing entries in a request, largely taken from jsonschema documentation example
|
|
# https://python-jsonschema.readthedocs.io/en/latest/faq/#why-doesn-t-my-schema-s-default-property-set-the-default-on-my-instance
|
|
def extend_with_default(validator_class):
|
|
|
|
validate_properties = validator_class.VALIDATORS["properties"]
|
|
def set_defaults(validator, properties, instance, schema):
|
|
if jsonschema.Draft4Validator(schema).is_valid(instance):
|
|
# only add default for the matching sub-schema (e.g. when using 'oneOf')
|
|
for property, subschema in properties.items():
|
|
if "default" in subschema:
|
|
instance.setdefault(property, subschema["default"])
|
|
|
|
for error in validate_properties(validator, properties, instance, schema,):
|
|
yield error
|
|
|
|
return jsonschema.validators.extend(
|
|
validator_class, {"properties" : set_defaults},
|
|
)
|
|
|
|
|
|
ValidatorWithDefaults = extend_with_default(jsonschema.Draft4Validator)
|
|
|
|
ID_TO_CATEGORY = {
|
|
3: "firewall",
|
|
2: "guest",
|
|
1: "switch",
|
|
0: "router"
|
|
}
|
|
|
|
TEMPLATE_TYPE_TO_SHEMA = {
|
|
"cloud": CLOUD_TEMPLATE_OBJECT_SCHEMA,
|
|
"ethernet_hub": ETHERNET_HUB_TEMPLATE_OBJECT_SCHEMA,
|
|
"ethernet_switch": ETHERNET_SWITCH_TEMPLATE_OBJECT_SCHEMA,
|
|
"docker": DOCKER_TEMPLATE_OBJECT_SCHEMA,
|
|
"dynamips": DYNAMIPS_TEMPLATE_OBJECT_SCHEMA,
|
|
"vpcs": VPCS_TEMPLATE_OBJECT_SCHEMA,
|
|
"traceng": TRACENG_TEMPLATE_OBJECT_SCHEMA,
|
|
"virtualbox": VIRTUALBOX_TEMPLATE_OBJECT_SCHEMA,
|
|
"vmware": VMWARE_TEMPLATE_OBJECT_SCHEMA,
|
|
"iou": IOU_TEMPLATE_OBJECT_SCHEMA,
|
|
"qemu": QEMU_TEMPLATE_OBJECT_SCHEMA
|
|
}
|
|
|
|
DYNAMIPS_PLATFORM_TO_SHEMA = {
|
|
"c7200": C7200_DYNAMIPS_TEMPLATE_OBJECT_SCHEMA,
|
|
"c3745": C3745_DYNAMIPS_TEMPLATE_OBJECT_SCHEMA,
|
|
"c3725": C3725_DYNAMIPS_TEMPLATE_OBJECT_SCHEMA,
|
|
"c3600": C3600_DYNAMIPS_TEMPLATE_OBJECT_SCHEMA,
|
|
"c2691": C2691_DYNAMIPS_TEMPLATE_OBJECT_SCHEMA,
|
|
"c2600": C2600_DYNAMIPS_TEMPLATE_OBJECT_SCHEMA,
|
|
"c1700": C1700_DYNAMIPS_TEMPLATE_OBJECT_SCHEMA
|
|
}
|
|
|
|
|
|
class Template:
|
|
|
|
def __init__(self, template_id, settings, builtin=False):
|
|
|
|
if template_id is None:
|
|
self._id = str(uuid.uuid4())
|
|
elif isinstance(template_id, uuid.UUID):
|
|
self._id = str(template_id)
|
|
else:
|
|
self._id = template_id
|
|
|
|
self._settings = copy.deepcopy(settings)
|
|
|
|
# Version of the gui before 2.1 use linked_base
|
|
# and the server linked_clone
|
|
if "linked_base" in self.settings:
|
|
linked_base = self._settings.pop("linked_base")
|
|
if "linked_clone" not in self._settings:
|
|
self._settings["linked_clone"] = linked_base
|
|
|
|
# Convert old GUI category to text category
|
|
try:
|
|
self._settings["category"] = ID_TO_CATEGORY[self._settings["category"]]
|
|
except KeyError:
|
|
pass
|
|
|
|
# The "server" setting has been replaced by "compute_id" setting in version 2.2
|
|
if "server" in self._settings:
|
|
self._settings["compute_id"] = self._settings.pop("server")
|
|
|
|
# The "node_type" setting has been replaced by "template_type" setting in version 2.2
|
|
if "node_type" in self._settings:
|
|
self._settings["template_type"] = self._settings.pop("node_type")
|
|
|
|
# Remove an old IOU setting
|
|
if self._settings["template_type"] == "iou" and "image" in self._settings:
|
|
del self._settings["image"]
|
|
|
|
self._builtin = builtin
|
|
|
|
if builtin is False:
|
|
self.validate_and_apply_defaults(TEMPLATE_TYPE_TO_SHEMA[self.template_type])
|
|
|
|
if self.template_type == "dynamips":
|
|
# special case for Dynamips to cover all platform types that contain specific settings
|
|
self.validate_and_apply_defaults(DYNAMIPS_PLATFORM_TO_SHEMA[self._settings["platform"]])
|
|
|
|
log.debug('Template "{name}" [{id}] loaded'.format(name=self.name, id=self._id))
|
|
|
|
@property
|
|
def id(self):
|
|
return self._id
|
|
|
|
@property
|
|
def settings(self):
|
|
return self._settings
|
|
|
|
@settings.setter
|
|
def settings(self, settings):
|
|
self._settings.update(settings)
|
|
|
|
@property
|
|
def name(self):
|
|
return self._settings["name"]
|
|
|
|
@property
|
|
def compute_id(self):
|
|
return self._settings["compute_id"]
|
|
|
|
@property
|
|
def template_type(self):
|
|
return self._settings["template_type"]
|
|
|
|
@property
|
|
def builtin(self):
|
|
return self._builtin
|
|
|
|
def update(self, **kwargs):
|
|
|
|
|
|
from gns3server.controller import Controller
|
|
controller = Controller.instance()
|
|
Controller.instance().check_can_write_config()
|
|
self._settings.update(kwargs)
|
|
controller.notification.controller_emit("template.updated", self.__json__())
|
|
controller.save()
|
|
|
|
def validate_and_apply_defaults(self, schema):
|
|
|
|
validator = ValidatorWithDefaults(schema)
|
|
try:
|
|
validator.validate(self.__json__())
|
|
except jsonschema.ValidationError as e:
|
|
message = "JSON schema error {}".format(e.message)
|
|
log.error(message)
|
|
log.debug("Input schema: {}".format(json.dumps(schema)))
|
|
raise
|
|
|
|
def __json__(self):
|
|
"""
|
|
Template settings.
|
|
"""
|
|
|
|
settings = self._settings
|
|
settings.update({"template_id": self._id,
|
|
"builtin": self.builtin})
|
|
|
|
if self.builtin:
|
|
# builin templates have compute_id set to None to tell clients
|
|
# to select a compute
|
|
settings["compute_id"] = None
|
|
else:
|
|
settings["compute_id"] = self.compute_id
|
|
|
|
return settings
|