Adding ethMon plugin
This commit is contained in:
		
							parent
							
								
									e384cab2a4
								
							
						
					
					
						commit
						c8ae2e0a3d
					
				
					 6 changed files with 606 additions and 0 deletions
				
			
		
							
								
								
									
										208
									
								
								ethMon/LICENSE
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										208
									
								
								ethMon/LICENSE
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,208 @@
 | 
			
		|||
Apache License
 | 
			
		||||
 | 
			
		||||
Version 2.0, January 2004
 | 
			
		||||
 | 
			
		||||
http://www.apache.org/licenses/ TERMS AND CONDITIONS FOR USE, REPRODUCTION,
 | 
			
		||||
AND DISTRIBUTION
 | 
			
		||||
 | 
			
		||||
   1. Definitions.
 | 
			
		||||
 | 
			
		||||
      
 | 
			
		||||
 | 
			
		||||
"License" shall mean the terms and conditions for use, reproduction, and distribution
 | 
			
		||||
as defined by Sections 1 through 9 of this document.
 | 
			
		||||
 | 
			
		||||
      
 | 
			
		||||
 | 
			
		||||
"Licensor" shall mean the copyright owner or entity authorized by the copyright
 | 
			
		||||
owner that is granting the License.
 | 
			
		||||
 | 
			
		||||
      
 | 
			
		||||
 | 
			
		||||
"Legal Entity" shall mean the union of the acting entity and all other entities
 | 
			
		||||
that control, are controlled by, or are under common control with that entity.
 | 
			
		||||
For the purposes of this definition, "control" means (i) the power, direct
 | 
			
		||||
or indirect, to cause the direction or management of such entity, whether
 | 
			
		||||
by contract or otherwise, or (ii) ownership of fifty percent (50%) or more
 | 
			
		||||
of the outstanding shares, or (iii) beneficial ownership of such entity.
 | 
			
		||||
 | 
			
		||||
      
 | 
			
		||||
 | 
			
		||||
"You" (or "Your") shall mean an individual or Legal Entity exercising permissions
 | 
			
		||||
granted by this License.
 | 
			
		||||
 | 
			
		||||
      
 | 
			
		||||
 | 
			
		||||
"Source" form shall mean the preferred form for making modifications, including
 | 
			
		||||
but not limited to software source code, documentation source, and configuration
 | 
			
		||||
files.
 | 
			
		||||
 | 
			
		||||
      
 | 
			
		||||
 | 
			
		||||
"Object" form shall mean any form resulting from mechanical transformation
 | 
			
		||||
or translation of a Source form, including but not limited to compiled object
 | 
			
		||||
code, generated documentation, and conversions to other media types.
 | 
			
		||||
 | 
			
		||||
      
 | 
			
		||||
 | 
			
		||||
"Work" shall mean the work of authorship, whether in Source or Object form,
 | 
			
		||||
made available under the License, as indicated by a copyright notice that
 | 
			
		||||
is included in or attached to the work (an example is provided in the Appendix
 | 
			
		||||
below).
 | 
			
		||||
 | 
			
		||||
      
 | 
			
		||||
 | 
			
		||||
"Derivative Works" shall mean any work, whether in Source or Object form,
 | 
			
		||||
that is based on (or derived from) the Work and for which the editorial revisions,
 | 
			
		||||
annotations, elaborations, or other modifications represent, as a whole, an
 | 
			
		||||
original work of authorship. For the purposes of this License, Derivative
 | 
			
		||||
Works shall not include works that remain separable from, or merely link (or
 | 
			
		||||
bind by name) to the interfaces of, the Work and Derivative Works thereof.
 | 
			
		||||
 | 
			
		||||
      
 | 
			
		||||
 | 
			
		||||
"Contribution" shall mean any work of authorship, including the original version
 | 
			
		||||
of the Work and any modifications or additions to that Work or Derivative
 | 
			
		||||
Works thereof, that is intentionally submitted to Licensor for inclusion in
 | 
			
		||||
the Work by the copyright owner or by an individual or Legal Entity authorized
 | 
			
		||||
to submit on behalf of the copyright owner. For the purposes of this definition,
 | 
			
		||||
"submitted" means any form of electronic, verbal, or written communication
 | 
			
		||||
sent to the Licensor or its representatives, including but not limited to
 | 
			
		||||
communication on electronic mailing lists, source code control systems, and
 | 
			
		||||
issue tracking systems that are managed by, or on behalf of, the Licensor
 | 
			
		||||
for the purpose of discussing and improving the Work, but excluding communication
 | 
			
		||||
that is conspicuously marked or otherwise designated in writing by the copyright
 | 
			
		||||
