remove: redundant code
This commit is contained in:
1 parent
e9a85f1ef8
commit
8987fd239f
2 files changed
+1
-33
No files matched your search
@@ -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):
|
||||
|
||||
@@ -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())
|
||||
|
||||
|
||||
Reference in new issue
Block a user