"""
This script makes it possible to use mitmproxy in scenarios where IP spoofing
has been used to redirect connections to mitmproxy. The way this works is that
we rely on either the TLS Server Name Indication (SNI) or the Host header of the
HTTP request. Of course, this is not foolproof - if an HTTPS connection comes
without SNI, we don't know the actual target and cannot construct a certificate
that looks valid. Similarly, if there's no Host header or a spoofed Host header,
we're out of luck as well. Using transparent mode is the better option most of
the time.

Usage:
    mitmproxy
        -p 443
        -s dns_spoofing.py
        # Used as the target location if neither SNI nor host header are present.
        --mode reverse:http://example.com/
        # To avoid auto rewriting of host header by the reverse proxy target.
        --set keep_host_header
    mitmdump
        -p 80
        --mode reverse:http://localhost:443/

    (Setting up a single proxy instance and using iptables to redirect to it
    works as well)
"""
import json
import re
from bubble_api import bubble_matchers, HEADER_BUBBLE_MATCHERS
from mitmproxy import ctx

# This regex extracts splits the host header into host and port.
# Handles the edge case of IPv6 addresses containing colons.
# https://bugzilla.mozilla.org/show_bug.cgi?id=45891
parse_host_header = re.compile(r"^(?P<host>[^:]+|\[.+\])(?::(?P<port>\d+))?$")


class Rerouter:
    @staticmethod
    def get_matcher_ids(flow, host):
        if host is None:
            return None

        remote_addr = flow.client_conn.address[0]
        host = host.decode('utf-8')
        matchers = bubble_matchers(remote_addr, flow, host)
        if not matchers:
            print("no matchers for remote_addr/host: "+remote_addr+'/'+host)
            return None
        matcher_ids = []
        for m in matchers:
            ctx.log.info('get_matcher_ids: checking for match of path='+flow.request.path+' against regex: '+m['regex'])
            if re.match(m['regex'], flow.request.path):
                ctx.log.info('get_matcher_ids: rule matched, adding rule: '+m['rule'])
                matcher_ids.append(m['uuid'])
        return matcher_ids

    def request(self, flow):
        if flow.client_conn.tls_established:
            flow.request.scheme = "https"
            sni = flow.client_conn.connection.get_servername()
            port = 443
        else:
            flow.request.scheme = "http"
            sni = None
            port = 80

        host_header = flow.request.host_header
        m = parse_host_header.match(host_header)
        if m:
            host_header = m.group("host").strip("[]")
            if m.group("port"):
                port = int(m.group("port"))

        # Determine if this request should be filtered
        if sni or host_header:
            matchers = self.get_matcher_ids(flow, sni or host_header)
            if matchers:
                ctx.log.info("dns_spoofing.request: found matchers: " + ' '.join(matchers))
                flow.request.headers[HEADER_BUBBLE_MATCHERS] = json.dumps(matchers)
            else:
                ctx.log.info('dns_spoofing.request: no rules returned, passing thru...')
        else:
            ctx.log.info('dns_spoofing.request: no sni/host found, not applying rules to path: ' + flow.request.path)

        flow.request.host_header = host_header
        flow.request.host = sni or host_header
        flow.request.port = port


addons = [Rerouter()]