123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154 |
- import inspect
- import unittest
- import scapy.utils as pcr
- import Core.Controller as Ctrl
- import ID2TLib.TestLibrary as Lib
- class ID2TAttackTest(unittest.TestCase):
- """
- Generic Test Class for Core attacks based on unittest.TestCase.
- """
- def checksum_test(self, attack_args, sha256_checksum, seed=5, cleanup=True, pcap=Lib.test_pcap,
- flag_write_file=False, flag_recalculate_stats=False, flag_print_statistics=False,
- attack_sub_dir=True, test_sub_dir=True, time=False):
- """
- Runs the attack against a given sha256 checksum.
- :param attack_args: A list of attacks with their attack parameters (as defined in Controller.process_attacks).
- :param sha256_checksum: The checksum to verify the result pcap.
- :param seed: A random seed to keep random values static (care for count and order of random generation).
- :param cleanup: Clean up attack output after testing.
- :param pcap: The input pcap for the attack.
- :param flag_write_file: Writes the statistics to a file.
- :param flag_recalculate_stats: Forces the recalculation of statistics.
- :param flag_print_statistics: Prints the statistics on the terminal.
- :param attack_sub_dir: create sub-directory for each attack-class if True
- :param test_sub_dir: create sub-directory for each test-function/case if True
- :param time: Measure time for packet generation.
- """
- controller = Ctrl.Controller(pcap_file_path=pcap, do_extra_tests=False, non_verbose=True)
- controller.load_pcap_statistics(flag_write_file, flag_recalculate_stats, flag_print_statistics)
- controller.process_attacks(attack_args, [[seed]], time)
- caller_function = inspect.stack()[1].function
- try:
- self.assertEqual(sha256_checksum, Lib.get_sha256(controller.pcap_dest_path))
- except self.failureException:
- Lib.rename_test_result_files(controller, caller_function, attack_sub_dir, test_sub_dir)
- raise
- if cleanup:
- Lib.clean_up(controller)
- else:
- Lib.rename_test_result_files(controller, caller_function, attack_sub_dir, test_sub_dir)
- def temporal_efficiency_test(self, attack_args, time_limit=15, factor=1, seed=None, cleanup=True,
- pcap=Lib.test_pcap,
- flag_write_file=False, flag_recalculate_stats=False, flag_print_statistics=False,
- attack_sub_dir=True, test_sub_dir=True):
- """
- Runs the attack with given aruments and monitors time efficiency.
- :param attack_args: A list of attacks with their attack parameters (as defined in Controller.process_attacks).
- :param time_limit: The given time limit in seconds.
- :param factor: A factor to scale the generation time (e.g. only 7 pkts generated -> *10000/7 for 15 seconds).
- :param seed: A random seed to keep random values static (care for count and order of random generation).
- :param cleanup: Clean up attack output after testing.
- :param pcap: The input pcap for the attack.
- :param flag_write_file: Writes the statistics to a file.
- :param flag_recalculate_stats: Forces the recalculation of statistics.
- :param flag_print_statistics: Prints the statistics on the terminal.
- :param attack_sub_dir: create sub-directory for each attack-class if True
- :param test_sub_dir: create sub-directory for each test-function/case if True
- """
- controller = Ctrl.Controller(pcap_file_path=pcap, do_extra_tests=False, non_verbose=True)
- controller.load_pcap_statistics(flag_write_file, flag_recalculate_stats, flag_print_statistics)
- if seed is None:
- controller.process_attacks(attack_args, time=True)
- else:
- controller.process_attacks(attack_args, [[seed]], time=True)
- duration = controller.durations[0] * factor / controller.attack_controller.total_packets
- print(attack_args[0][0] + ' needs ' + str(duration) + ' seconds to generate ' + str(factor) + ' packets.')
- caller_function = inspect.stack()[1].function
- try:
- self.assertLessEqual(duration, time_limit)
- except self.failureException:
- Lib.rename_test_result_files(controller, caller_function, attack_sub_dir, test_sub_dir)
- raise
- if cleanup:
- Lib.clean_up(controller)
- else:
- Lib.rename_test_result_files(controller, caller_function, attack_sub_dir, test_sub_dir)
- def order_test(self, attack_args, seed=None, cleanup=True, pcap=Lib.test_pcap,
- flag_write_file=False, flag_recalculate_stats=False, flag_print_statistics=False,
- attack_sub_dir=True, test_sub_dir=True):
- """
- Checks if the result of an attack includes all packets in correct order.
- :param attack_args: A list of attacks with their attack parameters (as defined in Controller.process_attacks).
- :param seed: A random seed to keep random values static (care for count and order of random generation).
- :param cleanup: Clean up attack output after testing.
- :param pcap: The input pcap for the attack.
- :param flag_write_file: Writes the statistics to a file.
- :param flag_recalculate_stats: Forces the recalculation of statistics.
- :param flag_print_statistics: Prints the statistics on the terminal.
- :param attack_sub_dir: create sub-directory for each attack-class if True
- :param test_sub_dir: create sub-directory for each test-function/case if True
- """
- controller = Ctrl.Controller(pcap_file_path=pcap, do_extra_tests=False, non_verbose=True)
- controller.load_pcap_statistics(flag_write_file, flag_recalculate_stats, flag_print_statistics)
- controller.process_attacks(attack_args, [[seed]])
- caller_function = inspect.stack()[1].function
- try:
- path = controller.pcap_dest_path
- file = pcr.RawPcapReader(path)
- packet_a = file.read_packet()
- packet_b = file.read_packet()
- i = 0
- while packet_b is not None:
- time_a = packet_a[2][0:2]
- time_b = packet_b[2][0:2]
- if time_a[0] > time_b[0]:
- file.close()
- self.fail("Packet order incorrect at: " + str(i + 1) + "-" + str(i + 2) +
- ". Current time: " + str(time_a) + " Next time: " + str(time_b))
- elif time_a[0] == time_b[0]:
- if time_a[1] > time_b[1]:
- file.close()
- self.fail("Packet order incorrect at: " + str(i + 1) + "-" + str(i + 2) +
- ". Current time: " + str(time_a) + " Next time: " + str(time_b))
- packet_a = packet_b
- packet_b = file.read_packet()
- i += 1
- file.close()
- except self.failureException:
- Lib.rename_test_result_files(controller, caller_function, attack_sub_dir, test_sub_dir)
- raise
- if cleanup:
- Lib.clean_up(controller)
- else:
- Lib.rename_test_result_files(controller, caller_function, attack_sub_dir, test_sub_dir)
|