owner as "Not a Contribution."
 | 
			
		||||
 | 
			
		||||
      
 | 
			
		||||
 | 
			
		||||
"Contributor" shall mean Licensor and any individual or Legal Entity on behalf
 | 
			
		||||
of whom a Contribution has been received by Licensor and subsequently incorporated
 | 
			
		||||
within the Work.
 | 
			
		||||
 | 
			
		||||
2. Grant of Copyright License. Subject to the terms and conditions of this
 | 
			
		||||
License, each Contributor hereby grants to You a perpetual, worldwide, non-exclusive,
 | 
			
		||||
no-charge, royalty-free, irrevocable copyright license to reproduce, prepare
 | 
			
		||||
Derivative Works of, publicly display, publicly perform, sublicense, and distribute
 | 
			
		||||
the Work and such Derivative Works in Source or Object form.
 | 
			
		||||
 | 
			
		||||
3. Grant of Patent License. Subject to the terms and conditions of this License,
 | 
			
		||||
each Contributor hereby grants to You a perpetual, worldwide, non-exclusive,
 | 
			
		||||
no-charge, royalty-free, irrevocable (except as stated in this section) patent
 | 
			
		||||
license to make, have made, use, offer to sell, sell, import, and otherwise
 | 
			
		||||
transfer the Work, where such license applies only to those patent claims
 | 
			
		||||
licensable by such Contributor that are necessarily infringed by their Contribution(s)
 | 
			
		||||
alone or by combination of their Contribution(s) with the Work to which such
 | 
			
		||||
Contribution(s) was submitted. If You institute patent litigation against
 | 
			
		||||
any entity (including a cross-claim or counterclaim in a lawsuit) alleging
 | 
			
		||||
that the Work or a Contribution incorporated within the Work constitutes direct
 | 
			
		||||
or contributory patent infringement, then any patent licenses granted to You
 | 
			
		||||
under this License for that Work shall terminate as of the date such litigation
 | 
			
		||||
is filed.
 | 
			
		||||
 | 
			
		||||
4. Redistribution. You may reproduce and distribute copies of the Work or
 | 
			
		||||
Derivative Works thereof in any medium, with or without modifications, and
 | 
			
		||||
in Source or Object form, provided that You meet the following conditions:
 | 
			
		||||
 | 
			
		||||
(a) You must give any other recipients of the Work or Derivative Works a copy
 | 
			
		||||
of this License; and
 | 
			
		||||
 | 
			
		||||
(b) You must cause any modified files to carry prominent notices stating that
 | 
			
		||||
You changed the files; and
 | 
			
		||||
 | 
			
		||||
(c) You must retain, in the Source form of any Derivative Works that You distribute,
 | 
			
		||||
all copyright, patent, trademark, and attribution notices from the Source
 | 
			
		||||
form of the Work, excluding those notices that do not pertain to any part
 | 
			
		||||
of the Derivative Works; and
 | 
			
		||||
 | 
			
		||||
(d) If the Work includes a "NOTICE" text file as part of its distribution,
 | 
			
		||||
then any Derivative Works that You distribute must include a readable copy
 | 
			
		||||
of the attribution notices contained within such NOTICE file, excluding those
 | 
			
		||||
notices that do not pertain to any part of the Derivative Works, in at least
 | 
			
		||||
one of the following places: within a NOTICE text file distributed as part
 | 
			
		||||
of the Derivative Works; within the Source form or documentation, if provided
 | 
			
		||||
along with the Derivative Works; or, within a display generated by the Derivative
 | 
			
		||||
Works, if and wherever such third-party notices normally appear. The contents
 | 
			
		||||
of the NOTICE file are for informational purposes only and do not modify the
 | 
			
		||||
License. You may add Your own attribution notices within Derivative Works
 | 
			
		||||
that You distribute, alongside or as an addendum to the NOTICE text from the
 | 
			
		||||
Work, provided that such additional attribution notices cannot be construed
 | 
			
		||||
as modifying the License.
 | 
			
		||||
 | 
			
		||||
You may add Your own copyright statement to Your modifications and may provide
 | 
			
		||||
additional or different license terms and conditions for use, reproduction,
 | 
			
		||||
or distribution of Your modifications, or for any such Derivative Works as
 | 
			
		||||
a whole, provided Your use, reproduction, and distribution of the Work otherwise
 | 
			
		||||
complies with the conditions stated in this License.
 | 
			
		||||
 | 
			
		||||
5. Submission of Contributions. Unless You explicitly state otherwise, any
 | 
			
		||||
Contribution intentionally submitted for inclusion in the Work by You to the
 | 
			
		||||
Licensor shall be under the terms and conditions of this License, without
 | 
			
		||||
