|
|
@@ -1,5 +1,6 @@
|
|
|
import unittest.mock as mock
|
|
|
|
|
|
+import ID2TLib.Utility as Util
|
|
|
import ID2TLib.TestLibrary as Lib
|
|
|
import Test.ID2TAttackTest as Test
|
|
|
|
|
|
@@ -10,6 +11,8 @@ sha_SMBScan_100_000 = '9e55f4c2f035ec52701eabed32757f427627ce5ccf53e30bfc084680a
|
|
|
sha_SMBScan_hosting_10_000 = '14f92d19535332bf523e94bcb2038309844abedaa848ca7195a343256adba5f3'
|
|
|
sha_SMBScan_hosting_100_000 = '029d064de82122202b6ae53d4efff2ea3318ded73a2987cb16e1e74606532766'
|
|
|
sha_FTPExploit = '75290f0135b13b9d570a484fc7c674b80921a9311cd1229243ea8c547d8c08f0'
|
|
|
+sha_Portscan_open = '2e1deda7b36fb39705dd44dcb2350cca547cfb5049a16fb383c522dbe7e1d4e9'
|
|
|
+sha_Portscan_close = '3a62f594b9cd31bf7b8c455d1bb5cff3ec2beb044f6da39b066317910f10be66'
|
|
|
|
|
|
|
|
|
class EfficiencyTests(Test.ID2TAttackTest):
|
|
|
@@ -50,3 +53,11 @@ class EfficiencyTests(Test.ID2TAttackTest):
|
|
|
self.checksum_test([['FTPWinaXeExploit', 'ip.src=192.168.178.1', 'ip.dst=192.168.178.10']],
|
|
|
sha_FTPExploit, time=True)
|
|
|
self.assertLessEqual(self.controller.durations[0]*10000/7, 15)
|
|
|
+
|
|
|
+ def test_PortscanAttack_open(self):
|
|
|
+ self.checksum_test([['PortscanAttack', 'ip.src=192.168.178.1', 'port.open=80']], sha_Portscan_open, time=True)
|
|
|
+ self.assertLessEqual(self.controller.durations[0]*10000/1002, 15)
|
|
|
+
|
|
|
+ def test_PortscanAttack_close(self):
|
|
|
+ self.checksum_test([['PortscanAttack', 'ip.src=192.168.178.1', 'port.open=20']], sha_Portscan_close, time=True)
|
|
|
+ self.assertLessEqual(self.controller.durations[0]*10, 15)
|
