Test interface is up before running

mami-project · Oct 20, 2016 · df6554a · df6554a
1 parent ea17f47
commit df6554a
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 11 deletions.
diff --git a/pathspider/Network/test_interface.py b/pathspider/Network/test_interface.py
diff --git a/pathspider/network/__init__.py b/pathspider/network/__init__.py
@@ -0,0 +1,9 @@
+
+from pyroute2 import IPDB # pylint: disable=no-name-in-module
+
+def interface_up(int):
+    with IPDB() as ipdb:
+        for interface in set(ipdb.interfaces.values()):
+            if interface.ifname == int and interface.operstate == 'UP':
+                return True
+        return False
diff --git a/pathspider/standalone.py b/pathspider/standalone.py
@@ -7,6 +7,7 @@
 
 from pathspider.base import SHUTDOWN_SENTINEL
 
+from pathspider.network import interface_up
 
 def job_feeder(inputfile, spider):
     logger = logging.getLogger("feeder")
@@ -28,7 +29,11 @@ def run_standalone(args):
 
     try:
         if hasattr(args, "spider"):
-            spider = args.spider(args.workers, "int:" + args.interface, args)
+            if interface_up(args.interface):
+                spider = args.spider(args.workers, "int:" + args.interface, args)
+            else:
+                logger.error("The chosen interface is not up! Cannot continue.")
+                sys.exit(1)
         else:
             logger.error("Plugin not found! Cannot continue.")
             logger.error("Use --help to list all plugins.")

diff --git a/requirements.txt b/requirements.txt
@@ -1,2 +1,3 @@
 straight.plugin
 dnspython
+pyroute2