dirbruter.py 文件源码

def dir_bruter(self, target_url, word_queue, user_agent):
        results = {}
        session = requests.Session()
        session.mount(target_url.split(':', 1)[0], HTTPAdapter(max_retries=3))
        while not word_queue.empty():
            # attempt = word_queue.get()
            attempt_list = [word_queue.get()]
            for brute in attempt_list:
                headers = {"User-Agent": user_agent}
                request = session.get(target_url + brute, headers=headers, verify=False)
                if request.status_code == 200:
                    print("{i}     [{r}] => {u}".format(i=ctinfo, r=request.status_code, u=request.url))
                    logging.info("{i}     [{r}] => {u}".format(i=ctinfo, r=request.status_code, u=request.url))
                    results[request.url] = request.status_code
                elif request.status_code != 404:
                    # TODO: add a setting `only_save_200` or something like that, if no, save these results.
                    logging.error("{e}     {c} => {u}".format(e=cterr, c=request.status_code, u=request.url))
                    pass
        return results