Exemple #1
0
parse.add_argument('-o',
                   '--sta_file',
                   type=str,
                   default='./example/statistics.pkl',
                   help="file saving the final statistics (.pkl)")

arg = parse.parse_args()

m = Manipulator(arg.mal_pcap, arg.mimic_set, arg.normalizer, arg.init_pcap)

max_iter, particle_num, local_grp_size = 3, 6, 3
# max_iter,particle_num,local_grp_size = 4,8,4
# max_iter,particle_num,local_grp_size = 5,10,5
# max_iter,particle_num,local_grp_size = 3,10,5

m.change_particle_params(w=0.7298, c1=1.49618, c2=1.49618)
m.change_pso_params(max_iter=max_iter,
                    particle_num=particle_num,
                    grp_size=local_grp_size)
m.change_manipulator_params(grp_size=100,
                            min_time_extend=3.,
                            max_time_extend=6.,
                            max_cft_pkt=1,
                            max_crafted_pkt_prob=0.01)

# m.save_configurations('./configurations.txt')

# tmp_pcap_file = "_crafted.pcap"
# m.process(tmp_pcap_file, arg.sta_file, limit=20)

m.process(arg.sta_file, limit=np.Inf, heuristic=False)