gns3-server/gns3server/controller/link.py

#!/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 os
import re
import uuid
import html

from .controller_error import ControllerError, ControllerNotFoundError
from gns3server.agent.web_wireshark.manager import WebWiresharkManager
from gns3server.config import Config
from gns3server.utils.packet_filter_validation import validate_all_filters, FilterValidationError

import logging

log = logging.getLogger(__name__)


FILTERS = [
    {
        "type": "frequency_drop",
        "name": "Frequency drop",
        "description": "It will drop everything with a -1 frequency, drop every Nth packet with a positive frequency, or drop nothing",
        "parameters": [{"name": "Frequency", "minimum": -1, "maximum": 32767, "type": "int", "unit": "th packet"}],
    },
    {
        "type": "packet_loss",
        "name": "Packet loss",
        "description": "The percentage represents the chance for a packet to be lost",
        "parameters": [{"name": "Chance", "minimum": 0, "maximum": 100, "type": "int", "unit": "%"}],
    },
    {
        "type": "delay",
        "name": "Delay",
        "description": "Delay packets in milliseconds. You can add jitter in milliseconds (+/-) of the delay",
        "parameters": [
            {"name": "Latency", "minimum": 0, "maximum": 32767, "unit": "ms", "type": "int"},
            {"name": "Jitter (-/+)", "minimum": 0, "maximum": 32767, "unit": "ms", "type": "int"},
        ],
    },
    {
        "type": "corrupt",
        "name": "Corrupt",
        "description": "The percentage represents the chance for a packet to be corrupted",
        "parameters": [{"name": "Chance", "minimum": 0, "maximum": 100, "unit": "%", "type": "int"}],
    },
    {
        "type": "bpf",
        "name": "Berkeley Packet Filter (BPF)",
        "description": "This filter will drop any packet matching a BPF expression. Put one expression per line",
        "parameters": [{"name": "Filters", "type": "text"}],
    },
]


