Folder removed

fuzzer/fuzz_payloads.py

@@ -0,0 +1,80 @@
+from typing import List, Dict
+from boofuzz import s_group, s_static
+from encodings_helper import Encoder, EncodingTypes
+from configuration_manager import ConfigurationManager
+
+
+class FuzzPayloads:
+    payloads: Dict[str, List[str]] = {}
+    CUSTOM_PAYLOADS_KEY = "custom"
+
+    @staticmethod
+    def add_payload_to_list(line, directory_name):
+        if directory_name not in FuzzPayloads.payloads:
+            FuzzPayloads.payloads[directory_name] = []
+
+        if line not in FuzzPayloads.payloads[directory_name]:
+            FuzzPayloads.payloads[directory_name].append(line)
+
+    @staticmethod
+    def _get_payloads_using_directory_names(directory_names: List[str]) -> List[str]:
+        directory_names.append(FuzzPayloads.CUSTOM_PAYLOADS_KEY)    # Always add custom payloads into any payloads set
+        payloads: List[str] = []
+        for directory_name in directory_names:
+            if directory_name in FuzzPayloads.payloads:
+                for line in FuzzPayloads.payloads[directory_name]:
+                    payloads.append(line)
+        return list(set(payloads))          # Remove duplicities
+
+    @staticmethod
+    def get_all_payloads():
+        return FuzzPayloads._get_payloads_using_directory_names(list(FuzzPayloads.payloads.keys()))
+
+    @staticmethod
+    def _get_specific_type_payloads(payload_folders):
+        return FuzzPayloads._get_payloads_using_directory_names(payload_folders) if payload_folders else FuzzPayloads.get_all_payloads()
+
+    @staticmethod
+    def get_string_payloads():
+        payload_folders = ConfigurationManager.get_payloads_folders_for_string_json_primitive()
+        return FuzzPayloads._get_specific_type_payloads(payload_folders)
+
+    @staticmethod
+    def get_number_payloads():
+        payload_folders = ConfigurationManager.get_payloads_folders_for_number_json_primitive()
+        return FuzzPayloads._get_specific_type_payloads(payload_folders)
+
+    @staticmethod
+    def get_boolean_payloads():
+        payload_folders = ConfigurationManager.get_payloads_folders_for_boolean_json_primitive()
+        return FuzzPayloads._get_specific_type_payloads(payload_folders)
+
+
+def s_http_general(value, payloads, fuzzable=True, encoding: EncodingTypes = EncodingTypes.ascii, name=None, add_quotation_marks=False):
+    # Encode all payloads
+    encoded_payloads: List[bytes] = []
+    for payload in payloads:
+        encoded = Encoder.encode_string(payload, encoding)
+        if add_quotation_marks:
+            encoded = Encoder.get_ascii_encoded_quotation_mark() + encoded + Encoder.get_ascii_encoded_quotation_mark()
+        encoded_payloads.append(encoded)
+
+    # Encode default value
+    default_value = Encoder.encode_string(value, encoding)
+    if fuzzable:
+        # noinspection PyTypeChecker
+        s_group(name, encoded_payloads, default_value)
+    else:
+        s_static(default_value)
+
+
+def s_http_string(value, fuzzable=True, encoding: EncodingTypes = EncodingTypes.ascii, name=None):
+    s_http_general(value, FuzzPayloads.get_string_payloads(), fuzzable, encoding, name)
+
+
+def s_http_number(value, fuzzable=True, encoding: EncodingTypes = EncodingTypes.ascii, name=None, add_quotation_marks=False):
+    s_http_general(value, FuzzPayloads.get_number_payloads(), fuzzable, encoding, name, add_quotation_marks)
+
+
+def s_http_boolean(value, fuzzable=True, encoding: EncodingTypes = EncodingTypes.ascii, name=None, add_quotation_marks=False):
+    s_http_general(value, FuzzPayloads.get_boolean_payloads(), fuzzable, encoding, name, add_quotation_marks)