Netstat

Complexity

Total Complexity

12

Size/Duplication

Total Lines	107
Duplicated Lines	55.14 %

Importance

Changes	1
Bugs	0	Features	0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	__init__()	0	12	1
D	check()	59	59	10
A	poll()	0	6	1

# -*- coding: utf-8 -*-

__author__ = 'Kenny Freeman'
__email__ = 'kenny.freeman@gmail.com'
__license__ = "ISCL"
__docformat__ = 'reStructuredText'

import time
from collections import deque

import plumd
import plumd.plugins
from plumd.calc import Differential
from plumd.util import get_file_list, Filter


class Netstat(plumd.plugins.Reader):
    """Plugin to measure various kernel metrics from /proc."""
    defaults = {
        'poll.interval': 10,
        'proc_path': '/proc',
        'proc_netstat_record': {
            'TcpExt': ["SyncookiesSent", "SyncookiesRecv",
                "SyncookiesFailed","OutOfWindowIcmps","TW","TWRecycled",
                "TWKilled","DelayedACKLost", "ListenOverflows",
                "ListenDrops","TCPLostRetransmit","TCPRenoFailures",
                "TCPSackFailures","TCPLossFailures","TCPFastRetrans",
                "TCPForwardRetrans","TCPSlowStartRetrans","TCPTimeouts",
                "TCPLossProbes","TCPLossProbeRecovery", "TCPAbortOnData",
                "TCPAbortOnClose", "TCPAbortOnMemory","TCPAbortOnTimeout",
                "TCPAbortOnLinger","TCPAbortFailed", "TCPMemoryPressures",
                "TCPBacklogDrop", "TCPMinTTLDrop","TCPTimeWaitOverflow",
                "TCPFastOpenActive","TCPFastOpenActiveFail",
                "TCPFastOpenPassive","TCPFastOpenPassiveFail",
                "TCPFastOpenListenOverflow","TCPSynRetrans"],
            'IpExt': ["InNoRoutes", "InTruncatedPkts", "InBcastPkts",
                "OutBcastPkts", "InOctets", "OutOctets", "InBcastOctets",
                "OutBcastOctets","InCsumErrors","InNoECTPkts","InECT1Pkts",
                "InECT0Pkts","InCEPkts"]
        }
    }

    def __init__(self, log, config):
        """Plugin to measure various kernel metrics from /proc.

        :param log: A logger
        :type log: logging.RootLogger
        :param config: a plumd.config.Conf configuration helper instance.
        :type config: plumd.config.Conf
        """
        super(Netstat, self).__init__(log, config)
        self.config.defaults(Netstat.defaults)
        self.calc = Differential()
        self.proc_file = "{0}/net/netstat".format(config.get('proc_path'))


    def poll(self):
        """Poll for kernel metrics under proc file net/netstat.

        :rtype: ResultSet
        """
        return plumd.ResultSet(self.check())


    def check(self):

This code seems to be duplicated in your project.

        """Return detailed network statitistics proc file net/netstat.

        Note: add entries to the configuration value 'proc_netstat_record'
        to record them. See /proc/net/netstat for a list of available metrics.

        Note: ECT1Pkts and ECT0Pkts relate to ECT congestion notifications.

        :rtype: plumd.Result
        """
        # what metrics do we want to record?
        record = self.config.get('proc_netstat_record')
        result = plumd.Result("netstat")

        # read the proc file
        dat = None
        with open(self.proc_file, 'r') as f:
            try:
                dat = f.read().strip().split("\n")
            except Exception as e:
                self.log.error("failed to read {0}".format(self.proc_file))
                return result

        # timestamp for Differential calculations
        ts = time.time()

        # should always have a header row and value row, ie. divisible by 2
        if len(dat) % 2 != 0:
            self.log.error("netstat: cannot parse {0}".format(self.proc_file))
            return result

        # split values into lists
        dlist = deque()
        for entry in dat:
            dlist.append(entry.split())

        # put lists into key: value dict
        metrics = {}
        while dlist:
            headers = dlist.popleft()
            values = dlist.popleft()
            # { 'IpExt': {'InNoRoutes': 0, ...} } - [:-1] on IpExt: removes :
            metrics[headers[0][:-1]] = dict(zip(headers, values))

        # now, we have a list of items to record, just need to record them
        for ext, mnames in record.items():
            if ext not in metrics:
                self.log.warn("netstat: unknown extension: {0}".format(ext))
                continue
            values = metrics[ext]
            for mname in mnames:
                if mname in values:
                    mstr = "{0}.{1}".format(ext, mname)
                    mval = self.calc.per_second(mstr, int(values[mname]), ts)
                    result.add(plumd.Int(mstr, mval))
                else:
                    self.log.warn("netstat: unknown metric {0}".format(mname))

        return [result]