1
0
mirror of https://github.com/GNS3/gns3-server synced 2024-12-04 22:28:12 +00:00
gns3-server/gns3server/compute/dynamips/nios/nio.py

353 lines
9.9 KiB
Python
Raw Normal View History

2013-12-22 00:42:33 +00:00
#
# Copyright (C) 2013 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/>.
"""
Base interface for Dynamips Network Input/Output (NIO) module ("nio").
http://github.com/GNS3/dynamips/blob/master/README.hypervisor#L451
"""
2015-02-10 01:24:13 +00:00
import asyncio
2013-12-22 00:42:33 +00:00
from ..dynamips_error import DynamipsError
import logging
2021-04-13 09:16:50 +00:00
log = logging.getLogger(__name__)
2013-12-22 00:42:33 +00:00
2015-02-10 01:24:13 +00:00
class NIO:
2015-02-13 13:43:28 +00:00
2013-12-22 00:42:33 +00:00
"""
Base NIO class
2014-02-28 04:50:46 +00:00
:param hypervisor: Dynamips hypervisor instance
2013-12-22 00:42:33 +00:00
"""
2015-02-10 01:24:13 +00:00
def __init__(self, name, hypervisor):
2013-12-22 00:42:33 +00:00
self._hypervisor = hypervisor
2015-02-13 22:11:14 +00:00
self._name = name
self._filters = {}
self._suspended = False
self._capturing = False
self._pcap_output_file = ""
self._pcap_data_link_type = ""
2013-12-22 00:42:33 +00:00
self._bandwidth = None # no bandwidth constraint by default
self._input_filter = None # no input filter applied by default
self._output_filter = None # no output filter applied by default
self._input_filter_options = None # no input filter options by default
self._output_filter_options = None # no output filter options by default
self._dynamips_direction = {"in": 0, "out": 1, "both": 2}
async def list(self):
2013-12-22 00:42:33 +00:00
"""
Returns all NIOs.
:returns: NIO list
"""
nio_list = await self._hypervisor.send("nio list")
2015-02-10 01:24:13 +00:00
return nio_list
2013-12-22 00:42:33 +00:00
async def delete(self):
2013-12-22 00:42:33 +00:00
"""
Deletes this NIO.
"""
if self._input_filter or self._output_filter:
await self.unbind_filter("both")
self._capturing = False
2021-04-13 09:07:58 +00:00
await self._hypervisor.send(f"nio delete {self._name}")
log.info(f"NIO {self._name} has been deleted")
2013-12-22 00:42:33 +00:00
async def rename(self, new_name):
2013-12-22 00:42:33 +00:00
"""
Renames this NIO
:param new_name: new NIO name
"""
2021-04-13 09:07:58 +00:00
await self._hypervisor.send(f"nio rename {self._name} {new_name}")
2021-04-13 09:07:58 +00:00
log.info(f"NIO {self._name} renamed to {new_name}")
2013-12-22 00:42:33 +00:00
self._name = new_name
async def debug(self, debug):
2013-12-22 00:42:33 +00:00
"""
Enables/Disables debugging for this NIO.
:param debug: debug value (0 = disable, enable = 1)
"""
2021-04-13 09:07:58 +00:00
await self._hypervisor.send(f"nio set_debug {self._name} {debug}")
2013-12-22 00:42:33 +00:00
async def start_packet_capture(self, pcap_output_file, pcap_data_link_type="DLT_EN10MB"):
"""
Starts a packet capture.
:param pcap_output_file: PCAP destination file for the capture
:param pcap_data_link_type: PCAP data link type (DLT_*), default is DLT_EN10MB
"""
await self.bind_filter("both", "capture")
2021-04-13 09:07:58 +00:00
await self.setup_filter("both", f'{pcap_data_link_type} "{pcap_output_file}"')
self._capturing = True
self._pcap_output_file = pcap_output_file
self._pcap_data_link_type = pcap_data_link_type
async def stop_packet_capture(self):
"""
Stops a packet capture.
"""
await self.unbind_filter("both")
self._capturing = False
self._pcap_output_file = ""
self._pcap_data_link_type = ""
async def bind_filter(self, direction, filter_name):
2013-12-22 00:42:33 +00:00
"""
Adds a packet filter to this NIO.
Filter "freq_drop" drops packets.
Filter "capture" captures packets.
:param direction: "in", "out" or "both"
:param filter_name: name of the filter to apply
"""
if direction not in self._dynamips_direction:
2021-04-13 09:07:58 +00:00
raise DynamipsError(f"Unknown direction {direction} to bind filter {filter_name}:")
2013-12-22 00:42:33 +00:00
dynamips_direction = self._dynamips_direction[direction]
2021-04-13 09:16:50 +00:00
await self._hypervisor.send(
"nio bind_filter {name} {direction} {filter}".format(
name=self._name, direction=dynamips_direction, filter=filter_name
)
)
2013-12-22 00:42:33 +00:00
if direction == "in":
self._input_filter = filter_name
elif direction == "out":
self._output_filter = filter_name
elif direction == "both":
self._input_filter = filter_name
self._output_filter = filter_name
async def unbind_filter(self, direction):
2013-12-22 00:42:33 +00:00
"""
Removes packet filter for this NIO.
:param direction: "in", "out" or "both"
"""
if direction not in self._dynamips_direction:
2021-04-13 09:07:58 +00:00
raise DynamipsError(f"Unknown direction {direction} to unbind filter:")
2013-12-22 00:42:33 +00:00
dynamips_direction = self._dynamips_direction[direction]
2021-04-13 09:16:50 +00:00
await self._hypervisor.send(
"nio unbind_filter {name} {direction}".format(name=self._name, direction=dynamips_direction)
)
2013-12-22 00:42:33 +00:00
if direction == "in":
self._input_filter = None
elif direction == "out":
self._output_filter = None
elif direction == "both":
self._input_filter = None
self._output_filter = None
self._capturing = False
2013-12-22 00:42:33 +00:00
async def setup_filter(self, direction, options):
2013-12-22 00:42:33 +00:00
"""
Setups a packet filter bound with this NIO.
2013-12-22 00:42:33 +00:00
Filter "freq_drop" has 1 argument "<frequency>". It will drop
everything with a -1 frequency, drop every Nth packet with a
positive frequency, or drop nothing.
Filter "capture" has 2 arguments "<link_type_name> <output_file>".
It will capture packets to the target output file. The link type
name is a case-insensitive DLT_ name from the PCAP library
constants with the DLT_ part removed.See http://www.tcpdump.org/linktypes.html
for a list of all available DLT values.
:param direction: "in", "out" or "both"
:param options: options for the packet filter (string)
"""
if direction not in self._dynamips_direction:
2021-04-13 09:07:58 +00:00
raise DynamipsError(f"Unknown direction {direction} to setup filter:")
2013-12-22 00:42:33 +00:00
dynamips_direction = self._dynamips_direction[direction]
2021-04-13 09:16:50 +00:00
await self._hypervisor.send(
"nio setup_filter {name} {direction} {options}".format(
name=self._name, direction=dynamips_direction, options=options
)
)
2013-12-22 00:42:33 +00:00
if direction == "in":
self._input_filter_options = options
elif direction == "out":
self._output_filter_options = options
elif direction == "both":
self._input_filter_options = options
self._output_filter_options = options
@property
def input_filter(self):
"""
Returns the input packet filter for this NIO.
:returns: tuple (filter name, filter options)
"""
2014-05-28 12:26:20 +00:00
return self._input_filter, self._input_filter_options
2013-12-22 00:42:33 +00:00
@property
def output_filter(self):
"""
Returns the output packet filter for this NIO.
:returns: tuple (filter name, filter options)
"""
2014-05-28 12:26:20 +00:00
return self._output_filter, self._output_filter_options
2013-12-22 00:42:33 +00:00
async def get_stats(self):
2013-12-22 00:42:33 +00:00
"""
Gets statistics for this NIO.
:returns: NIO statistics (string with packets in, packets out, bytes in, bytes out)
"""
2021-04-13 09:07:58 +00:00
stats = await self._hypervisor.send(f"nio get_stats {self._name}")
2015-02-10 01:24:13 +00:00
return stats[0]
2013-12-22 00:42:33 +00:00
async def reset_stats(self):
2013-12-22 00:42:33 +00:00
"""
Resets statistics for this NIO.
"""
2021-04-13 09:07:58 +00:00
await self._hypervisor.send(f"nio reset_stats {self._name}")
2013-12-22 00:42:33 +00:00
@property
def bandwidth(self):
"""
Returns the bandwidth constraint for this NIO.
:returns: bandwidth integer value (in Kb/s)
"""
return self._bandwidth
async def set_bandwidth(self, bandwidth):
2015-02-10 01:24:13 +00:00
"""
Sets bandwidth constraint.
:param bandwidth: bandwidth integer value (in Kb/s)
"""
2021-04-13 09:07:58 +00:00
await self._hypervisor.send(f"nio set_bandwidth {self._name} {bandwidth}")
2015-02-10 01:24:13 +00:00
self._bandwidth = bandwidth
@property
def suspend(self):
"""
Returns if this link is suspended or not.
:returns: boolean
"""
return self._suspended
@suspend.setter
def suspend(self, suspended):
"""
Suspend this link.
:param suspended: boolean
"""
self._suspended = suspended
@property
def filters(self):
"""
Returns the list of packet filters for this NIO.
:returns: packet filters (dictionary)
"""
return self._filters
@filters.setter
def filters(self, new_filters):
"""
Set a list of packet filters for this NIO.
:param new_filters: packet filters (dictionary)
"""
assert isinstance(new_filters, dict)
self._filters = new_filters
@property
def capturing(self):
"""
Returns either a capture is configured on this NIO.
:returns: boolean
"""
return self._capturing
@property
def pcap_output_file(self):
"""
Returns the path to the PCAP output file.
:returns: path to the PCAP output file
"""
return self._pcap_output_file
@property
def pcap_data_link_type(self):
"""
Returns the PCAP data link type
:returns: PCAP data link type (DLT_* value)
"""
return self._pcap_data_link_type
2013-12-22 00:42:33 +00:00
def __str__(self):
"""
NIO string representation.
:returns: NIO name
"""
return self._name
@property
def name(self):
"""
Returns the NIO name.
:returns: NIO name
"""
return self._name