ReconADComputers_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 ReconADComputersBOF(BaseBOFTask):
    def __init__(self):
        super().__init__("ReconAD-Computers", base_binary_name="ReconAD")

        self.parser.description = "Use ADSI to query Active Directory computer objects and attributes."
        self.parser.epilog = (
            "Use Active Directory Service Interfaces (ADSI) to query computer objects and corresponding attributes."
            "Example usage:\n"
            "   ReconAD-Computers *dc* --usegc\n"
            "   ReconAD-Computers *srv* --attributes name,operatingSystemVersion --count 20\n"
        )

        self.parser.add_argument("computername", help="The computer(s) to search for.")
        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, "computers"),
            (BOFArgumentEncoding.WSTR, parser_arguments.computername),
            (BOFArgumentEncoding.WSTR, parser_arguments.attributes),
            (BOFArgumentEncoding.INT, parser_arguments.count),
            (BOFArgumentEncoding.INT, parser_arguments.usegc),
            (BOFArgumentEncoding.WSTR, parser_arguments.server),
        ]