浅谈扫描器之插件篇 – 作者:dawner-安全小百科

在我们试图构建漏扫系统的时候，调整好插件的配置和格式，能更有效地帮助我们进行漏洞探测，以及提升对bug的进行fuzz的高效性。

在本文中，我们将简单谈谈插件相关的内容。下面的示例代码依旧会沿用python，但求行文精炼不赘言。

插件的基本格式

入口函数

def run():
    print 'I am the bone of my sword'

运行这类插件时，加载插件的入口函数run，就可以直接运行插件。

类 + 入口函数

class BaseVerify:

    def __init__(self, url):
        self.url = url
    
    def run(self):
        print 'I am the bone of my sword'

对于这种插件，在我们获取到漏洞库文件路径后，需要对服务类型进行匹配，最后再进行插件调用。此后，我们会获得BaseVerify类的实例，再引用里面的入口函数。

调用方式：

test = BaseVerify('http://www.baidu.com')
test.run()

入口函数 + 验证函数

def assign(service, arg):
    if service != "wordpress":
        return
    else:
        r = urlparse.urlparse(arg)
        return True, r.netloc
        
def audit(arg):
    print 'I am the bone of my sword'

这里也可以在类中引入函数，不过此处关键点在于，同时也使用了验证函数。

这样做的好处在于，即使不做插件目录分类，也能进行精准扫描，不至于在验证漏洞时处耗时过多。

不过顺便提一句，即使只运行了验证service类型的代码，在加载大量插件的情况下，也是会消耗一定的资源的。

调用方式(仅做参考)：

try:
    audit( assign('www', 'http://www.baidu.com')[1] )
except:
    pass

关于插件注释

在每个文件中，建议通过类属性或者直接头部注释，对每个插件进行细节标识。不然的话，他人在复现和修改的时候，很难对代码量较大的内容进行阅读调整。当然，对于某些故意加密的插件，那就另当别论了。

插件的加载

_import_

plugin_path = 'plugin_dir.plugin_name'
try:
    plugin = __import__(plugin_path, fromlist=[plugin_path])
    test = getattr(plugin,'audit')
    test('http://www.baidu.com')
except Exception,e:
    print e

这里在获取某个插件的路径后，可转换为__import__可识别的路径格式，然后再对其入口函数audit进行引用。

importlib

from importlib import import_module
plugin_path = 'plugin_dir.plugin_name'
split_dot = plugin_path.rindex('.')
module, name = plugin_path[:split_dot], plugin_path[split_dot+1:]
mod = import_module(module)
try:
    test = getattr(mod, name)
    test.audit('http://www.baidu.com')
except:
    pass

注意，这里的plugin_path如果不含’.’的话，可用下面的形式：

from importlib import import_module
plugin_path = 'plugin_name'
test = import_module(plugin_path)
test.audit('http://www.baidu.com')

import

import sys
sys.path.append('plugin_dir/')
#加入系统路径plugin_dir
import plugin_name
plugin_name.audit('http://www.baidu.com')

import imp
#这里需要正常路径名
test = imp.load_source('audit', 'plugin_dir/plugin_name.py')
test.audit('http://www.baidu.com')

关于导入细节

笔者还见过某框架，除了导入必要的核心库文件以外，还把所有分类插件里的验证、运行等函数，也在主文件头部一股脑导入的。

也就是说，在每次运行框架之前，就算只是-v看版本，也会预载入所有内容。

而在某些框架脚本，在每次运行前会自动下载一个巨大的封装库（作者diy的），而不是把它放在requirement文件里。

也不是说这样一定不好，不过个人窃以为，如果想要尽可能优化框架的效率，还是不太推荐大家这么做。

插件的存储

临时加载

在一次性加载插件不算多的时候，我们是可以这么做的，也不会太影响效率。比如metasploit就可以选择是否启用postgresql数据库。如果插件都放在一个目录下，进行文件遍历即可，大概可以这样写：

vuln_dir = 'plugin_dir/*'
vuln_paths = [f.replace('/','.') for f in glob.glob(vuln_dir)]
for vuln_path in vuln_paths:
    #vuln_path == 'plugin_dir.plugin_name'
    #下面省略
    ...
    ...