diff --git a/Python/attacker.py b/Python/attacker.py index 7dc83e0..bd3b66e 100644 --- a/Python/attacker.py +++ b/Python/attacker.py @@ -99,8 +99,9 @@ def generateSYN(self): def receivePacket(self, packet): if(packet.packetType == "PONG" and self.openPort != None): self.openPorts.append(packet.srcPortNumber) - else: - self.openPort = None + elif(packet.packetType == "PONG-RST"): + if(packet.srcPortNumber == self.openPort): + self.openPort = None diff --git a/Python/router.py b/Python/router.py index c85bb11..b128b7b 100644 --- a/Python/router.py +++ b/Python/router.py @@ -132,7 +132,7 @@ def filterPackets(self, packetT): #If the packets dst port number is not found in either list, then drop the packet if(allow == False): if(packetT.packetType == "PING"): - npacket = packet.packet(self.ipAddress,packetT.dstPortNumber,packetT.srcIP, packetT.srcPortNumber, "RST", random.randint(0, 10000), 0, "R" ,self.packetSize, self.ipAddress) + npacket = packet.packet(self.ipAddress,packetT.dstPortNumber,packetT.srcIP, packetT.srcPortNumber, "PONG-RST", random.randint(0, 10000), 0, "R" ,self.packetSize, self.ipAddress) self.fileGen.addToLog(npacket.packetID, "Generated at Router", npacket.sequenceNumber, npacket.ackNumber, npacket.packetType, self.curTime) self.queueAttacker.append(npacket) print("Packet dropped at the DDD Filter") @@ -318,7 +318,7 @@ def sendPackets(self): else: print("Packet Dropped at the Port Mapper") if(self.sendClient[0].packetType == "PING"): - npacket = packet.packet(self.ipAddress,self.sendClient[0].dstPortNumber,self.sendClient[0].srcIP, self.sendClient[0].srcPortNumber, "RST", random.randint(0, 10000), 0, "R" ,self.packetSize, self.ipAddress) + npacket = packet.packet(self.ipAddress,self.sendClient[0].dstPortNumber,self.sendClient[0].srcIP, self.sendClient[0].srcPortNumber, "PONG-RST", random.randint(0, 10000), 0, "R" ,self.packetSize, self.ipAddress) self.fileGen.addToLog(npacket.packetID, "Generated at Router", npacket.sequenceNumber, npacket.ackNumber, npacket.packetType, self.curTime) self.queueAttacker.append(npacket) self.fileGen.animateDrop(self.packetIDPrefix[0], self.sendClient[0].packetType, self.sendClient[0].sequenceNumber, self.sendClient[0].ackNumber, self.sendClient[0].dstPortNumber)