revengai 2.74.0__py3-none-any.whl

revengai/models/pe_model.py

@@ -0,0 +1,215 @@
+# coding: utf-8
+
+"""
+    RevEng.AI API
+
+    RevEng.AI is Similarity Search Engine for executable binaries
+
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import pprint
+import re  # noqa: F401
+import json
+
+from pydantic import BaseModel, ConfigDict, StrictBool, StrictInt, StrictStr
+from typing import Any, ClassVar, Dict, List, Optional
+from revengai.models.code_signature_model import CodeSignatureModel
+from revengai.models.entrypoint_model import EntrypointModel
+from revengai.models.export_model import ExportModel
+from revengai.models.icon_model import IconModel
+from revengai.models.import_model import ImportModel
+from revengai.models.pdb_debug_model import PDBDebugModel
+from revengai.models.section_model import SectionModel
+from revengai.models.security_model import SecurityModel
+from revengai.models.timestamp_model import TimestampModel
+from typing import Optional, Set
+from typing_extensions import Self
+
+class PEModel(BaseModel):
+    """
+    PEModel
+    """ # noqa: E501
+    type: StrictStr
+    timestamps: Optional[TimestampModel]
+    architecture: StrictStr
+    checksum: StrictInt
+    image_base: StrictInt
+    security: Optional[SecurityModel]
+    version_info: Optional[Dict[str, Any]]
+    debug_info: Optional[PDBDebugModel]
+    number_of_resources: Optional[StrictInt]
+    entry_point: Optional[EntrypointModel]
+    signature: Optional[CodeSignatureModel]
+    dotnet: StrictBool
+    debug_stripped: StrictBool
+    import_hash: StrictStr
+    export_hash: StrictStr
+    rich_header_hash: StrictStr
+    sections: Optional[SectionModel]
+    imports: Optional[ImportModel]
+    exports: Optional[ExportModel]
+    icon_data: Optional[IconModel]
+    __properties: ClassVar[List[str]] = ["type", "timestamps", "architecture", "checksum", "image_base", "security", "version_info", "debug_info", "number_of_resources", "entry_point", "signature", "dotnet", "debug_stripped", "import_hash", "export_hash", "rich_header_hash", "sections", "imports", "exports", "icon_data"]
+
+    model_config = ConfigDict(
+        populate_by_name=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
+
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of PEModel from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Return the dictionary representation of the model using alias.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        """
+        excluded_fields: Set[str] = set([
+        ])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        # override the default output from pydantic by calling `to_dict()` of timestamps
+        if self.timestamps:
+            _dict['timestamps'] = self.timestamps.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of security
+        if self.security:
+            _dict['security'] = self.security.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of debug_info
+        if self.debug_info:
+            _dict['debug_info'] = self.debug_info.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of entry_point
+        if self.entry_point:
+            _dict['entry_point'] = self.entry_point.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of signature
+        if self.signature:
+            _dict['signature'] = self.signature.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of sections
+        if self.sections:
+            _dict['sections'] = self.sections.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of imports
+        if self.imports:
+            _dict['imports'] = self.imports.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of exports
+        if self.exports:
+            _dict['exports'] = self.exports.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of icon_data
+        if self.icon_data:
+            _dict['icon_data'] = self.icon_data.to_dict()
+        # set to None if timestamps (nullable) is None
+        # and model_fields_set contains the field
+        if self.timestamps is None and "timestamps" in self.model_fields_set:
+            _dict['timestamps'] = None
+
+        # set to None if security (nullable) is None
+        # and model_fields_set contains the field
+        if self.security is None and "security" in self.model_fields_set:
+            _dict['security'] = None
+
+        # set to None if version_info (nullable) is None
+        # and model_fields_set contains the field
+        if self.version_info is None and "version_info" in self.model_fields_set:
+            _dict['version_info'] = None
+
+        # set to None if debug_info (nullable) is None
+        # and model_fields_set contains the field
+        if self.debug_info is None and "debug_info" in self.model_fields_set:
+            _dict['debug_info'] = None
+
+        # set to None if number_of_resources (nullable) is None
+        # and model_fields_set contains the field
+        if self.number_of_resources is None and "number_of_resources" in self.model_fields_set:
+            _dict['number_of_resources'] = None
+
+        # set to None if entry_point (nullable) is None
+        # and model_fields_set contains the field
+        if self.entry_point is None and "entry_point" in self.model_fields_set:
+            _dict['entry_point'] = None
+
+        # set to None if signature (nullable) is None
+        # and model_fields_set contains the field
+        if self.signature is None and "signature" in self.model_fields_set:
+            _dict['signature'] = None
+
+        # set to None if sections (nullable) is None
+        # and model_fields_set contains the field
+        if self.sections is None and "sections" in self.model_fields_set:
+            _dict['sections'] = None
+
+        # set to None if imports (nullable) is None
+        # and model_fields_set contains the field
+        if self.imports is None and "imports" in self.model_fields_set:
+            _dict['imports'] = None
+
+        # set to None if exports (nullable) is None
+        # and model_fields_set contains the field
+        if self.exports is None and "exports" in self.model_fields_set:
+            _dict['exports'] = None
+
+        # set to None if icon_data (nullable) is None
+        # and model_fields_set contains the field
+        if self.icon_data is None and "icon_data" in self.model_fields_set:
+            _dict['icon_data'] = None
+
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of PEModel from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "type": obj.get("type"),
+            "timestamps": TimestampModel.from_dict(obj["timestamps"]) if obj.get("timestamps") is not None else None,
+            "architecture": obj.get("architecture"),
+            "checksum": obj.get("checksum"),
+            "image_base": obj.get("image_base"),
+            "security": SecurityModel.from_dict(obj["security"]) if obj.get("security") is not None else None,
+            "version_info": obj.get("version_info"),
+            "debug_info": PDBDebugModel.from_dict(obj["debug_info"]) if obj.get("debug_info") is not None else None,
+            "number_of_resources": obj.get("number_of_resources"),
+            "entry_point": EntrypointModel.from_dict(obj["entry_point"]) if obj.get("entry_point") is not None else None,
+            "signature": CodeSignatureModel.from_dict(obj["signature"]) if obj.get("signature") is not None else None,
+            "dotnet": obj.get("dotnet"),
+            "debug_stripped": obj.get("debug_stripped"),
+            "import_hash": obj.get("import_hash"),
+            "export_hash": obj.get("export_hash"),
+            "rich_header_hash": obj.get("rich_header_hash"),
+            "sections": SectionModel.from_dict(obj["sections"]) if obj.get("sections") is not None else None,
+            "imports": ImportModel.from_dict(obj["imports"]) if obj.get("imports") is not None else None,
+            "exports": ExportModel.from_dict(obj["exports"]) if obj.get("exports") is not None else None,
+            "icon_data": IconModel.from_dict(obj["icon_data"]) if obj.get("icon_data") is not None else None
+        })
+        return _obj
+
+

