0x00 起因
接触网安快一年了，爱收集一些字典啊敏感目录文件啊什么的。收集多了难免有重复，并且有的也需要合并使用方便。自己就写了一个小小的脚步用来完成这个目的。
0x01 代码
第一次发表文章，希望大家多提意见。抱拳了！
代码我就贴出来了
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
|
import osimport time
import argparse
import optparse
def Time(n,p):
pat_time = p-n
minute = pat_time/60
hour = minute/60
print(int(hour),'小时',int(minute%60),'分钟',int(pat_time%60),'秒')
def __O__(path): #遍历目录，包括目录里的子目录
f = os.listdir(path) #查看路径下所有文件及目录
D = open('D.txt','w')
F = open('F.txt','w')
for i in f:
fil = os.path.join(path,i) #dir是绝对路径
if os.path.isdir(fil): #如果fil是目录加入到D里
D.write(fil)
__O__(fil) #继续执行当前函数
if os.path.isfile(fil): #如果是文件，加入到F里
F.write(fil)
D.close()
F.close()
def type_txt(file,path): #传入路径，文件名（列表）
try:
n = time.time()
print(' 准备执行!')
path_list = [] #存放需要打开的文件
di = {}
date_list = []
dict_li = ['new_php.txt', 'new_asp.txt', 'new_aspx.txt', 'new_jsp.txt', 'new_dir.txt', 'new_mdb.txt',] #合并的字典列表
for i in dict_li:
di = []
for h in file: #合并给的路径及文件名
fn = os.path.join(path, h)
path_list.append(fn)
for j in path_list: #循环需要打开的文件列表
print(' 文件为：',j)
with open(j, 'r') as f:
date = f.readlines()
print(' -t 当前执行文件为： ' + j,'数量：',len(date))
for i in date: #读取每一行
af, aq = os.path.splitext(i) #分裂成路径及后缀名
aq = aq.lower()
if i != '\n': #如果没有换行符会加上
if '\n' not in i:
i = i + '\n'
aq+='\n'
if aq == '.php\n' and i not in di['new_php.txt']: #判读后缀名并通过后缀名放到相应的字典中，同时判读字典对应的文件列表中是否存在，如果是False则加入进去
di['new_php.txt'].append(i)
elif aq == '.asp\n' and i not in di['new_asp.txt']:
di['new_asp.txt'].append(i)
elif aq == '.aspx\n' and i not in di['new_aspx.txt']:
di['new_aspx.txt'].append(i)
elif aq == '.jsp\n' and i not in di['new_jsp.txt'] :
di['new_jsp.txt'].append(i)
elif (aq == ('.ini\n' or '.xml\n' or '.cgi\n' or '.log\n' or '.db\n' or '.mdb\n' or '.sql\n' or '.xml\n' or '.htm\n' or '.html\n' or '.bak\n' or '.dbf\n' or '.zip\n' or '.rb\n' or '.js\n' or '.css\n' or '.html?\n')) and i not in di['new_mdb.txt']:
di['new_mdb.txt'].append(i)
elif i not in di['new_dir.txt'] :
di['new_dir.txt'].append(i)
f.close()
except (UnicodeDecodeError) as e:
print("字典文件中有错误请检查是否是字典",e)
di1,fi = os.path.split(path)
print(' 分类完毕......准备写入')
for key in di: #写入到文件中
o = open(key,'w')
for i in di[key]:
o.write(i)
Time(n, time.time())
def dir_txt(path): #传进去路径，返回来的是路径下的后缀为txt的文件。如果没有则是当前路径。
dir_txt = []
a = os.listdir(path)
for i in a:
dd, ff = os.path.splitext(i)
if ff == '.txt':
dir_txt.append(i)
return dir_txt
def password(file,path):
try:
path_list = []
password_list = []
for i in file:
pathl = os.path.join(path,i)
path_list.append(pathl)
print(" 开始去除重复.....")
for i in path_list: #重复每一个给定的密码文件
F = open(i,'r')
txt = F.readlines()
F.close
for j in txt: #读取每一个密码文件，并且判读是否存在于password_list列表，如不存在则放进列表
if '\n' not in j:
j+='\n'
if j not in password_list:
password_list.append(j)
print(" 完成，并且准备写入。")
file = open('Password.txt','w') #打开即将写入的Password.txt，把passowrd_list列表数据全部写入进去
for i in password_list:
file.write(i)
file.close
print(" 运行完毕。")
except :
print(" 请检查文件是否存在，路径是否正确。")
def Optparse():
try:
parser = argparse.ArgumentParser(
"""把字典分类：new_php.txt,new_asp.txt,new_aspx.txt,new_jsp.txt,new_mdb.txt,new_dir.
-t -path C:\\x\\x\*
-t -path C:\\x\\x -file xxx.txt,xx.txt """)
group = parser.add_mutually_exclusive_group()
group.add_argument('-p',action='store_true',help='密码合并') #-p 和-t参数是互相排斥的，并不能一起使用。
group.add_argument('-t',action='store_true',help='后台目录字典合并',)
parser.add_argument('-file', type=str, help='输入的文件',required=False)
parser.add_argument('-path',type=str,help='你所指定的目录',required=False)
args = parser.parse_args()
if args.t: #判读t是否真
if args.file == None:
path,file = os.path.split(args.path)
dir__txt = dir_txt(path)
type_txt(dir__txt,path)
elif args.file != None:
list_file = args.file.split(',')
type_txt(list_file,args.path)
elif args.p: #判读p是否真
if args.file == None:
path, file = os.path.split(args.path)
dir__txt = dir_txt(path)
password(dir__txt,path)
elif args.file != None:
list_file = args.file.split(',')
password(list_file,args.path)
else:
print(" 错误请检查输入。")
except (KeyboardInterrupt):
print(' 以退出')
|
以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持服务器之家。
原文链接：https://bbs.ichunqiu.com/thread-49190-1-1.html