any additional terms or conditions. Notwithstanding the above, nothing herein
 | 
			
		||||
shall supersede or modify the terms of any separate license agreement you
 | 
			
		||||
may have executed with Licensor regarding such Contributions.
 | 
			
		||||
 | 
			
		||||
6. Trademarks. This License does not grant permission to use the trade names,
 | 
			
		||||
trademarks, service marks, or product names of the Licensor, except as required
 | 
			
		||||
for reasonable and customary use in describing the origin of the Work and
 | 
			
		||||
reproducing the content of the NOTICE file.
 | 
			
		||||
 | 
			
		||||
7. Disclaimer of Warranty. Unless required by applicable law or agreed to
 | 
			
		||||
in writing, Licensor provides the Work (and each Contributor provides its
 | 
			
		||||
Contributions) on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 | 
			
		||||
KIND, either express or implied, including, without limitation, any warranties
 | 
			
		||||
or conditions of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR
 | 
			
		||||
A PARTICULAR PURPOSE. You are solely responsible for determining the appropriateness
 | 
			
		||||
of using or redistributing the Work and assume any risks associated with Your
 | 
			
		||||
exercise of permissions under this License.
 | 
			
		||||
 | 
			
		||||
8. Limitation of Liability. In no event and under no legal theory, whether
 | 
			
		||||
in tort (including negligence), contract, or otherwise, unless required by
 | 
			
		||||
applicable law (such as deliberate and grossly negligent acts) or agreed to
 | 
			
		||||
in writing, shall any Contributor be liable to You for damages, including
 | 
			
		||||
any direct, indirect, special, incidental, or consequential damages of any
 | 
			
		||||
character arising as a result of this License or out of the use or inability
 | 
			
		||||
to use the Work (including but not limited to damages for loss of goodwill,
 | 
			
		||||
work stoppage, computer failure or malfunction, or any and all other commercial
 | 
			
		||||
damages or losses), even if such Contributor has been advised of the possibility
 | 
			
		||||
of such damages.
 | 
			
		||||
 | 
			
		||||
9. Accepting Warranty or Additional Liability. While redistributing the Work
 | 
			
		||||
or Derivative Works thereof, You may choose to offer, and charge a fee for,
 | 
			
		||||
acceptance of support, warranty, indemnity, or other liability obligations
 | 
			
		||||
and/or rights consistent with this License. However, in accepting such obligations,
 | 
			
		||||
You may act only on Your own behalf and on Your sole responsibility, not on
 | 
			
		||||
behalf of any other Contributor, and only if You agree to indemnify, defend,
 | 
			
		||||
and hold each Contributor harmless for any liability incurred by, or claims
 | 
			
		||||
asserted against, such Contributor by reason of your accepting any such warranty
 | 
			
		||||
or additional liability. END OF TERMS AND CONDITIONS
 | 
			
		||||
 | 
			
		||||
APPENDIX: How to apply the Apache License to your work.
 | 
			
		||||
 | 
			
		||||
To apply the Apache License to your work, attach the following boilerplate
 | 
			
		||||
notice, with the fields enclosed by brackets "[]" replaced with your own identifying
 | 
			
		||||
