-
-
Notifications
You must be signed in to change notification settings - Fork 529
detect-it-easy analyzer, closes #1590 #2354
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Changes from 16 commits
11ae110
9fc11c6
475e4d2
39d84d0
04a303e
71e944f
35b4922
e763f7d
ae4fe4d
646e736
3a87aa6
55e03ca
ddad7a0
551da9b
530bb1d
ce95dea
3467275
f47dd88
bea64c1
ba7b288
14df652
File filter
Filter by extension
Conversations
Jump to
Diff view
Diff view
There are no files selected for viewing
| Original file line number | Diff line number | Diff line change |
|---|---|---|
|
|
@@ -45,3 +45,6 @@ coverage.xml | |
| *.cover | ||
| .hypothesis/ | ||
| /.env | ||
|
|
||
| # post run dev | ||
| integrations/malware_tools_analyzers/clamav/sigs | ||
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,60 @@ | ||
| import logging | ||
|
|
||
| from api_app.analyzers_manager.classes import DockerBasedAnalyzer, FileAnalyzer | ||
| from tests.mock_utils import MockUpResponse | ||
|
|
||
| logger = logging.getLogger(__name__) | ||
|
|
||
|
|
||
| class DetectItEasy(FileAnalyzer, DockerBasedAnalyzer): | ||
| name: str = "executable_analyzer" | ||
| url: str = "http://malware_tools_analyzers:4002/die" | ||
| # http request polling max number of tries | ||
| max_tries: int = 10 | ||
| # interval between http request polling (in secs) | ||
| poll_distance: int = 1 | ||
|
|
||
| def update(self): | ||
| pass | ||
|
|
||
| def run(self): | ||
| fname = str(self.filename).replace("/", "_").replace(" ", "_") | ||
| # get the file to send | ||
| binary = self.read_file_bytes() | ||
| args = [f"@{fname}", "--json"] | ||
| req_data = { | ||
| "args": args, | ||
| } | ||
| req_files = {fname: binary} | ||
| logger.info( | ||
| f"Running {self.analyzer_name} on {self.filename} with args: {args}" | ||
| ) | ||
| report = self._docker_run(req_data, req_files, analyzer_name=self.analyzer_name) | ||
| if not report: | ||
| self.report.errors.append("DIE does not support the file") | ||
mlodic marked this conversation as resolved.
Outdated
Show resolved
Hide resolved
|
||
| return {} | ||
| return report | ||
|
|
||
| @staticmethod | ||
| def mocked_docker_analyzer_get(*args, **kwargs): | ||
| return MockUpResponse( | ||
| { | ||
| "report": { | ||
| "arch": "NOEXEC", | ||
| "mode": "Unknown", | ||
| "type": "Unknown", | ||
| "detects": [ | ||
| { | ||
| "name": "Zip", | ||
| "type": "archive", | ||
| "string": "archive: Zip(2.0)[38.5%,1 file]", | ||
| "options": "38.5%,1 file", | ||
| "version": "2.0", | ||
| } | ||
| ], | ||
| "filetype": "Binary", | ||
| "endianess": "LE", | ||
| } | ||
| }, | ||
| 200, | ||
| ) | ||
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,117 @@ | ||
| from django.db import migrations | ||
| from django.db.models.fields.related_descriptors import ( | ||
| ForwardManyToOneDescriptor, | ||
| ForwardOneToOneDescriptor, | ||
| ManyToManyDescriptor, | ||
| ) | ||
|
|
||
| plugin = { | ||
| "python_module": { | ||
| "health_check_schedule": None, | ||
| "update_schedule": None, | ||
| "module": "detectiteasy.DetectItEasy", | ||
| "base_path": "api_app.analyzers_manager.file_analyzers", | ||
| }, | ||
| "name": "DetectItEasy", | ||
| "description": "[DetectItEasy](https://github.com/horsicq/Detect-It-Easy) is a program for determining types of files.", | ||
| "disabled": False, | ||
| "soft_time_limit": 10, | ||
| "routing_key": "default", | ||
| "health_check_status": True, | ||
| "type": "file", | ||
| "docker_based": True, | ||
| "maximum_tlp": "RED", | ||
| "observable_supported": [], | ||
| "supported_filetypes": [], | ||
| "run_hash": False, | ||
| "run_hash_type": "", | ||
| "not_supported_filetypes": [], | ||
| "model": "analyzers_manager.AnalyzerConfig", | ||
| } | ||
|
|
||
| params = [] | ||
|
||
|
|
||
| values = [] | ||
|
|
||
|
|
||
| def _get_real_obj(Model, field, value): | ||
| def _get_obj(Model, other_model, value): | ||
| if isinstance(value, dict): | ||
| real_vals = {} | ||
| for key, real_val in value.items(): | ||
| real_vals[key] = _get_real_obj(other_model, key, real_val) | ||
| value = other_model.objects.get_or_create(**real_vals)[0] | ||
| # it is just the primary key serialized | ||
| else: | ||
| if isinstance(value, int): | ||
| if Model.__name__ == "PluginConfig": | ||
| value = other_model.objects.get(name=plugin["name"]) | ||
| else: | ||
| value = other_model.objects.get(pk=value) | ||
| else: | ||
| value = other_model.objects.get(name=value) | ||
| return value | ||
|
|
||
| if ( | ||
| type(getattr(Model, field)) | ||
| in [ForwardManyToOneDescriptor, ForwardOneToOneDescriptor] | ||
| and value | ||
| ): | ||
| other_model = getattr(Model, field).get_queryset().model | ||
| value = _get_obj(Model, other_model, value) | ||
| elif type(getattr(Model, field)) in [ManyToManyDescriptor] and value: | ||
| other_model = getattr(Model, field).rel.model | ||
| value = [_get_obj(Model, other_model, val) for val in value] | ||
| return value | ||
|
|
||
|
|
||
| def _create_object(Model, data): | ||
| mtm, no_mtm = {}, {} | ||
| for field, value in data.items(): | ||
| value = _get_real_obj(Model, field, value) | ||
| if type(getattr(Model, field)) is ManyToManyDescriptor: | ||
| mtm[field] = value | ||
| else: | ||
| no_mtm[field] = value | ||
| try: | ||
| o = Model.objects.get(**no_mtm) | ||
| except Model.DoesNotExist: | ||
| o = Model(**no_mtm) | ||
| o.full_clean() | ||
| o.save() | ||
| for field, value in mtm.items(): | ||
| attribute = getattr(o, field) | ||
| if value is not None: | ||
| attribute.set(value) | ||
| return False | ||
| return True | ||
|
|
||
|
|
||
| def migrate(apps, schema_editor): | ||
| Parameter = apps.get_model("api_app", "Parameter") | ||
| PluginConfig = apps.get_model("api_app", "PluginConfig") | ||
| python_path = plugin.pop("model") | ||
| Model = apps.get_model(*python_path.split(".")) | ||
| if not Model.objects.filter(name=plugin["name"]).exists(): | ||
| exists = _create_object(Model, plugin) | ||
| if not exists: | ||
| for param in params: | ||
| _create_object(Parameter, param) | ||
| for value in values: | ||
| _create_object(PluginConfig, value) | ||
|
|
||
|
|
||
| def reverse_migrate(apps, schema_editor): | ||
| python_path = plugin.pop("model") | ||
| Model = apps.get_model(*python_path.split(".")) | ||
| Model.objects.get(name=plugin["name"]).delete() | ||
|
|
||
|
|
||
| class Migration(migrations.Migration): | ||
| atomic = False | ||
| dependencies = [ | ||
| ("api_app", "0062_alter_parameter_python_module"), | ||
| ("analyzers_manager", "0093_analyzer_config_ailtyposquatting"), | ||
| ] | ||
|
|
||
| operations = [migrations.RunPython(migrate, reverse_migrate)] | ||
Uh oh!
There was an error while loading. Please reload this page.