ReconAD_bof.s1.py

from typing import List, Tuple

from outflank_stage1.task.base_bof_task import BaseBOFTask
from outflank_stage1.task.enums import BOFArgumentEncoding


class ReconADBOF(BaseBOFTask):
    def __init__(self):
        super().__init__("ReconAD")

        self.parser.description = "Use ADSI to query Active Directory objects and attributes."
        self.parser.epilog = (
            "Use Active Directory Service Interfaces (ADSI) to query for AD objects and corresponding attributes.\n"
            "Example usage:\n"
            "  ReconAD (&(objectClass=user)(objectCategory=person)(sAMAccountName=*admin*)) --attributes "
            "displayName,sAMAccountName --count 10"
        )

        self.parser.add_argument("filter", help="Custom LDAP filter.")
        self.parser.add_argument("--attributes", default="-all", help="Comma separated LDAP attributes (default all).")
        self.parser.add_argument("--count", default=0, help="Limit the results (default all).", type=int)
        self.parser.add_argument("--usegc", action="store_true", help="Use global catalog.")
        self.parser.add_argument("--server", default="-noserver", help="Domain or server (Domain | Server [:port]).")

    def _encode_arguments_bof(self, arguments: List[str]) -> List[Tuple[BOFArgumentEncoding, str]]:
        parser_arguments = self.parser.parse_args(arguments)

        return [
            (BOFArgumentEncoding.WSTR, "custom"),
            (BOFArgumentEncoding.WSTR, parser_arguments.filter),
            (BOFArgumentEncoding.WSTR, parser_arguments.attributes),
            (BOFArgumentEncoding.INT, parser_arguments.count),
            (BOFArgumentEncoding.INT, parser_arguments.usegc),
            (BOFArgumentEncoding.WSTR, parser_arguments.server),
        ]