python安全编程-字典生成器

在渗透测试中，我们有时候会碰到拿到后台登录界面，而且没有其他漏洞可以利用，只能暴力破解用户名密码，现在直接使用弱口令的网站较少了，常见的字典一般是无法爆破成功，所以我们要针对网站内容来生成对应字典，进一步提升爆破成功率。

所以我们来写个小脚本来生成对应的字典

import exrex
import sys

web_white = ['com','cn','gov','www','edu','org']

def host_para(host):
    if '://' in host:
        host = host.split('://')[1].replace('/',' ')#将无用字符替换成空
    if '/' in host:
        host = host.replace('/','')
    return host
def dic_creat(hosts):
    web_dics = hosts.split('.')#将有用的东西去除，例如demo，webdic，放入字典生成的地方，生成字典
    #将核心的生成规则写入配置文件，方便以后使用
    #可以对生成密码进行过滤
    f_rule = open('ruler.ini','r',encoding='UTF-8')
    for i in f_rule:
        if '#' in  i[0]:
            rule = i
    f_pass_out = open('pass_0.txt','r')
    f_pass_out.close()
    for web_dic in web_dics:
        if web_dic not in web_white:
            f_pass =open('123.txt','r')#打开密码文件
            for dic_pass in f_pass:
                dics = list(exrex.generate( '{web_dic}[!@#]{dic_pass}'.format(web_dic = web_dic,dic_pass = dic_pass)))

                for dic in dics:
                    if len(dic) > 4:
                        f_pass_out =open('pass_1.txt','a+')
                        f_pass_out.write(dic ='\n')
                        f_pass_out.close()
                        print (dic)


#dic_creat(host_para('demo.123.com'))
if __name__ =='__main__':
    if len(sys.argv) == 2:
        dic_creat(host_para(sys.argv[1]))
        sys.exit(0)
    else:
        print('Usage: %s www.demo.com'%sys.argv[0])
        sys.exit(-1)

我们主要使用了exrex模块来生成字典，将所有可能都列举到字典中，首先先做一个根据网站域名来生成对应字典，并和常见的弱口令结合来生成对应的字典。

将域名中提取出来的内容进行过滤，将符号等无用的东西去掉，并将核心的生成规则写入配置文件，便于修改，后续再加入扩展功能。