class Link:
    """
    Base class for links.
    """

    def __init__(self, project, link_id=None):

        if link_id:
            self._id = link_id
        else:
            self._id = str(uuid.uuid4())
        self._nodes = []
        self._project = project
        self._capturing = False
        self._capture_node = None
        self._capture_file_name = None
        self._streaming_pcap = None
        self._created = False
        self._link_type = "ethernet"
        self._suspended = False
        self._filters = {}
        self._link_style = {}
        self._wireshark = False
        self._show_filters_icon = True

    @property
    def filters(self):
        """
        Get an array of filters
        """
        return self._filters

    @property
    def show_filters_icon(self):
        """
        Get whether to show filters icon in Web UI
        """
        return getattr(self, '_show_filters_icon', True)

    @property
    def project(self):
        """
        Get the project this link belongs to.

        :return: Project instance.
        """
        return self._project

    @property
    def capture_node(self):
        """
        Get the capturing node

        :return: Node instance.
        """
        return self._capture_node

    @property
    def compute(self):
        """
        Get the capturing node

        :return: Node instance.
        """
        assert self.capture_node
        return self.capture_node["node"].compute

    def get_active_filters(self):
        """
        Return the active filters.
        Filters are overridden if the link is suspended.
        """
        if self._suspended:
            # this is to allow all node types to support suspend link
            return {"frequency_drop": [-1]}
        return self._filters

    async def update_filters(self, filters):
        """
        Modify the filters list.

        Filter with value 0 will be dropped because not active
        """
        new_filters = {}
        for (filter, values) in filters.items():
            new_values = []
            for value in values:
                if isinstance(value, str):
                    new_values.append(value.strip("\n "))
                else:
                    new_values.append(int(value))
            values = new_values
            if len(values) != 0 and values[0] != 0 and values[0] != "":
                new_filters[filter] = values

        # Validate filter parameters before applying
        try:
            validate_all_filters(new_filters)
        except FilterValidationError as e:
            raise ControllerError(f"Invalid packet filter parameters: {str(e)}")

        if new_filters != self.filters:
            self._filters = new_filters
            if self._created:
                await self.update()
                self._project.emit_notification("link.updated", self.asdict())
                self._project.dump()

    async def update_suspend(self, value):
        if value != self._suspended:
            self._suspended = value
            await self.update()
            self._project.emit_notification("link.updated", self.asdict())
            self._project.dump()

    async def update_show_filters_icon(self, value):
        """
        Update the show_filters_icon property.

        :param value: Boolean indicating whether to show filters icon in Web UI
        """
        if value != self._show_filters_icon:
            self._show_filters_icon = value
            self._project.emit_notification("link.updated", self.asdict())
            self._project.dump()

    async def update_link_style(self, link_style):
        if link_style != self._link_style:
            self._link_style = link_style
            self._project.emit_notification("link.updated", self.asdict())
            self._project.dump()

    @property
    def created(self):
        """
        :returns: True the link has been created on the computes
        """
        return self._created

    async def add_node(self, node, adapter_number, port_number, label=None, dump=True):
        """
        Add a node to the link

        :param dump: Dump project on disk
        """

        port = node.get_port(adapter_number, port_number)
        if port is None:
            raise ControllerNotFoundError(f"Port {adapter_number}/{port_number} for {node.name} not found")
        if port.link is not None:
            raise ControllerError("Port is already used")

        self._link_type = port.link_type

        for other_node in self._nodes:
            if other_node["node"] == node:
                raise ControllerError("Cannot connect to itself")

            if node.node_type in ["nat", "cloud"]:
                if other_node["node"].node_type in ["nat", "cloud"]:
                    raise ControllerError(
                        "Connecting a {} to a {} is not allowed".format(other_node["node"].node_type, node.node_type)
                    )

            # Check if user is not connecting serial => ethernet
            other_port = other_node["node"].get_port(other_node["adapter_number"], other_node["port_number"])
            if other_port is None:
                raise ControllerNotFoundError(
                    "Port {}/{} for {} not found".format(
                        other_node["adapter_number"], other_node["port_number"], other_node["node"].name
                    )
                )
            if port.link_type != other_port.link_type:
                raise ControllerError(
                    f"Connecting a {other_port.link_type} interface to a {port.link_type} interface is not allowed"
                )

        if label is None:
            label = {
                "text": html.escape(f"{adapter_number}/{port_number}"),
                "style": "font-family: TypeWriter;font-size: 10.0;font-weight: bold;fill: #000000;fill-opacity: 1.0;",
            }

        self._nodes.append(
            {"node": node, "adapter_number": adapter_number, "port_number": port_number, "port": port, "label": label}
        )

        if len(self._nodes) == 2:
            await self.create()
            for n in self._nodes:
                n["node"].add_link(self)
                n["port"].link = self
            self._created = True
            self._project.emit_notification("link.created", self.asdict())

        if dump:
            self._project.dump()

    async def update_nodes(self, nodes):
        for node_data in nodes:
            node = self._project.get_node(node_data["node_id"])
            for port in self._nodes:
                if port["node"] == node:
                    label = node_data.get("label")
                    if label:
                        port["label"] = label
        self._project.emit_notification("link.updated", self.asdict())
        self._project.dump()

    async def create(self):
        """
        Create the link
        """

        raise NotImplementedError

    async def update(self):
        """
        Update a link
        """
        raise NotImplementedError

    async def delete(self):
        """
        Delete the link
        """
        for n in self._nodes:
            # It could be different of self if we rollback an already existing link
            if n["port"].link == self:
                n["port"].link = None
                n["node"].remove_link(self)

    async def reset(self):
        """
        Reset a link
        """

        raise NotImplementedError

    async def start_capture(self, data_link_type="DLT_EN10MB", capture_file_name=None, wireshark=False, jwt_token=None):
        """
        Start capture on the link

        :param data_link_type: PCAP data link type
        :param capture_file_name: PCAP capture file name
        :param wireshark: Enable Web Wireshark
        :param jwt_token: JWT token for authentication
        :returns: Capture object
        """

        self._capturing = True
        self._capture_file_name = capture_file_name
        self._project.emit_notification("link.updated", self.asdict())

        # Start Web Wireshark if requested
        if wireshark:
            await self._start_web_wireshark(jwt_token)

    async def stop_capture(self):
        """
        Stop capture on the link
        """

        # Stop Web Wireshark
        if self._capturing:
            await self._stop_web_wireshark()

        self._capturing = False
        self._project.emit_notification("link.updated", self.asdict())

    async def _start_web_wireshark(self, jwt_token: str):
        """Start Web Wireshark.

        Args:
            jwt_token: JWT authentication token

        Raises:
            ControllerError: If startup fails
        """
        if not jwt_token:
            raise ControllerError("JWT token is required for Web Wireshark")

        # Load WebWireshark configuration from config file
        webwireshark_config = Config.instance().settings.WebWireshark

        manager = WebWiresharkManager()
        try:
            log.info(f"Starting Web Wireshark for link {self.id}")

            result = await manager.start_wireshark_session(
                project_id=self._project.id,
                link_id=self.id,
                jwt_token=jwt_token,
                memory=webwireshark_config.memory,
                cpus=webwireshark_config.cpus,
                pids_limit=webwireshark_config.pids_limit
            )

            # Send notification
            self._project.emit_notification("link.web_wireshark_started", {
                "link_id": self.id,
                "ws_url": result.get("ws_url", result.get("url"))
            })

            log.info(f"Web Wireshark started for link {self.id}: {result.get('ws_url')}")
            self._wireshark = True

        except ControllerError:
            # Re-raise ControllerError to return error to client
            raise
        except Exception as e:
            error_msg = f"Error starting Web Wireshark: {str(e)}"
            log.error(error_msg)
            raise ControllerError(error_msg)
        finally:
            await manager.close()

    async def _stop_web_wireshark(self):
        """Stop Web Wireshark"""
        manager = WebWiresharkManager()
        try:
            log.info(f"Stopping Web Wireshark for link {self.id}")

            await manager.stop_wireshark_session(
                project_id=self._project.id,
                link_id=self.id
            )

            log.info(f"Web Wireshark stopped for link {self.id}")
            self._wireshark = False

        except Exception as e:
            log.error(f"Error stopping Web Wireshark: {e}")
        finally:
            await manager.close()

    async def _restart_web_wireshark(self, jwt_token: str):
        """Restart Web Wireshark after window was closed.

        Args:
            jwt_token: JWT authentication token

        Raises:
            ControllerError: If restart fails
        """
        # Get capture stream URL from compute
        manager = WebWiresharkManager()
        try:
            log.info(f"Restarting Web Wireshark for link {self.id}")

            result = await manager.restart_wireshark_session(
                project_id=self._project.id,
                link_id=self.id,
                jwt_token=jwt_token
            )

            self._project.emit_notification("link.web_wireshark_started", {
                "link_id": self.id,
                "ws_url": result.get("ws_url", result.get("url"))
            })

            log.info(f"Web Wireshark restarted for link {self.id}: {result.get('ws_url')}")

        except ControllerError:
            raise
        except Exception as e:
            error_msg = f"Error restarting Web Wireshark: {str(e)}"
            log.error(error_msg)
            raise ControllerError(error_msg)
        finally:
            await manager.close()

    def pcap_streaming_url(self):
        """
        Get the PCAP streaming URL on compute

        :returns: URL
        """

        assert self.capture_node
        compute = self.capture_node["node"].compute
        node_type = self.capture_node["node"].node_type
        node_id = self.capture_node["node"].id
        adapter_number = self.capture_node["adapter_number"]
        port_number = self.capture_node["port_number"]
        url = (
            "/projects/{project_id}/{node_type}/nodes/{node_id}/adapters/{adapter_number}/"
            "ports/{port_number}/capture/stream".format(
                project_id=self.project.id,
                node_type=node_type,
                node_id=node_id,
                adapter_number=adapter_number,
                port_number=port_number,
            )
        )

        return compute._getUrl(url)

    async def node_updated(self, node):
        """
        Called when a node member of the link is updated
        """
        raise NotImplementedError

    def default_capture_file_name(self):
        """
        :returns: File name for a capture on this link
        """

        capture_file_name = "{}_{}-{}_to_{}_{}-{}".format(
            self._nodes[0]["node"].name,
            self._nodes[0]["adapter_number"],
            self._nodes[0]["port_number"],
            self._nodes[1]["node"].name,
            self._nodes[1]["adapter_number"],
            self._nodes[1]["port_number"],
        )
        return re.sub(r"[^0-9A-Za-z_-]", "", capture_file_name) + ".pcap"

    @property
    def id(self):
        return self._id

    @property
    def nodes(self):
        return [node["node"] for node in self._nodes]

    @property
    def capturing(self):
        return self._capturing

    @property
    def capture_file_path(self):
        """
        Get the path of the capture
        """

        if self._capture_file_name:
            return os.path.join(self._project.captures_directory, self._capture_file_name)
        else:
            return None

    @property
    def capture_compute_id(self):
        """
        Get the capture compute ID.
        """

        if self._capture_node:
            return self.capture_node["node"].compute.id
        else:
            return None

    def available_filters(self):
        """
        Return the list of filters compatible with this link

        :returns: Array of filters
        """
        filter_node = self._get_filter_node()
        if filter_node:
            return FILTERS
        return []

    def _get_filter_node(self):
        """
        Return the node where the filter will run

        :returns: None if no node support filtering else the node
        """
        for node in self._nodes:
            if node["node"].node_type in (
                "vpcs",
                "vmware",
                "dynamips",
                "qemu",
                "iou",
                "cloud",
                "nat",
                "virtualbox",
                "docker",
            ):
                return node["node"]
        return None

    def __eq__(self, other):
        if not isinstance(other, Link):
            return False
        return self.id == other.id

    def __hash__(self):
        return hash(self._id)

    def asdict(self, topology_dump=False):
        """
        :param topology_dump: Filter to keep only properties require for saving on disk
        """
        res = []
        for side in self._nodes:
            res.append(
                {
                    "node_id": side["node"].id,
                    "adapter_number": side["adapter_number"],
                    "port_number": side["port_number"],
                    "label": side["label"],
                }
            )
        if topology_dump:
            return {
                "nodes": res,
                "link_id": self._id,
                "filters": self._filters,
                "link_style": self._link_style,
                "suspend": self._suspended,
                "show_filters_icon": getattr(self, '_show_filters_icon', True),
            }
        return {
            "nodes": res,
            "link_id": self._id,
            "project_id": self._project.id,
            "capturing": self._capturing,
            "capture_file_name": self._capture_file_name,
            "capture_file_path": self.capture_file_path,
            "capture_compute_id": self.capture_compute_id,
            "link_type": self._link_type,
            "filters": self._filters,
            "suspend": self._suspended,
            "link_style": self._link_style,
            "wireshark": self._wireshark,
            "show_filters_icon": getattr(self, '_show_filters_icon', True),
        }