Method match

fakedns.py:581–623

See if the request matches any rules in the rule list by calling the match function of each rule in the list The rule checks two things before it continues so I imagine this is probably still fast

(self, query, addr)

    def match(self, query, addr):
        """
        See if the request matches any rules in the rule list by calling the
        match function of each rule in the list
        The rule checks two things before it continues so I imagine this is
        probably still fast
        """
        for rule in self.rule_list:
            result = rule.match(query.type, query.domain, addr)
            if result is not None:
                response_data = result

                # Return Nonefound if the rule says "none"
                if response_data.lower() == 'none':
                    return NONEFOUND(query).make_packet()

                response = CASE[query.type](query, response_data)

                print(">> Matched Request - " + query.domain.decode())
                return response.make_packet()

        # if we got here, we didn't match.
        # Forward a request that we didnt have a rule for to someone else

        # if the user said not to forward requests, and we are here, it's time to send a NONEFOUND
        if args.noforward:
            print(">> Don't Forward %s" % query.domain.decode())
            return NONEFOUND(query).make_packet()
        try:
            s = socket.socket(type=socket.SOCK_DGRAM)
            s.settimeout(3.0)
            addr = ('%s' % (args.dns), 53)
            s.sendto(query.data, addr)
            data = s.recv(1024)
            s.close()
            print("Unmatched Request " + query.domain.decode())
            return data
        except socket.error as e:
            # We shouldn't wind up here but if we do, don't drop the request
            # send the client *something*
            print(">> Error was handled by sending NONEFOUND")
            print(e)
            return NONEFOUND(query).make_packet()


# Convenience method for threading.