-
Notifications
You must be signed in to change notification settings - Fork 3
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
4b0860f
commit 9775af6
Showing
11 changed files
with
26 additions
and
71 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,52 +1,20 @@ | ||
import matplotlib.pyplot as plt | ||
import numpy as np | ||
from scapy.all import (TCP, Packet, RandIP, RandIP6, RandMAC, Raw, rdpcap, | ||
sniff, wrpcap) | ||
|
||
from heifip.extractor import FIPExtractor | ||
from heifip.images.flow import FlowImage | ||
from heifip.images.flow_tiled_auto import FlowImageTiledAuto | ||
from heifip.images.flow_tiled_fixed import FlowImageTiledFixed | ||
from heifip.images.markovchain import (MarkovTransitionMatrixFlow, | ||
MarkovTransitionMatrixPacket) | ||
from heifip.images.packet import PacketImage | ||
from heifip.layers import PacketProcessorType | ||
from heifip.images.flow import FlowImage | ||
|
||
# extractor = FIPExtractor() | ||
# imgs = extractor.create_image_from_file( | ||
# "/home/smachmeier/data/better-split-binary/benign/BitTorrent-0710.pcap", | ||
# PacketProcessorType.NONE, | ||
# MarkovTransitionMatrixFlow, | ||
# 0, # min_image_dim | ||
# 0, # max_image_dim | ||
# 3, # min_packets | ||
# 0, # max_packets | ||
# True, # remove_duplicates, | ||
# 8 | ||
# # 30, # dim | ||
# # 0, # fill | ||
# # True # auto_dim | ||
# ) | ||
# i = 0 | ||
# for img in imgs: | ||
# extractor.save_image(img, f"/home/smachmeier/Documents/projects/heiFIP/data/benign/{i}.png") | ||
# i += 1 | ||
pcap = sniff(offline="/home/smachmeier/data/test-data/") | ||
for pkt in pcap: | ||
# if Raw in pkt: | ||
# pkt[TCP].remove_payload() | ||
pkt.show() | ||
# wrpcap("/home/smachmeier/test.pcap", pkt, append=True) | ||
|
||
# fig = plt.figure(figsize=(16, 16)) | ||
# columns = 4 | ||
# rows = 4 | ||
# for i in range(1, columns*rows +1): | ||
# fig.add_subplot(rows, columns, i) | ||
# plt.ylabel("Y") | ||
# plt.xlabel("X") | ||
# plt.imshow(imgs[i]) | ||
# plt.savefig('test.pdf', dpi=fig.dpi) | ||
|
||
# plt.imshow(imgs[0]) | ||
# plt.savefig('test.pdf') | ||
extractor = FIPExtractor() | ||
imgs = extractor.create_image_from_file( | ||
"./tests/pcaps/dns-caa.pcap", | ||
PacketProcessorType.NONE, | ||
FlowImage, | ||
0, # min_image_dim | ||
0, # max_image_dim | ||
1, # min_packets | ||
0, # max_packets | ||
True, # remove_duplicates, | ||
8 | ||
# 30, # dim | ||
# 0, # fill | ||
# True # auto_dim | ||
) | ||
print(imgs) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,5 +1,4 @@ | ||
import binascii | ||
import logging | ||
|
||
import numpy as np | ||
from scapy.all import Packet, raw | ||
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,5 +1,4 @@ | ||
import binascii | ||
import logging | ||
|
||
import numpy as np | ||
from scapy.all import Packet, raw | ||
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,5 +1,4 @@ | ||
import binascii | ||
import logging | ||
|
||
import numpy as np | ||
from scapy.all import Packet, raw | ||
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,5 +1,4 @@ | ||
import binascii | ||
import logging | ||
|
||
import numpy as np | ||
from scapy.all import Packet, raw | ||
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,6 +1,3 @@ | ||
import hashlib | ||
from typing import Type | ||
|
||
from scapy.all import Packet | ||
from scapy.layers.dns import DNS | ||
|
||
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters