OXIESEC PANEL

Name	Size	Modified	Perms
📁 ..	-	09/05/2025 09:34:01 AM	rwxr-xr-x
📄 __init__.py	137 bytes	05/08/2024 06:42:49 PM	rw-r--r--
📁 __pycache__	-	05/08/2024 06:42:49 PM	rwxr-xr-x
📄 query_type.py	1.29 KB	05/08/2024 06:42:49 PM	rw-r--r--
📄 resolver.py	2 KB	05/08/2024 06:42:49 PM	rw-r--r--

Editing: resolver.py

import random
import subprocess
import logging

from .query_type import QueryType

logger = logging.getLogger(__name__)

class Resolver:

def __init__(
        self,
        executable='dig',
        nameservers=None,
        additional_args=None,
        encoding='utf-8',
    ):
        """
        Stores some customisable options into this resolver instance
        """
        self.executable = executable
        self.nameservers = nameservers or []
        self.additional_args = additional_args or []
        self.encoding = encoding

@property
    def nameserver(self):
        """
        Returns a random nameserver we should query against
        """
        return random.choice(self.nameservers)

@staticmethod
    def _execute(args):
        """
        Calls out to subprocess with the passed in args

This method is normally mocked in tests
        """
        logger.info('Executing subprocess.check_output({})'.format(repr(args)))  # pragma: no cover
        return subprocess.check_output(args)  # pragma: no cover

def _args(self, domain, query_type):
        """
        Builds up the final arguments to pass into subprocess

dig @1.1.1.1 example.com A +short
        """
        yield self.executable

# Add in a random nameserver
        if self.nameservers:
            yield '@{}'.format(self.nameserver)

# The record we want to query
        yield domain

# Our query type
        yield query_type.name

# We only care about the result
        yield '+short'

# Add in any additional args
        yield from self.additional_args

def query(self, domain, query_type):
        """
        Queries the resolver for a specific domain and query type

Returns a list of records
        """
        domain = domain.lower()
        query_type = QueryType.get(query_type)

args = list(self._args(domain, query_type))
        output = self._execute(args).decode(self.encoding).strip()

return output.split('\n') if output else []