ID2TAttackTest.py 7.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152
  1. import unittest
  2. import inspect
  3. import ID2TLib.Controller as Ctrl
  4. import ID2TLib.TestLibrary as Lib
  5. import scapy.utils as pcr
  6. class ID2TAttackTest(unittest.TestCase):
  7. """
  8. Generic Test Class for ID2T attacks based on unittest.TestCase.
  9. """
  10. def checksum_test(self, attack_args, sha256_checksum, seed=5, cleanup=True, pcap=Lib.test_pcap,
  11. flag_write_file=False, flag_recalculate_stats=False, flag_print_statistics=False,
  12. attack_sub_dir=True, test_sub_dir=True, time=False):
  13. """
  14. Runs the attack against a given sha256 checksum.
  15. :param attack_args: A list of attacks with their attack parameters (as defined in Controller.process_attacks).
  16. :param sha256_checksum: The checksum to verify the result pcap.
  17. :param seed: A random seed to keep random values static (care for count and order of random generation).
  18. :param cleanup: Clean up attack output after testing.
  19. :param pcap: The input pcap for the attack.
  20. :param flag_write_file: Writes the statistics to a file.
  21. :param flag_recalculate_stats: Forces the recalculation of statistics.
  22. :param flag_print_statistics: Prints the statistics on the terminal.
  23. :param attack_sub_dir: create sub-directory for each attack-class if True
  24. :param test_sub_dir: create sub-directory for each test-function/case if True
  25. :param time: Measure time for packet generation.
  26. """
  27. controller = Ctrl.Controller(pcap_file_path=pcap, do_extra_tests=False)
  28. controller.load_pcap_statistics(flag_write_file, flag_recalculate_stats, flag_print_statistics)
  29. controller.process_attacks(attack_args, [[seed]], time)
  30. caller_function = inspect.stack()[1].function
  31. try:
  32. self.assertEqual(sha256_checksum, Lib.get_sha256(controller.pcap_dest_path))
  33. except self.failureException:
  34. Lib.rename_test_result_files(controller, caller_function, attack_sub_dir, test_sub_dir)
  35. raise
  36. if cleanup:
  37. Lib.clean_up(controller)
  38. else:
  39. Lib.rename_test_result_files(controller, caller_function, attack_sub_dir, test_sub_dir)
  40. def temporal_efficiency_test(self, attack_args, time_limit=15, factor=1, seed=None, cleanup=True, pcap=Lib.test_pcap,
  41. flag_write_file=False, flag_recalculate_stats=False, flag_print_statistics=False,
  42. attack_sub_dir=True, test_sub_dir=True):
  43. """
  44. Runs the attack with given aruments and monitors time efficiency.
  45. :param attack_args: A list of attacks with their attack parameters (as defined in Controller.process_attacks).
  46. :param time_limit: The given time limit in seconds.
  47. :param factor: A factor to scale the generation time (e.g. only 7 pkts generated -> *10000/7 for 15 seconds).
  48. :param seed: A random seed to keep random values static (care for count and order of random generation).
  49. :param cleanup: Clean up attack output after testing.
  50. :param pcap: The input pcap for the attack.
  51. :param flag_write_file: Writes the statistics to a file.
  52. :param flag_recalculate_stats: Forces the recalculation of statistics.
  53. :param flag_print_statistics: Prints the statistics on the terminal.
  54. :param attack_sub_dir: create sub-directory for each attack-class if True
  55. :param test_sub_dir: create sub-directory for each test-function/case if True
  56. """
  57. controller = Ctrl.Controller(pcap_file_path=pcap, do_extra_tests=False)
  58. controller.load_pcap_statistics(flag_write_file, flag_recalculate_stats, flag_print_statistics)
  59. if seed is None:
  60. controller.process_attacks(attack_args, time=True)
  61. else:
  62. controller.process_attacks(attack_args, [[seed]], time=True)
  63. duration = controller.durations[0]*factor/controller.attack_controller.total_packets
  64. print(attack_args[0][0] + ' needs ' + str(duration) + ' seconds to generate ' + str(factor) + ' packets.')
  65. caller_function = inspect.stack()[1].function
  66. try:
  67. self.assertLessEqual(duration, time_limit)
  68. except self.failureException:
  69. Lib.rename_test_result_files(controller, caller_function, attack_sub_dir, test_sub_dir)
  70. raise
  71. if cleanup:
  72. Lib.clean_up(controller)
  73. else:
  74. Lib.rename_test_result_files(self.controller, caller_function, attack_sub_dir, test_sub_dir)
  75. def order_test(self, attack_args, seed=None, cleanup=True, pcap=Lib.test_pcap,
  76. flag_write_file=False, flag_recalculate_stats=False, flag_print_statistics=False,
  77. attack_sub_dir=True, test_sub_dir=True):
  78. """
  79. Checks if the result of an attack includes all packets in correct order.
  80. :param attack_args: A list of attacks with their attack parameters (as defined in Controller.process_attacks).
  81. :param seed: A random seed to keep random values static (care for count and order of random generation).
  82. :param cleanup: Clean up attack output after testing.
  83. :param pcap: The input pcap for the attack.
  84. :param flag_write_file: Writes the statistics to a file.
  85. :param flag_recalculate_stats: Forces the recalculation of statistics.
  86. :param flag_print_statistics: Prints the statistics on the terminal.
  87. :param attack_sub_dir: create sub-directory for each attack-class if True
  88. :param test_sub_dir: create sub-directory for each test-function/case if True
  89. """
  90. controller = Ctrl.Controller(pcap_file_path=pcap, do_extra_tests=False)
  91. controller.load_pcap_statistics(flag_write_file, flag_recalculate_stats, flag_print_statistics)
  92. controller.process_attacks(attack_args, [[seed]])
  93. caller_function = inspect.stack()[1].function
  94. try:
  95. path = controller.pcap_dest_path
  96. file = pcr.RawPcapReader(path)
  97. packet_a = file.read_packet()
  98. packet_b = file.read_packet()
  99. i = 0
  100. while packet_b is not None:
  101. time_a = packet_a[2][0:2]
  102. time_b = packet_b[2][0:2]
  103. if time_a[0] > time_b[0]:
  104. file.close()
  105. self.fail("Packet order incorrect at: " + str(i+1) + "-" + str(i+2) +
  106. ". Current time: " + str(time_a) + " Next time: " + str(time_b))
  107. elif time_a[0] == time_b[0]:
  108. if time_a[1] > time_b[1]:
  109. file.close()
  110. self.fail("Packet order incorrect at: " + str(i + 1) + "-" + str(i + 2) +
  111. ". Current time: " + str(time_a) + " Next time: " + str(time_b))
  112. packet_a = packet_b
  113. packet_b = file.read_packet()
  114. i += 1
  115. file.close()
  116. except self.failureException:
  117. Lib.rename_test_result_files(controller, caller_function, attack_sub_dir, test_sub_dir)
  118. raise
  119. if cleanup:
  120. Lib.clean_up(controller)
  121. else:
  122. Lib.rename_test_result_files(controller, caller_function, attack_sub_dir, test_sub_dir)