Inject.py 文件源码

def initialize(self, options):
        '''Called if plugin is enabled, passed the options namespace'''
        self.options      = options
        self.html_src     = options.html_url
        self.js_src       = options.js_url
        self.rate_limit   = options.rate_limit
        self.count_limit  = options.count_limit
        self.per_domain   = options.per_domain
        self.black_ips    = options.black_ips
        self.white_ips    = options.white_ips
        self.match_str    = options.match_str
        self.html_payload = options.html_payload

        try:
            self.proxyip = get_if_addr(options.interface)
            if self.proxyip == "0.0.0.0":
                sys.exit("[-] Interface %s does not have an IP address" % options.interface)
        except Exception, e:
            sys.exit("[-] Error retrieving interface IP address: %s" % e)

        if self.white_ips:
            temp = []
            for ip in self.white_ips.split(','):
                temp.append(ip)
            self.white_ips = temp

        if self.black_ips:
            temp = []
            for ip in self.black_ips.split(','):
                temp.append(ip)
            self.black_ips = temp

        if self.options.preserve_cache:
            self.implements.remove("handleHeader")
            self.implements.remove("connectionMade")

        if options.html_file is not None:
            self.html_payload += options.html_file.read()

        self.ctable = {}
        self.dtable = {}
        self.count = 0
        self.mime = "text/html"
        print "[*] Inject plugin online"