3
0

test_SMBScan.py 3.9 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273
  1. import unittest.mock as mock
  2. import Test.ID2TAttackTest as Test
  3. sha_default = 'ef321877edfd828f6e6cd4abbffb5ade9cb66b3acd54ba9f3a5e2bfbeac9c964'
  4. sha_one_victim_linux = '4928d421caaec8f2c4e5c5bb835b5521b705478779cbc8f343b77143a5a66995'
  5. sha_victim_range_winxp_hosting = '57a0b7dd69a5bba35404af574d4f06ef52ac2b3b292703596dabd2d1c31721b0'
  6. sha_multiple_victims_macos = '82d6d7e0471e6395c77df7b5bac141e48d50afe22841c7c53747bbfdd0de184d'
  7. sha_port_shuffle = '85d4fd1b44e41cfb30d5758c7264f4d5509701c04a0f12495b4155011fc3aaaa'
  8. sha_dest_mac_only = '0814dadb666e0056ef5b3a572a4971f333376b61e602acb84cb99c851845f016'
  9. sha_ip_src_shuffle = 'f070db569ecf4e17003e60f9ac53c064279c732ccb2128c13c8a7e3b64adc846'
  10. sha_smb2 = '9d78ac62d76a811c62e0ba7f0ed88569fd133cc06756451a58021be5e1c9fb61'
  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',
  23. 'ip.dst.end=192.168.178.10', 'hosting.ip=192.168.178.5']],
  24. sha_victim_range_winxp_hosting)
  25. def test_smbscan_multiple_victims_macos(self):
  26. with mock.patch("ID2TLib.Utility.get_rnd_os", return_value="macos"):
  27. self.checksum_test([['SMBScanAttack', 'ip.src=192.168.178.1',
  28. 'ip.dst=192.168.178.10,192.168.178.15,192.168.178.20',
  29. 'hosting.ip=192.168.178.15,192.168.178.20']], sha_multiple_victims_macos)
  30. def test_smbscan_invalid_smb_version(self):
  31. with self.assertRaises(SystemExit):
  32. self.checksum_test([['SMBScanAttack', 'protocol.version=42']], 'somehash')
  33. def test_smbscan_invalid_smb_platform(self):
  34. with self.assertRaises(SystemExit):
  35. self.checksum_test([['SMBScanAttack', 'hosting.version=1337']], 'somehash')
  36. def test_smbscan_port_shuffle(self):
  37. with mock.patch("ID2TLib.Utility.get_rnd_os", return_value="win7"):
  38. self.checksum_test([['SMBScanAttack', 'ip.src=192.168.178.1', 'ip.dst=192.168.178.5',
  39. 'ip.dst.end=192.168.178.10', 'hosting.ip=192.168.178.5', 'port.src.shuffle=false']],
  40. sha_port_shuffle)
  41. def test_smbscan_dest_mac_only(self):
  42. with mock.patch("ID2TLib.Utility.get_rnd_os", return_value="win7"):
  43. self.checksum_test([['SMBScanAttack', 'ip.src=192.168.178.1',
  44. 'mac.dst=00:0C:29:9C:70:64']], sha_dest_mac_only)
  45. def test_smbscan_src_ip_shuffle(self):
  46. with mock.patch("ID2TLib.Utility.get_rnd_os", return_value="win7"):
  47. self.checksum_test([['SMBScanAttack', 'ip.src=192.168.178.1', 'ip.dst=192.168.178.5',
  48. 'ip.dst.end=192.168.178.10', 'hosting.ip=192.168.178.5', 'ip.src.shuffle=True']],
  49. sha_ip_src_shuffle)
  50. def test_smbscan_smb2(self):
  51. with mock.patch("ID2TLib.Utility.get_rnd_os", return_value="linux"):
  52. self.checksum_test([['SMBScanAttack', 'ip.src=192.168.178.1', 'ip.dst=192.168.178.5',
  53. 'ip.dst.end=192.168.178.10', 'hosting.ip=192.168.178.5', 'protocol.version=2.1',
  54. 'hosting.version=2.1']], sha_smb2)
  55. def test_smbscan_order(self):
  56. self.order_test([['SMBScanAttack']])