remove: redundant code

This commit is contained in:
Hieu Le committed 2021-03-31 23:47:39 +07:00
1 parent e9a85f1ef8
commit 8987fd239f
2 files changed
+1 -33

No files matched your search

+1 -1
View File
@@ -241,7 +241,7 @@ class Flow:
last_timestamp = (
self.latest_timestamp if self.latest_timestamp != 0 else packet.time
)
if (packet.time - (last_timestamp / 1e6)) > constants.CLUMP_TIMEOUT:
if (packet.time - last_timestamp) > constants.CLUMP_TIMEOUT:
self.update_active_idle(packet.time - last_timestamp)
def update_active_idle(self, current_time):
-32
View File
@@ -1,7 +1,6 @@
import csv
from collections import defaultdict
import requests
from scapy.sessions import DefaultSession
from .features.context.packet_direction import PacketDirection
@@ -116,37 +115,6 @@ class FlowSession(DefaultSession):
):
data = flow.get_data()
# POST Request to Model API
if self.url_model:
payload = {
"columns": list(data.keys()),
"data": [list(data.values())],
}
post = requests.post(
self.url_model,
json=payload,
headers={
"Content-Type": "application/json; format=pandas-split"
},
)
resp = post.json()
result = resp["result"].pop()
if result == 0:
result_print = "Benign"
else:
result_print = "Malicious"
print(
"{: <15}:{: <6} -> {: <15}:{: <6} \t {} (~{:.2f}%)".format(
resp["src_ip"],
resp["src_port"],
resp["dst_ip"],
resp["dst_port"],
result_print,
resp["probability"].pop()[result] * 100,
)
)
if self.csv_line == 0:
self.csv_writer.writerow(data.keys())