-
Notifications
You must be signed in to change notification settings - Fork 0
/
make_token_true.py
73 lines (58 loc) · 2.01 KB
/
make_token_true.py
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
import txtlist as mklist
import csv
csv_name = "./XSS_data_all.csv"
before_data_list = mklist.Maketxt(csv_name)
normal_list = before_data_list[0]
XSS_list = before_data_list[1]
for i in range(len(normal_list)):
for j in range(len(normal_list[i][1])):
try:
if normal_list[i][1][j] == "=":
normal_list[i][1][j-1] = "OreVar"
elif normal_list[i][1][j] == "(":
if normal_list[i][1][j-2] == "=":
normal_list[i][1][j-1] = "OreDef"
else:
normal_list[i][1][j+1] = "OreVar"
except:
pass
for i in range(len(XSS_list)):
for j in range(len(XSS_list[i][1])):
try:
if XSS_list[i][1][j] == "=":
XSS_list[i][1][j-1] = "OreVar"
elif XSS_list[i][1][j] == "(":
if XSS_list[i][1][j-2] == "=":
XSS_list[i][1][j-1] = "OreDef"
else:
XSS_list[i][1][j+1] = "OreVar"
except:
pass
for i in range(len(XSS_list)):
txt = " ".join(XSS_list[i][1])
XSS_list[i][1] = txt
for i in range(len(normal_list)):
txt = " ".join(normal_list[i][1])
normal_list[i][1] = txt
print(XSS_list)
print(normal_list)
n_normal = len(normal_list)
n_XSS = len(XSS_list)
diff = n_normal - n_XSS
if diff < 0:
all_num = n_normal
else:
all_num = n_XSS
num = int(all_num * 0.8)
train_normal_list = normal_list[:num]
test_normal_list = normal_list[num:all_num]
train_XSS_list = XSS_list[:num]
test_XSS_list = XSS_list[num:all_num]
train_list = train_normal_list + train_XSS_list
test_list = test_normal_list + test_XSS_list
with open("./train/train_all_true.csv","w",encoding="utf-8",newline="") as tr:
tr_wirter = csv.writer(tr)
tr_wirter.writerows(train_list)
with open("./test/test_all_true.csv","w",encoding="utf-8",newline="") as te:
te_wirter = csv.writer(te)
te_wirter.writerows(test_list)