2015-01-14 01:26:32 +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-01-23 01:04:24 +00:00
|
|
|
import sys
|
|
|
|
import os
|
|
|
|
import struct
|
|
|
|
import stat
|
2015-01-14 01:26:32 +00:00
|
|
|
import asyncio
|
|
|
|
import aiohttp
|
2015-01-23 01:04:24 +00:00
|
|
|
import socket
|
2015-02-09 01:10:04 +00:00
|
|
|
import shutil
|
2015-11-12 14:37:34 +00:00
|
|
|
import re
|
2015-01-23 01:04:24 +00:00
|
|
|
|
|
|
|
import logging
|
|
|
|
log = logging.getLogger(__name__)
|
2015-01-14 01:26:32 +00:00
|
|
|
|
2015-01-20 01:30:57 +00:00
|
|
|
from uuid import UUID, uuid4
|
2015-06-07 19:51:33 +00:00
|
|
|
from gns3server.utils.interfaces import is_interface_up
|
2015-01-21 02:02:22 +00:00
|
|
|
from ..config import Config
|
2015-02-09 01:10:04 +00:00
|
|
|
from ..utils.asyncio import wait_run_in_executor
|
2015-10-05 18:12:20 +00:00
|
|
|
from ..utils import force_unix_path
|
2015-01-20 11:46:15 +00:00
|
|
|
from .project_manager import ProjectManager
|
2015-01-14 01:26:32 +00:00
|
|
|
|
2015-02-24 02:00:34 +00:00
|
|
|
from .nios.nio_udp import NIOUDP
|
|
|
|
from .nios.nio_tap import NIOTAP
|
2016-06-01 23:50:31 +00:00
|
|
|
from .nios.nio_ethernet import NIOEthernet
|
2015-06-17 15:11:25 +00:00
|
|
|
from ..utils.images import md5sum, remove_checksum
|
2016-06-07 13:34:04 +00:00
|
|
|
from .error import NodeError, ImageMissingError
|
2015-01-23 01:04:24 +00:00
|
|
|
|
2015-01-14 01:26:32 +00:00
|
|
|
|
2015-01-14 17:52:02 +00:00
|
|
|
class BaseManager:
|
2015-01-20 12:24:00 +00:00
|
|
|
|
2015-01-14 01:26:32 +00:00
|
|
|
"""
|
2015-04-08 17:17:34 +00:00
|
|
|
Base class for all Manager classes.
|
2016-05-11 17:35:36 +00:00
|
|
|
Responsible of management of a node pool of the same type.
|
2015-01-14 01:26:32 +00:00
|
|
|
"""
|
|
|
|
|
2015-02-28 05:12:43 +00:00
|
|
|
_convert_lock = None
|
2015-02-26 01:55:35 +00:00
|
|
|
|
2015-01-14 01:26:32 +00:00
|
|
|
def __init__(self):
|
2015-01-19 21:43:35 +00:00
|
|
|
|
2015-02-28 05:12:43 +00:00
|
|
|
BaseManager._convert_lock = asyncio.Lock()
|
2016-05-11 17:35:36 +00:00
|
|
|
self._nodes = {}
|
2015-01-19 21:43:35 +00:00
|
|
|
self._port_manager = None
|
2015-01-21 02:02:22 +00:00
|
|
|
self._config = Config.instance()
|
2015-01-14 01:26:32 +00:00
|
|
|
|
|
|
|
@classmethod
|
|
|
|
def instance(cls):
|
|
|
|
"""
|
2015-01-18 22:41:53 +00:00
|
|
|
Singleton to return only one instance of BaseManager.
|
2015-01-14 01:26:32 +00:00
|
|
|
|
2015-01-19 21:43:35 +00:00
|
|
|
:returns: instance of BaseManager
|
2015-01-14 01:26:32 +00:00
|
|
|
"""
|
|
|
|
|
2015-01-16 16:09:45 +00:00
|
|
|
if not hasattr(cls, "_instance") or cls._instance is None:
|
2015-01-14 01:26:32 +00:00
|
|
|
cls._instance = cls()
|
|
|
|
return cls._instance
|
|
|
|
|
2015-01-21 22:21:15 +00:00
|
|
|
@property
|
|
|
|
def module_name(self):
|
|
|
|
"""
|
|
|
|
Returns the module name.
|
|
|
|
|
|
|
|
:returns: module name
|
|
|
|
"""
|
|
|
|
|
|
|
|
return self.__class__.__name__
|
|
|
|
|
2015-01-19 10:22:24 +00:00
|
|
|
@property
|
|
|
|
def port_manager(self):
|
|
|
|
"""
|
2015-01-21 02:02:22 +00:00
|
|
|
Returns the port manager.
|
2015-01-19 10:22:24 +00:00
|
|
|
|
|
|
|
:returns: Port manager
|
|
|
|
"""
|
|
|
|
|
|
|
|
return self._port_manager
|
|
|
|
|
|
|
|
@port_manager.setter
|
|
|
|
def port_manager(self, new_port_manager):
|
|
|
|
|
2015-01-19 21:43:35 +00:00
|
|
|
self._port_manager = new_port_manager
|
2015-01-19 10:22:24 +00:00
|
|
|
|
2015-01-21 02:02:22 +00:00
|
|
|
@property
|
|
|
|
def config(self):
|
|
|
|
"""
|
|
|
|
Returns the server config.
|
|
|
|
|
|
|
|
:returns: Config
|
|
|
|
"""
|
|
|
|
|
|
|
|
return self._config
|
|
|
|
|
2015-01-23 04:31:26 +00:00
|
|
|
@asyncio.coroutine
|
2015-01-23 06:40:51 +00:00
|
|
|
def unload(self):
|
|
|
|
|
2015-02-05 21:24:06 +00:00
|
|
|
tasks = []
|
2016-05-11 17:35:36 +00:00
|
|
|
for node_id in self._nodes.keys():
|
|
|
|
tasks.append(asyncio.async(self.close_node(node_id)))
|
2015-02-05 21:24:06 +00:00
|
|
|
|
|
|
|
if tasks:
|
|
|
|
done, _ = yield from asyncio.wait(tasks)
|
|
|
|
for future in done:
|
|
|
|
try:
|
|
|
|
future.result()
|
2015-07-25 22:46:23 +00:00
|
|
|
except (Exception, GeneratorExit) as e:
|
2016-05-11 17:35:36 +00:00
|
|
|
log.error("Could not close node {}".format(e), exc_info=1)
|
2015-02-05 21:24:06 +00:00
|
|
|
continue
|
2015-01-20 01:30:57 +00:00
|
|
|
|
2015-01-23 06:40:51 +00:00
|
|
|
if hasattr(BaseManager, "_instance"):
|
|
|
|
BaseManager._instance = None
|
2015-01-23 20:01:23 +00:00
|
|
|
log.debug("Module {} unloaded".format(self.module_name))
|
2015-01-14 01:26:32 +00:00
|
|
|
|
2016-05-11 17:35:36 +00:00
|
|
|
def get_node(self, node_id, project_id=None):
|
2015-01-14 01:26:32 +00:00
|
|
|
"""
|
2016-05-11 17:35:36 +00:00
|
|
|
Returns a Node instance.
|
2015-01-14 01:26:32 +00:00
|
|
|
|
2016-05-11 17:35:36 +00:00
|
|
|
:param node_id: Node identifier
|
2015-02-05 00:13:35 +00:00
|
|
|
:param project_id: Project identifier
|
2015-01-14 01:26:32 +00:00
|
|
|
|
2016-05-11 17:35:36 +00:00
|
|
|
:returns: Node instance
|
2015-01-14 01:26:32 +00:00
|
|
|
"""
|
|
|
|
|
2015-02-05 00:13:35 +00:00
|
|
|
if project_id:
|
|
|
|
# check the project_id exists
|
|
|
|
project = ProjectManager.instance().get_project(project_id)
|
|
|
|
|
2015-01-20 01:30:57 +00:00
|
|
|
try:
|
2016-05-11 17:35:36 +00:00
|
|
|
UUID(node_id, version=4)
|
2015-01-20 01:30:57 +00:00
|
|
|
except ValueError:
|
2016-05-11 17:35:36 +00:00
|
|
|
raise aiohttp.web.HTTPBadRequest(text="Node ID {} is not a valid UUID".format(node_id))
|
2015-01-20 01:30:57 +00:00
|
|
|
|
2016-05-11 17:35:36 +00:00
|
|
|
if node_id not in self._nodes:
|
|
|
|
raise aiohttp.web.HTTPNotFound(text="Node ID {} doesn't exist".format(node_id))
|
2015-02-05 00:13:35 +00:00
|
|
|
|
2016-05-11 17:35:36 +00:00
|
|
|
node = self._nodes[node_id]
|
2015-02-05 00:13:35 +00:00
|
|
|
if project_id:
|
2016-05-11 17:35:36 +00:00
|
|
|
if node.project.id != project.id:
|
|
|
|
raise aiohttp.web.HTTPNotFound(text="Project ID {} doesn't belong to node {}".format(project_id, node.name))
|
2015-02-05 00:13:35 +00:00
|
|
|
|
2016-05-11 17:35:36 +00:00
|
|
|
return node
|
2015-01-14 01:26:32 +00:00
|
|
|
|
2015-02-26 01:55:35 +00:00
|
|
|
@asyncio.coroutine
|
2015-03-01 20:05:51 +00:00
|
|
|
def convert_old_project(self, project, legacy_id, name):
|
2015-02-26 01:55:35 +00:00
|
|
|
"""
|
2015-03-01 20:05:51 +00:00
|
|
|
Convert projects made before version 1.3
|
2015-02-26 01:55:35 +00:00
|
|
|
|
|
|
|
:param project: Project instance
|
|
|
|
:param legacy_id: old identifier
|
2015-03-01 20:05:51 +00:00
|
|
|
:param name: node name
|
2015-02-26 01:55:35 +00:00
|
|
|
|
2015-03-01 20:05:51 +00:00
|
|
|
:returns: new identifier
|
2015-02-26 01:55:35 +00:00
|
|
|
"""
|
|
|
|
|
2015-03-01 20:05:51 +00:00
|
|
|
new_id = str(uuid4())
|
2015-03-09 01:13:01 +00:00
|
|
|
legacy_project_files_path = os.path.join(project.path, "{}-files".format(project.name))
|
2015-03-01 01:55:53 +00:00
|
|
|
new_project_files_path = os.path.join(project.path, "project-files")
|
|
|
|
if os.path.exists(legacy_project_files_path) and not os.path.exists(new_project_files_path):
|
|
|
|
# move the project files
|
2015-03-01 20:05:51 +00:00
|
|
|
log.info("Converting old project...")
|
2015-03-01 01:55:53 +00:00
|
|
|
try:
|
|
|
|
log.info('Moving "{}" to "{}"'.format(legacy_project_files_path, new_project_files_path))
|
|
|
|
yield from wait_run_in_executor(shutil.move, legacy_project_files_path, new_project_files_path)
|
|
|
|
except OSError as e:
|
|
|
|
raise aiohttp.web.HTTPInternalServerError(text="Could not move project files directory: {} to {} {}".format(legacy_project_files_path,
|
2015-03-01 20:05:51 +00:00
|
|
|
new_project_files_path, e))
|
2015-03-01 01:55:53 +00:00
|
|
|
|
2015-03-09 01:13:01 +00:00
|
|
|
if project.is_local() is False:
|
2015-03-10 03:46:23 +00:00
|
|
|
legacy_remote_project_path = os.path.join(project.location, project.name, self.module_name.lower())
|
|
|
|
new_remote_project_path = os.path.join(project.path, "project-files", self.module_name.lower())
|
|
|
|
if os.path.exists(legacy_remote_project_path) and not os.path.exists(new_remote_project_path):
|
|
|
|
# move the legacy remote project (remote servers only)
|
|
|
|
log.info("Converting old remote project...")
|
2015-03-09 01:13:01 +00:00
|
|
|
try:
|
2015-03-10 03:46:23 +00:00
|
|
|
log.info('Moving "{}" to "{}"'.format(legacy_remote_project_path, new_remote_project_path))
|
|
|
|
yield from wait_run_in_executor(shutil.move, legacy_remote_project_path, new_remote_project_path)
|
2015-03-09 01:13:01 +00:00
|
|
|
except OSError as e:
|
2015-03-10 03:46:23 +00:00
|
|
|
raise aiohttp.web.HTTPInternalServerError(text="Could not move directory: {} to {} {}".format(legacy_remote_project_path,
|
|
|
|
new_remote_project_path, e))
|
2015-03-08 23:45:29 +00:00
|
|
|
|
2015-02-26 01:55:35 +00:00
|
|
|
if hasattr(self, "get_legacy_vm_workdir"):
|
2016-05-11 17:35:36 +00:00
|
|
|
# rename old project node working dir
|
|
|
|
log.info("Converting old node working directory...")
|
2015-02-26 01:55:35 +00:00
|
|
|
legacy_vm_dir = self.get_legacy_vm_workdir(legacy_id, name)
|
2015-03-01 01:55:53 +00:00
|
|
|
legacy_vm_working_path = os.path.join(new_project_files_path, legacy_vm_dir)
|
2015-03-01 20:05:51 +00:00
|
|
|
new_vm_working_path = os.path.join(new_project_files_path, self.module_name.lower(), new_id)
|
2015-03-02 20:04:30 +00:00
|
|
|
if os.path.exists(legacy_vm_working_path) and not os.path.exists(new_vm_working_path):
|
|
|
|
try:
|
|
|
|
log.info('Moving "{}" to "{}"'.format(legacy_vm_working_path, new_vm_working_path))
|
|
|
|
yield from wait_run_in_executor(shutil.move, legacy_vm_working_path, new_vm_working_path)
|
|
|
|
except OSError as e:
|
2016-05-11 17:35:36 +00:00
|
|
|
raise aiohttp.web.HTTPInternalServerError(text="Could not move vm working directory: {} to {} {}".format(legacy_vm_working_path,
|
2016-05-16 12:30:09 +00:00
|
|
|
new_vm_working_path, e))
|
2015-02-28 22:00:00 +00:00
|
|
|
|
2015-03-01 20:05:51 +00:00
|
|
|
return new_id
|
2015-02-26 01:55:35 +00:00
|
|
|
|
2015-01-14 01:26:32 +00:00
|
|
|
@asyncio.coroutine
|
2016-05-11 17:35:36 +00:00
|
|
|
def create_node(self, name, project_id, node_id, *args, **kwargs):
|
2015-01-20 11:46:15 +00:00
|
|
|
"""
|
2016-05-11 17:35:36 +00:00
|
|
|
Create a new node
|
2015-01-20 11:46:15 +00:00
|
|
|
|
2016-05-11 17:35:36 +00:00
|
|
|
:param name: Node name
|
2015-02-04 20:48:29 +00:00
|
|
|
:param project_id: Project identifier
|
2016-05-11 17:35:36 +00:00
|
|
|
:param node_id: restore a node identifier
|
2015-01-20 11:46:15 +00:00
|
|
|
"""
|
2015-01-20 22:28:40 +00:00
|
|
|
|
2016-05-11 17:35:36 +00:00
|
|
|
if node_id in self._nodes:
|
|
|
|
return self._nodes[node_id]
|
2015-03-02 14:35:36 +00:00
|
|
|
|
2015-02-04 20:48:29 +00:00
|
|
|
project = ProjectManager.instance().get_project(project_id)
|
2016-05-11 17:35:36 +00:00
|
|
|
if node_id and isinstance(node_id, int):
|
|
|
|
# old project
|
2015-02-26 01:55:35 +00:00
|
|
|
with (yield from BaseManager._convert_lock):
|
2016-05-11 17:35:36 +00:00
|
|
|
node_id = yield from self.convert_old_project(project, node_id, name)
|
2015-01-20 01:30:57 +00:00
|
|
|
|
2016-05-11 17:35:36 +00:00
|
|
|
if not node_id:
|
|
|
|
node_id = str(uuid4())
|
2015-01-20 01:30:57 +00:00
|
|
|
|
2016-05-11 17:35:36 +00:00
|
|
|
node = self._NODE_CLASS(name, node_id, project, self, *args, **kwargs)
|
|
|
|
if asyncio.iscoroutinefunction(node.create):
|
|
|
|
yield from node.create()
|
2015-01-22 00:41:35 +00:00
|
|
|
else:
|
2016-05-11 17:35:36 +00:00
|
|
|
node.create()
|
|
|
|
self._nodes[node.id] = node
|
|
|
|
project.add_node(node)
|
|
|
|
return node
|
2015-01-22 10:34:10 +00:00
|
|
|
|
|
|
|
@asyncio.coroutine
|
2016-05-11 17:35:36 +00:00
|
|
|
def close_node(self, node_id):
|
2015-01-22 10:34:10 +00:00
|
|
|
"""
|
2016-05-11 17:35:36 +00:00
|
|
|
Close a node
|
2015-01-22 10:34:10 +00:00
|
|
|
|
2016-05-11 17:35:36 +00:00
|
|
|
:param node_id: Node identifier
|
2015-02-04 20:48:29 +00:00
|
|
|
|
2016-05-11 17:35:36 +00:00
|
|
|
:returns: Node instance
|
2015-01-22 10:34:10 +00:00
|
|
|
"""
|
|
|
|
|
2016-05-11 17:35:36 +00:00
|
|
|
node = self.get_node(node_id)
|
|
|
|
if asyncio.iscoroutinefunction(node.close):
|
|
|
|
yield from node.close()
|
2015-01-22 10:34:10 +00:00
|
|
|
else:
|
2016-05-11 17:35:36 +00:00
|
|
|
node.close()
|
|
|
|
return node
|
2015-01-23 10:28:58 +00:00
|
|
|
|
2015-03-02 02:20:33 +00:00
|
|
|
@asyncio.coroutine
|
|
|
|
def project_closing(self, project):
|
|
|
|
"""
|
|
|
|
Called when a project is about to be closed.
|
|
|
|
|
|
|
|
:param project: Project instance
|
|
|
|
"""
|
|
|
|
|
|
|
|
pass
|
|
|
|
|
2015-02-16 05:13:24 +00:00
|
|
|
@asyncio.coroutine
|
2015-02-27 23:51:17 +00:00
|
|
|
def project_closed(self, project):
|
2015-02-16 05:13:24 +00:00
|
|
|
"""
|
|
|
|
Called when a project is closed.
|
|
|
|
|
2015-02-27 23:51:17 +00:00
|
|
|
:param project: Project instance
|
|
|
|
"""
|
|
|
|
|
2016-05-11 17:35:36 +00:00
|
|
|
for node in project.nodes:
|
|
|
|
if node.id in self._nodes:
|
|
|
|
del self._nodes[node.id]
|
2015-02-27 23:51:17 +00:00
|
|
|
|
|
|
|
@asyncio.coroutine
|
|
|
|
def project_moved(self, project):
|
|
|
|
"""
|
|
|
|
Called when a project is moved
|
|
|
|
|
|
|
|
:param project: project instance
|
2015-02-16 05:13:24 +00:00
|
|
|
"""
|
|
|
|
|
|
|
|
pass
|
|
|
|
|
2015-03-21 20:52:17 +00:00
|
|
|
@asyncio.coroutine
|
|
|
|
def project_committed(self, project):
|
|
|
|
"""
|
|
|
|
Called when a project is committed.
|
|
|
|
|
|
|
|
:param project: Project instance
|
|
|
|
"""
|
|
|
|
|
|
|
|
pass
|
|
|
|
|
2015-01-23 10:28:58 +00:00
|
|
|
@asyncio.coroutine
|
2016-05-11 17:35:36 +00:00
|
|
|
def delete_node(self, node_id):
|
2015-01-23 10:28:58 +00:00
|
|
|
"""
|
2016-05-11 17:35:36 +00:00
|
|
|
Delete a node. The node working directory will be destroyed when a commit is received.
|
2015-01-23 10:28:58 +00:00
|
|
|
|
2016-05-11 17:35:36 +00:00
|
|
|
:param node_id: Node identifier
|
|
|
|
:returns: Node instance
|
2015-01-23 10:28:58 +00:00
|
|
|
"""
|
|
|
|
|
2016-05-11 17:35:36 +00:00
|
|
|
node = yield from self.close_node(node_id)
|
2016-05-16 12:30:09 +00:00
|
|
|
node.project.emit("node.deleted", node)
|
2016-05-11 17:35:36 +00:00
|
|
|
node.project.mark_node_for_destruction(node)
|
|
|
|
if node.id in self._nodes:
|
|
|
|
del self._nodes[node.id]
|
|
|
|
return node
|
2015-01-23 01:04:24 +00:00
|
|
|
|
|
|
|
@staticmethod
|
2015-09-14 21:05:25 +00:00
|
|
|
def has_privileged_access(executable):
|
2015-01-23 01:04:24 +00:00
|
|
|
"""
|
2016-06-01 23:50:31 +00:00
|
|
|
Check if an executable have the right to attach to Ethernet and TAP adapters.
|
2015-01-23 01:04:24 +00:00
|
|
|
|
|
|
|
:param executable: executable path
|
|
|
|
|
|
|
|
:returns: True or False
|
|
|
|
"""
|
|
|
|
|
|
|
|
if sys.platform.startswith("win"):
|
|
|
|
# do not check anything on Windows
|
|
|
|
return True
|
|
|
|
|
|
|
|
if os.geteuid() == 0:
|
|
|
|
# we are root, so we should have privileged access.
|
|
|
|
return True
|
2015-09-14 21:05:25 +00:00
|
|
|
|
|
|
|
if os.stat(executable).st_uid == 0 and (os.stat(executable).st_mode & stat.S_ISUID or os.stat(executable).st_mode & stat.S_ISGID):
|
2015-01-23 01:04:24 +00:00
|
|
|
# the executable has set UID bit.
|
|
|
|
return True
|
|
|
|
|
|
|
|
# test if the executable has the CAP_NET_RAW capability (Linux only)
|
2015-09-14 21:05:25 +00:00
|
|
|
try:
|
|
|
|
if sys.platform.startswith("linux") and "security.capability" in os.listxattr(executable):
|
2015-01-23 01:04:24 +00:00
|
|
|
caps = os.getxattr(executable, "security.capability")
|
|
|
|
# test the 2nd byte and check if the 13th bit (CAP_NET_RAW) is set
|
|
|
|
if struct.unpack("<IIIII", caps)[1] & 1 << 13:
|
|
|
|
return True
|
2015-09-14 21:05:25 +00:00
|
|
|
except OSError as e:
|
|
|
|
log.error("could not determine if CAP_NET_RAW capability is set for {}: {}".format(executable, e))
|
2015-01-23 01:04:24 +00:00
|
|
|
|
|
|
|
return False
|
|
|
|
|
|
|
|
def create_nio(self, executable, nio_settings):
|
|
|
|
"""
|
|
|
|
Creates a new NIO.
|
|
|
|
|
|
|
|
:param nio_settings: information to create the NIO
|
|
|
|
|
|
|
|
:returns: a NIO object
|
|
|
|
"""
|
|
|
|
|
|
|
|
nio = None
|
|
|
|
if nio_settings["type"] == "nio_udp":
|
|
|
|
lport = nio_settings["lport"]
|
|
|
|
rhost = nio_settings["rhost"]
|
|
|
|
rport = nio_settings["rport"]
|
|
|
|
try:
|
2015-06-07 03:37:34 +00:00
|
|
|
info = socket.getaddrinfo(rhost, rport, socket.AF_UNSPEC, socket.SOCK_DGRAM, 0, socket.AI_PASSIVE)
|
|
|
|
if not info:
|
|
|
|
raise aiohttp.web.HTTPInternalServerError(text="getaddrinfo returns an empty list on {}:{}".format(rhost, rport))
|
|
|
|
for res in info:
|
|
|
|
af, socktype, proto, _, sa = res
|
|
|
|
with socket.socket(af, socktype, proto) as sock:
|
|
|
|
sock.connect(sa)
|
2015-01-23 01:04:24 +00:00
|
|
|
except OSError as e:
|
|
|
|
raise aiohttp.web.HTTPInternalServerError(text="Could not create an UDP connection to {}:{}: {}".format(rhost, rport, e))
|
2015-02-24 02:00:34 +00:00
|
|
|
nio = NIOUDP(lport, rhost, rport)
|
2015-01-23 01:04:24 +00:00
|
|
|
elif nio_settings["type"] == "nio_tap":
|
|
|
|
tap_device = nio_settings["tap_device"]
|
2015-10-05 18:12:20 +00:00
|
|
|
# if not is_interface_up(tap_device):
|
2015-09-26 16:09:50 +00:00
|
|
|
# raise aiohttp.web.HTTPConflict(text="TAP interface {} does not exist or is down".format(tap_device))
|
2015-04-28 08:43:27 +00:00
|
|
|
# FIXME: check for permissions on tap device
|
2015-09-14 21:05:25 +00:00
|
|
|
# if not self.has_privileged_access(executable):
|
2015-04-27 20:38:15 +00:00
|
|
|
# raise aiohttp.web.HTTPForbidden(text="{} has no privileged access to {}.".format(executable, tap_device))
|
2015-02-24 02:00:34 +00:00
|
|
|
nio = NIOTAP(tap_device)
|
2016-06-01 23:50:31 +00:00
|
|
|
elif nio_settings["type"] in ("nio_generic_ethernet", "nio_ethernet"):
|
2015-06-07 19:51:33 +00:00
|
|
|
ethernet_device = nio_settings["ethernet_device"]
|
|
|
|
if not is_interface_up(ethernet_device):
|
2015-07-22 04:58:28 +00:00
|
|
|
raise aiohttp.web.HTTPConflict(text="Ethernet interface {} does not exist or is down".format(ethernet_device))
|
2016-06-01 23:50:31 +00:00
|
|
|
nio = NIOEthernet(ethernet_device)
|
2015-01-23 01:04:24 +00:00
|
|
|
assert nio is not None
|
|
|
|
return nio
|
2015-04-14 16:46:55 +00:00
|
|
|
|
2016-06-02 13:19:34 +00:00
|
|
|
def images_directories(self):
|
|
|
|
"""
|
|
|
|
Return all directory where we will look for images
|
|
|
|
by priority
|
|
|
|
"""
|
|
|
|
server_config = self.config.get_section_config("Server")
|
|
|
|
|
|
|
|
paths = []
|
|
|
|
img_directory = self.get_images_directory()
|
|
|
|
os.makedirs(img_directory, exist_ok=True)
|
|
|
|
paths.append(img_directory)
|
|
|
|
for directory in server_config.get("additional_images_path", "").split(":"):
|
|
|
|
paths.append(directory)
|
|
|
|
# Compatibility with old topologies we look in parent directory
|
|
|
|
paths.append(os.path.normpath(os.path.join(self.get_images_directory(), '..')))
|
|
|
|
# Return only the existings paths
|
|
|
|
return [force_unix_path(p) for p in paths if os.path.exists(p)]
|
|
|
|
|
2015-04-14 16:46:55 +00:00
|
|
|
def get_abs_image_path(self, path):
|
|
|
|
"""
|
|
|
|
Get the absolute path of an image
|
|
|
|
|
|
|
|
:param path: file path
|
|
|
|
:return: file path
|
|
|
|
"""
|
|
|
|
|
2015-04-24 23:27:32 +00:00
|
|
|
if not path:
|
|
|
|
return ""
|
2015-11-12 14:37:34 +00:00
|
|
|
|
2016-06-02 13:19:34 +00:00
|
|
|
server_config = self.config.get_section_config("Server")
|
2015-04-14 16:46:55 +00:00
|
|
|
img_directory = self.get_images_directory()
|
2015-11-12 14:37:34 +00:00
|
|
|
|
|
|
|
# Windows path should not be send to a unix server
|
|
|
|
if not sys.platform.startswith("win"):
|
|
|
|
if re.match(r"^[A-Z]:", path) is not None:
|
2016-05-11 17:35:36 +00:00
|
|
|
raise NodeError("{} is not allowed on this remote server. Please use only a filename in {}.".format(path, img_directory))
|
2015-11-12 14:37:34 +00:00
|
|
|
|
2015-04-14 16:46:55 +00:00
|
|
|
if not os.path.isabs(path):
|
2016-06-02 13:19:34 +00:00
|
|
|
orig_path = path
|
|
|
|
for directory in self.images_directories():
|
2016-06-02 14:44:38 +00:00
|
|
|
path = self._recursive_search_file_in_directory(directory, orig_path)
|
|
|
|
if path:
|
2016-06-02 13:19:34 +00:00
|
|
|
return force_unix_path(path)
|
2016-06-07 13:34:04 +00:00
|
|
|
|
|
|
|
# Not found we try the default directory
|
2016-06-02 14:44:38 +00:00
|
|
|
s = os.path.split(orig_path)
|
2016-06-07 13:34:04 +00:00
|
|
|
path = force_unix_path(os.path.join(self.get_images_directory(), *s))
|
|
|
|
if os.path.exists(path):
|
|
|
|
return path
|
|
|
|
raise ImageMissingError(path)
|
2015-05-04 08:57:08 +00:00
|
|
|
|
2016-06-02 13:19:34 +00:00
|
|
|
# For non local server we disallow using absolute path outside image directory
|
|
|
|
if server_config.get("local", False) is True:
|
2016-06-07 13:34:04 +00:00
|
|
|
path = force_unix_path(path)
|
|
|
|
if os.path.exists(path):
|
|
|
|
return path
|
|
|
|
raise ImageMissingError(path)
|
2015-11-09 18:02:10 +00:00
|
|
|
|
2016-06-02 13:19:34 +00:00
|
|
|
path = force_unix_path(path)
|
|
|
|
for directory in self.images_directories():
|
|
|
|
if os.path.commonprefix([directory, path]) == directory:
|
2016-06-07 13:34:04 +00:00
|
|
|
if os.path.exists(path):
|
|
|
|
return path
|
|
|
|
raise ImageMissingError(path)
|
2016-06-02 13:19:34 +00:00
|
|
|
raise NodeError("{} is not allowed on this remote server. Please use only a filename in {}.".format(path, self.get_images_directory()))
|
2015-04-14 16:46:55 +00:00
|
|
|
|
2016-06-02 14:44:38 +00:00
|
|
|
def _recursive_search_file_in_directory(self, directory, searched_file):
|
|
|
|
"""
|
|
|
|
Search for a file in directory and is subdirectories
|
|
|
|
|
|
|
|
:returns: Path or None if not found
|
|
|
|
"""
|
|
|
|
s = os.path.split(searched_file)
|
|
|
|
|
|
|
|
for root, dirs, files in os.walk(directory):
|
|
|
|
for file in files:
|
|
|
|
# If filename is the same
|
|
|
|
if s[1] == file and (s[0] == '' or s[0] == os.path.basename(root)):
|
|
|
|
path = os.path.normpath(os.path.join(root, s[1]))
|
|
|
|
if os.path.exists(path):
|
|
|
|
return path
|
|
|
|
return None
|
|
|
|
|
2015-04-14 16:46:55 +00:00
|
|
|
def get_relative_image_path(self, path):
|
|
|
|
"""
|
|
|
|
Get a path relative to images directory path
|
|
|
|
or an abspath if the path is not located inside
|
|
|
|
image directory
|
|
|
|
|
|
|
|
:param path: file path
|
|
|
|
:return: file path
|
|
|
|
"""
|
|
|
|
|
2015-04-24 23:27:32 +00:00
|
|
|
if not path:
|
|
|
|
return ""
|
2015-12-14 15:06:51 +00:00
|
|
|
path = force_unix_path(self.get_abs_image_path(path))
|
2016-06-02 13:19:34 +00:00
|
|
|
for directory in self.images_directories():
|
|
|
|
if os.path.commonprefix([directory, path]) == directory:
|
|
|
|
return os.path.relpath(path, directory)
|
2015-04-14 16:46:55 +00:00
|
|
|
return path
|
|
|
|
|
2015-04-13 12:33:13 +00:00
|
|
|
@asyncio.coroutine
|
|
|
|
def list_images(self):
|
|
|
|
"""
|
2016-05-11 17:35:36 +00:00
|
|
|
Return the list of available images for this node type
|
2015-04-13 12:33:13 +00:00
|
|
|
|
|
|
|
:returns: Array of hash
|
|
|
|
"""
|
|
|
|
|
|
|
|
images = []
|
2015-10-05 09:07:15 +00:00
|
|
|
img_dir = self.get_images_directory()
|
|
|
|
for root, dirs, files in os.walk(img_dir):
|
|
|
|
for filename in files:
|
|
|
|
if filename[0] != "." and not filename.endswith(".md5sum"):
|
|
|
|
path = os.path.relpath(os.path.join(root, filename), img_dir)
|
|
|
|
images.append({
|
|
|
|
"filename": filename,
|
|
|
|
"path": path})
|
2015-04-13 12:33:13 +00:00
|
|
|
return images
|
|
|
|
|
2015-04-14 16:46:55 +00:00
|
|
|
def get_images_directory(self):
|
|
|
|
"""
|
|
|
|
Get the image directory on disk
|
|
|
|
"""
|
2015-04-24 23:27:32 +00:00
|
|
|
|
2015-04-14 16:46:55 +00:00
|
|
|
raise NotImplementedError
|
2015-04-24 08:15:23 +00:00
|
|
|
|
|
|
|
@asyncio.coroutine
|
|
|
|
def write_image(self, filename, stream):
|
|
|
|
directory = self.get_images_directory()
|
2015-10-05 18:12:20 +00:00
|
|
|
path = os.path.abspath(os.path.join(directory, *os.path.split(filename)))
|
|
|
|
if os.path.commonprefix([directory, path]) != directory:
|
|
|
|
raise aiohttp.web.HTTPForbidden(text="Could not write image: {}, {} is forbiden".format(filename, path))
|
2015-04-24 08:15:23 +00:00
|
|
|
log.info("Writting image file %s", path)
|
|
|
|
try:
|
2015-06-17 15:11:25 +00:00
|
|
|
remove_checksum(path)
|
2016-05-02 15:25:46 +00:00
|
|
|
# We store the file under his final name only when the upload is finished
|
|
|
|
tmp_path = path + ".tmp"
|
2015-10-05 18:12:20 +00:00
|
|
|
os.makedirs(os.path.dirname(path), exist_ok=True)
|
2016-05-02 15:25:46 +00:00
|
|
|
with open(tmp_path, 'wb+') as f:
|
2015-04-24 08:15:23 +00:00
|
|
|
while True:
|
|
|
|
packet = yield from stream.read(512)
|
|
|
|
if not packet:
|
|
|
|
break
|
|
|
|
f.write(packet)
|
2016-05-02 15:25:46 +00:00
|
|
|
os.chmod(tmp_path, stat.S_IWRITE | stat.S_IREAD | stat.S_IEXEC)
|
|
|
|
shutil.move(tmp_path, path)
|
2015-06-17 15:11:25 +00:00
|
|
|
md5sum(path)
|
2015-04-24 08:15:23 +00:00
|
|
|
except OSError as e:
|
2015-10-05 18:12:20 +00:00
|
|
|
raise aiohttp.web.HTTPConflict(text="Could not write image: {} because {}".format(filename, e))
|