使用Python开发的POC多线程批量执行小框架

因为代码量非常少，所以就叫“小框架”吧。

接口非常简陋，但是好处是适配POC脚本的时候很灵活，兼容性高，不需要任何研究成本。

简单来说，你按照自己的想法和习惯开发一个POC验证程序，它只要做到三点，即：

1. 接收一个目标地址字符串作为输入参数。

2. 定义一个名称为 "exploit" 的函数供小框架来调用。

3. 当验证成功时返回一个自定义的字符串作为标志。

就完全可以满足使用小框架的条件了。

针对以上条件再说一下细节。

小框架负责将字符串参数传递给POC程序，逻辑上讲最少要传递目标地址，想再接一些其他参数也完全可以，只要你的POC程序可以正确解析和处理就可以，完全由你自己来控制。

小框架在读取POC程序之后会调用名为 "exploit" 的函数，并获取返回结果，就好比操作系统载入一个PE程序之后会调用 "main()" 函数的道理一样，至于这个函数里面又调用了其他哪些函数，就完全是由你自己来自由定制了。

POC程序，或者说 "exploit" 函数验证成功的时候返回一个表示成功的自定义字符串，这个不必解释了，那么验证失败的时候呢？当然也可以返回另一个标志字符串，但是我个人一般习惯直接丢弃验证失败的目标结果不做记录，这个看个人需求了，也是自由定制的。

小框架使用Python来开发，Python 2 和 3 无缝兼容执行。

上代码：

#!/usr/bin/env python

# -*- coding: utf-8 -*-

import argparse

import importlib

import sys

import threadpool

def main():

    parser = argparse.ArgumentParser(description="Multithreading Batch Processing Framework")

    parser.add_argument("-s", "--script", type=open, required=True, help="script file")

    parser.add_argument("-i", "--input", type=open, required=True, help="input file")

    parser.add_argument("-t", "--thread", default=1, type=int, help="number of threads")

    try:

        args = parser.parse_args()

    except IOError as e:

        print(e)

        sys.exit(1)

    poolsize = args.thread

    some_callable = get_some_callable(args.script)

    list_of_args = get_list_of_args(args.input)

    global log

    try:

        log = open("log_" + args.input.name, "w")

    except IOError as e:

        print(e)

        sys.exit(1)

    pool = threadpool.ThreadPool(poolsize)

    reqs = threadpool.makeRequests(some_callable, list_of_args, callback)

    [pool.putRequest(req) for req in reqs]

    pool.wait()

    pool.dismissWorkers(poolsize)

    pool.joinAllDismissedWorkers()

    log.close()

def get_some_callable(script):

    try:

        script = importlib.import_module(script.name.split(".")[0])

        some_callable = script.exploit

    except (AttributeError, ImportError) as e:

        print(e)

        sys.exit(1)

    else:

        return some_callable

def get_list_of_args(input):

    list_of_args = list()

    list_of_temp = input.readlines()

    for i in list_of_temp:

        list_of_args.append(i.strip())

    return list_of_args

def callback(request, result):

    if result:

        log.write(result + "\n")

        print(result)

if __name__ == "__main__":

    main()