diff options
Diffstat (limited to 'ParrotZik.py')
-rw-r--r-- | ParrotZik.py | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/ParrotZik.py b/ParrotZik.py index 2746259..05e316e 100644 --- a/ParrotZik.py +++ b/ParrotZik.py @@ -13,7 +13,8 @@ class ParrotZik(object): if len(service_matches) == 0: print "Couldn't find Parrot Zik" - return False + self.sock = '' + return first_match = service_matches[0] port = first_match["port"] @@ -86,7 +87,11 @@ class ParrotZik(object): return self.sendMessage(message) def sendMessage(self,message): - self.sock.send(str(message)) + try: + self.sock.send(str(message)) + except: + self.sock = '' + return data = self.sock.recv(7) len = struct.unpack('B', data[1])[0] data = self.sock.recv(1024) |