build.tracing.trace_manager.TraceManager.queue_probe_packet() - Code Metrics - Inspection of "Create unit test" - kytos-ng/sdntrace - Measure and Improve Code Quality continuously with Scrutinizer

Test Failed

Pull Request — master (#20)

by Rogerio

created 2022-07-18 20:36 UTC

TraceManager.queue_probe_packet() A

↳ Parent: build.tracing.trace_manager

Complexity

Conditions

Size

Total Lines	21
Code Lines	8

Duplication

Lines	0
Ratio	0 %

Code Coverage

Tests	1
CRAP Score	1.6699

Importance

Changes

Metric	Value
cc	1
eloc	8
nop	5
dl	0
loc	21
ccs	1
cts	8
cp	0.125
crap	1.6699
rs	10
c	0
b	0
f	0

"""
    Trace Manager Class
"""


import time
import dill
from _thread import start_new_thread as new_thread

from kytos.core import log
from napps.amlight.sdntrace import settings
from napps.amlight.sdntrace.shared.switches import Switches
from napps.amlight.sdntrace.shared.colors import Colors
from napps.amlight.sdntrace.tracing.tracer import TracePath
from napps.amlight.sdntrace.tracing.trace_pkt import process_packet
from napps.amlight.sdntrace.tracing.trace_entries import TraceEntries


class TraceManager(object):
    """
        The TraceManager class is the class responsible to
        manage all trace requests.
    """

    def __init__(self, controller):
        """Initialization of the TraceManager class
        Args:
             controller = Kytos.core.controller object
        """
        # Controller
        self.controller = controller

        # Trace ID used to distinguish each trace
        self._id = 30000

        # Trace queues
        self._request_queue = dict()
        self._results_queue = dict()
        self._running_traces = dict()

        # Counters
        self._total_traces_requested = 0

        # PacketIn queue with Probes
        self.trace_pkt_in = []

        self._is_tracing_running = True

        # Thread to start traces
        new_thread(self._run_traces, (settings.TRACE_INTERVAL,))

    def stop_traces(self):
        self._is_tracing_running = False

    def is_tracing_running(self):
        return self._is_tracing_running

    def _run_traces(self, trace_interval):
        """ Thread that will keep reading the self._request_queue
        queue looking for new trace requests to run.

        Args:
            trace_interval = sleeping time
        """
        while self.is_tracing_running():
            if self.number_pending_requests() > 0:
                try:
                    new_request_ids = []
                    for req_id in self._request_queue:
                        if not self.limit_traces_reached():
                            entries = self._request_queue[req_id]
                            self._running_traces[req_id] = entries
                            new_thread(self._spawn_trace, (req_id, entries,))
                            new_request_ids.append(req_id)
                        else:
                            break
                    # After starting traces for new requests,
                    # remove them from self._request_queue
                    for rid in new_request_ids:
                        del self._request_queue[rid]
                except Exception as error:  # pylint: disable=broad-except
                    log.error("Trace Error: %s" % error)
            time.sleep(trace_interval)

    def _spawn_trace(self, trace_id, trace_entries):
        """ Once a request is found by the run_traces method,
        instantiate a TracePath class and run the tracepath

        Args:
            trace_id: trace request id
            trace_entries: TraceEntries class
        """
        log.info("Creating thread to trace request id %s..." % trace_id)
        tracer = TracePath(self, trace_id, trace_entries)
        tracer.tracepath()

        del self._running_traces[trace_id]

    def add_result(self, trace_id, result):
        """Used to save trace results to self._results_queue

        Args:
            trace_id: trace ID
            result: trace result generated using tracer
        """
        self._results_queue[trace_id] = result

    def avoid_duplicated_request(self, entries):
        """Verify if any of the requested queries has the same entries.
        If so, ignore it

        Args:
            entries: entries provided by user via REST.
        Return:
            True: if exists a similar request
            False: otherwise
        """
        for request in self._request_queue:
            if entries == self._request_queue[request]:
                return True
        return False

    @staticmethod
    def is_entry_valid(entries):
        """ This method validates all params provided, including
        if the switch/dpid requested exists.

        Args:
            entries: dictionary with user request
        Returns:
            TraceEntries class
            Error msg
        """
        try:
            trace_entries = TraceEntries()
            trace_entries.load_entries(entries)
        except ValueError as msg:
            return str(msg)

        init_switch = Switches().get_switch(trace_entries.dpid)
        if isinstance(init_switch, bool):
            return "Unknown Switch"
        color = Colors().get_switch_color(init_switch.dpid)

        if len(color) == 0:
            return "Switch not Colored"

        # TODO: get Coloring API to confirm color_field

        return trace_entries

    def get_id(self):
        """ID generator for each trace. Useful in case
        of parallel requests

        Returns:
            integer to be the new request/trace id
        """
        self._id += 1
        return self._id

    def get_result(self, trace_id):
        """Used by external apps to get a trace result using the trace ID

        Returns:
            result from self._results_queue
            msg depending of the status (unknown, pending, or active)
        """
        trace_id = int(trace_id)
        try:
            return self._results_queue[trace_id]
        except (ValueError, KeyError):
            if trace_id in self._running_traces:
                return {'msg': 'trace in process'}
            elif trace_id in self._request_queue:
                return {'msg': 'trace pending'}
            return {'msg': 'unknown trace id'}

    def get_results(self):
        """Used by external apps to get all trace results. Useful
        to see all requests and results

        Returns:
            list of results
        """
        return self._results_queue

    def limit_traces_reached(self):
        """ Control the number of active traces running in parallel. Protects the
        switches and avoid DoS.

        Returns:
            True: if the number of traces running is equal/more
                than settings.PARALLEL_TRACES
            False: if it is not.
        """
        if len(self._running_traces) >= settings.PARALLEL_TRACES:
            return True
        return False

    def new_trace(self, trace_entries):
        """Receives external requests for traces.

        Args:
            trace_entries: TraceEntries Class
        Returns:
            int with the request/trace id
        """

        trace_id = self.get_id()

        # Add to request_queue
        self._request_queue[trace_id] = trace_entries

        # Statistics
        self._total_traces_requested += 1

        return trace_id

    def number_pending_requests(self):
        """Used to check if there are entries to be traced

        Returns:
            length of self._request_queue
        """
        return len(self._request_queue)

    def queue_probe_packet(self, event, ethernet, in_port, switch):
        """Used by sdntrace.packet_in_handler. Only tracing probes
        get to this point. Adds the PacketIn msg received to the
        trace_pkt_in queue.

        Args:
            event: PacketIn msg
            ethernet: ethernet frame
            in_port: in_port
            switch: kytos.core.switch.Switch() class
        """
        pkt_in = dict()

        pkt_in["dpid"] = switch.dpid
        pkt_in["in_port"] = in_port
        pkt_in["msg"] = dill.loads(process_packet(ethernet))
        pkt_in["ethernet"] = ethernet
        pkt_in["event"] = event

        # This queue stores all PacketIn message received
        self.trace_pkt_in.append(pkt_in)

    # REST calls

    def rest_new_trace(self, entries):
        """Used for the REST PUT call

        Args:
            entries: user provided parameters to trace
        Returns:
            Trace_ID in JSON format
            Error msg if entries has invalid data
        """
        result = dict()
        trace_entries = self.is_entry_valid(entries)
        if not isinstance(trace_entries, TraceEntries):
            result['result'] = {'error': trace_entries}
            return result

        if self.avoid_duplicated_request(entries):
            result['result'] = {'error': "Duplicated Trace Request ignored"}
            return result

        trace_id = self.new_trace(trace_entries)
        result['result'] = {'trace_id': trace_id}
        return result

    def rest_get_result(self, trace_id):
        """Used for the REST GET call

        Returns:
            get_result in JSON format
        """
        return self.get_result(trace_id)

    def rest_list_results(self):
        """Used for the REST GET call

        Returns:
            get_results in JSON format
        """
        return self.get_results()

    def rest_list_stats(self):
        """ Used to export some info about the TraceManager.
        Total number of requests, number of active traces, number of
        pending traces, list of traces pending
        Returns:
                Total number of requests
                number of active traces
                number of pending traces
                list of traces pending
        """
        stats = dict()
        stats['number_of_requests'] = self._total_traces_requested
        stats['number_of_running_traces'] = len(self._running_traces)
        stats['number_of_pending_traces'] = len(self._request_queue)
        stats['list_of_pending_traces'] = self._results_queue

        return stats


1		"""
2		Trace Manager Class
3		"""
4
5
6	1	import time
7	1	import dill
8	1	from _thread import start_new_thread as new_thread
9
10	1	from kytos.core import log
11	1	from napps.amlight.sdntrace import settings
12	1	from napps.amlight.sdntrace.shared.switches import Switches
13	1	from napps.amlight.sdntrace.shared.colors import Colors
14	1	from napps.amlight.sdntrace.tracing.tracer import TracePath
15	1	from napps.amlight.sdntrace.tracing.trace_pkt import process_packet
16	1	from napps.amlight.sdntrace.tracing.trace_entries import TraceEntries
17
18
19	1	class TraceManager(object):
20		"""
21		The TraceManager class is the class responsible to
22		manage all trace requests.
23		"""
24
25	1	def __init__(self, controller):
26		"""Initialization of the TraceManager class
27		Args:
28		controller = Kytos.core.controller object
29		"""
30		# Controller
31	1	self.controller = controller
32
33		# Trace ID used to distinguish each trace
34	1	self._id = 30000
35
36		# Trace queues
37	1	self._request_queue = dict()
38	1	self._results_queue = dict()
39	1	self._running_traces = dict()
40
41		# Counters
42	1	self._total_traces_requested = 0
43
44		# PacketIn queue with Probes
45	1	self.trace_pkt_in = []
46
47		self._is_tracing_running = True
48	1
49		# Thread to start traces
50	1	new_thread(self._run_traces, (settings.TRACE_INTERVAL,))
51
52		def stop_traces(self):
53		self._is_tracing_running = False
54
55		def is_tracing_running(self):
56		return self._is_tracing_running
57
58		def _run_traces(self, trace_interval):
59		""" Thread that will keep reading the self._request_queue
60		queue looking for new trace requests to run.
61
62		Args:
63		trace_interval = sleeping time
64		"""
65		while self.is_tracing_running():
66		if self.number_pending_requests() > 0:
67		try:
68		new_request_ids = []
69		for req_id in self._request_queue:
70		if not self.limit_traces_reached():
71		entries = self._request_queue[req_id]
72		self._running_traces[req_id] = entries
73		new_thread(self._spawn_trace, (req_id, entries,))
74		new_request_ids.append(req_id)
75		else:
76		break
77	1	# After starting traces for new requests,
78		# remove them from self._request_queue
79		for rid in new_request_ids:
80		del self._request_queue[rid]
81		except Exception as error: # pylint: disable=broad-except
82		log.error("Trace Error: %s" % error)
83		time.sleep(trace_interval)
84
85		def _spawn_trace(self, trace_id, trace_entries):
86		""" Once a request is found by the run_traces method,
87		instantiate a TracePath class and run the tracepath
88
89		Args:
90	1	trace_id: trace request id
91		trace_entries: TraceEntries class
92		"""
93		log.info("Creating thread to trace request id %s..." % trace_id)
94		tracer = TracePath(self, trace_id, trace_entries)
95		tracer.tracepath()
96
97		del self._running_traces[trace_id]
98
99	1	def add_result(self, trace_id, result):
100		"""Used to save trace results to self._results_queue
101
102		Args:
103		trace_id: trace ID
104		result: trace result generated using tracer
105		"""
106		self._results_queue[trace_id] = result
107
108		def avoid_duplicated_request(self, entries):
109		"""Verify if any of the requested queries has the same entries.
110		If so, ignore it
111
112		Args:
113		entries: entries provided by user via REST.
114	1	Return:
115	1	True: if exists a similar request
116		False: otherwise
117		"""
118		for request in self._request_queue:
119		if entries == self._request_queue[request]:
120		return True
121		return False
122
123		@staticmethod
124		def is_entry_valid(entries):
125		""" This method validates all params provided, including
126		if the switch/dpid requested exists.
127
128		Args:
129		entries: dictionary with user request
130		Returns:
131		TraceEntries class
132		Error msg
133		"""
134		try:
135		trace_entries = TraceEntries()
136		trace_entries.load_entries(entries)
137		except ValueError as msg:
138		return str(msg)
139
140		init_switch = Switches().get_switch(trace_entries.dpid)
141		if isinstance(init_switch, bool):
142	1	return "Unknown Switch"
143		color = Colors().get_switch_color(init_switch.dpid)
144
145		if len(color) == 0:
146		return "Switch not Colored"
147
148		# TODO: get Coloring API to confirm color_field
149
150		return trace_entries
151
152	1	def get_id(self):
153		"""ID generator for each trace. Useful in case
154		of parallel requests
155
156		Returns:
157		integer to be the new request/trace id
158		"""
159		self._id += 1
160		return self._id
161
162		def get_result(self, trace_id):
163		"""Used by external apps to get a trace result using the trace ID
164
165		Returns:
166		result from self._results_queue
167		msg depending of the status (unknown, pending, or active)
168		"""
169	1	trace_id = int(trace_id)
170		try:
171		return self._results_queue[trace_id]
172		except (ValueError, KeyError):
173		if trace_id in self._running_traces:
174		return {'msg': 'trace in process'}
175		elif trace_id in self._request_queue:
176		return {'msg': 'trace pending'}
177		return {'msg': 'unknown trace id'}
178	1
179		def get_results(self):
180		"""Used by external apps to get all trace results. Useful
181		to see all requests and results
182
183		Returns:
184		list of results
185		"""
186		return self._results_queue
187
188		def limit_traces_reached(self):
189		""" Control the number of active traces running in parallel. Protects the
190		switches and avoid DoS.
191	1
192		Returns:
193		True: if the number of traces running is equal/more
194		than settings.PARALLEL_TRACES
195		False: if it is not.
196		"""
197		if len(self._running_traces) >= settings.PARALLEL_TRACES:
198		return True
199		return False
200
201		def new_trace(self, trace_entries):
202		"""Receives external requests for traces.
203
204		Args:
205		trace_entries: TraceEntries Class
206		Returns:
207		int with the request/trace id
208		"""
209
210	1	trace_id = self.get_id()
211
212		# Add to request_queue
213		self._request_queue[trace_id] = trace_entries
214
215		# Statistics
216		self._total_traces_requested += 1
217
218	1	return trace_id
219
220		def number_pending_requests(self):
221		"""Used to check if there are entries to be traced
222
223		Returns:
224		length of self._request_queue
225		"""
226		return len(self._request_queue)
227
228		def queue_probe_packet(self, event, ethernet, in_port, switch):
229		"""Used by sdntrace.packet_in_handler. Only tracing probes
230		get to this point. Adds the PacketIn msg received to the
231		trace_pkt_in queue.
232
233		Args:
234		event: PacketIn msg
235		ethernet: ethernet frame
236		in_port: in_port
237		switch: kytos.core.switch.Switch() class
238		"""
239		pkt_in = dict()
240
241		pkt_in["dpid"] = switch.dpid
242	1	pkt_in["in_port"] = in_port
243		pkt_in["msg"] = dill.loads(process_packet(ethernet))
244		pkt_in["ethernet"] = ethernet
245		pkt_in["event"] = event
246
247		# This queue stores all PacketIn message received
248		self.trace_pkt_in.append(pkt_in)
249
250		# REST calls
251
252		def rest_new_trace(self, entries):
253		"""Used for the REST PUT call
254
255		Args:
256		entries: user provided parameters to trace
257		Returns:
258		Trace_ID in JSON format
259		Error msg if entries has invalid data
260		"""
261		result = dict()
262		trace_entries = self.is_entry_valid(entries)
263		if not isinstance(trace_entries, TraceEntries):
264		result['result'] = {'error': trace_entries}
265	1	return result
266
267		if self.avoid_duplicated_request(entries):
268		result['result'] = {'error': "Duplicated Trace Request ignored"}
269		return result
270
271		trace_id = self.new_trace(trace_entries)
272		result['result'] = {'trace_id': trace_id}
273	1	return result
274
275		def rest_get_result(self, trace_id):
276		"""Used for the REST GET call
277
278		Returns:
279		get_result in JSON format
280		"""
281	1	return self.get_result(trace_id)
282
283		def rest_list_results(self):
284		"""Used for the REST GET call
285
286		Returns:
287		get_results in JSON format
288		"""
289		return self.get_results()
290
291		def rest_list_stats(self):
292		""" Used to export some info about the TraceManager.
293		Total number of requests, number of active traces, number of
294		pending traces, list of traces pending
295		Returns:
296		Total number of requests
297		number of active traces
298		number of pending traces
299		list of traces pending
300		"""
301		stats = dict()
302		stats['number_of_requests'] = self._total_traces_requested
303		stats['number_of_running_traces'] = len(self._running_traces)
304		stats['number_of_pending_traces'] = len(self._request_queue)
305		stats['list_of_pending_traces'] = self._results_queue
306
307		return stats
308

kytos-ng / sdntrace

Pull Request — master (#20)

TraceManager.queue_probe_packet() A

Complexity

Size

Duplication

Code Coverage

Importance

Duplication Side-by-Side

Filter issues like