Source code for proxy.plugin.custom_dns_resolver
# -*- coding: utf-8 -*-
"""
proxy.py
~~~~~~~~
⚡⚡⚡ Fast, Lightweight, Pluggable, TLS interception capable proxy server focused on
Network monitoring, controls & Application development, testing, debugging.
:copyright: (c) 2013-present by Abhinav Singh and contributors.
:license: BSD, see LICENSE for more details.
.. spelling::
dns
"""
import socket
from typing import Tuple, Optional
from ..http.proxy import HttpProxyBasePlugin
from ..common.types import HostPort
[docs]class CustomDnsResolverPlugin(HttpProxyBasePlugin):
"""This plugin demonstrate how to use your own custom DNS resolver."""
[docs] def resolve_dns(self, host: str, port: int) -> Tuple[Optional[str], Optional[HostPort]]:
"""Here we are using in-built python resolver for demonstration.
Ideally you would like to query your custom DNS server or even
use :term:`DoH` to make real sense out of this plugin.
The second parameter returned is None. Return a 2-tuple to
configure underlying interface to use for connection to the
upstream server.
"""
try:
return socket.getaddrinfo(host, port, proto=socket.IPPROTO_TCP)[0][4][0], None
except socket.gaierror:
# Ideally we can also thrown HttpRequestRejected or HttpProtocolException here
# Returning None simply fallback to core generated exceptions.
return None, None