|
@@ -6,10 +6,10 @@ import Test.ID2TAttackTest as Test
|
|
|
|
|
|
class MemoryTests(Test.ID2TAttackTest):
|
|
|
def test_SMBLoris(self):
|
|
|
- self.memory_test([['SMBLorisAttack']])
|
|
|
+ self.memory_test([['SMBLorisAttack', 'attackers.count=30', 'packets.per-second=8.0']])
|
|
|
|
|
|
- def test_SMBScanself(self):
|
|
|
- self.memory_test([['SMBScanAttack']])
|
|
|
+ def test_SMBScan(self):
|
|
|
+ self.memory_test([['SMBScanAttack', 'ip.src=192.168.178.1', 'ip.dst=192.168.178.10-192.168.197.145']])
|
|
|
|
|
|
@mock.patch('ID2TLib.Utility.get_rnd_bytes', side_effect=Lib.get_bytes)
|
|
|
@mock.patch('ID2TLib.Utility.get_rnd_x86_nop', side_effect=Lib.get_x86_nop)
|
|
@@ -30,7 +30,7 @@ class MemoryTests(Test.ID2TAttackTest):
|
|
|
|
|
|
@mock.patch('Attack.BaseAttack.BaseAttack.write_attack_pcap', side_effect=Lib.write_attack_pcap)
|
|
|
def test_DDoS(self, mock_write_attack_pcap):
|
|
|
- self.memory_test([['DDoSAttack']])
|
|
|
+ self.memory_test([['DDoSAttack', 'attackers.count=100', 'packets.per-second=95', 'attack.duration=150']])
|
|
|
|
|
|
def test_EternalBlue(self):
|
|
|
self.memory_test([['EternalBlueExploit']])
|