(WIP #39) ARPspoof tests (need to improve).

m4n3dw0lf · Nov 27, 2018 · 5ada971 · 5ada971
1 parent 1f9a3a4
commit 5ada971
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 4 deletions.
diff --git a/pythem/modules/arpoisoner.py b/pythem/modules/arpoisoner.py
@@ -90,7 +90,7 @@ def get_range(self, targets):
     def resolve_mac(self, targetip):
         try:
             conf.verb = 0
-            ans, unans = srp(Ether(dst="ff:ff:ff:ff:ff:ff") / ARP(op="who-has", pdst=targetip), timeout=2)
+            ans, unans = srp(Ether(dst="ff:ff:ff:ff:ff:ff") / ARP(op="who-has", pdst=targetip), timeout=2, iface=self.interface)
             for snd, rcv in ans:
                 return str(rcv[Ether].src)
         except socket.gaierror:

diff --git a/pythem/modules/redirect.py b/pythem/modules/redirect.py
@@ -49,7 +49,7 @@ def start(self, host, port, js):
 <head>
 <script src="{}"></script>
 </head>
-""".format(self.js)
+""".format(js)
         self.response = self.js
         self.host = host
         self.port = int(port)
@@ -81,7 +81,7 @@ def server(self, host, port, js):
 <head>
 <script src="{}"></script>
 </head>
-""".format(self.js)
+""".format(js)
         self.response = self.js
         self.host = host
         self.port = int(port)
@@ -111,6 +111,7 @@ def server(self, host, port, js):
                     if self.dnsspoof:
                       self.dnsspoof.stop()
                 except AttributeError:
+                    domain = None
                     pass
                 try:
                     connection, client_address = server.accept()

diff --git a/...em/tests/test_modules_objects_creation.py → pythem/tests/prerequisites_for_tests.py b/...em/tests/test_modules_objects_creation.py → pythem/tests/prerequisites_for_tests.py
diff --git a/pythem/tests/test_redirect_module.py b/pythem/tests/test_redirect_module.py
@@ -10,7 +10,7 @@
 from time import sleep
 
 class TestModulesObjectsCreation(unittest.TestCase):
-    def test_redirect_startup(self):
+    def test_redirect(self):
         from pythem.modules.utils import get_myip
         myip = get_myip('lo')
         from pythem.modules.redirect import Redirect