revengai/models/platform.py

@@ -0,0 +1,36 @@
+# coding: utf-8
+
+"""
+    RevEng.AI API
+
+    RevEng.AI is Similarity Search Engine for executable binaries
+
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import json
+from enum import Enum
+from typing_extensions import Self
+
+
+class Platform(str, Enum):
+    """
+    Platform
+    """
+
+    """
+    allowed enum values
+    """
+    LINUX = 'linux'
+    WINDOWS = 'windows'
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Self:
+        """Create an instance of Platform from a JSON string"""
+        return cls(json.loads(json_str))
+
+

revengai/models/process.py

@@ -0,0 +1,103 @@
+# coding: utf-8
+
+"""
+    RevEng.AI API
+
+    RevEng.AI is Similarity Search Engine for executable binaries
+
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import pprint
+import re  # noqa: F401
+import json
+
+from pydantic import BaseModel, ConfigDict, StrictFloat, StrictInt, StrictStr
+from typing import Any, ClassVar, Dict, List, Optional, Union
+from typing import Optional, Set
+from typing_extensions import Self
+
+class Process(BaseModel):
+    """
+    Process
+    """ # noqa: E501
+    pid: StrictInt
+    procname: StrictStr
+    executable_name: StrictStr
+    args: List[StrictStr]
+    ts_from: Union[StrictFloat, StrictInt]
+    ts_to: Optional[Union[StrictFloat, StrictInt]]
+    children: List[Any]
+    __properties: ClassVar[List[str]] = ["pid", "procname", "executable_name", "args", "ts_from", "ts_to", "children"]
+
+    model_config = ConfigDict(
+        populate_by_name=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
+
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of Process from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Return the dictionary representation of the model using alias.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        """
+        excluded_fields: Set[str] = set([
+        ])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        # set to None if ts_to (nullable) is None
+        # and model_fields_set contains the field
+        if self.ts_to is None and "ts_to" in self.model_fields_set:
+            _dict['ts_to'] = None
+
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of Process from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "pid": obj.get("pid"),
+            "procname": obj.get("procname"),
+            "executable_name": obj.get("executable_name"),
+            "args": obj.get("args"),
+            "ts_from": obj.get("ts_from"),
+            "ts_to": obj.get("ts_to"),
+            "children": obj.get("children")
+        })
+        return _obj
+
+

revengai/models/process_dump.py

@@ -0,0 +1,96 @@
+# coding: utf-8
+
+"""
+    RevEng.AI API
+
+    RevEng.AI is Similarity Search Engine for executable binaries
+
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import pprint
+import re  # noqa: F401
+import json
+
+from pydantic import BaseModel, ConfigDict, StrictInt, StrictStr
+from typing import Any, ClassVar, Dict, List
+from revengai.models.process_dump_metadata import ProcessDumpMetadata
+from typing import Optional, Set
+from typing_extensions import Self
+
+class ProcessDump(BaseModel):
+    """
+    ProcessDump
+    """ # noqa: E501
+    base_address: StrictInt
+    actual_filename: StrictStr
+    filename_friendly: StrictStr
+    extended_metadata: ProcessDumpMetadata
+    __properties: ClassVar[List[str]] = ["base_address", "actual_filename", "filename_friendly", "extended_metadata"]
+
+    model_config = ConfigDict(
+        populate_by_name=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
+
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of ProcessDump from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Return the dictionary representation of the model using alias.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        """
+        excluded_fields: Set[str] = set([
+        ])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        # override the default output from pydantic by calling `to_dict()` of extended_metadata
+        if self.extended_metadata:
+            _dict['extended_metadata'] = self.extended_metadata.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of ProcessDump from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "base_address": obj.get("base_address"),
+            "actual_filename": obj.get("actual_filename"),
+            "filename_friendly": obj.get("filename_friendly"),
+            "extended_metadata": ProcessDumpMetadata.from_dict(obj["extended_metadata"]) if obj.get("extended_metadata") is not None else None
+        })
+        return _obj
+
+

revengai/models/process_dump_metadata.py

@@ -0,0 +1,90 @@
+# coding: utf-8
+
+"""
+    RevEng.AI API
+
+    RevEng.AI is Similarity Search Engine for executable binaries
+
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import pprint
+import re  # noqa: F401
+import json
+
+from pydantic import BaseModel, ConfigDict, StrictInt, StrictStr
+from typing import Any, ClassVar, Dict, List
+from typing import Optional, Set
+from typing_extensions import Self
+
+class ProcessDumpMetadata(BaseModel):
+    """
+    ProcessDumpMetadata
+    """ # noqa: E501
+    sha256: StrictStr
+    type: StrictStr
+    size: StrictInt
+    __properties: ClassVar[List[str]] = ["sha256", "type", "size"]
+
+    model_config = ConfigDict(
+        populate_by_name=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
+
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of ProcessDumpMetadata from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Return the dictionary representation of the model using alias.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        """
+        excluded_fields: Set[str] = set([
+        ])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of ProcessDumpMetadata from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "sha256": obj.get("sha256"),
+            "type": obj.get("type"),
+            "size": obj.get("size")
+        })
+        return _obj
+
+

revengai/models/process_dumps.py

@@ -0,0 +1,92 @@
+# coding: utf-8
+
+"""
+    RevEng.AI API
+
+    RevEng.AI is Similarity Search Engine for executable binaries
+
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import pprint
+import re  # noqa: F401
+import json
+
+from pydantic import BaseModel, ConfigDict, StrictBool
+from typing import Any, ClassVar, Dict, List
+from revengai.models.process_dumps_data import ProcessDumpsData
+from typing import Optional, Set
+from typing_extensions import Self
+
+class ProcessDumps(BaseModel):
+    """
+    ProcessDumps
+    """ # noqa: E501
+    success: StrictBool
+    data: ProcessDumpsData
+    __properties: ClassVar[List[str]] = ["success", "data"]
+
+    model_config = ConfigDict(
+        populate_by_name=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
+
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of ProcessDumps from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Return the dictionary representation of the model using alias.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        """
+        excluded_fields: Set[str] = set([
+        ])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        # override the default output from pydantic by calling `to_dict()` of data
+        if self.data:
+            _dict['data'] = self.data.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of ProcessDumps from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "success": obj.get("success"),
+            "data": ProcessDumpsData.from_dict(obj["data"]) if obj.get("data") is not None else None
+        })
+        return _obj
+
+