Added self as tear_down_monitor argument *blush*

[pwnitter] / pwnitter.py

diff --git a/pwnitter.py b/pwnitter.py
index faca125..67d3f75 100755 (executable)
--- a/pwnitter.py
+++ b/pwnitter.py
@@ -42,7 +42,10 @@ class Pwnitter(dbus.service.Object):
     def Start(self, filename=None):
         # FIXME: Prevent double Start()
         if filename is None: # Then we do *not* want to read from a PCap file but rather a monitor device
-            self.setup_monitor(self.device)
+            self.setup_monitor(device)
+            device = self.device
+        else: # We have given a filename, so let's make PCap read from the file
+            device = filename
         self.is_running = True
         try:
             self.cap = pcap.pcap(device)
@@ -80,7 +83,7 @@ class Pwnitter(dbus.service.Object):
         return self.status
 
 
-    def tear_down_monitor(device='mon0'):
+    def tear_down_monitor(self, device='mon0'):
         cmd = '/sbin/ifconfig mon0 down'.split()
         subprocess.call(cmd)
         cmd = '/usr/sbin/iw dev mon0 del'.split()