information. (Don't include the brackets!) The text should be enclosed in
 | 
			
		||||
the appropriate comment syntax for the file format. We also recommend that
 | 
			
		||||
a file or class name and description of purpose be included on the same "printed
 | 
			
		||||
page" as the copyright notice for easier identification within third-party
 | 
			
		||||
archives.
 | 
			
		||||
 | 
			
		||||
Copyright 2020 githubDante - <github -at- dante.tk>
 | 
			
		||||
 | 
			
		||||
Licensed under the Apache License, Version 2.0 (the "License");
 | 
			
		||||
 | 
			
		||||
you may not use this file except in compliance with the License.
 | 
			
		||||
 | 
			
		||||
You may obtain a copy of the License at
 | 
			
		||||
 | 
			
		||||
http://www.apache.org/licenses/LICENSE-2.0
 | 
			
		||||
 | 
			
		||||
Unless required by applicable law or agreed to in writing, software
 | 
			
		||||
 | 
			
		||||
distributed under the License is distributed on an "AS IS" BASIS,
 | 
			
		||||
 | 
			
		||||
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 | 
			
		||||
 | 
			
		||||
See the License for the specific language governing permissions and
 | 
			
		||||
 | 
			
		||||
limitations under the License.
 | 
			
		||||
							
								
								
									
										3
									
								
								ethMon/Makefile
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										3
									
								
								ethMon/Makefile
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,3 @@
 | 
			
		|||
DOCFILES := README.rst
 | 
			
		||||
 | 
			
		||||
include ../common.mk
 | 
			
		||||
							
								
								
									
										96
									
								
								ethMon/README.rst
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										96
									
								
								ethMon/README.rst
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,96 @@
 | 
			
		|||
***********
 | 
			
		||||
**ethMon**
 | 
			
		||||
***********
 | 
			
		||||
 | 
			
		||||
ethMon is an Icinga2/Nagios compatible Python plugin for network traffic monitoring on a specific interface.
 | 
			
		||||
Currently the carbon/graphite data is always in *bytes*.
 | 
			
		||||
 | 
			
		||||
Graph drawn from a collected data
 | 
			
		||||
 | 
			
		||||
.. image:: images/plugin_graph.png
 | 
			
		||||
 | 
			
		||||
In order to calculate the speed the plugin keeps the data from previous runs in ~/.ethMonCache.db
 | 
			
		||||
 | 
			
		||||
**Installation**
 | 
			
		||||
=================
 | 
			
		||||
* Download ethMon to the Icinga/Nagios plugin directory (`/usr/lib/nagios/plugins/` on Debian 10)
 | 
			
		||||
        * `Link <https://github.com/githubDante/ethMon/releases/download/v1.1.0/ethMon>`_
 | 
			
		||||
* Ensure that the executable bit of *ethMon* is set
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
**Configuration**
 | 
			
		||||
=================
 | 
			
		||||
 | 
			
		||||
* In order to use the plugin with Icinga the following configuration is needed:
 | 
			
		||||
    * in */usr/share/icinga2/include/command-plugins.conf* add the following
 | 
			
		||||
 | 
			
		||||
        .. code-block:: console
 | 
			
		||||
 | 
			
		||||
            template CheckCommand "ethMon-common" {
 | 
			
		||||
                command = [ PluginDir + "/ethMon" ]
 | 
			
		||||
                arguments = {
 | 
			
		||||
                        "-i" = {
 | 
			
		||||
                            value = "$interface$"
 | 
			
		||||
                            description = "Interface which will be monitored"
 | 
			
		||||
                        }
 | 
			
		||||
                        "-w" = {
 | 
			
		||||
                            value = "$speed_warn$"
 | 
			
		||||
                            description = "Warning threshold"
 | 
			
		||||
                        }
 | 
			
		||||
                        "-c" = {
 | 
			
		||||
                            value = "$speed_crit$"
 | 
			
		||||
                            description = "Critical threshold"
 | 
			
		||||
                        }
 | 
			
		||||
                        "-s" = {
 | 
			
		||||
                            value = "$speed_scale$"
 | 
			
		||||
                            description = "Speed scaling"
 | 
			
		||||
                        }
 | 
			
		||||
                        "--interval" = {
 | 
			
		||||
                            value = "$chk_interval$"
 | 
			
		||||
                            description = "Interval between the checks (in seconds)"
 | 
			
		||||
                        }
 | 
			
		||||
                }
 | 
			
		||||
                vars.speed_warn = 60
 | 
			
		||||
                vars.speed_crit = 100
 | 
			
		||||
                vars.speed_scale = "Mb"
 | 
			
		||||
                vars.chk_interval = 30
 | 
			
		||||
            }
 | 
			
		||||
            object CheckCommand "ethMon" {
 | 
			
		||||
                import "ethMon-common"
 | 
			
		||||
            }
 | 
			
		||||
        ..
 | 
			
		||||
 | 
			
		||||
    * then per host (or zone if the interface name match everywhere):
 | 
			
		||||
 | 
			
		||||
        .. code-block:: console
 | 
			
		||||
 | 
			
		||||
            apply Service "Traffic-enp3s5" {
 | 
			
		||||
                check_command = "ethMon"
 | 
			
		||||
                check_interval = 30
 | 
			
		||||
                vars.interface = "enp3s5"
 | 
			
		||||
            command_endpoint = host.vars.client_endpoint // Check executed on client node
 | 
			
		||||
            assign where host.vars.client_endpoint == "TheMonitoredHost" // The service is applied only to this node
 | 
			
		||||
            }
 | 
			
		||||
        ..
 | 
			
		||||
 | 
			
		||||
    * **NOTE**: The same value must be used for *check_interval & vars.chk_interval*
 | 
			
		||||
 | 
			
		||||
**Support for threshold ranges**
 | 
			
		||||
 | 
			
		||||
* `Icinga threshold ranges documentation <https://icinga.com/docs/icinga-2/latest/doc/05-service-monitoring/#threshold-ranges>`_
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
The plugin has partial support for the ranges described in the Icinga documentation. If used the ranges must be specified for both (warning & critical) parameters.
 | 
			
		||||
 Implemented are the following specifications:
 | 
			
		||||
    * min:max - The value must be outside the range in order the alert to be triggered
 | 
			
		||||
    * @min:max - The value must be inside the range in order the alert to be triggered
 | 
			
		||||
    * min: - The value must be below *min* or above 10*10^256 in order the alert to be triggered
 | 
			
		||||
    * @min: - The value must be between *min* and 10*10^256 in order the alert to be triggered
 | 
			
		||||
    * :max - The value must be below 0 and *max* in order the alert to be triggered
 | 
			
		||||
    * @:max - The value must be between 0 and *max* in order the alert to be triggered
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
**Motivation**
 | 
			
		||||
================
 | 
			
		||||
 | 
			
		||||
The plugin has been created as *check_eth* from Nagios Exchange refused to work in my current setup.
 | 
			
		||||
							
								
								
									
										6
									
								
								ethMon/control
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										6
									
								
								ethMon/control
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,6 @@
 | 
			
		|||
Uploaders: Jan Wagner <waja@cyconet.org>
 | 
			
		||||
Recommends: python3-minimal
 | 
			
		||||
Version: 1.1.0
 | 
			
		||||
Homepage: https://github.com/githubDante/ethMon
 | 
			
		||||
Watch: https://github.com/githubDante/ethMon/tags .*/v?(\d\S+)\.tar\.gz
 | 
			
		||||
Description: Plugin for network traffic monitoring on a specific interface
 | 
			
		||||
							
								
								
									
										5
									
								
								ethMon/copyright
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										5
									
								
								ethMon/copyright
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,5 @@
 | 
			
		|||
Copyright: (c) github@dante.tk
 | 
			
		||||
License: Apache License 2.0
 | 
			
		||||
 | 
			
		||||
On Debian systems, a copy of the Apache License 2.0 can be found in
 | 
			
		||||
/usr/share/common-licenses/Apache-2.0
 | 
			
		||||
							
								
								
									
										288
									
								
								ethMon/ethMon
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										288
									
								
								ethMon/ethMon
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,288 @@
 | 
			
		|||
#!/usr/bin/env python3
 | 
			
		||||
 | 
			
		||||
from argparse import ArgumentParser, Namespace
 | 
			
		||||
import re
 | 
			
		||||
from typing import Any, List, Dict, Tuple, Optional
 | 
			
		||||
import shelve
 | 
			
		||||
import sys
 | 
			
		||||
from pathlib import Path
 | 
			
		||||
 | 
			
		||||
__PREV_DATA__ = '{}/.ethMonCache'.format(Path.home().as_posix())
 | 
			
		||||
__total__ = 'tot_{}'
 | 
			
		||||
__old__ = 'old_{}'
 | 
			
		||||
 | 
			
		||||
__scale__ = {
 | 
			
		||||
    'KB': lambda x: x / 1000,
 | 
			
		||||
    'Kb': lambda x: (x*8) / 1000,
 | 
			
		||||
    'MB': lambda x: x / 1000**2,
 | 
			
		||||
    'Mb': lambda x: (x*8) / 1000**2,
 | 
			
		||||
    'GB': lambda x: x / 1000**3,
 | 
			
		||||
    'Gb': lambda x: (x*8) / 1000**3
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
__to_bytes__ = {
 | 
			
		||||
    'KB': lambda x: x * 1000,
 | 
			
		||||
    'Kb': lambda x: (x * 1000) / 8,
 | 
			
		||||
    'MB': lambda x: x * 1000**2,
 | 
			
		||||
    'Mb': lambda x: (x * 1000**2) / 8,
 | 
			
		||||
    'GB': lambda x: x * 1000**3,
 | 
			
		||||
    'Gb': lambda x: (x * 1000**3) / 8
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def options_parser() -> Namespace:
 | 
			
		||||
    _scalers = [i for i in __scale__.keys()]
 | 
			
		||||
    parser = ArgumentParser('ethMon')
 | 
			
		||||
    parser.add_argument('-i', '--interface', required=True, type=str, help='Network interface')
 | 
			
		||||
    parser.add_argument('-w', '--warning', type=str, required=True, help='Warning threshold')
 | 
			
		||||
    parser.add_argument('-c', '--critical', type=str, required=True, help='Critical threshold')
 | 
			
		||||
    parser.add_argument('-s', '--scale', choices=_scalers, help='Scaled results {}'.format(_scalers))
 | 
			
		||||
    parser.add_argument('--interval', type=int, help='Interval between the checks (in seconds)')
 | 
			
		||||
 | 
			
		||||
    return parser.parse_args()
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def threshold_spec(threshold: str) -> bool:
 | 
			
		||||
    """
 | 
			
		||||
    Check for ranges specified in the warning/critical options
 | 
			
		||||
    """
 | 
			
		||||
    try:
 | 
			
		||||
        x, y = threshold.split(':')
 | 
			
		||||
        return True
 | 
			
		||||
    except ValueError:
 | 
			
		||||
        return False
 | 
			
		||||
 | 
			
		||||
def to_int(x: str) -> int:
 | 
			
		||||
    """
 | 
			
		||||
    Try to convert a value to integer. Always returns zero on unsuccessfull conversion
 | 
			
		||||
 | 
			
		||||
    :param x: Any 
 | 
			
		||||
    """
 | 
			
		||||
    try:
 | 
			
		||||
        return int(x)
 | 
			
		||||
    except:
 | 
			
		||||
        return 0
 | 
			
		||||
 | 
			
		||||
def threshold_extract(threshold: str) -> Tuple[int, int, bool]:
 | 
			
		||||
    """
 | 
			
		||||
    Extract the range and the inner/outer parameter
 | 
			
		||||
    
 | 
			
		||||
 | 
			
		||||
    :return: upper (int), lower (int), inner (bool)
 | 
			
		||||
    """
 | 
			
		||||
    lower, upper = threshold.split(':')
 | 
			
		||||
    if lower == '':
 | 
			
		||||
        lower = 0
 | 
			
		||||
    if upper == '':
 | 
			
		||||
        """ Set the upper limit to a very large number if not specified """
 | 
			
		||||
        upper = 10*10**256
 | 
			
		||||
    if lower.startswith('@'):
 | 
			
		||||
        lower = lower.lstrip('@')
 | 
			
		||||
        inner = True
 | 
			
		||||
    else:
 | 
			
		||||
        inner = False
 | 
			
		||||
    
 | 
			
		||||
    if to_int(lower) > to_int(upper):
 | 
			
		||||
        raise ValueError('Invalid value in threshold specification. lower > upper')
 | 
			
		||||
    return to_int(lower), to_int(upper), inner
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def get_old_data(iface: str) -> Tuple[int, int, int, int]:
 | 
			
		||||
    """
 | 
			
		||||
    Get the old interface data from the storage (if available)
 | 
			
		||||
 | 
			
		||||
    :param iface: The name of the interface
 | 
			
		||||
    :return: RX bytes, TX bytes, RX Total, TX Total values from the previous run
 | 
			
		||||
    """
 | 
			
		||||
    storage = shelve.open(__PREV_DATA__)
 | 
			
		||||
    rx, tx = storage.get(__old__.format(iface), (0, 0))
 | 
			
		||||
    rx_total, tx_total = storage.get(__total__.format(iface), (0, 0))
 | 
			
		||||
    storage.close()
 | 
			
		||||
    return rx, tx, rx_total, tx_total
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def update_stats(iface: str, rx_bytes: int, tx_bytes: int) -> None:
 | 
			
		||||
    """
 | 
			
		||||
    Store the data from the current run
 | 
			
		||||
 | 
			
		||||
    :param iface: The name of the interface
 | 
			
		||||
    :param rx_bytes: RX bytes as read from the net/dev file
 | 
			
		||||
    :param tx_bytes: TX bytes values as read from the net/dev file
 | 
			
		||||
    :return:
 | 
			
		||||
    """
 | 
			
		||||
    storage = shelve.open(__PREV_DATA__)
 | 
			
		||||
    rx_total, tx_total = storage.get(__total__.format(iface), (0, 0))
 | 
			
		||||
    storage[__total__.format(iface)] = (rx_total + rx_bytes, tx_total + tx_bytes)
 | 
			
		||||
    storage[__old__.format(iface)] = (rx_bytes, tx_bytes)
 | 
			
		||||
    storage.close()
 | 
			
		||||
    return
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def get_iface_stats(iface: str) -> Tuple[int, int]:
 | 
			
		||||
    """
 | 
			
		||||
    Extract the interface statistics
 | 
			
		||||
 | 
			
		||||
    :param iface:
 | 
			
		||||
    :return:
 | 
			
		||||
    """
 | 
			
		||||
    """
 | 
			
		||||
    RX/TX slots
 | 
			
		||||
    bytes packets errs drop fifo frame compressed multicast
 | 
			
		||||
    """
 | 
			
		||||
    _rx_bytes = 0
 | 
			
		||||
    _tx_bytes = 0
 | 
			
		||||
    slots = []
 | 
			
		||||
    with open('/proc/net/dev', 'r') as stat:
 | 
			
		||||
        for line in stat:
 | 
			
		||||
            if iface not in line:
 | 
			
		||||
                continue
 | 
			
		||||
            line = line.rstrip()
 | 
			
		||||
            line = re.sub(r'\s\s+', ' ', line)
 | 
			
		||||
            line = line.lstrip(' ')
 | 
			
		||||
            slots = line.split(' ')
 | 
			
		||||
    if len(slots) > 10:
 | 
			
		||||
        _rx_bytes = int(slots[1])
 | 
			
		||||
        _tx_bytes = int(slots[9])
 | 
			
		||||
    return _rx_bytes, _tx_bytes
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def speed_calc(old_data: tuple, current_data: tuple) -> Tuple[int, int]:
 | 
			
		||||
    """
 | 
			
		||||
    Calculate the changes between the old and new data
 | 
			
		||||
 | 
			
		||||
    :param data:
 | 
			
		||||
    :param current_data:
 | 
			
		||||
    :return:
 | 
			
		||||
    """
 | 
			
		||||
    old_rx, old_tx = old_data[:2]
 | 
			
		||||
    cur_rx, cur_tx = current_data
 | 
			
		||||
 | 
			
		||||
    return cur_rx - old_rx, cur_tx - old_tx
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def speed_scaler(val: int, scaler: str) -> str:
 | 
			
		||||
    """
 | 
			
		||||
    available ['KB', 'Kb', 'MB', 'Mb', 'GB', 'Gb']
 | 
			
		||||
    """
 | 
			
		||||
    if scaler not in __scale__.keys():
 | 
			
		||||
        raise
 | 
			
		||||
    return str(round(__scale__[scaler](val), 3))
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def speed_normalizer(val: int, scaler: str) -> int:
 | 
			
		||||
    """
 | 
			
		||||
    available ['KB', 'Kb', 'MB', 'Mb', 'GB', 'Gb']
 | 
			
		||||
    """
 | 
			
		||||
    if scaler not in __to_bytes__.keys():
 | 
			
		||||
        raise
 | 
			
		||||
    return int(__to_bytes__[scaler](val))
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def final_string(rx_s: int, tx_s: int, warning_s: int, crit_s: int, code: int):
 | 
			
		||||
    status = 'OK'
 | 
			
		||||
    if code == 2:
 | 
			
		||||
        status = 'CRITICAL'
 | 
			
		||||
    elif code == 1:
 | 
			
		||||
        status = 'WARNING'
 | 
			
		||||
 | 
			
		||||
    return '{} bandwidth utilization | rx={}B;{}B;{}B tx={}B;{}B;{}B'.format(
 | 
			
		||||
        status, rx_s, warning_s, crit_s, tx_s, warning_s, crit_s
 | 
			
		||||
    )
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
if __name__ == '__main__':
 | 
			
		||||
    options = options_parser()
 | 
			
		||||
    old = get_old_data(options.interface)
 | 
			
		||||
    exit_c = 0
 | 
			
		||||
    if threshold_spec(options.warning) or threshold_spec(options.critical):
 | 
			
		||||
        use_thresholds = True
 | 
			
		||||
        try:
 | 
			
		||||
            warn_lower, warn_upper, warn_inner = threshold_extract(options.warning)
 | 
			
		||||
            crit_lower, crit_upper, crit_inner = threshold_extract(options.critical)
 | 
			
		||||
        except ValueError:
 | 
			
		||||
            print('Invalid range specification.')
 | 
			
		||||
            sys.exit(100)
 | 
			
		||||
    else:
 | 
			
		||||
        use_thresholds = False
 | 
			
		||||
 | 
			
		||||
    current = get_iface_stats(options.interface)
 | 
			
		||||
    rx_speed, tx_speed = speed_calc(old, current)
 | 
			
		||||
 | 
			
		||||
    if options.interval:
 | 
			
		||||
        interval_scaler = lambda x: int(x / options.interval)
 | 
			
		||||
        rx_speed = interval_scaler(rx_speed)
 | 
			
		||||
        tx_speed = interval_scaler(tx_speed)
 | 
			
		||||
 | 
			
		||||
    update_stats(options.interface, *current)
 | 
			
		||||
 | 
			
		||||
    if options.scale:
 | 
			
		||||
        _suffix = ' {}'.format(options.scale)
 | 
			
		||||
        c_0 = speed_scaler(current[0], options.scale) + _suffix
 | 
			
		||||
        c_1 = speed_scaler(current[1], options.scale) + _suffix
 | 
			
		||||
        rx_s = speed_scaler(rx_speed, options.scale) + _suffix
 | 
			
		||||
        tx_s = speed_scaler(tx_speed, options.scale) + _suffix
 | 
			
		||||
 | 
			
		||||
        if use_thresholds is False:
 | 
			
		||||
            warning_s = speed_normalizer(to_int(options.warning), options.scale)
 | 
			
		||||
            if warning_s < rx_speed or warning_s < tx_speed:
 | 
			
		||||
                exit_c = 1
 | 
			
		||||
            critical_s = speed_normalizer(to_int(options.critical), options.scale)
 | 
			
		||||
            if critical_s < rx_speed or critical_s < tx_speed:
 | 
			
		||||
                exit_c = 2
 | 
			
		||||
        else:
 | 
			
		||||
            warn_s_low = speed_normalizer(warn_lower, options.scale)
 | 
			
		||||
            warn_s_up = speed_normalizer(warn_upper, options.scale)
 | 
			
		||||
            crit_s_low = speed_normalizer(crit_lower, options.scale)
 | 
			
		||||
            crit_s_up = speed_normalizer(crit_upper, options.scale)
 | 
			
		||||
 | 
			
		||||
            if warn_inner is True:
 | 
			
		||||
                if (warn_s_low <= rx_speed <= warn_s_up) or (warn_s_low <= tx_speed <= warn_s_up):
 | 
			
		||||
                    exit_c = 1
 | 
			
		||||
            else:
 | 
			
		||||
                if (warn_s_low >= rx_speed or warn_s_up <= rx_speed) or (warn_s_low >= tx_speed or warn_s_up <= tx_speed):
 | 
			
		||||
                    exit_c = 1
 | 
			
		||||
            if crit_inner is True:
 | 
			
		||||
                if (crit_s_low <= rx_speed <= crit_s_up) or (crit_s_low <= tx_speed <= crit_s_up):
 | 
			
		||||
                    exit_c = 2
 | 
			
		||||
            else:
 | 
			
		||||
                if (crit_s_low >= rx_speed or crit_s_up <= rx_speed) or (crit_s_low >= tx_speed or crit_s_up <= tx_speed):
 | 
			
		||||
                    exit_c = 2
 | 
			
		||||
            warning_s = warn_s_low
 | 
			
		||||
            critical_s = crit_s_low
 | 
			
		||||
                
 | 
			
		||||
 | 
			
		||||
        print('RX {}: {}, TX {}: {}; RX speed: {} TX speed: {}'.format(
 | 
			
		||||
            options.scale, c_0, options.scale, c_1, rx_s, tx_s
 | 
			
		||||
        ), end='; ')
 | 
			
		||||
        print(final_string(rx_speed, tx_speed, warning_s, critical_s, exit_c))
 | 
			
		||||
 | 
			
		||||
    else:
 | 
			
		||||
        print('RX bytes: {}, TX bytes: {}; RX speed: {}, TX speed {}'.format(
 | 
			
		||||
            current[0], rx_speed, current[1], tx_speed
 | 
			
		||||
        ), end='; ')
 | 
			
		||||
        if use_thresholds is False:
 | 
			
		||||
            warning_s = to_int(options.warning)
 | 
			
		||||
            if warning_s < rx_speed or warning_s < tx_speed:
 | 
			
		||||
                exit_c = 1
 | 
			
		||||
            critical_s = to_int(options.critical)
 | 
			
		||||
            if critical_s < rx_speed or critical_s <= tx_speed:
 | 
			
		||||
                exit_c = 2
 | 
			
		||||
        else:
 | 
			
		||||
            if crit_inner is True:
 | 
			
		||||
                if (warn_lower <= rx_speed <= warn_upper) or (warn_lower <= tx_speed <= warn_upper):
 | 
			
		||||
                    exit_c = 1
 | 
			
		||||
            else:
 | 
			
		||||
                if (warn_lower >= rx_speed or warn_upper <= rx_speed) or (warn_lower >= tx_speed or warn_upper <= rx_speed):
 | 
			
		||||
                    exit_c = 1
 | 
			
		||||
            if crit_inner is True:
 | 
			
		||||
                if (crit_lower <= rx_speed <= crit_upper) or (crit_lower <= tx_speed <= crit_upper):
 | 
			
		||||
                    exit_c = 2
 | 
			
		||||
            else:
 | 
			
		||||
                if (crit_lower >= rx_speed or crit_upper <= rx_speed) or (crit_lower >= tx_speed or crit_upper <= tx_speed):
 | 
			
		||||
                    exit_c = 2
 | 
			
		||||
            warning_s = warn_lower
 | 
			
		||||
            critical_s = crit_lower
 | 
			
		||||
 | 
			
		||||
        print(final_string(rx_speed, tx_speed, warning_s, critical_s, exit_c))
 | 
			
		||||
 | 
			
		||||
    sys.exit(exit_c)
 | 
			
		||||
		Loading…
	
	Add table
		Add a link
		
	
		Reference in a new issue