3
0

test_SMBScanAttack.py 3.7 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667
  1. import unittest.mock as mock
  2. import Test.ID2TAttackTest as Test
  3. sha_default = 'c61cb8ce03e6b8b19132ec6a47adcfb02c4dba4234926653df5443d33b08f33b'
  4. sha_dest_mac_only = 'c42a1775db981a139abd42d031273805cbebd2316b0d8c097217c12193fb8a70'
  5. sha_multiple_victims_macos = 'b9a9f423d4154bc38723214124ad74dfdd07a39753563d21f5b453a8c069914a'
  6. sha_one_victim_linux = '3bb17444446334cf4feee9dd7cbeabd17acbb5ef48525fb3963591f30c37d17a'
  7. sha_port_shuffle = '08bdecc68fa1a2d1b0dd9802d7d025d42d90b9184d1fb6e1bcab234fac7db1b4'
  8. sha_smb2 = '315bc052fd045f8738021062e8b5f77a33c649adfed490d3c9da94c97ba32f95'
  9. sha_ip_src_shuffle = '1d699ca109c62000b77b53002f1087ebf5ccc2c2dead1dbc5c18b5f6311273d0'
  10. sha_victim_range_winxp_hosting = 'bd624da4e3b7a3f06b8154ed9d6274d498b589aaaa11c2d0dc207a80ab7205b9'
  11. # TODO: improve coverage
  12. class UnitTestSMBScan(Test.ID2TAttackTest):
  13. def test_smbscan_default(self):
  14. with mock.patch("ID2TLib.Utility.get_rnd_os", return_value="win7"):
  15. self.checksum_test([['SMBScanAttack']], sha_default)
  16. def test_smbscan_one_victim_linux(self):
  17. with mock.patch("ID2TLib.Utility.get_rnd_os", return_value="linux"):
  18. self.checksum_test([['SMBScanAttack', 'ip.src=192.168.178.1', 'ip.dst=192.168.178.10']],
  19. sha_one_victim_linux)
  20. def test_smbscan_victim_range_winxp_hosting(self):
  21. with mock.patch("ID2TLib.Utility.get_rnd_os", return_value="winxp"):
  22. self.checksum_test([['SMBScanAttack', 'ip.src=192.168.178.1', 'ip.dst=192.168.178.5-192.168.178.10',
  23. 'hosting.ip=192.168.178.5']], sha_victim_range_winxp_hosting)
  24. def test_smbscan_multiple_victims_macos(self):
  25. with mock.patch("ID2TLib.Utility.get_rnd_os", return_value="macos"):
  26. self.checksum_test([['SMBScanAttack', 'ip.src=192.168.178.1',
  27. 'ip.dst=192.168.178.10,192.168.178.15,192.168.178.20',
  28. 'hosting.ip=192.168.178.15,192.168.178.20']], sha_multiple_victims_macos)
  29. def test_smbscan_invalid_smb_version(self):
  30. with self.assertRaises(SystemExit):
  31. self.checksum_test([['SMBScanAttack', 'protocol.version=42']], 'somehash')
  32. def test_smbscan_invalid_smb_platform(self):
  33. with self.assertRaises(SystemExit):
  34. self.checksum_test([['SMBScanAttack', 'hosting.version=1337']], 'somehash')
  35. def test_smbscan_port_shuffle(self):
  36. with mock.patch("ID2TLib.Utility.get_rnd_os", return_value="win7"):
  37. self.checksum_test([['SMBScanAttack', 'ip.src=192.168.178.1', 'ip.dst=192.168.178.5-192.168.178.10',
  38. 'hosting.ip=192.168.178.5', 'port.src.shuffle=false']], sha_port_shuffle)
  39. def test_smbscan_dest_mac_only(self):
  40. with mock.patch("ID2TLib.Utility.get_rnd_os", return_value="win7"):
  41. self.checksum_test([['SMBScanAttack', 'ip.src=192.168.178.1', 'mac.dst=00:0C:29:9C:70:64']],
  42. sha_dest_mac_only)
  43. def test_smbscan_src_ip_shuffle(self):
  44. with mock.patch("ID2TLib.Utility.get_rnd_os", return_value="win7"):
  45. self.checksum_test([['SMBScanAttack', 'ip.src=192.168.178.1', 'ip.dst=192.168.178.5-192.168.178.10',
  46. 'hosting.ip=192.168.178.5', 'ip.src.shuffle=True']], sha_ip_src_shuffle)
  47. def test_smbscan_smb2(self):
  48. with mock.patch("ID2TLib.Utility.get_rnd_os", return_value="linux"):
  49. self.checksum_test([['SMBScanAttack', 'ip.src=192.168.178.1', 'ip.dst=192.168.178.5-192.168.178.10',
  50. 'hosting.ip=192.168.178.5', 'protocol.version=2.1', 'hosting.version=2.1']], sha_smb2)
  51. def test_smbscan_order(self):
  52. self.order_test([['SMBScanAttack']])