# # Copyright (C) 2020 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 . """ API routes for projects. """ import os import asyncio import tempfile import aiofiles import time import urllib.parse import gns3server.utils.zipfile_zstd as zipfile import logging log = logging.getLogger() from fastapi import APIRouter, Depends, Request, Response, Body, HTTPException, status, WebSocket, WebSocketDisconnect from fastapi.encoders import jsonable_encoder from fastapi.responses import StreamingResponse, FileResponse from websockets.exceptions import ConnectionClosed, WebSocketException from typing import List, Optional from uuid import UUID from pathlib import Path from gns3server import schemas from gns3server.controller import Controller from gns3server.controller.project import Project from gns3server.controller.controller_error import ControllerError, ControllerBadRequestError from gns3server.controller.import_project import import_project as import_controller_project from gns3server.controller.export_project import export_project as export_controller_project from gns3server.utils.asyncio import aiozipstream from gns3server.utils.path import is_safe_path from gns3server.config import Config from gns3server.db.repositories.rbac import RbacRepository from gns3server.db.repositories.templates import TemplatesRepository from gns3server.services.templates import TemplatesService from .dependencies.authentication import get_current_active_user, get_current_active_user_from_websocket from .dependencies.database import get_repository responses = {404: {"model": schemas.ErrorMessage, "description": "Could not find project"}} router = APIRouter(responses=responses) def dep_project(project_id: UUID) -> Project: """ Dependency to retrieve a project. """ project = Controller.instance().get_project(str(project_id)) return project CHUNK_SIZE = 1024 * 8 # 8KB @router.get("", response_model=List[schemas.Project], response_model_exclude_unset=True) async def get_projects( current_user: schemas.User = Depends(get_current_active_user), rbac_repo: RbacRepository = Depends(get_repository(RbacRepository)) ) -> List[schemas.Project]: """ Return all projects. """ controller = Controller.instance() if current_user.is_superadmin: return [p.asdict() for p in controller.projects.values()] else: user_projects = [] for project in controller.projects.values(): authorized = await rbac_repo.check_user_is_authorized( current_user.user_id, "GET", f"/projects/{project.id}") if authorized: user_projects.append(project.asdict()) return user_projects @router.post( "", status_code=status.HTTP_201_CREATED, response_model=schemas.Project, response_model_exclude_unset=True, responses={409: {"model": schemas.ErrorMessage, "description": "Could not create project"}}, ) async def create_project( project_data: schemas.ProjectCreate, current_user: schemas.User = Depends(get_current_active_user), rbac_repo: RbacRepository = Depends(get_repository(RbacRepository)) ) -> schemas.Project: """ Create a new project. """ controller = Controller.instance() project = await controller.add_project(**jsonable_encoder(project_data, exclude_unset=True)) await rbac_repo.add_permission_to_user_with_path(current_user.user_id, f"/projects/{project.id}/*") return project.asdict() @router.get("/{project_id}", response_model=schemas.Project) def get_project(project: Project = Depends(dep_project)) -> schemas.Project: """ Return a project. """ return project.asdict() @router.put("/{project_id}", response_model=schemas.Project, response_model_exclude_unset=True) async def update_project( project_data: schemas.ProjectUpdate, project: Project = Depends(dep_project) ) -> schemas.Project: """ Update a project. """ await project.update(**jsonable_encoder(project_data, exclude_unset=True)) return project.asdict() @router.delete("/{project_id}", status_code=status.HTTP_204_NO_CONTENT) async def delete_project( project: Project = Depends(dep_project), rbac_repo: RbacRepository = Depends(get_repository(RbacRepository)) ) -> Response: """ Delete a project. """ controller = Controller.instance() await project.delete() controller.remove_project(project) await rbac_repo.delete_all_permissions_with_path(f"/projects/{project.id}") return Response(status_code=status.HTTP_204_NO_CONTENT) @router.get("/{project_id}/stats") def get_project_stats(project: Project = Depends(dep_project)) -> dict: """ Return a project statistics. """ return project.stats() @router.post( "/{project_id}/close", status_code=status.HTTP_204_NO_CONTENT, responses={**responses, 409: {"model": schemas.ErrorMessage, "description": "Could not close project"}}, ) async def close_project(project: Project = Depends(dep_project)) -> Response: """ Close a project. """ await project.close() return Response(status_code=status.HTTP_204_NO_CONTENT) @router.post( "/{project_id}/open", status_code=status.HTTP_201_CREATED, response_model=schemas.Project, responses={**responses, 409: {"model": schemas.ErrorMessage, "description": "Could not open project"}}, ) async def open_project(project: Project = Depends(dep_project)) -> schemas.Project: """ Open a project. """ await project.open() return project.asdict() @router.post( "/load", status_code=status.HTTP_201_CREATED, response_model=schemas.Project, responses={**responses, 409: {"model": schemas.ErrorMessage, "description": "Could not load project"}}, ) async def load_project(path: str = Body(..., embed=True)) -> schemas.Project: """ Load a project (local server only). """ controller = Controller.instance() dot_gns3_file = path project = await controller.load_project(dot_gns3_file) return project.asdict() @router.get("/{project_id}/notifications") async def project_http_notifications(project_id: UUID) -> StreamingResponse: """ Receive project notifications about the controller from HTTP stream. """ controller = Controller.instance() project = controller.get_project(str(project_id)) log.info(f"New client has connected to the notification stream for project ID '{project.id}' (HTTP steam method)") async def event_stream(): try: with controller.notification.project_queue(project.id) as queue: while True: msg = await queue.get_json(5) yield (f"{msg}\n").encode("utf-8") finally: log.info(f"Client has disconnected from notification for project ID '{project.id}' (HTTP stream method)") if project.auto_close: # To avoid trouble with client connecting disconnecting we sleep few seconds before checking # if someone else is not connected await asyncio.sleep(5) if not controller.notification.project_has_listeners(project.id): log.info(f"Project '{project.id}' is automatically closing due to no client listening") await project.close() return StreamingResponse(event_stream(), media_type="application/json") @router.websocket("/{project_id}/notifications/ws") async def project_ws_notifications( project_id: UUID, websocket: WebSocket, current_user: schemas.User = Depends(get_current_active_user_from_websocket) ) -> None: """ Receive project notifications about the controller from WebSocket. """ if current_user is None: return controller = Controller.instance() project = controller.get_project(str(project_id)) log.info(f"New client has connected to the notification stream for project ID '{project.id}' (WebSocket method)") try: with controller.notification.project_queue(project.id) as queue: while True: notification = await queue.get_json(5) await websocket.send_text(notification) except (ConnectionClosed, WebSocketDisconnect): log.info(f"Client has disconnected from notification stream for project ID '{project.id}' (WebSocket method)") except WebSocketException as e: log.warning(f"Error while sending to project event to WebSocket client: {e}") finally: try: await websocket.close() except OSError: pass # ignore OSError: [Errno 107] Transport endpoint is not connected if project.auto_close: # To avoid trouble with client connecting disconnecting we sleep few seconds before checking # if someone else is not connected await asyncio.sleep(5) if not controller.notification.project_has_listeners(project.id): log.info(f"Project '{project.id}' is automatically closing due to no client listening") await project.close() @router.get("/{project_id}/export") async def export_project( project: Project = Depends(dep_project), include_snapshots: bool = False, include_images: bool = False, reset_mac_addresses: bool = False, compression: schemas.ProjectCompression = "zstd", compression_level: int = None, ) -> StreamingResponse: """ Export a project as a portable archive. """ compression_query = compression.lower() if compression_query == "zip": compression = zipfile.ZIP_DEFLATED if compression_level is not None and (compression_level < 0 or compression_level > 9): raise ControllerBadRequestError("Compression level must be between 0 and 9 for ZIP compression") elif compression_query == "none": compression = zipfile.ZIP_STORED elif compression_query == "bzip2": compression = zipfile.ZIP_BZIP2 if compression_level is not None and (compression_level < 1 or compression_level > 9): raise ControllerBadRequestError("Compression level must be between 1 and 9 for BZIP2 compression") elif compression_query == "lzma": compression = zipfile.ZIP_LZMA elif compression_query == "zstd": compression = zipfile.ZIP_ZSTANDARD if compression_level is not None and (compression_level < 1 or compression_level > 22): raise ControllerBadRequestError("Compression level must be between 1 and 22 for Zstandard compression") if compression_level is not None and compression_query in ("none", "lzma"): raise ControllerBadRequestError(f"Compression level is not supported for '{compression_query}' compression method") try: begin = time.time() # use the parent directory as a temporary working dir working_dir = os.path.abspath(os.path.join(project.path, os.pardir)) async def streamer(): log.info(f"Exporting project '{project.name}' with '{compression_query}' compression " f"(level {compression_level})") with tempfile.TemporaryDirectory(dir=working_dir) as tmpdir: with aiozipstream.ZipFile(compression=compression, compresslevel=compression_level) as zstream: await export_controller_project( zstream, project, tmpdir, include_snapshots=include_snapshots, include_images=include_images, reset_mac_addresses=reset_mac_addresses, ) async for chunk in zstream: yield chunk log.info(f"Project '{project.name}' exported in {time.time() - begin:.4f} seconds") # Will be raise if you have no space left or permission issue on your temporary directory # RuntimeError: something was wrong during the zip process except (ValueError, OSError, RuntimeError) as e: raise ConnectionError(f"Cannot export project: {e}") headers = {"CONTENT-DISPOSITION": f'attachment; filename="{project.name}.gns3project"'} return StreamingResponse(streamer(), media_type="application/gns3project", headers=headers) @router.post("/{project_id}/import", status_code=status.HTTP_201_CREATED, response_model=schemas.Project) async def import_project( project_id: UUID, request: Request, path: Optional[Path] = None, name: Optional[str] = None ) -> schemas.Project: """ Import a project from a portable archive. """ #TODO: import project remotely raise NotImplementedError() controller = Controller.instance() # We write the content to a temporary location and after we extract it all. # It could be more optimal to stream this but it is not implemented in Python. try: begin = time.time() # use the parent directory or projects dir as a temporary working dir if path: working_dir = os.path.abspath(os.path.join(path, os.pardir)) else: working_dir = controller.projects_directory() with tempfile.TemporaryDirectory(dir=working_dir) as tmpdir: temp_project_path = os.path.join(tmpdir, "project.zip") async with aiofiles.open(temp_project_path, "wb") as f: async for chunk in request.stream(): await f.write(chunk) with open(temp_project_path, "rb") as f: project = await import_controller_project(controller, str(project_id), f, location=path, name=name) log.info(f"Project '{project.name}' imported in {time.time() - begin:.4f} seconds") except OSError as e: raise ControllerError(f"Could not import the project: {e}") return project.asdict() @router.post( "/{project_id}/duplicate", status_code=status.HTTP_201_CREATED, response_model=schemas.Project, responses={**responses, 409: {"model": schemas.ErrorMessage, "description": "Could not duplicate project"}}, ) async def duplicate_project( project_data: schemas.ProjectDuplicate, project: Project = Depends(dep_project), current_user: schemas.User = Depends(get_current_active_user), rbac_repo: RbacRepository = Depends(get_repository(RbacRepository)) ) -> schemas.Project: """ Duplicate a project. """ reset_mac_addresses = project_data.reset_mac_addresses new_project = await project.duplicate( name=project_data.name, reset_mac_addresses=reset_mac_addresses ) await rbac_repo.add_permission_to_user_with_path(current_user.user_id, f"/projects/{new_project.id}/*") return new_project.asdict() @router.get("/{project_id}/files/{file_path:path}") async def get_file(file_path: str, project: Project = Depends(dep_project)) -> FileResponse: """ Return a file from a project. """ file_path = urllib.parse.unquote(file_path) path = os.path.normpath(file_path) # Raise error if user try to escape if not is_safe_path(path, project.path): raise HTTPException(status_code=status.HTTP_403_FORBIDDEN) path = os.path.join(project.path, path) if not os.path.exists(path): raise HTTPException(status_code=status.HTTP_404_NOT_FOUND) return FileResponse(path, media_type="application/octet-stream") @router.post("/{project_id}/files/{file_path:path}", status_code=status.HTTP_204_NO_CONTENT) async def write_file(file_path: str, request: Request, project: Project = Depends(dep_project)) -> Response: """ Write a file to a project. """ file_path = urllib.parse.unquote(file_path) path = os.path.normpath(file_path) # Raise error if user try to escape if not is_safe_path(path, project.path): raise HTTPException(status_code=status.HTTP_403_FORBIDDEN) path = os.path.join(project.path, path) try: async with aiofiles.open(path, "wb+") as f: async for chunk in request.stream(): await f.write(chunk) except FileNotFoundError: raise HTTPException(status_code=status.HTTP_404_NOT_FOUND) except PermissionError: raise HTTPException(status_code=status.HTTP_401_UNAUTHORIZED) except OSError as e: raise ControllerError(str(e)) return Response(status_code=status.HTTP_204_NO_CONTENT) @router.post( "/{project_id}/templates/{template_id}", response_model=schemas.Node, status_code=status.HTTP_201_CREATED, responses={404: {"model": schemas.ErrorMessage, "description": "Could not find project or template"}}, ) async def create_node_from_template( project_id: UUID, template_id: UUID, template_usage: schemas.TemplateUsage, templates_repo: TemplatesRepository = Depends(get_repository(TemplatesRepository)), ) -> schemas.Node: """ Create a new node from a template. """ template = await TemplatesService(templates_repo).get_template(template_id) controller = Controller.instance() project = controller.get_project(str(project_id)) node = await project.add_node_from_template( template, x=template_usage.x, y=template_usage.y, compute_id=template_usage.compute_id ) return node.asdict()